akartonplanning/app/main.js

import { app, Menu, BrowserWindow, dialog, nativeImage, shell } from "electron";
import { clearActivity, setActivity, loginToRPC } from "./config/rpc.js";
import { ElectronBlocker } from "@ghostery/adblocker-electron";
import { setValue, getValue } from "./config/store.js";
import { dirname, join } from "path";
import { fileURLToPath } from "url";

import { getScreenWidth, getScreenHeight } from "./config/dimensions.js";
import Windows from "./useragents.json" with { type: "json" };
import checkInternetConnected from "check-internet-connected";
import domains from "./domains.json" with { type: "json" };
import contextMenu from "electron-context-menu";
import updaterpkg from "electron-updater";
import ElectronDl from "electron-dl";
import menulayout from "./config/menu.js";
import logpkg from "electron-log";

const { transports, log: _log, functions } = logpkg;
const __filename = fileURLToPath(import.meta.url);
const windowHeight = getValue("windowHeight");
const windowWidth = getValue("windowWidth");
const __dirname = dirname(__filename);
const { autoUpdater } = updaterpkg;

transports.file.level = "verbose";
console.log = _log;
Object.assign(console, functions);

function createWindow() {
  const enterpriseOrNormal = getValue("enterprise-or-normal");
  const custompage = getValue("custompage");
  const partition = enterpriseOrNormal === "?auth=1" ? "persist:personal" : "persist:work";

  const win = new BrowserWindow({
    width: Math.round(getScreenWidth() * getValue("windowWidth")),
    height: Math.round(getScreenHeight() * getValue("windowHeight")),
    icon: join(__dirname, "/assets/icons/png/1024x1024.png"),
    show: false,
    webPreferences: {
      nodeIntegration: true,
      devTools: true,
      partition: partition,
    },
  });

  win.setAutoHideMenuBar(getValue("autohide-menubar") === "true");

  const splash = new BrowserWindow({
    width: Math.round(getScreenWidth() * 0.49),
    height: Math.round(getScreenHeight() * 0.65),
    transparent: true,
    frame: false,
    icon: join(__dirname, "/assets/icons/png/1024x1024.png"),
  });

  splash.loadURL(`https://agam778.github.io/MS-365-Electron/loading`);

  win.loadURL(`https://microsoft365.com/${custompage}/${enterpriseOrNormal}`, {
    userAgent: getValue("useragentstring") || Windows,
  });

  win.webContents.on("did-finish-load", () => {
    splash.destroy();
    win.show();
    if (getValue("discordrpcstatus") === "true") {
      setActivity(`On "${win.webContents.getTitle()}"`);
    }
    if (getValue("blockadsandtrackers") === "true") {
      ElectronBlocker.fromPrebuiltAdsAndTracking(fetch).then((blocker) => {
        blocker.enableBlockingInSession(win.webContents.session);
      });
    }
  });
}

ElectronDl({
  dlPath: "./downloads",
  onStarted: (item) => {
    dialog.showMessageBox({
      type: "info",
      title: "Downloading File",
      message: `Downloading "${item.getFilename()}" to "${item.getSavePath()}"`,
      buttons: ["OK"],
    });
  },
  onCompleted: () => {
    dialog.showMessageBox({
      type: "info",
      title: "Download Completed",
      message: `Downloading Completed! Please check your "Downloads" folder.`,
      buttons: ["OK"],
    });
  },
  onError: (item) => {
    dialog.showMessageBox({
      type: "error",
      title: "Download failed",
      message: `Downloading "${item.getFilename()}" failed :(`,
      buttons: ["OK"],
    });
  },
});

contextMenu({
  showInspectElement: false,
  showServices: false,
});

Menu.setApplicationMenu(Menu.buildFromTemplate(menulayout));

app.on("ready", () => {
  createWindow();
});

