aboutsummaryrefslogtreecommitdiffstats
path: root/internal
diff options
context:
space:
mode:
Diffstat (limited to 'internal')
-rw-r--r--internal/hash.go35
-rw-r--r--internal/hash_test.go27
-rw-r--r--internal/object.go113
3 files changed, 175 insertions, 0 deletions
diff --git a/internal/hash.go b/internal/hash.go
new file mode 100644
index 0000000..db55b24
--- /dev/null
+++ b/internal/hash.go
@@ -0,0 +1,35 @@
+package internal
+
+import (
+ "crypto/sha1"
+ "encoding/hex"
+ "strconv"
+)
+
+// Hash SHA1 hased content
+type Hash [20]byte
+
+// ComputeHash compute the hash for a given ObjectType and content
+func ComputeHash(t ObjectType, content []byte) Hash {
+ h := t.Bytes()
+ h = append(h, ' ')
+ h = strconv.AppendInt(h, int64(len(content)), 10)
+ h = append(h, 0)
+ h = append(h, content...)
+
+ return Hash(sha1.Sum(h))
+}
+
+// NewHash return a new Hash from a hexadecimal hash representation
+func NewHash(s string) Hash {
+ b, _ := hex.DecodeString(s)
+
+ var h Hash
+ copy(h[:], b)
+
+ return h
+}
+
+func (h Hash) String() string {
+ return hex.EncodeToString(h[:])
+}
diff --git a/internal/hash_test.go b/internal/hash_test.go
new file mode 100644
index 0000000..222f3b4
--- /dev/null
+++ b/internal/hash_test.go
@@ -0,0 +1,27 @@
+package internal
+
+import (
+ "testing"
+
+ . "gopkg.in/check.v1"
+)
+
+func Test(t *testing.T) { TestingT(t) }
+
+type HashSuite struct{}
+
+var _ = Suite(&HashSuite{})
+
+func (s *HashSuite) TestComputeHash(c *C) {
+ hash := ComputeHash(BlobObject, []byte(""))
+ c.Assert(hash.String(), Equals, "e69de29bb2d1d6434b8b29ae775ad8c2e48c5391")
+
+ hash = ComputeHash(BlobObject, []byte("Hello, World!\n"))
+ c.Assert(hash.String(), Equals, "8ab686eafeb1f44702738c8b0f24f2567c36da6d")
+}
+
+func (s *HashSuite) TestNewHash(c *C) {
+ hash := ComputeHash(BlobObject, []byte("Hello, World!\n"))
+
+ c.Assert(hash, Equals, NewHash(hash.String()))
+}
diff --git a/internal/object.go b/internal/object.go
new file mode 100644
index 0000000..0f11140
--- /dev/null
+++ b/internal/object.go
@@ -0,0 +1,113 @@
+package internal
+
+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
+ Commits map[Hash]*RAWObject
+ Trees map[Hash]*RAWObject
+ Blobs map[Hash]*RAWObject
+}
+
+func NewRAWObjectStorage() *RAWObjectStorage {
+ return &RAWObjectStorage{
+ Objects: make(map[Hash]*RAWObject, 0),
+ Commits: make(map[Hash]*RAWObject, 0),
+ Trees: make(map[Hash]*RAWObject, 0),
+ Blobs: 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)
+
+ switch obj.Type() {
+ case CommitObject:
+ o.Commits[h] = o.Objects[h]
+ case TreeObject:
+ o.Trees[h] = o.Objects[h]
+ case BlobObject:
+ o.Blobs[h] = o.Objects[h]
+ }
+
+ return h
+}
+
+func (o *RAWObjectStorage) Get(h Hash) (Object, bool) {
+ obj, ok := o.Objects[h]
+
+ return obj, ok
+}