yyewolf/bar

package api

import (
    "bar/autogen"
    "bar/internal/config"
    "bar/internal/db"
    "os"

    "github.com/gorilla/context"
    "github.com/gorilla/sessions"
    "github.com/labstack/echo/v4"
    "github.com/labstack/echo/v4/middleware"
    echoLog "github.com/labstack/gommon/log"
    middlewareL "github.com/neko-neko/echo-logrus/v2"
    "github.com/neko-neko/echo-logrus/v2/log"
)

type Server struct {
    db.DBackend
}

func NewServer(db db.DBackend) *Server {
    s := &Server{
        db,
    }
    return s
}

func (s *Server) getUserSess(c echo.Context) *sessions.Session {
    return c.Get("userSess").(*sessions.Session)
}

func (s *Server) getAdminSess(c echo.Context) *sessions.Session {
    return c.Get("adminSess").(*sessions.Session)
}
func (s *Server) getOnboardSess(c echo.Context) *sessions.Session {
    return c.Get("onBoardSess").(*sessions.Session)
}

func (s *Server) Serve(c *config.Config) error {
    e := echo.New()

    // Logger
    log.Logger().SetOutput(os.Stdout)
    log.Logger().SetLevel(echoLog.WARN)
    e.Logger = log.Logger()
    e.Use(middlewareL.Logger())
    e.Use(middleware.BodyLimit("15M"))

    userStore := sessions.NewCookieStore([]byte(c.ApiConfig.SessionSecret))
    adminStore := sessions.NewCookieStore([]byte(c.ApiConfig.AdminSessionSecret))
    onBoardStore := sessions.NewCookieStore([]byte(c.ApiConfig.AdminSessionSecret))
    e.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
        return func(c echo.Context) error {
            defer context.Clear(c.Request())
            c.Set("userStore", userStore)
            c.Set("adminStore", adminStore)
            c.Set("onBoardStore", onBoardStore)
            return next(c)
        }
    })

    // CORS
    e.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
        return func(c echo.Context) error {
            c.Response().Header().Set("Access-Control-Allow-Origin", c.Request().Header.Get("Origin"))
            c.Response().Header().Set("Access-Control-Allow-Credentials", "true")
            c.Response().Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS, PATCH")
            c.Response().Header().Set("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, Authorization, Cookie, Cookies, X-Local-Token")
            return next(c)
        }
    })

    // Fake timeout
    // e.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
    //     return func(c echo.Context) error {
    //         time.Sleep(1000 * time.Millisecond)
    //         return next(c)
    //     }
    // })

    // You can use h for intellisense and get the handlers' names
    e.Use(s.AuthMiddleware)

    autogen.RegisterHandlers(e, s)

    if err := e.Start(c.ApiConfig.Port); err != nil {
        return err
    }

    return nil
}