FilePool.cc

Go to the documentation of this file.

#include "FilePool.hh"
#include "File.hh"
#include "FileContext.hh"
#include "FileOperations.hh"
#include "TclObject.hh"
#include "CommandException.hh"
#include "Display.hh"
#include "EventDistributor.hh"
#include "CliComm.hh"
#include "Reactor.hh"
#include <memory>
 
using std::string;
using std::vector;
 
namespace openmsx {
 
class Sha1SumCommand final : public Command
{
public:
        Sha1SumCommand(CommandController& commandController, FilePool& filePool);
        void execute(span<const TclObject> tokens, TclObject& result) override;
        string help(const vector<string>& tokens) const override;
        void tabCompletion(vector<string>& tokens) const override;
private:
        FilePool& filePool;
};
 
 
static string initialFilePoolSettingValue()
{
        TclObject result;
 
        for (auto& p : systemFileContext().getPaths()) {
                result.addListElement(
                        makeTclDict("-path", FileOperations::join(p, "systemroms"),
                                    "-types", "system_rom"),
                        makeTclDict("-path", FileOperations::join(p, "software"),
                                    "-types", "rom disk tape"));
        }
        return string(result.getString());
}
 
FilePool::FilePool(CommandController& controller, Reactor& reactor_)
        : core(FileOperations::getUserDataDir() + "/.filecache",
               [&] { return getDirectories(); },
               [&](const std::string& message) { reportProgress(message); })
        , filePoolSetting(
                controller, "__filepool",
                "This is an internal setting. Don't change this directly, "
                "instead use the 'filepool' command.",
                initialFilePoolSettingValue())
        , reactor(reactor_)
{
        filePoolSetting.attach(*this);
        reactor.getEventDistributor().registerEventListener(OPENMSX_QUIT_EVENT, *this);
 
        sha1SumCommand = std::make_unique<Sha1SumCommand>(controller, *this);
}
 
FilePool::~FilePool()
{
        reactor.getEventDistributor().unregisterEventListener(OPENMSX_QUIT_EVENT, *this);
        filePoolSetting.detach(*this);
}
 
File FilePool::getFile(FileType fileType, const Sha1Sum& sha1sum)
{
        return core.getFile(fileType, sha1sum);
}
 
Sha1Sum FilePool::getSha1Sum(File& file)
{
        return core.getSha1Sum(file);
}
 
static FileType parseTypes(Interpreter& interp, const TclObject& list)
{
        auto result = FileType::NONE;
        unsigned num = list.getListLength(interp);
        for (unsigned i = 0; i < num; ++i) {
                std::string_view elem = list.getListIndex(interp, i).getString();
                if (elem == "system_rom") {
                        result |= FileType::SYSTEM_ROM;
                } else if (elem == "rom") {
                        result |= FileType::ROM;
                } else if (elem == "disk") {
                        result |= FileType::DISK;
                } else if (elem == "tape") {
                        result |= FileType::TAPE;
                } else {
                        throw CommandException("Unknown type: ", elem);
                }
        }
        return result;
}
 
FilePoolCore::Directories FilePool::getDirectories() const
{
        FilePoolCore::Directories result;
        try {
                auto& interp = filePoolSetting.getInterpreter();
                const TclObject& all = filePoolSetting.getValue();
                unsigned numLines = all.getListLength(interp);
                for (unsigned i = 0; i < numLines; ++i) {
                        FilePoolCore::Dir dir;
                        bool hasPath = false;
                        dir.types = FileType::NONE;
                        TclObject line = all.getListIndex(interp, i);
                        unsigned numItems = line.getListLength(interp);
                        if (numItems & 1) {
                                throw CommandException(
                                        "Expected a list with an even number "
                                        "of elements, but got ", line.getString());
                        }
                        for (unsigned j = 0; j < numItems; j += 2) {
                                std::string_view name  = line.getListIndex(interp, j + 0).getString();
                                TclObject value = line.getListIndex(interp, j + 1);
                                if (name == "-path") {
                                        dir.path = value.getString();
                                        hasPath = true;
                                } else if (name == "-types") {
                                        dir.types = parseTypes(interp, value);
                                } else {
                                        throw CommandException("Unknown item: ", name);
                                }
                        }
                        if (!hasPath) {
                                throw CommandException(
                                        "Missing -path item: ", line.getString());
                        }
                        if (dir.types == FileType::NONE) {
                                throw CommandException(
                                        "Missing -types item: ", line.getString());
                        }
                        result.push_back(dir);
                }
        } catch (CommandException& e) {
                reactor.getCliComm().printWarning(
                        "Error while parsing '__filepool' setting", e.getMessage());
        }
        return result;
}
 
void FilePool::update(const Setting& setting)
{
        assert(&setting == &filePoolSetting); (void)setting;
        getDirectories(); // check for syntax errors
}
 
void FilePool::reportProgress(const std::string& message)
{
        if (quit) core.abort();
        reactor.getCliComm().printProgress(message);
        reactor.getDisplay().repaintDelayed(0);
}
 
int FilePool::signalEvent(const std::shared_ptr<const Event>& event)
{
        (void)event; // avoid warning for non-assert compiles
        assert(event->getType() == OPENMSX_QUIT_EVENT);
        quit = true;
        return 0;
}
 
 
// class Sha1SumCommand
 
Sha1SumCommand::Sha1SumCommand(
                CommandController& commandController_, FilePool& filePool_)
        : Command(commandController_, "sha1sum")
        , filePool(filePool_)
{
}
 
void Sha1SumCommand::execute(span<const TclObject> tokens, TclObject& result)
{
        checkNumArgs(tokens, 2, "filename");
        File file(tokens[1].getString());
        result = filePool.getSha1Sum(file).toString();
}
 
string Sha1SumCommand::help(const vector<string>& /*tokens*/) const
{
        return "Calculate sha1 value for the given file. If the file is "
               "(g)zipped the sha1 is calculated on the unzipped version.";
}
 
void Sha1SumCommand::tabCompletion(vector<string>& tokens) const
{
        completeFileName(tokens, userFileContext());
}
 
} // namespace openmsx