#include "etfstockinfo.h"
#include "ui_etfstockinfo.h"

#include <QNetworkCookie>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <QDebug>
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonArray>

ETFStockInfo::ETFStockInfo(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::ETFStockInfo)
{
    ui->setupUi(this);

    m_model_market = new QStandardItemModel(0,17,this);
    m_model_position = new QStandardItemModel(0,7,this);

    ui->tableView_market->setModel(m_model_market);
//    ui->tableView_position->setModel(m_model_position);

    m_customModel = new CustomSortProxyModel; // 可自定义排序的模型
    m_customModel->setSourceModel(m_model_position); // 设置源模型
    ui->tableView_position->setModel(m_customModel);
    // 表格列排序
    ui->tableView_position->setSortingEnabled(true);

    QStringList labels_market;
    labels_market<<"基金代码"<<"基金名称"<<"成交额(亿)"<<"涨跌额"<<"累计净值"<<"etf类型"<<"涨跌幅"<<"成交量"<<"市价"
                <<"年涨跌幅"<<"粉丝"<<"市值（亿）"<<"手续费"<<"交易量-单批"<<"单位净值"<<"总股数"<<"时间";

    m_model_market->setHorizontalHeaderLabels(labels_market);

    QStringList labels_position;
    labels_position<<"股票名"<<"股票代码"<<"机构名称"<<"持股数(亿)"<<"持股比例%"<<"截止日期"<<"基金数量(家)";

    m_model_position->setHorizontalHeaderLabels(labels_position);


    // 处理数据爬取的操作
    // 模拟浏览器的参数
    QString userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36";
    m_cookie = "__utma=1.731742638.1647403301.1699341909.1700229030.32; device_id=196eef62baf016c7d95a22752d9bdbab; smidV2=20240414233939e95389ecf7ecd2f4d08524ce770aacd500753aa68e9640320; s=c611de27gr; cookiesu=651726298794778; xq_a_token=220b0abef0fac476d076c9f7a3938b7edac35f48; xqat=220b0abef0fac476d076c9f7a3938b7edac35f48; xq_r_token=1d46f0ed628506486164e5055a4993f9b54b2f4c; xq_id_token=eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJ1aWQiOi0xLCJpc3MiOiJ1YyIsImV4cCI6MTcyOTIxMjc4NCwiY3RtIjoxNzI3NDkwOTU3MjA0LCJjaWQiOiJkOWQwbjRBWnVwIn0.C_GmKEhTaaioDMLWkgZyMXDl4duYEVmsdJHsTi7gbcNz0Tohc-uxHsaw0yBT5k-qmbrJ_RaLMCSxy06v14-R3dwL-MsiKRHxHa5qvQZN4BjEgvPRkqqvPgE_fkPLte8qQOEgd5iVkhr-4mjip0-9WCeXYiH7DygxFOBXGlgoPtpPzAtOTm5TWJmXh0ipDsIZxfNOl8jipXYaIdkv_kqLul5gqiBi5qqnwONDa24Zx-Kvpm8ySWiPFBLzZBqTuRBs4oKAMpSdOiYGLVL7dcSDDZyWqAexmrN4f19hkmd6gBHL4dCczRMDGYc1e98sQtlbZ5lgeEuuM24jjcuwCxsxXQ; u=651726298794778; Hm_lvt_1db88642e346389874251b5a1eded6e3=1727099939,1727251802,1727488707,1727491020; HMACCOUNT=1628106D67895387; acw_tc=2760828017275244258732552e9880f861be6db0c40facbdd5f223490decc2; acw_sc__v2=66f7ee8290dc3f63112948801ef331b8c97ccb35; Hm_lpvt_1db88642e346389874251b5a1eded6e3=1727524485; .thumbcache_f24b8bbe5a5934237bbc0eda20c1b6e7=t00N841S/BEpxTGOoJrbm0blWik12om0ew/whcq/V2DNtyEA8um7J+yzeGOli+6iP/TrvrH0YKH2kHlsmDb5EQ%3D%3D; ssxmod_itna=YqAOBKYve+x0ODfxBcDB4DKM7RtAA4454DkDIEC+GAqGNK3DZDiqAPGhDC8RzeL4Ko7+e2aeQvSeopd5pYDk0Ge5TB0PIjaIDB3DEx06TCCQxiiSDCeDIDWeDiDG4Gm4qGtDpxG=DjDytZ9TtDm4GWGqDmDGYBWqDgDYQDGwIXD7QDIqtW07tQDDNQKpAKDiYeHlL5uMRMtOrK7DtDjdTD/3+kZCbCcPwVFeFO=nPcDB6wxBjZRq00Un+g4mNqLYf4IDAxQuYKtgheYYoAfDhTKhhLSGx4tGY4+GDKSiMS2DDAIvdKeD; ssxmod_itna2=YqAOBKYve+x0ODfxBcDB4DKM7RtAA4454DkDIEC+GDA69mqD/YttDFhxMltFKApT7bCMH+bytp7GQQyCbBWukeAF3+jxeN2wLjbQAFcdwbeNeoohjQ4qtEnLg4TKyX2LqVL=CGaV=GqZZqbDby8DFnFbgWiZEH8zoBzBCbzqWeqwoPq7/TpL=gIRY8sVhEpnWWHQU8sz+S8=+8oRerHq0wtQnSzyQDK7Dmvr3a+VY7fezRmxKF=bDeqexYUbuuzw2eR3In9evW6tzHbQY6vp=AIOUc9l6vc0vOl9plD6D07zGGQ41uxpii2Y5s7KvYQDWGYExD7=DYKKeD==";
    QByteArray cookieByte = m_cookie.toUtf8();
    QList<QNetworkCookie> list;
    list.push_back(QNetworkCookie(cookieByte));

    QVariant var;
    var.setValue(list);
    // 设置要访问的网址
    m_request.setUrl(QUrl("https://xueqiu.com/hq#hot"));
    // 设置请求头，用户代理，用来模拟浏览器
    m_request.setHeader(QNetworkRequest::UserAgentHeader,userAgent);
    // 设置cookie
    //m_request.setHeader(QNetworkRequest::CookieHeader, var);

    // 查看manager都支持哪些协议
    qDebug()<<"支持的协议:"<<m_manager.supportedSchemes();
    // 关联信号槽，当请求管理对象完全打开网页之后，会将数据给槽函数处理
    connect(&m_manager,SIGNAL(finished(QNetworkReply*)),this,SLOT(showAplyData(QNetworkReply*)));

    qDebug()<<"开始请求";
    m_manager.get(m_request);
    qDebug()<<"请求中...";

}

