@in-the-loop-labs/pair-review 1.6.2 → 2.0.1

package/src/database.js

@@ -20,7 +20,7 @@ function getDbPath() {
 /**
  * Current schema version - increment this when adding new migrations
  */
-const CURRENT_SCHEMA_VERSION = 19;
+const CURRENT_SCHEMA_VERSION = 24;
 
 /**
  * Database schema SQL statements
@@ -138,6 +138,7 @@ const SCHEMA_SQL = {
       default_model TEXT,
       default_council_id TEXT,
       default_tab TEXT,
+      default_chat_instructions TEXT,
       local_path TEXT,
       created_at TEXT DEFAULT CURRENT_TIMESTAMP,
       updated_at TEXT DEFAULT CURRENT_TIMESTAMP
@@ -150,6 +151,7 @@ const SCHEMA_SQL = {
       review_id INTEGER NOT NULL,
       provider TEXT,
       model TEXT,
+      tier TEXT,
       custom_instructions TEXT,
       repo_instructions TEXT,
       request_instructions TEXT,
@@ -203,6 +205,47 @@ const SCHEMA_SQL = {
       created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
       updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
     )
+  `,
+
+  chat_sessions: `
+    CREATE TABLE IF NOT EXISTS chat_sessions (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      review_id INTEGER NOT NULL,
+      context_comment_id INTEGER,
+      agent_session_id TEXT, -- Reserved: agent session ID for future reconnection support
+      provider TEXT NOT NULL,
+      model TEXT,
+      status TEXT DEFAULT 'active' CHECK(status IN ('active', 'closed', 'error')),
+      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+      FOREIGN KEY (review_id) REFERENCES reviews(id),
+      FOREIGN KEY (context_comment_id) REFERENCES comments(id)
+    )
+  `,
+
+  chat_messages: `
+    CREATE TABLE IF NOT EXISTS chat_messages (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      session_id INTEGER NOT NULL,
+      role TEXT NOT NULL,
+      type TEXT DEFAULT 'message',
+      content TEXT NOT NULL,
+      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+      FOREIGN KEY (session_id) REFERENCES chat_sessions(id) ON DELETE CASCADE
+    )
+  `,
+
+  context_files: `
+    CREATE TABLE IF NOT EXISTS context_files (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      review_id INTEGER NOT NULL,
+      file TEXT NOT NULL,
+      line_start INTEGER NOT NULL,
+      line_end INTEGER NOT NULL,
+      label TEXT,
+      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+      FOREIGN KEY (review_id) REFERENCES reviews(id) ON DELETE CASCADE
+    )
   `
 };
 
@@ -233,7 +276,12 @@ const INDEX_SQL = [
   'CREATE INDEX IF NOT EXISTS idx_councils_name ON councils(name)',
   // Voice tracking indexes
   'CREATE INDEX IF NOT EXISTS idx_comments_voice ON comments(voice_id)',
-  'CREATE INDEX IF NOT EXISTS idx_comments_is_raw ON comments(is_raw)'
+  'CREATE INDEX IF NOT EXISTS idx_comments_is_raw ON comments(is_raw)',
+  // Chat indexes
+  'CREATE INDEX IF NOT EXISTS idx_chat_sessions_review ON chat_sessions(review_id)',
+  'CREATE INDEX IF NOT EXISTS idx_chat_messages_session ON chat_messages(session_id)',
+  // Context files indexes
+  'CREATE INDEX IF NOT EXISTS idx_context_files_review ON context_files(review_id)'
 ];
 
 /**
@@ -942,6 +990,151 @@ const MIGRATIONS = {
     }
 
     console.log('Migration to schema version 19 complete');
+  },
+
+  // Migration to version 20: adds chat_sessions and chat_messages tables
+  20: (db) => {
+    console.log('Running migration to schema version 20...');
+
+    // Create chat_sessions table if it doesn't exist
+    if (!tableExists(db, 'chat_sessions')) {
+      db.exec(`
+        CREATE TABLE chat_sessions (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          review_id INTEGER NOT NULL,
+          context_comment_id INTEGER,
+          agent_session_id TEXT, -- Reserved: agent session ID for future reconnection support
+          provider TEXT NOT NULL,
+          model TEXT,
+          status TEXT DEFAULT 'active' CHECK(status IN ('active', 'closed', 'error')),
+          created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+          updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+          FOREIGN KEY (review_id) REFERENCES reviews(id),
+          FOREIGN KEY (context_comment_id) REFERENCES comments(id)
+        )
+      `);
+      console.log('  Created chat_sessions table');
+
+      // Create index
+      db.exec('CREATE INDEX IF NOT EXISTS idx_chat_sessions_review ON chat_sessions(review_id)');
+      console.log('  Created index for chat_sessions table');
+    } else {
+      console.log('  Table chat_sessions already exists');
+    }
+
+    // Create chat_messages table if it doesn't exist
+    if (!tableExists(db, 'chat_messages')) {
+      db.exec(`
+        CREATE TABLE chat_messages (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          session_id INTEGER NOT NULL,
+          role TEXT NOT NULL,
+          type TEXT DEFAULT 'message',
+          content TEXT NOT NULL,
+          created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+          FOREIGN KEY (session_id) REFERENCES chat_sessions(id) ON DELETE CASCADE
+        )
+      `);
+      console.log('  Created chat_messages table');
+
+      // Create index
+      db.exec('CREATE INDEX IF NOT EXISTS idx_chat_messages_session ON chat_messages(session_id)');
+      console.log('  Created index for chat_messages table');
+    } else {
+      console.log('  Table chat_messages already exists');
+    }
+
+    console.log('Migration to schema version 20 complete');
+  },
+
+  // Migration to version 21: adds type column to chat_messages for distinguishing context vs message
+  21: (db) => {
+    console.log('Running migration to schema version 21...');
+
+    const hasType = columnExists(db, 'chat_messages', 'type');
+    if (!hasType) {
+      try {
+        db.prepare(`ALTER TABLE chat_messages ADD COLUMN type TEXT DEFAULT 'message'`).run();
+        console.log('  Added type column to chat_messages');
+      } catch (error) {
+        if (!error.message.includes('duplicate column name')) {
+          throw error;
+        }
+        console.log('  Column type already exists (race condition)');
+      }
+    } else {
+      console.log('  Column type already exists');
+    }
+
+    console.log('Migration to schema version 21 complete');
+  },
+
+  22: (db) => {
+    console.log('Migrating to schema version 22: Add tier column to analysis_runs');
+
+    const columns = db.prepare('PRAGMA table_info(analysis_runs)').all();
+    if (!columns.some(c => c.name === 'tier')) {
+      try {
+        db.prepare('ALTER TABLE analysis_runs ADD COLUMN tier TEXT').run();
+        console.log('  Added tier column to analysis_runs');
+      } catch (error) {
+        if (!error.message.includes('duplicate column name')) {
+          throw error;
+        }
+        console.log('  Column tier already exists (race condition)');
+      }
+    } else {
+      console.log('  Column tier already exists');
+    }
+
+    console.log('Migration to schema version 22 complete');
+  },
+
+  23: (db) => {
+    console.log('Migrating to schema version 23: Add default_chat_instructions to repo_settings');
+
+    const columns = db.prepare('PRAGMA table_info(repo_settings)').all();
+    if (!columns.some(c => c.name === 'default_chat_instructions')) {
+      try {
+        db.prepare('ALTER TABLE repo_settings ADD COLUMN default_chat_instructions TEXT').run();
+        console.log('  Added default_chat_instructions column to repo_settings');
+      } catch (error) {
+        if (!error.message.includes('duplicate column name')) {
+          throw error;
+        }
+        console.log('  Column default_chat_instructions already exists (race condition)');
+      }
+    } else {
+      console.log('  Column default_chat_instructions already exists');
+    }
+
+    console.log('Migration to schema version 23 complete');
+  },
+
+  // Migration to version 24: adds context_files table for pinning non-diff file ranges to the diff panel
+  24: (db) => {
+    console.log('Migrating to schema version 24: Add context_files table');
+
+    if (!tableExists(db, 'context_files')) {
+      db.exec(`
+        CREATE TABLE IF NOT EXISTS context_files (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          review_id INTEGER NOT NULL,
+          file TEXT NOT NULL,
+          line_start INTEGER NOT NULL,
+          line_end INTEGER NOT NULL,
+          label TEXT,
+          created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+          FOREIGN KEY (review_id) REFERENCES reviews(id) ON DELETE CASCADE
+        )
+      `);
+      db.exec('CREATE INDEX IF NOT EXISTS idx_context_files_review ON context_files(review_id)');
+      console.log('  Created context_files table');
+    } else {
+      console.log('  Table context_files already exists');
+    }
+
+    console.log('Migration to schema version 24 complete');
   }
 };
 
@@ -1441,7 +1634,7 @@ class RepoSettingsRepository {
    */
   async getRepoSettings(repository) {
     const row = await queryOne(this.db, `
-      SELECT id, repository, default_instructions, default_provider, default_model, default_council_id, default_tab, local_path, created_at, updated_at
+      SELECT id, repository, default_instructions, default_provider, default_model, default_council_id, default_tab, default_chat_instructions, local_path, created_at, updated_at
       FROM repo_settings
       WHERE repository = ? COLLATE NOCASE
     `, [repository]);
@@ -1498,7 +1691,7 @@ class RepoSettingsRepository {
    * @returns {Promise<Object>} Saved settings object
    */
   async saveRepoSettings(repository, settings) {
-    const { default_instructions, default_provider, default_model, default_council_id, default_tab, local_path } = settings;
+    const { default_instructions, default_provider, default_model, default_council_id, default_tab, default_chat_instructions, local_path } = settings;
     const now = new Date().toISOString();
 
     // Check if settings already exist
@@ -1513,6 +1706,7 @@ class RepoSettingsRepository {
             default_model = ?,
             default_council_id = ?,
             default_tab = ?,
+            default_chat_instructions = ?,
             local_path = ?,
             updated_at = ?
         WHERE repository = ? COLLATE NOCASE
@@ -1522,6 +1716,7 @@ class RepoSettingsRepository {
         default_model !== undefined ? default_model : existing.default_model,
         default_council_id !== undefined ? default_council_id : existing.default_council_id,
         default_tab !== undefined ? default_tab : existing.default_tab,
+        default_chat_instructions !== undefined ? default_chat_instructions : existing.default_chat_instructions,
         local_path !== undefined ? local_path : existing.local_path,
         now,
         repository
@@ -1534,15 +1729,16 @@ class RepoSettingsRepository {
         default_model: default_model !== undefined ? default_model : existing.default_model,
         default_council_id: default_council_id !== undefined ? default_council_id : existing.default_council_id,
         default_tab: default_tab !== undefined ? default_tab : existing.default_tab,
+        default_chat_instructions: default_chat_instructions !== undefined ? default_chat_instructions : existing.default_chat_instructions,
         local_path: local_path !== undefined ? local_path : existing.local_path,
         updated_at: now
       };
     } else {
       // Insert new settings
       const result = await run(this.db, `
-        INSERT INTO repo_settings (repository, default_instructions, default_provider, default_model, default_council_id, default_tab, local_path, created_at, updated_at)
-        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
-      `, [repository, default_instructions || null, default_provider || null, default_model || null, default_council_id || null, default_tab || null, local_path || null, now, now]);
+        INSERT INTO repo_settings (repository, default_instructions, default_provider, default_model, default_council_id, default_tab, default_chat_instructions, local_path, created_at, updated_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      `, [repository, default_instructions || null, default_provider || null, default_model || null, default_council_id || null, default_tab || null, default_chat_instructions || null, local_path || null, now, now]);
 
       return {
         id: result.lastID,
@@ -1552,6 +1748,7 @@ class RepoSettingsRepository {
         default_model: default_model || null,
         default_council_id: default_council_id || null,
         default_tab: default_tab || null,
+        default_chat_instructions: default_chat_instructions || null,
         local_path: local_path || null,
         created_at: now,
         updated_at: now
@@ -2159,7 +2356,8 @@ class ReviewRepository {
   async getReview(id) {
     const row = await queryOne(this.db, `
       SELECT id, pr_number, repository, status, review_id,
-             created_at, updated_at, submitted_at, review_data, custom_instructions, summary
+             created_at, updated_at, submitted_at, review_data, custom_instructions, summary,
+             review_type, local_path, local_head_sha
       FROM reviews
       WHERE id = ?
     `, [id]);
@@ -2631,14 +2829,14 @@ class AnalysisRunRepository {
    * @param {string} [runInfo.status='running'] - Initial status (default 'running'; pass 'completed' for externally-produced results)
    * @returns {Promise<Object>} Created analysis run record
    */
-  async create({ id, reviewId, provider = null, model = null, customInstructions = null, repoInstructions = null, requestInstructions = null, headSha = null, status = 'running', parentRunId = null, configType = 'single', levelsConfig = null }) {
+  async create({ id, reviewId, provider = null, model = null, tier = null, customInstructions = null, repoInstructions = null, requestInstructions = null, headSha = null, status = 'running', parentRunId = null, configType = 'single', levelsConfig = null }) {
     const isTerminal = ['completed', 'failed', 'cancelled'].includes(status);
     const completedAt = isTerminal ? 'CURRENT_TIMESTAMP' : 'NULL';
     const levelsConfigJson = levelsConfig ? JSON.stringify(levelsConfig) : null;
     await run(this.db, `
-      INSERT INTO analysis_runs (id, review_id, provider, model, custom_instructions, repo_instructions, request_instructions, head_sha, status, completed_at, parent_run_id, config_type, levels_config)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ${completedAt}, ?, ?, ?)
-    `, [id, reviewId, provider, model, customInstructions, repoInstructions, requestInstructions, headSha, status, parentRunId, configType, levelsConfigJson]);
+      INSERT INTO analysis_runs (id, review_id, provider, model, tier, custom_instructions, repo_instructions, request_instructions, head_sha, status, completed_at, parent_run_id, config_type, levels_config)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ${completedAt}, ?, ?, ?)
+    `, [id, reviewId, provider, model, tier, customInstructions, repoInstructions, requestInstructions, headSha, status, parentRunId, configType, levelsConfigJson]);
 
     // Query back the inserted row to return actual database values (including timestamps)
     return await this.getById(id);
@@ -2713,7 +2911,7 @@ class AnalysisRunRepository {
    */
   async getById(id) {
     const row = await queryOne(this.db, `
-      SELECT id, review_id, provider, model, custom_instructions, repo_instructions, request_instructions,
+      SELECT id, review_id, provider, model, tier, custom_instructions, repo_instructions, request_instructions,
              head_sha, summary, status, total_suggestions, files_analyzed, started_at, completed_at,
              parent_run_id, config_type, levels_config
       FROM analysis_runs
@@ -2734,7 +2932,7 @@ class AnalysisRunRepository {
   async getByReviewId(reviewId, { limit } = {}) {
     const params = [reviewId];
     let sql = `
-      SELECT id, review_id, provider, model, custom_instructions, repo_instructions, request_instructions,
+      SELECT id, review_id, provider, model, tier, custom_instructions, repo_instructions, request_instructions,
              head_sha, summary, status, total_suggestions, files_analyzed, started_at, completed_at,
              parent_run_id, config_type, levels_config
       FROM analysis_runs
@@ -2764,7 +2962,7 @@ class AnalysisRunRepository {
    */
   async getChildRuns(parentRunId) {
     return query(this.db, `
-      SELECT id, review_id, provider, model, custom_instructions, repo_instructions, request_instructions,
+      SELECT id, review_id, provider, model, tier, custom_instructions, repo_instructions, request_instructions,
              head_sha, summary, status, total_suggestions, files_analyzed, started_at, completed_at,
              parent_run_id, config_type, levels_config
       FROM analysis_runs
@@ -3128,6 +3326,114 @@ class CouncilRepository {
   }
 }
 
+/**
+ * ContextFileRepository class for managing context file range records.
+ * Context files allow pinning specific line ranges from non-diff files
+ * into the diff panel for review.
+ */
+class ContextFileRepository {
+  /**
+   * Create a new ContextFileRepository instance
+   * @param {Database} db - Database instance
+   */
+  constructor(db) {
+    this.db = db;
+  }
+
+  /**
+   * Add a context file range for a review
+   * @param {number} reviewId - Review ID
+   * @param {string} file - File path
+   * @param {number} lineStart - Start line number
+   * @param {number} lineEnd - End line number
+   * @param {string|null} [label=null] - Optional label for the range
+   * @returns {Promise<Object>} The newly created context file record
+   */
+  async add(reviewId, file, lineStart, lineEnd, label = null) {
+    const result = await run(this.db, `
+      INSERT INTO context_files (review_id, file, line_start, line_end, label)
+      VALUES (?, ?, ?, ?, ?)
+    `, [reviewId, file, lineStart, lineEnd, label]);
+
+    return queryOne(this.db, `
+      SELECT id, review_id, file, line_start, line_end, label, created_at
+      FROM context_files
+      WHERE id = ?
+    `, [result.lastID]);
+  }
+
+  /**
+   * Get all context file ranges for a review, ordered by id
+   * @param {number} reviewId - Review ID
+   * @returns {Promise<Array<Object>>} Array of context file records
+   */
+  async getByReviewId(reviewId) {
+    return query(this.db, `
+      SELECT id, review_id, file, line_start, line_end, label, created_at
+      FROM context_files
+      WHERE review_id = ?
+      ORDER BY id
+    `, [reviewId]);
+  }
+
+  /**
+   * Get context file ranges for a specific file within a review, ordered by line_start
+   * @param {number} reviewId - Review ID
+   * @param {string} file - File path
+   * @returns {Promise<Array<Object>>} Array of context file records
+   */
+  async getByReviewIdAndFile(reviewId, file) {
+    return query(this.db, `
+      SELECT id, review_id, file, line_start, line_end, label, created_at
+      FROM context_files
+      WHERE review_id = ? AND file = ?
+      ORDER BY line_start
+    `, [reviewId, file]);
+  }
+
+  /**
+   * Update the line range of an existing context file record
+   * @param {number} id - Context file record ID
+   * @param {number} lineStart - New start line number
+   * @param {number} lineEnd - New end line number
+   * @returns {Promise<boolean>} True if record was updated
+   */
+  async updateRange(id, lineStart, lineEnd) {
+    const result = await run(this.db, `
+      UPDATE context_files SET line_start = ?, line_end = ? WHERE id = ?
+    `, [lineStart, lineEnd, id]);
+
+    return result.changes > 0;
+  }
+
+  /**
+   * Remove a context file range by ID, scoped to a specific review
+   * @param {number} id - Context file record ID
+   * @param {number} reviewId - Review ID (ensures deletion is scoped to the correct review)
+   * @returns {Promise<boolean>} True if record was deleted
+   */
+  async remove(id, reviewId) {
+    const result = await run(this.db, `
+      DELETE FROM context_files WHERE id = ? AND review_id = ?
+    `, [id, reviewId]);
+
+    return result.changes > 0;
+  }
+
+  /**
+   * Remove all context file ranges for a review
+   * @param {number} reviewId - Review ID
+   * @returns {Promise<number>} Number of records deleted
+   */
+  async removeAll(reviewId) {
+    const result = await run(this.db, `
+      DELETE FROM context_files WHERE review_id = ?
+    `, [reviewId]);
+
+    return result.changes;
+  }
+}
+
 module.exports = {
   initializeDatabase,
   closeDatabase,
@@ -3150,6 +3456,7 @@ module.exports = {
   AnalysisRunRepository,
   GitHubReviewRepository,
   CouncilRepository,
+  ContextFileRepository,
   generateWorktreeId,
   migrateExistingWorktrees
 };

package/src/main.js

@@ -14,6 +14,7 @@ const { normalizeRepository, resolveRenamedFile, resolveRenamedFileOld } = requi
 const logger = require('./utils/logger');
 const simpleGit = require('simple-git');
 const { getGeneratedFilePatterns } = require('./git/gitattributes');
+const { getEmoji: getCategoryEmoji } = require('./utils/category-emoji');
 const open = (...args) => import('open').then(({default: open}) => open(...args));
 
 let db = null;
@@ -554,7 +555,7 @@ async function handlePullRequest(args, config, db, flags = {}) {
             await new Promise(resolve => setTimeout(resolve, retryDelay * attempt));
           }
 
-          const response = await fetch(`http://localhost:${port}/api/analyze/${prInfo.owner}/${prInfo.repo}/${prInfo.number}`, {
+          const response = await fetch(`http://localhost:${port}/api/pr/${prInfo.owner}/${prInfo.repo}/${prInfo.number}/analyses`, {
             method: 'POST',
             headers: { 'Content-Type': 'application/json' }
           });
@@ -657,20 +658,6 @@ async function startServerWithPRContext(config, prInfo, flags = {}) {
   return actualPort;
 }
 
-/**
- * Category to emoji mapping for AI suggestions
- */
-const CATEGORY_EMOJI_MAP = {
-  'bug': '🐛',
-  'performance': '⚡',
-  'design': '📐',
-  'code-style': '🧹',
-  'improvement': '💡',
-  'praise': '⭐',
-  'security': '🔒',
-  'suggestion': '💬'
-};
-
 /**
  * Format AI suggestion with emoji and category prefix
  * @param {string} text - The suggestion text
@@ -681,7 +668,7 @@ function formatAISuggestion(text, category) {
   if (!category) {
     return text;
   }
-  const emoji = CATEGORY_EMOJI_MAP[category] || '💬';
+  const emoji = getCategoryEmoji(category);
   // Properly capitalize hyphenated categories (e.g., "code-style" -> "Code Style")
   const capitalizedCategory = category
     .split('-')
@@ -734,6 +721,7 @@ async function performHeadlessReview(args, config, db, flags, options) {
     let worktreePath;
     let diff;
     let changedFiles;
+    const repository = normalizeRepository(prInfo.owner, prInfo.repo);
 
     // Determine working directory: --use-checkout uses current directory
     if (flags.useCheckout) {
@@ -803,7 +791,6 @@ async function performHeadlessReview(args, config, db, flags, options) {
       } else {
         // Current directory is not the target repository - find or clone it
         console.log(`Current directory is not a checkout of ${prInfo.owner}/${prInfo.repo}, locating repository...`);
-        const repository = normalizeRepository(prInfo.owner, prInfo.repo);
         const result = await findRepositoryPath({
           db,
           owner: prInfo.owner,