#Requires -Version 5.1
<#
.SYNOPSIS
One-command update and restart for MCP Memory Service on Windows.
.DESCRIPTION
Streamlined workflow to update MCP Memory Service:
1. Pull latest changes from git
2. Install updated dependencies (editable mode)
3. Restart HTTP dashboard server (via scheduled task or direct)
4. Verify version and health
Target time: <2 minutes (typical: 60-90 seconds)
.PARAMETER NoRestart
Update code but don't restart the server
.PARAMETER Force
Force update even with uncommitted changes (auto-stash)
.EXAMPLE
.\update_and_restart.ps1
Standard update and restart with prompts
.EXAMPLE
.\update_and_restart.ps1 -Force
Force update and auto-stash uncommitted changes
.EXAMPLE
.\update_and_restart.ps1 -NoRestart
Update code only, don't restart server
.NOTES
File Name : update_and_restart.ps1
Author : Heinrich Krupp
Prerequisite : Git, Python/uv, pip
Copyright 2026 - Licensed under Apache License 2.0
#>
param(
[switch]$NoRestart,
[switch]$Force
)
$ErrorActionPreference = "Stop"
$StartTime = Get-Date
# Configuration
$ProjectRoot = Split-Path -Parent (Split-Path -Parent (Split-Path -Parent $PSScriptRoot))
$ManageServiceScript = Join-Path $PSScriptRoot "manage_service.ps1"
$HealthUrl = "http://127.0.0.1:8000/api/health"
# Color helpers
function Write-InfoLog { param($Message) Write-Host "ℹ $Message" -ForegroundColor Cyan }
function Write-SuccessLog { param($Message) Write-Host "✓ $Message" -ForegroundColor Green }
function Write-WarningLog { param($Message) Write-Host "⚠ $Message" -ForegroundColor Yellow }
function Write-ErrorLog { param($Message) Write-Host "✗ $Message" -ForegroundColor Red }
function Write-StepLog { param($Message) Write-Host "`n▶ $Message" -ForegroundColor Blue }
function Get-ElapsedSeconds {
return [int]((Get-Date) - $StartTime).TotalSeconds
}
function Get-CurrentVersion {
$VersionFile = Join-Path $ProjectRoot "src\mcp_memory_service\_version.py"
if (Test-Path $VersionFile) {
$Content = Get-Content $VersionFile -Raw
if ($Content -match '__version__\s*=\s*["\047]([^"\047]+)["\047]') {
return $Matches[1]
}
}
return "unknown"
}
function Get-ServerVersion {
try {
$Response = Invoke-RestMethod -Uri $HealthUrl -TimeoutSec 3 -ErrorAction SilentlyContinue
return $Response.version
} catch {
return "unknown"
}
}
# Banner
Write-Host ""
Write-Host "╔════════════════════════════════════════════╗" -ForegroundColor Cyan
Write-Host "║ MCP Memory Service - Update & Restart ║" -ForegroundColor Cyan
Write-Host "╚════════════════════════════════════════════╝" -ForegroundColor Cyan
Write-Host ""
Set-Location $ProjectRoot
# Step 1: Check for uncommitted changes
Write-StepLog "Checking repository status..."
$GitStatus = git status --porcelain 2>&1
if ($GitStatus -and -not $Force) {
Write-WarningLog "You have uncommitted changes:"
git status --short | Select-Object -First 10
Write-Host ""
$Response = Read-Host "Stash changes and continue? [y/N]"
if ($Response -match '^[Yy]$') {
Write-InfoLog "Stashing local changes..."
$StashMessage = "Auto-stash before update $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')"
git stash push -m $StashMessage
Write-SuccessLog "Changes stashed"
} else {
Write-ErrorLog "Update cancelled. Use -Force to override."
exit 1
}
}
# Step 2: Get current version
Write-StepLog "Recording current version..."
$CurrentVersion = Get-CurrentVersion
Write-InfoLog "Current version: $CurrentVersion"
# Step 3: Pull latest changes
Write-StepLog "Pulling latest changes from git..."
$BeforeCommit = git rev-parse HEAD
git pull --rebase 2>&1 | Out-Null
$AfterCommit = git rev-parse HEAD
if ($BeforeCommit -eq $AfterCommit) {
Write-InfoLog "Already up-to-date (no new commits)"
} else {
$CommitCount = (git rev-list --count "$BeforeCommit..$AfterCommit").Trim()
Write-SuccessLog "Pulled $CommitCount new commit(s)"
Write-Host ""
git log --oneline --graph "$BeforeCommit..$AfterCommit" | Select-Object -First 10
}
# Step 4: Get new version
$NewVersion = Get-CurrentVersion
if ($CurrentVersion -ne $NewVersion) {
Write-InfoLog "Version change: $CurrentVersion → $NewVersion"
}
# Step 5: Install dependencies (editable mode)
Write-StepLog "Installing dependencies (editable mode)..."
# Check for uv
$UseUv = Get-Command uv -ErrorAction SilentlyContinue
if ($UseUv) {
Write-InfoLog "Using uv for faster installation..."
& uv pip install -e . --quiet 2>&1 | Out-Null
} else {
Write-InfoLog "Using pip for installation..."
& pip install -e . --quiet 2>&1 | Out-Null
}
Write-SuccessLog "Dependencies installed"
# Verify installation
try {
$PipShow = pip show mcp-memory-service 2>&1 | Out-String
if ($PipShow -match 'Version:\s+(.+)') {
$InstalledVersion = $Matches[1].Trim()
} else {
$InstalledVersion = "unknown"
}
if ($InstalledVersion -ne $NewVersion) {
Write-WarningLog "Installation version mismatch! Expected: $NewVersion, Got: $InstalledVersion"
Write-WarningLog "Retrying installation..."
& pip install -e . --force-reinstall --quiet 2>&1 | Out-Null
# Re-check
$PipShow = pip show mcp-memory-service 2>&1 | Out-String
if ($PipShow -match 'Version:\s+(.+)') {
$InstalledVersion = $Matches[1].Trim()
}
}
Write-InfoLog "Installed version: $InstalledVersion"
} catch {
Write-WarningLog "Could not verify installation version"
}
# Step 6: Restart server (if requested)
if ($NoRestart) {
Write-WarningLog "Skipping server restart (-NoRestart flag)"
} else {
Write-StepLog "Restarting HTTP dashboard server..."
if (Test-Path $ManageServiceScript) {
# Use manage_service.ps1 for smart restart
& $ManageServiceScript restart
} else {
Write-WarningLog "manage_service.ps1 not found, using manual restart..."
# Fallback: manual restart
# Kill any existing server processes
Get-Process python -ErrorAction SilentlyContinue |
Where-Object { $_.CommandLine -like "*run_http_server*" } |
Stop-Process -Force -ErrorAction SilentlyContinue
Start-Sleep -Seconds 2
# Start server in background
$ServerScript = Join-Path $ProjectRoot "scripts\server\run_http_server.py"
$LogFile = Join-Path $env:TEMP "mcp-memory-update.log"
Start-Process python -ArgumentList $ServerScript -WindowStyle Hidden -RedirectStandardOutput $LogFile -RedirectStandardError $LogFile
Start-Sleep -Seconds 8
}
# Step 7: Health check
Write-StepLog "Verifying server health..."
$MaxWait = 15
$WaitCount = 0
$Healthy = $false
while ($WaitCount -lt $MaxWait) {
try {
$HealthResponse = Invoke-RestMethod -Uri $HealthUrl -TimeoutSec 2 -ErrorAction SilentlyContinue
$ServerVersion = $HealthResponse.version
if ($ServerVersion -eq $NewVersion) {
Write-SuccessLog "Server healthy and running version $ServerVersion"
$Healthy = $true
break
} else {
Write-WarningLog "Server running old version: $ServerVersion (expected: $NewVersion)"
Write-InfoLog "Waiting for server to reload... (${WaitCount}s)"
}
} catch {
# Server not ready yet
}
Start-Sleep -Seconds 1
$WaitCount++
}
if (-not $Healthy) {
Write-ErrorLog "Server health check timeout after ${MaxWait}s"
Write-InfoLog "Check logs in: $env:LOCALAPPDATA\mcp-memory\logs\"
exit 1
}
}
# Step 8: Summary
$TotalTime = Get-ElapsedSeconds
Write-Host ""
Write-Host "╔════════════════════════════════════════════╗" -ForegroundColor Green
Write-Host "║ Update Complete! 🎉 ║" -ForegroundColor Green
Write-Host "╚════════════════════════════════════════════╝" -ForegroundColor Green
Write-Host ""
Write-SuccessLog "Version: $CurrentVersion → $NewVersion"
Write-SuccessLog "Total time: ${TotalTime}s"
Write-Host ""
Write-InfoLog "Dashboard: http://localhost:8000"
Write-InfoLog "API Docs: http://localhost:8000/api/docs"
Write-Host ""
if ($CurrentVersion -ne $NewVersion) {
Write-InfoLog "New version deployed. Check CHANGELOG.md for details:"
Write-Host ""
$ChangelogPath = Join-Path $ProjectRoot "CHANGELOG.md"
if (Test-Path $ChangelogPath) {
$ChangelogContent = Get-Content $ChangelogPath -Raw
if ($ChangelogContent -match "(?s)## \[$NewVersion\].*?(?=## \[|\z)") {
$Matches[0] -split "`n" | Select-Object -First 20 | ForEach-Object { Write-Host $_ }
} else {
Write-WarningLog "CHANGELOG not updated for $NewVersion"
}
}
}
exit 0
