@virsanghavi/axis-server 1.6.0 → 1.7.1

package/dist/mcp-server.mjs

@@ -1035,12 +1035,13 @@ ${conventions}`;
   }
   // --- Billing & Usage ---
   async getSubscriptionStatus(email) {
-    logger.info(`[getSubscriptionStatus] Starting - email: ${email}`);
+    logger.info(`[getSubscriptionStatus] Starting - email: ${email || "(API key identity)"}`);
     logger.info(`[getSubscriptionStatus] Config - apiUrl: ${this.contextManager.apiUrl}, apiSecret: ${this.contextManager.apiSecret ? "SET" : "NOT SET"}, useSupabase: ${this.useSupabase}`);
     if (this.contextManager.apiUrl) {
       try {
-        logger.info(`[getSubscriptionStatus] Attempting API call to: usage?email=${encodeURIComponent(email)}`);
-        const result = await this.callCoordination(`usage?email=${encodeURIComponent(email)}`);
+        const endpoint = email ? `usage?email=${encodeURIComponent(email)}` : "usage";
+        logger.info(`[getSubscriptionStatus] Attempting API call to: ${endpoint}`);
+        const result = await this.callCoordination(endpoint);
         logger.info(`[getSubscriptionStatus] API call successful: ${JSON.stringify(result).substring(0, 200)}`);
         return result;
       } catch (e) {
@@ -1050,8 +1051,8 @@ ${conventions}`;
     } else {
       logger.warn("[getSubscriptionStatus] No API URL configured");
     }
