Added M365 connections to shared

Shared/Connect-M365.ps1

@@ -0,0 +1,159 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+function Test-EXOConnection {
+    [CmdletBinding(SupportsShouldProcess, ConfirmImpact = 'High')]
+    [OutputType([bool])]
+    param (
+        [Switch]$Force
+    )
+    #Validate EXO V2 is installed
+    if ((Get-Module -ListAvailable | Where-Object { $_.Name -like "ExchangeOnlineManagement" }).count -ge 1) {
+        Write-Host "ExchangeOnlineManagement Powershell Module installed"
+    } else {
+        if ($Force -or $PSCmdlet.ShouldContinue("Do you want to install the module?", "ExchangeOnlineManagement Powershell Module not installed")) {
+            Install-Module -Name ExchangeOnlineManagement -Force -ErrorAction SilentlyContinue -Scope CurrentUser
+            if ((Get-Module -ListAvailable | Where-Object { $_.Name -like "ExchangeOnlineManagement" }).count -ge 1) {
+                Write-Host "ExchangeOnlineManagement Powershell Module installed"
+            } else {
+                Write-Host "ExchangeOnlineManagement Powershell Module installation failed" -ForegroundColor Red
+                return $false
+            }
+        } else {
+            Write-Host "We cannot continue without ExchangeOnlineManagement Powershell module" -ForegroundColor Red
+            return $false
+        }
+    }
+
+    #Validate EXO V2 is loaded
+    if ((Get-Module | Where-Object { $_.Name -like "ExchangeOnlineManagement" }).count -ge 1) {
+        Write-Host "ExchangeOnlineManagement Powershell Module loaded"
+    } else {
+        Import-Module ExchangeOnlineManagement -ErrorAction SilentlyContinue -Force
+        if ((Get-Module | Where-Object { $_.Name -like "ExchangeOnlineManagement" }).count -ge 1) {
+            Write-Host "ExchangeOnlineManagement Powershell Module Imported"
+        } else {
+            Write-Host "ExchangeOnlineManagement Powershell Module Import failed" -ForegroundColor Red
+            return $false
+        }
+    }
+
+    #Validate EXO V2 is connected or try to connect
+    $connection = $null
+    $connection = Get-ConnectionInformation -ErrorAction SilentlyContinue
+    if ($null -eq $connection) {
+        Write-Host "Please use Global administrator credentials" -ForegroundColor Yellow
+        if ($Force -or $PSCmdlet.ShouldContinue("Do you want to connect?", "We need a ExchangeOnlineManagement connection")) {
+            Connect-ExchangeOnline -ErrorAction SilentlyContinue
+            $connection = Get-ConnectionInformation -ErrorAction SilentlyContinue
+            if ($null -eq $connection) {
+                Write-Host "Connection could not be established" -ForegroundColor Red
+                Write-Host "We cannot continue without ExchangeOnlineManagement Powershell session" -ForegroundColor Red
+                return $false
+            } else {
+                Write-Host "Connected to EXO V2"
+                Write-Host "Session details"
+                Write-Host "Tenant Id: $($connection.TenantId)"
+                Write-Host "User: $($connection.UserPrincipalName)"
+                return $true
+            }
+        } else {
+            Write-Host "We cannot continue without ExchangeOnlineManagement Powershell session" -ForegroundColor Red
+            return $false
+        }
+    } elseif ($connection.count -eq 1) {
+        Write-Host "Connected to EXO V2"
+        Write-Host "Session details"
+        Write-Host "Tenant Id: $($connection.TenantId)"
+        Write-Host "User: $($connection.UserPrincipalName)"
+        return $true
+    } else {
+        Write-Host "You have more than one EXO sessions please use just one session" -ForegroundColor Red
+        return $false
+    }
+}
+
+function Test-AADConnection {
+    [CmdletBinding(SupportsShouldProcess, ConfirmImpact = 'High')]
+    [OutputType([bool])]
+    param (
+        [Switch]$Force
+    )
+
+    #Validate AzureAD is installed
+    if ((Get-Module -ListAvailable | Where-Object { $_.Name -like "AzureAD" }).count -ge 1) {
+        Write-Host "AzureAD Powershell Module installed"
+    } else {
+        if ($Force -or $PSCmdlet.ShouldContinue("Do you want to install the module?", "AzureAD Powershell Module not installed")) {
+            Install-Module -Name AzureAD -Repository PSGallery -AllowClobber -Force -ErrorAction SilentlyContinue -Scope CurrentUser
+            if ((Get-Module -ListAvailable | Where-Object { $_.Name -like "AzureAD" }).count -ge 1) {
+                Write-Host "AzureAD Powershell Module installed"
+            } else {
+                Write-Host "AzureAD Powershell Module installation failed" -ForegroundColor Red
+                return $false
+            }
+        } else {
+            Write-Host "We cannot continue without AzureAD Powershell module" -ForegroundColor Red
+            return $false
+        }
+    }
+
+    #Validate AzureAD is loaded
+    if ((Get-Module | Where-Object { $_.Name -like "AzureAD" }).count -ge 1) {
+        Write-Host "AzureAD Powershell Module loaded"
+    } else {
+        Import-Module AzureAD -ErrorAction SilentlyContinue -Force
+        if ((Get-Module | Where-Object { $_.Name -like "AzureAD" }).count -ge 1) {
+            Write-Host "AzureAD Powershell Module Imported"
+        } else {
+            Write-Host "AzureAD Powershell Module Import failed" -ForegroundColor Red
+            return $false
+        }
+    }
+
+    #Validate AzureAD is connected or try to connect
+    try {
+        $connection = $null
+        $connection = Get-AzureADTenantDetail -ErrorAction SilentlyContinue
+        if ($null -eq $connection) {
+            Write-Host "Not connected to AzureAD" -ForegroundColor Red
+            Write-Host "We cannot continue without AzureAD Powershell session" -ForegroundColor Red
+            return $false
+        } else {
+            if ($connection.count -eq 1) {
+                Write-Host "Connected to AzureAD"
+                Write-Host "Session details"
+                Write-Host "Tenant: $($connection.DisplayName)"
+                return $true
+            } else {
+                Write-Host "You have more than one AzureAD sessions please use just one session" -ForegroundColor Red
+                return $false
+            }
+        }
+    } catch [Microsoft.Open.Azure.AD.CommonLibrary.AadNeedAuthenticationException] {
+        Write-Host "Please use Global administrator credentials" -ForegroundColor Yellow
+        if ($Force -or $PSCmdlet.ShouldContinue("Do you want to connect?", "We need a AzureAD connection")) {
+            Connect-AzureAD -ErrorAction SilentlyContinue
+            try {
+                $connection = Get-AzureADTenantDetail -ErrorAction SilentlyContinue
+                if ($null -eq $connection) {
+                    Write-Host "Connection could not be established" -ForegroundColor Red
+                    Write-Host "We cannot continue without AzureAD Powershell session" -ForegroundColor Red
+                    return $false
+                } else {
+                    Write-Host "Connected to AzureAD"
+                    Write-Host "Session details"
+                    Write-Host "Tenant: $($connection.DisplayName)"
+                    return $true
+                }
+            } catch [Microsoft.Open.Azure.AD.CommonLibrary.AadNeedAuthenticationException] {
+                Write-Host "Connection could not be established" -ForegroundColor Red
+                Write-Host "We cannot continue without AzureAD Powershell session" -ForegroundColor Red
+                return $false
+            }
+        } else {
+            Write-Host "We cannot continue without AzureAD Powershell session" -ForegroundColor Red
+            return $false
+        }
+    }
+}

