402 lines
14 KiB
Swift
402 lines
14 KiB
Swift
|
//
|
|||
|
|
// ChatModePickSheet.swift
|
|||
|
|
// Crush
|
|||
|
|
//
|
|||
|
|
// Created by Leon on 2025/8/17.
|
|||
|
|
//
|
|||
|
|
|
|||
|
|
import UIKit
|
|||
|
|
|
|||
|
|
/// 对话模型选择回调
|
|||
|
|
typealias ChatModelSelectionCallback = (AIChatModel) -> Void
|
|||
|
|
|
|||
|
|
/// 对话模型选择
|
|||
|
|
class ChatModePickSheet: EGPopBaseView {
|
|||
|
|
var closeButton: EPIconTertiaryButton!
|
|||
|
|
var titleLabel: UILabel!
|
|||
|
|
|
|||
|
|
// 滚动视图
|
|||
|
|
var scrollView: UIScrollView!
|
|||
|
|
|
|||
|
|
// 模型卡片视图数组
|
|||
|
|
var modelCardViews: [ChatModelCardView] = []
|
|||
|
|
|
|||
|
|
var stayTunedLabel: CLLabel!
|
|||
|
|
|
|||
|
|
// 数据源
|
|||
|
|
private var chatModels: [AIChatModel] = []
|
|||
|
|
private var currentSelectedModelCode: String?
|
|||
|
|
|
|||
|
|
var cardHeight = 200.0//168.0
|
|||
|
|
// 选择回调
|
|||
|
|
var selectionCallback: ChatModelSelectionCallback?
|
|||
|
|
|
|||
|
|
required init?(coder: NSCoder) {
|
|||
|
|
fatalError("init(coder:) has not been implemented")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
init(currentSelectedModelCode: String? = nil) {
|
|||
|
|
self.currentSelectedModelCode = currentSelectedModelCode
|
|||
|
|
super.init(direction: .bottom)
|
|||
|
|
|
|||
|
|
contentView.backgroundColor = .c.csbn
|
|||
|
|
// 初始高度，后续会根据数据动态调整
|
|||
|
|
contentLength = 200 + UIWindow.safeAreaBottom
|
|||
|
|
|
|||
|
|
setupViews()
|
|||
|
|
loadChatModels()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func setupViews() {
|
|||
|
|
closeButton = {
|
|||
|
|
let v = EPIconTertiaryButton(radius: .round, iconSize: .small, iconCode: .delete)
|
|||
|
|
v.addTarget(self, action: #selector(bgButtonPressed), for: .touchUpInside)
|
|||
|
|
contentView.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.top.equalToSuperview().offset(20)
|
|||
|
|
make.size.equalTo(v.bgImageSize())
|
|||
|
|
make.trailing.equalToSuperview().offset(-16)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
titleLabel = {
|
|||
|
|
let v = UILabel()
|
|||
|
|
v.font = .t.ttm
|
|||
|
|
v.textColor = .text
|
|||
|
|
v.textAlignment = .center
|
|||
|
|
contentView.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(24)
|
|||
|
|
make.trailing.equalToSuperview().offset(-24)
|
|||
|
|
make.top.equalToSuperview().offset(32)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
scrollView = {
|
|||
|
|
let v = UIScrollView()
|
|||
|
|
v.showsVerticalScrollIndicator = false
|
|||
|
|
v.showsHorizontalScrollIndicator = false
|
|||
|
|
v.delegate = self
|
|||
|
|
contentView.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview()
|
|||
|
|
make.trailing.equalToSuperview()
|
|||
|
|
make.top.equalTo(titleLabel.snp.bottom).offset(24)
|
|||
|
|
make.height.equalTo(cardHeight) // 每个卡片的高度
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
stayTunedLabel = {
|
|||
|
|
let v = CLLabel()
|
|||
|
|
v.font = .t.tbs
|
|||
|
|
v.textColor = .c.ctsn
|
|||
|
|
contentView.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(24)
|
|||
|
|
make.top.equalTo(scrollView.snp.bottom).offset(16)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
titleLabel.text = "Dialog Model"
|
|||
|
|
stayTunedLabel.text = "Stay tuned for more models"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func loadChatModels() {
|
|||
|
|
guard let models = AppDictManager.shared.chatModels, models.count > 0 else {
|
|||
|
|
// 如果没有数据，尝试加载
|
|||
|
|
AppDictManager.shared.loadChatModelDict { [weak self] success in
|
|||
|
|
if success {
|
|||
|
|
self?.loadChatModels()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
chatModels = models
|
|||
|
|
setupModelCards()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func setupModelCards() {
|
|||
|
|
// 清除之前的卡片
|
|||
|
|
modelCardViews.forEach { $0.removeFromSuperview() }
|
|||
|
|
modelCardViews.removeAll()
|
|||
|
|
|
|||
|
|
// 创建新的卡片
|
|||
|
|
for (index, model) in chatModels.enumerated() {
|
|||
|
|
let cardView = ChatModelCardView()
|
|||
|
|
cardView.configure(with: model, isSelected: model.code == currentSelectedModelCode)
|
|||
|
|
cardView.tag = index
|
|||
|
|
cardView.addGestureRecognizer(UITapGestureRecognizer(target: self, action: #selector(modelCardTapped(_:))))
|
|||
|
|
|
|||
|
|
scrollView.addSubview(cardView)
|
|||
|
|
modelCardViews.append(cardView)
|
|||
|
|
|
|||
|
|
cardView.snp.makeConstraints { make in
|
|||
|
|
make.leading.trailing.equalToSuperview()
|
|||
|
|
make.width.equalTo(scrollView.snp.width)
|
|||
|
|
make.height.equalTo(cardHeight) // 每个卡片固定高度
|
|||
|
|
if index == 0 {
|
|||
|
|
make.top.equalToSuperview()
|
|||
|
|
} else {
|
|||
|
|
make.top.equalTo(modelCardViews[index - 1].snp.bottom).offset(12)
|
|||
|
|
}
|
|||
|
|
if index == chatModels.count - 1 {
|
|||
|
|
make.bottom.equalToSuperview()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 动态调整contentLength
|
|||
|
|
updateContentLength()
|
|||
|
|
|
|||
|
|
// 延迟执行滚动操作，确保视图布局完成
|
|||
|
|
// DispatchQueue.main.async { [weak self] in
|
|||
|
|
// self?.scrollToSelectedModel()
|
|||
|
|
// }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func updateContentLength() {
|
|||
|
|
// 基础高度：标题 + 间距 + 底部标签
|
|||
|
|
let baseHeight: CGFloat = 32 + 24 + 16 + 20 + UIWindow.safeAreaBottom// titleLabel top + spacing + stayTunedLabel + bottom margin
|
|||
|
|
|
|||
|
|
// 卡片高度：每个卡片200 + 间距12（除了最后一个）
|
|||
|
|
let cardHeight: CGFloat = cardHeight
|
|||
|
|
let cardSpacing: CGFloat = 12
|
|||
|
|
let totalCardHeight = CGFloat(chatModels.count) * cardHeight + CGFloat(max(0, chatModels.count - 1)) * cardSpacing
|
|||
|
|
|
|||
|
|
// 滚动视图高度：根据卡片数量调整，最多显示2个卡片的高度
|
|||
|
|
let maxVisibleHeight = min(totalCardHeight, cardHeight * 2 + cardSpacing)
|
|||
|
|
|
|||
|
|
// 更新scrollView高度约束
|
|||
|
|
scrollView.snp.updateConstraints { make in
|
|||
|
|
make.height.equalTo(maxVisibleHeight)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 更新contentLength
|
|||
|
|
contentLength = baseHeight + maxVisibleHeight + UIWindow.safeAreaBottom
|
|||
|
|
|
|||
|
|
// 更新布局
|
|||
|
|
layoutIfNeeded()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func scrollToSelectedModel() {
|
|||
|
|
guard let selectedCode = currentSelectedModelCode,
|
|||
|
|
let selectedIndex = chatModels.firstIndex(where: { $0.code == selectedCode }),
|
|||
|
|
selectedIndex < chatModels.count else { return }
|
|||
|
|
|
|||
|
|
// 确保scrollView已经有正确的frame
|
|||
|
|
guard scrollView.frame.height > 0 else {
|
|||
|
|
// 如果frame还没有设置，再次延迟执行
|
|||
|
|
DispatchQueue.main.async { [weak self] in
|
|||
|
|
self?.scrollToSelectedModel()
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let offsetY = CGFloat(selectedIndex) * (cardHeight + 12) // 卡片高度 + 间距
|
|||
|
|
scrollView.setContentOffset(CGPoint(x: 0, y: offsetY), animated: false)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@objc private func modelCardTapped(_ gesture: UITapGestureRecognizer) {
|
|||
|
|
guard let cardView = gesture.view as? ChatModelCardView,
|
|||
|
|
let index = modelCardViews.firstIndex(of: cardView),
|
|||
|
|
index < chatModels.count else { return }
|
|||
|
|
|
|||
|
|
let selectedModel = chatModels[index]
|
|||
|
|
|
|||
|
|
// 更新选中状态
|
|||
|
|
modelCardViews.forEach { $0.setSelected(false) }
|
|||
|
|
cardView.setSelected(true)
|
|||
|
|
|
|||
|
|
// 调用回调
|
|||
|
|
selectionCallback?(selectedModel)
|
|||
|
|
|
|||
|
|
// 关闭弹窗
|
|||
|
|
dismiss()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MARK: - UIScrollViewDelegate
|
|||
|
|
extension ChatModePickSheet: UIScrollViewDelegate {
|
|||
|
|
func scrollViewDidScroll(_ scrollView: UIScrollView) {
|
|||
|
|
// 垂直滚动不需要页面指示器逻辑
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MARK: - ChatModelCardView
|
|||
|
|
class ChatModelCardView: UIView {
|
|||
|
|
private var block: UIView!
|
|||
|
|
private var modelName: UILabel!
|
|||
|
|
private var modelDesc: UILabel!
|
|||
|
|
private var selectMark: UIImageView!
|
|||
|
|
private var innerBlock: UIView!
|
|||
|
|
private var queryButton: EPIconTertiaryButton!
|
|||
|
|
private var priceStackView: UIStackView!
|
|||
|
|
|
|||
|
|
private var model: AIChatModel?
|
|||
|
|
|
|||
|
|
override init(frame: CGRect) {
|
|||
|
|
super.init(frame: frame)
|
|||
|
|
setupViews()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
required init?(coder: NSCoder) {
|
|||
|
|
fatalError("init(coder:) has not been implemented")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private func setupViews() {
|
|||
|
|
block = {
|
|||
|
|
let v = UIView()
|
|||
|
|
v.backgroundColor = .c.csen
|
|||
|
|
v.cornerRadius = 16
|
|||
|
|
addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
// make.edges.equalToSuperview()
|
|||
|
|
make.leading.equalToSuperview().offset(24)
|
|||
|
|
make.trailing.equalToSuperview().offset(-24)
|
|||
|
|
make.top.bottom.equalToSuperview()
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
selectMark = {
|
|||
|
|
let v = UIImageView()
|
|||
|
|
v.image = UIImage(named: "checkmark_tick")
|
|||
|
|
v.isHidden = true
|
|||
|
|
block.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.size.equalTo(CGSize(width: 20, height: 20))
|
|||
|
|
make.top.equalToSuperview().offset(18)
|
|||
|
|
make.trailing.equalToSuperview().offset(-16)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
modelName = {
|
|||
|
|
let v = CLLabel()
|
|||
|
|
v.font = .t.tts
|
|||
|
|
v.setContentCompressionResistancePriority(UILayoutPriority(749), for: .horizontal)
|
|||
|
|
block.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(16)
|
|||
|
|
make.top.equalToSuperview().offset(16)
|
|||
|
|
make.trailing.lessThanOrEqualTo(selectMark.snp.leading).offset(-8)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
queryButton = {
|
|||
|
|
let v = EPIconTertiaryButton(radius: .round, iconSize: .xs10, iconCode: .question)
|
|||
|
|
v.addTarget(self, action: #selector(tapQueryButton), for: .touchUpInside)
|
|||
|
|
block.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalTo(modelName.snp.trailing).offset(4)
|
|||
|
|
make.centerY.equalTo(modelName)
|
|||
|
|
make.trailing.lessThanOrEqualToSuperview().offset(-16)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
modelDesc = {
|
|||
|
|
let v = CLLabel()
|
|||
|
|
v.font = .t.tbs
|
|||
|
|
v.textColor = .c.ctsn
|
|||
|
|
block.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(16)
|
|||
|
|
make.trailing.equalToSuperview().offset(-16)
|
|||
|
|
make.top.equalTo(modelName.snp.bottom).offset(4)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
innerBlock = {
|
|||
|
|
let v = UIView()
|
|||
|
|
v.backgroundColor = .c.csdn
|
|||
|
|
v.cornerRadius = 12
|
|||
|
|
block.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(16)
|
|||
|
|
make.trailing.equalToSuperview().offset(-16)
|
|||
|
|
//make.top.equalTo(modelDesc.snp.bottom).offset(12)
|
|||
|
|
make.top.equalToSuperview().offset(76)
|
|||
|
|
make.height.equalTo(108) // 76
|
|||
|
|
//make.bottom.equalToSuperview().offset(-16)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
priceStackView = {
|
|||
|
|
let v = UIStackView()
|
|||
|
|
v.axis = .vertical
|
|||
|
|
v.spacing = 12
|
|||
|
|
v.alignment = .leading
|
|||
|
|
innerBlock.addSubview(v)
|
|||
|
|
v.snp.makeConstraints { make in
|
|||
|
|
make.leading.equalToSuperview().offset(12)
|
|||
|
|
make.trailing.equalToSuperview().offset(-12)
|
|||
|
|
make.top.equalToSuperview().offset(12)
|
|||
|
|
make.bottom.equalToSuperview().offset(-12)
|
|||
|
|
}
|
|||
|
|
return v
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func configure(with model: AIChatModel, isSelected: Bool = false) {
|
|||
|
|
self.model = model
|
|||
|
|
|
|||
|
|
modelName.text = model.name ?? ""
|
|||
|
|
modelDesc.text = model.description ?? ""
|
|||
|
|
|
|||
|
|
// 清除之前的价格标签
|
|||
|
|
priceStackView.arrangedSubviews.forEach { $0.removeFromSuperview() }
|
|||
|
|
|
|||
|
|
// 添加价格信息
|
|||
|
|
if let price = model.textPrice {
|
|||
|
|
let coin = Coin(cents: price)
|
|||
|
|
let priceLabel = CLIconLabel()
|
|||
|
|
priceLabel.iconImageView.image = UIImage(named: "icon_16_diamond")
|
|||
|
|
priceLabel.contentLabel.text = "\(coin.thousandthsFormatted)/Text Message"
|
|||
|
|
priceStackView.addArrangedSubview(priceLabel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if let price = model.voicePrice {
|
|||
|
|
let coin = Coin(cents: price)
|
|||
|
|
let priceLabel = CLIconLabel()
|
|||
|
|
priceLabel.iconImageView.image = UIImage(named: "icon_16_diamond")
|
|||
|
|
priceLabel.contentLabel.text = "\(coin.thousandthsFormatted)/Send or play voice"
|
|||
|
|
priceStackView.addArrangedSubview(priceLabel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if let price = model.voiceChatPrice {
|
|||
|
|
let coin = Coin(cents: price)
|
|||
|
|
let priceLabel = CLIconLabel()
|
|||
|
|
priceLabel.iconImageView.image = UIImage(named: "icon_16_diamond")
|
|||
|
|
priceLabel.contentLabel.text = "\(coin.thousandthsFormatted)/ 1 min Voice call"
|
|||
|
|
priceStackView.addArrangedSubview(priceLabel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
setSelected(isSelected)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func setSelected(_ selected: Bool) {
|
|||
|
|
selectMark.isHidden = !selected
|
|||
|
|
// block.layer.borderWidth = selected ? 2 : 0
|
|||
|
|
// block.layer.borderColor = selected ? UIColor.c.cpn.cgColor : UIColor.clear.cgColor
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@objc func tapQueryButton(){
|
|||
|
|
let content = "*文本消息价格是指与角色进行文本消息对话的价格，含发送文本，发送图片，发送礼物；按条计算\n\n*发送语音消息价格是指与角色发送语音或者播放角色的语音的价格；按次计算\n\n*语音通话消息价格是指与角色进行语音电话对话的价格；按分钟计算"
|
|||
|
|
let alert = Alert(title: "Tips", text: content)
|
|||
|
|
let action1 = AlertAction(title: "Got it", actionStyle: .confirm) {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
alert.addAction(action1)
|
|||
|
|
alert.show()
|
|||
|
|
}
|
|||
|
|
}
|