keyboard/keyBoard/Class/AiTalk/VC/KBAiMainVC.m

//
//  KBAiMainVC.m
//  keyBoard
//
//  Created by Mac on 2026/1/15.
//

#import "KBAiMainVC.h"
#import "ConversationOrchestrator.h"
#import "KBAICommentView.h"
#import "KBAiChatView.h"
#import "KBAiRecordButton.h"
#import "LSTPopView.h"

@interface KBAiMainVC () <KBAiRecordButtonDelegate>
@property(nonatomic, weak) LSTPopView *popView;

// UI
@property(nonatomic, strong) KBAiChatView *chatView;
@property(nonatomic, strong) KBAiRecordButton *recordButton;
@property(nonatomic, strong) UILabel *statusLabel;
@property(nonatomic, strong) UIButton *commentButton;
@property(nonatomic, strong) KBAICommentView *commentView;

// 核心模块
@property(nonatomic, strong) ConversationOrchestrator *orchestrator;

@end

@implementation KBAiMainVC

#pragma mark - Lifecycle

- (void)viewDidLoad {
  [super viewDidLoad];

  [self setupUI];
  [self setupOrchestrator];
}

- (void)viewWillDisappear:(BOOL)animated {
  [super viewWillDisappear:animated];

  // 页面消失时停止对话
  [self.orchestrator stop];
}

#pragma mark - UI Setup

- (void)setupUI {
  self.view.backgroundColor = [UIColor systemBackgroundColor];
  self.title = @"AI 助手";

  // 安全区域
  UILayoutGuide *safeArea = self.view.safeAreaLayoutGuide;

  // 状态标签
  self.statusLabel = [[UILabel alloc] init];
  self.statusLabel.text = @"按住按钮开始对话";
  self.statusLabel.font = [UIFont systemFontOfSize:14];
  self.statusLabel.textColor = [UIColor secondaryLabelColor];
  self.statusLabel.textAlignment = NSTextAlignmentCenter;
  self.statusLabel.translatesAutoresizingMaskIntoConstraints = NO;
  [self.view addSubview:self.statusLabel];

  // 聊天视图
  self.chatView = [[KBAiChatView alloc] init];
  self.chatView.backgroundColor = [UIColor systemBackgroundColor];
  self.chatView.translatesAutoresizingMaskIntoConstraints = NO;
  [self.view addSubview:self.chatView];

  // 录音按钮
  self.recordButton = [[KBAiRecordButton alloc] init];
  self.recordButton.delegate = self;
  self.recordButton.translatesAutoresizingMaskIntoConstraints = NO;
  [self.view addSubview:self.recordButton];

  // 评论按钮（聊天视图右侧居中）
  self.commentButton = [UIButton buttonWithType:UIButtonTypeCustom];
  [self.commentButton setImage:[UIImage systemImageNamed:@"bubble.right.fill"]
                      forState:UIControlStateNormal];
  self.commentButton.tintColor = [UIColor whiteColor];
  self.commentButton.backgroundColor = [UIColor systemBlueColor];
  self.commentButton.layer.cornerRadius = 25;
  self.commentButton.layer.shadowColor = [UIColor blackColor].CGColor;
  self.commentButton.layer.shadowOffset = CGSizeMake(0, 2);
  self.commentButton.layer.shadowOpacity = 0.3;
  self.commentButton.layer.shadowRadius = 4;
  self.commentButton.translatesAutoresizingMaskIntoConstraints = NO;
  [self.commentButton addTarget:self
                         action:@selector(showComment)
               forControlEvents:UIControlEventTouchUpInside];
  [self.view addSubview:self.commentButton];

  // 布局约束
  [NSLayoutConstraint activateConstraints:@[
    // 状态标签
    [self.statusLabel.topAnchor constraintEqualToAnchor:safeArea.topAnchor
                                               constant:8],
    [self.statusLabel.leadingAnchor
        constraintEqualToAnchor:safeArea.leadingAnchor
                       constant:16],
    [self.statusLabel.trailingAnchor
        constraintEqualToAnchor:safeArea.trailingAnchor
                       constant:-16],

    // 聊天视图
    [self.chatView.topAnchor
        constraintEqualToAnchor:self.statusLabel.bottomAnchor
                       constant:8],
    [self.chatView.leadingAnchor
        constraintEqualToAnchor:safeArea.leadingAnchor],
    [self.chatView.trailingAnchor
        constraintEqualToAnchor:safeArea.trailingAnchor],
    [self.chatView.bottomAnchor
        constraintEqualToAnchor:self.recordButton.topAnchor
                       constant:-16],

    // 录音按钮
    [self.recordButton.leadingAnchor
        constraintEqualToAnchor:safeArea.leadingAnchor
                       constant:20],
    [self.recordButton.trailingAnchor
        constraintEqualToAnchor:safeArea.trailingAnchor
                       constant:-20],
    [self.recordButton.bottomAnchor
        constraintEqualToAnchor:safeArea.bottomAnchor
                       constant:-16],
    [self.recordButton.heightAnchor constraintEqualToConstant:50],

    // 评论按钮（右侧居中）
    [self.commentButton.trailingAnchor
        constraintEqualToAnchor:safeArea.trailingAnchor
                       constant:-16],
    [self.commentButton.centerYAnchor
        constraintEqualToAnchor:self.chatView.centerYAnchor],
    [self.commentButton.widthAnchor constraintEqualToConstant:50],
    [self.commentButton.heightAnchor constraintEqualToConstant:50],
  ]];
}

