@fermindi/pwn-cli 0.1.0 → 0.2.0

package/src/core/inject.js

@@ -1,128 +1,208 @@
-import { existsSync, cpSync, renameSync, readFileSync, appendFileSync, writeFileSync } 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));
-
-/**
- * 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
- * @returns {object} Result with success status and message
- */
-export async function inject(options = {}) {
-  const {
-    cwd = process.cwd(),
-    force = false,
-    silent = false
-  } = options;
-
-  const templateDir = getTemplatePath();
-  const targetDir = join(cwd, '.ai');
-
-  const log = silent ? () => {} : console.log;
-
-  // Check if .ai/ already exists
-  if (existsSync(targetDir) && !force) {
-    return {
-      success: false,
-      error: 'ALREADY_EXISTS',
-      message: '.ai/ directory already exists. Use --force to overwrite.'
-    };
-  }
-
-  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
-    };
-
-  } catch (error) {
-    return {
-      success: false,
-      error: 'INJECTION_FAILED',
-      message: error.message
-    };
-  }
-}
-
-/**
- * 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 } 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
+    };
+  }
+
+  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');
+    }
+  }
+}

package/src/core/state.js

@@ -1,91 +1,91 @@
-import { readFileSync, writeFileSync, existsSync } from 'fs';
-import { join } from 'path';
-import { execSync } from 'child_process';
-
-/**
- * Get the path to state.json in a workspace
- * @param {string} cwd - Working directory (defaults to process.cwd())
- * @returns {string} Path to state.json
- */
-export function getStatePath(cwd = process.cwd()) {
-  return join(cwd, '.ai', 'state.json');
-}
-
-/**
- * Check if a PWN workspace exists in the given directory
- * @param {string} cwd - Working directory
- * @returns {boolean}
- */
-export function hasWorkspace(cwd = process.cwd()) {
-  return existsSync(join(cwd, '.ai'));
-}
-
-/**
- * Get the current git username
- * @returns {string} Git username or 'unknown'
- */
-export function getGitUser() {
-  try {
-    return execSync('git config user.name', { encoding: 'utf8' }).trim();
-  } catch {
-    return 'unknown';
-  }
-}
-
-/**
- * Read the current state from state.json
- * @param {string} cwd - Working directory
- * @returns {object|null} State object or null if not found
- */
-export function getState(cwd = process.cwd()) {
-  const statePath = getStatePath(cwd);
-
-  if (!existsSync(statePath)) {
-    return null;
-  }
-
-  try {
-    const content = readFileSync(statePath, 'utf8');
-    return JSON.parse(content);
-  } catch {
-    return null;
-  }
-}
-
-/**
- * Update the state.json file
- * @param {object} updates - Fields to update
- * @param {string} cwd - Working directory
- * @returns {object} Updated state
- */
-export function updateState(updates, cwd = process.cwd()) {
-  const statePath = getStatePath(cwd);
-  const currentState = getState(cwd) || {};
-
-  const newState = {
-    ...currentState,
-    ...updates,
-    last_updated: new Date().toISOString()
-  };
-
-  writeFileSync(statePath, JSON.stringify(newState, null, 2));
-  return newState;
-}
-
-/**
- * Initialize a new state.json file
- * @param {string} cwd - Working directory
- * @returns {object} Initial state
- */
-export function initState(cwd = process.cwd()) {
-  const state = {
-    developer: getGitUser(),
-    session_started: new Date().toISOString(),
-    current_task: null,
-    context_loaded: []
-  };
-
-  const statePath = getStatePath(cwd);
-  writeFileSync(statePath, JSON.stringify(state, null, 2));
-  return state;
-}
+import { readFileSync, writeFileSync, existsSync } from 'fs';
+import { join } from 'path';
+import { execSync } from 'child_process';
+
+/**
+ * Get the path to state.json in a workspace
+ * @param {string} cwd - Working directory (defaults to process.cwd())
+ * @returns {string} Path to state.json
+ */
+export function getStatePath(cwd = process.cwd()) {
+  return join(cwd, '.ai', 'state.json');
+}
+
+/**
+ * Check if a PWN workspace exists in the given directory
+ * @param {string} cwd - Working directory
+ * @returns {boolean}
+ */
+export function hasWorkspace(cwd = process.cwd()) {
+  return existsSync(join(cwd, '.ai'));
+}
+
+/**
+ * Get the current git username
+ * @returns {string} Git username or 'unknown'
+ */
+export function getGitUser() {
+  try {
+    return execSync('git config user.name', { encoding: 'utf8' }).trim();
+  } catch {
+    return 'unknown';
+  }
+}
+
+/**
+ * Read the current state from state.json
+ * @param {string} cwd - Working directory
+ * @returns {object|null} State object or null if not found
+ */
+export function getState(cwd = process.cwd()) {
+  const statePath = getStatePath(cwd);
+
+  if (!existsSync(statePath)) {
+    return null;
+  }
+
+  try {
+    const content = readFileSync(statePath, 'utf8');
+    return JSON.parse(content);
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * Update the state.json file
+ * @param {object} updates - Fields to update
+ * @param {string} cwd - Working directory
+ * @returns {object} Updated state
+ */
+export function updateState(updates, cwd = process.cwd()) {
+  const statePath = getStatePath(cwd);
+  const currentState = getState(cwd) || {};
+
+  const newState = {
+    ...currentState,
+    ...updates,
+    last_updated: new Date().toISOString()
+  };
+
+  writeFileSync(statePath, JSON.stringify(newState, null, 2));
+  return newState;
+}
+
+/**
+ * Initialize a new state.json file
+ * @param {string} cwd - Working directory
+ * @returns {object} Initial state
+ */
+export function initState(cwd = process.cwd()) {
+  const state = {
+    developer: getGitUser(),
+    session_started: new Date().toISOString(),
+    current_task: null,
+    context_loaded: []
+  };
+
+  const statePath = getStatePath(cwd);
+  writeFileSync(statePath, JSON.stringify(state, null, 2));
+  return state;
+}