Function NewWebSocketProxyWithSessionAndLogger

internal/vnc/proxy.go:69–113

NewWebSocketProxyWithSessionAndLogger creates a new WebSocket proxy with session notifications and shared logger.

(config *ProxyConfig, session SessionNotifier, sharedLogger *logger.Logger)

// NewWebSocketProxyWithSessionAndLogger creates a new WebSocket proxy with session notifications and shared logger.
func NewWebSocketProxyWithSessionAndLogger(config *ProxyConfig, session SessionNotifier, sharedLogger *logger.Logger) *WebSocketProxy {
	var proxyLogger *logger.Logger

	if sharedLogger != nil {
		proxyLogger = sharedLogger
	} else {
		// Use the global logger system for unified logging
		proxyLogger = logger.GetPackageLoggerConcrete("vnc-proxy")
	}

	proxyLogger.Info("Creating new WebSocket proxy for %s (Type: %s, Node: %s)",
		getTargetName(config), config.VMType, config.NodeName)
	proxyLogger.Debug("Proxy config - Port: %s, Proxmox Host: %s", config.Port, config.ProxmoxHost)

	return &WebSocketProxy{
		config:  config,
		logger:  proxyLogger,
		session: session,
		upgrader: websocket.Upgrader{
			CheckOrigin: func(r *http.Request) bool {
				// Allow connections from localhost only for security
				origin := r.Header.Get("Origin")
				if origin == "" {
					proxyLogger.Debug("WebSocket connection with no origin header (non-browser)")

					return true // Allow non-browser connections
				}
				u, err := url.Parse(origin)
				if err != nil {
					proxyLogger.Error("Failed to parse origin header: %s", origin)

					return false
				}
				allowed := u.Hostname() == "localhost" || u.Hostname() == "127.0.0.1"
				if !allowed {
					proxyLogger.Error("WebSocket connection rejected from origin: %s", origin)
				} else {
					proxyLogger.Debug("WebSocket connection allowed from origin: %s", origin)
				}

				return allowed
			},
		},
	}
}

// getTargetName returns a descriptive name for the target (VM name or node name).
func getTargetName(config *ProxyConfig) string {