Skip to content

Commit

Permalink
feat: create menu items with check for updates action
Browse files Browse the repository at this point in the history
  • Loading branch information
@louis-jan
louis-jan committed Oct 2, 2023
1 parent 34103b9 commit 0a5a39c
Show file tree
Hide file tree
Showing 2 changed files with 242 additions and 149 deletions.
293 changes: 144 additions & 149 deletions electron/main.ts
View file
Original file line number Diff line number Diff line change
@@ -1,30 +1,41 @@
import {
app,
BrowserWindow,
screen as electronScreen,
ipcMain,
dialog,
shell,
} from "electron";
import { app, BrowserWindow, ipcMain, dialog, shell } from "electron";
import { readdirSync } from "fs";
import { resolve, join, extname } from "path";
import { rmdir, unlink, createWriteStream } from "fs";
import isDev = require("electron-is-dev");
import { init } from "./core/plugin-manager/pluginMgr";
const { autoUpdater } = require("electron-updater");
const Store = require("electron-store");
import { setupMenu } from "./utils/menu";

import isDev = require("electron-is-dev");
// @ts-ignore
import request = require("request");
// @ts-ignore
import progress = require("request-progress");
const { autoUpdater } = require("electron-updater");
const Store = require("electron-store");

let mainWindow: BrowserWindow | undefined = undefined;
const store = new Store();

autoUpdater.autoDownload = false;
autoUpdater.autoInstallOnAppQuit = true;
app
.whenReady()
.then(migratePlugins)
.then(setupPlugins)
.then(setupMenu)
.then(handleIPCs)
.then(handleAppUpdates)
.then(createMainWindow)
.then(() => {
app.on("activate", () => {
if (!BrowserWindow.getAllWindows().length) {
createMainWindow();
}
});
});

app.on("window-all-closed", () => {
app.quit();
});

const createMainWindow = () => {
function createMainWindow() {
mainWindow = new BrowserWindow({
width: 1200,
height: 800,
Expand All @@ -37,26 +48,6 @@ const createMainWindow = () => {
},
});

ipcMain.handle(
"invokePluginFunc",
async (_event, modulePath, method, ...args) => {
const module = join(app.getPath("userData"), "plugins", modulePath);
return await import(/* webpackIgnore: true */ module)
.then((plugin) => {
if (typeof plugin[method] === "function") {
return plugin[method](...args);
} else {
console.log(plugin[method]);
console.error(`Function "${method}" does not exist in the module.`);
}
})
.then((res) => {
return res;
})
.catch((err) => console.log(err));
}
);

const startURL = isDev
? "http://localhost:3000"
: `file://${join(__dirname, "../renderer/index.html")}`;
Expand All @@ -69,139 +60,143 @@ const createMainWindow = () => {
});

if (isDev) mainWindow.webContents.openDevTools();
};
}

