Merge pull request #122 from hqwlkj/refactor/tidy-up

refactor(ui): 重构代码结构，调整文件路径和导入引用

Author: qorzj

src/cli.tsx

@@ -1,7 +1,7 @@
 import React from "react";
 import { render } from "ink";
 import { setShellIfWindows } from "./common/shell-utils";
-import { checkForNpmUpdate, promptForPendingUpdate, type PackageInfo } from "./updateCheck";
+import { checkForNpmUpdate, promptForPendingUpdate, type PackageInfo } from "./common/update-check";
 import { AppContainer } from "./ui";
 
 const args = process.argv.slice(2);

src/common/openai-client.ts

@@ -3,7 +3,7 @@ import * as os from "os";
 import * as path from "path";
 import OpenAI from "openai";
 import { Agent, fetch as undiciFetch } from "undici";
-import { resolveCurrentSettings } from "../ui/App";
+import { resolveCurrentSettings } from "../settings";
 
 // Custom undici Agent with a 180-second keepAlive timeout.  The default
 // global fetch (undici) only keeps connections alive for 4 seconds, which

src/updateCheck.ts src/common/update-check.tssrc/updateCheck.ts renamed to src/common/update-check.ts

@@ -5,8 +5,8 @@ import * as os from "os";
 import * as path from "path";
 import { render, type Instance } from "ink";
 import chalk from "chalk";