#pragma mark - Orchestrator Setup

- (void)setupOrchestrator {
  self.orchestrator = [[ConversationOrchestrator alloc] init];

  // 配置服务器地址（TODO: 替换为实际地址）
  // self.orchestrator.asrServerURL = @"wss://your-asr-server.com/ws/asr";
  // self.orchestrator.llmServerURL =
  // @"https://your-llm-server.com/api/chat/stream";
  // self.orchestrator.ttsServerURL = @"https://your-tts-server.com/api/tts";

  __weak typeof(self) weakSelf = self;

  // 状态变化回调
  self.orchestrator.onStateChange = ^(ConversationState state) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf updateStatusForState:state];
  };

  // 实时识别文本回调
  self.orchestrator.onPartialText = ^(NSString *text) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    strongSelf.statusLabel.text = text.length > 0 ? text : @"正在识别...";
  };

  // 用户最终文本回调
  self.orchestrator.onUserFinalText = ^(NSString *text) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    if (text.length > 0) {
      [strongSelf.chatView addUserMessage:text];
    }
  };

  // AI 可见文本回调（打字机效果）
  self.orchestrator.onAssistantVisibleText = ^(NSString *text) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf.chatView updateLastAssistantMessage:text];
  };

  // AI 完整回复回调
  self.orchestrator.onAssistantFullText = ^(NSString *text) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf.chatView updateLastAssistantMessage:text];
    [strongSelf.chatView markLastAssistantMessageComplete];
  };

  // 音量更新回调
  self.orchestrator.onVolumeUpdate = ^(float rms) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf.recordButton updateVolumeRMS:rms];
  };

  // AI 开始说话
  self.orchestrator.onSpeakingStart = ^{
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    // 添加空的 AI 消息占位
    [strongSelf.chatView addAssistantMessage:@""];
  };

  // AI 说话结束
  self.orchestrator.onSpeakingEnd = ^{
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf.chatView markLastAssistantMessageComplete];
  };

  // 错误回调
  self.orchestrator.onError = ^(NSError *error) {
    __strong typeof(weakSelf) strongSelf = weakSelf;
    if (!strongSelf)
      return;

    [strongSelf showError:error];
  };
}

