lasthourcloud/src/lib/docker.ts

import Dockerode from 'dockerode';
import { promises as fs } from 'fs';
import { saveBuildLog } from './common';

export async function buildCacheImageWithNode(data, imageForBuild) {
	const {
		applicationId,
		tag,
		workdir,
		docker,
		buildId,
		baseDirectory,
		installCommand,
		buildCommand,
		debug,
		secrets,
		pullmergeRequestId
	} = data;
	const Dockerfile: Array<string> = [];
	Dockerfile.push(`FROM ${imageForBuild}`);
	Dockerfile.push('WORKDIR /usr/src/app');
	if (secrets.length > 0) {
		secrets.forEach((secret) => {
			if (secret.isBuildSecret) {
				if (pullmergeRequestId) {
					if (secret.isPRMRSecret) {
						Dockerfile.push(`ARG ${secret.name} ${secret.value}`);
					}
				} else {
					if (!secret.isPRMRSecret) {
						Dockerfile.push(`ARG ${secret.name} ${secret.value}`);
					}
				}
			}
		});
	}
	// TODO: If build command defined, install command should be the default yarn install
	if (installCommand) {
		Dockerfile.push(`COPY ./${baseDirectory || ''}package*.json ./`);
		try {
			await fs.stat(`${workdir}/yarn.lock`);
			Dockerfile.push(`COPY ./${baseDirectory || ''}yarn.lock ./`);
		} catch (error) {}
		try {
			await fs.stat(`${workdir}/pnpm-lock.yaml`);
			Dockerfile.push(`COPY ./${baseDirectory || ''}pnpm-lock.yaml ./`);
		} catch (error) {}
		Dockerfile.push(`RUN ${installCommand}`);
	}
	Dockerfile.push(`COPY ./${baseDirectory || ''} ./`);
	Dockerfile.push(`RUN ${buildCommand}`);
	await fs.writeFile(`${workdir}/Dockerfile-cache`, Dockerfile.join('\n'));
	await buildImage({ applicationId, tag, workdir, docker, buildId, isCache: true, debug });
}

export async function buildCacheImageWithCargo(data, imageForBuild) {
	const {
		applicationId,
		tag,
		workdir,
		docker,
		buildId,
		baseDirectory,
		installCommand,
		buildCommand,
		debug,
		secrets
	} = data;
	const Dockerfile: Array<string> = [];
	Dockerfile.push(`FROM ${imageForBuild} as planner-${applicationId}`);
	Dockerfile.push('WORKDIR /usr/src/app');
	Dockerfile.push('RUN cargo install cargo-chef');
	Dockerfile.push('COPY . .');
	Dockerfile.push('RUN cargo chef prepare --recipe-path recipe.json');
	Dockerfile.push(`FROM ${imageForBuild}`);
	Dockerfile.push('WORKDIR /usr/src/app');
	Dockerfile.push('RUN cargo install cargo-chef');
	Dockerfile.push(`COPY --from=planner-${applicationId} /usr/src/app/recipe.json recipe.json`);
	Dockerfile.push('RUN cargo chef cook --release --recipe-path recipe.json');
	await fs.writeFile(`${workdir}/Dockerfile-cache`, Dockerfile.join('\n'));
	await buildImage({ applicationId, tag, workdir, docker, buildId, isCache: true, debug });
}

export async function buildImage({
	applicationId,
	tag,
	workdir,
	docker,
	buildId,
	isCache = false,
	debug = false
}) {
	if (isCache) {
		saveBuildLog({ line: `Building cache image started.`, buildId, applicationId });
	} else {
		saveBuildLog({ line: `Building image started.`, buildId, applicationId });
	}
	if (!debug && isCache) {
		saveBuildLog({
			line: `Debug turned off. To see more details, allow it in the configuration.`,
			buildId,
			applicationId
		});
	}

	const stream = await docker.engine.buildImage(
		{ src: ['.'], context: workdir },
		{
			dockerfile: isCache ? 'Dockerfile-cache' : 'Dockerfile',
			t: `${applicationId}:${tag}${isCache ? '-cache' : ''}`
		}
	);
	await streamEvents({ stream, docker, buildId, applicationId, debug });
}

export function dockerInstance({ destinationDocker }): { engine: Dockerode; network: string } {
	// new Docker({protocol:"ssh",  host: '188.34.164.25',port: 22, username:'root', sshOptions: {agentForward: true, agent:process.env.SSH_AUTH_SOCK}})
	return {
		engine: new Dockerode({
			socketPath: destinationDocker.engine
		}),
		network: destinationDocker.network
	};
}
export async function streamEvents({ stream, docker, buildId, applicationId, debug }) {
	await new Promise((resolve, reject) => {
		docker.engine.modem.followProgress(stream, onFinished, onProgress);
		function onFinished(err, res) {
			if (err) reject(err);
			resolve(res);
		}
		function onProgress(event) {
			if (event.error) {
				reject(event.error);
			} else if (event.stream) {
				if (event.stream !== '\n') {
					if (debug)
						saveBuildLog({ line: `${event.stream.replace('\n', '')}`, buildId, applicationId });
				}
			}
		}
	});
}

export const baseServiceConfigurationDocker = {
	restart_policy: {
		condition: 'any',
		max_attempts: 6
	}
};

export const baseServiceConfigurationSwarm = {
	replicas: 1,
	restart_policy: {
		condition: 'any',
		max_attempts: 6
	},
	update_config: {
		parallelism: 1,
		delay: '10s',
		order: 'start-first'
	},
	rollback_config: {
		parallelism: 1,
		delay: '10s',
		order: 'start-first',
		failure_action: 'rollback'
	}
};