#!/bin/bash

# LibreOffice 完整安装脚本
# 包含主程序安装和字体安装功能

# 脚本配置
RESOURCES_DIR="/data/resources"
LIBREOFFICE_ARCHIVE="$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm.tar.gz"
FONTS_ZIP="$RESOURCES_DIR/fonts.zip"
INSTALL_DIR="/opt"
LOG_FILE="/var/log/libreoffice_install.log"

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

# 日志函数
log_info() {
    echo -e "${BLUE}[INFO]${NC} $(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
}

# 检查root权限
check_root() {
    if [[ $EUID -ne 0 ]]; then
        log_error "此脚本需要root权限运行"
        exit 1
    fi
}

# 检查JDK是否安装
check_jdk() {
    log_info "检查JDK是否已安装..."
    
    local java_installed=false
    
    # 检查Java命令是否存在
    if command -v java &> /dev/null; then
        java_version=$(java -version 2>&1 | head -n1 | cut -d'"' -f2)
        log_success "检测到Java已安装，版本: $java_version"
        java_installed=true
    fi
    
    # 检查rpm包中的JDK
    if rpm -qa | grep -E "java-|jdk-" &> /dev/null; then
        log_success "检测到通过rpm安装的JDK"
        java_installed=true
    fi
    
    if [[ $java_installed == false ]]; then
        log_error "未检测到JDK安装，LibreOffice需要Java运行环境"
        log_info "请先安装JDK，然后重新运行此脚本"
        log_info "推荐安装OpenJDK: yum install -y java-11-openjdk"
        exit 1
	
    fi
}

# 安装系统依赖
install_dependencies() {
    log_info "安装系统依赖包..."
    
    # 安装必要的依赖包
    yum -y install ibus >> "$LOG_FILE" 2>&1
    yum -y install libX11-devel --nogpg >> "$LOG_FILE" 2>&1
    
    if [[ $? -eq 0 ]]; then
        log_success "系统依赖安装完成"
    else
        log_warning "系统依赖安装可能存在問題"
    fi
}

