From b1fbdcc566f5df739522659018ec1660825ef14b Mon Sep 17 00:00:00 2001
From: zhengkunwang <31820853+zhengkunwang223@users.noreply.github.com>
Date: Tue, 31 Dec 2024 17:57:55 +0800
Subject: [PATCH] =?UTF-8?q?feat(appstore):=20Add=20Synchronization=20of=20?=
 =?UTF-8?q?Main=20Node=20Application=20Store=20Pa=E2=80=A6=20(#7612)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 agent/app/task/task.go                        | 10 +++---
 core/buserr/errors.go                         | 11 +++++++
 core/utils/files/files.go                     |  8 +++++
 frontend/src/api/interface/app.ts             |  5 +++
 frontend/src/api/modules/app.ts               |  4 +++
 frontend/src/components/backup/index.vue      |  4 ++-
 frontend/src/components/dialog-pro/index.vue  |  2 +-
 frontend/src/components/log-file/index.vue    |  4 ++-
 frontend/src/lang/modules/zh.ts               |  1 +
 frontend/src/store/modules/global.ts          |  1 +
 frontend/src/utils/util.ts                    | 31 ++++++++++++++++---
 frontend/src/views/app-store/apps/index.vue   | 12 +++++--
 .../src/views/app-store/setting/index.vue     | 19 ++++++++++--
 frontend/src/views/cronjob/backup/index.vue   |  4 ++-
 .../src/views/host/file-management/index.vue  |  4 +--
 15 files changed, 100 insertions(+), 20 deletions(-)

diff --git a/agent/app/task/task.go b/agent/app/task/task.go
index 749387941..ce0bc4332 100644
--- a/agent/app/task/task.go
+++ b/agent/app/task/task.go
@@ -251,22 +251,22 @@ func (t *Task) Logf(format string, v ...any) {
 }
 
 func (t *Task) LogFailed(msg string) {
-	t.Logger.Printf(msg + i18n.GetMsgByKey("Failed"))
+	t.Logger.Println(msg + i18n.GetMsgByKey("Failed"))
 }
 
 func (t *Task) LogFailedWithErr(msg string, err error) {
-	t.Logger.Printf(fmt.Sprintf("%s %s : %s", msg, i18n.GetMsgByKey("Failed"), err.Error()))
+	t.Logger.Println(fmt.Sprintf("%s %s : %s", msg, i18n.GetMsgByKey("Failed"), err.Error()))
 }
 
 func (t *Task) LogSuccess(msg string) {
-	t.Logger.Printf(msg + i18n.GetMsgByKey("Success"))
+	t.Logger.Println(msg + i18n.GetMsgByKey("Success"))
 }
 func (t *Task) LogSuccessF(format string, v ...any) {
-	t.Logger.Printf(fmt.Sprintf(format, v...) + i18n.GetMsgByKey("Success"))
+	t.Logger.Println(fmt.Sprintf(format, v...) + i18n.GetMsgByKey("Success"))
 }
 
 func (t *Task) LogStart(msg string) {
-	t.Logger.Printf(fmt.Sprintf("%s%s", i18n.GetMsgByKey("Start"), msg))
+	t.Logger.Println(fmt.Sprintf("%s%s", i18n.GetMsgByKey("Start"), msg))
 }
 
 func (t *Task) LogWithOps(operate, msg string) {
diff --git a/core/buserr/errors.go b/core/buserr/errors.go
index d4805eedc..5015ff84b 100644
--- a/core/buserr/errors.go
+++ b/core/buserr/errors.go
@@ -65,3 +65,14 @@ func WithMap(Key string, maps map[string]interface{}, err error) BusinessError {
 		Err: err,
 	}
 }
+
+func WithName(Key string, name string) BusinessError {
+	paramMap := map[string]interface{}{}
+	if name != "" {
+		paramMap["name"] = name
+	}
+	return BusinessError{
+		Msg: Key,
+		Map: paramMap,
+	}
+}
diff --git a/core/utils/files/files.go b/core/utils/files/files.go
index 0aab25873..f0cab6e8a 100644
--- a/core/utils/files/files.go
+++ b/core/utils/files/files.go
@@ -161,3 +161,11 @@ func DownloadFileWithProxy(url, dst string) error {
 	}
 	return nil
 }
