From 20b340d01af7608f656f1f5be27c6c6657271c3b Mon Sep 17 00:00:00 2001
From: Erik Unger <erik@erikunger.com>
Date: Tue, 19 Nov 2024 16:14:40 +0100
Subject: [PATCH] added ReadHeader and ReadHeaderString

---
 utils.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/utils.go b/utils.go
index bcf234a..cbc1750 100644
--- a/utils.go
+++ b/utils.go
@@ -2,6 +2,7 @@ package fs
 
 import (
 	"context"
+	"errors"
 	"io"
 	"os"
 	"runtime"
@@ -70,6 +71,44 @@ func FileInfoToFileCallback(fileCallback func(File) error) func(*FileInfo) error
 	}
 }
 
+// ReadHeader reads up to maxNumBytes from the beginning of the passed FileReader.
+// If fr is a MemFile then a slice of its FileData is returned
+// without copying it.
+func ReadHeader(fr FileReader, maxNumBytes int) ([]byte, error) {
+	if maxNumBytes <= 0 {
+		return nil, nil
+	}
+
+	// Fast path for MemFile
+	if memFile, ok := fr.(MemFile); ok {
+		if len(memFile.FileData) <= maxNumBytes {
+			return memFile.FileData, nil
+		}
+		return memFile.FileData[:maxNumBytes], nil
+	}
+
+	// Generic case
+	r, err := fr.OpenReader()
+	if err != nil {
+		return nil, err
+	}
+	defer r.Close()
+
+	data := make([]byte, maxNumBytes)
+	n, err := r.Read(data)
+	if err != nil && !errors.Is(err, io.EOF) {
+		return data[:n], err
+	}
+	return data[:n], nil
+}
+
+// ReadHeaderString reads up to maxNumBytes as string from the beginning of the passed FileReader.
+// If fr is a MemFile then a slice of its FileData will be used as fast path.
+func ReadHeaderString(fr FileReader, maxNumBytes int) (string, error) {
+	data, err := ReadHeader(fr, maxNumBytes)
+	return string(data), err
+}
+
 // ReadAllContext reads all data from r until EOF is reached,
 // another error is returned, or the context got canceled.
 // It is identical to io.ReadAll except that