fr-cli 2.1.0__py3-none-any.whl

fr_cli/weapon/fs.py

@@ -0,0 +1,206 @@
+"""
+虚拟文件系统 (VFS) - 安全沙盒引擎
+限制AI和用户只能在允许的目录内操作
+"""
+import os
+from pathlib import Path
+from fr_cli.lang.i18n import T
+from fr_cli.ui.ui import GREEN, RED, CYAN, RESET
+
+class VFS:
+    def __init__(self, allowed_dirs):
+        self.ds = [str(Path(d).resolve()) for d in allowed_dirs]
+        self.cwd = self.ds[0] if self.ds else None
+
+    def _resolve(self, p):
+        """安全解析路径，防止../逃逸"""
+        if not self.cwd: return None
+        base = Path(self.cwd)
+        target = (base / p).resolve()
+        # 检查解析后的路径是否仍在允许的目录树内
+        for d in self.ds:
+            base_path = Path(d).resolve()
+            try:
+                # 使用 relative_to 精确判断是否为目标目录的子路径
+                # 可正确处理根目录（/）及避免 /foo 错误匹配 /foo-bar 的前缀问题
+                target.relative_to(base_path)
+                return target
+            except ValueError:
+                continue
+        return None
+
+    def add(self, p, l):
+        try:
+            rp = str(Path(p).resolve())
+            if not os.path.isdir(rp): return False, f"{RED}{T('err_dir_no', l)}{RESET}"
+            if rp not in self.ds:
+                self.ds.append(rp)
+                if not self.cwd: self.cwd = rp
+            return True, f"{GREEN}{T('ok_dir_add', l, rp)}{RESET}"
+        except Exception as e: return False, f"{RED}{e}{RESET}"
+
+    def cd(self, p, l):
+        if not p: return True, f"{GREEN}{self.cwd}{RESET}"
+        # 支持直接切换到已挂载的根目录
+        for d in self.ds:
+            if Path(p).resolve() == Path(d).resolve():
+                self.cwd = d; return True, f"{GREEN}{T('ok_cd', l, self.cwd)}{RESET}"
+        
+        target = self._resolve(p)
+        if not target: return False, f"{RED}{T('err_bound', l)}{RESET}"
+        if not target.is_dir(): return False, f"{RED}{T('err_no_file', l)}{RESET}"
+        self.cwd = str(target)
+        return True, f"{GREEN}{T('ok_cd', l, self.cwd)}{RESET}"
+
+    def ls(self, l):
+        if not self.cwd: return None, f"{RED}{T('no_dir', l)}{RESET}"
+        try:
+            p = Path(self.cwd)
+            items = []
+            for f in p.iterdir():
+                if f.name.startswith('.'): continue
+                items.append(f"{CYAN}{f.name}/" if f.is_dir() else f.name)
+            return sorted(items), None
+        except Exception as e: return None, f"{RED}{e}{RESET}"
+
+    def read(self, fn, l):
+        target = self._resolve(fn)
+        if not target: return None, f"{RED}{T('err_bound', l)}{RESET}"
+        if not target.is_file(): return None, f"{RED}{T('err_no_file', l)}{RESET}"
+        try:
+            # 尝试多种编码读取
+            for enc in ['utf-8', 'gbk', 'latin-1']:
+                try: return target.read_text(encoding=enc), None
+                except UnicodeDecodeError: continue
+            return None, f"{RED}Decode fail{RESET}"
+        except Exception as e: return None, f"{RED}{e}{RESET}"
+    
+    def write(self, fn, content, l, mode='w', encoding='utf-8'):
+        """安全写入文件
+        
+        Args:
+            fn: 文件名
+            content: 文件内容
+            l: 语言
+            mode: 写入模式 ('w'=覆盖, 'a'=追加)
+            encoding: 文件编码
+        
+        Returns:
+            (success, message)
+        """
+        target = self._resolve(fn)
+        if not target: return False, f"{RED}{T('err_bound', l)}{RESET}"
+        
+        try:
+            # 确保父目录存在（覆盖和追加模式都需要）
+            parent = target.parent
+            if not parent.exists():
+                parent.mkdir(parents=True, exist_ok=True)
+            
+            # 写入文件
+            with open(target, mode, encoding=encoding) as f:
+                f.write(content)
+            
+            return True, f"{GREEN}{T('ok_write', l, str(target))}{RESET}"
+        except PermissionError:
+            return False, f"{RED}{T('err_write_perm', l)}{RESET}"
+        except Exception as e:
+            return False, f"{RED}{e}{RESET}"
+    
+    def append(self, fn, content, l, encoding='utf-8'):
+        """追加内容到文件
+        
+        Args:
+            fn: 文件名
+            content: 要追加的内容
+            l: 语言
+            encoding: 文件编码
+        
+        Returns:
+            (success, message)
+        """
+        return self.write(fn, content, l, mode='a', encoding=encoding)
+    
+    def exists(self, fn):
+        """检查文件是否存在
+        
+        Args:
+            fn: 文件名
+        
+        Returns:
+            bool: 文件是否存在
+        """
+        target = self._resolve(fn)
+        return target is not None and target.exists()
+    
+    def delete(self, fn, l):
+        """删除文件
+        
+        Args:
+            fn: 文件名
+            l: 语言
+        
+        Returns:
+            (success, message)
+        """
+        target = self._resolve(fn)
+        if not target: return False, f"{RED}{T('err_bound', l)}{RESET}"
+        if not target.exists(): return False, f"{RED}{T('err_no_file', l)}{RESET}"
+        
+        try:
+            target.unlink()
+            return True, f"{GREEN}{T('ok_delete', l, str(target))}{RESET}"
+        except PermissionError:
+            return False, f"{RED}{T('err_write_perm', l)}{RESET}"
+        except Exception as e:
+            return False, f"{RED}{e}{RESET}"
+
+    def list_dirs(self, l):
+        """列出所有已挂载的工作目录（洞府）
+        
+        Returns:
+            (列表, None) 或 (None, 错误信息)
+        """
+        if not self.ds:
+            return None, f"{RED}{T('no_dir', l)}{RESET}"
+        items = []
+        for i, d in enumerate(self.ds):
+            marker = f" {GREEN}[{T('cur_dir', l)}]{RESET}" if d == self.cwd else ""
+            items.append(f"  [{i}] {CYAN}{d}{RESET}{marker}")
+        return items, None
+
+    def remove_dir(self, p, l):
+        """从允许列表中移除指定工作目录
+        
+        支持按索引或绝对/相对路径删除。
+        若移除的是当前 cwd，自动切换到剩余目录中的第一个。
+        
+        Args:
+            p: 索引字符串或路径
+            l: 语言
+        
+        Returns:
+            (success, message)
+        """
+        if not self.ds:
+            return False, f"{RED}{T('no_dir', l)}{RESET}"
+
+        # 尝试按索引解析
+        try:
+            idx = int(p)
+            if idx < 0 or idx >= len(self.ds):
+                return False, f"{RED}{T('err_dir_idx', l)}{RESET}"
+            removed = self.ds.pop(idx)
+        except ValueError:
+            # 按路径解析
+            rp = str(Path(p).resolve())
+            if rp not in self.ds:
+                return False, f"{RED}{T('err_dir_not_mounted', l, rp)}{RESET}"
+            self.ds.remove(rp)
+            removed = rp
+
+        # 若删除的是当前 cwd，自动切换
+        if self.cwd == removed:
+            self.cwd = self.ds[0] if self.ds else None
+
+        return True, f"{GREEN}{T('ok_dir_remove', l, removed)}{RESET}"

