Files
koodo-reader/main.js
troyeguo 568a1fb644 fix bug
Former-commit-id: 32105ddfb56341bc6da285d2b1ca06e168147f39
2021-09-19 23:29:53 +08:00

147 lines
3.9 KiB
JavaScript

const { app, BrowserWindow, Menu, remote, ipcMain } = require("electron");
const path = require("path");
const isDev = require("electron-is-dev");
const fs = require("fs");
const configDir = (app || remote.app).getPath("userData");
const dirPath = path.join(configDir, "uploads");
let mainWin;
let readerWindow;
const singleInstance = app.requestSingleInstanceLock();
var filePath = null;
if (process.platform == "win32" && process.argv.length >= 2) {
filePath = process.argv[1];
}
// Single Instance Lock
if (!singleInstance) {
app.quit();
if (filePath) {
fs.writeFileSync(
path.join(dirPath, "log.json"),
JSON.stringify({ filePath })
);
}
} else {
app.on("second-instance", (event, argv, workingDir) => {
if (mainWin) {
if (!mainWin.isVisible()) mainWin.show();
mainWin.focus();
}
});
}
app.on("ready", () => {
let option = {
width: 1050,
height: 660,
webPreferences: {
webSecurity: false,
nodeIntegration: true,
nativeWindowOpen: true,
enableRemoteModule: true,
nodeIntegrationInSubFrames: true,
allowRunningInsecureContent: true,
},
};
mainWin = new BrowserWindow(option);
if (!isDev) {
Menu.setApplicationMenu(null);
}
const urlLocation = isDev
? "http://localhost:3000"
: `file://${path.join(__dirname, "./build/index.html")}`;
mainWin.loadURL(urlLocation);
mainWin.on("close", () => {
mainWin = null;
});
ipcMain.on("open-book", (event, arg) => {
let url = arg;
let options = {
webPreferences: {
webSecurity: false,
nodeIntegration: true,
nativeWindowOpen: true,
enableRemoteModule: true,
nodeIntegrationInSubFrames: true,
allowRunningInsecureContent: true,
},
};
let pdfLocation = isDev
? "http://localhost:3000/" + url
: `file://${path.join(
__dirname,
"./build",
"lib",
"pdf",
"web",
"viewer.html"
)}?${url.split("?")[1]}`;
var urlParams;
var match,
pl = /\+/g,
search = /([^&=]+)=?([^&]*)/g,
decode = function (s) {
return decodeURIComponent(s.replace(pl, " "));
},
query = url.split("?").reverse()[0];
urlParams = {};
while ((match = search.exec(query)))
urlParams[decode(match[1])] = decode(match[2]);
if (url.indexOf("full") > -1) {
Object.assign(options, {
width: 1050,
height: 660,
});
readerWindow = new BrowserWindow(options);
readerWindow.loadURL(url.indexOf("pdf") > -1 ? pdfLocation : url);
readerWindow.maximize();
} else {
Object.assign(options, {
width: parseInt(urlParams.width),
height: parseInt(urlParams.height),
x: parseInt(urlParams.x),
y: parseInt(urlParams.y),
frame: urlParams.isMergeWord === "yes" ? false : true,
hasShadow: urlParams.isMergeWord === "yes" ? false : true,
transparent: urlParams.isMergeWord === "yes" ? true : false,
});
readerWindow = new BrowserWindow(options);
readerWindow.loadURL(url.indexOf("pdf") > -1 ? pdfLocation : url);
}
readerWindow.on("close", () => {
readerWindow && readerWindow.destroy();
});
event.returnValue = "success";
});
ipcMain.on("storage-location", (event, arg) => {
event.returnValue = path.join(dirPath, "data");
});
ipcMain.on("get-file-data", function (event) {
if (fs.existsSync(path.join(dirPath, "log.json"))) {
const _data = JSON.parse(
fs.readFileSync(path.join(dirPath, "log.json"), "utf8") || "{}"
);
if (_data && _data.filePath) {
filePath = _data.filePath;
fs.writeFileSync(path.join(dirPath, "log.json"), "");
}
}
event.returnValue = filePath;
filePath = null;
});
});
app.on("window-all-closed", () => {
app.quit();
});
app.on("open-file", (e, pathToFile) => {
filePath = pathToFile;
});