240 lines
10 KiB
PowerShell
240 lines
10 KiB
PowerShell
#Requires -Version 5.1
|
|
Set-StrictMode -Version 3.0
|
|
|
|
$scriptPath = $MyInvocation.MyCommand.Path
|
|
$scriptDir = Split-Path -Path $scriptPath -Parent
|
|
$logDir = Join-Path -Path $scriptDir -ChildPath ".logs"
|
|
$logName = "foam_sync.log"
|
|
$logFilePath = Join-Path -Path $logDir -ChildPath $logName
|
|
$scriptName = (Get-Item $scriptPath).Name
|
|
|
|
# --- Task Setup ---
|
|
$taskName = 'FoamGitSync'
|
|
$taskDescription = "Periodically synchronizes the Git repository at $scriptDir using $scriptName."
|
|
$fullCommand = "c:\windows\system32\conhost.exe --headless powershell.exe -WindowStyle Hidden -NoProfile -NonInteractive -ExecutionPolicy Bypass -File $scriptPath"
|
|
$executeCommand = $fullCommand.Split(" ")[0]
|
|
$frequencyMinutes = 2
|
|
$timeLimitSeconds = 30
|
|
|
|
if (-not (Test-Path -Path $logDir)) {
|
|
New-Item -ItemType Directory -Path $logDir -Force | Out-Null
|
|
}
|
|
Start-Transcript -Path $logFilePath -Append -IncludeInvocationHeader -Force
|
|
|
|
try {
|
|
Write-Host "Script: $scriptName at $scriptPath"
|
|
Write-Host "Repository directory: $scriptDir"
|
|
Write-Host "Sync frequency: Every $frequencyMinutes minutes"
|
|
Write-Host "Log file: $logFilePath"
|
|
|
|
# Run as the user who executes this script.
|
|
$taskPrincipal = New-ScheduledTaskPrincipal -UserId $env:USERNAME -LogonType Interactive
|
|
$trigger = New-ScheduledTaskTrigger -Once -At (Get-Date) -RepetitionInterval (New-TimeSpan -Minutes $frequencyMinutes)
|
|
$actionArgumentForRegistration = "-NoProfile -NonInteractive -WindowStyle Hidden -ExecutionPolicy Bypass -File $scriptPath"
|
|
$action = New-ScheduledTaskAction -Execute "powershell.exe" -Argument $actionArgumentForRegistration
|
|
|
|
# Task settings
|
|
$settings = New-ScheduledTaskSettingsSet -ExecutionTimeLimit $(New-TimeSpan -Seconds $timeLimitSeconds)
|
|
|
|
# Check and configure the scheduled task
|
|
try {
|
|
$existingTask = Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
|
|
|
|
if ($existingTask) {
|
|
Write-Host "Scheduled task '$taskName' already exists. Checking configuration..."
|
|
$currentTrigger = $existingTask.Triggers[0]
|
|
$currentAction = $existingTask.Actions[0]
|
|
|
|
# Check Trigger
|
|
$triggerMatches = $false
|
|
if ($currentTrigger -is [Microsoft.Management.Infrastructure.CimInstance] `
|
|
-and $currentTrigger.RepetitionInterval.TotalMinutes -eq $trigger.RepetitionInterval.TotalMinutes) {
|
|
$triggerMatches = $true
|
|
}
|
|
|
|
# Check Action
|
|
$actionMatches = $false
|
|
if ($currentAction -is [Microsoft.Management.Infrastructure.CimInstance] `
|
|
-and $currentAction.Execute -eq "powershell.exe" `
|
|
-and $currentAction.Argument -eq $expectedRetrievedActionArgument) {
|
|
$actionMatches = $true
|
|
}
|
|
|
|
# Check Principal (example, can be expanded)
|
|
$principalMatches = ($existingTask.Principal.UserId -eq $taskPrincipal.UserId)
|
|
|
|
# Check Settings (specifically ExecutionTimeLimit for this change)
|
|
$settingsMatch = $false
|
|
if ($existingTask.Settings.ExecutionTimeLimit -eq $settings.ExecutionTimeLimit) {
|
|
$settingsMatch = $true
|
|
}
|
|
else {
|
|
Write-Warning "settings do not match: want $($settings.ExecutionTimeLimit), got $($existingTask.Settings.ExecutionTimeLimit)"
|
|
}
|
|
|
|
if ($triggerMatches -and $actionMatches -and $principalMatches -and $settingsMatch) {
|
|
Write-Host "Scheduled task '$taskName' is already correctly configured."
|
|
}
|
|
else {
|
|
Write-Host "Scheduled task '$taskName' configuration differs."
|
|
Write-Host "`$triggerMatches: $triggerMatches, `$actionMatches: $actionMatches, `$principalMatches: $principalMatches, `$settingsMatch: $settingsMatch"
|
|
Write-Host "Attempting to update in-place..."
|
|
try {
|
|
Set-ScheduledTask -TaskName $taskName -Action $action -Trigger $trigger -Settings $settings -Principal $taskPrincipal -ErrorAction Stop
|
|
Write-Host "Scheduled task '$taskName' updated successfully."
|
|
}
|
|
catch {
|
|
Write-Warning "Failed to update scheduled task '$taskName' in-place. Error: $($_.Exception.Message)"
|
|
Write-Warning "The task remains in its previous state. Manual intervention may be required or re-run with Administrator privileges."
|
|
# We intentionally DO NOT unregister here to avoid the scenario you described.
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
Write-Host "Creating scheduled task '$taskName'..."
|
|
try {
|
|
Register-ScheduledTask -TaskName $taskName -Description $taskDescription -Principal $taskPrincipal -Trigger $trigger -Action $action -Settings $settings -ErrorAction Stop
|
|
Write-Host "Scheduled task '$taskName' created successfully."
|
|
}
|
|
catch {
|
|
Write-Warning "Failed to create scheduled task '$taskName'. Error: $($_.Exception.Message)"
|
|
Write-Warning "You may need to run this script as Administrator."
|
|
}
|
|
}
|
|
}
|
|
catch {
|
|
# This outer catch is for unexpected errors, e.g., if Get-ScheduledTask had -ErrorAction Stop
|
|
Write-Warning "An unexpected error occurred during scheduled task setup for '$taskName'. Error: $($_.Exception.Message)"
|
|
}
|
|
|
|
# --- Git Operations ---
|
|
Write-Host "Navigating to repository: $scriptDir"
|
|
try {
|
|
Set-Location -Path $scriptDir -ErrorAction Stop
|
|
}
|
|
catch {
|
|
Write-Error "Unable to find repository at $scriptDir. Exiting script."
|
|
exit 1
|
|
}
|
|
|
|
Write-Host "Staging all changes with 'git add .'"
|
|
git add .
|
|
if ($LASTEXITCODE -ne 0) {
|
|
Write-Warning "'git add .' command failed with exit code $LASTEXITCODE."
|
|
}
|
|
|
|
Write-Host "Checking for staged changes with 'git diff --staged --quiet'..."
|
|
git diff --staged --quiet
|
|
$changesStaged = ($LASTEXITCODE -ne 0)
|
|
|
|
if ($changesStaged) {
|
|
Write-Host "Staged changes detected. Creating commit..."
|
|
$commitMessage = "$scriptName ($($env:COMPUTERNAME)) $(Get-Date -Format 'yyyy-MM-ddTHH:mm:ssZ')"
|
|
Write-Host "Commit message: $commitMessage"
|
|
git commit -m $commitMessage
|
|
if ($LASTEXITCODE -ne 0) {
|
|
Write-Warning "'git commit' command failed with exit code $LASTEXITCODE."
|
|
}
|
|
else {
|
|
Write-Host "Commit created successfully."
|
|
}
|
|
}
|
|
else {
|
|
Write-Host "No relevant changes detected to commit."
|
|
Write-Host "Resetting staging area with 'git reset HEAD --quiet'."
|
|
git reset HEAD --quiet
|
|
}
|
|
|
|
Write-Host "Updating remote 'origin' with 'git remote update origin --prune'..."
|
|
git remote update origin --prune
|
|
if ($LASTEXITCODE -ne 0) {
|
|
Write-Error "Unable to update remote 'origin'. Exiting script."
|
|
exit 1
|
|
}
|
|
|
|
$localCommit = (git rev-parse HEAD 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $localCommit) { Write-Error "Failed to get local HEAD commit. Exiting script."; exit 1 }
|
|
|
|
$remoteBranch = "origin/main"
|
|
$remoteCommit = (git rev-parse $remoteBranch 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $remoteCommit) { Write-Error "Failed to get remote '$remoteBranch' commit. Exiting script."; exit 1 }
|
|
|
|
Write-Host "Local HEAD commit: $localCommit"
|
|
Write-Host "Remote '$remoteBranch' commit: $remoteCommit"
|
|
|
|
if ($localCommit -eq $remoteCommit) {
|
|
Write-Host "Local and remote are already in sync."
|
|
exit 0
|
|
}
|
|
|
|
$sleepyTime = Get-Random -Minimum 1 -Maximum 15
|
|
Write-Host "Local and remote differ. Sleeping for $sleepyTime seconds..."
|
|
Start-Sleep -Seconds $sleepyTime
|
|
|
|
$localCommitAfterSleep = (git rev-parse HEAD 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $localCommitAfterSleep) { Write-Error "Failed to re-get local HEAD commit. Exiting script."; exit 1 }
|
|
$remoteCommitAfterSleep = (git rev-parse $remoteBranch 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $remoteCommitAfterSleep) { Write-Error "Failed to re-get remote '$remoteBranch' commit. Exiting script."; exit 1 }
|
|
|
|
if ($localCommitAfterSleep -eq $remoteCommitAfterSleep) {
|
|
Write-Host "Local and remote became synchronized during sleep."
|
|
exit 0
|
|
}
|
|
$localCommit = $localCommitAfterSleep
|
|
$remoteCommit = $remoteCommitAfterSleep
|
|
|
|
Write-Host "Proceeding with sync logic..."
|
|
|
|
git merge-base --is-ancestor $localCommit $remoteCommit
|
|
$localIsAncestorOfRemote = ($LASTEXITCODE -eq 0)
|
|
|
|
git merge-base --is-ancestor $remoteCommit $localCommit
|
|
$remoteIsAncestorOfLocal = ($LASTEXITCODE -eq 0)
|
|
|
|
if ($localIsAncestorOfRemote) {
|
|
Write-Host "Remote '$remoteBranch' is ahead. Pulling with rebase..."
|
|
git pull --rebase origin main
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "'git pull --rebase' failed. Manual intervention may be required."; exit 1 }
|
|
}
|
|
elseif ($remoteIsAncestorOfLocal) {
|
|
Write-Host "Local HEAD is ahead. Pushing..."
|
|
git push origin main
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "'git push' failed. Manual intervention may be required."; exit 1 }
|
|
}
|
|
else {
|
|
Write-Host "Local HEAD and remote '$remoteBranch' have diverged."
|
|
$remoteTimestampStr = (git log --pretty=format:"%at" -n 1 $remoteBranch 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $remoteTimestampStr) { Write-Error "Failed to get remote commit timestamp for '$remoteBranch'. Exiting script."; exit 1 }
|
|
$remoteTimestamp = [long]$remoteTimestampStr
|
|
|
|
$localTimestampStr = (git log --pretty=format:"%at" -n 1 HEAD 2>$null).Trim()
|
|
if ($LASTEXITCODE -ne 0 -or -not $localTimestampStr) { Write-Error "Failed to get local commit timestamp for HEAD. Exiting script."; exit 1 }
|
|
$localTimestamp = [long]$localTimestampStr
|
|
|
|
# It's good practice to check if conversion was successful, though [long] will error on failure.
|
|
|
|
Write-Host "Local timestamp: $localTimestamp, Remote timestamp: $remoteTimestamp"
|
|
|
|
if ($remoteTimestamp -gt $localTimestamp) {
|
|
Write-Host "Remote is newer. Pulling with rebase, strategy 'theirs'..."
|
|
git pull --rebase -X theirs origin main
|
|
}
|
|
else {
|
|
Write-Host "Local is newer or same age. Pulling with rebase, strategy 'ours'..."
|
|
git pull --rebase -X ours origin main
|
|
}
|
|
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "Rebase during divergence failed. Manual intervention may be required."; exit 1 }
|
|
|
|
Write-Host "Pushing changes after rebase..."
|
|
git push origin main
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "'git push' after rebase failed. Manual intervention may be required."; exit 1 }
|
|
}
|
|
|
|
Write-Host "Synchronization process completed successfully."
|
|
exit 0
|
|
}
|
|
finally {
|
|
Stop-Transcript
|
|
}
|