/*
 * Copyright (c) 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>
#include "pgl_srv_shell.hpp"
#include "pgl_srv_shell_event_observer.hpp"
#include "pgl_srv_shell_host_utils.hpp"
#include "pgl_srv_tipc_utils.hpp"

namespace ams::pgl::srv {

    Result ShellInterfaceCommon::LaunchProgramImpl(os::ProcessId *out, const ncm::ProgramLocation &loc, u32 pm_flags, u8 pgl_flags) {
        R_RETURN(pgl::srv::LaunchProgram(out, loc, pm_flags, pgl_flags));
    }

    Result ShellInterfaceCommon::TerminateProcessImpl(os::ProcessId process_id) {
        R_RETURN(pgl::srv::TerminateProcess(process_id));
    }

    Result ShellInterfaceCommon::LaunchProgramFromHostImpl(os::ProcessId *out, const void *content_path, size_t content_path_size, u32 pm_flags) {
        AMS_UNUSED(content_path_size);
        R_RETURN(pgl::srv::LaunchProgramFromHost(out, static_cast<const char *>(content_path), pm_flags));
    }

    Result ShellInterfaceCommon::GetProgramLaunchPropertyFromHostImpl(pgl::ProgramLaunchProperty *out, const void *content_path, size_t content_path_size) {
        AMS_UNUSED(content_path_size);
        R_RETURN(pgl::srv::GetProgramLaunchPropertyFromHost(out, static_cast<const char *>(content_path)));
    }

    Result ShellInterfaceCommon::GetRunningApplicationProcessIdImpl(os::ProcessId *out) {
        R_RETURN(pgl::srv::GetRunningApplicationProcessId(out));
    }

    Result ShellInterfaceCommon::BoostSystemMemoryResourceLimitImpl(u64 size) {
        R_RETURN(pgl::srv::BoostSystemMemoryResourceLimit(size));
    }

    Result ShellInterfaceCommon::IsRunningProcessImpl(bool *out, os::ProcessId process_id) {
        *out = pgl::srv::IsRunningProcess(process_id);
        R_SUCCEED();
    }

    Result ShellInterfaceCommon::EnableApplicationCrashReportImpl(bool enabled) {
        pgl::srv::EnableApplicationCrashReport(enabled);
        R_SUCCEED();
    }

    Result ShellInterfaceCommon::IsApplicationCrashReportEnabledImpl(bool *out) {
        *out = pgl::srv::IsApplicationCrashReportEnabled();
        R_SUCCEED();
    }

    Result ShellInterfaceCommon::EnableApplicationAllThreadDumpOnCrashImpl(bool enabled) {
        pgl::srv::EnableApplicationAllThreadDumpOnCrash(enabled);
        R_SUCCEED();
    }

    Result ShellInterfaceCommon::GetProcessIdImpl(os::ProcessId *out, ncm::ProgramId program_id) {
        R_RETURN(pm::shell::GetProcessId(out, program_id));
    }

    Result ShellInterfaceCommon::TriggerSnapShotDumperImpl(SnapShotDumpType dump_type, const void *arg, size_t arg_size) {
        AMS_UNUSED(arg_size);
        R_RETURN(pgl::srv::TriggerSnapShotDumper(dump_type, static_cast<const char *>(arg)));
    }
    
    Result ShellInterfaceCommon::EnableApplicationCrashReport2Impl(os::ProcessId process_id, bool enabled) {
        pgl::srv::EnableApplicationCrashReport2(process_id, enabled);
        R_SUCCEED();
    }

    Result ShellInterfaceCmif::LaunchProgram(ams::sf::Out<os::ProcessId> out, const ncm::ProgramLocation &loc, u32 pm_flags, u8 pgl_flags) {
        R_RETURN(this->LaunchProgramImpl(out.GetPointer(), loc, pm_flags, pgl_flags));
    }

    Result ShellInterfaceCmif::TerminateProcess(os::ProcessId process_id) {
        R_RETURN(this->TerminateProcessImpl(process_id));
    }

    Result ShellInterfaceCmif::LaunchProgramFromHost(ams::sf::Out<os::ProcessId> out, const ams::sf::InBuffer &content_path, u32 pm_flags) {
        R_RETURN(this->LaunchProgramFromHostImpl(out.GetPointer(), content_path.GetPointer(), content_path.GetSize(), pm_flags));
    }

    Result ShellInterfaceCmif::GetProgramLaunchPropertyFromHost(ams::sf::Out<pgl::ProgramLaunchProperty> out, const ams::sf::InBuffer &content_path) {
        R_RETURN(this->GetProgramLaunchPropertyFromHostImpl(out.GetPointer(), content_path.GetPointer(), content_path.GetSize()));
    }

    Result ShellInterfaceCmif::GetRunningApplicationProcessId(ams::sf::Out<os::ProcessId> out) {
        R_RETURN(this->GetRunningApplicationProcessIdImpl(out.GetPointer()));
    }

    Result ShellInterfaceCmif::BoostSystemMemoryResourceLimit(u64 size) {
        R_RETURN(this->BoostSystemMemoryResourceLimitImpl(size));
    }

    Result ShellInterfaceCmif::IsRunningProcess(ams::sf::Out<bool> out, os::ProcessId process_id) {
        R_RETURN(this->IsRunningProcessImpl(out.GetPointer(), process_id));
    }

    Result ShellInterfaceCmif::EnableApplicationCrashReport(bool enabled) {
        R_RETURN(this->EnableApplicationCrashReportImpl(enabled));
    }

    Result ShellInterfaceCmif::IsApplicationCrashReportEnabled(ams::sf::Out<bool> out) {
        R_RETURN(this->IsApplicationCrashReportEnabledImpl(out.GetPointer()));
    }

    Result ShellInterfaceCmif::EnableApplicationAllThreadDumpOnCrash(bool enabled) {
        R_RETURN(this->EnableApplicationAllThreadDumpOnCrashImpl(enabled));
    }

    Result ShellInterfaceCmif::TriggerSnapShotDumper(SnapShotDumpType dump_type, const ams::sf::InBuffer &arg) {
        R_RETURN(this->TriggerSnapShotDumperImpl(dump_type, arg.GetPointer(), arg.GetSize()));
    }

    Result ShellInterfaceCmif::CreateShellEvent(ams::sf::Out<ams::sf::SharedPointer<pgl::sf::IEventObserver>> out) {
        /* Allocate a new interface. */
        auto session = ObjectFactory::CreateSharedEmplaced<pgl::sf::IEventObserver, ShellEventObserverCmif>(m_allocator);
        R_UNLESS(session != nullptr, pgl::ResultOutOfMemory());

        *out = std::move(session);
        R_SUCCEED();
    }

    Result ShellInterfaceCmif::Command21NotImplemented(ams::sf::Out<u64> out, u32 in, const ams::sf::InBuffer &buf1, const ams::sf::InBuffer &buf2) {
        AMS_UNUSED(out, in, buf1, buf2);
        R_THROW(pgl::ResultNotImplemented());
    }

    Result ShellInterfaceTipc::LaunchProgram(ams::tipc::Out<os::ProcessId> out, const ncm::ProgramLocation loc, u32 pm_flags, u8 pgl_flags) {
        R_RETURN(this->LaunchProgramImpl(out.GetPointer(), loc, pm_flags, pgl_flags));
    }

    Result ShellInterfaceTipc::TerminateProcess(os::ProcessId process_id) {
        R_RETURN(this->TerminateProcessImpl(process_id));
    }

    Result ShellInterfaceTipc::LaunchProgramFromHost(ams::tipc::Out<os::ProcessId> out, const ams::tipc::InBuffer content_path, u32 pm_flags) {
        R_RETURN(this->LaunchProgramFromHostImpl(out.GetPointer(), content_path.GetPointer(), content_path.GetSize(), pm_flags));
    }

    Result ShellInterfaceTipc::GetProgramLaunchPropertyFromHost(ams::tipc::Out<pgl::ProgramLaunchProperty> out, const ams::tipc::InBuffer content_path) {
        R_RETURN(this->GetProgramLaunchPropertyFromHostImpl(out.GetPointer(), content_path.GetPointer(), content_path.GetSize()));
    }

    Result ShellInterfaceTipc::GetRunningApplicationProcessId(ams::tipc::Out<os::ProcessId> out) {
        R_RETURN(this->GetRunningApplicationProcessIdImpl(out.GetPointer()));
    }

    Result ShellInterfaceTipc::BoostSystemMemoryResourceLimit(u64 size) {
        R_RETURN(this->BoostSystemMemoryResourceLimitImpl(size));
    }

    Result ShellInterfaceTipc::IsRunningProcess(ams::tipc::Out<bool> out, os::ProcessId process_id) {
        R_RETURN(this->IsRunningProcessImpl(out.GetPointer(), process_id));
    }

    Result ShellInterfaceTipc::EnableApplicationCrashReport(bool enabled) {
        R_RETURN(this->EnableApplicationCrashReportImpl(enabled));
    }

    Result ShellInterfaceTipc::IsApplicationCrashReportEnabled(ams::tipc::Out<bool> out) {
        R_RETURN(this->IsApplicationCrashReportEnabledImpl(out.GetPointer()));
    }

    Result ShellInterfaceTipc::EnableApplicationAllThreadDumpOnCrash(bool enabled) {
        R_RETURN(this->EnableApplicationAllThreadDumpOnCrashImpl(enabled));
    }

    Result ShellInterfaceTipc::GetProcessId(ams::tipc::Out<os::ProcessId> out, ncm::ProgramId program_id) {
        R_RETURN(this->GetProcessIdImpl(out.GetPointer(), program_id));
    }

    Result ShellInterfaceTipc::CreateShellEvent(ams::tipc::OutMoveHandle out) {
        R_RETURN(pgl::srv::AllocateShellEventObserverForTipc(out.GetPointer()));
    }
    
    Result ShellInterfaceTipc::EnableApplicationCrashReport2(os::ProcessId process_id, bool enabled) {
        R_RETURN(this->EnableApplicationCrashReport2Impl(process_id, enabled));
    }
}