claude-flow 2.0.0-alpha.79 → 2.0.0-alpha.80

package/src/cli/simple-commands/init/index.js

@@ -904,6 +904,107 @@ async function setupCoordinationSystem(workingDir, dryRun = false) {
   // This is a placeholder for future coordination setup logic
 }
 
+/**
+ * Setup monitoring and telemetry for token tracking
+ */
+async function setupMonitoring(workingDir) {
+  console.log('  📈 Configuring token usage tracking...');
+  
+  const fs = await import('fs/promises');
+  const path = await import('path');
+  
+  try {
+    // Create .claude-flow directory for tracking data
+    const trackingDir = path.join(workingDir, '.claude-flow');
+    await fs.mkdir(trackingDir, { recursive: true });
+    
+    // Create initial token usage file
+    const tokenUsageFile = path.join(trackingDir, 'token-usage.json');
+    const initialData = {
+      total: 0,
+      input: 0,
+      output: 0,
+      byAgent: {},
+      lastUpdated: new Date().toISOString()
+    };
+    
+    await fs.writeFile(tokenUsageFile, JSON.stringify(initialData, null, 2));
+    printSuccess('  ✓ Created token usage tracking file');
+    
+    // Add telemetry configuration to .claude/settings.json if it exists
+    const settingsPath = path.join(workingDir, '.claude', 'settings.json');
+    try {
+      const settingsContent = await fs.readFile(settingsPath, 'utf8');
+      const settings = JSON.parse(settingsContent);
+      
+      // Add telemetry hook
+      if (!settings.hooks) settings.hooks = {};
+      if (!settings.hooks['post-task']) settings.hooks['post-task'] = [];
+      
+      // Add token tracking hook
+      const tokenTrackingHook = 'npx claude-flow@alpha internal track-tokens --session-id {{session_id}} --tokens {{token_usage}}';
+      if (!settings.hooks['post-task'].includes(tokenTrackingHook)) {
+        settings.hooks['post-task'].push(tokenTrackingHook);
+      }
+      
+      await fs.writeFile(settingsPath, JSON.stringify(settings, null, 2));
+      printSuccess('  ✓ Added token tracking hooks to settings');
+    } catch (err) {
+      console.log('  ⚠️  Could not update settings.json:', err.message);
+    }
+    
+    // Create monitoring configuration
+    const monitoringConfig = {
+      enabled: true,
+      telemetry: {
+        claudeCode: {
+          env: 'CLAUDE_CODE_ENABLE_TELEMETRY',
+          value: '1',
+          description: 'Enable Claude Code OpenTelemetry metrics'
+        }
+      },
+      tracking: {
+        tokens: true,
+        costs: true,
+        agents: true,
+        sessions: true
+      },
+      storage: {
+        location: '.claude-flow/token-usage.json',
+        format: 'json',
+        rotation: 'monthly'
+      }
+    };
+    
+    const configPath = path.join(trackingDir, 'monitoring.config.json');
+    await fs.writeFile(configPath, JSON.stringify(monitoringConfig, null, 2));
+    printSuccess('  ✓ Created monitoring configuration');
+    
+    // Create shell profile snippet for environment variable
+    const envSnippet = `
+# Claude Flow Token Tracking
+# Add this to your shell profile (.bashrc, .zshrc, etc.)
+export CLAUDE_CODE_ENABLE_TELEMETRY=1
+
+# Optional: Set custom metrics path
+# export CLAUDE_METRICS_PATH="$HOME/.claude/metrics"
+`;
+    
+    const envPath = path.join(trackingDir, 'env-setup.sh');
+    await fs.writeFile(envPath, envSnippet.trim());
+    printSuccess('  ✓ Created environment setup script');
+    
+    console.log('\n  📋 To enable Claude Code telemetry:');
+    console.log('     1. Add to your shell profile: export CLAUDE_CODE_ENABLE_TELEMETRY=1');
+    console.log('     2. Or run: source .claude-flow/env-setup.sh');
+    console.log('\n  💡 Token usage will be tracked in .claude-flow/token-usage.json');
+    console.log('     Run: claude-flow analysis token-usage --breakdown --cost-analysis');
+    
+  } catch (err) {
+    printError(`  Failed to setup monitoring: ${err.message}`);
+  }
+}
+
 /**
  * Enhanced Claude Flow v2.0.0 initialization
  */
@@ -1095,7 +1196,7 @@ ${commands.map((cmd) => `- [${cmd}](./${cmd}.md)`).join('\n')}
     }
 
     // Create helper scripts
