#!/bin/bash

# 🔐 Auth Helper v2 - JavaScriptと同じ状態検出ロジック
# エージェントの認証状態確認と認証プロセス支援

set -e

# 共通ライブラリの読み込み
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$SCRIPT_DIR/../common/utils.sh"
setup_directories "$SCRIPT_DIR"

# スクリプトパスの設定
PANE_CONTROLLER="$SCRIPT_DIR/pane_controller.sh"
AUTH_DELEGATOR="$UTILITIES_DIR/auth_delegator.sh"

# ログ関数のエイリアス（後方互換性のため）
log_info() { log "INFO" "$1" "AUTH"; }
log_error() { log "ERROR" "$1" "ERROR"; }
log_success() { log "SUCCESS" "$1" "SUCCESS"; }
log_warn() { log "WARN" "$1" "WARN"; }

# ペイン番号取得は共通ライブラリの関数を使用
# get_pane_number() は utils.sh で定義済み

# エージェント状態を検出（JavaScriptのanalyzeAgentStateと同じロジック）
detect_agent_state() {
    local screen="$1"
    
    # デバッグ用
    # echo "[DEBUG] Screen length: $(echo "$screen" | wc -l) lines" >&2
    # echo "[DEBUG] Raw screen first 100 chars: $(echo "$screen" | head -c 100)" >&2
    
    # 空チェック
    if [ -z "$screen" ] || [ "$(echo "$screen" | grep -v '^$' | wc -l)" -eq 0 ]; then
        echo "stopped|none|停止中"
        return 0
    fi
    
    # =================================================================
    # 前処理: 画面内容を正規化（一度だけ実行）
    # 狭いペインで改行が多い場合でも確実にパターンマッチングできるよう、
    # 全ての改行とスペースを削除した文字列を作成
    # =================================================================
    # ANSIエスケープシーケンスを削除してから正規化
    local clean_screen=$(echo "$screen" | sed -E 's/\x1b\[[0-9;]*[mGKHF]//g' | sed -E 's/\x1b\[?[0-9;]*[a-zA-Z]//g')
    local compact_lower=$(echo "$clean_screen" | tr '[:upper:]' '[:lower:]' | tr -d '\n' | tr -d ' ')
    
    # 優先度0: 最優先 - bashプロンプトの検出を厳密化
    # 空でない最後の行を取得
    local last_valid_line=""
    local lines_array=()
    while IFS= read -r line; do
        lines_array+=("$line")
    done <<< "$screen"
    
    for ((i=${#lines_array[@]}-1; i>=0; i--)); do
        if [ -n "$(echo "${lines_array[$i]}" | tr -d '[:space:]')" ]; then
            last_valid_line=$(echo "${lines_array[$i]}" | tr '[:upper:]' '[:lower:]')
            break
        fi
    done
    
    # bashプロンプトの検出（元の簡単な判定に戻す）
    if echo "$last_valid_line" | grep -qE '\$[[:space:]]*$'; then
        echo "stopped|none|停止中（シェルプロンプト）"
        return 0
    fi
    
    # =================================================================
    # 以下、全ての検出は正規化済みの compact_lower を使用
    # =================================================================
    
    # 優先度0: Claudeプロンプトが表示されている場合（最優先）
    # プロンプト（>）と「for shortcuts」または「Bypassing」が同時に存在する場合は起動完了
    local has_prompt=$(echo "$screen" | grep -q "> *$" && echo "yes" || echo "no")
    local has_shortcuts=$(echo "$compact_lower" | grep -q "forshortcuts\|bypassingpermissions" && echo "yes" || echo "no")
    # echo "[DEBUG] has_prompt=$has_prompt, has_shortcuts=$has_shortcuts" >&2
    
    if [ "$has_prompt" = "yes" ] && [ "$has_shortcuts" = "yes" ]; then
        echo "running_claude|claude|Claude起動完了"
        return 0
    fi
    
    # プロンプトボックスが表示されている場合も起動完了
    if echo "$screen" | grep -q "╭─.*─╮" && echo "$screen" | grep -q "│ >" && echo "$screen" | grep -q "╰─.*─╯"; then
        # ボックス内に実際にプロンプトがあることを確認
        if echo "$compact_lower" | grep -q "forshortcuts\|bypassingpermissions\|try\|tip:"; then
            echo "running_claude|claude|Claude起動完了"
            return 0
        fi
    fi
    
    # 優先度1: Claude実行中の検出
    if echo "$compact_lower" | grep -q "esctointerrupt"; then
        echo "executing_claude|claude|Claude実行中"
        return 0
    fi
    
    # 優先度2: Claude認証中の検出（詳細タイプ付き）
    if echo "$compact_lower" | grep -q "bypasspermissions"; then
        echo "auth_claude|claude|Claude認証中 - 権限確認画面|permission_prompt"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "pressentertocontinue\|securitynotes"; then
        echo "auth_claude|claude|Claude認証中 - 続行確認画面|continue_prompt"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "useclaudecode'sterminalsetup\|dangerousmode"; then
        echo "auth_claude|claude|Claude認証中 - Terminal設定画面|terminal_setup"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "preview\|console\.log\|choosethetextstyle"; then
        echo "auth_claude|claude|Claude認証中 - テーマ選択画面|theme_selection"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "browserdidn'topen\|waitingforbrowser\|oautherror\|authenticate.*browser"; then
        echo "auth_claude|claude|Claude認証中 - ブラウザ認証待機|browser_auth"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "pastecodehere.*prompted"; then
        echo "auth_claude|claude|Claude認証中 - コード入力待機|code_input"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "selectlogin\|claudeaccountwithsubscription\|anthropicconsoleaccount"; then
        echo "auth_claude|claude|Claude認証中 - ログイン方法選択|login_selection"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "pastecodehere.*prompted"; then
        echo "auth_claude|claude|Claude認証中 - コード入力画面|code_input"
        return 0
    fi
    
    # 優先度3: Claude起動完了の検出
    if echo "$compact_lower" | grep -q "/helpforhelp.*foryourcurrentsetup"; then
        echo "running_claude|claude|Claude起動完了"
        return 0
    fi
    
    # その他のClaude起動完了パターン
    if echo "$compact_lower" | grep -q "howcanihelp\|try\"edit\|tip:" && \
       ! echo "$compact_lower" | grep -q "preview\|console\.log\|pressentertocontinue\|useclaudecode'sterminalsetup"; then
        echo "running_claude|claude|Claude起動完了"
        return 0
    fi
    
    # 優先度4: Gemini起動完了の検出
    if echo "$compact_lower" | grep -q "typeyourmessage" && \
       ! echo "$compact_lower" | grep -q "waitingforauth"; then
        echo "running_gemini|gemini|Gemini起動完了"
        return 0
    fi
    
    # Geminiバージョン検出
    if echo "$compact_lower" | grep -q "gemini-2\.\|gemini-1\.\|gemini-2\.5-pro\|gemini-2\.0-pro\|gemini-1\.5-pro"; then
        echo "running_gemini|gemini|Gemini起動完了"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "/help.*information" && ! echo "$compact_lower" | grep -q "waitingforauth"; then
        echo "running_gemini|gemini|Gemini起動完了"
        return 0
    fi
    
    # 優先度5: Gemini認証中の検出
    if echo "$compact_lower" | grep -q "waitingforauth\|loginwithgoogle\|vertexai\|geminiapikey"; then
        echo "auth_gemini|gemini|Gemini認証中"
        return 0
    fi
    
    # 優先度6: 停止中（エージェントなし）
    # Bashプロンプトのみの場合
    if (echo "$compact_lower" | grep -q "bash\|sh-") && \
       ! echo "$compact_lower" | grep -q "claude" && \
       ! echo "$compact_lower" | grep -q "gemini" && \
       ! echo "$compact_lower" | grep -q "preview"; then
        echo "stopped|none|停止中"
        return 0
    fi
    
    # 優先度7: その他の起動中状態
    if echo "$compact_lower" | grep -q "claude"; then
        echo "running_claude|claude|Claude起動中（初期化中）"
        return 0
    fi
    
    if echo "$compact_lower" | grep -q "gemini"; then
        echo "running_gemini|gemini|Gemini起動中（初期化中）"
        return 0
    fi
    
    # デフォルト
    echo "stopped|none|不明"
    return 0
}

# 認証状態確認（互換性のため残す）
check_agent_state() {
    local pane_num="$1"
    
    # ペイン番号検証
    pane_num=$(get_pane_number "$pane_num")
    if [ -z "$pane_num" ]; then
        log_error "無効なペイン指定"
        echo "unknown"
        return 1
    fi
    
    # 画面内容取得
    local target=$(get_tmux_target "$pane_num")
    local screen=$(tmux capture-pane -t "$target" -p -S -3000 2>/dev/null || echo "")
    
    # 新しい状態検出を使用
    local result=$(detect_agent_state "$screen")
    local state=$(echo "$result" | cut -d'|' -f1)
    local agent=$(echo "$result" | cut -d'|' -f2)
    local details=$(echo "$result" | cut -d'|' -f3)
    
    # 互換性のため、旧形式の状態名も返す
    case "$state" in
        "executing_claude"|"running_claude")
            echo "authenticated"
            ;;
        "auth_claude")
            # 詳細な認証状態を判定
            if echo "$details" | grep -q "権限確認画面"; then
                echo "permission_prompt"
            elif echo "$details" | grep -q "続行確認画面"; then
                echo "continue_prompt"
            elif echo "$details" | grep -q "Terminal設定画面"; then
                echo "terminal_setup"
            elif echo "$details" | grep -q "テーマ選択画面"; then
                echo "theme_selection"
            elif echo "$details" | grep -q "ブラウザ認証待機"; then
                echo "browser_auth"
            elif echo "$details" | grep -q "コード入力"; then
                echo "code_input"
            else
                echo "starting"
            fi
            ;;
        "running_gemini")
            echo "authenticated"
            ;;
        "auth_gemini")
            echo "browser_auth"
            ;;
        "stopped")
            echo "not_started"
            ;;
        *)
            echo "unknown"
            ;;
    esac
}

