mirror of
https://github.com/MatsuriDayo/nekoray.git
synced 2025-12-17 20:44:38 +03:00
1473 lines
51 KiB
C++
1473 lines
51 KiB
C++
#include "./ui_mainwindow.h"
|
||
#include "mainwindow.h"
|
||
|
||
#include "fmt/Preset.hpp"
|
||
#include "db/ProfileFilter.hpp"
|
||
#include "db/ConfigBuilder.hpp"
|
||
#include "sub/GroupUpdater.hpp"
|
||
#include "sys/ExternalProcess.hpp"
|
||
#include "sys/AutoRun.hpp"
|
||
|
||
#include "ui/ThemeManager.hpp"
|
||
#include "ui/TrayIcon.hpp"
|
||
#include "ui/edit/dialog_edit_profile.h"
|
||
#include "ui/dialog_basic_settings.h"
|
||
#include "ui/dialog_manage_groups.h"
|
||
#include "ui/dialog_manage_routes.h"
|
||
#include "ui/dialog_hotkey.h"
|
||
|
||
#include "3rdparty/qrcodegen.hpp"
|
||
#include "3rdparty/VT100Parser.hpp"
|
||
#include "qv2ray/v2/components/proxy/QvProxyConfigurator.hpp"
|
||
#include "qv2ray/v2/ui/LogHighlighter.hpp"
|
||
|
||
#ifndef NKR_NO_EXTERNAL
|
||
|
||
#include "3rdparty/ZxingQtReader.hpp"
|
||
|
||
#endif
|
||
|
||
#ifdef Q_OS_WIN
|
||
|
||
#include "3rdparty/WinCommander.hpp"
|
||
|
||
#else
|
||
|
||
#include <unistd.h>
|
||
|
||
#endif
|
||
|
||
#include <QClipboard>
|
||
#include <QLabel>
|
||
#include <QTextBlock>
|
||
#include <QScrollBar>
|
||
#include <QMutex>
|
||
#include <QScreen>
|
||
#include <QDesktopServices>
|
||
#include <QInputDialog>
|
||
#include <QThread>
|
||
#include <QTimer>
|
||
#include <QMessageBox>
|
||
#include <QDir>
|
||
#include <QFileInfo>
|
||
#include <QFileSystemWatcher>
|
||
|
||
MainWindow::MainWindow(QWidget *parent)
|
||
: QMainWindow(parent), ui(new Ui::MainWindow) {
|
||
mainwindow = this;
|
||
dialog_message = [=](const QString &a, const QString &b) {
|
||
runOnUiThread([=] {
|
||
dialog_message_impl(a, b);
|
||
});
|
||
};
|
||
|
||
// Load Manager
|
||
auto isLoaded = NekoRay::profileManager->Load();
|
||
if (!isLoaded) {
|
||
auto defaultGroup = NekoRay::ProfileManager::NewGroup();
|
||
defaultGroup->name = tr("Default");
|
||
NekoRay::profileManager->AddGroup(defaultGroup);
|
||
}
|
||
|
||
// Setup misc UI
|
||
themeManager->ApplyTheme(NekoRay::dataStore->theme);
|
||
ui->setupUi(this);
|
||
title_base = windowTitle();
|
||
connect(ui->menu_start, &QAction::triggered, this, [=]() { neko_start(); });
|
||
connect(ui->menu_stop, &QAction::triggered, this, [=]() { neko_stop(); });
|
||
connect(ui->tabWidget->tabBar(), &QTabBar::tabMoved, this, [=](int from, int to) {
|
||
// use tabData to track tab & gid
|
||
NekoRay::profileManager->_groups.clear();
|
||
for (int i = 0; i < ui->tabWidget->tabBar()->count(); i++) {
|
||
NekoRay::profileManager->_groups += ui->tabWidget->tabBar()->tabData(i).toInt();
|
||
}
|
||
NekoRay::profileManager->Save();
|
||
});
|
||
ui->label_running->installEventFilter(this);
|
||
ui->label_inbound->installEventFilter(this);
|
||
RegisterHotkey(false);
|
||
auto last_size = NekoRay::dataStore->mw_size.split("x");
|
||
if (last_size.length() == 2) {
|
||
auto w = last_size[0].toInt();
|
||
auto h = last_size[1].toInt();
|
||
if (w > 0 && h > 0) {
|
||
resize(w, h);
|
||
}
|
||
}
|
||
|
||
// top bar
|
||
ui->toolButton_program->setMenu(ui->menu_program);
|
||
ui->toolButton_preferences->setMenu(ui->menu_preferences);
|
||
ui->toolButton_server->setMenu(ui->menu_server);
|
||
ui->menubar->setVisible(false);
|
||
connect(ui->toolButton_document, &QToolButton::clicked, this,
|
||
[=] { QDesktopServices::openUrl(QUrl("https://matsuridayo.github.io/")); });
|
||
connect(ui->toolButton_ads, &QToolButton::clicked, this,
|
||
[=] { QDesktopServices::openUrl(QUrl("https://matsuricom.github.io/")); });
|
||
connect(ui->toolButton_update, &QToolButton::clicked, this, [=] { CheckUpdate(); });
|
||
|
||
// Setup log UI
|
||
new SyntaxHighlighter(false, qvLogDocument);
|
||
ui->masterLogBrowser->setDocument(qvLogDocument);
|
||
ui->masterLogBrowser->setFont(QFontDatabase::systemFont(QFontDatabase::FixedFont));
|
||
{
|
||
auto font = ui->masterLogBrowser->font();
|
||
font.setPointSize(9);
|
||
ui->masterLogBrowser->setFont(font);
|
||
qvLogDocument->setDefaultFont(font);
|
||
}
|
||
connect(ui->masterLogBrowser->verticalScrollBar(), &QSlider::valueChanged, this, [=](int value) {
|
||
if (ui->masterLogBrowser->verticalScrollBar()->maximum() == value)
|
||
qvLogAutoScoll = true;
|
||
else
|
||
qvLogAutoScoll = false;
|
||
});
|
||
connect(ui->masterLogBrowser, &QTextBrowser::textChanged, this, [=]() {
|
||
if (!qvLogAutoScoll)
|
||
return;
|
||
auto bar = ui->masterLogBrowser->verticalScrollBar();
|
||
bar->setValue(bar->maximum());
|
||
});
|
||
showLog = [=](const QString &log) {
|
||
runOnUiThread([=] {
|
||
show_log_impl(log);
|
||
});
|
||
};
|
||
showLog_ext = [=](const QString &tag, const QString &log) {
|
||
runOnUiThread([=] {
|
||
show_log_impl("[" + tag + "] " + log);
|
||
});
|
||
};
|
||
showLog_ext_vt100 = [=](const QString &log) {
|
||
runOnUiThread([=] {
|
||
show_log_impl(cleanVT100String(log));
|
||
});
|
||
};
|
||
|
||
// table UI
|
||
ui->proxyListTable->callback_save_order = [=] {
|
||
auto group = NekoRay::profileManager->CurrentGroup();
|
||
group->order = ui->proxyListTable->order;
|
||
group->Save();
|
||
};
|
||
connect(ui->proxyListTable->horizontalHeader(), &QHeaderView::sectionClicked, this,
|
||
[=](int logicalIndex) {
|
||
NekoRay::GroupSortAction action;
|
||
// 不正确的descending实现
|
||
if (proxy_last_order == logicalIndex) {
|
||
action.descending = true;
|
||
proxy_last_order = -1;
|
||
} else {
|
||
proxy_last_order = logicalIndex;
|
||
}
|
||
action.save_sort = true;
|
||
// 表头
|
||
if (logicalIndex == 1) {
|
||
action.method = NekoRay::GroupSortMethod::ByType;
|
||
} else if (logicalIndex == 2) {
|
||
action.method = NekoRay::GroupSortMethod::ByAddress;
|
||
} else if (logicalIndex == 3) {
|
||
action.method = NekoRay::GroupSortMethod::ByName;
|
||
} else if (logicalIndex == 4) {
|
||
action.method = NekoRay::GroupSortMethod::ByLatency;
|
||
} else if (logicalIndex == 0) {
|
||
action.method = NekoRay::GroupSortMethod::ById;
|
||
} else {
|
||
return;
|
||
}
|
||
refresh_proxy_list_impl(-1, action);
|
||
});
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(0, QHeaderView::ResizeToContents);
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(1, QHeaderView::ResizeToContents);
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(2, QHeaderView::Stretch);
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(3, QHeaderView::Stretch);
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(4, QHeaderView::ResizeToContents);
|
||
ui->proxyListTable->horizontalHeader()->setSectionResizeMode(5, QHeaderView::ResizeToContents);
|
||
ui->tableWidget_conn->horizontalHeader()->setSectionResizeMode(0, QHeaderView::ResizeToContents);
|
||
ui->tableWidget_conn->horizontalHeader()->setSectionResizeMode(1, QHeaderView::ResizeToContents);
|
||
ui->tableWidget_conn->horizontalHeader()->setSectionResizeMode(2, QHeaderView::Stretch);
|
||
|
||
// search box
|
||
ui->search->setVisible(false);
|
||
connect(shortcut_ctrl_f, &QShortcut::activated, this, [=] {
|
||
ui->search->setVisible(true);
|
||
ui->search->setFocus();
|
||
});
|
||
connect(shortcut_esc, &QShortcut::activated, this, [=] {
|
||
if (ui->search->isVisible()) {
|
||
ui->search->setText("");
|
||
ui->search->textChanged("");
|
||
ui->search->setVisible(false);
|
||
}
|
||
if (select_mode) {
|
||
emit profile_selected(-1);
|
||
select_mode = false;
|
||
refresh_status();
|
||
}
|
||
});
|
||
connect(ui->search, &QLineEdit::textChanged, this, [=](const QString &text) {
|
||
if (text.isEmpty()) {
|
||
for (int i = 0; i < ui->proxyListTable->rowCount(); i++) {
|
||
ui->proxyListTable->setRowHidden(i, false);
|
||
}
|
||
} else {
|
||
QList<QTableWidgetItem *> findItem = ui->proxyListTable->findItems(text, Qt::MatchContains);
|
||
for (int i = 0; i < ui->proxyListTable->rowCount(); i++) {
|
||
ui->proxyListTable->setRowHidden(i, true);
|
||
}
|
||
for (auto item: findItem) {
|
||
if (item != nullptr) ui->proxyListTable->setRowHidden(item->row(), false);
|
||
}
|
||
}
|
||
});
|
||
|
||
// refresh
|
||
this->refresh_groups();
|
||
|
||
// Setup Tray
|
||
tray = new QSystemTrayIcon(this);//初始化托盘对象tray
|
||
tray->setContextMenu(ui->menu_program);//创建托盘菜单
|
||
tray->show();//让托盘图标显示在系统托盘上
|
||
connect(tray, &QSystemTrayIcon::activated, this,
|
||
[=](QSystemTrayIcon::ActivationReason reason) {
|
||
switch (reason) {
|
||
case QSystemTrayIcon::Trigger:
|
||
if (this->isVisible()) {
|
||
hide();
|
||
} else {
|
||
this->showNormal();
|
||
this->raise();
|
||
this->activateWindow();
|
||
}
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
});
|
||
|
||
//
|
||
ui->menu_program_preference->addActions(ui->menu_preferences->actions());
|
||
connect(ui->menu_add_from_clipboard2, &QAction::triggered, ui->menu_add_from_clipboard, &QAction::trigger);
|
||
//
|
||
connect(ui->menu_program, &QMenu::aboutToShow, this, [=]() {
|
||
ui->actionRemember_last_proxy->setChecked(NekoRay::dataStore->remember_enable);
|
||
ui->actionStart_with_system->setChecked(GetProcessAutoRunSelf());
|
||
ui->actionAllow_LAN->setChecked(NekoRay::dataStore->inbound_address == "0.0.0.0");
|
||
// active server
|
||
for (const auto &old: ui->menuActive_Server->actions()) {
|
||
ui->menuActive_Server->removeAction(old);
|
||
old->deleteLater();
|
||
}
|
||
for (const auto &pf: NekoRay::profileManager->CurrentGroup()->ProfilesWithOrder()) {
|
||
auto a = new QAction(pf->bean->DisplayTypeAndName());
|
||
a->setProperty("id", pf->id);
|
||
a->setCheckable(true);
|
||
if (NekoRay::dataStore->started_id == pf->id) a->setChecked(true);
|
||
ui->menuActive_Server->addAction(a);
|
||
}
|
||
// active routing
|
||
for (const auto &old: ui->menuActive_Routing->actions()) {
|
||
ui->menuActive_Routing->removeAction(old);
|
||
old->deleteLater();
|
||
}
|
||
for (const auto &name: NekoRay::Routing::List()) {
|
||
auto a = new QAction(name);
|
||
a->setCheckable(true);
|
||
a->setChecked(name == NekoRay::dataStore->active_routing);
|
||
ui->menuActive_Routing->addAction(a);
|
||
}
|
||
});
|
||
connect(ui->menuActive_Server, &QMenu::triggered, this, [=](QAction *a) {
|
||
bool ok;
|
||
auto id = a->property("id").toInt(&ok);
|
||
if (!ok) return;
|
||
if (NekoRay::dataStore->started_id == id) {
|
||
neko_stop();
|
||
} else {
|
||
neko_start(id);
|
||
}
|
||
});
|
||
connect(ui->menuActive_Routing, &QMenu::triggered, this, [=](QAction *a) {
|
||
auto fn = a->text();
|
||
if (!fn.isEmpty()) {
|
||
NekoRay::Routing r;
|
||
r.load_control_force = true;
|
||
r.fn = "routes/" + fn;
|
||
if (r.Load()) {
|
||
auto btn = QMessageBox::question(GetMessageBoxParent(), "NekoRay",
|
||
tr("Load routing and apply: %1").arg(fn) + "\n" + r.toString());
|
||
if (btn == QMessageBox::Yes) {
|
||
NekoRay::Routing::SetToActive(fn);
|
||
if (NekoRay::dataStore->started_id >= 0) {
|
||
neko_start(NekoRay::dataStore->started_id);
|
||
} else {
|
||
refresh_status();
|
||
}
|
||
}
|
||
}
|
||
}
|
||
});
|
||
connect(ui->actionRemember_last_proxy, &QAction::triggered, this, [=](bool checked) {
|
||
NekoRay::dataStore->remember_enable = checked;
|
||
NekoRay::dataStore->Save();
|
||
});
|
||
connect(ui->actionStart_with_system, &QAction::triggered, this, [=](bool checked) {
|
||
SetProcessAutoRunSelf(checked);
|
||
});
|
||
connect(ui->actionAllow_LAN, &QAction::triggered, this, [=](bool checked) {
|
||
NekoRay::dataStore->inbound_address = checked ? "0.0.0.0" : "127.0.0.1";
|
||
dialog_message("", "UpdateDataStore");
|
||
});
|
||
//
|
||
connect(ui->checkBox_VPN, &QCheckBox::clicked, this, [=](bool checked) {
|
||
neko_set_spmode(checked ? NekoRay::SystemProxyMode::VPN : NekoRay::SystemProxyMode::DISABLE);
|
||
});
|
||
connect(ui->menu_spmode, &QMenu::aboutToShow, this, [=]() {
|
||
ui->menu_spmode_disabled->setChecked(title_spmode == NekoRay::SystemProxyMode::DISABLE);
|
||
ui->menu_spmode_system_proxy->setChecked(title_spmode == NekoRay::SystemProxyMode::SYSTEM_PROXY);
|
||
ui->menu_spmode_vpn->setChecked(title_spmode == NekoRay::SystemProxyMode::VPN);
|
||
});
|
||
connect(ui->menu_spmode_system_proxy, &QAction::triggered, this,
|
||
[=]() { neko_set_spmode(NekoRay::SystemProxyMode::SYSTEM_PROXY); });
|
||
connect(ui->menu_spmode_vpn, &QAction::triggered, this,
|
||
[=]() { neko_set_spmode(NekoRay::SystemProxyMode::VPN); });
|
||
connect(ui->menu_spmode_disabled, &QAction::triggered, this,
|
||
[=]() { neko_set_spmode(NekoRay::SystemProxyMode::DISABLE); });
|
||
connect(ui->menu_qr, &QAction::triggered, this, [=]() { display_qr_link(false); });
|
||
connect(ui->menu_tcp_ping, &QAction::triggered, this, [=]() { speedtest_current_group(0); });
|
||
connect(ui->menu_url_test, &QAction::triggered, this, [=]() { speedtest_current_group(1); });
|
||
connect(ui->menu_full_test, &QAction::triggered, this, [=]() { speedtest_current_group(2); });
|
||
refresh_status();
|
||
|
||
// Prepare core
|
||
NekoRay::dataStore->core_token = GetRandomString(32);
|
||
NekoRay::dataStore->core_port = MkPort();
|
||
if (NekoRay::dataStore->core_port <= 0) NekoRay::dataStore->core_port = 19810;
|
||
|
||
auto core_path = NekoRay::dataStore->core_path;
|
||
if (NekoRay::dataStore->flag_use_appdata) {
|
||
core_path = QApplication::applicationDirPath() + "/nekoray_core";
|
||
}
|
||
|
||
QStringList args;
|
||
args.push_back("nekoray");
|
||
args.push_back("-port");
|
||
args.push_back(Int2String(NekoRay::dataStore->core_port));
|
||
#ifdef NKR_DEBUG
|
||
args.push_back("-debug");
|
||
#endif
|
||
|
||
// Start core
|
||
core_process = new NekoRay::sys::CoreProcess(core_path, args);
|
||
core_process->Start();
|
||
|
||
setup_grpc();
|
||
|
||
// Start last
|
||
if (NekoRay::dataStore->remember_enable) {
|
||
if (NekoRay::dataStore->system_proxy_mode == NekoRay::SystemProxyMode::SYSTEM_PROXY) {
|
||
neko_set_spmode(NekoRay::dataStore->system_proxy_mode, false);
|
||
}
|
||
if (NekoRay::dataStore->remember_id >= 0) {
|
||
runOnUiThread([=] { neko_start(NekoRay::dataStore->remember_id); });
|
||
}
|
||
}
|
||
|
||
connect(qApp, &QGuiApplication::commitDataRequest, this, &MainWindow::on_commitDataRequest);
|
||
|
||
if (!NekoRay::dataStore->start_minimal) show();
|
||
}
|
||
|
||
void MainWindow::closeEvent(QCloseEvent *event) {
|
||
if (tray->isVisible()) {
|
||
hide(); //隐藏窗口
|
||
event->ignore(); //忽略事件
|
||
}
|
||
}
|
||
|
||
MainWindow::~MainWindow() {
|
||
delete ui;
|
||
}
|
||
|
||
// Group tab manage
|
||
|
||
inline int tabIndex2GroupId(int index) {
|
||
if (NekoRay::profileManager->_groups.length() <= index) return -1;
|
||
return NekoRay::profileManager->_groups[index];
|
||
}
|
||
|
||
inline int groupId2TabIndex(int gid) {
|
||
for (int key = 0; key < NekoRay::profileManager->_groups.count(); key++) {
|
||
if (NekoRay::profileManager->_groups[key] == gid) return key;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
void MainWindow::on_tabWidget_currentChanged(int index) {
|
||
if (NekoRay::dataStore->refreshing_group_list) return;
|
||
if (tabIndex2GroupId(index) == NekoRay::dataStore->current_group) return;
|
||
show_group(tabIndex2GroupId(index));
|
||
}
|
||
|
||
void MainWindow::show_group(int gid) {
|
||
auto group = NekoRay::profileManager->GetGroup(gid);
|
||
if (group == nullptr) {
|
||
MessageBoxWarning(tr("Error"), QString("No such group: %1").arg(gid));
|
||
return;
|
||
}
|
||
|
||
if (NekoRay::dataStore->current_group != gid) {
|
||
NekoRay::dataStore->current_group = gid;
|
||
NekoRay::dataStore->Save();
|
||
}
|
||
ui->tabWidget->widget(groupId2TabIndex(gid))->layout()->addWidget(ui->proxyListTable);
|
||
|
||
NekoRay::GroupSortAction gsa;
|
||
gsa.scroll_to_started = true;
|
||
refresh_proxy_list_impl(-1, gsa);
|
||
}
|
||
|
||
// callback
|
||
|
||
void MainWindow::dialog_message_impl(const QString &sender, const QString &info) {
|
||
if (info.contains("UpdateIcon")) {
|
||
icon_status = -1;
|
||
refresh_status();
|
||
}
|
||
if (info.contains("UpdateDataStore")) {
|
||
auto changed = NekoRay::dataStore->Save();
|
||
if (info.contains("RouteChanged")) changed = true;
|
||
refresh_proxy_list();
|
||
if (info.contains("VPNChanged") && title_spmode == NekoRay::SystemProxyMode::VPN) {
|
||
MessageBoxWarning(tr("VPN settings changed"), tr("Restart VPN to take effect."));
|
||
} else if (changed && NekoRay::dataStore->started_id >= 0 &&
|
||
QMessageBox::question(GetMessageBoxParent(), tr("Confirmation"),
|
||
tr("Settings changed, restart proxy?")) == QMessageBox::StandardButton::Yes) {
|
||
neko_start(NekoRay::dataStore->started_id);
|
||
}
|
||
refresh_status();
|
||
}
|
||
if (sender == Dialog_DialogEditProfile) {
|
||
if (info == "accept") {
|
||
refresh_proxy_list();
|
||
}
|
||
} else if (sender == Dialog_DialogManageGroups) {
|
||
if (info.startsWith("refresh")) {
|
||
this->refresh_groups();
|
||
}
|
||
} else if (sender == "SubUpdater") {
|
||
// 订阅完毕
|
||
refresh_proxy_list();
|
||
if (!info.contains("dingyue")) {
|
||
MessageBoxInfo("NekoRay", tr("Imported %1 profile(s)").arg(NekoRay::dataStore->imported_count));
|
||
}
|
||
} else if (sender == "ExternalProcess") {
|
||
if (info == "Crashed") {
|
||
neko_stop();
|
||
}
|
||
}
|
||
}
|
||
|
||
// top bar & tray menu
|
||
|
||
inline bool dialog_is_using = false;
|
||
|
||
#define USE_DIALOG(a) if (dialog_is_using) return; \
|
||
dialog_is_using = true; \
|
||
auto dialog = new a(this); \
|
||
dialog->exec(); \
|
||
dialog->deleteLater(); \
|
||
dialog_is_using = false;
|
||
|
||
void MainWindow::on_menu_basic_settings_triggered() {
|
||
USE_DIALOG(DialogBasicSettings)
|
||
}
|
||
|
||
void MainWindow::on_menu_manage_groups_triggered() {
|
||
USE_DIALOG(DialogManageGroups)
|
||
}
|
||
|
||
void MainWindow::on_menu_routing_settings_triggered() {
|
||
USE_DIALOG(DialogManageRoutes)
|
||
}
|
||
|
||
void MainWindow::on_menu_hotkey_settings_triggered() {
|
||
USE_DIALOG(DialogHotkey)
|
||
}
|
||
|
||
void MainWindow::on_commitDataRequest() {
|
||
qDebug() << "Start of data save";
|
||
if (!isMaximized()) {
|
||
auto olds = NekoRay::dataStore->mw_size;
|
||
auto news = QString("%1x%2").arg(size().width()).arg(size().height());
|
||
if (olds != news) {
|
||
NekoRay::dataStore->mw_size = news;
|
||
NekoRay::dataStore->Save();
|
||
}
|
||
}
|
||
//
|
||
auto last_id = NekoRay::dataStore->started_id;
|
||
neko_stop();
|
||
if (NekoRay::dataStore->remember_enable && last_id >= 0) {
|
||
NekoRay::dataStore->UpdateStartedId(last_id);
|
||
}
|
||
qDebug() << "End of data save";
|
||
}
|
||
|
||
void MainWindow::on_menu_exit_triggered() {
|
||
neko_set_spmode(NekoRay::SystemProxyMode::DISABLE, false);
|
||
if (title_spmode == NekoRay::SystemProxyMode::VPN) return;
|
||
RegisterHotkey(true);
|
||
//
|
||
on_commitDataRequest();
|
||
//
|
||
NekoRay::dataStore->core_prepare_exit = true;
|
||
hide();
|
||
ExitNekorayCore();
|
||
//
|
||
if (exit_update) {
|
||
QDir::setCurrent(QApplication::applicationDirPath());
|
||
QProcess::startDetached("./updater", QStringList{});
|
||
}
|
||
qApp->quit();
|
||
}
|
||
|
||
void MainWindow::neko_set_spmode(int mode, bool save) {
|
||
if (mode != title_spmode) {
|
||
// DISABLE
|
||
|
||
if (title_spmode == NekoRay::SystemProxyMode::SYSTEM_PROXY) {
|
||
ClearSystemProxy();
|
||
} else if (title_spmode == NekoRay::SystemProxyMode::VPN) {
|
||
if (!StopVPNProcess()) {
|
||
refresh_status();
|
||
return;
|
||
}
|
||
}
|
||
|
||
// ENABLE
|
||
|
||
if (mode == NekoRay::SystemProxyMode::SYSTEM_PROXY) {
|
||
#ifdef Q_OS_WIN
|
||
if (mode == NekoRay::SystemProxyMode::SYSTEM_PROXY &&
|
||
!InRange(NekoRay::dataStore->inbound_http_port, 0, 65535)) {
|
||
auto btn = QMessageBox::warning(this, "NekoRay", tr("Http inbound is not enabled, can't set system proxy."),
|
||
"OK", tr("Settings"), "", 0, 0);
|
||
if (btn == 1) {
|
||
on_menu_basic_settings_triggered();
|
||
}
|
||
return;
|
||
}
|
||
#endif
|
||
SetSystemProxy("127.0.0.1",
|
||
NekoRay::dataStore->inbound_http_port,
|
||
NekoRay::dataStore->inbound_socks_port);
|
||
} else if (mode == NekoRay::SystemProxyMode::VPN) {
|
||
if (!StartVPNProcess()) {
|
||
refresh_status();
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (save) {
|
||
NekoRay::dataStore->system_proxy_mode = mode;
|
||
NekoRay::dataStore->Save();
|
||
}
|
||
|
||
title_spmode = mode;
|
||
refresh_status();
|
||
}
|
||
|
||
void MainWindow::refresh_status(const QString &traffic_update) {
|
||
// From TrafficLooper
|
||
if (!traffic_update.isEmpty()) {
|
||
traffic_update_cache = traffic_update;
|
||
if (traffic_update == "STOP") {
|
||
traffic_update_cache = "";
|
||
} else {
|
||
ui->label_speed->setText(traffic_update);
|
||
return;
|
||
}
|
||
}
|
||
ui->label_speed->setText(traffic_update_cache);
|
||
|
||
// From UI
|
||
if (last_test_time.addSecs(2) < QTime::currentTime()) {
|
||
ui->label_running->setText(tr("Running: %1").arg(running.isNull()
|
||
? tr("None")
|
||
: running->bean->DisplayName().left(50)));
|
||
}
|
||
//
|
||
auto display_http = tr("None");
|
||
if (InRange(NekoRay::dataStore->inbound_http_port, 0, 65535)) {
|
||
display_http = DisplayAddress(NekoRay::dataStore->inbound_address, NekoRay::dataStore->inbound_http_port);
|
||
}
|
||
auto inbound_txt = QString("Socks: %1\nHTTP: %2").arg(
|
||
DisplayAddress(NekoRay::dataStore->inbound_address, NekoRay::dataStore->inbound_socks_port),
|
||
display_http
|
||
);
|
||
ui->label_inbound->setText(inbound_txt);
|
||
//
|
||
ui->checkBox_VPN->setChecked(title_spmode == NekoRay::SystemProxyMode::VPN);
|
||
if (select_mode) ui->label_running->setText("[" + tr("Select") + "]");
|
||
|
||
auto make_title = [=](bool isTray) {
|
||
QStringList tt;
|
||
if (select_mode) tt << "[" + tr("Select") + "]";
|
||
if (!title_error.isEmpty()) tt << "[" + title_error + "]";
|
||
if (title_spmode == NekoRay::SystemProxyMode::SYSTEM_PROXY) {
|
||
tt << "[" + tr("System Proxy") + "]";
|
||
} else if (title_spmode == NekoRay::SystemProxyMode::VPN) {
|
||
tt << "[" + tr("VPN Mode") + "]";
|
||
}
|
||
tt << title_base;
|
||
if (!isTray) tt << "(" + QString(NKR_VERSION) + ")";
|
||
if (!NekoRay::dataStore->active_routing.isEmpty() && NekoRay::dataStore->active_routing != "Default") {
|
||
tt << "[" + NekoRay::dataStore->active_routing + "]";
|
||
}
|
||
if (!running.isNull()) tt << running->bean->DisplayTypeAndName();
|
||
return tt.join(isTray ? "\n" : " ");
|
||
};
|
||
|
||
auto icon_status_new = TrayIcon::NONE;
|
||
if (title_spmode == NekoRay::SystemProxyMode::SYSTEM_PROXY) {
|
||
icon_status_new = TrayIcon::SYSTEM_PROXY;
|
||
} else if (title_spmode == NekoRay::SystemProxyMode::VPN) {
|
||
icon_status_new = TrayIcon::VPN;
|
||
} else if (!running.isNull()) {
|
||
icon_status_new = TrayIcon::RUNNING;
|
||
}
|
||
|
||
setWindowTitle(make_title(false));
|
||
if (icon_status_new != icon_status) setWindowIcon(TrayIcon::GetIcon(TrayIcon::NONE));
|
||
|
||
if (tray != nullptr) {
|
||
tray->setToolTip(make_title(true));
|
||
if (icon_status_new != icon_status) tray->setIcon(TrayIcon::GetIcon(icon_status_new));
|
||
}
|
||
|
||
icon_status = icon_status_new;
|
||
}
|
||
|
||
// table显示
|
||
|
||
// refresh_groups -> show_group -> refresh_proxy_list
|
||
void MainWindow::refresh_groups() {
|
||
NekoRay::dataStore->refreshing_group_list = true;
|
||
|
||
// refresh group?
|
||
for (int i = ui->tabWidget->count() - 1; i > 0; i--) {
|
||
ui->tabWidget->removeTab(i);
|
||
}
|
||
|
||
int index = 0;
|
||
for (const auto &gid: NekoRay::profileManager->_groups) {
|
||
auto group = NekoRay::profileManager->GetGroup(gid);
|
||
if (index == 0) {
|
||
ui->tabWidget->setTabText(0, group->name);
|
||
} else {
|
||
auto widget2 = new QWidget();
|
||
auto layout2 = new QVBoxLayout();
|
||
layout2->setContentsMargins(QMargins());
|
||
layout2->setSpacing(0);
|
||
widget2->setLayout(layout2);
|
||
ui->tabWidget->addTab(widget2, group->name);
|
||
}
|
||
ui->tabWidget->tabBar()->setTabData(index, gid);
|
||
index++;
|
||
}
|
||
|
||
// show after group changed
|
||
if (NekoRay::profileManager->CurrentGroup() == nullptr) {
|
||
NekoRay::dataStore->current_group = -1;
|
||
ui->tabWidget->setCurrentIndex(groupId2TabIndex(0));
|
||
show_group(NekoRay::profileManager->_groups.count() > 0 ? NekoRay::profileManager->_groups.first() : 0);
|
||
} else {
|
||
ui->tabWidget->setCurrentIndex(groupId2TabIndex(NekoRay::dataStore->current_group));
|
||
show_group(NekoRay::dataStore->current_group);
|
||
}
|
||
|
||
NekoRay::dataStore->refreshing_group_list = false;
|
||
}
|
||
|
||
|
||
void MainWindow::refresh_proxy_list_impl(const int &id, NekoRay::GroupSortAction groupSortAction) {
|
||
if (id < 0) {
|
||
//这样才能清空数据
|
||
ui->proxyListTable->setRowCount(0);
|
||
}
|
||
|
||
QTableWidgetItem *started_item = nullptr;
|
||
|
||
// 绘制或更新item(s)
|
||
int row = -1;
|
||
for (const auto &profile: NekoRay::profileManager->profiles) {
|
||
if (NekoRay::dataStore->current_group != profile->gid) continue;
|
||
|
||
row++;
|
||
if (id >= 0 && profile->id != id) continue; // update only one item
|
||
if (id < 0) {
|
||
ui->proxyListTable->insertRow(row);
|
||
} else {
|
||
// 排序过的，要找
|
||
row = ui->proxyListTable->id2Row[id];
|
||
}
|
||
|
||
auto f0 = std::make_unique<QTableWidgetItem>();
|
||
// auto font = f0->font();
|
||
// font.setPointSize(9);
|
||
// f0->setFont(font);
|
||
f0->setData(114514, profile->id);
|
||
|
||
// C0: is Running
|
||
auto f = f0->clone();
|
||
if (profile->id == NekoRay::dataStore->started_id) {
|
||
f->setText("✓");
|
||
if (groupSortAction.scroll_to_started) started_item = f;
|
||
} else {
|
||
f->setText(" ");
|
||
}
|
||
ui->proxyListTable->setItem(row, 0, f);
|
||
|
||
// C1: Type
|
||
f = f0->clone();
|
||
f->setText(profile->bean->DisplayType());
|
||
auto insecure_hint = DisplayInsecureHint(profile->bean);
|
||
if (!insecure_hint.isEmpty()) {
|
||
f->setBackground(Qt::red);
|
||
f->setToolTip(insecure_hint);
|
||
}
|
||
ui->proxyListTable->setItem(row, 1, f);
|
||
|
||
// C2: Address+Port
|
||
f = f0->clone();
|
||
f->setText(profile->bean->DisplayAddress());
|
||
ui->proxyListTable->setItem(row, 2, f);
|
||
|
||
// C3: Name
|
||
f = f0->clone();
|
||
f->setText(profile->bean->name);
|
||
ui->proxyListTable->setItem(row, 3, f);
|
||
|
||
// C4: Test Result
|
||
f = f0->clone();
|
||
f->setText(profile->DisplayLatency());
|
||
if (!profile->full_test_report.isEmpty()) f->setText(profile->full_test_report);
|
||
ui->proxyListTable->setItem(row, 4, f);
|
||
|
||
// C5: Traffic
|
||
f = f0->clone();
|
||
f->setText(profile->traffic_data->DisplayTraffic());
|
||
ui->proxyListTable->setItem(row, 5, f);
|
||
}
|
||
|
||
// 显示排序
|
||
if (id < 0) {
|
||
switch (groupSortAction.method) {
|
||
case NekoRay::GroupSortMethod::Raw: {
|
||
auto group = NekoRay::profileManager->CurrentGroup();
|
||
if (group == nullptr) return;
|
||
ui->proxyListTable->order = group->order;
|
||
break;
|
||
}
|
||
case NekoRay::GroupSortMethod::ById: {
|
||
std::sort(ui->proxyListTable->order.begin(), ui->proxyListTable->order.end(),
|
||
[=](int a, int b) {
|
||
if (groupSortAction.descending) {
|
||
return a > b;
|
||
} else {
|
||
return a < b;
|
||
}
|
||
});
|
||
break;
|
||
}
|
||
case NekoRay::GroupSortMethod::ByAddress:
|
||
case NekoRay::GroupSortMethod::ByName:
|
||
case NekoRay::GroupSortMethod::ByType: {
|
||
std::sort(ui->proxyListTable->order.begin(), ui->proxyListTable->order.end(),
|
||
[=](int a, int b) {
|
||
QString ms_a;
|
||
QString ms_b;
|
||
if (groupSortAction.method == NekoRay::GroupSortMethod::ByType) {
|
||
ms_a = NekoRay::profileManager->GetProfile(a)->bean->DisplayType();
|
||
ms_b = NekoRay::profileManager->GetProfile(b)->bean->DisplayType();
|
||
} else if (groupSortAction.method == NekoRay::GroupSortMethod::ByName) {
|
||
ms_a = NekoRay::profileManager->GetProfile(a)->bean->name;
|
||
ms_b = NekoRay::profileManager->GetProfile(b)->bean->name;
|
||
} else if (groupSortAction.method == NekoRay::GroupSortMethod::ByAddress) {
|
||
ms_a = NekoRay::profileManager->GetProfile(a)->bean->DisplayAddress();
|
||
ms_b = NekoRay::profileManager->GetProfile(b)->bean->DisplayAddress();
|
||
}
|
||
if (groupSortAction.descending) {
|
||
return ms_a > ms_b;
|
||
} else {
|
||
return ms_a < ms_b;
|
||
}
|
||
});
|
||
break;
|
||
}
|
||
case NekoRay::GroupSortMethod::ByLatency: {
|
||
std::sort(ui->proxyListTable->order.begin(), ui->proxyListTable->order.end(),
|
||
[=](int a, int b) {
|
||
auto ms_a = NekoRay::profileManager->GetProfile(a)->latency;
|
||
auto ms_b = NekoRay::profileManager->GetProfile(b)->latency;
|
||
if (ms_a <= 0) ms_a = 114514;
|
||
if (ms_b <= 0) ms_b = 114514;
|
||
if (groupSortAction.descending) {
|
||
return ms_a > ms_b;
|
||
} else {
|
||
return ms_a < ms_b;
|
||
}
|
||
});
|
||
break;
|
||
}
|
||
}
|
||
ui->proxyListTable->update_order(groupSortAction.save_sort);
|
||
}
|
||
|
||
if (started_item != nullptr) {
|
||
runOnUiThread([=] {
|
||
ui->proxyListTable->verticalScrollBar()->setSliderPosition(started_item->row());
|
||
});
|
||
}
|
||
}
|
||
|
||
// table菜单相关
|
||
|
||
void MainWindow::on_proxyListTable_itemDoubleClicked(QTableWidgetItem *item) {
|
||
auto id = item->data(114514).toInt();
|
||
if (select_mode) {
|
||
emit profile_selected(id);
|
||
select_mode = false;
|
||
refresh_status();
|
||
return;
|
||
}
|
||
auto dialog = new DialogEditProfile("", id, this);
|
||
connect(dialog, &QDialog::finished, dialog, &QDialog::deleteLater);
|
||
}
|
||
|
||
#define NO_ADD_TO_SUBSCRIPTION_GROUP \
|
||
if (!NekoRay::profileManager->CurrentGroup()->url.isEmpty()) { \
|
||
MessageBoxWarning("NekoRay", MainWindow::tr("Manual addition of profiles is not allowed in subscription groupings.")); \
|
||
return; \
|
||
}
|
||
|
||
void MainWindow::on_menu_add_from_input_triggered() {
|
||
NO_ADD_TO_SUBSCRIPTION_GROUP
|
||
auto dialog = new DialogEditProfile("socks", NekoRay::dataStore->current_group, this);
|
||
connect(dialog, &QDialog::finished, dialog, &QDialog::deleteLater);
|
||
}
|
||
|
||
void MainWindow::on_menu_add_from_clipboard_triggered() {
|
||
NO_ADD_TO_SUBSCRIPTION_GROUP
|
||
auto clipboard = QApplication::clipboard()->text();
|
||
NekoRay::sub::groupUpdater->AsyncUpdate(clipboard);
|
||
}
|
||
|
||
void MainWindow::on_menu_clone_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.isEmpty()) return;
|
||
|
||
auto btn = QMessageBox::question(this, tr("Clone"), tr("Clone %1 item(s)").arg(ents.count()));
|
||
if (btn != QMessageBox::Yes) return;
|
||
|
||
QStringList sls;
|
||
for (const auto &ent: ents) {
|
||
sls << ent->bean->ToNekorayShareLink(ent->type);
|
||
}
|
||
|
||
NekoRay::sub::groupUpdater->AsyncUpdate(sls.join("\n"));
|
||
}
|
||
|
||
void MainWindow::on_menu_move_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.isEmpty()) return;
|
||
|
||
auto items = QStringList{};
|
||
for (auto &&group: NekoRay::profileManager->groups) {
|
||
items += Int2String(group->id) + " " + group->name;
|
||
}
|
||
|
||
bool ok;
|
||
auto a = QInputDialog::getItem(nullptr,
|
||
tr("Move"),
|
||
tr("Move %1 item(s)").arg(ents.count()),
|
||
items, 0, false, &ok);
|
||
if (!ok) return;
|
||
auto gid = SubStrBefore(a, " ").toInt();
|
||
for (const auto &ent: ents) {
|
||
NekoRay::profileManager->MoveProfile(ent, gid);
|
||
}
|
||
refresh_proxy_list();
|
||
}
|
||
|
||
void MainWindow::on_menu_delete_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.count() == 0) return;
|
||
if (QMessageBox::question(this, tr("Confirmation"), QString(tr("Remove %1 item(s) ?")).arg(ents.count())) ==
|
||
QMessageBox::StandardButton::Yes) {
|
||
for (const auto &ent: ents) {
|
||
NekoRay::profileManager->DeleteProfile(ent->id);
|
||
}
|
||
refresh_proxy_list();
|
||
}
|
||
}
|
||
|
||
void MainWindow::on_menu_reset_traffic_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.count() == 0) return;
|
||
if (QMessageBox::question(this,
|
||
tr("Confirmation"),
|
||
QString(tr("Reset traffic of %1 item(s) ?")).arg(ents.count()))
|
||
== QMessageBox::StandardButton::Yes) {
|
||
for (const auto &ent: ents) {
|
||
ent->traffic_data->Reset();
|
||
ent->Save();
|
||
refresh_proxy_list(ent->id);
|
||
}
|
||
}
|
||
}
|
||
|
||
void MainWindow::on_menu_profile_debug_info_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.count() != 1) return;
|
||
auto btn = QMessageBox::information(this, "NekoRay", ents.first()->ToJsonBytes(), "OK", "Edit", "Reload", 0, 0);
|
||
if (btn == 1) {
|
||
QDesktopServices::openUrl(QUrl::fromLocalFile(
|
||
QFileInfo(QString("profiles/%1.json").arg(ents.first()->id)).absoluteFilePath()
|
||
));
|
||
} else if (btn == 2) {
|
||
ents.first()->Load();
|
||
refresh_proxy_list();
|
||
}
|
||
}
|
||
|
||
void MainWindow::on_menu_copy_links_triggered() {
|
||
auto ents = get_now_selected();
|
||
QStringList links;
|
||
for (const auto &ent: ents) {
|
||
links += ent->bean->ToShareLink();
|
||
}
|
||
if (links.length() == 0) return;
|
||
QApplication::clipboard()->setText(links.join("\n"));
|
||
MessageBoxInfo("NekoRay", tr("Copied %1 item(s)").arg(links.length()));
|
||
}
|
||
|
||
void MainWindow::on_menu_copy_links_nkr_triggered() {
|
||
auto ents = get_now_selected();
|
||
QStringList links;
|
||
for (const auto &ent: ents) {
|
||
links += ent->bean->ToNekorayShareLink(ent->type);
|
||
}
|
||
if (links.length() == 0) return;
|
||
QApplication::clipboard()->setText(links.join("\n"));
|
||
MessageBoxInfo("NekoRay", tr("Copied %1 item(s)").arg(links.length()));
|
||
}
|
||
|
||
void MainWindow::on_menu_export_config_triggered() {
|
||
auto ents = get_now_selected();
|
||
if (ents.count() != 1) return;
|
||
auto ent = ents.first();
|
||
auto result = NekoRay::BuildConfig(ent, false);
|
||
auto config_core = QJsonObject2QString(result->coreConfig, true);
|
||
QApplication::clipboard()->setText(config_core);
|
||
MessageBoxWarning(tr("Config copied"), config_core);
|
||
}
|
||
|
||
void MainWindow::display_qr_link(bool nkrFormat) {
|
||
auto ents = get_now_selected();
|
||
if (ents.count() != 1) return;
|
||
|
||
auto link = ents.first()->bean->ToShareLink();
|
||
if (nkrFormat) {
|
||
link = ents.first()->bean->ToNekorayShareLink(ents.first()->type);
|
||
}
|
||
|
||
qrcodegen::QrCode qr = qrcodegen::QrCode::encodeText(link.toUtf8().data(),
|
||
qrcodegen::QrCode::Ecc::MEDIUM);
|
||
qint32 sz = qr.getSize();
|
||
QImage im(sz, sz, QImage::Format_RGB32);
|
||
QRgb black = qRgb(0, 0, 0);
|
||
QRgb white = qRgb(255, 255, 255);
|
||
for (int y = 0; y < sz; y++)
|
||
for (int x = 0; x < sz; x++)
|
||
im.setPixel(x, y, qr.getModule(x, y) ? black : white);
|
||
|
||
class W : public QDialog {
|
||
public:
|
||
QLabel *l = nullptr;
|
||
QPlainTextEdit *l2 = nullptr;
|
||
QImage im;
|
||
|
||
void set(QLabel *qLabel, QPlainTextEdit *pl, QImage qImage) {
|
||
this->l = qLabel;
|
||
this->l2 = pl;
|
||
this->im = std::move(qImage);
|
||
}
|
||
|
||
void resizeEvent(QResizeEvent *resizeEvent) override {
|
||
auto size = resizeEvent->size();
|
||
auto side = size.height() - 20 - l2->size().height();
|
||
l->setPixmap(QPixmap::fromImage(im.scaled(side, side, Qt::KeepAspectRatio, Qt::FastTransformation),
|
||
Qt::MonoOnly));
|
||
l->resize(side, side);
|
||
}
|
||
};
|
||
|
||
auto w = new W();
|
||
auto l = new QLabel(w);
|
||
w->setLayout(new QVBoxLayout);
|
||
w->setMinimumSize(256, 256);
|
||
l->setMinimumSize(256, 256);
|
||
l->setMargin(6);
|
||
l->setAlignment(Qt::AlignmentFlag::AlignCenter);
|
||
l->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
|
||
l->setScaledContents(true);
|
||
w->layout()->addWidget(l);
|
||
auto l2 = new QPlainTextEdit(w);
|
||
l2->setPlainText(link);
|
||
l2->setReadOnly(true);
|
||
w->layout()->addWidget(l2);
|
||
w->set(l, l2, im);
|
||
w->setWindowTitle(ents.first()->bean->DisplayTypeAndName());
|
||
QSizePolicy sizePolicy(QSizePolicy::Preferred, QSizePolicy::Preferred);
|
||
sizePolicy.setHeightForWidth(true);
|
||
w->setSizePolicy(sizePolicy);
|
||
w->exec();
|
||
w->deleteLater();
|
||
}
|
||
|
||
void MainWindow::on_menu_scan_qr_triggered() {
|
||
NO_ADD_TO_SUBSCRIPTION_GROUP
|
||
#ifndef NKR_NO_EXTERNAL
|
||
using namespace ZXingQt;
|
||
|
||
hide();
|
||
QThread::sleep(1);
|
||
|
||
auto screen = QGuiApplication::primaryScreen();
|
||
auto geom = screen->geometry();
|
||
auto qpx = screen->grabWindow(0, geom.x(), geom.y(), geom.width(), geom.height());
|
||
|
||
show();
|
||
|
||
auto hints = DecodeHints()
|
||
.setFormats(BarcodeFormat::QRCode)
|
||
.setTryRotate(false)
|
||
.setBinarizer(Binarizer::FixedThreshold);
|
||
|
||
auto result = ReadBarcode(qpx.toImage(), hints);
|
||
const auto &text = result.text();
|
||
if (text.isEmpty()) {
|
||
MessageBoxInfo("NekoRay", tr("QR Code not found"));
|
||
} else {
|
||
NekoRay::sub::groupUpdater->AsyncUpdate(text);
|
||
}
|
||
#endif
|
||
}
|
||
|
||
void MainWindow::on_menu_clear_test_result_triggered() {
|
||
for (const auto &profile: NekoRay::profileManager->profiles) {
|
||
if (NekoRay::dataStore->current_group != profile->gid) continue;
|
||
profile->latency = 0;
|
||
profile->full_test_report = "";
|
||
}
|
||
refresh_proxy_list();
|
||
}
|
||
|
||
void MainWindow::on_menu_select_all_triggered() {
|
||
ui->proxyListTable->selectAll();
|
||
}
|
||
|
||
void MainWindow::on_menu_delete_repeat_triggered() {
|
||
QList<QSharedPointer<NekoRay::ProxyEntity>> out;
|
||
QList<QSharedPointer<NekoRay::ProxyEntity>> out_del;
|
||
|
||
NekoRay::ProfileFilter::Uniq(NekoRay::profileManager->CurrentGroup()->Profiles(), out, true, false);
|
||
NekoRay::ProfileFilter::OnlyInSrc_ByPointer(NekoRay::profileManager->CurrentGroup()->Profiles(), out, out_del);
|
||
|
||
QString remove_display;
|
||
for (const auto &ent: out_del) {
|
||
remove_display += ent->bean->DisplayTypeAndName() + "\n";
|
||
}
|
||
if (out_del.length() > 0 &&
|
||
QMessageBox::question(this,
|
||
tr("Confirmation"),
|
||
tr("Remove %1 item(s) ?").arg(out_del.length()) + "\n" + remove_display
|
||
) == QMessageBox::StandardButton::Yes) {
|
||
|
||
for (const auto &ent: out_del) {
|
||
NekoRay::profileManager->DeleteProfile(ent->id);
|
||
}
|
||
refresh_proxy_list();
|
||
}
|
||
}
|
||
|
||
bool mw_sub_updating = false;
|
||
|
||
void MainWindow::on_menu_update_subscripton_triggered() {
|
||
auto group = NekoRay::profileManager->CurrentGroup();
|
||
if (group->url.isEmpty()) return;
|
||
if (mw_sub_updating) return;
|
||
mw_sub_updating = true;
|
||
NekoRay::sub::groupUpdater->AsyncUpdate(group->url, group->id, this, [=] {
|
||
mw_sub_updating = false;
|
||
});
|
||
}
|
||
|
||
void MainWindow::on_menu_remove_unavailable_triggered() {
|
||
QList<QSharedPointer<NekoRay::ProxyEntity>> out_del;
|
||
|
||
for (const auto &profile: NekoRay::profileManager->profiles) {
|
||
if (NekoRay::dataStore->current_group != profile->gid) continue;
|
||
if (profile->latency < 0) out_del += profile;
|
||
}
|
||
|
||
QString remove_display;
|
||
for (const auto &ent: out_del) {
|
||
remove_display += ent->bean->DisplayTypeAndName() + "\n";
|
||
}
|
||
if (out_del.length() > 0 &&
|
||
QMessageBox::question(this,
|
||
tr("Confirmation"),
|
||
tr("Remove %1 item(s) ?").arg(out_del.length()) + "\n" + remove_display
|
||
) == QMessageBox::StandardButton::Yes) {
|
||
|
||
for (const auto &ent: out_del) {
|
||
NekoRay::profileManager->DeleteProfile(ent->id);
|
||
}
|
||
refresh_proxy_list();
|
||
}
|
||
}
|
||
|
||
void MainWindow::on_proxyListTable_customContextMenuRequested(const QPoint &pos) {
|
||
ui->menu_server->popup(ui->proxyListTable->viewport()->mapToGlobal(pos)); //弹出菜单
|
||
}
|
||
|
||
QMap<int, QSharedPointer<NekoRay::ProxyEntity>> MainWindow::get_now_selected() {
|
||
auto items = ui->proxyListTable->selectedItems();
|
||
QMap<int, QSharedPointer<NekoRay::ProxyEntity>> map;
|
||
for (auto item: items) {
|
||
auto id = item->data(114514).toInt();
|
||
auto ent = NekoRay::profileManager->GetProfile(id);
|
||
if (ent != nullptr) map[id] = ent;
|
||
}
|
||
return map;
|
||
}
|
||
|
||
void MainWindow::keyPressEvent(QKeyEvent *event) {
|
||
switch (event->key()) {
|
||
case Qt::Key_Escape:
|
||
// take over by shortcut_esc
|
||
break;
|
||
default:
|
||
QMainWindow::keyPressEvent(event);
|
||
}
|
||
}
|
||
|
||
// Log
|
||
|
||
inline void FastAppendTextDocument(const QString &message, QTextDocument *doc) {
|
||
QTextCursor cursor(doc);
|
||
cursor.movePosition(QTextCursor::End);
|
||
cursor.beginEditBlock();
|
||
cursor.insertBlock();
|
||
cursor.insertText(message);
|
||
cursor.endEditBlock();
|
||
}
|
||
|
||
void MainWindow::show_log_impl(const QString &log) {
|
||
if (log.trimmed().isEmpty()) return;
|
||
|
||
FastAppendTextDocument(log, qvLogDocument);
|
||
// qvLogDocument->setPlainText(qvLogDocument->toPlainText() + log);
|
||
// From https://gist.github.com/jemyzhang/7130092
|
||
auto maxLines = 200;
|
||
auto block = qvLogDocument->begin();
|
||
|
||
while (block.isValid()) {
|
||
if (qvLogDocument->blockCount() > maxLines) {
|
||
QTextCursor cursor(block);
|
||
block = block.next();
|
||
cursor.select(QTextCursor::BlockUnderCursor);
|
||
cursor.movePosition(QTextCursor::NextCharacter, QTextCursor::KeepAnchor);
|
||
cursor.removeSelectedText();
|
||
continue;
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
|
||
void MainWindow::on_masterLogBrowser_customContextMenuRequested(const QPoint &pos) {
|
||
QMenu *menu = ui->masterLogBrowser->createStandardContextMenu();
|
||
|
||
auto sep = new QAction;
|
||
sep->setSeparator(true);
|
||
menu->addAction(sep);
|
||
|
||
auto action_clear = new QAction;
|
||
action_clear->setText(tr("Clear"));
|
||
action_clear->setData(-1);
|
||
connect(action_clear, &QAction::triggered, this, [=] {
|
||
qvLogDocument->clear();
|
||
});
|
||
menu->addAction(action_clear);
|
||
|
||
menu->exec(ui->masterLogBrowser->viewport()->mapToGlobal(pos)); //弹出菜单
|
||
}
|
||
|
||
// eventFilter
|
||
|
||
bool MainWindow::eventFilter(QObject *obj, QEvent *event) {
|
||
if (event->type() == QEvent::MouseButtonPress) {
|
||
auto mouseEvent = dynamic_cast<QMouseEvent *>(event);
|
||
|
||
if (obj == ui->label_running && mouseEvent->button() == Qt::LeftButton && running != nullptr) {
|
||
test_current();
|
||
return true;
|
||
} else if (obj == ui->label_inbound && mouseEvent->button() == Qt::LeftButton) {
|
||
on_menu_basic_settings_triggered();
|
||
return true;
|
||
}
|
||
}
|
||
return QMainWindow::eventFilter(obj, event);
|
||
}
|
||
|
||
// profile selector
|
||
|
||
void MainWindow::start_select_mode(QObject *context, const std::function<void(int)> &callback) {
|
||
select_mode = true;
|
||
connectOnce(this, &MainWindow::profile_selected, context, callback);
|
||
refresh_status();
|
||
}
|
||
|
||
// 连接列表
|
||
|
||
inline QJsonArray last_arr;
|
||
|
||
void MainWindow::refresh_connection_list(const QJsonArray &arr) {
|
||
if (last_arr == arr) {
|
||
return;
|
||
}
|
||
last_arr = arr;
|
||
|
||
ui->tableWidget_conn->setRowCount(0);
|
||
|
||
int row = -1;
|
||
for (const auto &_item: arr) {
|
||
auto item = _item.toObject();
|
||
|
||
row++;
|
||
ui->tableWidget_conn->insertRow(row);
|
||
|
||
// C0: Status
|
||
auto *f = new QTableWidgetItem("");
|
||
f->setData(114514, item["ID"].toInt());
|
||
auto start_t = item["Start"].toInt();
|
||
auto end_t = item["End"].toInt();
|
||
if (end_t > 0) {
|
||
f->setText(tr("End"));
|
||
} else {
|
||
f->setText(tr("Active"));
|
||
}
|
||
f->setToolTip(tr("Start: %1\nEnd: %2").arg(
|
||
DisplayTime(start_t),
|
||
end_t > 0 ? DisplayTime(end_t) : ""
|
||
));
|
||
ui->tableWidget_conn->setItem(row, 0, f);
|
||
|
||
// C1: Outbound
|
||
f = f->clone();
|
||
f->setToolTip("");
|
||
f->setText(item["Tag"].toString());
|
||
ui->tableWidget_conn->setItem(row, 1, f);
|
||
|
||
// C2: Destination
|
||
f = f->clone();
|
||
QString target1 = item["Dest"].toString();
|
||
QString target2 = item["RDest"].toString();
|
||
if (target2.isEmpty() || target1 == target2) {
|
||
target2 = "";
|
||
}
|
||
f->setText("[" + target1 + "] " + target2);
|
||
ui->tableWidget_conn->setItem(row, 2, f);
|
||
}
|
||
}
|
||
|
||
|
||
// Hotkey
|
||
|
||
#ifndef NKR_NO_EXTERNAL
|
||
|
||
#include <QHotkey>
|
||
|
||
inline QList<QSharedPointer<QHotkey>> RegisteredHotkey;
|
||
|
||
void MainWindow::RegisterHotkey(bool unregister) {
|
||
while (!RegisteredHotkey.isEmpty()) {
|
||
auto hk = RegisteredHotkey.takeFirst();
|
||
hk->deleteLater();
|
||
}
|
||
if (unregister) return;
|
||
|
||
QStringList regstr;
|
||
regstr += NekoRay::dataStore->hotkey_mainwindow;
|
||
regstr += NekoRay::dataStore->hotkey_group;
|
||
regstr += NekoRay::dataStore->hotkey_route;
|
||
|
||
for (const auto &key: regstr) {
|
||
if (key.isEmpty()) continue;
|
||
if (regstr.count(key) > 1) return; // Conflict hotkey
|
||
}
|
||
for (const auto &key: regstr) {
|
||
QKeySequence k(key);
|
||
if (k.isEmpty()) continue;
|
||
auto hk = QSharedPointer<QHotkey>(new QHotkey(k, true));
|
||
if (hk->isRegistered()) {
|
||
RegisteredHotkey += hk;
|
||
connect(hk.get(), &QHotkey::activated, this, [=] { HotkeyEvent(key); });
|
||
} else {
|
||
hk->deleteLater();
|
||
}
|
||
}
|
||
}
|
||
|
||
void MainWindow::HotkeyEvent(const QString &key) {
|
||
if (key.isEmpty()) return;
|
||
runOnUiThread([=] {
|
||
if (key == NekoRay::dataStore->hotkey_mainwindow) {
|
||
tray->activated(QSystemTrayIcon::ActivationReason::Trigger);
|
||
} else if (key == NekoRay::dataStore->hotkey_group) {
|
||
on_menu_manage_groups_triggered();
|
||
} else if (key == NekoRay::dataStore->hotkey_route) {
|
||
on_menu_routing_settings_triggered();
|
||
}
|
||
});
|
||
}
|
||
|
||
#else
|
||
|
||
void MainWindow::RegisterHotkey(bool unregister) {}
|
||
|
||
void MainWindow::HotkeyEvent(const QString &key) {}
|
||
|
||
#endif
|
||
|
||
// VPN Launcher
|
||
|
||
//inline QString mshta_exec_uac = R"(vbscript:CreateObject("Shell.Application").ShellExecute("%1","%2","","runas",1)(window.close))";
|
||
|
||
bool MainWindow::StartVPNProcess() {
|
||
//
|
||
if (vpn_pid != 0) {
|
||
return true;
|
||
}
|
||
#ifdef Q_OS_WIN
|
||
auto configFn = ":/nekoray/vpn/sing-box-vpn.json";
|
||
if (QFile::exists("vpn/sing-box-vpn.json")) configFn = "vpn/sing-box-vpn.json";
|
||
auto config = ReadFileText(configFn)
|
||
.replace("%IPV6_ADDRESS%", NekoRay::dataStore->vpn_ipv6 ? "\"inet6_address\": \"fdfe:dcba:9876::1/126\"," : "")
|
||
.replace("%MTU%", Int2String(NekoRay::dataStore->vpn_mtu))
|
||
.replace("%STACK%", Preset::SingBox::VpnImplementation[NekoRay::dataStore->vpn_implementation])
|
||
.replace("%PORT%", Int2String(NekoRay::dataStore->inbound_socks_port));
|
||
#else
|
||
auto protectPath = QDir::currentPath() + "/protect";
|
||
auto configFn = ":/nekoray/vpn/vpn-run-root.sh";
|
||
if (QFile::exists("vpn/vpn-run-root.sh")) configFn = "vpn/vpn-run-root.sh";
|
||
auto config = ReadFileText(configFn)
|
||
.replace("$PORT", Int2String(NekoRay::dataStore->inbound_socks_port))
|
||
.replace("$USE_NEKORAY", "1")
|
||
.replace("$ENABLE_IPV6", NekoRay::dataStore->vpn_ipv6 ? "1" : "")
|
||
.replace("./nekoray_core", QApplication::applicationDirPath() + "/nekoray_core")
|
||
.replace("./tun2socks", QApplication::applicationDirPath() + "/tun2socks")
|
||
.replace("$PROTECT_LISTEN_PATH", protectPath)
|
||
.replace("$TUN_NAME", "nekoray-tun")
|
||
.replace("$USER_ID", Int2String((int) getuid()))
|
||
.replace("$MTU", Int2String(NekoRay::dataStore->vpn_mtu))
|
||
.replace("$TABLE_FWMARK", "514");
|
||
#endif
|
||
//
|
||
QFile file;
|
||
file.setFileName(QFileInfo(configFn).fileName());
|
||
file.open(QIODevice::ReadWrite | QIODevice::Truncate);
|
||
file.write(config.toUtf8());
|
||
file.close();
|
||
auto configPath = QFileInfo(file).absoluteFilePath();
|
||
//
|
||
#ifdef Q_OS_WIN
|
||
runOnNewThread([=] {
|
||
vpn_pid = 1; //TODO get pid?
|
||
WinCommander::runProcessElevated(QApplication::applicationDirPath() + "/sing-box.exe",
|
||
{"--disable-color", "run", "-c", configPath}); // blocking
|
||
vpn_pid = 0;
|
||
runOnUiThread([=] {
|
||
neko_set_spmode(NekoRay::SystemProxyMode::DISABLE);
|
||
});
|
||
});
|
||
#else
|
||
QFile::remove(protectPath);
|
||
if (QFile::exists(protectPath)) {
|
||
MessageBoxWarning("Error", "protect cannot be removed");
|
||
return false;
|
||
}
|
||
//
|
||
auto vpn_process = new QProcess;
|
||
QProcess::connect(vpn_process, &QProcess::stateChanged, mainwindow, [=](QProcess::ProcessState state) {
|
||
if (state == QProcess::NotRunning) {
|
||
if (watcher != nullptr) {
|
||
watcher->deleteLater();
|
||
watcher = nullptr;
|
||
}
|
||
vpn_pid = 0;
|
||
vpn_process->deleteLater();
|
||
GetMainWindow()->neko_set_spmode(NekoRay::SystemProxyMode::DISABLE);
|
||
}
|
||
});
|
||
if (watcher != nullptr) {
|
||
watcher->deleteLater();
|
||
watcher = nullptr;
|
||
}
|
||
watcher = new QFileSystemWatcher({QDir::currentPath()});
|
||
connect(watcher, &QFileSystemWatcher::directoryChanged, this, [=] {
|
||
if (!QFile::exists(protectPath)) return;
|
||
if (!Tun2rayStartStop(true)) {
|
||
neko_set_spmode(NekoRay::SystemProxyMode::DISABLE);
|
||
}
|
||
if (watcher != nullptr) {
|
||
watcher->deleteLater();
|
||
watcher = nullptr;
|
||
}
|
||
});
|
||
//
|
||
vpn_process->setProcessChannelMode(QProcess::ForwardedChannels);
|
||
vpn_process->start("pkexec", {"bash", configPath});
|
||
vpn_process->waitForStarted();
|
||
vpn_pid = vpn_process->processId(); // actually it's pkexec or bash PID
|
||
#endif
|
||
return true;
|
||
}
|
||
|
||
bool MainWindow::StopVPNProcess() {
|
||
if (vpn_pid != 0) {
|
||
bool ok;
|
||
#ifdef Q_OS_WIN
|
||
auto ret = WinCommander::runProcessElevated("taskkill", {"/IM", "sing-box.exe"});
|
||
ok = ret == 0;
|
||
#else
|
||
QProcess p;
|
||
p.start("pkexec", {"pkill", "-2", "-P", Int2String(vpn_pid)});
|
||
p.waitForFinished();
|
||
ok = p.exitCode() == 0;
|
||
#endif
|
||
if (ok) {
|
||
vpn_pid = 0;
|
||
Tun2rayStartStop(false);
|
||
} else {
|
||
MessageBoxWarning(tr("Error"), tr("Failed to stop VPN process"));
|
||
}
|
||
return ok;
|
||
}
|
||
return true;
|
||
}
|