#Region ;**** Directives created by AutoIt3Wrapper_GUI ****
#AutoIt3Wrapper_Icon=icon.ico
#AutoIt3Wrapper_Outfile=VideoFolderRenamer.exe
#AutoIt3Wrapper_Compression=4
#AutoIt3Wrapper_UseUpx=y
#EndRegion ;**** Directives created by AutoIt3Wrapper_GUI ****

#include <GUIConstantsEx.au3>
#include <WindowsConstants.au3>
#include <EditConstants.au3>
#include <FileConstants.au3>
#include <File.au3>
#include <Array.au3>
#include <Excel.au3>
#include <Inet.au3>
#include <MsgBoxConstants.au3>

; ============================================
; 全局变量（兼容 AutoIt 3.3.18.0）
; ============================================
Global $g_sSelectedPath = ""
Global $g_aMP4Files[0] ; 将在 _FileListToArrayRec 后赋值
Global $g_aResults[1][4] ; 预分配二维数组，列: [MP4文件名, MP4路径, 原文件夹名, 新UUID]
Global $g_iResultCount = 0
Global $g_bProcessing = False

; ============================================
; 主程序入口
; ============================================
Main()

Func Main()
    ; 创建GUI界面
    Local $hGUI = GUICreate("视频文件夹重命名工具 v1.0", 600, 400, -1, -1)

    ; 选择目录按钮
    Local $hBtnSelectDir = GUICtrlCreateButton("📁 选择视频目录", 20, 20, 150, 40)
    GUICtrlSetFont(-1, 10, 600)

    ; 开始执行按钮
    Local $hBtnStart = GUICtrlCreateButton("▶️ 开始执行", 200, 20, 150, 40)
    GUICtrlSetFont(-1, 10, 600)
    GUICtrlSetState(-1, $GUI_DISABLE) ; 初始禁用

    ; 日志显示框
    Local $hLogEdit = GUICtrlCreateEdit("", 20, 80, 560, 280, BitOR($ES_AUTOVSCROLL, $ES_READONLY, $WS_VSCROLL, $ES_MULTILINE))
    GUICtrlSetFont(-1, 9, 400, 0, "Consolas")
    GUICtrlSetBkColor(-1, 0xFFFFFF)
    GUICtrlSetLimit(-1, 100000)

    GUISetState(@SW_SHOW, $hGUI)

    ; 消息循环
    While 1
        Switch GUIGetMsg()
            Case $GUI_EVENT_CLOSE
                ExitLoop

            Case $hBtnSelectDir
                ; 选择目录
                Local $sPath = FileSelectFolder("请选择包含MP4视频的文件夹", "")
                If $sPath <> "" Then
                    $g_sSelectedPath = $sPath
                    _Log($hLogEdit, "✓ 已选择目录: " & $g_sSelectedPath)
                    GUICtrlSetState($hBtnStart, $GUI_ENABLE)
                EndIf

            Case $hBtnStart
                If $g_sSelectedPath = "" Then
                    MsgBox($MB_ICONWARNING, "警告", "请先选择视频目录！")
                    ContinueCase
                EndIf

                ; 禁用按钮，防止重复点击
                GUICtrlSetState($hBtnSelectDir, $GUI_DISABLE)
                GUICtrlSetState($hBtnStart, $GUI_DISABLE)
                $g_bProcessing = True

                ; 清空日志和结果
                GUICtrlSetData($hLogEdit, "")
                $g_iResultCount = 0
                ReDim $g_aResults[1][4] ; 重置结果数组

                ; 扫描MP4文件
                _Log($hLogEdit, "🔍 正在扫描目录中的MP4文件...")
                Local $aTemp = _FileListToArrayRec($g_sSelectedPath, "*.mp4", $FLTAR_FILES, $FLTAR_RECUR, $FLTAR_SORT, $FLTAR_FULLPATH)

                If Not @error And IsArray($aTemp) And UBound($aTemp) > 0 Then
					$g_aMP4Files = $aTemp
				Else
					Dim $g_aMP4Files[0] ; ✅ 正确创建空数组（UBound = 0）
				EndIf

                If UBound($g_aMP4Files) = 0 Or (UBound($g_aMP4Files) = 1 And $g_aMP4Files[0] = 0) Then
                    _Log($hLogEdit, "❌ 未找到任何MP4文件！")
                    GUICtrlSetState($hBtnSelectDir, $GUI_ENABLE)
                    GUICtrlSetState($hBtnStart, $GUI_ENABLE)
                    $g_bProcessing = False
                    ContinueCase
                EndIf

                _Log($hLogEdit, "✓ 找到 " & UBound($g_aMP4Files) & " 个MP4文件")
                _Log($hLogEdit, "")

                ; 处理每个MP4文件
                Local $iSuccessCount = 0
                Local $iFailCount = 0

                For $i = 0 To UBound($g_aMP4Files) - 1
                    If Not $g_bProcessing Then ExitLoop

                    Local $sMP4Path = $g_aMP4Files[$i]
                    Local $sMP4Name = StringRegExpReplace($sMP4Path, "^.+[\\/]", "")
                    Local $sMP4BaseName = StringRegExpReplace($sMP4Name, "\.mp4$", "", 1)
                    Local $sMP4Dir = StringRegExpReplace($sMP4Path, "[\\/][^\\/]+$", "")

                    ; 检查同名文件夹是否存在
                    Local $sFolderName = $sMP4BaseName
                    Local $sFolderPath = $sMP4Dir & "\" & $sFolderName

                    If DirGetSize($sFolderPath) = -1 Then
                        _Log($hLogEdit, "⚠️ [" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 未找到同名文件夹")
                        $iFailCount += 1
                        ContinueLoop
                    EndIf

                    ; 检查playlist.m3u8是否存在
                    Local $sM3U8Path = $sFolderPath & "\playlist.m3u8"
                    If Not FileExists($sM3U8Path) Then
                        _Log($hLogEdit, "⚠️ [" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 文件夹中不存在playlist.m3u8")
                        $iFailCount += 1
                        ContinueLoop
                    EndIf

                    ; 获取新的UUID
                    _Log($hLogEdit, "[" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 正在获取新UUID...")
                    Local $sNewUUID = _GetNewUUID()

                    If $sNewUUID = "" Then
                        _Log($hLogEdit, "❌ [" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 获取UUID失败")
                        $iFailCount += 1
                        ContinueLoop
                    EndIf

                    ; 重命名文件夹
                    Local $sNewFolderPath = $sMP4Dir & "\" & $sNewUUID

                    If DirMove($sFolderPath, $sNewFolderPath) Then
                        _Log($hLogEdit, "✓ [" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 重命名成功: " & $sFolderName & " → " & $sNewUUID)

                        ; 扩容结果数组（如果需要）
                        If $g_iResultCount >= UBound($g_aResults) Then
                            ReDim $g_aResults[$g_iResultCount + 10][4]
                        EndIf

                        ; 保存结果
                        $g_aResults[$g_iResultCount][0] = $sMP4Name
                        $g_aResults[$g_iResultCount][1] = $sMP4Path
                        $g_aResults[$g_iResultCount][2] = $sFolderName
                        $g_aResults[$g_iResultCount][3] = $sNewUUID
                        $g_iResultCount += 1

                        $iSuccessCount += 1
                    Else
                        _Log($hLogEdit, "❌ [" & ($i + 1) & "/" & UBound($g_aMP4Files) & "] " & $sMP4Name & " - 重命名失败")
                        $iFailCount += 1
                    EndIf

                    ; 等待0.2秒
                    Sleep(200)
                Next

                ; 调整结果数组大小
                If $g_iResultCount > 0 Then
                    ReDim $g_aResults[$g_iResultCount][4]
                EndIf

                ; 生成Excel报告
                If $g_iResultCount > 0 Then
                    _Log($hLogEdit, "")
                    _Log($hLogEdit, "📊 正在生成Excel报告...")
                    If _GenerateExcelReport() Then
                        _Log($hLogEdit, "✓ Excel报告已生成: " & @ScriptDir & "\VideoFolderMapping.xlsx")
                    Else
                        _Log($hLogEdit, "❌ Excel报告生成失败")
                    EndIf
                EndIf

                ; 显示统计信息
                _Log($hLogEdit, "")
                _Log($hLogEdit, "========================================")
                _Log($hLogEdit, "📊 处理完成！")
                _Log($hLogEdit, "   ✓ 成功: " & $iSuccessCount & " 个")
                _Log($hLogEdit, "   ⚠️ 失败: " & $iFailCount & " 个")
                _Log($hLogEdit, "   📁 总计: " & UBound($g_aMP4Files) & " 个MP4文件")
                _Log($hLogEdit, "========================================")

                ; 恢复按钮状态
                GUICtrlSetState($hBtnSelectDir, $GUI_ENABLE)
                GUICtrlSetState($hBtnStart, $GUI_ENABLE)
                $g_bProcessing = False

        EndSwitch
    WEnd

    GUIDelete($hGUI)
EndFunc

; ============================================
; 获取新的UUID
; ============================================
Func _GetNewUUID()
    Local $sURL = "https://api.yeomai.site/API/ApiGetShortUUID.php"
    Local $sResponse = ""

    ; 尝试最多3次
    For $i = 1 To 3
        $sResponse = BinaryToString(InetRead($sURL, 10))

        If @error = 0 And $sResponse <> "" Then
            ; 清理响应数据，移除空白字符
            $sResponse = StringStripWS($sResponse, $STR_STRIPLEADING + $STR_STRIPTRAILING)
            Return $sResponse
        EndIf

        Sleep(500)
    Next

    Return ""
EndFunc

; ============================================
; 日志输出函数
; ============================================
Func _Log($hEdit, $sText)
    Local $sCurrent = GUICtrlRead($hEdit)
    GUICtrlSetData($hEdit, $sCurrent & $sText & @CRLF)

    ; 滚动到最后一行
    Local $iLines = GUICtrlSendMsg($hEdit, $EM_GETLINECOUNT, 0, 0)
    GUICtrlSendMsg($hEdit, $EM_LINESCROLL, 0, $iLines)
EndFunc

; ============================================
; 生成Excel报告
; ============================================
Func _GenerateExcelReport()
    If $g_iResultCount = 0 Then Return False

    Local $oExcel = _Excel_Open(False, False, False, False, True)
    If @error Then Return SetError(1, 0, False)

    Local $oWorkbook = _Excel_BookNew($oExcel, 1)
    If @error Then
        _Excel_Close($oExcel)
        Return SetError(2, 0, False)
    EndIf

    ; 设置表头
    Local $aHeaders[1][4] = [["MP4文件名", "MP4完整路径", "原文件夹名", "新UUID文件夹名"]]
    _Excel_RangeWrite($oWorkbook, $oWorkbook.Activesheet, $aHeaders, "A1")

    ; 写入数据
    _Excel_RangeWrite($oWorkbook, $oWorkbook.Activesheet, $g_aResults, "A2")

    ; 格式化表格
    With $oWorkbook.Activesheet
        .Range("A1:D1").Font.Bold = True
        .Range("A1:D1").Interior.Color = 0xD3D3D3
        .Columns("A:A").ColumnWidth = 30
        .Columns("B:B").ColumnWidth = 60
        .Columns("C:C").ColumnWidth = 30
        .Columns("D:D").ColumnWidth = 35
        .UsedRange.Borders.Weight = 2
    EndWith

    ; 保存文件
    Local $sSavePath = @ScriptDir & "\VideoFolderMapping.xlsx"
    _Excel_BookSaveAs($oWorkbook, $sSavePath, $xlOpenXMLWorkbook, True)

    If @error Then
        _Excel_BookClose($oWorkbook, False)
        _Excel_Close($oExcel)
        Return SetError(3, 0, False)
    EndIf

    _Excel_BookClose($oWorkbook, False)
    _Excel_Close($oExcel)

    Return True
EndFunc