1
typeset -g _CONDA_CMD_REGISTRY_FILE="$HOME/.zsh/conda-commands.json"
2
typeset -g _CONDA_CMD_REGISTRY_LOADED=0
3
typeset -gA _CONDA_CMD_TYPE_MAP
4
typeset -gA _CONDA_CMD_TARGET_MAP
5
typeset -gA _CONDA_CMD_DESC_MAP

1
_conda_exec_mamba() {
2
    if command -v micromamba >/dev/null 2>&1; then
3
        micromamba "$@"
4
    elif [[ -n "$MAMBA_EXE" && -x "$MAMBA_EXE" ]]; then
5
        "$MAMBA_EXE" "$@"
6
    else
7
        echo "未找到 micromamba，请先完成 micromamba 初始化。" >&2
8
        return 127
9
    fi
10
}

1
_conda_load_command_registry() {
2
    (( _CONDA_CMD_REGISTRY_LOADED )) && return 0
3
    [[ -f "$_CONDA_CMD_REGISTRY_FILE" ]] || return 1
4

5
    local key type target desc
6
    while IFS=$'\t' read -r key type target desc; do
7
        [[ -z "$key" || -z "$type" || -z "$target" ]] && continue
8
        _CONDA_CMD_TYPE_MAP[$key]="$type"
9
        _CONDA_CMD_TARGET_MAP[$key]="$target"
10
        _CONDA_CMD_DESC_MAP[$key]="$desc"
11
    done < <(
12
        python3 - "$_CONDA_CMD_REGISTRY_FILE" <<'PY'
13
import json
14
import sys
15

16
path = sys.argv[1]
17
with open(path, "r", encoding="utf-8") as f:
18
    data = json.load(f)
19

20
for key, spec in data.get("commands", {}).items():
21
    cmd_type = str(spec.get("type", "")).strip()
22
    target = str(spec.get("target", "")).strip()
23
    desc = str(spec.get("description", "")).strip()
24
    desc = desc.replace("\t", " ").replace("\n", " ")
25
    if key and cmd_type and target:
26
        print(f"{key}\t{cmd_type}\t{target}\t{desc}")
27
PY
28
    )
29

30
    _CONDA_CMD_REGISTRY_LOADED=1
31
}

1
_conda_dispatch() {
2
    local subcommand="$1"
3
    shift
4

5
    _conda_load_command_registry || return 1
6

7
    local cmd_type="${_CONDA_CMD_TYPE_MAP[$subcommand]}"
8
    local target="${_CONDA_CMD_TARGET_MAP[$subcommand]}"
9
    [[ -z "$cmd_type" || -z "$target" ]] && return 1
10

11
    case "$cmd_type" in
12
        function)
13
            if typeset -f "$target" >/dev/null 2>&1; then
14
                "$target" "$@"
15
            else
16
                echo "conda 子命令 '$subcommand' 对应函数 '$target' 不存在。" >&2
17
                return 1
18
            fi
19
            ;;
20
        micromamba)
21
            _conda_exec_mamba "$target" "$@"
22
            ;;
23
        *)
24
            echo "conda 子命令 '$subcommand' 的类型 '$cmd_type' 不支持。" >&2
25
            return 1
26
            ;;
27
    esac
28
}

1
_conda_print_custom_help() {
2
    _conda_load_command_registry || return 0
3

4
    local cmd
5
    echo ""
6
    echo "Custom subcommands:"
7
    for cmd in ${(k)_CONDA_CMD_TYPE_MAP}; do
8
        local desc="${_CONDA_CMD_DESC_MAP[$cmd]}"
9
        if [[ -n "$desc" ]]; then
10
            echo "  ${cmd} - ${desc}"
11
        else
12
            echo "  ${cmd}"
13
        fi
14
    done
15
    echo ""
16
    echo "Use 'conda <subcommand> ...' to run a custom command."
17
}

1
conda() {
2
    if [[ $# -eq 0 ]]; then
3
        _conda_exec_mamba
4
        _conda_print_custom_help
5
        return 0
6
    fi
7

8
    local subcommand="$1"
9
    if [[ -n "$subcommand" ]] && _conda_dispatch "$subcommand" "${@:2}"; then
10
        return $?
11
    fi
12
    _conda_exec_mamba "$@"
13
}