electron踩坑系列之一

function archiveLog(file) {
  file = file.toString();
  const info = path.parse(file);
  try {
    fs.renameSync(file, path.join(info.dir, info.name + '.old' + info.ext));
  } catch (e) {
    console.warn('Could not rotate log', e);
  }
}
复制代码

// 50M
log.transports.file.maxSize = 50 * 1024 * 1024;
log.transports.file.archiveLog = function archiveLog(file: string) {
    file = file.toString();
    const info = path.parse(file);
    try {
        // 重命名
        const dateStr = getDateStr();
        const newFileName = path.join(info.dir, `${info.name}.${dateStr}${info.ext}`);
        console.log("archiveLog", file, newFileName);
        fs.renameSync(file, newFileName);
        // 删除旧文件
        deleteFiles(info.dir)
    } catch (e) {
        console.error('Could not rotate log', e);
    }
}
复制代码

function fixPeople(data: any) {
    modifyPeople(data, unParsePerson)
    modifyPeople(data, parsePerson)
  },
function unParsePerson (person) {
  if (typeof person === "string") return person
  var name = person.name || ""
  var u = person.url || person.web
  var url = u ? (" ("+u+")") : ""
  var e = person.email || person.mail
  var email = e ? (" <"+e+">") : ""
  return name+email+url
}
function parsePerson (person) {
  console.log("parsePerson person", person)
  if (typeof person !== "string") return person
  var name = person.match(/^([^\(<]+)/)
  var url = person.match(/\(([^\)]+)\)/)
  var email = person.match(/<([^>]+)>/)
  var obj = {}
  if (name && name[0].trim()) obj.name = name[0].trim()
  if (email) obj.email = email[1];
  if (url) obj.url = url[1];
  return obj
}
复制代码

import { createHash } from "crypto"
import { createReadStream, statSync , readFileSync, writeFileSync } from "fs"
import path from "path";
import YAML from "yaml";
const jsonConfig = require("../package.json");
const version = jsonConfig.version;
const exeFilePath = path.join(__dirname, `../packages/${version}.exe`);
const yamlFilePath = path.join(__dirname, "../packages/latest.yml");
const jsFilePath = path.join(__dirname, "../packages/latest.js");
const bakYamlFilePath =  path.join(__dirname, `../packages/latest.bak.yml`);
// packages\app-builder-lib\src\util\hash.ts
function hashFile(file: string, algorithm = "sha512", encoding: "base64" | "hex" = "base64", options?: any): Promise<string> {
    return new Promise<string>((resolve, reject) => {
      const hash = createHash(algorithm)
      hash.on("error", reject).setEncoding(encoding)
      createReadStream(file, { ...options, highWaterMark: 1024 * 1024 /* better to use more memory but hash faster */ })
        .on("error", reject)
        .on("end", () => {
          hash.end()
          resolve(hash.read() as string)
        })
        .pipe(hash, { end: false })
    })
}
function getFileSize(path: string){
    const state = statSync(path);
    return state.size;
}
function getYAML(path: string){
  const file = readFileSync(path, 'utf8')
  const data = YAML.parse(file);
  return data;
}
function saveYAML(path:string, content: any){
  writeFileSync(path,  YAML.stringify(content))
}
function saveJS(path:string, content: string){
  writeFileSync(path,  content)
}
;(async function updateLatestYML(){
  try{
    const hash = await hashFile(exeFilePath);
    const size = getFileSize(exeFilePath);
    console.log("hash:", hash);
    console.log("size:", size);
    const yamlData = getYAML(yamlFilePath);
    saveYAML(bakYamlFilePath, yamlData);
    console.log("yamlData:before", yamlData);
    const name = `${version}.exe`
    yamlData.version = version;
    const file = yamlData.files[0];
    file.url = name;
    file.sha512 = hash;
    file.size = size;
    yamlData.path = name;
    yamlData.sha512 = hash;
    console.log("yamlData:after", yamlData);
    saveYAML(yamlFilePath, yamlData);
    saveJS(jsFilePath, `window._lastest=${JSON.stringify(yamlData)}`);
  }catch(err){
    console.log("update latest.yml 失败", err);
  }
})();
复制代码

function checkUpdate() {
    // 开发环境
    if(!app.isPackaged){
        return autoUpdater.checkForUpdates();
    }
    return autoUpdater.checkForUpdatesAndNotify();
}
复制代码

win.webContents.on("did-frame-navigate",  (event: Electron.Event, url: string, httpResponseCode: number, httpStatusText: string, isMainFrame: boolean, frameProcessId: number, frameRoutingId: number) =>{
            console.log("did-frame-navigate", httpResponseCode, httpStatusText);
            if(httpResponseCode >= 400 && !this.isUsedFailedPage){
               return this.useFailedPage();
            }
        });
复制代码

const BLACK_LIST = ["KeyI", "KeyR"];
function preventForceRefreshAndTool() {
    const env = process.env.ENV;
    console.log("process.env.ENV", process.env.ENV);
    if (env !== "prod") {
        return console.log("非产线环境，允许刷新和打开开发者工具");
    }
    window.addEventListener('keydown', (e: KeyboardEvent) => {
        const { ctrlKey, code, shiftKey } = e;
        console.log("keydown", e);
        // ctrl +  r
        // crtl + shift + i
        if (ctrlKey && shiftKey && BLACK_LIST.includes(code)) {
            console.log("globalShortcut：被阻止", ctrlKey, shiftKey, code);
            e.preventDefault();
        }
    }, false);
}
复制代码

win.webContents.on('before-input-event', (event, input) => {
          console.log('before-input-event', input.control, input.shift, input.key)
          if (input.control && KEY_BLACK_LIST.includes(input.key.toUpperCase())) {
            event.preventDefault()
          }
     });
复制代码

BrowserWindow.webContents.setBackgroundThrottling(false);
复制代码

const { webContents } = require('electron')
const options = { extraHeaders: 'pragma: no-cache\n' }
webContents.loadURL('https://github.com', options)
复制代码

options.isAdminRightsRequired = true;
复制代码

if (updateInfo != null && isPerMachine && oneClick) {
    updateInfo.isAdminRightsRequired = true
}
复制代码

import iconv from 'iconv-lite';
 return new Promise((resolve, reject) => {
        let ch: childProcess.ChildProcess = exec(`${command}`, options, function (err, stdout: Buffer | string, stderr: Buffer | string) {
            log.log("startProcess exec: stdout:",
                Buffer.isBuffer(stdout) ? iconv.decode(stdout, "GBK") : stdout,
                " stderr;",
                Buffer.isBuffer(stderr) ? iconv.decode(stderr, "GBK") : stderr);
            if (err) {
                log.log("startProcess exec: error.message: ", err.message, " error.stack:", err.stack);
                return reject(err)
            }
            return resolve(ch);
        });
复制代码