settingsdialog.cpp

#include "settingsdialog.h"
#include "ui_settingsdialog.h"
#include "customdir.h"
#include "configsaver.h"
#include "llminterface.h"

#define ZOOM_ENABLE_REQ "(关闭鼠标缩放后可用)"

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

    this->setFixedSize(this->size());

    p = (VPetInterface *) this->parentWidget();
    orgSize = p->size();

    // 窗体设置
    this->move(150, 100);   // 设置窗体默认位置
    this->setWindowTitle("虚拟桌宠设置");
    this->setWindowIcon(QIcon(":/img/resources/icons/settings.png"));
    this->setWindowModality(Qt::WindowModal);
    this->setAttribute(Qt::WA_DeleteOnClose, true);

    ButtonBoxInit();
    GeneralSettingsInit();
    ModelSettingsInit();
    VoiceSettingsInit();
    LLMSettingsInit();
    AboutSettingsInit();
}

SettingsDialog::~SettingsDialog()
{
    ConfigSaver::writeConfig(p);
    qDebug() << QT_BACKGROUND_LOG << "Settings released";
    delete ui;
}

void SettingsDialog::ButtonBoxInit()
{
    // 设置文字
    ui->configureBox->button(QDialogButtonBox::Cancel)->setText("取消");
    ui->configureBox->button(QDialogButtonBox::Ok)->setText("确定");

    // 设置大小
    auto buttonList = ui->configureBox->buttons();
    for(auto button: buttonList)
        button->setFixedSize(72, 26);
}

void SettingsDialog::GeneralSettingsInit()
{
    isSliderZoomAvailable = !p->wheelZoomState();

    ui->checkBoxWindowTop->setChecked(p->windowOnTopState());
    ui->checkBoxWheelZoom->setChecked(p->wheelZoomState());

    ui->sliderZoom->setEnabled(isSliderZoomAvailable);
    ui->sliderZoom->setMaximum(MAX_MODEL_WIDTH);    // in width
    ui->sliderZoom->setMinimum(MIN_MODEL_WIDTH);    // in width
    ui->sliderZoom->setPageStep(5);     // in width

    // 设置添加自启动按钮状态
    if(p->isRegAutoRun())
    {
        ui->btnSetAutoRun->setDisabled(true);
        ui->labelAutoRunInfo->setText("开机自启动项已存在，请在系统启动设置中设置");
    } else {
        ui->labelAutoRunInfo->setText("将虚拟桌宠添加进开机自启动项");
        connect(ui->btnSetAutoRun, &QPushButton::clicked,
                this, &SettingsDialog::onAutoRunBtnClicked);
    }

    // 设置拖动条状态
    if(isSliderZoomAvailable)
    {
        auto size = p->size();
        QString sizeStr = QString("(%1×%2)").arg(size.width()).arg(size.height());
        ui->labelSize->setText(QString("缩放").append(sizeStr));
        ui->sliderZoom->setValue(size.width());

        // 缩放条时间槽连接:实现拖动缩放
        connect(ui->sliderZoom, &QSlider::sliderMoved, this, &SettingsDialog::onSliderChanged);
        connect(ui->sliderZoom, &QSlider::sliderReleased, this, &SettingsDialog::onSliderChanged);
    } else
        ui->labelSize->setText(QString("缩放").append(ZOOM_ENABLE_REQ));

    QStringList fpsValues;
    fpsValues << "30" << "60";
    ui->comboFPSSetting->addItems(fpsValues);
    ui->comboFPSSetting->setCurrentText(QString::number(p->fps()));

    connect(ui->checkBoxWheelZoom, &QCheckBox::stateChanged,
            this, &SettingsDialog::onZoomBoxChanged);
}

void SettingsDialog::ModelSettingsInit()
{
    fileDir = FileHandler::getModelDirList();
    ui->comboModel->addItems(fileDir);
    modelIndex = p->modelIndex();
    ui->comboModel->setCurrentIndex(modelIndex);

    connect(ui->comboModel, &QComboBox::textActivated,
            this, &SettingsDialog::onComboBoxChanged);
    connect(ui->importButton, &QPushButton::clicked,
            this, &SettingsDialog::onImportClicked);
    connect(ui->deleteButton, &QPushButton::clicked,
            this, &SettingsDialog::onDeleteModelClicked);
}

