; ##########################################################################
; ###                                                                    ###
; ### hostCleanUp                                                        ###
; ###                                                                    ###
; ##########################################################################
 
KillTries = 10

StepsAll = 1
If runeje0 > 0
  StepsAll++
If regsvr0 > 0
  StepsAll++
If dattxt0 > 0
  StepsAll++
If datexe0 > 0
  StepsAll += 2
If regbak0 > 0
  StepsAll += 2
IfExist %U3_APP_DATA_PATH%\regdataX.reg
  StepsAll++
If regdel0 > 0
  StepsAll++
If fildel0 > 0
  StepsAll++
StepsStep := 100/StepsAll
StepsPos = 0

Progress b2 x%PL% y%PT% w%PW% m FM%PFM% FS%PFS%, U3Helper %U3HVer% - (c)2006-2007 Markus Birth <mbirth@webwriters.de>, Cleaning up %AppName% ..., AHKProgress-%AppName%
WinSet Transparent, %PTrans%, AHKProgress-%AppName%

Loop %runeje0%
{
  CurCmd := runeje%A_Index%
  Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/runeje0, Running shutdown command ... %CurCmd%
  CurCmd := EnvParseStr(CurCmd)
  RunWait %CurCmd%
}
If runeje0 > 0
  StepsPos++

Progress % StepsPos*StepsStep, Checking running processes ...

EnumProcesses(procss)
StringSplit procs, procss, |
KillProcs0 = 0
Loop %U3_HOST_EXEC_PATH%, 2
  U3HEP := A_LoopFileLongPath
Loop %U3_DEVICE_EXEC_PATH%, 2
  U3DEP := A_LoopFileLongPath
Loop %U3_APP_DATA_PATH%, 2
  U3ADP := A_LoopFileLongPath
Loop %A_ScriptFullPath%
  ASFP := A_LoopFileLongPath
Loop %procs0%
{
  CurProc := procs%A_Index%
  CurFn := GetModuleFileNameEx(CurProc)
  if (StrLen(CurFn) > 0)
  {
    Loop %CurFn%
    {
      CurFn := A_LoopFileLongPath
      CurFnOnly := A_LoopFileName
    }
    if ( (CurFn <> ASFP) && ( (SubStr(CurFn, 1, StrLen(U3HEP)) = U3HEP) || ( (StrLen(U3DEP) > 0) && (SubStr(CurFn, 1, StrLen(U3DEP)) = U3DEP) ) || ( (StrLen(U3ADP) > 0) && (SubStr(CurFn, 1, StrLen(U3ADP)) = U3ADP) ) ) )
    {
      KillProcs0++
      KillProcs%KillProcs0% := CurProc . "|" . CurFnOnly
    }
  }
}
Loop %KillProcs0%
{
  OutIndex = %A_Index%
  CurProc := KillProcs%A_Index%
  SplitFirst(CurPID, CurFn, CurProc, "|")

  Loop %KillTries%
  {
    Process Exist, %CurPID%
    If ErrorLevel
      ProgPID = %ErrorLevel%
    Else
      Break

    If (A_Index < KillTries-1)
    {
      Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/KillTries))/KillProcs0, Stopping process ... %CurFn% [%CurPID%]
      WinClose ahk_pid %ProgPID%, , 0
    }
    Else
    {
      Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/KillTries))/KillProcs0, Killing process ... %CurFn% [%CurPID%]
      Process Close, %ProgPID%
    }
  }
}
StepsPos++

