@rhseung/ps-cli 1.7.5 → 1.9.0

package/dist/commands/stats.js

@@ -1,84 +1,184 @@
 #!/usr/bin/env node
 import {
-  source_default
-} from "../chunk-ASMT3CRD.js";
-import {
-  getUserStats
-} from "../chunk-A6STXEAE.js";
+  getUserProblemStats,
+  getUserStats,
+  getUserTagRatings,
+  getUserTop100,
+  scrapeUserStats
+} from "../chunk-YH6VDCNG.js";
 import {
   Command,
   CommandBuilder,
   CommandDef,
+  TIER_COLORS,
+  __decorateClass,
   calculateTierProgress,
+  defineFlags,
+  findProjectRoot,
+  getArchiveDir,
   getNextTierMinRating,
   getSolvedAcHandle,
+  getSolvingDir,
   getTierColor,
-  getTierName
-} from "../chunk-F4LZ6ENP.js";
-import {
-  __decorateClass
-} from "../chunk-7MQMPJ3X.js";
+  getTierName,
+  getTierShortName,
+  icons,
+  logger,
+  source_default
+} from "../chunk-JZK2464U.js";
 
 // src/commands/stats.tsx
-import { Alert } from "@inkjs/ui";
-import { Spinner } from "@inkjs/ui";
-import { Box, Text, Transform } from "ink";
+import { Alert, Spinner } from "@inkjs/ui";
+import { BarChart, StackedBarChart } from "@pppp606/ink-chart";
+import { Box, Text } from "ink";
 
 // src/hooks/use-user-stats.ts
+import { existsSync } from "fs";
+import { readdir, stat } from "fs/promises";
+import { join } from "path";
 import { useEffect, useState } from "react";
+async function countProblems(dir) {
+  let count = 0;
+  try {
+    if (!existsSync(dir)) return 0;
+    const entries = await readdir(dir);
+    for (const entry of entries) {
+      if (entry.startsWith(".")) continue;
+      const fullPath = join(dir, entry);
+      const s = await stat(fullPath);
+      if (s.isDirectory()) {
+        if (existsSync(join(fullPath, "meta.json"))) {
+          count++;
+        } else {
+          count += await countProblems(fullPath);
+        }
+      }
+    }
+  } catch {
+  }
+  return count;
+}
 function useUserStats({
   handle,
-  onComplete
+  onComplete,
+  fetchLocalCount = false
 }) {
   const [status, setStatus] = useState(
     "loading"
   );
   const [user, setUser] = useState(null);
+  const [top100, setTop100] = useState(null);
+  const [problemStats, setProblemStats] = useState(null);
+  const [tagRatings, setTagRatings] = useState(
+    null
+  );
+  const [bojStats, setBojStats] = useState(null);
+  const [localSolvedCount, setLocalSolvedCount] = useState(null);
   const [error, setError] = useState(null);
   useEffect(() => {
-    void getUserStats(handle).then((userData) => {
-      setUser(userData);
-      setStatus("success");
-      setTimeout(() => {
-        onComplete();
-      }, 5e3);
-    }).catch((err) => {
-      setError(err instanceof Error ? err.message : String(err));
-      setStatus("error");
-      setTimeout(() => {
-        onComplete();
-      }, 3e3);
-    });
-  }, [handle, onComplete]);
+    async function fetchData() {
+      try {
+        const userData = await getUserStats(handle).catch((err) => {
+          if (err instanceof Error && err.message.includes("404")) {
+            throw new Error(`\uC0AC\uC6A9\uC790 '${handle}'\uC744(\uB97C) \uCC3E\uC744 \uC218 \uC5C6\uC2B5\uB2C8\uB2E4.`);
+          }
+          throw err;
+        });
+        if (!userData) {
+          throw new Error(`\uC0AC\uC6A9\uC790 '${handle}'\uC744(\uB97C) \uCC3E\uC744 \uC218 \uC5C6\uC2B5\uB2C8\uB2E4.`);
+        }
+        setUser(userData);
+        const [top100Data, problemStatsData, tagRatingsData, bojStatsData] = await Promise.all([
+          getUserTop100(handle).catch((err) => {
+            console.error("Error fetching top 100:", err);
+            return null;
+          }),
+          getUserProblemStats(handle).catch((err) => {
+            console.error("Error fetching problem stats:", err);
+            return null;
+          }),
+          getUserTagRatings(handle).catch((err) => {
+            console.error("Error fetching tag ratings:", err);
+            return null;
+          }),
+          scrapeUserStats(handle).catch((err) => {
+            console.error("Error scraping BOJ stats:", err);
+            return null;
+          })
+        ]);
+        setTop100(top100Data);
+        setProblemStats(problemStatsData);
+        setTagRatings(tagRatingsData);
+        setBojStats(bojStatsData);
+        if (fetchLocalCount) {
+          const projectRoot = findProjectRoot();
+          if (projectRoot) {
+            const archiveDir = getArchiveDir();
+            const solvingDir = getSolvingDir();
+            const archivePath = join(projectRoot, archiveDir);
+            const solvingPath = join(projectRoot, solvingDir);
+            const [archiveCount, solvingCount] = await Promise.all([
+              countProblems(archivePath),
+              countProblems(solvingPath)
+            ]);
+            setLocalSolvedCount(archiveCount + solvingCount);
+          }
+        }
+        setStatus("success");
+        setTimeout(() => {
+          onComplete();
+        }, 5e3);
+      } catch (err) {
+        setError(err instanceof Error ? err.message : String(err));
+        setStatus("error");
+        setTimeout(() => {
+          onComplete();
+        }, 3e3);
+      }
+    }
+    void fetchData();
+  }, [fetchLocalCount, handle, onComplete]);
   return {
     status,
     user,
+    top100,
+    problemStats,
+    tagRatings,
+    bojStats,
+    localSolvedCount,
     error
   };
 }
 
 // src/commands/stats.tsx
 import { jsx, jsxs } from "react/jsx-runtime";
