pangolin/src/components/DomainInfoCard.tsx

"use client";

import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
import {
    InfoSection,
    InfoSectionContent,
    InfoSections,
    InfoSectionTitle
} from "@app/components/InfoSection";
import { useTranslations } from "next-intl";
import { useEnvContext } from "@app/hooks/useEnvContext";
import { useDomainContext } from "@app/hooks/useDomainContext";
import { SettingsContainer, SettingsSection, SettingsSectionBody, SettingsSectionDescription, SettingsSectionFooter, SettingsSectionForm, SettingsSectionHeader, SettingsSectionTitle } from "./Settings";
import { Button } from "./ui/button";
import {
    Form,
    FormControl,
    FormField,
    FormItem,
    FormLabel,
    FormMessage,
    FormDescription
} from "@app/components/ui/form";
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "./ui/select";
import { Input } from "./ui/input";
import { useForm } from "react-hook-form";
import z from "zod";
import { toASCII } from "punycode";
import { zodResolver } from "@hookform/resolvers/zod";
import { build } from "@server/build";
import { Switch } from "./ui/switch";
import { useEffect, useState } from "react";
import DNSRecordsTable, { DNSRecordRow } from "./DNSRecordTable";
import { createApiClient } from "@app/lib/api";
import { useToast } from "@app/hooks/useToast";
import { formatAxiosError } from "@app/lib/api";
import { Badge } from "./ui/badge";

type DomainInfoCardProps = {
    orgId?: string;
    domainId?: string;
};

// Helper functions for Unicode domain handling
function toPunycode(domain: string): string {
    try {
        const parts = toASCII(domain);
        return parts;
    } catch (error) {
        return domain.toLowerCase();
    }
}


function isValidDomainFormat(domain: string): boolean {
    const unicodeRegex = /^(?!:\/\/)([^\s.]+\.)*[^\s.]+$/;

    if (!unicodeRegex.test(domain)) {
        return false;
    }

    const parts = domain.split('.');
    for (const part of parts) {
        if (part.length === 0 || part.startsWith('-') || part.endsWith('-')) {
            return false;
        }
        if (part.length > 63) {
            return false;
        }
    }

    if (domain.length > 253) {
        return false;
    }

    return true;
}

const formSchema = z.object({
    baseDomain: z
        .string()
        .min(1, "Domain is required")
        .refine((val) => isValidDomainFormat(val), "Invalid domain format")
        .transform((val) => toPunycode(val)),
    type: z.enum(["ns", "cname", "wildcard"]),
    certResolver: z.string().nullable().optional(),
    preferWildcardCert: z.boolean().optional()
});

type FormValues = z.infer<typeof formSchema>;

const certResolverOptions = [
    { id: "default", title: "Default" },
    { id: "custom", title: "Custom Resolver" }
];


