# WK-Checklist

## Init ##
$wd = $(Split-Path $MyInvocation.MyCommand.Path)
pushd "$wd"
. .\init.ps1
. .\servers.ps1
. .\imaging.ps1
clear
$host.UI.RawUI.WindowTitle = "WK PE Tool"
$logpath = "$WKPath\Info\$date"
md "$logpath" 2>&1 | Out-Null
$log = "$logpath\winpe.log"

# Functions
function wk-exit {
    param([string]$action)
    switch ($action) {
        'Q' {PowerShell -ExecutionPolicy Bypass; break}
        'R' {wpeutil reboot; break}
        'S' {wpeutil shutdown; break}
        default {throw}
    }
    exit 0
}
function menu-tools {
    # Avail tools
    $tools = @(
        @{Name="Blue Screen View"; Folder="BlueScreenView"; File="BlueScreenView64.exe"},
        @{Name="Explorer++"; Folder="Explorer++"; File="Explorer++64.exe"},
        @{Name="Fast Copy"; Folder="FastCopy"; File="FastCopy64.exe"; Args=@('/cmd=noexist_only', '/utf8', '/skip_empty_dir', '/linkdest', '/exclude="desktop.ini;Thumbs.db"')},
        @{Name="HW Monitor"; Folder="HWMonitor"; File="HWMonitor64.exe"},
        @{Name="NT Password Editor"; Folder="NT Password Editor"; File="ntpwedit64.exe"},
        @{Name="Notepad2"; Folder="Notepad2"; File="Notepad2-Mod64.exe"},
        @{Name="Prime95"; Folder="Prime95"; File="prime95.exe"},
        @{Name="PhotoRec (GUI)"; Folder="TestDisk"; File="qphotorec_win.exe"},
        @{Name="PhotoRec (CLI)"; Folder="TestDisk"; File="photorec_win.exe"},
        @{Name="TestDisk (CLI)"; Folder="TestDisk"; File="testdisk_win.exe"}
    )
    
    # Build menu
    $selection = $null
    $actions = @(@{Name="Main Menu"; Letter="M"})
    
    # Run Loop
    $_done = $false
    do {
        $selection = (menu-select "Tools Menu" $tools $actions)
        
        if ($selection -imatch '^M$') {
            # User selected to return to the menu
            return $false
        } elseif ($selection -inotmatch '^\d+$') {
            # This shouldn't happen?
            throw
        } else {
            $selection -= 1
            $path = "{0}\{1}" -f $WKPath, $tools[$selection].Folder
            if ($tools[$selection].ContainsKey("Args")) {
                Start-Process $tools[$selection].File -ArgumentList $tools[$selection].Args -WorkingDirectory $path
            } else {
                Start-Process $tools[$selection].File -WorkingDirectory $path
            }
        }
    } until ($_done)
}
function menu-main {
    # Build menu
    $selection = $null
    $menus = @(
        @{Name="Drive Imaging"; Menu="menu-imaging"}
        @{Name="Windows Setup"; Menu="menu-setup"}
        @{Name="Misc Tools"; Menu="menu-tools"}
    )
    $actions = @(
        @{Name="Command Prompt"; Letter="C"}
        @{Name="PowerShell"; Letter="P"}
        @{Name="Reboot"; Letter="R"}
        @{Name="Shutdown"; Letter="S"}
    )
    
    # Show Menu
    $selection = (menu-select "Main Menu" $menus $actions -SecretExit $true)
    
    if ($selection -imatch '^C$') {
        Start-Process "$windir\System32\cmd.exe" -argumentlist @("-new_console:n") -WorkingDirectory "$WKPath"
        return
    } elseif ($selection -imatch '^P$') {
        Start-Process "$windir\System32\WindowsPowerShell\v1.0\powershell.exe" -argumentlist @("-ExecutionPolicy", "Bypass", "-new_console:n") -WorkingDirectory "$WKPath"
        return
    } elseif ($selection -imatch '^[QRS]$') {
        wk-exit $selection
        return
    } elseif ($selection -inotmatch '^\d+$') {
        # This shouldn't happen?
        throw
    } else {
        # Launch sub-menu
        $selection -= 1
        & $menus[$selection].Menu
    }
}

# Main Loop
do {
    menu-main
} while ($true)