+
+func Stat(path string) bool {
+	_, err := os.Stat(path)
+	if err != nil && os.IsNotExist(err) {
+		return false
+	}
+	return true
+}
diff --git a/frontend/src/api/interface/app.ts b/frontend/src/api/interface/app.ts
index 81ff75521..3b2acc535 100644
--- a/frontend/src/api/interface/app.ts
+++ b/frontend/src/api/interface/app.ts
@@ -268,4 +268,9 @@ export namespace App {
     export interface AppStoreConfig {
         defaultDomain: string;
     }
+
+    export interface CustomAppStoreConfig {
+        status: string;
+        imagePrefix: string;
+    }
 }
diff --git a/frontend/src/api/modules/app.ts b/frontend/src/api/modules/app.ts
index a7a6c6cfa..6183c28ec 100644
--- a/frontend/src/api/modules/app.ts
+++ b/frontend/src/api/modules/app.ts
@@ -122,3 +122,7 @@ export const UpdateAppStoreConfig = (req: App.AppStoreConfig) => {
 export const SyncCutomAppStore = (req: App.AppStoreSync) => {
     return http.post(`/custom/app/sync`, req);
 };
+
+export const getCurrentNodeCustomAppConfig = () => {
+    return http.get<App.CustomAppStoreConfig>(`/custom/app/config`);
+};
diff --git a/frontend/src/components/backup/index.vue b/frontend/src/components/backup/index.vue
index d68eeb893..64fe839ee 100644
--- a/frontend/src/components/backup/index.vue
+++ b/frontend/src/components/backup/index.vue
@@ -114,6 +114,8 @@ import { Backup } from '@/api/interface/backup';
 import router from '@/routers';
 import { MsgSuccess } from '@/utils/message';
 import TaskLog from '@/components/task-log/index.vue';
+import { GlobalStore } from '@/store';
+const globalStore = GlobalStore();
 
 const selects = ref<any>([]);
 const loading = ref();
@@ -306,7 +308,7 @@ const onDownload = async (row: Backup.RecordInfo) => {
         fileName: row.fileName,
     };
     await downloadBackupRecord(params).then(async (res) => {
-        downloadFile(res.data);
+        downloadFile(res.data, globalStore.currentNode);
     });
 };
 
diff --git a/frontend/src/components/dialog-pro/index.vue b/frontend/src/components/dialog-pro/index.vue
index 087af7103..6ddf6f86d 100644
--- a/frontend/src/components/dialog-pro/index.vue
+++ b/frontend/src/components/dialog-pro/index.vue
@@ -28,7 +28,7 @@
 
 <script lang="ts" setup>
 import { computed, useSlots } from 'vue';
