import Docker from 'dockerode'
import config, { getQuestImage } from '../config.js'

const engine = new Docker()

const containerOptions = {
  AttachStdin: true,
  AttachStdout: true,
  AttachStderr: true,
  Tty: true,
  OpenStdin: true,
  StdinOnce: false,
  HostConfig: {
    NetworkMode: config.docker.network,
  },
}

// const network = engine.getNetwork(config.dockerNetwork)

export function createContainer(name, questId) {
  const option = {
    ...containerOptions,
    name,
    Image: getQuestImage(questId),
  }
  const { hostPwd, mountQuest } = config.docker
  if (!config.isProduction && hostPwd && mountQuest) {
    console.log('Mounting quest folder', mountQuest)
    option.HostConfig.Binds = [
      `${hostPwd}/quests/${mountQuest}/home:/home/commander`,
      `${hostPwd}/packages/container:/usr/local/lib/container`,
    ]
  }
  return engine.createContainer(option)
}

export async function getAndStartContainer(id) {
  console.log(`Getting container: ${id}`)
  const container = engine.getContainer(id)
  if (!container) {
    throw new Error(`Container ${id} not found`)
  }
  if (!(await container.inspect()).State.Running) {
    await container.start()
  }
  return container
}

export async function attachContainer(container, { token }) {
  const exec = await container.exec({
    AttachStdin: true,
    AttachStdout: true,
    AttachStderr: true,
    Cmd: ['/bin/zsh'],
    Tty: true,
    Env: [
      `TOKEN=${token}`,
      `API_ENDPOINT=http://backend:3000`,
      'ZDOTDIR=/etc/zsh',
    ],
  })

  const execOutput = await exec.start({
    Detach: false,
    Tty: true,
  })

  return execOutput
}

export async function deleteContainer(id) {
  const container = engine.getContainer(id)
  if (!container) {
    return
  }
  try {
    await container.stop()
  } catch (error) {
    console.warn('Failed to stop container', id)
  }
  await container.remove()
}

export function buildQuestImage(questPath, questId) {
  return new Promise((resolve, reject) => {
    engine.buildImage(
      {
        context: questPath,
        src: ['Dockerfile', 'home'],
      },
      {
        t: getQuestImage(questId),
        networkmode: 'none',
        memory: 10 * 1e6,
      },
      (err, response) => {
        if (err) {
          reject(err)
        }
        response.on('data', (data) => {
          // Process the data (this could be Docker build output)
          const { stream, error } = JSON.parse(data.toString())
          if (stream) {
            console.log(stream)
          }
          if (error) {
            console.error(error)
            reject(error)
          }
        })

        response.on('end', () => {
          console.log(`Build completed for ${questId}`)
          resolve(questId)
        })

        response.on('error', (error) => {
          console.error(`Build failed for ${questId}:`, error)
          reject(error)
        })
      }
    )
  })
}