package ginext import ( "github.com/gin-gonic/gin" "net/http" ) var anyMethods = []string{ http.MethodGet, http.MethodPost, http.MethodPut, http.MethodPatch, http.MethodHead, http.MethodOptions, http.MethodDelete, http.MethodConnect, http.MethodTrace, } type GinRoutesWrapper struct { wrapper *GinWrapper routes gin.IRouter } type GinRouteBuilder struct { routes *GinRoutesWrapper methods []string relPath string handlers []gin.HandlerFunc } func (w *GinWrapper) Routes() *GinRoutesWrapper { return &GinRoutesWrapper{wrapper: w, routes: w.engine} } func (w *GinRoutesWrapper) Group(relativePath string) *GinRoutesWrapper { return &GinRoutesWrapper{wrapper: w.wrapper, routes: w.routes.Group(relativePath)} } func (w *GinRoutesWrapper) GET(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodGet}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) POST(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodPost}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) DELETE(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodDelete}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) PATCH(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodPatch}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) PUT(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodPut}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) OPTIONS(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodOptions}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) HEAD(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{http.MethodHead}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) COUNT(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: []string{"COUNT"}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRoutesWrapper) Any(relativePath string) *GinRouteBuilder { return &GinRouteBuilder{routes: w, methods: anyMethods, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} } func (w *GinRouteBuilder) Use(middleware ...gin.HandlerFunc) *GinRouteBuilder { w.handlers = append(w.handlers, middleware...) return w } func (w *GinRouteBuilder) Handle(handler WHandlerFunc) { w.handlers = append(w.handlers, Wrap(w.routes.wrapper, handler)) for _, m := range w.methods { w.routes.routes.Handle(m, w.relPath, w.handlers...) } } func (w *GinWrapper) NoRoute(handler WHandlerFunc) { w.engine.NoRoute(Wrap(w, handler)) }