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

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

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

export function isMapRasterDataset(dataset: DatasetCreateResponse): boolean {
  return dataset.dataset_type === 'raster'
    && dataset.status === 'ready'
    && MAP_RASTER_SOURCES.has(dataset.source_name ?? '')
}

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
}