objectstore/*, repository: Add ReadSize v0.1.28

For cases where knowing the type is unnecessary and incurs extra
overhead.

2 files changed, 56 insertions, 0 deletions

diff --git a/objectstore/packed/read_size.go b/objectstore/packed/read_size.go
new file mode 100644
index 00000000..e162586a
--- /dev/null
+++ b/objectstore/packed/read_size.go
@@ -0,0 +1,39 @@
+package packed
+
+import (
+	"fmt"
+
+	packfmt "codeberg.org/lindenii/furgit/format/pack"
+	"codeberg.org/lindenii/furgit/objectid"
+	"codeberg.org/lindenii/furgit/objecttype"
+)
+
+// ReadSize reads an object's declared content size.
+func (store *Store) ReadSize(id objectid.ObjectID) (int64, error) {
+	loc, err := store.lookup(id)
+	if err != nil {
+		return 0, err
+	}
+	return store.resolveSizeAt(loc)
+}
+
+// resolveSizeAt resolves one object's declared content size from location.
+func (store *Store) resolveSizeAt(start location) (int64, error) {
+	pack, meta, err := store.entryMetaAt(start)
+	if err != nil {
+		return 0, err
+	}
+	if packfmt.IsBaseObjectType(meta.ty) {
+		return meta.size, nil
+	}
+	switch meta.ty {
+	case objecttype.TypeRefDelta, objecttype.TypeOfsDelta:
+		return deltaDeclaredSizeAt(pack, meta.dataOffset)
+	case objecttype.TypeInvalid, objecttype.TypeFuture:
+		return 0, fmt.Errorf("objectstore/packed: unsupported pack type %d", meta.ty)
+	case objecttype.TypeCommit, objecttype.TypeTree, objecttype.TypeBlob, objecttype.TypeTag:
+		return 0, fmt.Errorf("objectstore/packed: internal invariant violation for base type %d", meta.ty)
+	default:
+		return 0, fmt.Errorf("objectstore/packed: unsupported pack type %d", meta.ty)
+	}
+}
diff --git a/objectstore/packed/read_test.go b/objectstore/packed/read_test.go
index 8cee3b62..9bfa6610 100644
--- a/objectstore/packed/read_test.go
+++ b/objectstore/packed/read_test.go
@@ -36,6 +36,13 @@ func TestPackedStoreReadAgainstGit(t *testing.T) {
 				if gotHeaderSize != int64(len(wantBody)) {
 					t.Fatalf("ReadHeader size = %d, want %d", gotHeaderSize, len(wantBody))
 				}
+				gotSize, err := store.ReadSize(id)
+				if err != nil {
+					t.Fatalf("ReadSize: %v", err)
+				}
+				if gotSize != int64(len(wantBody)) {
+					t.Fatalf("ReadSize = %d, want %d", gotSize, len(wantBody))
+				}
 
 				gotRaw, err := store.ReadBytesFull(id)
 				if err != nil {
@@ -108,6 +115,9 @@ func TestPackedStoreErrors(t *testing.T) {
 		if _, _, err := store.ReadHeader(notFoundID); !errors.Is(err, objectstore.ErrObjectNotFound) {
 			t.Fatalf("ReadHeader not-found error = %v", err)
 		}
+		if _, err := store.ReadSize(notFoundID); !errors.Is(err, objectstore.ErrObjectNotFound) {
+			t.Fatalf("ReadSize not-found error = %v", err)
+		}
 
 		var otherAlgo objectid.Algorithm
 		for _, candidate := range objectid.SupportedAlgorithms() {
@@ -182,6 +192,13 @@ func TestPackedStoreReadHeaderUsesResolvedObjectSizeForDelta(t *testing.T) {
 		if gotSize != wantResolvedSize {
 			t.Fatalf("ReadHeader(%s) size = %d, want resolved size %d", deltaID, gotSize, wantResolvedSize)
 		}
+		gotReadSize, err := store.ReadSize(deltaID)
+		if err != nil {
+			t.Fatalf("ReadSize(%s): %v", deltaID, err)
+		}
+		if gotReadSize != wantResolvedSize {
+			t.Fatalf("ReadSize(%s) = %d, want resolved size %d", deltaID, gotReadSize, wantResolvedSize)
+		}
 	})
 }