import { Sandbox } from "@e2b/code-interpreter";
import { TIMEOUT_MS } from "../constants.js";
import { getSandboxErrorFields } from "./sandbox-error-fields.js";

export async function readFile(
  sandbox: Sandbox,
  filePath: string,
): Promise<{
  success: boolean;
  output: string;
}> {
  try {
    const readOutput = await sandbox.commands.run(`cat "${filePath}"`);
    // Add an extra 5 min timeout to the sandbox.
    await sandbox.setTimeout(TIMEOUT_MS);

    if (readOutput.exitCode !== 0) {
      console.error(
        `\nError reading file '${filePath}' from sandbox via cat:`,
        readOutput,
      );
      return {
        success: false,
        output: `FAILED TO READ FILE from sandbox '${filePath}'. Exit code: ${readOutput.exitCode}.\nStderr: ${readOutput.stderr}.\nStdout: ${readOutput.stdout}`,
      };
    }
    if (readOutput.stderr) {
      console.warn(
        `\nStderr while reading file '${filePath}' from sandbox via cat: ${readOutput.stderr}`,
      );
    }
    return {
      success: true,
      output: readOutput.stdout,
    };
  } catch (e: any) {
    console.error(
      `\nException while trying to read file '${filePath}' from sandbox via cat:`,
      e,
    );
    let outputMessage = `FAILED TO EXECUTE READ COMMAND for sandbox '${filePath}'.`;
    const errorFields = getSandboxErrorFields(e);
    if (errorFields) {
      outputMessage += `\nExit code: ${errorFields.exitCode}.\nStderr: ${errorFields.stderr}.\nStdout: ${errorFields.stdout}`;
    } else {
      outputMessage += ` Error: ${(e as Error).message || String(e)}`;
    }

    return {
      success: false,
      output: outputMessage,
    };
  }
}

export async function writeFile(
  sandbox: Sandbox,
  filePath: string,
  content: string,
): Promise<{
  success: boolean;
  output: string;
}> {
  try {
    const delimiter = "EOF_" + Date.now() + "_" + Math.random().toString(36);
    const writeCommand = `cat > "${filePath}" << '${delimiter}'
${content}
${delimiter}`;
    const writeOutput = await sandbox.commands.run(writeCommand);
    // Add an extra 5 min timeout to the sandbox.
    await sandbox.setTimeout(TIMEOUT_MS);

    if (writeOutput.exitCode !== 0) {
      console.error(
        `\nError writing file '${filePath}' to sandbox via printf:`,
        writeOutput,
      );
      return {
        success: false,
        output: `FAILED TO WRITE FILE to sandbox '${filePath}'. Exit code: ${writeOutput.exitCode}. Stderr: ${writeOutput.stderr}. Stdout: ${writeOutput.stdout}`,
      };
    }
    if (writeOutput.stderr) {
      console.warn(
        `\nStderr while writing file '${filePath}' to sandbox via printf: ${writeOutput.stderr}`,
      );
    }
    return {
      success: true,
      output: `Successfully wrote file '${filePath}' to sandbox via printf.`,
    };
  } catch (e: any) {
    console.error(
      `\nException while trying to write file '${filePath}' to sandbox via printf:`,
      e,
    );

    let outputMessage = `FAILED TO EXECUTE WRITE COMMAND for sandbox '${filePath}'.`;
    const errorFields = getSandboxErrorFields(e);
    if (errorFields) {
      outputMessage += `\nExit code: ${errorFields.exitCode}.\nStderr: ${errorFields.stderr}.\nStdout: ${errorFields.stdout}`;
    } else {
      outputMessage += ` Error: ${(e as Error).message || String(e)}`;
    }

    return {
      success: false,
      output: outputMessage,
    };
  }
}