leonitousconforti/tinyburg

import * as Option from "effect/Option";
import * as MobyApi from "the-moby-effect";

import { DOCKER_IMAGE_TAG, SHARED_EMULATOR_DATA_VOLUME_NAME } from "../versions.js";

/** The port bindings that all architect emulator containers must have. */
export interface IArchitectPortBindings {
    /** Adb console port */
    "5554/tcp": readonly [{ HostPort: string }];

    /** Adb port */
    "5555/tcp": readonly [{ HostPort: string }];

    /** Mitmproxy web interface port */
    "8080/tcp": readonly [{ HostPort: string }];

    /** Envoy proxy admin web interface port */
    "8081/tcp": readonly [{ HostPort: string }];

    /** Emulator grpc port */
    "8554/tcp": readonly [{ HostPort: string }];

    /** Emulator grpc web port */
    "8555/tcp": readonly [{ HostPort: string }];

    /** Frida server port */
    "27042/tcp": readonly [{ HostPort: string }];
}

/**
 * Specifies the docker options for creating an architect container. You must
 * provide a name for the container and you can optionally provide the
 * entrypoint command, network mode, port bindings, and environment variables.
 *
 * @internal
 */
export const containerCreateOptions = ({
    containerName,
    command,
    networkMode,
    portBindings,
    environmentVariables,
}: {
    containerName: string;
    environmentVariables: string[];
    command: Option.Option<string[]>;
    networkMode: string | undefined;
    portBindings: Partial<IArchitectPortBindings>;
}): MobyApi.Containers.ContainerCreateOptions => ({
    name: containerName,
    spec: {
        Image: DOCKER_IMAGE_TAG,
        Cmd: Option.getOrNull(command),
        Volumes: { "/android/avd-home/Pixel2.avd/": {} },
        Env: environmentVariables.some((environmentVariable) => environmentVariable.startsWith("DISPLAY="))
            ? environmentVariables
            : ["DISPLAY=:0", ...environmentVariables],
        HostConfig: {
            NetworkMode: networkMode || undefined,
            DeviceRequests: [{ Count: -1, Driver: "nvidia", Capabilities: [["gpu"]] }],
            Devices: [{ CgroupPermissions: "mrw", PathInContainer: "/dev/kvm", PathOnHost: "/dev/kvm" }],
            PortBindings: portBindings,
            Binds: [
                "/tmp/.X11-unix:/tmp/.X11-unix",
                "/etc/timezone:/etc/timezone:ro",
                "/etc/localtime:/etc/localtime:ro",
                "/usr/share/vulkan/icd.d/nvidia_icd.json:/usr/share/vulkan/icd.d/nvidia_icd.json",
                "/usr/share/glvnd/egl_vendor.d/10_nvidia.json:/usr/share/glvnd/egl_vendor.d/10_nvidia.json",
                "/usr/share/vulkan/implicit_layer.d/nvidia_layers.json:/usr/share/vulkan/implicit_layer.d/nvidia_layers.json",
                `${SHARED_EMULATOR_DATA_VOLUME_NAME}:/android/avd-home/Pixel2.avd/`,
            ],
        },
    },
});

export default containerCreateOptions;