import { Container } from "dockerode"
import { Stream } from "stream"

export function promisifyStream(stream: Stream): Promise<string> {
  const data: Buffer[] = []
  return new Promise((resolve, reject) => {
    stream.on("data", (chunk) => data.push(chunk))
    stream.on("end", () => {
      const toReturn = Buffer.concat(data).toString()
      resolve(toReturn)
    })
    stream.on("error", () => reject())
  })
}

export async function pullImage(docker: any, imageName: string) {
  const stream = await docker.pull(imageName)
  await promisifyStream(stream)
}

export async function waitForMysql(_container: Container, _dbPass: String) {
  await wait(20000);
  //let connectionTries = 0
  //while (connectionTries < 40) {
  //  const exec = await container.exec({
  //    Cmd: [`systemctl status mysql`],
  //    AttachStdin: true,
  //    AttachStdout: true,
  //  })
  //  const { output } = await exec.start({ hijack: true, stdin: true })
  //  const data = await promisifyStream(output)
  //  console.log(data);
  //  const status = data.split(" - ")[1].trim()
  //  if (status === "accepting connections") {
  //    await wait(1000)
  //    break
  //  }
  //  connectionTries += 1
  //  await wait(500)
  //}
  //if (connectionTries === 30) {
  //  throw Error("Too many tries to connect to database")
  //}
}

export function wait(ms: number) {
  return new Promise((resolve) => {
    setTimeout(resolve, ms)
  })
}

export async function startMysqlContainer(
  docker: any,
  imageName: string,
  containerName: string,
  _dbName: string,
  _dbUser: string,
  dbPass: string,
) {
  const container = await docker.createContainer({
    Image: imageName,
    name: containerName,
    Tty: false,
    PortBindings: { "3306/tcp": [{ HostPort: "3306" }] },
    Env: [
      `MYSQL_ROOT_PASSWORD=${dbPass}`,
    ],
  })
  await container.start()
  await waitForMysql(container, dbPass)
}

export async function removeMysqlContainer(docker: any, containerName: string) {
  const containers = await docker.listContainers({ all: true })
  for (const containerInfo of containers) {
    if (containerInfo.Names[0] === `/${containerName}`) {
      const container = docker.getContainer(containerInfo.Id)
      if (containerInfo.State !== "exited") {
        await container.stop()
      }
      await container.remove()
    }
  }
}