45 lines
1.1 KiB
Go
45 lines
1.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) {
|
|
ref := chi.URLParam(r, "repo_ref")
|
|
|
|
var repo *store.Repo
|
|
var err error
|
|
if id, parseErr := strconv.ParseInt(ref, 10, 64); parseErr == nil {
|
|
repo, err = h.repoSvc.Get(r.Context(), id)
|
|
} else {
|
|
repo, err = h.repoSvc.GetByName(r.Context(), ref)
|
|
}
|
|
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, repo.ID, repo.Type)
|
|
prefix := fmt.Sprintf("/mirror/%s", ref)
|
|
http.StripPrefix(prefix, http.FileServer(http.Dir(localDir))).ServeHTTP(w, r)
|
|
}
|