#!/bin/bash

# 高端 Docker 端口查看工具 - 修复版
# 显示全部容器的端口映射信息

set -e

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
BOLD='\033[1m'
NC='\033[0m' # No Color

# 获取终端宽度
TERM_WIDTH=$(tput cols 2>/dev/null || echo 80)

print_header() {
    local title="🐳 DOCKER 容器端口映射表 🐳"
    local padding=$(( (TERM_WIDTH - ${#title}) / 2 ))
    printf "\n${BOLD}${CYAN}%*s${NC}\n" $((padding + ${#title})) "$title"
    echo
}

print_separator() {
    printf "${BLUE}%*s${NC}\n" $TERM_WIDTH | tr ' ' '═'
}

print_table_header() {
    printf "${BOLD}${YELLOW}%-25s %-12s %-35s %-20s${NC}\n" "容器名称" "状态" "主机端口 → 容器端口" "镜像"
    print_separator
}

format_ports() {
    local ports="$1"
    if [[ -z "$ports" || "$ports" == "" ]]; then
        echo -e "${RED}无端口映射${NC}"
        return
    fi
    
    # 提取并格式化端口信息
    echo "$ports" | sed -E '
        s/0\.0\.0\.0:([0-9]+)->([0-9]+)\/tcp/\1 → \2/g;
        s/\[::\]:([0-9]+)->([0-9]+)\/tcp/\1 → \2/g;
        s/, /\\n/g;
    ' | while IFS= read -r line; do
        if [[ -n "$line" ]]; then
            echo -e "  ${GREEN}${line}${NC}"
        fi
    done
}

print_container_info() {
    local name="$1"
    local status="$2"
    local ports="$3"
    local image="$4"
    
    # 状态颜色
    local status_color=$GREEN
    if [[ "$status" == *"Exited"* ]]; then
        status_color=$RED
    elif [[ "$status" == *"Restarting"* ]]; then
        status_color=$YELLOW
    fi
    
    # 缩短镜像名称
    local image_short=$(echo "$image" | cut -d':' -f1 | rev | cut -d'/' -f1 | rev)
    if [[ ${#image_short} -gt 18 ]]; then
        image_short="${image_short:0:15}..."
    fi
    
    # 打印容器基本信息
    printf "%-25s ${status_color}%-12s${NC} %-35s %-20s\n" \
        "${name:0:24}" \
        "${status:0:11}" \
        "" \
        "$image_short"
    
    # 打印端口信息
    format_ports "$ports"
}

get_total_info() {
    local total=$(docker ps -aq | wc -l | tr -d ' ')
    local running=$(docker ps -q | wc -l | tr -d ' ')
    local stopped=$((total - running))
    echo "$total $running $stopped"
}

main() {
    # 检查 Docker
    if ! command -v docker &> /dev/null; then
        echo -e "${RED}错误: Docker 未安装${NC}"
        exit 1
    fi
    
    # 显示头部
    print_header
    
    # 获取统计信息
    read total running stopped <<< $(get_total_info)
    
    # 显示统计
    echo -e "${BOLD}📊 容器统计: ${GREEN}运行: $running${NC} | ${YELLOW}停止: $stopped${NC} | ${BLUE}总计: $total${NC}"
    print_separator
    echo
    
    # 显示表格头部
    print_table_header
    
    # 获取并显示容器信息
    docker ps -a --format "{{.Names}}||{{.Status}}||{{.Ports}}||{{.Image}}" | while IFS='||' read -r name status ports image; do
        print_container_info "$name" "$status" "$ports" "$image"
        echo
    done
    
    # 显示底部信息
    echo
    print_separator
    echo -e "${CYAN}💡 提示:${NC}"
    echo -e "  ${GREEN}●${NC} 绿色状态: 容器正常运行"
    echo -e "  ${YELLOW}●${NC} 黄色状态: 容器重启中"
    echo -e "  ${RED}●${NC} 红色状态: 容器已停止"
    echo
    echo -e "${PURPLE}🔄 刷新: 重新运行此脚本${NC}"
    echo -e "${PURPLE}🐳 管理: docker ps -a | docker start/stop [容器名]${NC}"
}

# 运行主函数
main "$@"