#include <File.au3>
#include <GUIConstants.au3>

Opt("TrayIconHide",1)
Global $temppath = @ScriptDir

GUICreate("MP3 Inventory", 320, 205, -1, @DesktopHeight/10)
GUICtrlCreateLabel("This program is for Mp3 file(s) only!   -   Created by Rogue", 24, 5, 275, 20)
$Button_1 = GUICtrlCreateButton("Take Invenotry", 200, 34, 100, 50)
$Checkbox_1 = GUICtrlCreateCheckbox("Include Subfolders", 10, 30)
$Checkbox_2 = GUICtrlCreateCheckbox("Include Full File Path", 10, 55)
GUICtrlCreateLabel("Save Inventory File to:", 8, 87, 125, 20)
Global $Input1 = GUICtrlCreateInput(@ScriptDir, 25, 107, 240, 20)
$Browse_1 = GUICtrlCreateButton("...", 270, 107, 30, 20)
GUICtrlCreateLabel("Scan directory for Mp3's:", 8, 137, 125, 20)
Global $Input2 = GUICtrlCreateInput(@ScriptDir, 25, 157, 240, 20)
$Browse_2 = GUICtrlCreateButton("...", 270, 157, 30, 20)
GUICtrlCreateLabel("Progress:", 90, 180, 65, 20)
GUICtrlSetFont (-1, 10, 600)
Global $Label_1 = GUICtrlCreateLabel("Waiting", 165, 180, 95, 20)
GUICtrlSetFont (-1, 10, 600)
GUISetState (@SW_SHOW)

While 1
    $msg = GUIGetMsg()
    $Text = @ScriptDir & "\Inventory.txt"
    Select
        Case $msg = $GUI_EVENT_CLOSE
            ExitLoop
        Case $msg = $Button_1
            GUICtrlSetData($Label_1, "Scanning...")
            GUICtrlSetFont (-1, 10, 600)
            GUICtrlSetColor(-1, 16711680)
            If BitAND(GUICtrlRead($Checkbox_1), $GUI_CHECKED) = $GUI_CHECKED And BitAND(GUICtrlRead($Checkbox_2), $GUI_CHECKED) = $GUI_CHECKED Then
                $Inventory = _RecFileListToArray($temppath, "*.mp3", 1, 1, 1)
                    If @Error=4 Then
                        MsgBox (48,"Error","No MP3 Files Found.", 5)
                    EndIf
            ElseIf BitAND(GUICtrlRead($Checkbox_1), $GUI_CHECKED) = $GUI_CHECKED And BitAND(GUICtrlRead($Checkbox_2), $GUI_UNCHECKED) = $GUI_UNCHECKED Then
                $Inventory = _RecFileListToArray($temppath, "*.mp3", 1, 1)
                    If @Error=4 Then
                        MsgBox (48,"Error","No MP3 Files Found.", 5)
                    EndIf
            ElseIf BitAND(GUICtrlRead($Checkbox_1), $GUI_UNCHECKED) = $GUI_UNCHECKED And BitAND(GUICtrlRead($Checkbox_2), $GUI_CHECKED) = $GUI_CHECKED Then
                $Inventory = _RecFileListToArray($temppath, "*.mp3", 1, 0, 1)
                    If @Error=4 Then
                        MsgBox (48,"Error","No MP3 Files Found.", 5)
                    EndIf
            Else
                $Inventory = _RecFileListToArray($temppath, "*.mp3", 1)
                    If @Error=4 Then
                        MsgBox (48,"Error","No MP3 Files Found.", 5)
                    EndIf
            EndIf
            If FileExists($Text) Then FileDelete($Text)
            _FileWriteFromArray($Text, $Inventory, 1)
            GUICtrlSetData($Label_1, "Done")
            GUICtrlSetFont (-1, 10, 600)
            GUICtrlSetColor(-1, 32768)
            Sleep(2000)
            $CountLines = (_FileCountLines($Text) -1)
            If $CountLines > 0 Then
                MsgBox(1, "Info", "Number of Mp3's in folder: " & $Countlines & @CRLF & "List saved at: " & $Text, 8)
                Run("notepad.exe " & $Text)
            EndIf
        Case $msg = $Browse_1
            $temppath = FileSelectFolder("Save File Where", "", 1, @ScriptDir)
            GUICtrlSetData($Input1, $temppath)
        Case $msg = $Browse_2
            $temppath = FileSelectFolder("Choose a folder with Mp3's", "", 1, @ScriptDir)
            GUICtrlSetData($Input2, $temppath)
    EndSelect