ETFStockInfo::~ETFStockInfo()
{
    delete ui;
}

void ETFStockInfo::getEtfInfo(QByteArray &buffer)
{
    m_model_market->setRowCount(0);
//    qDebug()<<"ETF:"<<buffer;
    // 使用JSON进行数据处理
    QJsonDocument jd = QJsonDocument::fromJson(buffer);
    if(jd.isArray()){
        qDebug()<<"js array";

    }else if(jd.isObject()){
        qDebug()<<"js object";
        QJsonObject jobject = jd.object();
        QStringList jlist = jobject.keys();
        qDebug()<<"jlist:"<<jlist; // 输出一下一级键的情况
        // 通过键值对的方式提取数据
        int count = jobject["data"].toObject()["count"].toInt();
        qDebug()<<"count:"<<count;
        QJsonArray listData = jobject["data"].toObject()["list"].toArray();
        count = listData.size(); // 适应新版的处理
        for(int i=0;i<count;++i){
            QString symbol = listData.at(i).toObject().value("symbol").toString(); // 股票代号
            double amount = listData.at(i).toObject().value("amount").toDouble()/100000000; // 成交额 单位 亿
            double chg = listData.at(i).toObject().value("chg").toDouble(); // 涨跌额
            double acc_unit_nav = listData.at(i).toObject().value("acc_unit_nav").toDouble(); // 累计净值
            QString etf_types; // = listData.at(i).toObject().value("etf_types").toString();  //etf类型
            QJsonArray type_arr = listData.at(i).toObject().value("etf_types").toArray();
            for(int i=0;i<type_arr.size();++i){
                etf_types += type_arr.at(i).toString()+" ";
            }

            double percent = listData.at(i).toObject().value("percent").toDouble(); // 涨跌幅 百分比，大于0上涨，小于0下跌
            int volume = listData.at(i).toObject().value("volume").toInt(); // 成交量 单位 手
            double current = listData.at(i).toObject().value("current").toDouble(); // 收盘价
            double current_year_percent = listData.at(i).toObject().value("current_year_percent").toDouble(); // 年涨跌幅
            double followers = listData.at(i).toObject().value("followers").toInt(); // 粉丝数量
            QString name = listData.at(i).toObject().value("name").toString();  // 股票名字
            double market_capital = listData.at(i).toObject().value("market_capital").toDouble()/100000000; // 总市值 单位 亿
            double premium_rate = listData.at(i).toObject().value("premium_rate").toDouble(); // 手续费率
            double lot_size = listData.at(i).toObject().value("lot_size").toDouble(); // 单批-交易量
            double unit_nav = listData.at(i).toObject().value("unit_nav").toDouble(); // 单位净值
            int total_shares = listData.at(i).toObject().value("total_shares").toDouble(); // 单位净值
            double timestamp = listData.at(i).toObject().value("timestamp").toDouble();  // 时间戳


            QList<QStandardItem*> rowItems;
            rowItems.append(new QStandardItem(symbol));
            rowItems.append(new QStandardItem(name));
            rowItems.append(new QStandardItem(QString::number(amount)));
            rowItems.append(new QStandardItem(QString::number(chg)));
            rowItems.append(new QStandardItem(QString::number(acc_unit_nav)));
            rowItems.append(new QStandardItem(etf_types));

            QStandardItem *percentItem =  new QStandardItem(QString::number(percent));
            if(percent > 0){
                percentItem->setData(QColor("red"),Qt::DecorationRole); // 添加一个装饰的颜色为红色
                percentItem->setData(QColor("red"),Qt::TextColorRole);  // 将字体颜色设置为红色
                rowItems.at(0)->setData(QColor("red"),Qt::TextColorRole);  // 将股票名字设置为红色
             }
            else if(percent < 0){
                percentItem->setData(QColor("green"),Qt::BackgroundColorRole);
                rowItems.at(0)->setData(QColor("green"),Qt::TextColorRole);
            }
            rowItems.append(percentItem);

            rowItems.append(new QStandardItem(QString::number(volume)));
            rowItems.append(new QStandardItem(QString::number(current)));
            rowItems.append(new QStandardItem(QString::number(current_year_percent)));
            rowItems.append(new QStandardItem(QString::number(followers)));

            rowItems.append(new QStandardItem(QString::number(market_capital,'f',2)));
            rowItems.append(new QStandardItem(QString::number(premium_rate,'f',2)));
            rowItems.append(new QStandardItem(QString::number(lot_size)));
            rowItems.append(new QStandardItem(QString::number(unit_nav)));
            rowItems.append(new QStandardItem(QString::number(total_shares)));
            rowItems.append(new QStandardItem(QString::number(timestamp)));

            // 添加一行数据项到模型中
            //m_model->appendRow(rowItems);
            m_model_market->appendRow(rowItems);
        }
    }

}

