vibecodingmachine-core 1.0.0

package/src/utils/requirement-helpers.js

@@ -0,0 +1,432 @@
+const fs = require('fs-extra');
+const path = require('path');
+const { getRequirementsPath } = require('./repo-helpers.cjs');
+
+/**
+ * Get ordinal suffix for numbers (1st, 2nd, 3rd, 4th, etc.)
+ * @param {number} num - The number
+ * @returns {string} The ordinal suffix
+ */
+function getOrdinalSuffix(num) {
+  const j = num % 10;
+  const k = num % 100;
+  if (j === 1 && k !== 11) {
+    return 'st';
+  }
+  if (j === 2 && k !== 12) {
+    return 'nd';
+  }
+  if (j === 3 && k !== 13) {
+    return 'rd';
+  }
+  return 'th';
+}
+
+/**
+ * Add or increment TRY AGAIN prefix to requirement text
+ * @param {string} requirementText - The requirement text
+ * @returns {string} Requirement text with TRY AGAIN prefix
+ */
+function addTryAgainPrefix(requirementText) {
+  const tryAgainMatch = requirementText.match(/^TRY AGAIN \((\d+)(?:st|nd|rd|th) time\): (.+)$/);
+
+  if (tryAgainMatch) {
+    const currentCount = parseInt(tryAgainMatch[1]);
+    const baseRequirement = tryAgainMatch[2];
+    return `TRY AGAIN (${currentCount + 1}${getOrdinalSuffix(currentCount + 1)} time): ${baseRequirement}`;
+  } else {
+    return `TRY AGAIN (1st time): ${requirementText}`;
+  }
+}
+
+/**
+ * Parse requirement from markdown line
+ * @param {string} line - Markdown line (e.g., "- Requirement text")
+ * @returns {string} Requirement text without markdown prefix
+ */
+function parseRequirementLine(line) {
+  if (line.startsWith('- ')) {
+    return line.substring(2);
+  }
+  return line;
+}
+
+/**
+ * Move requirement from TO VERIFY section to VERIFIED (CHANGELOG)
+ * @param {string} reqPath - Path to REQUIREMENTS file
+ * @param {string} requirementTitle - Title of requirement to move
+ * @returns {Promise<boolean>} Success status
+ */
+async function promoteToVerified(reqPath, requirementTitle) {
+  try {
+    const content = await fs.readFile(reqPath, 'utf-8');
+    const lines = content.split('\n');
+    const updatedLines = [];
+    let inVerifySection = false;
+    let verifyCount = 0;
+    let requirementToMove = null;
+
+    for (const line of lines) {
+      if (line.includes('## ✅ Verified by AI screenshot')) {
+        inVerifySection = true;
+        updatedLines.push(line);
+        continue;
+      }
+
+      if (inVerifySection && line.startsWith('## ')) {
+        inVerifySection = false;
+        updatedLines.push(line);
+        continue;
+      }
+
+      if (inVerifySection && line.startsWith('- ')) {
+        const reqText = parseRequirementLine(line);
+        if (reqText === requirementTitle || reqText.includes(requirementTitle)) {
+          requirementToMove = reqText;
+          verifyCount++;
+          continue;
+        }
+        verifyCount++;
+      }
+
+      updatedLines.push(line);
+    }
+
+    if (requirementToMove) {
+      // CHANGELOG.md should be at repository root, not in .vibecodingmachine directory
+      const allnightDir = path.dirname(reqPath); // .vibecodingmachine directory
+      const repoRoot = path.dirname(allnightDir); // repository root (one level up)
+      const changelogPath = path.join(repoRoot, 'CHANGELOG.md');
+      const timestamp = new Date().toISOString().split('T')[0];
+      const changelogEntry = `- ${requirementToMove} (${timestamp})`;
+
+      let changelogContent = '';
+      if (await fs.pathExists(changelogPath)) {
+        changelogContent = await fs.readFile(changelogPath, 'utf-8');
+      } else {
+        changelogContent = '# Changelog\n\n## Verified Requirements\n\n';
+      }
+
+      if (changelogContent.includes('## Verified Requirements')) {
+        changelogContent = changelogContent.replace(
+          '## Verified Requirements\n',
+          `## Verified Requirements\n${changelogEntry}\n`
+        );
+      } else {
+        changelogContent += `\n## Verified Requirements\n${changelogEntry}\n`;
+      }
+
+      await fs.writeFile(changelogPath, changelogContent);
+      await fs.writeFile(reqPath, updatedLines.join('\n'));
+      return true;
+    }
+
+    return false;
+  } catch (error) {
+    throw new Error(`Failed to promote requirement to verified: ${error.message}`);
+  }
+}
+
+/**
+ * Move requirement from VERIFIED (CHANGELOG) back to TODO with TRY AGAIN prefix
+ * @param {string} reqPath - Path to REQUIREMENTS file
+ * @param {string} requirementTitle - Title of requirement to move
+ * @returns {Promise<boolean>} Success status
+ */
+async function demoteFromVerifiedToTodo(reqPath, requirementTitle) {
+  try {
+    // CHANGELOG.md should be at repository root, not in .vibecodingmachine directory
+    const allnightDir = path.dirname(reqPath); // .vibecodingmachine directory
+    const repoRoot = path.dirname(allnightDir); // repository root (one level up)
+    const changelogPath = path.join(repoRoot, 'CHANGELOG.md');
+
+    if (!(await fs.pathExists(changelogPath))) {
+      return false;
+    }
+
+    let changelogContent = await fs.readFile(changelogPath, 'utf-8');
+    const changelogLines = changelogContent.split('\n');
+    const updatedChangelogLines = [];
+    let requirementToMove = null;
+
+    for (const line of changelogLines) {
+      if (line.startsWith('- ')) {
+        const lineText = parseRequirementLine(line);
+        // Extract title part (before timestamp in parentheses)
+        const titleMatch = lineText.match(/^(.+?)\s*\([\d-]+\)$/);
+        const lineTitle = titleMatch ? titleMatch[1] : lineText;
+        
+        // Check if this line matches the requirement title
+        // Handle both cases: requirementTitle might include timestamp or not
+        const reqTitleMatch = requirementTitle.match(/^(.+?)\s*\([\d-]+\)$/);
+        const reqTitleOnly = reqTitleMatch ? reqTitleMatch[1] : requirementTitle;
+        
+        if (lineTitle === reqTitleOnly || lineTitle.includes(reqTitleOnly) || reqTitleOnly.includes(lineTitle)) {
+          requirementToMove = lineTitle;
+          continue;
+        }
+      }
+      updatedChangelogLines.push(line);
+    }
+
+    if (!requirementToMove) {
+      return false;
+    }
+
+    await fs.writeFile(changelogPath, updatedChangelogLines.join('\n'));
+
+    const content = await fs.readFile(reqPath, 'utf-8');
+    const lines = content.split('\n');
+    const updatedLines = [];
+    let foundTodoSection = false;
+
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      
+      if (line.includes('## ⏳ Requirements not yet completed')) {
+        foundTodoSection = true;
+        updatedLines.push(line);
+        const requirementText = addTryAgainPrefix(requirementToMove);
+        updatedLines.push(`- ${requirementText}`);
+        continue;
+      }
+
+      updatedLines.push(line);
+    }
+
+    if (!foundTodoSection) {
+      updatedLines.push('## ⏳ Requirements not yet completed');
+      const requirementText = addTryAgainPrefix(requirementToMove);
+      updatedLines.push(`- ${requirementText}`);
+    }
+
+    await fs.writeFile(reqPath, updatedLines.join('\n'));
+    return true;
+  } catch (error) {
+    throw new Error(`Failed to demote requirement from verified: ${error.message}`);
+  }
+}
+
+/**
+ * Move requirement from TODO to TO VERIFY section
+ * @param {string} reqPath - Path to REQUIREMENTS file
+ * @param {string} requirementTitle - Title of requirement to move
+ * @returns {Promise<boolean>} Success status
+ */
+async function promoteTodoToVerify(reqPath, requirementTitle) {
+  try {
+    const content = await fs.readFile(reqPath, 'utf-8');
+    const lines = content.split('\n');
+    
+    // Find the requirement block (### header format)
+    let requirementStartIndex = -1;
+    let requirementEndIndex = -1;
+    let inTodoSection = false;
+    
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i].trim();
+      
+      if (line.includes('## ⏳ Requirements not yet completed')) {
+        inTodoSection = true;
+        continue;
+      }
+      
+      if (inTodoSection && line.startsWith('##') && !line.startsWith('###')) {
+        break;
+      }
+      
+      if (inTodoSection && line.startsWith('###')) {
+        const title = line.replace(/^###\s*/, '').trim();
+        if (title && (title === requirementTitle || title.includes(requirementTitle) || requirementTitle.includes(title))) {
+          requirementStartIndex = i;
+          // Find the end of this requirement (next ### or ## header)
+          for (let j = i + 1; j < lines.length; j++) {
+            const nextLine = lines[j].trim();
+            if (nextLine.startsWith('###') || (nextLine.startsWith('##') && !nextLine.startsWith('###'))) {
+              requirementEndIndex = j;
+              break;
+            }
+          }
+          if (requirementEndIndex === -1) {
+            requirementEndIndex = lines.length;
+          }
+          break;
+        }
+      }
+    }
+    
+    if (requirementStartIndex === -1) {
+      return false;
+    }
+    
+    // Extract the requirement block
+    const requirementBlock = lines.slice(requirementStartIndex, requirementEndIndex);
+    
+    // Remove the requirement from its current location
+    const updatedLines = [
+      ...lines.slice(0, requirementStartIndex),
+      ...lines.slice(requirementEndIndex)
+    ];
+    
+    // Find or create TO VERIFY section
+    const verifySectionVariants = [
+      '## 🔍 TO VERIFY BY HUMAN',
+      '## 🔍 TO VERIFY',
+      '## TO VERIFY',
+      '## ✅ TO VERIFY',
+      '## ✅ Verified by AI screenshot'
+    ];
+    
+    let verifyIndex = -1;
+    for (let i = 0; i < updatedLines.length; i++) {
+      if (verifySectionVariants.some(variant => updatedLines[i].includes(variant))) {
+        verifyIndex = i;
+        break;
+      }
+    }
+    
+    if (verifyIndex === -1) {
+      // Create TO VERIFY section before CHANGELOG or at end
+      const changelogIndex = updatedLines.findIndex(line => line.includes('## CHANGELOG'));
+      const insertIndex = changelogIndex > 0 ? changelogIndex : updatedLines.length;
+      updatedLines.splice(insertIndex, 0, '', '## 🔍 TO VERIFY BY HUMAN', '');
+      verifyIndex = insertIndex + 1;
+    }
+    
+    // Insert requirement block after section header
+    let insertIndex = verifyIndex + 1;
+    while (insertIndex < updatedLines.length && updatedLines[insertIndex].trim() === '') {
+      insertIndex++;
+    }
+    updatedLines.splice(insertIndex, 0, ...requirementBlock);
+    // Add blank line after if needed
+    if (insertIndex + requirementBlock.length < updatedLines.length && updatedLines[insertIndex + requirementBlock.length].trim() !== '') {
+      updatedLines.splice(insertIndex + requirementBlock.length, 0, '');
+    }
+    
+    await fs.writeFile(reqPath, updatedLines.join('\n'));
+    return true;
+  } catch (error) {
+    throw new Error(`Failed to promote requirement from TODO to TO VERIFY: ${error.message}`);
+  }
+}
+
+/**
+ * Move requirement from TO VERIFY back to TODO section
+ * @param {string} reqPath - Path to REQUIREMENTS file
+ * @param {string} requirementTitle - Title of requirement to move
+ * @returns {Promise<boolean>} Success status
+ */
+async function demoteVerifyToTodo(reqPath, requirementTitle) {
+  try {
+    const content = await fs.readFile(reqPath, 'utf-8');
+    const lines = content.split('\n');
+    
+    // Find the requirement block in TO VERIFY section (### header format)
+    let requirementStartIndex = -1;
+    let requirementEndIndex = -1;
+    let inVerifySection = false;
+    
+    const verifySectionVariants = [
+      '## 🔍 TO VERIFY BY HUMAN',
+      '## 🔍 TO VERIFY',
+      '## TO VERIFY',
+      '## ✅ TO VERIFY',
+      '## ✅ Verified by AI screenshot'
+    ];
+    
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i].trim();
+      
+      if (verifySectionVariants.some(variant => line.includes(variant))) {
+        inVerifySection = true;
+        continue;
+      }
+      
+      if (inVerifySection && line.startsWith('##') && !line.startsWith('###')) {
+        break;
+      }
+      
+      if (inVerifySection && line.startsWith('###')) {
+        const title = line.replace(/^###\s*/, '').trim();
+        if (title && (title === requirementTitle || title.includes(requirementTitle) || requirementTitle.includes(title))) {
+          requirementStartIndex = i;
+          // Find the end of this requirement (next ### or ## header)
+          for (let j = i + 1; j < lines.length; j++) {
+            const nextLine = lines[j].trim();
+            if (nextLine.startsWith('###') || (nextLine.startsWith('##') && !nextLine.startsWith('###'))) {
+              requirementEndIndex = j;
+              break;
+            }
+          }
+          if (requirementEndIndex === -1) {
+            requirementEndIndex = lines.length;
+          }
+          break;
+        }
+      }
+    }
+    
+    if (requirementStartIndex === -1) {
+      return false;
+    }
+    
+    // Extract the requirement block
+    const requirementBlock = lines.slice(requirementStartIndex, requirementEndIndex);
+    
+    // Update title with TRY AGAIN prefix
+    const originalTitle = requirementBlock[0].replace(/^###\s*/, '').trim();
+    const titleWithPrefix = addTryAgainPrefix(originalTitle);
+    requirementBlock[0] = `### ${titleWithPrefix}`;
+    
+    // Remove the requirement from its current location
+    const updatedLines = [
+      ...lines.slice(0, requirementStartIndex),
+      ...lines.slice(requirementEndIndex)
+    ];
+    
+    // Find or create TODO section
+    let todoIndex = -1;
+    for (let i = 0; i < updatedLines.length; i++) {
+      if (updatedLines[i].includes('## ⏳ Requirements not yet completed')) {
+        todoIndex = i;
+        break;
+      }
+    }
+    
+    if (todoIndex === -1) {
+      // Create TODO section at the top (after initial headers)
+      const firstSectionIndex = updatedLines.findIndex(line => line.startsWith('##') && !line.startsWith('###'));
+      const insertIndex = firstSectionIndex > 0 ? firstSectionIndex : updatedLines.length;
+      updatedLines.splice(insertIndex, 0, '## ⏳ Requirements not yet completed', '');
+      todoIndex = insertIndex;
+    }
+    
+    // Insert requirement block after section header
+    let insertIndex = todoIndex + 1;
+    while (insertIndex < updatedLines.length && updatedLines[insertIndex].trim() === '') {
+      insertIndex++;
+    }
+    updatedLines.splice(insertIndex, 0, ...requirementBlock);
+    // Add blank line after if needed
+    if (insertIndex + requirementBlock.length < updatedLines.length && updatedLines[insertIndex + requirementBlock.length].trim() !== '') {
+      updatedLines.splice(insertIndex + requirementBlock.length, 0, '');
+    }
+    
+    await fs.writeFile(reqPath, updatedLines.join('\n'));
+    return true;
+  } catch (error) {
+    throw new Error(`Failed to demote requirement from TO VERIFY to TODO: ${error.message}`);
+  }
+}
+
+module.exports = {
+  getOrdinalSuffix,
+  addTryAgainPrefix,
+  parseRequirementLine,
+  promoteToVerified,
+  demoteFromVerifiedToTodo,
+  promoteTodoToVerify,
+  demoteVerifyToTodo
+};
+

