ai-toolkit/ui/src/hooks/useJob.tsx

'use client';

import { useEffect, useState } from 'react';
import { Job } from '@prisma/client';

export default function useJob(jobID: string, reloadInterval: null | number = null) {
  const [job, setJob] = useState<Job | null>(null);
  const [status, setStatus] = useState<'idle' | 'loading' | 'success' | 'error'>('idle');

  const refreshJobs = () => {
    setStatus('loading');
    fetch(`/api/jobs?id=${jobID}`)
      .then(res => res.json())
      .then(data => {
        console.log('Job:', data);
        setJob(data);
        setStatus('success');
      })
      .catch(error => {
        console.error('Error fetching datasets:', error);
        setStatus('error');
      });
  };

  useEffect(() => {
    refreshJobs();

    if (reloadInterval) {
      const interval = setInterval(() => {
        refreshJobs();
      }, reloadInterval);

      return () => {
        clearInterval(interval);
      }
    }
  }, [jobID]);

  return { job, setJob, status, refreshJobs };
}