#!/bin/bash
# ----------------------------------------------------------------------------
# Copyright (c) 2026 Huawei Technologies Co., Ltd.
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
# CANN Open Software License Agreement Version 2.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.
# ----------------------------------------------------------------------------
#
# CANNBot Agent Workspace Initialization Script
#
# CANNBot:   https://gitcode.com/cann/cannbot-skills
# This file: https://gitcode.com/cann/cannbot-skills/blob/main/plugins-community/cuda2ascend/example/init.sh

set -e

# ============================================================
# Configuration
# ============================================================
CANNBOT_URL="https://gitcode.com/cann/cannbot-skills.git"
PLUGIN_NAME="cuda2ascend"

# ============================================================
# Terminal output helpers
# ============================================================
if [ -t 1 ]; then
    GREEN='\033[0;32m'; YELLOW='\033[0;33m'; RED='\033[0;31m'
    CYAN='\033[0;36m';  BOLD='\033[1m';     DIM='\033[2m'; NC='\033[0m'
else
    GREEN=''; YELLOW=''; RED=''; CYAN=''; BOLD=''; DIM=''; NC=''
fi

ok()   { echo -e "  ${DIM}${GREEN}✓${NC}${DIM} $*${NC}"; }
warn() { echo -e "  ${YELLOW}⚠${NC}${DIM} $*${NC}"; }
err()  { echo -e "  ${RED}✗${NC}${DIM} $*${NC}"; }
info() { echo -e "  ${DIM}${CYAN}→${NC}${DIM} $*${NC}"; }
step() { echo -e "${DIM}$*${NC}"; }

# ============================================================
# Usage
# ============================================================
show_help() {
    cat << EOF
CANNBot Agent Workspace Initialization Script

Usage:
  ./init.sh opencode [options]

Arguments:
  opencode                  Target environment (currently the only supported one)

Options:
  -h, --help                Show this help message
  --repo <name>:<path>      Use a local repository instead of cloning.
                            Supported names:
                              cannbot-skills  - skills platform (consumed here)
                              asc-devkit / cann-samples / ops-tensor
                                            - passed through to the base plugin

Examples:
  ./init.sh opencode --repo cannbot-skills:~/cannbot-skills
  ./init.sh opencode --repo asc-devkit:~/asc-devkit --repo cann-samples:~/cann-samples
EOF
}

# ============================================================
# Parse target
# ============================================================
if [[ $# -lt 1 ]]; then
    err "Missing required argument 'opencode'"
    echo ""
    show_help
    exit 1
fi

TARGET_ENV=""
case "$1" in
    -h|--help)
        show_help; exit 0 ;;
    opencode)
        TARGET_ENV="opencode"; shift ;;
    *)
        err "First argument must be 'opencode', got: '$1'"
        echo ""
        show_help
        exit 1
        ;;
esac

# ============================================================
# Parse options
# ============================================================
CANNBOT_LOCAL_PATH=""
REPO_ARGS=()

while [[ $# -gt 0 ]]; do
    case "$1" in
        -h|--help)
            show_help; exit 0 ;;
        --repo)
            if [[ -z "${2:-}" ]] || [[ "$2" != *:* ]]; then
                err "--repo requires name:/path argument, got: '${2:-}'"
                exit 1
            fi
            repo_name="${2%%:*}"
            repo_path="${2#*:}"
            case "$repo_name" in
                cannbot-skills|asc-devkit|cann-samples|ops-tensor) ;;
                *)
                    err "Unknown repo name '$repo_name' " \
                        "(valid: cannbot-skills / asc-devkit / cann-samples / ops-tensor)"
                    exit 1
                    ;;
            esac
            if [ ! -d "$repo_path" ]; then
                err "Repo path not found: $repo_path (--repo $repo_name)"
                exit 1
            fi
            if [[ "$repo_name" == "cannbot-skills" ]]; then
                CANNBOT_LOCAL_PATH="$repo_path"
            else
                REPO_ARGS+=("--repo" "$2")
            fi
            shift 2
            ;;
        *)
            err "Unknown argument: '$1'"
            echo ""
            show_help
            exit 1
            ;;
    esac
done

# ============================================================
# Resolve paths
# ============================================================
SCRIPT_DIR="$(dirname "$(realpath "$0")")"
AGENT_DIR="${SCRIPT_DIR}"
REPO_DIR="$(realpath "${SCRIPT_DIR}/..")"
WORKSPACE_NAME="$(basename "$REPO_DIR")"

# ============================================================
# Validate cannbot-skills local path (if provided)
# ============================================================
if [ -n "$CANNBOT_LOCAL_PATH" ]; then
    CANNBOT_LOCAL_PATH="$(realpath "$CANNBOT_LOCAL_PATH" 2>/dev/null || echo "$CANNBOT_LOCAL_PATH")"
    if [ ! -d "$CANNBOT_LOCAL_PATH" ]; then
        err "cannbot-skills directory not found: $CANNBOT_LOCAL_PATH"
        exit 1
    fi
fi

# ============================================================
# Display configuration
# ============================================================
echo ""
echo -e "  ${BOLD}${WORKSPACE_NAME} agent${NC}"
echo -e "  ${BOLD}Configuration:${NC}"
echo -e "  target env:   ${CYAN}${TARGET_ENV}${NC}"
echo -e "  workspace:    ${CYAN}${WORKSPACE_NAME}${NC}"
echo -e "  repo root:    ${CYAN}${REPO_DIR}${NC}"
if [ -n "$CANNBOT_LOCAL_PATH" ]; then
    echo -e "  cannbot:      ${CYAN}${CANNBOT_LOCAL_PATH}${NC} (local)"
else
    echo -e "  cannbot:      ${CYAN}clone from ${CANNBOT_URL}${NC}"
fi
echo ""

# ============================================================
# Step 1: Setup cannbot-skills
# ============================================================
step "Setting up cannbot-skills..."
if [ -n "$CANNBOT_LOCAL_PATH" ]; then
    SKILLS_REPO="$CANNBOT_LOCAL_PATH"
    ok "Using local cannbot-skills at $SKILLS_REPO"
else
    SKILLS_REPO="$REPO_DIR/.cannbot/cannbot-skills"
    if [ -d "$SKILLS_REPO/.git" ]; then
        git -C "$SKILLS_REPO" pull --quiet 2>/dev/null || true
        ok "cannbot-skills updated (local cache)"
    else
        mkdir -p "$(dirname "$SKILLS_REPO")"
        if git clone --quiet "$CANNBOT_URL" "$SKILLS_REPO" 2>/dev/null; then
            ok "cannbot-skills cloned from $CANNBOT_URL"
        else
            err "Failed to clone cannbot-skills from $CANNBOT_URL"
            exit 1
        fi
    fi
fi
echo ""

# ============================================================
# Step 2: Invoke base plugin init.sh
# ============================================================
PLUGIN_INIT="$SKILLS_REPO/plugins-community/${PLUGIN_NAME}/init.sh"

if [ ! -f "$PLUGIN_INIT" ]; then
    err "Plugin init.sh not found at: $PLUGIN_INIT"
    exit 1
fi

PLUGIN_ARGS=("opencode" "$REPO_DIR" "--override" "$AGENT_DIR")
if [[ ${#REPO_ARGS[@]} -gt 0 ]]; then
    PLUGIN_ARGS+=("${REPO_ARGS[@]}")
fi

bash "$PLUGIN_INIT" "${PLUGIN_ARGS[@]}" || {
    err "Plugin '${PLUGIN_NAME}' init failed"
    exit 1
}