@fermindi/pwn-cli 0.1.1 → 0.3.0

package/src/core/inject.js

@@ -1,204 +1,300 @@
-import { existsSync, cpSync, renameSync, readFileSync, appendFileSync, writeFileSync, readdirSync } from 'fs';
-import { join, dirname } from 'path';
-import { fileURLToPath } from 'url';
-import { randomUUID } from 'crypto';
-import { initState } from './state.js';
-
-const __dirname = dirname(fileURLToPath(import.meta.url));
-
-/**
- * Known AI instruction files in the industry
- * PWN should detect these and warn/offer migration
- */
-export const KNOWN_AI_FILES = [
-  // Claude Code
-  { pattern: 'CLAUDE.md', type: 'claude', description: 'Claude Code instructions' },
-  { pattern: 'claude.md', type: 'claude', description: 'Claude Code instructions' },
-  { pattern: '.claude', type: 'claude', description: 'Claude config directory' },
-
-  // Cursor
-  { pattern: '.cursorrules', type: 'cursor', description: 'Cursor AI rules' },
-  { pattern: '.cursorignore', type: 'cursor', description: 'Cursor ignore file' },
-
-  // GitHub Copilot
-  { pattern: '.github/copilot-instructions.md', type: 'copilot', description: 'GitHub Copilot instructions' },
-
-  // Session files (common pattern)
-  { pattern: 'session.*.md', type: 'session', description: 'Session file (legacy pattern)' },
-
-  // Other AI tools
-  { pattern: '.aider', type: 'aider', description: 'Aider AI config' },
-  { pattern: '.continue', type: 'continue', description: 'Continue AI config' },
-  { pattern: '.codeium', type: 'codeium', description: 'Codeium config' },
-];
-
-/**
- * Detect known AI instruction files in a directory
- * @param {string} cwd - Directory to scan
- * @returns {Object[]} Array of detected files with metadata
- */
-export function detectKnownAIFiles(cwd = process.cwd()) {
-  const detected = [];
-
-  for (const known of KNOWN_AI_FILES) {
-    // Handle glob patterns like session.*.md
-    if (known.pattern.includes('*')) {
-      const regex = new RegExp('^' + known.pattern.replace('.', '\\.').replace('*', '.*') + '$');
-      try {
-        const files = readdirSync(cwd);
-        for (const file of files) {
-          if (regex.test(file)) {
-            detected.push({
-              ...known,
-              file,
-              path: join(cwd, file)
-            });
-          }
-        }
-      } catch {
-        // Ignore read errors
-      }
-    } else {
-      // Direct file check
-      const filePath = join(cwd, known.pattern);
-      if (existsSync(filePath)) {
-        detected.push({
-          ...known,
-          file: known.pattern,
-          path: filePath
-        });
-      }
-    }
-  }
-
-  return detected;
-}
-
-/**
- * Get the path to the workspace template
- * @returns {string} Path to template directory
- */
-export function getTemplatePath() {
-  return join(__dirname, '../../templates/workspace/.ai');
-}
-
-/**
- * Inject PWN workspace into a project
- * @param {object} options - Injection options
- * @param {string} options.cwd - Target directory (defaults to process.cwd())
- * @param {boolean} options.force - Force overwrite existing .ai/ directory
- * @param {boolean} options.silent - Suppress console output
- * @param {boolean} options.skipDetection - Skip detection of known AI files
- * @returns {object} Result with success status and message
- */
-export async function inject(options = {}) {
-  const {
-    cwd = process.cwd(),
-    force = false,
-    silent = false,
-    skipDetection = false
-  } = options;
-
-  const templateDir = getTemplatePath();
-  const targetDir = join(cwd, '.ai');
-
-  const log = silent ? () => {} : console.log;
-
-  // Detect known AI instruction files
-  const detectedFiles = skipDetection ? [] : detectKnownAIFiles(cwd);
-
-  // Check if .ai/ already exists
-  if (existsSync(targetDir) && !force) {
-    return {
-      success: false,
-      error: 'ALREADY_EXISTS',
-      message: '.ai/ directory already exists. Use --force to overwrite.',
-      detected: detectedFiles
-    };
-  }
-
-  try {
-    // Copy workspace template
-    log('📦 Copying workspace template...');
-    cpSync(templateDir, targetDir, { recursive: true });
-
-    // Rename state.template.json → state.json
-    const templateState = join(targetDir, 'state.template.json');
-    const stateFile = join(targetDir, 'state.json');
-
-    if (existsSync(templateState)) {
-      renameSync(templateState, stateFile);
-    }
-
-    // Rename notifications.template.json → notifications.json and generate unique topic
-    const templateNotify = join(targetDir, 'config', 'notifications.template.json');
-    const notifyFile = join(targetDir, 'config', 'notifications.json');
-
-    if (existsSync(templateNotify)) {
-      renameSync(templateNotify, notifyFile);
-      initNotifications(notifyFile);
-    }
-
-    // Initialize state.json with current user
-    initState(cwd);
-
-    // Update .gitignore
-    updateGitignore(cwd, silent);
-
-    return {
-      success: true,
-      message: 'PWN workspace injected successfully',
-      path: targetDir,
-      detected: detectedFiles
-    };
-
-  } catch (error) {
-    return {
-      success: false,
-      error: 'INJECTION_FAILED',
-      message: error.message,
-      detected: detectedFiles
-    };
-  }
-}
-
-/**
- * Initialize notifications.json with unique topic
- * @param {string} notifyFile - Path to notifications.json
- */
-function initNotifications(notifyFile) {
-  try {
-    const content = readFileSync(notifyFile, 'utf8');
-    const config = JSON.parse(content);
-
-    // Generate unique topic ID
-    const uniqueId = randomUUID().split('-')[0]; // First segment: 8 chars
-    config.channels.ntfy.topic = `pwn-${uniqueId}`;
-
-    writeFileSync(notifyFile, JSON.stringify(config, null, 2));
-  } catch {
-    // Ignore errors - notifications will use defaults
-  }
-}
-
-/**
- * Update .gitignore to exclude PWN personal files
- * @param {string} cwd - Working directory
- * @param {boolean} silent - Suppress output
- */
-function updateGitignore(cwd, silent = false) {
-  const gitignorePath = join(cwd, '.gitignore');
-  let gitignoreContent = '';
-
-  if (existsSync(gitignorePath)) {
-    gitignoreContent = readFileSync(gitignorePath, 'utf8');
-  }
-
-  if (!gitignoreContent.includes('.ai/state.json')) {
-    const pwnSection = '\n# PWN\n.ai/state.json\n.ai/config/notifications.json\n';
-    appendFileSync(gitignorePath, pwnSection);
-    if (!silent) {
-      console.log('📝 Updated .gitignore');
-    }
-  }
-}
+import { existsSync, cpSync, renameSync, readFileSync, appendFileSync, writeFileSync, readdirSync, mkdirSync } from 'fs';
+import { join, dirname } from 'path';
+import { fileURLToPath } from 'url';
+import { randomUUID } from 'crypto';
+import { initState } from './state.js';
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+
+/**
+ * Known AI instruction files in the industry
+ * PWN should detect these and warn/offer migration
+ */
+export const KNOWN_AI_FILES = [
+  // Claude Code
+  { pattern: 'CLAUDE.md', type: 'claude', description: 'Claude Code instructions', migratable: true },
+  { pattern: 'claude.md', type: 'claude', description: 'Claude Code instructions', migratable: true },
+  { pattern: '.claude', type: 'claude', description: 'Claude config directory', migratable: false },
+
+  // Cursor
+  { pattern: '.cursorrules', type: 'cursor', description: 'Cursor AI rules', migratable: true },
+  { pattern: '.cursor/rules', type: 'cursor', description: 'Cursor AI rules', migratable: true },
+  { pattern: '.cursorignore', type: 'cursor', description: 'Cursor ignore file', migratable: false },
+
+  // Cline memory-bank
+  { pattern: 'memory-bank', type: 'memory-bank', description: 'Cline memory bank', migratable: true },
+
+  // GitHub Copilot
+  { pattern: '.github/copilot-instructions.md', type: 'copilot', description: 'GitHub Copilot instructions', migratable: true },
+
+  // Session files (common pattern)
+  { pattern: 'session.*.md', type: 'session', description: 'Session file (legacy pattern)', migratable: false },
+
+  // Other AI tools
+  { pattern: '.aider', type: 'aider', description: 'Aider AI config', migratable: false },
+  { pattern: '.continue', type: 'continue', description: 'Continue AI config', migratable: false },
+  { pattern: '.codeium', type: 'codeium', description: 'Codeium config', migratable: false },
+];
+
+/**
+ * Detect known AI instruction files in a directory
+ * @param {string} cwd - Directory to scan
+ * @returns {Object[]} Array of detected files with metadata
+ */
+export function detectKnownAIFiles(cwd = process.cwd()) {
+  const detected = [];
+
+  for (const known of KNOWN_AI_FILES) {
+    // Handle glob patterns like session.*.md
+    if (known.pattern.includes('*')) {
+      const regex = new RegExp('^' + known.pattern.replace('.', '\\.').replace('*', '.*') + '$');
+      try {
+        const files = readdirSync(cwd);
+        for (const file of files) {
+          if (regex.test(file)) {
+            detected.push({
+              ...known,
+              file,
+              path: join(cwd, file)
+            });
+          }
+        }
+      } catch {
+        // Ignore read errors
+      }
+    } else {
+      // Direct file check
+      const filePath = join(cwd, known.pattern);
+      if (existsSync(filePath)) {
+        detected.push({
+          ...known,
+          file: known.pattern,
+          path: filePath
+        });
+      }
+    }
+  }
+
+  return detected;
+}
+
+/**
+ * Get the path to the workspace template
+ * @returns {string} Path to template directory
+ */
+export function getTemplatePath() {
+  return join(__dirname, '../../templates/workspace/.ai');
+}
+
+/**
+ * Inject PWN workspace into a project
+ * @param {object} options - Injection options
+ * @param {string} options.cwd - Target directory (defaults to process.cwd())
+ * @param {boolean} options.force - Force overwrite existing .ai/ directory
+ * @param {boolean} options.silent - Suppress console output
+ * @param {boolean} options.skipDetection - Skip detection of known AI files
+ * @returns {object} Result with success status and message
+ */
+export async function inject(options = {}) {
+  const {
+    cwd = process.cwd(),
+    force = false,
+    silent = false,
+    skipDetection = false
+  } = options;
+
+  const templateDir = getTemplatePath();
+  const targetDir = join(cwd, '.ai');
+
+  const log = silent ? () => {} : console.log;
+
+  // Detect known AI instruction files
+  const detectedFiles = skipDetection ? [] : detectKnownAIFiles(cwd);
+
+  // Check if .ai/ already exists
+  if (existsSync(targetDir) && !force) {
+    return {
+      success: false,
+      error: 'ALREADY_EXISTS',
+      message: '.ai/ directory already exists. Use --force to overwrite.',
+      detected: detectedFiles
+    };
+  }
+
+  // Backup existing AI files content before overwriting
+  const backedUpContent = {};
+  if (detectedFiles.length > 0 && detectedFiles.some(f => f.migratable)) {
+    log('📋 Backing up existing AI files...');
+    for (const file of detectedFiles.filter(f => f.migratable)) {
+      try {
+        backedUpContent[file.file] = {
+          content: readFileSync(file.path, 'utf8'),
+          type: file.type
+        };
+      } catch {
+        // Ignore read errors
+      }
+    }
+  }
+
+  try {
+    // Copy workspace template
+    log('📦 Copying workspace template...');
+    cpSync(templateDir, targetDir, { recursive: true });
+
+    // Rename state.template.json → state.json
+    const templateState = join(targetDir, 'state.template.json');
+    const stateFile = join(targetDir, 'state.json');
+
+    if (existsSync(templateState)) {
+      renameSync(templateState, stateFile);
+    }
+
+    // Rename notifications.template.json → notifications.json and generate unique topic
+    const templateNotify = join(targetDir, 'config', 'notifications.template.json');
+    const notifyFile = join(targetDir, 'config', 'notifications.json');
+
+    if (existsSync(templateNotify)) {
+      renameSync(templateNotify, notifyFile);
+      initNotifications(notifyFile);
+    }
+
+    // Initialize state.json with current user
+    initState(cwd);
+
+    // Update .gitignore
+    updateGitignore(cwd, silent);
+
+    // Handle CLAUDE.md: backup existing and copy PWN template to root
+    let backupInfo = { backed_up: [] };
+    const claudeMdPath = join(cwd, 'CLAUDE.md');
+    const backupClaudeMdPath = join(cwd, '~CLAUDE.md');
+    const templateClaudeMd = join(targetDir, 'agents', 'claude.md');
+
+    // Backup existing CLAUDE.md if present
+    if (backedUpContent['CLAUDE.md'] || backedUpContent['claude.md']) {
+      const originalName = backedUpContent['CLAUDE.md'] ? 'CLAUDE.md' : 'claude.md';
+      const originalPath = join(cwd, originalName);
+
+      if (existsSync(originalPath)) {
+        renameSync(originalPath, backupClaudeMdPath);
+        backupInfo.backed_up.push({ from: originalName, to: '~CLAUDE.md' });
+        if (!silent) {
+          console.log(`📦 Backed up ${originalName} → ~CLAUDE.md`);
+        }
+      }
+    }
+
+    // Copy PWN template to CLAUDE.md in root
+    if (existsSync(templateClaudeMd)) {
+      cpSync(templateClaudeMd, claudeMdPath);
+      if (!silent) {
+        console.log('📝 Created CLAUDE.md with PWN template');
+      }
+    }
+
+    // Backup other AI files (not CLAUDE.md) to .ai/
+    const otherFiles = Object.fromEntries(
+      Object.entries(backedUpContent).filter(([k]) => k.toLowerCase() !== 'claude.md')
+    );
+    if (Object.keys(otherFiles).length > 0) {
+      const otherBackups = backupExistingAIFiles(targetDir, otherFiles, silent);
+      backupInfo.backed_up.push(...otherBackups.backed_up);
+    }
+
+    return {
+      success: true,
+      message: 'PWN workspace injected successfully',
+      path: targetDir,
+      detected: detectedFiles,
+      backed_up: backupInfo.backed_up
+    };
+
+  } catch (error) {
+    return {
+      success: false,
+      error: 'INJECTION_FAILED',
+      message: error.message,
+      detected: detectedFiles
+    };
+  }
+}
+
+/**
+ * Initialize notifications.json with unique topic
+ * @param {string} notifyFile - Path to notifications.json
+ */
+function initNotifications(notifyFile) {
+  try {
+    const content = readFileSync(notifyFile, 'utf8');
+    const config = JSON.parse(content);
+
+    // Generate unique topic ID
+    const uniqueId = randomUUID().split('-')[0]; // First segment: 8 chars
+    config.channels.ntfy.topic = `pwn-${uniqueId}`;
+
+    writeFileSync(notifyFile, JSON.stringify(config, null, 2));
+  } catch {
+    // Ignore errors - notifications will use defaults
+  }
+}
+
+/**
+ * Backup existing AI files to .ai/ root with ~ prefix
+ * - PWN template stays as the base (it's the correct one)
+ * - User's existing files are backed up for manual merge
+ * @param {string} targetDir - Path to .ai/ directory
+ * @param {Object} backedUpContent - Map of filename -> {content, type}
+ * @param {boolean} silent - Suppress output
+ * @returns {Object} Backup info
+ */
+function backupExistingAIFiles(targetDir, backedUpContent, silent = false) {
+  const result = { backed_up: [] };
+
+  try {
+    // Backup all existing AI files with ~ prefix
+    for (const [filename, data] of Object.entries(backedUpContent)) {
+      // Convert filename to backup name: CLAUDE.md -> ~CLAUDE.md, .cursorrules -> ~cursorrules.md
+      const safeName = filename.replace(/^\./, '').replace(/[\/\\]/g, '-');
+      const backupName = `~${safeName}${safeName.endsWith('.md') ? '' : '.md'}`;
+      const backupPath = join(targetDir, backupName);
+
+      writeFileSync(backupPath, data.content);
+
+      result.backed_up.push({ from: filename, to: backupName });
+
+      if (!silent) {
+        console.log(`📦 Backed up ${filename} → .ai/${backupName}`);
+      }
+    }
+
+  } catch (error) {
+    if (!silent) {
+      console.error('⚠️  Backup warning:', error.message);
+    }
+  }
+
+  return result;
+}
+
+/**
+ * Update .gitignore to exclude PWN personal files
+ * @param {string} cwd - Working directory
+ * @param {boolean} silent - Suppress output
+ */
+function updateGitignore(cwd, silent = false) {
+  const gitignorePath = join(cwd, '.gitignore');
+  let gitignoreContent = '';
+
+  if (existsSync(gitignorePath)) {
+    gitignoreContent = readFileSync(gitignorePath, 'utf8');
+  }
+
+  if (!gitignoreContent.includes('.ai/state.json')) {
+    const pwnSection = '\n# PWN\n.ai/state.json\n.ai/config/notifications.json\n';
+    appendFileSync(gitignorePath, pwnSection);
+    if (!silent) {
+      console.log('📝 Updated .gitignore');
+    }
+  }
+}