-function ProgressBarWithColor({ value, colorFn }) {
-  const width = process.stdout.columns || 40;
-  const barWidth = Math.max(10, Math.min(30, width - 20));
-  const filled = Math.round(value / 100 * barWidth);
-  const empty = barWidth - filled;
-  const filledBar = "\u2588".repeat(filled);
-  const emptyBar = "\u2591".repeat(empty);
-  const barText = filledBar + emptyBar;
-  return /* @__PURE__ */ jsx(Transform, { transform: (output) => colorFn(output), children: /* @__PURE__ */ jsx(Text, { children: barText }) });
-}
-function StatsView({ handle, onComplete }) {
-  const { status, user, error } = useUserStats({
+var statsFlagsSchema = {
+  handle: {
+    type: "string",
+    shortFlag: "h",
+    description: "Solved.ac \uD578\uB4E4 (\uC124\uC815\uC5D0 \uC800\uC7A5\uB41C \uAC12 \uC0AC\uC6A9 \uAC00\uB2A5)"
+  }
+};
+function StatsView({ handle, onComplete, showLocalStats }) {
+  const {
+    status,
+    user,
+    top100,
+    problemStats,
+    tagRatings,
+    bojStats,
+    localSolvedCount,
+    error
+  } = useUserStats({
     handle,
-    onComplete
+    onComplete,
+    fetchLocalCount: showLocalStats
   });
   if (status === "loading") {
-    return /* @__PURE__ */ jsx(Box, { flexDirection: "column", children: /* @__PURE__ */ jsx(Spinner, { label: "\uD1B5\uACC4\uB97C \uBD88\uB7EC\uC624\uB294 \uC911..." }) });
+    return /* @__PURE__ */ jsx(Box, { flexDirection: "column", paddingY: 1, children: /* @__PURE__ */ jsx(Spinner, { label: "\uD1B5\uACC4\uB97C \uBD88\uB7EC\uC624\uB294 \uC911..." }) });
   }
   if (status === "error") {
-    return /* @__PURE__ */ jsx(Box, { flexDirection: "column", children: /* @__PURE__ */ jsxs(Alert, { variant: "error", children: [
+    return /* @__PURE__ */ jsx(Box, { flexDirection: "column", paddingY: 1, children: /* @__PURE__ */ jsxs(Alert, { variant: "error", children: [
       "\uD1B5\uACC4\uB97C \uBD88\uB7EC\uC62C \uC218 \uC5C6\uC2B5\uB2C8\uB2E4: ",
       error
     ] }) });
@@ -89,50 +189,209 @@ function StatsView({ handle, onComplete }) {
     const tierColorFn = typeof tierColor === "string" ? source_default.hex(tierColor) : tierColor.multiline;
     const nextTierMin = getNextTierMinRating(user.tier);
     const progress = user.tier === 31 ? 100 : calculateTierProgress(user.rating, user.tier);
-    return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", children: [
-      /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsxs(Text, { color: "cyan", bold: true, children: [
-        "\u2728 ",
-        user.handle
-      ] }) }),
-      /* @__PURE__ */ jsx(Box, { marginBottom: 1, flexDirection: "row", gap: 1, children: /* @__PURE__ */ jsxs(Text, { children: [
-        tierColorFn(tierName),
-        " ",
-        /* @__PURE__ */ jsx(Text, { bold: true, children: tierColorFn(user.rating.toLocaleString()) }),
-        nextTierMin !== null && /* @__PURE__ */ jsx(Text, { bold: true, children: " / " + nextTierMin.toLocaleString() })
-      ] }) }),
-      /* @__PURE__ */ jsx(Box, { flexDirection: "column", marginBottom: 1, children: /* @__PURE__ */ jsx(ProgressBarWithColor, { value: progress, colorFn: tierColorFn }) }),
+    const tierDistData = problemStats ? [
+      {
+        label: "Bronze",
+        value: problemStats.filter((s) => s.level >= 1 && s.level <= 5).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[3]
+        // Bronze III
+      },
+      {
+        label: "Silver",
+        value: problemStats.filter((s) => s.level >= 6 && s.level <= 10).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[8]
+        // Silver III
+      },
+      {
+        label: "Gold",
+        value: problemStats.filter((s) => s.level >= 11 && s.level <= 15).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[13]
+        // Gold III
+      },
+      {
+        label: "Platinum",
+        value: problemStats.filter((s) => s.level >= 16 && s.level <= 20).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[18]
+        // Platinum III
+      },
+      {
+        label: "Diamond",
+        value: problemStats.filter((s) => s.level >= 21 && s.level <= 25).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[23]
+        // Diamond III
+      },
+      {
+        label: "Ruby",
+        value: problemStats.filter((s) => s.level >= 26 && s.level <= 30).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[28]
+        // Ruby III
+      },
+      {
+        label: "Master",
+        value: problemStats.filter((s) => s.level === 31).reduce((a, b) => a + b.solved, 0),
+        color: TIER_COLORS[31]
+        // Master
+      }
+    ].filter((d) => d.value > 0) : [];
+    const tagChartData = tagRatings ? tagRatings.sort((a, b) => b.rating - a.rating).slice(0, 8).map((tr) => ({
+      label: tr.tag.displayNames.find((dn) => dn.language === "ko")?.name || tr.tag.key,
+      value: tr.rating
+    })) : [];
+    const bojSummaryData = bojStats ? [
+      { label: "\uC815\uB2F5", value: bojStats.accepted, color: "green" },
+      { label: "\uC624\uB2F5", value: bojStats.wrong, color: "red" },
+      {
+        label: "TLE/MLE",
+        value: bojStats.timeout + bojStats.memory,
+        color: "yellow"
+      },
+      {
+        label: "\uAE30\uD0C0",
+        value: bojStats.runtimeError + bojStats.compileError,
+        color: "gray"
+      }
+    ].filter((d) => d.value > 0) : [];
+    return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", gap: 1, children: [
       /* @__PURE__ */ jsx(
+        Box,
+        {
+          flexDirection: "row",
+          justifyContent: "space-between",
+          alignItems: "flex-end",
+          children: /* @__PURE__ */ jsxs(Box, { flexDirection: "column", children: [
+            /* @__PURE__ */ jsxs(Text, { bold: true, children: [
+              icons.user,
+              " ",
+              source_default.cyan(user.handle)
+            ] }),
+            /* @__PURE__ */ jsxs(Text, { color: "gray", children: [
+              "https://solved.ac/profile/",
+              user.handle
+            ] })
+          ] })
+        }
+      ),
+      /* @__PURE__ */ jsxs(
         Box,
         {
           flexDirection: "column",
           borderStyle: "round",
           borderColor: "gray",
-          alignSelf: "flex-start",
-          children: /* @__PURE__ */ jsxs(Box, { paddingX: 1, paddingY: 0, flexDirection: "column", children: [
+          paddingX: 1,
+          children: [
+            /* @__PURE__ */ jsxs(Box, { justifyContent: "space-between", marginBottom: 1, children: [
+              /* @__PURE__ */ jsxs(Box, { gap: 1, children: [
+                /* @__PURE__ */ jsx(Text, { bold: true, children: tierColorFn(tierName) }),
+                /* @__PURE__ */ jsx(Text, { bold: true, children: tierColorFn(user.rating.toLocaleString()) }),
+                nextTierMin !== null && /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
+                  " / ",
+                  nextTierMin.toLocaleString()
+                ] })
+              ] }),
+              /* @__PURE__ */ jsxs(Text, { color: "yellow", children: [
+                icons.trophy,
+                " Rank #",
+                user.rank.toLocaleString()
+              ] })
+            ] }),
+            /* @__PURE__ */ jsx(
+              StackedBarChart,
+              {
+                data: [
+                  {
+                    label: "Progress",
+                    value: progress,
+                    color: typeof tierColor === "string" ? tierColor : "#ff7ca8"
+                  },
+                  ...progress < 100 ? [{ label: "Remaining", value: 100 - progress, color: "#333" }] : []
+                ],
+                showLabels: false,
+                showValues: false,
+                width: "full"
+              }
+            )
+          ]
+        }
+      ),
+      /* @__PURE__ */ jsxs(Box, { flexDirection: "row", gap: 2, children: [
+        /* @__PURE__ */ jsxs(Box, { flexDirection: "column", flexGrow: 1, children: [
+          /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "[ \uC0C1\uC138 \uD1B5\uACC4 ]" }) }),
+          /* @__PURE__ */ jsxs(Box, { flexDirection: "column", gap: 0, children: [
             /* @__PURE__ */ jsxs(Text, { children: [
               "\uD574\uACB0\uD55C \uBB38\uC81C:",
               " ",
               /* @__PURE__ */ jsx(Text, { bold: true, color: "green", children: user.solvedCount.toLocaleString() }),
+              " ",
               "\uAC1C"
             ] }),
+            localSolvedCount !== null && /* @__PURE__ */ jsxs(Text, { color: "gray", children: [
+              " ",
+              "\u2514 \uB85C\uCEEC \uAD00\uB9AC: ",
+              /* @__PURE__ */ jsx(Text, { bold: true, children: localSolvedCount }),
+              " \uAC1C"
+            ] }),
             /* @__PURE__ */ jsxs(Text, { children: [
-              "\uD074\uB798\uC2A4: ",
-              /* @__PURE__ */ jsx(Text, { bold: true, children: user.class })
+              "\uD074\uB798\uC2A4:",
+              " ",
+              /* @__PURE__ */ jsxs(Text, { bold: true, color: "magenta", children: [
+                user.class,
+                user.classDecoration === "gold" ? "++" : user.classDecoration === "silver" ? "+" : ""
+              ] })
             ] }),
-            user.maxStreak > 0 && /* @__PURE__ */ jsxs(Text, { children: [
-              "\uCD5C\uB300 \uC5F0\uC18D \uD574\uACB0:",
+            /* @__PURE__ */ jsxs(Text, { children: [
+              "\uCD5C\uB300 \uC2A4\uD2B8\uB9AD:",
+              " ",
+              /* @__PURE__ */ jsx(Text, { bold: true, color: "orange", children: user.maxStreak }),
               " ",
-              /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: user.maxStreak }),
               "\uC77C"
             ] }),
             /* @__PURE__ */ jsxs(Text, { children: [
-              "\uC21C\uC704: ",
-              /* @__PURE__ */ jsx(Text, { bold: true, children: user.rank.toLocaleString() }),
-              "\uC704"
+              "\uAE30\uC5EC \uD69F\uC218:",
+              " ",
+              /* @__PURE__ */ jsx(Text, { bold: true, color: "blue", children: user.voteCount }),
+              " ",
+              "\uD68C"
             ] })
+          ] }),
+          bojSummaryData.length > 0 && /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginTop: 1, children: [
+            /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "[ \uBC31\uC900 \uC81C\uCD9C \uC694\uC57D ]" }) }),
+            /* @__PURE__ */ jsx(
+              StackedBarChart,
+              {
+                data: bojSummaryData,
+                mode: "absolute",
+                width: 30
+              }
+            )
           ] })
-        }
-      )
+        ] }),
+        tagChartData.length > 0 && /* @__PURE__ */ jsxs(Box, { flexDirection: "column", width: 40, children: [
+          /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "[ \uC54C\uACE0\uB9AC\uC998 \uAC15\uC810 (Tag Rating) ]" }) }),
+          /* @__PURE__ */ jsx(
+            BarChart,
+            {
+              data: tagChartData,
+              showValue: "right",
+              barChar: "\u2588",
+              color: "cyan"
+            }
+          )
+        ] })
+      ] }),
+      tierDistData.length > 0 && /* @__PURE__ */ jsxs(Box, { flexDirection: "column", children: [
+        /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "[ \uD2F0\uC5B4\uBCC4 \uD574\uACB0 \uBB38\uC81C \uBD84\uD3EC ]" }) }),
+        /* @__PURE__ */ jsx(StackedBarChart, { data: tierDistData, width: "full" })
+      ] }),
+      top100 && top100.length > 0 && /* @__PURE__ */ jsxs(Box, { flexDirection: "column", children: [
+        /* @__PURE__ */ jsx(Box, { marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "[ \uC0C1\uC704 100\uBB38\uC81C \uBD84\uD3EC ]" }) }),
+        /* @__PURE__ */ jsx(Box, { flexDirection: "column", children: Array.from({ length: Math.ceil(top100.length / 20) }).map(
+          (_, rowIndex) => /* @__PURE__ */ jsx(Box, { flexDirection: "row", children: top100.slice(rowIndex * 20, (rowIndex + 1) * 20).map((p, colIndex) => {
+            const tierColor2 = getTierColor(p.level);
+            const tierColorFn2 = typeof tierColor2 === "string" ? source_default.hex(tierColor2) : tierColor2.multiline;
+            return /* @__PURE__ */ jsx(Box, { width: 3, children: /* @__PURE__ */ jsx(Text, { children: tierColorFn2(getTierShortName(p.level)) }) }, colIndex);
+          }) }, rowIndex)
+        ) })
+      ] })
     ] });
   }
   return null;
