Agent generated files.

tools/optimizer/processes/scripts/utils.ts

@@ -0,0 +1,154 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { execSync } from 'node:child_process';
+import fs from 'node:fs/promises';
+import path from 'node:path';
+
+/**
+ * Fetches the current repository owner and name.
+ */
+export function getRepoInfo(): { owner: string; repo: string } {
+  try {
+    const output = execSync('gh repo view --json owner,name', { encoding: 'utf-8' });
+    const data = JSON.parse(output);
+    return {
+      owner: data.owner.login,
+      repo: data.name,
+    };
+  } catch (err) {
+    console.error('Error fetching repo info, falling back to default:', err);
+    return { owner: 'google-gemini', repo: 'gemini-cli' };
+  }
+}
+
+/**
+ * Fetches maintainers from CODEOWNERS file.
+ */
+export async function getMaintainers(): Promise<string[]> {
+  try {
+    const codeownersPath = path.join(process.cwd(), '.github', 'CODEOWNERS');
+    const content = await fs.readFile(codeownersPath, 'utf8');
+    const maintainers = new Set<string>();
+    
+    const lines = content.split('\n');
+    for (const line of lines) {
+      if (line.startsWith('#') || !line.trim()) continue;
+      // Match @user or @org/team
+      const matches = line.match(/@[\w-]+\/[\w-]+|@[\w-]+/g);
+      if (matches) {
+        for (const match of matches) {
+          if (match.includes('/')) {
+            // For team mentions, we might want to expand them, 
+            // but for now let's just skip them or handle them if needed.
+            continue;
+          }
+          maintainers.add(match.replace('@', ''));
+        }
+      }
+    }
+    
+    if (maintainers.size === 0) {
+      return ['gundermanc', 'jackwotherspoon', 'DavidAPierce'];
+    }
+    
+    return Array.from(maintainers);
+  } catch (err) {
+    console.warn('CODEOWNERS not found or unreadable, using fallback maintainers.');
+    return ['gundermanc', 'jackwotherspoon', 'DavidAPierce'];
+  }
+}
+
+/**
+ * Safely updates a CSV file by modifying specific columns for certain rows.
+ */
+export async function updateSimulationCsv(
+  filename: 'issues-after.csv' | 'prs-after.csv',
+  updates: Map<string, Record<string, string>>
+) {
+  if (updates.size === 0) return;
+
+  const filePath = path.join(process.cwd(), filename);
+  const beforeFilePath = path.join(process.cwd(), filename.replace('-after', '-before'));
+  
+  let content = '';
+  try {
+    content = await fs.readFile(filePath, 'utf8');
+  } catch {
+    try {
+      content = await fs.readFile(beforeFilePath, 'utf8');
+    } catch {
+      console.error(`Could not find ${filename} or ${beforeFilePath}`);
+      return;
+    }
+  }
+
+  const lines = content.split('\n');
+  if (lines.length === 0) return;
+
+  const header = lines[0].split(',');
+  const numberIndex = header.indexOf('number');
+  if (numberIndex === -1) {
+    console.error(`CSV ${filename} missing 'number' column`);
+    return;
+  }
+
+  const newLines = [lines[0]];
+  for (let i = 1; i < lines.length; i++) {
+    const rawLine = lines[i].trim();
+    if (!rawLine) continue;
+    
+    // Split by comma but respect quotes
+    const columns: string[] = [];
+    let current = '';
+    let inQuotes = false;
+    for (let charIndex = 0; charIndex < rawLine.length; charIndex++) {
+      const char = rawLine[charIndex];
+      if (char === '"') {
+        inQuotes = !inQuotes;
+        current += char;
+      } else if (char === ',' && !inQuotes) {
+        columns.push(current);
+        current = '';
+      } else {
+        current += char;
+      }
+    }
+    columns.push(current);
+    
+    const number = columns[numberIndex]?.replace(/"/g, '');
+    
+    if (updates.has(number)) {
+      const update = updates.get(number)!;
+      for (const [colName, newValue] of Object.entries(update)) {
+        const colIndex = header.indexOf(colName);
+        if (colIndex !== -1) {
+          columns[colIndex] = newValue;
+        }
+      }
+    }
+    newLines.push(columns.join(','));
+  }
+
+  await fs.writeFile(filePath, newLines.join('\n'));
+}
+
+/**
+ * Executes a gh command with logging and dry-run support.
+ */
+export function execGh(command: string, execute: boolean) {
+  if (!execute) {
+    console.log(`[DRY RUN] Would execute: gh ${command}`);
+    return;
+  }
+
+  console.log(`Executing: gh ${command}`);
+  try {
+    execSync(`gh ${command}`, { stdio: 'inherit' });
+  } catch (err) {
+    console.error(`Failed to execute gh ${command}:`, err);
+  }
+}