@raingor/pi-web-switch 0.4.1 → 0.4.3

package/vite.config.ts

@@ -3,21 +3,31 @@ import react from "@vitejs/plugin-react";
 import tailwindcss from "@tailwindcss/vite";
 import path from "path";
 import type { Connect } from "vite";
-import { chatApiPlugin } from "./server/chat-api-plugin";
 
 // ─── Pi Config API Plugin ───────────────────────────────
 
 function piApiPlugin(): Plugin {
-  // Lazy-load the server-side module (Node.js only)
-  let pi: typeof import("./server/pi-reader");
-  let builtins: typeof import("./src/data/builtin-providers");
-
   return {
     name: "pi-api",
     configureServer(server) {
-      // Load server-side modules
-      pi = require("./server/pi-reader");
-      builtins = require("./src/data/builtin-providers");
+      // Lazy-load the server-side module (Node.js only)
+      const pi = require("./server/pi-reader");
+      const builtins = require("./src/data/builtin-providers");
+
+      // Warm the usage cache in the background so the dashboard's first
+      // request doesn't block on scanning ~150MB of session JSONL.
+      setTimeout(() => {
+        try {
+          pi.readAllUsage();
+        } catch {
+          /* ignore warm-up failure */
+        }
+        try {
+          pi.readCopilotUsage();
+        } catch {
+          /* ignore warm-up failure */
+        }
+      }, 0);
 
       const routes: Record<string, (req: Connect.IncomingMessage, res: any) => void> = {
         "GET /api/pi/settings"(_, res) {
@@ -116,6 +126,28 @@ function piApiPlugin(): Plugin {
           res.setHeader("Content-Type", "application/json");
           res.end(JSON.stringify(pi.listTrash()));
         },
+        "GET /api/pi/copilot-config"(_, res) {
+          res.setHeader("Content-Type", "application/json");
+          res.end(JSON.stringify(pi.readCopilotConfig() ?? {}));
+        },
+        "POST /api/pi/copilot-config"(req, res) {
+          let body = "";
+          req.on("data", (chunk: string) => (body += chunk));
+          req.on("end", () => {
+            try {
+              const cfg = JSON.parse(body) as { username?: string; token?: string };
+              const ok = pi.writeCopilotConfig(cfg);
+              // Config changed → drop cached usage so the next view refetches.
+              pi.clearCopilotCaches();
+              res.setHeader("Content-Type", "application/json");
+              res.end(JSON.stringify({ success: ok }));
+            } catch {
+              res.statusCode = 400;
+              res.setHeader("Content-Type", "application/json");
+              res.end(JSON.stringify({ success: false, error: "Invalid request body" }));
+            }
+          });
+        },
         "POST /api/pi/session/trash"(req, res) {
           let body = "";
           req.on("data", (chunk: string) => (body += chunk));
@@ -283,8 +315,14 @@ function piApiPlugin(): Plugin {
           const toParam = parsedUrl.searchParams.get("to") || "";
 
           const now = new Date();
+          // Date buckets follow China time (UTC+8) regardless of system timezone
           const localDateStr = (dt: Date) =>
-            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+            new Intl.DateTimeFormat("en-CA", {
+              timeZone: "Asia/Shanghai",
+              year: "numeric",
+              month: "2-digit",
+              day: "2-digit",
+            }).format(dt);
           let fromDate: string;
           let toDate = localDateStr(now);
 
@@ -315,8 +353,14 @@ function piApiPlugin(): Plugin {
           const toParam = parsedUrl.searchParams.get("to") || "";
 
           const now = new Date();
+          // Date buckets follow China time (UTC+8) regardless of system timezone
           const localDateStr = (dt: Date) =>
-            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+            new Intl.DateTimeFormat("en-CA", {
+              timeZone: "Asia/Shanghai",
+              year: "numeric",
+              month: "2-digit",
+              day: "2-digit",
+            }).format(dt);
           let fromDate: string;
           let toDate = localDateStr(now);
 
@@ -347,8 +391,14 @@ function piApiPlugin(): Plugin {
           const toParam = parsedUrl.searchParams.get("to") || "";
 
           const now = new Date();
+          // Date buckets follow China time (UTC+8) regardless of system timezone
           const localDateStr = (dt: Date) =>
-            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+            new Intl.DateTimeFormat("en-CA", {
+              timeZone: "Asia/Shanghai",
+              year: "numeric",
+              month: "2-digit",
+              day: "2-digit",
+            }).format(dt);
           let fromDate: string;
           let toDate = localDateStr(now);
 
@@ -379,8 +429,14 @@ function piApiPlugin(): Plugin {
           const toParam = parsedUrl.searchParams.get("to") || "";
 
           const now = new Date();
+          // Date buckets follow China time (UTC+8) regardless of system timezone
           const localDateStr = (dt: Date) =>
-            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+            new Intl.DateTimeFormat("en-CA", {
+              timeZone: "Asia/Shanghai",
+              year: "numeric",
+              month: "2-digit",
+              day: "2-digit",
+            }).format(dt);
           let fromDate: string;
           let toDate = localDateStr(now);
 
@@ -406,8 +462,14 @@ function piApiPlugin(): Plugin {
         // Helper: resolve date range params
         const resolveDateRange = (range: string, fromParam: string, toParam: string) => {
           const now = new Date();
+          // Date buckets follow China time (UTC+8) regardless of system timezone
           const localDateStr = (dt: Date) =>
-            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+            new Intl.DateTimeFormat("en-CA", {
+              timeZone: "Asia/Shanghai",
+              year: "numeric",
+              month: "2-digit",
+              day: "2-digit",
+            }).format(dt);
           let fromDate: string;
           let toDate = localDateStr(now);
           if (range === "today") fromDate = toDate;
@@ -432,7 +494,10 @@ function piApiPlugin(): Plugin {
         }
 
         // Handle provider-filtered endpoints: /api/pi/{provider}-usage-range
-        const providerMatch = pathOnly.match(/^\/api\/pi\/(opencode|gemini|grok)-usage-range$/);
+        // Copilot is read from the local ~/.copilot/session-store.db (no
+        // GitHub REST API, no PAT), so it shares the same sync pipeline as
+        // the other local sources.
+        const providerMatch = pathOnly.match(/^\/api\/pi\/(atomcode|copilot|opencode|gemini|grok)-usage-range$/);
         if (method === "GET" && providerMatch) {
           const providerId = providerMatch[1]!;
           const parsedUrl = new URL(url, "http://localhost");
@@ -463,11 +528,13 @@ function piApiPlugin(): Plugin {
 // ─── Vite Config ────────────────────────────────────────
 
 export default defineConfig({
+  // Relative base so Electron can loadFile() the built HTML from disk —
+  // absolute "/assets/..." URLs would resolve to the filesystem root.
+  base: './',
   plugins: [
     react(),
     tailwindcss(),
     piApiPlugin(),
-    chatApiPlugin(),
   ],
   resolve: {
     alias: {
@@ -478,4 +545,11 @@ export default defineConfig({
     port: 5176,
     strictPort: true,
   },
+  build: {
+    rollupOptions: {
+      input: {
+        main: path.resolve(__dirname, "index.html"),
+      },
+    },
+  },
 });