If (U3_IS_DEVICE_AVAILABLE <> "true")
{
  ; U3 stick not plugged in!!
  MsgBox 4112, U3 Device Not Available (%U3_IS_DEVICE_AVAILABLE%), Your U3 Device seems to be disconnected. The settings cannot be saved!`n`nAll your changes made since plugging in the U3 Device are likely to be lost. Try to manually save them now.`n`n%U3_HOST_EXEC_PATH%`n`nAfter pressing OK, registry entries will be removed.
  If regbak0 > 0
    StepsPos++
  If datexe0 > 0
    StepsPos++
  
  ; Try to save to temporary directory ... maybe change U3_APP_DATA_PATH to new dir ... should work
}
Else
{
  ; U3 stick plugged in
  IniRead ForeignSettings, %INIFile%, U3Helper, ForeignSettings, 0
  If (ForeignSettings <> "0")
  {
    MsgBox 4132, %AppName%: Foreign settings, You chose to use the settings previously stored on this PC for %AppName%.`n`nDo you want to copy them to your U3 stick so that you have these settings available the next time?
    IfMsgBox Yes
    {
      IniDelete %INIFile%, U3Helper, ForeignSettings
    }
  }

  ;******************************************************************************

  IniRead ForeignSettings, %INIFile%, U3Helper, ForeignSettings, 0
  If (ForeignSettings = "0")
  {
    Loop %regbak0%
    {
      Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/regbak0, Translating paths in registry #%A_Index% ...
      CurBranch := regbak%A_Index%
      SplitFirst(RegRoot, RegSub, CurBranch, "\")
      Loop %RegRoot%, %RegSub%, 0, 1
      {
        If (A_LoopRegType = "REG_SZ" or A_LoopRegType = "REG_EXPAND_SZ" or A_LoopRegType = "REG_MULTI_SZ")
        {
          RegRead RegValue
          NewRegValue := EnvUnparseStr(RegValue)
          If (NewRegValue <> RegValue)
          {
            RegWrite %NewRegValue%
          }
        }
      }
      Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/regbak0, Saving registry settings #%A_Index% ...
      RunWait regedit /E "%U3_APP_DATA_PATH%\regdata%A_Index%.reg" "%CurBranch%"
    }
  }
  
  If regbak0 > 0
    StepsPos++

  ;******************************************************************************

  ;Copy data files
  SetWorkingDir %U3_HOST_EXEC_PATH%
  CopyErrors := ""
  Loop %datexe0%
  {
    OutIndex = %A_Index%
    CurMask := datexe%A_Index%
    Copied = 0
    Skipped = 0
    Errors = 0
    Dirs = 0

    FileGetAttrib FAttr, %CurMask%
    IfInString FAttr, D
      CurMask .= "\*.*"

    FileCount = 0
    Loop %CurMask%, 1, 1
    {
      FileCount++
    }
    Loop %CurMask%, 1, 1
    {
      CurFile := A_LoopFileFullPath
      TargFile := U3_APP_DATA_PATH . "\" . CurFile
      FileGetAttrib FAttr, %CurFile%
      IfInString FAttr, D
      {
        ; also create empty directories
        Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/FileCount))/datexe0, Creating directory %CurFile% ... (CPY:%Copied% / DIR:%Dirs% / SKP:%Skipped% / ERR:%Errors%)
        FileCreateDir %TargFile%
        Dirs++
      }
      Else 
      {
        Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/FileCount))/datexe0, Copying data %CurFile% ... (CPY:%Copied% / DIR:%Dirs% / SKP:%Skipped% / ERR:%Errors%)
        FileCopyNewer(CurFile, TargFile)
        If ErrorLevel = 2
        {
          CopyErrors .= "Dir-entry: " . CurFile . "\" . A_LoopFileFullPath . " (Error while copying)`n"
          Errors++
        }
        Else If ErrorLevel = 1
        {
          CopyErrors .= "Dir-entry: " . CurFile . "\" . A_LoopFileFullPath . " (File does not exist)`n"
          Errors++
        }
        Else If ErrorLevel = -1
          Skipped++
        Else
          Copied++
      }
    }
  }
  SetWorkingDir %A_ScriptDir%

  If (CopyErrors <> "")
    MsgBox 4112, Error while copying, Following files could not be backed up:`n`n%CopyErrors%`n`nTry to manually save them now.`n`n%U3_HOST_EXEC_PATH%`n`nAfter pressing OK, those files will be deleted.

  If datexe0 > 0
    StepsPos++

  ;******************************************************************************
  
  ;Cleaning deleted files from stick
  SetWorkingDir %U3_APP_DATA_PATH%
  Loop %datexe0%
  {
    OutIndex = %A_Index%
    CurMask := datexe%A_Index%
    Deleted = 0
    Skipped = 0
    Errors = 0

    FileGetAttrib FAttr, %CurMask%
    IfInString FAttr, D
      CurMask .= "\*.*"

    FileCount = 0
    Loop %CurMask%, 1, 1
    {
      FileCount++
    }
    Loop %CurMask%, 1, 1
    {
      CurFile := A_LoopFileFullPath
      Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/FileCount))/datexe0, Cleaning data %CurFile% ... (DEL:%Deleted% / SKP:%Skipped% / ERR:%Errors%)
      TargFile := U3_HOST_EXEC_PATH . "\" . CurFile
      IfExist %TargFile%
      {
        ; file still exists on host
        Skipped++
        Continue
      }
      ; target directory or file doesn't exist anymore, delete on U3
      FileSetAttrib -RSH, %CurFile%
      FileGetAttrib FAttr, %CurFile%
      IfInString FAttr, D
      {
        ; target is a directory
        FileSetAttrib -RSH, %CurFile%\*.*, 1, 1
        FileRemoveDir %CurFile%, 1
        If ErrorLevel
          Errors++
        Else
          Deleted++
      }
      Else
      {
        ; target is a single file only
        FileDelete %CurFile%
        If ErrorLevel
          Errors++
        Else
          Deleted++
      }
    }
  }
  SetWorkingDir %A_ScriptDir%
  
  If datexe0 > 0
    StepsPos++

  ;******************************************************************************

  ;Translate paths in text files
  SetWorkingDir %U3_APP_DATA_PATH%
  TransErrors := ""
  Loop %dattxt0%
  {
    OutIndex = %A_Index%
    CurMask := dattxt%A_Index%
      
    FileCount = 0
    Loop %CurMask%, 0, 0
    {
      FileCount++
    }
    Loop %CurMask%, 0, 0
    {
      CurFile := A_LoopFileFullPath
      TmpFile := A_LoopFileDir . "\$$$" . A_LoopFileName
      FileMove %U3_APP_DATA_PATH%\%CurFile%, %U3_APP_DATA_PATH%\%TmpFile%, 1
      Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/FileCount))/dattxt0, Translating paths in file %CurFile% ...
      Loop Read, %U3_APP_DATA_PATH%\%TmpFile%, %U3_APP_DATA_PATH%\%CurFile%
      {
        IfNotInString A_LoopReadLine, \
        {
          ; no paths to replace --- skip processing
          FileAppend %A_LoopReadLine%`n
          Continue
        }
        FileAppend % EnvUnparseStr(A_LoopReadLine) . "`n"
      }
      IfExist %U3_APP_DATA_PATH%\%CurFile%
      {
        FileDelete %U3_APP_DATA_PATH%\%TmpFile%
      }
      Else
      {
        FileMove %U3_APP_DATA_PATH%\%TmpFile%, %U3_APP_DATA_PATH%\%CurFile%
        TransErrors .= "File: " . CurFile . " (Error while translating)`n"
      }
    }
  }
  SetWorkingDir %A_ScriptDir%

  If (TransErrors <> "")
    MsgBox 4112, Error while translating, Following files could not be translated:`n`n%TransErrors%`n`nThe original state has been restored (hopefully).

  If dattxt0 > 0
    StepsPos++
}

