zcatalyst-cli/lib/shell/dependencies/invoker/job/node.mjs

Command Line Tool for CATALYST

'use strict';

import * as path from 'path';
import * as fs from 'fs';
import { pathToFileURL } from 'url';

const args = process.argv.slice(2);
const target = JSON.parse(args[0]);
const userData = JSON.parse(args[1]);
const projectJson = JSON.parse(args[2]);
const authJson = JSON.parse(args[3]);
const buildDir = JSON.parse(args[4]);

const requestFile = path.join(buildDir, '.catalyst', 'user_req_body');
const responseFile = path.join(buildDir, '.catalyst', 'user_res_body');
const metaFile = path.join(buildDir, '.catalyst', 'user_meta.json');

const writeToFile = (resp, status) => {
	fs.writeFileSync(responseFile, resp);
	fs.writeFileSync(metaFile, JSON.stringify({ response: { statusCode: status } }));
};

let body = {};

if (fs.existsSync(requestFile)) {
	const content = fs.readFileSync(requestFile);
	try {
		body = JSON.parse(content.toString());
	} catch (err) {
		body = {};
	}
}

/**
 * execution timeout 15 minutes
 */
const timeout = 15 * 60 * 1000;
const endTime = timeout + Date.now();

// exit on timeout
process.env.DEBUG === 'false' &&
	setTimeout(() => {
		writeToFile('TIMEOUT', 408);
		process.exit(0);
	}, timeout);

const context = {
	catalystHeaders: Object.assign(projectJson, authJson),
	getMaxExecutionTimeMs: () => timeout,
	getRemainingExecutionTimeMs: () => endTime - Date.now(),
	closeWithSuccess: () => {
		writeToFile('SUCCESS', 200);
		process.exit(0);
	},
	closeWithFailure: () => {
		writeToFile('FAILURE', 530);
		process.exit(0);
	}
};

const jobReq = {
	getJobDetails: () => ('job_details' in userData ? userData.job_details : {}),
	getJobMetaDetails: () => jobReq.getJobDetails()?.job_meta_details || {},
	getJobPoolDetails: () => jobReq.getJobMetaDetails()?.jobpool_details || {},
	getProjectDetails: () => jobReq.getJobPoolDetails()?.project_details || {},
	getJobCapacityAttributes: () => jobReq.getJobDetails()?.capacity || {},
	getAllJobParams: () => jobReq.getJobMetaDetails()?.params || {},
	getJobParam: (key) => (jobReq.getAllJobParams() || {})[key]
};

import(pathToFileURL(target.index))
	.then((module) => {
		try {
			if (!('default' in module)) {
				throw new Error('Could not find any default export');
			}
			if (typeof module.default !== 'function') {
				throw new Error('The default export is not a function');
			}
			module.default(jobReq, context);
		} catch (e) {
			// eslint-disable-next-line no-console
			console.error(e);
			writeToFile('CODE_EXCEPTION', 532);
			process.exit(0);
		}
	})
	.catch((e) => {
		// eslint-disable-next-line no-console
		console.error(e);
		writeToFile('INTERNAL_SERVER_ERROR', 500);
		process.exit(0);
	});