/* 响应情况
{"data":{"chg_date":"2025中报","fund_items":[{"to_float_shares_ratio":32.27,"org_name_or_fund_name":"全部合计","held_num":214204071},
{"to_float_shares_ratio":1.7792,"org_name_or_fund_name":"华泰柏瑞沪深300交易型开放式指数证券投资基金","held_num":11809096},
{"to_float_shares_ratio":1.6554,"org_name_or_fund_name":"华夏国证半导体芯片交易型开放式指数证券投资基金","held_num":10987221},
*/
void ETFStockInfo::getStockFundPosition(QByteArray &buffer,QString url)
{
    int index = url.indexOf(QRegularExpression("[A-Z]{2}\\d{6}"));
    qDebug()<<"index:"<<index<<url.mid(index,8);
    QString symbol = url.mid(index,8);
    QString name=" ";
    if(m_codeNames.size() > 0){
        name = m_codeNames[symbol];
    }

    //    m_model_position->setRowCount(0);
    //    qDebug()<<"ETF:"<<buffer;
    // 使用JSON进行数据处理
    QJsonDocument jd = QJsonDocument::fromJson(buffer);
    if(jd.isArray()){
        qDebug()<<"js array";

    }else if(jd.isObject()){
        qDebug()<<"js object";
        QJsonObject jobject = jd.object();
        QStringList jlist = jobject.keys();
        qDebug()<<"jlist:"<<jlist; // 输出一下一级键的情况
        // 通过键值对的方式提取数据
        QString chg_date = jobject["data"].toObject()["chg_date"].toString(); // 修改时间 按季度来显示

        QJsonArray fund_items = jobject["data"].toObject()["fund_items"].toArray();
        int count = fund_items.size(); // 适应新版的处理
        qDebug()<<"基金数量 count:"<<count;
        for(int i=0;i<1;++i){
            double to_float_shares_ratio = fund_items.at(i).toObject().value("to_float_shares_ratio").toDouble(); // 流通股比例
            QString org_name_or_fund_name = fund_items.at(i).toObject().value("org_name_or_fund_name").toString(); // 机构名
            double held_num = fund_items.at(i).toObject().value("held_num").toInt()/100000000.0; // 持股数 [按亿算]

            qDebug()<<to_float_shares_ratio<<org_name_or_fund_name<<held_num<<chg_date<<count;

            QList<QStandardItem*> items;
            items.append(new QStandardItem(name));
            items.append(new QStandardItem(symbol));
            items.append(new QStandardItem(org_name_or_fund_name));
            items.append(new QStandardItem(QString::number(held_num)));
            items.append(new QStandardItem(QString::number(to_float_shares_ratio)));
            items.append(new QStandardItem(chg_date));
            items.append(new QStandardItem(QString::number(count)));

            m_model_position->appendRow(items);
        }
    }
}