fr_cli/weapon/launcher.py

@@ -0,0 +1,249 @@
+"""
+本地应用启动器 —— 驭器神通
+支持跨平台调用浏览器、办公软件、通讯工具等本机程序。
+"""
+import platform
+import subprocess
+
+SYSTEM = platform.system()
+
+# 常用应用别名映射表 —— 按平台映射到可执行命令或应用包名
+_APP_ALIASES = {
+    "Darwin": {
+        # 浏览器
+        "chrome": "Google Chrome",
+        "googlechrome": "Google Chrome",
+        "safari": "Safari",
+        "firefox": "Firefox",
+        "edge": "Microsoft Edge",
+        "browser": "Safari",
+        "浏览器": "Safari",
+        # 办公
+        "word": "Microsoft Word",
+        "msword": "Microsoft Word",
+        "excel": "Microsoft Excel",
+        "msexcel": "Microsoft Excel",
+        "powerpoint": "Microsoft PowerPoint",
+        "ppt": "Microsoft PowerPoint",
+        "mspowerpoint": "Microsoft PowerPoint",
+        "wps": "WPS Office",
+        # 通讯
+        "wechat": "WeChat",
+        "微信": "WeChat",
+        "qq": "QQ",
+        "tim": "Tencent TIM",
+        "dingtalk": "DingTalk",
+        "钉钉": "DingTalk",
+        "飞书": "Lark",
+        "lark": "Lark",
+        # 编辑器 / 工具
+        "notepad": "TextEdit",
+        "textedit": "TextEdit",
+        "记事本": "TextEdit",
+        "notes": "Notes",
+        "备忘录": "Notes",
+        "vscode": "Visual Studio Code",
+        "code": "Visual Studio Code",
+        "terminal": "Terminal",
+        "iterm": "iTerm",
+        "终端": "Terminal",
+        "计算器": "Calculator",
+        "calc": "Calculator",
+        "calculator": "Calculator",
+        # 媒体
+        "music": "Music",
+        "itunes": "Music",
+        "播放器": "Music",
+        "spotify": "Spotify",
+        "vlc": "VLC",
+        "quicktime": "QuickTime Player",
+        "photos": "Photos",
+        "相册": "Photos",
+        # 系统
+        "finder": "Finder",
+        "访达": "Finder",
+        "systempreferences": "System Preferences",
+        "系统偏好设置": "System Preferences",
+        "appstore": "App Store",
+        "app store": "App Store",
+        "地图": "Maps",
+        "maps": "Maps",
+    },
+    "Windows": {
+        # 浏览器
+        "chrome": "chrome",
+        "googlechrome": "chrome",
+        "edge": "msedge",
+        "firefox": "firefox",
+        "browser": "start microsoft-edge:",
+        "浏览器": "start microsoft-edge:",
+        # 办公
+        "word": "winword",
+        "msword": "winword",
+        "excel": "excel",
+        "msexcel": "excel",
+        "powerpoint": "powerpnt",
+        "ppt": "powerpnt",
+        "mspowerpoint": "powerpnt",
+        "wps": "wps",
+        # 通讯
+        "wechat": "WeChat",
+        "微信": "WeChat",
+        "qq": "QQ",
+        "tim": "TIM",
+        "dingtalk": "DingTalk",
+        "钉钉": "DingTalk",
+        "飞书": "Lark",
+        "lark": "Lark",
+        # 编辑器 / 工具
+        "notepad": "notepad",
+        "记事本": "notepad",
+        "vscode": "code",
+        "code": "code",
+        "terminal": "wt",
+        "终端": "wt",
+        "计算器": "calc",
+        "calc": "calc",
+        "calculator": "calc",
+        # 媒体
+        "music": "mswindowsmusic:",
+        "itunes": "itunes",
+        "播放器": "mswindowsmusic:",
+        "spotify": "spotify",
+        "vlc": "vlc",
+        "photos": "ms-photos:",
+        "相册": "ms-photos:",
+        # 系统
+        "explorer": "explorer",
+        "文件资源管理器": "explorer",
+        "settings": "ms-settings:",
+        "系统设置": "ms-settings:",
+        "appstore": "ms-windows-store:",
+        "app store": "ms-windows-store:",
+    },
+    "Linux": {
+        # 浏览器
+        "chrome": "google-chrome",
+        "googlechrome": "google-chrome",
+        "chromium": "chromium-browser",
+        "firefox": "firefox",
+        "edge": "microsoft-edge",
+        "browser": "xdg-open",
+        "浏览器": "xdg-open",
+        # 办公
+        "word": "libreoffice --writer",
+        "excel": "libreoffice --calc",
+        "powerpoint": "libreoffice --impress",
+        "ppt": "libreoffice --impress",
+        "wps": "wps",
+        # 通讯
+        "wechat": "wechat",
+        "微信": "wechat",
+        "qq": "qq",
+        "lark": "lark",
+        "飞书": "lark",
+        # 编辑器 / 工具
+        "notepad": "gedit",
+        "记事本": "gedit",
+        "vscode": "code",
+        "code": "code",
+        "terminal": "gnome-terminal",
+        "终端": "gnome-terminal",
+        "计算器": "gnome-calculator",
+        "calc": "gnome-calculator",
+        "calculator": "gnome-calculator",
+        # 媒体
+        "vlc": "vlc",
+        "播放器": "vlc",
+        "spotify": "spotify",
+        # 系统
+        "files": "nautilus",
+        "文件管理器": "nautilus",
+        "settings": "gnome-control-center",
+        "系统设置": "gnome-control-center",
+    },
+}
+
+
+def _resolve_app(name):
+    """将别名解析为平台对应的应用标识"""
+    key = name.lower().strip()
+    plat = SYSTEM
+    aliases = _APP_ALIASES.get(plat, {})
+    return aliases.get(key, name)
+
+
+def open_file(path, lang="zh"):
+    """用系统默认程序打开文件或 URL（跨平台）"""
+    if not path:
+        return False, "路径为空" if lang == "zh" else "Empty path"
+    try:
+        if SYSTEM == "Darwin":
+            subprocess.Popen(["open", path], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        elif SYSTEM == "Windows":
+            subprocess.Popen(["start", "", path], shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        else:
+            subprocess.Popen(["xdg-open", path], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        return True, f"已打开: {path}" if lang == "zh" else f"Opened: {path}"
+    except Exception as e:
+        return False, str(e)
+
+
+def launch_app(name, target=None, lang="zh"):
+    """启动指定应用程序，可选传入文件/URL 作为目标"""
+    if not name:
+        return False, "应用名称为空" if lang == "zh" else "Empty app name"
+
+    app = _resolve_app(name)
+
+    try:
+        if SYSTEM == "Darwin":
+            cmd = ["open", "-a", app]
+            if target:
+                cmd.append(target)
+            subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        elif SYSTEM == "Windows":
+            # Windows 某些应用支持直接传目标，某些用 start
+            if target:
+                # 如果是 URL，直接用 start；如果是文件，用应用打开
+                if target.startswith("http"):
+                    subprocess.Popen(["start", "", target], shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                else:
+                    subprocess.Popen(["start", "", "", app, target], shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            else:
+                # 某些 Windows 命令本身就是 URI 协议或 shell 命令
+                if app.endswith(":") or " " in app:
+                    subprocess.Popen([app], shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                else:
+                    subprocess.Popen(["start", "", app], shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        else:
+            # Linux
+            parts = app.split()
+            cmd = parts.copy()
+            if target:
+                cmd.append(target)
+            subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+        msg = f"已启动: {name}" if lang == "zh" else f"Launched: {name}"
+        if target:
+            msg += f" ({target})"
+        return True, msg
+    except Exception as e:
+        return False, str(e)
+
+
+def list_apps(lang="zh"):
+    """列出当前平台支持的应用别名"""
+    plat = SYSTEM
+    aliases = _APP_ALIASES.get(plat, {})
+    if not aliases:
+        return None, "暂无预置应用列表" if lang == "zh" else "No app list available"
+
+    items = []
+    seen = set()
+    for alias, real in sorted(aliases.items()):
+        if real not in seen:
+            seen.add(real)
+            items.append(f"  {real:<30} ← {alias}")
+    header = "本机可用应用映射:" if lang == "zh" else "Available app mappings:"
+    return header + "\n" + "\n".join(items), None

fr_cli/weapon/loader.py

@@ -0,0 +1,98 @@
+"""
+法宝图谱加载器
+从统一注册表获取工具信息，替代 WEAPON.MD 的文本解析。
+保留 WEAPON.MD 作为人类可读文档，程序逻辑不再依赖其解析。
+"""
+from fr_cli.command.registry import get_registry
+
+
+# 旧类别到注册表工具名的映射（用于兼容旧接口）
+_LEGACY_CATEGORIES = {
+    "file_operations": (["write_file", "read_file", "list_files", "change_dir", "append_file", "delete_file"], "fr_cli/weapon/fs.py"),
+    "image_analysis": (["analyze_image", "generate_image"], "fr_cli/weapon/vision.py"),
+    "email_management": (["mail_inbox", "mail_read", "mail_send"], "fr_cli/weapon/mail.py"),
+    "web_search": (["search_web", "fetch_web"], "fr_cli/weapon/web.py"),
+    "scheduled_tasks": (["cron_add", "cron_list", "cron_del"], "fr_cli/weapon/cron.py"),
+    "cloud_storage": (["disk_ls", "disk_up", "disk_down"], "fr_cli/weapon/disk.py"),
+    "session_management": (["save_session", "list_sessions", "export_session"], "fr_cli/memory/history.py"),
+    "configuration": (["set_model", "set_key", "set_limit", "set_lang"], "fr_cli/conf/config.py"),
+    "launcher_system": (["open_file", "launch_app", "list_apps"], "fr_cli/weapon/launcher.py"),
+    "agent_system": (["agent_create", "agent_run"], "fr_cli/agent/"),
+    "data_scroll": (["read_excel", "read_csv"], "fr_cli/weapon/dataframe.py"),
+}
+
+
+def load_weapon_md(mcp_tools=None):
+    """
+    从统一注册表获取法宝图谱。
+    保持返回格式兼容旧接口：(tools:list, trigger_map:dict)
+    :param mcp_tools: MCP 外部神通列表，可选
+    """
+    reg = get_registry()
+    reg_tools = {t["name"]: t for t in reg.get_available_tools(plugins={})}
+
+    tools = []
+    trigger_map = {}
+
+    for cat_name, (tool_names, path) in _LEGACY_CATEGORIES.items():
+        commands = []
+        triggers = []
+        descriptions = []
+        for tn in tool_names:
+            if tn in reg_tools:
+                commands.append(tn)
+                info = reg_tools[tn]
+                if info.get("description"):
+                    descriptions.append(info["description"])
+                if info.get("triggers"):
+                    triggers.extend(info["triggers"])
+        if commands:
+            tools.append({
+                "name": cat_name,
+                "description": ", ".join(descriptions) if descriptions else cat_name,
+                "commands": commands,
+                "path": path,
+            })
+            # 去重 triggers
+            seen = set()
+            unique_triggers = []
+            for t in triggers:
+                if t not in seen:
+                    seen.add(t)
+                    unique_triggers.append(t)
+            trigger_map[cat_name] = unique_triggers
+
+    # 注入 MCP 外部神通
+    if mcp_tools:
+        mcp_commands = [t["name"] for t in mcp_tools]
+        tools.append({
+            "name": "mcp_tools",
+            "description": "MCP 外部神通: " + ", ".join([t["name"] for t in mcp_tools]),
+            "commands": mcp_commands,
+            "path": "fr_cli/weapon/mcp.py",
+        })
+        trigger_map["mcp_tools"] = ["mcp", "外部工具"]
+
+    return tools, trigger_map
+
+
+def get_available_tools(weapon_tools, plugins):
+    """
+    获取当前可用的工具列表（从传入的 weapon_tools 追加插件）
+    保持与旧接口完全兼容。
+    """
+    tools = [t.copy() for t in weapon_tools]
+    if plugins:
+        for plugin_name in plugins.keys():
+            tools.append({
+                "name": f"plugin_{plugin_name}",
+                "description": f"自定义插件: {plugin_name}",
+                "commands": [f"/{plugin_name}"],
+            })
+    return tools
+
+
+# should_inject_tools 已移除：
+# 主程序逻辑中从未调用此函数（main.py 使用 _should_force_tool + _classify_intent 做意图判定）。
+# MasterAgent 模式下直接注入全部工具列表，无需触发词匹配。
+# 保留 load_weapon_md 和 get_available_tools 作为兼容层。