@unifiedmemory/cli 1.0.0

package/bin/um-mcp-serve

@@ -0,0 +1,62 @@
+#!/usr/bin/env bash
+set -e
+
+# Find script location (works even when symlinked)
+if command -v readlink >/dev/null 2>&1; then
+  SCRIPT_PATH="$(readlink -f "$0" 2>/dev/null || echo "$0")"
+else
+  SCRIPT_PATH="$0"
+fi
+SCRIPT_DIR="$(cd "$(dirname "$SCRIPT_PATH")" && pwd)"
+CLI_ROOT="$(dirname "$SCRIPT_DIR")"
+
+# Config file paths
+CONFIG_FILE=".um/config.json"
+AUTH_FILE="$HOME/.um/auth.json"
+
+# Export org/project IDs from config if available
+if [ -f "$CONFIG_FILE" ]; then
+  if command -v jq >/dev/null 2>&1; then
+    UM_ORG_ID=$(jq -r '.org_id // empty' "$CONFIG_FILE" 2>/dev/null || echo "")
+    UM_PROJECT_ID=$(jq -r '.project_id // empty' "$CONFIG_FILE" 2>/dev/null || echo "")
+
+    if [ -n "$UM_ORG_ID" ]; then
+      export UM_ORG_ID
+      echo "✓ Loaded UM_ORG_ID: $UM_ORG_ID" >&2
+    fi
+
+    if [ -n "$UM_PROJECT_ID" ]; then
+      export UM_PROJECT_ID
+      echo "✓ Loaded UM_PROJECT_ID: $UM_PROJECT_ID" >&2
+    fi
+  else
+    echo "⚠️  jq not found, cannot parse config files" >&2
+    echo "   Install jq: brew install jq (macOS) or apt-get install jq (Linux)" >&2
+  fi
+fi
+
+# Export user ID from auth file if available
+if [ -f "$AUTH_FILE" ]; then
+  if command -v jq >/dev/null 2>&1; then
+    UM_USER_ID=$(jq -r '.decoded.sub // empty' "$AUTH_FILE" 2>/dev/null || echo "")
+
+    if [ -n "$UM_USER_ID" ]; then
+      export UM_USER_ID
+      echo "✓ Loaded UM_USER_ID: $UM_USER_ID" >&2
+    fi
+  fi
+fi
+
+# Warn if environment variables not set
+if [ -z "$UM_ORG_ID" ] || [ -z "$UM_USER_ID" ]; then
+  echo "⚠️  Warning: Some auth context not available for hooks" >&2
+  echo "   UM_ORG_ID: ${UM_ORG_ID:-<not set>}" >&2
+  echo "   UM_USER_ID: ${UM_USER_ID:-<not set>}" >&2
+  echo "   UM_PROJECT_ID: ${UM_PROJECT_ID:-<not set>}" >&2
+  echo "" >&2
+  echo "   This may cause hook-triggered vault tool calls to fail." >&2
+  echo "   Run 'um login' and 'um init' to configure authentication." >&2
+fi
+
+# Start MCP server
+exec node "$CLI_ROOT/index.js" mcp serve

package/commands/init.js

