@raingor/pi-web-switch 0.4.0 → 0.4.1

package/src/types/chat.ts

@@ -0,0 +1,217 @@
+// Chat-related type definitions — ported and simplified from pi-web's lib/types.ts
+
+export interface TextContent {
+  type: "text";
+  text: string;
+}
+
+export interface ImageContent {
+  type: "image";
+  source: {
+    type: "base64" | "url";
+    media_type?: string;
+    data?: string;
+    url?: string;
+  };
+}
+
+export interface ThinkingContent {
+  type: "thinking";
+  thinking: string;
+  deferred?: boolean;
+}
+
+export interface ToolCallContent {
+  type: "toolCall";
+  toolCallId: string;
+  toolName: string;
+  input: Record<string, unknown>;
+}
+
+export type AssistantContentBlock = TextContent | ImageContent | ThinkingContent | ToolCallContent;
+
+export interface UserMessage {
+  role: "user";
+  content: string | (TextContent | ImageContent)[];
+  timestamp?: number;
+}
+
+export interface AssistantMessage {
+  role: "assistant";
+  content: AssistantContentBlock[];
+  model: string;
+  provider: string;
+  stopReason?: string;
+  errorMessage?: string;
+  timestamp?: number;
+  usage?: {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+    cost: {
+      input: number;
+      output: number;
+      cacheRead: number;
+      cacheWrite: number;
+      total: number;
+    };
+  };
+}
+
+export interface ToolResultMessage {
+  role: "toolResult";
+  toolCallId: string;
+  toolName?: string;
+  content: (TextContent | ImageContent)[];
+  isError?: boolean;
+  details?: unknown;
+  timestamp?: number;
+}
+
+export interface CustomMessage {
+  role: "custom";
+  customType: string;
+  content: string | (TextContent | ImageContent)[];
+  display: boolean;
+  details?: unknown;
+  timestamp?: number;
+}
+
+export interface BashExecutionMessage {
+  role: "bashExecution";
+  command: string;
+  output: string;
+  exitCode?: number;
+  cancelled?: boolean;
+  truncated?: boolean;
+  fullOutputPath?: string;
+  excludeFromContext?: boolean;
+  timestamp?: number;
+}
+
+export type AgentMessage = UserMessage | AssistantMessage | ToolResultMessage | CustomMessage | BashExecutionMessage;
+
+export interface SessionInfo {
+  path: string;
+  id: string;
+  cwd: string;
+  name?: string;
+  created: string;
+  modified: string;
+  messageCount: number;
+  firstMessage: string;
+  parentSessionId?: string;
+  projectRoot?: string;
+  worktreeBranch?: string;
+}
+
+export interface SessionContext {
+  messages: AgentMessage[];
+  entryIds: string[];
+  thinkingLevel: string;
+  model: { provider: string; modelId: string } | null;
+}
+
+export interface SessionTreeNode {
+  entry: { id: string; type: string };
+  children: SessionTreeNode[];
+  label?: string;
+  compressedEntryIds?: string[];
+}
+
+export interface SessionData {
+  sessionId: string;
+  filePath: string;
+  tree: SessionTreeNode[];
+  leafId: string | null;
+  context: SessionContext;
+}
+
+export interface ContextUsage {
+  percent: number | null;
+  contextWindow: number;
+  tokens: number | null;
+}
+
+export interface SessionStatsInfo {
+  sessionFile?: string;
+  sessionId: string;
+  sessionName?: string;
+  userMessages: number;
+  assistantMessages: number;
+  toolCalls: number;
+  toolResults: number;
+  totalMessages: number;
+  tokens: {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+    total: number;
+  };
+  cost: number;
+  contextUsage?: ContextUsage;
+}
+
+export interface AttachedImage {
+  data: string;
+  mimeType: string;
+  previewUrl: string;
+}
+
+export interface ChatInputHandle {
+  insertText: (text: string) => void;
+  insertIfEmpty: (content: string) => void;
+  prependText: (text: string) => void;
+  addImages: (files: File[]) => void;
+}
+
+export interface ModelEntry {
+  id: string;
+  name: string;
+  provider: string;
+}
+
+export interface ModelsResponse {
+  models: Record<string, string>;
+  modelList?: ModelEntry[];
+  defaultModel?: { provider: string; modelId: string } | null;
+  thinkingLevels?: Record<string, string[]>;
+  modelError?: string;
+}
+
+export interface AgentStateResponse {
+  isStreaming?: boolean;
+  isPromptRunning?: boolean;
+  isBashRunning?: boolean;
+  isCompacting?: boolean;
+  contextUsage?: ContextUsage | null;
+  systemPrompt?: string;
+  thinkingLevel?: string;
+  model?: { id: string; provider: string };
+}
+
+export interface NoticeItem {
+  id: string;
+  message: string;
+  type: "info" | "success" | "warning" | "error";
+  exiting?: boolean;
+}
+
+export type AgentPhase =
+  | { kind: "waiting_model" }
+  | { kind: "running_command" }
+  | { kind: "running_tools"; tools: { id: string; name: string }[] }
+  | null;
+
+export interface QueuedMessages {
+  steering: string[];
+  followUp: string[];
+}
+
+export interface SlashCommandInfo {
+  name: string;
+  description?: string;
+  source: "extension" | "prompt" | "skill";
+}