export default function DomainInfoCard({ orgId, domainId }: DomainInfoCardProps) {
    const { domain, updateDomain } = useDomainContext();
    const t = useTranslations();
    const { env } = useEnvContext();
    const api = createApiClient(useEnvContext());
    const { toast } = useToast();

    const [dnsRecords, setDnsRecords] = useState<DNSRecordRow[]>([]);
    const [loadingRecords, setLoadingRecords] = useState(true);
    const [isRefreshing, setIsRefreshing] = useState(false);
    const [saveLoading, setSaveLoading] = useState(false);

    const form = useForm<FormValues>({
        resolver: zodResolver(formSchema),
        defaultValues: {
            baseDomain: "",
            type: build == "oss" || !env.flags.usePangolinDns ? "wildcard" : "ns",
            certResolver: null,
            preferWildcardCert: false
        }
    });

    useEffect(() => {
        if (domain.domainId) {
            const certResolverValue = domain.certResolver && domain.certResolver.trim() !== "" 
                ? domain.certResolver 
                : null;

            form.reset({
                baseDomain: domain.baseDomain || "",
                type: (domain.type as "ns" | "cname" | "wildcard") || "wildcard",
                certResolver: certResolverValue,
                preferWildcardCert: domain.preferWildcardCert || false
            });
        }
    }, [domain]);

    const fetchDNSRecords = async (showRefreshing = false) => {
        if (showRefreshing) {
            setIsRefreshing(true);
        } else {
            setLoadingRecords(true);
        }

        try {
            const response = await api.get<{ data: DNSRecordRow[] }>(
                `/org/${orgId}/domain/${domainId}/dns-records`
            );
            setDnsRecords(response.data.data);
        } catch (error) {
            // Only show error if records exist (not a 404)
            const err = error as any;
            if (err?.response?.status !== 404) {
                toast({
                    title: t("error"),
                    description: formatAxiosError(error),
                    variant: "destructive"
                });
            }
        } finally {
            setLoadingRecords(false);
            setIsRefreshing(false);
        }
    };

    useEffect(() => {
        if (domain.domainId) {
            fetchDNSRecords();
        }
    }, [domain.domainId]);

    const onSubmit = async (values: FormValues) => {
        if (!orgId || !domainId) {
            toast({
                title: t("error"),
                description: t("orgOrDomainIdMissing", { fallback: "Organization or Domain ID is missing" }),
                variant: "destructive"
            });
            return;
        }

        setSaveLoading(true);

        try {
            const response = await api.patch(
                `/org/${orgId}/domain/${domainId}`,
                {
                    certResolver: values.certResolver,
                    preferWildcardCert: values.preferWildcardCert
                }
            );

            updateDomain({
                ...domain,
                certResolver: values.certResolver || null,
                preferWildcardCert: values.preferWildcardCert || false
            });

            toast({
                title: t("success"),
                description: t("domainSettingsUpdated", { fallback: "Domain settings updated successfully" }),
                variant: "default"
            });
        } catch (error) {
            toast({
                title: t("error"),
                description: formatAxiosError(error),
                variant: "destructive"
            });
        } finally {
            setSaveLoading(false);
        }
    };

    const getTypeDisplay = (type: string) => {
        switch (type) {
            case "ns":
                return t("selectDomainTypeNsName");
            case "cname":
                return t("selectDomainTypeCnameName");
            case "wildcard":
                return t("selectDomainTypeWildcardName");
            default:
                return type;
        }
    };



    return (
        <>
            <Alert>
                <AlertDescription>
                    <InfoSections cols={3}>
                        <InfoSection>
                            <InfoSectionTitle>
                                {t("type")}
                            </InfoSectionTitle>
                            <InfoSectionContent>
                                <span>
                                    {getTypeDisplay(domain.type ? domain.type : "")}
                                </span>
                            </InfoSectionContent>
                        </InfoSection>
                        <InfoSection>
                            <InfoSectionTitle>
                                {t("status")}
                            </InfoSectionTitle>
                            <InfoSectionContent>
                                {domain.verified ? (
                                    <Badge variant="green">{t("verified")}</Badge>
                                ) : (
                                    <Badge variant="yellow">
                                        {t("pending", { fallback: "Pending" })}
                                    </Badge>
                                )}
                            </InfoSectionContent>
                        </InfoSection>
                    </InfoSections>
                </AlertDescription>
            </Alert>

            {domain.type !== "wildcard" && (
                loadingRecords ? (
                    <div className="space-y-4">
                        {t("loadingDNSRecords", { fallback: "Loading DNS Records..." })}
                    </div>
                ) : (
                    <DNSRecordsTable
                        domainId={domain.domainId}
                        records={dnsRecords}
                        isRefreshing={isRefreshing}
                    />
                )
            )}

            {/* Domain Settings - Only show for wildcard domains */}
            {domain.type === "wildcard" && (
                <SettingsContainer>
                    <SettingsSection>
                        <SettingsSectionHeader>
                            <SettingsSectionTitle>
                                {t("domainSetting")}
                            </SettingsSectionTitle>
                        </SettingsSectionHeader>

                        <SettingsSectionBody>
                            <SettingsSectionForm>
                                <Form {...form}>
                                    <form
                                        onSubmit={form.handleSubmit(onSubmit)}
                                        className="space-y-4"
                                        id="domain-settings-form"
                                    >
                                        <FormField
                                            control={form.control}
                                            name="certResolver"
                                            render={({ field }) => (
                                                <FormItem>
                                                    <FormLabel>{t("certResolver")}</FormLabel>
                                                    <FormControl>
                                                        <Select
                                                            value={
                                                                field.value === null ? "default" :
                                                                    (field.value === "" || (field.value && field.value !== "default")) ? "custom" :
                                                                        "default"
                                                            }
                                                            onValueChange={(val) => {
                                                                if (val === "default") {
                                                                    field.onChange(null);
                                                                } else if (val === "custom") {
                                                                    field.onChange("");
                                                                } else {
                                                                    field.onChange(val);
                                                                }
                                                            }}
                                                        >
                                                            <SelectTrigger>
                                                                <SelectValue placeholder={t("selectCertResolver")} />
                                                            </SelectTrigger>
                                                            <SelectContent>
                                                                {certResolverOptions.map((opt) => (
                                                                    <SelectItem key={opt.id} value={opt.id}>
                                                                        {opt.title}
                                                                    </SelectItem>
                                                                ))}
                                                            </SelectContent>
                                                        </Select>
                                                    </FormControl>
                                                    <FormMessage />
                                                    {field.value !== null && field.value !== "default" && (
                                                        <div className="space-y-2 mt-2">
                                                            <FormControl>
                                                                <Input
                                                                    placeholder={t("enterCustomResolver")}
                                                                    value={field.value || domain.certResolver || ""}
                                                                    onChange={(e) => field.onChange(e.target.value)}
                                                                />
                                                            </FormControl>
                                                            <FormField
                                                                control={form.control}
                                                                name="preferWildcardCert"
                                                                render={({ field: switchField }) => (
                                                                    <FormItem className="items-center space-y-2 mt-4">
                                                                        <FormControl>
                                                                            <div className="flex items-center space-x-2">
                                                                                <Switch
                                                                                    checked={switchField.value}
                                                                                    onCheckedChange={switchField.onChange}
                                                                                />
                                                                                <FormLabel>{t("preferWildcardCert")}</FormLabel>
                                                                            </div>
                                                                        </FormControl>

                                                                        <FormDescription>
                                                                            {t("preferWildcardCertDescription")}
                                                                        </FormDescription>
                                                                        <FormMessage />
                                                                    </FormItem>
                                                                )}
                                                            />
                                                        </div>
                                                    )}
                                                </FormItem>
                                            )}
                                        />
                                    </form>
                                </Form>
                            </SettingsSectionForm>
                        </SettingsSectionBody>

                        <SettingsSectionFooter>
                            <Button
                                type="submit"
                                loading={saveLoading}
                                disabled={saveLoading}
                                form="domain-settings-form"
                            >
                                {t("saveSettings")}
                            </Button>
                        </SettingsSectionFooter>
                    </SettingsSection>
                </SettingsContainer>
            )}
        </>
    );
}