refactor: eiblog

pkg/mid/language.go

@@ -0,0 +1,55 @@
+// Package mid provides ...
+package mid
+
+import (
+	"strings"
+
+	"github.com/gin-gonic/gin"
+)
+
+// LangOpts 语言选项
+type LangOpts struct {
+	CookieName string
+	Default    string
+	Supported  []string
+}
+
+// isExist language
+func (opts LangOpts) isExist(l string) bool {
+	for _, v := range opts.Supported {
+		if v == l {
+			return true
+		}
+	}
+	return false
+}
+
+// LangMiddleware set language
+func LangMiddleware(opts LangOpts) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		lang, err := c.Cookie(opts.CookieName)
+		// found cookie
+		if err == nil {
+			c.Set(opts.CookieName, lang)
+			return
+		}
+		// set cookie
+		al := strings.ToLower(c.GetHeader("Accept-Language"))
+		if al != "" {
+			// choose default if not supported
+			lang = opts.Default
+
+			langs := strings.Split(al, ",")
+			for _, v := range langs {
+				if opts.isExist(v) {
+					lang = v
+					break
+				}
+			}
+		} else {
+			lang = opts.Default
+		}
+		c.SetCookie(opts.CookieName, lang, 86400*365, "/", "", false, false)
+		c.Set(opts.CookieName, lang)
+	}
+}

pkg/mid/session.go

@@ -0,0 +1,34 @@
+// Package mid provides ...
+package mid
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-contrib/sessions/cookie"
+	"github.com/gin-gonic/gin"
+)
+
+// SessionOpts 设置选项
+type SessionOpts struct {
+	Name   string
+	Secure bool   // required
+	Secret []byte // required
+	// redis store
+	RedisAddr string
+	RedisPwd  string
+}
+
+// SessionMiddleware session中间件
+func SessionMiddleware(opts SessionOpts) gin.HandlerFunc {
+	store := cookie.NewStore(opts.Secret)
+	store.Options(sessions.Options{
+		MaxAge:   86400 * 30,
+		Path:     "/",
+		Secure:   opts.Secure,
+		HttpOnly: true,
+	})
+	name := "SESSIONID"
+	if opts.Name != "" {
+		name = opts.Name
+	}
+	return sessions.Sessions(name, store)
+}

pkg/mid/u.go

@@ -0,0 +1,18 @@
+// Package mid provides ...
+package mid
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/gofrs/uuid"
+)
+
+// UserMiddleware 用户cookie标记
+func UserMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		cookie, err := c.Cookie("u")
+		if err != nil || cookie == "" {
+			u1 := uuid.Must(uuid.NewV4()).String()
+			c.SetCookie("u", u1, 86400*730, "/", "", true, true)
+		}
+	}
+}