#!/bin/bash

# 俄罗斯方块游戏 - Bash版本
# 使用终端控制序列进行绘制

# 设置终端
stty -echo  # 关闭输入回显
tput civis  # 隐藏光标
trap "game_over; exit" INT TERM EXIT

# 游戏区域大小
width=10
height=20
score=0
level=1
lines_cleared=0

# 定义方块形状 (7种经典形状)
# 每种形状有4个旋转状态，每个状态是一个4x4网格的字符串表示
declare -A shapes
shapes[0]="0,1,0,0,1,1,1,0,0,0,0,0,0,0,0,0"  # T形
shapes[1]="1,1,0,0,1,1,0,0,0,0,0,0,0,0,0,0"  # O形
shapes[2]="0,1,0,0,0,1,0,0,1,1,0,0,0,0,0,0"  # L形
shapes[3]="0,1,0,0,0,1,0,0,0,1,1,0,0,0,0,0"  # J形
shapes[4]="1,0,0,0,1,1,0,0,0,1,0,0,0,0,0,0"  # S形
shapes[5]="0,0,1,0,0,1,1,0,0,1,0,0,0,0,0,0"  # Z形
shapes[6]="1,0,0,0,1,0,0,0,1,0,0,0,1,0,0,0"  # I形

# 颜色定义
colors=("\033[37m" "\033[31m" "\033[32m" "\033[33m" "\033[34m" "\033[35m" "\033[36m")
reset_color="\033[0m"

# 游戏板 (二维数组模拟)
declare -a board
for ((i=0; i<height*width; i++)); do
    board[$i]=0
done

# 当前方块
current_shape=0
current_rotation=0
current_x=$((width/2-2))
current_y=0
current_color=0

# 下一个方块
next_shape=$((RANDOM % 7))
next_color=$((RANDOM % 7))

# 初始化游戏
function init_game {
    clear
    echo -e "\033[?25l"  # 隐藏光标
    draw_border
    draw_next_piece
    generate_new_piece
}

# 绘制边框
function draw_border {
    tput cup 0 0
    echo -n "+"
    for ((i=0; i<width*2+1; i++)); do echo -n "-"; done
    echo "+"
    
    for ((row=0; row<height; row++)); do
        tput cup $((row+1)) 0
        echo -n "|"
        tput cup $((row+1)) $((width*2+2))
        echo "|"
    done
    
    tput cup $((height+1)) 0
    echo -n "+"
    for ((i=0; i<width*2+1; i++)); do echo -n "-"; done
    echo "+"
    
    # 分数显示区域
    tput cup 1 $((width*2+5))
    echo "Score: $score"
    tput cup 2 $((width*2+5))
    echo "Level: $level"
    tput cup 3 $((width*2+5))
    echo "Lines: $lines_cleared"
    tput cup 4 $((width*2+5))
    echo "Next:"
}

# 绘制下一个方块预览
function draw_next_piece {
    tput cup 5 $((width*2+5))
    local shape_str=${shapes[$next_shape]}
    IFS=',' read -ra shape_arr <<< "$shape_str"
    
    for ((i=0; i<16; i++)); do
        local row=$((i/4))
        local col=$((i%4))
        local val=${shape_arr[$i]}
        
        tput cup $((5+row)) $((width*2+5+col*2))
        if [ "$val" == "1" ]; then
            echo -en "${colors[$next_color]}[]${reset_color}"
        else
            echo -n "  "
        fi
    done
}

# 生成新方块
function generate_new_piece {
    current_shape=$next_shape
    current_color=$next_color
    current_rotation=0
    current_x=$((width/2-2))
    current_y=0
    
    next_shape=$((RANDOM % 7))
    next_color=$((RANDOM % 7))
    
    draw_next_piece
    
    # 检查游戏是否结束
    if check_collision $current_x $current_y $current_shape $current_rotation; then
        game_over
    fi
}

# 检查碰撞
function check_collision {
    local x=$1
    local y=$2
    local shape=$3
    local rotation=$4
    
    local shape_str=${shapes[$shape]}
    IFS=',' read -ra shape_arr <<< "$shape_str"
    
    for ((i=0; i<16; i++)); do
        local row=$((i/4))
        local col=$((i%4))
        local val=${shape_arr[$i]}
        
        if [ "$val" == "1" ]; then
            local board_x=$((x+col))
            local board_y=$((y+row))
            
            # 检查边界
            if [ $board_x -lt 0 ] || [ $board_x -ge $width ] || [ $board_y -ge $height ]; then
                return 0  # 碰撞
            fi
            
            # 检查与已固定方块的碰撞
            if [ $board_y -ge 0 ] && [ ${board[$((board_y*width+board_x))]} -ne 0 ]; then
                return 0  # 碰撞
            fi
        fi
    done
    
    return 1  # 无碰撞
}

# 绘制游戏板
function draw_board {
    for ((row=0; row<height; row++)); do
        tput cup $((row+1)) 1
        for ((col=0; col<width; col++)); do
            local cell=${board[$((row*width+col))]}
            if [ $cell -ne 0 ]; then
                echo -en "${colors[$cell]}[]${reset_color}"
            else
                echo -n "  "
            fi
        done
    done
}

# 绘制当前下落的方块
function draw_current_piece {
    local shape_str=${shapes[$current_shape]}
    IFS=',' read -ra shape_arr <<< "$shape_str"
    
    for ((i=0; i<16; i++)); do
        local row=$((i/4))
        local col=$((i%4))
        local val=${shape_arr[$i]}
        
        if [ "$val" == "1" ]; then
            local draw_y=$((current_y+row))
            local draw_x=$((current_x+col))
            
            if [ $draw_y -ge 0 ]; then
                tput cup $((draw_y+1)) $((draw_x*2+1))
                echo -en "${colors[$current_color]}[]${reset_color}"
            fi
        fi
    done
}

