@lyy0709/contextweaver 1.0.0

package/dist/server-DENFYPME.js

@@ -0,0 +1,263 @@
+import {
+  codebaseRetrievalSchema,
+  handleCodebaseRetrieval
+} from "./chunk-XVKMTPCT.js";
+import "./chunk-LPFRFKFW.js";
+import {
+  logger
+} from "./chunk-YVLGQTLG.js";
+import "./chunk-V2USKRIC.js";
+
+// src/mcp/server.ts
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+
+// src/mcp/tools/enhancePrompt.ts
+import { z } from "zod";
+var enhancePromptSchema = z.object({
+  prompt: z.string().describe("The original prompt to enhance"),
+  conversation_history: z.string().optional().describe(
+    "Recent conversation history for context. Format: 'User: ...\\nAssistant: ...'"
+  ),
+  project_root_path: z.string().optional().describe("Project root path for context")
+});
+function formatEnvMissingResponse(missingVars) {
+  const configPath = "~/.contextweaver/.env";
+  const text = `## \u26A0\uFE0F \u914D\u7F6E\u7F3A\u5931
+
+ContextWeaver \u7684 Prompt Enhancer \u9700\u8981\u914D\u7F6E\u5916\u90E8 LLM API\u3002
+
+### \u7F3A\u5931\u7684\u73AF\u5883\u53D8\u91CF
+${missingVars.map((v) => `- \`${v}\``).join("\n")}
+
+### \u914D\u7F6E\u6B65\u9AA4
+
+\u8BF7\u7F16\u8F91\u914D\u7F6E\u6587\u4EF6\uFF1A\`${configPath}\`\uFF0C\u6DFB\u52A0\u6216\u53D6\u6D88\u6CE8\u91CA\u4EE5\u4E0B\u914D\u7F6E\uFF1A
+
+\`\`\`bash
+# Prompt Enhancer \u914D\u7F6E\uFF08\u4F7F\u7528 enhance-prompt \u5DE5\u5177\u65F6\u9700\u8981\uFF09
+# PROMPT_ENHANCER_ENDPOINT=openai
+# PROMPT_ENHANCER_BASE_URL=
+PROMPT_ENHANCER_TOKEN=your-api-key-here
+# PROMPT_ENHANCER_MODEL=
+# PROMPT_ENHANCER_TEMPLATE=
+\`\`\`
+
+\u4FDD\u5B58\u6587\u4EF6\u540E\u91CD\u65B0\u8C03\u7528\u6B64\u5DE5\u5177\u5373\u53EF\u3002
+`;
+  return {
+    content: [
+      {
+        type: "text",
+        text
+      }
+    ]
+  };
+}
+async function handleEnhancePrompt(args) {
+  logger.info({ hasHistory: Boolean(args.conversation_history) }, "MCP enhance-prompt \u8C03\u7528\u5F00\u59CB");
+  const { checkEnhancerEnv } = await import("./config-WTC56Y2R.js");
+  const envCheck = checkEnhancerEnv();
+  if (!envCheck.isValid) {
+    logger.warn({ missingVars: envCheck.missingVars }, "Prompt Enhancer \u73AF\u5883\u53D8\u91CF\u672A\u914D\u7F6E");
+    return formatEnvMissingResponse(envCheck.missingVars);
+  }
+  try {
+    const { enhancePrompt } = await import("./enhancer-QHNMR35J.js");
+    const result = await enhancePrompt({
+      prompt: args.prompt,
+      conversationHistory: args.conversation_history,
+      projectRootPath: args.project_root_path
+    });
+    return {
+      content: [
+        {
+          type: "text",
+          text: result.enhanced
+        }
+      ]
+    };
+  } catch (err) {
+    const error = err;
+    logger.error({ error: error.message, stack: error.stack }, "MCP enhance-prompt \u8C03\u7528\u5931\u8D25");
+    return {
+      content: [
+        {
+          type: "text",
+          text: `Error: ${error.message || "unknown error"}`
+        }
+      ],
+      isError: true
+    };
+  }
+}
+
+// src/mcp/server.ts
+var SERVER_NAME = "contextweaver";
+var TOOLS = [
+  {
+    name: "codebase-retrieval",
+    description: `
+IMPORTANT: This is the PRIMARY tool for searching the codebase. 
+It uses a hybrid engine (Semantic + Exact Match) to find relevant code.
+Think of it as the "Google Search" for this repository.
+
+Capabilities:
+1. Semantic Search: Understands "what code does" (e.g., "auth logic") via high-dimensional embeddings.
+2. Exact Match: Filters by precise symbols (e.g., class names) via FTS (Full Text Search).
+3. Zen Context: Returns code with localized context (breadcrumbs) to avoid token overflow.
+
+<RULES>
+# 1. Tool Selection (When to use)
+- ALWAYS use this tool FIRST for any code exploration or understanding task.
+- DO NOT try to guess file paths. If you don't have the exact path, use this tool.
+- DO NOT use 'grep' or 'find' for semantic understanding. Only use them for exhaustive text matching (e.g. "Find ALL occurrences of string 'foo'").
+
+# 2. Before Editing (Critical)
+- Before creating a plan or editing any file, YOU MUST call this tool to gather context.
+- Ask for ALL symbols involved in the edit (classes, functions, types, constants).
+- Do not assume you remember the code structure. Verify it with this tool.
+
+# 3. Query Strategy (How to use)
+- Split your intent:
+  - Put the "Goal/Context" in 'information_request'.
+  - Put "Known Class/Func Names" in 'technical_terms'.
+- If the first search is too broad, add more specific 'technical_terms'.
+</RULES>
+
+Examples of GOOD queries:
+* [Goal: Understand Auth] 
+  information_request: "How is user authentication flow handled?"
+* [Goal: Fix DB Pool bug] 
+  information_request: "Logic for database connection pooling and error handling" 
+  technical_terms: ["PoolConfig", "Connection", "release"]
+
+Examples of BAD queries:
+* "Show me src/main.ts" (Use 'read_file' instead)
+* "Find definition of constructor of class Foo" (Use this tool, but put "Foo" in technical_terms)
+* "Find all references to function bar across the whole project" (Use 'grep' tool for exhaustive reference counting)
+`,
+    inputSchema: {
+      type: "object",
+      properties: {
+        repo_path: {
+          type: "string",
+          description: "The absolute file system path to the repository root."
+        },
+        information_request: {
+          type: "string",
+          description: "The SEMANTIC GOAL. Describe the functionality, logic, or behavior you are looking for in full natural language sentences. Focus on 'how it works' rather than exact names. (e.g., 'Trace the execution flow of the login process')"
+        },
+        technical_terms: {
+          type: "array",
+          items: { type: "string" },
+          description: "HARD FILTERS. An optional list of EXACT, KNOWN identifiers (class/function names, constants) that MUST appear in the code. Only use terms you are 100% sure exist. Leave empty if exploring."
+        }
+      },
+      required: ["repo_path", "information_request"]
+    }
+  },
+  {
+    name: "enhance-prompt",
+    description: `
+Enhance a user prompt by calling an external LLM API.
+
+Use this tool when you want to turn a rough requirement into a clearer, more specific, and actionable prompt.
+
+Features:
+- Customizable template via env var
+- Automatic language detection (Chinese input -> Chinese output)
+- Optional conversation history support
+`,
+    inputSchema: {
+      type: "object",
+      properties: {
+        prompt: {
+          type: "string",
+          description: "The original prompt to enhance"
+        },
+        conversation_history: {
+          type: "string",
+          description: "Recent conversation history for context. Format: 'User: ...\\nAssistant: ...'"
+        },
+        project_root_path: {
+          type: "string",
+          description: "Project root path for context"
+        }
+      },
+      required: ["prompt"]
+    }
+  }
+];
+async function startMcpServer() {
+  logger.info({ name: SERVER_NAME }, "\u542F\u52A8 MCP \u670D\u52A1\u5668");
+  const server = new Server(
+    {
+      name: SERVER_NAME,
+      version: "1.0.0"
+    },
+    {
+      capabilities: {
+        tools: {}
+      }
+    }
+  );
+  server.setRequestHandler(ListToolsRequestSchema, async () => {
+    logger.debug("\u6536\u5230 list_tools \u8BF7\u6C42");
+    return { tools: TOOLS };
+  });
+  server.setRequestHandler(CallToolRequestSchema, async (request, extra) => {
+    const { name, arguments: args } = request.params;
+    logger.info({ tool: name }, "\u6536\u5230 call_tool \u8BF7\u6C42");
+    const rawToken = extra._meta?.progressToken;
+    const progressToken = typeof rawToken === "string" || typeof rawToken === "number" ? rawToken : void 0;
+    const onProgress = progressToken ? async (current, total, message) => {
+      try {
+        await extra.sendNotification({
+          method: "notifications/progress",
+          params: {
+            progressToken,
+            progress: current,
+            total,
+            message
+          }
+        });
+      } catch (err) {
+        logger.debug({ error: err.message }, "\u53D1\u9001\u8FDB\u5EA6\u901A\u77E5\u5931\u8D25");
+      }
+    } : void 0;
+    try {
+      switch (name) {
+        case "codebase-retrieval": {
+          const parsed = codebaseRetrievalSchema.parse(args);
+          return await handleCodebaseRetrieval(parsed, void 0, onProgress);
+        }
+        case "enhance-prompt": {
+          const parsed = enhancePromptSchema.parse(args);
+          return await handleEnhancePrompt(parsed);
+        }
+        default:
+          throw new Error(`Unknown tool: ${name}`);
+      }
+    } catch (err) {
+      const error = err;
+      logger.error({ error: error.message, stack: error.stack, tool: name }, "\u5DE5\u5177\u8C03\u7528\u5931\u8D25");
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Error: ${error.message}`
+          }
+        ],
+        isError: true
+      };
+    }
+  });
+  const transport = new StdioServerTransport();
+  logger.info("MCP \u670D\u52A1\u5668\u5DF2\u542F\u52A8\uFF0C\u7B49\u5F85\u8FDE\u63A5...");
+  await server.connect(transport);
+}
+export {
+  startMcpServer
+};

package/package.json

@@ -0,0 +1,82 @@
+{
+  "name": "@lyy0709/contextweaver",
+  "version": "1.0.0",
+  "description": "A context weaving tool for LLMs — with Prompt Enhancer",
+  "license": "MIT",
+  "author": "lyy0709",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/lyy0709/ContextWeaver.git"
+  },
+  "homepage": "https://github.com/lyy0709/ContextWeaver#readme",
+  "private": false,
+  "type": "module",
+  "bin": {
+    "contextweaver": "dist/index.js",
+    "cw": "dist/index.js"
+  },
+  "files": [
+    "dist/**/*.js",
+    "dist/**/*.d.ts"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "scripts": {
+    "build": "tsup src/index.ts src/mcp/main.ts --format esm --dts --out-dir dist --sourcemap --clean",
+    "build:release": "tsup src/index.ts src/mcp/main.ts --format esm --dts --out-dir dist --clean",
+    "dev": "tsup src/index.ts src/mcp/main.ts --format esm --dts --out-dir dist --sourcemap --watch",
+    "fmt": "biome check --write ./src",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  },
+  "dependencies": {
+    "@keqingmoe/tree-sitter": "^0.26.2",
+    "@lancedb/lancedb": "^0.22.0",
+    "@modelcontextprotocol/sdk": "^1.25.1",
+    "better-sqlite3": "^12.2.0",
+    "cac": "^6.7.14",
+    "chardet": "^2.1.1",
+    "dotenv": "^17.2.3",
+    "fdir": "^6.5.0",
+    "iconv-lite": "^0.7.1",
+    "ignore": "^7.0.5",
+    "p-limit": "^7.2.0",
+    "pino": "^10.1.0",
+    "tree-sitter-c": "^0.24.1",
+    "tree-sitter-c-sharp": "^0.23.1",
+    "tree-sitter-cpp": "^0.23.4",
+    "tree-sitter-go": "0.25.0",
+    "tree-sitter-java": "0.23.5",
+    "tree-sitter-javascript": "0.25.0",
+    "tree-sitter-python": "0.25.0",
+    "tree-sitter-rust": "0.24.0",
+    "tree-sitter-typescript": "0.23.2",
+    "zod": "^4.2.1"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "2.3.10",
+    "@types/better-sqlite3": "^7.6.13",
+    "@types/node": "^24.0.0",
+    "knip": "^5.78.0",
+    "tsup": "^8.5.1",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18"
+  },
+  "pnpm": {
+    "onlyBuiltDependencies": [
+      "@keqingmoe/tree-sitter",
+      "better-sqlite3",
+      "esbuild",
+      "tree-sitter-c",
+      "tree-sitter-c-sharp",
+      "tree-sitter-cpp",
+      "tree-sitter-go",
+      "tree-sitter-java",
+      "tree-sitter-javascript",
+      "tree-sitter-python",
+      "tree-sitter-rust",
+      "tree-sitter-typescript"
+    ]
+  }
+}