# 新しい詳細状態取得関数
get_detailed_state() {
    local pane_num="$1"
    
    # ペイン番号検証
    pane_num=$(get_pane_number "$pane_num")
    if [ -z "$pane_num" ]; then
        echo "error|none|無効なペイン指定"
        return 1
    fi
    
    # 画面内容取得
    local target=$(get_tmux_target "$pane_num")
    local screen=$(tmux capture-pane -t "$target" -p -S -3000 2>/dev/null || echo "")
    
    # 新しい状態検出を使用
    detect_agent_state "$screen"
}

# その他の関数は元のauth_helper.shから継承...
# （認証待機、認証代行、ステータス表示など）

# ヘルプ表示
show_usage() {
    cat << EOF
🔐 Auth Helper v2 - Claude Code認証支援ツール

使用方法:
  $(basename $0) <command> [options]

コマンド:
  check <pane>              認証状態確認（旧形式）
  state <pane>              詳細状態確認（新形式）
  wait <pane> [timeout]     認証完了待機（デフォルト:150秒）
  delegate <pane>           認証代行依頼
  status                    全ペイン認証状態表示
  
ペイン指定:
  - 番号: 0, 1, 2, ... (実際のペイン数に依存)

状態（新形式）:
  stopped           - 停止中
  executing_claude  - Claude実行中
  running_claude    - Claude起動完了
  auth_claude      - Claude認証中
  running_gemini   - Gemini起動完了
  auth_gemini      - Gemini認証中

例:
  $(basename $0) state 0      # 新形式で詳細状態取得
  $(basename $0) check 0      # 旧形式で状態確認
  $(basename $0) wait 1 180
  $(basename $0) delegate 2
  $(basename $0) status
EOF
}

