package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "strings"
)

const version = "0.1"

var (
    overrideDisplay = flag.String("display", "", "Override DISPLAY")
    overrideSocket  = flag.String("proxy-socket", "/tmp/.X11-unix/X5", "Proxy socket path")
    showVersion     = flag.Bool("version", false, "Show version and exit")
)

func init() {
    flag.Usage = func() {
        fmt.Fprintf(os.Stderr, "Usage of %s:\n\n", os.Args[0])
        flag.PrintDefaults()
    }
}

func main() {
    flag.Parse()

    if *showVersion {
        fmt.Printf("X11Proxy version %s\n", version)
        return
    }
    fmt.Printf("version %s\n", version)

    display := os.Getenv("DISPLAY")
    if *overrideDisplay != "" {
        display = *overrideDisplay
    }

    connType, target := resolveDisplay(display)
    fmt.Printf("Proxying to %s (%s)\n", target, connTypeString(connType))

    err := StartProxy(*overrideSocket, target, connType, display)
    if err != nil {
        log.Fatalf("Proxy error: %v", err)
    }
}

func connTypeString(t ConnType) string {
    if t == Unix {
        return "Unix socket"
    }
    return "TCP"
}

func hexDump(buf []byte) string {
    var out strings.Builder
    for i := 0; i < len(buf); i += 16 {
        line := fmt.Sprintf("%08x  ", i)

        for j := 0; j < 16; j++ {
            if i+j < len(buf) {
                line += fmt.Sprintf("%02x ", buf[i+j])
            } else {
                line += "   "
            }
            if j == 7 {
                line += " "
            }
        }

        line += " |"

        for j := 0; j < 16 && i+j < len(buf); j++ {
            b := buf[i+j]
            if b >= 32 && b <= 126 {
                line += string(b)
            } else {
                line += "."
            }
        }

        line += "|\n"
        out.WriteString(line)
    }
    return out.String()
}