97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package universal
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/debridmediamanager.com/zurg/internal/config"
|
|
"github.com/debridmediamanager.com/zurg/internal/torrent"
|
|
"github.com/debridmediamanager.com/zurg/pkg/logutil"
|
|
"github.com/hashicorp/golang-lru/v2/expirable"
|
|
)
|
|
|
|
func HandleHeadRequest(w http.ResponseWriter, r *http.Request, t *torrent.TorrentManager, c config.ConfigInterface, cache *expirable.LRU[string, string]) {
|
|
rlog := logutil.NewLogger()
|
|
log := rlog.Named("unihead")
|
|
|
|
requestPath := path.Clean(r.URL.Path)
|
|
requestPath = strings.Replace(requestPath, "/http", "", 1)
|
|
if requestPath == "/favicon.ico" {
|
|
return
|
|
}
|
|
|
|
segments := strings.Split(requestPath, "/")
|
|
// If there are less than 3 segments, return an error or adjust as needed
|
|
if len(segments) < 4 {
|
|
log.Errorf("Request %s %s not supported yet", r.Method, r.URL.Path)
|
|
http.Error(w, "Method not implemented", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
if data, exists := cache.Get("head:" + requestPath); exists {
|
|
splits := strings.Split(data, " ")
|
|
contentType := splits[0]
|
|
contentLength := splits[1]
|
|
w.Header().Set("Content-Type", contentType)
|
|
w.Header().Set("Content-Length", contentLength)
|
|
w.WriteHeader(http.StatusOK)
|
|
return
|
|
}
|
|
|
|
baseDirectory := segments[len(segments)-3]
|
|
torrentName := segments[len(segments)-2]
|
|
filename := segments[len(segments)-1]
|
|
|
|
torrents := t.FindAllTorrentsWithName(baseDirectory, torrentName)
|
|
if torrents == nil {
|
|
log.Errorf("Cannot find torrent %s in the directory %s", requestPath, baseDirectory)
|
|
http.Error(w, "Cannot find file", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
_, file := getFile(torrents, filename)
|
|
if file == nil {
|
|
log.Errorf("Cannot find file from path %s", requestPath)
|
|
http.Error(w, "Cannot find file", http.StatusNotFound)
|
|
return
|
|
}
|
|
if file.Link == "" {
|
|
// This is a dead file, serve an alternate file
|
|
log.Errorf("File %s is no longer available", filename)
|
|
http.Error(w, "Cannot find file", http.StatusNotFound)
|
|
return
|
|
}
|
|
contentType := getContentMimeType(filename)
|
|
contentLength := fmt.Sprintf("%d", file.Bytes)
|
|
w.Header().Set("Content-Type", contentType)
|
|
w.Header().Set("Content-Length", contentLength)
|
|
cache.Add("head:"+requestPath, contentType+" "+contentLength)
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func getContentMimeType(filePath string) string {
|
|
switch filepath.Ext(filePath) {
|
|
case ".mkv":
|
|
return "video/x-matroska"
|
|
case ".mp4":
|
|
return "video/mp4"
|
|
case ".avi":
|
|
return "video/x-msvideo"
|
|
case ".mp3":
|
|
return "audio/mpeg"
|
|
case ".rar":
|
|
return "application/x-rar-compressed"
|
|
case ".zip":
|
|
return "application/zip"
|
|
case ".7z":
|
|
return "application/x-7z-compressed"
|
|
case ".srt":
|
|
return "text/plain"
|
|
default:
|
|
return "application/octet-stream"
|
|
}
|
|
}
|