← vibesdk

import { tool, t, type } from '../types';
import { StructuredLogger } from '../../../logger';
import { ICodingAgent } from 'worker/agents/services/interfaces/ICodingAgent';
import { z } from 'zod';

type GitCommand = 'commit' | 'log' | 'show' | 'reset';

export function createGitTool(
	agent: ICodingAgent,
	logger: StructuredLogger,
	options?: { excludeCommands?: GitCommand[] }
) {
	const allCommands: GitCommand[] = ['commit', 'log', 'show', 'reset'];
	const allowedCommands = options?.excludeCommands
		? allCommands.filter(cmd => !options.excludeCommands!.includes(cmd))
		: allCommands;

	const hasReset = allowedCommands.includes('reset');
	const commandsList = allowedCommands.join(', ');
	const description = hasReset
		? `Execute git commands. Commands: ${commandsList}. WARNING: reset is destructive!`
		: `Execute git commands. Commands: ${commandsList}.`;

	const commandType = type(
		z.enum(allowedCommands as [GitCommand, ...GitCommand[]]),
		(cmd: GitCommand) => {
			if (cmd === 'commit' || cmd === 'reset') {
				return { gitCommit: true };
			}
			return {};
		}
	);

	return tool({
		name: 'git',
		description,
		args: {
			command: commandType.describe('Git command to execute'),
			message: t.string().optional().describe('Commit message (required for commit command, e.g., "fix: resolve authentication bug")'),
			limit: t.number().optional().describe('Number of commits to show (for log command, default: 10)'),
			oid: t.string().optional().describe(hasReset
				? 'Commit hash/OID (required for show and reset commands)'
				: 'Commit hash/OID (required for show command)'),
			includeDiff: t.boolean().optional().describe('Include file diffs in show command output (default: false). Use ONLY when you need to see actual code changes. WARNING: Slower for commits with many/large files.'),
		},
		run: async ({ command, message, limit, oid, includeDiff }) => {
			try {
				const gitInstance = agent.git;

				switch (command) {
					case 'commit': {
						if (!message) {
							return {
								success: false,
								message: 'Commit message is required for commit command'
							};
						}

						const unescapedMessage = message.replace(/\\n/g, '\n').replace(/\\t/g, '\t');

						logger.info('Git commit', { message: unescapedMessage });
						const commitOid = await gitInstance.commit([], unescapedMessage);

						return {
							success: true,
							data: { oid: commitOid },
							message: commitOid ? `Committed: ${message}` : 'No changes to commit'
						};
					}

					case 'log': {
						logger.info('Git log', { limit: limit || 10 });
						const commits = await gitInstance.log(limit || 10);

						return {
							success: true,
							data: { commits },
							message: `Retrieved ${commits.length} commits`
						};
					}

					case 'show': {
						if (!oid) {
							return {
								success: false,
								message: 'Commit OID is required for show command'
							};
						}

						logger.info('Git show', { oid, includeDiff });
						const result = await gitInstance.show(oid, { includeDiff });

						return {
							success: true,
							data: result,
							message: `Commit ${result.oid.substring(0, 7)}: ${result.message} (${result.files} files)`
						};
					}

					case 'reset': {
						if (!oid) {
							return {
								success: false,
								message: 'Commit OID is required for reset command'
							};
						}

						logger.info('Git reset', { oid });
						const result = await gitInstance.reset(oid, { hard: true });

						return {
							success: true,
							data: result,
							message: `Reset to commit ${result.ref.substring(0, 7)}. ${result.filesReset} files updated. HEAD moved.`
						};
					}

					default:
						return {
							success: false,
							message: `Unknown git command: ${command}`
						};
				}
			} catch (error) {
				logger.error('Git command failed', { command, error });
				return {
					success: false,
					message: `Git ${command} failed: ${error instanceof Error ? error.message : String(error)}`
				};
			}
		},
	});
}