refactor: move repo and delta in their own package

1 files changed, 113 insertions, 0 deletions

diff --git a/repo/chunk.go b/repo/chunk.go
new file mode 100644
index 0000000..618cc9b
--- /dev/null
+++ b/repo/chunk.go
@@ -0,0 +1,113 @@
+package repo
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"path/filepath"
+)
+
+type Chunk interface {
+	Reader() io.ReadSeeker
+	Len() int
+}
+
+type IdentifiedChunk interface {
+	Chunk
+	GetId() *ChunkId
+}
+
+type BufferedChunk interface {
+	Chunk
+	Bytes() []byte
+}
+
+type RepoChunk interface {
+	Chunk
+	SetRepo(r *Repo)
+}
+
+type ChunkId struct {
+	Ver int
+	Idx uint64
+}
+
+func (i *ChunkId) Path(repo string) string {
+	return filepath.Join(repo, fmt.Sprintf(versionFmt, i.Ver), chunksName, fmt.Sprintf(chunkIdFmt, i.Idx))
+}
+
+func NewStoredChunk(repo *Repo, id *ChunkId) *StoredChunk {
+	return &StoredChunk{repo: repo, Id: id}
+}
+
+type StoredChunk struct {
+	repo *Repo
+	Id   *ChunkId
+}
+
+func (c *StoredChunk) GetId() *ChunkId {
+	return c.Id
+}
+
+func (c *StoredChunk) SetRepo(r *Repo) {
+	c.repo = r
+}
+
+func (c *StoredChunk) Reader() io.ReadSeeker {
+	// log.Printf("Chunk %d: Reading from file\n", c.id)
+	return c.repo.LoadChunkContent(c.Id)
+}
+
+func (c *StoredChunk) Len() int {
+	return c.repo.chunkSize
+}
+
+func NewTempChunk(value []byte) *TempChunk {
+	return &TempChunk{Value: value}
+}
+
+type TempChunk struct {
+	Value []byte
+}
+
+func (c *TempChunk) Reader() io.ReadSeeker {
+	return bytes.NewReader(c.Value)
+}
+
+func (c *TempChunk) Len() int {
+	return len(c.Value)
+}
+
+func (c *TempChunk) Bytes() []byte {
+	return c.Value
+}
+
+func (c *TempChunk) AppendFrom(r io.Reader) {
+	buff, err := io.ReadAll(r)
+	if err != nil {
+		println("Chunk: error appending to temp chunk:", err)
+	}
+	c.Value = append(c.Value, buff...)
+}
+
+type DeltaChunk struct {
+	repo   *Repo
+	Source *ChunkId
+	Patch  []byte
+	Size   int
+}
+
+func (c *DeltaChunk) SetRepo(r *Repo) {
+	c.repo = r
+}
+
+func (c *DeltaChunk) Reader() io.ReadSeeker {
+	var buff bytes.Buffer
+	c.repo.Patcher().Patch(c.repo.LoadChunkContent(c.Source), &buff, bytes.NewReader(c.Patch))
+	return bytes.NewReader(buff.Bytes())
+}
+
+// TODO: Maybe return the size of the patch instead ?
+func (c *DeltaChunk) Len() int {
+	return c.Size
+}