DeepCodeGeniusWeb-vscode/src/contributes/commandsBase.ts

// src/contributes/commandsBase.ts

import { UiUtilWrapper } from "../util/uiUtil";
import { runCommand } from "../util/commonUtil";
import { logger } from "../util/logger";


export function checkDevChatDependency(pythonCommand: string): boolean {
	try {
		const binPath = getPipxEnvironmentPath(pythonCommand);

		if (binPath) {
			updateEnvironmentPath(binPath);
		} else {
			logger.channel()?.info(`Failed to obtain the pipx environment path.`);
		}
	} catch (error) {
		// DevChat dependency check failed
		// log out detail error message
		logger.channel()?.info(`Failed to check DevChat dependency due to error: ${error}`);
	}

	try {
		// Check if DevChat is installed
		runCommand('devchat --help');
		return true;
	} catch(error) {
		logger.channel()?.error(`Failed to check DevChat dependency due to error: ${error}`);
		return false;
	}
}

function getDefaultPythonCommand(): string | undefined {
	try {
		runCommand('python3 -V');
		try {
			return runCommand('where python3').toString().trim();
		} catch (error) {
			return runCommand('which python3').toString().trim();
		}
	} catch (error) {
		try {
			const version = runCommand('python -V');
			if (version.includes('Python 3')) {
				try {
					return runCommand('where python').toString().trim();
				} catch (error) {
					return runCommand('which python').toString().trim();
				}
			}
			return undefined;
		} catch (error) {
			return undefined;
		}
	}
}

export function getValidPythonCommand(): string | undefined {
	try {
		const pythonCommand = UiUtilWrapper.getConfiguration('DevChat', 'PythonPath');
		if (pythonCommand) {
			return pythonCommand;
		}

		const defaultPythonCommand = getDefaultPythonCommand();
		if (defaultPythonCommand) {
			UiUtilWrapper.updateConfiguration('DevChat', 'PythonPath', defaultPythonCommand);
		}

		return defaultPythonCommand;
	} catch (error) {
		return undefined;
	}
}

export function getPipxEnvironmentPath(pythonCommand: string): string | null {
	// Get pipx environment
	const pipxEnvOutput = runCommand(`"${pythonCommand}" -m pipx environment`).toString();
	const binPathRegex = /PIPX_BIN_DIR=\s*(.*)/;

	// Get BIN path from pipx environment
	const match = pipxEnvOutput.match(binPathRegex);
	if (match && match[1]) {
		return match[1];
	} else {
		return null;
	}
}

function updateEnvironmentPath(binPath: string): void {
	// Add BIN path to PATH
	if (process.env.PATH?.indexOf(binPath) === undefined || process.env.PATH?.indexOf(binPath) < 0) {
		process.env.PATH = `${binPath}:${process.env.PATH}`;
		logger.channel()?.info(`Added ${binPath} to PATH.`);
	}
}