@gramatr/client 0.5.1

package/gemini/lib/gemini-install-utils.ts

@@ -0,0 +1,169 @@
+import { existsSync, readFileSync } from 'fs';
+import { join } from 'path';
+
+export interface GeminiExtensionManifest {
+  name: string;
+  version: string;
+  description: string;
+  mcpServers?: Record<string, GeminiMcpServerConfig>;
+  settings?: GeminiSettingDef[];
+  contextFileName?: string;
+}
+
+export interface GeminiMcpServerConfig {
+  httpUrl?: string;
+  headers?: Record<string, string>;
+  timeout?: number;
+}
+
+export interface GeminiSettingDef {
+  name: string;
+  envVar: string;
+  sensitive?: boolean;
+  description?: string;
+}
+
+export interface GeminiHooksFileHook {
+  type: 'command';
+  command: string;
+  name?: string;
+  timeout?: number;
+  description?: string;
+}
+
+export interface GeminiHooksFileEntry {
+  matcher?: string;
+  sequential?: boolean;
+  hooks: GeminiHooksFileHook[];
+}
+
+export interface GeminiHooksFile {
+  hooks: Record<string, GeminiHooksFileEntry[]>;
+}
+
+/**
+ * Build the gemini-extension.json manifest for gramatr.
+ *
+ * The mcpServers block points to the production MCP endpoint.
+ * Auth is handled via the GRAMATR_API_KEY setting which Gemini stores
+ * in its extension .env / system keychain.
+ */
+export function buildExtensionManifest(): GeminiExtensionManifest {
+  return {
+    name: 'gramatr',
+    version: '1.0.0',
+    description: 'gramatr intelligence layer — decision routing, vector memory, and pattern learning for Gemini CLI',
+    mcpServers: {
+      gramatr: {
+        httpUrl: 'https://api.gramatr.com/mcp',
+        headers: {
+          Authorization: 'Bearer ${GRAMATR_API_KEY}',
+        },
+        timeout: 30000,
+      },
+    },
+    settings: [
+      {
+        name: 'API Key',
+        envVar: 'GRAMATR_API_KEY',
+        sensitive: true,
+        description: 'gramatr API key (starts with gmtr_sk_). Get one at gramatr.com or via gmtr-login.',
+      },
+    ],
+  };
+}
+
+/**
+ * Build the hooks/hooks.json for the gramatr Gemini extension.
+ *
+ * Maps Gemini CLI lifecycle events to gramatr hook scripts.
+ * Uses ${extensionPath} for portability — Gemini CLI expands this
+ * to the installed extension directory at runtime.
+ */
+export function buildGeminiHooksFile(): GeminiHooksFile {
+  return {
+    hooks: {
+      SessionStart: [
+        {
+          hooks: [
+            {
+              type: 'command',
+              command: 'npx tsx "${extensionPath}/hooks/session-start.ts"',
+              name: 'gramatr-session-start',
+              timeout: 15,
+              description: 'Load gramatr session context and handoff',
+            },
+          ],
+        },
+      ],
+      BeforeAgent: [
+        {
+          hooks: [
+            {
+              type: 'command',
+              command: 'npx tsx "${extensionPath}/hooks/user-prompt-submit.ts"',
+              name: 'gramatr-prompt-routing',
+              timeout: 15,
+              description: 'Route prompt through gramatr intelligence',
+            },
+          ],
+        },
+      ],
+      SessionEnd: [
+        {
+          hooks: [
+            {
+              type: 'command',
+              command: 'npx tsx "${extensionPath}/hooks/stop.ts"',
+              name: 'gramatr-session-end',
+              timeout: 10,
+              description: 'Submit classification feedback to gramatr',
+            },
+          ],
+        },
+      ],
+    },
+  };
+}
+
+/**
+ * Read existing Gemini settings.json and extract the mcpServers block.
+ * Returns null if file doesn't exist or can't be parsed.
+ */
+export function readGeminiSettings(geminiHome: string): Record<string, unknown> | null {
+  const settingsPath = join(geminiHome, 'settings.json');
+  if (!existsSync(settingsPath)) return null;
+  try {
+    return JSON.parse(readFileSync(settingsPath, 'utf8'));
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * Resolve the Gemini extensions directory.
+ */
+export function getGeminiExtensionsDir(home: string): string {
+  return join(home, '.gemini', 'extensions');
+}
+
+/**
+ * Resolve the gramatr extension install path.
+ */
+export function getGramatrExtensionDir(home: string): string {
+  return join(getGeminiExtensionsDir(home), 'gramatr');
+}
+
+/**
+ * Read the stored API key from ~/.gmtr.json (shared with other platforms).
+ */
+export function readStoredApiKey(home: string): string | null {
+  const configPath = join(home, '.gmtr.json');
+  if (!existsSync(configPath)) return null;
+  try {
+    const config = JSON.parse(readFileSync(configPath, 'utf8'));
+    return config.token || null;
+  } catch {
+    return null;
+  }
+}