# 安装自定义字体
install_and_verify_fonts() {
    local fonts_temp="/tmp/fonts_install"
    local system_fonts_dir="/usr/share/fonts/custom_fonts"
    
    log_info "开始安装自定义字体..."
    
    # 1. 检查字体包是否存在
    if [[ ! -f "$FONTS_ZIP" ]]; then
        log_warning "字体包不存在: $FONTS_ZIP，跳过字体安装"
        return 0
    fi
    
    # 2. 检查文件格式
    if ! file "$FONTS_ZIP" | grep -q "Zip archive"; then
        log_error "文件不是有效的ZIP格式: $FONTS_ZIP"
        return 1
    fi
    
    # 3. 清理并创建临时目录
    rm -rf "$fonts_temp"
    mkdir -p "$fonts_temp"
    
    # 4. 解压字体包
    log_info "解压字体包..."
    if ! unzip -q "$FONTS_ZIP" -d "$fonts_temp"; then
        log_error "字体包解压失败"
        rm -rf "$fonts_temp"
        return 1
    fi
    
    # 5. 查找字体文件
    local font_files=()
    while IFS= read -r -d '' file; do
        font_files+=("$file")
    done < <(find "$fonts_temp" -type f \( -name "*.ttf" -o -name "*.otf" -o -name "*.ttc" -o -name "*.TTF" -o -name "*.OTF" \) -print0)
    
    if [[ ${#font_files[@]} -eq 0 ]]; then
        log_warning "未找到任何字体文件(.ttf/.otf/.ttc)，跳过字体安装"
        rm -rf "$fonts_temp"
        return 0
    fi
    
    log_success "找到 ${#font_files[@]} 个字体文件"
    
    # 6. 创建系统字体目录
    mkdir -p "$system_fonts_dir"
    
    # 7. 复制字体文件并设置权限
    log_info "安装字体文件..."
    for font_file in "${font_files[@]}"; do
        local font_name=$(basename "$font_file")
        cp "$font_file" "$system_fonts_dir/"
        chmod 644 "$system_fonts_dir/$font_name"
        log_info "安装字体: $font_name"
    done
    
    # 8. 更新字体缓存
    log_info "更新字体缓存..."
    if command -v fc-cache >/dev/null 2>&1; then
        if fc-cache -fv "$system_fonts_dir" >> "$LOG_FILE" 2>&1; then
            log_success "字体缓存更新成功"
        else
            log_error "字体缓存更新失败"
            rm -rf "$fonts_temp"
            return 1
        fi
    else
        log_warning "未找到fc-cache命令，跳过缓存更新"
    fi
    
    # 9. 验证字体安装
    log_info "验证字体安装..."
    if command -v fc-list >/dev/null 2>&1; then
        local installed_count=$(fc-list | grep -c "$system_fonts_dir")
        log_success "系统已识别 $installed_count 个新字体"
        
        # 显示部分安装的字体
        fc-list | grep "$system_fonts_dir" | head -5 | while read font; do
            log_info "字体示例: $font"
        done
    fi
    
    # 10. 清理临时文件
    rm -rf "$fonts_temp"
    
    log_success "自定义字体安装完成"
    return 0
}

# 安装LibreOffice主程序
install_libreoffice_main() {
    log_info "开始安装LibreOffice主程序..."
    
    # 检查安装包是否存在
    if [[ ! -f "$LIBREOFFICE_ARCHIVE" ]]; then
        log_error "未找到LibreOffice安装包: $LIBREOFFICE_ARCHIVE"
        exit 1
    fi
    
    # 切换到资源目录
    cd "$RESOURCES_DIR"
    
    # 解压安装包
    log_info "解压安装包..."
    if ! tar -zxvf "LibreOffice_7.6.7.2_Linux_x86-64_rpm.tar.gz" >> "$LOG_FILE" 2>&1; then
        log_error "解压安装包失败"
        exit 1
    fi
    
    # 进入RPM包目录
    local rpm_dir="$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm/RPMS"
    if [[ ! -d "$rpm_dir" ]]; then
        log_error "未找到RPM包目录: $rpm_dir"
        exit 1
    fi
    
    log_info "找到RPM包目录: $rpm_dir"
    cd "$rpm_dir"
    
    # 安装RPM包
    log_info "安装RPM包..."
    if yum -y localinstall *.rpm >> "$LOG_FILE" 2>&1; then
        log_success "RPM包安装成功"
    else
        log_error "RPM包安装失败"
        exit 1
    fi
    
    # 创建符号链接
    if [[ ! -f "/usr/local/bin/soffice" ]]; then
        ln -sf /opt/libreoffice7.6/program/soffice /usr/local/bin/soffice
        log_info "创建符号链接: /usr/local/bin/soffice"
    fi
}

# 配置环境变量
setup_environment() {
    log_info "配置环境变量..."
    
    # 添加到/etc/profile
    if ! grep -q "LibreOffice_PATH" /etc/profile; then
        cat >> /etc/profile << EOF

# LibreOffice Configuration
export LibreOffice_PATH=/opt/libreoffice7.6/program
export PATH=\$LibreOffice_PATH:\$PATH
EOF
        log_success "环境变量已添加到/etc/profile"
    fi
    
    # 设置当前环境变量
    export LibreOffice_PATH=/opt/libreoffice7.6/program
    export PATH=$LibreOffice_PATH:$PATH
}

# 测试安装
test_installation() {
    log_info "开始测试安装..."
    
    local test_pass=0
    local test_total=0
    
    # 测试1: 检查soffice命令
    echo "=== 测试1: 检查soffice命令 ==="
    if command -v soffice >/dev/null 2>&1; then
        log_success "soffice命令可用: $(which soffice)"
        ((test_pass++))
    else
        log_error "soffice命令不可用"
    fi
    ((test_total++))
    
    # 测试2: 检查版本信息
    echo "=== 测试2: 检查版本信息 ==="
    if soffice --version >> "$LOG_FILE" 2>&1; then
        local version=$(soffice --version | head -1)
        log_success "版本检查通过: $version"
        ((test_pass++))
    else
        log_error "版本检查失败"
    fi
    ((test_total++))
    
    # 测试3: 测试文档转换
    echo "=== 测试3: 测试文档转换功能 ==="
    local test_file="/tmp/test_install.txt"
    echo "LibreOffice安装测试文档" > "$test_file"
    echo "创建时间: $(date)" >> "$test_file"
    
    if soffice --headless --convert-to pdf --outdir /tmp "$test_file" >> "$LOG_FILE" 2>&1; then
        if [[ -f "/tmp/test_install.pdf" ]]; then
            local pdf_size=$(stat -c%s "/tmp/test_install.pdf")
            log_success "文档转换功能正常 (PDF大小: ${pdf_size}字节)"
            ((test_pass++))
            rm -f "/tmp/test_install.pdf"
        else
            log_error "文档转换失败 - 未生成PDF文件"
        fi
    else
        log_error "文档转换测试执行失败"
    fi
    ((test_total++))
    rm -f "$test_file"
    
    # 测试4: 检查字体
echo "=== 测试4: 检查字体安装 ==="
    if command -v fc-list >/dev/null 2>&1; then
        local font_count=$(fc-list | wc -l)
        # 修正：使用更可靠的方法统计自定义字体
        local custom_font_count=0
        if [[ -d "/usr/share/fonts/custom_fonts" ]]; then
            custom_font_count=$(find /usr/share/fonts/custom_fonts -name "*.ttf" -o -name "*.otf" -o -name "*.ttc" 2>/dev/null | wc -l)
        fi
        
        log_success "系统字体总数: $font_count, 自定义字体: $custom_font_count"
        ((test_pass++))
        
        # 显示前几个自定义字体（如果存在）
        if [[ $custom_font_count -gt 0 ]]; then
            log_info "自定义字体示例:"
            fc-list | grep "/usr/share/fonts/custom_fonts" 2>/dev/null | head -3 | while read font; do
                log_info "  - $font"
            done
        else
            log_info "未安装自定义字体"
        fi
    else
        log_warning "无法检查字体数量 - fc-list命令不可用"
        ((test_pass++))  # 跳过这个测试，不记失败
    fi
    ((test_total++))

    # 输出测试结果
    echo "================================================"
    log_info "测试结果: $test_pass/$test_total 项测试通过"
    
    if [[ $test_pass -eq $test_total ]]; then
        log_success "🎉 LibreOffice安装测试完全通过！"
        return 0
    else
        log_warning "⚠️  LibreOffice安装存在部分问题"
        return 1
    fi
}

# 显示安装信息
show_installation_info() {
    echo
    echo "================================================"
    echo "LibreOffice 安装完成"
    echo "================================================"
    echo "安装目录: /opt/libreoffice7.6"
    echo "程序路径: /opt/libreoffice7.6/program/soffice"
    echo "符号链接: /usr/local/bin/soffice"
    echo "字体目录: /usr/share/fonts/custom_fonts"
    echo "日志文件: $LOG_FILE"
    echo
    echo "使用示例:"
    echo "  soffice --writer          # 启动Writer"
    echo "  soffice --calc            # 启动Calc"
    echo "  soffice --headless --convert-to pdf input.doc  # 文档转换"
    echo
    echo "即将自动进行安装测试..."
    echo "================================================"
    echo
}

# 清理安装文件
cleanup_installation() {
    log_info "清理安装文件..."
    
    # 删除解压的目录
    if [[ -d "$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm" ]]; then
        rm -rf "$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm"
        log_success "清理安装目录完成"
    fi
}

# 主安装函数
main_installation() {
    log_info "开始LibreOffice完整安装流程..."
    
    # 检查root权限
    check_root
    #检查是否存在Jdk
    check_jdk

    # 创建日志目录
    mkdir -p $(dirname "$LOG_FILE")
    
    # 安装系统依赖
    install_dependencies
    
    # 安装LibreOffice主程序
    install_libreoffice_main
    
    # 安装字体
    install_and_verify_fonts
    
    # 配置环境变量
    setup_environment
    
    # 显示安装信息
    show_installation_info

    # 测试安装
    test_installation
    
    # 清理安装文件
    cleanup_installation

    log_success "LibreOffice安装完成！"
}

# 脚本使用说明
usage() {
    echo "用法: $0 [选项]"
    echo "选项:"
    echo "  -h, --help    显示帮助信息"
    echo "  -t, --test    仅测试安装是否成功，不进行安装"
    echo "  -f, --fonts   仅安装字体"
    echo "  -c, --clean   清理安装文件"
    echo
    echo "示例:"
    echo " bash $0             # 完整安装"
    echo " bash $0 --test      # 仅测试"
    echo " bash $0 --fonts     # 仅安装字体"
    echo " bash $0 --clean     # 清理安装文件"
    echo
    echo "文件位置:"
    echo "  资源目录: $RESOURCES_DIR"
    echo "  安装包: $LIBREOFFICE_ARCHIVE"
    echo "  字体包: $FONTS_ZIP"
    echo "  脚本位置: /data/scripts/install_libreoffice.sh"
}

# 清理函数
cleanup_files() {
    log_info "开始清理安装文件..."
    
    if [[ -d "$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm" ]]; then
        rm -rf "$RESOURCES_DIR/LibreOffice_7.6.7.2_Linux_x86-64_rpm"
        log_success "已清理安装目录"
    else
        log_info "未找到需要清理的安装目录"
    fi
}

# 参数处理
case "$1" in
    -h|--help)
        usage
        exit 0
        ;;
    -t|--test)
        log_info "开始测试LibreOffice安装..."
        test_installation
        exit 0
        ;;
    -f|--fonts)
        log_info "开始安装字体..."
        check_root
        install_and_verify_fonts
        test_installation
        exit 0
        ;;
    -c|--clean)
        log_info "开始清理安装文件..."
        check_root
        cleanup_files
        exit 0
        ;;
    "")
        main_installation
        ;;
    *)
        log_error "未知参数: $1"
        usage
        exit 1
        ;;
esac