abyss-cli 0.1.0__tar.gz

abyss_cli-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: abyss-cli
+Version: 0.1.0
+Summary: 终端AI开发助手
+Author: Abyss Team
+Requires-Python: >=3.9
+Requires-Dist: openai>=1.0.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: prompt-toolkit>=3.0.0
+Requires-Dist: ddgs>=9.0.0
+Dynamic: requires-python

abyss_cli-0.1.0/pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "abyss-cli"
+version = "0.1.0"
+description = "终端AI开发助手"
+authors = [{name = "Abyss Team"}]
+requires-python = ">=3.9"
+dependencies = [
+    "openai>=1.0.0",
+    "click>=8.0.0",
+    "prompt-toolkit>=3.0.0",
+    "ddgs>=9.0.0",
+]
+
+[project.scripts]
+abyss = "abyss.main:cli"
+
+[tool.setuptools]
+packages = [
+    "abyss", "abyss.tools", "abyss.prompts",
+    "abyss.commands", "abyss.extensions",
+    "abyss.hooks", "abyss.mcp", "abyss.skills",
+]

abyss_cli-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

abyss_cli-0.1.0/setup.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+"""
+打包配置文件
+"""
+from setuptools import setup
+
+setup(
+    name="abyss-cli",
+    version="0.1.0",
+    packages=[
+        "abyss", "abyss.tools", "abyss.prompts",
+        "abyss.commands", "abyss.extensions",
+        "abyss.hooks", "abyss.mcp", "abyss.skills",
+    ],
+    package_dir={"": "src"},
+    install_requires=[
+        "openai>=1.0.0",
+        "click>=8.0.0",
+        "ddgs>=9.0.0",
+    ],
+    entry_points={
+        "console_scripts": [
+            "abyss=abyss.main:cli",
+        ],
+    },
+    python_requires=">=3.9",
+    description="终端AI开发助手",
+    author="Abyss Team",
+)

abyss_cli-0.1.0/src/abyss/__init__.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+"""Abyss - 终端AI开发助手"""
+__version__ = "0.1.0"

abyss_cli-0.1.0/src/abyss/ansi_menu.py