-defineOptions({ name: 'DrawerPro' });
+defineOptions({ name: 'DialogPro' });
 
 const props = defineProps({
     title: String,
diff --git a/frontend/src/components/log-file/index.vue b/frontend/src/components/log-file/index.vue
index 580ffd43b..f2e48ea0b 100644
--- a/frontend/src/components/log-file/index.vue
+++ b/frontend/src/components/log-file/index.vue
@@ -34,6 +34,8 @@
 import { nextTick, onMounted, onUnmounted, reactive, ref } from 'vue';
 import { downloadFile } from '@/utils/util';
 import { ReadByLine } from '@/api/modules/files';
+import { GlobalStore } from '@/store';
+const globalStore = GlobalStore();
 
 interface LogProps {
     id?: number;
@@ -146,7 +148,7 @@ const changeLoading = () => {
 
 const onDownload = async () => {
     changeLoading();
-    downloadFile(logPath.value);
+    downloadFile(logPath.value, globalStore.currentNode);
     changeLoading();
 };
 
diff --git a/frontend/src/lang/modules/zh.ts b/frontend/src/lang/modules/zh.ts
index c308c1679..0f6a6a907 100644
--- a/frontend/src/lang/modules/zh.ts
+++ b/frontend/src/lang/modules/zh.ts
@@ -1859,6 +1859,7 @@ const message = {
         defaultWebDomainHepler: '默认访问用于应用端口跳转，例如应用端口为 8080 则跳转地址为http(s)://默认访问地址:8080',
         webUIConfig: '请在应用参数或者应用商店设置处添加访问地址',
         toLink: '跳转',
+        customAppHelper: '当前使用的是主节点应用商店包，修改配置请在主节点操作',
     },
     website: {
         website: '网站',
diff --git a/frontend/src/store/modules/global.ts b/frontend/src/store/modules/global.ts
index de1d90815..84faf690c 100644
--- a/frontend/src/store/modules/global.ts
+++ b/frontend/src/store/modules/global.ts
@@ -51,6 +51,7 @@ const GlobalStore = defineStore({
             state.themeConfig.isGold ||
             (state.themeConfig.theme === 'auto' && window.matchMedia('(prefers-color-scheme: dark)').matches),
         isDarkGoldTheme: (state) => state.themeConfig.isGold && state.isMasterProductPro,
+        isMaster: (state) => state.currentNode === 'local',
     },
     actions: {
         setOpenMenuTabs(openMenuTabs: boolean) {
diff --git a/frontend/src/utils/util.ts b/frontend/src/utils/util.ts
index 56ce8953b..714e51e60 100644
--- a/frontend/src/utils/util.ts
+++ b/frontend/src/utils/util.ts
@@ -519,10 +519,33 @@ export function toLowerCase(str: string) {
     return str.toLowerCase();
 }
 
-export function downloadFile(filePath: string) {
-    let url = `${import.meta.env.VITE_API_URL as string}/files/download?`;
-    let path = encodeURIComponent(filePath);
-    window.open(url + 'path=' + path, '_blank');
+export function downloadFile(filePath: string, currentNode: string) {
+    const url = `${import.meta.env.VITE_API_URL as string}/files/download`;
+    const path = encodeURIComponent(filePath);
+
+    fetch(`${url}?path=${path}`, {
+        method: 'GET',
+        headers: {
+            CurrentNode: currentNode,
+        },
+    })
+        .then((response) => {
+            if (!response.ok) {
+                throw new Error('Network response was not ok');
+            }
+            return response.blob();
+        })
+        .then((blob) => {
+            const downloadUrl = window.URL.createObjectURL(blob);
+            const a = document.createElement('a');
+            a.href = downloadUrl;
+            a.download = filePath.split('/').pop() || 'downloaded-file';
+            document.body.appendChild(a);
+            a.click();
+            a.remove();
+            window.URL.revokeObjectURL(downloadUrl);
+        })
+        .catch(() => {});
 }
 
 export function downloadWithContent(content: string, fileName: string) {
diff --git a/frontend/src/views/app-store/apps/index.vue b/frontend/src/views/app-store/apps/index.vue
index d6432ab12..27d3c1f8a 100644
--- a/frontend/src/views/app-store/apps/index.vue
+++ b/frontend/src/views/app-store/apps/index.vue
@@ -175,7 +175,14 @@
 <script lang="ts" setup>
 import { App } from '@/api/interface/app';
 import { onMounted, reactive, ref, computed } from 'vue';
-import { GetAppTags, SearchApp, SyncApp, SyncCutomAppStore, SyncLocalApp } from '@/api/modules/app';
+import {
+    GetAppTags,
+    SearchApp,
+    SyncApp,
+    SyncCutomAppStore,
+    SyncLocalApp,
+    getCurrentNodeCustomAppConfig,
+} from '@/api/modules/app';
 import Install from '../detail/install/index.vue';
 import router from '@/routers';
 import { MsgSuccess } from '@/utils/message';
@@ -184,7 +191,6 @@ import { getLanguage, newUUID } from '@/utils/util';
 import Detail from '../detail/index.vue';
 import TaskLog from '@/components/task-log/index.vue';
 import { storeToRefs } from 'pinia';
-import { GetCustomAppStoreConfig } from '@/xpack/api/modules/app';
 
 const globalStore = GlobalStore();
 const { isProductPro } = storeToRefs(globalStore);
@@ -350,7 +356,7 @@ onMounted(async () => {
     }
     search(req);
     if (isProductPro.value) {
-        const res = await GetCustomAppStoreConfig();
+        const res = await getCurrentNodeCustomAppConfig();
         if (res && res.data) {
             syncCustomAppstore.value = res.data.status === 'enable';
         }
diff --git a/frontend/src/views/app-store/setting/index.vue b/frontend/src/views/app-store/setting/index.vue
index a6fc5fbdb..5fb93b7b2 100644
--- a/frontend/src/views/app-store/setting/index.vue
+++ b/frontend/src/views/app-store/setting/index.vue
@@ -28,7 +28,10 @@
                             </el-input>
                             <span class="input-help">{{ $t('app.defaultWebDomainHepler') }}</span>
                         </el-form-item>
-                        <CustomSetting v-if="isProductPro" />
+                        <CustomSetting v-if="isProductPro && globalStore.isMaster" />
+                        <el-form-item v-if="!globalStore.isMaster && useCustomApp">
+                            <el-text type="warning">{{ $t('app.customAppHelper') }}</el-text>
+                        </el-form-item>
                     </el-col>
                 </el-row>
             </el-form>
@@ -38,7 +41,7 @@
 </template>
 
 <script setup lang="ts">
-import { GetAppStoreConfig } from '@/api/modules/app';
+import { GetAppStoreConfig, getCurrentNodeCustomAppConfig } from '@/api/modules/app';
 import { Rules } from '@/global/form-rules';
 import { FormRules } from 'element-plus';
 import CustomSetting from '@/xpack/views/appstore/index.vue';
@@ -59,6 +62,7 @@ const loading = ref(false);
 const configForm = ref();
 const protocol = ref('http://');
 const domainRef = ref();
+const useCustomApp = ref(false);
 
 function getUrl(url: string) {
     const regex = /^(https?:\/\/)(.*)/;
@@ -99,7 +103,18 @@ const setDefaultDomain = () => {
     });
 };
 
+const getNodeConfig = async () => {
+    if (globalStore.isMaster) {
+        return;
+    }
+    const res = await getCurrentNodeCustomAppConfig();
+    if (res && res.data) {
+        useCustomApp.value = res.data.status === 'enable';
+    }
+};
+
 onMounted(() => {
     search();
+    getNodeConfig();
 });
 </script>
diff --git a/frontend/src/views/cronjob/backup/index.vue b/frontend/src/views/cronjob/backup/index.vue
index 42db0724b..a450207fd 100644
--- a/frontend/src/views/cronjob/backup/index.vue
+++ b/frontend/src/views/cronjob/backup/index.vue
@@ -54,6 +54,8 @@ import i18n from '@/lang';
 import { downloadBackupRecord, searchBackupRecordsByCronjob } from '@/api/modules/backup';
 import { Backup } from '@/api/interface/backup';
 import { MsgError } from '@/utils/message';
+import { GlobalStore } from '@/store';
+const globalStore = GlobalStore();
 
 const selects = ref<any>([]);
 const loading = ref();
@@ -116,7 +118,7 @@ const onDownload = async (row: Backup.RecordInfo) => {
     await downloadBackupRecord(params)
         .then(async (res) => {
             loading.value = false;
-            downloadFile(res.data);
+            downloadFile(res.data, globalStore.currentNode);
         })
         .catch(() => {
             loading.value = false;
diff --git a/frontend/src/views/host/file-management/index.vue b/frontend/src/views/host/file-management/index.vue
index be6646314..8628febe1 100644
--- a/frontend/src/views/host/file-management/index.vue
+++ b/frontend/src/views/host/file-management/index.vue
@@ -311,7 +311,7 @@ import {
     RemoveFavorite,
     SearchFavorite,
 } from '@/api/modules/files';
-import { computeSize, copyText, dateFormat, downloadFile, getFileType, getIcon, getRandomStr } from '@/utils/util';
+import { computeSize, copyText, dateFormat, getFileType, getIcon, getRandomStr, downloadFile } from '@/utils/util';
 import { StarFilled, Star, Top, Right, Close } from '@element-plus/icons-vue';
 import { File } from '@/api/interface/file';
 import { Mimetypes, Languages } from '@/global/mimetype';
@@ -810,7 +810,7 @@ const openPaste = () => {
 };
 
 const openDownload = (file: File.File) => {
-    downloadFile(file.path);
+    downloadFile(file.path, globalStore.currentNode);
 };
 
 const openDetail = (row: File.File) => {