AutoLibrary/src/gui/ALAutoScriptOrchDialog/_helpers.py

# -*- coding: utf-8 -*-
"""
Copyright (c) 2026 KenanZhu.
All rights reserved.

This software is provided "as is", without any warranty of any kind.
You may use, modify, and distribute this file under the terms of the MIT License.
See the LICENSE file for details.
"""
"""
    Helper utilities and constants for the AutoScript orchestration dialog.
"""
import re

from PySide6.QtCore import (
    QObject,
    QDate,
    QTime
)
from PySide6.QtWidgets import (
    QComboBox,
    QDateEdit,
    QDoubleSpinBox,
    QHBoxLayout,
    QLabel,
    QLineEdit,
    QSizePolicy,
    QSpinBox,
    QStackedWidget,
    QTimeEdit,
    QWidget,
)

from autoscript import (
    ALL_VARIABLES,
)

# Types that support arithmetic operations (add/sub)
ARITH_TYPES = {"Date", "Time", "Int", "Float"}
VAR_TYPE_ORDER = [
    "String",
    "Int",
    "Float",
    "Boolean",
    "Date",
    "Time"
]
PRESET_VARIABLES = [
    {
        "name": name.upper(),
        "type": vtype,
        "display": display
    }
    for display, (name, vtype) in ALL_VARIABLES.items()
]
PRESET_NAMES = {
    p["name"] for p in PRESET_VARIABLES
}
# Operator display names (UI-specific), using Lua operator symbols
_COMPARE_DISPLAY_MAP = {
    "==": "等于",
    "~=": "不等于",
    ">": "大于",
    "<": "小于",
    ">=": "大于等于",
    "<=": "小于等于",
}
COMPARE_OPERATORS = sorted(
    [(name, op) for op, name in _COMPARE_DISPLAY_MAP.items()],
    key=lambda x: len(x[1]), reverse=True
)
LOGIC_OPERATORS = [
    ("并且 (and)", "and"),
    ("或者 (or)", "or"),
]
ACTION_TYPES = [
    ("设置为", "set"),
    ("增加", "add"),
    ("减少", "sub"),
]
DATE_RELATIVE_OPTIONS = [
    ("前天", "day_before_yesterday"),
    ("昨天", "yesterday"),
    ("今天", "today"),
    ("明天", "tomorrow"),
    ("后天", "day_after_tomorrow")
]
DATE_OFFSET_UNITS = [
    ("天", "days"),
    ("周", "weeks"),
    ("月", "months"),
    ("年", "years"),
]


class VariableManager(QObject):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)
        self._vars = []
        self._nameMap = {}

        self._initPresetVars()


    def _initPresetVars(
        self
    ):

        for p in PRESET_VARIABLES:
            entry = {"name": p["name"], "type": p["type"], "display": p["display"]}
            self._vars.append(entry)
            self._nameMap[p["name"]] = entry


    def getInfoByName(
        self,
        name: str
    ):

        return self._nameMap.get(name.upper().strip())


    def populateCombo(
        self,
        combo: QComboBox
    ):

        currentData = combo.currentData()
        combo.blockSignals(True)
        combo.clear()
        for entry in self._vars:
            combo.addItem(
                entry["display"],
                (entry["name"], entry["type"])
            )
        if currentData:
            for i in range(combo.count()):
                d = combo.itemData(i)
                if d and d[0] == currentData[0]:
                    combo.setCurrentIndex(i)
                    break
        combo.blockSignals(False)


    def findExactNameEntry(
        self,
        combo: QComboBox,
        name: str
    ) -> int:

        name = name.upper().strip()
        for i in range(combo.count()):
            d = combo.itemData(i)
            if d and len(d) >= 1 and d[0].upper().strip() == name:
                return i
        return -1


