/*
 *   This file is part of Checkpoint
 *   Copyright (C) 2017-2019 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"

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

Result Account::init(void)
{
    return accountInitialize();
}

void Account::exit(void)
{
    for (auto& value : mUsers) {
        SDL_DestroyTexture(value.second.icon);
    }
    accountExit();
}

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

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

    if (R_SUCCEEDED(accountGetProfile(&profile, id)) && R_SUCCEEDED(accountProfileGet(&profile, NULL, &profilebase))) {
        user.name = std::string(profilebase.username, 0x20);

        // load icon
        u8* buffer;
        size_t image_size, real_size;
        if (R_SUCCEEDED(accountProfileGetImageSize(&profile, &image_size)) && (buffer = (u8*)malloc(image_size)) != NULL &&
            R_SUCCEEDED(accountProfileLoadImage(&profile, buffer, image_size, &real_size))) {
            SDLH_LoadImage(&user.icon, buffer, image_size);
            free(buffer);
        }
    }

    accountProfileClose(&profile);
    return user;
}

std::string Account::username(u128 id)
{
    std::map<u128, 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;
}

SDL_Texture* Account::icon(u128 id)
{
    std::map<u128, User>::const_iterator got = mUsers.find(id);
    if (got == mUsers.end()) {
        User user = getUser(id);
        mUsers.insert({id, user});
        return user.icon;
    }
    return got->second.icon;
}

u128 Account::selectAccount(void)
{
    struct UserReturnData {
        u64 result;
        u128 UID;
    } PACKED;

    struct UserReturnData outdata;

    AppletHolder aph;
    AppletStorage ast;
    AppletStorage hast1;
    LibAppletArgs args;

    u8 indata[0xA0] = {0};
    indata[0x96]    = 1;

    appletCreateLibraryApplet(&aph, AppletId_playerSelect, LibAppletMode_AllForeground);
    libappletArgsCreate(&args, 0);
    libappletArgsPush(&args, &aph);

    appletCreateStorage(&hast1, 0xA0);

    appletStorageWrite(&hast1, 0, indata, 0xA0);
    appletHolderPushInData(&aph, &hast1);
    appletHolderStart(&aph);

    while (appletHolderWaitInteractiveOut(&aph))
        ;

    appletHolderJoin(&aph);
    appletHolderPopOutData(&aph, &ast);
    appletStorageRead(&ast, 0, &outdata, 24);

    appletHolderClose(&aph);
    appletStorageClose(&ast);
    appletStorageClose(&hast1);

    return outdata.UID;
}