# Check if the VeeamCatalogSvc service exists and throw an error if not installed
$veeamService = Get-Service -Name "VeeamCatalogSvc" -ErrorAction SilentlyContinue
if (-not $veeamService) {
    throw "This script cannot be used if the Veeam Catalog Service is not installed."
}

# Check if the VeeamCatalogSvc service is running and throw error if it is
if ($veeamService.Status -eq "Running") {
    throw "Stop all Veeam services before running this script."
}

# Function to open a folder browser dialog
Function Get-Folder {
    Add-Type -AssemblyName System.Windows.Forms
    $folderBrowser = New-Object System.Windows.Forms.FolderBrowserDialog
    $folderBrowser.Description = "Select the new VBRCatalog folder"
    $folderBrowser.RootFolder = [System.Environment+SpecialFolder]::MyComputer
    if ($folderBrowser.ShowDialog() -eq [System.Windows.Forms.DialogResult]::OK) {
        $folderBrowser.SelectedPath
    } else {
        throw "No folder selected. Exiting."
    }
}

# Prompt user to select new VBRCatalog folder
$newPath = Get-Folder

# Create or update SMB share
$existingShare = Get-SmbShare -Name VBRCatalog -ErrorAction SilentlyContinue
if ($existingShare -and $existingShare.Path -ne $newPath) {
    Copy-Item -Path "$($existingShare.Path)\*" -Destination $newPath -Recurse -Force
    Remove-SmbShare -Name VBRCatalog -Force
}
New-SmbShare -Name VBRCatalog -Path $newPath -ReadAccess "BUILTIN\Administrators" -ErrorAction SilentlyContinue

# Set new VBRCatalog Registry value
Set-ItemProperty -Path "HKLM:\SOFTWARE\Veeam\Veeam Backup Catalog" -Name "CatalogPath" -Value $newPath

# Find Veeam Backup Catalog's product ID and update its secondary location
$productID = (Get-ChildItem 'HKLM:\SOFTWARE\Classes\Installer\Products' | Where-Object {
    $keyProperties = Get-ItemProperty $_.PSPath -ErrorAction SilentlyContinue
    $keyProperties.ProductName -eq "Veeam Backup Catalog"
} | Select-Object -ExpandProperty PSChildName)

if ($productID) {
    Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Installer\UserData\S-1-5-18\Components\548FDDBB83EB487478FBC0FD5AC7CDCB" -Name $productID -Value $newPath
} else {
    throw "Veeam Backup Catalog product not found."
}