init...

2022-03-01 13:50:13 +08:00
parent 8eab9943e9
commit d2cabd7d58
113 changed files with 7925 additions and 0 deletions
@@ -0,0 +1,119 @@
+package v1
+
+import (
+	"cutego/core/api/v1/request"
+	cache2 "cutego/core/cache"
+	"cutego/core/entity"
+	service2 "cutego/core/service"
+	"cutego/pkg/common"
+	"cutego/pkg/page"
+	"cutego/pkg/resp"
+	"cutego/pkg/tree/tree_menu"
+	"cutego/pkg/util"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"strings"
+	"time"
+)
+
+type LoginApi struct {
+	loginService      service2.LoginService
+	roleService       service2.RoleService
+	permissionService service2.PermissionService
+	menuService       service2.MenuService
+	loginInfoService  service2.LoginInfoService
+}
+
+// Login 登录
+func (a LoginApi) Login(c *gin.Context) {
+	loginBody := request.LoginBody{}
+	if c.BindJSON(&loginBody) == nil {
+		m := make(map[string]string)
+		loginStatus, token := a.loginService.Login(loginBody.UserName, loginBody.Password)
+		if loginStatus {
+			a.handleLoginInfo(c, loginBody, loginStatus)
+			// 将token存入到redis中
+			cache2.SetRedisToken(loginBody.UserName, token)
+			m["token"] = token
+			c.JSON(http.StatusOK, resp.Success(m))
+		} else {
+			a.handleLoginInfo(c, loginBody, loginStatus)
+			c.JSON(http.StatusOK, resp.ErrorResp(token))
+		}
+	} else {
+		c.JSON(http.StatusOK, resp.ErrorResp(500, "参数绑定错误"))
+	}
+}
+
+func (a LoginApi) handleLoginInfo(c *gin.Context, body request.LoginBody, loginStatus bool) {
+	status := common.If(loginStatus, "1", "0")
+
+	clientIp := a.loginInfoService.GetRequestClientIp(c)
+	var location string
+	if "127.0.0.1" == clientIp {
+		location = "本机"
+	} else {
+		location = a.loginInfoService.GetLocationByIp(clientIp).Country
+	}
+
+	a.loginInfoService.Save(entity.SysLoginInfo{
+		UserName:      body.UserName,
+		IpAddr:        clientIp,
+		LoginLocation: location,
+		Browser:       c.GetHeader("User-Agent"),
+		OS:            strings.ReplaceAll(c.GetHeader("sec-ch-ua-platform"), "\"", ""),
+		Status:        status.(string),
+		LoginTime:     time.Now(),
+	})
+}
+
+// GetUserInfo 获取用户信息
+func (a LoginApi) GetUserInfo(c *gin.Context) {
+	m := make(map[string]interface{})
+	user := a.loginService.GetCurrentUser(c)
+	// 查询用户角色集合
+	roleKeys := a.permissionService.GetRolePermissionByUserId(user)
+	// 权限集合
+	perms := a.permissionService.GetMenuPermission(user)
+	m["roles"] = roleKeys
+	m["permissions"] = perms
+	m["user"] = user
+	c.JSON(http.StatusOK, resp.Success(m))
+}
+
+// GetRouters 根据用户ID查询菜单
+func (a LoginApi) GetRouters(c *gin.Context) {
+	// 获取当前登录用户
+	user := a.loginService.GetCurrentUser(c)
+	menus := a.menuService.GetMenuTreeByUserId(user)
+	systemMenus := tree_menu.SystemMenus{}
+	systemMenus = *menus
+	array := systemMenus.ConvertToINodeArray(menus)
+	generateTree := tree_menu.GenerateTree(array, nil)
+	c.JSON(http.StatusOK, resp.Success(generateTree))
+}
+
+// GetLoginHistory 根据用户名称查询登录记录
+func (a LoginApi) GetLoginHistory(c *gin.Context) {
+	// 获取当前登录用户
+	user := a.loginService.GetCurrentUser(c)
+	// 配置参数
+	query := request.LoginInfoQuery{}
+	query.UserName = user.UserName
+	// 查询
+	list, i := a.loginInfoService.FindPage(query)
+	success := resp.Success(page.Page{
+		Size:  query.PageSize,
+		Total: i,
+		List:  list,
+	}, "查询成功")
+	c.JSON(http.StatusOK, success)
+}
+
+// Logout 退出登录
+func (a LoginApi) Logout(c *gin.Context) {
+	// 删除Redis缓存
+	name := util.GetUserInfo(c).UserName
+	cache2.RemoveRedisToken(name)
+	resp.OK(c)
+}