@@ -144,17 +403,15 @@ var StatsCommand = class extends Command {
       handle = getSolvedAcHandle();
     }
     if (!handle) {
-      console.error("\uC624\uB958: Solved.ac \uD578\uB4E4\uC744 \uC785\uB825\uD574\uC8FC\uC138\uC694.");
-      console.error(`\uC0AC\uC6A9\uBC95: ps stats <\uD578\uB4E4>`);
-      console.error(`\uB3C4\uC6C0\uB9D0: ps stats --help`);
-      console.error(
-        `\uD78C\uD2B8: \uC124\uC815\uC5D0 \uD578\uB4E4\uC744 \uC800\uC7A5\uD558\uBA74 \uB9E4\uBC88 \uC785\uB825\uD560 \uD544\uC694\uAC00 \uC5C6\uC2B5\uB2C8\uB2E4.`
-      );
+      logger.error("Solved.ac \uD578\uB4E4\uC744 \uC785\uB825\uD574\uC8FC\uC138\uC694.");
+      console.log(`\uB3C4\uC6C0\uB9D0: ps stats --help`);
       process.exit(1);
       return;
     }
+    const showLocalStats = !args[0] && !flags.handle;
     await this.renderView(StatsView, {
-      handle
+      handle,
+      showLocalStats
     });
   }
 };
