Worklet: Predictable Reboot Notifications for Windows

osascript -e 'display notification "hello world!"'

softwareupdate -l > /tmp/output.txt & proccess_id=$!

        wait $process_id

grep -w "Tool" /tmp/output.txt > /dev/null

if [[ $? -eq 0 ]]; then

#shutdown -h +5

message1="Automox will reboot your system in 2 hours. Please take time to save and close any open documents."
osascript -e 'display notification "'"$message1"'"'

        sleep 10s

message2="Automox will reboot your system in 1 hour. Please take time to save and close any open documents."
osascript -e 'display notification "'"$message2"'"'

        sleep 10s

message3="Automox will reboot your system in 30 minutes. Please take time to save and close any open documents."
osascript -e 'display alert "'"$message3"'"'

        sleep 10s

message4="Automox will reboot your system in 10 minutes. Please take time to save and close any open documents."
osascript -e 'display alert "'"$message4"'"'


else 
        exit 1
fi

Function Show-BalloonTip {
    Param(
        [Parameter(Mandatory = $true, Position = 0)]
        [ValidateNotNull()]
        [String]
        $BalloonTipText,
        [Parameter(Position = 1)]
        [String]
        $BalloonTipTitle = 'Tip',
        [Parameter(Position = 2)]
        [ValidateSet('Error', 'Info', 'None', 'Warning')]
        [String]
        $BalloonTipIcon = 'None',
        [Parameter(Position = 3)]
        [Int]
        $BalloonTipTime = 1000
    )
    end {
        Add-Type -AssemblyName System.Windows.Forms
        Add-Type -AssemblyName System.Drawing
        [Windows.Forms.ToolTipIcon]$BalloonTipIcon = $BalloonTipIcon
        $ContextMenu = New-Object System.Windows.Forms.ContextMenu
        $MenuItem = New-Object System.Windows.Forms.MenuItem

        $NotifyIcon = New-Object Windows.Forms.NotifyIcon -Property @{
            BalloonTipIcon = $BalloonTipIcon
            BalloonTipText = $BalloonTipText
            BalloonTipTitle = $BalloonTipTitle
            ContextMenu = $ContextMenu
            Icon = [Drawing.Icon]::ExtractAssociatedIcon((Get-Command powershell).Path)
            Text = -join $BalloonTipText[0..62]
            Visible = $false
        }
        $NotifyIcon.contextMenu.MenuItems.AddRange($MenuItem)
        $NotifyIcon.Visible = $True
        $MenuItem.Text = "Exit"
        $MenuItem.add_Click({
           $NotifyIcon.Visible = $false
           $NotifyIcon.ShowInTaskbar = $false
        })
        $NotifyIcon.ShowBalloonTip($BalloonTipTime)     
        switch ($Host.Runspace.ApartmentState) {
            STA {
                $null = Register-ObjectEvent -InputObject $NotifyIcon -EventName BalloonTipClosed -Action {                     
                    $Sender.Dispose()
                    Unregister-Event $EventSubscriber.SourceIdentifier
                    Remove-Job $EventSubscriber.Action
                }
            }
            default {
                continue
            }
        }
    }
}

$sysInfo = New-Object -ComObject "Microsoft.Update.SystemInfo"

### Variables ###

#$wait30Minutes = 1800
#$wait15Minutes = 900
#$rebootTimer = 300

$wait30Minutes = 18
$wait15Minutes = 9
$rebootTimer = 3

$balloonTipIcon = "None"
$balloonTipTitle = "Your IT Department"

if($sysInfo)
#if($sysInfo.RebootRequired)
    {
    Show-BalloonTip -BalloonTipText "We have installed security updates. Your device will automatically reboot within 90 minutes. Please save your work." -BalloonTipTitle $balloonTipTitle -BalloonTipIcon $balloonTipIcon -BalloonTipTime 7500
    Start-Sleep -Seconds $wait30Minutes

    Show-BalloonTip -BalloonTipText "We have installed security updates. Your device will automatically reboot within 60 minutes. Please save your work." -BalloonTipTitle $balloonTipTitle -BalloonTipIcon $balloonTipIcon -BalloonTipTime 7500    
    Start-Sleep -Seconds $wait30Minutes

    Show-BalloonTip -BalloonTipText "We have installed security updates. Your device will automatically reboot within 30 minutes. Please save your work." -BalloonTipTitle $balloonTipTitle -BalloonTipIcon $balloonTipIcon -BalloonTipTime 7500
    Start-Sleep -Seconds $wait15Minutes

    Show-BalloonTip -BalloonTipText "We have installed security updates. Your device will automatically reboot within 15 minutes. Please save your work." -BalloonTipTitle $balloonTipTitle -BalloonTipIcon $balloonTipIcon -BalloonTipTime 15000
    Start-Sleep -Seconds 6

    ### Initate Reboot ###
    #shutdown /r /t $rebootTimer /d p:4:1
    Write-Output "System Rebooted!"
    }

Add-Type @'
using System;
using System.Diagnostics;
using System.Runtime.InteropServices;

