console/integration-tests/testkit/utils.ts

import Docker from 'dockerode';
import { humanId } from 'human-id';

let docker: Docker | null = null;

function getDockerConnection() {
  if (!docker) {
    docker = new Docker();
  }

  return docker;
}

const LOCAL_SERVICES = {
  server: 3001,
  clickhouse: 8123,
  emails: 6260,
  composition_federation_2: 3069,
  usage: 4001,
  schema: 6500,
  external_composition: 3012,
} as const;

export type KnownServices = keyof typeof LOCAL_SERVICES;

export function getCDNPort() {
  if (process.env.RUN_AGAINST_LOCAL_SERVICES === '1') {
    return 9000;
  }

  return 8083;
}

export function getAppBaseUrl() {
  if (process.env.RUN_AGAINST_LOCAL_SERVICES === '1') {
    return 'localhost:3000';
  }

  return 'localhost:8080';
}

export async function getServiceHost(
  serviceName: KnownServices,
  servicePort: number,
  localhost = true,
): Promise<string> {
  if (process.env.RUN_AGAINST_LOCAL_SERVICES === '1') {
    return `localhost:${LOCAL_SERVICES[serviceName]}`;
  }

  const actualHost = localhost ? 'localhost' : serviceName;
  const docker = getDockerConnection();
  const containers = await docker.listContainers({
    filters: JSON.stringify({
      label: [`com.docker.compose.service=${serviceName}`],
    }),
  });

  if (containers.length === 0) {
    throw new Error(`Failed to locate Docker container for service "${serviceName}"!`);
  }

  const container = containers[0];
  const ports = container.Ports || [];

  if (ports.length === 0) {
    throw new Error(
      `Container "${container.Id}" for service "${serviceName}" does not expose any ports!`,
    );
  }

  const publicPort = ports.find(p => p.PublicPort === servicePort);
  const privatePort = ports.find(p => p.PrivatePort === servicePort);

  if (!publicPort && !privatePort) {
    throw new Error(
      `Container "${container.Id}" for service "${serviceName}" does not expose port "${servicePort}"!`,
    );
  }

  if (publicPort) {
    return `${actualHost}:${publicPort.PublicPort}`;
  }

  if (privatePort) {
    console.warn(
      `Container "${container.Id}" (service: "${serviceName}") expose port "${servicePort}" as "${privatePort.PublicPort}", please consider to update your setup!`,
    );

    return `${actualHost}:${privatePort.PublicPort}`;
  }

  return `${actualHost}:${servicePort}`;
}

export function generateUnique() {
  return humanId({
    separator: '',
    adjectiveCount: 1,
    addAdverb: true,
    capitalize: false,
  });
}