package/vite.config.ts

@@ -3,6 +3,7 @@ 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 ───────────────────────────────
 
@@ -306,6 +307,145 @@ function piApiPlugin(): Plugin {
           return res.end(JSON.stringify(usage));
         }
 
+        // Handle GET /api/pi/cindy-usage-range?range=today|7d|30d|custom&from=...&to=...
+        if (method === "GET" && pathOnly === "/api/pi/cindy-usage-range") {
+          const parsedUrl = new URL(url, "http://localhost");
+          const range = parsedUrl.searchParams.get("range") || "today";
+          const fromParam = parsedUrl.searchParams.get("from") || "";
+          const toParam = parsedUrl.searchParams.get("to") || "";
+
+          const now = new Date();
+          const localDateStr = (dt: Date) =>
+            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+          let fromDate: string;
+          let toDate = localDateStr(now);
+
+          if (range === "today") {
+            fromDate = toDate;
+          } else if (range === "7d") {
+            const d = new Date(now); d.setDate(d.getDate() - 6); fromDate = localDateStr(d);
+          } else if (range === "30d") {
+            const d = new Date(now); d.setDate(d.getDate() - 29); fromDate = localDateStr(d);
+          } else if (range === "custom" && fromParam) {
+            fromDate = fromParam;
+            if (toParam) toDate = toParam;
+          } else {
+            fromDate = toDate;
+          }
+
+          const allRecords = pi.readCindyUsage();
+          const usage = pi.getUsageByRange(allRecords, fromDate, toDate);
+          res.setHeader("Content-Type", "application/json");
+          return res.end(JSON.stringify(usage));
+        }
+
+        // Handle GET /api/pi/claude-usage-range?range=today|7d|30d|custom&from=...&to=...
+        if (method === "GET" && pathOnly === "/api/pi/claude-usage-range") {
+          const parsedUrl = new URL(url, "http://localhost");
+          const range = parsedUrl.searchParams.get("range") || "today";
+          const fromParam = parsedUrl.searchParams.get("from") || "";
+          const toParam = parsedUrl.searchParams.get("to") || "";
+
+          const now = new Date();
+          const localDateStr = (dt: Date) =>
+            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+          let fromDate: string;
+          let toDate = localDateStr(now);
+
+          if (range === "today") {
+            fromDate = toDate;
+          } else if (range === "7d") {
+            const d = new Date(now); d.setDate(d.getDate() - 6); fromDate = localDateStr(d);
+          } else if (range === "30d") {
+            const d = new Date(now); d.setDate(d.getDate() - 29); fromDate = localDateStr(d);
+          } else if (range === "custom" && fromParam) {
+            fromDate = fromParam;
+            if (toParam) toDate = toParam;
+          } else {
+            fromDate = toDate;
+          }
+
+          const allRecords = pi.readClaudeUsage();
+          const usage = pi.getUsageByRange(allRecords, fromDate, toDate);
+          res.setHeader("Content-Type", "application/json");
+          return res.end(JSON.stringify(usage));
+        }
+
+        // Handle GET /api/pi/codex-usage-range?range=today|7d|30d|custom&from=...&to=...
+        if (method === "GET" && pathOnly === "/api/pi/codex-usage-range") {
+          const parsedUrl = new URL(url, "http://localhost");
+          const range = parsedUrl.searchParams.get("range") || "today";
+          const fromParam = parsedUrl.searchParams.get("from") || "";
+          const toParam = parsedUrl.searchParams.get("to") || "";
+
+          const now = new Date();
+          const localDateStr = (dt: Date) =>
+            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+          let fromDate: string;
+          let toDate = localDateStr(now);
+
+          if (range === "today") {
+            fromDate = toDate;
+          } else if (range === "7d") {
+            const d = new Date(now); d.setDate(d.getDate() - 6); fromDate = localDateStr(d);
+          } else if (range === "30d") {
+            const d = new Date(now); d.setDate(d.getDate() - 29); fromDate = localDateStr(d);
+          } else if (range === "custom" && fromParam) {
+            fromDate = fromParam;
+            if (toParam) toDate = toParam;
+          } else {
+            fromDate = toDate;
+          }
+
+          const allRecords = pi.readCodexUsage();
+          const usage = pi.getUsageByRange(allRecords, fromDate, toDate);
+          res.setHeader("Content-Type", "application/json");
+          return res.end(JSON.stringify(usage));
+        }
+
+        // Helper: resolve date range params
+        const resolveDateRange = (range: string, fromParam: string, toParam: string) => {
+          const now = new Date();
+          const localDateStr = (dt: Date) =>
+            `${dt.getFullYear()}-${String(dt.getMonth() + 1).padStart(2, "0")}-${String(dt.getDate()).padStart(2, "0")}`;
+          let fromDate: string;
+          let toDate = localDateStr(now);
+          if (range === "today") fromDate = toDate;
+          else if (range === "7d") { const d = new Date(now); d.setDate(d.getDate() - 6); fromDate = localDateStr(d); }
+          else if (range === "30d") { const d = new Date(now); d.setDate(d.getDate() - 29); fromDate = localDateStr(d); }
+          else if (range === "custom" && fromParam) { fromDate = fromParam; if (toParam) toDate = toParam; }
+          else fromDate = toDate;
+          return { fromDate, toDate };
+        };
+
+        // Handle GET /api/pi/all-usage-range
+        if (method === "GET" && pathOnly === "/api/pi/all-usage-range") {
+          const parsedUrl = new URL(url, "http://localhost");
+          const range = parsedUrl.searchParams.get("range") || "today";
+          const fromParam = parsedUrl.searchParams.get("from") || "";
+          const toParam = parsedUrl.searchParams.get("to") || "";
+          const { fromDate, toDate } = resolveDateRange(range, fromParam, toParam);
+          const allRecords = pi.readAllCombinedUsage();
+          const usage = pi.getUsageByRange(allRecords, fromDate, toDate);
+          res.setHeader("Content-Type", "application/json");
+          return res.end(JSON.stringify(usage));
+        }
+
+        // Handle provider-filtered endpoints: /api/pi/{provider}-usage-range
+        const providerMatch = pathOnly.match(/^\/api\/pi\/(opencode|gemini|grok)-usage-range$/);
+        if (method === "GET" && providerMatch) {
+          const providerId = providerMatch[1]!;
+          const parsedUrl = new URL(url, "http://localhost");
+          const range = parsedUrl.searchParams.get("range") || "today";
+          const fromParam = parsedUrl.searchParams.get("from") || "";
+          const toParam = parsedUrl.searchParams.get("to") || "";
+          const { fromDate, toDate } = resolveDateRange(range, fromParam, toParam);
+          const allRecords = pi.filterByProvider(pi.readAllCombinedUsage(), providerId);
+          const usage = pi.getUsageByRange(allRecords, fromDate, toDate);
+          res.setHeader("Content-Type", "application/json");
+          return res.end(JSON.stringify(usage));
+        }
+
         const key = `${method} ${pathOnly}`;
         const handler = routes[key];
         if (handler) {
@@ -327,6 +467,7 @@ export default defineConfig({
     react(),
     tailwindcss(),
     piApiPlugin(),
+    chatApiPlugin(),
   ],
   resolve: {
     alias: {