12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package lfs
-
- import (
- "bytes"
- "context"
- "io"
- "net/http"
-
- "code.gitea.io/gitea/modules/json"
- "code.gitea.io/gitea/modules/log"
- )
-
- // TransferAdapter represents an adapter for downloading/uploading LFS objects.
- type TransferAdapter interface {
- Name() string
- Download(ctx context.Context, l *Link) (io.ReadCloser, error)
- Upload(ctx context.Context, l *Link, p Pointer, r io.Reader) error
- Verify(ctx context.Context, l *Link, p Pointer) error
- }
-
- // BasicTransferAdapter implements the "basic" adapter.
- type BasicTransferAdapter struct {
- client *http.Client
- }
-
- // Name returns the name of the adapter.
- func (a *BasicTransferAdapter) Name() string {
- return "basic"
- }
-
- // Download reads the download location and downloads the data.
- func (a *BasicTransferAdapter) Download(ctx context.Context, l *Link) (io.ReadCloser, error) {
- req, err := createRequest(ctx, http.MethodGet, l.Href, l.Header, nil)
- if err != nil {
- return nil, err
- }
- resp, err := performRequest(ctx, a.client, req)
- if err != nil {
- return nil, err
- }
- return resp.Body, nil
- }
-
- // Upload sends the content to the LFS server.
- func (a *BasicTransferAdapter) Upload(ctx context.Context, l *Link, p Pointer, r io.Reader) error {
- req, err := createRequest(ctx, http.MethodPut, l.Href, l.Header, r)
- if err != nil {
- return err
- }
- if req.Header.Get("Content-Type") == "" {
- req.Header.Set("Content-Type", "application/octet-stream")
- }
- if req.Header.Get("Transfer-Encoding") == "chunked" {
- req.TransferEncoding = []string{"chunked"}
- }
- req.ContentLength = p.Size
-
- res, err := performRequest(ctx, a.client, req)
- if err != nil {
- return err
- }
- defer res.Body.Close()
- return nil
- }
-
- // Verify calls the verify handler on the LFS server
- func (a *BasicTransferAdapter) Verify(ctx context.Context, l *Link, p Pointer) error {
- b, err := json.Marshal(p)
- if err != nil {
- log.Error("Error encoding json: %v", err)
- return err
- }
-
- req, err := createRequest(ctx, http.MethodPost, l.Href, l.Header, bytes.NewReader(b))
- if err != nil {
- return err
- }
- req.Header.Set("Content-Type", MediaType)
- res, err := performRequest(ctx, a.client, req)
- if err != nil {
- return err
- }
- defer res.Body.Close()
- return nil
- }
|