void SettingsDialog::VoiceSettingsInit()
{
    auto voiceList = *LlmInterface::getVoiceList(); // 取值
    ui->comboVoice->addItems(voiceList);

    // 初始化音量条、语速条及相关标签
    ui->volumeSlider->setMaximum(100);
    ui->volumeSlider->setMinimum(0);
    ui->volumeSlider->setPageStep(1);
    ui->volumeSlider->setValue(p->volume());
    ui->volume->setText(QString::number(p->volume()) + "%");
    ui->comboVoice->setCurrentIndex(p->voice());

    ui->paceSlider->setMaximum(100);
    ui->paceSlider->setMinimum(0);
    ui->volumeSlider->setPageStep(1);
    ui->paceSlider->setValue(p->pace());
    ui->pace->setText(QString::number(p->pace()) + "%");

    // 音量条语速条信号和槽
    connect(ui->volumeSlider, &OptimizedSlider::sliderMoved,
            this, &SettingsDialog::onVolumeChanged);
    connect(ui->volumeSlider, &OptimizedSlider::sliderReleased,
            this, &SettingsDialog::onVolumeChanged);

    connect(ui->paceSlider, &OptimizedSlider::sliderMoved,
            this, &SettingsDialog::onPaceChanged);
    connect(ui->paceSlider, &OptimizedSlider::sliderReleased,
            this, &SettingsDialog::onPaceChanged);

    connect(ui->previewButton, &QPushButton::clicked,
            this, &SettingsDialog::testVoice);

    connect(ui->voiceBindButton, &QPushButton::clicked,
            this, &SettingsDialog::onSaveToModelClicked);
}

void SettingsDialog::LLMSettingsInit()
{
    QStringList api = p->api();
    if(!api[0].isEmpty() && !api[1].isEmpty())
    {
        ui->textEditAPIKey->setText(api[0]);
        ui->textEditSecretKey->setText(api[1]);

        ui->checkBoxTTS->setDisabled(!p->LLMEnable());
        ui->checkBoxLLM->setChecked(p->LLMEnable());
        ui->checkBoxTTS->setChecked(p->TTSEnable());
    } else {
        ui->checkBoxTTS->setDisabled(true);
        ui->checkBoxLLM->setDisabled(true);
    }

    connect(ui->checkBoxLLM, &QCheckBox::stateChanged,
            this, &SettingsDialog::onLLMBoxChanged);
    connect(ui->checkBoxTTS, &QCheckBox::stateChanged,
            this, &SettingsDialog::onTTSBoxChanged);

    connect(ui->btnAPISave, &QPushButton::pressed,
            this, &SettingsDialog::onAPISaveClicked);
}

void SettingsDialog::AboutSettingsInit()
{
    connect(ui->btnWebRaw, &QPushButton::clicked, this, [=]() {
        QString url = "https://github.com/Positron114514/GunShipVPet";
        QDesktopServices::openUrl(QUrl(url.toLatin1()));
    });

    connect(ui->tabWidget, &QTabWidget::currentChanged, this, &SettingsDialog::onChangedAboutPage);

    // if(p->isRegAutoRun())
    //     ui->btnRegAutoRun->setDisabled(true);

    // connect(ui->btnRegAutoRun, &QPushButton::clicked,
    //         this, &SettingsDialog::onRegBtnClicked);
}

void SettingsDialog::accept()
{
    p->setWindowOnTopState(ui->checkBoxWindowTop->isChecked());
    p->setWheelZoomState(ui->checkBoxWheelZoom->isChecked());

    p->setModelIndex(ui->comboModel->currentIndex());

    p->setFps(ui->comboFPSSetting->currentText().toInt());

    p->setLLMEnable(ui->checkBoxLLM->isChecked());
    p->setTTSEnable(ui->checkBoxTTS->isChecked());

    p->setVoice(ui->comboVoice->currentIndex());

    int voiceIndex = FileHandler::getModelVoiceIndex();
    if(voiceIndex != -1)
    {
        qDebug() << "Find voice from model. index: " << voiceIndex;
        p->setVoice(voiceIndex);
    }

    qDebug() << QT_BACKGROUND_LOG << "settings carded";

    done(Accepted);
}

void SettingsDialog::reject()
{
    p->resizeWindow(orgSize);

    qDebug() << QT_BACKGROUND_LOG << "action rejected";

    done(Rejected);
}

void SettingsDialog::onZoomBoxChanged()
{
    bool state = ui->checkBoxWheelZoom->isChecked();

    if(!state)
    {
        auto size = p->size();
        QString sizeStr = QString("(%1×%2)").arg(size.width()).arg(size.height());
        ui->labelSize->setText(QString("缩放").append(sizeStr));

        ui->sliderZoom->setValue(p->size().width());

        // 缩放条时间槽连接:实现拖动缩放
        connect(ui->sliderZoom, &QSlider::sliderMoved, this, &SettingsDialog::onSliderChanged);
        connect(ui->sliderZoom, &QSlider::sliderReleased, this, &SettingsDialog::onSliderChanged);

        isSliderZoomAvailable = true;
    } else {
        ui->labelSize->setText(QString("缩放").append(ZOOM_ENABLE_REQ));
        isSliderZoomAvailable = false;
    }

    ui->sliderZoom->setDisabled(!isSliderZoomAvailable);
}

