TempFile.hxx

#pragma once
 
// cosmos
#include <cosmos/fs/filesystem.hxx>
#include <cosmos/fs/FileBase.hxx>
 
namespace cosmos {
 
 
class COSMOS_API TempFile :
                public FileBase {
public: // functions
 
        TempFile() = default;
 
        explicit TempFile(const SysString _template, const OpenFlags flags = OpenFlags{OpenFlag::CLOEXEC}) {
                open(_template, flags);
        }
 
        ~TempFile();
 
        TempFile& operator=(TempFile &&other) noexcept {
                m_tmp_path = other.m_tmp_path;
                other.m_tmp_path.clear();
 
                FileBase::operator=(std::move(other));
                return *this;
        }
 
        // Prevent copying due to the path deletion responsibility.
 
        void close() override {
                try {
                        FileBase::close();
                } catch(...) {
                        unlinkPath();
                        throw;
                }
 
                unlinkPath();
        }
 
        void open(const SysString _template, const OpenFlags flags = OpenFlags{OpenFlag::CLOEXEC}) {
                close();
 
                auto [fd, path] = fs::make_tempfile(_template, flags);
                m_fd = fd;
                m_tmp_path = path;
        }
 
 
        const std::string& path() const;
 
protected: // functions
 
        void unlinkPath() {
                if (!m_tmp_path.empty()) {
                        try {
                                fs::unlink_file(m_tmp_path);
                        } catch(...) {
                                m_tmp_path.clear();
                                throw;
                        }
                        m_tmp_path.clear();
                }
        }
 
protected: // data
 
        std::string m_tmp_path;
};
 
} // end ns