electron-incremental-update 1.3.0 → 2.0.0-beta.2

package/dist/index.js

@@ -1,186 +1,97 @@
 import {
   __require,
+  getAppVersion,
+  getEntryVersion,
   getPathFromAppNameAsar,
-  getVersions,
-  is,
+  isDev,
   isUpdateJSON,
-  parseVersion,
   restartApp,
-  unzipFile,
-  waitAppReady
-} from "./chunk-SBPTSLG7.js";
+  unzipFile
+} from "./chunk-BG22XZAB.js";
 
-// src/index.ts
-import { resolve } from "node:path";
+// src/entry.ts
+import { join } from "node:path";
 import { existsSync as existsSync2, renameSync } from "node:fs";
 import { app as app2 } from "electron";
 
 // src/updater/core.ts
 import { existsSync, rmSync, writeFileSync } from "node:fs";
+import { EventEmitter } from "node:stream";
 import { app } from "electron";
 
-// src/crypto/dec.ts
-import { createDecipheriv, createVerify } from "node:crypto";
-
-// src/crypto/utils.ts
-import { createHash } from "node:crypto";
-function hashString(data, length) {
-  const hash = createHash("SHA256").update(data).digest("binary");
-  return Buffer.from(hash).subarray(0, length);
-}
-
-// src/crypto/dec.ts
-function decrypt(encryptedText, key, iv) {
-  const decipher = createDecipheriv("aes-256-cbc", key, iv);
-  let decrypted = decipher.update(encryptedText, "base64url", "utf8");
-  decrypted += decipher.final("utf8");
-  return decrypted;
-}
-var verify = (buffer, signature, cert) => {
-  try {
-    const [sig, version] = decrypt(signature, hashString(cert, 32), hashString(buffer, 16)).split("%");
-    const result = createVerify("RSA-SHA256").update(buffer).verify(cert, sig, "base64");
-    return result ? version : false;
-  } catch (error) {
-    return false;
-  }
-};
-
 // src/updater/types.ts
 var ErrorInfo = {
-  downlaod: "Download failed",
+  download: "Download failed",
   validate: "Validate failed",
-  param: "Missing params",
-  version: "Unsatisfied version"
+  param: "Missing params"
 };
 var UpdaterError = class extends Error {
+  code;
   constructor(msg, info) {
-    super(msg + ": " + info);
+    super(ErrorInfo[msg] + ": " + info);
+    this.code = msg;
   }
 };
 
-// src/updater/defaultFunctions/download.ts
-import { net } from "electron";
-async function downlaodFn(url, headers, onResponse) {
-  await waitAppReady();
-  return new Promise((resolve2, reject) => {
-    const request = net.request({ url, method: "GET", redirect: "follow" });
-    Object.keys(headers).forEach((key) => request.setHeader(key, headers[key]));
-    request.on("response", (res) => onResponse(res, resolve2, reject));
-    request.on("error", reject);
-    request.end();
-  });
-}
-var downloadJSONDefault = async (url, headers) => {
-  return await downlaodFn(url, headers, (resp, resolve2, reject) => {
-    let data = "";
-    resp.on("data", (chunk) => data += chunk);
-    resp.on("end", () => {
-      try {
-        const json = JSON.parse(data);
-        if (isUpdateJSON(json)) {
-          resolve2(json);
-        } else {
-          throw Error;
-        }
-      } catch (ignore) {
-        reject(new Error("invalid update json"));
-      }
-    });
-    resp.on("aborted", () => reject(new Error("aborted")));
-    resp.on("error", () => reject(new Error("download error")));
-  });
-};
-var downloadBufferDefault = async (url, headers, total, onDownloading) => {
-  let current = 0;
-  return await downlaodFn(url, headers, (resp, resolve2, reject) => {
-    let data = [];
-    resp.on("data", (chunk) => {
-      current += chunk.length;
-      onDownloading?.({ percent: `${+(current / total).toFixed(2) * 100}%`, total, current });
-      data.push(chunk);
-    });
-    resp.on("end", () => resolve2(Buffer.concat(data)));
-    resp.on("aborted", () => reject(new Error("aborted")));
-    resp.on("error", () => reject(new Error("download error")));
-  });
-};
-
-// src/updater/defaultFunctions/compareVersion.ts
-var isLowerVersionDefault = (version1, version2) => {
-  const oldV = parseVersion(version1);
-  const newV = parseVersion(version2);
-  function compareStrings(str1, str2) {
-    if (str1 === "") {
-      return str2 !== "";
-    } else if (str2 === "") {
-      return true;
-    }
-    return str1 < str2;
-  }
-  for (let key of Object.keys(oldV)) {
-    if (key === "stage" && compareStrings(oldV[key], newV[key])) {
-      return true;
-    } else if (oldV[key] !== newV[key]) {
-      return oldV[key] < newV[key];
-    }
-  }
-  return false;
-};
-
 // src/updater/core.ts
