SNFUtility/Common/FileBackup.hpp

// FileBackup.hpp
//
// Copyright (C) 2011 ARM Research Labs, LLC.
// See www.armresearch.com for the copyright terms.
//
// This file defines the interface used by the FileBackup class.
//

#ifndef FileBackuphpp_included
#define FileBackuphpp_included

#include <string>
#include <vector>

/// FileBackup class.
//
// This class provides capability to manage the backup and restore of
// disk files.
//
//////////////////////////////////////////////////////////////////////////////////////////////////////////
class FileBackup {

public:

    /// Constructor.
    //
    // \param[in] Suffix is the suffix to append to create the name of
    // the backup file.
    //
    FileBackup(std::string Suffix);

    /// Create a backup of the specified file.
    //
    // This method creates a backup of the specified file The name of
    // the link is the name of the file appended by the BackupSuffix
    // data member.
    //
    // The file can be restored from the backup by RestoreFromBackup.
    //
    // \param[in] File is the file name to create a backup for.
    //
    // \throws std::runtime_error in case of error.
    //
    // \see RestoreAllFilesFromBackup().
    //
    void CreateBackupFile(std::string File);

    /// Remove the backups of all the specified files.
    //
    // This method removes the backup of the files specified by the
    // CreateBackup() method.
    //
    // If an error is encountered, a message is output to std::cerr.
    //
    // \returns true if all backups were removed with no error, false otherwise.
    //
    bool RemoveAllBackupFiles();

    /// Restore the all the specified files from the backup.
    //
    // This method restores the backup of the files specified by the
    // CreateBackupFile() method.
    //
    // If an error is encountered, a message is output to std::cerr.
    //
    // \returns true if all files were restored with no error, false otherwise.
    //
    // \see CreateBackupFile().
    //
    bool RestoreAllFilesFromBackup();

private:

    /// Get the name of the backup file.
    //
    // \param[in] File is the name of the file to back up.
    //
    // \returns the name of the backup file.
    //
    std::string GetBackupFileName(std::string File);

    /// Copy a file.
    //
    // \param[in] From is the name of the file to copy from.
    //
    // \param[in] To is the name of the file to copy to.
    //
    // \throws runtime_error in case of error.
    //
    void CopyFile(std::string From, std::string To);

    /// Suffix to append to the file name to obtain the backup file
    /// name.
    std::string BackupSuffix;

    /// Typedef for container of names of files.
    typedef std::vector<std::string> FilenameContainer;

    /// Container of files that have backups.
    FilenameContainer BackedUpFile;

};

#endif