#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QDebug>
#include <QMessageBox>
#include <QTimer>
#include <QDir>
#include <QDateTime>
#include <QTextCodec>
#include <QInputDialog>
#include <QLineEdit>
#include <QDir>
#include <QDesktopServices>

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

 connect(ui->treeWidget,SIGNAL(itemClicked(QTreeWidgetItem*,int)), this,SLOT(connectionClicked(QTreeWidgetItem* ,int)));

 qTreeWidgetItemTcpClient=ui->treeWidget->topLevelItem(0);
 qTreeWidgetItemTcpServer=ui->treeWidget->topLevelItem(1);
 qTreeWidgetItemWsClient=ui->treeWidget->topLevelItem(2);
 qTreeWidgetItemWsServer=ui->treeWidget->topLevelItem(3);

 QVariant language = MainWindow::qSettings->value("language");
 if (!language.isNull()) {
 QString languageName = language.toString();
 if (languageName == "chs") {
 ui->actionLanguageChinese->setChecked(true);
 } else if (languageName == "en") {
 ui->actionLanguageEnglish->setChecked(true);
 }
 }
}

MainWindow::~MainWindow()
{
 delete ui;
// delete qSettings;
}

void MainWindow::on_createTcpClient_triggered()
{
 hideAllConnectionWidget();
 QTreeWidgetItem *qTreeWidgetItem1=new QTreeWidgetItem();
 qTreeWidgetItem1->setText(0,tr("新的客户端"));
 qTreeWidgetItemTcpClient->insertChild(0,qTreeWidgetItem1);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem1);
 long key=reinterpret_cast<long>(qTreeWidgetItem1);
// qDebug()<<qTreeWidgetItem1;
 TcpClient *client=new TcpClient(qTreeWidgetItem1,ui->gridLayout_2);
 ui->treeWidget->expandAll();
 tcpClientList.insert(key,client);
 ui->deleteConnection->setEnabled(true);
// qDebug()<<client;
}

void MainWindow::on_createWsClient_triggered()
{
 hideAllConnectionWidget();
 QTreeWidgetItem *qTreeWidgetItem1=new QTreeWidgetItem();
 qTreeWidgetItem1->setText(0, tr("新的客户端"));
 qTreeWidgetItemWsClient->insertChild(0,qTreeWidgetItem1);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem1);
 long key=reinterpret_cast<long>(qTreeWidgetItem1);
// qDebug()<<qTreeWidgetItem1;
 WsClient *client=new WsClient(qTreeWidgetItem1,ui->gridLayout_2);
 ui->treeWidget->expandAll();
 wsClientList.insert(key,client);
 ui->deleteConnection->setEnabled(true);
// qDebug()<<client;
}

void MainWindow::hideAllConnectionWidget(){
 for (int i=0;i<ui->gridLayout_2->layout()->count();i++)
 {
 QLayoutItem *item = ui->gridLayout_2->layout()->itemAt(i);
 item->widget()->hide();
 }
}

void MainWindow::on_createTcpServer_triggered()
{
 createTcpServerDialog = new CreateServerDialog(this);
 connect(createTcpServerDialog, SIGNAL(confirm(QString, quint16)), this, SLOT(dialog_createTcpServer_confirm(QString, quint16)));
 createTcpServerDialog->exec();
}

void MainWindow::dialog_createTcpServer_confirm(QString ip, quint16 port)
{
 hideAllConnectionWidget();
 QTreeWidgetItem *qTreeWidgetItem1=new QTreeWidgetItem();
 qTreeWidgetItem1->setText(0,ip + ":" + QString::number(port));
 TcpServer *server=new TcpServer(qTreeWidgetItem1, ui->gridLayout_2, ip, port);
 if(!server->start()){
 delete qTreeWidgetItem1;
 qTreeWidgetItem1=nullptr;
 return;
 }
 createTcpServerDialog->close();
 qTreeWidgetItemTcpServer->addChild(qTreeWidgetItem1);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem1);
 long key=reinterpret_cast<long>(qTreeWidgetItem1);
 ui->treeWidget->expandAll();
 tcpServerList.insert(key,server);
 ui->deleteConnection->setEnabled(true);
}

