43 lines
1 KiB
Go
43 lines
1 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/syonad/clonepack/internal/core"
|
|
"github.com/syonad/clonepack/internal/store"
|
|
)
|
|
|
|
type ProxyHandler struct {
|
|
repoSvc *core.RepoService
|
|
dataDir string
|
|
}
|
|
|
|
func NewProxyHandler(repoSvc *core.RepoService, dataDir string) *ProxyHandler {
|
|
return &ProxyHandler{repoSvc: repoSvc, dataDir: dataDir}
|
|
}
|
|
|
|
func (h *ProxyHandler) ServeFile(w http.ResponseWriter, r *http.Request) {
|
|
repoID, err := strconv.ParseInt(chi.URLParam(r, "repo_id"), 10, 64)
|
|
if err != nil {
|
|
http.Error(w, "invalid repo_id", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
repo, err := h.repoSvc.Get(r.Context(), repoID)
|
|
if errors.Is(err, store.ErrNotFound) {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
localDir := fmt.Sprintf("%s/repos/%d/%s", h.dataDir, repoID, repo.Type)
|
|
prefix := fmt.Sprintf("/mirror/%d", repoID)
|
|
http.StripPrefix(prefix, http.FileServer(http.Dir(localDir))).ServeHTTP(w, r)
|
|
}
|