From 42b85f74a9b585878dd11525cd31a6a8fb0175ce Mon Sep 17 00:00:00 2001
From: UUBulb <35923940+uubulb@users.noreply.github.com>
Date: Thu, 30 Jan 2025 12:15:21 +0800
Subject: [PATCH] ddns: allow overriding domains per configuration (#111)

---
 src/components/server.tsx       | 74 ++++++++++++++++++++++++++-------
 src/lib/utils.ts                |  2 +-
 src/locales/en/translation.json |  3 +-
 src/types/api.ts                |  3 ++
 4 files changed, 65 insertions(+), 17 deletions(-)

diff --git a/src/components/server.tsx b/src/components/server.tsx
index 99dee14..97864dc 100644
--- a/src/components/server.tsx
+++ b/src/components/server.tsx
@@ -49,6 +49,22 @@ const serverFormSchema = z.object({
     enable_ddns: asOptionalField(z.boolean()),
     ddns_profiles: asOptionalField(z.array(z.number())),
     ddns_profiles_raw: asOptionalField(z.string()),
+    override_ddns_domains: asOptionalField(z.record(z.coerce.number().int(), z.array(z.string()))),
+    override_ddns_domains_raw: asOptionalField(
+        z.string().refine(
+            (val) => {
+                try {
+                    JSON.parse(val)
+                    return true
+                } catch (e) {
+                    return false
+                }
+            },
+            {
+                message: "Invalid JSON string",
+            },
+        ),
+    ),
 })
 
 export const ServerCard: React.FC<ServerCardProps> = ({ data, mutate }) => {
@@ -58,6 +74,9 @@ export const ServerCard: React.FC<ServerCardProps> = ({ data, mutate }) => {
         defaultValues: {
             ...data,
             ddns_profiles_raw: data.ddns_profiles ? conv.arrToStr(data.ddns_profiles) : undefined,
+            override_ddns_domains_raw: data.override_ddns_domains
+                ? JSON.stringify(data.override_ddns_domains)
+                : undefined,
         },
         resetOptions: {
             keepDefaultValues: false,
@@ -71,6 +90,9 @@ export const ServerCard: React.FC<ServerCardProps> = ({ data, mutate }) => {
             values.ddns_profiles = values.ddns_profiles_raw
                 ? conv.strToArr(values.ddns_profiles_raw).map(Number)
                 : undefined
+            values.override_ddns_domains = values.override_ddns_domains_raw
+                ? JSON.parse(values.override_ddns_domains_raw)
+                : undefined
             await updateServer(data!.id!, values)
         } catch (e) {
             console.error(e)
@@ -124,21 +146,43 @@ export const ServerCard: React.FC<ServerCardProps> = ({ data, mutate }) => {
                                         </FormItem>
                                     )}
                                 />
-                                <FormField
-                                    control={form.control}
-                                    name="ddns_profiles_raw"
-                                    render={({ field }) => (
-                                        <FormItem>
-                                            <FormLabel>
-                                                {t("DDNSProfiles") + t("SeparateWithComma")}
-                                            </FormLabel>
-                                            <FormControl>
-                                                <Input placeholder="1,2,3" {...field} />
-                                            </FormControl>
-                                            <FormMessage />
-                                        </FormItem>
-                                    )}
-                                />
+                                {form.watch("enable_ddns") ? (
+                                    <>
+                                        <FormField
+                                            control={form.control}
+                                            name="ddns_profiles_raw"
+                                            render={({ field }) => (
+                                                <FormItem>
+                                                    <FormLabel>
+                                                        {t("DDNSProfiles") + t("SeparateWithComma")}
+                                                    </FormLabel>
+                                                    <FormControl>
+                                                        <Input placeholder="1,2,3" {...field} />
+                                                    </FormControl>
+                                                    <FormMessage />
+                                                </FormItem>
+                                            )}
+                                        />
+                                        <FormField
+                                            control={form.control}
+                                            name="override_ddns_domains_raw"
+                                            render={({ field }) => (
+                                                <FormItem>
+                                                    <FormLabel>
+                                                        {t("OverrideDDNSDomains")}
+                                                    </FormLabel>
+                                                    <FormControl>
+                                                        <Textarea className="resize-y" {...field} />
+                                                    </FormControl>
+                                                    <FormMessage />
+                                                </FormItem>
+                                            )}
+                                        />
+                                    </>
+                                ) : (
+                                    <></>
+                                )}
+
                                 <FormField
                                     control={form.control}
                                     name="enable_ddns"
diff --git a/src/lib/utils.ts b/src/lib/utils.ts
index 5b07416..c532f19 100644
--- a/src/lib/utils.ts
+++ b/src/lib/utils.ts
@@ -17,7 +17,7 @@ export function asOptionalField<T extends z.ZodTypeAny>(schema: T) {
 }
 
 export const conv = {
-    recordToStr: (rec: Record<string, boolean>) => {
+    recordToStr: <T>(rec: Record<string, T>) => {
         const arr: string[] = []
         for (const key in rec) {
             arr.push(key)
diff --git a/src/locales/en/translation.json b/src/locales/en/translation.json
index 57207ac..d36cbda 100644
--- a/src/locales/en/translation.json
+++ b/src/locales/en/translation.json
@@ -179,5 +179,6 @@
     "ConfirmBlock": "Confirm Block",
     "RejectPassword": "Reject Password Login",
     "EmptyText": "Text is empty",
-    "EmptyNote": "You didn't have any note."
+    "EmptyNote": "You didn't have any note.",
+    "OverrideDDNSDomains": "Override DDNS Domains (per configuration)"
 }
diff --git a/src/types/api.ts b/src/types/api.ts
index f10cf60..5fffd37 100644
--- a/src/types/api.ts
+++ b/src/types/api.ts
@@ -411,6 +411,7 @@ export interface ModelNAT {
 
 export interface ModelNATForm {
     domain: string
+    enabled: boolean
     host: string
     /** @minLength 1 */
     name: string
@@ -560,6 +561,7 @@ export interface ModelServer {
     name: string
     /** 管理员可见备注 */
     note: string
+    override_ddns_domains?: Record<string, string[]>
     /** 公开备注 */
     public_note: string
     state: ModelHostState
@@ -582,6 +584,7 @@ export interface ModelServerForm {
     name: string
     /** 管理员可见备注 */
     note?: string
+    override_ddns_domains?: Record<string, string[]>
     /** 公开备注 */
     public_note?: string
 }