void SettingsDialog::onSliderChanged()
{
    int value = ui->sliderZoom->value();
    p->resizeWindow(value, value * MODEL_PROPORTION);

    auto size = p->size();
    QString sizeStr = QString("(%1×%2)").arg(size.width()).arg(size.height());
    ui->labelSize->setText(QString("缩放").append(sizeStr));
}

void SettingsDialog::onComboBoxChanged()
{
    modelIndex = ui->comboModel->currentIndex();
    qDebug() << QT_DEBUG_OUTPUT << "current index:" << modelIndex;
}

void SettingsDialog::onImportClicked()
{
    QString modelDir = QFileDialog::getOpenFileName(this,
                                                    "选择模型",
                                                    "/",
                                                    tr("模型文件(*.model3.json)"));

    QFileInfo modelInfo(modelDir);
    // qDebug() << QT_DEBUG_OUTPUT << info.path();
    FileHandler::addModel(modelInfo.path());

    ui->comboModel->clear();
    fileDir = FileHandler::getModelDirList();
    ui->comboModel->addItems(fileDir);
}

void SettingsDialog::onLLMBoxChanged()
{
    p->setLLMEnable(ui->checkBoxLLM->isChecked());
    if(!ui->checkBoxLLM->isChecked())
    {
        ui->checkBoxTTS->setDisabled(true);
        ui->checkBoxTTS->setChecked(false);
        p->setTTSEnable(false);
    } else {
        ui->checkBoxTTS->setDisabled(false);
    }
}

void SettingsDialog::onTTSBoxChanged()
{
    p->setTTSEnable(ui->checkBoxTTS->isChecked());
}

void SettingsDialog::onAutoRunBtnClicked()
{
    p->regAutoRun();
    if(p->isRegAutoRun())
    {
        QMessageBox msgBox(this);
        msgBox.setIcon(QMessageBox::Information);
        msgBox.setWindowTitle("操作成功");
        msgBox.setText("开机自启动项添加成功！\n如果需要关闭请在系统启动设置中手动关闭");
        // msgBox.button(QMessageBox::Ok)->setText("确定");
        msgBox.exec();

        ui->btnSetAutoRun->setDisabled(true);
        ui->labelAutoRunInfo->setText("开机自启动项已存在，请在系统启动设置中设置");
    } else {
        qDebug() << QT_BACKGROUND_LOG << "AUTORUN regedit creation failed! Time"
                 << autoRunErrorTime + 1;

        if(autoRunErrorTime <= 3)
        {
            QMessageBox msgBox(this);
            msgBox.setIcon(QMessageBox::Warning);
            msgBox.setWindowTitle("操作失败");
            msgBox.setText("开机自启动项添加失败！\n请重试...");
            // msgBox.button(QMessageBox::Ok)->setText("确定");
            msgBox.exec();

            qDebug() << QT_INTERFACE_LOG << "Error window exec and notified";
            autoRunErrorTime++;
        } else {
            QMessageBox msgBox(this);
            msgBox.setIcon(QMessageBox::Critical);
            msgBox.setWindowTitle("操作多次失败...");
            msgBox.setText("开机自启动项多次添加失败！\n请向我们提交log与Issues，谢谢！");
            // msgBox.button(QMessageBox::Ok)->setText("确定");
            connect(msgBox.button(QMessageBox::Ok), &QPushButton::clicked, this, [=]() {
                QString url = "https://github.com/Positron114514/GunShipVPet/issues";
                QDesktopServices::openUrl(url);
            });
            msgBox.exec();

            qDebug() << QT_INTERFACE_LOG << "Multi-error window exec";

            ui->btnSetAutoRun->setDisabled(true);
            ui->labelAutoRunInfo->setText("看起来功能出现了问题，暂时不可用...");
        }
    }
}

void SettingsDialog::onChangedAboutPage()
{
    QWidget *tab = ui->tabWidget->currentWidget();
    if(tab == ui->aboutTab)
    {
        QFont font;
        font.setFamily("MiSans");
        font.setPointSize(16);
        ui->ourName->setFont(font);
    }
}

