read token from config

internal/dav/aero.go

@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"log"
 	"net/http"
-	"os"
 	"strings"
 
 	"github.com/aerogo/aero"
@@ -87,7 +86,7 @@ func Setup(app *aero.Application, c config.ConfigInterface, t *torrent.TorrentMa
 			}
 
 			unrestrictFn := func() (*realdebrid.UnrestrictResponse, error) {
-				return realdebrid.UnrestrictLink(os.Getenv("RD_TOKEN"), link)
+				return realdebrid.UnrestrictLink(c.GetToken(), link)
 			}
 			resp := realdebrid.RetryUntilOk(unrestrictFn)
 			if resp == nil {

internal/dav/getfile.go

@@ -4,18 +4,18 @@ import (
 	"fmt"
 	"log"
 	"net/http"
-	"os"
 	"path"
 	"path/filepath"
 	"strings"
 
+	"github.com/debridmediamanager.com/zurg/internal/config"
 	"github.com/debridmediamanager.com/zurg/internal/torrent"
 	"github.com/debridmediamanager.com/zurg/pkg/davextra"
 	"github.com/debridmediamanager.com/zurg/pkg/realdebrid"
 )
 
 // HandleGetRequest handles a GET request to a file
-func HandleGetRequest(w http.ResponseWriter, r *http.Request, t *torrent.TorrentManager) {
+func HandleGetRequest(w http.ResponseWriter, r *http.Request, t *torrent.TorrentManager, c config.ConfigInterface) {
 	requestPath := path.Clean(r.URL.Path)
 
 	segments := strings.Split(requestPath, "/")
@@ -48,13 +48,15 @@ func HandleGetRequest(w http.ResponseWriter, r *http.Request, t *torrent.Torrent
 	}
 
 	unrestrictFn := func() (*realdebrid.UnrestrictResponse, error) {
-		return realdebrid.UnrestrictLink(os.Getenv("RD_TOKEN"), link)
+		return realdebrid.UnrestrictLink(c.GetToken(), link)
 	}
 	resp := realdebrid.RetryUntilOk(unrestrictFn)
 	if resp == nil {
 		// TODO: Readd the file
 		// when unrestricting fails, it means the file is not available anymore, but still in their database
 		// if it's the only file, tough luck
+		// if it's the only file, try to readd it
+		// delete the old one, add a new one
 		log.Println("Cannot unrestrict link", link, filenameV2)
 		http.Error(w, "Cannot find file", http.StatusNotFound)
 		return

internal/dav/router.go

@@ -16,7 +16,7 @@ func Router(mux *http.ServeMux, c config.ConfigInterface, t *torrent.TorrentMana
 			HandlePropfindRequest(w, r, t, c)
 
 		case http.MethodGet:
-			HandleGetRequest(w, r, t)
+			HandleGetRequest(w, r, t, c)
 
 		case http.MethodOptions:
 			w.WriteHeader(http.StatusOK)