Added check to see if JSON folder exists

Author: sanderstad

functions/clone/New-PSDCClone.ps1

@@ -360,7 +360,7 @@
                             $command = [ScriptBlock]::Create("New-VHD -ParentPath $ParentVhd -Path `"$Destination\$CloneName.vhdx`" -Differencing")
                             $vhd = Invoke-PSFCommand -ComputerName $computer -ScriptBlock $command -Credential $Credential
 
-                            if(-not $vhd){
+                            if (-not $vhd) {
                                 return
                             }
                         }
@@ -562,6 +562,19 @@
                                 FQDN      = $fqdn
                             }
 
+                            # Test if the JSON folder can be reached
+                            if (-not (Test-Path -Path "PSDCJSONFolder:\")) {
+                                $command = [scriptblock]::Create("Import-Module PSDatabaseClone -Force")
+
+                                try {
+                                    Invoke-PSFCommand -ComputerName $computer -ScriptBlock $command -Credential $Credential
+                                }
+                                catch {
+                                    Stop-PSFFunction -Message "Couldn't import module remotely" -Target $command
+                                    return
+                                }
+                            }
+
                             # Setup the json file
                             $jsonHostFile = "PSDCJSONFolder:\hosts.json"
 
@@ -668,6 +681,19 @@
                         IsEnabled     = $active
                     }
 
+                    # Test if the JSON folder can be reached
+                    if (-not (Test-Path -Path "PSDCJSONFolder:\")) {
+                        $command = [scriptblock]::Create("Import-Module PSDatabaseClone -Force")
+
+                        try {
+                            Invoke-PSFCommand -ComputerName $computer -ScriptBlock $command -Credential $Credential
+                        }
+                        catch {
+                            Stop-PSFFunction -Message "Couldn't import module remotely" -Target $command
+                            return
+                        }
+                    }
+
                     # Set the clone file
                     $jsonCloneFile = "PSDCJSONFolder:\clones.json"
 

functions/image/New-PSDCImage.ps1

@@ -578,6 +578,19 @@
                     CreatedOn         = (Get-Date -format "yyyyMMddHHmmss")
                 }
 
+                # Test if the JSON folder can be reached
+                if (-not (Test-Path -Path "PSDCJSONFolder:\")) {
+                    $command = [scriptblock]::Create("Import-Module PSDatabaseClone -Force")
+
+                    try {
+                        Invoke-PSFCommand -ComputerName $computer -ScriptBlock $command -Credential $Credential
+                    }
+                    catch {
+                        Stop-PSFFunction -Message "Couldn't import module remotely" -Target $command
+                        return
+                    }
+                }
+
                 # Set the image file
                 $jsonImageFile = "PSDCJSONFolder:\images.json"