summaryrefslogtreecommitdiff
path: root/utils/shortcuts/flash.go
blob: da1507a79105fcb9a9f069d48743ce06b8333dfa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package shortcuts

import (
	"lain/session"

	"github.com/gofiber/fiber/v2"
)

const flashKey = "__flash__"

func Flash(ctx *fiber.Ctx, data any) error {
	normalized, err := normalizeBind(data)
	if err != nil {
		return err
	}

	return session.Set(ctx, flashKey, normalized)
}

func ConsumeFlash(ctx *fiber.Ctx) (any, error) {
	value, err := session.Get(ctx, flashKey)
	if err != nil || value == nil {
		return nil, err
	}

	if err := session.Delete(ctx, flashKey); err != nil {
		return nil, err
	}

	return value, nil
}