<#
.NOTES
Author              : @ChrisTitusTech   
Runspace Author     : @DeveloperDurp
Version 0.1
#>

#region Variables
    $sync = [Hashtable]::Synchronized(@{})
    $sync.logfile = "$env:TEMP\winutil.log"
    
    $sync.taskrunning = $false
    $sync.taskmessage = "There is currently a task running. Please try again once previous task is complete."
    $sync.tasktitle = "Task in progress"

    $VerbosePreference = "Continue"
    if(!$env:args){$gui = $true}

#endregion Variables
 
#region Functions

    #===========================================================================
    # Button clicks
    #===========================================================================

    function Invoke-Button {

        <#
        
            .DESCRIPTION
            Meant to make creating buttons easier. There is a section below in the gui that will assign this function to every button.
            This way you can dictate what each button does from this function. 
        
            Input will be the name of the button that is clicked. 
        #>
        
        Param ([string]$Button) 
        Switch -Wildcard ($Button){

            "*Tab*BT*" {switchtab $Button}
            "*InstallUpgrade*" {Invoke-command $sync.GUIInstallPrograms -ArgumentList "Upgrade"}
            "*desktop*" {Tweak-Buttons $Button}
            "*laptop*" {Tweak-Buttons $Button}
            "*minimal*" {Tweak-Buttons $Button}
            "*undoall*" {Invoke-command $Sync.GUIUndoTweaks}
            "install" {Invoke-command $sync.GUIInstallPrograms -ArgumentList "$(uncheckall "Install")"}
            "tweaksbutton" {Invoke-command $Sync.GUITweaks -ArgumentList "$(uncheckall "tweaks")"}
            "FeatureInstall" {Invoke-command $Sync.GUIFeatures -ArgumentList "$(uncheckall "feature")"}
            "Panelcontrol" {cmd /c control}
            "Panelnetwork" {cmd /c ncpa.cpl}
            "Panelpower" {cmd /c powercfg.cpl}
            "Panelsound" {cmd /c mmsys.cpl}
            "Panelsystem" {cmd /c sysdm.cpl}
            "Paneluser" {cmd /c "control userpasswords2"}
            "Updates*" {Invoke-command $sync.GUIUpdates -ArgumentList "$button"}
        }
    }

    function uncheckall {

        <#
        
            .DESCRIPTION
            Function is meant to find all checkboxes that are checked on the specefic tab and input them into a script.

            Outputed data will be the names of the checkboxes comma seperated. 

            "Installadvancedip,Installbitwarden"

            .EXAMPLE

            uncheckall "Install"
        
        #>

        param($group)

        if ($sync.taskrunning -eq $true){
            return
        }

        $sync.keys | Where-Object {$psitem -like "*$($group)?*" `
                                -and $psitem -notlike "$($group)Install" `
                                -and $psitem -notlike "*GUI*" `
                                -and $psitem -notlike "*Script*"
                            } | ForEach-Object {
            if ($sync["$psitem"].IsChecked -eq $true){
                $output += ",$psitem"
                $sync["$psitem"].IsChecked = $false
            }
        }
        
        if($output){Write-Output $output.Substring(1)}
    }

    function Invoke-Runspace {

        <#
        
            .DESCRIPTION
            Simple function to make it easier to invoke a runspace from inside the script. 

            .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptsInstallPrograms
                ArgumentList = "Installadvancedip,Installbitwarden"
                Verbose = $true
            }

            Invoke-Runspace @params
        
        #>

        [CmdletBinding()]
        Param (
            $ScriptBlock,
            $ArgumentList
        ) 

        $Script = [PowerShell]::Create().AddScript($ScriptBlock).AddArgument($ArgumentList)

        $Script.Runspace = $runspace
        $Script.BeginInvoke()
    }

    #===========================================================================
    # Navigation Controls
    #===========================================================================

    function switchtab {

        <#
        
            .DESCRIPTION
            Sole purpose of this fuction reduce duplicated code for switching between tabs. 
        
        #>

        Param ($button)
        $x = [int]($button -replace "Tab","" -replace "BT","") - 1

        0..3 | ForEach-Object {
            
            if ($x -eq $psitem){$sync["TabNav"].Items[$psitem].IsSelected = $true}
            else{$sync["TabNav"].Items[$psitem].IsSelected = $false}
        }
    }

    Function Tweak-Buttons {

        <#
        
            .DESCRIPTION
            Meant to make settings presets easier in the tweaks tab. Will pull the data from config/preset.json
        
        #>

        Param ($button)
        $preset = $sync.preset.$button

        $sync.keys | Where-Object {$psitem -like "*tweaks?*" -and $psitem -notlike "tweaksbutton"} | ForEach-Object {
            if ($preset -contains $psitem ){$sync["$psitem"].IsChecked = $True}
            Else{$sync["$psitem"].IsChecked = $false} 
        }
    }

#endregion Functions

#===========================================================================
# Scritps to be ran inside a runspace
#===========================================================================

#region Scripts

    #===========================================================================
    # Generic Scripts
    #===========================================================================

    $sync.WriteLogs = {

        <#
        
            .DESCRIPTION
            Simple function to write logs to a temp directory.

            .EXAMPLE

            $Level = "INFO"
            $Message = "This is a test message!"
            $LogPath = "$ENV:TEMP\winutil.log"
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message,$LogPath)
        
        #>

        [cmdletbinding()]
        param(
            $Level = "Info", 
            $Message, 
            $LogPath = "$env:TEMP\winutil.log"
        )

        $date = get-date
        $delimiter = '|'
        write-output "$date $delimiter $Level $delimiter $message" |  out-file -Append -Encoding ascii -FilePath $LogPath
        if($Level -eq "ERROR" -or $Level -eq "FAILURE"){
            write-Error "$date $delimiter $Level $delimiter $message"
            return
        }
        if($Level -eq "Warning"){
            Write-Warning "$date $delimiter $Level $delimiter $message"
            return
        }
        Write-Verbose "$date $delimiter $Level $delimiter $message"
    }

    #===========================================================================
    # Install Tab
    #===========================================================================

    <#

        This section is working as expected and logs output to console and $ENV:Temp\winutil.log

        TODO: Error Handling with winget. Currently it does not handle errors as expected.
    
    #>

    $Sync.GUIInstallPrograms = {

        <#

            .DESCRIPTION
            This Scriptblock is meant to be ran from inside the GUI and will prevent the user from starting another install task. 

            Input data will look like below and link with the name of the check box. This will then look to the config/applications.json file to find
            the winget install commands for the selected applications.

            Installadvancedip,Installbitwarden

            .EXAMPLE

            Invoke-command $sync.GUIInstallPrograms -ArgumentList "Installadvancedip,Installbitwarden"

        #>

        Param ($programstoinstall)

        #Check if any check boxes have been checked and if a task is currently running

            if ($sync.taskrunning -eq $true){
                [System.Windows.MessageBox]::Show($sync.taskmessage,$sync.tasktitle,"OK","Info")
                return
            }

            if($programstoinstall -notlike "*install*"){
                [System.Windows.MessageBox]::Show("Please check the applications you wish to install",'Nothing to do',"OK","Info")
                return
            }

        #Section to see if winget will upgrade all installs or which winget commands to run from  config/applications.json
        
            $programstoinstall = $programstoinstall -split ","

            if($programstoinstall -eq "Upgrade"){
                $winget = ",Upgrade"
            }
            else{
                foreach ($program in $programstoinstall){

                    $($sync.applications.install.$program.winget) -split ";" | ForEach-Object {
                        if($psitem){
                            $winget += ",$psitem"
                        }Else{
                            Invoke-command $sync.WriteLogs -ArgumentList ("WARNING","$Program Not found") 
                        }
                    }
                }
            }
            
            if($winget -eq $null){
                [System.Windows.MessageBox]::Show("No found applications to install",'Nothing to do',"OK","Info")
                return
            }            

            
        #Invoke a runspace so that the GUI does not lock up

            $sync.taskrunning = $true

            $params = @{
                ScriptBlock = $sync.ScriptsInstallPrograms
                ArgumentList = "$($winget.substring(1))"
                Verbose = $true
            }
            Invoke-Runspace @params

    }

    $sync.ScriptsInstallPrograms = {

        <#

            .DESCRIPTION
            This scriptblock will detect if winget is installed and if not attempt to install it. Once ready it will then either upgrade any installs or attempt to install any applications provided.

            .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptsInstallPrograms
                ArgumentList = "git.git,WinDirStat.WinDirStat"
            }
            VerbosePreference = "Continue"
            Invoke-Command @params

        .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptsInstallPrograms
                ArgumentList = "Upgrade"
            }

            VerbosePreference = "Continue"
            Invoke-Command @params

        #>

        Param ($programstoinstall)
        $programstoinstall = $programstoinstall -split ","

        function Write-Logs {
            param($Level, $Message, $LogPath)
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message,$LogPath)
        }

        #region Check for WinGet and install if not present

            if (Test-Path $env:userprofile\AppData\Local\Microsoft\WindowsApps\winget.exe) {
                #Checks if winget executable exists and if the Windows Version is 1809 or higher
                Write-Logs -Level INFO -Message "WinGet was detected" -LogPath $sync.logfile
            }
            else {

                if (($sync.ComputerInfo.WindowsVersion) -lt "1809") {
                    #Checks if Windows Version is too old for winget
                    Write-Logs -Level Warning -Message "Winget is not supported on this version of Windows (Pre-1809). Stopping installs" -LogPath $sync.logfile
                    return
                }

                Write-Logs -Level INFO -Message "WinGet was not detected" -LogPath $sync.logfile

                if (((($sync.ComputerInfo.OSName.IndexOf("LTSC")) -ne -1) -or ($sync.ComputerInfo.OSName.IndexOf("Server") -ne -1)) -and (($sync.ComputerInfo.WindowsVersion) -ge "1809")) {
                    Try{
                        #Checks if Windows edition is LTSC/Server 2019+
                        #Manually Installing Winget
                        Write-Logs -Level INFO -Message "LTSC/Server Edition detected. Running Alternative Installer" -LogPath $sync.logfile
        
                        #Download Needed Files
                        $step = "Downloading the required files"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile
                        Start-BitsTransfer -Source "https://aka.ms/Microsoft.VCLibs.x64.14.00.Desktop.appx" -Destination "$ENV:TEMP\Microsoft.VCLibs.x64.14.00.Desktop.appx" -ErrorAction Stop
                        Start-BitsTransfer -Source "https://github.com/microsoft/winget-cli/releases/download/v1.2.10271/Microsoft.DesktopAppInstaller_8wekyb3d8bbwe.msixbundle" -Destination "$ENV:TEMP/Microsoft.DesktopAppInstaller_8wekyb3d8bbwe.msixbundle" -ErrorAction Stop
                        Start-BitsTransfer -Source "https://github.com/microsoft/winget-cli/releases/download/v1.2.10271/b0a0692da1034339b76dce1c298a1e42_License1.xml" -Destination "$ENV:TEMP/b0a0692da1034339b76dce1c298a1e42_License1.xml" -ErrorAction Stop
        
                        #Installing Packages
                        $step = "Installing Packages"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile
                        Add-AppxProvisionedPackage -Online -PackagePath "$ENV:TEMP\Microsoft.VCLibs.x64.14.00.Desktop.appx" -SkipLicense -ErrorAction Stop
                        Add-AppxProvisionedPackage -Online -PackagePath "$ENV:TEMP\Microsoft.DesktopAppInstaller_8wekyb3d8bbwe.msixbundle" -LicensePath "$ENV:TEMP\b0a0692da1034339b76dce1c298a1e42_License1.xml" -ErrorAction Stop
                        
                        #Sleep for 5 seconds to maximize chance that winget will work without reboot
                        Start-Sleep -s 5
        
                        #Removing no longer needed Files
                        $step = "Removing Files"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile
                        Remove-Item -Path "$ENV:TEMP\Microsoft.VCLibs.x64.14.00.Desktop.appx" -Force
                        Remove-Item -Path "$ENV:TEMP\Microsoft.DesktopAppInstaller_8wekyb3d8bbwe.msixbundle" -Force
                        Remove-Item -Path "$ENV:TEMP\b0a0692da1034339b76dce1c298a1e42_License1.xml" -Force

                        $step = "WinGet Sucessfully installed"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile

                    }Catch{Write-Logs -Level FAILURE -Message "WinGet Install failed at $step" -LogPath $sync.logfile}
                }
                else {
                    Try{
                        #Installing Winget from the Microsoft Store                       
                        $step = "Installing WinGet"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile
                        Start-Process "ms-appinstaller:?source=https://aka.ms/getwinget"
                        $nid = (Get-Process AppInstaller).Id
                        Wait-Process -Id $nid

                        $step = "Winget Installed"
                        Write-Logs -Level INFO -Message $step -LogPath $sync.logfile
                    }Catch{Write-Logs -Level FAILURE -Message "WinGet Install failed at $step" -LogPath $sync.logfile}
                }
                Write-Logs -Level INFO -Message "WinGet has been installed" -LogPath $sync.logfile
                Start-Sleep -Seconds 15
            }

        #endregion Check for WinGet and install if not present

        $results = @()
        foreach ($program in $programstoinstall){
            if($programstoinstall -eq "Upgrade"){
                $Message = "Attempting to upgrade packages"
                $ErrorMessage = "Failed to upgrade packages"
                $SuccessMessage = "Upgardes have completed"
                $ArgumentList = "upgrade --all"
            }
            else{
                $Message = "$($program) was selected to be installed."
                $ErrorMessage = "$($program) failed to installed."
                $SuccessMessage = "$($program) has been installed"
                $ArgumentList = "install -e --accept-source-agreements --accept-package-agreements --silent $($program)"
            }

            try {
                Write-Logs -Level INFO -Message "$Message" -LogPath $sync.logfile
                Write-Host ""
                
                $installs = Start-Process -FilePath winget -ArgumentList $ArgumentList -ErrorAction Stop -Wait -PassThru -NoNewWindow
            }
            catch {
                Write-Logs -Level FAILURE -Message $ErrorMessage -LogPath $sync.logfile
                $results += $program
            }
        }

        Write-Logs -Level INFO -Message "Installs have completed" -LogPath $sync.logfile
        if($sync["Form"]){
            $sync.taskrunning = $false
            [System.Windows.MessageBox]::Show("All applications have been installed",'Installs are done!',"OK","Info")
        }
    }

    #===========================================================================
    # Tab 2 - Tweaks Buttons
    #===========================================================================

    <#

        This section is working as expected and logs output to console and $ENV:Temp\winutil.log

        TODO: Error Handling as Try blocks and -erroraction stop causes runspace to lock up
    
    #>

    $Sync.GUITweaks = {

        <#

            .DESCRIPTION
            This Scriptblock is meant to be ran from inside the GUI and will prevent the user from starting another install task. 

            Input data will look like below and link with the name of the check box. This will then look to the config/applications.json file to find
            the modifications for the selected task.

            EssTweaksDeBloat,MiscTweaksUTC

            .EXAMPLE

            Invoke-command $sync.GUIInstallPrograms -ArgumentList "EssTweaksDeBloat,MiscTweaksUTC"

        #>

        Param($Tweakstorun)

        #Check if any check boxes have been checked and if a task is currently running

            if ($sync.taskrunning -eq $true){
                [System.Windows.MessageBox]::Show($sync.taskmessage,$sync.tasktitle,"OK","Info")
                return
            }

            if($Tweakstorun -notlike "*Tweaks*"){
                [System.Windows.MessageBox]::Show("Please check the applications you wish to install",'Nothing to do',"OK","Info")
                return
            }

            $sync.taskrunning = $true

        #Invoke a runspace so that the GUI does not lock up

            $params = @{
                ScriptBlock = $sync.ScriptTweaks
                ArgumentList = ("$Tweakstorun")
            }
            
            Invoke-Runspace @params
    
    }

    $Sync.ScriptTweaks = {

        <#

            .DESCRIPTION
            This scriptblock will run a series of modifications included in the config/tweaks.json file. 

            TODO: Figure out error handling as any errors in this runspace will crash the powershell session.

            .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptsInstallPrograms
                ArgumentList = "EssTweaksTele,EssTweaksServices"
                Verbose = $true
            }
            VerbosePreference = "Continue"
            Invoke-Command @params

        #>

        Param($Tweakstorun)
        $Tweakstorun = $Tweakstorun -split ","

        $ErrorActionPreference = "SilentlyContinue"

        function Write-Logs {
            param($Level, $Message, $LogPath)
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message, $LogPath)
        }

        Write-Logs -Level INFO -Message "Gathering required modifications" -LogPath $sync.logfile

        $RegistryToModify = $Tweakstorun | ForEach-Object {
            $sync.tweaks.$psitem.registry
        }

        $ServicesToModify = $Tweakstorun | ForEach-Object {
            $sync.tweaks.$psitem.service
        }

        $ScheduledTaskToModify = $Tweakstorun | ForEach-Object {
            $sync.tweaks.$psitem.ScheduledTask
        }

        $AppxToModify = $Tweakstorun | ForEach-Object {
            $sync.tweaks.$psitem.appx
        }
    
        $ScriptsToRun = $Tweakstorun | ForEach-Object {
            $sync.tweaks.$psitem.InvokeScript
        }

        if($RegistryToModify){
            Write-Logs -Level INFO -Message "Starting Registry Modification" -LogPath $sync.logfile

            $RegistryToModify | ForEach-Object {
                if(!(Test-Path $psitem.path)){
                    $Step = "create"
                    Write-Logs -Level INFO -Message "$($psitem.path) did not exist. Creating" -LogPath $sync.logfile
                    New-Item -Path $psitem.path -Force | Out-Null
                }

                $step = "set"
                Write-Logs -Level INFO -Message "Setting $("$($psitem.path)\$($psitem.name)") to $($psitem.value)" -LogPath $sync.logfile
                Set-ItemProperty -Path $psitem.path -Name $psitem.name -Type $psitem.type -Value $psitem.value
            }

            Write-Logs -Level INFO -Message "Finished setting registry" -LogPath $sync.logfile
        }

        if($ServicesToModify){
            Write-Logs -Level INFO -Message "Starting Services Modification" -LogPath $sync.logfile

            $ServicesToModify | ForEach-Object {
                    $service = Get-Service -Name W32Time -ErrorAction SilentlyContinue
                    if ($service.Length -gt 0) {
                        Stop-Service "$($psitem.name)" 
                        Set-Service "$($psitem.name)" -StartupType $($psitem.StartupType)
                        Write-Logs -Level INFO -Message "Service $($psitem.name) set to  $($psitem.StartupType)" -LogPath $sync.logfile
                    }
            }
    
            Write-Logs -Level INFO -Message "Finished setting Services" -LogPath $sync.logfile
        }
        
        if($ScheduledTaskToModify){
            Write-Logs -Level INFO -Message "Starting ScheduledTask Modification" -LogPath $sync.logfile

            $ScheduledTaskToModify | ForEach-Object {
                Try{
                    if($($psitem.State) -eq "Disabled"){
                        Disable-ScheduledTask -TaskName "$($psitem.name)" -ErrorAction Stop | Out-Null
                    }
                    if($($psitem.State) -eq "Enabled"){
                        Enable-TaskName "$($psitem.name)" -ErrorAction Stop | Out-Null
                    }
                    Write-Logs -Level INFO -Message "Scheduled Task $($psitem.name) set to  $($psitem.State)" -LogPath $sync.logfile
                }Catch{Write-Logs -Level ERROR -Message "Unable to set Scheduled Task $($psitem.name) set to  $($psitem.State)" -LogPath $sync.logfile}
            }
    
            Write-Logs -Level INFO -Message "Finished setting ScheduledTasks" -LogPath $sync.logfile
        }

        if($AppxToModify){
            Write-Logs -Level INFO -Message "Starting Appx Modification" -LogPath $sync.logfile

            $AppxToModify | ForEach-Object {
                Try{
                    Get-AppxPackage -Name $psitem| Remove-AppxPackage -ErrorAction Stop
                    Get-AppxProvisionedPackage -Online | Where-Object DisplayName -like $psitem | Remove-AppxProvisionedPackage -ErrorAction stop -Online
                    Write-Logs -Level INFO -Message "Uninstalled $psitem" -LogPath $sync.logfile
                }Catch{Write-Logs -Level ERROR -Message "Failed to uninstall $psitem" -LogPath $sync.logfile }
            }
    
            Write-Logs -Level INFO -Message "Finished uninstalling Appx" -LogPath $sync.logfile
        }

        if($ScriptsToRun){
            Write-Logs -Level INFO -Message "Running Scripts" -LogPath $sync.logfile

            $ScriptsToRun | ForEach-Object {
                $Scriptblock = [scriptblock]::Create($psitem)
                #Invoke-Command -ScriptBlock $Scriptblock
                Start-Process $PSHOME\powershell.exe -Verb runas -ArgumentList "-Command  $scriptblock" -Wait
            }
    
            Write-Logs -Level INFO -Message "Finished Scripts" -LogPath $sync.logfile
        }

        Write-Logs -Level INFO -Message "Tweaks finished" -LogPath $sync.logfile
        
        if($sync["Form"]){
            $sync.taskrunning = $false
            [System.Windows.MessageBox]::Show("All modifications have finished",'Tweaks are done!',"OK","Info")
        }
    }

    $Sync.GUIUndoTweaks = {

        <#

            .DESCRIPTION
            This Scriptblock is meant to be ran from inside the GUI and will prevent the user from starting another tweak task. 

            .EXAMPLE

            Invoke-command $sync.GUIUndoTweaks

        #>

        #Check if any check boxes have been checked and if a task is currently running

            if ($sync.taskrunning -eq $true){
                [System.Windows.MessageBox]::Show($sync.taskmessage,$sync.tasktitle,"OK","Info")
                return
            }

            $sync.taskrunning = $true

        #Invoke a runspace so that the GUI does not lock up            

            Invoke-Runspace $sync.ScriptUndoTweaks
    }

    $sync.ScriptUndoTweaks = {
        
        <#

            .DESCRIPTION
            This scriptblock will undo all modifications from this script. 

            TODO: Figure out error handling as any errors in this runspace will crash the powershell session.

            .EXAMPLE

            VerbosePreference = "Continue"
            Invoke-Command -ScriptBlock $sync.ScriptUndoTweaks
        #>

        $ErrorActionPreference = "SilentlyContinue"

        function Write-Logs {
            param($Level, $Message, $LogPath)
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message, $LogPath)
        }

        Write-Logs -Level INFO -Message "Creating Restore Point incase something bad happens" -LogPath $sync.logfile
        Enable-ComputerRestore -Drive "C:\"
        Checkpoint-Computer -Description "RestorePoint1" -RestorePointType "MODIFY_SETTINGS"

        foreach ($tweak in $($sync.tweaks.psobject.properties)) {

                #registry reset
                Foreach ($registries in $($tweak.value.registry)){
                    foreach($registry in $registries){
                        Write-Logs -Level INFO -Message "Setting $("$($registry.path)\$($registry.name)") to $($registry.OriginalValue)" -LogPath $sync.logfile
                        Set-ItemProperty -Path $registry.path -Name $registry.name -Type $registry.type -Value $registry.OriginalValue
                    }
                }
                Write-Logs -Level INFO -Message "Finished reseting $($tweak.name) registries" -LogPath $sync.logfile

                #Services modification 
                Foreach ($services in $($tweak.value.service)){
                    foreach($service in $services) {
                        Stop-Service "$($service.name)"
                        Set-Service "$($service.name)" -StartupType $($service.OriginalType)
                        Write-Logs -Level INFO -Message "Service $($service.name) set to  $($service.OriginalType)" -LogPath $sync.logfile
                    }
                }
                Write-Logs -Level INFO -Message "Finished reseting $($tweak.name) Services" -LogPath $sync.logfile

                #Scheduled Tasks Modification
                Foreach ($ScheduledTasks in $($tweak.value.ScheduledTask)){
                    foreach($ScheduledTask in $ScheduledTasks) {
                        if($($ScheduledTask.OriginalState) -eq "Disabled"){
                            Disable-ScheduledTask -TaskName "$($ScheduledTask.name)" | Out-Null
                        }
                        if($($ScheduledTask.OriginalState) -eq "Enabled"){
                            Enable-TaskName "$($ScheduledTask.name)"  | Out-Null
                        }
                        Write-Logs -Level INFO -Message "Scheduled Task $($ScheduledTask.name) set to  $($ScheduledTask.OriginalState)" -LogPath $sync.logfile
                    }
                }
                Write-Logs -Level INFO -Message "Finished reseting $($tweak.name) Scheduled Tasks" -LogPath $sync.logfile                  
        }

        Set-ItemProperty -Path "HKLM:\System\CurrentControlSet\Control\Session Manager\Power" -Name "HibernteEnabled" -Type Dword -Value 1
        Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\FlyoutMenuSettings" -Name "ShowHibernateOption" -Type Dword -Value 1
        Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\Personalization" -Name "NoLockScreen" -ErrorAction SilentlyContinue
        
        If (!(Test-Path "HKCU:\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\OperationStatusManager")) {
            Remove-Item -Path "HKCU:\SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\OperationStatusManager" -Recurse -ErrorAction SilentlyContinue
        }
        If (!(Test-Path "HKCU:\SOFTWARE\Microsoft\Siuf\Rules")) {
            Remove-Item -Path "HKCU:\SOFTWARE\Microsoft\Siuf\Rules" -Recurse -ErrorAction SilentlyContinue
        }
        If (!(Test-Path "HKCU:\SOFTWARE\Policies\Microsoft\Windows\CloudContent")) {
            Remove-Item -Path "HKCU:\SOFTWARE\Policies\Microsoft\Windows\CloudContent" -Recurse -ErrorAction SilentlyContinue
        }
        If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\AdvertisingInfo")) {
            Remove-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\AdvertisingInfo" -Recurse -ErrorAction SilentlyContinue
        }
        If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\CloudContent")) {
            Remove-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\CloudContent" -Recurse -ErrorAction SilentlyContinue
        }
        If (!(Test-Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\CapabilityAccessManager\ConsentStore\location")) {
            Remove-Item -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\CapabilityAccessManager\ConsentStore\location" -Recurse -ErrorAction SilentlyContinue
        }

        Write-Logs -Level INFO -Message "Unrestricting AutoLogger directory" -LogPath $sync.logfile
        $autoLoggerDir = "$env:PROGRAMDATA\Microsoft\Diagnosis\ETLLogs\AutoLogger"
        icacls $autoLoggerDir /grant:r SYSTEM:`(OI`)`(CI`)F | Out-Null

        Write-Logs -Level INFO -Message "Reset Local Group Policies to Stock Defaults" -LogPath $sync.logfile        
        # cmd /c secedit /configure /cfg %windir%\inf\defltbase.inf /db defltbase.sdb /verbose
        cmd /c RD /S /Q "%WinDir%\System32\GroupPolicyUsers"
        cmd /c RD /S /Q "%WinDir%\System32\GroupPolicy"
        cmd /c gpupdate /force

        Write-Logs -Level INFO -Message "Restoring Clipboard History..." -LogPath $sync.logfile        
        Remove-ItemProperty -Path "HKCU:\SOFTWARE\Microsoft\Clipboard" -Name "EnableClipboardHistory" -ErrorAction SilentlyContinue
        Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\System" -Name "AllowClipboardHistory" -ErrorAction SilentlyContinue

        Set-ItemProperty -Path "HKCU:\Control Panel\Desktop" -Name "UserPreferencesMask" -Type Binary -Value ([byte[]](158,30,7,128,18,0,0,0))

        if($sync["Form"]){
            $sync.taskrunning = $false
            [System.Windows.MessageBox]::Show("All tweaks have been removed",'Undo is done!',"OK","Info")
        }
    }

    #===========================================================================
    # Tab 3 - Config Buttons
    #===========================================================================

    <#

        This section is working as expected and logs output to console and $ENV:Temp\winutil.log

        TODO: Error Handling as Try blocks and -erroraction stop causes runspace to lock up
    
    #>

    $Sync.GUIFeatures = {
        
        <#

            .DESCRIPTION
            This Scriptblock is meant to be ran from inside the GUI and will prevent the user from starting another install task. 

            Input data will look like below and link with the name of the check box. This will then look to the config/features.json file to find
            the install commands for the selected features.

            Featureshyperv,Featureslegacymedia

            .EXAMPLE

            Invoke-command $sync.GUIInstallPrograms -ArgumentList "Featureshyperv,Featureslegacymedia"

        #>
        
        param ($featuretoinstall)

        #Check if any check boxes have been checked and if a task is currently running 

            if ($sync.taskrunning -eq $true){
                [System.Windows.MessageBox]::Show($sync.taskmessage,$sync.tasktitle,"OK","Info")
                return
            }

            if($featuretoinstall -notlike "*Features*"){
                [System.Windows.MessageBox]::Show("Please check the features you wish to install",'Nothing to do',"OK","Info")
                return
            }

            $sync.taskrunning = $true

        #Invoke a runspace so that the GUI does not lock up  
            
            $params = @{
                ScriptBlock = $sync.ScriptFeatureInstall
                ArgumentList = ("$featuretoinstall")
            }
            
            Invoke-Runspace @params

    }

    $sync.ScriptFeatureInstall = {

        <#

            .DESCRIPTION
            This scriptblock will install the selected features from the config/features.json file. 

            TODO: Figure out error handling as any errors in this runspace will crash the powershell session.

            .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptFeatureInstall
                ArgumentList = "Featureshyperv,Featureslegacymedia"
                Verbose = $true
            }
            VerbosePreference = "Continue"
            Invoke-Command @params

        #>

        param ($featuretoinstall)

        $featuretoinstall = $featuretoinstall -split ","

        function Write-Logs {
            param($Level, $Message, $LogPath)
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message, $LogPath)
        }
        
        Foreach ($feature in $featuretoinstall){

            $sync.feature.$feature | ForEach-Object {
                Try{
                    Write-Logs -Level INFO -Message "Installing Windows Feature $psitem" -LogPath $sync.logfile
                    Enable-WindowsOptionalFeature -Online -FeatureName "$psitem" -All -NoRestart
                    Write-output $psitem
                }Catch{Write-Logs -Level ERROR -Message "Failed to install $psitem" -LogPath $sync.logfile}

            }

        }

        Write-Logs -Level INFO -Message "Finished Installing features" -LogPath $sync.logfile

        if($sync["Form"]){
            $sync.taskrunning = $false
            [System.Windows.MessageBox]::Show("Features have been installed",'Installs are done!',"OK","Info")
        }
        
    }

    #===========================================================================
    # Tab 4 - Updates Buttons
    #===========================================================================
    
    $Sync.GUIUpdates = {
        
        <#

            .DESCRIPTION

            Current Options

            "Updatesdefault" 
            "Updatesdisable" 
            "Updatessecurity"

            .EXAMPLE

            Invoke-command $sync.GUIUpdates -ArgumentList "Updatesdefault"

        #>
        
        param ($updatestoconfigure)

        #Check if any check boxes have been checked and if a task is currently running 

            if ($sync.taskrunning -eq $true){
                [System.Windows.MessageBox]::Show($sync.taskmessage,$sync.tasktitle,"OK","Info")
                return
            }

            $sync.taskrunning = $true

        #Invoke a runspace so that the GUI does not lock up  
            
            $params = @{
                ScriptBlock = $sync.ScriptUpdates
                ArgumentList = ("$updatestoconfigure")
            }
            
            Invoke-Runspace @params

    }

    $sync.ScriptUpdates = {

        <#

            .DESCRIPTION
            This scriptblock will install the selected features from the config/features.json file. 

            TODO: Figure out error handling as any errors in this runspace will crash the powershell session.

            .EXAMPLE

            $params = @{
                ScriptBlock = $sync.ScriptFeatureInstall
                ArgumentList = "Featureshyperv,Featureslegacymedia"
                Verbose = $true
            }
            VerbosePreference = "Continue"
            Invoke-Command @params

        #>

        param ($updatestoconfigure)

        function Write-Logs {
            param($Level, $Message, $LogPath)
            Invoke-command $sync.WriteLogs -ArgumentList ($Level,$Message, $LogPath)
        }
        if($updatestoconfigure -eq "Updatesdefault"){
            # Source: https://github.com/rgl/windows-vagrant/blob/master/disable-windows-updates.ps1 reversed! 
            Set-StrictMode -Version Latest
            $ProgressPreference = 'SilentlyContinue'
            $ErrorActionPreference = 'Stop'
            trap {
                Write-Logs -Level "ERROR" -LogPath $sync.logfile -Message $psitem
                Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Sleeping for 60m to give you time to look around the virtual machine before self-destruction..."
            }

            # disable automatic updates.
            # XXX this does not seem to work anymore.
            # see How to configure automatic updates by using Group Policy or registry settings
            #     at https://support.microsoft.com/en-us/help/328010
            function New-Directory($path) {
                $p, $components = $path -split '[\\/]'
                $components | ForEach-Object {
                    $p = "$p\$psitem"
                    if (!(Test-Path $p)) {
                        New-Item -ItemType Directory $p | Out-Null
                    }
                }
                $null
            }
            $auPath = 'HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU'
            New-Directory $auPath 
            # set NoAutoUpdate.
            # 0: Automatic Updates is enabled (default).
            # 1: Automatic Updates is disabled.
            New-ItemProperty `
                -Path $auPath `
                -Name NoAutoUpdate `
                -Value 0 `
                -PropertyType DWORD `
                -Force `
                | Out-Null
            # set AUOptions.
            # 1: Keep my computer up to date has been disabled in Automatic Updates.
            # 2: Notify of download and installation.
            # 3: Automatically download and notify of installation.
            # 4: Automatically download and scheduled installation.
            New-ItemProperty `
                -Path $auPath `
                -Name AUOptions `
                -Value 3 `
                -PropertyType DWORD `
                -Force `
                | Out-Null

            # disable Windows Update Delivery Optimization.
            # NB this applies to Windows 10.
            # 0: Disabled
            # 1: PCs on my local network
            # 3: PCs on my local network, and PCs on the Internet
            $deliveryOptimizationPath = 'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\DeliveryOptimization\Config'
            if (Test-Path $deliveryOptimizationPath) {
                New-ItemProperty `
                    -Path $deliveryOptimizationPath `
                    -Name DODownloadMode `
                    -Value 0 `
                    -PropertyType DWORD `
                    -Force `
                    | Out-Null
            }
            # Service tweaks for Windows Update

            $services = @(
                "BITS"
                "wuauserv"
            )

            foreach ($service in $services) {
                # -ErrorAction SilentlyContinue is so it doesn't write an error to stdout if a service doesn't exist

                Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Setting $service StartupType to Automatic"
                Get-Service -Name $service -ErrorAction SilentlyContinue | Set-Service -StartupType Automatic
            }
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Enabling driver offering through Windows Update..."
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\Device Metadata" -Name "PreventDeviceMetadataFromNetwork" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DontPromptForWindowsUpdate" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DontSearchWindowsUpdate" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DriverUpdateWizardWuSearchEnabled" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate" -Name "ExcludeWUDriversInQualityUpdate" -ErrorAction SilentlyContinue
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Enabling Windows Update automatic restart..."
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU" -Name "NoAutoRebootWithLoggedOnUsers" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU" -Name "AUPowerManagement" -ErrorAction SilentlyContinue
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Enabled driver offering through Windows Update"
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "BranchReadinessLevel" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "DeferFeatureUpdatesPeriodInDays" -ErrorAction SilentlyContinue
            Remove-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "DeferQualityUpdatesPeriodInDays " -ErrorAction SilentlyContinue

            ### Reset Windows Update Script - reregister dlls, services, and remove registry entires.
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "1. Stopping Windows Update Services..." 
            Stop-Service -Name BITS 
            Stop-Service -Name wuauserv 
            Stop-Service -Name appidsvc 
            Stop-Service -Name cryptsvc 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "2. Remove QMGR Data file..." 
            Remove-Item "$env:allusersprofile\Application Data\Microsoft\Network\Downloader\qmgr*.dat" -ErrorAction SilentlyContinue 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "3. Renaming the Software Distribution and CatRoot Folder..." 
            Rename-Item $env:systemroot\SoftwareDistribution SoftwareDistribution.bak -ErrorAction SilentlyContinue 
            Rename-Item $env:systemroot\System32\Catroot2 catroot2.bak -ErrorAction SilentlyContinue 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "4. Removing old Windows Update log..." 
            Remove-Item $env:systemroot\WindowsUpdate.log -ErrorAction SilentlyContinue 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "5. Resetting the Windows Update Services to defualt settings..." 
            "sc.exe sdset bits D:(A;;CCLCSWRPWPDTLOCRRC;;;SY)(A;;CCDCLCSWRPWPDTLOCRSDRCWDWO;;;BA)(A;;CCLCSWLOCRRC;;;AU)(A;;CCLCSWRPWPDTLOCRRC;;;PU)" 
            "sc.exe sdset wuauserv D:(A;;CCLCSWRPWPDTLOCRRC;;;SY)(A;;CCDCLCSWRPWPDTLOCRSDRCWDWO;;;BA)(A;;CCLCSWLOCRRC;;;AU)(A;;CCLCSWRPWPDTLOCRRC;;;PU)" 

            Set-Location $env:systemroot\system32 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "6. Registering some DLLs..." 
            regsvr32.exe /s atl.dll 
            regsvr32.exe /s urlmon.dll 
            regsvr32.exe /s mshtml.dll 
            regsvr32.exe /s shdocvw.dll 
            regsvr32.exe /s browseui.dll 
            regsvr32.exe /s jscript.dll 
            regsvr32.exe /s vbscript.dll 
            regsvr32.exe /s scrrun.dll 
            regsvr32.exe /s msxml.dll 
            regsvr32.exe /s msxml3.dll 
            regsvr32.exe /s msxml6.dll 
            regsvr32.exe /s actxprxy.dll 
            regsvr32.exe /s softpub.dll 
            regsvr32.exe /s wintrust.dll 
            regsvr32.exe /s dssenh.dll 
            regsvr32.exe /s rsaenh.dll 
            regsvr32.exe /s gpkcsp.dll 
            regsvr32.exe /s sccbase.dll 
            regsvr32.exe /s slbcsp.dll 
            regsvr32.exe /s cryptdlg.dll 
            regsvr32.exe /s oleaut32.dll 
            regsvr32.exe /s ole32.dll 
            regsvr32.exe /s shell32.dll 
            regsvr32.exe /s initpki.dll 
            regsvr32.exe /s wuapi.dll 
            regsvr32.exe /s wuaueng.dll 
            regsvr32.exe /s wuaueng1.dll 
            regsvr32.exe /s wucltui.dll 
            regsvr32.exe /s wups.dll 
            regsvr32.exe /s wups2.dll 
            regsvr32.exe /s wuweb.dll 
            regsvr32.exe /s qmgr.dll 
            regsvr32.exe /s qmgrprxy.dll 
            regsvr32.exe /s wucltux.dll 
            regsvr32.exe /s muweb.dll 
            regsvr32.exe /s wuwebv.dll 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "7) Removing WSUS client settings..." 
            #Fix to stop runspace from locking up if values not found
            start-process powershell.exe -Verb RunAs -ArgumentList "-c `"
                REG DELETE `"HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate`" /v AccountDomainSid /f
                REG DELETE `"HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate`" /v PingID /f
                REG DELETE `"HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate`" /v SusClientId /f`"
            "

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "8) Resetting the WinSock..." 
            netsh winsock reset 
            netsh winhttp reset proxy 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "9) Delete all BITS jobs..." 
            Get-BitsTransfer | Remove-BitsTransfer 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "10) Attempting to install the Windows Update Agent..." 
            if([System.Environment]::Is64BitOperatingSystem){ 
                wusa Windows8-RT-KB2937636-x64 /quiet
            } 
            else{ 
                wusa Windows8-RT-KB2937636-x86 /quiet
            } 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "11) Starting Windows Update Services..." 
            Start-Service -Name BITS 
            Start-Service -Name wuauserv 
            Start-Service -Name appidsvc 
            Start-Service -Name cryptsvc 

            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "12) Forcing discovery..." 
            wuauclt /resetauthorization /detectnow 
        }
        if($updatestoconfigure -eq "Updatesdisable"){
            # Source: https://github.com/rgl/windows-vagrant/blob/master/disable-windows-updates.ps1
            Set-StrictMode -Version Latest
            $ProgressPreference = 'SilentlyContinue'
            $ErrorActionPreference = 'Stop'
            trap {
                Write-Logs -Level "ERROR" -LogPath $sync.logfile -Message $psitem
                Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Sleeping for 60m to give you time to look around the virtual machine before self-destruction..."
            }

            # disable automatic updates.
            # XXX this does not seem to work anymore.
            # see How to configure automatic updates by using Group Policy or registry settings
            #     at https://support.microsoft.com/en-us/help/328010
            function New-Directory($path) {
                $p, $components = $path -split '[\\/]'
                $components | ForEach-Object {
                    $p = "$p\$_"
                    If (!(Test-Path $p)) {
                        New-Item -ItemType Directory $p | Out-Null
                    }
                }
                $null
            }
            $auPath = 'HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU'
            New-Directory $auPath 
            # set NoAutoUpdate.
            # 0: Automatic Updates is enabled (default).
            # 1: Automatic Updates is disabled.
            New-ItemProperty `
                -Path $auPath `
                -Name NoAutoUpdate `
                -Value 1 `
                -PropertyType DWORD `
                -Force `
            | Out-Null
            # set AUOptions.
            # 1: Keep my computer up to date has been disabled in Automatic Updates.
            # 2: Notify of download and installation.
            # 3: Automatically download and notify of installation.
            # 4: Automatically download and scheduled installation.
            New-ItemProperty `
                -Path $auPath `
                -Name AUOptions `
                -Value 1 `
                -PropertyType DWORD `
                -Force `
            | Out-Null

            # disable Windows Update Delivery Optimization.
            # NB this applies to Windows 10.
            # 0: Disabled
            # 1: PCs on my local network
            # 3: PCs on my local network, and PCs on the Internet
            $deliveryOptimizationPath = 'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\DeliveryOptimization\Config'
            If (Test-Path $deliveryOptimizationPath) {
                New-ItemProperty `
                    -Path $deliveryOptimizationPath `
                    -Name DODownloadMode `
                    -Value 0 `
                    -PropertyType DWORD `
                    -Force `
                | Out-Null
            }
            # Service tweaks for Windows Update

            $services = @(
                "BITS"
                "wuauserv"
            )

            foreach ($service in $services) {
                # -ErrorAction SilentlyContinue is so it doesn't write an error to stdout if a service doesn't exist

                Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Setting $service StartupType to Disabled"
                Get-Service -Name $service -ErrorAction SilentlyContinue | Set-Service -StartupType Disabled
            }
        }    
        if($updatestoconfigure -eq "Updatessecurity"){
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Disabling driver offering through Windows Update..."
            If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\Device Metadata")) {
                New-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\Device Metadata" -Force | Out-Null
            }
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\Device Metadata" -Name "PreventDeviceMetadataFromNetwork" -Type DWord -Value 1
            If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching")) {
                New-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Force | Out-Null
            }
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DontPromptForWindowsUpdate" -Type DWord -Value 1
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DontSearchWindowsUpdate" -Type DWord -Value 1
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\DriverSearching" -Name "DriverUpdateWizardWuSearchEnabled" -Type DWord -Value 0
            If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate")) {
                New-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate" | Out-Null
            }
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate" -Name "ExcludeWUDriversInQualityUpdate" -Type DWord -Value 1
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Disabling Windows Update automatic restart..."
            If (!(Test-Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU")) {
                New-Item -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU" -Force | Out-Null
            }
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU" -Name "NoAutoRebootWithLoggedOnUsers" -Type DWord -Value 1
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU" -Name "AUPowerManagement" -Type DWord -Value 0
            Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Disabled driver offering through Windows Update"
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "BranchReadinessLevel" -Type DWord -Value 20
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "DeferFeatureUpdatesPeriodInDays" -Type DWord -Value 365
            Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\WindowsUpdate\UX\Settings" -Name "DeferQualityUpdatesPeriodInDays " -Type DWord -Value 4
        }    

        Write-Logs -Level "INFO" -LogPath $sync.logfile -Message "Process complete. Please reboot your computer."

        if($sync["Form"]){
            $sync.taskrunning = $false
            [System.Windows.MessageBox]::Show("Updates have been configured",'Configuration is done!',"OK","Info")
        }
        
    }

