@ateriss_/aiv-cli 0.1.0

package/src/context/tree.ts

@@ -0,0 +1,90 @@
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { TreeNode } from '../types';
+
+const IGNORED = new Set([
+  'node_modules', '.git', 'dist', 'build', '.next', '.nuxt', 'coverage',
+  '.aiv', '.cache', 'vendor', '__pycache__', '.venv', 'venv', '.DS_Store',
+]);
+
+const MODULE_TYPE_MAP: Array<[RegExp, TreeNode['module_type']]> = [
+  [/auth|login|session|jwt|oauth|passport/i, 'auth'],
+  [/sql|database|db|migration|seed|schema|prisma|typeorm|sequelize|knex/i, 'sql'],
+  [/component|view|page|template|ui|frontend|client|web/i, 'frontend'],
+  [/service|controller|handler|route|api|endpoint|backend|server/i, 'backend'],
+  [/infra|terraform|k8s|kubernetes|helm|docker|deploy|ci|cd/i, 'infra'],
+  [/config|setting|env|constant/i, 'config'],
+  [/test|spec|__test__|__mock__|fixture/i, 'test'],
+];
+
+const SENSITIVITY_MAP: Array<[RegExp, TreeNode['sensitivity']]> = [
+  [/auth|login|password|jwt|oauth|session|token|secret|crypto|encrypt/i, 'high'],
+  [/payment|billing|payroll|salary|invoice|stripe|financial/i, 'high'],
+  [/admin|permission|role|acl|rbac|sudo/i, 'high'],
+  [/migration|seed|database|schema/i, 'medium'],
+  [/service|controller|api|route/i, 'medium'],
+  [/test|spec|mock|fixture|config/i, 'low'],
+];
+
+export async function buildTree(cwd: string, maxDepth: number = 4): Promise<TreeNode> {
+  return scanDir(cwd, cwd, maxDepth);
+}
+
+function scanDir(root: string, dir: string, maxDepth: number, depth: number = 0): TreeNode {
+  const name = path.basename(dir);
+  const rel = path.relative(root, dir) || '.';
+
+  const node: TreeNode = {
+    path: rel,
+    type: 'directory',
+    module_type: detectModuleType(name),
+    sensitivity: detectSensitivity(name),
+    children: [],
+  };
+
+  if (depth >= maxDepth) return node;
+
+  let entries: fs.Dirent[];
+  try {
+    entries = fs.readdirSync(dir, { withFileTypes: true });
+  } catch {
+    return node;
+  }
+
+  for (const entry of entries) {
+    if (IGNORED.has(entry.name)) continue;
+
+    const fullPath = path.join(dir, entry.name);
+    if (entry.isDirectory()) {
+      node.children!.push(scanDir(root, fullPath, maxDepth, depth + 1));
+    } else if (depth < maxDepth) {
+      node.children!.push({
+        path: path.relative(root, fullPath),
+        type: 'file',
+        module_type: detectModuleType(entry.name),
+        sensitivity: detectSensitivity(entry.name),
+      });
+    }
+  }
+
+  // Limit children to keep tree.json manageable
+  if (node.children!.length > 50) {
+    node.children = node.children!.slice(0, 50);
+  }
+
+  return node;
+}
+
+function detectModuleType(name: string): TreeNode['module_type'] {
+  for (const [pattern, type] of MODULE_TYPE_MAP) {
+    if (pattern.test(name)) return type;
+  }
+  return 'other';
+}
+
+function detectSensitivity(name: string): TreeNode['sensitivity'] {
+  for (const [pattern, level] of SENSITIVITY_MAP) {
+    if (pattern.test(name)) return level;
+  }
+  return 'low';
+}

package/src/git/github.ts

