Move code to pkg

pkg/middlewares/replacepath/replace_path.go

@@ -0,0 +1,46 @@
+package replacepath
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/containous/traefik/pkg/config"
+	"github.com/containous/traefik/pkg/middlewares"
+	"github.com/containous/traefik/pkg/tracing"
+	"github.com/opentracing/opentracing-go/ext"
+)
+
+const (
+	// ReplacedPathHeader is the default header to set the old path to.
+	ReplacedPathHeader = "X-Replaced-Path"
+	typeName           = "ReplacePath"
+)
+
+// ReplacePath is a middleware used to replace the path of a URL request.
+type replacePath struct {
+	next http.Handler
+	path string
+	name string
+}
+
+// New creates a new replace path middleware.
+func New(ctx context.Context, next http.Handler, config config.ReplacePath, name string) (http.Handler, error) {
+	middlewares.GetLogger(ctx, name, typeName).Debug("Creating middleware")
+
+	return &replacePath{
+		next: next,
+		path: config.Path,
+		name: name,
+	}, nil
+}
+
+func (r *replacePath) GetTracingInformation() (string, ext.SpanKindEnum) {
+	return r.name, tracing.SpanKindNoneEnum
+}
+
+func (r *replacePath) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	req.Header.Add(ReplacedPathHeader, req.URL.Path)
+	req.URL.Path = r.path
+	req.RequestURI = req.URL.RequestURI()
+	r.next.ServeHTTP(rw, req)
+}

pkg/middlewares/replacepath/replace_path_test.go

@@ -0,0 +1,46 @@
+package replacepath
+
+import (
+	"context"
+	"net/http"
+	"testing"
+
+	"github.com/containous/traefik/pkg/config"
+	"github.com/containous/traefik/pkg/testhelpers"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestReplacePath(t *testing.T) {
+	var replacementConfig = config.ReplacePath{
+		Path: "/replacement-path",
+	}
+
+	paths := []string{
+		"/example",
+		"/some/really/long/path",
+	}
+
+	for _, path := range paths {
+		t.Run(path, func(t *testing.T) {
+
+			var expectedPath, actualHeader, requestURI string
+			next := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+				expectedPath = r.URL.Path
+				actualHeader = r.Header.Get(ReplacedPathHeader)
+				requestURI = r.RequestURI
+			})
+
+			handler, err := New(context.Background(), next, replacementConfig, "foo-replace-path")
+			require.NoError(t, err)
+
+			req := testhelpers.MustNewRequest(http.MethodGet, "http://localhost"+path, nil)
+
+			handler.ServeHTTP(nil, req)
+
+			assert.Equal(t, expectedPath, replacementConfig.Path, "Unexpected path.")
+			assert.Equal(t, path, actualHeader, "Unexpected '%s' header.", ReplacedPathHeader)
+			assert.Equal(t, expectedPath, requestURI, "Unexpected request URI.")
+		})
+	}
+}