forked from TrueCloudLab/distribution
558ace1391
This allows to rewrite 'URLFor' of the storage driver to use a specific host/trim the base path. It is different from the 'redirect' middleware, as it still calls the storage driver URLFor. For example, with Azure storage provider, this allows to transform the SAS Azure Blob Storage URL into the URL compatible with Azure Front Door. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
storagedriver "github.com/distribution/distribution/v3/registry/storage/driver"
|
|
storagemiddleware "github.com/distribution/distribution/v3/registry/storage/driver/middleware"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func init() {
|
|
if err := storagemiddleware.Register("rewrite", newRewriteStorageMiddleware); err != nil {
|
|
logrus.Errorf("tailed to register redirect storage middleware: %v", err)
|
|
}
|
|
}
|
|
|
|
type rewriteStorageMiddleware struct {
|
|
storagedriver.StorageDriver
|
|
overrideScheme string
|
|
overrideHost string
|
|
trimPathPrefix string
|
|
}
|
|
|
|
var _ storagedriver.StorageDriver = &rewriteStorageMiddleware{}
|
|
|
|
func getStringOption(key string, options map[string]interface{}) (string, error) {
|
|
o, ok := options[key]
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
s, ok := o.(string)
|
|
if !ok {
|
|
return "", fmt.Errorf("%s must be a string", key)
|
|
}
|
|
return s, nil
|
|
}
|
|
|
|
func newRewriteStorageMiddleware(ctx context.Context, sd storagedriver.StorageDriver, options map[string]interface{}) (storagedriver.StorageDriver, error) {
|
|
var err error
|
|
|
|
r := &rewriteStorageMiddleware{StorageDriver: sd}
|
|
|
|
if r.overrideScheme, err = getStringOption("scheme", options); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if r.overrideHost, err = getStringOption("host", options); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if r.trimPathPrefix, err = getStringOption("trimpathprefix", options); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return r, nil
|
|
}
|
|
|
|
func (r *rewriteStorageMiddleware) RedirectURL(req *http.Request, path string) (string, error) {
|
|
storagePath, err := r.StorageDriver.RedirectURL(req, path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
u, err := url.Parse(storagePath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if r.overrideScheme != "" {
|
|
u.Scheme = r.overrideScheme
|
|
}
|
|
|
|
if r.overrideHost != "" {
|
|
u.Host = r.overrideHost
|
|
}
|
|
|
|
if r.trimPathPrefix != "" {
|
|
u.Path = strings.TrimPrefix(u.Path, r.trimPathPrefix)
|
|
}
|
|
|
|
return u.String(), nil
|
|
}
|