@@ -0,0 +1,559 @@
+# -*- coding: utf-8 -*-
+"""
+自定义 ANSI 补全菜单 — 完全绕过 prompt_toolkit 的灰色背景问题
+使用纯 ANSI 转义码渲染，保留 prompt_toolkit 的行编辑能力。
+
+用法：
+    from .ansi_menu import ansi_prompt
+    user_input = ansi_prompt("> ")
+"""
+import os
+import sys
+import msvcrt
+import time
+from typing import List, Tuple, Optional
+
+
+# ─── 模糊匹配 ──────────────────────────────────────────────
+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)
+
+
+# ─── 命令列表 ──────────────────────────────────────────────
+COMMANDS = [
+    ("/help", "显示帮助"),
+    ("/clear", "清空上下文"),
+    ("/exit", "退出"),
+    ("/quit", "退出"),
+    ("/config", "查看当前配置"),
+    ("/config set api-key ", "设置 API Key"),
+    ("/config set model ", "设置模型"),
+    ("/config set thinking ", "开关思考模式 (on/off)"),
+    ("/config set max-tokens ", "设置最大 token 数"),
+    ("/model", "查看/切换模型"),
+    ("/thinking", "查看/开关思考模式"),
+    ("/show-reasoning", "开关思考过程显示"),
+]
+
+
+# ─── 输入历史（↑↓ 翻看用）────────────────────────────────
+# 模块级变量，跨多次 ansi_prompt 调用共享。
+HISTORY: list = []
+HISTORY_MAX = 100
+
+
+# ─── 补全菜单最大可见行数（根据终端高度动态计算）────────────
+# 旧版本写死 8，COMMANDS 列表有 12 个 → 后 4 个被截断看不见。
+# 现在根据 os.get_terminal_size().lines 算：留 3 行给 prompt / spinner / status。
+_MIN_VISIBLE = 3
+_RESERVED_LINES = 3
+
+
+def compute_max_visible() -> int:
+    """根据终端高度算补全菜单最大可见行数。
+    - 大屏：尽量显示所有项（避免截断）
+    - 小屏：至少留 3 行供 prompt 等使用
+    - 非 TTY 环境：fallback 到 20（够所有命令 + 文件）
+    """
+    try:
+        term_h = os.get_terminal_size().lines
+    except (OSError, AttributeError):
+        return 20
+    return max(_MIN_VISIBLE, term_h - _RESERVED_LINES)
+
+
+# ── 补全数据源 ───────────────────────────────────────
+def get_file_completions(partial: str) -> List[Tuple[str, 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) -> List[Tuple[str, str]]:
+    results = []
+    for cmd_name, cmd_desc in COMMANDS:
+        if _fuzzy_match(partial, cmd_name):
+            results.append((cmd_name, cmd_desc))
+    return results
+
+
+# ─── @ 上下文检测 ─────────────────────────────────────────
+def find_at_context(text_before_cursor: str) -> Optional[Tuple[int, 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
+
+
+# ─── ANSI 颜色工具 ─────────────────────────────────────
+def ansi_white(s: str) -> str:
+    return f"\x1b[97m{s}\x1b[0m"
+
+
+def ansi_selected(s: str) -> str:
+    return f"\x1b[7;97m{s}\x1b[0m"
+
+
+def ansi_dim(s: str) -> str:
+    return f"\x1b[2m{s}\x1b[0m"
+
+
+def ansi_move_up(n: int) -> str:
+    return f"\x1b[{n}A" if n > 0 else ""
+
+
+def ansi_move_down(n: int) -> str:
+    return f"\x1b[{n}B" if n > 0 else ""
+
+
+def ansi_move_left(n: int) -> str:
+    return f"\x1b[{n}D" if n > 0 else ""
+
+
+def ansi_move_right(n: int) -> str:
+    return f"\x1b[{n}C" if n > 0 else ""
+
+
+def ansi_clear_line() -> str:
+    return "\r\x1b[K"
+
+
+def ansi_erase_lines_below(count: int) -> str:
+    out = ""
+    for _ in range(count):
+        out += "\r\x1b[K\n"
+    out += ansi_move_up(count)
+    return out
+
+
+# ─── 交互式选择器 ──────────────────────────────────────
+def ansi_select(prompt: str, options: list, default_idx: int = 0) -> "str | None":
+    """上下键选择 + Enter 确认 + Esc 取消。
+
+    Args:
+        prompt: 提示语（显示在选项上方）
+        options: 选项列表（字符串）
+        default_idx: 初始高亮项索引（默认 0）
+
+    Returns:
+        选中的字符串；按 Esc 取消返回 None。
+    """
+    if not options:
+        return None
+
+    selected = max(0, min(default_idx, len(options) - 1))
+
+    def render():
+        """重绘整个选择区域：prompt + 选项列表 + 高亮当前"""
+        # 第一行先清掉
+        sys.stdout.write('\r\x1b[K')
+        sys.stdout.write(f"  \033[1m{prompt}\033[0m\n")
+        for i, opt in enumerate(options):
+            sys.stdout.write('\r\x1b[K')
+            if i == selected:
+                # 选中行：反白 + 箭头
+                sys.stdout.write(f"  \x1b[7;97m> {opt}\x1b[0m\n")
+            else:
+                sys.stdout.write(f"    {opt}\n")
+        # 把光标移回 prompt 行（len(options) + 1：1 行 prompt + N 行选项）。
+        # 旧代码用 len(options) 会落到第一选项行，下次 render() 在选项行写新 prompt，
+        # 旧 prompt 永远不被清掉 → 屏幕上堆积多个 prompt。
+        sys.stdout.write(ansi_move_up(len(options) + 1))
+        sys.stdout.flush()
+
+    render()
+
+    while True:
+        try:
+            key = msvcrt.getwch()
+        except (OSError, ValueError):
+            time.sleep(0.01)
+            continue
+
+        # Ctrl+C
+        if key == '\x03':
+            sys.stdout.write(ansi_erase_lines_below(len(options) + 1))
+            raise KeyboardInterrupt
+
+        # 特殊键
+        if key == '\x00' or key == '\xe0':
+            special = msvcrt.getwch()
+            if special == 'H':  # 上箭头
+                selected = max(0, selected - 1)
+                render()
+            elif special == 'P':  # 下箭头
+                selected = min(len(options) - 1, selected + 1)
+                render()
+            continue
+
+        # Esc 取消
+        if key == '\x1b':
+            sys.stdout.write(ansi_erase_lines_below(len(options) + 1))
+            sys.stdout.write('\r\x1b[K')
+            sys.stdout.flush()
+            return None
+
+        # Enter 确认
+        if key == '\r' or key == '\n':
+            # 渲染最终状态（保持高亮在选中行）
+            sys.stdout.write(ansi_erase_lines_below(len(options) + 1))
+            sys.stdout.write('\r\x1b[K')
+            sys.stdout.write(f"  \033[1m{prompt}\033[0m \033[32m> {options[selected]}\033[0m\n")
+            sys.stdout.flush()
+            return options[selected]
+
+        # 其他键忽略
+
+
+# ── 主输入函数 ────────────────────────────────────────────
+def ansi_prompt(prompt_str: str = "> ") -> str:
+    """
+    自定义输入循环，支持 / 和 @ 补全菜单（ANSI 渲染）。
+    返回用户输入的字符串。
+    Ctrl+C 抛出 KeyboardInterrupt。
+    """
+    # Windows: 清空控制台输入缓冲，避免上次输出污染导致首次按键被吞
+    if sys.platform == "win32":
+        try:
+            import ctypes
+            STD_INPUT_HANDLE = -10
+            handle = ctypes.windll.kernel32.GetStdHandle(STD_INPUT_HANDLE)
+            ctypes.windll.kernel32.FlushConsoleInputBuffer(handle)
+        except Exception:
+            pass
+
+    buf: list = list(prompt_str)
+    cursor_pos: int = len(buf)
+    prev_line_len: int = len(buf)
+    completions: List[Tuple[str, str]] = []
+    selected_idx: int = 0
+    showing_menu: bool = False
+    max_visible: int = compute_max_visible()  # 根据终端高度动态算，避免截断
+    menu_row_count: int = 0
+    max_rows_ever: int = 0  # 追踪渲染过的最大行数，确保清除干净
+    # 历史翻看：history_idx = -1 表示不在导航态（停在草稿）
+    history_idx: int = -1
+    history_draft: str = ""  # 进入导航前保存的当前 buffer，便于 ↓ 回到
+
+    def redraw_line():
+        """重绘当前输入行。
+        关键：必须 \r\x1b[K（回行首 + 清行尾），不能只 \r。
+        否则本行原本有内容（如上一轮的 '，请重试)' 残留）会被新输入部分覆盖，
+        用户看到 '> /show-reasoning，请重试)' 这种新旧混杂的乱象。
+        """
+        nonlocal cursor_pos, buf, prev_line_len
+        # 回行首 + 清行尾：本行原内容（残留的 '，请重试)' 之类）彻底抹掉
+        sys.stdout.write('\r\x1b[K')
+        line = ''.join(buf)
+        sys.stdout.write(line)
+        prev_line_len = len(buf)
+        target_col = cursor_pos
+        current_col = len(buf)
+        if target_col < current_col:
+            sys.stdout.write(ansi_move_left(current_col - target_col))
+        sys.stdout.flush()
+
+    def clear_menu_area():
+        """清除菜单区域的所有行，光标回到输入行 cursor_pos 位置。
+        只在已有菜单空间内操作（假定 show_menu 已用 \\n 创建了空间），使用 \\x1b[B 避免额外滚动。
+        """
+        nonlocal menu_row_count
+        if menu_row_count > 0:
+            # \x1b[B 到菜单首行（空间已由 show_menu 首次激活时创建）
+            sys.stdout.write('\x1b[B')
+            for i in range(menu_row_count):
+                sys.stdout.write('\r\x1b[K')
+                if i < menu_row_count - 1:
+                    sys.stdout.write('\x1b[B')
+            # 上移回输入行
+            sys.stdout.write(ansi_move_up(menu_row_count))
+            sys.stdout.write('\r')
+            sys.stdout.write(ansi_move_right(cursor_pos))
+            sys.stdout.flush()
+            menu_row_count = 0
+
+    def hide_menu():
+        nonlocal showing_menu, menu_row_count, max_rows_ever, completions, selected_idx
+        # 用 menu_row_count 判断是否有菜单需要清除（showing_menu 可能已被 update_completions 置 False）
+        if menu_row_count > 0:
+            clear_menu_area()
+        showing_menu = False
+        completions = []
+        selected_idx = 0
+        max_rows_ever = 0  # 重置，下次激活重新滚动创建空间
+
+    def show_menu():
+        nonlocal showing_menu, menu_row_count, completions, selected_idx, max_rows_ever
+        if not completions:
+            hide_menu()
+            return
+
+        showing_menu = True
+        visible_count = min(len(completions), max_visible)
+        start_idx = 0
+        if selected_idx >= max_visible:
+            start_idx = selected_idx - max_visible + 1
+
+        # 仅在需要更多空间时滚动（首次激活或匹配项增多）
+        if visible_count > max_rows_ever:
+            extra = visible_count - max_rows_ever
+            for _ in range(extra):
+                sys.stdout.write('\n')
+            sys.stdout.write(ansi_move_up(extra))
+            max_rows_ever = visible_count
+        # 菜单已存在：清除旧内容（在已有空间内，用 \x1b[B 避免额外滚动）
+        elif menu_row_count > 0:
+            sys.stdout.write('\x1b[B')
+            for i in range(menu_row_count):
+                sys.stdout.write('\r\x1b[K')
+                if i < menu_row_count - 1:
+                    sys.stdout.write('\x1b[B')
+            sys.stdout.write(ansi_move_up(menu_row_count))
+            sys.stdout.write('\r')
+            sys.stdout.write(ansi_move_right(cursor_pos))
+
+        # 重绘输入行
+        sys.stdout.write('\r\x1b[K' + ''.join(buf))
+
+        # 绘制菜单项（在已有空间内，用 \x1b[B）
+        sys.stdout.write('\x1b[B')
+        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:
+                line = f"  {ansi_selected(name)}  {ansi_dim(meta)}"
+            else:
+                line = f"  {ansi_white(name)}  {ansi_dim(meta)}"
+            sys.stdout.write('\r\x1b[K' + line)
+            if i < visible_count - 1:
+                sys.stdout.write('\x1b[B')
+
+        # 上移回输入行
+        sys.stdout.write(ansi_move_up(visible_count))
+        sys.stdout.write('\r')
+        sys.stdout.write(ansi_move_right(cursor_pos))
+        sys.stdout.flush()
+
+        menu_row_count = visible_count
+
+    def update_completions():
+        nonlocal completions, showing_menu, selected_idx
+        text_before = ''.join(buf[len(prompt_str):cursor_pos])
+
+        at_ctx = find_at_context(text_before)
+        if at_ctx:
+            _, partial = at_ctx
+            new_comps = get_file_completions(partial)
+            if new_comps:
+                completions = new_comps
+                selected_idx = 0
+                showing_menu = True
+                return
+            else:
+                completions = []
+                showing_menu = False
+                return
+
+        if text_before.startswith('/'):
+            new_comps = get_command_completions(text_before)
+            if new_comps:
+                completions = new_comps
+                selected_idx = 0
+                showing_menu = True
+                return
+            else:
+                completions = []
+                showing_menu = False
+                return
+
+        completions = []
+        showing_menu = False
+
+    # 打印初始提示符
+    sys.stdout.write(prompt_str)
+    sys.stdout.flush()
+
+    while True:
+        # 直接阻塞读取 msvcrt.getwch()，不要用 kbhit() + 紧密轮询。
+        # 5ms 紧密轮询会和 Windows 中文 IME 冲突，导致 IME 提交字符后
+        # 程序仍然读不到 Enter，从而"按回车无反应、卡死"。
+        # getwch() 在 Windows 上 Ctrl+C 不会自动抛异常，需靠 \x03 字符判断（已处理）。
+        try:
+            key = msvcrt.getwch()
+        except (OSError, ValueError):
+            time.sleep(0.01)
+            continue
+
+        # 特殊键（两字节序列）
+        if key == '\x00' or key == '\xe0':
+            special = msvcrt.getwch()
+            if special == 'H':  # 上箭头
+                if showing_menu and completions:
+                    selected_idx = max(0, selected_idx - 1)
+                    show_menu()
+                elif HISTORY:
+                    # 历史翻看：↑ 调到更早一条
+                    if history_idx == -1:
+                        # 第一次按 ↑：保存当前 buffer 为草稿
+                        history_draft = ''.join(buf[len(prompt_str):])
+                    history_idx = max(0, history_idx - 1) if history_idx != -1 else len(HISTORY) - 1
+                    recall = HISTORY[history_idx]
+                    buf = list(prompt_str) + list(recall)
+                    cursor_pos = len(buf)
+                    redraw_line()
+                continue
+            elif special == 'P':  # 下箭头
+                if showing_menu and completions:
+                    vis = min(len(completions), max_visible)
+                    selected_idx = min(vis - 1, selected_idx + 1)
+                    show_menu()
+                elif history_idx != -1:
+                    # 历史翻看：↓ 调到更新一条（越界则回草稿）
+                    history_idx += 1
+                    if history_idx >= len(HISTORY):
+                        # 越界：恢复草稿
+                        buf = list(prompt_str) + list(history_draft)
+                        cursor_pos = len(buf)
+                        history_idx = -1
+                    else:
+                        recall = HISTORY[history_idx]
+                        buf = list(prompt_str) + list(recall)
+                        cursor_pos = len(buf)
+                    redraw_line()
+                continue
+            elif special == 'K':  # 左箭头
+                if cursor_pos > len(prompt_str):
+                    cursor_pos -= 1
+                    redraw_line()
+                continue
+            elif special == 'M':  # 右箭头
+                if cursor_pos < len(buf):
+                    cursor_pos += 1
+                    redraw_line()
+                continue
+            continue
+
+        # Ctrl+C
+        if key == '\x03':
+            sys.stdout.write('\r\x1b[K\n')
+            raise KeyboardInterrupt
+
+        # Enter
+        if key == '\r' or key == '\n':
+            if showing_menu and completions:
+                sel_text = completions[selected_idx][0]
+                text_before = ''.join(buf[len(prompt_str):cursor_pos])
+                at_ctx = find_at_context(text_before)
+                if at_ctx:
+                    start = len(prompt_str) + at_ctx[0]
+                elif text_before.startswith('/'):
+                    start = len(prompt_str)
+                else:
+                    start = len(prompt_str)
+                buf[start:cursor_pos] = list(sel_text)
+                cursor_pos = start + len(sel_text)
+                hide_menu()
+                redraw_line()
+            else:
+                result = ''.join(buf[len(prompt_str):])
+                # 入历史（非空 + 不与最后一条重复）
+                if result and (not HISTORY or HISTORY[-1] != result):
+                    HISTORY.append(result)
+                    if len(HISTORY) > HISTORY_MAX:
+                        HISTORY.pop(0)
+                # 重置历史导航状态
+                history_idx = -1
+                history_draft = ""
+                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()
+                update_completions()
+                if showing_menu:
+                    show_menu()
+                else:
+                    hide_menu()
+            continue
+
+        # Tab - 确认补全
+        if key == '\t':
+            if showing_menu and completions:
+                sel_text = completions[selected_idx][0]
+                text_before = ''.join(buf[len(prompt_str):cursor_pos])
+                at_ctx = find_at_context(text_before)
+                if at_ctx:
+                    start = len(prompt_str) + at_ctx[0]
+                elif text_before.startswith('/'):
+                    start = len(prompt_str)
+                else:
+                    start = len(prompt_str)
+                buf[start:cursor_pos] = list(sel_text)
+                cursor_pos = start + len(sel_text)
+                hide_menu()
+                redraw_line()
+            continue
+
+        # Escape - 关闭菜单
+        if key == '\x1b':
+            if showing_menu:
+                hide_menu()
+                redraw_line()
+            continue
+
+        # 普通可打印字符
+        if len(key) == 1 and key.isprintable():
+            buf.insert(cursor_pos, key)
+            cursor_pos += 1
+            redraw_line()
+            update_completions()
+            if showing_menu:
+                show_menu()
+            continue