/*
 * 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 <stratosphere.hpp>

namespace sts::ams {

    namespace {

        FirmwareVersion g_firmware_version;
        bool g_has_cached;
        os::Mutex g_mutex;

        void CacheValues() {
            if (__atomic_load_n(&g_has_cached, __ATOMIC_SEQ_CST)) {
                return;
            }

            std::scoped_lock lk(g_mutex);

            if (g_has_cached) {
                return;
            }

            /* TODO: spl::smc:: */
            u32 target_fw = 0;
            {
                SecmonArgs args = {0};
                args.X[0] = 0xC3000002; /* smcGetConfig */
                args.X[1] = 65000; /* ConfigItem_ExosphereVersion */
                R_ASSERT(svcCallSecureMonitor(&args));
                STS_ASSERT(args.X[0] == 0);

                target_fw = (args.X[1] >> 0x08) & 0xFF;
            }

            switch (static_cast<ams::TargetFirmware>(target_fw)) {
                case ams::TargetFirmware_100:
                    g_firmware_version = FirmwareVersion_100;
                    break;
                case ams::TargetFirmware_200:
                    g_firmware_version = FirmwareVersion_200;
                    break;
                case ams::TargetFirmware_300:
                    g_firmware_version = FirmwareVersion_300;
                    break;
                case ams::TargetFirmware_400:
                    g_firmware_version = FirmwareVersion_400;
                    break;
                case ams::TargetFirmware_500:
                    g_firmware_version = FirmwareVersion_500;
                    break;
                case ams::TargetFirmware_600:
                case ams::TargetFirmware_620:
                    g_firmware_version = FirmwareVersion_600;
                    break;
                case ams::TargetFirmware_700:
                    g_firmware_version = FirmwareVersion_700;
                    break;
                case ams::TargetFirmware_800:
                    g_firmware_version = FirmwareVersion_800;
                    break;
                case ams::TargetFirmware_810:
                    g_firmware_version = FirmwareVersion_810;
                    break;
                case ams::TargetFirmware_900:
                    g_firmware_version = FirmwareVersion_900;
                    break;
                STS_UNREACHABLE_DEFAULT_CASE();
            }

            __atomic_store_n(&g_has_cached, true, __ATOMIC_SEQ_CST);
        }

    }

    FirmwareVersion GetRuntimeFirmwareVersion() {
        CacheValues();
        return g_firmware_version;
    }

    void SetFirmwareVersionForLibnx() {
        u32 major = 0, minor = 0, micro = 0;
        switch (GetRuntimeFirmwareVersion()) {
            case FirmwareVersion_100:
                major = 1;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_200:
                major = 2;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_300:
                major = 3;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_400:
                major = 4;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_500:
                major = 5;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_600:
                major = 6;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_700:
                major = 7;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_800:
                major = 8;
                minor = 0;
                micro = 0;
                break;
            case FirmwareVersion_810:
                major = 8;
                minor = 1;
                micro = 0;
                break;
            case FirmwareVersion_900:
                major = 9;
                minor = 0;
                micro = 0;
                break;
            STS_UNREACHABLE_DEFAULT_CASE();
        }
        hosversionSet(MAKEHOSVERSION(major, minor, micro));
    }

}