Qv2ray/src/ui/w_RoutesEditor.cpp

// WARNING
// Since it's required for *extra.cpp to know the content of those macros defined below.
// We include this CPP file instead of the proper HPP file. Adding #pragma once to prevent duplicate function instances
#pragma once

#include "w_RoutesEditor.hpp"
#include "QvCoreConfigOperations.hpp"
#include "w_OutboundEditor.hpp"
#include "w_JsonEditor.hpp"
#include "w_InboundEditor.hpp"
#include "w_ImportConfig.hpp"

#include "NodeStyle.hpp"
#include "FlowViewStyle.hpp"

static bool isLoading = false;
#define CurrentRule this->rules[this->currentRuleTag]
#define LOADINGCHECK if(isLoading) return;
#define GetFirstNodeData(node, nodeModel, dataModel) (static_cast<dataModel *>(static_cast<nodeModel *>((node).nodeDataModel())->outData(0).get()))

#define CHECKEMPTYRULES if (this->rules.isEmpty()) { \
        LOG(MODULE_UI, "No rules currently, we add one.") \
        on_addRouteBtn_clicked(); \
    }

#define GRAPH_GLOBAL_OFFSET_X -80
#define GRAPH_GLOBAL_OFFSET_Y -10

#define LOAD_FLAG_BEGIN isLoading = true;
#define LOAD_FLAG_END isLoading = false;

void RouteEditor::SetupNodeWidget()
{
    if (GetGlobalConfig().uiConfig.useDarkTheme) {
        ConnectionStyle::setConnectionStyle(R"({"ConnectionStyle": {"ConstructionColor": "gray","NormalColor": "black","SelectedColor": "gray",
                                            "SelectedHaloColor": "deepskyblue","HoveredColor": "deepskyblue","LineWidth": 3.0,
                                            "ConstructionLineWidth": 2.0,"PointDiameter": 10.0,"UseDataDefinedColors": true}})");
    } else {
        QtNodes::NodeStyle::setNodeStyle(R"({"NodeStyle": {"NormalBoundaryColor": "darkgray","SelectedBoundaryColor": "deepskyblue",
                                            "GradientColor0": "mintcream","GradientColor1": "mintcream","GradientColor2": "mintcream",
                                            "GradientColor3": "mintcream","ShadowColor": [200, 200, 200],"FontColor": [10, 10, 10],
                                            "FontColorFaded": [100, 100, 100],"ConnectionPointColor": "white","PenWidth": 2.0,"HoveredPenWidth": 2.5,
                                            "ConnectionPointDiameter": 10.0,"Opacity": 1.0}})");
        QtNodes::FlowViewStyle::setStyle(R"({"FlowViewStyle": {"BackgroundColor": [255, 255, 240],"FineGridColor": [245, 245, 230],"CoarseGridColor": [235, 235, 220]}})");
        ConnectionStyle::setConnectionStyle(R"({"ConnectionStyle": {"ConstructionColor": "gray","NormalColor": "black","SelectedColor": "gray",
                                            "SelectedHaloColor": "deepskyblue","HoveredColor": "deepskyblue","LineWidth": 3.0,"ConstructionLineWidth": 2.0,
                                            "PointDiameter": 10.0,"UseDataDefinedColors": false}})");
    }

    QVBoxLayout *l = new QVBoxLayout(nodeGraphWidget);
    nodeScene = new FlowScene(nodeGraphWidget);
    connect(nodeScene, &FlowScene::nodeClicked, this, &RouteEditor::onNodeClicked);
    connect(nodeScene, &FlowScene::connectionCreated, this, &RouteEditor::onConnectionCreated);
    connect(nodeScene, &FlowScene::connectionDeleted, this, &RouteEditor::onConnectionDeleted);
    auto view = new FlowView(nodeScene);
    view->scaleDown();
    l->addWidget(view);
    l->setContentsMargins(0, 0, 0, 0);
    l->setSpacing(0);
}

