Class QuestionFeature

ai/ai.ts:4362–4539

class QuestionFeature extends BaseFeatureHandler {
  readonly name = "AI 提问";
  readonly command = "";
  readonly description = "向 AI 提问";

  private aiService: AIService;
  private messageUtils: MessageUtils;
  private activeToken?: AbortToken;

  constructor(
    aiService: AIService,
    configManagerPromise: Promise<ConfigManager>,
    httpClient: HttpClient,
  ) {
    super(configManagerPromise);
    this.aiService = aiService;
    this.messageUtils = new MessageUtils(configManagerPromise, httpClient);
  }

  cancelCurrentOperation(): void {
    if (this.activeToken && !this.activeToken.aborted)
      this.activeToken.abort("操作被取消");
    this.activeToken = undefined;
  }

  private async runQuestion(
    msg: Api.Message,
    question: string,
    trigger?: Api.Message,
  ): Promise<void> {
    this.cancelCurrentOperation();

    const token = this.aiService.createAbortToken();
    this.activeToken = token;

    try {
      await this.handleQuestion(msg, question, trigger, token);
    } finally {
      this.activeToken = undefined;
      this.aiService.releaseToken(token);
    }
  }

  async execute(
    msg: Api.Message,
    args: string[],
    _prefixes: string[],
  ): Promise<void> {
    const question = args.join(" ").trim();
    await this.runQuestion(msg, question);
  }

  async askFromReply(msg: Api.Message, trigger?: Api.Message): Promise<void> {
    const replyMsg = await safeGetReplyMessage(msg);
    requireUser(!!replyMsg, "至少需要一条提示");
    const question = getMessageText(replyMsg).trim();
    await this.runQuestion(msg, question, trigger);
  }