package msg

import (
	"github.com/gorilla/sessions"
	"github.com/labstack/echo-contrib/session"
	"github.com/labstack/echo/v4"
)

type Type string

const (
	TypeSuccess Type = "success"
	TypeInfo    Type = "info"
	TypeWarning Type = "warning"
	TypeDanger  Type = "danger"
)

// TODO: Error handling and cleanup

func Success(c echo.Context, message string) {
	Set(c, TypeSuccess, message)
}

func Info(c echo.Context, message string) {
	Set(c, TypeInfo, message)
}

func Warning(c echo.Context, message string) {
	Set(c, TypeWarning, message)
}

func Danger(c echo.Context, message string) {
	Set(c, TypeDanger, message)
}

// Set adds a new message into the cookie storage.
func Set(c echo.Context, typ Type, message string) {
	sess := getSession(c)
	sess.AddFlash(message, string(typ))
	_ = sess.Save(c.Request(), c.Response())
}

// Get gets flash messages from the cookie storage.
func Get(c echo.Context, typ Type) []string {
	sess := getSession(c)
	fm := sess.Flashes(string(typ))
	// If we have some messages.
	if len(fm) > 0 {
		_ = sess.Save(c.Request(), c.Response())
		// Initiate a strings slice to return messages.
		var flashes []string
		for _, fl := range fm {
			// Add message to the slice.
			flashes = append(flashes, fl.(string))
		}

		return flashes
	}
	return nil
}

func getSession(c echo.Context) *sessions.Session {
	sess, _ := session.Get("msg", c)
	return sess
}