RouteEditor::RouteEditor(QJsonObject connection, QWidget *parent) : QDialog(parent), root(connection), original(connection)
{
    setupUi(this);
    isLoading = true;
    setWindowFlags(windowFlags() | Qt::WindowMaximizeButtonHint);
    //
    SetupNodeWidget();
    //
    // Setup icons according to the theme settings.
    addInboundBtn->setIcon(QICON_R("add.png"));
    addOutboundBtn->setIcon(QICON_R("add.png"));
    editBtn->setIcon(QICON_R("edit.png"));
    addRouteBtn->setIcon(QICON_R("add.png"));
    delBtn->setIcon(QICON_R("delete.png"));
    balancerAddBtn->setIcon(QICON_R("add.png"));
    balancerDelBtn->setIcon(QICON_R("delete.png"));
    //
    domainStrategy = root["routing"].toObject()["domainStrategy"].toString();
    domainStrategyCombo->setCurrentText(domainStrategy);

    // Show connections in the node graph
    for (auto in : root["inbounds"].toArray()) {
        INBOUND _in = INBOUND(in.toObject());
        AddNewInbound(_in);
    }

    for (auto out : root["outbounds"].toArray()) {
        OUTBOUND _out = OUTBOUND(out.toObject());
        AddNewOutbound(_out);
    }

    for (auto item : root["routing"].toObject()["rules"].toArray()) {
        auto _rule = StructFromJsonString<RuleObject>(JsonToString(item.toObject()));
        AddNewRule(_rule);
    }

    // Set default outboung combo text AFTER adding all outbounds.
    defaultOutboundCombo->setCurrentText(root["outbounds"].toArray().first().toObject()["tag"].toString());

    // Find and add balancers.
    for (auto _balancer : root["routing"].toObject()["balancers"].toArray()) {
        auto _balancerObject = _balancer.toObject();

        if (!_balancerObject["tag"].toString().isEmpty()) {
            balancers[_balancerObject["tag"].toString()] = _balancerObject["selector"].toVariant().toStringList();
        }
    }

    isLoading = false;
}

void RouteEditor::onNodeClicked(Node &n)
{
    LOADINGCHECK

    if (isExiting()) return;

    auto isOut = outboundNodes.values().contains(&n);
    auto isIn = inboundNodes.values().contains(&n);
    auto isRule = ruleNodes.values().contains(&n);

    if (isRule) {
        // It's a rule object
        currentRuleTag = GetFirstNodeData(n, QvRuleNodeDataModel, RuleNodeData)->GetRuleTag();
        DEBUG(MODULE_GRAPH, "Selecting rule: " + currentRuleTag.toStdString())
        ShowCurrentRuleDetail();
        toolBox->setCurrentIndex(1);
    } else if (isOut || isIn) {
        // It's an inbound or an outbound.
        QString alias;
        QString host;
        int port;
        QString protocol;

        if (isOut) {
            alias = GetFirstNodeData(n, QvOutboundNodeModel, OutboundNodeData)->GetOutbound();
            QJsonObject _root = outbounds[alias].raw();
            auto result = GetOutboundData(OUTBOUND(_root), &host, &port, &protocol);
            Q_UNUSED(result)
        } else {
            alias = GetFirstNodeData(n, QvInboundNodeModel, InboundNodeData)->GetInbound();
            QJsonObject _root = inbounds[alias].raw();
            host = _root["listen"].toString();
            protocol = _root["protocol"].toString();
            port = _root["port"].toInt();
        }

        tagLabel->setText(alias);
        protocolLabel->setText(protocol);
        portLabel->setNum(port);
        hostLabel->setText(host);
    } else {
        LOG(MODULE_GRAPH, "Selected an unknown node, RARE.")
    }
}

