mono-pilot 0.2.9 → 0.2.12

package/dist/src/extensions/session-hints.js

@@ -1,49 +1,41 @@
 import { existsSync, readFileSync } from "node:fs";
-import { readdir } from "node:fs/promises";
 import { homedir } from "node:os";
-import { basename, dirname, join, resolve } from "node:path";
+import { dirname, resolve } from "node:path";
 import { fileURLToPath } from "node:url";
 import { Text } from "@mariozechner/pi-tui";
 import { hasMessageEntries } from "./mode-runtime.js";
+import { isServerEnabled, loadMcpConfig, toNonEmptyString } from "../mcp/config.js";
+import { discoverRules } from "../rules/discovery.js";
 const SESSION_HINTS_MESSAGE_TYPE = "SessionHints";
-const RULES_RELATIVE_DIR = join(".pi", "rules");
 const MONO_PILOT_NAME = "mono-pilot";
 const MAX_VERSION_SEARCH_DEPTH = 6;
 let cachedVersion = null;
-/** List *.rule.txt full paths from a directory (empty array if dir missing). */
-async function listRuleFiles(dirPath) {
-    if (!existsSync(dirPath))
-        return [];
+async function discoverMcpServers(cwd) {
+    let config;
     try {
-        const entries = await readdir(dirPath, { withFileTypes: true, encoding: "utf8" });
-        return entries
-            .filter((e) => e.isFile() && e.name.endsWith(".rule.txt"))
-            .map((e) => resolve(dirPath, e.name))
-            .sort((a, b) => a.localeCompare(b));
+        config = await loadMcpConfig(cwd);
     }
     catch {
-        return [];
+        return { userMcpServers: [], projectMcpServers: [] };
     }
-}
-/** Discover rules files grouped by scope, deduped by rule name. */
-async function discoverRules(cwd) {
-    const workspaceRulesDir = resolve(cwd, RULES_RELATIVE_DIR);
-    const userRulesDir = resolve(homedir(), RULES_RELATIVE_DIR);
-    const [workspaceRules, userRules] = await Promise.all([
-        listRuleFiles(workspaceRulesDir),
-        listRuleFiles(userRulesDir),
-    ]);
-    const seenNames = new Set();
-    const dedupeByName = (rules) => rules.filter((filePath) => {
-        const name = basename(filePath, ".rule.txt");
-        if (seenNames.has(name))
-            return false;
-        seenNames.add(name);
-        return true;
-    });
-    const uniqueWorkspaceRules = dedupeByName(workspaceRules);
-    const uniqueUserRules = dedupeByName(userRules);
-    return { userRules: uniqueUserRules, projectRules: uniqueWorkspaceRules };
+    if (!config)
+        return { userMcpServers: [], projectMcpServers: [] };
+    const groups = { user: [], project: [] };
+    for (const [serverName, serverConfig] of Object.entries(config.servers)) {
+        if (!isServerEnabled(serverConfig))
+            continue;
+        const source = config.sourceByServer[serverName];
+        if (!source)
+            continue;
+        groups[source.scope].push({ name: serverName, url: toNonEmptyString(serverConfig.url) });
+    }
+    for (const servers of Object.values(groups)) {
+        servers.sort((a, b) => a.name.localeCompare(b.name));
+    }
+    return {
+        userMcpServers: groups.user,
+        projectMcpServers: groups.project,
+    };
 }
 function shortenHome(filePath) {
     const home = homedir();
@@ -94,8 +86,14 @@ export default function sessionHintsExtension(pi) {
         lines.push(theme.fg("dim", "option+m") + theme.fg("muted", " to cycle Plan/Ask/Agent mode"));
         const userRules = details?.userRules ?? [];
         const projectRules = details?.projectRules ?? [];
-        if (userRules.length > 0 || projectRules.length > 0) {
+        const userMcpServers = details?.userMcpServers ?? [];
+        const projectMcpServers = details?.projectMcpServers ?? [];
+        const hasRules = userRules.length > 0 || projectRules.length > 0;
+        const hasMcp = userMcpServers.length > 0 || projectMcpServers.length > 0;
+        if (hasRules || hasMcp) {
             lines.push("");
+        }
+        if (hasRules) {
             lines.push(theme.fg("mdHeading", "[Rules]"));
             if (userRules.length > 0) {
                 lines.push(`  ${theme.fg("accent", "user")}`);
@@ -110,6 +108,26 @@ export default function sessionHintsExtension(pi) {
                 }
             }
         }
+        if (hasRules && hasMcp) {
+            lines.push("");
+        }
+        if (hasMcp) {
+            lines.push(theme.fg("mdHeading", "[MCP Servers]"));
+            if (userMcpServers.length > 0) {
+                lines.push(`  ${theme.fg("accent", "user")}`);
+                for (const server of userMcpServers) {
+                    const urlPart = server.url ? `  ${theme.fg("muted", server.url)}` : "";
+                    lines.push(`    ${theme.fg("dim", server.name)}${urlPart}`);
+                }
+            }
+            if (projectMcpServers.length > 0) {
+                lines.push(`  ${theme.fg("accent", "project")}`);
+                for (const server of projectMcpServers) {
+                    const urlPart = server.url ? `  ${theme.fg("muted", server.url)}` : "";
+                    lines.push(`    ${theme.fg("dim", server.name)}${urlPart}`);
+                }
+            }
+        }
         return new Text(lines.join("\n"), 0, 0);
     });
     pi.on("session_start", async (_event, ctx) => {
@@ -118,7 +136,14 @@ export default function sessionHintsExtension(pi) {
         const entries = ctx.sessionManager.getEntries();
         if (hasMessageEntries(entries))
             return;
-        const details = await discoverRules(ctx.cwd);
+        const [rulesDetails, mcpDetails] = await Promise.all([
+            discoverRules(ctx.cwd),
+            discoverMcpServers(ctx.cwd),
+        ]);
+        const details = {
+            ...rulesDetails,
+            ...mcpDetails,
+        };
         pi.sendMessage({
             customType: SESSION_HINTS_MESSAGE_TYPE,
             content: "",