-var Updater = class {
-  CERT = __SIGNATURE_CERT__;
+var Updater = class extends EventEmitter {
+  CERT = __EIU_SIGNATURE_CERT__;
   info;
-  option;
+  options;
   asarPath;
   gzipPath;
   tmpFilePath;
+  provider;
   /**
    * updater logger
    */
   logger;
   /**
-   * downloading progress hook
-   * @param progress download progress
-   * @example
-   * updater.onDownloading = ({ percent, total, current }) => {
-   *   console.log(`download progress: ${percent}, total: ${total}, current: ${current}`)
-   * }
+   * URL handler hook
+   *
+   * for Github, there are some {@link https://github.com/XIU2/UserScript/blob/master/GithubEnhanced-High-Speed-Download.user.js#L34 public CDNs}
+   * @param url source url
+   * @param isDownloadAsar whether is download asar
    */
-  onDownloading;
+  handleURL;
   /**
    * whether receive beta version
    */
   get receiveBeta() {
-    return !!this.option.receiveBeta;
+    return !!this.options.receiveBeta;
   }
   set receiveBeta(receiveBeta) {
-    this.option.receiveBeta = receiveBeta;
+    this.options.receiveBeta = receiveBeta;
   }
   /**
    * initialize incremental updater
+   * @param provider update provider
    * @param option UpdaterOption
    */
-  constructor(option = {}) {
-    this.option = option;
+  constructor(provider, option = {}) {
+    super();
+    this.provider = provider;
+    this.options = option;
     if (option.SIGNATURE_CERT) {
       this.CERT = option.SIGNATURE_CERT;
     }
+    if (option.logger) {
+      this.logger = option.logger;
+    }
+    if (isDev && !this.logger) {
+      this.logger = {
+        info: (...args) => console.log("[EIU-INFO ]", ...args),
+        debug: (...args) => console.log("[EIU-DEBUG]", ...args),
+        warn: (...args) => console.log("[EIU-WARN ]", ...args),
+        error: (...args) => console.error("[EIU-ERROR]", ...args)
+      };
+      this.logger.info("no logger set, enable dev-only logger");
+    }
     this.asarPath = getPathFromAppNameAsar();
     this.gzipPath = `${this.asarPath}.gz`;
     this.tmpFilePath = `${this.asarPath}.tmp`;
   }
-  async needUpdate(version, minVersion) {
-    const isLowerVersion = this.option.overrideFunctions?.isLowerVersion ?? isLowerVersionDefault;
-    const { appVersion, entryVersion } = getVersions();
-    if (await isLowerVersion(entryVersion, minVersion)) {
-      throw new UpdaterError(ErrorInfo.version, `entry version (${entryVersion}) < minimumVersion (${minVersion})`);
-    }
-    this.logger?.info(`check update: current version is ${appVersion}, new version is ${version}`);
-    return await isLowerVersion(appVersion, version);
-  }
-  async parseData(format, data) {
+  async fetch(format, data) {
     if (existsSync(this.tmpFilePath)) {
       this.logger?.warn(`remove tmp file: ${this.tmpFilePath}`);
       rmSync(this.tmpFilePath);
@@ -193,99 +104,96 @@ var Updater = class {
       if (format === "json" && isUpdateJSON(data) || format === "buffer" && Buffer.isBuffer(data)) {
         return data;
       } else {
-        throw new UpdaterError(ErrorInfo.param, `invalid type at format '${format}': ${JSON.stringify(data)}`);
+        this.err("invalid type", "param", `invalid type at format '${format}': ${JSON.stringify(data)}`);
+        return;
       }
     }
-    const ua = this.option.downloadConfig?.userAgent || "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.183 Safari/537.36";
-    const headers = {
-      Accept: `application/${format === "json" ? "json" : "octet-stream"}`,
-      UserAgent: ua,
-      ...this.option.downloadConfig?.extraHeader
-    };
-    this.logger?.debug(`download headers: ${JSON.stringify(headers)}`);
-    const config = format === "json" ? {
-      name: "updateJsonURL",
-      url: this.option.updateJsonURL,
-      repoFallback: `${this.option.repository?.replace("github.com", "raw.githubusercontent.com")}/master/version.json`,
-      fn: this.option.overrideFunctions?.downloadJSON ?? downloadJSONDefault
-    } : {
-      name: "releaseAsarURL",
-      url: this.option.releaseAsarURL,
-      repoFallback: `${this.option.repository}/releases/download/v${this.info?.version}/${app.name}-${this.info?.version}.asar.gz`,
-      fn: this.option.overrideFunctions?.downloadBuffer ?? downloadBufferDefault
-    };
-    data ??= config.url;
-    if (!data) {
-      this.logger?.debug(`no ${config.name}, fallback to use repository`);
-      if (!this.option.repository) {
-        throw new UpdaterError(ErrorInfo.param, `${config.name} or repository is not set`);
-      }
-      if (format === "buffer" && !this.info?.version) {
-        throw new UpdaterError(ErrorInfo.param, "version is not set");
-      }
-      data = config.repoFallback;
-    }
-    this.logger?.debug(`download ${format} from ${data}`);
+    this.logger?.debug(`download from ${this.provider.name}`);
     try {
-      const ret = format === "json" ? await config.fn(data, headers) : await config.fn(data, headers, this.info.size, this.onDownloading);
-      this.logger?.debug(`download ${format} success${format === "buffer" ? `, file size: ${ret.length}` : ""}`);
-      return ret;
+      const result = format === "json" ? await this.provider.downloadJSON(data ?? __EIU_VERSION_PATH__) : await this.provider.downloadAsar(app.name, this.info, (data2) => this.emit("download-progress", data2));
+      this.logger?.debug(`download ${format} success${format === "buffer" ? `, file size: ${result.length}` : ""}`);
+      return result;
     } catch (e) {
-      throw new UpdaterError(ErrorInfo.downlaod, e.toString());
+      this.err(`download ${format} failed`, "download", `download ${format} failed: ${e}`);
     }
   }
+  /**
+   * handle error message and emit error event
+   */
+  err(msg, code, errorInfo) {
+    const err = new UpdaterError(code, errorInfo);
+    this.logger?.error(msg, err);
+    this.emit("error", err);
+  }
   async checkUpdate(data) {
-    try {
-      let { signature, size, version, minimumVersion, beta } = await this.parseData("json", data);
-      if (this.receiveBeta) {
-        version = beta.version;
-        signature = beta.signature;
-        minimumVersion = beta.minimumVersion;
-        size = beta.size;
-      }
-      this.logger?.debug(`checked version: ${version}, size: ${size}, signature: ${signature}`);
-      if (!await this.needUpdate(version, minimumVersion)) {
-        this.logger?.info(`update unavailable: ${version} is the latest version`);
-        return void 0;
-      } else {
-        this.logger?.info(`update available: ${version}`);
-        this.info = {
-          signature,
-          minimumVersion,
-          version,
-          size
-        };
-        return this.info;
-      }
-    } catch (error) {
-      this.logger?.error("check update failed", error);
-      return error;
+    const emitUnavailable = (msg) => {
+      this.logger?.info(msg);
+      this.emit("update-unavailable", msg);
+    };
+    const _data = await this.fetch("json", data);
+    if (!_data) {
+      emitUnavailable("failed to get update info");
+      return false;
+    }
+    let { signature, size, version, minimumVersion, beta } = _data;
+    if (this.receiveBeta) {
+      version = beta.version;
+      signature = beta.signature;
+      minimumVersion = beta.minimumVersion;
+      size = beta.size;
+    }
+    this.logger?.debug(`checked update, version: ${version}, size: ${size}, signature: ${signature}`);
+    if (isDev) {
+      emitUnavailable("in dev mode, skip check update");
+      return false;
     }
+    const isLowerVersion = this.provider.isLowerVersion;
+    const entryVersion = getEntryVersion();
+    const appVersion = getAppVersion();
+    if (isLowerVersion(entryVersion, minimumVersion)) {
+      emitUnavailable(`entry version (${entryVersion}) < minimumVersion (${minimumVersion})`);
+      return false;
+    }
+    this.logger?.info(`check update: current version is ${appVersion}, new version is ${version}`);
+    if (!isLowerVersion(appVersion, version)) {
+      emitUnavailable(`current version (${appVersion}) < new version (${version})`);
+      return false;
+    }
+    this.logger?.info(`update available: ${version}`);
+    this.info = { signature, minimumVersion, version, size };
+    this.emit("update-available", this.info);
+    return true;
   }
   async download(data, sig) {
+    if (!this.info) {
+      this.err("download failed", "param", "no update info, call `checkUpdate` first");
+      return false;
+    }
+    const _sig = sig ?? this.info.signature;
+    const buffer = await this.fetch("buffer", data ? Buffer.from(data) : void 0);
+    if (!buffer) {
+      this.err("download failed", "param", "no update asar file buffer");
+      return false;
+    }
+    this.logger?.debug("verify start");
+    const _ver = await this.provider.verifySignaure(buffer, _sig, this.CERT);
+    if (!_ver) {
+      this.err("verify failed", "validate", "invalid signature / certificate pair");
+      return false;
+    }
+    this.logger?.debug("verify success");
     try {
-      const _sig = sig ?? this.info?.signature;
-      if (!_sig) {
-        throw new UpdaterError(ErrorInfo.param, "signature is empty");
-      }
-      const buffer = await this.parseData("buffer", data);
-      this.logger?.debug("verify start");
-      const _verify = this.option.overrideFunctions?.verifySignaure ?? verify;
-      const _ver = await _verify(buffer, _sig, this.CERT);
-      if (!_ver) {
-        throw new UpdaterError(ErrorInfo.validate, "invalid signature or certificate");
-      }
-      this.logger?.debug("verify success");
       this.logger?.debug(`write to ${this.gzipPath}`);
       writeFileSync(this.gzipPath, buffer);
       this.logger?.debug(`extract to ${this.tmpFilePath}`);
       await unzipFile(this.gzipPath, this.tmpFilePath);
       this.logger?.info(`download success, version: ${_ver}`);
       this.info = void 0;
+      this.emit("update-downloaded");
       return true;
     } catch (error) {
-      this.logger?.error("download asar failed", error);
-      return error;
+      this.err("unwrap asar failed", "download", `fail to unwrap asar file, ${error}`);
+      return false;
     }
   }
   /**
@@ -297,66 +205,53 @@ var Updater = class {
   }
 };
 
-// src/updater/index.ts
-function createUpdater(option) {
-  return new Updater(option);
-}
-
-// src/index.ts
+// src/entry.ts
 function startupWithUpdater(fn) {
   return fn;
 }
 var defaultOnInstall = (install, _, __, logger) => {
   install();
-  logger?.info(`update success!`);
+  logger.info(`update success!`);
 };
-async function initApp(appOptions = {}) {
+async function initApp(appOptions) {
   const {
+    provider,
     updater,
-    electronDevDistPath = "../dist-electron",
-    mainPath = "main/index.js",
-    hooks
-  } = appOptions || {};
-  const {
     onInstall = defaultOnInstall,
     beforeStart,
     onStartError
-  } = hooks || {};
-  function handleError(err, logger2) {
-    console.error(err);
-    onStartError?.(err, logger2);
-    app2.quit();
-  }
+  } = appOptions;
   let updaterInstance;
   if (typeof updater === "object" || !updater) {
-    updaterInstance = createUpdater(updater);
+    updaterInstance = new Updater(provider, updater);
   } else {
     updaterInstance = await updater();
   }
-  const logger = updaterInstance.logger;
+  const logger = updaterInstance.logger || console;
   try {
     const appNameAsarPath = getPathFromAppNameAsar();
     const tempAsarPath = `${appNameAsarPath}.tmp`;
     if (existsSync2(tempAsarPath)) {
-      logger?.info(`installing new asar: ${tempAsarPath}`);
+      logger.info(`installing new asar: ${tempAsarPath}`);
       await onInstall(() => renameSync(tempAsarPath, appNameAsarPath), tempAsarPath, appNameAsarPath, logger);
     }
-    const mainDir = is.dev ? electronDevDistPath : appNameAsarPath;
-    const entry = resolve(__dirname, mainDir, mainPath);
-    await beforeStart?.(entry, logger);
-    __require(entry)(updaterInstance);
+    const mainFilePath = join(
+      isDev ? join(app2.getAppPath(), __EIU_MAIN_DEV_DIR__) : appNameAsarPath,
+      "main",
+      __EIU_MAIN_FILE__
+    );
+    await beforeStart?.(mainFilePath, logger);
+    __require(mainFilePath)(updaterInstance);
   } catch (error) {
-    handleError(error, logger);
+    logger.error("startup error", error);
+    onStartError?.(error, logger);
+    app2.quit();
   }
 }
 export {
   ErrorInfo,
   Updater,
   UpdaterError,
-  createUpdater,
-  downloadBufferDefault,
-  downloadJSONDefault,
   initApp,
-  isLowerVersionDefault,
   startupWithUpdater
 };