add directory index feature (#56)

Author: ubogdan

swagger.go

@@ -55,7 +55,7 @@ var WrapHandler = EchoWrapHandler()
 func EchoWrapHandler(configFns ...func(c *Config)) echo.HandlerFunc {
 	var once sync.Once
 
-	handler := swaggerFiles.Handler
+	h := swaggerFiles.Handler
 
 	config := &Config{
 		URL:          "doc.json",
@@ -72,21 +72,20 @@ func EchoWrapHandler(configFns ...func(c *Config)) echo.HandlerFunc {
 	t := template.New("swagger_index.html")
 	index, _ := t.Parse(indexTemplate)
 
-	var re = regexp.MustCompile(`(.*)(index\.html|doc\.json|favicon-16x16\.png|favicon-32x32\.png|/oauth2-redirect\.html|swagger-ui\.css|swagger-ui\.css\.map|swagger-ui\.js|swagger-ui\.js\.map|swagger-ui-bundle\.js|swagger-ui-bundle\.js\.map|swagger-ui-standalone-preset\.js|swagger-ui-standalone-preset\.js\.map)[?|.]*`)
+	var re = regexp.MustCompile(`^(.*/)([^?].*)?[?|.]*$`)
 
 	return func(c echo.Context) error {
-		var matches []string
-		if matches = re.FindStringSubmatch(c.Request().RequestURI); len(matches) != 3 {
-			return c.String(http.StatusNotFound, "404 page not found")
-		}
+		matches := re.FindStringSubmatch(c.Request().RequestURI)
 		path := matches[2]
+
 		once.Do(func() {
-			handler.Prefix = matches[1]
+			h.Prefix = matches[1]
 		})
 
 		defer c.Response().Flush()
-
 		switch path {
+		case "":
+			c.Redirect(301, h.Prefix+"index.html")
 		case "index.html":
 			_ = index.Execute(c.Response().Writer, config)
 		case "doc.json":
@@ -100,7 +99,7 @@ func EchoWrapHandler(configFns ...func(c *Config)) echo.HandlerFunc {
 			c.Response().Writer.Header().Set("Content-Type", "application/json; charset=utf-8")
 			_, _ = c.Response().Writer.Write([]byte(doc))
 		default:
-			handler.ServeHTTP(c.Response().Writer, c.Request())
+			h.ServeHTTP(c.Response().Writer, c.Request())
 		}
 
 		return nil

swagger_test.go

@@ -235,6 +235,7 @@ func TestWrapHandler(t *testing.T) {
 
 	router := echo.New()
 
+	//RegisterEchoHandler(router.GET, "/", nil)
 	router.GET("/*", EchoWrapHandler(DocExpansion("none"), DomID("#swagger-ui")))
 
 	w1 := performRequest("GET", "/index.html", router)
@@ -253,6 +254,8 @@ func TestWrapHandler(t *testing.T) {
 	w4 := performRequest("GET", "/notfound", router)
 	assert.Equal(t, 404, w4.Code)
 
+	w5 := performRequest("GET", "/", router)
+	assert.Equal(t, 301, w5.Code)
 }
 
 func performRequest(method, target string, e *echo.Echo) *httptest.ResponseRecorder {