Added check for module loaded

Get-VB365Data/GetVB365Data.ps1

@@ -23,11 +23,13 @@
     .OUTPUTS
         vb365-environment-info.json
     .Notes 
-    Version:        1.0
+    Version:        1.2
     Author:         Ed Howard (edward.x.howard@veeam.com)
     Creation Date:  06.06.2022
     Purpose/Change: 06.06.2022 - 1.0 - Initial script development
     Purpose/Change: 20.11.2022 - 1.1 - Update of output file
+    Contributor:    David Tosoff (https://github.com/drtcode)
+    Purpose/Change: 12.14.2022 - 1.2 - Added check for loaded Graph modules, and Try-Catch block for associated cmdlets.
 #>
 
 [CmdletBinding()]
@@ -47,35 +49,49 @@ if ($Days -notin $options) {
 }
 
 if ($Local -eq $False) {
-    Connect-MgGraph -Scopes "user.read.all", "reports.read.all"
 
-    # GetOffice365ActiveUserDetail
-    Get-MgReportOffice365ActiveUserDetail -Period D$Days -OutFile active_user_detail.csv
+    # Add check for modules loaded
+    if (!(Get-Module -Name Microsoft.Graph*)) { # if no modules found with this name.
+        Write-Host "Microsoft.Graph module does not appear to be loaded."
+        Write-Host "Refer to README @ https://github.com/VeeamHub/veeam-calculators/tree/master/Get-VB365Data for guidance on installing/loading these modules."
+        Exit
+    }
+
+    try {
+        Connect-MgGraph -Scopes "user.read.all", "reports.read.all"
 
-    # GetOffice365ActiveUserCounts
-    Get-MgReportOffice365ActiveUserCount -Period D$Days -OutFile active_user_counts.csv
+        # GetOffice365ActiveUserDetail
+        Get-MgReportOffice365ActiveUserDetail -Period D$Days -OutFile active_user_detail.csv
 
-    # GetOffice365GroupsActivityGroupCounts
-    Get-MgReportOffice365GroupActivityCount -Period D$Days -OutFile group_activity_counts.csv
+        # GetOffice365ActiveUserCounts
+        Get-MgReportOffice365ActiveUserCount -Period D$Days -OutFile active_user_counts.csv
 
-    # GetMailboxUsageDetail
-    Get-MgReportMailboxUsageDetail -Period D$Days -OutFile mailbox_usage_detail.csv 
-
-    # GetMailboxUsageStorage
-    Get-MgReportMailboxUsageStorage -Period D$Days -OutFile mailbox_usage_storage.csv
+        # GetOffice365GroupsActivityGroupCounts
+        Get-MgReportOffice365GroupActivityCount -Period D$Days -OutFile group_activity_counts.csv
 
-    # GetOneDriveUsageStorage
-    Get-MgReportOneDriveUsageStorage -Period D$Days -OutFile onedrive_usage_storage.csv
+        # GetMailboxUsageDetail
+        Get-MgReportMailboxUsageDetail -Period D$Days -OutFile mailbox_usage_detail.csv 
+
+        # GetMailboxUsageStorage
+        Get-MgReportMailboxUsageStorage -Period D$Days -OutFile mailbox_usage_storage.csv
 
-    # GetSharePointSiteUsageStorage
-    Get-MgReportSharePointSiteUsageStorage -Period D$Days -OutFile sharepoint_site_storage.csv
+        # GetOneDriveUsageStorage
+        Get-MgReportOneDriveUsageStorage -Period D$Days -OutFile onedrive_usage_storage.csv
 
-    # GetSharePointSiteUseageSiteCounts
-    Get-MgReportSharePointSiteUsageSiteCount -Period D$Days -OutFile sharepoint_site_counts.csv
+        # GetSharePointSiteUsageStorage
+        Get-MgReportSharePointSiteUsageStorage -Period D$Days -OutFile sharepoint_site_storage.csv
 
-    # GetSharePointSitesDetail
-    Get-MgReportSharePointSiteUsageDetail -Period D$Days -OutFile sharepoint_sites_detail.csv
-
+        # GetSharePointSiteUseageSiteCounts
+        Get-MgReportSharePointSiteUsageSiteCount -Period D$Days -OutFile sharepoint_site_counts.csv
+
+        # GetSharePointSitesDetail
+        Get-MgReportSharePointSiteUsageDetail -Period D$Days -OutFile sharepoint_sites_detail.csv
+    } catch {
+        Write-Host "There was an issue executing Microsoft.Graph cmdlets."
+        Write-Host "Check to ensure the modules are loaded, you are authenticated, and you have the required Graph permissions."
+        Write-Host "Refer to README @ https://github.com/VeeamHub/veeam-calculators/tree/master/Get-VB365Data for details."
+        Exit
+    }
 }
 else {
     Write-Host "Running in Local mode"