@in-the-loop-labs/pair-review 2.0.3 → 2.1.1

package/src/setup/pr-setup.js

@@ -16,7 +16,7 @@ const { GitWorktreeManager } = require('../git/worktree');
 const { GitHubClient } = require('../github/client');
 const { normalizeRepository } = require('../utils/paths');
 const { findMainGitRoot } = require('../local-review');
-const { getConfigDir, getMonorepoPath } = require('../config');
+const { getConfigDir, getMonorepoPath, resolveMonorepoOptions, DEFAULT_CHECKOUT_TIMEOUT_MS } = require('../config');
 const logger = require('../utils/logger');
 const simpleGit = require('simple-git');
 const fs = require('fs').promises;
@@ -202,10 +202,13 @@ async function registerRepositoryLocation(db, currentDir, owner, repo) {
  * @param {number} params.prNumber - PR number (used for worktree lookup)
  * @param {Object} [params.config] - Application config (used for monorepo path lookup)
  * @param {Function} [params.onProgress] - Optional progress callback
- * @returns {Promise<{ repositoryPath: string, knownPath: string|null, worktreeSourcePath: string|null }>}
+ * @returns {Promise<{ repositoryPath: string, knownPath: string|null, worktreeSourcePath: string|null, checkoutScript: string|null, checkoutTimeout: number, worktreeConfig: Object|null }>}
  *   - repositoryPath: the main git root (bare repo or .git parent)
  *   - knownPath: the known path from database (if any)
  *   - worktreeSourcePath: path to use as cwd for `git worktree add` (may be a worktree with sparse-checkout)
+ *   - checkoutScript: path to the checkout script (if configured)
+ *   - checkoutTimeout: timeout in ms for checkout script (default: 300000 = 5 minutes)
+ *   - worktreeConfig: { worktreeBaseDir, nameTemplate } if configured, null otherwise
  */
 async function findRepositoryPath({ db, owner, repo, repository, prNumber, config, onProgress }) {
   const worktreeManager = new GitWorktreeManager(db);
@@ -267,6 +270,19 @@ async function findRepositoryPath({ db, owner, repo, repository, prNumber, confi
     }
   }
 
+  // ------------------------------------------------------------------
+  // Resolve monorepo worktree options (checkout_script, worktree_directory, worktree_name_template)
+  // ------------------------------------------------------------------
+  const resolved = config ? resolveMonorepoOptions(config, repository) : { checkoutScript: null, checkoutTimeout: DEFAULT_CHECKOUT_TIMEOUT_MS, worktreeConfig: null };
+  const { checkoutScript, checkoutTimeout, worktreeConfig } = resolved;
+
+  // When a checkout script is configured, null out worktreeSourcePath —
+  // the script handles all sparse-checkout setup, so we don't want to
+  // inherit from an existing worktree.
+  if (checkoutScript) {
+    worktreeSourcePath = null;
+  }
+
   // ------------------------------------------------------------------
   // Tier 0: Check known local path from repo_settings
   // ------------------------------------------------------------------
@@ -335,7 +351,7 @@ async function findRepositoryPath({ db, owner, repo, repository, prNumber, confi
     }
   }
 
-  return { repositoryPath, knownPath, worktreeSourcePath };
+  return { repositoryPath, knownPath, worktreeSourcePath, checkoutScript, checkoutTimeout, worktreeConfig };
 }
 
 /**
@@ -381,7 +397,7 @@ async function setupPRReview({ db, owner, repo, prNumber, githubToken, config, o
   // Step: repo - Find (or clone) a local repository
   // ------------------------------------------------------------------
   progress({ step: 'repo', status: 'running', message: 'Locating repository...' });
-  const { repositoryPath, knownPath, worktreeSourcePath } = await findRepositoryPath({
+  const { repositoryPath, knownPath, worktreeSourcePath, checkoutScript, checkoutTimeout, worktreeConfig } = await findRepositoryPath({
     db,
     owner,
     repo,
@@ -396,10 +412,10 @@ async function setupPRReview({ db, owner, repo, prNumber, githubToken, config, o
   // Step: worktree - Create git worktree for the PR
   // ------------------------------------------------------------------
   progress({ step: 'worktree', status: 'running', message: 'Setting up git worktree...' });
-  const worktreeManager = new GitWorktreeManager(db);
+  const worktreeManager = new GitWorktreeManager(db, worktreeConfig || {});
   const prInfo = { owner, repo, number: prNumber };
   // Use worktreeSourcePath as cwd for git worktree add (if available) to inherit sparse-checkout
-  const worktreePath = await worktreeManager.createWorktreeForPR(prInfo, prData, repositoryPath, { worktreeSourcePath });
+  const worktreePath = await worktreeManager.createWorktreeForPR(prInfo, prData, repositoryPath, { worktreeSourcePath, checkoutScript, checkoutTimeout });
   progress({ step: 'worktree', status: 'completed', message: `Worktree created at ${worktreePath}` });
 
   // ------------------------------------------------------------------
@@ -415,7 +431,11 @@ async function setupPRReview({ db, owner, repo, prNumber, githubToken, config, o
   //
   // NOTE: prData.changed_files is an INTEGER (count) from the GitHub pulls.get
   // API, not an array. We must fetch the actual file list via pulls.listFiles.
-  if (prData.changed_files > 0) {
+  if (checkoutScript) {
+    // checkout_script handles all sparse-checkout setup — skip built-in expansion
+    logger.info('Skipping built-in sparse-checkout expansion (checkout_script configured)');
+    progress({ step: 'sparse', status: 'completed', message: 'Sparse-checkout managed by checkout_script' });
+  } else if (prData.changed_files > 0) {
     const isSparse = await worktreeManager.isSparseCheckoutEnabled(worktreePath);
     if (isSparse) {
       progress({ step: 'sparse', status: 'running', message: 'Expanding sparse-checkout for PR directories...' });