mirror of
https://github.com/opencloud-eu/opencloud.git
synced 2026-01-24 22:08:58 -05:00
4f26424db6
* enhancement: use reva client pool selectors register mock service to registry and pass tests * enhancement: bump reva * Fix a couple of linter issues --------- Co-authored-by: Ralf Haferkamp <rhaferkamp@owncloud.com>
107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
package imgsource
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
gateway "github.com/cs3org/go-cs3apis/cs3/gateway/v1beta1"
|
|
rpc "github.com/cs3org/go-cs3apis/cs3/rpc/v1beta1"
|
|
provider "github.com/cs3org/go-cs3apis/cs3/storage/provider/v1beta1"
|
|
revactx "github.com/cs3org/reva/v2/pkg/ctx"
|
|
"github.com/cs3org/reva/v2/pkg/rgrpc/todo/pool"
|
|
"github.com/cs3org/reva/v2/pkg/rhttp"
|
|
"github.com/cs3org/reva/v2/pkg/storagespace"
|
|
"github.com/owncloud/ocis/v2/services/thumbnails/pkg/config"
|
|
"github.com/pkg/errors"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
const (
|
|
// "github.com/cs3org/reva/v2/internal/http/services/datagateway" is internal so we redeclare it here
|
|
// TokenTransportHeader holds the header key for the reva transfer token
|
|
TokenTransportHeader = "X-Reva-Transfer"
|
|
)
|
|
|
|
// CS3 implements a CS3 image source
|
|
type CS3 struct {
|
|
gatewaySelector pool.Selectable[gateway.GatewayAPIClient]
|
|
insecure bool
|
|
}
|
|
|
|
// NewCS3Source configures a new CS3 image source
|
|
func NewCS3Source(cfg config.Thumbnail, gatewaySelector pool.Selectable[gateway.GatewayAPIClient]) CS3 {
|
|
return CS3{
|
|
gatewaySelector: gatewaySelector,
|
|
insecure: cfg.CS3AllowInsecure,
|
|
}
|
|
}
|
|
|
|
// Get downloads the file from a cs3 service
|
|
// The caller MUST make sure to close the returned ReadCloser
|
|
func (s CS3) Get(ctx context.Context, path string) (io.ReadCloser, error) {
|
|
auth, ok := ContextGetAuthorization(ctx)
|
|
if !ok {
|
|
return nil, errors.New("cs3source: authorization missing")
|
|
}
|
|
ref, err := storagespace.ParseReference(path)
|
|
if err != nil {
|
|
// If the path is not a spaces reference try to handle it like a plain
|
|
// path reference.
|
|
ref = provider.Reference{
|
|
Path: path,
|
|
}
|
|
}
|
|
|
|
ctx = metadata.AppendToOutgoingContext(context.Background(), revactx.TokenHeader, auth)
|
|
gwc, err := s.gatewaySelector.Next()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rsp, err := gwc.InitiateFileDownload(ctx, &provider.InitiateFileDownloadRequest{Ref: &ref})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if rsp.Status.Code != rpc.Code_CODE_OK {
|
|
return nil, fmt.Errorf("could not load image: %s", rsp.Status.Message)
|
|
}
|
|
var ep, tk string
|
|
for _, p := range rsp.Protocols {
|
|
if p.Protocol == "spaces" {
|
|
ep, tk = p.DownloadEndpoint, p.Token
|
|
break
|
|
}
|
|
}
|
|
if (ep == "" || tk == "") && len(rsp.Protocols) > 0 {
|
|
ep, tk = rsp.Protocols[0].DownloadEndpoint, rsp.Protocols[0].Token
|
|
}
|
|
|
|
httpReq, err := rhttp.NewRequest(ctx, "GET", ep, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
httpReq.Header.Set(revactx.TokenHeader, auth)
|
|
httpReq.Header.Set(TokenTransportHeader, tk)
|
|
|
|
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{
|
|
MinVersion: tls.VersionTLS12,
|
|
InsecureSkipVerify: s.insecure, //nolint:gosec
|
|
}
|
|
client := &http.Client{}
|
|
|
|
resp, err := client.Do(httpReq) // nolint:bodyclose
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, fmt.Errorf("could not get the image \"%s\". Request returned with statuscode %d ", path, resp.StatusCode)
|
|
}
|
|
|
|
return resp.Body, nil
|
|
}
|