/*
 * Copyright (c) 2018-2019 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <unordered_map>
#include "ldr_ecs.hpp"

namespace sts::ldr::ecs {

    namespace {

        /* Convenience definition. */
        constexpr size_t DeviceNameSizeMax = 0x20;
        constexpr size_t MaxExternalContentSourceCount = 0x10;

        /* Types. */
        class ExternalContentSource {
            NON_COPYABLE(ExternalContentSource);
            NON_MOVEABLE(ExternalContentSource);
            private:
                char device_name[DeviceNameSizeMax];
            public:
                ExternalContentSource(const char *dn) {
                    std::strncpy(this->device_name, dn, sizeof(this->device_name));
                    this->device_name[sizeof(this->device_name) - 1] = '\0';
                }

                ~ExternalContentSource() {
                    fsdevUnmountDevice(this->device_name);
                }

                const char *GetDeviceName() const {
                    return this->device_name;
                }
        };

        /* Global storage. */
        std::unordered_map<u64, ExternalContentSource> g_map;
    }

    /* API. */
    const char *Get(ncm::TitleId title_id) {
        auto it = g_map.find(static_cast<u64>(title_id));
        if (it == g_map.end()) {
            return nullptr;
        }
        return it->second.GetDeviceName();
    }

    Result Set(Handle *out, ncm::TitleId title_id) {
        /* TODO: Is this an appropriate error? */
        R_UNLESS(g_map.size() < MaxExternalContentSourceCount, ldr::ResultTooManyArguments());

        /* Clear any sources. */
        R_ASSERT(Clear(title_id));

        /* Generate mountpoint. */
        char device_name[DeviceNameSizeMax];
        std::snprintf(device_name, DeviceNameSizeMax, "ecs-%016lx", static_cast<u64>(title_id));

        /* Create session. */
        os::ManagedHandle server, client;
        R_TRY(svcCreateSession(server.GetPointer(), client.GetPointer(), 0, 0));

        /* Create service. */
        Service srv;
        serviceCreate(&srv, client.Move());
        FsFileSystem fs = { srv };
        auto fs_guard = SCOPE_GUARD { fsFsClose(&fs); };

        /* Try to mount. */
        R_UNLESS(fsdevMountDevice(device_name, fs) >= 0, fs::ResultMountNameAlreadyExists());
        fs_guard.Cancel();

        /* Add to map. */
        g_map.emplace(static_cast<u64>(title_id), device_name);
        *out = server.Move();
        return ResultSuccess();
    }

    Result Clear(ncm::TitleId title_id) {
        /* Delete if present. */
        g_map.erase(static_cast<u64>(title_id));
        return ResultSuccess();
    }

}