@@ -0,0 +1,112 @@
+import { PullRequest, PRDiff, PRFile } from '../types';
+
+const GITHUB_API = 'https://api.github.com';
+
+export class GithubClient {
+  private headers: Record<string, string>;
+
+  constructor(token: string) {
+    this.headers = {
+      Authorization: `Bearer ${token}`,
+      Accept: 'application/vnd.github.v3+json',
+      'X-GitHub-Api-Version': '2022-11-28',
+      'User-Agent': 'aiv-cli/0.1.0',
+    };
+  }
+
+  async listPRs(owner: string, repo: string, limit: number = 20): Promise<PullRequest[]> {
+    const url = `${GITHUB_API}/repos/${owner}/${repo}/pulls?state=open&per_page=${Math.min(limit, 100)}&sort=updated&direction=desc`;
+    const res = await this.fetch(url);
+    if (!res.ok) await this.throwError(res, 'list PRs');
+
+    const data = await res.json() as any[];
+    return data.map(pr => this.mapPR(pr));
+  }
+
+  async getPR(owner: string, repo: string, prNumber: number): Promise<PullRequest> {
+    const url = `${GITHUB_API}/repos/${owner}/${repo}/pulls/${prNumber}`;
+    const res = await this.fetch(url);
+    if (!res.ok) await this.throwError(res, `get PR #${prNumber}`);
+    return this.mapPR(await res.json() as any);
+  }
+
+  async getPRFiles(owner: string, repo: string, prNumber: number): Promise<PRFile[]> {
+    const url = `${GITHUB_API}/repos/${owner}/${repo}/pulls/${prNumber}/files?per_page=100`;
+    const res = await this.fetch(url);
+    if (!res.ok) await this.throwError(res, 'get PR files');
+
+    const data = await res.json() as any[];
+    return data.map(f => ({
+      filename: f.filename,
+      status: f.status,
+      additions: f.additions,
+      deletions: f.deletions,
+      patch: f.patch,
+    }));
+  }
+
+  async getPRDiff(owner: string, repo: string, prNumber: number): Promise<PRDiff> {
+    const [pr, files] = await Promise.all([
+      this.getPR(owner, repo, prNumber),
+      this.getPRFiles(owner, repo, prNumber),
+    ]);
+
+    const rawDiff = files
+      .filter(f => f.patch)
+      .map(f => `diff --git a/${f.filename} b/${f.filename}\n${f.patch}`)
+      .join('\n');
+
+    return { pr, files, rawDiff };
+  }
+
+  async submitReview(
+    owner: string,
+    repo: string,
+    prNumber: number,
+    event: 'APPROVE' | 'REQUEST_CHANGES',
+    body: string = '',
+  ): Promise<void> {
+    const url = `${GITHUB_API}/repos/${owner}/${repo}/pulls/${prNumber}/reviews`;
+    const { default: fetch } = await import('node-fetch');
+    const res = await fetch(url, {
+      method: 'POST',
+      headers: { ...this.headers, 'Content-Type': 'application/json' },
+      body: JSON.stringify({ event, body }),
+    }) as unknown as Response;
+    if (!res.ok) await this.throwError(res, `submit review on PR #${prNumber}`);
+  }
+
+  private mapPR(pr: any): PullRequest {
+    return {
+      id: pr.id,
+      number: pr.number,
+      title: pr.title,
+      author: pr.user?.login ?? 'unknown',
+      branch: pr.head?.ref ?? '',
+      base: pr.base?.ref ?? '',
+      url: pr.html_url,
+      state: pr.state,
+      createdAt: pr.created_at,
+      updatedAt: pr.updated_at,
+      additions: pr.additions ?? 0,
+      deletions: pr.deletions ?? 0,
+      changedFiles: pr.changed_files ?? 0,
+      labels: (pr.labels ?? []).map((l: any) => l.name),
+      description: pr.body ?? undefined,
+    };
+  }
+
+  private async fetch(url: string): Promise<Response> {
+    const { default: fetch } = await import('node-fetch');
+    return fetch(url, { headers: this.headers }) as unknown as Response;
+  }
+
+  private async throwError(res: Response, action: string): Promise<never> {
+    let msg = `GitHub API error (${res.status}) while trying to ${action}`;
+    try {
+      const body = await res.json() as any;
+      if (body.message) msg += `: ${body.message}`;
+    } catch {}
+    throw new Error(msg);
+  }
+}