-    const helpers = ['setup-mcp.sh', 'quick-start.sh', 'github-setup.sh', 'github-safe.js'];
+    const helpers = ['setup-mcp.sh', 'quick-start.sh', 'github-setup.sh', 'github-safe.js', 'standard-checkpoint-hooks.sh', 'checkpoint-manager.sh'];
     for (const helper of helpers) {
       if (!dryRun) {
         const content = createHelperScript(helper);
@@ -1123,6 +1224,7 @@ ${commands.map((cmd) => `- [${cmd}](./${cmd}.md)`).join('\n')}
       'coordination/orchestration',
       '.swarm', // Add .swarm directory for shared memory
       '.hive-mind', // Add .hive-mind directory for hive-mind system
+      '.claude/checkpoints', // Add checkpoints directory for Git checkpoint system
     ];
 
     for (const dir of standardDirs) {
@@ -1327,6 +1429,13 @@ ${commands.map((cmd) => `- [${cmd}](./${cmd}.md)`).join('\n')}
       console.log('  [DRY RUN] Would create agent system with 64 specialized agents');
     }
 
+    // Optional: Setup monitoring and telemetry
+    const enableMonitoring = flags.monitoring || flags['enable-monitoring'];
+    if (enableMonitoring && !dryRun) {
+      console.log('\n📊 Setting up monitoring and telemetry...');
+      await setupMonitoring(workingDir);
+    }
+    
     // Final instructions
     console.log('\n🎉 Claude Flow v2.0.0 initialization complete!');
     console.log('\n📚 Quick Start:');
@@ -1347,6 +1456,8 @@ ${commands.map((cmd) => `- [${cmd}](./${cmd}.md)`).join('\n')}
     console.log('• Use --help with any command for options');
     console.log('• Run commands with --claude flag for best Claude Code integration');
     console.log('• Enable GitHub integration with .claude/helpers/github-setup.sh');
+    console.log('• Git checkpoints are automatically enabled in settings.json');
+    console.log('• Use .claude/helpers/checkpoint-manager.sh for easy rollback');
   } catch (err) {
     printError(`Failed to initialize Claude Flow v2.0.0: ${err.message}`);
   }

package/src/cli/simple-commands/init/template-copier.js

@@ -311,7 +311,7 @@ async function copyHelperScripts(templatesDir, targetDir, options, results) {
       await fs.mkdir(helpersDir, { recursive: true });
     }
 
