LorenEteval_Furious/Furious/Window/UserRoutingWindow.py

# Copyright (C) 2024-present  Loren Eteval & contributors <loren.eteval@proton.me>
#
# This file is part of Furious.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

from __future__ import annotations

from Furious.Frozenlib import *
from Furious.Core.CoreManager import cleanRoutingRule
from Furious.Library import *
from Furious.Qt import *
from Furious.Qt import gettext as _

from PySide6 import QtCore
from PySide6.QtGui import *
from PySide6.QtWidgets import *

import copy
import uuid
import logging
import functools

__all__ = ['UserRoutingWindow']

logger = logging.getLogger(__name__)

registerAppSettings('UserRoutingWindowGeometry')
registerAppSettings('UserRoutingWindowState')
registerAppSettings('UserRoutingHeaderViewState')


DOMAIN_STRATEGIES, ROUTING_STATES = (
    ['AsIs', 'IPIfNonMatch', 'IPOnDemand'],
    ['Enabled', 'Disabled'],
)


def valuesToList(text: str) -> list[str]:
    values = list()

    for line in text.splitlines():
        for value in line.split(','):
            value = value.strip()

            if value:
                values.append(value)

    return values


def listToText(value) -> str:
    if isinstance(value, list):
        return '\n'.join(str(item) for item in value)

    return ''


def routingStateText(state: str) -> str:
    if state == 'Enabled':
        return _('Enabled')
    elif state == 'Disabled':
        return _('Disabled')

    return _(state)


def addRoutingStateItems(combo: QComboBox):
    combo.clear()

    for state in ROUTING_STATES:
        combo.addItem(routingStateText(state), state)


def comboCurrentData(combo: QComboBox, default=''):
    data = combo.currentData()

    if data is None:
        return default

    return data


def routingObjectFromProfile(routingProfile: dict):
    if not isinstance(routingProfile, dict):
        routingProfile = dict()

    domainStrategy = routingProfile.get('domainStrategy', 'AsIs')

    if domainStrategy not in DOMAIN_STRATEGIES:
        domainStrategy = 'AsIs'

    rules = list(
        filter(
            lambda rule: rule is not None,
            list(cleanRoutingRule(rule) for rule in routingProfile.get('rules', [])),
        )
    )

    return {
        'domainStrategy': domainStrategy,
        'domainMatcher': 'hybrid',
        'rules': rules,
    }


