@virsanghavi/axis-server 1.4.0 → 1.6.0

package/dist/mcp-server.mjs

@@ -1166,6 +1166,7 @@ import fs4 from "fs";
 
 // ../../src/local/local-search.ts
 import fs3 from "fs/promises";
+import fsSync2 from "fs";
 import path3 from "path";
 var SKIP_DIRS = /* @__PURE__ */ new Set([
   "node_modules",
@@ -1321,34 +1322,61 @@ var STOP_WORDS = /* @__PURE__ */ new Set([
   "most",
   "some",
   "any",
-  "find",
-  "show",
-  "get",
-  "look",
-  "search",
-  "locate",
-  "check",
-  "file",
-  "files",
-  "code",
-  "function",
-  "class",
-  "method",
   "there",
   "here",
   "just",
   "also",
   "very",
   "really",
-  "quite"
+  "quite",
+  "show",
+  "look",
+  "locate",
+  "using",
+  "used",
+  "need",
+  "want"
 ]);
 var MAX_FILE_SIZE = 256 * 1024;
 var MAX_RESULTS = 20;
 var CONTEXT_LINES = 2;
 var MAX_MATCHES_PER_FILE = 6;
 function extractKeywords(query) {
-  const raw = query.toLowerCase().replace(/[^\w\s\-_.]/g, " ").split(/\s+/).filter((w) => w.length >= 2 && !STOP_WORDS.has(w));
-  return [...new Set(raw)];
+  const words = query.toLowerCase().replace(/[^\w\s\-_.]/g, " ").split(/\s+/).filter((w) => w.length >= 2);
+  const filtered = words.filter((w) => !STOP_WORDS.has(w));
+  const result = filtered.length > 0 ? filtered : words.filter((w) => w.length >= 3);
+  return [...new Set(result)];
+}
+var PROJECT_ROOT_MARKERS = [
+  ".git",
+  ".axis",
+  "package.json",
+  "Cargo.toml",
+  "go.mod",
+  "pyproject.toml",
+  "setup.py",
+  "Gemfile",
+  "pom.xml",
+  "tsconfig.json",
+  ".cursorrules",
+  "AGENTS.md"
+];
+function detectProjectRoot(startDir) {
+  let current = startDir;
+  const root = path3.parse(current).root;
+  while (current !== root) {
+    for (const marker of PROJECT_ROOT_MARKERS) {
+      try {
+        fsSync2.accessSync(path3.join(current, marker));
+        return current;
+      } catch {
+      }
+    }
+    const parent = path3.dirname(current);
+    if (parent === current) break;
+    current = parent;
+  }
+  return startDir;
 }
 async function walkDir(dir, maxDepth = 12) {
   const results = [];
@@ -1428,8 +1456,12 @@ async function searchFile(filePath, rootDir, keywords) {
   return { filePath, relativePath, score, matchedKeywords, regions };
 }
 async function localSearch(query, rootDir) {
-  const cwd = rootDir || process.cwd();
+  const rawCwd = rootDir || process.cwd();
+  const cwd = detectProjectRoot(rawCwd);
   const keywords = extractKeywords(query);
+  if (cwd !== rawCwd) {
+    logger.info(`[localSearch] Detected project root: ${cwd} (CWD was: ${rawCwd})`);
+  }
   if (keywords.length === 0) {
     return "Could not extract meaningful search terms from the query. Try being more specific (e.g. 'authentication middleware' instead of 'how does it work').";
   }
@@ -1545,6 +1577,120 @@ var nerveCenter = new NerveCenter(manager, {
   projectName: process.env.PROJECT_NAME || "default"
 });
 logger.info("=== Axis MCP Server Initialized ===");
+var RECHECK_INTERVAL_MS = 30 * 60 * 1e3;
+var GRACE_PERIOD_MS = 5 * 60 * 1e3;
+var subscription = {
+  checked: false,
+  valid: true,
+  // Assume valid until proven otherwise (for startup)
+  plan: "unknown",
+  reason: "",
+  checkedAt: 0
+};
+async function verifySubscription() {
+  if (!apiSecret) {
+    const hasDirectSupabase = !useRemoteApiOnly && !!process.env.NEXT_PUBLIC_SUPABASE_URL && !!process.env.SUPABASE_SERVICE_ROLE_KEY;
+    if (hasDirectSupabase) {
+      subscription = { checked: true, valid: true, plan: "developer", reason: "Direct Supabase mode \u2014 no API key needed", checkedAt: Date.now() };
+      logger.info("[subscription] Direct Supabase credentials found \u2014 developer mode, skipping verification");
+      return subscription;
+    }
+    subscription = {
+      checked: true,
+      valid: false,
+      plan: "none",
+      reason: "no_api_key",
+      checkedAt: Date.now()
+    };
+    logger.error("[subscription] No API key configured. Axis requires an API key from https://useaxis.dev/dashboard");
+    return subscription;
+  }
+  const verifyUrl = apiUrl.endsWith("/v1") ? `${apiUrl}/verify` : `${apiUrl}/v1/verify`;
+  logger.info(`[subscription] Verifying subscription at ${verifyUrl}`);
+  const controller = new AbortController();
+  const timeout = setTimeout(() => controller.abort(), 1e4);
+  try {
+    const response = await fetch(verifyUrl, {
+      method: "GET",
+      headers: {
+        "Authorization": `Bearer ${apiSecret}`
+      },
+      signal: controller.signal
+    });
+    clearTimeout(timeout);
+    const data = await response.json();
+    logger.info(`[subscription] Verify response: ${JSON.stringify(data)}`);
+    if (data.valid === true) {
+      subscription = {
+        checked: true,
+        valid: true,
+        plan: data.plan || "Pro",
+        reason: "",
+        checkedAt: Date.now(),
+        validUntil: data.validUntil
+      };
+    } else {
+      subscription = {
+        checked: true,
+        valid: false,
+        plan: data.plan || "Free",
+        reason: data.reason || "subscription_invalid",
+        checkedAt: Date.now()
+      };
+      logger.warn(`[subscription] Subscription NOT valid: ${data.reason}`);
+    }
+  } catch (e) {
+    clearTimeout(timeout);
+    logger.warn(`[subscription] Verification failed (network): ${e.message}`);
+    if (!subscription.checked) {
+      subscription = {
+        checked: true,
+        valid: true,
+        // Grace period
+        plan: "unverified",
+        reason: "Verification endpoint unreachable \u2014 grace period active",
+        checkedAt: Date.now()
+      };
+      logger.warn("[subscription] First check failed \u2014 allowing grace period");
+    }
+  }
+  return subscription;
+}
+function isSubscriptionStale() {
+  return Date.now() - subscription.checkedAt > RECHECK_INTERVAL_MS;
+}
+function getSubscriptionBlockMessage() {
+  if (subscription.reason === "no_api_key") {
+    return [
+      "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550",
+      "  Axis API key required",
+      "",
+      "  No API key found. Axis requires an active subscription",
+      "  and a valid API key to operate.",
+      "",
+      "  1. Sign up or log in at https://useaxis.dev",
+      "  2. Subscribe to Axis Pro",
+      "  3. Generate an API key from the dashboard",
+      "  4. Add AXIS_API_KEY to your mcp.json configuration",
+      "  5. Restart your IDE",
+      "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550"
+    ].join("\n");
+  }
+  return [
+    "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550",
+    "  Axis Pro subscription required",
+    "",
+    `  Status: ${subscription.reason || "subscription_expired"}`,
+    `  Current plan: ${subscription.plan}`,
+    "",
+    "  Your Axis Pro subscription has expired or is inactive.",
+    "  All Axis MCP tools are disabled until the subscription is renewed.",
+    "",
+    "  \u2192 Renew at https://useaxis.dev/dashboard",
+    "  \u2192 After renewing, restart your IDE to re-verify.",
+    "\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550"
+  ].join("\n");
+}
 var ragEngine;
 if (!useRemoteApiOnly && process.env.NEXT_PUBLIC_SUPABASE_URL && process.env.SUPABASE_SERVICE_ROLE_KEY) {
   ragEngine = new RagEngine(
@@ -1558,7 +1704,7 @@ async function ensureFileSystem() {
   try {
     const fs5 = await import("fs/promises");
     const path5 = await import("path");
-    const fsSync2 = await import("fs");
+    const fsSync3 = await import("fs");
     const cwd = process.cwd();
     logger.info(`Server CWD: ${cwd}`);
     const historyDir = path5.join(cwd, "history");
@@ -1567,7 +1713,7 @@ async function ensureFileSystem() {
     const axisDir = path5.join(cwd, ".axis");
     const axisInstructions = path5.join(axisDir, "instructions");
     const legacyInstructions = path5.join(cwd, "agent-instructions");
-    if (fsSync2.existsSync(legacyInstructions) && !fsSync2.existsSync(axisDir)) {
+    if (fsSync3.existsSync(legacyInstructions) && !fsSync3.existsSync(axisDir)) {
       logger.info("Using legacy agent-instructions directory");
     } else {
       await fs5.mkdir(axisInstructions, { recursive: true }).catch(() => {
@@ -1894,6 +2040,16 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
   const { name, arguments: args } = request.params;
   logger.info("Tool call", { name });
+  if (isSubscriptionStale()) {
+    await verifySubscription();
+  }
+  if (!subscription.valid) {
+    logger.warn(`[subscription] Blocking tool call "${name}" \u2014 subscription invalid`);
+    return {
+      content: [{ type: "text", text: getSubscriptionBlockMessage() }],
+      isError: true
+    };
+  }
   if (name === READ_CONTEXT_TOOL) {
     const filename = String(args?.filename);
     try {
@@ -1940,41 +2096,51 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
   }
   if (name === SEARCH_CONTEXT_TOOL) {
     const query = String(args?.query);
-    let ragResults = null;
-    try {
-      const remote = await manager.searchContext(query, nerveCenter.currentProjectName);
-      if (remote && !remote.includes("No results found") && remote.trim().length > 20) {
-        ragResults = remote;
-      }
-    } catch {
-    }
-    if (!ragResults && ragEngine) {
-      try {
-        const localRag = await ragEngine.search(query);
-        if (localRag.length > 0) {
-          ragResults = localRag.join("\n---\n");
-        }
-      } catch {
-      }
-    }
-    let localResults = null;
+    logger.info(`[search_codebase] Query: "${query}"`);
+    let localResults = "";
     try {
       localResults = await localSearch(query);
+      logger.info(`[search_codebase] Local search completed: ${localResults.length} chars`);
     } catch (e) {
       logger.warn(`[search_codebase] Local search error: ${e}`);
+      localResults = "";
     }
-    const parts = [];
-    if (ragResults) {
-      parts.push("## Indexed Results (RAG)\n\n" + ragResults);
-    }
-    if (localResults && !localResults.startsWith("No matches found") && !localResults.startsWith("Could not extract")) {
-      parts.push("## Local Codebase Search\n\n" + localResults);
-    } else if (!ragResults) {
-      return { content: [{ type: "text", text: localResults || "No results found." }] };
+    let ragResults = null;
+    const RAG_TIMEOUT_MS = 3e3;
+    try {
+      const ragPromise = (async () => {
+        try {
+          const remote = await manager.searchContext(query, nerveCenter.currentProjectName);
+          if (remote && !remote.includes("No results found") && remote.trim().length > 20) {
+            return remote;
+          }
+        } catch {
+        }
+        if (ragEngine) {
+          try {
+            const results = await ragEngine.search(query);
+            if (results.length > 0) return results.join("\n---\n");
+          } catch {
+          }
+        }
+        return null;
+      })();
+      ragResults = await Promise.race([
+        ragPromise,
+        new Promise((resolve) => setTimeout(() => resolve(null), RAG_TIMEOUT_MS))
+      ]);
+      if (ragResults) {
+        logger.info(`[search_codebase] RAG returned results (${ragResults.length} chars)`);
+      }
+    } catch {
     }
-    if (parts.length === 0) {
-      return { content: [{ type: "text", text: "No results found for this query." }] };
+    const hasLocal = localResults && !localResults.startsWith("No matches found") && !localResults.startsWith("Could not extract");
+    if (!hasLocal && !ragResults) {
+      return { content: [{ type: "text", text: localResults || "No results found for this query." }] };
     }
+    const parts = [];
+    if (hasLocal) parts.push(localResults);
+    if (ragResults) parts.push("## Indexed Results (RAG)\n\n" + ragResults);
     return { content: [{ type: "text", text: parts.join("\n\n---\n\n") }] };
   }
   if (name === "get_subscription_status") {
@@ -2069,6 +2235,21 @@ async function main() {
     ragEngine.setProjectId(nerveCenter.projectId);
     logger.info(`Local RAG Engine linked to Project ID: ${nerveCenter.projectId}`);
   }
+  await verifySubscription();
+  if (!subscription.valid) {
+    logger.error("[subscription] Subscription invalid at startup \u2014 all tools will be blocked");
+    logger.error(`[subscription] Reason: ${subscription.reason} | Plan: ${subscription.plan}`);
+  } else {
+    logger.info(`[subscription] Subscription verified: ${subscription.plan} (valid until: ${subscription.validUntil || "N/A"})`);
+  }
+  setInterval(async () => {
+    try {
+      await verifySubscription();
+      logger.info(`[subscription] Periodic re-check: valid=${subscription.valid}, plan=${subscription.plan}`);
+    } catch (e) {
+      logger.warn(`[subscription] Periodic re-check failed: ${e}`);
+    }
+  }, RECHECK_INTERVAL_MS);
   logger.info("MCP server ready - all tools and resources registered");
   const transport = new StdioServerTransport();
   await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@virsanghavi/axis-server",
-    "version": "1.4.0",
+    "version": "1.6.0",
     "description": "Axis MCP Server CLI",
     "main": "dist/index.js",
     "bin": {