jarvis-ai-assistant 0.3.17__py3-none-any.whl → 0.3.19__py3-none-any.whl

jarvis/jarvis_smart_shell/main.py

@@ -10,6 +10,7 @@ from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.config import get_shell_name, set_config
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 app = typer.Typer(
     help="将自然语言要求转换为shell命令",
@@ -58,38 +59,95 @@ def _get_markers() -> Tuple[str, str]:
     )
 
 
+def _check_bash_shell() -> bool:
+    """Check if current shell is bash
+
+    Returns:
+        bool: True if bash shell, False otherwise
+    """
+    return get_shell_name() == "bash"
+
+
+def _get_bash_config_file() -> str:
+    """Get bash config file path
+
+    Returns:
+        str: Path to bash config file (~/.bashrc)
+    """
+    return os.path.expanduser("~/.bashrc")
+
+
+def _get_bash_markers() -> Tuple[str, str]:
+    """Get start and end markers for JSS completion in bash
+
+    Returns:
+        Tuple[str, str]: (start_marker, end_marker)
+    """
+    return (
+        "# ===== JARVIS JSS BASH COMPLETION START =====",
+        "# ===== JARVIS JSS BASH COMPLETION END =====",
+    )
+
+
+def _check_zsh_shell() -> bool:
+    """Check if current shell is zsh
+
+    Returns:
+        bool: True if zsh shell, False otherwise
+    """
+    return get_shell_name() == "zsh"
+
+
+def _get_zsh_config_file() -> str:
+    """Get zsh config file path
+
+    Returns:
+        str: Path to zsh config file (~/.zshrc)
+    """
+    return os.path.expanduser("~/.zshrc")
+
+
+def _get_zsh_markers() -> Tuple[str, str]:
+    """Get start and end markers for JSS completion in zsh
+
+    Returns:
+        Tuple[str, str]: (start_marker, end_marker)
+    """
+    return (
+        "# ===== JARVIS JSS ZSH COMPLETION START =====",
+        "# ===== JARVIS JSS ZSH COMPLETION END =====",
+    )
+
+
 @app.command("install")
 def install_jss_completion(
-    shell: str = typer.Option("fish", help="指定shell类型(仅支持fish)"),
+    shell: str = typer.Option("fish", help="指定shell类型(支持fish, bash, zsh)"),
 ) -> None:
-    """为fish shell安装'命令未找到'处理器，实现自然语言命令执行"""
-    if shell != "fish":
-        print(f"错误: 不支持的shell类型: {shell}, 仅支持fish")
+    """为指定的shell安装'命令未找到'处理器，实现自然语言命令建议"""
+    if shell not in ("fish", "bash", "zsh"):
+        PrettyOutput.print(f"错误: 不支持的shell类型: {shell}, 仅支持fish, bash, zsh", OutputType.ERROR)
         raise typer.Exit(code=1)
 
-    if not _check_fish_shell():
-        print("当前不是fish shell，无需安装")
-        return
+    if shell == "fish":
+        config_file = _get_config_file()
+        start_marker, end_marker = _get_markers()
 
-    config_file = _get_config_file()
-    start_marker, end_marker = _get_markers()
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 config.fish 文件，将创建新文件", OutputType.INFO)
+            os.makedirs(os.path.dirname(config_file), exist_ok=True)
+            with open(config_file, "w") as f:
+                f.write("")
 
-    if not os.path.exists(config_file):
-        print("未找到config.fish文件，将创建新文件")
-        os.makedirs(os.path.dirname(config_file), exist_ok=True)
-        with open(config_file, "w") as f:
-            f.write("")
+        with open(config_file, "r") as f:
+            content = f.read()
 
-    with open(config_file, "r") as f:
-        content = f.read()
+        if start_marker in content:
+            PrettyOutput.print("JSS fish completion 已安装，请执行: source ~/.config/fish/config.fish", OutputType.SUCCESS)
+            return
 
