#include "mainwindow.h"
#include <dirent.h>
#include <stdio.h>
#include <unistd.h>
#include <DWidgetUtil> //加入此头文件方可使用moveToCenter
#include <DSearchEdit>
#include <DTitlebar>
#include <QDebug>
#include <QMenu>
#include <QAction>
#include <DLineEdit>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <DPushButton>
#include <DFileDialog>
#include <iostream>
#include <QByteArray>
#include <QCheckBox>
#include <QMessageBox>
#include <QFrame>
#include <QTabWidget>
#include <QRadioButton>
#include <QButtonGroup>
#include <QGridLayout>
#include <QSlider>
#include <QGroupBox>
#include <QTextBrowser>
#include <fstream>
#include <QProcess>
#include <QTcpSocket>

#define MAXPATH 1000

MainWindow::MainWindow(DMainWindow *parent)
    : DMainWindow(parent)
{
    // 获取程序目录
    getcwd(programPath, 100);
    std::clog << "Path: " << programPath << std::endl;
    //初始化主窗口 Initialize the main window
    setCentralWidget(w);//将w作为窗口的用户部分（整个窗口除了标题栏的部分）
    moveToCenter(this); //把窗口移动到屏幕中间
    w->setWindowTitle(tr("远程桌面连接器"));
    this->setWindowFlags(windowFlags() &~ Qt::WindowMaximizeButtonHint);
    MainWindow::resize(500, 100); //改变窗口大小应当改变MainWindow的大小

    ip = new DLineEdit;
    ip->setParent(w);
    ip->setToolTip(tr("请在这里输入IP地址"));
    ip->setPlaceholderText(tr("请在这里输入IP地址"));

    ip->show();

    DPushButton *connectButton = new DPushButton;
    connectButton->setText(tr("连接"));
    connectButton->setParent(w);
    connect(connectButton, &DPushButton::clicked, this, &MainWindow::ConnectIp);
    connectButton->show();

    moreSetting = new QCheckBox(tr("更多选项"));
    moreSetting->setParent(w);
    moreSetting->setCheckable(true);
    moreSetting->show();
    connect(moreSetting, &QCheckBox::stateChanged, this, &MainWindow::ShowMoreSetting);

    QVBoxLayout *moreSettingLayout = new QVBoxLayout;
    moreSettingLayout->setParent(moreSettingFrame);

    moreSettingFrame = new QFrame;
    moreSettingFrame->setParent(w);
    moreSettingFrame->setEnabled(false);
    moreSettingFrame->setLayout(moreSettingLayout);

    QHBoxLayout *connectChooserLayout = new QHBoxLayout();
    connectChooser = new QComboBox();
    connectChooser->addItems(QStringList() << "RDP" << "VNC");
    connect(connectChooser, &QComboBox::currentTextChanged, this, [this](){
        moreSettingTab->setDisabled(connectChooser->currentIndex());
    });
    connectChooserLayout->addWidget(new DLabel(tr("远程协议：")));
    connectChooserLayout->addWidget(connectChooser);
    moreSettingLayout->addLayout(connectChooserLayout);

    QWidget *showTab = new QWidget;
    showTab->setParent(moreSettingFrame);
    showTab->show();
    QWidget *userPasswordTab = new QWidget();
    userPasswordTab->setParent(moreSettingFrame);
    userPasswordTab->show();

    user = new DLineEdit();
    user->setParent(userPasswordTab);
    user->setPlaceholderText("在这里输入用户名（不输入则以当前电脑登录用户的用户名）");
    user->show();

    password = new DPasswordEdit();
    password->show();

    showScreen = new QButtonGroup();
    QRadioButton *fullScreen = new QRadioButton("全屏访问（按Ctrl+Alt+Enter退出全屏）");
    fullScreen->show();
    fullScreen->setParent(showTab);
    QRadioButton *sizeScreen = new QRadioButton("指定分辨率访问");
    sizeScreen->setChecked(true);
    sizeScreen->setParent(showTab);
    sizeScreen->show();

    showScreen->addButton(fullScreen, 0);
    showScreen->addButton(sizeScreen, 1);
    connect(moreSetting, &QCheckBox::stateChanged, this, &MainWindow::ShowMoreSetting);
    connect(fullScreen, &QRadioButton::clicked, this, &MainWindow::ShowScreenEnabled);
    connect(sizeScreen, &QRadioButton::clicked, this, &MainWindow::ShowScreenEnabled);


    sizeScreenWidth = new DLineEdit();
    sizeScreenWidth->setParent(showTab);
    sizeScreenWidth->setText("800");
    sizeScreenWidth->show();
    sizeScreenHeight = new DLineEdit();
    DLabel *cheng = new DLabel(tr("×"));
    cheng->setParent(showTab);
    cheng->show();
    sizeScreenHeight->setParent(showTab);
    sizeScreenHeight->setText("600");
    sizeScreenWidth->show();

    DLabel *showTitleTips = new DLabel("标题：");
    showTitleTips->setParent(showTab);
    showTitleTips->show();

    showTitle = new DLineEdit();
    showTitle->setParent(showTab);
    showTitle->show();

    DLabel *colorTips = new DLabel(tr("色彩位数："));
    colorTips->setParent(showTab);
    colorTips->show();

    QHBoxLayout *colorLayout = new QHBoxLayout();

    color = new DSlider();
    color->setMaximum(3);
    color->setMinimum(0);
    color->setValue(3);
    color->setParent(showTab);
    color->show();
    connect(color, &DSlider::valueChanged, this, [&](){
        QString number = QString::number(MainWindow::color->value() * 8 + 8);
        number.append("位");
        colorShow->setText(number);
    });

    colorShow = new DLabel("32位");
    colorShow->setParent(showTab);
    colorShow->show();

    colorLayout->addWidget(color);
    colorLayout->addWidget(colorShow);

    QGridLayout *showTabLayout = new QGridLayout;
    showTabLayout->setParent(showTab);
    showTabLayout->addWidget(fullScreen, 0, 0, 1, 4);
    showTabLayout->addWidget(sizeScreen, 1, 0);
    showTabLayout->addWidget(sizeScreenWidth, 1, 1);
    showTabLayout->addWidget(cheng, 1, 2);
    showTabLayout->addWidget(sizeScreenHeight, 1, 3);
    showTabLayout->addWidget(showTitleTips, 2, 0);
    showTabLayout->addWidget(showTitle, 2, 1, 1, 3);
    showTabLayout->addWidget(colorTips, 3, 0);
    showTabLayout->addLayout(colorLayout, 3, 1, 1, 3);
    showTab->setLayout(showTabLayout);

    QGridLayout *userPasswordLayout = new QGridLayout();
    userPasswordLayout->addWidget(new DLabel(tr("用户名：")), 0, 0);
    userPasswordLayout->addWidget(user, 0, 1);
    userPasswordLayout->addWidget(new DLabel(tr("密码：")), 1, 0);
    userPasswordLayout->addWidget(password, 1, 1);
    userPasswordTab->setLayout(userPasswordLayout);
    QVBoxLayout *connectLayout = new QVBoxLayout();
    QVBoxLayout *rdpVersionLayout = new QVBoxLayout();
    QVBoxLayout *rdesktopVersionLayout = new QVBoxLayout();
    QVBoxLayout *rdesktopConnectLayout = new QVBoxLayout();

    QWidget *connectTab = new QWidget();
    connectTab->setParent(moreSettingFrame);
    connectTab->setLayout(connectLayout);
    connectTab->show();

    QGroupBox *rdpVersionBox = new QGroupBox();
    rdpVersionBox->setParent(connectTab);
    rdpVersionBox->show();
    rdpVersionBox->setTitle("rdp 版本");
    rdpVersionBox->setLayout(rdpVersionLayout);
    connectLayout->addWidget(rdpVersionBox);

    QRadioButton *rdp4 = new QRadioButton();
    rdp4->setParent(rdpVersionBox);
    rdp4->setText("设置为rdp4协议连接");
    rdp4->show();

    QRadioButton *rdp5 = new QRadioButton();
    rdp5->setParent(rdpVersionBox);
    rdp5->setText("设置为rdp5协议连接（推荐）");
    rdp5->setChecked(true);
    rdp5->show();

    rdpVersion = new QButtonGroup();
    rdpVersion->addButton(rdp4, 0);
    rdpVersion->addButton(rdp5, 1);

    rdpVersionLayout->addWidget(rdp4);
    rdpVersionLayout->addWidget(rdp5);

    QGroupBox *rdesktopVersionBox = new QGroupBox();
    rdesktopVersionBox->setParent(connectTab);
    rdesktopVersionBox->show();
    rdesktopVersionBox->setTitle("rdesktop 版本");
    rdesktopVersionBox->setLayout(rdesktopVersionLayout);
    connectLayout->addWidget(rdesktopVersionBox);

    QRadioButton *systemRdesktop = new QRadioButton();
    systemRdesktop->setParent(rdesktopVersionBox);
    systemRdesktop->setText("设置为系统版本的rdesktop（可能会不支持连接Windows XP）");
    systemRdesktop->setChecked(true);
    systemRdesktop->show();

    QRadioButton *rdesktop190 = new QRadioButton();
    rdesktop190->setParent(rdesktopVersionBox);
    rdesktop190->setText("设置为rdesktop 1.9.0（测试支持连接Windows XP，只支持 amd64）");
    rdesktop190->setEnabled(GetRunCommand("arch").replace(" ", "").replace("\n", "") == QString("x86_64"));
    //GetRunCommand("arch");
    rdesktop190->show();


    rdesktopVersion = new QButtonGroup();
    rdesktopVersion->addButton(systemRdesktop, 0);
    rdesktopVersion->addButton(rdesktop190, 1);

    rdesktopVersionLayout->addWidget(systemRdesktop);
    rdesktopVersionLayout->addWidget(rdesktop190);

    QGroupBox *rdesktopConnect = new QGroupBox();
    rdesktopConnect->setParent(connectTab);
    rdesktopConnect->show();
    rdesktopConnect->setTitle("连接选项");
    rdesktopConnect->setLayout(rdesktopConnectLayout);
    connectLayout->addWidget(rdesktopConnect);

    paste = new QCheckBox();
    paste->setParent(rdesktopConnect);
    paste->setText("剪切板共享");
    paste->setChecked(true);
    paste->show();

    remoteSound = new QCheckBox();
    remoteSound->setParent(rdesktopConnect);
    remoteSound->setText("播放远程连接声音");
    remoteSound->setChecked(true);
    remoteSound->show();

    shareRootFile = new QCheckBox();
    shareRootFile->setParent(rdesktopConnect);
    shareRootFile->setText("共享系统根目录");
    shareRootFile->show();

    shareHomeFile = new QCheckBox();
    shareHomeFile->setParent(rdesktopConnect);
    shareHomeFile->setText("共享用户目录");
    shareHomeFile->setChecked(true);
    shareHomeFile->show();

    rdesktopConnectLayout->addWidget(paste);
    rdesktopConnectLayout->addWidget(remoteSound);
    rdesktopConnectLayout->addWidget(shareRootFile);
    rdesktopConnectLayout->addWidget(shareHomeFile);

    QWidget *highSetting = new QWidget();
    highSetting->setParent(moreSettingFrame);
    highSetting->show();

    QVBoxLayout *highSettingLayout = new QVBoxLayout();
    highSettingLayout->setParent(highSetting);
    highSetting->setLayout(highSettingLayout);

    QTextBrowser *help = new QTextBrowser();
    help->setParent(highSetting);
    QProcess rdesktopHelp;
    rdesktopHelp.start("rdesktop");
    rdesktopHelp.waitForFinished();
    help->setText(rdesktopHelp.readAllStandardError());
    help->show();
    highSettingLayout->addWidget(help);

    command = new DLineEdit();
    command->setPlaceholderText("请在这里输入需要的参数（上面是参数示例）");
    command->setParent(highSetting);
    command->show();
    highSettingLayout->addWidget(command);

    moreSettingTab = new QTabWidget;
    moreSettingTab->setParent(moreSettingFrame);
    moreSettingTab->addTab(showTab, tr("显示"));
    moreSettingTab->addTab(userPasswordTab, tr("用户"));
    moreSettingTab->addTab(connectTab, tr("连接"));
    moreSettingTab->addTab(highSetting, tr("高级"));
    moreSettingTab->show();

    moreSettingLayout->addWidget(moreSettingTab);

    QVBoxLayout *AllWidget = new QVBoxLayout;
    QHBoxLayout *firstLineWidget = new QHBoxLayout;
    firstLineWidget->addWidget(new DLabel("IP地址："));
    firstLineWidget->addWidget(ip);
    firstLineWidget->addWidget(connectButton);

    AllWidget->addLayout(firstLineWidget);
    AllWidget->setStretchFactor(firstLineWidget, 1);
    AllWidget->addWidget(moreSetting);
    AllWidget->setStretchFactor(moreSetting, 1);
    AllWidget->addWidget(moreSettingFrame);
    AllWidget->setStretchFactor(moreSettingFrame, 6);

    w->setLayout(AllWidget);

    //在标题栏上添加一个菜单/菜单项 Add a menu/menu item to the title bar
    QMenu *menu=new QMenu;
    QAction *giteeAction=new QAction("Gitee 网站");
    menu->addAction(giteeAction);
    titlebar()->setMenu(menu);
    connect(giteeAction, &QAction::triggered, this, [&](){std::system("xdg-open https://gitee.com/gfdgd-xi/simple-remote-desktop-accessor");});


    QAction *githubAction=new QAction("Github 网站");
    menu->addAction(githubAction);
    titlebar()->setMenu(menu);
    connect(githubAction, &QAction::triggered, this, [&](){std::system("xdg-open https://github.com/gfdgd-xi/simple-remote-desktop-accessor");});

    //设置标题栏上的图标 Set the icon on the title bar
    titlebar()->setIcon(QIcon::fromTheme("/opt/durapps/spark-simple-remote-desktop-accessor/icon.svg"));

    //让程序适配浅色模式和深色模式 Adapt the program to light and dark model
    //connect(DGuiApplicationHelper::instance(), &DGuiApplicationHelper::themeTypeChanged,this,&MainWindow::setTheme);

    // 判断是否为 !amd64 + !rdesktop
    if(GetRunCommand("arch").replace(" ", "").replace("\n", "") != QString("x86_64") && system("which rdesktop")){
        if(QMessageBox::question(w, "提示", "您似乎还没用安装 rdesktop，是否安装？") == QMessageBox::Yes){
            RunDTKCommand("deepin-terminal -C \"pkexec apt install rdesktop\" ");
        }
    }

}
QString MainWindow::GetRunCommand(QString command){
    QProcess process;
    process.start(command);
    process.waitForStarted();
    process.waitForFinished();
    QString re = process.readAllStandardOutput();
    process.close();
    return re;
}

