llama-swap/proxy/ui_compress.go

package proxy

import (
	"net/http"
	"strings"
)

// selectEncoding chooses the best encoding based on Accept-Encoding header
// Returns the encoding ("br", "gzip", or "") and the corresponding file extension
func selectEncoding(acceptEncoding string) (encoding, ext string) {
	if acceptEncoding == "" {
		return "", ""
	}

	for _, part := range strings.Split(acceptEncoding, ",") {
		enc := strings.TrimSpace(strings.SplitN(part, ";", 2)[0])
		if enc == "br" {
			return "br", ".br"
		}
	}

	for _, part := range strings.Split(acceptEncoding, ",") {
		enc := strings.TrimSpace(strings.SplitN(part, ";", 2)[0])
		if enc == "gzip" {
			return "gzip", ".gz"
		}
	}

	return "", ""
}

// ServeCompressedFile serves a file with compression support.
// It checks for pre-compressed versions and serves them with proper headers.
func ServeCompressedFile(fs http.FileSystem, w http.ResponseWriter, r *http.Request, name string) {
	encoding, ext := selectEncoding(r.Header.Get("Accept-Encoding"))

	// Try to serve compressed version if client supports it
	if encoding != "" {
		if cf, err := fs.Open(name + ext); err == nil {
			defer cf.Close()

			// Verify it's a regular file (not a directory)
			if stat, err := cf.Stat(); err == nil && !stat.IsDir() {
				// Set the content encoding header
				w.Header().Set("Content-Encoding", encoding)
				w.Header().Add("Vary", "Accept-Encoding")

				// Get original file info for content type detection
				origFile, err := fs.Open(name)
				if err == nil {
					origFile.Close()
				}

				// Serve the compressed file
				http.ServeContent(w, r, name, stat.ModTime(), cf)
				return
			}
		}
	}

	// Fall back to serving the uncompressed file
	file, err := fs.Open(name)
	if err != nil {
		http.Error(w, err.Error(), http.StatusNotFound)
		return
	}
	defer file.Close()

	stat, err := file.Stat()
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	if stat.IsDir() {
		http.Error(w, "is a directory", http.StatusForbidden)
		return
	}

	http.ServeContent(w, r, name, stat.ModTime(), file)
}