aboutsummaryrefslogtreecommitdiffstats
path: root/common/object.go
diff options
context:
space:
mode:
authorMáximo Cuadros <mcuadros@gmail.com>2015-10-25 20:30:36 +0100
committerMáximo Cuadros <mcuadros@gmail.com>2015-10-25 20:30:36 +0100
commit9a44cd8ccff143a112436c38bfe5581e74b68f07 (patch)
treef4d2f38cc61647bf159a7c870913e6f6b60828b2 /common/object.go
parentbe69a505926451bf10450ac68d40265a6f43e150 (diff)
downloadgo-git-9a44cd8ccff143a112436c38bfe5581e74b68f07.tar.gz
formats/packfile: new reader API
Diffstat (limited to 'common/object.go')
-rw-r--r--common/object.go95
1 files changed, 95 insertions, 0 deletions
diff --git a/common/object.go b/common/object.go
new file mode 100644
index 0000000..60c44da
--- /dev/null
+++ b/common/object.go
@@ -0,0 +1,95 @@
+package common
+
+import (
+ "bytes"
+ "io"
+)
+
+// Object is a generic representation of any git object
+type Object interface {
+ Type() ObjectType
+ SetType(ObjectType)
+ Size() int64
+ SetSize(int64)
+ Hash() Hash
+ Reader() io.Reader
+ Writer() io.Writer
+}
+
+// ObjectStorage generic storage of objects
+type ObjectStorage interface {
+ New() Object
+ Set(Object) Hash
+ Get(Hash) (Object, bool)
+}
+
+// ObjectType internal object type's
+type ObjectType int8
+
+const (
+ CommitObject ObjectType = 1
+ TreeObject ObjectType = 2
+ BlobObject ObjectType = 3
+ TagObject ObjectType = 4
+ OFSDeltaObject ObjectType = 6
+ REFDeltaObject ObjectType = 7
+)
+
+func (t ObjectType) String() string {
+ switch t {
+ case CommitObject:
+ return "commit"
+ case TreeObject:
+ return "tree"
+ case BlobObject:
+ return "blob"
+ default:
+ return "-"
+ }
+}
+
+func (t ObjectType) Bytes() []byte {
+ return []byte(t.String())
+}
+
+type RAWObject struct {
+ b []byte
+ t ObjectType
+ s int64
+}
+
+func (o *RAWObject) Type() ObjectType { return o.t }
+func (o *RAWObject) SetType(t ObjectType) { o.t = t }
+func (o *RAWObject) Size() int64 { return o.s }
+func (o *RAWObject) SetSize(s int64) { o.s = s }
+func (o *RAWObject) Reader() io.Reader { return bytes.NewBuffer(o.b) }
+func (o *RAWObject) Hash() Hash { return ComputeHash(o.t, o.b) }
+func (o *RAWObject) Writer() io.Writer { return o }
+func (o *RAWObject) Write(p []byte) (n int, err error) {
+ o.b = append(o.b, p...)
+ return len(p), nil
+}
+
+type RAWObjectStorage struct {
+ Objects map[Hash]*RAWObject
+}
+
+func NewRAWObjectStorage() *RAWObjectStorage {
+ return &RAWObjectStorage{make(map[Hash]*RAWObject, 0)}
+}
+
+func (o *RAWObjectStorage) New() Object {
+ return &RAWObject{}
+}
+
+func (o *RAWObjectStorage) Set(obj Object) Hash {
+ h := obj.Hash()
+ o.Objects[h] = obj.(*RAWObject)
+
+ return h
+}
+
+func (o *RAWObjectStorage) Get(h Hash) (Object, bool) {
+ obj, ok := o.Objects[h]
+ return obj, ok
+}