Mike Swanson
8b6f0bcc96
SolverBot: - Inject active project path into agent system prompts so agents know which directory to scope file operations to GuruRMM: - Bump agent version to 0.6.0 - Add serde aliases for PowerShell/ClaudeTask command types - Add typed CommandType enum on server for proper serialization - Support claude_task command type in send_command API Dataforth: - Fix SCP space-escaping in Sync-FromNAS.ps1 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
"""Create scheduled task on AD2 to swap agent binary and restart service."""
|
|
import requests, time
|
|
|
|
# Auth
|
|
token_r = requests.post('http://172.16.3.30:3001/api/auth/login', json={
|
|
'email': 'claude-api@azcomputerguru.com',
|
|
'password': 'ClaudeAPI2026!@#'
|
|
})
|
|
token = token_r.json()['token']
|
|
headers = {'Authorization': 'Bearer ' + token, 'Content-Type': 'application/json'}
|
|
agent_id = 'd28a1c90-47d7-448f-a287-197bc8892234'
|
|
|
|
# Create a scheduled task to swap the binary and restart
|
|
# Uses schtasks for simplicity - runs a PowerShell script that:
|
|
# 1. Stops the agent service
|
|
# 2. Backs up old binary
|
|
# 3. Copies new binary in place
|
|
# 4. Starts the service
|
|
# 5. Writes result to file
|
|
cmd = (
|
|
"$script = @'\n"
|
|
"Stop-Service GuruRMMAgent -Force\n"
|
|
"Start-Sleep -Seconds 3\n"
|
|
"Copy-Item \"C:\\Program Files\\GuruRMM\\gururmm-agent.exe\" \"C:\\Program Files\\GuruRMM\\gururmm-agent.exe.bak\" -Force\n"
|
|
"Copy-Item \"C:\\Temp\\gururmm-agent-new.exe\" \"C:\\Program Files\\GuruRMM\\gururmm-agent.exe\" -Force\n"
|
|
"Start-Sleep -Seconds 1\n"
|
|
"Start-Service GuruRMMAgent\n"
|
|
"\"Agent updated at $(Get-Date)\" | Out-File C:\\Temp\\update_result.txt\n"
|
|
"'@\n"
|
|
"$script | Out-File C:/Temp/update_agent.ps1 -Encoding UTF8\n"
|
|
"Write-Output ('Script written: ' + (Get-Item C:/Temp/update_agent.ps1).Length.ToString() + ' bytes')"
|
|
)
|
|
|
|
print("Step 1: Writing update script to AD2...")
|
|
r = requests.post(
|
|
'http://172.16.3.30:3001/api/agents/' + agent_id + '/command',
|
|
headers=headers,
|
|
json={'command_type': 'powershell', 'command': cmd, 'timeout_seconds': 30}
|
|
)
|
|
print('Send:', r.json())
|
|
cmd_id = r.json()['command_id']
|
|
time.sleep(10)
|
|
r2 = requests.get('http://172.16.3.30:3001/api/commands/' + cmd_id, headers=headers)
|
|
d = r2.json()
|
|
print('Status:', d['status'])
|
|
print('stdout:', d.get('stdout', ''))
|
|
if d.get('stderr'):
|
|
print('stderr:', (d.get('stderr', '') or '')[:300])
|
|
|
|
if d['status'] != 'completed':
|
|
print("FAILED to write script")
|
|
exit(1)
|
|
|
|
# Step 2: Create and run scheduled task
|
|
print("\nStep 2: Creating scheduled task to run update...")
|
|
cmd2 = (
|
|
"schtasks /create /tn AgentBinaryUpdate /tr "
|
|
"\"powershell.exe -ExecutionPolicy Bypass -File C:\\Temp\\update_agent.ps1\" "
|
|
"/sc ONCE /st 00:00 /sd 01/01/2030 /ru SYSTEM /rl HIGHEST /f; "
|
|
"schtasks /run /tn AgentBinaryUpdate; "
|
|
"Write-Output 'Update task started - agent will restart momentarily'"
|
|
)
|
|
|
|
r = requests.post(
|
|
'http://172.16.3.30:3001/api/agents/' + agent_id + '/command',
|
|
headers=headers,
|
|
json={'command_type': 'powershell', 'command': cmd2, 'timeout_seconds': 30}
|
|
)
|
|
print('Send:', r.json())
|
|
cmd_id = r.json()['command_id']
|
|
time.sleep(10)
|
|
r2 = requests.get('http://172.16.3.30:3001/api/commands/' + cmd_id, headers=headers)
|
|
d = r2.json()
|
|
print('Status:', d['status'])
|
|
print('stdout:', d.get('stdout', ''))
|
|
if d.get('stderr'):
|
|
print('stderr:', (d.get('stderr', '') or '')[:300])
|
|
|
|
print("\nAgent will stop, binary will be swapped, then service restarts.")
|
|
print("Wait ~30 seconds then check if agent reconnects.")
|