openmw/components/files/collections.cpp

#include "collections.hpp"

#include <components/misc/strings/algorithm.hpp>
#include <components/misc/strings/lower.hpp>

namespace Files
{
    Collections::Collections()
        : mDirectories()
        , mFoldCase(false)
        , mCollections()
    {
    }

    Collections::Collections(const Files::PathContainer& directories, bool foldCase)
        : mDirectories(directories)
        , mFoldCase(foldCase)
        , mCollections()
    {
    }

    const MultiDirCollection& Collections::getCollection(const std::string& extension) const
    {
        std::string ext = Misc::StringUtils::lowerCase(extension);
        auto iter = mCollections.find(ext);
        if (iter==mCollections.end())
        {
            std::pair<MultiDirCollectionContainer::iterator, bool> result =
                mCollections.emplace(ext, MultiDirCollection(mDirectories, ext, mFoldCase));

            iter = result.first;
        }

        return iter->second;
    }

    std::filesystem::path Collections::getPath(const std::string& file) const
    {
        for (const auto & mDirectorie : mDirectories)
        {
            for (const auto& iter2 :
                std::filesystem::directory_iterator (mDirectorie))
            {
                const auto& path = iter2.path();
                if (mFoldCase)
                {
                    if (Misc::StringUtils::ciEqual(file, path.filename().string())) //TODO(Project579): This will probably break in windows with unicode paths
                        return path;
                }
                else if (path.filename().string() == file) //TODO(Project579): This will probably break in windows with unicode paths
                    return path;
            }
        }

        throw std::runtime_error ("file " + file + " not found");
    }

    bool Collections::doesExist(const std::string& file) const
    {
        for (const auto & mDirectorie : mDirectories)
        {
            for (const auto& iter2 :
                std::filesystem::directory_iterator (mDirectorie))
            {
                const auto& path = iter2.path();

                if (mFoldCase)
                {
                    if (Misc::StringUtils::ciEqual(file, path.filename().string())) //TODO(Project579): This will probably break in windows with unicode paths
                        return true;
                }
                else if (path.filename().string() == file) //TODO(Project579): This will probably break in windows with unicode paths
                    return true;
            }
        }

        return false;
    }

    const Files::PathContainer& Collections::getPaths() const
    {
        return mDirectories;
    }
}