import * as cp from "child_process";
import { Client, useBuffer } from "@coder/protocol";
import { client } from "./client";
import { promisify } from "./util";

class CP {
	public constructor(
		private readonly client: Client,
	) { }

	public exec = (
		command: string,
		options?: { encoding?: BufferEncoding | string | "buffer" | null } & cp.ExecOptions | null | ((error: Error | null, stdout: string, stderr: string) => void) | ((error: Error | null, stdout: Buffer, stderr: Buffer) => void),
		callback?: ((error: Error | null, stdout: string, stderr: string) => void) | ((error: Error | null, stdout: Buffer, stderr: Buffer) => void),
	): cp.ChildProcess => {
		// TODO: Probably should add an `exec` instead of using `spawn`, especially
		// since bash might not be available.
		const childProcess = this.client.spawn("bash", ["-c", command.replace(/"/g, "\\\"")]);

		let stdout = "";
		childProcess.stdout.on("data", (data) => {
			stdout += data.toString();
		});

		let stderr = "";
		childProcess.stderr.on("data", (data) => {
			stderr += data.toString();
		});

		childProcess.on("exit", (exitCode) => {
			const error = exitCode !== 0 ? new Error(stderr) : null;
			if (typeof options === "function") {
				callback = options;
			}
			if (callback) {
				// @ts-ignore not sure how to make this work.
				callback(
					error,
					useBuffer(options) ? Buffer.from(stdout) : stdout,
					useBuffer(options) ? Buffer.from(stderr) : stderr,
				);
			}
		});

		// @ts-ignore TODO: not fully implemented
		return childProcess;
	}

	public fork = (modulePath: string, args?: string[] | cp.ForkOptions, options?: cp.ForkOptions): cp.ChildProcess => {
		if (options && options.env && options.env.AMD_ENTRYPOINT) {
			// @ts-ignore TODO: not fully implemented
			return this.client.bootstrapFork(
				options.env.AMD_ENTRYPOINT,
				Array.isArray(args) ? args : [],
				// @ts-ignore TODO: env is a different type
				Array.isArray(args) || !args ? options : args,
			);
		}

		// @ts-ignore TODO: not fully implemented
		return this.client.fork(
			modulePath,
			Array.isArray(args) ? args : [],
			// @ts-ignore TODO: env is a different type
			Array.isArray(args) || !args ? options : args,
		);
	}

	public spawn = (command: string, args?: string[] | cp.SpawnOptions, options?: cp.SpawnOptions): cp.ChildProcess => {
		// @ts-ignore TODO: not fully implemented
		return this.client.spawn(
			command,
			Array.isArray(args) ? args : [],
			// @ts-ignore TODO: env is a different type
			Array.isArray(args) || !args ? options : args,
		);
	}
}

const fillCp = new CP(client);

// tslint:disable-next-line no-any makes util.promisify return an object
(fillCp as any).exec[promisify.customPromisifyArgs] = ["stdout", "stderr"];

export = fillCp;