util/Sawyer/FileSystem.h

// WARNING: Changes to this file must be contributed back to Sawyer or else they will
//          be clobbered by the next update from Sawyer.  The Sawyer repository is at
//          https://github.com/matzke1/sawyer.
 
 
 
 
#ifndef Sawyer_FileSystem_H
#define Sawyer_FileSystem_H
 
#include <Sawyer/Sawyer.h>
#include <boost/filesystem.hpp>
#include <boost/noncopyable.hpp>
#include <fstream>
 
namespace Sawyer {
 
namespace FileSystem {
 
class SAWYER_EXPORT TemporaryFile: private boost::noncopyable {
    boost::filesystem::path name_;
    std::ofstream stream_;
    bool keep_;
 
public:
    TemporaryFile(): keep_(false) {
        name_ = boost::filesystem::temp_directory_path() / boost::filesystem::unique_path();
        stream_.open(name_.string().c_str());
    }
 
    explicit TemporaryFile(const boost::filesystem::path &name): keep_(false) {
        name_ = name;
        stream_.open(name.string().c_str());
    }
 
    ~TemporaryFile() {
        stream_.close();
        if (!keep_)
            boost::filesystem::remove(name_);
    }
 
    const boost::filesystem::path& name() const { return name_; }
 
    std::ofstream& stream() { return stream_; }
 
    bool keep() const { return keep_; }
    void keep(bool b) { keep_ = b; }
};
 
class SAWYER_EXPORT TemporaryDirectory: private boost::noncopyable {
    boost::filesystem::path name_;
    bool keep_;
 
public:
    TemporaryDirectory()
        : name_(boost::filesystem::temp_directory_path() / boost::filesystem::unique_path()), keep_(false) {
        createOrThrow();
    }
 
    explicit TemporaryDirectory(const boost::filesystem::path &name)
        : name_(name), keep_(false) {
        createOrThrow();
    }
 
    ~TemporaryDirectory() {
        if (!keep_)
            boost::filesystem::remove_all(name_);
    }
 
    const boost::filesystem::path& name() const { return name_; }
 
    bool keep() const { return keep_; }
    void keep(bool b) { keep_ = b; }
private:
    // Create directory or throw exception
    void createOrThrow() {
        boost::system::error_code ec;
        if (!boost::filesystem::create_directory(name_, ec))
            throw boost::filesystem::filesystem_error("cannot create directory", name_, ec);
    }
};
 
} // namespace
} // namespace
 
#endif