From 50898906c1da3027aabd1643809807f5bc74ef9f Mon Sep 17 00:00:00 2001
From: pompurin404 <pompurin404@mihomo.party>
Date: Wed, 11 Sep 2024 23:59:56 +0800
Subject: [PATCH] support global override

---
 changelog.md                                  |  9 +-
 src/main/core/factory.ts                      |  7 +-
 .../components/override/edit-info-modal.tsx   | 12 ++-
 .../src/components/override/override-item.tsx | 93 ++++++++++---------
 .../components/profiles/edit-info-modal.tsx   | 30 ++++--
 src/shared/types.d.ts                         |  1 +
 6 files changed, 93 insertions(+), 59 deletions(-)

diff --git a/changelog.md b/changelog.md
index ba33b4a..3994f40 100644
--- a/changelog.md
+++ b/changelog.md
@@ -4,9 +4,6 @@
 
 ### New Features
 
-- 支持自动开启轻量模式
-
-### Bug Fixes
-
-- 修复日志等级为静默时无法正常启动的问题
-- 修复后台运行时的内存泄漏问题
+- 代理组图标支持SVG格式
+- 优化覆写启用逻辑
+- 覆写支持全局启用
diff --git a/src/main/core/factory.ts b/src/main/core/factory.ts
index a7c1bde..9d7b90f 100644
--- a/src/main/core/factory.ts
+++ b/src/main/core/factory.ts
@@ -4,7 +4,8 @@ import {
   getProfile,
   getProfileItem,
   getOverride,
-  getOverrideItem
+  getOverrideItem,
+  getOverrideConfig
 } from '../config'
 import { mihomoWorkConfigPath, overridePath } from '../utils/dirs'
 import yaml from 'yaml'
