diff options
Diffstat (limited to 'internal')
-rw-r--r-- | internal/hash.go | 35 | ||||
-rw-r--r-- | internal/hash_test.go | 27 | ||||
-rw-r--r-- | internal/object.go | 113 |
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 +} |