app
.whenReady()
.then(migratePlugins)
.then(() => {
createMainWindow();
setupPlugins();
autoUpdater.checkForUpdates();

ipcMain.handle("basePlugins", async (event) => {
const basePluginPath = join(
__dirname,
"../",
isDev ? "/core/pre-install" : "../app.asar.unpacked/core/pre-install"
);
return readdirSync(basePluginPath)
.filter((file) => extname(file) === ".tgz")
.map((file) => join(basePluginPath, file));
function handleAppUpdates() {
/*New Update Available*/
autoUpdater.on("update-available", async (_info: any) => {
const action = await dialog.showMessageBox({
message: `Update available. Do you want to download the latest update?`,
buttons: ["Download", "Later"],
});
if (action.response === 0) await autoUpdater.downloadUpdate();
});

ipcMain.handle("pluginPath", async (event) => {
return join(app.getPath("userData"), "plugins");
});
ipcMain.handle("appVersion", async (event) => {
return app.getVersion();
});
ipcMain.handle("openExternalUrl", async (event, url) => {
shell.openExternal(url);
/*App Update Completion Message*/
autoUpdater.on("update-downloaded", async (_info: any) => {
mainWindow?.webContents.send("APP_UPDATE_COMPLETE", {});
const action = await dialog.showMessageBox({
message: `Update downloaded. Please restart the application to apply the updates.`,
buttons: ["Restart", "Later"],
});
if (action.response === 0) {
autoUpdater.quitAndInstall();
}
});

/**
* Used to delete a file from the user data folder
*/
ipcMain.handle("deleteFile", async (_event, filePath) => {
const userDataPath = app.getPath("userData");
const fullPath = join(userDataPath, filePath);

let result = "NULL";
unlink(fullPath, function (err) {
if (err && err.code == "ENOENT") {
result = `File not exist: ${err}`;
} else if (err) {
result = `File delete error: ${err}`;
} else {
result = "File deleted successfully";
}
console.log(
`Delete file ${filePath} from ${fullPath} result: ${result}`
);
});
/*App Update Error */
autoUpdater.on("error", (info: any) => {
dialog.showMessageBox({ message: info.message });
mainWindow?.webContents.send("APP_UPDATE_ERROR", {});
});

return result;
/*App Update Progress */
autoUpdater.on("download-progress", (progress: any) => {
console.log("app update progress: ", progress.percent);
mainWindow?.webContents.send("APP_UPDATE_PROGRESS", {
percent: progress.percent,
});
});
autoUpdater.autoDownload = false;
autoUpdater.autoInstallOnAppQuit = true;
autoUpdater.checkForUpdates();
}

/**
* Used to download a file from a given url
*/
ipcMain.handle("downloadFile", async (_event, url, fileName) => {
const userDataPath = app.getPath("userData");
const destination = resolve(userDataPath, fileName);

progress(request(url), {})
.on("progress", function (state: any) {
mainWindow?.webContents.send("FILE_DOWNLOAD_UPDATE", {
...state,
fileName,
});
})
.on("error", function (err: Error) {
mainWindow?.webContents.send("FILE_DOWNLOAD_ERROR", {
fileName,
err,
});
function handleIPCs() {
ipcMain.handle(
"invokePluginFunc",
async (_event, modulePath, method, ...args) => {
const module = join(app.getPath("userData"), "plugins", modulePath);
return await import(/* webpackIgnore: true */ module)
.then((plugin) => {
if (typeof plugin[method] === "function") {
return plugin[method](...args);
} else {
console.log(plugin[method]);
console.error(`Function "${method}" does not exist in the module.`);
}
})
.on("end", function () {
mainWindow?.webContents.send("FILE_DOWNLOAD_COMPLETE", {
fileName,
});
.then((res) => {
return res;
})
.pipe(createWriteStream(destination));
});

app.on("activate", () => {
if (!BrowserWindow.getAllWindows().length) {
createMainWindow();
}
});
.catch((err) => console.log(err));
}
);
ipcMain.handle("basePlugins", async (_event) => {
const basePluginPath = join(
__dirname,
"../",
isDev ? "/core/pre-install" : "../app.asar.unpacked/core/pre-install"
);
return readdirSync(basePluginPath)
.filter((file) => extname(file) === ".tgz")
.map((file) => join(basePluginPath, file));
});

/*New Update Available*/
autoUpdater.on("update-available", async (info: any) => {
const action = await dialog.showMessageBox({
message: `Update available. Do you want to download the latest update?`,
buttons: ["Download", "Later"],
ipcMain.handle("pluginPath", async (_event) => {
return join(app.getPath("userData"), "plugins");
});
if (action.response === 0) await autoUpdater.downloadUpdate();
});

/*App Update Completion Message*/
autoUpdater.on("update-downloaded", async (info: any) => {
mainWindow?.webContents.send("APP_UPDATE_COMPLETE", {});
const action = await dialog.showMessageBox({
message: `Update downloaded. Please restart the application to apply the updates.`,
buttons: ["Restart", "Later"],
ipcMain.handle("appVersion", async (_event) => {
return app.getVersion();
});
ipcMain.handle("openExternalUrl", async (_event, url) => {
shell.openExternal(url);
});
if (action.response === 0) {
autoUpdater.quitAndInstall();
}
});

/*App Update Error */
autoUpdater.on("error", (info: any) => {
dialog.showMessageBox({ message: info.message });
mainWindow?.webContents.send("APP_UPDATE_ERROR", {});
});
/**
* Used to delete a file from the user data folder
*/
ipcMain.handle("deleteFile", async (_event, filePath) => {
const userDataPath = app.getPath("userData");
const fullPath = join(userDataPath, filePath);

let result = "NULL";
unlink(fullPath, function (err) {
if (err && err.code == "ENOENT") {
result = `File not exist: ${err}`;
} else if (err) {
result = `File delete error: ${err}`;
} else {
result = "File deleted successfully";
}
console.log(`Delete file ${filePath} from ${fullPath} result: ${result}`);
});

/*App Update Progress */
autoUpdater.on("download-progress", (progress: any) => {
console.log("app update progress: ", progress.percent);
mainWindow?.webContents.send("APP_UPDATE_PROGRESS", {
percent: progress.percent,
return result;
});
});

app.on("window-all-closed", () => {
if (process.platform !== "darwin") {
app.quit();
}
});
/**
* Used to download a file from a given url
*/
ipcMain.handle("downloadFile", async (_event, url, fileName) => {
const userDataPath = app.getPath("userData");
const destination = resolve(userDataPath, fileName);

progress(request(url), {})
.on("progress", function (state: any) {
mainWindow?.webContents.send("FILE_DOWNLOAD_UPDATE", {
...state,
fileName,
});
})
.on("error", function (err: Error) {
mainWindow?.webContents.send("FILE_DOWNLOAD_ERROR", {
fileName,
err,
});
})
.on("end", function () {
mainWindow?.webContents.send("FILE_DOWNLOAD_COMPLETE", {
fileName,
});
})
.pipe(createWriteStream(destination));
});
}

function migratePlugins() {
return new Promise((resolve) => {
const store = new Store();
if (store.get("migrated_version") !== app.getVersion()) {
console.log("start migration:", store.get("migrated_version"));
const userDataPath = app.getPath("userData");
Expand All @@ -217,12 +212,12 @@ function migratePlugins() {
resolve(undefined);
}
});
};
}

function setupPlugins() {
init({
// Function to check from the main process that user wants to install a plugin
confirmInstall: async (plugins: string[]) => {
confirmInstall: async (_plugins: string[]) => {
return true;
},
// Path to install plugin to
Expand Down
Loading

0 comments on commit 0a5a39c

Please sign in to comment.