runDocker.ts

/*!
 * Copyright © 2023 United States Government as represented by the
 * Administrator of the National Aeronautics and Space Administration.
 * All Rights Reserved.
 *
 * SPDX-License-Identifier: Apache-2.0
 */

import Dockerode from 'dockerode'
import { fork } from 'node:child_process'
import { type SearchEngineLauncherFunction } from './run.js'

const [, , command, jsonifiedArgs] = process.argv

if (command === 'launch-docker-subprocess') {
  const { dataDir, logsDir, engine, port, options } = JSON.parse(jsonifiedArgs)
  const Image =
    engine === 'elasticsearch'
      ? 'elastic/elasticsearch:8.6.2'
      : 'opensearchproject/opensearch:2.11.0'
  console.log('Launching Docker container', Image)
  const docker = new Dockerode()

  const container = await docker.createContainer({
    Env: [...options, 'path.data=/var/lib/search', 'path.logs=/var/log/search'],
    HostConfig: {
      AutoRemove: true,
      Mounts: [
        { Source: dataDir, Target: '/var/lib/search', Type: 'bind' },
        { Source: logsDir, Target: '/var/log/search', Type: 'bind' },
      ],
      PortBindings: {
        [`${port}/tcp`]: [{ HostIP: '127.0.0.1', HostPort: `${port}` }],
      },
    },
    Image,
  })
  const stream = await container.attach({ stream: true, stderr: true })
  stream.pipe(process.stderr)
  await container.start()

  const signals = ['message', 'SIGTERM', 'SIGINT']
  signals.forEach((signal) => {
    process.on(signal, async () => {
      await container.kill()
    })
  })

  await container.wait()
}

export const launchDocker: SearchEngineLauncherFunction = async ({
  dataDir,
  logsDir,
  engine,
  port,
  options,
}) => {
  const argv = {
    dataDir,
    logsDir,
    engine,
    port,
    options,
  }
  const subprocess = fork(new URL(import.meta.url), [
    'launch-docker-subprocess',
    JSON.stringify(argv),
  ])

  return {
    async kill() {
      console.log('Killing Docker container')
      subprocess.send({ action: 'kill' })
    },
    async waitUntilStopped() {
      return new Promise((resolve) => {
        subprocess.on('exit', () => {
          console.log('Docker container exited')
          resolve()
        })
      })
    },
  }
}