123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package lfs
-
- import (
- "bytes"
- "context"
- "errors"
- "fmt"
- "net/http"
- "net/url"
- "strings"
-
- "code.gitea.io/gitea/modules/json"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/proxy"
- )
-
- const batchSize = 20
-
- // HTTPClient is used to communicate with the LFS server
- // https://github.com/git-lfs/git-lfs/blob/main/docs/api/batch.md
- type HTTPClient struct {
- client *http.Client
- endpoint string
- transfers map[string]TransferAdapter
- }
-
- // BatchSize returns the preferred size of batchs to process
- func (c *HTTPClient) BatchSize() int {
- return batchSize
- }
-
- func newHTTPClient(endpoint *url.URL, httpTransport *http.Transport) *HTTPClient {
- if httpTransport == nil {
- httpTransport = &http.Transport{
- Proxy: proxy.Proxy(),
- }
- }
-
- hc := &http.Client{
- Transport: httpTransport,
- }
-
- client := &HTTPClient{
- client: hc,
- endpoint: strings.TrimSuffix(endpoint.String(), "/"),
- transfers: make(map[string]TransferAdapter),
- }
-
- basic := &BasicTransferAdapter{hc}
-
- client.transfers[basic.Name()] = basic
-
- return client
- }
-
- func (c *HTTPClient) transferNames() []string {
- keys := make([]string, len(c.transfers))
-
- i := 0
- for k := range c.transfers {
- keys[i] = k
- i++
- }
-
- return keys
- }
-
- func (c *HTTPClient) batch(ctx context.Context, operation string, objects []Pointer) (*BatchResponse, error) {
- log.Trace("BATCH operation with objects: %v", objects)
-
- url := fmt.Sprintf("%s/objects/batch", c.endpoint)
-
- request := &BatchRequest{operation, c.transferNames(), nil, objects}
-
- payload := new(bytes.Buffer)
- err := json.NewEncoder(payload).Encode(request)
- if err != nil {
- log.Error("Error encoding json: %v", err)
- return nil, err
- }
-
- log.Trace("Calling: %s", url)
-
- req, err := http.NewRequestWithContext(ctx, "POST", url, payload)
- if err != nil {
- log.Error("Error creating request: %v", err)
- return nil, err
- }
- req.Header.Set("Content-type", MediaType)
- req.Header.Set("Accept", MediaType)
-
- res, err := c.client.Do(req)
- if err != nil {
- select {
- case <-ctx.Done():
- return nil, ctx.Err()
- default:
- }
- log.Error("Error while processing request: %v", err)
- return nil, err
- }
- defer res.Body.Close()
-
- if res.StatusCode != http.StatusOK {
- return nil, fmt.Errorf("Unexpected server response: %s", res.Status)
- }
-
- var response BatchResponse
- err = json.NewDecoder(res.Body).Decode(&response)
- if err != nil {
- log.Error("Error decoding json: %v", err)
- return nil, err
- }
-
- if len(response.Transfer) == 0 {
- response.Transfer = "basic"
- }
-
- return &response, nil
- }
-
- // Download reads the specific LFS object from the LFS server
- func (c *HTTPClient) Download(ctx context.Context, objects []Pointer, callback DownloadCallback) error {
- return c.performOperation(ctx, objects, callback, nil)
- }
-
- // Upload sends the specific LFS object to the LFS server
- func (c *HTTPClient) Upload(ctx context.Context, objects []Pointer, callback UploadCallback) error {
- return c.performOperation(ctx, objects, nil, callback)
- }
-
- func (c *HTTPClient) performOperation(ctx context.Context, objects []Pointer, dc DownloadCallback, uc UploadCallback) error {
- if len(objects) == 0 {
- return nil
- }
-
- operation := "download"
- if uc != nil {
- operation = "upload"
- }
-
- result, err := c.batch(ctx, operation, objects)
- if err != nil {
- return err
- }
-
- transferAdapter, ok := c.transfers[result.Transfer]
- if !ok {
- return fmt.Errorf("TransferAdapter not found: %s", result.Transfer)
- }
-
- for _, object := range result.Objects {
- if object.Error != nil {
- objectError := errors.New(object.Error.Message)
- log.Trace("Error on object %v: %v", object.Pointer, objectError)
- if uc != nil {
- if _, err := uc(object.Pointer, objectError); err != nil {
- return err
- }
- } else {
- if err := dc(object.Pointer, nil, objectError); err != nil {
- return err
- }
- }
- continue
- }
-
- if uc != nil {
- if len(object.Actions) == 0 {
- log.Trace("%v already present on server", object.Pointer)
- continue
- }
-
- link, ok := object.Actions["upload"]
- if !ok {
- log.Debug("%+v", object)
- return errors.New("Missing action 'upload'")
- }
-
- content, err := uc(object.Pointer, nil)
- if err != nil {
- return err
- }
-
- err = transferAdapter.Upload(ctx, link, object.Pointer, content)
-
- content.Close()
-
- if err != nil {
- return err
- }
-
- link, ok = object.Actions["verify"]
- if ok {
- if err := transferAdapter.Verify(ctx, link, object.Pointer); err != nil {
- return err
- }
- }
- } else {
- link, ok := object.Actions["download"]
- if !ok {
- log.Debug("%+v", object)
- return errors.New("Missing action 'download'")
- }
-
- content, err := transferAdapter.Download(ctx, link)
- if err != nil {
- return err
- }
-
- if err := dc(object.Pointer, content, nil); err != nil {
- return err
- }
- }
- }
-
- return nil
- }
|