-    const helpers = ['setup-mcp.sh', 'quick-start.sh', 'github-setup.sh', 'github-safe.js'];
+    const helpers = ['setup-mcp.sh', 'quick-start.sh', 'github-setup.sh', 'github-safe.js', 'checkpoint-manager.sh', 'standard-checkpoint-hooks.sh'];
     const { createHelperScript } = await import('./templates/enhanced-templates.js');
     
     for (const helper of helpers) {

package/src/cli/simple-commands/init/templates/commands/helpers/standard-checkpoint-hooks.sh

@@ -0,0 +1,179 @@
+#!/bin/bash
+# Standard checkpoint hook functions for Claude settings.json (without GitHub features)
+
+# Function to handle pre-edit checkpoints
+pre_edit_checkpoint() {
+    local tool_input="$1"
+    local file=$(echo "$tool_input" | jq -r '.file_path // empty')
+    
+    if [ -n "$file" ]; then
+        local checkpoint_branch="checkpoint/pre-edit-$(date +%Y%m%d-%H%M%S)"
+        local current_branch=$(git branch --show-current)
+        
+        # Create checkpoint
+        git add -A
+        git stash push -m "Pre-edit checkpoint for $file" >/dev/null 2>&1
+        git branch "$checkpoint_branch"
+        
+        # Store metadata
+        mkdir -p .claude/checkpoints
+        cat > ".claude/checkpoints/$(date +%s).json" <<EOF
+{
+  "branch": "$checkpoint_branch",
+  "file": "$file",
+  "timestamp": "$(date -u +%Y-%m-%dT%H:%M:%SZ)",
+  "type": "pre-edit",
+  "original_branch": "$current_branch"
+}
+EOF
+        
+        # Restore working directory
+        git stash pop --quiet >/dev/null 2>&1 || true
+        
+        echo "✅ Created checkpoint: $checkpoint_branch for $file"
+    fi
+}
+
+# Function to handle post-edit checkpoints
+post_edit_checkpoint() {
+    local tool_input="$1"
+    local file=$(echo "$tool_input" | jq -r '.file_path // empty')
+    
+    if [ -n "$file" ] && [ -f "$file" ]; then
+        # Check if file was modified - first check if file is tracked
+        if ! git ls-files --error-unmatch "$file" >/dev/null 2>&1; then
+            # File is not tracked, add it first
+            git add "$file"
+        fi
+        
+        # Now check if there are changes
+        if git diff --cached --quiet "$file" 2>/dev/null && git diff --quiet "$file" 2>/dev/null; then
+            echo "ℹ️  No changes to checkpoint for $file"
+        else
+            local tag_name="checkpoint-$(date +%Y%m%d-%H%M%S)"
+            local current_branch=$(git branch --show-current)
+            
+            # Create commit
+            git add "$file"
+            if git commit -m "🔖 Checkpoint: Edit $file
+
+Automatic checkpoint created by Claude
+- File: $file
+- Branch: $current_branch
+- Timestamp: $(date -u +%Y-%m-%dT%H:%M:%SZ)
+
+[Auto-checkpoint]" --quiet; then
+                # Create tag only if commit succeeded
+                git tag -a "$tag_name" -m "Checkpoint after editing $file"
+                
+                # Store metadata
+                mkdir -p .claude/checkpoints
+                local diff_stats=$(git diff HEAD~1 --stat | tr '\n' ' ' | sed 's/"/\\"/g')
+                cat > ".claude/checkpoints/$(date +%s).json" <<EOF
+{
+  "tag": "$tag_name",
+  "file": "$file",
+  "timestamp": "$(date -u +%Y-%m-%dT%H:%M:%SZ)",
+  "type": "post-edit",
+  "branch": "$current_branch",
+  "diff_summary": "$diff_stats"
+}
+EOF
+                
+                echo "✅ Created checkpoint: $tag_name for $file"
+            else
+                echo "ℹ️  No commit created (no changes or commit failed)"
+            fi
+        fi
+    fi
+}
+
+# Function to handle task checkpoints
+task_checkpoint() {
+    local user_prompt="$1"
+    local task=$(echo "$user_prompt" | head -c 100 | tr '\n' ' ')
+    
+    if [ -n "$task" ]; then
+        local checkpoint_name="task-$(date +%Y%m%d-%H%M%S)"
+        
+        # Commit current state
+        git add -A
+        git commit -m "🔖 Task checkpoint: $task..." --quiet || true
+        
+        # Store metadata
+        mkdir -p .claude/checkpoints
+        cat > ".claude/checkpoints/task-$(date +%s).json" <<EOF
+{
+  "checkpoint": "$checkpoint_name",
+  "task": "$task",
+  "timestamp": "$(date -u +%Y-%m-%dT%H:%M:%SZ)",
+  "commit": "$(git rev-parse HEAD)"
+}
+EOF
+        
+        echo "✅ Created task checkpoint: $checkpoint_name"
+    fi
+}
+
+# Function to handle session end
+session_end_checkpoint() {
+    local session_id="session-$(date +%Y%m%d-%H%M%S)"
+    local summary_file=".claude/checkpoints/summary-$session_id.md"
+    
+    mkdir -p .claude/checkpoints
+    
+    # Create summary
+    cat > "$summary_file" <<EOF
+# Session Summary - $(date +'%Y-%m-%d %H:%M:%S')
+
+## Checkpoints Created
+$(find .claude/checkpoints -name '*.json' -mtime -1 -exec basename {} \; | sort)
+
+## Files Modified
+$(git diff --name-only $(git log --format=%H -n 1 --before="1 hour ago" 2>/dev/null) 2>/dev/null || echo "No files tracked")
+
+## Recent Commits
+$(git log --oneline -10 --grep="Checkpoint" || echo "No checkpoint commits")
+
+## Rollback Instructions
+To rollback to a specific checkpoint:
+\`\`\`bash
+# List all checkpoints
+git tag -l 'checkpoint-*' | sort -r
+
+# Rollback to a checkpoint
+git checkout checkpoint-YYYYMMDD-HHMMSS
+
+# Or reset to a checkpoint (destructive)
+git reset --hard checkpoint-YYYYMMDD-HHMMSS
+\`\`\`
+EOF
+    
+    # Create final checkpoint
+    git add -A
+    git commit -m "🏁 Session end checkpoint: $session_id" --quiet || true
+    git tag -a "session-end-$session_id" -m "End of Claude session"
+    
+    echo "✅ Session summary saved to: $summary_file"
+    echo "📌 Final checkpoint: session-end-$session_id"
+}
+
+# Main entry point
+case "$1" in
+    pre-edit)
+        pre_edit_checkpoint "$2"
+        ;;
+    post-edit)
+        post_edit_checkpoint "$2"
+        ;;
+    task)
+        task_checkpoint "$2"
+        ;;
+    session-end)
+        session_end_checkpoint
+        ;;
+    *)
+        echo "Usage: $0 {pre-edit|post-edit|task|session-end} [input]"
+        exit 1
+        ;;
+esac