abyss-cli 0.1.0__py3-none-any.whl

abyss/commands/slash.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+"""
+Slash Commands 模块
+支持从 .md 文件加载用户自定义斜杠命令，文件名即命令名，内容为 Prompt 模板。
+模板支持 {{var}} 变量替换和 $USER_INPUT 占位符（用户在命令后追加的输入）。
+"""
+import re
+from pathlib import Path
+from typing import Dict, List, Optional
+
+
+class SlashCommand:
+    """单个 slash 命令：名称 + Prompt 模板，支持变量展开。"""
+
+    def __init__(self, name: str, template: str):
+        self.name = name
+        self.template = template
+
+    def expand(self, variables: Dict[str, str]) -> str:
+        """展开模板变量。
+        - {{var}} 替换为 variables[var]，缺失保留原占位符
+        - $VAR 替换为 variables[VAR]，缺失保留原占位符
+        """
+        result = self.template
+        for key, value in variables.items():
+            result = result.replace("{{" + key + "}}", value)
+            result = result.replace("$" + key, value)
+        return result
+
+
+class SlashCommandRegistry:
+    """从指定目录加载 .md 文件作为 slash 命令的注册表。"""
+
+    def __init__(self, commands_dir: Optional[Path] = None):
+        self._commands: Dict[str, SlashCommand] = {}
+        if commands_dir is not None:
+            self._load_from_dir(commands_dir)
+
+    def _load_from_dir(self, commands_dir: Path) -> None:
+        """从目录加载所有顶层 .md 文件作为命令"""
+        if not commands_dir.exists() or not commands_dir.is_dir():
+            return
+        for md_file in commands_dir.iterdir():
+            if not md_file.is_file() or md_file.suffix != ".md":
+                continue
+            name = md_file.stem
+            try:
+                template = md_file.read_text(encoding="utf-8")
+            except Exception:
+                continue
+            self._commands[name] = SlashCommand(name, template)
+
+    def get(self, name: str) -> Optional[SlashCommand]:
+        """按名获取命令，未定义返回 None"""
+        return self._commands.get(name)
+
+    def list_names(self) -> List[str]:
+        """返回所有已加载命令名"""
+        return list(self._commands.keys())
+
+    def expand_command(self, name: str, variables: Dict[str, str]) -> Optional[str]:
+        """获取并展开指定命令，未定义返回 None"""
+        cmd = self.get(name)
+        if cmd is None:
+            return None
+        return cmd.expand(variables)
+
+
+def create_default_registry() -> SlashCommandRegistry:
+    """创建指向 ~/.abyss/commands/ 的默认注册表"""
+    commands_dir = Path.home() / ".abyss" / "commands"
+    return SlashCommandRegistry(commands_dir=commands_dir)

