context.go

package service

import (
	"encoding/json"
	"net/http"

	"github.com/gin-contrib/sessions"
	"github.com/gin-gonic/gin"
	"github.com/hhandhuan/ku-bbs/pkg/config"

	"github.com/hhandhuan/ku-bbs/internal/model"
)

const (
	errKey     = "err"
	msgKey     = "msg"
	dataKey    = "data"
	flashKey   = "flash"
	userKey    = "user"
	unreadKey  = "unread"
	versionKey = "version"
	nameKey    = "name"
	titleKey   = "title"
)

func Context(ctx *gin.Context) *BaseContext {
	stx := &BaseContext{
		Ctx:     ctx,
		path:    "/",
		session: sessions.Default(ctx),
		name:    config.GetInstance().App.Name,
	}
	return stx
}

type BaseContext struct {
	Ctx     *gin.Context
	session sessions.Session
	path    string
	name    string
	title   string
}

// Redirect 处理跳转
func (c *BaseContext) Redirect() {
	c.Ctx.Redirect(http.StatusFound, c.path)
}

// clear 清除闪存消息
func (c *BaseContext) clear() {
	c.session.Delete(errKey)
	c.session.Delete(msgKey)
	c.session.Delete(flashKey)
	_ = c.session.Save()
}

// Back 返回上一页
func (c *BaseContext) Back() *BaseContext {
	c.path = c.Ctx.Request.RequestURI
	return c
}

// To 设置跳转路径
func (c *BaseContext) To(to string) *BaseContext {
	c.path = to
	return c
}

// WithData 闪存消息
func (c *BaseContext) WithData(data interface{}) *BaseContext {
	r, _ := json.Marshal(data)
	c.session.Set(flashKey, string(r))
	_ = c.session.Save()
	return c
}

// ParseFlash 解析闪存数据
func (c *BaseContext) ParseFlash() map[string]interface{} {
	flashData := make(map[string]interface{})
	if str := c.session.Get(flashKey); str != nil {
		if v, ok := str.(string); ok {
			_ = json.Unmarshal([]byte(v), &flashData)
		}
	}
	return flashData
}

// WithError 错误信息跳转
func (c *BaseContext) WithError(err interface{}) *BaseContext {
	errStr := ""
	switch v := err.(type) {
	case error:
		errStr = v.Error()
	case string:
		errStr = v
	}
	c.session.Set(errKey, errStr)
	_ = c.session.Save()
	return c
}

// WithMsg 提示消息跳转
func (c *BaseContext) WithMsg(msg string) *BaseContext {
	c.session.Set(msgKey, msg)
	_ = c.session.Save()
	return c
}

// SetAuth 设置授权
func (c *BaseContext) SetAuth(users model.Users) {
	s, _ := json.Marshal(users)
	c.session.Set(userKey, string(s))
	_ = c.session.Save()
}

// Auth 获取授权
func (c *BaseContext) Auth() *model.Users {
	var user *model.Users
	str := c.session.Get(userKey)
	if str == nil {
		return user
	}
	if v, ok := str.(string); ok {
		_ = json.Unmarshal([]byte(v), &user)
	}
	return user
}

// Refresh 刷新授权
func (c *BaseContext) Refresh() {
	var user model.Users
	model.User().Where("id", c.Auth().ID).Find(&user)
	c.SetAuth(user)
}

// Check 检查授权
func (c *BaseContext) Check() bool {
	user := c.Auth()
	if user == nil {
		return false
	} else {
		return user.ID > 0
	}
}

// IsAdmin 检查授权
func (c *BaseContext) IsAdmin() bool {
	user := c.Auth()
	if user == nil {
		return false
	} else {
		return user.IsAdmin > 0
	}
}

// Forget 清除授权
func (c *BaseContext) Forget() {
	c.session.Delete(userKey)
	_ = c.session.Save()
}

// SetTitle 设置模版标题
func (c *BaseContext) SetTitle(title string) *BaseContext {
	c.title = title
	return c
}

// unread 消息未读数
func (c *BaseContext) unread() bool {
	if !c.Check() {
		return false
	}
	var (
		remind *model.Reminds
		notice *model.SystemUserNotices
	)

	UID := c.Auth().ID

	// 提醒消息
	r := model.Remind().Where("receiver", UID).Where("readed_at is null").Find(&remind)
	if r.Error == nil && r.RowsAffected > 0 {
		return true
	}

	// 未读系统消息
	s := model.SystemUserNotice().Where("user_id", UID).Where("readed_at is null").Find(&notice)
	if s.Error == nil && s.RowsAffected > 0 {
		return true
	}

	return false
}

// View 模版返回
func (c *BaseContext) View(tpl string, data interface{}) {
	obj := gin.H{
		versionKey: config.GetInstance().App.Version,
		errKey:     c.session.Get(errKey),
		msgKey:     c.session.Get(msgKey),
		userKey:    c.Auth(),
		unreadKey:  c.unread(),
		dataKey:    data,
		flashKey:   c.ParseFlash(),
		nameKey:    c.name,
		titleKey:   c.title,
	}
	c.clear()
	c.Ctx.HTML(http.StatusOK, tpl, obj)
}

// Json 通用 JSON 响应
func (c *BaseContext) Json(data interface{}) {
	c.Ctx.JSON(http.StatusOK, data)
}

// MDFileJson markdown 上传图片响应
func (c *BaseContext) MDFileJson(ok int, msg, url string) {
	c.Json(gin.H{"success": ok, "message": msg, "url": url})
}