From c8e1328164e9ffbd681c3c0e449f1e6b9856b896 Mon Sep 17 00:00:00 2001
From: Gabriel Arakaki Giovanini <mail@gabrielgio.me>
Date: Sun, 26 Feb 2023 19:54:48 +0100
Subject: feat: Inicial commit

It contains rough template for the server and runners.

It contains rough template for the server and runners.
---
 pkg/ext/router.go | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 pkg/ext/router.go

(limited to 'pkg/ext/router.go')

diff --git a/pkg/ext/router.go b/pkg/ext/router.go
new file mode 100644
index 0000000..74f0a95
--- /dev/null
+++ b/pkg/ext/router.go
@@ -0,0 +1,51 @@
+package ext
+
+import (
+	"github.com/fasthttp/router"
+	"github.com/valyala/fasthttp"
+)
+
+type (
+	Router struct {
+		middlewares []Middleware
+		fastRouter  *router.Router
+	}
+	Middleware          func(next fasthttp.RequestHandler) fasthttp.RequestHandler
+	ErrorRequestHandler func(ctx *fasthttp.RequestCtx) error
+)
+
+func NewRouter(nestedRouter *router.Router) *Router {
+	return &Router{
+		fastRouter: nestedRouter,
+	}
+}
+
+func (self *Router) AddMiddleware(middleware Middleware) {
+	self.middlewares = append(self.middlewares, middleware)
+}
+
+func wrapError(next ErrorRequestHandler) fasthttp.RequestHandler {
+	return func(ctx *fasthttp.RequestCtx) {
+		if err := next(ctx); err != nil {
+			ctx.Response.SetStatusCode(500)
+			InternalServerError(ctx, err)
+		}
+	}
+}
+
+func (self *Router) run(next ErrorRequestHandler) fasthttp.RequestHandler {
+	return func(ctx *fasthttp.RequestCtx) {
+		req := wrapError(next)
+		for _, r := range self.middlewares {
+			req = r(req)
+		}
+		req(ctx)
+	}
+}
+
+func (self *Router) GET(path string, handler ErrorRequestHandler) {
+	self.fastRouter.GET(path, self.run(handler))
+}
+func (self *Router) POST(path string, handler ErrorRequestHandler) {
+	self.fastRouter.POST(path, self.run(handler))
+}
-- 
cgit v1.2.3