From 6a7a0be67106d9d592e03199a2492fafc5f99ff0 Mon Sep 17 00:00:00 2001
From: Jonathan Butler <xxbutler86xx@gmail.com>
Date: Sat, 11 Jan 2025 11:38:43 -0500
Subject: [PATCH] Append investigation folder with seconds to avoid duplicate
 folder creation error.

---
 .../functions/Initialize-HawkGlobalObject.ps1 | 48 ++++++++++---------
 1 file changed, 26 insertions(+), 22 deletions(-)

diff --git a/Hawk/internal/functions/Initialize-HawkGlobalObject.ps1 b/Hawk/internal/functions/Initialize-HawkGlobalObject.ps1
index fd26ac9..a428092 100644
--- a/Hawk/internal/functions/Initialize-HawkGlobalObject.ps1
+++ b/Hawk/internal/functions/Initialize-HawkGlobalObject.ps1
@@ -91,36 +91,40 @@
         param([string]$RootPath)
     
         $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
-
-        Write-Information "[$timestamp UTC] - [ACTION] - Connecting to Microsoft Graph"
     
         try {
-            # Test Graph connection
-            $null = Test-GraphConnection
+            # Test Graph connection first to see if we're already connected
+            try {
+                $null = Get-MgOrganization -ErrorAction Stop
+                Write-Information "[$timestamp UTC] - [INFO]   - Already connected to Microsoft Graph"
+            }
+            catch {
+                # Only show connecting message if we actually need to connect
+                Write-Information "[$timestamp UTC] - [ACTION] - Connecting to Microsoft Graph"
+                $null = Test-GraphConnection
+                Write-Information "[$timestamp UTC] - [INFO]   - Connected to Microsoft Graph Successfully"
+            }
     
-            # If successful, display a success message
-            Write-Information "[$timestamp UTC] - [INFO]   - Connected to Microsoft Graph Successfully"
-        }
-        catch {
-            # If it fails, display an error message
-            Write-Error "[$timestamp UTC] - [ERROR] - Failed to connect to Microsoft Graph"
-        }
+            # Get tenant name 
+            $TenantName = (Get-MGDomain -ErrorAction Stop | Where-Object { $_.isDefault }).ID
+            [string]$FolderID = "Hawk_" + $TenantName.Substring(0, $TenantName.IndexOf('.')) + "_" + (Get-Date).ToUniversalTime().ToString("yyyyMMdd_HHmmss")
     
-        # Get tenant name
-        $TenantName = (Get-MGDomain -ErrorAction Stop | Where-Object { $_.isDefault }).ID
-        [string]$FolderID = "Hawk_" + $TenantName.Substring(0, $TenantName.IndexOf('.')) + "_" + (Get-Date).ToUniversalTime().ToString("yyyyMMdd_HHmm")
+            $FullOutputPath = Join-Path $RootPath $FolderID
     
-        $FullOutputPath = Join-Path $RootPath $FolderID
+            if (Test-Path $FullOutputPath) {
+                Write-Information "[$timestamp UTC] - [ERROR] - Path $FullOutputPath already exists"
+            }
+            else {
+                Write-Information "[$timestamp UTC] - [ACTION] - Creating subfolder $FullOutputPath"
+                $null = New-Item $FullOutputPath -ItemType Directory -ErrorAction Stop
+            }
     
-        if (Test-Path $FullOutputPath) {
-            Write-Information "[$timestamp UTC] - [ERROR] - Path $FullOutputPath already exists"
+            Return $FullOutputPath
         }
-        else {
-            Write-Information "[$timestamp UTC] - [ACTION] - Creating subfolder $FullOutputPath"
-            $null = New-Item $FullOutputPath -ItemType Directory -ErrorAction Stop
+        catch {
+            # If it fails at any point, display an error message
+            Write-Error "[$timestamp UTC] - [ERROR] - Failed to create logging folder: $_"
         }
-    
-        Return $FullOutputPath
     }
     
     Function Set-LoggingPath {