From 72417e8ef6f923f5865095d4aefadfcbe54ff003 Mon Sep 17 00:00:00 2001
From: pompurin404 <pompurin404@mihomo.party>
Date: Sat, 10 Aug 2024 15:35:47 +0800
Subject: [PATCH] fix dns

---
 src/main/config/controledMihomo.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main/config/controledMihomo.ts b/src/main/config/controledMihomo.ts
index 34d6aa5..9af2ae5 100644
--- a/src/main/config/controledMihomo.ts
+++ b/src/main/config/controledMihomo.ts
@@ -3,6 +3,7 @@ import yaml from 'yaml'
 import fs from 'fs'
 import { getAxios, startMihomoMemory, startMihomoTraffic } from '../core/mihomoApi'
 import { generateProfile } from '../resolve/factory'
+import { getAppConfig } from './app'
 
 export let controledMihomoConfig: Partial<IMihomoConfig> // mihomo.yaml
 
@@ -14,6 +15,7 @@ export function getControledMihomoConfig(force = false): Partial<IMihomoConfig>
 }
 
 export function setControledMihomoConfig(patch: Partial<IMihomoConfig>): void {
+  const { useNameserverPolicy } = getAppConfig()
   if (patch.tun) {
     const oldTun = controledMihomoConfig.tun || {}
     const newTun = Object.assign(oldTun, patch.tun)
@@ -21,8 +23,10 @@ export function setControledMihomoConfig(patch: Partial<IMihomoConfig>): void {
   }
   if (patch.dns) {
     const oldDns = controledMihomoConfig.dns || {}
-    const newDns = { ...patch.dns }
-    newDns.enable = oldDns.enable
+    const newDns = Object.assign(oldDns, patch.dns)
+    if (!useNameserverPolicy) {
+      delete newDns['nameserver-policy']
+    }
     patch.dns = newDns
   }
   if (patch.sniffer) {