diff options
Diffstat (limited to 'plumbing/transport/fetch_pack.go')
-rw-r--r-- | plumbing/transport/fetch_pack.go | 177 |
1 files changed, 0 insertions, 177 deletions
diff --git a/plumbing/transport/fetch_pack.go b/plumbing/transport/fetch_pack.go deleted file mode 100644 index 14d3e19..0000000 --- a/plumbing/transport/fetch_pack.go +++ /dev/null @@ -1,177 +0,0 @@ -package transport - -import ( - "bytes" - "fmt" - "io" - "io/ioutil" - "strings" - - "gopkg.in/src-d/go-git.v4/plumbing" - "gopkg.in/src-d/go-git.v4/plumbing/format/pktline" - "gopkg.in/src-d/go-git.v4/plumbing/protocol/packp" - "gopkg.in/src-d/go-git.v4/plumbing/storer" - "gopkg.in/src-d/go-git.v4/storage/memory" -) - -//TODO: Replace this by packpAdvRefs. -type UploadPackInfo struct { - Capabilities *packp.Capabilities - Refs memory.ReferenceStorage -} - -func NewUploadPackInfo() *UploadPackInfo { - return &UploadPackInfo{ - Capabilities: packp.NewCapabilities(), - Refs: make(memory.ReferenceStorage, 0), - } -} - -func (i *UploadPackInfo) Decode(r io.Reader) error { - d := packp.NewAdvRefsDecoder(r) - ar := packp.NewAdvRefs() - if err := d.Decode(ar); err != nil { - if err == packp.ErrEmpty { - return err - } - return plumbing.NewUnexpectedError(err) - } - - i.Capabilities = ar.Capabilities - - if err := i.addRefs(ar); err != nil { - return plumbing.NewUnexpectedError(err) - } - - return nil -} - -func (i *UploadPackInfo) addRefs(ar *packp.AdvRefs) error { - for name, hash := range ar.References { - ref := plumbing.NewReferenceFromStrings(name, hash.String()) - i.Refs.SetReference(ref) - } - - return i.addSymbolicRefs(ar) -} - -func (i *UploadPackInfo) addSymbolicRefs(ar *packp.AdvRefs) error { - if !hasSymrefs(ar) { - return nil - } - - for _, symref := range ar.Capabilities.Get("symref").Values { - chunks := strings.Split(symref, ":") - if len(chunks) != 2 { - err := fmt.Errorf("bad number of `:` in symref value (%q)", symref) - return plumbing.NewUnexpectedError(err) - } - name := plumbing.ReferenceName(chunks[0]) - target := plumbing.ReferenceName(chunks[1]) - ref := plumbing.NewSymbolicReference(name, target) - i.Refs.SetReference(ref) - } - - return nil -} - -func hasSymrefs(ar *packp.AdvRefs) bool { - return ar.Capabilities.Supports("symref") -} - -func (i *UploadPackInfo) Head() *plumbing.Reference { - ref, _ := storer.ResolveReference(i.Refs, plumbing.HEAD) - return ref -} - -func (i *UploadPackInfo) String() string { - return string(i.Bytes()) -} - -func (i *UploadPackInfo) Bytes() []byte { - var buf bytes.Buffer - e := pktline.NewEncoder(&buf) - - _ = e.EncodeString("# service=git-upload-pack\n") - - // inserting a flush-pkt here violates the protocol spec, but some - // servers do it, like Github.com - e.Flush() - - _ = e.Encodef("%s HEAD\x00%s\n", i.Head().Hash(), i.Capabilities.String()) - - for _, ref := range i.Refs { - if ref.Type() != plumbing.HashReference { - continue - } - - _ = e.Encodef("%s %s\n", ref.Hash(), ref.Name()) - } - - e.Flush() - - return buf.Bytes() -} - -type UploadPackRequest struct { - Wants []plumbing.Hash - Haves []plumbing.Hash - Depth int -} - -func (r *UploadPackRequest) Want(h ...plumbing.Hash) { - r.Wants = append(r.Wants, h...) -} - -func (r *UploadPackRequest) Have(h ...plumbing.Hash) { - r.Haves = append(r.Haves, h...) -} - -func (r *UploadPackRequest) IsEmpty() bool { - return isSubset(r.Wants, r.Haves) -} - -func isSubset(needle []plumbing.Hash, haystack []plumbing.Hash) bool { - for _, h := range needle { - found := false - for _, oh := range haystack { - if h == oh { - found = true - break - } - } - - if !found { - return false - } - } - - return true -} - -func (r *UploadPackRequest) String() string { - b, _ := ioutil.ReadAll(r.Reader()) - return string(b) -} - -func (r *UploadPackRequest) Reader() *strings.Reader { - var buf bytes.Buffer - e := pktline.NewEncoder(&buf) - - for _, want := range r.Wants { - _ = e.Encodef("want %s\n", want) - } - - for _, have := range r.Haves { - _ = e.Encodef("have %s\n", have) - } - - if r.Depth != 0 { - _ = e.Encodef("deepen %d\n", r.Depth) - } - - _ = e.Flush() - _ = e.EncodeString("done\n") - - return strings.NewReader(buf.String()) -} |