beegin.go

package beegin

import (
	"fmt"
	"net/http"
)

type (
	HandlerFunc func(*Context) error
	routerMap   map[string]HandlerFunc

	Engine struct {
		router                  *router
		DefaultErrorHandlerFunc HandlerFunc
	}
)

func New() *Engine {
	return &Engine{
		router: newRouter(),
		DefaultErrorHandlerFunc:
		defaultErrorHandlerFunc,
	}
}

func defaultErrorHandlerFunc(c *Context) error {
	return c.String(http.StatusNotFound, fmt.Errorf("404 Not Found"))
}

func (e *Engine) GET(pattern string, h HandlerFunc) {
	e.router.addRoute(http.MethodGet, pattern, h)
}

func (e *Engine) POST(pattern string, h HandlerFunc) {
	e.router.addRoute(http.MethodPost, pattern, h)
}

func (e *Engine) Run(ip string, port int) error {
	fmt.Printf("=> listen on %s port %d\n", ip, port)
	return http.ListenAndServe(fmt.Sprintf("%s:%d", ip, port), e)
}

func (e *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	c := newContext(w, req)
	e.router.handle(c)
}

func (e *Engine) WrapHandler(h http.Handler) HandlerFunc {
	return func(c *Context) error {
		h.ServeHTTP(c.Writer, c.Req)
		return nil
	}
}