import { tool } from 'ai';
import { z } from 'zod';
import { exec } from 'node:child_process';
import { promisify } from 'node:util';

const run = promisify(exec);

export const bash = tool({
  description: 'Run a shell command on the local machine and return stdout / stderr.',
  inputSchema: z.object({
    command: z.string().describe('The shell command to execute'),
  }),
  execute: async ({ command }) => {
    const { stdout, stderr } = await run(command);
    return { stdout, stderr };
  },
});

import { tool } from 'ai';
import { z } from 'zod';
import { readFile, writeFile } from 'node:fs/promises';

export const read = tool({
  description: 'Read a file. Returns content with line numbers.',
  inputSchema: z.object({ path: z.string() }),
  execute: async ({ path }) => {
    const text = await readFile(path, 'utf8');
    return text.split('\n')
      .map((line, i) => `${i + 1}\t${line}`)
      .join('\n');
  },
});

export const write = tool({
  description: 'Create a new file or overwrite an existing one.',
  inputSchema: z.object({ path: z.string(), content: z.string() }),
  execute: async ({ path, content }) => {
    await writeFile(path, content, 'utf8');
    return { ok: true, path };
  },
});

export const edit = tool({
  description: 'Apply a diff: replace one exact slice of a file.',
  inputSchema: z.object({
    path: z.string(),
    old_string: z.string(),
    new_string: z.string(),
  }),
  execute: async ({ path, old_string, new_string }) => {
    const text = await readFile(path, 'utf8');
    if (!text.includes(old_string)) throw new Error('old_string not found');
    await writeFile(path, text.replace(old_string, new_string), 'utf8');
    return { ok: true };
  },
});