package/src/utils/update-checker.js

@@ -0,0 +1,167 @@
+const https = require('https');
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * Check for updates from npm registry
+ * @param {string} packageName - Name of the package to check (e.g., 'allnightai-cli')
+ * @param {string} currentVersion - Current version (e.g., '1.0.0')
+ * @returns {Promise<Object>} Update info or null if no update available
+ */
+async function checkForUpdates(packageName, currentVersion) {
+  return new Promise((resolve, reject) => {
+    const registryUrl = `https://registry.npmjs.org/${packageName}`;
+
+    https.get(registryUrl, (res) => {
+      let data = '';
+
+      res.on('data', (chunk) => {
+        data += chunk;
+      });
+
+      res.on('end', () => {
+        try {
+          const packageInfo = JSON.parse(data);
+          const latestVersion = packageInfo['dist-tags'].latest;
+
+          if (latestVersion !== currentVersion) {
+            const versionInfo = packageInfo.versions[latestVersion];
+            const publishedDate = new Date(packageInfo.time[latestVersion]);
+
+            resolve({
+              hasUpdate: true,
+              currentVersion,
+              latestVersion,
+              publishedDate: publishedDate.toLocaleString('en-US', {
+                year: 'numeric',
+                month: '2-digit',
+                day: '2-digit',
+                hour: '2-digit',
+                minute: '2-digit',
+                timeZoneName: 'short'
+              }),
+              packageName,
+              description: versionInfo.description || ''
+            });
+          } else {
+            resolve({
+              hasUpdate: false,
+              currentVersion,
+              latestVersion,
+              packageName
+            });
+          }
+        } catch (error) {
+          console.error('Error parsing registry response:', error);
+          resolve({ hasUpdate: false, currentVersion, packageName, error: error.message });
+        }
+      });
+    }).on('error', (error) => {
+      console.error('Error checking for updates:', error);
+      resolve({ hasUpdate: false, currentVersion, packageName, error: error.message });
+    });
+  });
+}
+
+/**
+ * Get update info cache file path
+ * @param {string} packageName - Package name
+ * @returns {string} Path to cache file
+ */
+function getUpdateCachePath(packageName) {
+  const homeDir = require('os').homedir();
+  const cacheDir = path.join(homeDir, '.config', 'allnightai');
+
+  // Ensure cache directory exists
+  if (!fs.existsSync(cacheDir)) {
+    fs.mkdirSync(cacheDir, { recursive: true });
+  }
+
+  return path.join(cacheDir, `${packageName}-update-cache.json`);
+}
+
+/**
+ * Load cached update info
+ * @param {string} packageName - Package name
+ * @returns {Object|null} Cached update info or null
+ */
+function loadUpdateCache(packageName) {
+  try {
+    const cachePath = getUpdateCachePath(packageName);
+    if (fs.existsSync(cachePath)) {
+      const data = fs.readFileSync(cachePath, 'utf8');
+      const cache = JSON.parse(data);
+
+      // Cache is valid for 1 hour
+      const cacheAge = Date.now() - cache.checkedAt;
+      if (cacheAge < 3600000) { // 1 hour
+        return cache.updateInfo;
+      }
+    }
+  } catch (error) {
+    console.error('Error loading update cache:', error);
+  }
+  return null;
+}
+
+/**
+ * Save update info to cache
+ * @param {string} packageName - Package name
+ * @param {Object} updateInfo - Update info to cache
+ */
+function saveUpdateCache(packageName, updateInfo) {
+  try {
+    const cachePath = getUpdateCachePath(packageName);
+    const cache = {
+      checkedAt: Date.now(),
+      updateInfo
+    };
+    fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2));
+  } catch (error) {
+    console.error('Error saving update cache:', error);
+  }
+}
+
+/**
+ * Check for updates with caching
+ * @param {string} packageName - Package name
+ * @param {string} currentVersion - Current version
+ * @param {boolean} forceCheck - Force check even if cache is valid
+ * @returns {Promise<Object>} Update info
+ */
+async function checkForUpdatesWithCache(packageName, currentVersion, forceCheck = false) {
+  if (!forceCheck) {
+    const cached = loadUpdateCache(packageName);
+    if (cached) {
+      return cached;
+    }
+  }
+
+  const updateInfo = await checkForUpdates(packageName, currentVersion);
+  saveUpdateCache(packageName, updateInfo);
+  return updateInfo;
+}
+
+/**
+ * Clear update notification (mark as dismissed)
+ * @param {string} packageName - Package name
+ */
+function clearUpdateNotification(packageName) {
+  try {
+    const cachePath = getUpdateCachePath(packageName);
+    if (fs.existsSync(cachePath)) {
+      fs.unlinkSync(cachePath);
+    }
+  } catch (error) {
+    console.error('Error clearing update notification:', error);
+  }
+}
+
+module.exports = {
+  checkForUpdates,
+  checkForUpdatesWithCache,
+  loadUpdateCache,
+  saveUpdateCache,
+  clearUpdateNotification,
+  getUpdateCachePath
+};