From 60ba6bd20cf089808165e2979529dd66525f0e92 Mon Sep 17 00:00:00 2001 From: Javier Date: Wed, 17 Jun 2020 09:34:53 -0500 Subject: [PATCH] feat(*) add back header and body filter phases (#31) --- instance.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/instance.go b/instance.go index 2c68879..e5cf113 100644 --- a/instance.go +++ b/instance.go @@ -23,6 +23,8 @@ type ( certificater interface{ Certificate(*pdk.PDK) } rewriter interface{ Rewrite(*pdk.PDK) } accesser interface{ Access(*pdk.PDK) } + headerFilter interface{ HeaderFilter(*pdk.PDK) } + bodyFilter interface{ BodyFilter(*pdk.PDK) } prereader interface{ Preread(*pdk.PDK) } logger interface{ Log(*pdk.PDK) } ) @@ -33,6 +35,8 @@ func getHandlers(config interface{}) map[string]func(kong *pdk.PDK) { if h, ok := config.(certificater); ok { handlers["certificate"] = h.Certificate } if h, ok := config.(rewriter) ; ok { handlers["rewrite"] = h.Rewrite } if h, ok := config.(accesser) ; ok { handlers["access"] = h.Access } + if h, ok := config.(headerFilter); ok { handlers["header_filter"] = h.HeaderFilter } + if h, ok := config.(bodyFilter) ; ok { handlers["body_filter"] = h.BodyFilter } if h, ok := config.(prereader) ; ok { handlers["preread"] = h.Preread } if h, ok := config.(logger) ; ok { handlers["log"] = h.Log }