def makeValueWidget(
    var_type: str,
    parent: QWidget = None
) -> QWidget:

    if var_type == "Int":
        w = QSpinBox(parent)
        w.setRange(-999999, 999999)
        w.setFixedHeight(25)
        w.setMinimumWidth(100)
        return w
    if var_type == "Float":
        w = QDoubleSpinBox(parent)
        w.setRange(-999999.0, 999999.0)
        w.setDecimals(2)
        w.setFixedHeight(25)
        w.setMinimumWidth(100)
        return w
    if var_type == "String":
        w = QLineEdit(parent)
        w.setPlaceholderText("输入值")
        w.setFixedHeight(25)
        w.setMinimumWidth(120)
        return w
    if var_type == "Boolean":
        w = QComboBox(parent)
        w.addItem("是 (true)", "true")
        w.addItem("否 (false)", "false")
        w.setFixedHeight(25)
        w.setMinimumWidth(100)
        return w
    if var_type == "Date":
        return _DateInputContainer(parent)
    if var_type == "Time":
        return _TimeInputContainer(parent)
    w = QLineEdit(parent)
    w.setPlaceholderText("输入值")
    w.setFixedHeight(25)
    w.setMinimumWidth(120)
    return w


def makeOffsetWidget(
    var_type: str,
    parent: QWidget = None
) -> QWidget:

    if var_type == "Int":
        w = QSpinBox(parent)
        w.setRange(-999999, 999999)
        w.setFixedHeight(25)
        w.setMinimumWidth(100)
        return w
    if var_type == "Float":
        w = QDoubleSpinBox(parent)
        w.setRange(-999999.0, 999999.0)
        w.setDecimals(2)
        w.setFixedHeight(25)
        w.setMinimumWidth(100)
        return w
    if var_type == "Date":
        return _DateOffsetContainer(parent)
    if var_type == "Time":
        return _TimeOffsetContainer(parent)
    w = QLabel("(不支持该操作)", parent)
    w.setFixedHeight(25)
    return w


def makeVarRefCombo(
    parent: QWidget = None
) -> QComboBox:

    cb = QComboBox(parent)
    cb.setFixedHeight(25)
    cb.setMinimumWidth(120)
    cb.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
    return cb


def makeComboWidget(
    items,
    min_width: int = 80,
    parent: QWidget = None
) -> QComboBox:

    cb = QComboBox(parent)
    for display, data in items:
        cb.addItem(display, data)
    cb.setFixedHeight(25)
    cb.setMinimumWidth(min_width)
    return cb


def makeLabel(
    text: str,
    parent: QWidget = None,
    width: int = None
) -> QLabel:

    lbl = QLabel(text, parent)
    lbl.setFixedHeight(25)
    if width:
        lbl.setFixedWidth(width)
    return lbl


class _DateInputContainer(QWidget):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)
        self._dynamicItems = {}  # index -> raw expression, for one-way parsed items
        self.setupUi()


    def setupUi(
        self
    ):

        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(4)
        self._modeCombo = QComboBox(self)
        self._modeCombo.addItem("相对日期", "relative")
        self._modeCombo.addItem("绝对日期", "absolute")
        self._modeCombo.setFixedHeight(25)
        self._stack = QStackedWidget(self)
        self._relCombo = QComboBox(self)
        for display, data in DATE_RELATIVE_OPTIONS:
            self._relCombo.addItem(display, data)
        self._relCombo.setFixedHeight(25)
        self._stack.addWidget(self._relCombo)
        self._dateEdit = QDateEdit(self)
        self._dateEdit.setDisplayFormat("yyyy-MM-dd")
        self._dateEdit.setCalendarPopup(True)
        self._dateEdit.setFixedHeight(25)
        self._stack.addWidget(self._dateEdit)
        self._modeCombo.currentIndexChanged.connect(
            lambda i: self._stack.setCurrentIndex(i)
        )
        layout.addWidget(self._modeCombo)
        layout.addWidget(self._stack)
        layout.addStretch()

    _RE_DATE_ADD_CURRENT = re.compile(
        r'^date_add\(CURRENT_DATE\(\),\s*(-?\d+)\)$', re.IGNORECASE
    )

    def getValue(
        self
    ) -> str:

        mode = self._modeCombo.currentData()
        if mode == "relative":
            idx = self._relCombo.currentIndex()
            if idx in self._dynamicItems:
                return self._dynamicItems[idx]
            return self._relCombo.currentText()
        return self._dateEdit.date().toString("yyyy-MM-dd")


    def setValue(
        self,
        expr: str
    ):

        s = expr.strip()
        up = s.upper()
        if up == "CURRENT_DATE()":
            self._modeCombo.setCurrentIndex(0)
            self._relCombo.setCurrentIndex(2)
            return
        m_add = self._RE_DATE_ADD_CURRENT.match(up)
        if m_add:
            n = int(m_add.group(1))
            _OFFSET_IDX = {-2: 0, -1: 1, 0: 2, 1: 3, 2: 4}
            idx = _OFFSET_IDX.get(n)
            if idx is not None:
                self._modeCombo.setCurrentIndex(0)
                self._relCombo.setCurrentIndex(idx)
                return
            label = f"{n}天后" if n >= 0 else f"{-n}天前"
            raw = f"CURRENT_DATE {'+' if n >= 0 else '-'} {abs(n)}"
            self._modeCombo.setCurrentIndex(0)
            for ci in range(self._relCombo.count()):
                if ci in self._dynamicItems and self._dynamicItems[ci] == raw:
                    self._relCombo.setCurrentIndex(ci)
                    return
            idx = self._relCombo.count()
            self._relCombo.addItem(label)
            self._dynamicItems[idx] = raw
            self._relCombo.setCurrentIndex(idx)
            return
        m_date_ctor = re.match(r"^DATE\((\d+),\s*(\d+),\s*(\d+)\)$", up)
        if m_date_ctor:
            self._modeCombo.setCurrentIndex(1)
            self._dateEdit.setDate(QDate(
                int(m_date_ctor.group(1)),
                int(m_date_ctor.group(2)),
                int(m_date_ctor.group(3)),
            ))
            return
        m_date = re.match(r'^"(\d{4}-\d{2}-\d{2})"$', s)
        if m_date:
            self._modeCombo.setCurrentIndex(1)
            parts = m_date.group(1).split("-")
            self._dateEdit.setDate(QDate(int(parts[0]), int(parts[1]), int(parts[2])))


