diff --git a/tools/add_odbc_connection.ps1 b/tools/add_odbc_connection.ps1 index 29095b3..e49e082 100644 --- a/tools/add_odbc_connection.ps1 +++ b/tools/add_odbc_connection.ps1 @@ -1,3 +1,22 @@ -$targetdir="DefaultDir="+(Split-Path (Get-Location))+"\components" +$script = $MyInvocation.MyCommand.Path + +$isAdmin = [Security.Principal.WindowsPrincipal]::new([Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator) +if (-not $isAdmin) { + Write-Warning "Not running as administrator. Restarting with elevated privileges..." + Start-Process powershell -Verb RunAs -ArgumentList "-ExecutionPolicy Bypass -File `"$script`"" + exit +} + +$targetdir=Join-Path (Split-Path (Split-Path $script)) "components" Write-Host 'Installing GhOST Library to' $targetdir -Add-OdbcDsn -Name "GhOST Library" -DriverName "Microsoft Access Text Driver (*.txt, *.csv)" -DsnType "User" -Platform "64-bit" -SetPropertyValue $targetdir \ No newline at end of file + +try { + Add-OdbcDsn -Name "GhOST Library" -DriverName "Microsoft Access Text Driver (*.txt, *.csv)" -DsnType "User" -Platform "64-bit" -SetPropertyValue "DefaultDir=$targetdir" -ErrorAction Stop + Write-Host "`nSUCCESS: ODBC connection 'GhOST Library' created." -ForegroundColor Green +} +catch { + Write-Host "`nERROR: $_" -ForegroundColor Red +} + +Write-Host "`nPress any key to exit..." -ForegroundColor Yellow +$null = $Host.UI.RawUI.ReadKey("NoEcho,IncludeKeyDown") diff --git a/tools/remove_odbc_connection.ps1 b/tools/remove_odbc_connection.ps1 index f61ffc8..892e754 100644 --- a/tools/remove_odbc_connection.ps1 +++ b/tools/remove_odbc_connection.ps1 @@ -1 +1,19 @@ -Remove-OdbcDsn -Name "GhOST Library" -DsnType "User" -Platform "64-bit" -ErrorAction 'SilentlyContinue' \ No newline at end of file +$script = $MyInvocation.MyCommand.Path + +$isAdmin = [Security.Principal.WindowsPrincipal]::new([Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator) +if (-not $isAdmin) { + Write-Warning "Not running as administrator. Restarting with elevated privileges..." + Start-Process powershell -Verb RunAs -ArgumentList "-ExecutionPolicy Bypass -File `"$script`"" + exit +} + +try { + Remove-OdbcDsn -Name "GhOST Library" -DsnType "User" -Platform "64-bit" + Write-Host "`nSUCCESS: ODBC connection 'GhOST Library' removed." -ForegroundColor Green +} +catch { + Write-Host "`nERROR: $_" -ForegroundColor Red +} + +Write-Host "`nPress any key to exit..." -ForegroundColor Yellow +$null = $Host.UI.RawUI.ReadKey("NoEcho,IncludeKeyDown")