void SettingsDialog::onAPISaveClicked()
{
    QString apiKey = ui->textEditAPIKey->text();
    QString secretKey = ui->textEditSecretKey->text();
    if(!apiKey.isEmpty() && !secretKey.isEmpty())
    {
        p->setAPI(apiKey, secretKey);
        QMessageBox msg(this);
        msg.setWindowTitle("提示");

        if(!LlmInterface::getAccessToken(p->api()[0], p->api()[1]))
        {
            qDebug() << QT_DEBUG_OUTPUT << "failed to load access token";
            msg.setText("错误：获取accessToken失败（api不可用）");
            p->setTokenState(false);
        }else
        {
            msg.setText("API设置成功");
        }
        msg.exec();
        ui->checkBoxLLM->setDisabled(false);
    } else {
        QMessageBox msg(this);
        msg.setWindowTitle("失败");
        msg.setText("请完整填写文心一言API并重试");
        msg.exec();

        QStringList cur = p->api();
        if(!cur[0].isEmpty() && !cur[1].isEmpty())
        {
            ui->textEditAPIKey->setText(cur[0]);
            ui->textEditSecretKey->setText(cur[1]);
        }
    }
}

void SettingsDialog::onDeleteModelClicked()
{
    // 弹窗初始化
    QMessageBox msgBox(this);
    msgBox.setWindowTitle("删除");
    QString modelName = ui->comboModel->currentText();
    msgBox.setText("确定要删除模型" + modelName + "？");
    msgBox.setIcon(QMessageBox::Question);
    QPushButton *btnAccept = new QPushButton("确认");
    QPushButton *btnReject = new QPushButton("取消");
    msgBox.addButton(btnAccept, QMessageBox::AcceptRole);
    msgBox.addButton(btnReject, QMessageBox::RejectRole);

    // qDebug() << QT_DEBUG_OUTPUT << msgBox.exec();
    if(!msgBox.exec())
    {
        int activatedIndex = p->modelIndex();
        int deletedIndex = ui->comboModel->currentIndex();

        qDebug() << QT_DEBUG_OUTPUT << "activated" << activatedIndex << " deleted" << deletedIndex;

        if(activatedIndex == deletedIndex)
        {
            QMessageBox warning(this);
            warning.setWindowTitle("错误");
            warning.setIcon(QMessageBox::Warning);
            warning.setText("请先选择其他模型再尝试删除");
            warning.exec();
            return;
        }

        QString path = CustomDir::resourcesDir() + modelName;
        qDebug() << QT_BACKGROUND_LOG << "start deleting" << path;
        if(!FileHandler::deleteFolder(path))
        {
            QMessageBox warning(this);
            warning.setWindowTitle("错误");
            warning.setIcon(QMessageBox::Warning);
            warning.setText("删除文件夹失败，请阅读log并反馈");
            warning.exec();
            return;
        }

        FileHandler::resetModel();
        p->setModelIndex((activatedIndex < deletedIndex) ?
                                         activatedIndex : activatedIndex - 1);

        // 重新建立模型列表
        ui->comboModel->clear();
        fileDir = FileHandler::getModelDirList();
        ui->comboModel->addItems(fileDir);

        ui->comboModel->setCurrentIndex(p->modelIndex());
    }
}

void SettingsDialog::onVolumeChanged()
{
    int volume = ui->volumeSlider->value();
    int returnValue;

    if(volume > HIGH_VOLUME)
    {
        if(!isHighVolumeNotified)
        {
            QMessageBox notify(this);
            notify.setWindowTitle("高音量警告");
            notify.setText("音量高于" + QString::number(HIGH_VOLUME) + "%，可能会损伤听力，是否继续？");
            notify.setIcon(QMessageBox::Warning);
            notify.addButton("确定", QMessageBox::AcceptRole);
            notify.addButton("取消", QMessageBox::RejectRole);
            returnValue = notify.exec();
            if(returnValue)
                isHighVolumeNotified = true;
            else
                return;
        }
    }

    p->setVolume(volume);
    ui->volume->setText(QString::number(p->volume()) + "%");
}

void SettingsDialog::onPaceChanged()
{
    int pace = ui->paceSlider->value();

    p->setPace(pace);
    ui->pace->setText(QString::number(p->pace()) + "%");
}

void SettingsDialog::testVoice()
{
    qDebug() << QT_DEBUG_OUTPUT << "test triggered";
    ui->previewButton->setEnabled(false);
    thread = new VoiceThread(this, VoiceThread::Test, ui->comboVoice->currentIndex());
    thread->start();
    connect(thread, &VoiceThread::done, this, [=]() {
        ui->previewButton->setEnabled(true);
    });
}

void SettingsDialog::onSaveToModelClicked()
{
    qDebug() << QT_DEBUG_OUTPUT << "Save voice to model triggered";

    FileHandler::saveVoiceToModel(p->voice());
}