kimi-dev-workflow 0.1.0__py3-none-any.whl

kimi_dev_workflow/detector.py

@@ -0,0 +1,130 @@
+"""Project type auto-detection."""
+
+from __future__ import annotations
+
+import json
+import os
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+
+@dataclass
+class ProjectInfo:
+    """Detected project information."""
+
+    project_type: str
+    framework: str
+    language: str
+    build_tool: str
+    state_manager: str = ""
+    router: str = ""
+    style: str = ""
+    http_client: str = ""
+    validator: str = ""
+    form_lib: str = ""
+
+
+def _read_json(path: Path) -> dict:
+    """Safely read a JSON file."""
+    try:
+        return json.loads(path.read_text(encoding="utf-8"))
+    except (json.JSONDecodeError, OSError):
+        return {}
+
+
+def _has_dep(package_json: dict, name: str) -> bool:
+    """Check if a dependency exists in package.json."""
+    deps = package_json.get("dependencies", {})
+    dev_deps = package_json.get("devDependencies", {})
+    return name in deps or name in dev_deps
+
+
+def detect_project(cwd: Optional[Path] = None) -> ProjectInfo:
+    """Auto-detect project type from current directory."""
+    cwd = cwd or Path.cwd()
+
+    # Check for package.json (Node.js projects)
+    package_json_path = cwd / "package.json"
+    if package_json_path.exists():
+        package = _read_json(package_json_path)
+
+        # Detect framework
+        if _has_dep(package, "react"):
+            framework = "React"
+            language = "TypeScript" if _has_dep(package, "typescript") else "JavaScript"
+            build_tool = "Vite" if (cwd / "vite.config.ts").exists() or (cwd / "vite.config.js").exists() else "Webpack"
+            state = "Zustand" if _has_dep(package, "zustand") else "Redux" if _has_dep(package, "redux") else "Context"
+            router = "React Router" if _has_dep(package, "react-router-dom") else ""
+            style = "Tailwind" if _has_dep(package, "tailwindcss") else "Styled Components" if _has_dep(package, "styled-components") else "CSS Modules"
+            http = "TanStack Query" if _has_dep(package, "@tanstack/react-query") else "Axios" if _has_dep(package, "axios") else "Fetch"
+            validator = "Zod" if _has_dep(package, "zod") else "Yup" if _has_dep(package, "yup") else ""
+            form_lib = "react-hook-form" if _has_dep(package, "react-hook-form") else ""
+            return ProjectInfo("react", framework, language, build_tool, state, router, style, http, validator, form_lib)
+
+        elif _has_dep(package, "vue"):
+            framework = "Vue"
+            language = "TypeScript" if _has_dep(package, "typescript") else "JavaScript"
+            build_tool = "Vite" if (cwd / "vite.config.ts").exists() or (cwd / "vite.config.js").exists() else "Webpack"
+            state = "Pinia" if _has_dep(package, "pinia") else "Vuex" if _has_dep(package, "vuex") else ""
+            router = "Vue Router" if _has_dep(package, "vue-router") else ""
+            style = "Tailwind" if _has_dep(package, "tailwindcss") else "SCSS" if _has_dep(package, "sass") else "CSS"
+            http = "Axios" if _has_dep(package, "axios") else "Fetch"
+            validator = "Zod" if _has_dep(package, "zod") else ""
+            form_lib = "VeeValidate" if _has_dep(package, "vee-validate") else ""
+            return ProjectInfo("vue", framework, language, build_tool, state, router, style, http, validator, form_lib)
+
+        else:
+            # Generic Node.js project
+            language = "TypeScript" if _has_dep(package, "typescript") else "JavaScript"
+            return ProjectInfo("node", "Node.js", language, "Unknown")
+
+    # Check for Go
+    if (cwd / "go.mod").exists():
+        return ProjectInfo("go", "Go", "Go", "go build")
+
+    # Check for Python
+    if (cwd / "pyproject.toml").exists() or (cwd / "requirements.txt").exists() or (cwd / "setup.py").exists():
+        build_tool = "hatchling" if (cwd / "pyproject.toml").exists() else "setuptools"
+        return ProjectInfo("python", "Python", "Python", build_tool, "", "", "", "", "", "")
+
+    # Check for Rust
+    if (cwd / "Cargo.toml").exists():
+        return ProjectInfo("rust", "Rust", "Rust", "Cargo", "", "", "", "", "", "")
+
+    # Unknown
+    return ProjectInfo("unknown", "Unknown", "Unknown", "Unknown", "", "", "", "", "", "")
+
+
+def detect_project_name(cwd: Optional[Path] = None) -> str:
+    """Detect project name from directory or package files."""
+    cwd = cwd or Path.cwd()
+
+    # Try package.json name
+    package_json = cwd / "package.json"
+    if package_json.exists():
+        data = _read_json(package_json)
+        name = data.get("name", "")
+        if name:
+            return name
+
+    # Try go.mod module name
+    go_mod = cwd / "go.mod"
+    if go_mod.exists():
+        content = go_mod.read_text(encoding="utf-8")
+        for line in content.splitlines():
+            if line.startswith("module "):
+                return line.split()[1].split("/")[-1]
+
+    # Try pyproject.toml name
+    pyproject = cwd / "pyproject.toml"
+    if pyproject.exists():
+        import tomllib
+        try:
+            data = tomllib.loads(pyproject.read_text(encoding="utf-8"))
+            return data.get("project", {}).get("name", "")
+        except Exception:
+            pass
+
+    # Fallback to directory name
+    return cwd.name

