add restore

apps/admin.saladeaula.digital/app/routes/_.$orgid.enrollments.add/route.tsx

@@ -19,7 +19,7 @@ import {
 } from 'lucide-react'
 import { redirect, Link, useParams, useFetcher } from 'react-router'
 import { Controller, useFieldArray, useForm } from 'react-hook-form'
-import { Fragment, use, useMemo, useState } from 'react'
+import { Fragment, use, useEffect, useMemo, useState } from 'react'
 import { format } from 'date-fns'
 import { ptBR } from 'react-day-picker/locale'
 import { zodResolver } from '@hookform/resolvers/zod'
@@ -75,12 +75,15 @@ import { cloudflareContext } from '@repo/auth/context'
 import { SearchFilter } from '@repo/ui/components/search-filter'
 
 import { formSchema, type Schema, MAX_ITEMS } from './data'
+import { pick } from 'ramda'
 
 export function meta({}: Route.MetaArgs) {
   return [{ title: 'Adicionar matrícula' }]
 }
 
-export async function loader({ context }: Route.LoaderArgs) {
+export async function loader({ params, context, request }: Route.LoaderArgs) {
+  const url = new URL(request.url)
+  const submissionId = url.searchParams.get('submission')
   const cloudflare = context.get(cloudflareContext)
   const courses = createSearch({
     index: 'saladeaula_courses',
@@ -90,7 +93,15 @@ export async function loader({ context }: Route.LoaderArgs) {
     env: cloudflare.env
   })
 
-  return { courses }
+  const submission = submissionId
+    ? req({
+        url: `/orgs/${params.orgid}/enrollments/${submissionId}/submitted`,
+        context,
+        request
+      }).then((r) => r.json())
+    : Promise.resolve({ enrolled: [] })
+
+  return { courses, submission }
 }
 
 export async function action({ params, request, context }: Route.ActionArgs) {
@@ -116,9 +127,10 @@ const emptyRow = {
 }
 
 export default function Route({
-  loaderData: { courses }
+  loaderData: { courses, submission }
 }: Route.ComponentProps) {
   const { orgid } = useParams()
+  const { enrolled } = use(submission)
   const fetcher = useFetcher()
   const form = useForm({
     resolver: zodResolver(formSchema),
@@ -156,6 +168,18 @@ export default function Route({
     })
   }
 
+  useEffect(() => {
+    if (enrolled.length === 0) {
+      return
+    }
+
+    reset({
+      enrollments: enrolled.map(({ input_record }) =>
+        pick(['course', 'user'], input_record)
+      )
+    })
+  }, [enrolled, reset])
+
   return (
     <div className="space-y-2.5">
       <Breadcrumb>