bitool 0.1.2__py3-none-any.whl

bitool/core/app.py

@@ -0,0 +1,142 @@
+"""核心应用入口 - 协调 GUI 和 Web 两种模式."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Any
+
+from .commands import CommandCategory, CommandRegistry
+from .plugin import PluginContext
+from .plugin import PluginRegistry as PluginRegistryCore
+from .workspace import WorkspaceManager
+
+
+class BitoolApp:
+    """Bitool 主应用."""
+
+    def __init__(self, config: Any = None) -> None:
+        """初始化应用.
+
+        Args:
+            config: 应用配置（可选）
+        """
+        self._config = config
+        self._workspace = WorkspaceManager()
+        self._plugin_registry = PluginRegistryCore()
+        self._command_registry = CommandRegistry()
+
+        # 设置插件上下文
+        self._plugin_context = PluginContext(
+            workspace=self._workspace,
+            config=self._config,
+            registry=self._plugin_registry,
+        )
+
+    @property
+    def config(self) -> Any:
+        """获取配置."""
+        return self._config
+
+    @property
+    def workspace(self) -> WorkspaceManager:
+        """获取工作空间管理器."""
+        return self._workspace
+
+    @property
+    def registry(self) -> CommandRegistry:
+        """获取命令注册表."""
+        return self._command_registry
+
+    def start(self, mode: str = "gui") -> None:
+        """启动应用.
+
+        Args:
+            mode: 运行模式 (gui/web)
+        """
+        # 加载插件
+        self._load_plugins()
+
+        # 注册内建命令
+        self._register_builtin_commands()
+
+        # 创建模式
+        if mode == "gui":
+            self._run_gui()
+        elif mode == "web":
+            self._run_web()
+        else:
+            raise ValueError(f"未知模式: {mode}，支持的模式: gui, web")
+
+    def _run_gui(self) -> None:
+        """运行 GUI 模式."""
+        # TODO: 实现 GUI 启动逻辑
+
+    def _run_web(self, host: str = "127.0.0.1", port: int = 8000) -> None:
+        """运行 Web 模式.
+
+        Args:
+            host: 监听地址
+            port: 监听端口
+        """
+
+        # TODO: 实现 Web 启动逻辑
+
+    def _load_plugins(self) -> None:
+        """加载插件."""
+        # 从配置目录加载插件
+        plugin_dirs = ["~/.bitool/plugins"]
+        if self._config and hasattr(self._config, "plugin_dirs"):
+            plugin_dirs = self._config.plugin_dirs
+
+        for plugin_dir in plugin_dirs:
+            self._plugin_registry.load_plugins_from_directory(plugin_dir)
+
+        # 初始化所有插件
+        for plugin_name in self._plugin_registry.list_plugins():
+            plugin = self._plugin_registry.get_plugin(plugin_name)
+            if plugin and hasattr(plugin, "initialize"):
+                plugin.initialize(self._plugin_context)
+
+    def _register_builtin_commands(self) -> None:
+        """注册内建命令."""
+
+        def system_info(**kwargs: Any) -> str:
+            """显示系统信息."""
+            plugins = ", ".join(self._plugin_registry.list_plugins()) or "无"
+            return f"Bitool Maturin v0.1.0\n工作空间: {self.workspace.root}\n插件: {plugins}"
+
+        self.registry.register(
+            name="system_info",
+            description="显示系统信息",
+            category=CommandCategory.SYSTEM,
+            handler=system_info,
+        )
+
+        def health_check(**kwargs: Any) -> str:
+            """系统健康检查."""
+            issues = []
+
+            # 检查工作空间
+            workspace_root = self.workspace.root
+            if workspace_root and not Path(workspace_root).exists():
+                issues.append(f"工作空间目录不存在: {workspace_root}")
+
+            if issues:
+                return "健康检查发现问题:\n" + "\n".join(f"  - {issue}" for issue in issues)
+            return "系统健康状态: 正常"
+
+        self.registry.register(
+            name="health_check",
+            description="系统健康检查",
+            category=CommandCategory.SYSTEM,
+            handler=health_check,
+        )
+
+    def run(self, mode: str = "gui", **kwargs: Any) -> None:
+        """运行应用 (同步入口).
+
+        Args:
+            mode: 运行模式
+            **kwargs: 其他参数
+        """
+        self.start(mode)

bitool/core/commands.py

@@ -0,0 +1,194 @@
+"""命令注册与执行系统."""
+
+from __future__ import annotations
+
+import asyncio
+import inspect
+from dataclasses import dataclass, field
+from enum import Enum
+from typing import Any, Callable
+
+
+class CommandCategory(str, Enum):
+    """命令类别."""
+
+    PDF = "pdf"
+    WORKSPACE = "workspace"
+    SYSTEM = "system"
+    FILE = "file"
+    PLUGIN = "plugin"
+
+
+@dataclass
+class CommandResult:
+    """命令执行结果."""
+
+    success: bool
+    output: str = ""
+    error: str = ""
+    data: Any = None
+
+
+@dataclass
+class CommandInfo:
+    """命令信息."""
+
+    name: str
+    description: str
+    category: CommandCategory
+    handler: Callable
+    parameters: dict[str, Any] = field(default_factory=dict)
+
+
+class CommandRegistry:
+    """命令注册表."""
+
+    def __init__(self) -> None:
+        """初始化命令注册表."""
+        self._commands: dict[str, CommandInfo] = {}
+
+    def register(
+        self,
+        name: str,
+        description: str,
+        category: CommandCategory,
+        handler: Callable,
+        parameters: dict[str, Any] | None = None,
+    ) -> None:
+        """注册命令.
+
+        Args:
+            name: 命令名称
+            description: 命令描述
+            category: 命令类别
+            handler: 命令处理器
+            parameters: 命令参数
+        """
+        if name in self._commands:
+            raise ValueError(f"命令 {name} 已注册")
+
+        self._commands[name] = CommandInfo(
+            name=name,
+            description=description,
+            category=category,
+            handler=handler,
+            parameters=parameters or {},
+        )
+
+    def unregister(self, name: str) -> None:
+        """注销命令.
+
+        Args:
+            name: 命令名称
+        """
+        self._commands.pop(name, None)
+
+    def get(self, name: str) -> CommandInfo | None:
+        """获取命令信息.
+
+        Args:
+            name: 命令名称
+
+        Returns:
+            命令信息，如果不存在则返回 None
+        """
+        return self._commands.get(name)
+
+    def list_commands(self, category: CommandCategory | None = None) -> list[CommandInfo]:
+        """列出命令.
+
+        Args:
+            category: 命令类别（可选）
+
+        Returns:
+            命令信息列表
+        """
+        if category is None:
+            return list(self._commands.values())
+        return [cmd for cmd in self._commands.values() if cmd.category == category]
+
+    def list(self, category: CommandCategory | None = None) -> list[CommandInfo]:
+        """列出命令 (别名)."""
+        return self.list_commands(category)
+
+    def list_names(self, category: CommandCategory | None = None) -> list[str]:
+        """列出命令名称.
+
+        Args:
+            category: 命令类别（可选）
+
+        Returns:
+            命令名称列表
+        """
+        return [cmd.name for cmd in self.list_commands(category)]
+
+
+class CommandExecutor:
+    """命令执行器."""
+
+    def __init__(self, registry: CommandRegistry) -> None:
+        """初始化命令执行器.
+
+        Args:
+            registry: 命令注册表
+        """
+        self._registry = registry
+
+    async def execute(
+        self,
+        name: str,
+        **kwargs: Any,
+    ) -> CommandResult:
+        """执行命令.
+
+        Args:
+            name: 命令名称
+            **kwargs: 命令参数
+
+        Returns:
+            命令执行结果
+        """
+        cmd_info = self._registry.get(name)
+        if cmd_info is None:
+            return CommandResult(
+                success=False,
+                error=f"命令 '{name}' 不存在",
+            )
+
+        handler = cmd_info.handler
+
+        try:
+            # 判断是否是异步函数
+            if inspect.iscoroutinefunction(handler):
+                result = await handler(**kwargs)
+            else:
+                # 在事件循环中运行同步函数
+                loop = asyncio.get_event_loop()
+                result = await loop.run_in_executor(None, lambda: handler(**kwargs))
+
+            if isinstance(result, CommandResult):
+                return result
+            return CommandResult(success=True, output=str(result))
+
+        except Exception as e:  # noqa: BLE001
+            return CommandResult(
+                success=False,
+                error=f"执行命令 '{name}' 时出错: {e}",
+            )
+
+    def execute_sync(self, name: str, **kwargs: Any) -> CommandResult:
+        """同步执行命令 (用于本地模式).
+
+        Args:
+            name: 命令名称
+            **kwargs: 命令参数
+
+        Returns:
+            命令执行结果
+        """
+        try:
+            return asyncio.run(self.execute(name, **kwargs))
+        except RuntimeError:
+            # 如果已有事件循环运行，使用当前循环
+            loop = asyncio.get_event_loop()
+            return loop.run_until_complete(self.execute(name, **kwargs))