Dodawanie drukarki dodanej przez OptimiDoc Server

param (
    [switch]$Debug,
    [string]$DefaultServerAddress = "192.168.8.39",
    [string]$PrinterName = "OptimiDocPrinter",
    [string]$QueueUser = "$env:USERNAME",     # Changed to use Windows username as default
    [string]$QueueProject = "",  # New parameter for project part of queue name
    [string]$QueueCustom = "",   # New parameter for custom part of queue name
    [string]$DefaultDriver = ""  # New parameter for default driver
)

# Enable debug output if the -Debug switch is used
if ($Debug) { $DebugPreference = "Continue" }

Write-Debug "DefaultServerAddress: $DefaultServerAddress"
Write-Debug "PrinterName         : $PrinterName"
Write-Debug "QueueUser           : $QueueUser"
Write-Debug "QueueProject        : $QueueProject"
Write-Debug "QueueCustom         : $QueueCustom"
Write-Debug "DefaultDriver       : $DefaultDriver"
Write-Debug ""

# Load WPF module for GUI
Add-Type -AssemblyName PresentationFramework

# Ensure the logo file exists, download if necessary
$LogoPath = Join-Path $PSScriptRoot "optimidoc-logo.png"
$LogoUrl = "https://optimidoc.com/wp-content/uploads/2023/02/[email protected]"
if (-not (Test-Path $LogoPath)) {
    Write-Debug "Logo file not found. Downloading..."
    try {
        [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
        Invoke-WebRequest -Uri $LogoUrl -OutFile $LogoPath -ErrorAction Stop
        Write-Debug "Logo downloaded successfully to $LogoPath"
    } catch {
        Write-Warning "Failed to download logo: $_.Exception.Message"
    }
}
else {
        Write-Debug "Logo was already downloaded"
}

Write-Debug ""

# Function to list available printer drivers excluding Microsoft and default system printers
Function Get-FilteredPrinterDrivers {
    Write-Debug "Fetching filtered printer drivers..."
    Get-PrinterDriver | Where-Object {
        $_.Name -notmatch "Microsoft" `
                -and $_.Name `
                -notmatch "XPS" `
                -and $_.Name `
                -notmatch "PDF" `
                -and $_.Name `
                -notmatch "Fax" `
                -and $_.Name `
                -notmatch "Anydesk"
    } | Select-Object Name
}

# Function to display a combined GUI dialog for all inputs
Function Show-CombinedInputDialog {
    Param (
        [array]$DriverList,
        [string]$ServerAddress
    )
    Write-Debug "Showing combined input dialog..."
    Write-Debug "DriverList: "
    $DriverList | ForEach-Object { Write-Debug "    $($_.Name)" }
    Write-Debug "PassedToInputServerAddress: $ServerAddress"

    $Window = New-Object System.Windows.Window
    $Window.Title = "Printer Setup"
    $Window.Width = 500
    $Window.Height = 600 # Increased height to accommodate the new checkbox
    $Window.WindowStartupLocation = "CenterScreen"
    $Window.Background = "#333333" # Dark gray background

    $StackPanel = New-Object System.Windows.Controls.StackPanel
    $StackPanel.Margin = 20
    $StackPanel.HorizontalAlignment = "Center"
    $Window.Content = $StackPanel

    # Add OptimiDoc logo
    $Logo = New-Object System.Windows.Controls.Image
    try {
        $BitmapImage = New-Object System.Windows.Media.Imaging.BitmapImage
        $BitmapImage.BeginInit()
        $BitmapImage.UriSource = New-Object Uri($LogoPath, [UriKind]::Absolute)
        $BitmapImage.CacheOption = [System.Windows.Media.Imaging.BitmapCacheOption]::OnLoad
        $BitmapImage.EndInit()
        $Logo.Source = $BitmapImage
    } catch {
        Write-Warning "Failed to load logo image: $_.Exception.Message"
    }
    $Logo.Width = 200
    $Logo.Margin = "0,0,0,10"
    $StackPanel.Children.Add($Logo)

    # Check if printer exists

    Write-Debug "Using printer name: $PrinterName"
    # $PrinterExists = Get-Printer | Where-Object {$_.Name -eq $PrinterName}
    # $DeleteExistingPrinter = $false

    # Checkbox to delete existing printer
    $CheckBoxDeletePrinter = New-Object System.Windows.Controls.CheckBox
    $CheckBoxDeletePrinter.Content = "Delete existing printer if it exists"
    $CheckBoxDeletePrinter.IsChecked = $false
    $CheckBoxDeletePrinter.Margin = "0,0,0,10"
    $CheckBoxDeletePrinter.Foreground = "White"
    $StackPanel.Children.Add($CheckBoxDeletePrinter)

    # Printer Name Input
    $TextBlockPrinterName = New-Object System.Windows.Controls.TextBlock
    $TextBlockPrinterName.Text = "Enter the printer name:"
    $TextBlockPrinterName.FontSize = 16
    $TextBlockPrinterName.FontWeight = "Bold"
    $TextBlockPrinterName.Margin = "0,0,0,5"
    $TextBlockPrinterName.Foreground = "White"
    $StackPanel.Children.Add($TextBlockPrinterName)

    $TextBoxPrinterName = New-Object System.Windows.Controls.TextBox
    $TextBoxPrinterName.Width = 400
    $TextBoxPrinterName.Height = 30
    $TextBoxPrinterName.Margin = "0,0,0,20"
    $TextBoxPrinterName.FontSize = 14
    $TextBoxPrinterName.Background = "#F5F5F5"
    $TextBoxPrinterName.BorderBrush = "#BDBDBD"
    $TextBoxPrinterName.Foreground = "Gray"
    $TextBoxPrinterName.Text = $PrinterName # Default printer name
    $TextBoxPrinterName.Add_GotFocus({
        if ($TextBoxPrinterName.Foreground -eq "Gray") {
            $TextBoxPrinterName.Text = ""
            $TextBoxPrinterName.Foreground = "Black"
        }
    })
    $TextBoxPrinterName.Add_LostFocus({
        if ([string]::IsNullOrWhiteSpace($TextBoxPrinterName.Text)) {
            $TextBoxPrinterName.Text = $PrinterName
            $TextBoxPrinterName.Foreground = "Gray"
        }
    })
    $StackPanel.Children.Add($TextBoxPrinterName)

    # Printer Driver Selection
    $TextBlockDriver = New-Object System.Windows.Controls.TextBlock
    $TextBlockDriver.Text = "Select a printer driver:"
    $TextBlockDriver.FontSize = 16
    $TextBlockDriver.FontWeight = "Bold"
    $TextBlockDriver.Margin = "0,0,0,10"
    $TextBlockDriver.Foreground = "White"
    $StackPanel.Children.Add($TextBlockDriver)

    $RadioButtonGroup = New-Object System.Windows.Controls.StackPanel
    $RadioButtonGroup.Orientation = "Vertical"
    $RadioButtonGroup.Margin = "0,0,0,20"
    $StackPanel.Children.Add($RadioButtonGroup)

    # Initialize variables
    $SelectedDriver = ""
    $Window.Tag = $null

    # Radio Button Click Action
    $rbClickAction = {
        Param($sender)
        $script:SelectedDriver = $sender.Content
        Write-Debug "Selected driver: $($script:SelectedDriver)"
    }

    foreach ($Driver in $DriverList) {
        $rb = New-Object System.Windows.Controls.RadioButton
        Write-Debug "Building radiobutton for: $($Driver.Name)"
        $rb.Content = $Driver.Name
        $rb.GroupName = "printerDriver"
        $rb.FontSize = 14
        $rb.Foreground = "White"
        $rb.Margin = "0,2,0,2"
        $rb.Add_Click($rbClickAction)
        if ($Driver.Name -eq $DefaultDriver) {
            $rb.IsChecked = $true
            $script:SelectedDriver = $Driver.Name
            Write-Debug "Auto-selected default driver: $($Driver.Name)"
        }
        $RadioButtonGroup.Children.Add($rb)
    }

    # User Input (Queue Name Part 1)
    $TextBlockUser = New-Object System.Windows.Controls.TextBlock
    $TextBlockUser.Text = "Enter user ('L prefix will be added):"
    $TextBlockUser.FontSize = 16
    $TextBlockUser.FontWeight = "Bold"
    $TextBlockUser.Margin = "0,10,0,5"
    $TextBlockUser.Foreground = "White"
    $StackPanel.Children.Add($TextBlockUser)

    $TextBoxUser = New-Object System.Windows.Controls.TextBox
    $TextBoxUser.Width = 400
    $TextBoxUser.Height = 30
    $TextBoxUser.Margin = "0,0,0,20"
    $TextBoxUser.FontSize = 14
    $TextBoxUser.Background = "#F5F5F5"
    $TextBoxUser.BorderBrush = "#BDBDBD"
    $TextBoxUser.Text = $QueueUser  # Set default value from parameter
    $TextBoxUser.Foreground = if ([string]::IsNullOrWhiteSpace($QueueUser)) { "Gray" } else { "Black" }
    $StackPanel.Children.Add($TextBoxUser)

    # Project Input (Queue Name Part 2)
    $TextBlockProject = New-Object System.Windows.Controls.TextBlock
    $TextBlockProject.Text = "Enter project ('P prefix will be added):"
    $TextBlockProject.FontSize = 16
    $TextBlockProject.FontWeight = "Bold"
    $TextBlockProject.Margin = "0,10,0,5"
    $TextBlockProject.Foreground = "White"
    $StackPanel.Children.Add($TextBlockProject)

    $TextBoxProject = New-Object System.Windows.Controls.TextBox
    $TextBoxProject.Width = 400
    $TextBoxProject.Height = 30
    $TextBoxProject.Margin = "0,0,0,20"
    $TextBoxProject.FontSize = 14
    $TextBoxProject.Background = "#F5F5F5"
    $TextBoxProject.BorderBrush = "#BDBDBD"
    $TextBoxProject.Text = $QueueProject  # Set default value from parameter
    $TextBoxProject.Foreground = if ([string]::IsNullOrWhiteSpace($QueueProject)) { "Gray" } else { "Black" }
    $StackPanel.Children.Add($TextBoxProject)

    # Custom Queue Input (Queue Name Part 3)
    $TextBlockCustomQueue = New-Object System.Windows.Controls.TextBlock
    $TextBlockCustomQueue.Text = "Enter custom queue name ('Q prefix will be added):"
    $TextBlockCustomQueue.FontSize = 16
    $TextBlockCustomQueue.FontWeight = "Bold"
    $TextBlockCustomQueue.Margin = "0,10,0,5"
    $TextBlockCustomQueue.Foreground = "White"
    $StackPanel.Children.Add($TextBlockCustomQueue)

    $TextBoxCustomQueue = New-Object System.Windows.Controls.TextBox
    $TextBoxCustomQueue.Width = 400
    $TextBoxCustomQueue.Height = 30
    $TextBoxCustomQueue.Margin = "0,0,0,20"
    $TextBoxCustomQueue.FontSize = 14
    $TextBoxCustomQueue.Background = "#F5F5F5"
    $TextBoxCustomQueue.BorderBrush = "#BDBDBD"
    $TextBoxCustomQueue.Text = $QueueCustom  # Set default value from parameter
    $TextBoxCustomQueue.Foreground = if ([string]::IsNullOrWhiteSpace($QueueCustom)) { "Gray" } else { "Black" }
    $StackPanel.Children.Add($TextBoxCustomQueue)

    # Server Address Input
    $TextBlockServer = New-Object System.Windows.Controls.TextBlock
    $TextBlockServer.Text = "Enter the server address:"
    $TextBlockServer.FontSize = 16
    $TextBlockServer.FontWeight = "Bold"
    $TextBlockServer.Margin = "0,10,0,5"
    $TextBlockServer.Foreground = "White"
    $StackPanel.Children.Add($TextBlockServer)

    $TextBoxServer = New-Object System.Windows.Controls.TextBox
    $TextBoxServer.Width = 400
    $TextBoxServer.Height = 30
    $TextBoxServer.Margin = "0,0,0,20"
    $TextBoxServer.FontSize = 14
    $TextBoxServer.Background = "#F5F5F5"
    $TextBoxServer.BorderBrush = "#BDBDBD"
    $TextBoxServer.Foreground = "Gray"
    $TextBoxServer.Text = $ServerAddress
    $TextBoxServer.Add_GotFocus({
        if ($TextBoxServer.Foreground -eq "Gray") {
            $TextBoxServer.Text = ""
            $TextBoxServer.Foreground = "Black"
        }
    })
    $TextBoxServer.Add_LostFocus({
        if ([string]::IsNullOrWhiteSpace($TextBoxServer.Text)) {
            $TextBoxServer.Text = $ServerAddress
            $TextBoxServer.Foreground = "Gray"
        }
    })
    $StackPanel.Children.Add($TextBoxServer)

    # OK Button
    $Button = New-Object System.Windows.Controls.Button
    $Button.Content = "Install Printer"
    $Button.Width = 100
    $Button.Height = 40
    $Button.Background = "#2196F3"
    $Button.Foreground = "White"
    $Button.HorizontalAlignment = "Center"
    $StackPanel.Children.Add($Button)

    $Button.Add_Click({
        if ([string]::IsNullOrEmpty($script:SelectedDriver)) {
            [System.Windows.MessageBox]::Show("Please select a printer driver.", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
            return
        }

        # Build Queue Name from three parts
        $queueName = ""
        if (-not [string]::IsNullOrWhiteSpace($TextBoxUser.Text)) {
            $queueName += "'L$($TextBoxUser.Text.Trim())"
        }
        if (-not [string]::IsNullOrWhiteSpace($TextBoxProject.Text)) {
            $queueName += "'P$($TextBoxProject.Text.Trim())"
        }
        if (-not [string]::IsNullOrWhiteSpace($TextBoxCustomQueue.Text)) {
            $queueName += "'Q$($TextBoxCustomQueue.Text.Trim())"
        }

        Write-Debug "Generated Queue Name: $queueName"

        $Window.Tag = @{
            DeleteExisting = $CheckBoxDeletePrinter.IsChecked
            PrinterName = $TextBoxPrinterName.Text.Trim()
            Driver = $script:SelectedDriver
            ServerAddress = $TextBoxServer.Text.Trim()
            QueueName = $queueName
        }
        $Window.Close()
    })

    $Window.ShowDialog() | Out-Null
    return $Window.Tag
}

Write-Debug ""

# Main script execution
$DriverList = Get-FilteredPrinterDrivers
$inputResult = Show-CombinedInputDialog -DriverList $DriverList -ServerAddress $DefaultServerAddress

# Validate inputs
If ([string]::IsNullOrEmpty($inputResult.Driver)) {
    Write-Host "No printer driver selected."
    Exit
}
If ([string]::IsNullOrWhiteSpace($inputResult.QueueName)) {
    [System.Windows.MessageBox]::Show("Printer queue name cannot be empty.", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
    Exit
}
If ([string]::IsNullOrWhiteSpace($inputResult.ServerAddress)) {
    [System.Windows.MessageBox]::Show("Server address cannot be empty.", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
    Exit
}

# Check if printer exists and prompt to delete if requested
if (-not $inputResult.DeleteExisting) {
    if (Get-Printer | Where-Object {$_.Name -eq $inputResult.PrinterName}) {
        $result = [System.Windows.MessageBox]::Show("Printer '$($inputResult.PrinterName)' already exists. Do you want to delete it?", "Confirm Delete", [System.Windows.MessageBoxButton]::YesNo, [System.Windows.MessageBoxImage]::Question)
        if ($result -eq "Yes") {
            Write-Host "Deleting existing printer and port..."
            try {
                # Delete printer
                Remove-Printer -Name $inputResult.PrinterName -ErrorAction Stop
                Write-Debug "Printer '$($inputResult.PrinterName)' deleted successfully."

                # Delete printer port
                $portNameToDelete = "lpr://$($inputResult.ServerAddress)/$($inputResult.QueueName)"
                if (Get-PrinterPort | Where-Object {$_.Name -eq $portNameToDelete}) {
                    Remove-PrinterPort -Name $portNameToDelete -ErrorAction Stop
                    Write-Debug "Printer port '$portNameToDelete' deleted successfully."
                }
            } catch {
                Write-Warning "Failed to delete existing printer or port: $_"
                [System.Windows.MessageBox]::Show("Failed to delete existing printer or port: $_", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
            }
        }
    }
}

# Delete existing printer and port if requested
if ($inputResult.DeleteExisting) {
    Write-Host "Deleting existing printer and port..."
    try {
        # Delete printer
        if (Get-Printer | Where-Object {$_.Name -eq $inputResult.PrinterName}) {
            Remove-Printer -Name $inputResult.PrinterName -ErrorAction Stop
            Write-Debug "Printer '$($inputResult.PrinterName)' deleted successfully."
        }

        # Delete printer port
        $portNameToDelete = "lpr://$($inputResult.ServerAddress)/$($inputResult.QueueName)"
        if (Get-PrinterPort | Where-Object {$_.Name -eq $portNameToDelete}) {
            Remove-PrinterPort -Name $portNameToDelete -ErrorAction Stop
            Write-Debug "Printer port '$portNameToDelete' deleted successfully."
        }
    } catch {
        Write-Warning "Failed to delete existing printer or port: $_"
        [System.Windows.MessageBox]::Show("Failed to delete existing printer or port: $_", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
    }
}

# Add the LPR printer port and printer
$printerName = $inputResult.PrinterName
$queue       = $inputResult.QueueName # Queue name
$portName    = "$($printerName) $($queue)"

Write-Debug "Adding printer port with the following details:"
Write-Debug "Port Name      : $portName"
Write-Debug "LprHostAddress : $($inputResult.ServerAddress)"
Write-Debug "LprQueueName   : $queue"

Try {
    # Check if the port already exists
    $existingPort = Get-PrinterPort | Where-Object {$_.Name -eq $portName}
    if ($existingPort) {
        Write-Warning "Port '$portName' already exists. Attempting to delete it."
        Remove-PrinterPort -Name $portName -ErrorAction Stop
    }

    $printerAddPrinterPortArguments = @{
                  Name = $portName
        LprHostAddress = $inputResult.ServerAddress
          LprQueueName = $queue
    }

    # Write-Debug ""
    # Write-Debug @printerAddPrinterPortArguments
    # Write-Debug ""

    Add-PrinterPort @printerAddPrinterPortArguments


    Write-Debug "Printer port '$portName' added successfully."
} Catch {
    Write-Debug "Failed to add printer port. $_"
    [System.Windows.MessageBox]::Show("Failed to add printer port. $_", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
    Exit
}

Write-Debug "Adding printer with the following details:"
Write-Debug "Printer Name  : $printerName"
Write-Debug "Driver Name   : $($inputResult.Driver)"
Write-Debug "Port Name     : $portName"
Write-Debug "Server addr   : $($inputResult.ServerAddress)"
Write-Debug "Queue         : $($inputResult.QueueName)"

Write-Host "Adding printer : $printerName
        driver : $($inputResult.Driver)
   server addr : $($inputResult.ServerAddress)
         queue : $($inputResult.QueueName)"

Try {
    Add-Printer -Name           $printerName `
                -PortName       $portName `
                -DriverName     $inputResult.Driver `
                -ErrorAction    Stop
    [System.Windows.MessageBox]::Show("Printer '$printerName' added successfully.", "Success", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Information) | Out-Null
} Catch {
    [System.Windows.MessageBox]::Show("Failed to add printer. $_", "Error", [System.Windows.MessageBoxButton]::OK, [System.Windows.MessageBoxImage]::Error) | Out-Null
}