internal/format/packrev: Add basic writer HEAD main

TODO: extract stickywriter to its own package

1 files changed, 101 insertions, 0 deletions

diff --git a/internal/format/packrev/write.go b/internal/format/packrev/write.go
new file mode 100644
index 00000000..a30f89ad
--- /dev/null
+++ b/internal/format/packrev/write.go
@@ -0,0 +1,101 @@
+package packrev
+
+import (
+	"bufio"
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"io"
+	"math"
+
+	"lindenii.org/go/furgit/object/id"
+)
+
+// ErrInvalidPositions reports that
+// positions supplied for a reverse index write
+// are out of range or too numerous.
+var ErrInvalidPositions = errors.New("internal/format/packrev: invalid positions")
+
+// Write writes one pack reverse index to w.
+//
+// positions holds, for each object in pack offset order,
+// the object's pack index position.
+// packHash must be the pack's trailer hash;
+// Write panics when its length does not match the object format.
+func Write(w io.Writer, objectFormat id.ObjectFormat, positions []uint32, packHash []byte) error {
+	hashID, err := hashFunctionID(objectFormat)
+	if err != nil {
+		return err
+	}
+
+	if len(packHash) != objectFormat.Size() {
+		panic("internal/format/packrev: invalid pack hash length")
+	}
+
+	if len(positions) > math.MaxUint32 {
+		return fmt.Errorf("%w: too many positions", ErrInvalidPositions)
+	}
+
+	for _, position := range positions {
+		if uint64(position) >= uint64(len(positions)) {
+			return fmt.Errorf("%w: index position out of range", ErrInvalidPositions)
+		}
+	}
+
+	hashImpl, err := objectFormat.New()
+	if err != nil {
+		return fmt.Errorf("internal/format/packrev: %w", err)
+	}
+
+	bw := bufio.NewWriter(io.MultiWriter(w, hashImpl))
+	sw := &stickyWriter{w: bw}
+
+	sw.writeUint32(signature)
+	sw.writeUint32(version)
+	sw.writeUint32(hashID)
+
+	for _, position := range positions {
+		sw.writeUint32(position)
+	}
+
+	sw.write(packHash)
+
+	if sw.err != nil {
+		return fmt.Errorf("internal/format/packrev: %w", sw.err)
+	}
+
+	err = bw.Flush()
+	if err != nil {
+		return fmt.Errorf("internal/format/packrev: %w", err)
+	}
+
+	_, err = w.Write(hashImpl.Sum(nil))
+	if err != nil {
+		return fmt.Errorf("internal/format/packrev: %w", err)
+	}
+
+	return nil
+}
+
+// stickyWriter forwards writes to w
+// and retains the first error,
+// turning subsequent writes into no-ops.
+type stickyWriter struct {
+	w   io.Writer
+	err error
+}
+
+func (sw *stickyWriter) write(p []byte) {
+	if sw.err != nil {
+		return
+	}
+
+	_, sw.err = sw.w.Write(p)
+}
+
+func (sw *stickyWriter) writeUint32(v uint32) {
+	var buf [4]byte
+
+	binary.BigEndian.PutUint32(buf[:], v)
+	sw.write(buf[:])
+}