#include "positionmanagement.h"
#include "ui_positionmanagement.h"
#include <QVBoxLayout>
#include <QLineEdit>
#include <QPushButton>
#include <QComboBox>
#include <QDebug>
#include <QMessageBox>
#include <QDateTime>
#include <QMessageBox>

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

    m_modelAll = new QStandardItemModel(this);
    m_modelShort = new QStandardItemModel(this);
    m_modelMid = new QStandardItemModel(this);
    m_modelLong = new QStandardItemModel(this);

    QStringList labels; // 14列
    labels<<"id"<<"股票"<<"code"<<"入池时间"<<"入池价格"<<"购买时间"<<"购买价格"
         <<"卖出时间"<<"卖出价格"<<"入池收益"<<"买入收益"<<"卖出收益"<<"类型"<<"状态"<<"当前价格";

    m_modelAll->setHorizontalHeaderLabels(labels);
    m_modelShort->setHorizontalHeaderLabels(labels);
    m_modelMid->setHorizontalHeaderLabels(labels);
    m_modelLong->setHorizontalHeaderLabels(labels);

    ui->tableView_all->setModel(m_modelAll);
    ui->tableView_short->setModel(m_modelShort);
    ui->tableView_mid->setModel(m_modelMid);
    ui->tableView_long->setModel(m_modelLong);

    // 初始化数据库
    initMySQL();

    // 添加股票信息相关
    m_addDlg = new QDialog(this);
    m_addDlg->setWindowTitle("添加股票到仓位池");
    QVBoxLayout *vb = new QVBoxLayout(m_addDlg);
    QStringList placetext;
    placetext<<"输入股票名字"<<"输入股票代号"<<"输入价格";
    for(int i=0;i<3;++i){
        QLineEdit *lt = new QLineEdit(this);
        lt->setPlaceholderText(placetext.at(i));
        vb->addWidget(lt);
        m_lineEdits.append(lt);
    }

    QComboBox *comBox = new QComboBox(this);
    comBox->addItem("短线");
    comBox->addItem("中线");
    comBox->addItem("长线");
    vb->addWidget(comBox);

    QPushButton *addBtn = new QPushButton("添加",this);
    vb->addWidget(addBtn);

    // 使用匿名函数来实现槽
    connect(addBtn,&QPushButton::clicked,this,[=]{
        QString name = m_lineEdits.at(0)->text();
        QString code = m_lineEdits.at(1)->text();
        QString price = m_lineEdits.at(2)->text();
        QString type = comBox->currentText();
        qDebug()<<name<<code<<price<<type;
        QString cur_time = QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss");

        QString sql = QString("insert into position_pool (name,code,time_in,price_in,type,cur_price) values ('%1','%2','%3',%4,'%5',%6)")
                .arg(name).arg(code).arg(cur_time).arg(price).arg(type).arg(price);
        qDebug()<<"添加股票的sql:"<<sql;

        QSqlQuery que(db);

        if(que.exec(sql)){
            qDebug()<<"insert ok";
            QMessageBox::information(m_addDlg,"添加成功",name+" "+price);

        }else{
            qDebug()<<"insert fail"<<que.lastError().text();
            QMessageBox::information(m_addDlg,"添加失败",name+" "+que.lastError().text());
        }

    });

    // 右击菜单: 准备好菜单--给目标对象设置菜单策略为自定义上下文菜单--连接信号槽
    m_menu = new QMenu(this);
    m_menu->addAction("查询最新价格");
    m_menu->addAction("买入当前股票");
    m_menu->addAction("卖出当前股票");

    // 连接菜单动作和槽
    connect(m_menu->actions().at(0),SIGNAL(triggered(bool)),this,SLOT(searchPriceAction()));
    connect(m_menu->actions().at(1),SIGNAL(triggered(bool)),this,SLOT(buyStockAction()));
    connect(m_menu->actions().at(2),SIGNAL(triggered(bool)),this,SLOT(saleStockAction()));


    this->setContextMenuPolicy(Qt::CustomContextMenu); // 设定菜单策略
    ui->tableView_all->setContextMenuPolicy(Qt::CustomContextMenu);
    ui->tableView_short->setContextMenuPolicy(Qt::CustomContextMenu);
    ui->tableView_mid->setContextMenuPolicy(Qt::CustomContextMenu);
    ui->tableView_long->setContextMenuPolicy(Qt::CustomContextMenu);

    // 和窗口连接，在整个窗口中都可以右击显示菜单
