Process - Phase 1 done: Container initialization & file migration

Sam Goto · Sam Goto · commit f2934ebf6036 · 2023-03-19T13:08:14.000-06:00
diff --git a/src/index.ts b/src/index.ts
@@ -2,10 +2,6 @@ import Process from "./process";
 
 const process = new Process();
 process
-	.prepTransfer("print('Hi')", "python3")
-	.then((containerID) => console.log(containerID))
+	.initContainer("python3", "print('Hi')")
+	.then(() => console.log("OK"))
 	.catch((e) => console.log(e));
-// process
-// 	.initContainer("python3")
-// 	.then((containerID) => console.log(containerID))
-// 	.catch((e) => console.log(e));
diff --git a/src/process.ts b/src/process.ts
@@ -4,6 +4,7 @@ import fs from "fs";
 import path from "path";
 import {
 	ContainerInitialization,
+	JobStatus,
 	container,
 	fileFormat,
 } from "./types/process";
@@ -19,8 +20,8 @@ class Process {
 	 * Creates an appropriate docker container
 	 * to execute the target code
 	 */
-	async initContainer(container: container): Promise<ContainerInitialization> {
-		return new Promise((reject, resolve) => {
+	createContainer(container: container): Promise<ContainerInitialization> {
+		return new Promise((resolve, reject) => {
 			const initCommand = `docker create ${container}`;
 
 			child.exec(initCommand, (error, containerID, stderr) => {
@@ -38,8 +39,8 @@ class Process {
 	/**
 	 * writes a code into a temp file
 	 */
-	async prepTransfer(context: string, container: container): Promise<string> {
-		return new Promise((reject, resolve) => {
+	async writeFile(container: container, context: string): Promise<string> {
+		return new Promise((resolve, reject) => {
 			const fileName = crypto.randomBytes(32).toString("hex");
 			const filePath = path.join(
 				__dirname,
@@ -62,18 +63,63 @@ class Process {
 	 * copies the target code into the newly created
 	 * isolated container
 	 */
-	async copyContext(containerID: string): Promise<void> {
-		return new Promise(async (reject, resolve) => {
-			// const initCommand = `docker cp ${} ${containerID}/src`;
-			// child.exec(initCommand, (error, containerID, stderr) => {
-			// 	if (error) {
-			// 		reject(error.message);
-			// 	} else if (stderr) {
-			// 		reject(stderr);
-			// 	} else {
-			// 		resolve(containerID);
-			// 	}
-			// });
+	copyContext(
+		containerID: string,
+		container: container,
+		context: string
+	): Promise<string> {
+		return new Promise(async (resolve, reject) => {
+			this.writeFile(container, context)
+				.then((filePath) => {
+					const initCommand = `docker cp ${filePath} ${containerID}:/src`;
+					console.log(filePath);
+					child.exec(initCommand, (error, containerID, stderr) => {
+						if (error) {
+							reject(error.message);
+						} else if (stderr) {
+							reject(stderr);
+						} else {
+							resolve(containerID.trim());
+						}
+					});
+				})
+				.catch((e) => {
+					reject(e);
+				});
+		});
+	}
+
+	/**
+	 *  Excecutes phase1
+	 *  1] Creates a new container
+	 *  2] Copes the code into the contaienr
+	 */
+	initContainer(container: container, context: string): Promise<JobStatus> {
+		return new Promise(async (resolve, reject) => {
+			this.createContainer(container)
+				.then(async (containerID) => {
+					return this.copyContext(containerID, container, context)
+						.then(() => {
+							resolve({
+								message: `Job has succedded.`,
+								jobFailed: false,
+								retryable: true,
+								context: context,
+								containerID: containerID,
+							});
+						})
+						.catch((e) => {
+							throw new Error(e);
+						});
+				})
+				.catch((e) => {
+					reject({
+						message: `Job has failed for the following reason(s): ${e}.`,
+						jobFailed: true,
+						retryable: true,
+						context: context,
+					});
+				});
 		});
 	}
 }
diff --git a/src/types/process.ts b/src/types/process.ts
@@ -1,3 +1,35 @@
 export type container = "python3" | "javascript";
 export type fileFormat = "py" | "js";
 export type ContainerInitialization = string;
+
+export interface JobStatus {
+	/**
+	 * A brief message that describes that job status
+	 */
+	message: string;
+
+	/**
+	 * Whether the job failed
+	 */
+	jobFailed: boolean;
+
+	/**
+	 * indicates if the job is retryable
+	 */
+	retryable: boolean;
+
+	/**
+	 * the raw data of the target code
+	 */
+	context: string;
+
+	/**
+	 * Newly created ontainer ID if exists
+	 */
+	containerID?: string;
+
+	/**
+	 * Newly created temp file that contain the context if exists
+	 */
+	filePath?: string;
+}
diff --git a/temp/60f1f0ec0277413a86bb31656c6ccc024f6495c31262fd43b7749d3522da08a4.py b/temp/60f1f0ec0277413a86bb31656c6ccc024f6495c31262fd43b7749d3522da08a4.py
