Ycode/install.sh at main · liyimil/Ycode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/usr/bin/env bash
set -euo pipefail

REPO="${PICO_REPO:-https://github.com/martin-los/pico.git}"
INSTALL_DIR="${PICO_INSTALL_DIR:-$HOME/.pico-agent}"
BRANCH="${PICO_BRANCH:-main}"

RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; CYAN='\033[0;36m'
BOLD='\033[1m'; RESET='\033[0m'
info()    { printf "${CYAN}[pico]${RESET} %s\n" "$*"; }
success() { printf "${GREEN}[pico]${RESET} ${BOLD}%s${RESET}\n" "$*"; }
warn()    { printf "${YELLOW}[pico]${RESET} %s\n" "$*" >&2; }
die()     { printf "${RED}[pico] ERROR:${RESET} %s\n" "$*" >&2; exit 1; }

find_python() {
    for cmd in python3.13 python3.12 python3.11 python3.10 python3 python; do
        if command -v "$cmd" &>/dev/null; then
            if "$cmd" -c 'import sys; sys.exit(0 if sys.version_info >= (3,10) else 1)' 2>/dev/null; then
                echo "$cmd"
                return 0
            fi
        fi
    done
    return 1
}

main() {
    printf "\n${BOLD}╔══════════════════════════════════════════╗${RESET}\n"
    printf   "${BOLD}║        pico  一键安装                    ║${RESET}\n"
    printf   "${BOLD}╚══════════════════════════════════════════╝${RESET}\n\n"

    command -v git &>/dev/null || die "找不到 git，请先安装。"

    PYTHON=$(find_python) || die "需要 Python 3.10 或以上版本。
  安装方法：sudo apt install python3.11   (Debian/Ubuntu)
            brew install python@3.11       (macOS)"

    PY_VER=$("$PYTHON" -c 'import sys; v=sys.version_info; print(f"{v.major}.{v.minor}.{v.micro}")')
    info "使用 Python ${PY_VER} (${PYTHON})"

    if [[ -d "$INSTALL_DIR/.git" ]]; then
        info "更新已有安装 ${INSTALL_DIR} ..."
        git -C "$INSTALL_DIR" fetch --quiet origin
        git -C "$INSTALL_DIR" reset --hard "origin/${BRANCH}" --quiet
    else
        info "克隆 pico 到 ${INSTALL_DIR} ..."
        rm -rf "$INSTALL_DIR"
        git clone --depth 1 --branch "$BRANCH" "$REPO" "$INSTALL_DIR" --quiet
    fi

    VENV_DIR="$INSTALL_DIR/.venv"
    if [[ ! -d "$VENV_DIR" ]]; then
        info "创建虚拟环境 ..."
        "$PYTHON" -m venv "$VENV_DIR"
    fi

    info "安装依赖 ..."
    "$VENV_DIR/bin/pip" install --quiet --upgrade pip
    "$VENV_DIR/bin/pip" install --quiet -e "$INSTALL_DIR"

    BIN_DIR="${PICO_BIN_DIR:-$HOME/.local/bin}"
    mkdir -p "$BIN_DIR"
    LAUNCHER="$BIN_DIR/pico"

    cat > "$LAUNCHER" <<EOF
#!/usr/bin/env bash
exec "${VENV_DIR}/bin/pico" "\$@"
EOF
    chmod +x "$LAUNCHER"

    printf "\n"
    success "pico 安装完成！"
    printf "\n"

    if ! echo "$PATH" | tr ':' '\n' | grep -qx "$BIN_DIR"; then
        warn "${BIN_DIR} 不在 PATH 里。"
        printf "  添加方法（任选一条）：\n\n"
        printf "    ${BOLD}echo 'export PATH=\"\$HOME/.local/bin:\$PATH\"' >> ~/.bashrc && source ~/.bashrc${RESET}\n"
        printf "    ${BOLD}echo 'export PATH=\"\$HOME/.local/bin:\$PATH\"' >> ~/.zshrc  && source ~/.zshrc${RESET}\n\n"
    else
        printf "  运行：${BOLD}pico${RESET}\n\n"
    fi

    printf "  使用前请设置 API key（三种 provider 任选一个）：\n"
    printf "    ${BOLD}export ANTHROPIC_API_KEY=sk-ant-...${RESET}        # 使用 Claude\n"
    printf "    ${BOLD}export OPENAI_API_KEY=sk-...${RESET}                # 使用 GPT\n"
    printf "    ${BOLD}export DEEPSEEK_API_KEY=sk-...${RESET}              # 使用 DeepSeek\n\n"
    printf "  安装位置：${CYAN}${INSTALL_DIR}${RESET}\n"
    printf "  启动器：  ${CYAN}${LAUNCHER}${RESET}\n\n"
}

main "$@"