import { getNodeGroupList, getNodeList } from '@/services/admin/server';
import { getSubscribeGroupList } from '@/services/admin/subscribe';
import { zodResolver } from '@hookform/resolvers/zod';
import { Icon } from '@iconify/react';
import { useQuery } from '@tanstack/react-query';
import {
  Accordion,
  AccordionContent,
  AccordionItem,
  AccordionTrigger,
} from '@workspace/ui/components/accordion';
import { Button } from '@workspace/ui/components/button';
import { Checkbox } from '@workspace/ui/components/checkbox';
import {
  Form,
  FormControl,
  FormDescription,
  FormField,
  FormItem,
  FormLabel,
  FormMessage,
} from '@workspace/ui/components/form';
import { Label } from '@workspace/ui/components/label';
import { ScrollArea } from '@workspace/ui/components/scroll-area';
import {
  Sheet,
  SheetContent,
  SheetFooter,
  SheetHeader,
  SheetTitle,
  SheetTrigger,
} from '@workspace/ui/components/sheet';
import { Combobox } from '@workspace/ui/custom-components/combobox';
import { ArrayInput } from '@workspace/ui/custom-components/dynamic-Inputs';
import { JSONEditor } from '@workspace/ui/custom-components/editor';
import { EnhancedInput } from '@workspace/ui/custom-components/enhanced-input';
import { evaluateWithPrecision, unitConversion } from '@workspace/ui/utils';
import { useTranslations } from 'next-intl';
import { assign, shake } from 'radash';
import { useEffect, useState } from 'react';
import { useForm } from 'react-hook-form';
import { z } from 'zod';

interface SubscribeFormProps<T> {
  onSubmit: (data: T) => Promise<boolean> | boolean;
  initialValues?: T;
  loading?: boolean;
  trigger: string;
  title: string;
}

const defaultValues = {
  inventory: -1,
  speed_limit: 0,
  device_limit: 0,
  traffic: 0,
  quota: 0,
  discount: [],
  server_group: [],
  server: [],
  unit_time: 'Month',
};

