clients and remote: fetch implementation

2 files changed, 64 insertions, 7 deletions

diff --git a/clients/common/common.go b/clients/common/common.go
index 3527d7c..c3d15c1 100644
--- a/clients/common/common.go
+++ b/clients/common/common.go
@@ -2,6 +2,7 @@ package common
 
 import (
 	"fmt"
+	"io/ioutil"
 	"net/url"
 	"strings"
 
@@ -72,9 +73,14 @@ func (r Capabilities) SymbolicReference(sym string) string {
 	return ""
 }
 
+type RemoteHead struct {
+	Id   string
+	Name string
+}
+
 type GitUploadPackInfo struct {
 	Capabilities Capabilities
-	Branches     map[string]string
+	Refs         map[string]*RemoteHead
 }
 
 func NewGitUploadPackInfo(d *pktline.Decoder) (*GitUploadPackInfo, error) {
@@ -92,7 +98,7 @@ func (r *GitUploadPackInfo) read(d *pktline.Decoder) error {
 		return err
 	}
 
-	r.Branches = map[string]string{}
+	r.Refs = map[string]*RemoteHead{}
 	for _, line := range lines {
 		if !r.isValidLine(line) {
 			continue
@@ -118,15 +124,41 @@ func (r *GitUploadPackInfo) isValidLine(line string) bool {
 }
 
 func (r *GitUploadPackInfo) readLine(line string) {
-	commit, branch := r.getCommitAndBranch(line)
-	r.Branches[branch] = commit
+	rh := r.getRemoteHead(line)
+	r.Refs[rh.Name] = rh
 }
 
-func (r *GitUploadPackInfo) getCommitAndBranch(line string) (string, string) {
+func (r *GitUploadPackInfo) getRemoteHead(line string) *RemoteHead {
 	parts := strings.Split(strings.Trim(line, " \n"), " ")
 	if len(parts) != 2 {
-		return "", ""
+		return nil
 	}
 
-	return parts[0], parts[1]
+	return &RemoteHead{parts[0], parts[1]}
+}
+
+type GitUploadPackRequest struct {
+	Want []string
+	Have []string
+}
+
+func (r *GitUploadPackRequest) String() string {
+	b, _ := ioutil.ReadAll(r.Reader())
+	return string(b)
+}
+
+func (r *GitUploadPackRequest) Reader() *strings.Reader {
+	e := pktline.NewEncoder()
+	for _, want := range r.Want {
+		e.AddLine(fmt.Sprintf("want %s", want))
+	}
+
+	for _, have := range r.Have {
+		e.AddLine(fmt.Sprintf("have %s", have))
+	}
+
+	e.AddFlush()
+	e.AddLine("done")
+
+	return e.GetReader()
 }
diff --git a/clients/common/common_test.go b/clients/common/common_test.go
index 0b9b60a..37d4ff9 100644
--- a/clients/common/common_test.go
+++ b/clients/common/common_test.go
@@ -1,9 +1,12 @@
 package common
 
 import (
+	"bytes"
+	"encoding/base64"
 	"testing"
 
 	. "gopkg.in/check.v1"
+	"gopkg.in/src-d/go-git.v2/pktline"
 )
 
 func Test(t *testing.T) { TestingT(t) }
@@ -35,3 +38,25 @@ func (s *SuiteCommon) TestCapabilitiesSymbolicReference(c *C) {
 	cap := parseCapabilities(CapabilitiesFixture)
 	c.Assert(cap.SymbolicReference("HEAD"), Equals, "refs/heads/master")
 }
+
+const GitUploadPackInfoFixture = "MDAxZSMgc2VydmljZT1naXQtdXBsb2FkLXBhY2sKMDAwMDAxMGM2ZWNmMGVmMmMyZGZmYjc5NjAzM2U1YTAyMjE5YWY4NmVjNjU4NGU1IEhFQUQAbXVsdGlfYWNrIHRoaW4tcGFjayBzaWRlLWJhbmQgc2lkZS1iYW5kLTY0ayBvZnMtZGVsdGEgc2hhbGxvdyBuby1wcm9ncmVzcyBpbmNsdWRlLXRhZyBtdWx0aV9hY2tfZGV0YWlsZWQgbm8tZG9uZSBzeW1yZWY9SEVBRDpyZWZzL2hlYWRzL21hc3RlciBhZ2VudD1naXQvMjoyLjQuOH5kYnVzc2luay1maXgtZW50ZXJwcmlzZS10b2tlbnMtY29tcGlsYXRpb24tMTE2Ny1nYzcwMDZjZgowMDNmZThkM2ZmYWI1NTI4OTVjMTliOWZjZjdhYTI2NGQyNzdjZGUzMzg4MSByZWZzL2hlYWRzL2JyYW5jaAowMDNmNmVjZjBlZjJjMmRmZmI3OTYwMzNlNWEwMjIxOWFmODZlYzY1ODRlNSByZWZzL2hlYWRzL21hc3RlcgowMDNlYjhlNDcxZjU4YmNiY2E2M2IwN2JkYTIwZTQyODE5MDQwOWMyZGI0NyByZWZzL3B1bGwvMS9oZWFkCjAwMDA="
+
+func (s *SuiteCommon) TestGitUploadPackInfo(c *C) {
+	b, _ := base64.StdEncoding.DecodeString(GitUploadPackInfoFixture)
+	info, err := NewGitUploadPackInfo(pktline.NewDecoder(bytes.NewBuffer(b)))
+	c.Assert(err, IsNil)
+
+	ref := info.Capabilities.SymbolicReference("HEAD")
+	c.Assert(ref, Equals, "refs/heads/master")
+	c.Assert(info.Refs[ref].Id, Equals, "6ecf0ef2c2dffb796033e5a02219af86ec6584e5")
+	c.Assert(info.Refs[ref].Name, Equals, "refs/heads/master")
+}
+
+func (s *SuiteCommon) TestGitUploadPackRequest(c *C) {
+	r := &GitUploadPackRequest{
+		Want: []string{"foo", "qux"},
+		Have: []string{"bar"},
+	}
+
+	c.Assert(r.String(), Equals, "000dwant foo\n000dwant qux\n000dhave bar\n00000009done\n")
+}