void RouteEditor::onConnectionCreated(QtNodes::Connection const &c)
{
    LOADINGCHECK

    if (isExiting()) return;

    // Connection Established
    auto const sourceNode = c.getNode(PortType::Out);
    auto const targetNode = c.getNode(PortType::In);
    auto conns = mapExt::Values(nodeScene->connections());

    if (inboundNodes.values().contains(sourceNode) && ruleNodes.values().contains(targetNode)) {
        // It's a inbound-rule connection
        onNodeClicked(*sourceNode);
        onNodeClicked(*targetNode);
        LOG(MODULE_GRAPH, "Inbound-rule new connection.")
        // Get all connected inbounds to this rule node.
        // QStringList has an helper to let us remove duplicates, see below.
        QStringList _inbounds;

        for (auto conn : conns) {
            auto _connection = conn.get();

            if (_connection->getNode(PortType::In) == targetNode && _connection->getNode(PortType::Out) == sourceNode && _connection->id() != c.id()) {
                nodeScene->deleteConnection(*_connection);
            }
            // Append all inbounds
            else if (_connection->getNode(PortType::In) == targetNode) {
                _inbounds.append(GetFirstNodeData(*_connection->getNode(PortType::Out), QvInboundNodeModel, InboundNodeData)->GetInbound());
            }
        }

        // caused by multi-in connection
        _inbounds.removeDuplicates();
        CurrentRule.inboundTag = _inbounds;
    } else if (ruleNodes.values().contains(sourceNode) && outboundNodes.values().contains(targetNode)) {
        // It's a rule-outbound connection
        onNodeClicked(*sourceNode);
        onNodeClicked(*targetNode);
        CurrentRule.outboundTag = GetFirstNodeData((*targetNode), QvOutboundNodeModel, OutboundNodeData)->GetOutbound();
        // Connecting to an outbound will disable the balancer feature.
        CurrentRule.QV2RAY_RULE_USE_BALANCER = false;
        // Update balancer settings.
        ShowCurrentRuleDetail();
        LOG(MODULE_GRAPH, "Updated outbound: " + CurrentRule.outboundTag.toStdString())
    } else {
        // It's an impossible connection
        LOG(MODULE_GRAPH, "Unrecognized connection, RARE.")
    }
}

void RouteEditor::onConnectionDeleted(QtNodes::Connection const &c)
{
    LOADINGCHECK

    if (isExiting()) return;

    // Connection Deleted
    auto const source = c.getNode(PortType::Out);
    auto const target = c.getNode(PortType::In);

    if (inboundNodes.values().contains(source) && ruleNodes.values().contains(target)) {
        // It's a inbound-rule connection
        onNodeClicked(*source);
        onNodeClicked(*target);
        currentRuleTag = GetFirstNodeData(*target, QvRuleNodeDataModel, RuleNodeData)->GetRuleTag();
        auto _inboundTag = GetFirstNodeData(*source, QvInboundNodeModel, InboundNodeData)->GetInbound();
        LOG(MODULE_UI, "Removing inbound: " + _inboundTag.toStdString() + " from rule: " + currentRuleTag.toStdString())
        CurrentRule.inboundTag.removeAll(_inboundTag);
    } else if (ruleNodes.values().contains(source) && outboundNodes.values().contains(target)) {
        // It's a rule-outbound connection
        onNodeClicked(*source);
        onNodeClicked(*target);
        currentRuleTag = GetFirstNodeData(*source, QvRuleNodeDataModel, RuleNodeData)->GetRuleTag();
        auto _outboundTag = GetFirstNodeData(*target, QvOutboundNodeModel, OutboundNodeData)->GetOutbound();

        if (!CurrentRule.QV2RAY_RULE_USE_BALANCER && CurrentRule.outboundTag == _outboundTag) {
            CurrentRule.outboundTag.clear();
        }

        LOG(MODULE_GRAPH, "Removing an outbound: " + _outboundTag.toStdString())
    } else {
        // It's an impossible connection
        LOG(MODULE_GRAPH, "Selected an unknown node, RARE.")
    }
}