Shared/M365/GraphConnection.ps1

@@ -0,0 +1,104 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+. $PSScriptRoot\..\ModuleHandle.ps1
+
+function Connect-GraphAdvanced {
+    param (
+        [Parameter(Mandatory = $true)]
+        [string[]]$Scopes,
+        [Parameter(Mandatory = $true)]
+        [string[]]$Modules,
+        [Parameter(Mandatory = $false)]
+        [switch]$DoNotShowConnectionDetails
+    )
+
+    #Validate Graph is installed and loaded
+    $requestModule = $false
+    $requestModule = Request-Module -Modules $Modules
+    if (-not $requestModule) {
+        Write-Host "We cannot continue without $Modules Powershell module" -ForegroundColor Red
+        return $null
+    }
+
+    #Validate Graph is connected or try to connect
+    $connection = $null
+    $connection = Get-MgContext -ErrorAction SilentlyContinue
+    if ($null -eq $connection) {
+        Write-Host "Not connected to Graph" -ForegroundColor Yellow
+        $connection = Add-GraphConnection -Scopes $Scopes
+    } else {
+        Write-Verbose "You have a Graph sessions"
+        Write-Verbose "Checking scopes"
+        if (-not (Test-GraphContext -Scopes $connection.Scopes -ExpectedScopes $Scopes)) {
+            Write-Host "Not connected to Graph with expected scopes" -ForegroundColor Yellow
+            $connection = Add-GraphConnection -Scopes $Scopes
+        }
+    }
+    if ($null -ne $connection -and -not $DoNotShowConnectionDetails) {
+        $connection.PSObject.Properties | ForEach-Object { Write-Verbose "$($_.Name): $($_.Value)" }
+        Show-GraphContext -Context $connection
+    }
+    return $connection
+}
+
+function Add-GraphConnection {
+    [CmdletBinding(SupportsShouldProcess = $true, ConfirmImpact = "High")]
+    param (
+        [Parameter(Mandatory = $true)]
+        [string[]]$Scopes
+    )
+
+    if ($PSCmdlet.ShouldProcess("Do you want to connect?", "We need a Graph connection with scopes $Scopes")) {
+        Write-Verbose "Connecting to Microsoft Graph API using scopes $Scopes"
+        Connect-MgGraph -Scopes $Scopes -NoWelcome -ErrorAction SilentlyContinue
+        $connection = $null
+        $connection = Get-MgContext -ErrorAction SilentlyContinue
+        Write-Verbose "Checking scopes"
+        if (-not $connection) {
+            Write-Host "We cannot continue without Graph Powershell session" -ForegroundColor Red
+            return $null
+        }
+        if (-not (Test-GraphContext -Scopes $connection.Scopes -ExpectedScopes $Scopes)) {
+            Write-Host "We cannot continue without Graph Powershell session without Expected Scopes" -ForegroundColor Red
+            return $null
+        }
+        return $connection
+    }
+}
+
+function Test-GraphContext {
+    [OutputType([bool])]
+    param (
+        [Parameter(Mandatory = $true)]
+        [string[]]$ExpectedScopes,
+        [Parameter(Mandatory = $true)]
+        [string[]]$Scopes
+    )
+
+    $foundError = $false
+    foreach ($expectedScope in $ExpectedScopes) {
+        if ($Scopes -notcontains $expectedScope) {
+            Write-Host "The following scope is missing: $expectedScope" -ForegroundColor Red
+            $foundError = $true
+        }
+    }
+
+    if ($foundError) {
+        return $false
+    } else {
+        Write-Verbose "All expected scopes are present."
+        return $true
+    }
+}
+
+function Show-GraphContext {
+    param (
+        [Parameter(Mandatory = $true)]
+        [Microsoft.Graph.PowerShell.Authentication.AuthContext]$Context
+    )
+    Write-Host "`nConnected to Graph"
+    Write-Host "Session details"
+    Write-Host "Tenant Id: $($Context.TenantId)"
+    Write-Host "Account: $($Context.Account)"
+}