void MainWindow::on_createWsServer_triggered()
{
 createWsServerDialog = new CreateServerDialog(this);
 connect(createWsServerDialog, SIGNAL(confirm(QString, quint16)), this, SLOT(dialog_createWsServer_confirm(QString, quint16)));
 createWsServerDialog->exec();
}

void MainWindow::dialog_createWsServer_confirm(QString ip, quint16 port)
{
 hideAllConnectionWidget();
 QTreeWidgetItem *qTreeWidgetItem1=new QTreeWidgetItem();
 qTreeWidgetItem1->setText(0,ip + ":" + QString::number(port));
 WsServer *server=new WsServer(qTreeWidgetItem1, ui->gridLayout_2, ip, port);
 if(!server->start()){
 delete qTreeWidgetItem1;
 qTreeWidgetItem1=nullptr;
 return;
 }
 createWsServerDialog->close();
 qTreeWidgetItemWsServer->addChild(qTreeWidgetItem1);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem1);
 long key=reinterpret_cast<long>(qTreeWidgetItem1);
 ui->treeWidget->expandAll();
 wsServerList.insert(key,server);
 ui->deleteConnection->setEnabled(true);
}

void MainWindow::connectionClicked(QTreeWidgetItem *qTreeWidgetItem, int){
 bool deleteButtonEnable=false;
 if(qTreeWidgetItem->parent()!=nullptr){
 if(qTreeWidgetItem->parent()==qTreeWidgetItemTcpClient){ //客户端
 deleteButtonEnable=true;
 hideAllConnectionWidget();
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 TcpClient *client = tcpClientList.value(key);
 client->qWidget->show();
 }
 if(qTreeWidgetItem->parent()==qTreeWidgetItemWsClient){ //客户端
 deleteButtonEnable=true;
 hideAllConnectionWidget();
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 WsClient *client = wsClientList.value(key);
 client->qWidget->show();
 }
 else if(qTreeWidgetItem->parent()==qTreeWidgetItemTcpServer){ //服务端
 deleteButtonEnable=true;
 }
 else if(qTreeWidgetItem->parent()==qTreeWidgetItemWsServer){ //服务端
 deleteButtonEnable=true;
 }
 else if(qTreeWidgetItem->parent()->parent()==qTreeWidgetItemTcpServer){ //服务端的连接
 deleteButtonEnable=true;
 hideAllConnectionWidget();
 long serverKey=reinterpret_cast<long>(qTreeWidgetItem->parent());
 long serverConnectionKey=reinterpret_cast<long>(qTreeWidgetItem);
 TcpServer *server = tcpServerList.value(serverKey);
 server->serverConnectionList.value(serverConnectionKey)->qWidget->show();
 }
 else if(qTreeWidgetItem->parent()->parent()==qTreeWidgetItemWsServer){ //服务端的连接
 deleteButtonEnable=true;
 hideAllConnectionWidget();
 long serverKey=reinterpret_cast<long>(qTreeWidgetItem->parent());
 long serverConnectionKey=reinterpret_cast<long>(qTreeWidgetItem);
 WsServer *server = wsServerList.value(serverKey);
 server->serverConnectionList.value(serverConnectionKey)->qWidget->show();
 }
 }
 if(deleteButtonEnable){
 ui->deleteConnection->setEnabled(true);
 }
 else{
 ui->deleteConnection->setEnabled(false);
 }
}