class _TimeInputContainer(QWidget):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)
        self._timeEdit = QTimeEdit(self)
        self._timeEdit.setDisplayFormat("HH:mm")
        self._timeEdit.setFixedHeight(25)

        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.addWidget(self._timeEdit)


    def getValue(
        self
    ) -> str:

        return self._timeEdit.time().toString("HH:mm")


    def setValue(
        self,
        expr: str
    ):

        s = expr.strip()
        up = s.upper()
        m_time_ctor = re.match(r"^TIME\((\d+),\s*(\d+)\)$", up)
        if m_time_ctor:
            self._timeEdit.setTime(QTime(
                int(m_time_ctor.group(1)),
                int(m_time_ctor.group(2)),
            ))
            return
        m = re.match(r'^"(\d{1,2}:\d{2})"$', s)
        if m:
            parts = m.group(1).split(":")
            self._timeEdit.setTime(QTime(int(parts[0]), int(parts[1])))


class _DateOffsetContainer(QWidget):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)
        self._spinBox = QSpinBox(self)
        self._spinBox.setRange(0, 99999)
        self._spinBox.setFixedHeight(25)
        self._unitCombo = QComboBox(self)
        for display, data in DATE_OFFSET_UNITS:
            self._unitCombo.addItem(display, data)
        self._unitCombo.setFixedHeight(25)

        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(4)
        layout.addWidget(self._spinBox)
        layout.addWidget(self._unitCombo)
        layout.addStretch()


    def getValue(
        self
    ) -> str:

        return str(self.getOffsetDays())


    def setValue(
        self,
        expr: str
    ):

        s = expr.strip().lstrip("+")
        try:
            self._spinBox.setValue(int(s))
        except ValueError:
            pass


    def getOffsetDays(
        self
    ) -> int:

        val = self._spinBox.value()
        unit = self._unitCombo.currentData()
        if unit == "weeks":
            return val * 7
        if unit == "months":
            return val * 30
        if unit == "years":
            return val * 365
        return val


    def getRawValue(
        self
    ) -> str:

        return str(self._spinBox.value())


class _TimeOffsetContainer(QWidget):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)
        self._spinBox = QSpinBox(self)
        self._spinBox.setRange(0, 99999)
        self._spinBox.setSuffix(" 小时")
        self._spinBox.setFixedHeight(25)

        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.addWidget(self._spinBox)


    def getValue(
        self
    ) -> str:

        return str(self.getOffsetHours())


    def setValue(
        self,
        expr: str
    ):

        s = expr.strip().lstrip("+")
        try:
            self._spinBox.setValue(int(s))
        except ValueError:
            pass


    def getOffsetHours(
        self
    ) -> int:

        return self._spinBox.value()


    def getRawValue(
        self
    ) -> str:

        return str(self._spinBox.value())


