import { resolve } from "path";
import { existsSync, mkdirSync, writeFileSync } from "fs";
import http from "http";

async function setup() {
  await new Promise<void>((res, rej) => {
    http
      .get("http://127.0.0.1:8188/object_info", (resp) => {
        let data = "";
        resp.on("data", (chunk) => {
          data += chunk;
        });
        resp.on("end", () => {
          // Modify the response data to add some checkpoints
          const objectInfo = JSON.parse(data);
          objectInfo.CheckpointLoaderSimple.input.required.ckpt_name[0] = ["model1.safetensors", "model2.ckpt"];
          objectInfo.VAELoader.input.required.vae_name[0] = ["vae1.safetensors", "vae2.ckpt"];

          data = JSON.stringify(objectInfo, undefined, "\t");

          const outDir = resolve("./tests-ui/data");
          if (!existsSync(outDir)) {
            mkdirSync(outDir);
          }

          const outPath = resolve(outDir, "object_info.json");
          console.log(`Writing ${Object.keys(objectInfo).length} nodes to ${outPath}`);
          writeFileSync(outPath, data, {
            encoding: "utf8",
          });
          res();
        });
      })
      .on("error", rej);
  });
}

setup();