@@ -32,8 +33,10 @@ async function overrideProfile(
   current: string | undefined,
   profile: IMihomoConfig
 ): Promise<IMihomoConfig> {
+  const { items = [] } = (await getOverrideConfig()) || {}
+  const globalOverride = items.filter((item) => item.global).map((item) => item.id)
   const { override = [] } = (await getProfileItem(current)) || {}
-  for (const ov of override) {
+  for (const ov of globalOverride.concat(override)) {
     const item = await getOverrideItem(ov)
     const content = await getOverride(ov, item?.ext || 'js')
     switch (item?.ext) {
diff --git a/src/renderer/src/components/override/edit-info-modal.tsx b/src/renderer/src/components/override/edit-info-modal.tsx
index 20dc346..ba9a5e4 100644
--- a/src/renderer/src/components/override/edit-info-modal.tsx
+++ b/src/renderer/src/components/override/edit-info-modal.tsx
@@ -5,7 +5,8 @@ import {
   ModalBody,
   ModalFooter,
   Button,
-  Input
+  Input,
+  Switch
 } from '@nextui-org/react'
 import React, { useState } from 'react'
 import SettingItem from '../base/base-setting-item'
@@ -57,6 +58,15 @@ const EditInfoModal: React.FC<Props> = (props) => {
               />
             </SettingItem>
           )}
+          <SettingItem title="全局启用">
+            <Switch
+              size="sm"
+              isSelected={values.global}
+              onValueChange={(v) => {
+                setValues({ ...values, global: v })
+              }}
+            />
+          </SettingItem>
         </ModalBody>
         <ModalFooter>
           <Button variant="light" onPress={onClose}>
diff --git a/src/renderer/src/components/override/override-item.tsx b/src/renderer/src/components/override/override-item.tsx
index b482cda..7531d9a 100644
--- a/src/renderer/src/components/override/override-item.tsx
+++ b/src/renderer/src/components/override/override-item.tsx
@@ -10,7 +10,7 @@ import {
 } from '@nextui-org/react'
 import { IoMdMore, IoMdRefresh } from 'react-icons/io'
 import dayjs from 'dayjs'
-import React, { Key, useEffect, useState } from 'react'
+import React, { Key, useEffect, useMemo, useState } from 'react'
 import EditFileModal from './edit-file-modal'
 import EditInfoModal from './edit-info-modal'
 import { useSortable } from '@dnd-kit/sortable'
@@ -34,44 +34,6 @@ interface MenuItem {
   className: string
 }
 
-const menuItems: MenuItem[] = [
-  {
-    key: 'edit-info',
-    label: '编辑信息',
-    showDivider: false,
-    color: 'default',
-    className: ''
-  } as MenuItem,
-  {
-    key: 'edit-file',
-    label: '编辑文件',
-    showDivider: false,
-    color: 'default',
-    className: ''
-  } as MenuItem,
-  {
-    key: 'open-file',
-    label: '打开文件',
-    showDivider: false,
-    color: 'default',
-    className: ''
-  } as MenuItem,
-  {
-    key: 'exec-log',
-    label: '执行日志',
-    showDivider: true,
-    color: 'default',
-    className: ''
-  } as MenuItem,
-  {
-    key: 'delete',
-    label: '删除',
-    showDivider: false,
-    color: 'danger',
-    className: 'text-danger'
-  } as MenuItem
-]
-
 const OverrideItem: React.FC<Props> = (props) => {
   const { info, addOverrideItem, removeOverrideItem, mutateOverrideConfig, updateOverrideItem } =
     props
@@ -91,7 +53,49 @@ const OverrideItem: React.FC<Props> = (props) => {
   })
   const transform = tf ? { x: tf.x, y: tf.y, scaleX: 1, scaleY: 1 } : null
   const [disableOpen, setDisableOpen] = useState(false)
-
+  const menuItems: MenuItem[] = useMemo(() => {
+    const list = [
+      {
+        key: 'edit-info',
+        label: '编辑信息',
+        showDivider: false,
+        color: 'default',
+        className: ''
+      } as MenuItem,
+      {
+        key: 'edit-file',
+        label: '编辑文件',
+        showDivider: false,
+        color: 'default',
+        className: ''
+      } as MenuItem,
+      {
+        key: 'open-file',
+        label: '打开文件',
+        showDivider: false,
+        color: 'default',
+        className: ''
+      } as MenuItem,
+      {
+        key: 'exec-log',
+        label: '执行日志',
+        showDivider: true,
+        color: 'default',
+        className: ''
+      } as MenuItem,
+      {
+        key: 'delete',
+        label: '删除',
+        showDivider: false,
+        color: 'danger',
+        className: 'text-danger'
+      } as MenuItem
+    ]
+    if (info.ext === 'yaml') {
+      list.splice(3, 1)
+    }
+    return list
+  }, [info])
   const onMenuAction = (key: Key): void => {
     switch (key) {
       case 'edit-info': {
@@ -219,10 +223,15 @@ const OverrideItem: React.FC<Props> = (props) => {
           </div>
           <div className="flex justify-between">
             <div className={`mt-2 flex justify-start`}>
-              <Chip size="sm" variant="bordered">
+              {info.global && (
+                <Chip size="sm" variant="dot" color="primary" className="mr-2">
+                  全局
+                </Chip>
+              )}
+              <Chip size="sm" variant="bordered" className="mr-2">
                 {info.type === 'local' ? '本地' : '远程'}
               </Chip>
-              <Chip size="sm" variant="bordered" className="ml-2">
+              <Chip size="sm" variant="bordered">
                 {info.ext === 'yaml' ? 'YAML' : 'JavaScript'}
               </Chip>
             </div>
diff --git a/src/renderer/src/components/profiles/edit-info-modal.tsx b/src/renderer/src/components/profiles/edit-info-modal.tsx
index d802a01..fa48e41 100644
--- a/src/renderer/src/components/profiles/edit-info-modal.tsx
+++ b/src/renderer/src/components/profiles/edit-info-modal.tsx
@@ -31,7 +31,13 @@ const EditInfoModal: React.FC<Props> = (props) => {
 
   const onSave = async (): Promise<void> => {
     try {
-      await updateProfileItem(values)
+      await updateProfileItem({
+        ...values,
+        override: values.override?.filter(
+          (i) =>
+            overrideItems.find((t) => t.id === i) && !overrideItems.find((t) => t.id === i)?.global
+        )
+      })
       await restartCore()
       onClose()
     } catch (e) {
@@ -97,8 +103,20 @@ const EditInfoModal: React.FC<Props> = (props) => {
           )}
           <SettingItem title="覆写">
             <div>
+              {overrideItems
+                .filter((i) => i.global)
+                .map((i) => {
+                  return (
+                    <div className="flex mb-2" key={i.id}>
+                      <Button disabled fullWidth variant="flat" size="sm">
+                        {i.name} (全局)
+                      </Button>
+                    </div>
+                  )
+                })}
               {values.override?.map((i) => {
                 if (!overrideItems.find((t) => t.id === i)) return null
+                if (overrideItems.find((t) => t.id === i)?.global) return null
                 return (
                   <div className="flex mb-2" key={i}>
                     <Button disabled fullWidth variant="flat" size="sm">
@@ -112,9 +130,7 @@ const EditInfoModal: React.FC<Props> = (props) => {
                       onPress={() => {
                         setValues({
                           ...values,
-                          override: values.override
-                            ?.filter((i) => overrideItems.find((t) => t.id === i))
-                            .filter((t) => t !== i)
+                          override: values.override?.filter((t) => t !== i)
                         })
                       }}
                     >
@@ -134,14 +150,12 @@ const EditInfoModal: React.FC<Props> = (props) => {
                   onAction={(key) => {
                     setValues({
                       ...values,
-                      override: Array.from(values.override || [])
-                        .filter((i) => overrideItems.find((t) => t.id === i))
-                        .concat(key.toString())
+                      override: Array.from(values.override || []).concat(key.toString())
                     })
                   }}
                 >
                   {overrideItems
-                    .filter((i) => !values.override?.includes(i.id))
+                    .filter((i) => !values.override?.includes(i.id) && !i.global)
                     .map((i) => (
                       <DropdownItem key={i.id}>{i.name}</DropdownItem>
                     ))}
diff --git a/src/shared/types.d.ts b/src/shared/types.d.ts
index 9dea974..299f337 100644
--- a/src/shared/types.d.ts
+++ b/src/shared/types.d.ts
@@ -389,6 +389,7 @@ interface IOverrideItem {
   ext: 'js' | 'yaml'
   name: string
   updated: number
+  global?: boolean
   url?: string
   file?: string
 }