import type { DatasetCreateResponse } from '../types'

const NON_IMAGERY_RASTER_SOURCES = new Set([
  'department_omgeving_thematic_raster',
  'digitaal_vlaanderen_dhmv',
  'vmm_flood_hazard',
  'spw_bathymetry',
  'spw_walous_land_cover',
])

export function isDetectionImageryDataset(dataset: DatasetCreateResponse): boolean {
  if (dataset.dataset_type !== 'raster' || dataset.status !== 'ready') {
    return false
  }
  if (NON_IMAGERY_RASTER_SOURCES.has(dataset.source_name ?? '')) {
    return false
  }
  return true
}