'use client';
import React, { useCallback } from 'react';
import Link from 'next/link';
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 { TiptapControl } from '@/components/form/control/tiptap.control';
import { Card, CardContent } from '@/components/ui/card';
import { api } from '@/lib/trpc/client';
import type { AppRouterOutput } from '@/lib/trpc/router';
import { Routes } from "@/config/routes";

type Block = NonNullable<AppRouterOutput['block']['get']>;

const BlockEditorSchema = z.object({
  key: z.string().min(1, 'Key is required').max(45),
  title: z.string().min(1, 'Title is required').max(255),
  content: z.any()
});

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

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

  const createMutation = api.block.create.useMutation({
    onSuccess(block) {
      router.replace(Routes.DASHBOARD.BLOCK_EDIT(block.id));
    }
  });

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

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

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

  return (
    <Form
      className='flex gap-6'
      resolver={zodResolver(BlockEditorSchema)}
      values={props.mode === 'edit' ? props.block : undefined}
      onSubmit={handleSubmit}
    >
      <div className='max-w-4xl flex-1'>
        <div className='flex flex-col gap-4 px-4'>
          <InputControl label='Title' name='title' />
          <TiptapControl label='Content' name='content' />
        </div>

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

            <FormSubmit>
              {props.mode === 'create' ? 'Create Block' : 'Update Block'}
            </FormSubmit>
          </div>
        </div>
      </div>

      <div className='w-80 flex-shrink-0'>
        <Card className='shadow-none'>
          <CardContent className='space-y-4'>
            <InputControl label='Key' name='key' />
          </CardContent>
        </Card>
      </div>
    </Form>
  );
};