fixes

2026-06-10 23:29:22 +00:00 · 2026-01-02 04:02:02 +00:00
commit 3b8e171fdb
117 changed files with 21570 additions and 0 deletions
@@ -0,0 +1,290 @@
+package goproxy
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+
+	"github.com/lukaszraczylo/gohoarder/pkg/cache"
+	"github.com/lukaszraczylo/gohoarder/pkg/network"
+	"github.com/rs/zerolog/log"
+)
+
+// Handler implements the GOPROXY protocol
+type Handler struct {
+	cache    *cache.Manager
+	client   *network.Client
+	upstream string
+	sumDBURL string
+}
+
+// Config holds Go proxy configuration
+type Config struct {
+	Upstream string // Upstream Go proxy (e.g., proxy.golang.org)
+	SumDBURL string // Checksum database URL
+}
+
+// New creates a new Go proxy handler
+func New(cacheManager *cache.Manager, client *network.Client, config Config) *Handler {
+	if config.Upstream == "" {
+		config.Upstream = "https://proxy.golang.org"
+	}
+
+	if config.SumDBURL == "" {
+		config.SumDBURL = "https://sum.golang.org"
+	}
+
+	return &Handler{
+		cache:    cacheManager,
+		client:   client,
+		upstream: config.Upstream,
+		sumDBURL: config.SumDBURL,
+	}
+}
+
+// ServeHTTP handles GOPROXY protocol requests
+func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	// Path is already stripped by http.StripPrefix in app.go
+	path := r.URL.Path
+
+	log.Debug().
+		Str("path", path).
+		Msg("Processing Go proxy request")
+
+	// Parse GOPROXY request
+	// Formats:
+	// /@v/list - list versions
+	// /@v/$version.info - version info
+	// /@v/$version.mod - go.mod file
+	// /@v/$version.zip - module zip
+	// /@latest - latest version
+
+	log.Debug().Str("path", path).Msg("Go proxy request")
+
+	// Route request based on path
+	if strings.HasPrefix(path, "/sumdb/") {
+		h.handleSumDB(ctx, w, r, path)
+	} else if strings.HasSuffix(path, "/@v/list") {
+		h.handleList(ctx, w, r, path)
+	} else if strings.Contains(path, "/@v/") && strings.HasSuffix(path, ".info") {
+		h.handleInfo(ctx, w, r, path)
+	} else if strings.Contains(path, "/@v/") && strings.HasSuffix(path, ".mod") {
+		h.handleMod(ctx, w, r, path)
+	} else if strings.Contains(path, "/@v/") && strings.HasSuffix(path, ".zip") {
+		h.handleZip(ctx, w, r, path)
+	} else if strings.HasSuffix(path, "/@latest") {
+		h.handleLatest(ctx, w, r, path)
+	} else {
+		http.Error(w, "Invalid Go proxy request", http.StatusBadRequest)
+	}
+}
+
+// handleList handles /@v/list requests
+func (h *Handler) handleList(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	url := h.upstream + path
+	modulePath := h.extractModulePath(path)
+
+	entry, err := h.cache.Get(ctx, "go", modulePath, "list", func(ctx context.Context) (io.ReadCloser, string, error) {
+		body, statusCode, err := h.client.Get(ctx, url, nil)
+		if err != nil {
+			return nil, "", err
+		}
+		if statusCode != http.StatusOK {
+			body.Close()
+			return nil, "", fmt.Errorf("upstream returned status %d", statusCode)
+		}
+		return body, url, nil
+	})
+
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch version list")
+		http.Error(w, "Failed to fetch version list", http.StatusBadGateway)
+		return
+	}
+	defer entry.Data.Close()
+
+	w.Header().Set("Content-Type", "text/plain; charset=UTF-8")
+	io.Copy(w, entry.Data)
+}
+
+// handleInfo handles /@v/$version.info requests
+func (h *Handler) handleInfo(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	url := h.upstream + path
+	modulePath := h.extractModulePath(path)
+	version := h.extractVersion(path, ".info")
+	// Use .info suffix to distinguish from .mod and .zip in cache
+	cacheKey := modulePath + "/@v/" + version + ".info"
+
+	entry, err := h.cache.Get(ctx, "go", cacheKey, version, func(ctx context.Context) (io.ReadCloser, string, error) {
+		body, statusCode, err := h.client.Get(ctx, url, nil)
+		if err != nil {
+			return nil, "", err
+		}
+		if statusCode != http.StatusOK {
+			body.Close()
+			return nil, "", fmt.Errorf("upstream returned status %d", statusCode)
+		}
+		return body, url, nil
+	})
+
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch version info")
+		http.Error(w, "Failed to fetch version info", http.StatusBadGateway)
+		return
+	}
+	defer entry.Data.Close()
+
+	w.Header().Set("Content-Type", "application/json; charset=UTF-8")
+	io.Copy(w, entry.Data)
+}
+
+// handleMod handles /@v/$version.mod requests
+func (h *Handler) handleMod(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	url := h.upstream + path
+	modulePath := h.extractModulePath(path)
+	version := h.extractVersion(path, ".mod")
+	// Use .mod suffix to distinguish from .info and .zip in cache
+	cacheKey := modulePath + "/@v/" + version + ".mod"
+
+	entry, err := h.cache.Get(ctx, "go", cacheKey, version, func(ctx context.Context) (io.ReadCloser, string, error) {
+		body, statusCode, err := h.client.Get(ctx, url, nil)
+		if err != nil {
+			return nil, "", err
+		}
+		if statusCode != http.StatusOK {
+			body.Close()
+			return nil, "", fmt.Errorf("upstream returned status %d", statusCode)
+		}
+		return body, url, nil
+	})
+
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch go.mod")
+		http.Error(w, "Failed to fetch go.mod", http.StatusBadGateway)
+		return
+	}
+	defer entry.Data.Close()
+
+	w.Header().Set("Content-Type", "text/plain; charset=UTF-8")
+	io.Copy(w, entry.Data)
+}
+
+// handleZip handles /@v/$version.zip requests
+func (h *Handler) handleZip(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	url := h.upstream + path
+	modulePath := h.extractModulePath(path)
+	version := h.extractVersion(path, ".zip")
+	// Use .zip suffix to distinguish from .info and .mod in cache
+	cacheKey := modulePath + "/@v/" + version + ".zip"
+
+	entry, err := h.cache.Get(ctx, "go", cacheKey, version, func(ctx context.Context) (io.ReadCloser, string, error) {
+		body, statusCode, err := h.client.Get(ctx, url, nil)
+		if err != nil {
+			return nil, "", err
+		}
+		if statusCode != http.StatusOK {
+			body.Close()
+			return nil, "", fmt.Errorf("upstream returned status %d", statusCode)
+		}
+		return body, url, nil
+	})
+
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch module zip")
+		http.Error(w, "Failed to fetch module zip", http.StatusBadGateway)
+		return
+	}
+	defer entry.Data.Close()
+
+	w.Header().Set("Content-Type", "application/zip")
+	io.Copy(w, entry.Data)
+}
+
+// handleLatest handles /@latest requests
+func (h *Handler) handleLatest(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	url := h.upstream + path
+	modulePath := h.extractModulePath(path)
+
+	entry, err := h.cache.Get(ctx, "go", modulePath, "latest", func(ctx context.Context) (io.ReadCloser, string, error) {
+		body, statusCode, err := h.client.Get(ctx, url, nil)
+		if err != nil {
+			return nil, "", err
+		}
+		if statusCode != http.StatusOK {
+			body.Close()
+			return nil, "", fmt.Errorf("upstream returned status %d", statusCode)
+		}
+		return body, url, nil
+	})
+
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch latest version")
+		http.Error(w, "Failed to fetch latest version", http.StatusBadGateway)
+		return
+	}
+	defer entry.Data.Close()
+
+	w.Header().Set("Content-Type", "application/json; charset=UTF-8")
+	io.Copy(w, entry.Data)
+}
+
+// handleSumDB handles sumdb requests (checksum database)
+func (h *Handler) handleSumDB(ctx context.Context, w http.ResponseWriter, r *http.Request, path string) {
+	// path format: /sumdb/sum.golang.org/...
+	// Remove /sumdb/ prefix and proxy to sumdb URL
+	sumdbPath := strings.TrimPrefix(path, "/sumdb/sum.golang.org")
+	url := h.sumDBURL + sumdbPath
+
+	log.Debug().Str("url", url).Msg("Proxying sumdb request")
+
+	// Sumdb requests should not be cached, proxy directly
+	body, statusCode, err := h.client.Get(ctx, url, nil)
+	if err != nil {
+		log.Error().Err(err).Str("url", url).Msg("Failed to fetch from sumdb")
+		http.Error(w, "Failed to fetch from sumdb", http.StatusBadGateway)
+		return
+	}
+	defer body.Close()
+
+	if statusCode != http.StatusOK {
+		log.Error().Int("status", statusCode).Str("url", url).Msg("Sumdb returned non-OK status")
+		http.Error(w, "Sumdb error", statusCode)
+		return
+	}
+
+	w.Header().Set("Content-Type", "text/plain; charset=UTF-8")
+	io.Copy(w, body)
+}
+
+// extractVersion extracts version from path
+func (h *Handler) extractVersion(path, suffix string) string {
+	// path format: /module/path/@v/v1.2.3.suffix
+	parts := strings.Split(path, "/@v/")
+	if len(parts) != 2 {
+		return ""
+	}
+	return strings.TrimSuffix(parts[1], suffix)
+}
+
+// extractModulePath extracts the clean module path from a GOPROXY path
+// Examples:
+//
+//	/github.com/avast/retry-go/v4/@v/v4.6.1.zip -> github.com/avast/retry-go/v4
+//	/golang.org/x/net/@v/v0.40.0.mod -> golang.org/x/net
+//	/github.com/user/repo/@v/list -> github.com/user/repo
+func (h *Handler) extractModulePath(path string) string {
+	// Remove leading slash
+	path = strings.TrimPrefix(path, "/")
+
+	// Split on /@v/ to get the module path
+	parts := strings.Split(path, "/@v/")
+	if len(parts) > 0 {
+		return parts[0]
+	}
+
+	// Fallback: remove /@latest suffix if present
+	return strings.TrimSuffix(path, "/@latest")
+}