void MainWindow::on_deleteConnection_triggered()
{
 QTreeWidgetItem *qTreeWidgetItem=ui->treeWidget->currentItem();
 QTreeWidgetItem *parent=qTreeWidgetItem->parent();
 if(qTreeWidgetItem->parent()==qTreeWidgetItemTcpClient){
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 TcpClient *client = tcpClientList.value(key);
 tcpClientList.remove(key);
 if(client!=nullptr){
 delete client;
 client=nullptr;
 }
 int p=parent->indexOfChild(qTreeWidgetItem);
 if(parent->childCount()>0 && p!=parent->childCount()-1){
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(parent->child(p));
 long key=reinterpret_cast<long>(parent->child(p));
 TcpClient *client = tcpClientList.value(key);
 client->qWidget->show();
 }
 else{
 ui->deleteConnection->setEnabled(false);
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem->parent());
 }
 }
 else if(qTreeWidgetItem->parent()==qTreeWidgetItemWsClient){
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 WsClient *client = wsClientList.value(key);
 wsClientList.remove(key);
 if(client!=nullptr){
 delete client;
 client=nullptr;
 }
 int p=parent->indexOfChild(qTreeWidgetItem);
 if(parent->childCount()>0 && p!=parent->childCount()-1){
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(parent->child(p));
 long key=reinterpret_cast<long>(parent->child(p));
 WsClient *client = wsClientList.value(key);
 client->qWidget->show();
 }
 else{
 ui->deleteConnection->setEnabled(false);
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem->parent());
 }
 }
 else if(qTreeWidgetItem->parent()==qTreeWidgetItemTcpServer){
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 TcpServer *server = tcpServerList.value(key);
 tcpServerList.remove(key);
 if(server!=nullptr){
 delete server;
 server=nullptr;
 }
 int p=parent->indexOfChild(qTreeWidgetItem);
 if(parent->childCount()>0 && p!=parent->childCount()-1){
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(parent->child(p));
 }
 else{
 ui->deleteConnection->setEnabled(false);
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem->parent());
 }
 }
 else if(qTreeWidgetItem->parent()==qTreeWidgetItemWsServer){
 long key=reinterpret_cast<long>(qTreeWidgetItem);
 WsServer *server = wsServerList.value(key);
 wsServerList.remove(key);
 if(server!=nullptr){
 delete server;
 server=nullptr;
 }
 int p=parent->indexOfChild(qTreeWidgetItem);
 if(parent->childCount()>0 && p!=parent->childCount()-1){
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(parent->child(p));
 }
 else{
 ui->deleteConnection->setEnabled(false);
 parent->removeChild(qTreeWidgetItem);
 ui->treeWidget->setCurrentItem(qTreeWidgetItem->parent());
 }
 }
 else if(qTreeWidgetItem->parent()->parent()==qTreeWidgetItemTcpServer){
 long parentKey=reinterpret_cast<long>(qTreeWidgetItem->parent());
 long childKey=reinterpret_cast<long>(qTreeWidgetItem);
 int p=parent->indexOfChild(qTreeWidgetItem);
 TcpServer *server = tcpServerList.value(parentKey);
 server->deleteServerConnection(childKey);
 if(parent->childCount()>0){
 ui->treeWidget->setCurrentItem(parent->child(p));
 }
 else{
 ui->deleteConnection->setEnabled(false);
 ui->treeWidget->setCurrentItem(parent);
 }
 }
 else if(qTreeWidgetItem->parent()->parent()==qTreeWidgetItemWsServer){
 long parentKey=reinterpret_cast<long>(qTreeWidgetItem->parent());
 long childKey=reinterpret_cast<long>(qTreeWidgetItem);
 int p=parent->indexOfChild(qTreeWidgetItem);
 WsServer *server = wsServerList.value(parentKey);
 server->deleteServerConnection(childKey);
 if(parent->childCount()>0){
 ui->treeWidget->setCurrentItem(parent->child(p));
 }
 else{
 ui->deleteConnection->setEnabled(false);
 ui->treeWidget->setCurrentItem(parent);
 }
 }

// QGridLayout *qGridLayout=reinterpret_cast<QGridLayout *>(test);
// qDebug()<<qGridLayout->layout()->count();

}

void MainWindow::on_actionLanguageChinese_triggered()
{
 qSettings->setValue("language", "chs");
 uncheckedAllActionLanguage();
 ui->actionLanguageChinese->setChecked(true);
 QMessageBox::information(this,tr("提示"),tr("重启应用后生效"));
}

void MainWindow::on_actionLanguageEnglish_triggered()
{
 qSettings->setValue("language", "en");
 uncheckedAllActionLanguage();
 ui->actionLanguageEnglish->setChecked(true);
 QMessageBox::information(this,tr("提示"),tr("重启应用后生效"));
}

void MainWindow::uncheckedAllActionLanguage() {
 ui->actionLanguageEnglish->setChecked(false);
 ui->actionLanguageChinese->setChecked(false);
}

void MainWindow::on_actionFeedback_triggered()
{
 QDesktopServices::openUrl(QUrl(QLatin1String("https://github.com/magicdam/MagicSocketDebugger/issues")));
}