64 lines
1.4 KiB
TypeScript
64 lines
1.4 KiB
TypeScript
|
import type { Plugin } from "@opencode-ai/plugin";
|
||
|
|
|
||
|
|
const MAX_LABEL_LENGTH = 30;
|
||
|
|
|
||
|
|
interface QuestionOption {
|
||
|
|
label: string;
|
||
|
|
description?: string;
|
||
|
|
}
|
||
|
|
|
||
|
|
interface Question {
|
||
|
|
question: string;
|
||
|
|
header?: string;
|
||
|
|
options: QuestionOption[];
|
||
|
|
multiSelect?: boolean;
|
||
|
|
}
|
||
|
|
|
||
|
|
interface AskUserQuestionArgs {
|
||
|
|
questions: Question[];
|
||
|
|
}
|
||
|
|
|
||
|
|
function truncateLabel(label: string, maxLength: number = MAX_LABEL_LENGTH): string {
|
||
|
|
if (label.length <= maxLength) {
|
||
|
|
return label;
|
||
|
|
}
|
||
|
|
return label.substring(0, maxLength - 3) + "...";
|
||
|
|
}
|
||
|
|
|
||
|
|
function truncateQuestionLabels(args: AskUserQuestionArgs): AskUserQuestionArgs {
|
||
|
|
if (!args.questions || !Array.isArray(args.questions)) {
|
||
|
|
return args;
|
||
|
|
}
|
||
|
|
|
||
|
|
return {
|
||
|
|
...args,
|
||
|
|
questions: args.questions.map((question) => ({
|
||
|
|
...question,
|
||
|
|
options: question.options?.map((option) => ({
|
||
|
|
...option,
|
||
|
|
label: truncateLabel(option.label),
|
||
|
|
})) ?? [],
|
||
|
|
})),
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
export function createQuestionLabelTruncatorHook(): Pick<
|
||
|
|
Plugin,
|
||
|
|
"tool.execute.before"
|
||
|
|
> {
|
||
|
|
return {
|
||
|
|
"tool.execute.before": async (input, output) => {
|
||
|
|
const toolName = input.tool?.toLowerCase();
|
||
|
|
|
||
|
|
if (toolName === "askuserquestion" || toolName === "ask_user_question") {
|
||
|
|
const args = output.args as AskUserQuestionArgs | undefined;
|
||
|
|
|
||
|
|
if (args?.questions) {
|
||
|
|
const truncatedArgs = truncateQuestionLabels(args);
|
||
|
|
Object.assign(output.args as object, truncatedArgs);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
};
|
||
|
|
}
|