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

#pragma once
#include <switch.h>

#include <stratosphere.hpp>
#include <stratosphere/ro.hpp>

namespace sts::ro {

    /* Access utilities. */
    void SetDevelopmentHardware(bool is_development_hardware);
    void SetDevelopmentFunctionEnabled(bool is_development_function_enabled);

    class Service final : public IServiceObject {
        protected:
            enum class CommandId {
                LoadNro     = 0,
                UnloadNro   = 1,
                LoadNrr     = 2,
                UnloadNrr   = 3,
                Initialize  = 4,
                LoadNrrEx   = 10,
            };
        private:
            size_t context_id;
            ModuleType type;
        public:
            explicit Service(ModuleType t);
            virtual ~Service() override;
        private:
            /* Actual commands. */
            Result LoadNro(Out<u64> load_address, PidDescriptor pid_desc, u64 nro_address, u64 nro_size, u64 bss_address, u64 bss_size);
            Result UnloadNro(PidDescriptor pid_desc, u64 nro_address);
            Result LoadNrr(PidDescriptor pid_desc, u64 nrr_address, u64 nrr_size);
            Result UnloadNrr(PidDescriptor pid_desc, u64 nrr_address);
            Result Initialize(PidDescriptor pid_desc, CopiedHandle process_h);
            Result LoadNrrEx(PidDescriptor pid_desc, u64 nrr_address, u64 nrr_size, CopiedHandle process_h);
        public:
            DEFINE_SERVICE_DISPATCH_TABLE {
                MAKE_SERVICE_COMMAND_META(Service, LoadNro),
                MAKE_SERVICE_COMMAND_META(Service, UnloadNro),
                MAKE_SERVICE_COMMAND_META(Service, LoadNrr),
                MAKE_SERVICE_COMMAND_META(Service, UnloadNrr),
                MAKE_SERVICE_COMMAND_META(Service, Initialize),
                MAKE_SERVICE_COMMAND_META(Service, LoadNrrEx,   FirmwareVersion_700),
            };

    };

}