feat: 支持在脚本链中开关隐私屏

.gitignore

@@ -13,4 +13,6 @@ __pycache__/
 config/script_chain/
 config/custom.yml
 config/push.yml
-config/env.yml
+config/env.yml
+config/run.yml
+.pai/

src/one_dragon/base/controller/pc_button/pc_button_listener.py

@@ -0,0 +1,83 @@
+from collections.abc import Callable
+from concurrent.futures import Future, ThreadPoolExecutor
+
+from pynput import keyboard, mouse
+
+from one_dragon.utils import thread_utils
+
+
+class PcButtonListener:
+
+    def __init__(self,
+                 on_button_tap: Callable[[str], None],
+                 listen_keyboard: bool = False,
+                 listen_mouse: bool = False,
+                 listen_gamepad: bool = False,
+                 ):
+        self.keyboard_listener = keyboard.Listener(on_press=self._on_keyboard_press)
+        self.mouse_listener = mouse.Listener(on_click=self._on_mouse_click)
+        self.gamepad_listener = None
+
+        self.on_button_tap: Callable[[str], None] = on_button_tap
+
+        self.listen_keyboard: bool = listen_keyboard
+        self.listen_mouse: bool = listen_mouse
+        self.listen_gamepad: bool = listen_gamepad
+
+        self._executor = ThreadPoolExecutor(thread_name_prefix='od_key_mouse_btn_listener', max_workers=8)
+
+    def _on_keyboard_press(self, event):
+        if isinstance(event, keyboard.Key):
+            k = event.name
+        elif isinstance(event, keyboard.KeyCode):
+            # 处理小键盘按键和特殊按键
+            if event.char is not None:
+                k = event.char
+            elif hasattr(event, 'vk') and event.vk is not None:
+                # 使用虚拟键码来识别小键盘按键
+                k = self._get_numpad_key_name(event.vk)
+            elif hasattr(event, 'vk'):
+                k = f'vk_{event.vk}'  # vk 为 None 的情况
+            else:
+                k = 'unknown'  # 没有 vk 属性
+        else:
+            return
+
+        # 确保按键名称不为空
+        if k is None or k == '':
+            return
+
+        self._call_button_tap_callback(k)
+
+    def _get_numpad_key_name(self, vk: int) -> str:
+        """
+        根据虚拟键码获取小键盘按键名称
+        :param vk: 虚拟键码
+        :return: 按键名称
+        """
+        # 小键盘数字键: vk 96-105 对应 numpad_0 到 numpad_9
+        if 96 <= vk <= 105:
+            return f'numpad_{vk - 96}'
+
+        # 其他按键返回通用格式
+        return f'vk_{vk}'
+
+    def _on_mouse_click(self, x, y, button: mouse.Button, pressed):
+        if pressed == 1:
+            self._call_button_tap_callback('mouse_' + button.name)
+
+    def _call_button_tap_callback(self, key: str) -> None:
+        if self.on_button_tap is not None:
+            future: Future = self._executor.submit(self.on_button_tap, key)
+            future.add_done_callback(thread_utils.handle_future_result)
+
+    def start(self):
+        if self.listen_keyboard:
+            self.keyboard_listener.start()
+        if self.listen_mouse:
+            self.mouse_listener.start()
+
+    def stop(self):
+        self.keyboard_listener.stop()
+        self.mouse_listener.stop()
+        self._executor.shutdown(wait=False, cancel_futures=True)

src/one_dragon_qt/widgets/setting_card/key_setting_card.py