// 运行 DTK应用（以修复在 Ubuntu 下运行 DTK 应用时 UI 显示问题）
QString MainWindow::RunDTKCommand(QString command){
    QProcess process;
    process.setEnvironment({{"XDG_CURRENT_DESKTOP", "Deepin"}});
    process.start(command + " -platformtheme deepin");
    process.waitForStarted();
    process.waitForFinished();
    QString re = process.readAllStandardOutput();
    process.close();
    return re;
}

void MainWindow::ShowScreenEnabled(){
    MainWindow::sizeScreenWidth->setEnabled(MainWindow::showScreen->checkedId());
    MainWindow::sizeScreenHeight->setEnabled(MainWindow::showScreen->checkedId());
}
void MainWindow::ShowMoreSetting(){
    if (MainWindow::moreSetting->checkState()){
        moreSettingFrame->setEnabled(true);
        return;
    }
    moreSettingFrame->setEnabled(false);
}
void MainWindow::ConnectIp(){
    if(MainWindow::ip->text() == ""){
        QMessageBox::information(w, tr("提示"), tr("没有输入IP地址，无法继续"));
        return;
    }
    QString ip = MainWindow::ip->text();
    int port = 0;
    bool isVNC = false;
    bool isRDP = false;
    // 如果有用户选择了高级模式 + VNC 协议，则无需分析
    if(moreSetting->isChecked() && connectChooser->currentIndex()) {
        isVNC = true;
    }
    else if(ip.contains(":")) {
        // 含有端口
        int index = ip.indexOf(":");
        port = ip.mid(index + 1).toInt();
        ip = ip.mid(0, index);
        // 如果端口号过小，则为 VNC
        if(port < 30) {
            isVNC = true;
        }
        // 如果端口号在 5900 <= port <= 5930 附近则为 VNC
        if(port >= 5900 && port <= 5930) {
            isVNC = true;
        }
        // 如果端口为 3389,则为 rdp
        if(port == 3389) {
            isRDP = true;
        }
    }
    /*if(!(isVNC + isRDP)) {
        // 判断是否为 rdp
        QTcpSocket socket;
        if(!port) {
            socket.connectToHost(ip, 3389);
            if(socket.waitForConnected()) {
                qDebug() << "Error";
            }
            qDebug() << socket.errorString();
        }
    }*/
    if(isVNC) {
        QProcess process;
        process.start("gvncviewer", QStringList() << ip);
        process.waitForStarted();
        process.waitForFinished();
        return;
    }
    QStringList option;
    QString App = "padsp";
    if (moreSetting->isChecked()){
        // 高级选项
        switch (MainWindow::rdesktopVersion->checkedId()) {
            case 0:
                option << "rdesktop";
                break;
            case 1:
                option << "/opt/durapps/spark-simple-remote-desktop-accessor/rdesktop";
                break;
        }
        option << MainWindow::ip->text() << "-T" << MainWindow::showTitle->text() << "-a" << QString("%1").arg(MainWindow::color->value() * 8 + 8);
        switch (MainWindow::showScreen->checkedId()){
            case 0:
                option << "-f";
                break;
            case 1:
                option << "-g" << MainWindow::sizeScreenWidth->text() + "x" + MainWindow::sizeScreenHeight->text();
                break;
        }
        if(MainWindow::user->text() != ""){
            option << "-u" << MainWindow::user->text();
        }
        if(MainWindow::password->text()!=""){
            option << "-p" << MainWindow::password->text();
        }
        switch (MainWindow::rdpVersion->checkedId()) {
            case 0:
                option << "-4";
                break;
            case 1:
                option << "-5";
                break;
        }
        if(MainWindow::paste->isChecked()){
            option << "-r" << "clipboard:PRIMARYCLIPBOARD";
        }
        if(MainWindow::remoteSound->isChecked()){
            option << "-r" << "sound:local";
        }
        if(MainWindow::shareRootFile->isChecked()){
            option << "-r" << "disk:ROOT=/";
        }
        if(MainWindow::shareHomeFile->isChecked()){
            option  << "-r" << "disk:HOME=" + QDir::homePath();
        }
        if(MainWindow::command->text() != ""){
            option << MainWindow::command->text().split(" ");
        }
    }
    else {
        // 默认模式
        option << "rdesktop" << MainWindow::ip->text() << "-T" << MainWindow::ip->text() << "-r" << "clipboard:PRIMARYCLIPBOARD" << "-r" << "sound:local" << "-r" << "disk:HOME=" + QDir::homePath();
    }
    w->setEnabled(false);   
    QProcess *process = new QProcess();
    process->start(App, option);
    process->write("yes\n");
    process->waitForStarted();
    process->waitForFinished();
    //process->close();
    if(process->exitCode() != 0 && process->exitCode() != 62){
        QMessageBox::critical(this, "错误", "程序返回值：" + QString("%1").arg(process->exitCode()) + "\n具体错误：\n" + process->readAllStandardError());
        qDebug() << QString::fromLocal8Bit(process->readAllStandardError());
    }
    w->setEnabled(true);
}
MainWindow::~MainWindow()
{
    delete w;
}

/*void MainWindow::setTheme(DGuiApplicationHelper::ColorType theme)
{
    //这个函数在程序运行之初就会被调用一次，而后每次更换主题颜色时被调用。 This function is called at the beginning of the program and then every time the theme color is changed.
    if(theme==DGuiApplicationHelper::LightType){ //在这个位置设置两种不同的样式 Set two different styles at this location
        //浅色模式 Light Mode
        qDebug()<<"Lighe";
    }else {
        //深色模式 Dark Mode
        qDebug()<<"Dark";
    }
}*/