kimi_dev_workflow/renderer.py

@@ -0,0 +1,107 @@
+"""AGENTS.md template rendering."""
+
+from __future__ import annotations
+
+import importlib.resources as pkg_resources
+from pathlib import Path
+from typing import Optional
+
+from .config import AGENTS_FILE, PACKAGE_NAME
+from .detector import ProjectInfo
+
+
+def _get_template_content() -> str:
+    """Read the AGENTS.md template from package resources."""
+    template_module = f"{PACKAGE_NAME}.templates"
+    try:
+        return pkg_resources.files(template_module).joinpath("AGENTS.md.template").read_text(encoding="utf-8")
+    except (AttributeError, OSError):
+        # Fallback for older Python versions
+        with pkg_resources.path(template_module, "AGENTS.md.template") as p:
+            return p.read_text(encoding="utf-8")
+
+
+def _get_python_replacements(info: ProjectInfo, project_name: str) -> dict[str, str]:
+    """Get replacements for Python projects."""
+    return {
+        "{项目名}": project_name,
+        "{React/Vue/Svelte} {版本} + {TypeScript/JavaScript}": f"{info.framework} + {info.build_tool}",
+        "{React/Vue/Svelte}": info.framework,
+        "{版本}": "",
+        "{TypeScript/JavaScript}": info.language,
+        "{Vite/Webpack/Rollup}": info.build_tool,
+        "{Zustand/Redux/Pinia}": "—",
+        "{React Router/Vue Router/TanStack Router}": "—",
+        "{Tailwind/Styled Components/SCSS/Less}": "—",
+        "{Axios/Fetch/TanStack Query}": "requests / httpx",
+        "{Zod/Yup/Joi}": "pydantic",
+        "{react-hook-form/VeeValidate}": "—",
+        "{store}": "—",
+        "{Context/useState}": "—",
+        "{ErrorClass}": "自定义 Exception",
+        "{校验库}": "pydantic",
+        "{路由文件路径}": "src/",
+        "{store 目录路径}": "—",
+        "{API 封装路径}": "src/",
+        "{认证相关文件}": "—",
+        "{布局组件路径}": "—",
+        "{路由文件}": "—",
+    }
+
+
+def _get_frontend_replacements(info: ProjectInfo, project_name: str) -> dict[str, str]:
+    """Get replacements for frontend projects."""
+    return {
+        "{项目名}": project_name,
+        "{React/Vue/Svelte}": info.framework,
+        "{版本}": "",
+        "{TypeScript/JavaScript}": info.language,
+        "{Vite/Webpack/Rollup}": info.build_tool,
+        "{Zustand/Redux/Pinia}": info.state_manager or "Zustand/Redux/Pinia",
+        "{React Router/Vue Router/TanStack Router}": info.router or "React Router/Vue Router",
+        "{Tailwind/Styled Components/SCSS/Less}": info.style or "Tailwind/CSS",
+        "{Axios/Fetch/TanStack Query}": info.http_client or "Axios/Fetch",
+        "{Zod/Yup/Joi}": info.validator or "Zod/Yup",
+        "{react-hook-form/VeeValidate}": info.form_lib or "react-hook-form/VeeValidate",
+        "{store}": info.state_manager or "store",
+        "{Context/useState}": "Context/useState",
+        "{ErrorClass}": "ErrorBoundary",
+        "{校验库}": info.validator or "校验库",
+        "{路由文件路径}": "src/router/",
+        "{store 目录路径}": "src/stores/",
+        "{API 封装路径}": "src/api/",
+        "{认证相关文件}": "src/api/auth.ts",
+        "{布局组件路径}": "src/components/layout/",
+        "{路由文件}": "src/router/index.ts",
+    }
+
+
+def render_agents(info: ProjectInfo, project_name: str) -> str:
+    """Render AGENTS.md template with detected project info."""
+    content = _get_template_content()
+
+    if info.project_type == "python":
+        replacements = _get_python_replacements(info, project_name)
+    else:
+        replacements = _get_frontend_replacements(info, project_name)
+
+    for placeholder, value in replacements.items():
+        content = content.replace(placeholder, value)
+
+    return content
+
+
+def write_agents(cwd: Optional[Path] = None, info: Optional[ProjectInfo] = None, project_name: str = "") -> Path:
+    """Write rendered AGENTS.md to the project directory."""
+    cwd = cwd or Path.cwd()
+    if info is None:
+        from .detector import detect_project
+        info = detect_project(cwd)
+    if not project_name:
+        from .detector import detect_project_name
+        project_name = detect_project_name(cwd)
+
+    content = render_agents(info, project_name)
+    path = cwd / AGENTS_FILE
+    path.write_text(content, encoding="utf-8")
+    return path