abyss/config.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+"""
+配置管理
+管理用户配置，存储在 ~/.abyss/config.json
+"""
+import os
+import json
+from pathlib import Path
+
+CONFIG_DIR = Path.home() / ".abyss"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+
+class Config:
+    """管理用户配置，存储在 ~/.abyss/config.json。
+
+    使用原子写入防止配置文件损坏导致数据丢失：
+    - save() 先写临时文件，成功后再替换正式文件
+    - _load() 遇到损坏文件时，自动创建 .bak 备份后回退默认值
+    """
+
+    DEFAULTS = {
+        "api_key": "",
+        "base_url": "https://api.deepseek.com",
+        "model": "deepseek-v4-pro",
+        "thinking": True,
+        "reasoning_effort": "high",
+        "auto_confirm": False,
+        "max_tokens": 65536,  # 默认 64K；deepseek-v4 输出上限 384K，可按需调大
+        "show_reasoning": False,
+        "user_id": "default",
+        "max_retries": 3,
+        "retry_delay": 1.0,
+    }
+
+    MODELS = {
+        "deepseek-v4-pro": {"context": "1M", "output": "384K", "thinking": True},
+        "deepseek-v4-flash": {"context": "1M", "output": "384K", "thinking": True},
+    }
+
+    def __init__(self):
+        self._config = {}
+        self._load()
+
+    def _load(self):
+        """从配置文件加载配置。文件损坏时备份后回退默认值。"""
+        if not CONFIG_FILE.exists():
+            self._config = dict(self.DEFAULTS)
+            return
+
+        try:
+            with open(CONFIG_FILE, "r", encoding="utf-8") as f:
+                self._config = json.load(f)
+            # 确保 _config 至少是个 dict
+            if not isinstance(self._config, dict):
+                self._backup_broken_file("非 dict 类型")
+                self._config = dict(self.DEFAULTS)
+        except json.JSONDecodeError:
+            self._backup_broken_file("JSON 解析失败")
+            self._config = dict(self.DEFAULTS)
+        except (IOError, OSError):
+            self._backup_broken_file("文件不可读")
+            self._config = dict(self.DEFAULTS)
+
+    def _backup_broken_file(self, reason: str):
+        """将损坏的配置文件重命名为 .bak，避免被默认配置覆盖丢失数据。"""
+        try:
+            bak = CONFIG_FILE.with_suffix(".json.bak")
+            CONFIG_FILE.replace(bak)
+        except Exception:
+            pass
+
+    def save(self):
+        """原子保存配置：写入临时文件后替换正式文件，防止写入中断导致文件损坏。"""
+        CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        tmp_path = CONFIG_FILE.with_suffix(".tmp")
+        try:
+            with open(tmp_path, "w", encoding="utf-8") as f:
+                json.dump(self._config, f, indent=2, ensure_ascii=False)
+            tmp_path.replace(CONFIG_FILE)
+        except Exception:
+            # 清理临时文件
+            try:
+                tmp_path.unlink(missing_ok=True)
+            except Exception:
+                pass
+            raise
+
+    def get(self, key: str):
+        """获取配置项"""
+        return self._config.get(key, self.DEFAULTS.get(key))
+
+    def set(self, key: str, value):
+        """设置配置项"""
+        self._config[key] = value
+        self.save()
+
+    def delete(self, key: str) -> bool:
+        """删除配置项（用于 /config unset 清理垃圾数据）。
+
+        幂等：key 不存在时不抛异常。
+        返回 True 表示删除了 key，False 表示 key 本来就不在。
+        """
+        if key in self._config:
+            del self._config[key]
+            self.save()
+            return True
+        return False
+
+    def is_ready(self) -> bool:
+        """检查是否已配置 API Key"""
+        return bool(self.get("api_key"))
+
+    def to_dict(self) -> dict:
+        """返回配置字典"""
+        return dict(self._config)
+
+    def get_model_info(self) -> dict:
+        """获取当前模型信息"""
+        model = self.get("model")
+        return self.MODELS.get(model, {})

abyss/custom_input.py