//    connect(this,&PositionManagement::customContextMenuRequested,this,[=](QPoint p){
//        //m_menu->exec(QCursor::pos()); // 方式1
//        //m_menu->exec(this->mapToGlobal(p)); // 方式2,如果是子部件，容易坐标偏移
//        m_menu->popup(QCursor::pos()); // 换一个方式3 也可以
//    });

    // 和每个视图连接，只在视图中显示菜单
    connect(ui->tableView_all,SIGNAL(customContextMenuRequested(QPoint)),this,SLOT(showMenu(QPoint)));
    connect(ui->tableView_short,SIGNAL(customContextMenuRequested(QPoint)),this,SLOT(showMenu(QPoint)));
    connect(ui->tableView_mid,SIGNAL(customContextMenuRequested(QPoint)),this,SLOT(showMenu(QPoint)));
    connect(ui->tableView_long,SIGNAL(customContextMenuRequested(QPoint)),this,SLOT(showMenu(QPoint)));



}

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

void PositionManagement::initMySQL()
{
    //添加一个数据库
    db=QSqlDatabase::addDatabase("QMYSQL");    //括号内要写出数据库的类型
    //设置数据库
    db.setHostName("127.0.0.1"); //设置数据库的主机ip
    //设置数据库的用户名
    db.setUserName("root");
    //设置数据库的密码
    db.setPassword("root");    //这个就是安装MySQL时设置的密码
    //设置数据库的名字
    db.setDatabaseName("stock_plan");
    //打开数据库（已经安装过mysql驱动了）
    if(db.open()==false){
        QMessageBox::warning(this,"waring",db.lastError().text());
    }else{
        qDebug()<<"mysql conn ok";
    }
}

void PositionManagement::on_pushButton_add_clicked()
{
    m_addDlg->show();
}

void PositionManagement::on_pushButton_refresh_clicked()
{
    m_modelAll->setRowCount(0);
    m_modelShort->setRowCount(0);
    m_modelMid->setRowCount(0);
    m_modelLong->setRowCount(0);


    QString sql = "select * from position_pool";
    QSqlQuery que(db);
    if(que.exec(sql)){
        qDebug()<<"查询成功";

        qDebug()<<"size:"<<que.size();
        while (que.next()) {
            QList<QStandardItem*> items;
            QList<QStandardItem*> items_short;
            QList<QStandardItem*> items_mid;
            QList<QStandardItem*> items_long;
            for(int i=0;i<15;++i){
                qDebug()<<i<<que.value(i).toString();
                items.append(new QStandardItem(que.value(i).toString()));
                items_short.append(new QStandardItem(que.value(i).toString()));
                items_mid.append(new QStandardItem(que.value(i).toString()));
                items_long.append(new QStandardItem(que.value(i).toString()));

            }
            m_modelAll->appendRow(items);
            if(items.at(12)->text() == "短线"){
                m_modelShort->appendRow(items_short);
            }else if(items.at(12)->text() == "中线"){
                m_modelMid->appendRow(items_mid);
            }else{
                m_modelLong->appendRow(items_long);
            }
        }

    }else{
        qDebug()<<"查询失败";
    }
}