Shared/ModuleHandle.ps1

@@ -0,0 +1,51 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+function Request-Module {
+    [OutputType([bool])]
+    param (
+        [Parameter(Mandatory = $true)]
+        [string[]]$Modules,
+        [Parameter(Mandatory = $false)]
+        [System.Version]$MinModuleVersion = $null
+    )
+
+    $installed = $null
+    Write-Verbose "Checking $Modules PowerShell Module"
+    if ($MinModuleVersion) {
+        Write-Verbose "with minimum version $minModuleVersion"
+        $installed = Get-InstalledModule -Name $Modules -MinimumVersion $MinModuleVersion -ErrorAction SilentlyContinue
+    } else {
+        Write-Verbose "without minimum version"
+        $installed = Get-InstalledModule -Name $Modules -ErrorAction SilentlyContinue
+    }
+
+    $foundError = $false
+    foreach ($module in $Modules) {
+        if ($null-eq $installed -or $installed.Name -notcontains $module) {
+            Write-Host "The following module is missing: $module" -ForegroundColor Yellow
+            $confirmed = $null
+            if ($MinModuleVersion) {
+                Write-Verbose "Installing $module with minimum version $minModuleVersion"
+                Install-Module -Name $module -Scope CurrentUser -MinimumVersion $MinModuleVersion
+                $confirmed = Get-InstalledModule -Name $module -MinimumVersion $MinModuleVersion -ErrorAction SilentlyContinue
+                if (-not $confirmed) {
+                    Write-Host "We could not install module: $module with minimum version $minModuleVersion" -ForegroundColor Red
+                    $foundError = $true
+                }
+            } else {
+                Write-Verbose "Installing $module"
+                Install-Module -Name $module -Scope CurrentUser
+                $confirmed = Get-InstalledModule -Name $module -ErrorAction SilentlyContinue
+                if (-not $confirmed) {
+                    Write-Host "We could not install module: $module" -ForegroundColor Red
+                    $confirmed = $true
+                }
+            }
+        }
+    }
+    if ($foundError) {
+        return $false
+    }
+    return $true
+}