@nxuss/lemma 0.9.0 → 0.9.2

package/dist/esm/mcp/tools.js

@@ -10,6 +10,38 @@ import { ComplexityRouter } from "../proxy/ComplexityRouter";
 import { squeezeCode, squeezePrompt, pruneHistoryMessages, generateAstDiff } from "../utils/ContextSqueezer";
 import { reportSavings } from "../utils/reportSavings";
 import { getProxyPort, safeResolvePath, logError, logWarn, estimateTokensFromResult, isCommandAllowed, loadAllowedCommands } from "./utils";
+import { runAnalysis, parseDiff } from "../pr-review/PRReviewEngine";
+import { ReviewStore } from "../pr-review/ReviewStore";
+import { runPRReview } from "../pr-review/reviewRunner";
+// ── Pro License Gate ──────────────────────────────────────────────────────────
+const LICENSE_FILE = path.join(os.homedir(), '.lemma-cache', 'license.json');
+let _proStatusCache = null;
+function isProUser() {
+    if (_proStatusCache !== null)
+        return _proStatusCache;
+    try {
+        if (fs.existsSync(LICENSE_FILE)) {
+            const data = JSON.parse(fs.readFileSync(LICENSE_FILE, 'utf8'));
+            _proStatusCache = !!(data.isPro || data.tier === 'pro');
+            return _proStatusCache;
+        }
+    }
+    catch { }
+    _proStatusCache = false;
+    return false;
+}
+const PRO_GATE_MESSAGE = [
+    "🔒 Lemma MCP requires a Pro license.",
+    "",
+    "All 32 MCP tools — including The Brain, zero-token static analysis,",
+    "PreCrime predictor, Coupling Radar, and Entropy Score — are Pro features.",
+    "",
+    "✨ Activate your license:",
+    "  lemma activate <your-key>",
+    "",
+    "🚀 Get Lemma Pro:",
+    "  https://lemma.nxus.studio/upgrade",
+].join("\n");
 const toolDefinitions = [
     {
         name: "scrub_privacy",
@@ -403,6 +435,62 @@ const toolDefinitions = [
             },
         },
     },
