/*
 *   This file is part of Checkpoint
 *   Copyright (C) 2017-2021 Bernardo Giordano, FlagBrew
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that 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/>.
 *
 *   Additional Terms 7.b and 7.c of GPLv3 apply to this file:
 *       * Requiring preservation of specified reasonable legal notices or
 *         author attributions in that material or in the Appropriate Legal
 *         Notices displayed by works containing it.
 *       * Prohibiting misrepresentation of the origin of that material,
 *         or requiring that modified versions of such material be marked in
 *         reasonable ways as different from the original version.
 */

#include "account.hpp"
#include <main.hpp>
#include <sys/stat.h>
#include <cstdio>

static std::map<AccountUid, User> mUsers;

Result Account::init(void)
{
    Result res = accountInitialize(AccountServiceType_Application);
    if (R_FAILED(res)) return res;

    AccountUid uids[8];
    s32 count = 0;
    accountListAllUsers(uids, 8, &count);
    for (s32 i = 0; i < count; i++) {
        Account::username(uids[i]);  // populates mUsers as side effect
    }
    return 0;
}

void Account::exit(void)
{
    accountExit();
}

std::vector<AccountUid> Account::ids(void)
{
    std::vector<AccountUid> v;
    for (auto& value : mUsers) {
        v.push_back(value.second.id);
    }
    return v;
}

static User getUser(AccountUid id)
{
    User user{id, ""};
    AccountProfile profile;
    AccountProfileBase profilebase;
    memset(&profilebase, 0, sizeof(profilebase));

    if (R_SUCCEEDED(accountGetProfile(&profile, id))) {
        if (R_SUCCEEDED(accountProfileGet(&profile, NULL, &profilebase))) {
            user.name = std::string(profilebase.nickname);
        }
        accountProfileClose(&profile);
    }
    return user;
}

std::string Account::username(AccountUid id)
{
    std::map<AccountUid, User>::const_iterator got = mUsers.find(id);
    if (got == mUsers.end()) {
        User user = getUser(id);
        mUsers.insert({id, user});
        return user.name;
    }

    return got->second.name;
}

std::string Account::iconPath(AccountUid id)
{
    char path[128];
    snprintf(path, sizeof(path), "sdmc:/switch/NXST/cache/%016lX%016lX.jpg",
             id.uid[0], id.uid[1]);

    struct stat st;
    if (stat(path, &st) == 0 && st.st_size > 0) return std::string(path);

    mkdir("sdmc:/switch", 0755);
    mkdir("sdmc:/switch/NXST", 0755);
    mkdir("sdmc:/switch/NXST/cache", 0755);

    AccountProfile profile;
    if (R_FAILED(accountGetProfile(&profile, id))) return "";

    u32 imgSize = 0;
    if (R_FAILED(accountProfileGetImageSize(&profile, &imgSize)) || imgSize == 0) {
        accountProfileClose(&profile);
        return "";
    }

    std::vector<u8> buf(imgSize);
    u32 outSize = 0;
    Result r = accountProfileLoadImage(&profile, buf.data(), imgSize, &outSize);
    accountProfileClose(&profile);
    if (R_FAILED(r) || outSize == 0) return "";

    FILE* f = fopen(path, "wb");
    if (!f) return "";
    fwrite(buf.data(), 1, outSize, f);
    fclose(f);
    return std::string(path);
}

AccountUid Account::selectAccount(void)
{
    LibAppletArgs args;
    libappletArgsCreate(&args, 0x10000);
    u8 st_in[0xA0]  = {0};
    u8 st_out[0x18] = {0};
    size_t repsz;

    Result res = libappletLaunch(AppletId_LibraryAppletPlayerSelect, &args, st_in, 0xA0, st_out, 0x18, &repsz);
    if (R_SUCCEEDED(res)) {
        u64 lres       = *(u64*)st_out;
        AccountUid uid = *(AccountUid*)&st_out[8];
        if (lres == 0)
            return uid;
    }

    return g_currentUId;
}