Merge pull request #1402 from microsoftgraph/fehintolaobafemi/parameter-completers

Microsoft Graph Authentication parameter completers

Author: FehintolaObafemi

src/Authentication/Authentication/custom/ParameterCompleters.ps1

@@ -9,7 +9,7 @@ Set-StrictMode -Version 2
 $scopesParameterBlock = {
     param($commandName,
     $parameterName,
-    $wordToComplete,
+    $wordToCompleteScopes,
     $commandAst,
     $fakeBoundParameters
     )
@@ -18,8 +18,25 @@ $scopesParameterBlock = {
     $permissionsOAuthQuery = Permissions_GetOauthData (Permissions_GetPermissionsData)
     $permissions += $permissionsOAuthQuery.Name
     $permissions | Where-Object {
-        $_ -like "$wordToComplete*"
+        $_ -like "$wordToCompleteScopes*"
     }
 }
 
-Register-ArgumentCompleter -CommandName Connect-MgGraph -ParameterName Scopes -ScriptBlock $scopesParameterBlock
+$environmentParameterBlock = {
+    param($commandName,
+    $parameterName,
+    $wordToCompleteEnvironment,
+    $commandAst,
+    $fakeBoundParameters
+    )
+
+    $environments = Get-MgEnvironment | Select-Object -ExpandProperty Name
+    $environments | Where-Object {
+        $_ -like "$wordToCompleteEnvironment*"
+    }
+}
+
+Register-ArgumentCompleter -CommandName Connect-MgGraph -ParameterName Scopes -ScriptBlock $scopesParameterBlock
+Register-ArgumentCompleter -CommandName Connect-MgGraph -ParameterName Environment -ScriptBlock $environmentParameterBlock
+
+Register-ArgumentCompleter -CommandName Get-MgEnvironment -ParameterName Name -ScriptBlock $environmentParameterBlock