import { loadFiles } from "utils/functions";

type Pyodide = {
  runPythonAsync: (code: string) => Promise<string>;
};

type PyError = Error & { message: string };

declare global {
  interface Window {
    loadPyodide: (cfg: { indexURL: string }) => Promise<Pyodide>;
    pyodide?: Pyodide;
  }
}

const config = {
  fullStdLib: false,
  indexURL: "/Program Files/Pyodide/",
};

const versionCommand = "import sys\r\nsys.version\r\n";

const captureStdOut =
  "import sys\r\nimport io\r\nsys.stdout = io.StringIO()\r\n";

export const runPython = async (
  code: string,
  printLn: (message: string) => void
): Promise<void> => {
  await loadFiles(["/Program Files/Pyodide/pyodide.js"]);

  if (!window.pyodide && window.loadPyodide) {
    window.pyodide = await window.loadPyodide(config);
  }

  if (window.pyodide) {
    const getVersion = code === "ver" || code === "version";

    try {
      let result = await window.pyodide.runPythonAsync(
        getVersion ? versionCommand : captureStdOut + code
      );

      if (!result) {
        result = await window.pyodide.runPythonAsync("sys.stdout.getvalue()");
      }

      if (result) printLn(result.toString());
    } catch (error) {
      const { message } = error as PyError;

      if (message) printLn(message);
    }
  }
};