# ===== 以下は元のauth_helper.shから移植 =====

# Claude起動完了チェック（新しい状態検出を使用）
check_claude_startup() {
    local pane="$1"
    local state_result=$(get_detailed_state "$pane")
    local state=$(echo "$state_result"  < /dev/null |  cut -d'|' -f1)
    
    if [[ "$state" == "running_claude" ]] || [[ "$state" == "executing_claude" ]]; then
        return 0  # 起動完了
    fi
    return 1  # 未完了
}

# Gemini起動完了チェック（新しい状態検出を使用）
check_gemini_startup() {
    local pane="$1"
    local state_result=$(get_detailed_state "$pane")
    local state=$(echo "$state_result" | cut -d'|' -f1)
    
    if [[ "$state" == "running_gemini" ]]; then
        return 0  # 起動完了
    fi
    return 1  # 未完了
}

# エージェント起動待機
wait_for_agent_startup() {
    local pane="$1"
    local timeout="${2:-150}"
    local agent_type="${3:-claude}"
    
    log_info "エージェント起動待機中 (ペイン: $pane, タイプ: $agent_type, タイムアウト: ${timeout}秒)"
    
    local elapsed=0
    local check_interval=1  # チェック間隔を短縮
    
    while [ $elapsed -lt $timeout ]; do
        local state_result=$(get_detailed_state "$pane")
        local state=$(echo "$state_result" | cut -d'|' -f1)
        local detected_agent=$(echo "$state_result" | cut -d'|' -f2)
        local details=$(echo "$state_result" | cut -d'|' -f3)
        # auth_typeは詳細から抽出
        local auth_type=""
        if [[ "$state" == "auth_claude" ]]; then
            # 詳細テキストから認証タイプを判定
            if echo "$details" | grep -q "権限確認画面"; then
                auth_type="permission_prompt"
            elif echo "$details" | grep -q "続行確認画面"; then
                auth_type="continue_prompt"  
            elif echo "$details" | grep -q "Terminal設定画面"; then
                auth_type="terminal_setup"
            elif echo "$details" | grep -q "テーマ選択画面"; then
                auth_type="theme_selection"
            elif echo "$details" | grep -q "ログイン方法選択"; then
                auth_type="login_selection"
            elif echo "$details" | grep -q "ブラウザ認証待機"; then
                auth_type="browser_auth"
            elif echo "$details" | grep -q "コード入力"; then
                auth_type="code_input"
            fi
        fi
        
        # エージェント起動完了チェック
        case "$agent_type" in
            "claude")
                if [[ "$state" == "running_claude" ]] || [[ "$state" == "executing_claude" ]]; then
                    log_success "Claude起動完了"
                    return 0
                elif [[ "$state" == "auth_claude" ]]; then
                    # デバッグログ
                    log_info "[DEBUG] auth_claude detected - details: $details, auth_type: $auth_type"
                    
                    # 認証画面の自動処理（統一されたauth_typeを使用）
                    case "$auth_type" in
                        "permission_prompt")
                            log_info "Bypass Permissions画面検出 - 自動同意"
                            tmux send-keys -t "$(get_tmux_target $pane)" Down
                            sleep 0.5
                            tmux send-keys -t "$(get_tmux_target $pane)" C-m
                            sleep 2
                            ;;
                        "continue_prompt")
                            log_info "続行確認画面検出 - Enter送信"
                            tmux send-keys -t "$(get_tmux_target $pane)" C-m
                            sleep 2
                            ;;
                        "terminal_setup")
                            log_info "Terminal設定画面検出 - Yes選択"
                            tmux send-keys -t "$(get_tmux_target $pane)" C-m
                            sleep 2
                            ;;
                        "theme_selection")
                            log_info "テーマ選択画面検出 - デフォルト選択"
                            tmux send-keys -t "$(get_tmux_target $pane)" C-m
                            sleep 1
                            ;;
                        "login_selection")
                            log_info "ログイン方法選択画面検出 - Claude account選択"
                            tmux send-keys -t "$(get_tmux_target $pane)" C-m
                            sleep 2
                            ;;
                        "browser_auth"|"code_input")
                            if [[ "$auth_type" == "browser_auth" ]]; then
                                log_info "ブラウザ認証画面検出 - 自動認証代行開始"
                            else
                                log_info "コード入力画面検出 - 自動認証代行開始"
                            fi
                            # 認証済みのペインを探す
                            local auth_pane=""
                            local pane_list=$(get_all_panes)
                            for p in $pane_list; do
                                if [ "$p" != "$pane" ]; then
                                    local p_state=$(get_detailed_state "$p" | cut -d'|' -f1)
                                    if [[ "$p_state" == "running_claude" ]] || [[ "$p_state" == "executing_claude" ]]; then
                                        auth_pane="$p"
                                        break
                                    fi
                                fi
                            done
                            
                            if [ -n "$auth_pane" ]; then
                                log_info "認証済みペイン $auth_pane を使用して認証代行"
                                # auth_delegator.shを呼び出し
                                "$SCRIPT_DIR/../utilities/auth_delegator.sh" delegate "$pane"
                            else
                                log_warn "認証済みペインが見つかりません - 手動認証が必要です"
                            fi
                            sleep 3
                            ;;
                        *)
                            log_warn "[DEBUG] Unknown auth_type: '$auth_type' for details: '$details'"
                            ;;
                    esac
                fi
                ;;
            "gemini")
                if [[ "$state" == "running_gemini" ]]; then
                    log_success "Gemini起動完了"
                    return 0
                fi
                ;;
            *)
                # その他のエージェントは起動チェックのみ
                if [[ "$state" != "stopped" ]]; then
                    log_success "$agent_type 起動完了"
                    return 0
                fi
                ;;
        esac
        
        sleep $check_interval
        elapsed=$((elapsed + check_interval))
        
        # 進捗表示
        if [ $((elapsed % 10)) -eq 0 ]; then
            log_info "待機中... ($elapsed/${timeout}秒) 現在の状態: $state"
        fi
    done
    
    log_error "タイムアウト: エージェント起動が完了しませんでした"
    return 1
}


# メイン処理
if [ "${BASH_SOURCE[0]}" = "${0}" ]; then
    case "${1:-}" in
        "check")
            if [ -z "$2" ]; then
                echo "エラー: ペイン番号が必要です"
                show_usage
                exit 1
            fi
            check_agent_state "$2"
            ;;
        "state")
            if [ -z "$2" ]; then
                echo "エラー: ペイン番号が必要です"
                show_usage
                exit 1
            fi
            get_detailed_state "$2"
            ;;
        "wait")
            if [ -z "$2" ]; then
                echo "エラー: ペイン番号が必要です"
                show_usage
                exit 1
            fi
            wait_for_agent_startup "$2" "${3:-150}" "${4:-claude}"
            ;;
        "delegate"|"status")
            # TODO: これらの関数も移植する必要がある
            echo "エラー: この機能は元のauth_helper.shを使用してください"
            exit 1
            ;;
        "help"|"-h"|"--help")
            show_usage
            ;;
        *)
            echo "エラー: 不明なコマンド '$1'"
            show_usage
            exit 1
            ;;
    esac
fi