Added M365 connections to shared

Shared/M365/EXOConnection.ps1

@@ -0,0 +1,96 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+. $PSScriptRoot\..\ModuleHandle.ps1
+
+function Connect-EXOAdvanced {
+    [CmdletBinding(SupportsShouldProcess = $true, ConfirmImpact = "High")]
+    param (
+        [Parameter(Mandatory = $false, ParameterSetName = 'SingleSession')]
+        [Parameter(Mandatory = $false, ParameterSetName = 'AllowMultipleSessions')]
+        [switch]$DoNotShowConnectionDetails,
+        [Parameter(Mandatory = $true, ParameterSetName = 'AllowMultipleSessions')]
+        [switch]$AllowMultipleSessions,
+        [Parameter(Mandatory = $true, ParameterSetName = 'AllowMultipleSessions')]
+        [string]$Prefix = $null
+    )
+
+    #Validate EXO 3.0 is installed and loaded
+    $requestModule = $false
+    $requestModule = Request-Module -Modules "ExchangeOnlineManagement" -MinModuleVersion 3.0.0
+
+    if (-not $requestModule) {
+        Write-Host "We cannot continue without ExchangeOnlineManagement Powershell module" -ForegroundColor Red
+        return $null
+    }
+
+    #Validate EXO is connected or try to connect
+    $connections = $null
+    $newConnection = $null
+    try {
+        $connections = Get-ConnectionInformation -ErrorAction Stop | Where-Object { $_.State -eq 'Connected' }
+    } catch {
+        Write-Host "We cannot check connections. Error:`n$_" -ForegroundColor Red
+        return $null
+    }
+
+    if ($null -eq $connections -or $AllowMultipleSessions) {
+        if ($connections.ModulePrefix -contains $Prefix) {
+            Write-Host "You already have a session with the prefix $Prefix" -ForegroundColor Red
+            return $null
+        } else {
+            $prefixString = "."
+            if ($Prefix) { $prefixString = " with Prefix $Prefix." }
+            Write-Host "Not connected to Exchange Online$prefixString" -ForegroundColor Yellow
+
+            if ($PSCmdlet.ShouldProcess("Do you want to add it?", "Adding an Exchange Online Session")) {
+                Write-Verbose "Connecting to Exchange Online session"
+                try {
+                    Connect-ExchangeOnline -ShowBanner:$false -Prefix $Prefix -ErrorAction Stop
+                } catch {
+                    Write-Host "We cannot connect to Exchange Online. Error:`n$_" -ForegroundColor Red
+                    return $null
+                }
+                try {
+                    $newConnections = Get-ConnectionInformation -ErrorAction Stop
+                } catch {
+                    Write-Host "We cannot check connections. Error:`n$_" -ForegroundColor Red
+                    return $null
+                }
+                foreach ($testConnection in $newConnections) {
+                    if ($connections -notcontains $testConnection) {
+                        $newConnection = $testConnection
+                    }
+                }
+            }
+        }
+    } else {
+        Write-Verbose "You already have an Exchange Online session"
+        if ($connections.count -gt 1) {
+            Write-Host "You have more than one Exchange Online sessions please use just one session. You are not using AllowMultipleSessions" -ForegroundColor Red
+            return $null
+        }
+        $newConnection = $connections
+    }
+
+    Write-Verbose "Connected session to Exchange Online"
+    $newConnection.PSObject.Properties | ForEach-Object { Write-Verbose "$($_.Name): $($_.Value)" }
+    if (-not $DoNotShowConnectionDetails) {
+        Show-EXOConnection -Connection $newConnection
+    }
+    return $newConnection
+}
+
+function Show-EXOConnection {
+    param (
+        [Parameter(Mandatory = $true)]
+        [Microsoft.Exchange.Management.ExoPowershellSnapin.ConnectionInformation]$Connection
+    )
+    Write-Host "`nConnected to Exchange Online"
+    Write-Host "Session details"
+    Write-Host "Tenant Id: $($Connection.TenantId)"
+    Write-Host "User: $($Connection.UserPrincipalName)"
+    if ($($Connection.ModulePrefix)) {
+        Write-Host "Prefix: $($Connection.ModulePrefix)"
+    }
+}

Shared/M365/GraphConnection.ps1

@@ -0,0 +1,137 @@
+# 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)]
+        [string[]]$TenantId = $null,
+        [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
+    try {
+        $connection = Get-MgContext -ErrorAction Stop
+    } catch {
+        Write-Host "We cannot check context. Error:`n$_" -ForegroundColor Red
+        return $null
+    }
+
+    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
+        } else {
+            Write-Verbose "All scopes are present"
+        }
+    }
+
+    if ($connection) {
+        Write-Verbose "Checking TenantId"
+        if ($TenantId) {
+            if ($connection.TenantId -ne $TenantId) {
+                Write-Host "Connected to $($connection.TenantId). Not expected tenant: $TenantId" -ForegroundColor Red
+                return $null
+            } else {
+                Write-Verbose "TenantId is correct"
+            }
+        }
+
+        if (-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"
+        try {
+            Connect-MgGraph -Scopes $Scopes -NoWelcome -ErrorAction Stop
+        } catch {
+            Write-Host "We cannot connect to Graph. Error:`n$_" -ForegroundColor Red
+            return $null
+        }
+        $connection = $null
+        try {
+            $connection = Get-MgContext -ErrorAction Stop
+        } catch {
+            Write-Host "We cannot check context. Error:`n$_" -ForegroundColor Red
+            return $null
+        }
+        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,56 @@
+# 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
+            try {
+                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 Stop
+                    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 Stop
+                    if (-not $confirmed) {
+                        Write-Host "We could not install module: $module" -ForegroundColor Red
+                        $confirmed = $true
+                    }
+                }
+            } catch {
+                Write-Host "Installation process fails. Error: `n$_" -ForegroundColor Red
+                return $false
+            }
+        }
+    }
+    if ($foundError) {
+        return $false
+    }
+    return $true
+}