@@ -0,0 +1,382 @@
+# -*- coding: utf-8 -*-
+"""
+方案A: 自定义终端输入 + ANSI 补全菜单
+完全绕过 prompt_toolkit 的补全系统，自己控制一切渲染。
+"""
+import os
+import sys
+import msvcrt
+
+
+def _move_cursor_up(n: int):
+    if n > 0:
+        sys.stdout.write(f"\x1b[{n}A")
+
+
+def _move_cursor_down(n: int):
+    if n > 0:
+        sys.stdout.write(f"\x1b[{n}B")
+
+
+def _move_cursor_right(n: int):
+    if n > 0:
+        sys.stdout.write(f"\x1b[{n}C")
+
+
+def _move_cursor_left(n: int):
+    if n > 0:
+        sys.stdout.write(f"\x1b[{n}D")
+
+
+def _clear_line():
+    sys.stdout.write("\r\x1b[K")
+
+
+def _erase_lines_below(count: int):
+    for _ in range(count):
+        sys.stdout.write("\r\x1b[K")
+        sys.stdout.write("\n")
+    _move_cursor_up(count)
+
+
+def _ansi_white(text: str) -> str:
+    return f"\x1b[97m{text}\x1b[0m"
+
+
+def _ansi_gray(text: str) -> str:
+    return f"\x1b[37m{text}\x1b[0m"
+
+
+def _ansi_yellow(text: str) -> str:
+    return f"\x1b[33m{text}\x1b[0m"
+
+
+def _ansi_bold_white(text: str) -> str:
+    return f"\x1b[1;97m{text}\x1b[0m"
+
+
+def _fuzzy_match(search: str, target: str) -> bool:
+    if not search:
+        return True
+    si = 0
+    for ch in target:
+        if si < len(search) and ch.lower() == search[si].lower():
+            si += 1
+    return si == len(search)
+
+
+def _get_file_completions(partial: str):
+    """获取当前目录文件补全列表"""
+    cwd = os.getcwd()
+    try:
+        entries = sorted(os.listdir(cwd))
+    except OSError:
+        return []
+
+    results = []
+    search = partial.lower()
+    for name in entries:
+        full = os.path.join(cwd, name)
+        if not os.path.isfile(full):
+            continue
+        if name.startswith("."):
+            continue
+        if _fuzzy_match(search, name):
+            try:
+                size = os.path.getsize(full)
+                if size < 1024:
+                    meta = f"{size}B"
+                elif size < 1024 * 1024:
+                    meta = f"{size // 1024}KB"
+                else:
+                    meta = f"{size // (1024 * 1024)}MB"
+            except OSError:
+                meta = ""
+            results.append((name, meta))
+    return results
+
+
+def _get_command_completions(partial: str):
+    """获取 / 命令补全列表"""
+    from .main import COMMANDS
+    results = []
+    for cmd_name, cmd_desc in COMMANDS:
+        if _fuzzy_match(partial, cmd_name):
+            results.append((cmd_name, cmd_desc))
+    return results
+
+
+def custom_prompt(prompt_str: str = "> "):
+    """
+    自定义输入循环，支持 / 和 @ 补全菜单。
+    返回用户输入的字符串。
+    """
+    buf = list(prompt_str)
+    cursor_pos = len(buf)
+    completions = []
+    completion_start = -1
+    selected_idx = 0
+    showing_menu = False
+    max_visible = 8
+
+    # 打印初始提示符
+    sys.stdout.write(prompt_str)
+    sys.stdout.flush()
+
+    while True:
+        if not msvcrt.kbhit():
+            import time
+            time.sleep(0.01)
+            continue
+
+        key = msvcrt.getwch()
+
+        # 特殊键（两字节）
+        if key == '\x00' or key == '\xe0':
+            special = msvcrt.getwch()
+            if special == 'H':  # 上箭头
+                if showing_menu and completions:
+                    selected_idx = max(0, selected_idx - 1)
+                    _redraw_menu(prompt_str, buf, cursor_pos, completions,
+                                 selected_idx, max_visible, completion_start)
+                continue
+            elif special == 'P':  # 下箭头
+                if showing_menu and completions:
+                    visible_count = min(len(completions), max_visible)
+                    selected_idx = min(visible_count - 1, selected_idx + 1)
+                    _redraw_menu(prompt_str, buf, cursor_pos, completions,
+                                 selected_idx, max_visible, completion_start)
+                continue
+            elif special == 'K':  # 左箭头
+                if cursor_pos > len(prompt_str):
+                    cursor_pos -= 1
+                    _redraw_line(prompt_str, buf, cursor_pos)
+                continue
+            elif special == 'M':  # 右箭头
+                if cursor_pos < len(buf):
+                    cursor_pos += 1
+                    _redraw_line(prompt_str, buf, cursor_pos)
+                continue
+            continue
+
+        # Ctrl+C
+        if key == '\x03':
+            _clear_line()
+            raise KeyboardInterrupt
+
+        # Enter
+        if key == '\r' or key == '\n':
+            if showing_menu and completions:
+                # 选中补全项
+                sel_text = completions[selected_idx][0]
+                prefix_len = cursor_pos - completion_start
+                buf[completion_start:cursor_pos] = list(sel_text)
+                cursor_pos = completion_start + len(sel_text)
+                showing_menu = False
+                completions = []
+                _redraw_line(prompt_str, buf, cursor_pos)
+            else:
+                # 提交输入
+                result = ''.join(buf[len(prompt_str):])
+                sys.stdout.write('\n')
+                sys.stdout.flush()
+                return result
+            continue
+
+        # Backspace
+        if key == '\x08' or key == '\x7f':
+            if cursor_pos > len(prompt_str):
+                cursor_pos -= 1
+                buf.pop(cursor_pos)
+                _redraw_line(prompt_str, buf, cursor_pos)
+                _check_completions(prompt_str, buf, cursor_pos)
+                if showing_menu:
+                    _update_completions_and_redraw(prompt_str, buf, cursor_pos,
+                                                   completions, selected_idx,
+                                                   max_visible, completion_start)
+                else:
+                    showing_menu = False
+                    completions = []
+            continue
+
+        # Tab - 触发/确认补全
+        if key == '\t':
+            if showing_menu and completions:
+                sel_text = completions[selected_idx][0]
+                prefix_len = cursor_pos - completion_start
+                buf[completion_start:cursor_pos] = list(sel_text)
+                cursor_pos = completion_start + len(sel_text)
+                showing_menu = False
+                completions = []
+                _redraw_line(prompt_str, buf, cursor_pos)
+            continue
+
+        # Escape - 关闭菜单
+        if key == '\x1b':
+            if showing_menu:
+                showing_menu = False
+                completions = []
+                _erase_lines_below(max_visible + 1)
+                _redraw_line(prompt_str, buf, cursor_pos)
+            continue
+
+        # 普通字符
+        if len(key) == 1 and key.isprintable():
+            buf.insert(cursor_pos, key)
+            cursor_pos += 1
+            _redraw_line(prompt_str, buf, cursor_pos)
+
+            # 检查是否需要显示补全菜单
+            _check_completions(prompt_str, buf, cursor_pos)
+            if showing_menu:
+                _update_completions_and_redraw(prompt_str, buf, cursor_pos,
+                                               completions, selected_idx,
+                                               max_visible, completion_start)
+            continue
+
+
+def _check_completions(prompt_str, buf, cursor_pos):
+    """检查当前光标位置是否需要显示补全菜单"""
+    global _completions_cache, _completion_start_cache, _showing_menu_cache
+
+    text_before = ''.join(buf[len(prompt_str):cursor_pos])
+
+    # 检查 @ 文件引用
+    at_ctx = _find_at_context(text_before)
+    if at_ctx:
+        at_pos_in_text, partial = at_ctx
+        actual_start = len(prompt_str) + at_pos_in_text
+        completions = _get_file_completions(partial)
+        if completions:
+            _completions_cache = completions
+            _completion_start_cache = actual_start
+            _showing_menu_cache = True
+            return
+        else:
+            _showing_menu_cache = False
+            _completions_cache = []
+            return
+
+    # 检查 / 命令
+    if text_before.startswith('/'):
+        completions = _get_command_completions(text_before)
+        if completions:
+            _completions_cache = completions
+            _completion_start_cache = len(prompt_str)
+            _showing_menu_cache = True
+            return
+        else:
+            _showing_menu_cache = False
+            _completions_cache = []
+            return
+
+    _showing_menu_cache = False
+    _completions_cache = []
+
+
+def _update_completions_and_redraw(prompt_str, buf, cursor_pos,
+                                    completions, selected_idx,
+                                    max_visible, completion_start):
+    """更新补全列表并重绘菜单"""
+    text_before = ''.join(buf[len(prompt_str):cursor_pos])
+
+    at_ctx = _find_at_context(text_before)
+    if at_ctx:
+        _, partial = at_ctx
+        new_completions = _get_file_completions(partial)
+    elif text_before.startswith('/'):
+        new_completions = _get_command_completions(text_before)
+    else:
+        new_completions = []
+
+    if not new_completions:
+        _erase_lines_below(max_visible + 1)
+        _showing_menu_cache = False
+        return
+
+    completions.clear()
+    completions.extend(new_completions)
+    selected_idx = 0
+    _redraw_menu(prompt_str, buf, cursor_pos, completions,
+                 selected_idx, max_visible, completion_start)
+
+
+# 全局状态
+_completions_cache = []
+_completion_start_cache = -1
+_showing_menu_cache = False
+
+
+def _find_at_context(text_before_cursor: str):
+    """查找最后一个有效的 @ 引用位置"""
+    at_positions = []
+    for i, ch in enumerate(text_before_cursor):
+        if ch == "@":
+            if i > 0 and text_before_cursor[i - 1] not in (" ", "\n", "\t"):
+                continue
+            at_positions.append(i)
+
+    if not at_positions:
+        return None
+
+    last_at = at_positions[-1]
+    partial = text_before_cursor[last_at + 1:]
+
+    if " " in partial:
+        return None
+
+    return last_at, partial
+
+
+def _redraw_line(prompt_str, buf, cursor_pos):
+    """重绘输入行"""
+    sys.stdout.write('\r\x1b[K')
+    line = ''.join(buf)
+    sys.stdout.write(line)
+    # 移动光标到正确位置
+    target_col = cursor_pos
+    current_col = len(buf)
+    if target_col < current_col:
+        _move_cursor_left(current_col - target_col)
+    sys.stdout.flush()
+
+
+def _redraw_menu(prompt_str, buf, cursor_pos, completions,
+                 selected_idx, max_visible, completion_start):
+    """重绘补全菜单"""
+    global _showing_menu_cache
+    _showing_menu_cache = True
+
+    # 清除旧菜单
+    old_count = min(len(completions), max_visible) + 1
+    _erase_lines_below(old_count)
+
+    # 重新定位到输入行末尾
+    _redraw_line(prompt_str, buf, cursor_pos)
+    sys.stdout.write('\n')
+
+    # 绘制菜单项
+    visible_count = min(len(completions), max_visible)
+    start_idx = 0
+    if selected_idx >= max_visible:
+        start_idx = selected_idx - max_visible + 1
+
+    for i in range(visible_count):
+        idx = start_idx + i
+        if idx >= len(completions):
+            break
+        name, meta = completions[idx]
+        is_selected = (idx == selected_idx)
+
+        if is_selected:
+            # 选中项：白色粗体 + 灰色元信息
+            display = f"  {_ansi_bold_white(name)}  {_ansi_gray(meta)}"
+        else:
+            # 未选中项：白色 + 灰色元信息
+            display = f"  {_ansi_white(name)}  {_ansi_gray(meta)}"
+
+        sys.stdout.write(display + '\n')
+
+    # 移回输入行
+    _move_cursor_up(visible_count + 1)
+    sys.stdout.flush()

abyss/extensions/__init__.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+"""
+扩展包管理模块
+提供从 Git 仓库安装 Skill/MCP/Hook/Slash Command 的能力
+"""
+from .url_parser import PackageSource, parse_source_url
+from .registry import (
+    ExtensionRegistry, InstalledExtension, ExtensionType,
+    create_default_registry,
+)
+from .installer import PackageInstaller
+
+__all__ = [
+    "PackageSource",
+    "parse_source_url",
+    "ExtensionRegistry",
+    "InstalledExtension",
+    "ExtensionType",
+    "create_default_registry",
+    "PackageInstaller",
+]