hooks useModelType.ts

0% Statements 0/39

0% Branches 0/1

0% Functions 0/1

0% Lines 0/39

Press n or j to go to the next uncovered block, b, p or k for the previous block.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  'use client'

import { useParams } from 'next/navigation'
import { notFound } from 'next/navigation'
import type { ModelType } from '../train/components/wizard/types'

const VALID_MODELS: ModelType[] = ['boundary-detector', 'column-classifier']

/**
 * Hook to get the current model type from the URL path.
 * Must be used within a route that has a [model] segment.
 *
 * @returns The validated ModelType from the URL
 * @throws notFound() if the model is invalid
 */
export function useModelType(): ModelType {
  const params = useParams()
  const model = params.model as string

  if (!model || !VALID_MODELS.includes(model as ModelType)) {
    notFound()
  }

  return model as ModelType
}

/**
 * Type guard to check if a string is a valid ModelType
 */
export function isValidModelType(value: string): value is ModelType {
  return VALID_MODELS.includes(value as ModelType)
}

/**
 * Get all valid model types
 */
export function getValidModelTypes(): ModelType[] {
  return [...VALID_MODELS]
}