From 4f79141d1b1808896a9eee5e9ffeba112d0cd35a Mon Sep 17 00:00:00 2001
From: chn <chn@chn.moe>
Date: Mon, 6 May 2024 14:36:58 +0800
Subject: [PATCH] add subaccount to login message

---
 local/pkgs/hpcstat/src/main.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/local/pkgs/hpcstat/src/main.cpp b/local/pkgs/hpcstat/src/main.cpp
index d6456615..c7d30c01 100644
--- a/local/pkgs/hpcstat/src/main.cpp
+++ b/local/pkgs/hpcstat/src/main.cpp
@@ -27,6 +27,7 @@ int main(int argc, const char** argv)
         return 1;
       else
       {
+        auto sub_account = env::env("HPCSTAT_SUBACCOUNT");
         sql::LoginData data
         {
           .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;
         data.Signature = *signature;
         sql::writedb(data);
-        if (env::interactive())
-          std::cout << fmt::format("\33[2K\rLogged in as {} (fingerprint: SHA256:{}).\n", Keys[*fp].Username, *fp);
+        if (env::interactive()) std::cout << fmt::format
+        (
+          "\33[2K\rLogged in as {} (Fingerprint: SHA256:{}{}).\n", Keys[*fp].Username, *fp,
+          sub_account ? fmt::format(" Subaccount {}", *sub_account) : ""
+        );
       }
     }
     else if (args[1] == "logout")