azuredatastudio/extensions/resource-deployment/src/utils.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/
import * as path from 'path';
import { ToolsInstallPath } from './constants';
import { ITool, NoteBookEnvironmentVariablePrefix } from './interfaces';

export function getErrorMessage(error: any): string {
	return (error instanceof Error)
		? (typeof error.message === 'string' ? error.message : '')
		: typeof error === 'string' ? error : `${JSON.stringify(error, undefined, '\t')}`;
}

export function getDateTimeString(): string {
	return new Date().toISOString().slice(0, 19).replace(/[^0-9]/g, ''); // Take the date time information and only leaving the numbers
}


export function getRuntimeBinaryPathEnvironmentVariableName(toolName: string): string {
	return `${NoteBookEnvironmentVariablePrefix}${toolName.toUpperCase().replace(/ |-/g, '_')}`;
}

export function setEnvironmentVariablesForInstallPaths(tools: ITool[], env: NodeJS.ProcessEnv = process.env): void {
	// Use Set class to make sure the collection only contains unique values.
	let installationPaths: Set<string> = new Set<string>();
	tools.forEach(t => {
		if (t.installationPathOrAdditionalInformation) {

			// construct an env variable name with NoteBookEnvironmentVariablePrefix prefix
			// and tool.name as suffix, making sure of using all uppercase characters and only _ as separator
			const envVarName = getRuntimeBinaryPathEnvironmentVariableName(t.name);
			env[envVarName] = t.installationPathOrAdditionalInformation;
			installationPaths.add(path.dirname(t.installationPathOrAdditionalInformation));
		}
	});
	if (installationPaths.size > 0) {
		const envVarToolsInstallationPath: string = [...installationPaths.values()].join(path.delimiter);
		env[ToolsInstallPath] = envVarToolsInstallationPath;
	}
}

export function assert(condition: boolean, message?: string): asserts condition {
	if (!condition) {
		throw new Error(message);
	}
}