;******************************************************************************

IniRead KeepSettings, %INIFile%, U3Helper, KeepSettings, 0
RevertSettings := "0"
If (KeepSettings <> "0")
{
  If (Unattended = "0")
  {
    MsgBox 4131, %AppName%: Foreign settings, Do you want to keep the changed settings for %AppName%?`n`n(Select No to revert to the former settings. Cancel to erase settings.)
    IfMsgBox No
    {
      RevertSettings := "1"
    }
    IfMsgBox Cancel
    {
      KeepSettings := "0"
    }
  }
  Else
  {
    RevertSettings := "1"
  }
}
If (KeepSettings = "0" or RevertSettings = "1")
{
  Loop %regbak0%
  {
    Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/regbak0, Removing registry settings #%A_Index% from host ...
    CurBranch := regbak%A_Index%
    SplitFirst(RegRoot, RegSub, CurBranch, "\")
    RegDeleteX(RegRoot, RegSub)
    If (RevertSettings = "1")
    {
      Progress % StepsPos*StepsStep+StepsStep*(A_Index-0.5)/regbak0, Restoring registry settings #%A_Index% from backup ...
      RunWait regedit /S "%U3_HOST_EXEC_PATH%\U3Hregbak%A_Index%.reg"
    }
  }
}

If regbak0 > 0
  StepsPos++
  
;******************************************************************************

If (KeepSettings = "0" or Unattended = "1")
{

  ;******************************************************************************

  Loop %regdel0%
  {
    Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/regdel0, Removing add. registry settings #%A_Index% from host ...
    CurBranch := regdel%A_Index%
    SplitFirst(RegRoot, RegSub, CurBranch, "\")
    RegDeleteX(RegRoot, RegSub)
    If ErrorLevel
    {
      SplitLast(RegSub, RegKey, RegSub, "\")
      RegDelete %RegRoot%, %RegSub%, %RegKey%
    }
  }
  
  If regdel0 > 0
    StepsPos++

  ;******************************************************************************

  ; regsvr32 stuff
  Loop %regsvr0%
  {
    CurDLL := regsvr%A_Index%
    Progress % StepsPos*StepsStep+StepsStep*(A_Index-1)/regsvr0, Unregistering file %CurDLL% ...
    RunWait regsvr32 /S /U "%U3_HOST_EXEC_PATH%\%CurDLL%"
  }
  
  If regsvr0 > 0
    StepsPos++
  
  ;******************************************************************************

  ; remove files
  ; if not specified, all file operations run on A_ScriptDir which is U3_HOST_EXEC_PATH
  Loop %fildel0%
  {
    OutIndex = %A_Index%
    CurMask := fildel%A_Index%
    Deleted = 0
    Errors = 0
    Skipped = 0
    
    CurMask := EnvParseStr(CurFile)
    FileGetAttrib FAttr, %CurMask%
    IfInString FAttr, D
      CurMask .= "\*.*"
    
    FileCount = 0
    Loop %CurMask%, 1, 1
    {
      FileCount++
    }
    Loop %CurMask%, 1, 1
    {
      CurFile := A_LoopFileFullPath
      Progress % StepsPos*StepsStep+StepsStep*(OutIndex-1.00+(A_Index/FileCount))/fildel0, Removing data %CurFile% from host ... (DEL:%Deleted% / SKP:%Skipped% / ERR:%Errors%)
      IfNotExist %CurFile%
      {
        Skipped++
        Continue
      }
      FileSetAttrib -RSH, %CurFile%
      FileGetAttrib FAttr, %CurFile%
      IfInString FAttr, D
      {
        ; target is a directory
        FileSetAttrib -RSH, %CurFile%\*.*, 1, 1
        FileRemoveDir %CurFile%, 1
        If ErrorLevel
          Errors++
        Else
          Deleted++
      }
      Else
      {
        ; target is a single file only
        FileDelete %CurFile%
        If ErrorLevel
          Errors++
        Else
          Deleted++
      }
    }
  }
  
  If fildel0 > 0
    StepsPos++

  ;******************************************************************************

}
Else
{
  If regdel0 > 0
    StepsPos++
  If regsvr0 > 0
    StepsPos++
  If fildel0 > 0
    StepsPos++
}

;******************************************************************************

Progress 100, hostCleanUp done.

If (U3_IS_DEVICE_AVAILABLE = "true")
{
  IniDelete %INIFile%, U3Helper, ForeignSettings
  IniDelete %INIFile%, U3Helper, KeepSettings
}