#endregion Scripts

$runspace = [RunspaceFactory]::CreateRunspace()
$runspace.ApartmentState = "STA"
$runspace.ThreadOptions = "ReuseThread"
$runspace.Open()
$runspace.SessionStateProxy.SetVariable("sync", $sync)

#Get ComputerInfo in the background
Invoke-Runspace -ScriptBlock {$sync.ComputerInfo = Get-ComputerInfo} | Out-Null

#region form

    #WinForms dependancies 
    [Reflection.Assembly]::LoadWithPartialName("System.Windows.Forms") | Out-Null
    Add-Type -AssemblyName System.Windows.Forms
    Add-Type -AssemblyName PresentationFramework
    [System.Windows.Forms.Application]::EnableVisualStyles()

    #List of config files to import
    $configs = (
        "applications", 
        "tweaks",
        "preset", 
        "feature"
    )

    #Test for admin credentials
    if (!([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
        $IsAdmin = $false
    }

    #To use local files run $env:environment = "dev" before starting the ps1 file
    if($env:environment -eq "dev"){

        if($IsAdmin -eq $false){
            [System.Windows.MessageBox]::Show("This application needs to be run as Admin",'Administrative privileges required',"OK","Info")
            return
        }
        
        $confirm = [System.Windows.MessageBox]::Show('$ENV:Evnronment is set to dev. Do you wish to load the dev environment?','Dev Environment tag detected',"YesNo","Info")
    }

    if($confirm -eq "yes"){
        $inputXML = Get-Content "MainWindow.xaml"
        $configs | ForEach-Object {
            $sync["$PSItem"] = Get-Content .\config\$PSItem.json | ConvertFrom-Json
        }        
    }
    else{

        #Select the working branch
        if($env:branch){
            $branch = $env:branch
        }
        Else {$branch = "main"}

        if($IsAdmin -eq $false){
            Write-Output "This application needs to be run as an administrator. Attempting relaunch"
            Start-Process -Verb runas -FilePath powershell.exe -ArgumentList "iwr -useb https://christitus.com/win | iex"
            break
        }

        $inputXML = (new-object Net.WebClient).DownloadString("https://raw.githubusercontent.com/ChrisTitusTech/winutil/$branch/MainWindow.xaml")
        $configs | ForEach-Object {
            $sync["$psitem"] = Invoke-RestMethod "https://raw.githubusercontent.com/ChrisTitusTech/winutil/$branch/config/$psitem.json"
        }
    }

#endregion form    

write-host ""                                                                                                                             
write-host "    CCCCCCCCCCCCCTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTTT   "
write-host " CCC::::::::::::CT:::::::::::::::::::::TT:::::::::::::::::::::T   "
write-host "CC:::::::::::::::CT:::::::::::::::::::::TT:::::::::::::::::::::T  "
write-host "C:::::CCCCCCCC::::CT:::::TT:::::::TT:::::TT:::::TT:::::::TT:::::T "
write-host "C:::::C       CCCCCCTTTTTT  T:::::T  TTTTTTTTTTTT  T:::::T  TTTTTT"
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C                     T:::::T                T:::::T        "
write-host "C:::::C       CCCCCC        T:::::T                T:::::T        "
write-host "C:::::CCCCCCCC::::C      TT:::::::TT            TT:::::::TT       "
write-host "CC:::::::::::::::C       T:::::::::T            T:::::::::T       "
write-host "CCC::::::::::::C         T:::::::::T            T:::::::::T       "
write-host "  CCCCCCCCCCCCC          TTTTTTTTTTT            TTTTTTTTTTT       "
write-host ""
write-host "====Chris Titus Tech====="
write-host "=====Windows Toolbox====="

if($gui -eq $true){
    $inputXML = $inputXML -replace 'mc:Ignorable="d"','' -replace "x:N",'N' -replace '^<Win.*', '<Window'
    [xml]$XAML = $inputXML
    $reader=(New-Object System.Xml.XmlNodeReader $xaml) 
    
    try{$sync["Form"]=[Windows.Markup.XamlReader]::Load( $reader )}
    catch [System.Management.Automation.MethodInvocationException] {
        Write-Warning "We ran into a problem with the XAML code.  Check the syntax for this control..."
        write-host $error[0].Exception.Message -ForegroundColor Red
        if ($error[0].Exception.Message -like "*button*"){
            write-warning "Ensure your &lt;button in the `$inputXML does NOT have a Click=ButtonClick property.  PS can't handle this`n`n`n`n"}
    }
    catch{#if it broke some other way <img draggable="false" role="img" class="emoji" alt="ðŸ˜€" src="https://s0.wp.com/wp-content/mu-plugins/wpcom-smileys/twemoji/2/svg/1f600.svg">
        Write-Host "Unable to load Windows.Markup.XamlReader. Double-check syntax and ensure .net is installed."
    }

    # Store Form Objects In PowerShell
    $xaml.SelectNodes("//*[@Name]") | ForEach-Object {$sync["$("$($psitem.Name)")"] = $sync["Form"].FindName($psitem.Name)}

    #Gives every button the invoke-button function
    $sync.keys | ForEach-Object {
        if($sync.$psitem){
            if($($sync["$psitem"].GetType() | Select-Object -ExpandProperty Name) -eq "Button"){
                $sync["$psitem"].Add_Click({
                    [System.Object]$Sender = $args[0]
                    Invoke-Button $Sender.name
                })
            }
        }
    }

    $sync["Form"].ShowDialog() | out-null
}

<#
    How to run Arguments

    First step is to set the $env:args variable with the setups you wish to do. To do multiple items put a " " space between each command. 
    For commands that require input seperate the command with a semicolon ":" and provide the values to pass to that argument seperated by a comma ",". (IE: Install:git.git,windirstat.windirstat)

    Supported arguments:

        InstallUpgrade
        UndoTweaks
        PanelControl
        PanelNetwork
        PanelPower
        PanelSound
        PanelSystem
        PanelUser
        DefaultUpdates
        DisableUpdates
        EnableSecurity
        QuitAfter
        Install:value1,values2,...
            - Values should be the same values you would use for winget
        Tweaks:value1,values2,...
            - Values should be what you find inside the tweaks.json file

    Example usage: 
      
        $env:args = "Install:git.git,WinDirStat.WinDirStat "; iwr -useb https://christitus.com/win | iex

        $env:args = "Tweaks:EssTweaksLoc,EssTweaksServices"; iwr -useb https://christitus.com/win | iex

        $env:args = "DefaultUpdates"; iwr -useb https://christitus.com/win | iex

        $env:args = "Install:git.git,WinDirStat.WinDirStat Tweaks:EssTweaksLoc,EssTweaksServices DefaultUpdates"; iwr -useb https://christitus.com/win | iex

#>

If($env:args){
    Write-Verbose "Arguments Detected, Running Args"
    If($env:args -match '\bInstallUpgrade\b'){Invoke-command $sync.ScriptsInstallPrograms -ArgumentList "Upgrade"}
    If($env:args -match '\bUndoTweaks\b'){Invoke-command $sync.ScriptUndoTweaks}
    If($env:args -match '\bPanelControl\b'){cmd /c control}
    If($env:args -match '\bPanelNetwork\b'){cmd /c ncpa.cpl}
    If($env:args -match '\bPanelPower\b'){cmd /c powercfg.cpl}
    If($env:args -match '\bPanelSound\b'){cmd /c mmsys.cpl}
    If($env:args -match '\bPanelSystem\b'){cmd /c sysdm.cpl}
    If($env:args -match '\bPanelUser\b'){cmd /c "control userpasswords2"}
    If($env:args -match '\bDefaultUpdates\b'){Invoke-command $sync.ScriptUpdates -ArgumentList "Updatesdefault"}
    If($env:args -match '\bDisableUpdates\b'){Invoke-command $sync.ScriptUpdates -ArgumentList "Updatesdisable"}
    If($env:args -match '\bEnableSecurity\b'){Invoke-command $sync.ScriptUpdates -ArgumentList "Updatessecurity"}
    If($env:args -match '\bQuitAfter\b'){Break}
    If($env:args -match '\bInstall\b'){
        $ProgramstoInstall = (($env:args-split " " | Where-Object {$_ -like "install*"} ) -split ":")[1]
        Write-Verbose "Installing $ProgramstoInstall."
        Invoke-command $sync.ScriptsInstallPrograms -ArgumentList "$ProgramstoInstall"
    }
    If($env:args -match '\bTweaks\b'){
        $Tweakstorun = (($env:args-split " " | Where-Object {$_ -like "Tweaks*"} ) -split ":")[1]
        Write-Verbose "Running the following tweaks $Tweakstorun."
        Invoke-command $sync.ScriptTweaks -ArgumentList "$Tweakstorun"
    }
}

Write-Host "Thank you for using winutil!"