import windowStateManager from "electron-window-state"; import { app, BrowserWindow, ipcMain, net, dialog } from "electron"; import { setupTitlebar, attachTitlebarToWindow } from "custom-electron-titlebar/main"; import * as Sentry from "@sentry/electron"; import contextMenu from "electron-context-menu"; import serve from "electron-serve"; import { getInstalledPackages } from "./libs/tea-dir"; import { readSessionData, writeSessionData } from "./libs/auth"; import type { Session } from "../src/libs/types"; import { installPackage, openTerminal } from "./libs/cli"; import { autoUpdater } from "electron-updater"; import * as log from "electron-log"; import path from "path"; /* TODO: - fix global mutable variable - organize the ipc handlers into its own module - create auto updater initialization module */ let teaProtocolPath = ""; // this should be empty string autoUpdater.logger = log; log.info("App starting..."); if (app.isPackaged) { Sentry.init({ dsn: "https://5ff29bb5b3b64cd4bd4f4960ef1db2e3@o4504750197899264.ingest.sentry.io/4504750206746624", debug: true, transportOptions: { maxQueueAgeDays: 30, maxQueueCount: 30, beforeSend: async () => { const ol = await net.isOnline(); log.log("isOnline", ol); return ol ? "send" : "queue"; } } }); } const serveURL = serve({ directory: "." }); const port = process.env.PORT || 3000; const allowDebug = !app.isPackaged || process.env.DEBUG_BUILD === "1"; let mainWindow: BrowserWindow | null; // todo: this is awful, there should be a way to check where all windows are closed in mac let macWindowClosed = false; setupTitlebar(); function createWindow() { const windowState = windowStateManager({ defaultWidth: 1000, defaultHeight: 600 }); const mainWindow = new BrowserWindow({ titleBarStyle: "hidden", backgroundColor: "black", autoHideMenuBar: true, trafficLightPosition: { x: 14, y: 13 }, minHeight: 600, minWidth: 1000, webPreferences: { contextIsolation: false, nodeIntegration: true, spellcheck: false, webSecurity: false, devTools: allowDebug, preload: path.join(app.getAppPath(), "preload.cjs") }, x: windowState.x, y: windowState.y, width: windowState.width, height: windowState.height }); windowState.manage(mainWindow); mainWindow.webContents.openDevTools(); mainWindow.once("ready-to-show", () => { mainWindow.show(); mainWindow.focus(); }); mainWindow.on("close", () => { windowState.saveState(mainWindow); }); attachTitlebarToWindow(mainWindow); return mainWindow; } function sendStatusToWindow(text: string, params?: { [key: string]: any }) { log.info(text); mainWindow?.webContents.send("message", text, params || {}); } autoUpdater.on("checking-for-update", () => { log.info("checking for tea gui update"); }); autoUpdater.on("update-available", (info) => { sendStatusToWindow( `A new tea gui(${info.version}) is being downloaded. Please don't close the app.`, { i18n_key: "notification.gui-downloading", version: info.version, } ); }); autoUpdater.on("update-not-available", () => { log.info("no update for tea gui"); }); autoUpdater.on("error", (err) => { log.error("auto update:", err); }); autoUpdater.on("download-progress", (progressObj) => { let log_message = "Download speed: " + progressObj.bytesPerSecond; log_message = log_message + " - Downloaded " + progressObj.percent + "%"; log_message = log_message + " (" + progressObj.transferred + "/" + progressObj.total + ")"; log.info("tea gui:", log_message); }); autoUpdater.on("update-downloaded", (info) => { sendStatusToWindow(`A new tea gui(${info.version}) is available. Relaunch the app to update.`, { i18n_key: "notification.gui-downloaded", version: info.version, action: "relaunch" }); }); contextMenu({ showLookUpSelection: false, showSearchWithGoogle: false, showCopyImage: false, prepend: (defaultActions, params, browserWindow) => [ { label: "Make App 💻" } ] }); function loadVite(port) { mainWindow?.loadURL(`http://localhost:${port}?is-vite=true`).catch((e) => { console.log("Error loading URL, retrying", e); setTimeout(() => { loadVite(port); }, 200); }); } function createMainWindow() { autoUpdater.checkForUpdatesAndNotify(); if (mainWindow) { if (mainWindow.isMinimized()) mainWindow.restore(); mainWindow.focus(); } else { mainWindow = createWindow(); } mainWindow.once("close", () => { mainWindow = null; }); if (!app.isPackaged) { // dev loadVite(port); } else { serveURL(mainWindow); } } if (process.defaultApp) { app.setAsDefaultProtocolClient("tea", process.execPath, [path.resolve(process.argv[1])]); } else { app.setAsDefaultProtocolClient("tea"); } app.once("ready", createMainWindow); app.on("activate", () => { if (!mainWindow) { createMainWindow(); } }); app.on("window-all-closed", async () => { // mac ux is just minimize them when closed unless forced quite CMD+Q macWindowClosed = true; if (process.platform !== "darwin") { app.quit(); } }); // NOTE: this doesnt work in linux // you have to loop through process.argv to figure out which url launched the app app.on("open-url", (event, url) => { // ie url: tea://packages/slug event.preventDefault(); const packagesPrefix = "/packages/"; let rawPath = url.replace("tea:/", ""); const isPackage = url.includes(packagesPrefix); if (isPackage) { // /packages/github.com/pypa/twine -> /packages/github_com_pypa_twine const packageSlug = rawPath .replace(packagesPrefix, "") .replace(/[^\w\s]/gi, "_") .toLocaleLowerCase(); rawPath = [packagesPrefix, packageSlug].join(""); } teaProtocolPath = rawPath; if (mainWindow && mainWindow.isMinimized()) { mainWindow.restore(); log.info("restored"); mainWindow?.webContents.send("sync-path"); } else if (macWindowClosed) { log.info("open new window"); createMainWindow(); } }); ipcMain.handle("get-installed-packages", async () => { const pkgs = await getInstalledPackages(); return pkgs; }); ipcMain.handle("get-session", async () => { const session = await readSessionData(); return session; }); ipcMain.handle("update-session", async (_, data) => { await writeSessionData(data as Session); }); ipcMain.handle("install-package", async (_, data) => { const result = await installPackage(data.full_name); return result; }); ipcMain.handle("open-terminal", async (_, data) => { const { cmd } = data as { cmd: string }; try { // TODO: detect if mac or linux // current openTerminal is only design for Mac await openTerminal(cmd); } catch (error) { console.error("elast:", error); } }); ipcMain.handle("relaunch", async () => { await autoUpdater.quitAndInstall(); }); ipcMain.handle("get-protocol-path", async () => { const path = teaProtocolPath; teaProtocolPath = ""; return path; });