app.on("web-contents-created", (event, contents) => {
  contents.setWindowOpenHandler(({ url }) => {
    const urlObject = new URL(url);
    const domain = urlObject.hostname;
    const protocol = urlObject.protocol;

    if (getValue("externalLinks") === "true") {
      if (protocol === "http:" || protocol === "https:") {
        const isAllowedDomain = domains.domains.some((allowedDomain) =>
          new RegExp(`^${allowedDomain.replace("*.", ".*")}$`).test(domain)
        );

        if (isAllowedDomain) {
          if (getValue("websites-in-new-window") === "false") {
            if (url.includes("page=Download")) return { action: "allow" };
            BrowserWindow.getFocusedWindow().loadURL(url).catch();
            if (getValue("discordrpcstatus") === "true") {
              setActivity(`On "${BrowserWindow.getFocusedWindow().webContents.getTitle()}"`);
            }
            return { action: "deny" };
          } else {
            if (getValue("discordrpcstatus") === "true") {
              setActivity(`On "${BrowserWindow.getFocusedWindow().webContents.getTitle()}"`);
            }
            return {
              action: "allow",
              overrideBrowserWindowOptions: {
                width: Math.round(getScreenWidth() * (windowWidth - 0.07)),
                height: Math.round(getScreenHeight() * (windowHeight - 0.07)),
              },
            };
          }
        } else {
          shell.openExternal(url);
          return { action: "deny" };
        }
      } else {
        return { action: "deny" };
      }
    } else {
      if (getValue("websites-in-new-window") === "false") {
        if (url.includes("page=Download")) return { action: "allow" };
        BrowserWindow.getFocusedWindow().loadURL(url).catch();
        if (getValue("discordrpcstatus") === "true") {
          setActivity(`On "${BrowserWindow.getFocusedWindow().webContents.getTitle()}"`);
        }
        return { action: "deny" };
      } else {
        if (getValue("discordrpcstatus") === "true") {
          setActivity(`On "${BrowserWindow.getFocusedWindow().webContents.getTitle()}"`);
        }
        return {
          action: "allow",
          overrideBrowserWindowOptions: {
            width: Math.round(getScreenWidth() * (windowWidth - 0.07)),
            height: Math.round(getScreenHeight() * (windowHeight - 0.07)),
          },
        };
      }
    }
  });
  contents.on("did-finish-load", () => {
    if (getValue("dynamicicons") === "true") {
      if (BrowserWindow.getFocusedWindow()) {
        if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("&ithint=file%2cpptx") ||
          BrowserWindow.getFocusedWindow().webContents.getTitle().includes(".pptx")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/powerpoint-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/powerpoint.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "PowerPoint");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("&ithint=file%2cdocx") ||
          BrowserWindow.getFocusedWindow().webContents.getTitle().includes(".docx")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/word-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/word.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "Word");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("&ithint=file%2cxlsx") ||
          BrowserWindow.getFocusedWindow().webContents.getTitle().includes(".xlsx")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/excel-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/excel.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "Excel");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("outlook.live.com") ||
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("outlook.office.com")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/outlook-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/outlook.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "Outlook");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("onedrive.live.com") ||
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("onedrive.aspx")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/onedrive-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/onedrive.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "OneDrive");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("teams.live.com")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/teams-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/teams.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "Teams");
            });
          }
        } else if (
          BrowserWindow.getFocusedWindow().webContents.getURL().includes("&ithint=onenote")
        ) {
          if (process.platform === "darwin") {
            app.dock.setIcon(join(__dirname, "../assets/icons/apps/onenote-mac.png"));
          } else if (process.platform === "win32") {
            let nimage = nativeImage.createFromPath(
              join(__dirname, "../assets/icons/apps/onenote.png")
            );
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(nimage, "OneNote");
            });
          }
        } else {
          if (process.platform === "darwin") {
            app.dock.setIcon(null);
          } else {
            BrowserWindow.getAllWindows().forEach((window) => {
              window.setOverlayIcon(null, "");
            });
          }
        }
      }
    }
    BrowserWindow.getAllWindows().forEach((window) => {
      if (window.webContents.getURL().includes("outlook.live.com")) {
        window.webContents
          .executeJavaScript(
            `
            const observer = new MutationObserver((mutationsList) => {
              let adElementFound = false;
              for (const mutation of mutationsList) {
                if (mutation.type === 'childList') {
                  const adElement = document.querySelector('div.GssDD');
                  if (adElement) {
                    adElement.remove();
                    adElementFound = true;
                  }
                }
              }
              if (adElementFound) {
                observer.disconnect();
              }
            });

            observer.observe(document.body, { childList: true, subtree: true });

            const adElement = document.querySelector('div.GssDD');
            if (adElement) {
              adElement.remove();
              observer.disconnect();
            }
            `
          )
          .catch();
      }
    });
    contents.insertCSS(
      `
      ::-webkit-scrollbar {
        width: 8px;
        height: 8px;
      }

      ::-webkit-scrollbar-track {
        background: transparent;
      }

      ::-webkit-scrollbar-thumb {
        background: transparent;
        border-radius: 5px;
      }

      ::-webkit-scrollbar-thumb:hover {
        background: #555;
      }      
      `
    );
  });
});

app.on("browser-window-created", (event, window) => {
  if (getValue("autohide-menubar") === "true") {
    window.setAutoHideMenuBar(true);
  } else {
    window.setAutoHideMenuBar(false);
  }
  window.webContents.on("did-finish-load", () => {
    if (getValue("discordrpcstatus") === "true") {
      setActivity(`On "${window.webContents.getTitle()}"`);
    }
  });
  if (getValue("blockadsandtrackers") === "true") {
    ElectronBlocker.fromPrebuiltAdsAndTracking(fetch).then((blocker) => {
      blocker.enableBlockingInSession(window.webContents.session);
    });
  }
});

app.on("window-all-closed", () => {
  if (process.platform !== "darwin") {
    app.quit();
  }
  if (process.platform === "darwin") {
    app.dock.setIcon(null);
  }
  clearActivity();
});

app.on("activate", () => {
  if (BrowserWindow.getAllWindows().length === 0) {
    createWindow();
  }
});

app.on("ready", function () {
  checkInternetConnected().catch(() => {
    const options = {
      type: "warning",
      buttons: ["Ok"],
      defaultId: 2,
      title: "Warning",
      message: "You appear to be offline!",
      detail:
        "Please check your Internet Connectivity. This app cannot run without an Internet Connection!",
    };
    dialog.showMessageBox(null, options, (response) => {
      console.log(response);
    });
  });
  if (getValue("autoupdater") === "true") {
    autoUpdater.checkForUpdatesAndNotify();
  }
  if (getValue("discordrpcstatus") === "true") {
    loginToRPC();
    setActivity(`Opening Microsoft 365...`);
  }
});