From 05c71c93446d365b930e239d74c664dda4adc77a Mon Sep 17 00:00:00 2001
From: icexb <icexb_baka@163.com>
Date: Thu, 27 Jul 2023 14:19:02 +0800
Subject: [PATCH] update core replace function

---
 electron/main/index.ts | 27 +++++++++++++++++++--------
 package.json           |  3 ++-
 2 files changed, 21 insertions(+), 9 deletions(-)

diff --git a/electron/main/index.ts b/electron/main/index.ts
index 530408f..e981394 100644
--- a/electron/main/index.ts
+++ b/electron/main/index.ts
@@ -1,9 +1,11 @@
-import {app, BrowserWindow, shell, ipcMain, dialog, Notification, ipcRenderer} from 'electron'
+import {app, BrowserWindow, dialog, ipcMain, Notification, shell} from 'electron'
 import {release} from 'node:os'
 import {join} from 'node:path'
 import path from 'path';
 import cp from 'child_process';
 import * as fs from 'fs';
+import crypto from 'crypto';
+import getPort from "./port"
 
 const axios = require("axios")
 const WebSocket = require('ws')
@@ -55,7 +57,14 @@ const indexHtml = join(process.env.DIST, 'index.html')
 let running = true;
 const CoreBin: string = path.join(PROGRAM_DIR, "core", process.platform === 'win32' ? 'core.exe' : 'core.bin');
 
-let CoreVersion;
+let CoreVersion: string;
+
+function getFileHash(filepath: string) {
+    const buffer = fs.readFileSync(filepath);
+    const hash = crypto.createHash('md5');
+    hash.update(buffer);
+    return hash.digest('hex')
+}
 
 function initCoreVersion() {
     const CoreBinPacked: string = getExtraResourcesPath(process.platform === 'win32' ? 'core.exe' : 'core.bin');
@@ -63,18 +72,22 @@ function initCoreVersion() {
         CoreVersion = cp.execSync(`"${CoreBin}" -v`).toString()
     }
     if (fs.existsSync(CoreBinPacked)) {
-        let CorePackedVersion;
+        let CorePackedVersion: string;
         cp.exec(`"${CoreBinPacked}" -v`, (error, stdout, stderr) => {
             CorePackedVersion = stdout;
             let replace = false;
             if (!fs.existsSync(CoreBin)) replace = true;
             else if (semver.gt(CorePackedVersion, CoreVersion, true)) replace = true;
-            else if (fs.statSync(CoreBinPacked).mtimeMs > fs.statSync(CoreBin).mtimeMs) replace = true;
+            else if (getFileHash(CoreBinPacked) != getFileHash(CoreBin)) replace = true;
 
             if (replace) {
                 CoreVersion = CorePackedVersion;
                 fs.cp(CoreBinPacked, CoreBin, (err) => {
-                    if (err == null) if (app.isPackaged) fs.unlinkSync(CoreBinPacked);
+                    if (err == null) {
+                        if (app.isPackaged) fs.unlinkSync(CoreBinPacked);
+                        app.relaunch();
+                        app.exit(0);
+                    }
                 });
             }
         })
@@ -212,8 +225,6 @@ function startReleaseCore() {
         console.log("Core Not Found")
 }
 
-import getPort from "./port"
-
 function initCore() {
     initCoreVersion()
     let devPort = 50000;
@@ -335,7 +346,7 @@ ipcMain.on('select-file-exist-story', function (event) {
     dialog.showOpenDialog({
         title: '选择数据文件',
         properties: ['openFile'],
-        filters: [{name: '世界计划数据文件', extensions: ['json', 'asset','pjs.txt']}]
+        filters: [{name: '世界计划数据文件', extensions: ['json', 'asset', 'pjs.txt']}]
     }).then(result => {
         event.sender.send('selected-story', result)
     })
diff --git a/package.json b/package.json
index 4b5d32e..f586a2a 100644
--- a/package.json
+++ b/package.json
@@ -16,7 +16,8 @@
   "scripts": {
     "dev": "vite",
     "build": "vue-tsc --noEmit && vite build && electron-builder --config electron-builder.without-core.json && electron-builder --config electron-builder.with-core.json",
-    "preview": "vite preview"
+    "preview": "vite preview",
+    "build-core": "vue-tsc --noEmit && vite build && electron-builder --config electron-builder.with-core.json"
   },
   "devDependencies": {
     "@vicons/antd": "^0.12.0",