export default function SubscribeForm<T extends Record<string, any>>({
  onSubmit,
  initialValues,
  loading,
  trigger,
  title,
}: SubscribeFormProps<T>) {
  const t = useTranslations('subscribe');

  const [open, setOpen] = useState(false);

  const formSchema = z.object({
    name: z.string(),
    description: z.string().optional(),
    unit_price: z.number(),
    unit_time: z.string().default('Month'),
    replacement: z.number().optional(),
    discount: z
      .array(
        z.object({
          quantity: z.number(),
          discount: z.number(),
        }),
      )
      .optional(),
    inventory: z.number().optional(),
    speed_limit: z.number().optional(),
    device_limit: z.number().optional(),
    traffic: z.number().optional(),
    quota: z.number().optional(),
    group_id: z.number().optional().nullish(),
    server_group: z.array(z.number()).optional(),
    server: z.array(z.number()).optional(),
  });

  const form = useForm({
    resolver: zodResolver(formSchema),
    defaultValues: assign(
      defaultValues,
      shake(initialValues, (value) => value === null) as Record<string, any>,
    ),
  });
  useEffect(() => {
    form?.reset(
      assign(defaultValues, shake(initialValues, (value) => value === null) as Record<string, any>),
    );
  }, [form, initialValues]);

  async function handleSubmit(data: { [x: string]: any }) {
    const bool = await onSubmit(data as T);

    if (bool) setOpen(false);
  }
  const { data: group } = useQuery({
    queryKey: ['getSubscribeGroupList'],
    queryFn: async () => {
      const { data } = await getSubscribeGroupList();
      return data.data?.list as API.SubscribeGroup[];
    },
  });

  const { data: server } = useQuery({
    queryKey: ['getNodeList', 'all'],
    queryFn: async () => {
      const { data } = await getNodeList({
        page: 1,
        size: 9999,
      });
      return data.data?.list;
    },
  });

  const { data: server_groups } = useQuery({
    queryKey: ['getNodeGroupList'],
    queryFn: async () => {
      const { data } = await getNodeGroupList();
      return (data.data?.list || []) as API.ServerGroup[];
    },
  });

  const unit_time = form.watch('unit_time');
  return (
    <Sheet open={open} onOpenChange={setOpen}>
      <SheetTrigger asChild>
        <Button
          onClick={() => {
            form.reset();
            setOpen(true);
          }}
        >
          {trigger}
        </Button>
      </SheetTrigger>
      <SheetContent className='w-screen max-w-full md:max-w-full'>
        <SheetHeader>
          <SheetTitle>{title}</SheetTitle>
        </SheetHeader>
        <ScrollArea className='-mx-6 h-[calc(100dvh-48px-36px-36px-env(safe-area-inset-top))]'>
          <Form {...form}>
            <form
              onSubmit={form.handleSubmit(handleSubmit)}
              className='grid gap-4 px-6 pt-4 lg:grid-cols-3'
            >
              <div className='flex flex-col gap-4'>
                <FormField
                  control={form.control}
                  name='name'
                  render={({ field }) => (
                    <FormItem>
                      <FormLabel>{t('form.name')}</FormLabel>
                      <FormControl>
                        <EnhancedInput
                          {...field}
                          onValueChange={(value) => {
                            form.setValue(field.name, value);
                          }}
                        />
                      </FormControl>
                      <FormMessage />
                    </FormItem>
                  )}
                />
                <FormField
                  control={form.control}
                  name='description'
                  render={({ field }) => (
                    <FormItem>
                      <FormControl>
                        <JSONEditor
                          title={t('form.description')}
                          value={field.value && JSON.parse(field.value)}
                          onChange={(value) => {
                            form.setValue(field.name, JSON.stringify(value));
                          }}
                          placeholder={{
                            description: 'description',
                            features: [
                              {
                                type: 'default',
                                icon: '',
                                label: 'label',
                              },
                            ],
                          }}
                          schema={{
                            type: 'object',
                            properties: {
                              description: {
                                type: 'string',
                                description: 'A brief description of the item.',
                              },
                              features: {
                                type: 'array',
                                items: {
                                  type: 'object',
                                  properties: {
                                    icon: {
                                      type: 'string',
                                      description:
                                        "Enter an Iconify icon identifier (e.g., 'mdi:account').",
                                      pattern: '^[a-z0-9]+:[a-z0-9-]+$',
                                      examples: [
                                        'uil:shield-check',
                                        'uil:shield-exclamation',
                                        'uil:database',
                                        'uil:server',
                                        'Visit https://icon-sets.iconify.design to browse available icons.',
                                      ],
                                    },
                                    label: {
                                      type: 'string',
                                      description: 'The label describing the feature.',
                                    },
                                    type: {
                                      type: 'string',
                                      enum: ['default', 'success', 'destructive'],
                                      description:
                                        'The type of feature, limited to specific values.',
                                    },
                                  },
                                },
                                description: 'A list of feature objects.',
                              },
                            },
                            required: ['description', 'features'],
                            additionalProperties: false,
                          }}
                        />
                      </FormControl>
                      <FormMessage />
                    </FormItem>
                  )}
                />
              </div>
              <div className='flex flex-col gap-4'>
                <div className='grid grid-cols-3 gap-4'>
                  <FormField
                    control={form.control}
                    name='unit_price'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.unitPrice')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            type='number'
                            {...field}
                            min={0}
                            formatInput={(value) => unitConversion('centsToDollars', value)}
                            formatOutput={(value) => unitConversion('dollarsToCents', value)}
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='unit_time'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.unitTime')}</FormLabel>
                        <FormControl>
                          <Combobox
                            placeholder={t('form.selectUnitTime')}
                            {...field}
                            onChange={(value) => {
                              if (value) {
                                form.setValue(field.name, value);
                              }
                            }}
                            options={[
                              { label: t('form.Year'), value: 'Year' },
                              { label: t('form.Month'), value: 'Month' },
                              { label: t('form.Day'), value: 'Day' },
                              { label: t('form.Hour'), value: 'Hour' },
                              { label: t('form.Minute'), value: 'Minute' },
                            ]}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='replacement'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.replacement')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            type='number'
                            {...field}
                            min={0}
                            formatInput={(value) => unitConversion('centsToDollars', value)}
                            formatOutput={(value) => unitConversion('dollarsToCents', value)}
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />

                  <FormField
                    control={form.control}
                    name='traffic'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.traffic')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            placeholder={t('form.noLimit')}
                            type='number'
                            {...field}
                            formatInput={(value) => unitConversion('bytesToGb', value)}
                            formatOutput={(value) => unitConversion('gbToBytes', value)}
                            suffix='GB'
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='speed_limit'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.speedLimit')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            placeholder={t('form.noLimit')}
                            type='number'
                            {...field}
                            formatInput={(value) => unitConversion('bitsToMb', value)}
                            formatOutput={(value) => unitConversion('mbToBits', value)}
                            suffix='MB'
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='device_limit'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.deviceLimit')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            placeholder={t('form.noLimit')}
                            type='number'
                            {...field}
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='inventory'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.inventory')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            placeholder={`-1 ${t('form.noLimit')}`}
                            type='number'
                            {...field}
                            min={-1}
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='quota'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.quota')}</FormLabel>
                        <FormControl>
                          <EnhancedInput
                            placeholder={t('form.noLimit')}
                            type='number'
                            {...field}
                            onValueChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <FormField
                    control={form.control}
                    name='group_id'
                    render={({ field }) => (
                      <FormItem>
                        <FormLabel>{t('form.subscribeGroup')}</FormLabel>
                        <FormControl>
                          <Combobox<number, false>
                            placeholder={t('form.selectSubscribeGroup')}
                            {...field}
                            onChange={(value) => {
                              form.setValue(field.name, value);
                            }}
                            options={group?.map((item) => ({
                              label: item.name,
                              value: item.id,
                            }))}
                          />
                        </FormControl>
                        <FormMessage />
                      </FormItem>
                    )}
                  />
                  <div className='col-span-3'>
                    <FormField
                      control={form.control}
                      name='discount'
                      render={({ field }) => (
                        <FormItem>
                          <FormLabel>{t('form.discount')}</FormLabel>
                          <FormControl>
                            <ArrayInput<API.SubscribeDiscount>
                              fields={[
                                {
                                  name: 'quantity',
                                  type: 'number',
                                  min: 1,
                                  suffix: unit_time && t(`form.${unit_time}`),
                                },
                                {
                                  name: 'discount',
                                  type: 'number',
                                  min: 1,
                                  max: 100,
                                  placeholder: t('form.discountPercent'),
                                  suffix: '%',
                                  calculateValue: function (data) {
                                    const { unit_price } = form.getValues();
                                    return {
                                      ...data,
                                      price: evaluateWithPrecision(
                                        `${unit_price} * ${data.quantity} * ${data.discount} / 100`,
                                      ),
                                    };
                                  },
                                },
                                {
                                  name: 'price',
                                  placeholder: t('form.discount_price'),
                                  type: 'number',
                                  min: 0,
                                  formatInput: (value) => unitConversion('centsToDollars', value),
                                  formatOutput: (value) => unitConversion('dollarsToCents', value),
                                  internal: true,
                                  calculateValue: function (data) {
                                    const { unit_price } = form.getValues();
                                    return {
                                      ...data,
                                      discount: evaluateWithPrecision(
                                        `${data.price} / ${data.quantity} / ${unit_price} * 100`,
                                      ),
                                    };
                                  },
                                },
                              ]}
                              value={field.value}
                              onChange={(value) => {
                                form.setValue(field.name, value);
                              }}
                            />
                          </FormControl>
                          <FormDescription>{t('form.discountDescription')}</FormDescription>
                          <FormMessage />
                        </FormItem>
                      )}
                    />
                  </div>
                </div>
              </div>
              <div className='flex flex-col gap-4'>
                <FormField
                  control={form.control}
                  name='server_group'
                  render={({ field }) => (
                    <FormItem>
                      <FormLabel>{t('form.serverGroup')}</FormLabel>
                      <FormControl>
                        <Accordion type='single' collapsible className='w-full'>
                          {server_groups?.map((group: API.ServerGroup) => {
                            const value = field.value || [];

                            return (
                              <AccordionItem key={group.id} value={String(group.id)}>
                                <AccordionTrigger>
                                  <div className='flex items-center gap-2'>
                                    <Checkbox
                                      checked={value.includes(group.id!)}
                                      onCheckedChange={(checked) => {
                                        return checked
                                          ? form.setValue(field.name, [...value, group.id])
                                          : form.setValue(
                                              field.name,
                                              value.filter((value: number) => value !== group.id),
                                            );
                                      }}
                                    />
                                    <Label>{group.name}</Label>
                                  </div>
                                </AccordionTrigger>
                                <AccordionContent>
                                  <ul className='list-disc [&>li]:mt-2'>
                                    {server
                                      ?.filter((server: API.Server) => server.group_id === group.id)
                                      ?.map((node: API.Server) => {
                                        return (
                                          <li
                                            key={node.id}
                                            className='flex items-center justify-between *:flex-1'
                                          >
                                            <span>{node.name}</span>
                                            <span>{node.server_addr}</span>
                                            <span className='text-right'>{node.protocol}</span>
                                          </li>
                                        );
                                      })}
                                  </ul>
                                </AccordionContent>
                              </AccordionItem>
                            );
                          })}
                        </Accordion>
                      </FormControl>
                      <FormMessage />
                    </FormItem>
                  )}
                />
                <FormField
                  control={form.control}
                  name='server'
                  render={({ field }) => (
                    <FormItem>
                      <FormLabel>{t('form.server')}</FormLabel>
                      <FormControl>
                        <div className='flex flex-col gap-2'>
                          {server
                            ?.filter((item: API.Server) => !item.group_id)
                            ?.map((item: API.Server) => {
                              const value = field.value || [];

                              return (
                                <div className='flex items-center gap-2' key={item.id}>
                                  <Checkbox
                                    checked={value.includes(item.id!)}
                                    onCheckedChange={(checked) => {
                                      return checked
                                        ? form.setValue(field.name, [...value, item.id])
                                        : form.setValue(
                                            field.name,
                                            value.filter((value: number) => value !== item.id),
                                          );
                                    }}
                                  />
                                  <Label className='flex w-full items-center justify-between *:flex-1'>
                                    <span>{item.name}</span>
                                    <span>{item.server_addr}</span>
                                    <span className='text-right'>{item.protocol}</span>
                                  </Label>
                                </div>
                              );
                            })}
                        </div>
                      </FormControl>
                      <FormMessage />
                    </FormItem>
                  )}
                />
              </div>
            </form>
          </Form>
        </ScrollArea>
        <SheetFooter className='flex-row justify-end gap-2 pt-3'>
          <Button
            variant='outline'
            disabled={loading}
            onClick={() => {
              setOpen(false);
            }}
          >
            {t('form.cancel')}
          </Button>
          <Button disabled={loading} onClick={form.handleSubmit(handleSubmit)}>
            {loading && <Icon icon='mdi:loading' className='mr-2 animate-spin' />}{' '}
            {t('form.confirm')}
          </Button>
        </SheetFooter>
      </SheetContent>
    </Sheet>
  );
}