From aa11889601166a40cfd9b1c478c7133045a052bf Mon Sep 17 00:00:00 2001
From: Ben Adrian Sarmiento <me@bensarmiento.com>
Date: Wed, 5 Jun 2024 09:31:50 +0200
Subject: [PATCH] Only copy the content-range header

---
 internal/torrent/dump.go         | 4 ++--
 internal/universal/downloader.go | 8 +++-----
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/internal/torrent/dump.go b/internal/torrent/dump.go
index d02fb0b..7009506 100644
--- a/internal/torrent/dump.go
+++ b/internal/torrent/dump.go
@@ -3,9 +3,9 @@ package torrent
 import "github.com/debridmediamanager/zurg/internal/config"
 
 func (t *TorrentManager) loadDumpedTorrents() {
-	// TODO: Paywall
+	// TODO: Paywall?
 	count := 0
-	t.getTorrentFiles("data").Each(func(filePath string) bool {
+	t.getTorrentFiles("dump").Each(func(filePath string) bool {
 		torrent := t.readTorrentFromFile(filePath)
 		if torrent != nil {
 			accessKey := t.GetKey(torrent)
diff --git a/internal/universal/downloader.go b/internal/universal/downloader.go
index 995d88a..33e7d03 100644
--- a/internal/universal/downloader.go
+++ b/internal/universal/downloader.go
@@ -177,11 +177,9 @@ func (dl *Downloader) streamFileToResponse(
 		return
 	}
 
-	// Copy the headers from the download response to the response
-	for k, vv := range downloadResp.Header {
-		for _, v := range vv {
-			resp.Header().Add(k, v)
-		}
+	// Copy the content-range header from the download response to the response
+	if cr := downloadResp.Header.Get("Content-Range"); cr != "" {
+		resp.Header().Set("Content-Range", cr)
 	}
 
 	buf := make([]byte, cfg.GetNetworkBufferSize())