展示账号邀请码

controllers/welcome.go

@@ -1,7 +1,14 @@
 package controllers
 
 import (
+	"box-gm/models"
+	"box-gm/utils"
+	"fmt"
+	"log"
+	"time"
+
 	"github.com/astaxie/beego"
+	"github.com/gomodule/redigo/redis"
 )
 
 type WelcomeController struct {
@@ -17,7 +24,71 @@ func (c *WelcomeController) Get() {
 		c.Ctx.Redirect(302, "/accountlogin")
 		return
 	}
+	username := userinfo.(*LoginInfo).Username
 
+	c.Data["rolename"] = models.GetUserRoleName(username)
+	c.Data["username"] = username
+	c.Data["sharecode"] = models.GetUserShareCode(username)
+	c.Data["token"] = models.GetToken()
+	c.Data["sysTime"] = time.Unix(time.Now().Unix(), 0).Format("2006-01-02 15:04:05")
 	//界面模板文件
 	c.TplName = "welcome.tpl"
 }
+
+func (c *WelcomeController) Post() {
+	if !models.CheckToken(c.GetString("token")) {
+		c.Data["json"] = &map[string]interface{}{"status": false, "info": "操作失败，请重新刷新界面"}
+		c.ServeJSON()
+		return
+	}
+
+	userinfo := c.GetSession("user")
+	if userinfo == nil {
+		c.Ctx.Redirect(302, "/accountlogin")
+		return
+	}
+
+	username := userinfo.(*LoginInfo).Username
+	log.Printf("WelcomeController Post username[%s]", username)
+	nUser := models.GetUserByUsername(username)
+	if nUser == nil {
+		c.Data["json"] = &map[string]interface{}{"status": false, "info": fmt.Sprintf("用户[%s]已存在", username)}
+		c.ServeJSON()
+		return
+	}
+
+	sharecode := c.GetString("sharecode")
+	log.Printf("WelcomeController Post sharecode[%s]", sharecode)
+	// 校验邀请码
+	pool := utils.GetAccountRedisPool()
+	if pool == nil {
+		c.Data["json"] = &map[string]interface{}{"status": false, "info": "get redis pool fail"}
+		c.ServeJSON()
+		return
+	}
+	rd := pool.Get()
+	defer rd.Close()
+	result, _ := redis.Bool(rd.Do("sismember", "sharecode", sharecode))
+	log.Printf("WelcomeController Post result[%v]", result)
+	if result == true {
+		c.Data["json"] = &map[string]interface{}{"status": false, "info": fmt.Sprintf("邀请码[%s]已占有!", sharecode)}
+		c.ServeJSON()
+		return
+	}
+
+	nUser.ShareCode = sharecode
+	_, err := models.UpdateUser(nUser)
+	if err != nil {
+		c.Data["json"] = &map[string]interface{}{"status": false, "info": err}
+		c.ServeJSON()
+		return
+	}
+
+	rd.Do("sadd", "sharecode", sharecode)
+	rd.Do("sadd", "sharecode:promoter", sharecode)
+
+	c.Data["json"] = &map[string]interface{}{"status": true, "info": "更新用户成功"}
+	c.ServeJSON()
+
+	go models.SaveLog(username, "修改账号推广码", &map[string]interface{}{"username": username, "ShareCode": nUser.ShareCode})
+}

models/user.go