-import { UpdatePrompt, type UpdatePromptChoice } from "./ui";
-import { killProcessTree } from "./common/process-tree";
+import { UpdatePrompt, type UpdatePromptChoice } from "../ui";
+import { killProcessTree } from "./process-tree";
 
 export type PackageInfo = {
   name: string;

src/common/runtime.ts src/common/validate.tssrc/common/runtime.ts renamed to src/common/validate.ts

@@ -2,8 +2,8 @@ import { execFileSync, execSync } from "child_process";
 import * as fs from "fs";
 import * as os from "os";
 import * as path from "path";
-import { fileURLToPath } from "url";
 import ejs from "ejs";
+import { fileURLToPath } from "url";
 import type { SessionMessage } from "./session";
 import { findGitBashPath, resolveShellPath } from "./common/shell-utils";
 import { supportsMultimodal } from "./common/model-capabilities";
@@ -166,8 +166,7 @@ function getCurrentDateAndModelPrompt(model?: string): string {
 
 export function getSystemPrompt(_projectRoot: string, options: PromptToolOptions = {}): string {
   const toolDocs = readToolDocs(getExtensionRoot(), options);
-  const basePrompt = toolDocs ? `${SYSTEM_PROMPT_BASE}\n\n# Available Tools\n\n${toolDocs}` : SYSTEM_PROMPT_BASE;
-  return basePrompt;
+  return toolDocs ? `${SYSTEM_PROMPT_BASE}\n\n# Available Tools\n\n${toolDocs}` : SYSTEM_PROMPT_BASE;
 }
 
 export function getCompactPrompt(sessionMessages: SessionMessage[]): string {
@@ -287,7 +286,7 @@ function getUnameInfo(): string {
   }
 }
 
-function getExtensionRoot(): string {
+export function getExtensionRoot(): string {
   // Prefer `__dirname` which is always available in the CJS bundle output.
   // Fall back to `import.meta.url` for ESM test environments (tsx --test).
   if (typeof __dirname !== "undefined") {

src/prompt.ts

@@ -2,7 +2,6 @@ import * as fs from "fs";
 import * as path from "path";
 import * as os from "os";
 import * as crypto from "crypto";
-import { fileURLToPath } from "url";
 import matter from "gray-matter";
 import ejs from "ejs";
 import type { ChatCompletionMessageParam, ChatCompletionContentPart } from "openai/resources/chat/completions";
@@ -12,6 +11,7 @@ import { DEEPSEEK_V4_MODELS, supportsMultimodal } from "./common/model-capabilit
 import {
   getCompactPrompt,
   getDefaultSkillPrompt,
+  getExtensionRoot,
   getRuntimeContext,
   getSystemPrompt,
   getTools,
@@ -135,15 +135,6 @@ function accumulateUsagePerModel(
   return usagePerModel;
 }
 
-function getExtensionRoot(): string {
-  if (typeof __dirname !== "undefined") {
-    return path.resolve(__dirname, "..");
-  }
-
-  const currentFilePath = fileURLToPath(import.meta.url);
-  return path.resolve(path.dirname(currentFilePath), "..");
-}
-
 function getTotalTokens(usage: ModelUsage | null | undefined): number {
   if (!isUsageRecord(usage)) {
     return 0;

src/session.ts

@@ -1,4 +1,7 @@
 import { defaultsToThinkingMode } from "./common/model-capabilities";
+import * as fs from "fs";
+import * as os from "os";
+import * as path from "path";
 
 export type DeepcodingEnv = Record<string, string | undefined> & {
   MODEL?: string;
@@ -370,3 +373,88 @@ export function applyModelConfigSelection(
 
   return { settings: next, changed: true };
 }
+
+// ---------------------------------------------------------------------------
+// Default constants
+// ---------------------------------------------------------------------------
+
+export const DEFAULT_MODEL = "deepseek-v4-pro";
+export const DEFAULT_BASE_URL = "https://api.deepseek.com";
+
+// ---------------------------------------------------------------------------
+// Settings file I/O
+// ---------------------------------------------------------------------------
+
+export function getUserSettingsPath(): string {
+  return path.join(os.homedir(), ".deepcode", "settings.json");
+}
+
+export function getProjectSettingsPath(projectRoot: string): string {
+  return path.join(projectRoot, ".deepcode", "settings.json");
+}
+
+export function readSettingsFile(settingsPath: string): DeepcodingSettings | null {
+  try {
+    if (!fs.existsSync(settingsPath)) {
+      return null;
+    }
+    const raw = fs.readFileSync(settingsPath, "utf8");
+    return JSON.parse(raw) as DeepcodingSettings;
+  } catch {
+    return null;
+  }
+}
+
+export function readSettings(): DeepcodingSettings | null {
+  return readSettingsFile(getUserSettingsPath());
+}
+
+export function readProjectSettings(projectRoot: string = process.cwd()): DeepcodingSettings | null {
+  return readSettingsFile(getProjectSettingsPath(projectRoot));
+}
+
+function writeSettingsFile(settingsPath: string, settings: DeepcodingSettings): void {
+  fs.mkdirSync(path.dirname(settingsPath), { recursive: true });
+  fs.writeFileSync(settingsPath, `${JSON.stringify(settings, null, 2)}\n`, "utf8");
+}
+
+export function writeSettings(settings: DeepcodingSettings): void {
+  const settingsPath = getUserSettingsPath();
+  writeSettingsFile(settingsPath, settings);
+}
+
+export function writeProjectSettings(settings: DeepcodingSettings, projectRoot: string = process.cwd()): void {
+  const settingsPath = getProjectSettingsPath(projectRoot);
+  writeSettingsFile(settingsPath, settings);
+}
+
+export function writeModelConfigSelection(
+  selection: ModelConfigSelection,
+  current: ModelConfigSelection = resolveCurrentSettings(),
+  projectRoot: string = process.cwd()
+): { changed: boolean; settings: DeepcodingSettings } {
+  const projectSettingsPath = getProjectSettingsPath(projectRoot);
+  const shouldWriteProjectSettings = fs.existsSync(projectSettingsPath);
+  const rawSettings = shouldWriteProjectSettings ? readProjectSettings(projectRoot) : readSettings();
+  const result = applyModelConfigSelection(rawSettings, current, selection);
+  if (result.changed) {
+    if (shouldWriteProjectSettings) {
+      writeProjectSettings(result.settings, projectRoot);
+    } else {
+      writeSettings(result.settings);
+    }
+  }
+  return result;
+}
+
+export function resolveCurrentSettings(projectRoot: string = process.cwd()): ResolvedDeepcodingSettings {
+  return resolveSettingsSources(
+    readSettings(),
+    readProjectSettings(projectRoot),
+    {
+      model: DEFAULT_MODEL,
+      baseURL: DEFAULT_BASE_URL,
+    },
+    process.env
+  );
+}

src/settings.ts

@@ -5,7 +5,7 @@ import * as os from "os";
 import * as path from "path";
 
 // eslint-disable-next-line @typescript-eslint/consistent-type-imports
-type ClipboardModule = typeof import("../ui/clipboard");
+type ClipboardModule = typeof import("../ui/core/clipboard");
 
 const ORIGINAL_PATH = process.env.PATH;
 const ORIGINAL_PLATFORM = process.platform;
@@ -30,7 +30,7 @@ function withPlatform<T>(platform: NodeJS.Platform, fn: () => T): T {
 
 test("readClipboardImage returns null when no clipboard helpers are installed", async () => {
   // Reload module so it picks up the patched PATH at spawn time.
-  const moduleUrl = new URL(`../ui/clipboard.ts?t=${Date.now()}`, import.meta.url).href;
+  const moduleUrl = new URL(`../ui/core/clipboard.ts?t=${Date.now()}`, import.meta.url).href;
   const { readClipboardImage } = (await import(moduleUrl)) as ClipboardModule;
   const result = withCleanPath(() => readClipboardImage());
   assert.equal(result, null);
@@ -63,7 +63,7 @@ test(
         { mode: 0o755 }
       );
 
-      const moduleUrl = new URL(`../ui/clipboard.ts?t=${Date.now()}`, import.meta.url).href;
+      const moduleUrl = new URL(`../ui/core/clipboard.ts?t=${Date.now()}`, import.meta.url).href;
       const { readClipboardImage } = (await import(moduleUrl)) as ClipboardModule;
 
       process.env.PATH = binDir;

src/tests/askUserQuestion.test.ts src/tests/ask-user-question.test.tssrc/tests/askUserQuestion.test.ts renamed to src/tests/ask-user-question.test.ts

@@ -1,6 +1,6 @@
 import { test } from "node:test";
 import assert from "node:assert/strict";
-import { calculateVisibleStart } from "../ui/DropdownMenu";
+import { calculateVisibleStart } from "../ui/components/DropdownMenu";
 
 test("calculateVisibleStart centers active item when possible", () => {
   // 10 items, max 5 visible, active index 4 (middle)