Initial commit 🌈

2023-10-16 21:31:51 +02:00
commit da2c53bf86
15 changed files with 802 additions and 0 deletions
--- a/internal/dav/router.go
+++ b/internal/dav/router.go
@@ -0,0 +1,127 @@
+package dav
+
+import (
+	"encoding/xml"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"path"
+	"strings"
+
+	"github.com/debridmediamanager.com/zurg/pkg/dav"
+	"github.com/debridmediamanager.com/zurg/pkg/realdebrid"
+	"github.com/debridmediamanager.com/zurg/pkg/repo"
+)
+
+func findTorrentByFilename(torrents []realdebrid.Torrent, filename string) *realdebrid.Torrent {
+	for _, torrent := range torrents {
+		if torrent.Filename == filename {
+			return &torrent
+		}
+	}
+	return nil
+}
+
+func Router(mux *http.ServeMux, db *repo.Database) {
+	torrents, err := realdebrid.GetTorrents(os.Getenv("RD_TOKEN"))
+	if err != nil {
+		log.Printf("Cannot get torrents: %v", err.Error())
+		return
+	}
+
+	rootResponse := dav.MultiStatus{
+		XMLNS: "DAV:",
+		Response: []dav.Response{
+			dav.Directory("/"),
+			dav.Directory("/torrents"),
+		},
+	}
+
+	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		requestPath := path.Clean(r.URL.Path)
+
+		switch r.Method {
+		case "PROPFIND":
+			log.Println("PROPFIND", requestPath)
+			var output []byte
+			var err error
+
+			if requestPath == "/" {
+				output, err = xml.MarshalIndent(rootResponse, "", "  ")
+			} else if requestPath == "/torrents" {
+				var allTorrentsResponse *dav.MultiStatus
+				allTorrentsResponse, err = createMultiTorrentResponse(torrents)
+				if err != nil {
+					log.Printf("Cannot read directory: %v", err.Error())
+					http.Error(w, fmt.Sprintf("Cannot read directory: %v", err.Error()), http.StatusInternalServerError)
+					return
+				}
+				output, err = xml.MarshalIndent(allTorrentsResponse, "", "  ")
+			} else {
+				lastSegment := path.Base(requestPath)
+				torrent := findTorrentByFilename(torrents, lastSegment)
+				if torrent == nil {
+					log.Println("Cannot find directory")
+					http.Error(w, "Cannot find directory", http.StatusNotFound)
+					return
+				}
+
+				var torrentResponse *dav.MultiStatus
+				torrentResponse, err = createSingleTorrentResponse(*torrent, db)
+				if err != nil {
+					log.Printf("Cannot read directory: %v", err.Error())
+					http.Error(w, fmt.Sprintf("Cannot read directory: %v", err.Error()), http.StatusInternalServerError)
+					return
+				}
+				output, err = xml.MarshalIndent(torrentResponse, "", "  ")
+			}
+
+			if err != nil {
+				log.Printf("Cannot marshal xml: %v", err.Error())
+				http.Error(w, fmt.Sprintf("Cannot marshal xml: %v", err.Error()), http.StatusInternalServerError)
+				return
+			}
+
+			w.Header().Set("Content-Type", "text/xml; charset=\"utf-8\"")
+			w.WriteHeader(http.StatusMultiStatus)
+			fmt.Fprintf(w, "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n%s\n", output)
+
+		case http.MethodOptions:
+			log.Println("OPTIONS", requestPath)
+			w.WriteHeader(http.StatusOK)
+
+		case http.MethodGet:
+			log.Println("GET", requestPath)
+			segments := strings.Split(requestPath, "/")
+
+			// If there are less than 2 segments, return an error or adjust as needed
+			if len(segments) < 2 {
+				log.Println("Cannot find file")
+				http.Error(w, "Cannot find file", http.StatusNotFound)
+			}
+
+			// Get the last two segments
+			secondLast := segments[len(segments)-2]
+			last := segments[len(segments)-1]
+			unrestrict, dbErr := db.Get(secondLast, last)
+			if dbErr != nil {
+				log.Printf("Cannot find file in db: %v", dbErr.Error())
+				http.Error(w, fmt.Sprintf("Cannot find file in db: %v", dbErr.Error()), http.StatusInternalServerError)
+				return
+			}
+
+			resp, err := realdebrid.UnrestrictLink(os.Getenv("RD_TOKEN"), unrestrict.Link)
+			if err != nil {
+				log.Printf("Cannot unrestrict link: %v", err.Error())
+				http.Error(w, fmt.Sprintf("Cannot unrestrict link: %v", err.Error()), http.StatusInternalServerError)
+				return
+			}
+			http.Redirect(w, r, resp.Download, http.StatusFound)
+
+		default:
+			log.Println("Method not implemented")
+			http.Error(w, "Method not implemented", http.StatusMethodNotAllowed)
+		}
+	})
+}