CONFIGROOT RouteEditor::OpenEditor()
{
    auto result = this->exec();

    if (rules.isEmpty()) {
        // Prevent empty rule list causing mis-detection of config type to simple.
        on_addRouteBtn_clicked();
    }

    // If clicking OK
    if (result == QDialog::Accepted) {
        QJsonArray rulesArray;
        QJsonArray _balancers;

        // Append rules by order
        for (auto i = 0; i < ruleListWidget->count(); i++) {
            auto _rule = rules[ruleListWidget->item(i)->text()];
            auto ruleJsonObject = GetRootObject(_rule);

            // Process balancer for a rule
            if (_rule.QV2RAY_RULE_USE_BALANCER) {
                // Do not use outbound tag.
                ruleJsonObject.remove("outboundTag");

                // Find balancer list
                if (!_balancers.contains(_rule.balancerTag)) {
                    LOG(MODULE_UI, "Cannot find a balancer for tag: " + _rule.balancerTag.toStdString())
                } else {
                    auto _balancerList = balancers[_rule.balancerTag];
                    QJsonObject balancerEntry;
                    balancerEntry["tag"] = _rule.balancerTag;
                    balancerEntry["selector"] = QJsonArray::fromStringList(_balancerList);
                    _balancers.append(balancerEntry);
                }
            }

            // Remove some empty fields.
            if (_rule.port.isEmpty()) {
                ruleJsonObject.remove("port");
            }

            if (_rule.network.isEmpty()) {
                ruleJsonObject.remove("network");
            }

            rulesArray.append(ruleJsonObject);
        }

        QJsonObject routing;
        routing["domainStrategy"] = domainStrategy;
        routing["rules"] = rulesArray;
        routing["balancers"] = _balancers;
        //
        QJsonArray _inbounds;
        QJsonArray _outbounds;

        // Convert our internal data format to QJsonArray
        for (auto x : inbounds) {
            _inbounds.append(x.raw());
        }

        for (auto x : outbounds) {
            if (getTag(x) == defaultOutbound) {
                // Put the default outbound to the first.
                _outbounds.push_front(x.raw());
            } else {
                _outbounds.push_back(x.raw());
            }
        }

        root["inbounds"] = _inbounds;
        root["outbounds"] = _outbounds;
        root["routing"] = routing;
        return root;
    } else {
        return original;
    }
}

RouteEditor::~RouteEditor() {}
void RouteEditor::on_buttonBox_accepted() {}

