uweb/misc/ebrowser/webview.js
2024-08-04 14:07:23 +08:00

720 lines
21 KiB
JavaScript

/* Copyright (C) 2024 Richard Hao Cao
Ebrowser is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
Ebrowser is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
const {
app, BrowserWindow, Menu, shell, clipboard,
session, protocol, net, dialog, ipcMain
} = require('electron')
let win;
if(!app.requestSingleInstanceLock())
app.quit()
else {
app.on('ready', createWindow);
app.on('second-instance', (event, args, cwd) => {
if (win) {
if (win.isMinimized()) {
win.restore()
}
win.show()
win.focus()
cmdlineProcess(args,cwd,1);
}else
createWindow();
})
}
Menu.setApplicationMenu(null);
const fs = require('fs');
const path = require('path')
var translateRes;
{
let langs = app.getPreferredSystemLanguages();
if(langs.length==0 || langs[0].startsWith('en'))
topMenu();
else
initTranslateRes(langs[0]);
}
var repositoryurl = "https://gitlab.com/jamesfengcao/uweb/-/raw/master/misc/ebrowser/";
const readline = require('readline');
const process = require('process')
var gredirects = [];
var gredirect;
var redirects;
var bRedirect = true;
var bJS = true;
var bForwardCookie = false;
var proxies = {};
var proxy;
var useragents = {};
var downloadMenus; //[]
var selectMenus = [];
var defaultUA =
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/" +
process.versions.chrome +" Safari/537.36";
app.userAgentFallback = defaultUA;
fs.readFile(path.join(__dirname,'redirect.json'), 'utf8', (err, jsonString) => {
if (err) return;
try {
redirects = JSON.parse(jsonString);
} catch (e){console.log(e)}
});
async function createWindow () {
try {
let json = await fs.promises.readFile(path.join(__dirname,'uas.json'), 'utf8');
useragents = JSON.parse(json);
} catch (e){console.log(e)}
protocol.handle("i",(req)=>{return null;});
await (async ()=>{
try{
const readInterface = readline.createInterface ({
input: fs.createReadStream (path.join(__dirname,'config'), 'utf8'),
});
for await (const line of readInterface) {
addrCommand(line);
}
}catch(e){console.log(e);}
})();
win = new BrowserWindow(
{width: 800, height: 600,autoHideMenuBar: true,
webPreferences: {
nodeIntegration: true,
contextIsolation: false,
webviewTag: true,
}});
win.setMenuBarVisibility(false);
win.on('closed', function () {
win = null
})
win.loadFile('index.html');
fs.readFile(path.join(__dirname,'gredirect.json'), 'utf8', (err, jsonString) => {
if (err) return;
try {
gredirects = JSON.parse(jsonString);
} catch (e){console.log(e)}
});
fs.readFile(path.join(__dirname,'proxy.json'), 'utf8', (err, jsonString) => {
if (err) return;
try {
proxies = JSON.parse(jsonString, (key,val)=>{
if(!proxy && key==="proxyRules"){
proxy = {proxyRules:val};
}
return val;
});
} catch (e){console.log(e)}
});
cmdlineProcess(process.argv, process.cwd(), 0);
//app.commandLine.appendSwitch ('trace-warnings');
fs.readFile(path.join(__dirname,'download.json'), 'utf8', (err, jsonStr) => {
if (err) return;
try {
downloadMenus = JSON.parse(jsonStr);
}catch (e){console.log(e)}
});
fs.readFile(path.join(__dirname,'select.json'), 'utf8', (err, jsonStr) => {
if (err) return;
try {
selectMenus = JSON.parse(jsonStr);
}catch (e){console.log(e)}
});
win.webContents.on('page-title-updated',(event,cmd)=>{
addrCommand(cmd);
});
session.defaultSession.on("will-download", async (e, item) => {
//item.setSavePath(save)
if(!downloadMenus) return;
e.preventDefault();
let menuT = downloadContextMenuTemp(item.getURL());
let button = await promiseContextMenu(menuT);
if(-1===button) return;
e.preventDefault();
});
win.webContents.on('console-message',cbConsoleMsg);
}
app.on('window-all-closed', function () {
app.quit()
})
app.on('activate', function () {
if (win === null) {
createWindow()
}
})
app.on('will-quit', () => {
})
app.on ('web-contents-created', (event, contents) => {
if (contents.getType () === 'webview') {
contents.setWindowOpenHandler(cbWindowOpenHandler);
contents.on('context-menu',onContextMenu);
contents.on('page-title-updated',cbTitleUpdate);
contents.session.webRequest.onBeforeRequest(interceptRequest);
}
});
ipcMain.on('command', (event, cmd) => {
addrCommand(cmd);
});
function addrCommand(cmd){
if(cmd.length<3) return;
let c0 = cmd.charCodeAt(0);
switch(c0){
case 58: //':'
args = cmd.substring(1).split(/\s+/);
switch(args[0]){
case "cert":
if(args.length==1)
session.defaultSession.setCertificateVerifyProc((request, callback) => {
callback(0);
});
else
session.defaultSession.setCertificateVerifyProc(null);
return;
case "clear":
if(args.length==1){
session.defaultSession.clearData();
return;
}
switch(args[1]){
case "cache":
session.defaultSession.clearCache();
return;
case "dns":
session.defaultSession.clearHostResolverCache();
return;
case "storage":
session.defaultSession.clearStorageData();
return;
default:
try {
let opts = JSON.parse(args.slice(1).join(""));
session.defaultSession.clearData(opts);
}catch(e){console.log(e)}
}
return;
case "ext":
session.defaultSession.loadExtension(args[1]);
return;
case "gr":
if(args.length<2) {
gredirect_enable(0);
return;
}
let i = parseInt(args[1]);
if(i>=0 && i<gredirects.length)
gredirect_enable(i);
else
gredirect_disable();
return;
case "js"://execute js
eval(cmd.slice(4));
return;
case "nc":
bForwardCookie = false;
msgbox_info("Cookie forwarding disabled");
return;
case "uc":
if(bForwardCookie) {
msgbox_info("Cookie forwarding enabled for global redirection");
return;
}
forwardCookie();
return;
case "np":
session.defaultSession.setProxy ({mode:"direct"});
bRedirect = true;
return;
case "up":
if(args.length>1)
proxy = proxies[args[1]]; //retrieve proxy
if(proxy){
gredirect_disable();
session.defaultSession.setProxy(proxy);
}
return;
case "nr":
bRedirect = false; return;
case "ur":
bRedirect = true; return;
case "ua":
if(args.length==2)
session.defaultSession.setUserAgent(useragents[args[1]]);
else
session.defaultSession.setUserAgent(defaultUA);
return;
case "update":
let updateurl;
if(1==args.length)
updateApp(repositoryurl);
else {
filename = args[1];
let iSlash = filename.lastIndexOf('/');
if(iSlash>0){
let folder = path.join(__dirname,filename.substring(0,iSlash));
fs.mkdirSync(folder,{ recursive: true });
}
fetch2file(repositoryurl,filename);
}
return;
}
}
}
function gredirect_disable(){
if(gredirect){
gredirect=null;
unregisterHandler();
}
bRedirect = false;
}
function gredirect_enable(i){
if(i>=gredirects.length) return;
if(!gredirect) registerHandler();
gredirect=gredirects[i];
}
function cbConsoleMsg(e, level, msg, line, sourceid){
console.log(line);
console.log(sourceid);
console.log(msg);
}
function interceptRequest(details, callback){
let url = details.url;
if(58===url.charCodeAt(1) || (!bJS && url.endsWith(".js"))){
callback({ cancel: true });
return;
}
do {
if(gredirect || !bRedirect ||(details.resourceType !== 'mainFrame' &&
details.resourceType !== 'subFrame')) break;
let oURL = new URL(url);
let domain = oURL.hostname;
let newUrl;
try{
let newDomain = redirects[domain];
if(!newDomain) break;
newUrl = "https://"+newDomain+oURL.pathname+oURL.search+oURL.hash;
}catch(e){break;}
callback({ cancel: false, redirectURL: newUrl });
return;
}while(false);
callback({ cancel: false });
}
function cbWindowOpenHandler(details){
let url = details.url;
let js = "newTab();tabs.children[tabs.children.length-1].src='"+
url+"';";
switch(details.disposition){
case "foreground-tab":
case "new-window":
js = js + "switchTab(tabs.children.length-1)";
}
win.webContents.executeJavaScript(js,false);
return { action: "deny" };
}
function cbTitleUpdate(event,title){
win.setTitle(title);
}
function menuSelection(menuTemplate, text){
for(let i=0; i<selectMenus.length-1;i++){
menuTemplate.push({
label: selectMenus[i],
click: () => {
let cmd = selectMenus[i+1].replace('%s',text);
let js = `handleQueries(\`${cmd}\`)`;
win.webContents.executeJavaScript(js,false);
}
});
}
}
function menuDownload(menuTemplate, labelprefix, linkUrl){
for(let i=0; i<downloadMenus.length-1;i++){
menuTemplate.push({
label: labelprefix+downloadMenus[i],
click: () => {
let cmd = downloadMenus[i+1].replace('%u',linkUrl);
let js = `handleQueries(\`${cmd}\`)`;
win.webContents.executeJavaScript(js,false);
}
});
}
}
function menuArray(labelprefix, linkUrl){
let menuTemplate = [
{
label: labelprefix+translate('Open'),
click: () => {
shell.openExternal(linkUrl);
}
},
{
label: labelprefix+translate('Copy'),
click: () => {
clipboard.writeText(linkUrl);
}
},
{
label: labelprefix+translate('Download'),
click: () => {
win.contentView.children[i].webContents.downloadURL(linkUrl);
}
},
];
if(downloadMenus)
menuDownload(menuTemplate, labelprefix, linkUrl);
return menuTemplate;
}
function onContextMenu(event, params){
let url = params.linkURL;
let mTemplate = [];
if (url) {
mTemplate.push({label:url,enabled:false});
mTemplate.push.apply(mTemplate,menuArray("",url));
if((url=params.srcURL))
mTemplate.push.apply(mTemplate,menuArray("src: ",url));
}else if((url=params.srcURL)){
mTemplate.push({label:url,enabled:false});
mTemplate.push.apply(mTemplate,menuArray("src: ",url));
}else if((url=params.selectionText)){
menuSelection(mTemplate,url);
}else
return;
const contextMenu = Menu.buildFromTemplate(mTemplate);
contextMenu.popup();
}
async function topMenu(){
const menuTemplate = [];
try {
let json = await fs.promises.readFile(path.join(__dirname,'menu.json'), 'utf8');
let menus = JSON.parse(json);
if(menus.length>1){
let submenu = [];
for(let i=0;i<menus.length-1; i=i+2){
let cmd = menus[i+1];
let js = `handleQueries("${cmd}")`;
submenu.push({
label: menus[i], click: ()=>{
win.webContents.executeJavaScript(js,false);
}});
}
menuTemplate.push({
label: translate('Tools'),
submenu: submenu,
});
}
}catch(e){console.log(e)}
menuTemplate.push(
{
label: translate('Edit'),
submenu: [
{ label: translate('Config folder'), click: ()=>{
shell.openPath(__dirname);
}},
]
},
{
label: translate('Help'),
submenu: [
{ label: translate('Check for updates'), click: ()=>{
addrCommand(":update");
}},
{ label: translate('Help'), accelerator: 'F1', click: ()=>{
help();
}},
{ label: translate('Stop'), accelerator: 'Ctrl+C', click: ()=>{
let js="tabs.children[iTab].stop()"
win.webContents.executeJavaScript(js,false)
}},
{ label: translate('getURL'), accelerator: 'Ctrl+G', click: ()=>{
let js="{let q=document.forms[0].q;q.focus();q.value=tabs.children[iTab].getURL();getWinTitle()}"
win.webContents.executeJavaScript(js,false).then((r)=>{
win.setTitle(r);
});
}},
{ label: translate('Select'), accelerator: 'Ctrl+L', click:()=>{
win.webContents.executeJavaScript("document.forms[0].q.select()",false);
}},
{ label: translate('New Tab'), accelerator: 'Ctrl+T', click:()=>{
let js = "newTab();document.forms[0].q.select();switchTab(tabs.children.length-1)";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Restore Tab'), accelerator: 'Ctrl+Shift+T', click:()=>{
let js = "{let u=closedUrls.pop();if(u){newTab();switchTab(tabs.children.length-1);tabs.children[iTab].src=u}}";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('No redirect'), accelerator: 'Ctrl+R', click: ()=>{
gredirect_disable();
}},
{ label: translate('Redirect'), accelerator: 'Ctrl+Shift+R', click: ()=>{
gredirect_enable(0);
}},
{ label: translate('Close tab'), accelerator: 'Ctrl+W', click: ()=>{
win.webContents.executeJavaScript("tabClose()",false).then((r)=>{
if(""===r) win.close();
else win.setTitle(r);
});
}},
{ label: translate('Next Tab'), accelerator: 'Ctrl+Tab', click: ()=>{
let js="tabInc(1);getWinTitle()";
win.webContents.executeJavaScript(js,false).then((r)=>{
win.setTitle(r);
});
}},
{ label: translate('Previous Tab'), accelerator: 'Ctrl+Shift+Tab', click: ()=>{
let js="tabDec(-1);getWinTitle()";
win.webContents.executeJavaScript(js,false).then((r)=>{
win.setTitle(r);
});
}},
{ label: translate('Go backward'), accelerator: 'Alt+Left', click: ()=>{
let js="tabs.children[iTab].goBack()";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Go forward'), accelerator: 'Alt+Right', click: ()=>{
let js="tabs.children[iTab].goForward()";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Zoom in'), accelerator: 'Ctrl+Shift+=', click: ()=>{
let js="{let t=tabs.children[iTab];let s=t.getZoomFactor()*1.2;t.setZoomFactor(s)}";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Zoom out'), accelerator: 'Ctrl+-', click: ()=>{
let js="{let t=tabs.children[iTab];let s=t.getZoomFactor()/1.2;t.setZoomFactor(s)}";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Default zoom'), accelerator: 'Ctrl+0', click: ()=>{
let js="tabs.children[iTab].setZoomFactor(1)";
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('No focus'), accelerator: 'Esc', click: ()=>{
let js = `{let e=document.activeElement;
if(e)e.blur();try{tabs.children[iTab].stopFindInPage('clearSelection')}catch(er){}}`;
win.webContents.executeJavaScript(js,false);
}},
{ label: translate('Reload'), accelerator: 'F5', click: ()=>{
win.webContents.executeJavaScript("tabs.children[iTab].reload()",false);
}},
{ label: translate('Devtools'), accelerator: 'F12', click: ()=>{
let js = "try{tabs.children[iTab].openDevTools()}catch(e){console.log(e)}";
win.webContents.executeJavaScript(js,false);
}},
],
},
);
const menu = Menu.buildFromTemplate(menuTemplate);
Menu.setApplicationMenu(menu);
}
function cmdlineProcess(argv,cwd,extra){
let i1st = 2+extra; //index for the first query item
if(argv.length>i1st){
if(i1st+1==argv.length){//local file
let fname = path.join(cwd, argv[i1st]);
if(fs.existsSync(fname)){
let js = "tabs.children[iTab].src='file://"+fname+"'";
win.webContents.executeJavaScript(js,false);
win.setTitle(argv[i1st]);
return;
}
}
let url=argv.slice(i1st).join(" ");
win.webContents.executeJavaScript("handleQuery(`"+url+"`)",false);
win.setTitle(url);
}
}
async function cbScheme_redir(req){
if(!gredirect) return null;
let oUrl = req.url;
let newurl = gredirect+oUrl;
let options = {
body: req.body,
headers: req.headers,
method: req.method,
referer: req.referer,
duplex: "half",
bypassCustomProtocolHandlers: true
};
if(bForwardCookie){
let cookies = await session.defaultSession.cookies.get({url: oUrl});
let cookieS = cookies.map (cookie => cookie.name + '=' + cookie.value ).join(';');
options.headers['Cookie'] = cookieS;
}
return fetch(newurl, options);
}
function registerHandler(){
protocol.handle("http",cbScheme_redir);
protocol.handle("https",cbScheme_redir);
protocol.handle("ws",cbScheme_redir);
protocol.handle("wss",cbScheme_redir);
}
function unregisterHandler(){
protocol.unhandle("http",cbScheme_redir);
protocol.unhandle("https",cbScheme_redir);
protocol.unhandle("ws",cbScheme_redir);
protocol.unhandle("wss",cbScheme_redir);
}
function forwardCookie(){
const choice = dialog.showMessageBoxSync(null, {
type: 'warning',
title: 'Confirm cookie forwarding with global redirection',
message: 'Cookies are used to access your account. Forwarding cookies is vulnerable to global redirection server, proceed to enable cookie forwarding with global redirection?',
buttons: ['No','Yes']
})
if(1===choice) bForwardCookie=true;
}
function msgbox_info(msg){
dialog.showMessageBoxSync(null, {
type: 'info',
title: msg,
message: msg,
buttons: ['OK']
})
}
async function updateApp(url){//url must ending with "/"
let msg;
do {
try {
let res = await fetch(url+"package.json");
let packageS = await res.text();
{//the last part of version string is the version number, must keep increasing
let head = packageS.slice(2,40);
let iV = head.indexOf("version");
if(iV<0) {
msg = "remote package.json corrupted"
break;
}
iV = iV + 10;
let iE = head.indexOf('"',iV+4);
let iS = head.lastIndexOf('.',iE-1);
let nLatestVer = parseInt(head.substring(iS+1,iE));
let ver = app.getVersion();
iS = ver.lastIndexOf('.');
let nVer = parseInt(ver.substring(iS+1));
if(nVer>=nLatestVer){
msg = `Current version ${ver} is already up to date`;
break;
}
const choice = dialog.showMessageBoxSync(null, {
type: 'warning',
title: `Update from ${url}`,
message: `Proceed to update from ${ver} to ${head.substring(iV,iE)}?`,
buttons: ['YES','NO']
});
if(1===choice) return;
}
writeFile("package.json", packageS);
fetch2file(url,"webview.js");
fetch2file(url,"index.html");
msg = "Update completed";
}catch(e){
msg = "Fail to update"
}
}while(false);
dialog.showMessageBoxSync(null, {
type: 'info',
title: `Update from ${url}`,
message: msg,
buttons: ['OK']
})
}
async function fetch2file(urlFolder, filename, bOverwritten=true){
let pathname=path.join(__dirname,filename);
if(!bOverwritten && fs.existsSync(pathname)) return;
let res = await fetch(urlFolder+filename);
let str = await res.text();
writeFile(pathname, str);
}
async function writeFile(filename, str){
let pathname=filename+".new";
fs.writeFile(pathname, str, (err) => {
if(err) throw "Fail to write";
fs.rename(pathname,filename,(e1)=>{
if(e1) throw "Fail to rename";
});
});
}
function help(){
const readme = "README.md";
const htmlFN = path.join(__dirname,readme);
let js=`{let t=tabs.children[iTab];t.dataset.jsonce=BML_md;t.src="file://${htmlFN}"}`;
win.webContents.executeJavaScript(js,false)
}
function downloadContextMenuTemp(url){
let mTemplate =
[{label:url,enabled:false},
{label: translate('Download')},
{
label: translate('Copy'),
click: () => {
clipboard.writeText(url);
}
},
];
menuDownload(mTemplate, "", url);
return mTemplate;
}
async function initTranslateRes(lang){
let basename=path.join(__dirname,"translate.");
let fname = basename+lang;
if(!fs.existsSync(fname))
fname = basename+lang.slice(0,2);
try {
let json = await fs.promises.readFile(fname,'utf8');
translateRes = JSON.parse(json);
} catch (e){}
topMenu();
}
function translate(str){
let result;
if(translateRes && (result=translateRes[str])) return result;
return str;
}
function promiseContextMenu(menuTemplate) {
return new Promise((resolve, reject) => {
menuTemplate[1].click = () => resolve(-1);
const menu = Menu.buildFromTemplate(menuTemplate);
menu.on('menu-will-close', () => resolve(-2));
menu.popup();
});
}