vision_watchdog.vbs

' vision_watchdog.vbs — Always-on background process manager for vision-tool.
' Copyright (C) 2026 Farhan Dhrubo
'
' Licensed under GPLv3 — see LICENSE.
'
' Monitors for ANY AI coding assistant process via WMI every 10 seconds.
' When ANY supported tool runs -> launches vision MCP server (hidden, no window).
' When ALL tools exit -> kills child process, cleans up PID file.
'
' Supported tools (13 monitored processes):
'   opencode.exe       — OpenCode CLI/Desktop
'   claude.exe         — Claude Code / Claude Desktop
'   cursor.exe         — Cursor AI editor
'   windsurf.exe       — Windsurf AI editor
'   aider.exe          — Aider AI pair programming
'   continue.exe       — Continue.dev
'   code.exe           — VS Code / VS Studio Code
'   vscode.exe         — VSCode variant (some forks)
'   codium.exe         — VSCodium (open-source VS Code fork)
'   studio.exe         — Various Studio AI editors
'   antigravity.exe    — Antigravity 1.x / 2.x (Google AI IDE)
'   claudecode.exe     — Claude Code CLI variants
'   ghcopilot.exe      — GitHub Copilot CLI
'
' Auto-start with Windows (Task Scheduler):
'   schtasks /create /tn "vision-tool-watchdog" /tr "wscript.exe //nologo \"%PATH%\vision_watchdog.vbs\"" /sc onstart /delay 0000:30 /ru %USERNAME% /f
'
' Usage:
'   wscript.exe //nologo vision_watchdog.vbs
'   wscript.exe //nologo vision_watchdog.vbs "python C:\path\to\vision_mcp_server.py --http 3789"
'   wscript.exe //nologo vision_watchdog.vbs "my_command" "my_pid_file.pid"
'
' For zero-flash (no wscript icon): compile vision_watchdog.cs into vision_watchdog.exe
'   csc.exe /target:winexe vision_watchdog.cs

Dim args, childCmd, pidFileName, shell, fso, wmi, pidFilePath

Set shell = CreateObject("WScript.Shell")
Set fso   = CreateObject("Scripting.FileSystemObject")
' WMI connection with retry — handles boot race condition
Dim wmiConnected, retryCount, maxRetries
wmiConnected = False
maxRetries = 5
retryCount = 0

Do While Not wmiConnected And retryCount < maxRetries
    On Error Resume Next
    Set wmi = GetObject("winmgmts:\\.\root\cimv2")
    If Err.Number = 0 Then
        wmiConnected = True
        On Error GoTo 0
    Else
        Err.Clear
        On Error GoTo 0
        retryCount = retryCount + 1
        If retryCount < maxRetries Then
            WScript.Sleep 5000
        End If
    End If
Loop

If Not wmiConnected Then
    WScript.Echo "Error: Cannot connect to WMI (root\cimv2). Exiting."
    WScript.Quit 1
End If

' -- AI tool processes to watch -----------------------------------------
' Add ANY process name here. WMI queries are case-insensitive.
Dim AI_TOOLS
AI_TOOLS = Array( _
    "opencode.exe", _
    "claude.exe", _
    "cursor.exe", _
    "windsurf.exe", _
    "aider.exe", _
    "continue.exe", _
    "code.exe", _
    "vscode.exe", _
    "codium.exe", _
    "studio.exe", _
    "antigravity.exe", _
    "claudecode.exe", _
    "ghcopilot.exe" _
)

' -- Parse arguments ----------------------------------------------------
Set args = WScript.Arguments

' Default: run vision_mcp_server.py (assumes next to this script)
scriptDir = fso.GetParentFolderName(WScript.ScriptFullName)
defaultCmd = "python """ & scriptDir & "\vision_mcp_server.py"" --http 3789"

childCmd    = defaultCmd
pidFileName = "vision_watchdog.pid"

If args.Count > 0 Then childCmd    = args(0)
If args.Count > 1 Then pidFileName = args(1)

pidFilePath = shell.ExpandEnvironmentStrings("%TEMP%") & "\" & pidFileName

' -- Helper: check if any AI tool process is running --------------------
Function IsAnyAiToolRunning()
    Dim proc, anyRunning
    anyRunning = False
    On Error Resume Next
    For Each tool In AI_TOOLS
        Set proc = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name='" & tool & "'")
        If Err.Number = 0 And proc.Count > 0 Then
            anyRunning = True
            Exit For
        End If
        Err.Clear
    Next
    On Error GoTo 0
    IsAnyAiToolRunning = anyRunning
End Function

' -- Main loop ----------------------------------------------------------
Do While True
    Dim anyRunning
    anyRunning = IsAnyAiToolRunning()

    If anyRunning Then
        ' Start child if not already running
        If Not fso.FileExists(pidFilePath) Then
            Dim pidFileOut
            ' Launch hidden (window style 0 = invisible)
            Dim procId
            procId = shell.Run(childCmd, 0, False)

            ' Write PID file so we can kill later
            Set pidFileOut = fso.CreateTextFile(pidFilePath, True)
            pidFileOut.WriteLine(procId)
            pidFileOut.Close
        End If
    Else
        ' Kill child if running
        If fso.FileExists(pidFilePath) Then
            Dim pidFileIn, pid
            Set pidFileIn = fso.OpenTextFile(pidFilePath, 1)
            pid = Trim(pidFileIn.ReadLine())
            pidFileIn.Close

            On Error Resume Next
            Dim proc
            Set proc = wmi.Get("Win32_Process.Handle='" & pid & "'")
            If Err.Number <> 0 Then
                Err.Clear
                Dim procs
                Set procs = wmi.ExecQuery("SELECT * FROM Win32_Process WHERE Name='python.exe' AND CommandLine LIKE '%vision_mcp_server%'")
                If Err.Number = 0 Then
                    For Each p In procs
                        p.Terminate()
                    Next
                End If
                Err.Clear
            Else
                proc.Terminate()
            End If
            On Error GoTo 0

            On Error Resume Next
            fso.DeleteFile pidFilePath, True
            On Error GoTo 0
        End If
    End If

    WScript.Sleep 10000
Loop