chn/nixos
1
0
Fork 0
mirror of https://github.com/CHN-beta/nixos.git synced 2024-10-23 06:18:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

add subaccount to login message

Browse Source
This commit is contained in:
陈浩南 2024-05-06 14:36:58 +08:00
parent 9ab1997e78
commit 4f79141d1b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
local/pkgs/hpcstat/src/main.cpp
View File

@ -27,6 +27,7 @@ int main(int argc, const char** argv)
return 1; return 1;
else else
{ {
auto sub_account = env::env("HPCSTAT_SUBACCOUNT");
sql::LoginData data sql::LoginData data
{ {
.Time = now(), .Key = *fp, .SessionId = *session, .Subaccount = env::env("HPCSTAT_SUBACCOUNT"), .Time = now(), .Key = *fp, .SessionId = *session, .Subaccount = env::env("HPCSTAT_SUBACCOUNT"),
@ -36,8 +37,11 @@ int main(int argc, const char** argv)
if (!signature) return 1; if (!signature) return 1;
data.Signature = *signature; data.Signature = *signature;
sql::writedb(data); sql::writedb(data);
if (env::interactive()) if (env::interactive()) std::cout << fmt::format
std::cout << fmt::format("\33[2K\rLogged in as {} (fingerprint: SHA256:{}).\n", Keys[*fp].Username, *fp); (
"\33[2K\rLogged in as {} (Fingerprint: SHA256:{}{}).\n", Keys[*fp].Username, *fp,
sub_account ? fmt::format(" Subaccount {}", *sub_account) : ""
);
} }
} }
else if (args[1] == "logout") else if (args[1] == "logout")