-    if (this.useSupabase && this.supabase) {
-      const { data: profile, error } = await this.supabase.from("profiles").select("subscription_status, stripe_customer_id, current_period_end").eq("email", email).single();
+    if (this.useSupabase && this.supabase && email) {
+      const { data: profile, error } = await this.supabase.from("profiles").select("subscription_status, stripe_customer_id, current_period_end").ilike("email", email).single();
       if (error || !profile) {
         return { status: "unknown", message: "Profile not found." };
       }
@@ -1066,21 +1067,22 @@ ${conventions}`;
     return { error: "Coordination not configured. API URL not set and Supabase not available." };
   }
   async getUsageStats(email) {
-    logger.info(`[getUsageStats] Starting - email: ${email}`);
+    logger.info(`[getUsageStats] Starting - email: ${email || "(API key identity)"}`);
     logger.info(`[getUsageStats] Config - apiUrl: ${this.contextManager.apiUrl}, apiSecret: ${this.contextManager.apiSecret ? "SET" : "NOT SET"}, useSupabase: ${this.useSupabase}`);
     if (this.contextManager.apiUrl) {
       try {
-        logger.info(`[getUsageStats] Attempting API call to: usage?email=${encodeURIComponent(email)}`);
-        const result = await this.callCoordination(`usage?email=${encodeURIComponent(email)}`);
+        const endpoint = email ? `usage?email=${encodeURIComponent(email)}` : "usage";
+        logger.info(`[getUsageStats] Attempting API call to: ${endpoint}`);
+        const result = await this.callCoordination(endpoint);
         logger.info(`[getUsageStats] API call successful: ${JSON.stringify(result).substring(0, 200)}`);
-        return { email, usageCount: result.usageCount || 0 };
+        return { email: email || result.email, usageCount: result.usageCount || 0 };
       } catch (e) {
         logger.error(`[getUsageStats] API call failed: ${e.message}`, e);
         return { error: `API call failed: ${e.message}` };
       }
     }
-    if (this.useSupabase && this.supabase) {
-      const { data: profile } = await this.supabase.from("profiles").select("usage_count").eq("email", email).single();
+    if (this.useSupabase && this.supabase && email) {
+      const { data: profile } = await this.supabase.from("profiles").select("usage_count").ilike("email", email).single();
       return { email, usageCount: profile?.usage_count || 0 };
     }
     return { error: "Coordination not configured. API URL not set and Supabase not available." };
@@ -1424,11 +1426,18 @@ async function searchFile(filePath, rootDir, keywords) {
   const relativePath = path3.relative(rootDir, filePath);
   const matchedKeywords = keywords.filter((kw) => contentLower.includes(kw));
   if (matchedKeywords.length === 0) return null;
+  const coverage = matchedKeywords.length / keywords.length;
+  if (keywords.length >= 3 && coverage < 0.4) return null;
+  if (keywords.length === 2 && matchedKeywords.length < 1) return null;
   const lines = content.split("\n");
-  let score = matchedKeywords.length;
+  let score = coverage * coverage * matchedKeywords.length;
   const relLower = relativePath.toLowerCase();
+  let pathMatches = 0;
   for (const kw of keywords) {
-    if (relLower.includes(kw)) score += 2;
+    if (relLower.includes(kw)) {
+      score += 3;
+      pathMatches++;
+    }
   }
   const matchingLineIndices = [];
   for (let i = 0; i < lines.length; i++) {
@@ -1437,6 +1446,22 @@ async function searchFile(filePath, rootDir, keywords) {
       matchingLineIndices.push(i);
     }
   }
+  let proximityBonus = 0;
+  for (let i = 0; i < matchingLineIndices.length; i++) {
+    const windowStart = matchingLineIndices[i];
+    const windowEnd = windowStart + 10;
+    const keywordsInWindow = /* @__PURE__ */ new Set();
+    for (let j = i; j < matchingLineIndices.length && matchingLineIndices[j] <= windowEnd; j++) {
+      const lineLower = lines[matchingLineIndices[j]].toLowerCase();
+      for (const kw of matchedKeywords) {
+        if (lineLower.includes(kw)) keywordsInWindow.add(kw);
+      }
+    }
+    if (keywordsInWindow.size >= 2) {
+      proximityBonus = Math.max(proximityBonus, keywordsInWindow.size * 1.5);
+    }
+  }
+  score += proximityBonus;
   score += Math.min(matchingLineIndices.length, 20) * 0.1;
   const regions = [];
   let lastEnd = -1;
@@ -1888,24 +1913,22 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
     // --- Billing & Usage ---
     {
       name: "get_subscription_status",
-      description: "**BILLING CHECK**: specific to the Axis business logic.\n- Returns the user's subscription tier (Pro vs Free), Stripe customer ID, and current period end.\n- Critical for gating features behind paywalls.\n- Returns 'Profile not found' if the user doesn't exist in the database.",
+      description: "**BILLING CHECK**: Returns the user's subscription tier (Pro vs Free), Stripe customer ID, and current period end.\n- If no email is provided, returns the subscription status of the current API key owner.\n- Critical for gating features behind paywalls.",
       inputSchema: {
         type: "object",
         properties: {
-          email: { type: "string", description: "User email to check." }
-        },
-        required: ["email"]
+          email: { type: "string", description: "Optional. User email to check. If omitted, checks the subscription of the current API key owner." }
+        }
       }
     },
     {
       name: "get_usage_stats",
-      description: "**API USAGE**: Returns a user's token usage and request counts.\n- Useful for debugging rate limits or explaining quota usage to users.",
+      description: "**API USAGE**: Returns token usage and request counts.\n- If no email is provided, returns usage for the current API key owner.\n- Useful for debugging rate limits or explaining quota usage to users.",
       inputSchema: {
         type: "object",
         properties: {
-          email: { type: "string", description: "User email to check." }
-        },
-        required: ["email"]
+          email: { type: "string", description: "Optional. User email to check. If omitted, checks usage of the current API key owner." }
+        }
       }
     },
     {
@@ -2144,8 +2167,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     return { content: [{ type: "text", text: parts.join("\n\n---\n\n") }] };
   }
   if (name === "get_subscription_status") {
-    const email = String(args?.email);
-    logger.info(`[get_subscription_status] Called with email: ${email}`);
+    const email = args?.email ? String(args.email) : void 0;
+    logger.info(`[get_subscription_status] Called with email: ${email || "(using API key identity)"}`);
     try {
       const result = await nerveCenter.getSubscriptionStatus(email);
       logger.info(`[get_subscription_status] Result: ${JSON.stringify(result).substring(0, 200)}`);
@@ -2156,8 +2179,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     }
   }
   if (name === "get_usage_stats") {
-    const email = String(args?.email);
-    logger.info(`[get_usage_stats] Called with email: ${email}`);
+    const email = args?.email ? String(args.email) : void 0;
+    logger.info(`[get_usage_stats] Called with email: ${email || "(using API key identity)"}`);
     try {
       const result = await nerveCenter.getUsageStats(email);
       logger.info(`[get_usage_stats] Result: ${JSON.stringify(result).substring(0, 200)}`);

package/package.json

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