@@ -163,16 +420,8 @@ StatsCommand = __decorateClass([
     name: "stats",
     description: `Solved.ac\uC5D0\uC11C \uC0AC\uC6A9\uC790 \uD1B5\uACC4\uB97C \uC870\uD68C\uD569\uB2C8\uB2E4.
 - \uD2F0\uC5B4, \uB808\uC774\uD305, \uD574\uACB0\uD55C \uBB38\uC81C \uC218 \uB4F1 \uD45C\uC2DC
-- \uADF8\uB77C\uB370\uC774\uC158\uC73C\uB85C \uC2DC\uAC01\uC801\uC73C\uB85C \uD45C\uC2DC`,
-    flags: [
-      {
-        name: "handle",
-        options: {
-          shortFlag: "h",
-          description: "Solved.ac \uD578\uB4E4 (\uC124\uC815\uC5D0 \uC800\uC7A5\uB41C \uAC12 \uC0AC\uC6A9 \uAC00\uB2A5)"
-        }
-      }
-    ],
+- \uAE30\uBCF8 Solved.ac \uD578\uB4E4\uC740 ps config\uC5D0\uC11C \uC124\uC815 \uAC00\uB2A5\uD569\uB2C8\uB2E4.`,
+    flags: defineFlags(statsFlagsSchema),
     autoDetectProblemId: false,
     examples: ["stats myhandle", "stats --handle myhandle"]
   })