berget 2.2.6 → 2.2.8

package/src/commands/code/__tests__/setup-flow.test.ts

@@ -1,274 +1,630 @@
-import { describe, expect, it } from 'vitest'
-import { runSetup } from '../setup'
-import { CancelledError, CommandFailedError, PrerequisiteError } from '../errors'
-import { FakePrompter, CANCEL, select, confirm } from './fake-prompter'
-import { FakeFileStore } from './fake-file-store'
-import { FakeCommandRunner } from './fake-command-runner'
-
-const makeDeps = (overrides: Partial<Parameters<typeof runSetup>[0]> = {}) => ({
-	prompter: new FakePrompter([]),
-	files: new FakeFileStore(),
-	commands: new FakeCommandRunner()
-		.handle('opencode --version', 'mocked')
-		.handle('pi --version', 'mocked'),
-	homeDir: '/home/user',
-	cwd: '/home/user/project',
-	...overrides,
-})
+import { describe, expect, it } from 'vitest';
+
+import type { ApiKeyServicePort, AuthServicePort } from '../ports/auth-services';
+
+import { CancelledError, CommandFailedError, PrerequisiteError } from '../errors';
+import { runSetup } from '../setup';
+import { FakeApiKeyService } from './fake-api-key-service';
+import { FakeAuthService } from './fake-auth-service';
+import { FakeCommandRunner } from './fake-command-runner';
+import { FakeFileStore } from './fake-file-store';
+import { CANCEL, confirm, FakePrompter, multiselect, select } from './fake-prompter';
+
+const makeDeps = (
+  overrides: Partial<Parameters<typeof runSetup>[0]> = {},
+): Parameters<typeof runSetup>[0] => {
+  return {
+    apiKeyService:
+      (overrides.apiKeyService as ApiKeyServicePort) ?? new FakeApiKeyService('sk_ber_test'),
+    authService: (overrides.authService as AuthServicePort) ?? new FakeAuthService(false),
+    commands:
+      overrides.commands ??
+      new FakeCommandRunner()
+        .handle('opencode --version', 'mocked')
+        .handle('pi --version', 'mocked'),
+    cwd: '/home/user/project',
+    files: overrides.files ?? new FakeFileStore(),
+    homeDir: '/home/user',
+    prompter: overrides.prompter ?? new FakePrompter([]),
+    ...Object.fromEntries(
+      Object.entries(overrides).filter(
+        ([k]) =>
+          k !== 'prompter' &&
+          k !== 'files' &&
+          k !== 'commands' &&
+          k !== 'authService' &&
+          k !== 'apiKeyService',
+      ),
+    ),
+  };
+};
+
+function base64urlEncode(data: string): string {
+  return Buffer.from(data).toString('base64url');
+}
+
+function makeJwt(payload: Record<string, unknown>): string {
+  const header = base64urlEncode(JSON.stringify({ alg: 'none', typ: 'JWT' }));
+  const body = base64urlEncode(JSON.stringify(payload));
+  return `${header}.${body}.signature`;
+}
 
 describe('runSetup', () => {
-	describe('happy path', () => {
-		it('sets up opencode project without existing config', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-					confirm(true, 'Create'),
-				]),
-			})
-
-			await runSetup(deps)
-			
-			const files = deps.files as FakeFileStore
-			const written = files.getWrittenFiles()
-			expect(written.has('/home/user/project/opencode.json')).toBe(true)
-			const config = JSON.parse(written.get('/home/user/project/opencode.json')!)
-			expect(config.plugin).toContain('@bergetai/opencode-auth@1.0.16')
-		})
-
-		it('sets up opencode globally without existing config', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('global'),
-					confirm(true, 'Create'),
-				]),
-			})
-
-			await runSetup(deps)
-			
-			const files = deps.files as FakeFileStore
-			const written = files.getWrittenFiles()
-			expect(written.has('/home/user/.config/opencode/opencode.json')).toBe(true)
-		})
-
-		it('sets up pi project with fresh install', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('pi'),
-					select('project'),
-					confirm(true, 'Proceed'),
-				]),
-				commands: new FakeCommandRunner()
-					.handle('pi --version', 'mocked') // For checkInstalled
-					.handle('pi install', ''), // For actual install
-			})
-
-			await runSetup(deps)
-			
-			const commands = deps.commands as FakeCommandRunner
-			expect(commands.calls.length).toBeGreaterThan(0)
-			const installCall = commands.calls.find(c => c.command === 'pi')
-			expect(installCall?.args).toContain('npm:@bergetai/pi-provider')
-		})
-	})
-
-	describe('prerequisites', () => {
-		it('throws PrerequisiteError when opencode is not installed', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-				]),
-				commands: new FakeCommandRunner(),
-			})
-			
-			// Simulate opencode not being installed
-			await expect(runSetup(deps)).rejects.toBeInstanceOf(PrerequisiteError)
-		})
-	})
-
-	describe('cancellation', () => {
-		it('throws CancelledError when user cancels at tool selection', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select(CANCEL),
-				]),
-			})
-
-			await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError)
-		})
-
-		it('throws CancelledError when user cancels at write confirmation', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-					confirm(false, 'Create'),
-				]),
-			})
-
-			await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError)
-		})
-	})
-
-	describe('file operations', () => {
-		it('preserves existing configuration keys when updating', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-					confirm(true, 'Write'),
-				]),
-			})
-			
-			const files = deps.files as FakeFileStore
-			files.seed('/home/user/project/opencode.json', JSON.stringify({
-				customField: 'should-preserve',
-				plugin: ['other-plugin'],
-			}))
-
-			await runSetup(deps)
-
-			const written = files.getWrittenFiles()
-			const config = JSON.parse(written.get('/home/user/project/opencode.json')!)
-			expect(config.customField).toBe('should-preserve')
-			expect(config.plugin).toContain('other-plugin')
-			expect(config.plugin).toContain('@bergetai/opencode-auth@1.0.16')
-		})
-
-		it('preserves jsonc comments when updating', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-					confirm(true, 'Write'),
-				]),
-			})
-			
-			const files = deps.files as FakeFileStore
-			files.seed('/home/user/project/opencode.jsonc', `{
+  describe('happy path', () => {
+    it('sets up opencode project without existing config', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Create'), // Config write
+          multiselect([]), // No agents selected
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/project/opencode.json')).toBe(true);
+      const config = JSON.parse(written.get('/home/user/project/opencode.json')!);
+      expect(config.plugin).toContain('@bergetai/opencode-auth');
+    });
+
+    it('sets up opencode globally without existing config', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('global'),
+          confirm(true, 'Create'), // Config write
+          multiselect([]), // No agents selected
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/.config/opencode/opencode.json')).toBe(true);
+    });
+
+    it('sets up pi project with fresh install', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner()
+          .handle('pi --version', 'mocked') // For checkInstalled
+          .handle('pi install', ''), // For actual install
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'), // Agent selection
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const commands = deps.commands as FakeCommandRunner;
+      expect(commands.calls.length).toBeGreaterThan(0);
+      const installCall = commands.calls.find((c) => c.command === 'pi');
+      expect(installCall?.args).toContain('npm:@bergetai/pi-provider');
+    });
+
+    it('skips agent selection for pi project', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner()
+          .handle('pi --version', 'mocked') // For checkInstalled
+          .handle('pi install', ''), // For actual install
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('__skip__'), // Skip agent selection
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      // Should not create any agent files
+      for (const path of written.keys()) {
+        expect(path).not.toContain('SYSTEM.md');
+      }
+    });
+  });
+
+  describe('prerequisites', () => {
+    it('throws PrerequisiteError when opencode is not installed', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner(),
+        prompter: new FakePrompter([select('opencode'), select('project')]),
+      });
+
+      // Simulate opencode not being installed
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(PrerequisiteError);
+    });
+  });
+
+  describe('cancellation', () => {
+    it('throws CancelledError when user cancels at tool selection', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([select(CANCEL)]),
+      });
+
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError);
+    });
+
+    it('throws CancelledError when user cancels at write confirmation', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(false, 'Create'),
+        ]),
+      });
+
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError);
+    });
+
+    it('throws CancelledError when user cancels at agent write confirmation (opencode)', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Create'),
+          multiselect(['backend', 'frontend']),
+          confirm(false, 'agent'),
+        ]),
+      });
+
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError);
+    });
+
+    it('throws CancelledError when user cancels at agent write confirmation (pi)', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(false, /Create|Overwrite/),
+        ]),
+      });
+
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(CancelledError);
+    });
+  });
+
+  describe('file operations', () => {
+    it('preserves existing configuration keys when updating', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Write'),
+          multiselect([]),
+        ]),
+      });
+
+      const files = deps.files as FakeFileStore;
+      files.seed(
+        '/home/user/project/opencode.json',
+        JSON.stringify({
+          customField: 'should-preserve',
+          plugin: ['other-plugin'],
+        }),
+      );
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      const config = JSON.parse(written.get('/home/user/project/opencode.json')!);
+      expect(config.customField).toBe('should-preserve');
+      expect(config.plugin).toContain('other-plugin');
+      expect(config.plugin).toContain('@bergetai/opencode-auth');
+    });
+
+    it('preserves jsonc comments when updating', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Write'),
+          multiselect([]),
+        ]),
+      });
+
+      const files = deps.files as FakeFileStore;
+      files.seed(
+        '/home/user/project/opencode.jsonc',
+        `{
   // This is my custom config
   "customField": "should-preserve",
   /* block comment explaining plugin */
   "plugin": ["other-plugin"]
-}`)
-
-			await runSetup(deps)
-
-			const written = files.getWrittenFiles()
-			const content = written.get('/home/user/project/opencode.jsonc')!
-			expect(content).toContain('// This is my custom config')
-			expect(content).toContain('/* block comment explaining plugin */')
-			expect(content).toContain('"customField": "should-preserve"')
-			expect(content).toContain('@bergetai/opencode-auth@1.0.16')
-		})
-
-		it('shows no changes needed when config is already up to date', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('project'),
-				]),
-			})
-			
-			const files = deps.files as FakeFileStore
-			// Already has the exact same plugin version
-			files.seed('/home/user/project/opencode.json', JSON.stringify({
-				$schema: 'https://opencode.ai/config.json',
-				plugin: ['@bergetai/opencode-auth@1.0.16'],
-			}, null, 2) + '\n')
-
-			await runSetup(deps)
-
-			// Check that no write happened — content should be unchanged
-			const written = files.getWrittenFiles()
-			const content = written.get('/home/user/project/opencode.json')!
-			const config = JSON.parse(content)
-			expect(config.plugin).toEqual(['@bergetai/opencode-auth@1.0.16'])
-			expect(content).toContain('$schema')
-		})
-
-		it('preserves existing Pi settings when setting defaultProvider', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('pi'),
-					select('project'),
-					confirm(true, 'Proceed'),
-				]),
-				commands: new FakeCommandRunner()
-					.handle('pi --version', 'mocked')
-					.handle('pi install', ''),
-			})
-
-			const files = deps.files as FakeFileStore
-			files.seed('/home/user/project/.pi/settings.json', JSON.stringify({
-				existingKey: 'should-preserve',
-				anotherSetting: true,
-			}))
-
-			await runSetup(deps)
-
-			const written = files.getWrittenFiles()
-			const settings = JSON.parse(written.get('/home/user/project/.pi/settings.json')!)
-			expect(settings.existingKey).toBe('should-preserve')
-			expect(settings.anotherSetting).toBe(true)
-			expect(settings.defaultProvider).toBe('berget')
-		})
-
-		it('creates parent directories when writing files', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('opencode'),
-					select('global'),
-					confirm(true, 'Create'),
-				]),
-			})
-
-			await runSetup(deps)
-
-			const files = deps.files as FakeFileStore
-			const written = files.getWrittenFiles()
-			expect(written.has('/home/user/.config/opencode/opencode.json')).toBe(true)
-		})
-	})
-
-	describe('command execution', () => {
-		it('passes arguments as array (no shell injection)', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('pi'),
-					select('project'),
-					confirm(true, 'Proceed'),
-				]),
-				commands: new FakeCommandRunner()
-					.handle('pi --version', 'mocked')
-					.handle('pi install', ''),
-			})
-
-			await runSetup(deps)
-
-			const commands = deps.commands as FakeCommandRunner
-			const installCall = commands.calls.find(c => c.command === 'pi')
-			expect(installCall?.args).toContain('npm:@bergetai/pi-provider')
-			expect(installCall?.args).toContain('-l')
-		})
-	})
-
-	describe('error handling', () => {
-		it('throws CommandFailedError when pi install fails', async () => {
-			const deps = makeDeps({
-				prompter: new FakePrompter([
-					select('pi'),
-					select('project'),
-					confirm(true, 'Proceed'),
-				]),
-				commands: new FakeCommandRunner()
-					.handle('pi --version', 'mocked')
-					.handle('pi install', new Error('npm error')),
-			})
-
-			await expect(runSetup(deps)).rejects.toBeInstanceOf(CommandFailedError)
-		})
-	})
-})
+}`,
+      );
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      const content = written.get('/home/user/project/opencode.jsonc')!;
+      expect(content).toContain('// This is my custom config');
+      expect(content).toContain('/* block comment explaining plugin */');
+      expect(content).toContain('"customField": "should-preserve"');
+      expect(content).toContain('@bergetai/opencode-auth');
+    });
+
+    it('shows no changes needed when config is already up to date', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([select('opencode'), select('project'), multiselect([])]),
+      });
+
+      const files = deps.files as FakeFileStore;
+      // Already has the exact same plugin version
+      files.seed(
+        '/home/user/project/opencode.json',
+        JSON.stringify(
+          {
+            $schema: 'https://opencode.ai/config.json',
+            plugin: ['@bergetai/opencode-auth'],
+          },
+          null,
+          2,
+        ) + '\n',
+      );
+
+      await runSetup(deps);
+
+      // Check that no write happened — content should be unchanged
+      const written = files.getWrittenFiles();
+      const content = written.get('/home/user/project/opencode.json')!;
+      const config = JSON.parse(content);
+      expect(config.plugin).toEqual(['@bergetai/opencode-auth']);
+      expect(content).toContain('$schema');
+    });
+
+    it('preserves existing Pi settings when setting defaultProvider', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      const files = deps.files as FakeFileStore;
+      files.seed(
+        '/home/user/project/.pi/settings.json',
+        JSON.stringify({
+          anotherSetting: true,
+          existingKey: 'should-preserve',
+        }),
+      );
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      const settings = JSON.parse(written.get('/home/user/project/.pi/settings.json')!);
+      expect(settings.existingKey).toBe('should-preserve');
+      expect(settings.anotherSetting).toBe(true);
+      expect(settings.defaultProvider).toBe('berget');
+    });
+
+    it('creates parent directories when writing files', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('global'),
+          confirm(true, 'Create'),
+          multiselect([]),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/.config/opencode/opencode.json')).toBe(true);
+    });
+  });
+
+  describe('command execution', () => {
+    it('passes arguments as array (no shell injection)', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const commands = deps.commands as FakeCommandRunner;
+      const installCall = commands.calls.find((c) => c.command === 'pi');
+      expect(installCall?.args).toContain('npm:@bergetai/pi-provider');
+      expect(installCall?.args).toContain('-l');
+    });
+  });
+
+  describe('error handling', () => {
+    it('throws CommandFailedError when pi install fails', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner()
+          .handle('pi --version', 'mocked')
+          .handle('pi install', new Error('npm error')),
+        prompter: new FakePrompter([select('pi'), select('project')]),
+      });
+
+      await expect(runSetup(deps)).rejects.toBeInstanceOf(CommandFailedError);
+    });
+  });
+
+  describe('auth integration', () => {
+    it('already authenticated shows simplified message', async () => {
+      const files = new FakeFileStore();
+      files.seed(
+        '/home/user/.local/share/opencode/auth.json',
+        JSON.stringify({ berget: { type: 'oauth' } }),
+      );
+
+      const deps = makeDeps({
+        files,
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          select('keep'), // New: keep existing auth
+          confirm(true, 'Create'), // Config write
+          multiselect([]),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const prompter = deps.prompter as FakePrompter;
+      const notes = prompter.calls.filter((c) => c.method === 'note');
+      const lastNote = notes.at(-1);
+      expect(JSON.stringify(lastNote)).toContain('Run: opencode');
+      expect(JSON.stringify(lastNote)).not.toContain('/connect');
+    });
+
+    it('login failure shows manual auth instructions', async () => {
+      const deps = makeDeps({
+        authService: new FakeAuthService(false),
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        files: new FakeFileStore(), // No pre-seeded auth → auth flow runs
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const prompter = deps.prompter as FakePrompter;
+      const notes = prompter.calls.filter((c) => c.method === 'note');
+      const lastNote = notes.at(-1);
+      expect(JSON.stringify(lastNote)).toContain('/login');
+    });
+
+    it('creates api key for pi when no seat', async () => {
+      const files = new FakeFileStore();
+
+      const deps = makeDeps({
+        authService: new FakeAuthService(true, false), // succeed, no seat
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        files,
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          confirm(true), // API key creation prompt
+          select('fullstack'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/.pi/agent/auth.json')).toBe(true);
+      const parsed = JSON.parse(written.get('/home/user/.pi/agent/auth.json')!);
+      expect(parsed.berget.type).toBe('api_key');
+    });
+
+    it('uses subscription when berget_code_seat present', async () => {
+      const files = new FakeFileStore();
+      const farFuture = Math.floor(Date.now() / 1000) + 3600 * 24 * 365; // 1 year from now in seconds
+      files.seed(
+        '/home/user/.berget/auth.json',
+        JSON.stringify({
+          access_token: makeJwt({ exp: farFuture, realm_access: { roles: ['berget_code_seat'] } }),
+          expires_at: farFuture * 1000,
+          refresh_token: 'ref',
+        }),
+      );
+
+      const deps = makeDeps({
+        files,
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          select('subscription'),
+          confirm(true, 'Create'),
+          multiselect([]),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      const parsed = JSON.parse(written.get('/home/user/.local/share/opencode/auth.json')!);
+      expect(parsed.berget.type).toBe('oauth');
+    });
+  });
+
+  describe('agent configuration', () => {
+    it('sets up multiple agents for opencode project', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Create'),
+          multiselect(['backend', 'frontend']),
+          confirm(true, 'agent'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/project/.opencode/agents/backend.md')).toBe(true);
+      expect(written.has('/home/user/project/.opencode/agents/frontend.md')).toBe(true);
+    });
+
+    it('sets up no agents for opencode when none selected', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Create'),
+          multiselect([]),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      for (const path of written.keys()) {
+        expect(path).not.toMatch(/agents\/\w+\.md$/);
+      }
+    });
+
+    it('sets up agent globally for opencode', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('global'),
+          confirm(true, 'Create'),
+          multiselect(['fullstack']),
+          confirm(true, 'agent'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/.config/opencode/agents/fullstack.md')).toBe(true);
+    });
+
+    it('sets up agent for pi project', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/project/.pi/SYSTEM.md')).toBe(true);
+    });
+
+    it('sets up agent for pi globally', async () => {
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        prompter: new FakePrompter([
+          select('pi'),
+          select('global'),
+          select('backend'),
+          confirm(true, 'Create'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const written = files.getWrittenFiles();
+      expect(written.has('/home/user/.pi/agent/SYSTEM.md')).toBe(true);
+    });
+
+    it('skips writing identical opencode agent files', async () => {
+      const deps = makeDeps({
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          confirm(true, 'Create'),
+          multiselect(['backend', 'frontend']),
+          confirm(true, 'agent'),
+        ]),
+      });
+
+      // First run writes the files
+      await runSetup(deps);
+
+      const files = deps.files as FakeFileStore;
+      const firstBackend = files
+        .getWrittenFiles()
+        .get('/home/user/project/.opencode/agents/backend.md');
+      const firstFrontend = files
+        .getWrittenFiles()
+        .get('/home/user/project/.opencode/agents/frontend.md');
+
+      // Second run with exact same content should not prompt for overwrite
+      const deps2 = makeDeps({
+        files,
+        prompter: new FakePrompter([
+          select('opencode'),
+          select('project'),
+          multiselect(['backend', 'frontend']),
+        ]),
+      });
+
+      await runSetup(deps2);
+
+      // Content should be unchanged
+      expect(files.getWrittenFiles().get('/home/user/project/.opencode/agents/backend.md')).toBe(
+        firstBackend,
+      );
+      expect(files.getWrittenFiles().get('/home/user/project/.opencode/agents/frontend.md')).toBe(
+        firstFrontend,
+      );
+    });
+
+    it('overwrites pi SYSTEM.md when content differs', async () => {
+      const files = new FakeFileStore();
+      files.seed('/home/user/project/.pi/SYSTEM.md', 'old agent content');
+
+      const deps = makeDeps({
+        commands: new FakeCommandRunner().handle('pi --version', 'mocked').handle('pi install', ''),
+        files,
+        prompter: new FakePrompter([
+          select('pi'),
+          select('project'),
+          select('fullstack'),
+          confirm(true, 'Overwrite'),
+        ]),
+      });
+
+      await runSetup(deps);
+
+      const written = files.getWrittenFiles();
+      const content = written.get('/home/user/project/.pi/SYSTEM.md');
+      expect(content).not.toBe('old agent content');
+      // Pi doesn't use front matter, so check for system prompt content
+      expect(content).toContain('Fullstack Agent');
+    });
+  });
+});