@virsanghavi/axis-server 1.4.0 → 1.5.0

package/dist/mcp-server.mjs

@@ -1545,6 +1545,92 @@ 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) {
+    subscription = { checked: true, valid: true, plan: "local", reason: "No API key configured \u2014 local mode", checkedAt: Date.now() };
+    logger.info("[subscription] No API key \u2014 running in local/dev mode, skipping verification");
+    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() {
+  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(
@@ -1894,6 +1980,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 {
@@ -2069,6 +2165,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.5.0",
     "description": "Axis MCP Server CLI",
     "main": "dist/index.js",
     "bin": {