Class ImageFeature

ai/ai.ts:4694–4805

class ImageFeature extends BaseFeatureHandler {
  readonly name = "图片生成";
  readonly command = "image";
  readonly description = "生成图片";

  private aiService: AIService;
  private messageUtils: MessageUtils;
  private httpClient: HttpClient;

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

  async execute(
    msg: Api.Message,
    args: string[],
    _prefixes: string[],
  ): Promise<void> {
    const prefixes = getPrefixes();
    const configManager = await this.getConfigManager();
    const config = configManager.getConfig();
    const replyMsg = await safeGetReplyMessage(msg);
    const replyToId = replyMsg?.id;

    const subCommand = args[1]?.toLowerCase();
    if (subCommand === "preview") {
      const state = args[2]?.toLowerCase();
      if (!state) {
        await this.editMessage(
          msg,
          `🖼️ <b>图片预览状态:</b>\n\n📄 当前状态: ${config.imagePreview ? "开启" : "关闭"}`,
        );
        return;
      }
      requireUser(state === "on" || state === "off", "参数必须是 on 或 off");
      await configManager.updateConfig((cfg) => {
        cfg.imagePreview = state === "on";
      });
      await this.editMessage(
        msg,
        `✅ 图片预览已${state === "on" ? "开启" : "关闭"}`,
      );
      return;
    }

    const promptInput = args.slice(1).join(" ").trim();
    const replyText = getMessageText(replyMsg).trim();
    const replyImageParts = await getMessageImageParts(replyMsg);
    const messageImageParts = await getMessageImageParts(msg);
    const imageParts = [...replyImageParts, ...messageImageParts];

    const hasPrompt = !!promptInput || !!replyText;