Skip to content

Commit 69cf846

Browse files
Migrate FirmwareAnalysis from generation to main (#24494)
* Move FirmwareAnalysis to main * Update ChangeLog.md --------- Co-authored-by: Vincent Dai <23257217+vidai-msft@users.noreply.github.com>
1 parent b0f8cab commit 69cf846

File tree

80 files changed

+1591
-168
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

80 files changed

+1591
-168
lines changed

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/Az.FirmwareAnalysis.psm1

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -47,37 +47,41 @@
4747

4848
# Ask for the shared functionality table
4949
$VTable = Register-AzModule
50-
50+
5151
# Tweaks the pipeline on module load
5252
$instance.OnModuleLoad = $VTable.OnModuleLoad
5353

5454
# Following two delegates are added for telemetry
5555
$instance.GetTelemetryId = $VTable.GetTelemetryId
5656
$instance.Telemetry = $VTable.Telemetry
57-
57+
58+
# Delegate to sanitize the output object
59+
$instance.SanitizeOutput = $VTable.SanitizerHandler
60+
61+
# Delegate to get the telemetry info
62+
$instance.GetTelemetryInfo = $VTable.GetTelemetryInfo
5863

5964
# Tweaks the pipeline per call
6065
$instance.OnNewRequest = $VTable.OnNewRequest
61-
66+
6267
# Gets shared parameter values
6368
$instance.GetParameterValue = $VTable.GetParameterValue
64-
69+
6570
# Allows shared module to listen to events from this module
6671
$instance.EventListener = $VTable.EventListener
67-
72+
6873
# Gets shared argument completers
6974
$instance.ArgumentCompleter = $VTable.ArgumentCompleter
70-
75+
7176
# The name of the currently selected Azure profile
7277
$instance.ProfileName = $VTable.ProfileName
7378

74-
7579
# Load the custom module
7680
$customModulePath = Join-Path $PSScriptRoot './custom/Az.FirmwareAnalysis.custom.psm1'
7781
if(Test-Path $customModulePath) {
7882
$null = Import-Module -Name $customModulePath
7983
}
80-
84+
8185
# Export nothing to clear implicit exports
8286
Export-ModuleMember
8387

@@ -97,12 +101,12 @@
97101
# Load the last folder if no profile is selected
98102
$profileDirectory = $directories | Select-Object -Last 1
99103
}
100-
104+
101105
if($profileDirectory) {
102106
Write-Information "Loaded Azure profile '$($profileDirectory.Name)' for module '$($instance.Name)'"
103107
$exportsPath = $profileDirectory.FullName
104108
}
105-
109+
106110
if($exportsPath) {
107111
Get-ChildItem -Path $exportsPath -Recurse -Include '*.ps1' -File | ForEach-Object { . $_.FullName }
108112
$cmdletNames = Get-ScriptCmdlet -ScriptFolder $exportsPath

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisBinaryHardening.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisBinaryHardening_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisCryptoCertificate.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisCryptoCertificate_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisCryptoKey.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisCryptoKey_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisCve.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisCve_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisFirmware.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,13 @@ begin {
180180
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisFirmware_List';
181181
}
182182
if (('Get', 'List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
183-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
183+
$testPlayback = $false
184+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
185+
if ($testPlayback) {
186+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
187+
} else {
188+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
189+
}
184190
}
185191
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
186192
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisPasswordHash.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisPasswordHash_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisSbomComponent.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,13 @@ begin {
136136
List = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisSbomComponent_List';
137137
}
138138
if (('List') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
139-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
139+
$testPlayback = $false
140+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
141+
if ($testPlayback) {
142+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
143+
} else {
144+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
145+
}
140146
}
141147
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
142148
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisSummary.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,13 @@ begin {
200200
GetViaIdentityWorkspace = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisSummary_GetViaIdentityWorkspace';
201201
}
202202
if (('Get') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
203-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
203+
$testPlayback = $false
204+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
205+
if ($testPlayback) {
206+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
207+
} else {
208+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
209+
}
204210
}
205211
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
206212
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/Get-AzFirmwareAnalysisWorkspace.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,13 @@ begin {
159159
List1 = 'Az.FirmwareAnalysis.private\Get-AzFirmwareAnalysisWorkspace_List1';
160160
}
161161
if (('Get', 'List', 'List1') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
162-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
162+
$testPlayback = $false
163+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
164+
if ($testPlayback) {
165+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
166+
} else {
167+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
168+
}
163169
}
164170
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
165171
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/New-AzFirmwareAnalysisFirmware.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,13 @@ begin {
201201
$PSBoundParameters['Id'] = New-Guid
202202
}
203203
if (('CreateExpanded') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
204-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
204+
$testPlayback = $false
205+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
206+
if ($testPlayback) {
207+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
208+
} else {
209+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
210+
}
205211
}
206212
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
207213
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/New-AzFirmwareAnalysisWorkspace.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,13 @@ begin {
144144
CreateExpanded = 'Az.FirmwareAnalysis.private\New-AzFirmwareAnalysisWorkspace_CreateExpanded';
145145
}
146146
if (('CreateExpanded') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
147-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
147+
$testPlayback = $false
148+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
149+
if ($testPlayback) {
150+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
151+
} else {
152+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
153+
}
148154
}
149155
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
150156
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

src/FirmwareAnalysis/FirmwareAnalysis.Autorest/exports/New-AzFirmwareAnalysisWorkspaceUploadUrl.ps1

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,13 @@ begin {
196196
GenerateViaJsonString = 'Az.FirmwareAnalysis.private\New-AzFirmwareAnalysisWorkspaceUploadUrl_GenerateViaJsonString';
197197
}
198198
if (('Generate', 'GenerateExpanded', 'GenerateViaJsonFilePath', 'GenerateViaJsonString') -contains $parameterSet -and -not $PSBoundParameters.ContainsKey('SubscriptionId') ) {
199-
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
199+
$testPlayback = $false
200+
$PSBoundParameters['HttpPipelinePrepend'] | Foreach-Object { $testPlayback = $testPlayback -or ('Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.PipelineMock' -eq $_.Target.GetType().FullName -and 'Playback' -eq $_.Target.Mode) }
201+
if ($testPlayback) {
202+
$PSBoundParameters['SubscriptionId'] = . (Join-Path $PSScriptRoot '..' 'utils' 'Get-SubscriptionIdTestSafe.ps1')
203+
} else {
204+
$PSBoundParameters['SubscriptionId'] = (Get-AzContext).Subscription.Id
205+
}
200206
}
201207
$cmdInfo = Get-Command -Name $mapping[$parameterSet]
202208
[Microsoft.Azure.PowerShell.Cmdlets.FirmwareAnalysis.Runtime.MessageAttributeHelper]::ProcessCustomAttributesAtRuntime($cmdInfo, $MyInvocation, $parameterSet, $PSCmdlet)

0 commit comments

Comments
 (0)