kimi_dev_workflow/scripts/kimi-index

@@ -0,0 +1,231 @@
+#!/bin/bash
+# kimi-index — Kimi CLI 代码索引一键管理脚本
+# 项目地址: https://github.com/<GITHUB_USERNAME>/kimi-dev-workflow
+# 用法: kimi-index [start|stop|status|search "query"|reset|reindex|logs]
+
+set -e
+
+INDEX_DIR=".grepai"
+GREPAI_BIN="${HOME}/.local/bin/grepai"
+OLLAMA_HOST="http://localhost:11434"
+
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
+log_ok()   { echo -e "${GREEN}[OK]${NC} $1"; }
+log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
+log_err()  { echo -e "${RED}[ERR]${NC} $1"; }
+
+check_grepai() {
+    if ! command -v grepai &> /dev/null; then
+        if [ -x "$GREPAI_BIN" ]; then
+            export PATH="${HOME}/.local/bin:$PATH"
+        else
+            log_err "grepai 未安装"
+            echo "  安装: curl -fsSL https://raw.githubusercontent.com/yoanbernabeu/grepai/main/install.sh | INSTALL_DIR=\$HOME/.local/bin sh"
+            exit 1
+        fi
+    fi
+}
+
+check_ollama() {
+    if ! curl -s "$OLLAMA_HOST/api/tags" > /dev/null 2>&1; then
+        log_warn "Ollama 未运行，尝试启动..."
+        if command -v ollama &> /dev/null; then
+            export OLLAMA_NUM_PARALLEL=4
+            ollama serve > /tmp/ollama-serve.log 2>&1 &
+            local pid=$!
+            sleep 3
+            if ! curl -s "$OLLAMA_HOST/api/tags" > /dev/null 2>&1; then
+                log_err "Ollama 启动失败"
+                exit 1
+            fi
+            log_ok "Ollama 已启动 (PID: $pid)"
+        else
+            log_err "Ollama 未安装"
+            echo "  https://ollama.com/download"
+            exit 1
+        fi
+    fi
+}
+
+check_model() {
+    local model="${1:-all-minilm:l6-v2}"
+    if ! curl -s "$OLLAMA_HOST/api/tags" | grep -q "$model"; then
+        log_warn "模型 $model 未下载，正在拉取..."
+        ollama pull "$model"
+        log_ok "模型 $model 已就绪"
+    fi
+}
+
+check_project() {
+    if [ ! -d "$INDEX_DIR" ]; then
+        log_warn "当前目录未初始化 grepai"
+        echo "  运行: grepai init"
+        exit 1
+    fi
+}
+
+cleanup_locks() {
+    if [ -d "$INDEX_DIR" ]; then
+        local count=$(find "$INDEX_DIR" -name "*.lock" -type f 2>/dev/null | wc -l)
+        if [ "$count" -gt 0 ]; then
+            rm -f "$INDEX_DIR"/*.lock
+            log_warn "已清理 $count 个残留锁文件"
+        fi
+    fi
+}
+
+cmd_start() {
+    log_info "启动 Kimi 代码索引..."
+    check_grepai
+    check_ollama
+    check_project
+    cleanup_locks
+    
+    local model=$(grep -A1 "^model:" "$INDEX_DIR/config.yaml" 2>/dev/null | tail -1 | xargs || echo "all-minilm:l6-v2")
+    check_model "$model"
+    
+    export PATH="${HOME}/.local/bin:$PATH"
+    export OLLAMA_NUM_PARALLEL=4
+    nohup grepai watch > /tmp/grepai-watch.log 2>&1 &
+    local pid=$!
+    
+    sleep 2
+    if ps -p $pid > /dev/null 2>&1; then
+        log_ok "索引守护进程已启动！PID: $pid"
+        echo ""
+        echo "  📊 查看进度: tail -f /tmp/grepai-watch.log"
+        echo "  🔍 查看状态: kimi-index status"
+        echo "  💡 开始开发: cd $(pwd) && kimi"
+        echo ""
+        sleep 3
+        cmd_status
+    else
+        log_err "启动失败"
+        echo "  tail -f /tmp/grepai-watch.log"
+        exit 1
+    fi
+}
+
+cmd_stop() {
+    log_info "停止索引监听..."
+    local found=0
+    if pgrep -f "grepai watch" > /dev/null 2>&1; then
+        pkill -f "grepai watch" 2>/dev/null || true
+        log_ok "索引进程已停止"
+        found=1
+    fi
+    cleanup_locks
+    [ "$found" -eq 0 ] && log_warn "没有找到运行中的索引进程"
+}
+
+cmd_status() {
+    if [ ! -d "$INDEX_DIR" ]; then
+        log_warn "当前目录未初始化索引"
+        return 1
+    fi
+    
+    echo ""
+    echo "┌─────────────────────────────────────────┐"
+    echo "│         Kimi 代码索引状态               │"
+    echo "├─────────────────────────────────────────┤"
+    
+    if pgrep -f "grepai watch" > /dev/null 2>&1; then
+        echo "│ 🟢 守护进程    运行中                   │"
+    else
+        echo "│ 🔴 守护进程    未运行                   │"
+    fi
+    
+    if grepai status 2>/dev/null | grep -q "indexed"; then
+        grepai status 2>/dev/null | head -6 | sed 's/^/│ /'
+    else
+        echo "│ ⚪ 索引数据    未建立                     │"
+    fi
+    
+    echo "│                                         │"
+    echo "│ 项目: $(basename "${PWD}")"
+    echo "└─────────────────────────────────────────┘"
+    echo ""
+}
+
+cmd_search() {
+    check_grepai; check_project
+    [ $# -eq 0 ] && { log_err "用法: kimi-index search \"关键词\""; exit 1; }
+    log_info "搜索: $*"
+    echo ""
+    grepai search "$*"
+}
+
+cmd_reset() {
+    local force=0
+    [ "$1" = "--force" ] || [ "$1" = "-f" ] && force=1
+    
+    if [ "$force" -eq 0 ]; then
+        log_warn "即将删除所有索引数据！"
+        echo "  当前目录: ${PWD}"
+        echo "  输入 'yes' 确认:"
+        read -r confirm
+        [ "$confirm" != "yes" ] && { log_info "已取消"; return 0; }
+    fi
+    
+    cmd_stop
+    [ -d "$INDEX_DIR" ] && rm -f "$INDEX_DIR"/*.gob "$INDEX_DIR"/*.lock 2>/dev/null || true
+    log_ok "索引数据已清除"
+}
+
+cmd_reindex() {
+    log_info "重新索引..."
+    cmd_stop
+    [ -d "$INDEX_DIR" ] && rm -f "$INDEX_DIR"/*.gob "$INDEX_DIR"/*.lock 2>/dev/null || true
+    cmd_start
+}
+
+cmd_logs() {
+    if [ -f /tmp/grepai-watch.log ]; then
+        log_info "索引日志:"
+        tail -20 /tmp/grepai-watch.log
+    else
+        log_warn "暂无日志"
+    fi
+}
+
+cmd_help() {
+    cat << 'EOF'
+
+  kimi-index — Kimi CLI 代码索引管理
+
+  用法:
+    kimi-index start        启动索引监听（后台）
+    kimi-index stop         停止索引监听
+    kimi-index status       查看索引状态
+    kimi-index search "xxx" 手动语义搜索
+    kimi-index reset        重置索引数据
+    kimi-index reset -f     强制重置（免确认）
+    kimi-index reindex      重置并重新索引
+    kimi-index logs         查看索引日志
+    kimi-index help         帮助
+
+EOF
+}
+
+main() {
+    local cmd="${1:-help}"
+    case "$cmd" in
+        start)     cmd_start ;;
+        stop)      cmd_stop ;;
+        status)    cmd_status ;;
+        search)    shift; cmd_search "$@" ;;
+        reset)     shift; cmd_reset "$@" ;;
+        reindex)   cmd_reindex ;;
+        logs)      cmd_logs ;;
+        help|--help|-h) cmd_help ;;
+        *) log_err "未知命令: $cmd"; cmd_help; exit 1 ;;
+    esac
+}
+
+main "$@"