'use client';
import React, { useCallback } from 'react';
import { useRouter } from 'next/navigation';
import { z } from 'zod';
import { zodResolver } from '@hookform/resolvers/zod';
import { Form } from 'react-compose-form';
import { FormReset } from '@/components/form/form-reset';
import { FormSubmit } from '@/components/form/form-submit';
import { InputControl } from '@/components/form/control/input.control';
import { api } from '@/lib/trpc/client';
import type { AppRouterOutput } from '@/lib/trpc/router';
import Link from "next/link";

type PaywallBypassIp = NonNullable<AppRouterOutput['paywallBypassIp']['get']>;

const PaywallBypassIpEditorSchema = z.object({
  ip: z.string().ip('Invalid IP address')
});

type Props =
  | {
      mode: 'create';
      id?: undefined;
    }
  | {
      mode: 'edit';
      id: string;
      item: PaywallBypassIp;
    };

export const PaywallBypassIpEditor: React.FC<Props> = (props) => {
  const router = useRouter();

  const createMutation = api.paywallBypassIp.create.useMutation({
    onSuccess(item) {
      router.replace(`/dashboard/paywall-bypass/ip/${item.id}`);
    }
  });

  const updateMutation = api.paywallBypassIp.update.useMutation();

  const handleSubmit = useCallback(
    async (data: z.infer<typeof PaywallBypassIpEditorSchema>) => {
      switch (props.mode) {
        case 'create': {
          await createMutation.mutateAsync({
            ipWhitelist: data
          });
          break;
        }

        case 'edit': {
          await updateMutation.mutateAsync({
            id: props.id,
            ipWhitelist: data
          });
          break;
        }
      }
    },
    [createMutation, updateMutation, props.mode, props.id]
  );

  return (
    <Form
      className='max-w-md space-y-4 px-4'
      resolver={zodResolver(PaywallBypassIpEditorSchema)}
      values={props.mode === 'edit' ? { ip: props.item.ip } : undefined}
      onSubmit={handleSubmit}
    >
      <InputControl
        label='IP Address'
        name='ip'
        placeholder='e.g. 192.168.1.1'
      />

      <div className='flex justify-between pt-5'>
        <div className='flex gap-2'>
          {props.mode === 'edit' && (
            <FormReset asChild>
              <Link href='.'>Cancel</Link>
            </FormReset>
          )}

          <FormSubmit>
            {props.mode === 'create' ? 'Create IP Bypass' : 'Update IP Bypass'}
          </FormSubmit>
        </div>
      </div>
    </Form>
  );
};