DesignCache.cpp

#include "ArchiveExtractor.h"
#include "DesignCache.h"
#include "Logger.h"
#include <exception>
#include <filesystem>
#include <utility>
 
using namespace Utils;
using namespace std::filesystem;
 
namespace Odb::Lib::App
{    
    DesignCache::DesignCache(std::string directory) :
        m_directory(std::move(directory))
    {
    }
 
    DesignCache::~DesignCache()
    {
        Clear();
    }
 
    std::shared_ptr<ProductModel::Design> DesignCache::GetDesign(const std::string& designName)
    {
        auto findIt = m_designsByName.find(designName);
        if (findIt == m_designsByName.end())
        {
            auto pDesign = LoadDesign(designName);             
            return pDesign;
        }
 
        return m_designsByName[designName];
    }
 
    std::shared_ptr<FileModel::Design::FileArchive> DesignCache::GetFileArchive(const std::string& designName)
    {
        std::stringstream ss;
        ss << "Retrieving design \"" << designName << "\" from cache... ";
        loginfo(ss.str());
 
        auto findIt = m_fileArchivesByName.find(designName);
        if (findIt == m_fileArchivesByName.end())
        {
            loginfo("Not found in cache, attempting to load from file...");
 
            auto pFileArchive = LoadFileArchive(designName);
            if (pFileArchive == nullptr)
            {
                logwarn("Failed loading from file");
            }
            else
            {
                loginfo("Loaded from file");
            }
            return pFileArchive;
        }
        else
        {
            loginfo("Found. Returning from cache.");
        }
 
        return m_fileArchivesByName[designName];        
    }
 
    std::vector<std::string> DesignCache::getLoadedDesignNames(const std::string& filter) const
    {
        std::vector<std::string> loadedDesigns;
        for (const auto& kv : m_designsByName)
        {
            loadedDesigns.push_back(kv.second->GetFileModel()->GetProductName());
        }
        return loadedDesigns;
    }
 
    std::vector<std::string> DesignCache::getLoadedFileArchiveNames(const std::string& filter) const
    {
        std::vector<std::string> loadedFileArchives;
        for (const auto& kv : m_fileArchivesByName)
        {
            loadedFileArchives.push_back(kv.second->GetProductName());
        }
        return loadedFileArchives;
    }
 
    std::vector<std::string> DesignCache::getUnloadedDesignNames(const std::string& filter) const
    {        
        std::vector<std::string> unloadedNames;
 
        //try
        {
            path dir(m_directory);
            for (const auto& entry : directory_iterator(dir))
            {
                if (entry.is_regular_file())
                {
                    unloadedNames.push_back(entry.path().stem().string());
                }
            }
        }
        //catch (std::filesystem::filesystem_error& fe)
        //{
        //    logexception(fe);
        //    // re-throw it so we get a HTTP 500 response to the client
        //    throw fe;
        //}
 
        return unloadedNames;
    }
 
    int DesignCache::loadAllFileArchives(bool stopOnError)
    {
        int loaded = 0;
 
        for (const auto& entry : directory_iterator(m_directory))
        {
            if (entry.is_regular_file())
            {
                if (ArchiveExtractor::IsArchiveTypeSupported(entry.path().filename()))
                {
                    try
                    {
                        auto pFileArchive = LoadFileArchive(entry.path().stem().string());
                        if (pFileArchive != nullptr)
                        {
                            loaded++;
                        }
                    }
                    catch (std::exception& e)
                    {
                        // continue if we encounter an error loading one
                        logexception(e);
                        if (stopOnError) throw e;
                    }
                }
            }
       }
 
        return loaded;
    }
 
    int DesignCache::loadAllDesigns(bool stopOnError)
    {
        int loaded = 0;
 
        for (const auto& entry : directory_iterator(m_directory))
        {
            if (entry.is_regular_file())
            {
                if (ArchiveExtractor::IsArchiveTypeSupported(entry.path().filename()))
                {
                    try
                    {
                        auto pDesign = LoadDesign(entry.path().stem().string());
                        if (pDesign != nullptr)
                        {
                            loaded++;
                        }
                    }
                    catch (std::exception& e)
                    {
                        logexception(e);
                        if (stopOnError)
                        {
                            throw e;
                        }
                    }
                }
            }
        }
 
        return loaded;
    }
 
    int DesignCache::loadFileArchives(const StringVector& names)
    {
        int loaded = 0;
 
        for (const auto& name : names)
        {
            try
            {
                auto pFileArchive = LoadFileArchive(name);
                if (pFileArchive != nullptr)
                {
                    loaded++;
                }
            }
            catch (std::exception& e)
            {
                // continue on error
                logexception(e);
            }
        }
 
        return loaded;
    }
 
    int DesignCache::loadDesigns(const StringVector& names)
    {
        int loaded = 0;
 
        for (const auto& name : names)
        {
            try
            {
                auto pDesign = LoadDesign(name);
                if (pDesign != nullptr)
                {
                    loaded++;
                }
            }
            catch (std::exception& e)
            {
                // continue on error
                logexception(e);
            }
        }
 
        return loaded;
    }
 
    void DesignCache::setDirectory(const std::string& directory)
    {
        m_directory = directory;
    }
 
    const std::string& DesignCache::getDirectory() const
    {
        return m_directory;
    }
 
    //bool DesignCache::isQueryValid(const std::string& query) const
    //{
    //    return false;
    //}
 
    void DesignCache::Clear()
    {
        m_fileArchivesByName.clear();
        m_designsByName.clear();
    }
 
    std::shared_ptr<ProductModel::Design> DesignCache::LoadDesign(const std::string& designName)
    { 
        for (const auto& entry : directory_iterator(m_directory))
        {
            if (entry.is_regular_file())
            {
                if (entry.path().stem() == designName)
                {
                    auto pFileModel = GetFileArchive(designName);
                    if (pFileModel != nullptr)
                    {
                        auto pDesign = std::make_shared<ProductModel::Design>();
                        if (pDesign->Build(pFileModel))
                        {
                            // overwrite any existing design with the same name
                            m_designsByName[pFileModel->GetProductName()] =  pDesign;
                            return pDesign;
                        }
                        else
                        {
                            break;
                        }
                    }
                    else
                    {
                        break;
                    }
                }
            }
        }    
 
        return nullptr;
    }
 
    std::shared_ptr<FileModel::Design::FileArchive> DesignCache::LoadFileArchive(const std::string& designName)
    {
        auto fileFound = false;
 
        // skip inaccessible files and do not follow symlinks
        const auto options = directory_options::skip_permission_denied;
        for (const auto& entry : directory_iterator(m_directory, options))
        {
            if (entry.is_regular_file())
            {
                if (entry.path().stem() == designName)
                {                    
                    fileFound = true;
 
                    loginfo("file found: [" + entry.path().string() + "], attempting to parse...");
 
                    auto pFileArchive = std::make_shared<FileModel::Design::FileArchive>(entry.path().string());
                    if (pFileArchive->ParseFileModel())
                    {
                        // overwrite any existing file archive with the same name
                        m_fileArchivesByName[pFileArchive->GetProductName()] = pFileArchive;
                        return pFileArchive;
                    }
                    else
                    {
                        break;
                    }
                }
            }
        }       
 
        if (!fileFound)
        {
            logwarn("Failed to find file for design \"" + designName + "\"");
        }
 
        return nullptr;
    }
}