2022-04-06 13:00:19 +02:00
|
|
|
package swift
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2022-06-24 12:50:08 +02:00
|
|
|
"errors"
|
2022-04-06 13:00:19 +02:00
|
|
|
"fmt"
|
2022-06-24 12:50:08 +02:00
|
|
|
"io"
|
2022-04-06 13:00:19 +02:00
|
|
|
"net/http"
|
|
|
|
"strings"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/ncw/swift/v2"
|
2022-04-06 13:35:44 +02:00
|
|
|
|
|
|
|
"github.com/imgproxy/imgproxy/v3/config"
|
2022-04-06 13:00:19 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type transport struct {
|
|
|
|
con *swift.Connection
|
|
|
|
}
|
|
|
|
|
|
|
|
func New() (http.RoundTripper, error) {
|
|
|
|
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,
|
|
|
|
}
|
|
|
|
|
|
|
|
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, "/")
|
|
|
|
|
2022-04-06 13:35:44 +02:00
|
|
|
object, objectHeaders, err := t.con.ObjectOpen(req.Context(), container, objectName, false, make(swift.Headers))
|
2022-04-06 13:00:19 +02:00
|
|
|
|
2022-06-24 12:50:08 +02:00
|
|
|
header := make(http.Header)
|
|
|
|
|
2022-04-06 13:00:19 +02:00
|
|
|
if err != nil {
|
2022-06-24 12:50:08 +02:00
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2022-04-06 13:00:19 +02:00
|
|
|
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 {
|
2022-04-06 13:00:19 +02:00
|
|
|
header.Set("ETag", etag)
|
|
|
|
|
|
|
|
if len(etag) > 0 && etag == req.Header.Get("If-None-Match") {
|
|
|
|
object.Close()
|
2022-04-06 13:35:44 +02:00
|
|
|
|
2022-04-06 13:00:19 +02:00
|
|
|
return &http.Response{
|
|
|
|
StatusCode: http.StatusNotModified,
|
|
|
|
Proto: "HTTP/1.0",
|
|
|
|
ProtoMajor: 1,
|
|
|
|
ProtoMinor: 0,
|
|
|
|
Header: header,
|
|
|
|
ContentLength: 0,
|
|
|
|
Body: nil,
|
|
|
|
Close: false,
|
|
|
|
Request: req,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-04-06 13:35:44 +02:00
|
|
|
for k, v := range objectHeaders {
|
|
|
|
header.Set(k, v)
|
|
|
|
}
|
|
|
|
|
2022-04-06 13:00:19 +02:00
|
|
|
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
|
|
|
|
}
|