From e246bcf12b6ed0f449917c945d6b64434d523ce9 Mon Sep 17 00:00:00 2001
From: Tw93 <tw93@qq.com>
Date: Mon, 28 Aug 2023 18:31:59 +0800
Subject: [PATCH] :bug: Fix icon download

---
 script/build_with_pake_cli.mjs | 40 +++++++++++++++++++++++++---------
 1 file changed, 30 insertions(+), 10 deletions(-)

diff --git a/script/build_with_pake_cli.mjs b/script/build_with_pake_cli.mjs
index a334339fc..98ebbb458 100644
--- a/script/build_with_pake_cli.mjs
+++ b/script/build_with_pake_cli.mjs
@@ -1,6 +1,7 @@
 import shelljs from 'shelljs';
 import axios from 'axios';
 import fs from 'fs';
+import os from 'os';
 import { fileURLToPath } from 'url';
 import path, { dirname } from 'path';
 
@@ -27,16 +28,35 @@ console.log('===========================\n');
 cd('node_modules/pake-cli');
 let params = `node cli.js ${process.env.URL} --name ${process.env.NAME}`;
 
-if (process.env.ICON) {
-  const iconPath = path.join(__dirname, 'icon');
-  axios({
-    method: 'get',
-    url: process.env.ICON,
-    responseType: 'stream',
-  }).then(function (response) {
-    response.data.pipe(fs.createWriteStream(iconPath));
-    params = `${params} --icon ${iconPath}`;
-  });
+// Download Icons
+if (process.env.ICON && process.env.ICON !== '') {
+  let iconFile;
+  switch (os.type()) {
+    case 'Linux':
+      iconFile = 'icon.png';
+      break;
+    case 'Darwin':
+      iconFile = 'icon.icns';
+      break;
+    case 'Windows_NT':
+      iconFile = 'icon.ico';
+      break;
+    default:
+      console.log("Unable to detect your OS system, won't download the icon!");
+      return;
+  }
+
+  axios
+    .get(process.env.ICON, { responseType: 'arraybuffer' })
+    .then(response => {
+      fs.writeFileSync(iconFile, response.data);
+      params = `${params} --icon ${iconFile}`;
+    })
+    .catch(error => {
+      console.error('Error occurred during icon download: ', error);
+    });
+} else {
+  console.log("Won't download the icon as ICON environment variable is not defined!");
 }
 
 params = `${params} --height ${process.env.HEIGHT} --width ${process.env.WIDTH}`;