34 lines
1.3 KiB
TypeScript
34 lines
1.3 KiB
TypeScript
import * as path from 'path';
|
|
import { ChatContext } from './contextManager';
|
|
import { createTempSubdirectory, runCommandStringAndWriteOutput } from '../util/commonUtil';
|
|
import { logger } from '../util/logger';
|
|
import { UiUtilWrapper } from '../util/uiUtil';
|
|
|
|
|
|
export const customCommandContext: ChatContext = {
|
|
name: '<Custom Local Command>',
|
|
description: 'Click this and enter your desired command to run. The return will be added to the context.',
|
|
handler: async () => {
|
|
// popup a dialog to ask for the command line to run
|
|
const customCommand = await UiUtilWrapper.showInputBox({
|
|
prompt: 'Input your custom command',
|
|
placeHolder: 'for example: ls -l'
|
|
});
|
|
|
|
// 检查用户是否输入了命令
|
|
if (customCommand) {
|
|
const tempDir = await createTempSubdirectory('devchat/context');
|
|
const diffFile = path.join(tempDir, 'custom.txt');
|
|
|
|
logger.channel()?.trace(`Your custom command is: ${customCommand}`);
|
|
const result = await runCommandStringAndWriteOutput(customCommand, diffFile);
|
|
logger.channel()?.trace(` exit code:`, result.exitCode);
|
|
|
|
logger.channel()?.trace(` stdout:`, result.stdout);
|
|
logger.channel()?.trace(` stderr:`, result.stderr);
|
|
return [`[context|${diffFile}]`];
|
|
}
|
|
return [];
|
|
},
|
|
};
|