stats/apps/start/src/modals/edit-client.tsx

import { ButtonContainer } from '@/components/button-container';
import { InputWithLabel } from '@/components/forms/input-with-label';
import { Button } from '@/components/ui/button';
import { useTRPC } from '@/integrations/trpc/react';
import { handleError } from '@/integrations/trpc/react';
import { zodResolver } from '@hookform/resolvers/zod';
import { useMutation, useQueryClient } from '@tanstack/react-query';
import { useRouter } from '@tanstack/react-router';
import { useForm } from 'react-hook-form';
import { toast } from 'sonner';
import { z } from 'zod';

import type { IServiceClient } from '@openpanel/db';

import { popModal } from '.';
import { ModalContent, ModalHeader } from './Modal/Container';

type EditClientProps = IServiceClient;

const validator = z.object({
  id: z.string().min(1),
  name: z.string().min(1),
});

type IForm = z.infer<typeof validator>;

export default function EditClient({ id, name }: EditClientProps) {
  const queryClient = useQueryClient();
  const { register, handleSubmit, reset, formState, control, setError } =
    useForm<IForm>({
      resolver: zodResolver(validator),
      defaultValues: {
        id,
        name,
      },
    });

  const trpc = useTRPC();
  const mutation = useMutation(
    trpc.client.update.mutationOptions({
      onError: handleError,
      onSuccess() {
        reset();
        toast('Success', {
          description: 'Client updated.',
        });
        popModal();
        queryClient.invalidateQueries(trpc.client.list.pathFilter());
      },
    }),
  );

  return (
    <ModalContent>
      <ModalHeader title="Edit client" />
      <form onSubmit={handleSubmit((values) => mutation.mutate(values))}>
        <InputWithLabel label="Name" placeholder="Name" {...register('name')} />
        <ButtonContainer>
          <Button type="button" variant="outline" onClick={() => popModal()}>
            Cancel
          </Button>
          <Button type="submit" disabled={!formState.isDirty}>
            Save
          </Button>
        </ButtonContainer>
      </form>
    </ModalContent>
  );
}