def getValueFromWidget(
    w: QWidget
) -> str:

    if hasattr(w, "getValue"):
        return w.getValue()
    if isinstance(w, QTimeEdit):
        return w.time().toString("HH:mm")
    if isinstance(w, QDateEdit):
        return w.date().toString("yyyy-MM-dd")
    if isinstance(w, QComboBox):
        return w.currentData() or w.currentText()
    if isinstance(w, QSpinBox):
        return str(w.value())
    if isinstance(w, QDoubleSpinBox):
        return str(w.value())
    if isinstance(w, QLineEdit):
        return w.text()
    return ""


def setWidgetValue(
    w: QWidget,
    var_type: str,
    expr: str
):
    """
        Set a widget's value from a Lua script expression.
    """

    if hasattr(w, "setValue"):
        w.setValue(expr)
        return
    s = expr.strip()
    up = s.upper()
    if isinstance(w, QTimeEdit):
        m_time_ctor = re.match(r"^TIME\((\d+),\s*(\d+)\)$", up)
        if m_time_ctor:
            w.setTime(QTime(int(m_time_ctor.group(1)), int(m_time_ctor.group(2))))
        else:
            m = re.match(r'^"(\d{1,2}:\d{2})"$', s)
            if m:
                parts = m.group(1).split(":")
                w.setTime(QTime(int(parts[0]), int(parts[1])))
    elif isinstance(w, QDateEdit):
        m_date_ctor = re.match(r"^DATE\((\d+),\s*(\d+),\s*(\d+)\)$", up)
        if m_date_ctor:
            w.setDate(QDate(
                int(m_date_ctor.group(1)),
                int(m_date_ctor.group(2)),
                int(m_date_ctor.group(3)),
            ))
        else:
            m = re.match(r'^"(\d{4}-\d{2}-\d{2})"$', s)
            if m:
                parts = m.group(1).split("-")
                w.setDate(QDate(int(parts[0]), int(parts[1]), int(parts[2])))
    elif isinstance(w, QComboBox):
        for i in range(w.count()):
            d = w.itemData(i)
            if d is not None:
                if str(d).upper() == up:
                    w.setCurrentIndex(i)
                    return
            if w.itemText(i).upper() == up:
                w.setCurrentIndex(i)
                return
    elif isinstance(w, QSpinBox):
        try:
            w.setValue(int(expr))
        except ValueError:
            pass
    elif isinstance(w, QDoubleSpinBox):
        try:
            w.setValue(float(expr))
        except ValueError:
            pass
    elif isinstance(w, QLineEdit):
        inner = expr.strip()
        if inner.startswith('"') and inner.endswith('"'):
            inner = inner[1:-1].replace('\\"', '"')
        w.setText(inner)


def encodeValueStr(
    raw_value: str,
    var_type: str
) -> str:
    """
        Encode a raw widget value as a Lua expression.

        Arithmetic expressions (A + B) are passed through for numeric types;
        Date/Time arithmetic is translated to ``date_add()`` / ``time_add()`` calls.
    """

    if var_type in ("Date", "Time"):
        return _encodeDateOrTime(str(raw_value), var_type)
    if isinstance(raw_value, bool):
        return "true" if raw_value else "false"
    s = str(raw_value)
    if isArithExpr(s):
        return s
    if var_type == "Boolean":
        up = s.upper().strip()
        if up in ("TRUE", "FALSE"):
            return up.lower()
        return "true" if raw_value else "false"
    if var_type == "String":
        escaped = s.replace("\\", "\\\\").replace('"', '\\"')
        return f'"{escaped}"'
    return s