+    {
+        name: "review_diff",
+        description: "Analyze a raw git diff locally for regressions, security issues, code quality, and static analysis. Zero-cost static analysis using local TypeScript compiler API — no LLM calls. Returns score, verdict, and detailed findings with suggestions.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                diff: { type: "string", description: "The git diff content to analyze" },
+                baseBranch: { type: "string", description: "Base branch for regression context", default: "main" },
+                threshold: { type: "number", description: "Quality score threshold for approval (0-100)", default: 90 },
+            },
+            required: ["diff"],
+        },
+    },
+    {
+        name: "review_pr",
+        description: "Analyze a Pull Request via GitHub, Azure DevOps, or GitLab API. Detects regressions, security issues, code quality problems, and breaking changes. Posts results as PR comments and can auto-approve if quality threshold is met. Pro license required.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                platform: { type: "string", enum: ["github", "azure", "gitlab"], description: "Git platform" },
+                repo: { type: "string", description: "Full repo name (e.g., 'owner/repo' or 'org/project/repo')" },
+                prNumber: { type: "number", description: "Pull request number" },
+                autoApprove: { type: "boolean", description: "Auto-approve if score >= threshold", default: true },
+                threshold: { type: "number", description: "Quality score threshold (0-100)", default: 90 },
+                context: { type: "string", description: "Optional context from The Brain about this project" },
+            },
+            required: ["platform", "repo", "prNumber"],
+        },
+    },
+    {
+        name: "pr_status",
+        description: "Get the status and detailed results of a previously reviewed PR. Returns score, verdict, findings, and approval status.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                reviewId: { type: "string", description: "Review ID returned by review_pr" },
+                repo: { type: "string", description: "Optional repo to find by PR number" },
+                prNumber: { type: "number", description: "Optional PR number to find by repo" },
+            },
+            required: [],
+        },
+    },
+    {
+        name: "generate_pr_workflow",
+        description: "Generate a CI/CD workflow file (GitHub Actions or Azure Pipelines) that runs Lemma PR Review on every pull request. Writes the file to the workspace.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                platform: { type: "string", enum: ["github", "azure"], description: "CI platform" },
+                outputPath: { type: "string", description: "Output path for the workflow file", default: ".github/workflows/lemma-pr-review.yml" },
+                autoApprove: { type: "boolean", description: "Enable auto-approve in the workflow", default: true },
+                threshold: { type: "number", description: "Quality threshold for auto-approve", default: 90 },
+            },
+            required: ["platform"],
+        },
+    },
 ];
 const toolHandlers = {
     scrub_privacy: handleScrubPrivacy,
@@ -437,6 +525,10 @@ const toolHandlers = {
     precrime_static: handlePrecrimeStatic,
     semantic_dedup_guard: handleSemanticDedupGuard,
     dead_export_necromancer: handleDeadExportNecromancer,
+    review_diff: handleReviewDiff,
+    review_pr: handleReviewPR,
+    pr_status: handlePRStatus,
+    generate_pr_workflow: handleGeneratePRWorkflow,
 };
 export function setupToolsHandlers(server, onToolCall) {
     server.setRequestHandler(ListToolsRequestSchema, async () => ({
@@ -445,6 +537,20 @@ export function setupToolsHandlers(server, onToolCall) {
     server.setRequestHandler(CallToolRequestSchema, async (request) => {
         const { name, arguments: args } = request.params;
         const startTime = Date.now();
+        // ── Pro License Gate ────────────────────────────────────────────────────
+        if (!isProUser()) {
+            onToolCall?.({
+                tool: name,
+                args: args || {},
+                result: "ERROR",
+                latency: Date.now() - startTime,
+                error: "Pro license required",
+            });
+            return {
+                content: [{ type: "text", text: PRO_GATE_MESSAGE }],
+            };
+        }
+        // ── End Pro Gate ────────────────────────────────────────────────────────
         const handler = toolHandlers[name];
         if (!handler) {
             const latency = Date.now() - startTime;
@@ -2319,4 +2425,200 @@ async function handleDeadExportNecromancer(args) {
         return { content: [{ type: "text", text: `Dead export necromancer failed: ${err.message}` }] };
     }
 }
+// ─── review_diff ─────────────────────────────────────────────────────────────
+async function handleReviewDiff(args) {
+    const diff = args?.diff;
+    const baseBranch = args?.baseBranch || "main";
+    const threshold = args?.threshold || 90;
+    if (!diff)
+        throw new Error("diff is required");
+    const workspaceRoot = process.cwd();
+    try {
+        const diffFiles = parseDiff(diff);
+        if (diffFiles.length === 0) {
+            return { content: [{ type: "text", text: "No parseable diff files found. Ensure the input is a valid git diff output." }] };
+        }
+        const result = runAnalysis(diffFiles, { threshold, autoApprove: false }, workspaceRoot, baseBranch);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: [
+                        `# 🔍 Lemma PR Review — Local Diff Analysis`,
+                        ``,
+                        `**Files analyzed:** ${diffFiles.length}`,
+                        `**Score:** ${result.score}/100 | **Verdict:** ${result.verdict}`,
+                        ``,
+                        result.summary,
+                    ].join("\n"),
+                },
+            ],
+        };
+    }
+    catch (err) {
+        logError("review_diff", err);
+        return { content: [{ type: "text", text: `PR review analysis failed: ${err.message}` }] };
+    }
+}
+// ─── review_pr ─────────────────────────────────────────────────────────────
+async function handleReviewPR(args) {
+    const platform = args?.platform;
+    const repo = args?.repo;
+    const prNumber = args?.prNumber;
+    const autoApprove = args?.autoApprove !== false;
+    const threshold = args?.threshold || 90;
+    if (!platform || !repo || !prNumber) {
+        throw new Error("platform, repo, and prNumber are required");
+    }
+    try {
+        const result = await runPRReview(platform, repo, prNumber, { autoApprove, threshold });
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify(result, null, 2),
+                },
+            ],
+        };
+    }
+    catch (err) {
+        logError("review_pr", err);
+        return { content: [{ type: "text", text: `PR review failed: ${err.message}` }] };
+    }
+}
+// ─── pr_status ─────────────────────────────────────────────────────────────
+async function handlePRStatus(args) {
+    const reviewId = args?.reviewId;
+    const repo = args?.repo;
+    const prNumber = args?.prNumber;
+    let review;
+    if (reviewId) {
+        review = ReviewStore.get(reviewId);
+    }
+    else if (repo && prNumber) {
+        review = ReviewStore.findByPR(repo, prNumber);
+    }
+    if (!review) {
+        const allReviews = ReviewStore.getAll();
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: allReviews.length === 0
+                        ? "No PR reviews found in this session."
+                        : `PR reviews in this session:\n\n${allReviews.slice(-5).map(r => `- ${r.metadata?.repo || "?"} #${r.metadata?.prNumber || "?"}: ${r.score}/100 — ${r.verdict} (${r.reviewId})`).join("\n")}`,
+                },
+            ],
+        };
+    }
+    return {
+        content: [
+            {
+                type: "text",
+                text: [
+                    `# 📋 PR Review Status`,
+                    ``,
+                    `**Review ID:** ${review.reviewId}`,
+                    `**Repo:** ${review.metadata?.repo || "N/A"}`,
+                    `**PR:** #${review.metadata?.prNumber || "N/A"}`,
+                    `**Score:** ${review.score}/100`,
+                    `**Verdict:** ${review.verdict}`,
+                    `**Approved:** ${review.approved ? "✅ Yes" : "❌ No"}`,
+                    `**Timestamp:** ${review.timestamp}`,
+                    ``,
+                    review.summary,
+                ].join("\n"),
+            },
+        ],
+    };
+}
+// ─── generate_pr_workflow ─────────────────────────────────────────────────
+async function handleGeneratePRWorkflow(args) {
+    const platform = args?.platform;
+    const outputPath = args?.outputPath || ".github/workflows/lemma-pr-review.yml";
+    const autoApprove = args?.autoApprove !== false;
+    const threshold = args?.threshold || 90;
+    if (!platform)
+        throw new Error("platform is required");
+    if (!["github", "azure"].includes(platform)) {
+        throw new Error("platform must be 'github' or 'azure'");
+    }
+    const workspaceRoot = process.cwd();
+    const { resolved } = safeResolvePath(workspaceRoot, outputPath);
+    const dir = path.dirname(resolved);
+    try {
+        if (!fs.existsSync(dir)) {
+            fs.mkdirSync(dir, { recursive: true });
+        }
+        const tokenName = platform === "github" ? "GITHUB_TOKEN" : "AZURE_DEVOPS_TOKEN";
+        const yaml = platform === "github"
+            ? generateGitHubWorkflowYaml(threshold)
+            : generateAzurePipelineYaml(threshold);
+        fs.writeFileSync(resolved, yaml, "utf8");
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✅ Workflow file generated at \`${outputPath}\`\n\nMake sure to add your Lemma license key and ${tokenName} as secrets in your ${platform === "github" ? "GitHub repository" : "Azure Pipeline"}.`,
+                },
+            ],
+        };
+    }
+    catch (err) {
+        logError("generate_pr_workflow", err);
+        return { content: [{ type: "text", text: `Failed to generate workflow: ${err.message}` }] };
+    }
+}
+function generateGitHubWorkflowYaml(threshold) {
+    return [
+        `name: Lemma PR Review`,
+        `on:`,
+        `  pull_request:`,
+        `    types: [opened, synchronize]`,
+        ``,
+        `jobs:`,
+        `  lemma-review:`,
+        `    runs-on: ubuntu-latest`,
+        `    steps:`,
+        `      - uses: actions/checkout@v4`,
+        `        with:`,
+        `          fetch-depth: 0`,
+        `      - uses: actions/setup-node@v4`,
+        `      - run: npm install -g @nxuss/lemma`,
+        `      - run: lemma pr-review --threshold ${threshold}`,
+        `        env:`,
+        `          LEMMA_LICENSE_KEY: \${{ secrets.LEMMA_LICENSE_KEY }}`,
+        `          GITHUB_TOKEN: \${{ secrets.GITHUB_TOKEN }}`,
+        `          LEMMA_PLATFORM: github`,
+        `          LEMMA_REPO: \${{ github.repository }}`,
+        `          LEMMA_PR_NUMBER: \${{ github.event.pull_request.number }}`,
+    ].join("\n");
+}
+function generateAzurePipelineYaml(threshold) {
+    return [
+        `trigger: none`,
+        `pr:`,
+        `  branches:`,
+        `    include:`,
+        `      - '*'`,
+        ``,
+        `pool:`,
+        `  vmImage: ubuntu-latest`,
+        ``,
+        `steps:`,
+        `  - checkout: self`,
+        `    fetchDepth: 0`,
+        `  - task: NodeTool@0`,
+        `    inputs:`,
+        `      versionSpec: '18'`,
+        `  - script: npm install -g @nxuss/lemma`,
+        `  - script: lemma pr-review --threshold ${threshold}`,
+        `    env:`,
+        `      LEMMA_LICENSE_KEY: \$(LEMMA_LICENSE_KEY)`,
+        `      LEMMA_PLATFORM: azure`,
+        `      LEMMA_REPO: \$(Build.Repository.Name)`,
+        `      LEMMA_PR_NUMBER: \$(System.PullRequest.PullRequestId)`,
+        `      AZURE_DEVOPS_TOKEN: \$(AZURE_DEVOPS_TOKEN)`,
+    ].join("\n");
+}
 //# sourceMappingURL=tools.js.map