code-server-2/src/cli.ts

import * as os from "os";
import * as path from "path";

import { validatePaths } from "vs/code/node/paths";
import { parseMainProcessArgv } from "vs/platform/environment/node/argvHelper";
import { ParsedArgs } from "vs/platform/environment/common/environment";
import { buildHelpMessage, buildVersionMessage, options } from "vs/platform/environment/node/argv";
import product from "vs/platform/product/node/product";
import pkg from "vs/platform/product/node/package";

import { MainServer, WebviewServer } from "vs/server/src/server";
import "vs/server/src/tar";
import { generateCertificate, generatePassword } from "vs/server/src/util";

interface Args extends ParsedArgs {
	"allow-http"?: boolean;
	auth?: boolean;
	cert?: string;
	"cert-key"?: string;
	"extra-builtin-extensions-dir"?: string;
	"extra-extensions-dir"?: string;
	host?: string;
	open?: string;
	port?: string;
	socket?: string;
	"webview-port"?: string;
	"webview-socket"?: string;
}

// The last item is _ which is like -- so our options need to come before it.
const last = options.pop()!;

// Remove options that won't work or don't make sense.
let i = options.length;
while (i--) {
	switch (options[i].id) {
		case "add":
		case "diff":
		case "file-uri":
		case "folder-uri":
		case "goto":
		case "new-window":
		case "reuse-window":
		case "wait":
		case "disable-gpu":
		// TODO: pretty sure these don't work but not 100%.
		case "max-memory":
		case "prof-startup":
		case "inspect-extensions":
		case "inspect-brk-extensions":
			options.splice(i, 1);
			break;
	}
}

options.push({ id: "allow-http", type: "boolean", cat: "o", description: "Allow http connections." });
options.push({ id: "cert", type: "string", cat: "o", description: "Path to certificate." });
options.push({ id: "cert-key", type: "string", cat: "o", description: "Path to certificate key." });
options.push({ id: "extra-builtin-extensions-dir", type: "string", cat: "o", description: "Path to extra builtin extension directory." });
options.push({ id: "extra-extensions-dir", type: "string", cat: "o", description: "Path to extra user extension directory." });
options.push({ id: "host", type: "string", cat: "o", description: "Host for the main and webview servers." });
options.push({ id: "no-auth", type: "boolean", cat: "o", description: "Disable password authentication." });
options.push({ id: "open", type: "boolean", cat: "o", description: "Open in the browser on startup." });
options.push({ id: "port", type: "string", cat: "o", description: "Port for the main server." });
options.push({ id: "socket", type: "string", cat: "o", description: "Listen on a socket instead of host:port." });
options.push({ id: "webview-port", type: "string", cat: "o", description: "Port for the webview server." });
options.push({ id: "webview-socket", type: "string", cat: "o", description: "Listen on a socket instead of host:port." });

options.push(last);

interface IMainCli {
	main: (argv: ParsedArgs) => Promise<void>;
}

const main = async (): Promise<void> => {
	const args = validatePaths(parseMainProcessArgv(process.argv)) as Args;
	["extra-extensions-dir", "extra-builtin-extensions-dir"].forEach((key) => {
		if (typeof args[key] === "string") {
			args[key] = [args[key]];
		}
	});

	if (!product.extensionsGallery) {
		product.extensionsGallery = {
			serviceUrl: process.env.SERVICE_URL || "https://v1.extapi.coder.com",
			itemUrl: process.env.ITEM_URL || "",
			controlUrl: "",
			recommendationsUrl: "",
		};
	}

	const version = `${(pkg as any).codeServerVersion || "development"}-vsc${pkg.version}`;
	if (args.help) {
		const executable = `${product.applicationName}${os.platform() === "win32" ? ".exe" : ""}`;
		return console.log(buildHelpMessage(
			product.nameLong, executable,
			version,
			undefined,
			false,
		));
	}

	if (args.version) {
		return console.log(buildVersionMessage(version, product.commit));
	}

	const shouldSpawnCliProcess = (): boolean => {
		return !!args["install-source"]
			|| !!args["list-extensions"]
			|| !!args["install-extension"]
			|| !!args["uninstall-extension"]
			|| !!args["locate-extension"]
			|| !!args["telemetry"];
	};

	if (shouldSpawnCliProcess()) {
		const cli = await new Promise<IMainCli>((c, e) => require(["vs/code/node/cliProcessMain"], c, e));
		await cli.main(args);
		// There is some WriteStream instance keeping it open so force an exit.
		return process.exit(0);
	}

	const options = {
		host: args.host,
		allowHttp: args["allow-http"],
		cert: args.cert,
		certKey: args["cert-key"],
		auth: typeof args.auth !== "undefined" ? args.auth : true,
		password: process.env.PASSWORD,
	};

	if (!options.host) {
		options.host = !options.auth || options.allowHttp
			? "localhost"
			: "0.0.0.0";
	}

	let usingGeneratedCert = false;
	if (!options.allowHttp && (!options.cert || !options.certKey)) {
		const { cert, certKey } = await generateCertificate();
		options.cert = cert;
		options.certKey = certKey;
		usingGeneratedCert = true;
	}

	let usingGeneratedPassword = false;
	if (options.auth && !options.password) {
		options.password = await generatePassword();
		usingGeneratedPassword = true;
	}

	const webviewPort = typeof args["webview-port"] !== "undefined"
		&& parseInt(args["webview-port"], 10) || 8444;
	const webviewServer = new WebviewServer({
		...options,
		port: webviewPort,
		socket: args["webview-socket"],
	});

	const port = typeof args.port !== "undefined" && parseInt(args.port, 10) || 8443;
	const server = new MainServer({
		...options,
		port,
		socket: args.socket,
	}, webviewServer, args);

	const [webviewAddress, serverAddress] = await Promise.all([
		webviewServer.listen(),
		server.listen()
	]);
	console.log(`Main server listening on ${serverAddress}`);
	console.log(`Webview server listening on ${webviewAddress}`);

	if (usingGeneratedPassword) {
		console.log("  - Password is", options.password);
		console.log("  - To use your own password, set the PASSWORD environment variable");
	} else if (options.auth) {
		console.log("  - Using custom password for authentication");
	} else {
		console.log("  - No authentication");
	}

	if (!options.allowHttp && options.cert && options.certKey) {
		console.log(
			usingGeneratedCert
				? `  - Using generated certificate and key in ${path.dirname(options.cert)} for HTTPS`
				: "  - Using provided certificate and key for HTTPS",
		);
	} else {
		console.log("  - Not serving HTTPS");
	}
};

main().catch((error) => {
	console.error(error);
	process.exit(1);
});