def _encodeDateOrTime(
    raw_value: str,
    var_type: str
) -> str:
    """
        Translate a date/time widget value into a Lua expression.
    """

    s = raw_value.strip()
    up = s.upper()
    m_arith_spaced = re.match(r'^(.+?)\s+([+-])\s+(.+)$', s)
    m_arith_nospace = re.match(r'^([A-Za-z_]\w*)([+-])(\d+|[A-Za-z_]\w*)$', s)
    m_arith = m_arith_spaced or m_arith_nospace
    if m_arith:
        left = m_arith.group(1).strip().upper()
        sign = m_arith.group(2)
        right = m_arith.group(3).strip()
        operand = right if sign == "+" else f"-{right}"
        if left == "CURRENT_DATE":
            return f"date_add(CURRENT_DATE(), {operand})"
        if left == "CURRENT_TIME":
            return f"time_add(CURRENT_TIME(), {operand})"
        if var_type == "Date":
            return f"date_add({left}, {operand})"
        if var_type == "Time":
            return f"time_add({left}, {operand})"
        return f"{left} {sign} {right}"
    if up == "CURRENT_DATE":
        return "CURRENT_DATE()"
    if up == "CURRENT_TIME":
        return "CURRENT_TIME()"
    _REL_MAP = {
        "前天": "date_add(CURRENT_DATE(), -2)",
        "昨天": "date_add(CURRENT_DATE(), -1)",
        "今天": "CURRENT_DATE()",
        "明天": "date_add(CURRENT_DATE(), 1)",
        "后天": "date_add(CURRENT_DATE(), 2)",
    }
    if s in _REL_MAP:
        return _REL_MAP[s]
    if var_type == "Date":
        m_date = re.match(r"^(\d{4})-(\d{2})-(\d{2})$", s)
        if m_date:
            y, m, d = int(m_date.group(1)), int(m_date.group(2)), int(m_date.group(3))
            return f"date({y}, {m}, {d})"
    if var_type == "Time":
        m_time = re.match(r"^(\d{1,2}):(\d{2})$", s)
        if m_time:
            h, m = int(m_time.group(1)), int(m_time.group(2))
            return f"time({h}, {m})"
    if re.match(r"^[+-]?\d+$", s):
        return s
    if re.match(r"^[A-Za-z_]\w*$", s):
        return s
    return f'"{s}"'


def stripOuterParens(
    s: str
) -> str:

    s = s.strip()
    if s.startswith("(") and s.endswith(")"):
        depth = 0
        for i, ch in enumerate(s):
            if ch == "(":
                depth += 1
            elif ch == ")":
                depth -= 1
                if depth == 0 and i < len(s) - 1:
                    return s
        return s[1:-1].strip()
    return s


# Pre-compiled patterns for detecting arithmetic expressions (A + B / A - B)
_RE_ARITH_SPACED = re.compile(r'^(.+?)\s+([+-])\s+(.+)$')
_RE_ARITH_NOSPACE = re.compile(r'^([A-Za-z_]\w*)([+-])(\d+|[A-Za-z_]\w*)$')


def isArithExpr(
    expr: str
) -> bool:
    """
        Return True if expr looks like a two-operand arithmetic expression (A ± B).
    """

    s = expr.strip()
    return bool(_RE_ARITH_SPACED.match(s) or _RE_ARITH_NOSPACE.match(s))


def isVarReference(
    expr: str
) -> bool:
    """
        Return True if *expr* looks like a variable name reference
        (as opposed to a literal value or function call).
    """

    s = expr.strip()
    up = s.upper()
    if up in ("TRUE", "FALSE"):
        return False
    if re.match(r"^DATE\(|^TIME\(|^DATE_ADD\(|^TIME_ADD\(|^CURRENT_DATE\(|^CURRENT_TIME\(|^CURRENT_", up):
        return False
    if up.startswith('"') or up.startswith("'"):
        return False
    if re.match(r"^[+-]?\d", s):
        return False
    if isArithExpr(s):
        return False
    return bool(re.match(r"^[A-Z_][A-Z0-9_]*$", up))


def findOperatorIn(
    text: str,
    operators: list
) -> tuple[int, str] | None:

    for op in operators:
        op_upper = op.upper()
        start = 0
        while True:
            idx = text.upper().find(op_upper, start)
            if idx < 0:
                break
            if _isInsideLiteral(text, idx):
                start = idx + 1
                continue
            return (idx, op)
    return None


def _isInsideLiteral(
    text: str,
    pos: int
) -> bool:

    in_single = False
    in_double = False
    for i, ch in enumerate(text):
        if i >= pos:
            break
        if ch == "'" and not in_double:
            in_single = not in_single
        elif ch == '"' and not in_single:
            in_double = not in_double
    return in_single or in_double