-
Notifications
You must be signed in to change notification settings - Fork 112
/
Reset-AutomationSchedules.ps1
186 lines (165 loc) · 9.27 KB
/
Reset-AutomationSchedules.ps1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
param(
[Parameter(Mandatory = $false)]
[String] $AzureEnvironment = "AzureCloud",
[Parameter(Mandatory = $true)]
[String] $AutomationAccountName,
[Parameter(Mandatory = $true)]
[String] $ResourceGroupName
)
$ErrorActionPreference = "Stop"
$ctx = Get-AzContext
if (-not($ctx)) {
Connect-AzAccount -Environment $AzureEnvironment
$ctx = Get-AzContext
}
else {
if ($ctx.Environment.Name -ne $AzureEnvironment) {
Disconnect-AzAccount -ContextName $ctx.Name
Connect-AzAccount -Environment $AzureEnvironment
$ctx = Get-AzContext
}
}
try {
$scheduledRunbooks = Get-AzAutomationScheduledRunbook -AutomationAccountName $AutomationAccountName -ResourceGroupName $ResourceGroupName
}
catch {
throw "$AutomationAccountName Automation Account not found in Resource Group $ResourceGroupName in Subscription $($ctx.Subscription.Name). If we are not in the right subscription, use Set-AzContext to switch to the correct one."
}
if (-not($scheduledRunbooks)) {
throw "The $AutomationAccountName Automation Account does not contain any scheduled runbook. It might not be associated to the Azure Optimization Engine."
}
$subscriptionId = (Get-AzContext).Subscription.Id
$schedules = Get-AzAutomationSchedule -ResourceGroupName $ResourceGroupName -AutomationAccountName $AutomationAccountName
$weeklySchedules = $schedules | Where-Object { $_.Name.StartsWith("AzureOptimization") -and $_.Name.EndsWith("Weekly") }
if ($weeklySchedules.Count -gt 0) {
$originalBaseTime = ($weeklySchedules | Sort-Object -Property StartTime | Select-Object -First 1).StartTime.AddHours(-1.25).DateTime
$now = (Get-Date).ToUniversalTime()
$diff = $now.AddHours(-1.25) - $originalBaseTime
$nextWeekDays = [Math]::Ceiling($diff.TotalDays / 7) * 7
$baseDateTime = $now.AddHours(-1.25).AddDays($nextWeekDays - $diff.TotalDays)
$baseTimeStr = $baseDateTime.ToString("u")
Write-Host "Existing schedules found. Weekly base time is $($baseDateTime.DayOfWeek) at $($baseDateTime.ToString('T')) (UTC)." -ForegroundColor Green
}
else {
throw "The $AutomationAccountName Automation Account does not contain Azure Optimization Engine schedules."
}
$newBaseTimeStr = Read-Host "Please, enter a new base time for the *weekly* schedules in UTC (YYYY-MM-dd HH:mm:ss). If you want to keep the current one, just press ENTER"
if (-not($newBaseTimeStr)) {
$newBaseTimeStr = $baseTimeStr
}
else {
try {
$newBaseTimeStr += "Z"
$newBaseTime = [DateTime]::Parse($newBaseTimeStr)
}
catch {
throw "$newBaseTimeStr is an invalid base time. Use the following format: YYYY-MM-dd HH:mm:ss. For example: 1977-09-08 06:14:15"
}
if ($newBaseTime -lt (Get-Date).ToUniversalTime().AddHours(-1)) {
throw "$newBaseTimeStr is an invalid base time. It can't be sooner than $((Get-Date).ToUniversalTime().AddHours(-1).ToString('u'))"
}
}
$baseTimeUtc = [DateTime]::Parse($newBaseTimeStr).ToUniversalTime()
if ($newBaseTimeStr -ne $baseTimeStr) {
Write-Host "Updating current base schedule to every $($baseTimeUtc.DayOfWeek) at $($baseTimeUtc.TimeOfDay.ToString()) UTC..." -ForegroundColor Green
$continueInput = Read-Host "Continue (Y/N)?"
if ("Y", "y" -contains $continueInput) {
$upgradeManifest = Get-Content -Path "./upgrade-manifest.json" | ConvertFrom-Json
$manifestSchedules = $upgradeManifest.schedules
foreach ($schedule in $schedules) {
$manifestSchedule = $manifestSchedules | Where-Object { $_.name -eq $schedule.Name }
if ($manifestSchedule) {
if ($schedule.Frequency -eq "Week") {
$newStartTime = $baseTimeUtc.Add([System.Xml.XmlConvert]::ToTimeSpan($manifestSchedule.offset))
}
else {
$now = (Get-Date).ToUniversalTime()
$newStartTime = [System.DateTimeOffset]::Parse($now.ToString("yyyy-MM-ddT00:00:00Z"))
$newStartTime = $newStartTime.AddHours($baseTimeUtc.Hour).AddMinutes($baseTimeUtc.Minute).AddSeconds($baseTimeUtc.Second)
if ($newStartTime -lt $now.AddMinutes(15))
{
$newStartTime = $newStartTime.AddDays(1)
}
$newStartTime = $newStartTime.Add([System.Xml.XmlConvert]::ToTimeSpan($manifestSchedule.offset))
}
$expiryTime = $schedule.ExpiryTime.ToString("yyyy-MM-ddTHH:mm:ssZ")
$startTime = $newStartTime.ToString("yyyy-MM-ddTHH:mm:ssZ")
$automationPath = "/subscriptions/$subscriptionId/resourceGroups/$resourceGroupName/providers/Microsoft.Automation/automationAccounts/$automationAccountName/schedules/$($schedule.Name)?api-version=2015-10-31"
$body = "{
`"name`": `"$($schedule.Name)`",
`"properties`": {
`"description`": `"$($schedule.Description)`",
`"startTime`": `"$startTime`",
`"expiryTime`": `"$expiryTime`",
`"interval`": $($schedule.Interval),
`"frequency`": `"$($schedule.Frequency)`",
`"advancedSchedule`": {}
}
}"
Invoke-AzRestMethod -Path $automationPath -Method PUT -Payload $body | Out-Null
Write-Host "Re-scheduled $($schedule.Name)." -ForegroundColor Blue
}
else {
Write-Host "$($schedule.Name) not found in schedules manifest." -ForegroundColor Yellow
}
}
}
else
{
throw "Interrupting schedules reset due to user input."
}
}
else {
Write-Host "Kept current base schedule (every $($baseTimeUtc.DayOfWeek) at $($baseTimeUtc.TimeOfDay.ToString()) UTC)." -ForegroundColor Green
}
$exportHybridWorkerOption = ($scheduledRunbooks | Where-Object { $_.RunbookName.StartsWith("Export") })[0].HybridWorker
$ingestHybridWorkerOption = ($scheduledRunbooks | Where-Object { $_.RunbookName.StartsWith("Ingest") })[0].HybridWorker
$recommendHybridWorkerOption = ($scheduledRunbooks | Where-Object { $_.RunbookName.StartsWith("Recommend") })[0].HybridWorker
if ($scheduledRunbooks | Where-Object { $_.RunbookName.StartsWith("Remediate") })
{
$remediateHybridWorkerOption = ($scheduledRunbooks | Where-Object { $_.RunbookName.StartsWith("Remediate") })[0].HybridWorker
}
$hybridWorkerOption = "None"
if ($exportHybridWorkerOption -or $ingestHybridWorkerOption -or $recommendHybridWorkerOption -or $remediateHybridWorkerOption) {
$hybridWorkerOption = "Export: $exportHybridWorkerOption; Ingest: $ingestHybridWorkerOption; Recommend: $recommendHybridWorkerOption; Remediate: $remediateHybridWorkerOption"
}
Write-Host "Current Hybrid Worker option: $hybridWorkerOption" -ForegroundColor Green
$newHybridWorker = Read-Host "If you want all schedules to use the same Hybrid Worker, please enter the Hybrid Worker Group name (if you want to keep the current option, just press ENTER)"
if ($newHybridWorker)
{
$hybridWorker = Get-AzAutomationHybridWorkerGroup -ResourceGroupName $resourceGroupName -AutomationAccountName $automationAccountName -Name $newHybridWorker -ErrorAction SilentlyContinue
if (-not($hybridWorker))
{
throw "Hybrid Worker $newHybridWorker was not found in Automation Account $automationAccountName."
}
Write-Host "Updating Hybrid Worker Group in every runbook schedule to $newHybridWorker..."
$continueInput = Read-Host "Continue (Y/N)?"
if ("Y", "y" -contains $continueInput)
{
Write-Host "Re-registering previous runbook schedules associations from $automationAccountName..." -ForegroundColor Green
foreach ($jobSchedule in $scheduledRunbooks) {
if ($jobSchedule.ScheduleName.StartsWith("AzureOptimization")) {
$automationPath = "/subscriptions/$subscriptionId/resourceGroups/$resourceGroupName/providers/Microsoft.Automation/automationAccounts/$automationAccountName/jobSchedules/$($jobSchedule.JobScheduleId)?api-version=2015-10-31"
$jobSchedule = (Invoke-AzRestMethod -Path $automationPath -Method GET).Content | ConvertFrom-Json
Unregister-AzAutomationScheduledRunbook -ResourceGroupName $resourceGroupName -AutomationAccountName $automationAccountName `
-JobScheduleId $jobSchedule.id.Split('/')[10] -Force
$params = @{}
$jobSchedule.properties.parameters.PSObject.Properties | ForEach-Object {
$params[$_.Name] = $_.Value
}
Register-AzAutomationScheduledRunbook -ResourceGroupName $resourceGroupName -AutomationAccountName $automationAccountName `
-RunbookName $jobSchedule.properties.runbook.name -ScheduleName $jobSchedule.properties.schedule.name -RunOn $newHybridWorker -Parameters $params | Out-Null
Write-Host "Re-registered $($jobSchedule.properties.runbook.name) for schedule $($jobSchedule.properties.schedule.name)." -ForegroundColor Blue
}
}
}
else
{
throw "Interrupting schedules reset due to user input."
}
}
else
{
Write-Host "Kept current Hybrid Worker option: $hybridWorkerOption" -ForegroundColor Green
}
Write-Host "DONE" -ForegroundColor Green