package/src/git/utils.ts

@@ -0,0 +1,51 @@
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { execSync } from 'node:child_process';
+
+export interface RepoInfo {
+  owner: string;
+  repo: string;
+}
+
+export function detectRepoInfo(cwd: string): RepoInfo | null {
+  try {
+    const remoteUrl = execSync('git remote get-url origin', { cwd, stdio: 'pipe' }).toString().trim();
+    return parseGithubUrl(remoteUrl);
+  } catch {
+    return null;
+  }
+}
+
+export function parseGithubUrl(url: string): RepoInfo | null {
+  const sshMatch = /github\.com[:/]([^/]+)\/([^/.]+)/.exec(url);
+  if (sshMatch) return { owner: sshMatch[1], repo: sshMatch[2] };
+
+  const httpsMatch = /github\.com\/([^/]+)\/([^/.]+)/.exec(url);
+  if (httpsMatch) return { owner: httpsMatch[1], repo: httpsMatch[2] };
+
+  return null;
+}
+
+export function appendGitignore(cwd: string): void {
+  const gitignorePath = path.join(cwd, '.gitignore');
+  const entry = '.aiv/';
+
+  if (!fs.existsSync(gitignorePath)) {
+    fs.writeFileSync(gitignorePath, entry + '\n', 'utf8');
+    return;
+  }
+
+  const content = fs.readFileSync(gitignorePath, 'utf8');
+  if (!content.includes('.aiv')) {
+    fs.appendFileSync(gitignorePath, `\n# aiv local context\n${entry}\n`, 'utf8');
+  }
+}
+
+export function isGitRepo(cwd: string): boolean {
+  try {
+    execSync('git rev-parse --git-dir', { cwd, stdio: 'pipe' });
+    return true;
+  } catch {
+    return false;
+  }
+}

package/src/i18n/en.ts

