Merge pull request #21 from kepkin/feature/custom-routing

Add custom router to register middleware.

internal/generator/router.go

@@ -24,6 +24,20 @@ func RegisterRoutes(r *gin.Engine, api {{ .InterfaceName }}) {
             {{- end }}
         {{- end }}
 	{{ end -}}
+}
+
+type HandlerRegister func(operationID, httpMethod, relativePath string, handler gin.HandlerFunc)
+
+// CustomRouter
+func RegisterRoutesCustom(handlerRegister HandlerRegister, api {{ .InterfaceName }}) {
+	e := &{{ .InterfaceName }}Server{api}
+	{{ range $url, $m := .Paths -}}
+        {{- range $methodName, $method := (dict "GET" $m.Get "POST" $m.Post "PATCH" $m.Patch "DELETE" $m.Delete "PUT" $m.Put "OPTIONS" $m.Options) }}
+			{{- with $method }}
+				handlerRegister("{{ .OperationID }}", "{{ $methodName }}", "{{ ConvertUrl $url }}", e._{{ $.InterfaceName }}_{{ .OperationID }}_Handler)
+            {{- end }}
+        {{- end }}
+	{{ end -}}
 }`))
 
 func (Generator) makeRouter(wr io.Writer, sp openapi3.Spec) error {

test/generated_api_test.go

@@ -116,3 +116,29 @@ func TestCreateUser(t *testing.T) {
 
 	assert.Equal(t, response.Body.Bytes(), photoBytes)
 }
+
+
+func TestCustomRouteWithMiddleWare(t *testing.T) {
+	r := gin.New()
+	api.RegisterRoutesCustom(func(operationID, httpMethod, relativePath string, handler gin.HandlerFunc) {
+		handlers := make([]gin.HandlerFunc, 0)
+
+		if operationID == "GetUser" {
+			handlers = append(handlers, func(c *gin.Context){
+				c.AbortWithStatus(401)
+			})
+		}
+
+		handlers = append(handlers, handler)
+		r.Handle(httpMethod, relativePath, handlers...)
+	}, api.NewPaymentGatewayAPI())
+
+	// Go to route with middle ware
+	request := httptest.NewRequest(http.MethodGet, "/v1/user/123", nil)
+	response := httptest.NewRecorder()
+
+	r.ServeHTTP(response, request)
+	if !assert.Equal(t, http.StatusUnauthorized, response.Code, response.Body.String()) {
+		return
+	}
+}