Try reading rar contents

internal/rar/contents.go

@@ -0,0 +1,69 @@
+package rar
+
+import (
+	"fmt"
+	"net/http"
+	"strings"
+
+	zurghttp "github.com/debridmediamanager/zurg/pkg/http"
+	"github.com/debridmediamanager/zurg/pkg/logutil"
+	"github.com/nwaples/rardecode"
+)
+
+type RarReader struct {
+	client *zurghttp.HTTPClient
+	log    *logutil.Logger
+}
+
+func NewRarReader(client *zurghttp.HTTPClient, log *logutil.Logger) *RarReader {
+	return &RarReader{
+		client: client,
+		log:    log,
+	}
+}
+
+// GetRARContents extracts the file names from a RAR archive.
+func (r *RarReader) GetRARContents(downloadLink string) ([]string, error) {
+	dlReq, err := http.NewRequest(http.MethodGet, downloadLink, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	downloadResp, err := r.client.Do(dlReq)
+	if err != nil {
+		return nil, err
+	}
+	defer downloadResp.Body.Close()
+
+	if downloadResp.StatusCode/100 != 2 {
+		return nil, fmt.Errorf("received a %s status code", downloadResp.Status)
+	}
+
+	var fileList []string
+
+	rarReader, err := rardecode.NewReader(downloadResp.Body, "")
+	if err != nil {
+		return nil, err
+	}
+
+	for {
+		r.log.Debugf("Reading next file from RAR archive")
+		header, err := rarReader.Next()
+		if err != nil {
+			if strings.Contains(err.Error(), "unexpected EOF") {
+				break
+			}
+			return nil, err
+		}
+
+		if header == nil {
+			break
+		}
+
+		fileList = append(fileList, header.Name)
+		r.log.Debugf("Found file: %s", header.Name)
+		r.log.Debugf("File size: %d MB", header.PackedSize/1024/1024)
+	}
+
+	return fileList, nil
+}