# 设置输出编码为UTF-8
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
# 目录配置
$OUTPUT_DIR = "./model"
$TARGET_DIR = "../../app/main/model"

# 表名列表
$tables = @(
    # ============================================
    # 新代理系统表
    # ============================================
    # "agent",
    # "agent_audit",
    # "agent_wallet",
    # "agent_relation",
    # "agent_link",
    # "agent_order",
    # "agent_commission",
    # "agent_rebate",
    # "agent_upgrade",
    # "agent_withdrawal",
    # "agent_config"
    # "agent_product_config"
    # "agent_invite_code_usage"
    # "agent_freeze_task"
    "agent_short_link"
    # "agent_real_name"
    # "agent_withdrawal_tax"
    # "agent_invite_code"
    # ============================================
    # 其他业务表
    # ============================================
    # "feature",
    # "global_notifications"
    # "order",
    # "order_refund"
    # "product",
    # "product_feature",
    # "query",
    # "query_cleanup_log"
    # "query_cleanup_detail"
    # "query_cleanup_config"
    # "user"
    # "user_auth"
    # "user_temp"
    # "example"
    # "admin_user"
    # "admin_user_role"
    # "admin_api",
    # "admin_menu"
    # "admin_role",
    # "admin_role_api",
    # "admin_role_menu",
    # "admin_dict_data"
    # "admin_dict_type"
    # "admin_promotion_link"
    # "admin_promotion_link_stats_total"
    # "admin_promotion_link_stats_history"
    # "admin_promotion_order"

)

# 为每个表生成模型
foreach ($table in $tables) {
    Write-Host "正在生成表: $table" -ForegroundColor Green
    goctl model mysql datasource -url="ycc:5vg67b3UNHu8@tcp(127.0.0.1:21001)/ycc" -table="$table" -dir="./model" --home="../template" -cache=true --style=goZero
    
    # 移动生成的文件到目标目录
    if (Test-Path $OUTPUT_DIR) {
        $sourceFiles = Get-ChildItem -Path $OUTPUT_DIR -File
        
        foreach ($file in $sourceFiles) {
            $fileName = $file.Name
            $targetPath = Join-Path $TARGET_DIR $fileName
            $sourcePath = $file.FullName
            
            # 检查文件类型并决定是否移动
            $shouldMove = $false
            $shouldOverwrite = $false
            
            if ($fileName -eq "vars.go") {
                # vars.go: 如果目标目录不存在才移动
                if (-not (Test-Path $targetPath)) {
                    $shouldMove = $true
                    Write-Host "  移动 $fileName (vars.go 不存在于目标目录)" -ForegroundColor Yellow
                }
                else {
                    Write-Host "  跳过 $fileName (vars.go 已存在于目标目录，防止覆盖)" -ForegroundColor Cyan
                }
            }
            elseif ($fileName -match "_gen\.go$") {
                # 带 _gen 后缀的文件: 直接覆盖
                $shouldMove = $true
                $shouldOverwrite = $true
                Write-Host "  移动 $fileName (覆盖 _gen 文件)" -ForegroundColor Yellow
            }
            else {
                # 不带 _gen 后缀的文件: 如果目标目录不存在才移动
                if (-not (Test-Path $targetPath)) {
                    $shouldMove = $true
                    Write-Host "  移动 $fileName (非 _gen 文件不存在于目标目录)" -ForegroundColor Yellow
                }
                else {
                    Write-Host "  跳过 $fileName (非 _gen 文件已存在于目标目录，防止覆盖)" -ForegroundColor Cyan
                }
            }
            
            # 执行移动操作
            if ($shouldMove) {
                # 确保目标目录存在
                if (-not (Test-Path $TARGET_DIR)) {
                    New-Item -ItemType Directory -Path $TARGET_DIR -Force | Out-Null
                }
                
                # 如果目标文件存在且需要覆盖，先删除
                if ($shouldOverwrite -and (Test-Path $targetPath)) {
                    Remove-Item -Path $targetPath -Force
                }
                
                # 移动文件
                Move-Item -Path $sourcePath -Destination $targetPath -Force
                Write-Host "    ✓ 已移动到: $targetPath" -ForegroundColor Green
            }
        }
    }
}

Write-Host ""
Write-Host '所有模型文件生成并移动完成!' -ForegroundColor Green