envionment.cpp

#include <cstring>
#include <iostream>
#include "config.h"
#include "environment.h"
#include "interop/interop.hpp"
#include "ui.h"
#include "utility/collection.hpp"
#include "utility/string.hpp"
#include "platform/platform.h"

using namespace openloco::interop;

namespace openloco::environment
{
    loco_global_array<char, 260, 0x009D118E> _path_buffer;
    loco_global_array<char, 257, 0x0050B0CE> _path_install;
    loco_global_array<char, 257, 0x0050B1CF> _path_saves_single_player;
    loco_global_array<char, 257, 0x0050B2EC> _path_saves_two_player;
    loco_global_array<char, 257, 0x0050B406> _path_scenarios;
    loco_global_array<char, 257, 0x0050B518> _path_landscapes;
    loco_global_array<char, 257, 0x0050B635> _path_objects;

    static fs::path get_sub_path(path_id id);

    static bool validate_loco_install_path(const fs::path &path)
    {
        if (path.empty())
        {
            return false;
        }
        else
        {
            auto g1Path = path / get_sub_path(path_id::g1);
            return fs::exists(g1Path);
        }
    }

    static fs::path auto_detect_loco_install_path()
    {
        static constexpr const char * searchPaths[] =
        {
            "C:/Program Files (x86)/Atari/Locomotion",
            "C:/GOG Games/Chris Sawyer's Locomotion",
        };

        std::cout << "Searching for Locomotion install path..." << std::endl;
        for (auto path : searchPaths)
        {
            if (validate_loco_install_path(path))
            {
                std::cout << "  found: " << path << std::endl;
                return path;
            }
        }
        return fs::path();
    }

    static fs::path resolve_loco_install_path()
    {
        auto &cfg = config::read_new_config();
        auto path = fs::path(cfg.loco_install_path);
        if (!path.empty())
        {
            if (validate_loco_install_path(path))
            {
                config::write_new_config();
                return path;
            }
            std::cerr << "Configured install path for Locomotion is missing data/g1.dat." << std::endl;
        }

        path = auto_detect_loco_install_path();
        if (!path.empty())
        {
#ifdef _OPENLOCO_USE_BOOST_FS_
            cfg.loco_install_path = path.make_preferred().string();
#else
            cfg.loco_install_path = path.make_preferred().u8string();
#endif
            config::write_new_config();
            return path;
        }
        else
        {
            std::cerr << "Unable to find install path for Locomotion." << std::endl <<
                         "You will need to manually provide it." << std::endl;
            ui::show_message_box("OpenLoco", "Select your Locomotion install path.");
            path = platform::prompt_directory("Select your Locomotion install path.");
            if (validate_loco_install_path(path))
            {
#ifdef _OPENLOCO_USE_BOOST_FS_
                cfg.loco_install_path = path.make_preferred().string();
#else
                cfg.loco_install_path = path.make_preferred().u8string();
#endif
                config::write_new_config();
                return path;
            }
            
            std::cerr << "Path is missing g1.dat." << std::endl;
            ui::show_message_box("OpenLoco", "Path is missing data/g1.dat.");
            std::exit(-1);
        }
    }

    static fs::path get_loco_install_path()
    {
        return _path_install.get();
    }

    // 0x004416B5
    fs::path get_path(path_id id)
    {
        auto basePath = get_loco_install_path();
        auto subPath = get_sub_path(id);
        auto result = basePath / subPath;
        return result;
    }

    template<size_t TSize, uint32_t TAddress>
    static void set_directory(loco_global_array<char, TSize, TAddress> buffer, fs::path path)
    {
#ifdef _OPENLOCO_USE_BOOST_FS_
        utility::strcpy_safe(buffer, path.make_preferred().string().c_str());
#else
        utility::strcpy_safe(buffer, path.make_preferred().u8string().c_str());
#endif
    }

    // 0x004412CE
    void resolve_paths()
    {
        auto basePath = resolve_loco_install_path();
        set_directory(_path_install, basePath);
        set_directory(_path_saves_single_player, basePath / "Single Player Saved Games/");
        set_directory(_path_saves_two_player, basePath / "Two Player Saved Games/");
        set_directory(_path_scenarios, basePath / "Scenarios/*.SC5");
        set_directory(_path_landscapes, basePath / "Scenarios/Landscapes/*.SC5");
        set_directory(_path_objects, basePath / "ObjData/*.DAT");
    }

    static fs::path get_sub_path(path_id id)
    {
        static constexpr const char * paths[] =
        {
            "Data/g1.DAT",
            "Data/PLUGIN.DAT",
            "Data/PLUGIN2.DAT",
            "Data/CSS1.DAT",
            "Data/CSS2.DAT",
            "Data/CSS3.DAT",
            "Data/CSS4.DAT",
            "Data/CSS5.DAT",
            "Data/GAME.CFG",
            "Data/KANJI.DAT",
            "Data/20s1.DAT",
            "Data/20s2.DAT",
            "Data/20s4.DAT",
            "Data/50s1.DAT",
            "Data/50s2.DAT",
            "Data/70s1.DAT",
            "Data/70s2.DAT",
            "Data/70s3.DAT",
            "Data/80s1.DAT",
            "Data/90s1.DAT",
            "Data/90s2.DAT",
            "Data/rag3.DAT",
            "Data/Chrysanthemum.DAT",
            "Data/Eugenia.DAT",
            "Data/Rag2.DAT",
            "Data/Rag1.DAT",
            "Data/20s3.DAT",
            "Data/40s1.DAT",
            "Data/40s2.DAT",
            "Data/50s3.DAT",
            "Data/40s3.DAT",
            "Data/80s2.DAT",
            "Data/60s1.DAT",
            "Data/80s3.DAT",
            "Data/60s2.DAT",
            "Data/60s3.DAT",
            "Data/80s4.DAT",
            "Data/20s5.DAT",
            "Data/20s6.DAT",
            "Data/title.dat",
            "Data/Scores.DAT",
            "Scenarios/Boulder Breakers.SC5",
            "Data/TUT1024_1.DAT",
            "Data/TUT1024_2.DAT",
            "Data/TUT1024_3.DAT",
            "Data/TUT800_1.DAT",
            "Data/TUT800_2.DAT",
            "Data/TUT800_3.DAT"
        };

        size_t index = (size_t)id;
        if (index >= utility::length(paths))
        {
            throw std::runtime_error("Invalid path_id: " + std::to_string((int32_t)id));
        }
        return paths[(size_t)id];
    }
}