-    if start_marker in content:
-        print("JSS fish completion已安装，请执行: source ~/.config/fish/config.fish")
-        return
-
-    with open(config_file, "a") as f:
-        f.write(
-            f"""
+        with open(config_file, "a") as f:
+            f.write(
+                f"""
 {start_marker}
 function fish_command_not_found
     if test (string length "$argv") -lt 10
@@ -103,43 +161,245 @@ function __fish_command_not_found_handler --on-event fish_command_not_found
 end
 {end_marker}
 """
-        )
-    print("JSS fish completion已安装，请执行: source ~/.config/fish/config.fish")
+            )
+        PrettyOutput.print("JSS fish completion 已安装，请执行: source ~/.config/fish/config.fish", OutputType.SUCCESS)
+    elif shell == "bash":
+        config_file = _get_bash_config_file()
+        start_marker, end_marker = _get_bash_markers()
+
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 ~/.bashrc 文件，将创建新文件", OutputType.INFO)
+            os.makedirs(os.path.dirname(config_file), exist_ok=True)
+            with open(config_file, "w") as f:
+                f.write("")
+
+        with open(config_file, "r") as f:
+            content = f.read()
+
+        if start_marker in content:
+            PrettyOutput.print("JSS bash completion 已安装，请执行: source ~/.bashrc", OutputType.SUCCESS)
+            return
+        else:
+            with open(config_file, "a") as f:
+                f.write(
+                    f"""
+{start_marker}
+# Bash 'command not found' handler for JSS
+# 行为：
+# - 生成可编辑的建议命令，用户可直接编辑后回车执行
+# - 非交互模式下仅打印建议
+command_not_found_handle() {{
+    local cmd="$1"
+    shift || true
+    local text="$cmd $*"
+
+    # 与 fish 行为保持一致：对过短输入不处理
+    if [ ${{#text}} -lt 10 ]; then
+        return 127
+    fi
+
+    local suggestion edited
+    suggestion=$(jss request "$text")
+    if [ -n "$suggestion" ]; then
+        # 交互式：用 readline 预填命令，用户可直接回车执行或编辑
+        if [[ $- == *i* ]]; then
+            edited="$suggestion"
+            # -e 启用 readline；-i 预填默认值；无提示前缀，使体验更接近 fish 的“替换命令行”
+            read -e -i "$edited" edited
+            if [ -n "$edited" ]; then
+                eval "$edited"
+                return $?
+            fi
+        else
+            # 非交互：仅打印建议
+            printf '%s\n' "$suggestion"
+        fi
+    fi
+    return 127
+}}
+{end_marker}
+"""
+                )
+            PrettyOutput.print("JSS bash completion 已安装，请执行: source ~/.bashrc", OutputType.SUCCESS)
+    elif shell == "zsh":
+        config_file = _get_zsh_config_file()
+        start_marker, end_marker = _get_zsh_markers()
+
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 ~/.zshrc 文件，将创建新文件", OutputType.INFO)
+            os.makedirs(os.path.dirname(config_file), exist_ok=True)
+            with open(config_file, "w") as f:
+                f.write("")
+
+        with open(config_file, "r") as f:
+            content = f.read()
+
+        if start_marker in content:
+            PrettyOutput.print("JSS zsh completion 已安装，请执行: source ~/.zshrc", OutputType.SUCCESS)
+            return
+
+        with open(config_file, "a") as f:
+            f.write(
+                f"""
+{start_marker}
+# Zsh 'command not found' handler for JSS
+# 行为：
+# - 生成可编辑的建议命令，用户可直接编辑后回车执行
+# - 非交互模式下仅打印建议
+command_not_found_handler() {{
+    local cmd="$1"
+    shift || true
+    local text="$cmd $*"
+
+    # 与 fish 行为保持一致：对过短输入不处理
+    if [ ${{#text}} -lt 10 ]; then
+        return 127
+    fi
+
+    local suggestion edited
+    suggestion=$(jss request "$text")
+    if [ -n "$suggestion" ]; then
+        if [[ -o interactive ]]; then
+            local editor="${{VISUAL:-${{EDITOR:-vi}}}}"
+            local tmpfile edited
+            tmpfile="$(mktemp -t jss-edit-XXXXXX)"
+            printf '%s\n' "$suggestion" > "$tmpfile"
+            "$editor" "$tmpfile"
+            edited="$(sed -n '/./{{p;q;}}' "$tmpfile" | tr -d '\r')"
+            rm -f "$tmpfile"
+            if [ -z "$edited" ]; then
+                edited="$suggestion"
+            fi
+            eval "$edited"
+            return $?
+        else
+            # 非交互：仅打印建议
+            print -r -- "$suggestion"
+        fi
+    fi
+    return 127
+}}
+{end_marker}
+"""
+            )
+        PrettyOutput.print("JSS zsh completion 已安装，请执行: source ~/.zshrc", OutputType.SUCCESS)
+        return
+
+        with open(config_file, "a") as f:
+            f.write(
+                f"""
+{start_marker}
+# Bash 'command not found' handler for JSS
+# 行为：
+# - 生成可编辑的建议命令，用户可直接编辑后回车执行
+# - 非交互模式下仅打印建议
+command_not_found_handle() {{
+    local cmd="$1"
+    shift || true
+    local text="$cmd $*"
+
+    # 与 fish 行为保持一致：对过短输入不处理
+    if [ ${{#text}} -lt 10 ]; then
+        return 127
+    fi
+
+    local suggestion edited
+    suggestion=$(jss request "$text")
+    if [ -n "$suggestion" ]; then
+        # 交互式：用 readline 预填命令，用户可直接回车执行或编辑
+        if [[ $- == *i* ]]; then
+            edited="$suggestion"
+            # -e 启用 readline；-i 预填默认值；无提示前缀，使体验更接近 fish 的“替换命令行”
+            read -e -i "$edited" edited
+            if [ -n "$edited" ]; then
+                eval "$edited"
+                return $?
+            fi
+        else
+            # 非交互：仅打印建议
+            printf '%s\n' "$suggestion"
+        fi
+    fi
+    return 127
+}}
+{end_marker}
+"""
+            )
+        PrettyOutput.print("JSS bash completion 已安装，请执行: source ~/.bashrc", OutputType.SUCCESS)
 
 
 @app.command("uninstall")
 def uninstall_jss_completion(
-    shell: str = typer.Option("fish", help="指定shell类型(仅支持fish)"),
+    shell: str = typer.Option("fish", help="指定shell类型(支持fish, bash, zsh)"),
 ) -> None:
-    """卸载JSS fish shell'命令未找到'处理器"""
-    if shell != "fish":
-        print(f"错误: 不支持的shell类型: {shell}, 仅支持fish")
+    """卸载JSS shell'命令未找到'处理器"""
+    if shell not in ("fish", "bash", "zsh"):
+        PrettyOutput.print(f"错误: 不支持的shell类型: {shell}, 仅支持fish, bash, zsh", OutputType.ERROR)
         raise typer.Exit(code=1)
 
-    if not _check_fish_shell():
-        print("当前不是fish shell，无需卸载")
-        return
+    if shell == "fish":
+        config_file = _get_config_file()
+        start_marker, end_marker = _get_markers()
 
-    config_file = _get_config_file()
-    start_marker, end_marker = _get_markers()
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 JSS fish completion 配置，无需卸载", OutputType.INFO)
+            return
 
-    if not os.path.exists(config_file):
-        print("未找到JSS fish completion配置，无需卸载")
-        return
+        with open(config_file, "r") as f:
+            content = f.read()
 
-    with open(config_file, "r") as f:
-        content = f.read()
+        if start_marker not in content:
+            PrettyOutput.print("未找到 JSS fish completion 配置，无需卸载", OutputType.INFO)
+            return
 
-    if start_marker not in content:
-        print("未找到JSS fish completion配置，无需卸载")
-        return
+        new_content = content.split(start_marker)[0] + content.split(end_marker)[-1]
+
+        with open(config_file, "w") as f:
+            f.write(new_content)
+
+        PrettyOutput.print("JSS fish completion 已卸载，请执行: source ~/.config/fish/config.fish", OutputType.SUCCESS)
+    elif shell == "bash":
+        config_file = _get_bash_config_file()
+        start_marker, end_marker = _get_bash_markers()
+
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 JSS bash completion 配置，无需卸载", OutputType.INFO)
+            return
+
+        with open(config_file, "r") as f:
+            content = f.read()
+
+        if start_marker not in content:
+            PrettyOutput.print("未找到 JSS bash completion 配置，无需卸载", OutputType.INFO)
+            return
 
-    new_content = content.split(start_marker)[0] + content.split(end_marker)[-1]
+        new_content = content.split(start_marker)[0] + content.split(end_marker)[-1]
 
-    with open(config_file, "w") as f:
-        f.write(new_content)
+        with open(config_file, "w") as f:
+            f.write(new_content)
+
+        PrettyOutput.print("JSS bash completion 已卸载，请执行: source ~/.bashrc", OutputType.SUCCESS)
+    elif shell == "zsh":
+        config_file = _get_zsh_config_file()
+        start_marker, end_marker = _get_zsh_markers()
+
+        if not os.path.exists(config_file):
+            PrettyOutput.print("未找到 JSS zsh completion 配置，无需卸载", OutputType.INFO)
+            return
+
+        with open(config_file, "r") as f:
+            content = f.read()
+
+        if start_marker not in content:
+            PrettyOutput.print("未找到 JSS zsh completion 配置，无需卸载", OutputType.INFO)
+            return
+
+        new_content = content.split(start_marker)[0] + content.split(end_marker)[-1]
+
+        with open(config_file, "w") as f:
+            f.write(new_content)
 
-    print("JSS fish completion已卸载，请执行: source ~/.config/fish/config.fish")
+        PrettyOutput.print("JSS zsh completion 已卸载，请执行: source ~/.zshrc", OutputType.SUCCESS)
 
 
 def process_request(request: str) -> Optional[str]:

jarvis/jarvis_stats/stats.py

@@ -93,6 +93,46 @@ class StatsManager:
         storage = StatsManager._get_storage()
         return storage.list_metrics()
 
+    @staticmethod
+    def get_metric_total(metric_name: str) -> float:
+        """
+        获取指标的累计总量（快速路径）
+        优先从总量缓存读取；若不存在则回溯历史数据计算一次后缓存
+        """
+        storage = StatsManager._get_storage()
+        try:
+            return float(storage.get_metric_total(metric_name))
+        except Exception:
+            return 0.0
+
+    @staticmethod
+    def get_metric_info(metric_name: str) -> Optional[Dict[str, Any]]:
+        """
+        获取指标元信息（包含unit、created_at、last_updated、group等）
+        - 若缺少group，会尝试基于历史记录的tags进行推断并写回，然后返回最新信息
+        """
+        storage = StatsManager._get_storage()
+        info = storage.get_metric_info(metric_name)
+        if not info or not info.get("group"):
+            try:
+                grp = storage.resolve_metric_group(metric_name)  # 触发一次分组解析与回填
+                if grp:
+                    info = storage.get_metric_info(metric_name)
+            except Exception:
+                pass
+        return info
+
+    @staticmethod
+    def resolve_metric_group(metric_name: str) -> Optional[str]:
+        """
+        主动解析并写回某个指标的分组信息（调用底层存储的推断逻辑）
+        """
+        storage = StatsManager._get_storage()
+        try:
+            return storage.resolve_metric_group(metric_name)
+        except Exception:
+            return None
+
     @staticmethod
     def show(
         metric_name: Optional[str] = None,