New version

npmproxy/internalProxy.go

@@ -0,0 +1,67 @@
+package npmproxy
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"path/filepath"
+	"time"
+
+	"github.com/trading-peter/gowebbuild/fsutils"
+)
+
+func (p *Proxy) internalHTTPServer(ctx context.Context) {
+	mux := http.NewServeMux()
+
+	srv := &http.Server{
+		Addr:    p.internalProxyHost,
+		Handler: mux,
+	}
+
+	go func() {
+		<-ctx.Done()
+		shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		if err := srv.Shutdown(shutdownCtx); err != nil {
+			fmt.Printf("Failed to shutdown internal server for npm proxy: %v\n", err)
+		}
+	}()
+
+	mux.HandleFunc("GET /{pkg}", func(w http.ResponseWriter, r *http.Request) {
+		pkgName := r.PathValue("pkg")
+		override, ok := p.matchingOverride(pkgName)
+
+		if !ok {
+			http.NotFound(w, r)
+			return
+		}
+
+		pkg, err := p.findPackageSource(override, pkgName)
+		if err != nil {
+			serveReverseProxy(override.Upstream, w, r)
+			return
+		}
+
+		json.NewEncoder(w).Encode(pkg)
+	})
+
+	mux.HandleFunc("GET /files/{file}", func(w http.ResponseWriter, r *http.Request) {
+		fileName := r.PathValue("file")
+		filePath := filepath.Join(p.pkgCachePath, fileName)
+
+		if !fsutils.IsFile(filePath) {
+			http.NotFound(w, r)
+			return
+		}
+
+		http.ServeFile(w, r, filePath)
+	})
+
+	if err := srv.ListenAndServe(); err != nil {
+		if err != http.ErrServerClosed {
+			panic(err)
+		}
+	}
+}