@@ -15,6 +15,7 @@ type User struct {
 	Role        int64  `form:"role"`
 	CreateTs    int64  `form:"create_ts"`
 	LastLoginTs int64  `form:"last_login_ts"`
+	ShareCode   string `orm:"size(100)" form:"sharecode"`
 }
 
 func init() {
@@ -25,7 +26,7 @@ func (u *User) TableName() string {
 	return "tb_user"
 }
 
-//验证用户信息
+// 验证用户信息
 func checkUser(u *User) (err error) {
 	valid := validation.Validation{}
 	b, _ := valid.Valid(&u)
@@ -38,7 +39,7 @@ func checkUser(u *User) (err error) {
 	return nil
 }
 
-//get user list
+// get user list
 func GetUserlist() (users []orm.Params, count int64) {
 	o := orm.NewOrm()
 	user := new(User)
@@ -48,7 +49,7 @@ func GetUserlist() (users []orm.Params, count int64) {
 	return users, count
 }
 
-//添加用户
+// 添加用户
 func AddUser(u *User) (int64, error) {
 	if err := checkUser(u); err != nil {
 		return 0, err
@@ -59,7 +60,7 @@ func AddUser(u *User) (int64, error) {
 	return id, err
 }
 
-//更新用户
+// 更新用户
 func UpdateUser(u *User) (int64, error) {
 	if err := checkUser(u); err != nil {
 		return 0, err
@@ -70,7 +71,7 @@ func UpdateUser(u *User) (int64, error) {
 		user["Username"] = u.Username
 	}
 	if len(u.Password) > 0 {
-		user["Password"] = u.Password//Strtomd5(u.Password)
+		user["Password"] = u.Password //Strtomd5(u.Password)
 	}
 	if u.Role != 0 {
 		user["Role"] = u.Role
@@ -81,6 +82,9 @@ func UpdateUser(u *User) (int64, error) {
 	if u.LastLoginTs != 0 {
 		user["LastLoginTs"] = u.LastLoginTs
 	}
+	if len(u.ShareCode) > 0 {
+		user["ShareCode"] = u.ShareCode
+	}
 	if len(user) == 0 {
 		return 0, errors.New("update field is empty")
 	}
@@ -127,7 +131,7 @@ func GetPermission(username string) int64 {
 	return p.Permission
 }
 
-//检查用户是否满足权限
+// 检查用户是否满足权限
 func CheckPermission(username string, permission int) bool {
 	user := GetUserByUsername(username)
 	if user == nil {
@@ -139,3 +143,26 @@ func CheckPermission(username string, permission int) bool {
 	}
 	return (p.Permission & (1 << uint(permission))) != 0
 }
+
+// 获取账号推广码
+func GetUserShareCode(username string) string {
+	user := GetUserByUsername(username)
+	log.Printf("GetUserShareCode username[%s] user[%v]", username, user)
+	if user == nil {
+		return ""
+	}
+	return user.ShareCode
+}
+
+// 获取账号角色名称
+func GetUserRoleName(username string) string {
+	user := GetUserByUsername(username)
+	if user == nil {
+		return ""
+	}
+	p := GetRoleById(user.Role)
+	if p == nil {
+		return ""
+	}
+	return p.Name
+}

views/index.tpl

@@ -73,7 +73,7 @@
     <div class="page-content">
         <div class="layui-tab tab" lay-filter="xbs_tab" lay-allowclose="false">
           <ul class="layui-tab-title">
-            <li class="home"><i class="layui-icon">&#xe68e;</i>数据</li>
+            <li class="home"><i class="layui-icon">&#xe68e;</i>账号信息</li>
           </ul>
           <div class="layui-tab-content">
             <div class="layui-tab-item layui-show">

views/welcome.tpl

@@ -9,11 +9,86 @@
         <link rel="shortcut icon" href="/favicon.ico" type="image/x-icon" />
         <link rel="stylesheet" href="./static/css/font.css">
         <link rel="stylesheet" href="./static/css/xadmin.css">
+        
+        <script type="text/javascript" src="https://cdn.bootcss.com/jquery/3.2.1/jquery.min.js"></script>
+        <script src="./static/lib/layui/layui.js" charset="utf-8"></script>
+        <script type="text/javascript" src="./static/js/xadmin.js"></script>
     </head>
     <body>
+    <input type="text" id="token" value={{.token}} hidden=true>
     <div class="x-body layui-anim layui-anim-up">
-        <blockquote class="layui-elem-quote">欢迎管理员：
-            <span class="x-red">test</span>！当前时间:2018-04-25 20:50:53</blockquote>
+        <blockquote class="layui-elem-quote">欢迎{{.rolename}}：
+            <span class="x-red">{{.username}}</span>！当前时间:{{.sysTime}}</blockquote>
     </div>
+    <div class="x-body layui-anim layui-anim-up">
+        <blockquote class="layui-elem-quote">推广码：
+            <input id="sharecode" style="width:128px;" value={{.sharecode}} type="text" class="layui-input">
+            <br>
+            <input value="刷新" class="layui-btn" style="width:128px;" type="button" onclick="randCode()">
+            <br>
+            <br>
+            <input value="更新" class="layui-btn" style="width:128px;" type="button" onclick="UpdateShareCode()">
+        </blockquote>
+    </div>
+
+    <script language="JavaScript">
+      var randChats = ['A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z'];
+      function generateMixed(n) {
+          var res = "";
+          for(var i = 0; i < n ; i ++) {
+              var id = Math.ceil(Math.random()*(randChats.length-1));
+              res += randChats[id];
+          }
+          return res;
+      }
+      // 随机兑换码
+      function randCode() {
+        document.getElementById("sharecode").value = generateMixed(6);
+      }
+      
+      function UpdateShareCode () {
+        var token = document.getElementById("token").value;
+        var sharecode = document.getElementById("sharecode").value;
+
+        if (sharecode.length != 6){
+          alert("兑换码长度错误！6位大写字母");
+          return;
+        }
+        // 兑换码字符检查
+        for (var i = 0; i < sharecode.length; i ++){
+          var isMatch = false
+          for (var j = 0; j < randChats.length; j ++) {
+            if (sharecode[i] == randChats[j]){
+              isMatch = true;
+              break;
+            }
+          }
+          if (isMatch == false){
+            alert("兑换码错误！请使用大写字母");
+            return;
+          }
+        }
+
+        var text = "token=" + encodeURI(token);
+        text += "&sharecode=" + encodeURI(sharecode);
+
+        $.ajax({
+          type:"post",
+          url:"/welcome?" + text,
+          success:function (data) {
+            // alert(data.status);
+            if (data.status){
+              window.location.href="/welcome";
+            }else{
+              if (data.info){
+                alert(data.info);
+              }else{
+                window.location.href="/accountlogin";
+              }
+            }
+          }
+        });
+      }
+    </script>
     </body>
 </html>