upscayl/main/index.js

// Native
const { join, parse } = require("path");
const { format } = require("url");
const { spawn } = require("child_process");
const fs = require("fs");

const { execPath, modelsPath } = require("./binaries");

// Packages
const {
  BrowserWindow,
  app,
  ipcMain,
  dialog,
  ipcRenderer,
} = require("electron");
const isDev = require("electron-is-dev");
const prepareNext = require("electron-next");

// Prepare the renderer once the app is ready
let mainWindow;
app.on("ready", async () => {
  await prepareNext("./renderer");

  mainWindow = new BrowserWindow({
    width: 1100,
    height: 700,
    webPreferences: {
      autoHideMenuBar: true,
      nodeIntegration: true,
      preload: join(__dirname, "preload.js"),
    },
  });
  const url = isDev
    ? "http://localhost:8000"
    : format({
        pathname: join(__dirname, "../renderer/out/index.html"),
        protocol: "file:",
        slashes: true,
      });

  mainWindow.setMenuBarVisibility(false);
  // mainWindow.maximize();
  mainWindow.loadURL(url);
});

// Quit the app once all windows are closed
app.on("window-all-closed", app.quit);

// ! DONT FORGET TO RESTART THE APP WHEN YOU CHANGE CODE HERE
ipcMain.on("sendMessage", (_, message) => {
  console.log(message);
});

ipcMain.handle("open", async () => {
  const { canceled, filePaths } = await dialog.showOpenDialog({
    properties: ["openFile", "multiSelections"],
  });

  if (canceled) {
    console.log("operation cancelled");
    return "cancelled";
  } else {
    console.log(filePaths[0]);
    // CREATE input AND upscaled FOLDER 
    return filePaths[0];
  }
})
ipcMain.handle("output", async (event, message) => {
  const { canceled, filePaths } = await dialog.showOpenDialog({
    properties: ["openDirectory"],
  });
  if (canceled) {
    console.log("operation cancelled");
    return "cancelled";
  }
  else {
    console.log(filePaths[0])
    return filePaths[0];
  }
})

ipcMain.on("upscayl", async (event, paths) => {
  const scale = "4";
  let inputDir = paths[0].match(/(.*)[\/\\]/)[1]||'';
  /*if (!fs.existsSync(inputDir)) {
    fs.mkdirSync(inputDir);
  }*/
  let outputDir = paths[1];
  /*if (!fs.existsSync(outputDir)) {
    fs.mkdirSync(outputDir);
  }*/

  // COPY IMAGE TO upscaled FOLDER
  const fullfileName = paths[0].split("/").slice(-1)[0];
  const fileName = parse(fullfileName).name;
  const fileExt = parse(fullfileName).ext;

  // UPSCALE
  let upscayl = spawn(
    execPath,
    [
      "-i",
      inputDir+'/'+fullfileName,
      "-o",
      outputDir+'/'+fileName+"_upscaled_"+scale+'x'+fileExt,
      "-s",
      scale,
      "-m",
      modelsPath,
      "-n",
      "realesrgan-x4plus",
    ],
    {
      cwd: null,
      detached: false,
    }
  );

  upscayl.stderr.on("data", (stderr) => {
    console.log(stderr.toString());
    stderr = stderr.toString();
    mainWindow.webContents.send("output", stderr.toString());
  });

  upscayl.on("close", (code) => {
    console.log("Done upscaling");
    mainWindow.webContents.send("done");
  });
})