class RoutingPreviewDialog(AppQDialog):
    def __init__(self, routingProfile: dict, parent=None):
        super().__init__(parent)

        self.setWindowTitle(_('Preview Routing'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        if PLATFORM == 'Darwin':
            self.toolbar = AppQToolBar(
                AppQAction(
                    _('Close'),
                    icon=bootstrapIcon('window-x.svg'),
                    callback=lambda: self.close(),
                ),
                parent=self,
            )
        else:
            self.toolbar = None

        self.textEditor = DraculaJSONTextEditor(fontFamily=AppFontName())
        self.textEditor.setLineWrapMode(DraculaJSONTextEditor.LineWrapMode.NoWrap)
        self.textEditor.setPlainText(
            UJSONEncoder.encode(routingObjectFromProfile(routingProfile), indent=4)
        )
        self.textEditor.setReadOnly(True)

        layout = QVBoxLayout()

        if self.toolbar is not None:
            layout.addWidget(self.toolbar)

        layout.addWidget(self.textEditor)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.setFixedSize(400, int(400 * GOLDEN_RATIO))


class RoutingTextEditDialog(AppQDialog):
    def __init__(self, text='', parent=None):
        super().__init__(parent)

        self.setWindowTitle(_('Edit Text'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        self.textEdit = QTextEdit(text)

        self.dialogBtns = AppQDialogButtonBox(QtCore.Qt.Orientation.Horizontal)
        self.dialogBtns.addButton(_('OK'), AppQDialogButtonBox.ButtonRole.AcceptRole)
        self.dialogBtns.addButton(
            _('Cancel'),
            AppQDialogButtonBox.ButtonRole.RejectRole,
        )
        self.dialogBtns.accepted.connect(self.accept)
        self.dialogBtns.rejected.connect(self.reject)

        layout = QVBoxLayout()
        layout.addWidget(self.textEdit)
        layout.addWidget(self.dialogBtns)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.setFixedSize(760, 470)

    def text(self):
        return self.textEdit.toPlainText()


class RoutingTextEdit(Mixins.QTranslatable, QTextEdit):
    def __init__(self, text='', parent=None):
        super().__init__(text, parent)

        self.setToolTip(_('Double-click to enlarge'))

    def mouseDoubleClickEvent(self, event):
        dialog = RoutingTextEditDialog(self.toPlainText(), parent=self)

        def handleResultCode(code):
            if code == PySide6Legacy.enumValueWrapper(AppQDialog.DialogCode.Accepted):
                self.setPlainText(dialog.text())

        dialog.finished.connect(handleResultCode)
        dialog.open()

        event.accept()

    def retranslate(self):
        self.setToolTip(_(self.toolTip()))


class RoutingDocumentationURL(AppQLabel):
    URL = 'https://xtls.github.io/config/routing.html'

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.setWebsiteURL()
        self.linkActivated.connect(self.handleLinkActivated)

    def setWebsiteURL(self):
        self.setText(
            '<html><head/><body><p>'
            f'<a href="{self.URL}">'
            '<span style=" text-decoration: underline; color:#007ad6;">'
            + _('Routing Documentation')
            + '</span></a></p></body></html>'
        )

    @staticmethod
    def handleLinkActivated(link: str):
        if QDesktopServices.openUrl(QtCore.QUrl(link)):
            logger.info(f'open link \'{link}\' success')
        else:
            logger.error(f'open link \'{link}\' failed')

    def retranslate(self):
        self.setWebsiteURL()


class RoutingProfilesModel(QtCore.QAbstractTableModel):
    Headers = ['Remark', 'Domain Strategy', 'State']

    def rowCount(self, parent=QtCore.QModelIndex()) -> int:
        if parent.isValid():
            return 0

        return len(Storage.UserRoutings())

    def columnCount(self, parent=QtCore.QModelIndex()) -> int:
        if parent.isValid():
            return 0

        return len(self.Headers)

    def flags(self, index):
        if not index.isValid():
            return QtCore.Qt.ItemFlag.NoItemFlags

        return QtCore.Qt.ItemFlag.ItemIsEnabled | QtCore.Qt.ItemFlag.ItemIsSelectable

    def headerData(self, section, orientation, role=QtCore.Qt.ItemDataRole.DisplayRole):
        if role != QtCore.Qt.ItemDataRole.DisplayRole:
            return None

        if orientation == QtCore.Qt.Orientation.Horizontal:
            header = self.Headers[section]

            if header in ['Remark', 'State']:
                return _(header)

            return header

        return section + 1

    def routingUniqueByRow(self, row: int):
        return list(Storage.UserRoutings().keys())[row]

    def routingByRow(self, row: int):
        return Storage.UserRoutings()[self.routingUniqueByRow(row)]

    def data(self, index, role=QtCore.Qt.ItemDataRole.DisplayRole):
        if not index.isValid():
            return None

        row, column = index.row(), index.column()

        if row < 0 or row >= len(Storage.UserRoutings()):
            return None

        routing = self.routingByRow(row)

        if role in [
            QtCore.Qt.ItemDataRole.DisplayRole,
            QtCore.Qt.ItemDataRole.EditRole,
        ]:
            if column == 0:
                return routing.get('remark', '')
            elif column == 1:
                return routing.get('domainStrategy', 'AsIs')
            elif column == 2:
                return routingStateText(
                    'Enabled' if routing.get('enabled', True) else 'Disabled'
                )

        return None

    def setData(self, index, value, role=QtCore.Qt.ItemDataRole.EditRole):
        return False

    def emitAllChanged(self):
        if self.rowCount() == 0:
            return

        self.dataChanged.emit(
            self.index(0, 0),
            self.index(self.rowCount() - 1, self.columnCount() - 1),
        )


class RoutingRuleEditDialog(AppQDialog):
    MatchInputHeight = 72
    ShortInputWidth = 240

    @staticmethod
    def textEdit(text=''):
        widget = RoutingTextEdit(text)
        widget.setFixedHeight(RoutingRuleEditDialog.MatchInputHeight)

        return widget

    @staticmethod
    def lineEdit(text=''):
        widget = QLineEdit(text)
        widget.setFixedHeight(RoutingRuleEditDialog.MatchInputHeight)

        return widget

    def __init__(self, rule: dict, parent=None):
        super().__init__(parent)

        self.rule = dict(rule)

        self.setWindowTitle(_('Edit Routing Rule'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        self.ruleTagEdit = self.lineEdit(self.rule.get('ruleTag', ''))

        self.outboundEdit = self.lineEdit(self.rule.get('outboundTag', 'proxy'))
        self.outboundEdit.setMaximumWidth(360)

        self.balancerTagEdit = self.lineEdit(str(self.rule.get('balancerTag', '')))

        self.networkCombo = AppQComboBox(translatable=False)
        self.networkCombo.addItems(['', 'tcp', 'udp', 'tcp,udp'])
        self.networkCombo.setCurrentText(self.rule.get('network', ''))
        self.networkCombo.setMaximumWidth(self.ShortInputWidth)

        self.portEdit = self.textEdit(str(self.rule.get('port', '')))
        self.portEdit.setMaximumWidth(360)

        self.sourcePortEdit = self.textEdit(str(self.rule.get('sourcePort', '')))
        self.sourcePortEdit.setMaximumWidth(360)

        self.localPortEdit = self.textEdit(str(self.rule.get('localPort', '')))
        self.localPortEdit.setMaximumWidth(360)

        self.vlessRouteEdit = self.textEdit(str(self.rule.get('vlessRoute', '')))
        self.vlessRouteEdit.setMaximumWidth(360)

        self.domainEdit = self.textEdit(listToText(self.rule.get('domain', [])))
        self.ipEdit = self.textEdit(listToText(self.rule.get('ip', [])))
        self.sourceIPEdit = self.textEdit(listToText(self.rule.get('sourceIP', [])))
        self.localIPEdit = self.textEdit(listToText(self.rule.get('localIP', [])))
        self.userEdit = self.textEdit(listToText(self.rule.get('user', [])))
        self.protocolEdit = self.textEdit(listToText(self.rule.get('protocol', [])))
        self.inboundTagEdit = self.textEdit(listToText(self.rule.get('inboundTag', [])))
        self.processEdit = self.textEdit(listToText(self.rule.get('process', [])))

        self.dialogBtns = AppQDialogButtonBox(QtCore.Qt.Orientation.Horizontal)
        self.dialogBtns.addButton(_('OK'), AppQDialogButtonBox.ButtonRole.AcceptRole)
        self.dialogBtns.addButton(
            _('Cancel'),
            AppQDialogButtonBox.ButtonRole.RejectRole,
        )
        self.dialogBtns.accepted.connect(self.accept)
        self.dialogBtns.rejected.connect(self.reject)

        generalLayout = QFormLayout()
        generalLayout.addRow(
            AppQLabel('Rule Tag', translatable=False),
            self.ruleTagEdit,
        )
        generalLayout.addRow(
            AppQLabel(_('OutBound (e.g. proxy/direct/block)')),
            self.outboundEdit,
        )
        generalLayout.addRow(
            AppQLabel('Balancer Tag', translatable=False),
            self.balancerTagEdit,
        )
        generalLayout.addRow(
            AppQLabel('Network', translatable=False),
            self.networkCombo,
        )

        generalGroup = AppQGroupBox(
            'General',
            translatable=False,
        )
        generalGroup.setLayout(generalLayout)

        destinationLayout = QFormLayout()
        destinationLayout.addRow(
            AppQLabel(_('Domain (one per line or comma-separated)')),
            self.domainEdit,
        )
        destinationLayout.addRow(
            AppQLabel(_('IP (one per line or comma-separated)')),
            self.ipEdit,
        )
        destinationLayout.addRow(
            AppQLabel(_('Port (comma/range, e.g. 53,443,1000-2000)')),
            self.portEdit,
        )
        destinationLayout.addRow(
            AppQLabel(_('VLESS Route (comma/range, e.g. 53,443,1000-2000)')),
            self.vlessRouteEdit,
        )

        destinationGroup = AppQGroupBox(
            'Destination Match',
            translatable=False,
        )
        destinationGroup.setLayout(destinationLayout)

        sourceLayout = QFormLayout()
        sourceLayout.addRow(
            AppQLabel(_('Source IP (one per line or comma-separated)')),
            self.sourceIPEdit,
        )
        sourceLayout.addRow(
            AppQLabel(_('Source Port (comma/range, e.g. 53,443,1000-2000)')),
            self.sourcePortEdit,
        )
        sourceLayout.addRow(
            AppQLabel(_('Local IP (one per line or comma-separated)')),
            self.localIPEdit,
        )
        sourceLayout.addRow(
            AppQLabel(_('Local Port (comma/range, e.g. 53,443,1000-2000)')),
            self.localPortEdit,
        )

        sourceGroup = AppQGroupBox(
            'Source / Local Match',
            translatable=False,
        )
        sourceGroup.setLayout(sourceLayout)

        identityLayout = QFormLayout()
        identityLayout.addRow(
            AppQLabel(_('User (one per line or comma-separated)')),
            self.userEdit,
        )
        identityLayout.addRow(
            AppQLabel(_('Inbound Tag (one per line or comma-separated)')),
            self.inboundTagEdit,
        )
        identityLayout.addRow(
            AppQLabel(_('Process (one per line or comma-separated)')),
            self.processEdit,
        )
        identityLayout.addRow(
            AppQLabel(_('Protocol (one per line or comma-separated)')),
            self.protocolEdit,
        )

        identityGroup = AppQGroupBox(
            'Identity / Process / Protocol Match',
            translatable=False,
        )
        identityGroup.setLayout(identityLayout)

        matchGrid = QGridLayout()
        matchGrid.addWidget(generalGroup, 0, 0)
        matchGrid.addWidget(destinationGroup, 0, 1)
        matchGrid.addWidget(sourceGroup, 1, 0)
        matchGrid.addWidget(identityGroup, 1, 1)
        matchGrid.setColumnStretch(0, 1)
        matchGrid.setColumnStretch(1, 1)

        bottomLayout = QHBoxLayout()
        bottomLayout.addWidget(RoutingDocumentationURL())
        bottomLayout.addStretch(1)
        bottomLayout.addWidget(self.dialogBtns)

        layout = QVBoxLayout()
        layout.addLayout(matchGrid)
        layout.addLayout(bottomLayout)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.resize(int(800 * GOLDEN_RATIO), 800)

    def routingRule(self):
        rule = {
            'type': 'field',
            'outboundTag': self.outboundEdit.text().strip() or 'proxy',
        }

        for key, value in [
            ('ruleTag', self.ruleTagEdit.text().strip()),
            ('network', self.networkCombo.currentText().strip()),
            ('port', self.portEdit.toPlainText().strip()),
            ('sourcePort', self.sourcePortEdit.toPlainText().strip()),
            ('localPort', self.localPortEdit.toPlainText().strip()),
            ('vlessRoute', self.vlessRouteEdit.toPlainText().strip()),
            ('balancerTag', self.balancerTagEdit.text().strip()),
        ]:
            if value:
                rule[key] = value

        for key, widget in [
            ('domain', self.domainEdit),
            ('ip', self.ipEdit),
            ('sourceIP', self.sourceIPEdit),
            ('localIP', self.localIPEdit),
            ('user', self.userEdit),
            ('protocol', self.protocolEdit),
            ('inboundTag', self.inboundTagEdit),
            ('process', self.processEdit),
        ]:
            values = valuesToList(widget.toPlainText())

            if values:
                rule[key] = values

        return rule


class RoutingRemarkEditDialog(AppQDialog):
    def __init__(self, remark: str, parent=None):
        super().__init__(parent)

        self.setWindowTitle(_('Edit Routing Remark'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        self.remarkEdit = QLineEdit(remark)

        self.dialogBtns = AppQDialogButtonBox(QtCore.Qt.Orientation.Horizontal)
        self.dialogBtns.addButton(_('OK'), AppQDialogButtonBox.ButtonRole.AcceptRole)
        self.dialogBtns.addButton(
            _('Cancel'),
            AppQDialogButtonBox.ButtonRole.RejectRole,
        )
        self.dialogBtns.accepted.connect(self.accept)
        self.dialogBtns.rejected.connect(self.reject)

        layout = QFormLayout()
        layout.addRow(AppQLabel(_('Remark')), self.remarkEdit)
        layout.addRow(self.dialogBtns)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.resize(420, 120)

    def remark(self):
        return self.remarkEdit.text().strip()


class RoutingProfileEditDialog(AppQDialog):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.setWindowTitle(_('Add Routing'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        self.remarkEdit = QLineEdit(_('New Routing'))

        self.domainStrategyCombo = AppQComboBox(translatable=False)
        self.domainStrategyCombo.addItems(DOMAIN_STRATEGIES)
        self.domainStrategyCombo.setCurrentText('AsIs')

        self.enabledCombo = AppQComboBox(translatable=False)
        addRoutingStateItems(self.enabledCombo)
        self.enabledCombo.setCurrentIndex(ROUTING_STATES.index('Enabled'))

        self.dialogBtns = AppQDialogButtonBox(QtCore.Qt.Orientation.Horizontal)
        self.dialogBtns.addButton(_('OK'), AppQDialogButtonBox.ButtonRole.AcceptRole)
        self.dialogBtns.addButton(
            _('Cancel'),
            AppQDialogButtonBox.ButtonRole.RejectRole,
        )
        self.dialogBtns.accepted.connect(self.accept)
        self.dialogBtns.rejected.connect(self.reject)

        layout = QFormLayout()
        layout.addRow(AppQLabel(_('Remark')), self.remarkEdit)
        layout.addRow(
            AppQLabel('Domain Strategy', translatable=False),
            self.domainStrategyCombo,
        )
        layout.addRow(AppQLabel(_('State')), self.enabledCombo)
        layout.addRow(self.dialogBtns)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.resize(460, 160)

    def routing(self):
        return {
            'remark': self.remarkEdit.text().strip() or _('New Routing'),
            'domainStrategy': self.domainStrategyCombo.currentText(),
            'enabled': comboCurrentData(self.enabledCombo, 'Enabled') == 'Enabled',
            'rules': [],
        }


class RoutingRulesQListWidget(AppQListWidget):
    editRequested, deleteRequested = (
        QtCore.Signal(),
        QtCore.Signal(),
    )

    def __init__(self, routing: dict, parent=None):
        super().__init__(parent)

        self.routing = routing

        self.setAlternatingRowColors(True)
        self.setSelectionBehavior(AppQListWidget.SelectionBehavior.SelectRows)
        self.setSelectionMode(AppQListWidget.SelectionMode.ExtendedSelection)
        self.setContextMenuPolicy(QtCore.Qt.ContextMenuPolicy.CustomContextMenu)

        self.itemDoubleClicked.connect(lambda _item: self.editRequested.emit())
        self.customContextMenuRequested.connect(self.handleCustomContextMenuRequested)

        self.contextDeleteAction = AppQAction(
            _('Delete'),
            callback=lambda: self.deleteRequested.emit(),
        )
        self.contextMenu = AppQMenu(self.contextDeleteAction, parent=self)

        self.flushAll()

    def rules(self):
        rules = self.routing.setdefault('rules', list())

        if not isinstance(rules, list):
            self.routing['rules'] = rules = list()

        return rules

    def ruleAt(self, index: int):
        return self.rules()[index]

    def ruleText(self, rule: dict) -> str:
        name, outbound, domains, ips = (
            rule.get('ruleTag', '') or 'Untitled Rule',
            rule.get('outboundTag', 'proxy'),
            len(rule.get('domain', [])),
            len(rule.get('ip', [])),
        )

        return f'{name} -> {outbound} ({domains} domains, {ips} IPs)'

    def selectedRuleText(self):
        indexes = self.selectedIndex

        if not indexes:
            return ''

        return self.ruleText(self.ruleAt(indexes[0]))

    def appendRule(self, rule: dict):
        self.rules().append(rule)
        self.flushAll()

    def setRule(self, index: int, rule: dict):
        self.rules()[index] = rule
        self.flushAll()

    def deleteRules(self, indexes: list[int]):
        for i in range(len(indexes)):
            self.rules().pop(indexes[i] - i)

        self.flushAll()

    def flushAll(self):
        self.clear()

        for rule in self.rules():
            self.addItem(self.ruleText(rule))

    @QtCore.Slot(QtCore.QPoint)
    def handleCustomContextMenuRequested(self, point):
        self.contextMenu.exec(self.viewport().mapToGlobal(point))


class RoutingRulesDialog(AppQDialog):
    def __init__(self, routing: dict, parent=None):
        super().__init__(parent)

        self.routing = routing
        self.setWindowTitle(_('Routing Rules'))
        self.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        if PLATFORM == 'Darwin':
            self.toolbar = AppQToolBar(
                AppQAction(
                    _('Close'),
                    icon=bootstrapIcon('window-x.svg'),
                    callback=lambda: self.close(),
                ),
                parent=self,
            )
        else:
            self.toolbar = None

        self.listWidget = RoutingRulesQListWidget(self.routing, parent=self)
        self.listWidget.editRequested.connect(self.editRule)
        self.listWidget.deleteRequested.connect(self.deleteRule)

        self.addButton = AppQPushButton(_('Add'))
        self.addButton.clicked.connect(self.addRule)

        self.deleteButton = AppQPushButton(_('Delete'))
        self.deleteButton.clicked.connect(self.deleteRule)

        buttonLayout = QHBoxLayout()
        buttonLayout.addWidget(self.addButton)
        buttonLayout.addWidget(self.deleteButton)

        layout = QVBoxLayout()

        if self.toolbar is not None:
            layout.addWidget(self.toolbar)

        layout.addWidget(self.listWidget)
        layout.addLayout(buttonLayout)

        self.setLayout(layout)

    def setWidthAndHeight(self):
        self.setFixedSize(760, 470)

    def addRule(self):
        rule = {'type': 'field', 'outboundTag': 'proxy', 'ruleTag': 'New Rule'}
        dialog = RoutingRuleEditDialog(rule, parent=None)

        def handleResultCode(code):
            if code == PySide6Legacy.enumValueWrapper(AppQDialog.DialogCode.Accepted):
                self.listWidget.appendRule(dialog.routingRule())

        dialog.finished.connect(handleResultCode)
        dialog.open()

    def editRule(self):
        indexes = self.listWidget.selectedIndex

        if len(indexes) != 1:
            return

        index = indexes[0]
        dialog = RoutingRuleEditDialog(self.listWidget.ruleAt(index), parent=None)

        def handleResultCode(_index, code):
            if code == PySide6Legacy.enumValueWrapper(AppQDialog.DialogCode.Accepted):
                self.listWidget.setRule(_index, dialog.routingRule())

        dialog.finished.connect(functools.partial(handleResultCode, index))
        dialog.open()

    def deleteRule(self):
        indexes = self.listWidget.selectedIndex

        if len(indexes) == 0:
            return

        def handleResultCode(_indexes, code):
            if code == PySide6Legacy.enumValueWrapper(
                AppQMessageBox.StandardButton.Yes
            ):
                self.listWidget.deleteRules(_indexes)
            else:
                # Do not delete
                pass

        if PLATFORM == 'Windows':
            # Windows
            mbox = MBoxQuestionDelete(icon=AppQMessageBox.Icon.Question)
        else:
            # macOS & linux
            mbox = MBoxQuestionDelete(
                icon=AppQMessageBox.Icon.Question,
                parent=self,
            )
            mbox.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        mbox.isMulti = bool(len(indexes) > 1)
        mbox.possibleRemark = self.listWidget.selectedRuleText()
        mbox.setText(mbox.customText())
        mbox.finished.connect(functools.partial(handleResultCode, indexes))

        # Show the MessageBox asynchronously
        mbox.open()


class UserRoutingQTableViewHorizontalHeader(AppQHeaderView):
    def __init__(self, *args, **kwargs):
        super().__init__(QtCore.Qt.Orientation.Horizontal, *args, **kwargs)


class UserRoutingTableView(Mixins.QTranslatable, AppQTableView):
    RowHeight = 42

    def __init__(self, parent=None):
        super().__init__(parent)

        self.sourceModel = RoutingProfilesModel(parent=self)
        self.setModel(self.sourceModel)
        self.setHorizontalHeader(
            UserRoutingQTableViewHorizontalHeader(
                parent=self,
                sectionSizeSettingsName='UserRoutingHeaderViewState',
            )
        )
        self.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
        self.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
        self.setDragDropMode(QAbstractItemView.DragDropMode.NoDragDrop)
        self.setEditTriggers(QAbstractItemView.EditTrigger.NoEditTriggers)
        self.setSortingEnabled(False)
        self.doubleClicked.connect(self.editSelectedRules)
        self.setContextMenuPolicy(QtCore.Qt.ContextMenuPolicy.CustomContextMenu)
        self.customContextMenuRequested.connect(self.handleCustomContextMenuRequested)

        self.contextPreviewAction = AppQAction(
            _('Preview'),
            icon=bootstrapIcon('file-earmark-text.svg'),
            callback=lambda: self.previewSelectedItem(),
        )
        self.contextRenameAction = AppQAction(
            _('Rename'),
            callback=lambda: self.renameSelectedItem(),
        )
        self.contextDeleteAction = AppQAction(
            _('Delete'),
            callback=lambda: self.deleteSelectedItem(),
        )
        self.contextMenu = AppQMenu(
            self.contextPreviewAction,
            self.contextRenameAction,
            AppQSeperator(),
            self.contextDeleteAction,
            parent=self,
        )

        self.setDefaultRowHeight(self.RowHeight)
        self.configureHeader()
        self.flushAll()

    def configureHeader(self):
        header = self.horizontalHeader()
        header.setSectionsMovable(True)
        header.setFirstSectionMovable(True)
        header.setCustomSectionResizeMode()
        header.restoreSectionSize()

    @property
    def selectedIndex(self):
        return sorted(
            list(set(index.row() for index in self.selectionModel().selectedRows()))
        )

    def routingUniqueByRow(self, row):
        return self.sourceModel.routingUniqueByRow(row)

    @QtCore.Slot(QtCore.QPoint)
    def handleCustomContextMenuRequested(self, point):
        indexes = self.selectedIndex

        isUniqueFlag = len(indexes) == 1

        for action in [self.contextPreviewAction, self.contextRenameAction]:
            action.setEnabled(isUniqueFlag)

        self.contextMenu.exec(self.viewport().mapToGlobal(point))

    def flushItem(self, row: int, column: int):
        index = self.sourceModel.index(row, column)

        if column == 1:
            combo = AppQComboBox(parent=self.viewport(), translatable=False)
            combo.addItems(DOMAIN_STRATEGIES)
            combo.setMinimumWidth(0)
            combo.setCurrentText(
                self.sourceModel.routingByRow(row).get('domainStrategy', 'AsIs')
            )
            combo.currentTextChanged.connect(
                lambda text, _row=row: self.setDomainStrategy(_row, text)
            )

            self.setIndexWidget(index, combo)
        elif column == 2:
            combo = AppQComboBox(parent=self.viewport(), translatable=False)
            addRoutingStateItems(combo)
            combo.setMinimumWidth(0)
            combo.setCurrentIndex(
                ROUTING_STATES.index(
                    'Enabled'
                    if self.sourceModel.routingByRow(row).get('enabled', True)
                    else 'Disabled'
                )
            )
            combo.currentIndexChanged.connect(
                lambda _index, _row=row, _combo=combo: self.setEnabled(
                    _row,
                    comboCurrentData(_combo, 'Enabled'),
                )
            )

            self.setIndexWidget(index, combo)

    def flushRow(self, row: int):
        for column in range(self.sourceModel.columnCount()):
            self.flushItem(row, column)

    def flushAll(self):
        self.sourceModel.emitAllChanged()

        for row in range(self.sourceModel.rowCount()):
            self.flushRow(row)

    def retranslate(self):
        self.sourceModel.headerDataChanged.emit(
            QtCore.Qt.Orientation.Horizontal,
            0,
            self.sourceModel.columnCount() - 1,
        )
        self.flushAll()

    def setDomainStrategy(self, row: int, text: str):
        if row < 0 or row >= self.sourceModel.rowCount():
            return

        self.sourceModel.routingByRow(row)['domainStrategy'] = text
        self.sourceModel.emitAllChanged()

    def setEnabled(self, row: int, state: str):
        if row < 0 or row >= self.sourceModel.rowCount():
            return

        self.sourceModel.routingByRow(row)['enabled'] = state == 'Enabled'
        self.sourceModel.emitAllChanged()

    def appendNewItem(self):
        dialog = RoutingProfileEditDialog(parent=self)

        def handleResultCode(code):
            if code != PySide6Legacy.enumValueWrapper(AppQDialog.DialogCode.Accepted):
                return

            row = self.sourceModel.rowCount()
            unique = str(uuid.uuid4())

            self.sourceModel.beginInsertRows(QtCore.QModelIndex(), row, row)

            Storage.UserRoutings()[unique] = dialog.routing()

            self.sourceModel.endInsertRows()
            self.flushRow(row)

        dialog.finished.connect(handleResultCode)
        dialog.open()

    def deleteSelectedItem(self):
        indexes = self.selectedIndex

        if not indexes:
            # Nothing selected. Do nothing
            return

        def handleResultCode(_indexes, code):
            if code != PySide6Legacy.enumValueWrapper(
                AppQMessageBox.StandardButton.Yes
            ):
                return

            for i in range(len(_indexes)):
                deleteIndex = _indexes[i] - i
                deleteUnique = self.routingUniqueByRow(deleteIndex)

                self.sourceModel.beginRemoveRows(
                    QtCore.QModelIndex(),
                    deleteIndex,
                    deleteIndex,
                )

                Storage.UserRoutings().pop(deleteUnique)

                self.sourceModel.endRemoveRows()

            self.flushAll()

        if PLATFORM == 'Windows':
            # Windows
            mbox = MBoxQuestionDelete(icon=AppQMessageBox.Icon.Question)
        else:
            # macOS & linux
            mbox = MBoxQuestionDelete(
                icon=AppQMessageBox.Icon.Question,
                parent=self,
            )
            mbox.setWindowModality(QtCore.Qt.WindowModality.WindowModal)

        mbox.isMulti = bool(len(indexes) > 1)
        mbox.possibleRemark = self.sourceModel.data(
            self.sourceModel.index(indexes[0], 0),
            QtCore.Qt.ItemDataRole.DisplayRole,
        )
        mbox.setText(mbox.customText())
        mbox.finished.connect(functools.partial(handleResultCode, indexes))

        # Show the MessageBox asynchronously
        mbox.open()

    def renameSelectedItem(self):
        indexes = self.selectedIndex

        if not indexes:
            # Nothing selected. Do nothing
            return

        if len(indexes) > 1:
            # Multiple items selected. Do nothing
            return

        routing = self.sourceModel.routingByRow(indexes[0])

        dialog = RoutingRemarkEditDialog(routing.get('remark', ''), parent=self)

        def handleResultCode(code):
            if code != PySide6Legacy.enumValueWrapper(AppQDialog.DialogCode.Accepted):
                return

            remark = dialog.remark()

            if remark:
                routing['remark'] = remark
                self.sourceModel.emitAllChanged()

        dialog.finished.connect(handleResultCode)
        dialog.open()

    def previewSelectedItem(self):
        indexes = self.selectedIndex

        if len(indexes) != 1:
            return

        routing = copy.deepcopy(self.sourceModel.routingByRow(indexes[0]))

        dialog = RoutingPreviewDialog(routing, parent=self)
        dialog.open()

    def editSelectedRules(self):
        indexes = self.selectedIndex

        if not indexes:
            # Nothing selected. Do nothing
            return

        if len(indexes) > 1:
            # Multiple items selected. Do nothing
            return

        routing = self.sourceModel.routingByRow(indexes[0])

        dialog = RoutingRulesDialog(routing, parent=self)
        dialog.finished.connect(lambda _code: self.flushAll())
        dialog.open()


class UserRoutingWindow(AppQMainWindow):
    DEFAULT_WINDOW_SIZE = QtCore.QSize(980, 560)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.setWindowTitle(_('Edit Routing'))

        self.tableView = UserRoutingTableView(parent=self)
        self.toolbar = AppQToolBar(
            AppQAction(
                _('Add'),
                icon=bootstrapIcon('plus-lg.svg'),
                callback=lambda: self.tableView.appendNewItem(),
            ),
            AppQAction(
                _('Delete'),
                icon=bootstrapIcon('dash-lg.svg'),
                callback=lambda: self.tableView.deleteSelectedItem(),
            ),
            parent=self,
        )
        self.toolbar.setObjectName('UserRoutingWindow_AppQToolBar')
        self.addToolBar(self.toolbar)

        centralWidget = QWidget()

        layout = QVBoxLayout(centralWidget)
        layout.addWidget(self.tableView)

        self.setCentralWidget(centralWidget)

    def setWidthAndHeight(self):
        if AppSettings.get('UserRoutingWindowGeometry') is None:
            self.resize(UserRoutingWindow.DEFAULT_WINDOW_SIZE)
        else:
            if PLATFORM == 'Darwin':
                self.resize(UserRoutingWindow.DEFAULT_WINDOW_SIZE)
            else:
                try:
                    self.restoreGeometry(AppSettings.get('UserRoutingWindowGeometry'))
                except Exception:
                    # Any non-exit exceptions

                    self.resize(UserRoutingWindow.DEFAULT_WINDOW_SIZE)

                try:
                    self.restoreState(AppSettings.get('UserRoutingWindowState'))
                except Exception:
                    # Any non-exit exceptions

                    pass

    def cleanup(self):
        AppSettings.set('UserRoutingWindowGeometry', self.saveGeometry())
        AppSettings.set('UserRoutingWindowState', self.saveState())