# 固定当前方块到游戏板
function lock_piece {
    local shape_str=${shapes[$current_shape]}
    IFS=',' read -ra shape_arr <<< "$shape_str"
    
    for ((i=0; i<16; i++)); do
        local row=$((i/4))
        local col=$((i%4))
        local val=${shape_arr[$i]}
        
        if [ "$val" == "1" ]; then
            local board_y=$((current_y+row))
            local board_x=$((current_x+col))
            
            if [ $board_y -ge 0 ] && [ $board_y -lt $height ]; then
                board[$((board_y*width+board_x))]=$current_color
            fi
        fi
    done
}

# 清除当前方块（绘制前清除）
function clear_current_piece {
    local shape_str=${shapes[$current_shape]}
    IFS=',' read -ra shape_arr <<< "$shape_str"
    
    for ((i=0; i<16; i++)); do
        local row=$((i/4))
        local col=$((i%4))
        local val=${shape_arr[$i]}
        
        if [ "$val" == "1" ]; then
            local draw_y=$((current_y+row))
            local draw_x=$((current_x+col))
            
            if [ $draw_y -ge 0 ]; then
                tput cup $((draw_y+1)) $((draw_x*2+1))
                echo -n "  "
            fi
        fi
    done
}

# 检查并清除完整的行
function check_lines {
    local lines_to_clear=()
    
    for ((row=height-1; row>=0; row--)); do
        local full=1
        for ((col=0; col<width; col++)); do
            if [ ${board[$((row*width+col))]} -eq 0 ]; then
                full=0
                break
            fi
        done
        
        if [ $full -eq 1 ]; then
            lines_to_clear+=($row)
        fi
    done
    
    # 清除行并下移
    for row in "${lines_to_clear[@]}"; do
        # 从当前行开始，将上面的行下移
        for ((r=row; r>0; r--)); do
            for ((c=0; c<width; c++)); do
                board[$((r*width+c))]=${board[$(((r-1)*width+c))]}
            done
        done
        # 清空顶行
        for ((c=0; c<width; c++)); do
            board[$c]=0
        done
        
        ((lines_cleared++))
        ((score+=100 * level))
        
        # 每清除10行升一级
        if [ $((lines_cleared % 10)) -eq 0 ]; then
            ((level++))
        fi
    done
    
    if [ ${#lines_to_clear[@]} -gt 0 ]; then
        draw_board
        update_score
    fi
}

# 更新分数显示
function update_score {
    tput cup 1 $((width*2+5))
    echo "Score: $score    "
    tput cup 2 $((width*2+5))
    echo "Level: $level    "
    tput cup 3 $((width*2+5))
    echo "Lines: $lines_cleared    "
}

# 移动方块
function move_piece {
    local dx=$1
    local dy=$2
    
    clear_current_piece
    
    local new_x=$((current_x+dx))
    local new_y=$((current_y+dy))
    
    if ! check_collision $new_x $new_y $current_shape $current_rotation; then
        current_x=$new_x
        current_y=$new_y
        draw_current_piece
        return 0
    else
        # 如果是向下移动发生碰撞，则固定方块
        if [ $dy -eq 1 ]; then
            lock_piece
            check_lines
            generate_new_piece
        fi
        draw_current_piece
        return 1
    fi
}

# 旋转方块
function rotate_piece {
    clear_current_piece
    
    local new_rotation=$(((current_rotation+1)%4))
    
    if ! check_collision $current_x $current_y $current_shape $new_rotation; then
        current_rotation=$new_rotation
    fi
    
    draw_current_piece
}

# 快速下落
function hard_drop {
    clear_current_piece
    
    while ! check_collision $current_x $((current_y+1)) $current_shape $current_rotation; do
        ((current_y++))
    done
    
    draw_current_piece
    lock_piece
    check_lines
    generate_new_piece
}

# 游戏结束
function game_over {
    tput cup $((height+2)) 0
    echo "Game Over! Final Score: $score"
    echo "Press any key to exit..."
    stty echo
    tput cnorm
    exit 0
}

# 主游戏循环
function game_loop {
    init_game
    draw_board
    draw_current_piece
    
    local last_drop=$(date +%s%N)
    local drop_interval=$((1000000000 - (level-1)*100000000))  # 纳秒，随等级加快
    if [ $drop_interval -lt 200000000 ]; then
        drop_interval=200000000
    fi
    
    while true; do
        # 检查输入（非阻塞）
        if read -t 0.01 -n 1 key; then
            case "$key" in
                'q'|'Q') game_over ;;
                'a'|'A') move_piece -1 0 ;;  # 左移
                'd'|'D') move_piece 1 0 ;;   # 右移
                's'|'S') move_piece 0 1 ;;   # 下移
                'w'|'W') rotate_piece ;;     # 旋转
                ' ') hard_drop ;;            # 空格键快速下落
                $'\e')                       # 方向键处理
                    read -t 0.01 -n 2 arrow
                    case "$arrow" in
                        '[A') rotate_piece ;;    # 上箭头旋转
                        '[B') move_piece 0 1 ;; # 下箭头下移
                        '[C') move_piece 1 0 ;; # 右箭头右移
                        '[D') move_piece -1 0 ;; # 左箭头左移
                    esac
                    ;;
            esac
        fi
        
        # 自动下落
        local current_time=$(date +%s%N)
        if [ $((current_time - last_drop)) -ge $drop_interval ]; then
            move_piece 0 1
            last_drop=$current_time
        fi
        
        # 更新显示
        update_score
        sleep 0.01
    done
}

# 启动游戏
game_loop