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