Function ensureVenv

lib/cli.js:235–321

* Ensure the virtual environment exists and dependencies are installed. * Returns true if all setup steps were already satisfied (fast path). * * @param {object} python - The result of findPython() * @param {boolean} forceRecreate - If true, delete and recreate the venv

(python, forceRecreate)

 * @param {boolean} forceRecreate - If true, delete and recreate the venv
 */
function ensureVenv(python, forceRecreate) {
  mkdirSync(CONFIG_HOME, { recursive: true });

  const marker = readMarker();
  const reqHash = requirementsHash();
  const pyExe = venvPython();

  // Determine if the venv itself needs to be (re)created
  let needsCreate = forceRecreate || !existsSync(pyExe);

  if (!needsCreate && marker) {
    // Recreate if Python major.minor changed
    const markerMinor = marker.python_version;
    const currentMinor = `${python.version.major}.${python.version.minor}`;
    if (markerMinor && markerMinor !== currentMinor) {
      needsCreate = true;
    }

    // Recreate if the recorded python path no longer exists
    if (marker.python_path && !existsSync(marker.python_path)) {
      needsCreate = true;
    }
  }

  let depsUpToDate = false;
  if (!needsCreate && marker && marker.requirements_hash === reqHash) {
    depsUpToDate = true;
  }

  // Fast path: nothing to do
  if (!needsCreate && depsUpToDate) {
    return true;
  }

  // --- Slow path: show setup progress ---

  log('[2/3] Setting up environment...');

  if (needsCreate) {
    if (existsSync(VENV_DIR)) {
      log('      Removing old virtual environment...');
      rmSync(VENV_DIR, { recursive: true, force: true });
    }

    log(`      Creating virtual environment at ~/.autoforge/venv/`);
    const exeParts = python.exe.split(' ');
    try {
      execFileSync(exeParts[0], [...exeParts.slice(1), '-m', 'venv', VENV_DIR], {
        encoding: 'utf8',
        timeout: 120_000,
        stdio: ['pipe', 'pipe', 'pipe'],
      });
    } catch (err) {
      die(`Failed to create virtual environment: ${err.message}`);
    }
  }

  // Install / update dependencies