@@ -0,0 +1,203 @@
+export const en = {
+  // ── General ────────────────────────────────────────────────────────────────
+  notInitialized: 'Not initialized. Run `aiv init` first.',
+  invalidProvider: 'Invalid provider. Choose: claude, openai, mock',
+  invalidPrNumber: 'Invalid PR number.',
+  repoNotDetected: 'Could not detect GitHub owner/repo. Use --owner and --repo flags or configure in .aiv/config.yml',
+
+  // ── init ───────────────────────────────────────────────────────────────────
+  initAlreadyDone: 'aiv is already initialized. Use --force to reinitialize.',
+  initTitle: '  aiv — AI PR Reviewer\n',
+  initWritingGlobalConfig: 'Writing global config (~/.aiv/config.yml)...',
+  initGlobalConfigCreated: 'Global config created (~/.aiv/config.yml)',
+  initGlobalConfigExists: 'Global config already exists (~/.aiv/config.yml)',
+  initWritingConfig: 'Writing repo config (.aiv/config.yml)...',
+  initConfigCreated: 'Repo config created (.aiv/config.yml)',
+  initRulesCreated: 'rules.yml created',
+  initScanningTree: 'Scanning project structure...',
+  initTreeCreated: 'tree.json created',
+  initTreeSkipped: 'tree.json skipped (could not scan)',
+  initBuildingContext: 'Building project context...',
+  initContextCreated: 'context.md created',
+  initContextSkipped: 'context.md skipped (could not analyze)',
+  initGitignoreUpdated: '.gitignore updated',
+  initSuccessTitle: '\n  Initialized! Next steps:\n',
+  initStep1: (envVar: string) => `  Set your API key:    export ${envVar}=your-key`,
+  initStep2: (envVar: string) => `  Set GitHub token:    export ${envVar}=your-token`,
+  initStep3: '  List PRs:            aiv prs',
+  initStep4: '  Review a PR:         aiv review <pr-number>',
+  initEditContext: (path: string) => `  Edit ${path} to add business context.`,
+  initEditRules: (path: string) => `  Edit ${path} to define your rules.`,
+  initGlobalHint: '  Global config: ~/.aiv/config.yml',
+  initAddAccountHint: '  Add accounts:  aiv config add-account <name> --token-env <VAR>',
+
+  // ── selector ───────────────────────────────────────────────────────────────
+  selectorSelectPR: 'Select a PR to review:',
+  selectorConfirmReview: 'Review',
+  selectorCancelled: '  Cancelled.\n',
+
+  // ── prs ────────────────────────────────────────────────────────────────────
+  prsFetching: (repo: string) => `Fetching PRs for ${repo}...`,
+  prsNoneFound: '\n  No open pull requests found.\n',
+  prsColPR: 'PR',
+  prsColTitle: 'Title',
+  prsColAuthor: 'Author',
+  prsColBranch: 'Branch',
+  prsColChanges: 'Changes',
+  prsColCreated: 'Created',
+  prsFooter: (count: number) => `  Showing ${count} open PR(s). Run aiv review <pr-number> to analyze.\n`,
+  prsMissingToken: (envVar: string) => `Missing env var: ${envVar}. Set your GitHub token.`,
+  prsFailed: (msg: string) => `Failed: ${msg}`,
+
+  // ── review ─────────────────────────────────────────────────────────────────
+  reviewTitle: (n: number) => `\n  aiv review — PR #${n}\n`,
+  reviewFetching: (n: number, repo: string) => `Fetching PR #${n} from ${repo}...`,
+  reviewLoaded: (title: string, files: number) => `PR loaded: ${title} (${files} files)`,
+  reviewFetchFailed: (msg: string) => `Failed to fetch PR: ${msg}`,
+  reviewLoadingContext: 'Loading project context...',
+  reviewContextLoaded: 'Context loaded',
+  reviewRunningAgents: (agents: string) => `\n  Running agents: ${agents}\n`,
+  reviewFailed: (msg: string) => `\n  Review failed: ${msg}\n`,
+  reviewAccount: (name: string, envVar: string) => `Account: ${name} (${envVar})`,
+
+  // ── context ────────────────────────────────────────────────────────────────
+  contextRefreshTitle: '\n  Refreshing context...\n',
+  contextScanningTree: 'Scanning project structure...',
+  contextTreeUpdated: 'tree.json updated',
+  contextTreeFailed: (msg: string) => `tree.json failed: ${msg}`,
+  contextRebuildingCtx: 'Rebuilding context.md...',
+  contextCtxUpdated: 'context.md updated',
+  contextCtxFailed: (msg: string) => `context.md failed: ${msg}`,
+  contextEditHint: (path: string) => `\n  Edit ${path} to add custom context.\n`,
+  contextNoFile: 'No context.md found. Run `aiv context refresh`.',
+
+  // ── config ─────────────────────────────────────────────────────────────────
+  configGlobalTitle: '  ~/.aiv/config.yml  (global)',
+  configNotCreated: '  (not created yet)',
+  configRepoConfigTitle: '  .aiv/config.yml  (repo)',
+  configRulesTitle: '  .aiv/rules.yml',
+  configProviderSet: (p: string) => `  Default provider set to: ${p}`,
+  configRepoSet: (o: string, r: string) => `  Repo set to: ${o}/${r}`,
+  configNoRules: '  No rules.yml found.',
+  configLangSet: (lang: string) => `  Language set to: ${lang}`,
+  configInvalidLang: 'Invalid language. Choose: en, es',
+  configTokenEnvHint: (v: string) => `  Token env var: ${v}`,
+  configUsernameHint: (u: string) => `  Username: ${u}`,
+  configRepoAccountHint: (a: string) => `  Repo account:   ${a}`,
+  configSavedToRepo: '  Saved to .aiv/config.yml',
+
+  // ── accounts ───────────────────────────────────────────────────────────────
+  accountsTitle: '\n  GitHub Accounts\n',
+  accountsNone: '  No accounts configured. Add one with: aiv config add-account <name>',
+  accountsColName: 'Name',
+  accountsColUser: 'Username',
+  accountsColEnvVar: 'Token Env Var',
+  accountsColToken: 'Token',
+  accountsColDesc: 'Description',
+  accountsColDefault: 'Default',
+  accountsTokenFound: 'found',
+  accountsTokenMissing: 'missing',
+  accountsDefaultMark: '✔ default',
+  accountsAdded: (name: string) => `  Account "${name}" added to global config.`,
+  accountsRemoved: (name: string) => `  Account "${name}" removed.`,
+  accountsDefaultSet: (name: string) => `  Default account set to: ${name}`,
+  accountsRepoSet: (name: string) => `  This repo will use account: ${name}`,
+  accountsNotFound: (name: string) => `  Account "${name}" not found.`,
+  accountsAlreadyExists: (name: string) => `  Account "${name}" already exists. Use --force to overwrite.`,
+  accountsGlobalHint: '\n  Global config: ~/.aiv/config.yml',
+  accountsRepoHint: '  Repo config:   .aiv/config.yml\n',
+
+  // ── orchestrator ───────────────────────────────────────────────────────────
+  orchestratorRunning: (agent: string) => `  Running ${agent} agent...`,
+  orchestratorDone: (agent: string, count: number, score: number) => `  ${agent} — ${count} finding(s) [score: ${score}]`,
+  orchestratorAgentFailed: (agent: string, msg: string) => `  ${agent} failed: ${msg}`,
+  orchestratorFailedMsg: (msg: string) => `Agent failed: ${msg}`,
+  orchestratorFailedUnexpected: 'Agent failed unexpectedly.',
+  orchestratorOverallRisk: (label: string, score: number, total: number, agents: number) =>
+    `Overall risk: ${label} (${score}/100). ${total} finding(s) across ${agents} agent(s).`,
+  orchestratorCriticalFound: (count: number) => `${count} high/critical issue(s) require attention.`,
+  orchestratorNoCritical: 'No critical issues detected.',
+
+  // ── renderer ───────────────────────────────────────────────────────────────
+  renderRiskScore: 'Risk Score:',
+  renderGenerated: 'Generated:',
+  renderExecutiveSummary: '  Executive Summary',
+  renderSecurityIssues: '  Security Issues',
+  renderBusinessRisks: '  Business Risks',
+  renderArchitectureIssues: '  Architecture Issues',
+  renderRegressions: '  Possible Regressions',
+  renderAgentSummaries: '  Agent Summaries',
+  renderReviewTitle: (n: number, title: string) => `  Review: PR #${n} — ${title}`,
+  renderSuggestion: 'Suggestion:',
+  severityCritical: 'CRITICAL',
+  severityHigh: 'HIGH',
+  severityMedium: 'MEDIUM',
+  severityLow: 'LOW',
+  severityInfo: 'INFO',
+
+  // ── errors from config layer ────────────────────────────────────────────────
+  errorMissingToken: (envVar: string, account: string) => `Missing env var: ${envVar} (account: ${account})`,
+  errorAccountNotFound: (name: string) => `Account "${name}" not found.`,
+  errorAccountNotFoundGlobal: (name: string) =>
+    `Account "${name}" not found in global config. Add it first with: aiv config add-account ${name}`,
+
+  // ── custom providers ──────────────────────────────────────────────────────
+  customProviderAdded: (name: string) => `  Provider "${name}" added to global config.`,
+  customProviderRemoved: (name: string) => `  Provider "${name}" removed.`,
+  customProviderNotFound: (name: string) => `  Provider "${name}" not found in custom_providers.`,
+  customProviderAlreadyExists: (name: string) => `  Provider "${name}" already exists. Use --force to overwrite.`,
+  customProviderBaseUrlRequired: '  --base-url is required for custom (non-built-in) providers.',
+  customProviderTitle: '\n  Custom Providers (OpenAI-compatible)\n',
+  customProviderNone: '  No custom providers configured. Add one with: aiv config add-provider <name> --base-url <url> --api-key-env <VAR> --model <model>',
+  customProviderColName: 'Name',
+  customProviderColUrl: 'Base URL',
+  customProviderColModel: 'Default Model',
+  customProviderColEnvVar: 'Token Env Var',
+  customProviderColToken: 'Token',
+  builtinProviderSet: (name: string, model: string, envVar: string) => `  ${name}: model=${model}, key_env=${envVar}`,
+
+  // ── provider routing ───────────────────────────────────────────────────────
+  providerFallback: (from: string, to: string) => `Provider "${from}" quota/rate limit — switching to "${to}"`,
+  providerAllFailed: 'All providers in the fallback chain failed. Check your API keys and quotas.',
+  configAgentProviderSet: (agent: string, spec: string) => `  Agent "${agent}" will use: ${spec}`,
+  configFallbackSet: (chain: string) => `  Fallback chain set: ${chain}`,
+  configFallbackCleared: '  Fallback chain cleared.',
+  configAgentProviderShow: '\n  Agent provider assignments\n',
+  configAgentProviderNone: '  No per-agent providers configured. All agents use the default provider.',
+  configInvalidAgentName: (name: string) => `  Unknown agent: "${name}". Valid: business, architecture, security, context`,
+  configInvalidProviderSpec: (spec: string) => `  Invalid spec: "${spec}". Use format: provider or provider/model (e.g. claude/claude-haiku-4-5)`,
+
+  // ── post-review action ─────────────────────────────────────────────────────
+  postReviewSelectAction: 'What would you like to do with this PR?',
+  postReviewSubmitting: 'Submitting review to GitHub...',
+  postReviewApproved: (n: number) => `  PR #${n} approved on GitHub.`,
+  postReviewChangesRequested: (n: number) => `  Changes requested on PR #${n}.`,
+  postReviewFailed: (msg: string) => `Failed to submit review: ${msg}`,
+  postReviewRefreshing: 'Refreshing project context...',
+  postReviewRefreshed: 'Context updated.',
+
+  // ── context generate ───────────────────────────────────────────────────────
+  contextGenerateTitle: '\n  Generating context and rules with AI...\n',
+  contextGenerating: 'Analyzing project with AI...',
+  contextGenerateDone: 'Done. Edit the files if needed.',
+  contextGenerateFailed: (msg: string) => `Generation failed: ${msg}`,
+  contextGenerateConfirmOverwrite: (file: string) => `${file} already exists. Overwrite?`,
+  contextGenerateSkipped: (file: string) => `  ${file} skipped (existing file kept).`,
+  contextGenerateWritten: (file: string) => `  ${file} written.`,
+  contextGenerateProviderError: (envVar: string) => `Missing AI key: ${envVar}. Check your provider config.`,
+
+  // ── agents command ─────────────────────────────────────────────────────────
+  agentsTitle: '\n  aiv — Available Agents\n',
+  agentsColAgent: 'Agent',
+  agentsColDesc: 'Description',
+  agentsColFocus: 'Focus Areas',
+  agentsFooter: 'Run specific agents with: aiv review <pr> --agent business security',
+  agentBusinessDesc: 'Analyzes business logic, domain rules, and functional correctness',
+  agentBusinessFocus: 'Business logic, domain invariants, rule violations, functional regressions',
+  agentArchDesc: 'Reviews structural patterns, module coupling, and design decisions',
+  agentArchFocus: 'Layer violations, coupling, SRP, dependency direction, abstraction quality',
+  agentSecDesc: 'Detects security vulnerabilities and data exposure risks',
+  agentSecFocus: 'Auth bypass, injection, data leakage, OWASP Top 10, sensitive data handling',
+};
+
+export type TranslationKeys = typeof en;