Added a basic homepage

internal/msg/msg.go

@@ -0,0 +1,97 @@
+package msg
+
+import (
+	"github.com/gorilla/sessions"
+	"github.com/labstack/echo/v4"
+	"github.com/camzawacki/personal-site/internal/log"
+	"github.com/camzawacki/personal-site/internal/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,
+		)
+	}
+}

internal/msg/msg_test.go

@@ -0,0 +1,46 @@
+package msg
+
+import (
+	"testing"
+
+	"github.com/camzawacki/personal-site/internal/tests"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/labstack/echo/v4"
+)
+
+func TestMsg(t *testing.T) {
+	e := echo.New()
+	ctx, _ := tests.NewContext(e, "/")
+	tests.InitSession(ctx)
+
+	assertMsg := func(typ Type, message string) {
+		ret := Get(ctx, typ)
+		require.Len(t, ret, 1)
+		assert.Equal(t, message, ret[0])
+		ret = Get(ctx, typ)
+		require.Len(t, ret, 0)
+	}
+
+	text := "aaa"
+	Success(ctx, text)
+	assertMsg(TypeSuccess, text)
+
+	text = "bbb"
+	Info(ctx, text)
+	assertMsg(TypeInfo, text)
+
+	text = "ccc"
+	Error(ctx, text)
+	assertMsg(TypeError, text)
+
+	text = "ddd"
+	Warning(ctx, text)
+	assertMsg(TypeWarning, text)
+
+	text = "eee"
+	Set(ctx, TypeSuccess, text)
+	assertMsg(TypeSuccess, text)
+}