@@ -0,0 +1,315 @@
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+import fs from 'fs-extra';
+import path from 'path';
+import axios from 'axios';
+import { getToken, saveToken } from '../lib/token-storage.js';
+import { loadAndRefreshToken } from '../lib/token-validation.js';
+import { login } from './login.js';
+import { ProviderDetector } from '../lib/provider-detector.js';
+
+export async function init(options = {}) {
+  console.log(chalk.cyan('\n🚀 UnifiedMemory Initialization\n'));
+
+  // Step 1: Check/refresh authentication
+  const authData = await ensureAuthenticated(options);
+  if (!authData) {
+    console.error(chalk.red('❌ Authentication failed'));
+    process.exit(1);
+  }
+
+  console.log(chalk.green(`✓ Logged in as: ${authData.user_id}`));
+  if (authData.org_id) {
+    console.log(chalk.green(`✓ Organization: ${authData.org_id}`));
+  }
+
+  // Step 2: Select/create project
+  const projectData = await selectOrCreateProject(authData, options);
+  if (!projectData) {
+    console.error(chalk.red('❌ Project setup failed'));
+    process.exit(1);
+  }
+
+  console.log(chalk.green(`✓ Project: ${projectData.project_name} (${projectData.project_id})`));
+
+  // Step 3: Save project config
+  await saveProjectConfig(authData, projectData);
+
+  // Step 4: Configure AI tools
+  if (!options.skipConfigure) {
+    await configureProviders(authData, projectData);
+  }
+
+  console.log(chalk.green('\n✅ Initialization complete!\n'));
+  console.log(chalk.cyan('Next steps:'));
+  console.log('  1. Restart your AI code assistant (Claude, Cursor, Cline)');
+  console.log('  2. Start working - context is automatically shared!');
+  console.log('  3. Run `um status` to verify configuration\n');
+}
+
+async function ensureAuthenticated(options) {
+  // Try to load and refresh token if expired
+  const stored = await loadAndRefreshToken(false); // Don't throw, allow new login
+
+  if (stored && stored.decoded) {
+    console.log(chalk.gray('Using saved session'));
+
+    // Extract user_id and org_id from token
+    const userId = stored.decoded.sub;
+    const orgId = stored.selectedOrg?.id || userId; // Fallback to user_id if no org
+    const expirationTime = stored.decoded.exp * 1000;
+
+    return {
+      user_id: userId,
+      org_id: orgId,
+      access_token: stored.idToken || stored.accessToken,
+      api_url: 'https://rose-asp-main-1c0b114.d2.zuplo.dev',
+      expires_at: expirationTime,
+    };
+  }
+
+  // Need to login
+  console.log(chalk.yellow('⚠ Authentication required\n'));
+
+  if (options.apiKey) {
+    return await loginWithApiKey(options.apiKey);
+  }
+
+  // For now, just use OAuth
+  console.log(chalk.blue('Initiating OAuth login...'));
+  const tokenData = await login();
+
+  if (!tokenData) {
+    return null;
+  }
+
+  // Extract from saved token
+  const savedToken = getToken();
+  const userId = savedToken?.decoded?.sub;
+  const orgId = savedToken?.selectedOrg?.id || userId;
+
+  return {
+    user_id: userId,
+    org_id: orgId,
+    access_token: savedToken.idToken || savedToken.accessToken,
+    api_url: 'https://rose-asp-main-1c0b114.d2.zuplo.dev',
+    expires_at: savedToken.decoded?.exp * 1000,
+  };
+}
+
+async function selectOrCreateProject(authData, options) {
+  const apiUrl = authData.api_url || 'https://rose-asp-main-1c0b114.d2.zuplo.dev';
+
+  // Fetch existing projects
+  const projects = await fetchProjects(authData, apiUrl);
+
+  console.log(chalk.cyan('\n📋 Project setup:\n'));
+  console.log(chalk.green(`  1. Create new project`));
+  console.log(chalk.green(`  2. Select existing project (${projects.length} available)`));
+
+  const { selection } = await inquirer.prompt([{
+    type: 'input',
+    name: 'selection',
+    message: 'Choose option (1-2):',
+    default: projects.length > 0 ? '2' : '1',
+    validate: (input) => {
+      const num = parseInt(input, 10);
+      if (isNaN(num) || num < 1 || num > 2) {
+        return 'Please enter 1 or 2';
+      }
+      return true;
+    },
+  }]);
+
+  const action = parseInt(selection, 10) === 1 ? 'create' : 'select';
+
+  if (action === 'create') {
+    const { name, description } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'name',
+        message: 'Project name:',
+        validate: input => input.length > 0 || 'Name is required',
+      },
+      {
+        type: 'input',
+        name: 'description',
+        message: 'Project description (optional):',
+      },
+    ]);
+
+    return await createProject(authData, apiUrl, name, description);
+  } else {
+    if (projects.length === 0) {
+      console.log(chalk.yellow('No projects found. Creating first project...'));
+      return await selectOrCreateProject(authData, { ...options, action: 'create' });
+    }
+
+    // Display numbered list
+    console.log(chalk.cyan('\n📋 Available projects:\n'));
+    projects.forEach((p, index) => {
+      console.log(chalk.green(`  ${index + 1}. ${p.display_name || p.name}`) + chalk.gray(` (${p.slug || p.id})`));
+    });
+
+    const { projectSelection } = await inquirer.prompt([{
+      type: 'input',
+      name: 'projectSelection',
+      message: `Choose project (1-${projects.length}):`,
+      default: '1',
+      validate: (input) => {
+        const num = parseInt(input, 10);
+        if (isNaN(num) || num < 1 || num > projects.length) {
+          return `Please enter a number between 1 and ${projects.length}`;
+        }
+        return true;
+      },
+    }]);
+
+    const selectedIndex = parseInt(projectSelection, 10) - 1;
+    const project = projects[selectedIndex];
+
+    return {
+      project_id: project.id,
+      project_name: project.display_name || project.name,
+      project_slug: project.slug,
+    };
+  }
+}
+
+async function fetchProjects(authData, apiUrl) {
+  try {
+    const response = await axios.get(
+      `${apiUrl}/v1/orgs/${authData.org_id}/projects`,
+      {
+        headers: {
+          'Authorization': `Bearer ${authData.access_token}`,
+          'X-Org-Id': authData.org_id,
+        },
+      }
+    );
+    // API returns {items: [...], next_cursor, has_more}
+    return response.data?.items || [];
+  } catch (error) {
+    console.error(chalk.red(`Failed to fetch projects: ${error.message}`));
+    return [];
+  }
+}
+
+async function createProject(authData, apiUrl, name, description) {
+  try {
+    const response = await axios.post(
+      `${apiUrl}/v1/orgs/${authData.org_id}/projects`,
+      {
+        display_name: name,
+        description: description || '',
+      },
+      {
+        headers: {
+          'Authorization': `Bearer ${authData.access_token}`,
+          'X-Org-Id': authData.org_id,
+        },
+      }
+    );
+
+    const project = response.data;
+    return {
+      project_id: project.id,
+      project_name: project.display_name || project.name,
+      project_slug: project.slug,
+    };
+  } catch (error) {
+    console.error(chalk.red(`Failed to create project: ${error.message}`));
+    return null;
+  }
+}
+
+async function saveProjectConfig(authData, projectData) {
+  const umDir = path.join(process.cwd(), '.um');
+  const configPath = path.join(umDir, 'config.json');
+
+  const config = {
+    version: '1.0',
+    org_id: authData.org_id,
+    project_id: projectData.project_id,
+    project_name: projectData.project_name,
+    api_url: authData.api_url || 'https://rose-asp-main-1c0b114.d2.zuplo.dev',
+    created_at: new Date().toISOString(),
+    mcp_config: {
+      inject_headers: true,
+      auto_context: true,
+    },
+  };
+
+  fs.ensureDirSync(umDir);
+  fs.writeJSONSync(configPath, config, { spaces: 2 });
+
+  console.log(chalk.green(`✓ Project config saved: ${configPath}`));
+
+  // Add .um and .claude to .gitignore
+  const gitignorePath = path.join(process.cwd(), '.gitignore');
+  if (fs.existsSync(gitignorePath)) {
+    let gitignore = fs.readFileSync(gitignorePath, 'utf-8');
+    let modified = false;
+
+    if (!gitignore.includes('.um/')) {
+      gitignore += '\n# UnifiedMemory CLI\n.um/\n';
+      modified = true;
+    }
+
+    if (!gitignore.includes('.claude/')) {
+      gitignore += '.claude/\n';
+      modified = true;
+    }
+
+    if (modified) {
+      fs.writeFileSync(gitignorePath, gitignore);
+      console.log(chalk.gray('  Updated .gitignore'));
+    }
+  } else {
+    // Create .gitignore if it doesn't exist
+    fs.writeFileSync(gitignorePath, '# UnifiedMemory CLI\n.um/\n.claude/\n');
+    console.log(chalk.gray('  Created .gitignore with .um/ and .claude/ entries'));
+  }
+}
+
+async function configureProviders(authData, projectData) {
+  console.log(chalk.cyan('\n🔧 Configuring AI code assistants...\n'));
+
+  // Pass current directory for project-level configs (like Claude Code)
+  const detector = new ProviderDetector(process.cwd());
+  const detected = detector.detectAll();
+
+  if (detected.length === 0) {
+    console.log(chalk.yellow('⚠ No AI code assistants detected'));
+    console.log(chalk.gray('  Install Claude Code, Cursor, Cline, Codex CLI, or Gemini CLI'));
+    console.log(chalk.gray('  to enable auto-configuration'));
+    return;
+  }
+
+  // NEW APPROACH: Configure local MCP server (no tokens in config files)
+  for (const provider of detected) {
+    const success = provider.configureMCP();
+
+    if (success) {
+      console.log(chalk.green(`✓ Configured ${provider.name}`));
+    } else {
+      console.log(chalk.red(`✗ Failed to configure ${provider.name}`));
+    }
+  }
+
+  console.log(chalk.cyan('\n💡 Important: Restart your AI assistant to load the new configuration'));
+  console.log(chalk.gray('   The MCP server will automatically use your authentication and project context'));
+}
+
+async function loginWithApiKey(apiKey) {
+  // TODO: Implement API key authentication
+  // Exchange API key for JWT token
+  console.log(chalk.yellow('API key authentication not yet implemented'));
+  return null;
+}
+
+async function refreshToken(refreshToken) {
+  // TODO: Implement token refresh
+  console.log(chalk.yellow('Token refresh not yet implemented'));
+  return null;
+}