void RouteEditor::ShowCurrentRuleDetail()
{
    LOADINGCHECK

    if (!rules.contains(currentRuleTag)) {
        LOG(MODULE_UI, "WARNING, trying to access a non-exist rule entry. return.")
        return;
    }

    // Switch to the detailed page.
    ruleEnableCB->setEnabled(true);
    ruleEnableCB->setChecked(CurrentRule.QV2RAY_RULE_ENABLED);
    LOAD_FLAG_BEGIN
    ruleTagLineEdit->setText(CurrentRule.QV2RAY_RULE_TAG);
    balancerSelectionCombo->clear();

    // BUG added the wrong items, should be outbound list.
    for (auto out : outbounds) {
        balancerSelectionCombo->addItem((out)["tag"].toString());
    }

    //
    // Balancers combo and balancer list.
    enableBalancerCB->setChecked(CurrentRule.QV2RAY_RULE_USE_BALANCER);
    balancersWidget->setEnabled(CurrentRule.QV2RAY_RULE_USE_BALANCER);

    if (!CurrentRule.balancerTag.isEmpty()) {
        balancerList->clear();
        balancerList->addItems(balancers[CurrentRule.balancerTag]);
    }

    isLoading = false;
    // Networks
    auto network = CurrentRule.network.toLower();
    bool isBoth = (network.contains("tcp") && network.contains("udp")) || network.isEmpty();
    netUDPRB->setChecked(network.contains("udp"));
    netTCPRB->setChecked(network.contains("tcp"));
    netBothRB->setChecked(isBoth);
    //
    // Set protocol checkboxes.
    auto protocol = CurrentRule.protocol;
    routeProtocolHTTPCB->setChecked(protocol.contains("http"));
    routeProtocolTLSCB->setChecked(protocol.contains("tls"));
    routeProtocolBTCB->setChecked(protocol.contains("bittorrent"));
    //
    // Port
    routePortTxt->setText(CurrentRule.port);
    //
    // Users
    QString users = Stringify(CurrentRule.user, NEWLINE);
    routeUserTxt->setPlainText(users);
    //
    // Incoming Sources
    QString sources = Stringify(CurrentRule.source, NEWLINE);
    sourceIPList->setPlainText(sources);
    //
    // Domains
    QString domains = Stringify(CurrentRule.domain, NEWLINE);
    hostList->setPlainText(domains);
    //
    // Outcoming IPs
    QString ips = Stringify(CurrentRule.ip, NEWLINE);
    ipList->setPlainText(ips);
    LOAD_FLAG_END
}
void RouteEditor::on_insertDirectBtn_clicked()
{
    auto freedom = GenerateFreedomOUT("as-is", "", 0);
    auto tag = "Freedom_" + QString::number(QTime::currentTime().msecsSinceStartOfDay());
    auto out = GenerateOutboundEntry("freedom", freedom, QJsonObject(), QJsonObject(), "0.0.0.0", tag);
    // ADD NODE
    AddNewOutbound(out);
    statusLabel->setText(tr("Added DIRECT outbound"));
}
void RouteEditor::on_routeProtocolHTTPCB_stateChanged(int arg1)
{
    LOADINGCHECK
    QStringList protocols;

    if (arg1 == Qt::Checked) protocols.push_back("http");

    if (routeProtocolTLSCB->isChecked()) protocols.push_back("tls");

    if (routeProtocolBTCB->isChecked()) protocols.push_back("bittorrent");

    CurrentRule.protocol = protocols;
    statusLabel->setText(tr("Protocol list changed: ") + Stringify(protocols));
}
void RouteEditor::on_routeProtocolTLSCB_stateChanged(int arg1)
{
    LOADINGCHECK
    QStringList protocols;

    if (arg1 == Qt::Checked) protocols.push_back("tls");

    if (routeProtocolHTTPCB->isChecked()) protocols.push_back("http");

    if (routeProtocolBTCB->isChecked()) protocols.push_back("bittorrent");

    CurrentRule.protocol = protocols;
    statusLabel->setText(tr("Protocol list changed: ") + Stringify(protocols));
}
void RouteEditor::on_routeProtocolBTCB_stateChanged(int arg1)
{
    LOADINGCHECK
    QStringList protocols;

    if (arg1 == Qt::Checked) protocols.push_back("bittorrent");

    if (routeProtocolHTTPCB->isChecked()) protocols.push_back("http");

    if (routeProtocolTLSCB->isChecked()) protocols.push_back("tls");

    CurrentRule.protocol = protocols;
    statusLabel->setText(tr("Protocol list changed: ") + Stringify(protocols));
}
void RouteEditor::on_balancerAddBtn_clicked()
{
    LOADINGCHECK
    auto balancerTx = balancerSelectionCombo->currentText();

    if (!balancerTx.isEmpty()) {
        this->balancers[CurrentRule.balancerTag].append(balancerSelectionCombo->currentText());
        balancerList->addItem(balancerTx);
        balancerSelectionCombo->setEditText("");
        statusLabel->setText(tr("OK"));
    } else {
        statusLabel->setText(tr("Balancer is empty, not processing."));
    }
}
void RouteEditor::on_balancerDelBtn_clicked()
{
    LOADINGCHECK

    if (balancerList->currentRow() < 0) {
        return;
    }

    balancers[CurrentRule.balancerTag].removeAt(balancerList->currentRow());
    balancerList->takeItem(balancerList->currentRow());
    statusLabel->setText(tr("Removed a balancer entry."));
}
void RouteEditor::on_hostList_textChanged()
{
    LOADINGCHECK
    CurrentRule.domain = SplitLines(hostList->toPlainText());
}
void RouteEditor::on_ipList_textChanged()
{
    LOADINGCHECK
    CurrentRule.ip = SplitLines(ipList->toPlainText());
}
void RouteEditor::on_routePortTxt_textEdited(const QString &arg1)
{
    LOADINGCHECK
    CurrentRule.port = arg1;
}
void RouteEditor::on_routeUserTxt_textEdited(const QString &arg1)
{
    LOADINGCHECK
    CurrentRule.user = SplitLines(arg1);
}
void RouteEditor::on_addRouteBtn_clicked()
{
    LOADINGCHECK
    // Add Route
    RuleObject rule;
    //
    rule.QV2RAY_RULE_ENABLED = true;
    rule.QV2RAY_RULE_USE_BALANCER = false;
    // Default balancer tag, it's a random string.
    auto bTag = GenerateRandomString();
    rule.QV2RAY_RULE_TAG = GenerateRandomString(5);
    rule.balancerTag = bTag;
    balancers[bTag] = QStringList();
    AddNewRule(rule);
}
void RouteEditor::on_netBothRB_clicked()
{
    LOADINGCHECK
    CurrentRule.network = "tcp,udp";
}
void RouteEditor::on_netUDPRB_clicked()
{
    LOADINGCHECK
    CurrentRule.network = "udp";
}
void RouteEditor::on_netTCPRB_clicked()
{
    LOADINGCHECK
    CurrentRule.network = "tcp";
}
void RouteEditor::on_routeUserTxt_textChanged()
{
    LOADINGCHECK
    CurrentRule.user = SplitLines(routeUserTxt->toPlainText());
}
void RouteEditor::on_sourceIPList_textChanged()
{
    LOADINGCHECK
    CurrentRule.source = SplitLines(sourceIPList->toPlainText());
}
void RouteEditor::on_enableBalancerCB_stateChanged(int arg1)
{
    LOADINGCHECK
    auto useBalancer = arg1 == Qt::Checked;
    CurrentRule.QV2RAY_RULE_USE_BALANCER = useBalancer;
    balancersWidget->setEnabled(useBalancer);

    if (CurrentRule.balancerTag.isEmpty()) {
        LOG(MODULE_UI, "Creating a new balancer tag.")
        CurrentRule.balancerTag = GenerateRandomString(6);
        balancers[CurrentRule.balancerTag] = QStringList();
    }

    DEBUG(MODULE_UI, "Balancer: " + CurrentRule.balancerTag.toStdString())

    if (useBalancer) {
        LOG(MODULE_UI, "A rule has been set to use balancer, disconnect it to any outbound.")
        auto ruleNode = ruleNodes[currentRuleTag];

        for (auto conn : mapExt::Values(nodeScene->connections())) {
            if (conn.get()->getNode(PortType::Out) == ruleNode) {
                nodeScene->deleteConnection(*conn);
            }
        }
    } else {
        QvMessageBox(this, tr("Route Editor"), tr("To make this rule ready to use, you need to connect it to an outbound node."));
    }
}
void RouteEditor::on_addDefaultBtn_clicked()
{
    LOADINGCHECK
    // Add default connection from GlobalConfig
    auto conf = GetGlobalConfig();
    //
    auto _Inconfig = conf.inboundConfig;
    //
    auto _in_httpConf = GenerateHTTPIN(QList<AccountObject>() << _Inconfig.httpAccount);
    auto _in_socksConf = GenerateSocksIN((_Inconfig.socks_useAuth ? "password" : "noauth"),
                                         QList<AccountObject>() << _Inconfig.socksAccount,
                                         _Inconfig.socksUDP, _Inconfig.socksLocalIP);
    //
    auto _in_HTTP = GenerateInboundEntry(_Inconfig.listenip, _Inconfig.http_port, "http", _in_httpConf, "HTTP_gConf");
    auto _in_SOCKS = GenerateInboundEntry(_Inconfig.listenip, _Inconfig.socks_port, "socks", _in_socksConf, "SOCKS_gConf");
    //
    AddNewInbound(_in_HTTP);
    AddNewInbound(_in_SOCKS);
    CHECKEMPTYRULES
}
void RouteEditor::on_insertBlackBtn_clicked()
{
    LOADINGCHECK
    auto blackHole = GenerateBlackHoleOUT(false);
    auto tag = "blackhole_" + QString::number(QTime::currentTime().msecsSinceStartOfDay());
    auto _blackHoleOutbound = GenerateOutboundEntry("blackhole", blackHole, QJsonObject(), QJsonObject(), "0.0.0.0", tag);
    AddNewOutbound(_blackHoleOutbound);
}
void RouteEditor::on_addInboundBtn_clicked()
{
    LOADINGCHECK
    InboundEditor w(INBOUND(), this);
    auto _result = w.OpenEditor();

    if (w.result() == QDialog::Accepted) {
        AddNewInbound(_result);
    }

    CHECKEMPTYRULES
}
void RouteEditor::on_addOutboundBtn_clicked()
{
    LOADINGCHECK
    ImportConfigWindow *w = new ImportConfigWindow(this);
    // True here for not keep the inbounds.
    auto configs = w->OpenImport(true);

    for (auto i = 0; i < configs.count(); i++) {
        auto conf = configs.values()[i];
        auto name = configs.key(conf, "");

        if (name.isEmpty())
            continue;

        // conf is rootObject, needs to unwrap it.
        auto confList = conf["outbounds"].toArray();

        for (int i = 0; i < confList.count(); i++) {
            AddNewOutbound(OUTBOUND(confList[i].toObject()));
        }
    }

    CHECKEMPTYRULES
}
void RouteEditor::on_ruleEnableCB_stateChanged(int arg1)
{
    bool _isEnabled = arg1 == Qt::Checked;
    CurrentRule.QV2RAY_RULE_ENABLED = _isEnabled;
    routeEditGroupBox->setEnabled(_isEnabled);
    routeRuleGroupBox->setEnabled(_isEnabled);
}
void RouteEditor::on_delBtn_clicked()
{
    if (nodeScene->selectedNodes().empty()) {
        QvMessageBox(this, tr("Remove Items"), tr("Please select a node from the graph to continue."));
    }

    auto firstNode = nodeScene->selectedNodes()[0];
    auto isInbound = inboundNodes.values().contains(firstNode);
    auto isOutbound = outboundNodes.values().contains(firstNode);
    auto isRule = ruleNodes.values().contains(firstNode);

    // Get the tag first, and call inbounds/outbounds/rules container variable remove()
    // Remove the node last since some events may trigger.
    // Then remove the node container.
    if (isInbound) {
        currentInboundOutboundTag = GetFirstNodeData(*firstNode, QvInboundNodeModel, InboundNodeData)->GetInbound();
        inbounds.remove(currentInboundOutboundTag);
        nodeScene->removeNode(*inboundNodes[currentInboundOutboundTag]);
        inboundNodes.remove(currentInboundOutboundTag);
    } else if (isOutbound) {
        currentInboundOutboundTag = GetFirstNodeData(*firstNode, QvOutboundNodeModel, OutboundNodeData)->GetOutbound();
        outbounds.remove(currentInboundOutboundTag);
        nodeScene->removeNode(*outboundNodes[currentInboundOutboundTag]);
        outboundNodes.remove(currentInboundOutboundTag);
        //
        defaultOutboundCombo->removeItem(defaultOutbound.indexOf(currentInboundOutboundTag));

        if (currentInboundOutboundTag == defaultOutbound) {
            // Set default outbound to the new one since the current has been removed.
            defaultOutbound = outbounds.firstKey();
            defaultOutboundCombo->setCurrentText(defaultOutbound);
        }
    } else if (isRule) {
        ruleEnableCB->setEnabled(false);
        currentRuleTag = GetFirstNodeData(*firstNode, QvRuleNodeDataModel, RuleNodeData)->GetRuleTag();
        routeRuleGroupBox->setEnabled(false);
        routeEditGroupBox->setEnabled(false);
        rules.remove(currentRuleTag);
        nodeScene->removeNode(*ruleNodes[currentRuleTag]);
        ruleNodes.remove(currentRuleTag);
        //
        // Remove item from the rule order list widget.
        ruleListWidget->takeItem(ruleListWidget->row(ruleListWidget->findItems(currentRuleTag, Qt::MatchExactly).first()));
        currentRuleTag = rules.firstKey();
        ShowCurrentRuleDetail();
    } else {
        LOG(MODULE_UI, "Unknown node selected.")
    }
}
void RouteEditor::on_editBtn_clicked()
{
    if (nodeScene->selectedNodes().empty()) {
        QvMessageBox(this, tr("Edit Inbound/Outbound"), tr("Please select a node from the graph to continue."));
    }

    auto firstNode = nodeScene->selectedNodes()[0];
    auto isInbound = inboundNodes.values().contains(firstNode);
    auto isOutbound = outboundNodes.values().contains(firstNode);

    if (isInbound) {
        currentInboundOutboundTag = GetFirstNodeData(*firstNode, QvInboundNodeModel, InboundNodeData)->GetInbound();
        auto _in = inbounds[currentInboundOutboundTag];
        INBOUND _result;
        auto protocol = _in["protocol"].toString();
        int _code;

        if (protocol != "http" && protocol != "mtproto" && protocol != "socks" && protocol != "dokodemo-door") {
            QvMessageBox(this, tr("Cannot Edit"), tr("Currently, this type of outbound is not supported by the editor.") + "\r\n" +
                         tr("We will launch Json Editor instead."));
            statusLabel->setText(tr("Opening JSON editor"));
            JsonEditor *w = new JsonEditor(_in, this);
            _result = INBOUND(w->OpenEditor());
            _code = w->result();
            delete w;
        } else {
            InboundEditor *w = new InboundEditor(_in, this);
            statusLabel->setText(tr("Opening default inbound editor"));
            _result = w->OpenEditor();
            _code = w->result();
            delete w;
        }

        statusLabel->setText(tr("OK"));

        if (_code == QDialog::Accepted) {
            bool isTagChanged = getTag(_in) != getTag(_result);

            if (isTagChanged) {
                RenameItemTag(RENAME_INBOUND, getTag(_in), getTag(_result));
            }

            inbounds[getTag(_result)] = _result;
        }
    } else if (isOutbound) {
        currentInboundOutboundTag = GetFirstNodeData(*firstNode, QvOutboundNodeModel, OutboundNodeData)->GetOutbound();
        OUTBOUND _result;
        auto _out = outbounds[currentInboundOutboundTag];
        auto protocol =  _out["protocol"].toString();
        int _code;

        if (protocol != "vmess" && protocol != "shadowsocks" && protocol != "socks") {
            QvMessageBox(this, tr("Unsupported Outbound Type"),
                         tr("This outbound entry is not supported by the GUI editor.") + NEWLINE +
                         tr("We will launch Json Editor instead."));
            JsonEditor w(_out, this);
            statusLabel->setText(tr("Opening JSON editor"));
            _result = OUTBOUND(w.OpenEditor());
            _code = w.result();
        } else {
            OutboundEditor w(_out, this);
            statusLabel->setText(tr("Opening default outbound editor."));
            _result = w.OpenEditor();
            _code = w.result();
        }

        if (_code == QDialog::Accepted) {
            bool isTagChanged = getTag(_out) != getTag(_result);

            if (isTagChanged) {
                RenameItemTag(RENAME_OUTBOUND, getTag(_out), getTag(_result));
            }

            outbounds[getTag(_result)] = _result;
            statusLabel->setText(tr("OK"));
        }
    } else {
        LOG(MODULE_UI, "Cannot apply 'edit' operation to non-inbound and non-outbound")
    }
}

void RouteEditor::on_domainStrategyCombo_currentIndexChanged(const QString &arg1)
{
    LOADINGCHECK
    domainStrategy = arg1;
}

void RouteEditor::on_defaultOutboundCombo_currentIndexChanged(const QString &arg1)
{
    LOADINGCHECK
    defaultOutbound = arg1;
}

void RouteEditor::on_ruleTagLineEdit_textEdited(const QString &arg1)
{
    RenameItemTag(RENAME_RULE, CurrentRule.QV2RAY_RULE_TAG, arg1);
}