From d9997054795ed2b4141fae9de276e35fb1f4783a Mon Sep 17 00:00:00 2001
From: 233boy <31657650+233boy@users.noreply.github.com>
Date: Wed, 8 May 2024 20:02:55 +0800
Subject: [PATCH] add dns set

---
 README.md   |  1 +
 src/core.sh | 10 ++++++++-
 src/dns.sh  | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/help.sh |  1 +
 v2ray.sh    |  2 +-
 5 files changed, 70 insertions(+), 2 deletions(-)
 create mode 100644 src/dns.sh

diff --git a/README.md b/README.md
index 73806d0e4..d451dc2ca 100644
--- a/README.md
+++ b/README.md
@@ -86,6 +86,7 @@ Usage: v2ray [options]... [args]...
    web [name] [domain]                             更改伪装网站
 
 进阶:
+   dns [...]                                       设置 DNS
    dd, ddel [name...]                              删除多个配置**
    fix [name]                                      修复一个配置
    fix-all                                         修复全部配置
diff --git a/src/core.sh b/src/core.sh
index 4b1e771b0..902fa8074 100644
--- a/src/core.sh
+++ b/src/core.sh
@@ -1700,7 +1700,7 @@ is_main_menu() {
         show_help
         ;;
     9)
-        ask list is_do_other "启用BBR 查看日志 查看错误日志 测试运行 重装脚本"
+        ask list is_do_other "启用BBR 查看日志 查看错误日志 测试运行 重装脚本 设置DNS"
         case $REPLY in
         1)
             load bbr.sh
@@ -1718,6 +1718,10 @@ is_main_menu() {
         5)
             get reinstall
             ;;
+        6)
+            load dns.sh
+            dns_set
+            ;;
         esac
         ;;
     10)
@@ -1795,6 +1799,10 @@ main() {
         [[ $is_api_fail ]] && manage restart &
         [[ $is_del_host ]] && manage restart caddy &
         ;;
+    dns)
+        load dns.sh
+        dns_set ${@:2}
+        ;;
     debug)
         is_debug=1
         get info $2
diff --git a/src/dns.sh b/src/dns.sh
new file mode 100644
index 000000000..420ab3fe8
--- /dev/null
+++ b/src/dns.sh
@@ -0,0 +1,58 @@
+is_dns_list=(
+    1.1.1.1
+    8.8.8.8
+    https://dns.google/dns-query
+    https://cloudflare-dns.com/dns-query
+    https://family.cloudflare-dns.com/dns-query
+    set
+    none
+)
+dns_set() {
+    if [[ $1 ]]; then
+        case ${1,,} in
+        11 | 1111)
+            is_dns_use=${is_dns_list[0]}
+            ;;
+        88 | 8888)
+            is_dns_use=${is_dns_list[1]}
+            ;;
+        gg | google)
+            is_dns_use=${is_dns_list[2]}
+            ;;
+        cf | cloudflare)
+            is_dns_use=${is_dns_list[3]}
+            ;;
+        nosex | family)
+            is_dns_use=${is_dns_list[4]}
+            ;;
+        set)
+            if [[ $2 ]]; then
+                is_dns_use=${2,,}
+            else
+                ask string is_dns_use "请输入 DNS: "
+            fi
+            ;;
+        none)
+            is_dns_use=none
+            ;;
+        *)
+            err "无法识别 DNS 参数: $@"
+            ;;
+        esac
+    else
+        is_tmp_list=(${is_dns_list[@]})
+        ask list dns_pick
+        is_dns_use=${is_dns_list[$REPLY - 1]}
+        if [[ $is_dns_use == "set" ]]; then
+            ask string is_dns_use "请输入 DNS: "
+        fi
+    fi
+    is_new_dns=$(sed s/https/https+local/ <<<$is_dns_use)
+    if [[ $is_new_dns == "none" ]]; then
+        cat <<<$(jq '.dns={}' $is_config_json) >$is_config_json
+    else
+        cat <<<$(jq '.dns.servers=["'$is_new_dns'"]' $is_config_json) >$is_config_json
+    fi
+    manage restart &
+    msg "\n已更新 DNS 为: $(_green $is_new_dns)\n"
+}
\ No newline at end of file
diff --git a/src/help.sh b/src/help.sh
index b83fab152..4a83dc48d 100644
--- a/src/help.sh
+++ b/src/help.sh
@@ -40,6 +40,7 @@ show_help() {
             "   new [name] [...]                                更改协议"
             "   web [name] [domain]                             更改伪装网站\n"
             "进阶:"
+            "   dns [...]                                       设置 DNS"
             "   dd, ddel [name...]                              删除多个配置**"
             "   fix [name]                                      修复一个配置"
             "   fix-all                                         修复全部配置"
diff --git a/v2ray.sh b/v2ray.sh
index d8c1d45eb..d85a492fd 100644
--- a/v2ray.sh
+++ b/v2ray.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
 args=$@
-is_sh_ver=v4.18
+is_sh_ver=v4.19
 
 . /etc/v2ray/sh/src/init.sh