# ====================== SSH Copy ID Tool (English Version) ======================
# Place this in your $PROFILE together with the SSH Host Management tools

<#
.SYNOPSIS
    PowerShell version of ssh-copy-id - Copy your SSH public key to a remote Linux server

.DESCRIPTION
    Exactly like the Linux ssh-copy-id command.
    Automatically creates ~/.ssh directory with correct permissions (700/600),
    supports user@host format, custom ports, force mode, and duplicate detection.

    Options (same as Linux):
    -i, --identity-file <file>   Public key file (auto-detects ed25519 > rsa by default)
    -p, --port <port>            SSH port (default 22)
    -o, --ssh-option <option>    Extra options passed to ssh
    -f, --force                  Force add key even if it already exists
    -h, --help, -?               Show this help message
.EXAMPLE
    ssh-copy-id user@192.168.1.100
    ssh-copy-id -p 2222 root@server.com -i ~/.ssh/id_ed25519.pub
    ssh-copy-id --force user@host
    ssh-copy-id --help
#>
function ssh-copy-id {
    [CmdletBinding()]
    param(
        [Parameter(Position = 0)]
        [string]$Destination,

        [Alias('i', '--identity-file')]
        [string]$IdentityFile,

        [Alias('p', '--port')]
        [int]$Port = 22,

        [Alias('o', '--ssh-option')]
        [string[]]$SshOption = @(),

        [Alias('f', '--force')]
        [switch]$Force,

        [Alias('h', '--help', '?')]
        [switch]$Help
    )

    # ====================== Help priority ======================
    if ($Help -or [string]::IsNullOrWhiteSpace($Destination)) {
        Get-Help $MyInvocation.MyCommand.Name -Detailed
        return
    }

    # ====================== Parse user@host ======================
    if ($Destination -match '^(.+)@(.+)$') {
        $UserName = $matches[1]
        $HostName = $matches[2]
    } else {
        Write-Host "ssh-copy-id: You must specify a username!" -ForegroundColor Red
        Write-Host "Correct format example: ssh-copy-id user@host" -ForegroundColor Cyan
        Write-Host "Show full help: ssh-copy-id --help" -ForegroundColor Cyan
        return
    }

    # ====================== Auto-select default public key ======================
    if (-not $IdentityFile) {
        $possibleKeys = @(
            "$env:USERPROFILE\.ssh\id_ed25519.pub",
            "$env:USERPROFILE\.ssh\id_rsa.pub",
            "$env:USERPROFILE\.ssh\id_ecdsa.pub"
        )
        foreach ($key in $possibleKeys) {
            if (Test-Path $key) {
                $IdentityFile = $key
                break
            }
        }
    }

    if (-not $IdentityFile -or -not (Test-Path $IdentityFile)) {
        Write-Error "Public key file not found!`nRun ssh-keygen first or specify with -i parameter."
        return
    }

    # ====================== Read public key ======================
    $pubKey = Get-Content -Path $IdentityFile -Raw -ErrorAction Stop

    # ====================== Build remote command ======================
    $forceString = if ($Force) { "true" } else { "false" }
    
    $remoteBashScript = @"
mkdir -p ~/.ssh && chmod 700 ~/.ssh
if [ "$forceString" = "false" ] && grep -qF "$($pubKey.Trim())" ~/.ssh/authorized_keys 2>/dev/null; then
    echo "Key already exists, skipping."
    exit 0
fi
echo "Adding new public key..."
echo "$($pubKey.Trim())" >> ~/.ssh/authorized_keys
chmod 600 ~/.ssh/authorized_keys
echo "✅ Public key has been successfully added to ~/.ssh/authorized_keys"
"@

    # Strip Windows Carriage Returns (CRLF -> LF) before encoding
    $remoteBashScript = $remoteBashScript.Replace("`r", "")

    # Convert script to Base64 to prevent quoting/escape character issues over SSH
    $scriptBytes = [System.Text.Encoding]::UTF8.GetBytes($remoteBashScript)
    $base64Script = [Convert]::ToBase64String($scriptBytes)

    # Remote shell decodes the payload and pipes it directly into bash
    # Supports both Linux (base64 -d) and macOS (base64 -D)
    $sshCmd = "echo '$base64Script' | { base64 -d 2>/dev/null || base64 -D 2>/dev/null; } | bash"

    # ====================== Execute ssh ======================
    Write-Host "Copying public key to ${UserName}@${HostName} (port $Port) ..." -ForegroundColor Green

    $sshArgs = @()
    if ($Port -ne 22) { $sshArgs += "-p", $Port }
    if ($SshOption.Count -gt 0) { $sshArgs += $SshOption }

    & ssh $sshArgs "${UserName}@${HostName}" $sshCmd

    if ($LASTEXITCODE -eq 0) {
        Write-Host "🎉 Operation completed successfully!" -ForegroundColor Green
        Write-Host "   You can now log in without a password:" -ForegroundColor Cyan
        Write-Host "   ssh $UserName@$HostName" -ForegroundColor Cyan
    } else {
        Write-Warning "⚠️  Execution failed. Please check network, password, or SSH server settings."
    }
}

Write-Host "✅ ssh-copy-id loaded successfully!" -ForegroundColor Cyan