armcord/src/window.ts

// To allow seamless switching between custom titlebar and native os titlebar,
// I had to add most of the window creation code here to split both into seperete functions
// WHY? Because I can't use the same code for both due to annoying bug with value `frame` not responding to variables
// I'm sorry for this mess but I'm not sure how to fix it.
import { BrowserWindow, shell, app, ipcMain } from "electron";
import path from "path";
import { contentPath } from "./main";
import { checkIfConfigIsNew, firstRun, getConfigUnsafe } from "./utils";
import { registerIpc } from "./ipc";
import contextMenu from "electron-context-menu";
export let mainWindow: BrowserWindow;
import * as glasstron from "glasstron";

let guestWindows: BrowserWindow [] = [];
contextMenu({
  showSaveImageAs: true,
  showCopyImageAddress: true,
  showSearchWithGoogle: true,

});

function doAfterDefiningTheWindow() {
  checkIfConfigIsNew();
  registerIpc();
  mainWindow.webContents.userAgent =
    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36"; //fake useragent for screenshare to work
  mainWindow.webContents.setWindowOpenHandler(({ url }) => {
    shell.openExternal(url);
    return { action: "deny" };
  });
  mainWindow.webContents.session.webRequest.onBeforeRequest(
    (details, callback) => {
      if (/api\/v\d\/science$/g.test(details.url))
        return callback({ cancel: true });
      return callback({});
    }
  );
  mainWindow.on("close", async (e) => {
    if (await getConfigUnsafe("minimizeToTray")) {
      e.preventDefault();
      mainWindow.hide();
    } else if (!await getConfigUnsafe("minimizeToTray")) {
      e.preventDefault();
      app.exit();
      app.quit()
    }
  });
  console.log(contentPath);
  try {
    mainWindow.loadFile(contentPath);
  } catch (e) {
    console.log(
      "Major error detected while starting up. User is most likely on Windows platform. Fallback to alternative startup."
    );
    console.log(process.platform);
    if (process.platform === "win32") {
      if (firstRun) {
        mainWindow.loadURL(`file://${__dirname}/content/setup.html`);
      } else {
        mainWindow.loadURL(`file://${__dirname}/content/splash.html`);
      }
    } else {
      if (firstRun) {
        mainWindow.loadURL(`file://${__dirname}/ts-out/content/setup.html`);
      } else {
        mainWindow.loadURL(`file://${__dirname}/ts-out/content/splash.html`);
      }
    }
  }
}
export function createCustomWindow() {
  mainWindow = new BrowserWindow({
    width: 300,
    height: 350,
    title: "ArmCord",
    darkTheme: true,
    icon: path.join(__dirname, "/assets/icon_transparent.png"),
    frame: false,
    autoHideMenuBar: true,
    webPreferences: {
      preload: path.join(__dirname, "preload/preload.js"),
      spellcheck: true,
    },
  });
  doAfterDefiningTheWindow();
}
export function createNativeWindow() {
  mainWindow = new BrowserWindow({
    width: 300,
    height: 350,
    title: "ArmCord",
    darkTheme: true,
    icon: path.join(__dirname, "/assets/icon_transparent.png"),
    frame: true,
    autoHideMenuBar: true,
    webPreferences: {
      preload: path.join(__dirname, "preload/preload.js"),
      spellcheck: true,
    },
  });
  doAfterDefiningTheWindow();
}
export function createGlasstronWindow() {
  mainWindow = new glasstron.BrowserWindow({
    width: 300,
    height: 350,
    title: "ArmCord",
    darkTheme: true,
    icon: path.join(__dirname, "/assets/icon_transparent.png"),
    frame: true,
    autoHideMenuBar: true,
    webPreferences: {
      preload: path.join(__dirname, "preload/preload.js"),
      spellcheck: true,
    },
  });

  //@ts-expect-error
  mainWindow.blurType = getConfigUnsafe("blurType");
  //@ts-expect-error
  mainWindow.setBlur(true);
  doAfterDefiningTheWindow();
}
export function createTabsHost() {
  guestWindows[1] = mainWindow;
  mainWindow = new BrowserWindow({
    width: 300,
    height: 350,
    title: "ArmCord",
    darkTheme: true,
    icon: path.join(__dirname, "/assets/icon_transparent.png"),
    frame: true,
    autoHideMenuBar: true,
    webPreferences: {
      preload: path.join(__dirname, "preload/preload.js"),
    },
  });
  doAfterDefiningTheWindow();
}
export function createTabsGuest(number: number) {
  console.log(guestWindows)
  if (guestWindows[number] !== undefined || null) {
  try {
      console.log("Showing Guest Window " + number);
      mainWindow.hide()
      guestWindows[number].show();
      mainWindow = guestWindows[number];
    } catch (e) {
      console.error(e);
    }
  } else {
  console.log("Creating Guest Window " + number);
  mainWindow.hide();

  guestWindows[number] = new BrowserWindow({
    width: 800,
    height: 600,
    title: "ArmCord Guest Window " + number,
    darkTheme: true,
    icon: path.join(__dirname, "/assets/icon_transparent.png"),
    frame: true,
    autoHideMenuBar: true,
    webPreferences: {
      preload: path.join(__dirname, "preload/preload.js"),
    },
  });
  
  mainWindow = guestWindows[number];
  ipcMain.on("tab" + number, (event) => {
    event.returnValue = true; //return true so we know the tab exists
  });
  
  guestWindows[number].webContents.userAgent =
    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36"; //fake useragent for screenshare to work
  
  guestWindows[number].webContents.setWindowOpenHandler(({ url }) => {
    shell.openExternal(url);
    return { action: "deny" };
  });
  
  guestWindows[number].webContents.session.webRequest.onBeforeRequest(
    (
      details: { url: string },
      callback: (arg0: { cancel?: boolean }) => any
    ) => {
      if (/api\/v\d\/science$/g.test(details.url))
        return callback({ cancel: true });
      return callback({});
    }
  );
  
  guestWindows[number].loadURL("https://discord.com/app");
  }
}