From 14293109ed59596c34e491b1486b355e7b331a2d Mon Sep 17 00:00:00 2001 From: Zhomart Mukhamejanov Date: Fri, 13 Jan 2023 13:50:14 -0800 Subject: [PATCH] Allow providing multiple BeforeScript and AfterScript Summary: Previsouly this code would allow only a single script to run: ``` r.Get("/swagger/*", httpSwagger.Handler( httpSwagger.BeforeScript("console.log('before 1')"), httpSwagger.BeforeScript("console.log('before 2')"), httpSwagger.URL("http://localhost:1323/swagger/doc.json"), //The url pointing to API definition )) // Only 'before 2' will be printed ``` Now both scripts are run: ``` 'before 1' 'before 2' ``` This improves behavior of BeforeScript and AfterScript. If users want to provide multiple scripts, they are forced to combine them using strings. Closes #81 Test Plan: Updated existing example and tests. Added new test TestMultipleScripts --- example/go-chi/go.mod | 4 ++-- example/go-chi/go.sum | 16 ++++++---------- example/go-chi/main.go | 3 +++ example/gorilla/go.mod | 4 ++-- example/gorilla/go.sum | 16 ++++++---------- example/gorilla/main.go | 2 +- swagger.go | 18 +++++++++--------- swagger_test.go | 33 +++++++++++++++++++++++++-------- 8 files changed, 54 insertions(+), 42 deletions(-) diff --git a/example/go-chi/go.mod b/example/go-chi/go.mod index ccdb297..9189412 100644 --- a/example/go-chi/go.mod +++ b/example/go-chi/go.mod @@ -6,6 +6,6 @@ require ( github.com/go-chi/chi v4.1.2+incompatible github.com/swaggo/http-swagger v1.2.6 github.com/swaggo/swag v1.8.1 - golang.org/x/net v0.0.0-20220425223048-2871e0cb64e4 // indirect - golang.org/x/tools v0.1.10 // indirect ) + +replace github.com/swaggo/http-swagger => ../.. diff --git a/example/go-chi/go.sum b/example/go-chi/go.sum index 880f50d..b4db2bf 100644 --- a/example/go-chi/go.sum +++ b/example/go-chi/go.sum @@ -1,9 +1,7 @@ github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= github.com/KyleBanks/depth v1.2.1 h1:5h8fQADFrWtarTdtDudMmGsC7GPbOAu6RVB3ffsVFHc= github.com/KyleBanks/depth v1.2.1/go.mod h1:jzSb9d0L43HxTQfT+oSA1EEp2q+ne2uh6XgeJcm8brE= -github.com/PuerkitoBio/purell v1.1.1 h1:WEQqlqaGbrPkxLJWfBwQmfEAE1Z7ONdDLqrN38tNFfI= github.com/PuerkitoBio/purell v1.1.1/go.mod h1:c11w/QuzBsJSee3cPx9rAFu61PvFxuPbtSwDGJws/X0= -github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 h1:d+Bc7a5rLufV/sSk/8dngufqelfh6jnri85riMAaF/M= github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578/go.mod h1:uGdkoq3SwY9Y+13GIhn11/XLaGBb4BfwItxLd5jeuXE= github.com/agiledragon/gomonkey/v2 v2.3.1 h1:k+UnUY0EMNYUFUAQVETGY9uUTxjMdnUkP0ARyJS1zzs= github.com/agiledragon/gomonkey/v2 v2.3.1/go.mod h1:ap1AmDzcVOAz1YpeJ3TCzIgstoaWLA6jbbgxfB4w2iY= @@ -18,10 +16,12 @@ github.com/go-chi/chi v4.1.2+incompatible/go.mod h1:eB3wogJHnLi3x/kFX2A+IbTBlXxm github.com/go-openapi/jsonpointer v0.19.3/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg= github.com/go-openapi/jsonpointer v0.19.5 h1:gZr+CIYByUqjcgeLXnQu2gHYQC9o73G2XUeOFYEICuY= github.com/go-openapi/jsonpointer v0.19.5/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg= -github.com/go-openapi/jsonreference v0.19.6 h1:UBIxjkht+AWIgYzCDSv2GN+E/togfwXUJFRTWhl2Jjs= github.com/go-openapi/jsonreference v0.19.6/go.mod h1:diGHMEHg2IqXZGKxqyvWdfWU/aim5Dprw5bqpKkTvns= -github.com/go-openapi/spec v0.20.4 h1:O8hJrt0UMnhHcluhIdUgCLRWyM2x7QkBXRvOs7m+O1M= +github.com/go-openapi/jsonreference v0.20.0 h1:MYlu0sBgChmCfJxxUKZ8g1cPWFOB37YSZqewK7OKeyA= +github.com/go-openapi/jsonreference v0.20.0/go.mod h1:Ag74Ico3lPc+zR+qjn4XBUmXymS4zJbYVCZmcgkasdo= github.com/go-openapi/spec v0.20.4/go.mod h1:faYFR1CvsJZ0mNsmsphTMSoRrNV3TEDoAM7FOEWeq8I= +github.com/go-openapi/spec v0.20.6 h1:ich1RQ3WDbfoeTqTAb+5EIxNmpKVJZWBNah9RAT0jIQ= +github.com/go-openapi/spec v0.20.6/go.mod h1:2OpW+JddWPrpXSCIX8eOx7lZ5iyuWj3RYR6VaaBKcWA= github.com/go-openapi/swag v0.19.5/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk= github.com/go-openapi/swag v0.19.15 h1:D2NRCBzS9/pEY3gP9Nl8aDqGUcPFrwG2p+CNFrLyrCM= github.com/go-openapi/swag v0.19.15/go.mod h1:QYRuS/SOXUCsnplDa677K7+DxSOj6IPNl/eQntq43wQ= @@ -59,11 +59,8 @@ github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UV github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= -github.com/swaggo/files v0.0.0-20210815190702-a29dd2bc99b2 h1:+iNTcqQJy0OZ5jk6a5NLib47eqXK8uYcPX+O4+cBpEM= -github.com/swaggo/files v0.0.0-20210815190702-a29dd2bc99b2/go.mod h1:lKJPbtWzJ9JhsTN1k1gZgleJWY/cqq0psdoMmaThG3w= -github.com/swaggo/http-swagger v1.2.6 h1:ihTjChUoSRMpFMjWw+0AkL1Ti4r6v8pCgVYLmQVRlRw= -github.com/swaggo/http-swagger v1.2.6/go.mod h1:CcoICgY3yVDk2u1LQUCMHbAj0fjlxIX+873psXlIKNA= -github.com/swaggo/swag v1.7.9/go.mod h1:gZ+TJ2w/Ve1RwQsA2IRoSOTidHz6DX+PIG8GWvbnoLU= +github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe h1:K8pHPVoTgxFJt1lXuIzzOX7zZhZFldJQK/CgKx9BFIc= +github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe/go.mod h1:lKJPbtWzJ9JhsTN1k1gZgleJWY/cqq0psdoMmaThG3w= github.com/swaggo/swag v1.8.1 h1:JuARzFX1Z1njbCGz+ZytBR15TFJwF2Q7fu8puJHhQYI= github.com/swaggo/swag v1.8.1/go.mod h1:ugemnJsPZm/kRwFUnzBlbHRd0JY9zE1M4F+uy2pAaPQ= github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI= @@ -101,7 +98,6 @@ golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuX golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= -golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk= golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= diff --git a/example/go-chi/main.go b/example/go-chi/main.go index 80dc90c..436cb95 100644 --- a/example/go-chi/main.go +++ b/example/go-chi/main.go @@ -27,8 +27,11 @@ func main() { r := chi.NewRouter() r.Get("/swagger/*", httpSwagger.Handler( + httpSwagger.BeforeScript("console.log('before 1')"), + httpSwagger.BeforeScript("console.log('before 2')"), httpSwagger.URL("http://localhost:1323/swagger/doc.json"), //The url pointing to API definition )) + log.Println("Listening :1323") log.Fatal(http.ListenAndServe(":1323", r)) } diff --git a/example/gorilla/go.mod b/example/gorilla/go.mod index 8e78c3a..93a6e9f 100644 --- a/example/gorilla/go.mod +++ b/example/gorilla/go.mod @@ -6,6 +6,6 @@ require ( github.com/gorilla/mux v1.8.0 github.com/swaggo/http-swagger v1.2.6 github.com/swaggo/swag v1.8.1 - golang.org/x/net v0.0.0-20220425223048-2871e0cb64e4 // indirect - golang.org/x/tools v0.1.10 // indirect ) + +replace github.com/swaggo/http-swagger => ../.. diff --git a/example/gorilla/go.sum b/example/gorilla/go.sum index 5da80d3..7775443 100644 --- a/example/gorilla/go.sum +++ b/example/gorilla/go.sum @@ -1,9 +1,7 @@ github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= github.com/KyleBanks/depth v1.2.1 h1:5h8fQADFrWtarTdtDudMmGsC7GPbOAu6RVB3ffsVFHc= github.com/KyleBanks/depth v1.2.1/go.mod h1:jzSb9d0L43HxTQfT+oSA1EEp2q+ne2uh6XgeJcm8brE= -github.com/PuerkitoBio/purell v1.1.1 h1:WEQqlqaGbrPkxLJWfBwQmfEAE1Z7ONdDLqrN38tNFfI= github.com/PuerkitoBio/purell v1.1.1/go.mod h1:c11w/QuzBsJSee3cPx9rAFu61PvFxuPbtSwDGJws/X0= -github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 h1:d+Bc7a5rLufV/sSk/8dngufqelfh6jnri85riMAaF/M= github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578/go.mod h1:uGdkoq3SwY9Y+13GIhn11/XLaGBb4BfwItxLd5jeuXE= github.com/agiledragon/gomonkey/v2 v2.3.1 h1:k+UnUY0EMNYUFUAQVETGY9uUTxjMdnUkP0ARyJS1zzs= github.com/agiledragon/gomonkey/v2 v2.3.1/go.mod h1:ap1AmDzcVOAz1YpeJ3TCzIgstoaWLA6jbbgxfB4w2iY= @@ -16,10 +14,12 @@ github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeME github.com/go-openapi/jsonpointer v0.19.3/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg= github.com/go-openapi/jsonpointer v0.19.5 h1:gZr+CIYByUqjcgeLXnQu2gHYQC9o73G2XUeOFYEICuY= github.com/go-openapi/jsonpointer v0.19.5/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg= -github.com/go-openapi/jsonreference v0.19.6 h1:UBIxjkht+AWIgYzCDSv2GN+E/togfwXUJFRTWhl2Jjs= github.com/go-openapi/jsonreference v0.19.6/go.mod h1:diGHMEHg2IqXZGKxqyvWdfWU/aim5Dprw5bqpKkTvns= -github.com/go-openapi/spec v0.20.4 h1:O8hJrt0UMnhHcluhIdUgCLRWyM2x7QkBXRvOs7m+O1M= +github.com/go-openapi/jsonreference v0.20.0 h1:MYlu0sBgChmCfJxxUKZ8g1cPWFOB37YSZqewK7OKeyA= +github.com/go-openapi/jsonreference v0.20.0/go.mod h1:Ag74Ico3lPc+zR+qjn4XBUmXymS4zJbYVCZmcgkasdo= github.com/go-openapi/spec v0.20.4/go.mod h1:faYFR1CvsJZ0mNsmsphTMSoRrNV3TEDoAM7FOEWeq8I= +github.com/go-openapi/spec v0.20.6 h1:ich1RQ3WDbfoeTqTAb+5EIxNmpKVJZWBNah9RAT0jIQ= +github.com/go-openapi/spec v0.20.6/go.mod h1:2OpW+JddWPrpXSCIX8eOx7lZ5iyuWj3RYR6VaaBKcWA= github.com/go-openapi/swag v0.19.5/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk= github.com/go-openapi/swag v0.19.15 h1:D2NRCBzS9/pEY3gP9Nl8aDqGUcPFrwG2p+CNFrLyrCM= github.com/go-openapi/swag v0.19.15/go.mod h1:QYRuS/SOXUCsnplDa677K7+DxSOj6IPNl/eQntq43wQ= @@ -59,11 +59,8 @@ github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UV github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= -github.com/swaggo/files v0.0.0-20210815190702-a29dd2bc99b2 h1:+iNTcqQJy0OZ5jk6a5NLib47eqXK8uYcPX+O4+cBpEM= -github.com/swaggo/files v0.0.0-20210815190702-a29dd2bc99b2/go.mod h1:lKJPbtWzJ9JhsTN1k1gZgleJWY/cqq0psdoMmaThG3w= -github.com/swaggo/http-swagger v1.2.6 h1:ihTjChUoSRMpFMjWw+0AkL1Ti4r6v8pCgVYLmQVRlRw= -github.com/swaggo/http-swagger v1.2.6/go.mod h1:CcoICgY3yVDk2u1LQUCMHbAj0fjlxIX+873psXlIKNA= -github.com/swaggo/swag v1.7.9/go.mod h1:gZ+TJ2w/Ve1RwQsA2IRoSOTidHz6DX+PIG8GWvbnoLU= +github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe h1:K8pHPVoTgxFJt1lXuIzzOX7zZhZFldJQK/CgKx9BFIc= +github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe/go.mod h1:lKJPbtWzJ9JhsTN1k1gZgleJWY/cqq0psdoMmaThG3w= github.com/swaggo/swag v1.8.1 h1:JuARzFX1Z1njbCGz+ZytBR15TFJwF2Q7fu8puJHhQYI= github.com/swaggo/swag v1.8.1/go.mod h1:ugemnJsPZm/kRwFUnzBlbHRd0JY9zE1M4F+uy2pAaPQ= github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI= @@ -101,7 +98,6 @@ golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuX golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= -golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk= golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= diff --git a/example/gorilla/main.go b/example/gorilla/main.go index ab7b10f..89d5cb7 100644 --- a/example/gorilla/main.go +++ b/example/gorilla/main.go @@ -30,7 +30,7 @@ func main() { httpSwagger.URL("http://localhost:1323/swagger/doc.json"), //The url pointing to API definition httpSwagger.DeepLinking(true), httpSwagger.DocExpansion("none"), - httpSwagger.DomID("#swagger-ui"), + httpSwagger.DomID("swagger-ui"), )).Methods(http.MethodGet) log.Fatal(http.ListenAndServe(":1323", r)) diff --git a/swagger.go b/swagger.go index 5eb250d..969e2a2 100644 --- a/swagger.go +++ b/swagger.go @@ -21,8 +21,8 @@ type Config struct { DocExpansion string DomID string InstanceName string - BeforeScript template.JS - AfterScript template.JS + BeforeScripts []template.JS + AfterScripts []template.JS Plugins []template.JS UIConfig map[template.JS]template.JS DeepLinking bool @@ -50,7 +50,7 @@ func DocExpansion(docExpansion string) func(*Config) { } } -// DomID #swagger-ui. +// DomID #swagger-ui, without #. func DomID(domID string) func(*Config) { return func(c *Config) { c.DomID = domID @@ -98,7 +98,7 @@ func UIConfig(props map[string]string) func(*Config) { // BeforeScript holds JavaScript to be run right before the Swagger UI object is created. func BeforeScript(js string) func(*Config) { return func(c *Config) { - c.BeforeScript = template.JS(js) + c.BeforeScripts = append(c.BeforeScripts, template.JS(js)) } } @@ -106,7 +106,7 @@ func BeforeScript(js string) func(*Config) { // and set on the window. func AfterScript(js string) func(*Config) { return func(c *Config) { - c.AfterScript = template.JS(js) + c.AfterScripts = append(c.AfterScripts, template.JS(js)) } } @@ -261,8 +261,8 @@ const indexTempl = ` diff --git a/swagger_test.go b/swagger_test.go index d08210f..9001593 100644 --- a/swagger_test.go +++ b/swagger_test.go @@ -212,9 +212,9 @@ func TestConfigURL(t *testing.T) { { desc: "configure BeforeScript", exp: &Config{ - BeforeScript: `const SomePlugin = (system) => ({ + BeforeScripts: []template.JS{`const SomePlugin = (system) => ({ // Some plugin - });`, + });`}, }, cfgfn: BeforeScript(`const SomePlugin = (system) => ({ // Some plugin @@ -223,9 +223,9 @@ func TestConfigURL(t *testing.T) { { desc: "configure AfterScript", exp: &Config{ - AfterScript: `const SomePlugin = (system) => ({ + AfterScripts: []template.JS{`const SomePlugin = (system) => ({ // Some plugin - });`, + });`}, }, cfgfn: AfterScript(`const SomePlugin = (system) => ({ // Some plugin @@ -242,6 +242,23 @@ func TestConfigURL(t *testing.T) { } } +func TestMultipleScripts(t *testing.T) { + router := http.NewServeMux() + + router.Handle("/", Handler( + BeforeScript("before 1"), + BeforeScript("before 2"), + AfterScript("after 1"), + AfterScript("after 2"), + )) + + w1 := performRequest(http.MethodGet, "/index.html", router) + assert.Contains(t, w1.Body.String(), "before 1") + assert.Contains(t, w1.Body.String(), "before 2") + assert.Contains(t, w1.Body.String(), "after 1") + assert.Contains(t, w1.Body.String(), "after 2") +} + func TestUIConfigOptions(t *testing.T) { type fixture struct { @@ -367,14 +384,14 @@ func TestUIConfigOptions(t *testing.T) { PersistAuthorization: true, DocExpansion: "none", DomID: "swagger-ui-id", - BeforeScript: `const SomePlugin = (system) => ({ + BeforeScripts: []template.JS{`const SomePlugin = (system) => ({ // Some plugin }); -`, - AfterScript: `const someOtherCode = function(){ +`}, + AfterScripts: []template.JS{`const someOtherCode = function(){ // Do something }; - someOtherCode();`, + someOtherCode();`}, Plugins: []template.JS{ "SomePlugin", "AnotherPlugin",