diff options
Diffstat (limited to 'contrib/sharedhost')
| -rw-r--r-- | contrib/sharedhost/replacement.go | 6 | ||||
| -rw-r--r-- | contrib/sharedhost/replacement_test.go | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/contrib/sharedhost/replacement.go b/contrib/sharedhost/replacement.go index 1fb2a0d..9267530 100644 --- a/contrib/sharedhost/replacement.go +++ b/contrib/sharedhost/replacement.go @@ -19,14 +19,14 @@ import ( // "users/", "domain.com/~jim/index.gmi" maps to "domain.com/users/jim/index.gmi". func ReplaceTilde(replacement string) gus.Middleware { return func(inner gus.Handler) gus.Handler { - return func(ctx context.Context, request *gus.Request) *gus.Response { + return gus.HandlerFunc(func(ctx context.Context, request *gus.Request) *gus.Response { if len(request.Path) > 1 && request.Path[0] == '/' && request.Path[1] == '~' { request = cloneRequest(request) request.Path = "/" + replacement + request.Path[2:] } - return inner(ctx, request) - } + return inner.Handle(ctx, request) + }) } } diff --git a/contrib/sharedhost/replacement_test.go b/contrib/sharedhost/replacement_test.go index cab80bb..67c3754 100644 --- a/contrib/sharedhost/replacement_test.go +++ b/contrib/sharedhost/replacement_test.go @@ -43,12 +43,12 @@ func TestReplaceTilde(t *testing.T) { replacer := sharedhost.ReplaceTilde(test.replacement) request := &gus.Request{URL: u} - handler := replacer(func(_ context.Context, request *gus.Request) *gus.Response { + handler := replacer(gus.HandlerFunc(func(_ context.Context, request *gus.Request) *gus.Response { assert.Equal(t, test.replacedPath, request.Path) return &gus.Response{} - }) + })) - handler(context.Background(), request) + handler.Handle(context.Background(), request) // original request was unmodified assert.Equal(t, originalPath, request.Path) |