From 0e9991a84600ac0fff9986058a874067308fee6e Mon Sep 17 00:00:00 2001
From: mercury <mercury.kd@gmail.com>
Date: Mon, 5 Jun 2023 12:46:45 +0400
Subject: [PATCH] default DNS for wireguard

---
 app/bot.php  | 45 +++++++++++++++++++++++++++++++++++++++++++--
 app/i18n.php |  4 ++++
 2 files changed, 47 insertions(+), 2 deletions(-)

diff --git a/app/bot.php b/app/bot.php
index ad3297f..d162407 100644
--- a/app/bot.php
+++ b/app/bot.php
@@ -14,6 +14,7 @@ class Bot
         $this->ip       = getenv('IP');
         $this->i18n     = $i18n;
         $this->language = $this->getPacConf()['language'] ?: 'en';
+        $this->dns      = '1.1.1.1, 8.8.8.8';
     }
 
     public function input()
@@ -129,6 +130,9 @@ class Bot
             case preg_match('~^/setSecret$~', $this->input['callback'], $m):
                 $this->setSecret();
                 break;
+            case preg_match('~^/defaultDNS (?P<arg>\d+(?:_(?:-)?\d+)?)$~', $this->input['callback'], $m):
+                $this->defaultDNS(...explode('_', $m['arg']));
+                break;
             case preg_match('~^/subnet (?P<arg>\d+(?:_(?:-)?\d+)?)$~', $this->input['callback'], $m):
                 $this->subnet(...explode('_', $m['arg']));
                 break;
@@ -1530,7 +1534,8 @@ DNS-over-HTTPS with IP:
             }
         }
         $text = "Menu -> Wireguard\n\n<code>" . implode(PHP_EOL, $text) . '</code>';
-        $bt = $this->getPacConf()['blocktorrent'];
+        $bt   = $this->getPacConf()['blocktorrent'];
+        $dns  = $this->getPacConf()['dns'];
         $data = [
             [
                 [
@@ -1552,6 +1557,12 @@ DNS-over-HTTPS with IP:
                     'callback_data' => "/subnet $page",
                 ],
             ],
+            [
+                [
+                    'text'          =>  $this->i18n('defaultDNS') . ': ' . ($dns ?: $this->dns),
+                    'callback_data' => "/defaultDNS $page",
+                ],
+            ],
         ];
         if ($clients = $this->getClients($page)) {
             $data = array_merge($data, $clients);
@@ -1566,6 +1577,34 @@ DNS-over-HTTPS with IP:
         ];
     }
 
+    public function defaultDNS($page = 0)
+    {
+        $r = $this->send(
+            $this->input['chat'],
+            "@{$this->input['username']} enter dns separated by commas",
+            $this->input['message_id'],
+            reply: 'enter dns separated by commas',
+        );
+        $_SESSION['reply'][$r['result']['message_id']] = [
+            'start_message'  => $this->input['message_id'],
+            'start_callback' => $this->input['callback_id'],
+            'callback'       => 'setDNS',
+            'args'           => [$page],
+        ];
+    }
+
+    public function setDNS($text, $page = 0)
+    {
+        $c = $this->getPacConf();
+        if ($text) {
+            $c['dns'] = $text;
+        } else {
+            unset($c['dns']);
+        }
+        $this->setPacConf($c);
+        $this->menu('wg', $page);
+    }
+
     public function subnetAdd($page = 0)
     {
         $r = $this->send(
@@ -2623,6 +2662,9 @@ DNS-over-HTTPS with IP:
     public function createConfig($data)
     {
         $conf[] = "[Interface]";
+        if (empty($data['interface']['DNS'])) {
+            $data['interface']['DNS'] = $this->getPacConf()['dns'] ?: $this->dns;
+        }
         foreach ($data['interface'] as $k => $v) {
             $conf[] = "$k = $v";
         }
@@ -2673,7 +2715,6 @@ DNS-over-HTTPS with IP:
                 'PrivateKey' => $private_peer_key,
                 'Address'    => "$client_ip/32",
                 'MTU'        => 1350,
-                'DNS'        => '10.10.0.5',
             ],
             'peers' => [
                 [
diff --git a/app/i18n.php b/app/i18n.php
index 7c14785..6f2b77e 100644
--- a/app/i18n.php
+++ b/app/i18n.php
@@ -233,4 +233,8 @@ $i = [
         'en' => 'setSecret',
         'ru' => 'установить ключ',
     ],
+    'defaultDNS'  => [
+        'en' => 'defaultDNS',
+        'ru' => 'дефолтный днс',
+    ],
 ];