import Service from '#models/service'
import { DockerService } from '#services/docker_service'
import { BaseSeeder } from '@adonisjs/lucid/seeders'
import { ModelAttributes } from '@adonisjs/lucid/types/model'

export default class ServiceSeeder extends BaseSeeder {
  private static NOMAD_STORAGE_DIR = '/opt/project-nomad/storage'
  private static DEFAULT_SERVICES: Omit<ModelAttributes<Service>, 'created_at' | 'updated_at' | 'metadata' | 'id'>[] = [
    {
      service_name: DockerService.KIWIX_SERVICE_NAME,
      container_image: 'ghcr.io/kiwix/kiwix-serve',
      container_command: '*.zim --address=0.0.0.0',
      container_config: JSON.stringify({
        HostConfig: {
          RestartPolicy: { Name: 'unless-stopped' },
          Binds: [`${ServiceSeeder.NOMAD_STORAGE_DIR}/zim:/data`],
          PortBindings: { '8080/tcp': [{ HostPort: '8090' }] }
        },
        ExposedPorts: { '8080/tcp': {} }
      }),
      ui_location: '8090',
      installed: false,
      is_dependency_service: false,
      depends_on: null,
    },
    {
      service_name: DockerService.OPENSTREETMAP_SERVICE_NAME,
      container_image: 'overv/openstreetmap-tile-server',
      container_command: 'run --shm-size="192m"',
      container_config: JSON.stringify({
        HostConfig: {
          RestartPolicy: { Name: 'unless-stopped' },
          Binds: [
            `${ServiceSeeder.NOMAD_STORAGE_DIR}/osm/db:/data/database`,
            `${ServiceSeeder.NOMAD_STORAGE_DIR}/osm/tiles:/data/tiles`
          ],
          PortBindings: { '80/tcp': [{ HostPort: '9000' }] }
        }
      }),
      ui_location: '9000',
      installed: false,
      is_dependency_service: false,
      depends_on: null,
    },
    {
      service_name: DockerService.OLLAMA_SERVICE_NAME,
      container_image: 'ollama/ollama:latest',
      container_command: 'serve',
      container_config: JSON.stringify({
        HostConfig: {
          RestartPolicy: { Name: 'unless-stopped' },
          Binds: [`${ServiceSeeder.NOMAD_STORAGE_DIR}/ollama:/root/.ollama`],
          PortBindings: { '11434/tcp': [{ HostPort: '11434' }] }
        },
        ExposedPorts: { '11434/tcp': {} }
      }),
      ui_location: null,
      installed: false,
      is_dependency_service: true,
      depends_on: null,
    },
    {
      service_name: DockerService.OPEN_WEBUI_SERVICE_NAME,
      container_image: 'ghcr.io/open-webui/open-webui:main',
      container_command: null,
      container_config: JSON.stringify({
        HostConfig: {
          RestartPolicy: { Name: 'unless-stopped' },
          NetworkMode: 'host',
          Binds: [`${ServiceSeeder.NOMAD_STORAGE_DIR}/open-webui:/app/backend/data`]
        }
      }),
      ui_location: '3000',
      installed: false,
      is_dependency_service: false,
      depends_on: DockerService.OLLAMA_SERVICE_NAME,
    },
  ]

  async run() {
    const existingServices = await Service.query().select('service_name')
    const existingServiceNames = new Set(existingServices.map(service => service.service_name))

    const newServices = ServiceSeeder.DEFAULT_SERVICES.filter(service => !existingServiceNames.has(service.service_name))

    await Service.createMany([...newServices])
  }
}