void ETFStockInfo::showAplyData(QNetworkReply *reply)
{
    qDebug()<<"收到响应etf";
    // 将响应的数据，一把读取完，放到字节数组里面来处理
    QByteArray buffer = reply->readAll();


    // 下面使用JSON进行数据处理
    if(reply->url() == QUrl("https://xueqiu.com/")){
        qDebug()<<"发现首页url";
        qDebug()<<buffer;

    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/chart/kline.json") != -1){
        qDebug()<<"查看个股情况:";
//        getStockOne(buffer);

    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/screener/quote/list.json?page=1&size=100&order=desc&order_by=percent&market=CN&ind_code") != -1){
        qDebug()<<"根据行业获取股票信息"<<reply->url();
        QString industryCode = reply->url().toString().split("=").back();
        qDebug()<<"行业信息:"<<industryCode;
//        getStockByIndustry(buffer,industryCode);

    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/screener/quote/list.json") != -1){
        qDebug()<<"获取所有沪深股票 可以干活了";
//        getStockAllCode(buffer);
    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/screener/industries.json") != -1){
        qDebug()<<"获取行业信息";
//        getIndustryInfo(buffer);
    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/screener/fund/list.json") != -1){
        qDebug()<<"获取到ETF基金信息";
        getEtfInfo(buffer);
    }else if(reply->url().toString().indexOf("https://stock.xueqiu.com/v5/stock/f10/cn/org_holding/detail.json") != -1){
        qDebug()<<"获取到股票基金持仓信息";
        qDebug()<<"基金持仓 url:"<<reply->url().toString();
        getStockFundPosition(buffer,reply->url().toString());
    }

    reply->deleteLater();
}

void ETFStockInfo::on_pushButton_market_clicked()
{
    QString url = "https://stock.xueqiu.com/v5/stock/screener/fund/list.json?page=1&size=3000&order=desc&order_by=percent&type=18&parent_type=1";

    m_request.setUrl(QUrl(url));
    m_manager.get(m_request);
}

// 查询基金持仓情况 按总量来查询
void ETFStockInfo::on_pushButton_position_search_clicked()
{
    QString type = ui->comboBox_type->currentText();
    QString count_rank = ui->comboBox_rank->currentText(); // 前N名
    QString url = "https://stock.xueqiu.com/v5/stock/f10/cn/org_holding/detail.json?symbol=SH603986&timestamp=1759161600000&extend=true";

    // 拿出前N名，进行请求，然后获取对应股票的基金持仓情况
//    m_request.setUrl(QUrl(url));
//    m_manager.get(m_request);

    emit sendSymbolNums(count_rank.toInt()); // 发获取股票代号的信号

}

void ETFStockInfo::saveCodeNames(QMap<QString, QString> &cns)
{
    m_codeNames = cns; // 拷贝股票代号和名字过来
    qDebug()<<__FUNCTION__<<"收到股票代号和名字的信息";
}

void ETFStockInfo::getStockFundPosSlot(QString symbol)
{
    QStringList tts = {"1759161600000","1751212800000","1743350400000"};
    for(auto &tt:tts){
        QString timestamp = tt;
        QString url = QString("https://stock.xueqiu.com/v5/stock/f10/cn/org_holding/detail.json?symbol=%1&timestamp=%2&extend=true").arg(symbol).arg(timestamp);
        m_request.setUrl(QUrl(url));
        m_manager.get(m_request);
    }

}