ComfyUI_frontend/tests-ui/setup.js

const { resolve } = require("path");
const { existsSync, mkdirSync, writeFileSync } = require("fs");
const http = require("http");

async function setup() {
	await new Promise((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("./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();