namespace PInvoke.Win32 {

    public static class UserInput {

        [DllImport("user32.dll", SetLastError=false)]
        private static extern bool GetLastInputInfo(ref LASTINPUTINFO plii);

        [StructLayout(LayoutKind.Sequential)]
        private struct LASTINPUTINFO {
            public uint cbSize;
            public int dwTime;
        }

        public static DateTime LastInput {
            get {
                DateTime bootTime = DateTime.UtcNow.AddMilliseconds(-Environment.TickCount);
                DateTime lastInput = bootTime.AddMilliseconds(LastInputTicks);
                return lastInput;
            }
        }

        public static TimeSpan IdleTime {
            get {
                return DateTime.UtcNow.Subtract(LastInput);
            }
        }

        public static int LastInputTicks {
            get {
                LASTINPUTINFO lii = new LASTINPUTINFO();
                lii.cbSize = (uint)Marshal.SizeOf(typeof(LASTINPUTINFO));
                GetLastInputInfo(ref lii);
                return lii.dwTime;
            }
        }
    }
}
'@

Function IsAutomaticReboot
{
    <# 
    
    Two conditions must be met for automatic reboot
    1. The local time must be between $rebootStartTime and $rebootEndTime
    2. There has not been any user input for at least $idleTime minutes

    #>

    $boolTimeConditionMet = $False
    $minTime = Get-Date $rebootStartTime
    $maxTime = Get-Date $rebootEndTime
    $now = Get-Date
    if ($minTime.TimeOfDay -le $now.TimeOfDay -and $maxTime.TimeOfDay -ge $now.TimeOfDay) 
    {
        $boolTimeConditionMet = $True
    }

    $boolIdleConditionMet = $False
    $idleTime = [PInvoke.Win32.UserInput]::IdleTime
    if ($idleTime.Minutes -ge $minIdleTime)
    {
        $boolIdleConditionMet = $True
    }

    return ($boolIdleConditionMet -and $boolTimeConditionMet)
}

Function RebootComputer
{
    Restart-Computer -Force
}

#---------------------------------------------------------------------------------------
# These parameters control the time of day window within which automatic reboot will be initiated
[string]$rebootStartTime = "1am"
[string]$rebootEndTime = "5am"

# If there is no user input detected for $minIdleTime minutes within $rebootStarTime and $rebootEndTime, automatic reboot
# will be initiated
[int]$minIdleTime = 30

# Check if a reboot is pending
$sysInfo = New-Object -ComObject "Microsoft.Update.SystemInfo"
if($sysInfo.RebootRequired)
{
  if (IsAutomaticReboot)
  {
    RebootComputer
  }
}

$action = New-ScheduledTaskAction -Execute wscript.exe -Argument "//B C:\ProgramData\amagent\rebootNotification\Hidden.vbs C:\ProgramData\amagent\rebootNotification\RunToastHidden.cmd" 
$trigger = New-ScheduledTaskTrigger -Daily -At 9am
$principal = New-ScheduledTaskPrincipal -UserID "NT AUTHORITY\SYSTEM" -LogonType ServiceAccount -RunLevel Highest
$settings = New-ScheduledTaskSettingsSet -MultipleInstances Parallel
$task = Register-ScheduledTask -TaskName "Reboot No Reminder" -Trigger $trigger -Action $action -User "NT AUTHORITY\SYSTEM"
$task.Triggers.Repetition.Interval = "PT1H"
$task.Triggers.Repetition.Duration = "P1D"
$task.Settings.StartWhenAvailable = $True
$task | Set-ScheduledTask -ErrorAction SilentlyContinue

function Test-WindowsPushNotificationsEnabled() {
    $ToastEnabledKey = (Get-ItemProperty "HKCU:\Software\Microsoft\Windows\CurrentVersion\PushNotifications" -Name ToastEnabled -ErrorAction Ignore).ToastEnabled
    if ($ToastEnabledKey -eq "1") {
        Write-Host "Toast notifications for the logged on user are enabled in Windows"
        $true
    }
    elseif ($ToastEnabledKey -eq "0") {
        Write-Host "Toast notifications for the logged on user are not enabled in Windows. The script will try to enable toast notifications for the logged on user"
        $false
    }
}

# Create Enable-WindowsPushNotifications
# This is used to re-enable toast notifications if the user disabled them generally in Windows
function Enable-WindowsPushNotifications() {
    $ToastEnabledKeyPath = "HKCU:\SOFTWARE\Microsoft\Windows\CurrentVersion\PushNotifications"
    Write-Host "Trying to enable toast notifications for the logged on user"
    try {
        Set-ItemProperty -Path $ToastEnabledKeyPath -Name ToastEnabled -Value 1 -Force
        Get-Service -Name WpnUserService** | Restart-Service -Force
        Write-Host "Successfully enabled toast notifications for the logged on user"
    }
    catch {
        Write-Host "Failed to enable toast notifications for the logged on user. Toast notifications will probably not be displayed"
    }
}