#pragma mark - 事件
- (void)showComment {
    CGFloat customViewHeight = KB_SCREEN_HEIGHT * (0.8);
    KBAICommentView *customView = [[KBAICommentView alloc]
                                   initWithFrame:CGRectMake(0, 0, KB_SCREEN_WIDTH, customViewHeight)];
    LSTPopView *popView =
    [LSTPopView initWithCustomView:customView
                        parentView:nil
                          popStyle:LSTPopStyleSmoothFromBottom
                      dismissStyle:LSTDismissStyleSmoothToBottom];
    self.popView = popView;
    popView.priority = 1000;
    popView.isAvoidKeyboard = false;
    popView.hemStyle = LSTHemStyleBottom;
    popView.dragStyle = LSTDragStyleY_Positive;
    popView.dragDistance = customViewHeight * 0.5;
    popView.sweepStyle = LSTSweepStyleY_Positive;
    popView.swipeVelocity = 1600;
    popView.sweepDismissStyle = LSTSweepDismissStyleSmooth;
    
    [popView pop];
}

- (void)showCommentDirectly {
  if (self.commentView.superview) {
    [self.view bringSubviewToFront:self.commentView];
    return;
  }

  CGFloat customViewHeight = KB_SCREEN_HEIGHT * (0.8);
  KBAICommentView *customView = [[KBAICommentView alloc] initWithFrame:CGRectZero];
  customView.translatesAutoresizingMaskIntoConstraints = NO;
  [self.view addSubview:customView];
  [NSLayoutConstraint activateConstraints:@[
    [customView.leadingAnchor constraintEqualToAnchor:self.view.leadingAnchor],
    [customView.trailingAnchor constraintEqualToAnchor:self.view.trailingAnchor],
    [customView.bottomAnchor constraintEqualToAnchor:self.view.bottomAnchor],
    [customView.heightAnchor constraintEqualToConstant:customViewHeight],
  ]];
  self.commentView = customView;
}

#pragma mark - UI Updates

- (void)updateStatusForState:(ConversationState)state {
  switch (state) {
  case ConversationStateIdle:
    self.statusLabel.text = @"按住按钮开始对话";
    self.recordButton.state = KBAiRecordButtonStateNormal;
    break;

  case ConversationStateListening:
    self.statusLabel.text = @"正在聆听...";
    self.recordButton.state = KBAiRecordButtonStateRecording;
    break;

  case ConversationStateRecognizing:
    self.statusLabel.text = @"正在识别...";
    self.recordButton.state = KBAiRecordButtonStateNormal;
    break;

  case ConversationStateThinking:
    self.statusLabel.text = @"AI 正在思考...";
    self.recordButton.state = KBAiRecordButtonStateNormal;
    break;

  case ConversationStateSpeaking:
    self.statusLabel.text = @"AI 正在回复...";
    self.recordButton.state = KBAiRecordButtonStateNormal;
    break;
  }
}

- (void)showError:(NSError *)error {
  UIAlertController *alert =
      [UIAlertController alertControllerWithTitle:@"错误"
                                          message:error.localizedDescription
                                   preferredStyle:UIAlertControllerStyleAlert];
  [alert addAction:[UIAlertAction actionWithTitle:@"确定"
                                            style:UIAlertActionStyleDefault
                                          handler:nil]];
  [self presentViewController:alert animated:YES completion:nil];
}

#pragma mark - KBAiRecordButtonDelegate

- (void)recordButtonDidBeginPress:(KBAiRecordButton *)button {
  [self.orchestrator userDidPressRecord];
}

- (void)recordButtonDidEndPress:(KBAiRecordButton *)button {
  [self.orchestrator userDidReleaseRecord];
}

- (void)recordButtonDidCancelPress:(KBAiRecordButton *)button {
  // 取消录音（同样调用 release，ASR 会返回空或部分结果）
  [self.orchestrator userDidReleaseRecord];
}

@end