1
0
mirror of https://github.com/imgproxy/imgproxy.git synced 2024-12-04 09:42:31 +02:00
imgproxy/transport/swift/swift.go

110 lines
2.5 KiB
Go
Raw Normal View History

package swift
import (
"context"
"errors"
"fmt"
"io"
"net/http"
"strings"
"time"
"github.com/ncw/swift/v2"
2022-04-06 13:35:44 +02:00
"github.com/imgproxy/imgproxy/v3/config"
defaultTransport "github.com/imgproxy/imgproxy/v3/transport"
2023-04-10 19:01:44 +02:00
"github.com/imgproxy/imgproxy/v3/transport/notmodified"
)
type transport struct {
con *swift.Connection
}
func New() (http.RoundTripper, error) {
trans, err := defaultTransport.New(false)
if err != nil {
return nil, err
}
c := &swift.Connection{
UserName: config.SwiftUsername,
ApiKey: config.SwiftAPIKey,
AuthUrl: config.SwiftAuthURL,
AuthVersion: config.SwiftAuthVersion,
Domain: config.SwiftDomain, // v3 auth only
Tenant: config.SwiftTenant, // v2 auth only
Timeout: time.Duration(config.SwiftTimeoutSeconds) * time.Second,
ConnectTimeout: time.Duration(config.SwiftConnectTimeoutSeconds) * time.Second,
Transport: trans,
}
ctx := context.Background()
err = c.Authenticate(ctx)
if err != nil {
return nil, fmt.Errorf("swift authentication error: %s", err)
}
return transport{con: c}, nil
}
func (t transport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
// Users should have converted the object storage URL in the format of swift://{container}/{object}
container := req.URL.Host
objectName := strings.TrimPrefix(req.URL.Path, "/")
reqHeaders := make(swift.Headers)
if r := req.Header.Get("Range"); len(r) > 0 {
reqHeaders["Range"] = r
}
object, objectHeaders, err := t.con.ObjectOpen(req.Context(), container, objectName, false, reqHeaders)
header := make(http.Header)
if err != nil {
if errors.Is(err, swift.ObjectNotFound) || errors.Is(err, swift.ContainerNotFound) {
return &http.Response{
StatusCode: http.StatusNotFound,
Proto: "HTTP/1.0",
ProtoMajor: 1,
ProtoMinor: 0,
Header: header,
Body: io.NopCloser(strings.NewReader(err.Error())),
Close: false,
Request: req,
}, nil
}
return nil, fmt.Errorf("error opening object: %v", err)
}
if config.ETagEnabled {
2022-04-06 13:35:44 +02:00
if etag, ok := objectHeaders["Etag"]; ok {
header.Set("ETag", etag)
2023-04-10 19:01:44 +02:00
}
2023-04-10 19:01:44 +02:00
if resp := notmodified.Response(req, header); resp != nil {
object.Close()
return resp, nil
}
}
2022-04-06 13:35:44 +02:00
for k, v := range objectHeaders {
header.Set(k, v)
}
return &http.Response{
Status: "200 OK",
StatusCode: 200,
Proto: "HTTP/1.0",
ProtoMajor: 1,
ProtoMinor: 0,
Header: header,
Body: object,
Close: true,
Request: req,
}, nil
}