void PositionManagement::showMenu(QPoint p)
{
    m_curPoint = p;  // 记录当前坐标--子部件内的坐标，刚好可用，不用再转换了
    m_curView = (QTableView*)sender(); // 记录被点击的视图，用来取索引--模型--数据
//    qDebug()<<sender()<<p;
//    qDebug()<<"显示右击菜单";
//    m_menu->exec(this->mapFromGlobal(p)); // 方式1,容易坐标跑偏，特别是子部件显示菜单时
    m_menu->exec(QCursor::pos());



}

void PositionManagement::searchPriceAction()
{
    qDebug()<<"查询最新价格"<<sender();

}

void PositionManagement::buyStockAction()
{
    qDebug()<<"买入当前股票";
    // 判断当前位置是否为选中的有效内容行，然后按当前价格买入即可，应该拿到最新价买卖
    // 一定要先判断状态是否为观察：观察的话--当前行买入，修改记录即可；如果是已经买入或卖出的状态，再添加一条新的买入记录

    QModelIndex index = m_curView->indexAt(m_curPoint);
    if(index.row() == -1) return; // 防止异常

    QStandardItemModel * curModel = (QStandardItemModel*)index.model();
    int row = index.row();
    QString id = curModel->item(row,0)->text();
    QString name = curModel->item(row,1)->text();
    QString code = curModel->item(row,2)->text();
    QString price = curModel->item(row,14)->text();
    QString state = curModel->item(row,13)->text();
    qDebug()<<id<<name<<code<<price<<"state:"<<state;

    if(state.toInt() == 1){
        // 更新当前即可,通过id来对比处理即可更新
        QString buy_time = QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss");
        QString sql = QString("update position_pool set time_buy='%1',price_buy=%2,state=%4 where id=%3")
                .arg(buy_time).arg(price).arg(id).arg("2");

        qDebug()<<"buy sql:"<<sql;

        QSqlQuery que(db);
        if(que.exec(sql)){
            qDebug()<<"buy 成功:"<<name<<price;
            on_pushButton_refresh_clicked(); // 刷新股票池
        }else{
            qDebug()<<"buy 失败:"<<name<<price;
        }

    }else{
        // 创建新的记录，把当前价格和时间设置为入池、买入相应的价格和时间
    }

}

void PositionManagement::saleStockAction()
{
    qDebug()<<"卖出当前股票";
    // 判断当前位置是否为选中的有效内容行，然后按当前价格卖出即可，应该拿到最新价买卖
    // 一定要先判断出目前是买入持仓的状态，否则卖出失败

    QModelIndex index = m_curView->indexAt(m_curPoint);
    if(index.row() == -1) return; // 防止异常

    QStandardItemModel * curModel = (QStandardItemModel*)index.model();
    int row = index.row();
    QString id = curModel->item(row,0)->text();
    QString name = curModel->item(row,1)->text();
    QString code = curModel->item(row,2)->text();
    QString price = curModel->item(row,14)->text();
    QString state = curModel->item(row,13)->text();
    QString buy_price = curModel->item(row,6)->text();
    qDebug()<<id<<name<<code<<price<<"state:"<<state;

    if(state.toInt() == 2){
        // 更新当前即可,通过id来对比处理即可更新
        QString sell_time = QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss");
        double rate = (price.toDouble() / buy_price.toDouble() - 1.0)*100; // 计算卖出盈利比
        QString sql = QString("update position_pool set time_sale='%1',price_sale=%2,state=%4,rate_sale=%7 where id=%3")
                .arg(sell_time).arg(price).arg(id).arg("3").arg(QString::number(rate));

        qDebug()<<"sell sql:"<<sql;

        QSqlQuery que(db);
        if(que.exec(sql)){
            qDebug()<<"sell 成功:"<<name<<price;
            on_pushButton_refresh_clicked(); // 刷新股票池
            QMessageBox::information(this,"卖出成功",name+" 盈利百分比:"+QString::number(rate));
        }else{
            qDebug()<<"sell 失败:"<<name<<price;
        }

    }else{
        qDebug()<<"不适合卖出,没有持仓";
        QMessageBox::information(this,"卖出失败","无持仓，不能卖出");
    }
}