package msg

import (
	"github.com/gorilla/sessions"
	"github.com/labstack/echo/v4"
	"github.com/camzawacki/personal-site/pkg/log"
	"github.com/camzawacki/personal-site/pkg/session"
)

// Type is a message type.
type Type string

const (
	// TypeSuccess represents a success message type.
	TypeSuccess Type = "success"

	// TypeInfo represents a info message type.
	TypeInfo Type = "info"

	// TypeWarning represents a warning message type.
	TypeWarning Type = "warning"

	// TypeError represents an error message type.
	TypeError Type = "error"
)

const (
	// sessionName stores the name of the session which contains flash messages.
	sessionName = "msg"
)

// Success sets a success flash message.
func Success(ctx echo.Context, message string) {
	Set(ctx, TypeSuccess, message)
}

// Info sets an info flash message.
func Info(ctx echo.Context, message string) {
	Set(ctx, TypeInfo, message)
}

// Warning sets a warning flash message.
func Warning(ctx echo.Context, message string) {
	Set(ctx, TypeWarning, message)
}

// Error sets an error flash message.
func Error(ctx echo.Context, message string) {
	Set(ctx, TypeError, message)
}

// Set adds a new flash message of a given type into the session storage.
// Errors will be logged and not returned.
func Set(ctx echo.Context, typ Type, message string) {
	if sess, err := getSession(ctx); err == nil {
		sess.AddFlash(message, string(typ))
		save(ctx, sess)
	}
}

// Get gets flash messages of a given type from the session storage.
// Errors will be logged and not returned.
func Get(ctx echo.Context, typ Type) []string {
	if sess, err := getSession(ctx); err == nil {
		if flash := sess.Flashes(string(typ)); len(flash) > 0 {
			save(ctx, sess)

			msgs := make([]string, 0, len(flash))
			for _, m := range flash {
				msgs = append(msgs, m.(string))
			}
			return msgs
		}
	}

	return nil
}

// getSession gets the flash message session.
func getSession(ctx echo.Context) (*sessions.Session, error) {
	sess, err := session.Get(ctx, sessionName)
	if err != nil {
		log.Ctx(ctx).Error("cannot load flash message session",
			"error", err,
		)
	}
	return sess, err
}

// save saves the flash message session.
func save(ctx echo.Context, sess *sessions.Session) {
	if err := sess.Save(ctx.Request(), ctx.Response()); err != nil {
		log.Ctx(ctx).Error("failed to set flash message",
			"error", err,
		)
	}
}