Wend

Func _RecFileListToArray($sPath, $sFilter = "*", $iFlag = 0, $iRecur = 0, $iFullPath = 0)
    Local $asFileList[1], $sFullPath
    If Not FileExists($sPath) Then Return SetError(1, 1, "")
    If StringRight($sPath, 1) <> "\" Then $sPath = $sPath & "\"
    If (StringInStr($sFilter, "\")) Or (StringInStr($sFilter, "/")) Or (StringInStr($sFilter, ":")) Or (StringInStr($sFilter, ">")) Or (StringInStr($sFilter, "<")) Or (StringInStr($sFilter, "|")) Or (StringStripWS($sFilter, 8) = "") Then Return SetError(2, 2, "")
    If Not ($iFlag = 0 Or $iFlag = 1 Or $iFlag = 2) Then Return SetError(3, 3, "")
    If Not ($iRecur = 0 Or $iRecur = 1) Then Return SetError(4, 4, "")
    If $iFullPath = 0 Then
        $sFullPath = $sPath
    ElseIf $iFullPath = 1 Then
        $sFullPath = ""
        $sFullPath = StringTrimLeft($sPath, (StringLen(@ScriptDir) + 1))
    Else
        Return SetError(5, 5, "")
    EndIf
    _FLTA_Search($sPath, $sFilter, $iFlag, $iRecur, $sFullPath, $asFileList)
    If $asFileList[0] = 0 Then Return SetError(6, 6, "")
    Return $asFileList
EndFunc  

Func _FLTA_Search($sStartFolder, $sFilter, $iFlag, $iRecur, $sFullPath, ByRef $asFileList)
    
    Local $hSearch, $sFile

    If StringRight($sStartFolder, 1) <> "\" Then $sStartFolder = $sStartFolder & "\"
; First look for filtered files/folders in folder
    $hSearch = FileFindFirstFile($sStartFolder & $sFilter)
    If $hSearch > 0 Then
        While 1
            $sFile = FileFindNextFile($hSearch)
            If @error Then ExitLoop
            Switch $iFlag
                Case 0; Both files and folders
                    If $iRecur And StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") <> 0 Then ContinueLoop
                Case 1; Files Only
                    If StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") <> 0 Then ContinueLoop
                Case 2; Folders only
                    If StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") = 0 Then ContinueLoop
            EndSwitch
            If $iFlag = 1 And StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") <> 0 Then ContinueLoop
            If $iFlag = 2 And StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") = 0 Then ContinueLoop
            _FLTA_Add($asFileList, $sFullPath, $sStartFolder, $sFile)
        WEnd
        FileClose($hSearch)
        ReDim $asFileList[$asFileList[0] + 1]
    EndIf
    
    If $iRecur = 1 Then
   ; Now look for subfolders
        $hSearch = FileFindFirstFile($sStartFolder & "*.*")
        If $hSearch > 0 Then
            While 1
                $sFile = FileFindNextFile($hSearch)
                If @error Then ExitLoop
                If StringInStr(FileGetAttrib($sStartFolder & $sFile), "D") And ($sFile <> "." Or $sFile <> "..") Then
               ; If folders needed, add subfolder to array
                    If $iFlag <> 1 Then _FLTA_Add($asFileList, $sFullPath, $sStartFolder, $sFile)
               ; Recursive search of this subfolder
                    _FLTA_Search($sStartFolder & $sFile, $sFilter, $iFlag, $iRecur, $sFullPath, $asFileList)
                EndIf
            WEnd
            FileClose($hSearch)
        EndIf
    EndIf
    
EndFunc

Func _FLTA_Add(ByRef $asFileList, $sFullPath, $sStartFolder, $sFile)
    
    Local $sAddFolder
    
    $asFileList[0] += 1
    If UBound($asFileList) <= $asFileList[0] Then ReDim $asFileList[UBound($asFileList) * 2]
    If $sFullPath = "" Then
        $sAddFolder = $sStartFolder
    Else
        $sAddFolder = StringReplace($sStartFolder, $sFullPath, "")
    EndIf
    $asFileList[$asFileList[0]] = $sAddFolder & $sFile
    
EndFunc