@@ -36,6 +36,7 @@ def __init__(self,
                  icon: Union[str, QIcon, FluentIconBase], title: str, content: Optional[str]=None,
                  icon_size: IconSize = IconSize(16, 16),
                  margins: Margins = Margins(16, 16, 0, 16),
+                 listen_mouse: bool = True,
                  parent=None):
 
         SettingCardBase.__init__(
@@ -56,6 +57,7 @@ def __init__(self,
         self.btn.clicked.connect(self._on_btn_clicked)
 
         # 初始化监听器和键盘事件工作者
+        self.listen_mouse: bool = listen_mouse
         self.button_listener = None  # 按键监听
         self.key_worker = KeyEventWorker()
         self.key_worker.key_pressed.connect(self._on_key_signal)
@@ -71,7 +73,7 @@ def _on_btn_clicked(self) -> None:
         """
         if self.button_listener is None:
             self.button_listener = PcButtonListener(
-                self._on_key_press, listen_keyboard=True, listen_mouse=True
+                self._on_key_press, listen_keyboard=True, listen_mouse=self.listen_mouse
             )
             self.btn.setText(gt('请按键'))
             self.button_listener.start()

src/script_chainer/config/run_config.py

@@ -0,0 +1,21 @@
+from one_dragon.base.config.yaml_config import YamlConfig
+
+
+class RunConfig(YamlConfig):
+    """千机链专有的运行行为配置，落在 `config/run.yml`。"""
+
+    def __init__(self, is_mock: bool = False):
+        YamlConfig.__init__(
+            self,
+            'run',
+            is_mock=is_mock, sample=False, copy_from_sample=False,
+        )
+
+    @property
+    def privacy_screen_close_key(self) -> str:
+        """隐私屏开关热键的主键（修饰键固定为 Ctrl+Alt+Shift）。"""
+        return self.get('privacy_screen_close_key', 'f12')
+
+    @privacy_screen_close_key.setter
+    def privacy_screen_close_key(self, new_value: str) -> None:
+        self.update('privacy_screen_close_key', new_value)

src/script_chainer/config/script_config.py

@@ -31,6 +31,12 @@ class ScriptProcessName(Enum):
     OK_SCRIPTS = ConfigItem(label='OK', value=['pythonw.exe'])
 
 
+class ComputerOperation(Enum):
+
+    OPEN_PRIVACY_SCREEN = ConfigItem(label='打开隐私屏', value='open_privacy_screen', desc='打开全屏隐私屏，屏幕变黑且截图录屏拍不到画面')
+    CLOSE_PRIVACY_SCREEN = ConfigItem(label='关闭隐私屏', value='close_privacy_screen', desc='关闭已打开的隐私屏，恢复屏幕显示')
+
+
 class GameProcessName(Enum):
 
     GENSHIN_IMPACT_CN = ConfigItem(label='原神', value='YuanShen.exe')
@@ -47,6 +53,7 @@ class GameProcessName(Enum):
 class ScriptType:
     EXTERNAL = 'external'
     PYTHON = 'python'
+    COMPUTER_OP = 'computer_op'
 
 
 class AttachDirection:
@@ -141,6 +148,7 @@ class ScriptConfig:
     game_label: str = ''
     script_type: str = ScriptType.EXTERNAL
     script_path: str = ''
+    computer_operation: str = ''
     script_process_name: list[str] = field(default_factory=list)
     game_process_name: str = ''
     launcher_mode: bool = False
@@ -180,6 +188,15 @@ def create_python_default(cls) -> 'ScriptConfig':
             notify_done=False,
         )
 
+    @classmethod
+    def create_computer_op_default(cls) -> 'ScriptConfig':
+        """创建电脑操作类型的默认配置。"""
+        return cls(
+            script_type=ScriptType.COMPUTER_OP,
+            notify_start=False,
+            notify_done=False,
+        )
+
     def copy(self) -> 'ScriptConfig':
         """深拷贝（保留 idx）。"""
         new = ScriptConfig(**self.to_dict())
@@ -243,8 +260,20 @@ def check_done_display_name(self) -> str:
         else:
             return ''
 
+    @property
+    def computer_operation_display_name(self) -> str:
+        config = get_config_item_from_enum(ComputerOperation, self.computer_operation)
+        if config is not None:
+            return config.label
+        return ''
+
     @property
     def invalid_message(self) -> str | None:
+        if self.script_type == ScriptType.COMPUTER_OP:
+            if get_config_item_from_enum(ComputerOperation, self.computer_operation) is None:
+                return '未选择电脑操作'
+            return None
+
         if self.script_type == ScriptType.PYTHON:
             if not self.script_path:
                 return 'Python 脚本路径为空'
@@ -433,6 +462,18 @@ def add_one(self) -> ScriptConfig:
         self.save()
         return new_config
 
+    def add_computer_op(self) -> ScriptConfig:
+        """新增一个电脑操作配置并返回。
+
+        Returns:
+            新创建的 ScriptConfig。
+        """
+        new_config = ScriptConfig.create_computer_op_default()
+        self.script_list.append(new_config)
+        self.init_idx()
+        self.save()
+        return new_config
+
     def delete_one(self, index: int) -> None:
         """删除一个配置。
 

src/script_chainer/context/script_chainer_context.py

@@ -10,6 +10,7 @@
 from one_dragon.envs.project_config import ProjectConfig
 from one_dragon.utils import os_utils
 from one_dragon.utils.log_utils import log
+from script_chainer.config.run_config import RunConfig
 from script_chainer.config.script_config import ScriptChainConfig
 from script_chainer.services.github_update_service import GithubUpdateService
 
@@ -26,6 +27,7 @@ def __init__(self):
         self.github_update_service: GithubUpdateService = GithubUpdateService(self)
         self.push_service: PushService = PushService(self)
         self.notify_config: NotifyConfig = NotifyConfig(None, {})
+        self.run_config: RunConfig = RunConfig()
         self._init_lock = threading.Lock()
 
     def init(self) -> None:

src/script_chainer/gui/page/editor_setting_interface.py

@@ -23,6 +23,7 @@
 from one_dragon_qt.widgets.setting_card.combo_box_setting_card import (
     ComboBoxSettingCard,
 )
+from one_dragon_qt.widgets.setting_card.key_setting_card import KeySettingCard
 from one_dragon_qt.widgets.setting_card.multi_push_setting_card import (
     MultiPushSettingCard,
 )
@@ -280,6 +281,7 @@ def get_content_widget(self) -> QWidget:
         content_widget = Column()
 
         content_widget.add_widget(self.get_appearance_group())
+        content_widget.add_widget(self.get_run_group())
         content_widget.add_widget(self.get_update_group())
 
         content_widget.add_stretch(1)
@@ -302,6 +304,19 @@ def get_appearance_group(self) -> SettingCardGroup:
 
         return group
 
+    def get_run_group(self) -> SettingCardGroup:
+        group = SettingCardGroup(gt('运行'))
+
+        self.privacy_screen_key_input = KeySettingCard(
+            icon=FluentIcon.POWER_BUTTON,
+            title='隐私屏开关快捷键',
+            content='组合键固定 Ctrl+Alt+Shift，按下后在显示和隐藏之间切换；点击右侧按钮后按下主键完成录制',
+            listen_mouse=False,
+        )
+        group.addSettingCard(self.privacy_screen_key_input)
+
+        return group
+
     def get_update_group(self) -> SettingCardGroup:
         group = SettingCardGroup(gt('更新'))
 
@@ -359,6 +374,9 @@ def get_update_group(self) -> SettingCardGroup:
     def on_interface_shown(self) -> None:
         VerticalScrollInterface.on_interface_shown(self)
         self.theme_opt.init_with_adapter(self.ctx.custom_config.get_prop_adapter('theme'))
+        self.privacy_screen_key_input.init_with_adapter(
+            self.ctx.run_config.get_prop_adapter('privacy_screen_close_key')
+        )
         self.proxy_type_opt.init_with_adapter(self.ctx.env_config.get_prop_adapter('proxy_type'))
         self.personal_proxy_input.init_with_adapter(self.ctx.env_config.get_prop_adapter('personal_proxy'))
         self.gh_proxy_url_opt.init_with_adapter(self.ctx.env_config.get_prop_adapter('gh_proxy_url'))

src/script_chainer/gui/page/script_setting_cards.py

@@ -19,7 +19,7 @@
     ScriptChainConfig,
     ScriptConfig,
 )
-from script_chainer.gui.page.script_setting_dialogs import ScriptRenameDialog
+from script_chainer.gui.page.script_setting_dialogs import ComputerOpEditDialog, ScriptRenameDialog
 from script_chainer.gui.page.script_setting_utils import (
     show_error,
     show_success,
@@ -344,3 +344,90 @@ def _update_display(self) -> None:
         self._external_tag.setVisible(is_external)
         self.edit_btn.setIcon(FluentIcon.FOLDER.icon() if is_external else FluentIcon.EDIT.icon())
         self.edit_btn.setToolTip('定位文件' if is_external else '编辑')
+
+
+class ComputerOpSettingCard(ScriptCardMixin, DraggableListItem):
+    """电脑操作卡片，通过铅笔编辑选择要执行的操作。"""
+
+    value_changed = Signal(ScriptConfig)
+    deleted = Signal(int)
+
+    def __init__(self, config: ScriptConfig, chain_name: str, index: int = 0, parent=None,
+                 enable_opacity_effect: bool = True):
+        self.config: ScriptConfig = config
+        self.chain_name: str = chain_name
+        self._setup_common_widgets()
+
+        self.debug_btn = TransparentToolButton(FluentIcon.PLAY, None)
+        self.debug_btn.setToolTip('调试运行')
+        self.debug_btn.clicked.connect(self.on_debug_clicked)
+
+        content_widget = MultiPushSettingCard(
+            icon=FluentIcon.COMMAND_PROMPT,
+            title=self._get_title(),
+            content=self._get_operation_display(),
+            parent=parent,
+            btn_list=[
+                self.debug_btn,
+                self.edit_btn,
+                self.delete_btn,
+                self.enable_switch,
+            ]
+        )
+
+        self._setup_rename_btn(content_widget)
+
+        DraggableListItem.__init__(
+            self,
+            data=config,
+            index=index,
+            content_widget=content_widget,
+            parent=parent,
+            enable_opacity_effect=enable_opacity_effect,
+        )
+
+        self.content_widget: MultiPushSettingCard
+        self._update_display()
+
+    def _get_title(self) -> str:
+        if self.config.display_name:
+            return self.config.display_name
+        return '电脑操作'
+
+    def _get_operation_display(self) -> str:
+        label = self.config.computer_operation_display_name
+        return label if label else '未设置'
+
+    def on_debug_clicked(self) -> None:
+        """调试运行当前电脑操作"""
+        invalid_msg = self.config.invalid_message
+        if invalid_msg is not None:
+            show_warning(self.window(), '配置不合法', invalid_msg)
+            return
+
+        display = self._get_title()
+
+        try:
+            cmd, cwd = build_runner_command(self.chain_name, self.index)
+            launch_in_terminal(
+                command=cmd,
+                cwd=cwd,
+                title=f'调试 {display}',
+            )
+            show_success(self.window(), '调试运行', f'已在终端启动 {display}')
+        except Exception as e:
+            show_error(self.window(), '启动失败', str(e))
+
+    def on_edit_clicked(self) -> None:
+        """弹窗选择要执行的电脑操作。"""
+        dialog = ComputerOpEditDialog(self.config.computer_operation, parent=self.window())
+        if dialog.exec():
+            self.config.computer_operation = dialog.get_operation()
+            self._update_display()
+            self.value_changed.emit(self.config)
+
+    def _update_display(self) -> None:
+        """更新卡片显示内容"""
+        self.content_widget.setTitle(self._get_title())
+        self.content_widget.setContent(self._get_operation_display())
+        self.enable_switch.setChecked(self.config.enabled)