#include "ctai_history_widget.h"

ctai_history_widget::ctai_history_widget(QWidget *parent)
    : QTableWidget(parent)
{
    setup_table();
}

void ctai_history_widget::setup_table()
{
    setColumnCount(1);
    setShowGrid(false);
    verticalHeader()->hide();
    horizontalHeader()->hide();
    horizontalHeader()->setSectionResizeMode(0, QHeaderView::Stretch);
    verticalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents);
    // 设置自动滚动属性
    setVerticalScrollMode(QAbstractItemView::ScrollPerPixel);
    // 滚动步长
    verticalScrollBar()->setSingleStep(10);
}

ctai_history_widget::~ctai_history_widget()
{
    for (auto &pair : message_map_)
    {
        delete pair.second.first; // 删除model_data
    }
    clear();
}

void ctai_history_widget::add_message(const model_data &message)
{
    QString send_id;
    if (message.msg_type_mode == SYSTEM)
    {
        send_id = QSL(message.postback_send_id);
    }
    else
    {
        send_id = QSL(message.send_user_id);
    }

    auto it = message_map_.find(send_id);
    if (it != message_map_.end() && it->second.first->msg_type_mode == SYSTEM)
    {
        // 更新现有消息
        model_data *stored_data = it->second.first;
        int row = it->second.second;

        //如果传入的对象属性为null或者[DONE]表示传输完毕或者其他情况
        //此时需要将tokens更新掉所有其他属性不更新传输数据
        QWidget *widget = cellWidget(row, 0);
        if (!QSL(message.postback_model_data).isEmpty() &&
            !QSL(message.postback_model_data).contains("[DONE]"))
        {
            //更新传入信息
            stored_data->postback_model_data += message.postback_model_data;
            if (auto message_widget = qobject_cast<ctai_history_textedit *>(widget))
            {
                message_widget->add_system_message(message);
            }
        }
        else
        {
            //更新tokens信息
            update_stored_data(stored_data, message);
            if (auto message_widget = qobject_cast<ctai_history_textedit *>(widget))
            {
                message_widget->update_tokens_message(message);
            }
        }
    }
    else
    {
        // 添加新消息
        model_data *new_data = new model_data(message);
        ctai_history_textedit *message_widget = new ctai_history_textedit(this);
        message_widget->init_layout(new_data->msg_type_mode);
        message_widget->show();
        if(new_data->msg_type_mode==SYSTEM){
            message_widget->add_header_message(*new_data);
            message_widget->update_tokens_message(*new_data);
            message_widget->add_system_message(*new_data);
        }else{
            message_widget->add_header_message(*new_data);
            message_widget->add_user_message(*new_data);
        }
        
        // 创建新行
        int row = rowCount();
        insertRow(row);
        setCellWidget(row, 0, message_widget);
        // 存储到map中
        message_map_[send_id] = std::make_pair(new_data, row);
        // 连接删除信号
        connect(message_widget, SIGNAL(delete_requested(QString)), this, SLOT(on_msg_remove(QString)));
        // 连接高度变化信号
        connect(message_widget, &ctai_history_textedit::current_row_height_changed,this, &ctai_history_widget::on_current_rows_height_changed);
        connect(message_widget, &ctai_history_textedit::fold_row_height_changed,this, &ctai_history_widget::on_rows_height_changed);
        // 滚动到新消息
        scrollToItem(item(row, 0));
    }
    // 立即滚动到底部
    scrollToBottom();
}
//适合信息折叠
void ctai_history_widget::on_rows_height_changed()
{
    //全部行高自适应
    for (int row = 0; row < rowCount(); ++row) {
        resizeRowToContents(row);
    }
    scrollToBottom();
}
//适合自适应流式输出内容
void ctai_history_widget::on_current_rows_height_changed()
{
    // 当前行高自适应
    resizeRowToContents(rowCount());
    scrollToBottom();
}
void ctai_history_widget::on_msg_remove(QString send_id)
{
    int row = this->rowCount();
    if (row >= 0)
    {
        // 从map中删除
        qDebug() << "remove_row:" << row;
        auto it = message_map_.find(send_id);
        if (it != message_map_.end())
        {
            qDebug() << "remove it:" << it->second.first;
            delete it->second.first; // 删除model_data
            message_map_.erase(it);
        }
        qDebug() << "current_row:" << this->currentRow();
        // 移除并删除该项
        removeRow(this->currentRow());
    }
}

ctai_history_textedit *ctai_history_widget::current_message() const
{
    int current_row = currentRow();
    if (current_row >= 0)
    {
        return qobject_cast<ctai_history_textedit *>(cellWidget(current_row, 0));
    }
    return nullptr;
}

void ctai_history_widget::update_stored_data(model_data *stored_data, const model_data &message)
{
    stored_data->api_key = message.api_key;
    stored_data->api_url = message.api_url;
    stored_data->postback_completion_tokens = message.postback_completion_tokens;
    stored_data->postback_finish_reason = message.postback_finish_reason;
    stored_data->postback_prompt_cache_hit_tokens = message.postback_prompt_cache_hit_tokens;
    stored_data->postback_prompt_cache_miss_tokens = message.postback_prompt_cache_miss_tokens;
    stored_data->postback_prompt_tokens = message.postback_prompt_tokens;
    stored_data->postback_send_id = message.postback_send_id;
    stored_data->postback_server_model = message.postback_server_model;
    stored_data->postback_stream_mode = message.postback_stream_mode;
    stored_data->postback_system_fingerprint = message.postback_system_fingerprint;
    stored_data->postback_time = message.postback_time;
    stored_data->postback_total_tokens = message.postback_total_tokens;
    stored_data->request_body = message.request_body;
    stored_data->send_accept_header = message.send_accept_header;
    stored_data->send_content_header = message.send_content_header;
    stored_data->send_user_data = message.send_user_data;
    stored_data->user_model = message.user_model;
    // postback_model_data 不在这里赋值
}