56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ewpt3ch/pkgstash/internal/cache"
|
|
)
|
|
|
|
func (s *Server) handlePackage(w http.ResponseWriter, req *http.Request) {
|
|
|
|
// most mirrors don't have a *db.sig so we 404 it here instead of spamming the mirror
|
|
if strings.HasSuffix(req.PathValue("file"), ".db.sig") {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
// record the useragent from requestor
|
|
log.Printf("Requestors UA: %s", req.Header.Get("User-Agent"))
|
|
|
|
// build file paths from the request, they follow archlinux repo
|
|
// <mirrorroot>/[core, extra, etc]/os/[x86_64, arm, etc]/package.pkg.tar.zst[.sig]
|
|
repo := req.PathValue("repo")
|
|
arch := req.PathValue("arch")
|
|
file := req.PathValue("file")
|
|
repoPath := filepath.Join(repo, "os", arch, file) //path from mirror root to pkg or db file
|
|
|
|
cachedFile, err := s.c.Fetch(repoPath)
|
|
if err != nil {
|
|
var upstreamErr *cache.UpstreamError
|
|
if errors.As(err, &upstreamErr) {
|
|
log.Printf("upstream error: %v", err)
|
|
http.Error(w, "Not found upstream", upstreamErr.StatusCode)
|
|
return
|
|
}
|
|
log.Printf("fetch error: %v", err)
|
|
http.Error(w, "Failed to fetch from upstream", http.StatusBadGateway)
|
|
return
|
|
}
|
|
defer cachedFile.Reader.Close()
|
|
|
|
w.Header().Set("Content-Type", "application/octet-stream")
|
|
w.Header().Set("Content-Disposition", "attachment; filename="+cachedFile.Filename)
|
|
w.Header().Set("Content-Length", strconv.FormatInt(cachedFile.Size, 10))
|
|
_, err = io.Copy(w, cachedFile.Reader)
|
|
if err != nil {
|
|
log.Printf("error streaming file to client: %v", err)
|
|
}
|
|
|
|
}
|