phase 5: transfer service

src/domain/account.cpp

@@ -0,0 +1,141 @@
+/*
+ *   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 <nxst/domain/account.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 AccountUid{};
+}