iedc-go/example/app/wechath5.go

package app

import (
	. "code.hoteas.com/golang/hotime"
	. "code.hoteas.com/golang/hotime/common"
	"code.hoteas.com/golang/hotime/dri/wechat"
)

var Wechath5 = Ctr{

	//微信注册，0已经完整的注册了，1还没有注册
	"code": func(that *Context) {

		if that.Req.FormValue("code") == "" {
			that.Display(3, "参数不足")
			return
		}

		appid, resToken, userInfo, err := wechat.H5Program.GetUserInfo(that.Req.FormValue("code"))

		if err != nil {
			that.Display(4, err)
			return
		}
		//此次获取的微信信息
		wechatInfo := Map{
			"openid":         userInfo.OpenID,
			"acttoken":       resToken.AccessToken,
			"retoken":        resToken.RefreshToken,
			"appid":          appid,
			"unionid":        userInfo.Unionid,
			"nickname":       userInfo.Nickname,
			"avatar":         userInfo.HeadImgURL,
			"create_time[#]": "now()",
			"modify_time[#]": "now()",
			"del_flag":       0,
			"type":           1,
		}

		userId := 0

		//最后验证服务商是否绑定
		defer func() {
			//有sn就关联业务员
			sn := that.Req.FormValue("sn")
			if sn == "" {
				return
			}
			if userId == 0 {
				return
			}

			salesman := that.Db.Get("salesman", "*", Map{"sn": sn})
			if salesman.GetCeilInt("id") == 0 {
				return
			}

			user := that.Db.Get("user", "*", Map{"id": userId})
			if user == nil {
				return
			}
			//用户都有企服人员关联
			if user.GetCeilInt("salesman_id") != 0 {

				return
			}
			//用户没有企服商id
			that.Db.Update("user", Map{"salesman_id": salesman.GetCeilInt64("id"), "provider_id": salesman.GetCeilInt64("provider_id")}, Map{"id": user.GetCeilInt64("id")})
			if user.GetCeilInt("company_id") == 0 {
				return
			}
			//绑定企业
			that.Db.Update("company", Map{"salesman_id": salesman.GetCeilInt64("id"), "provider_id": salesman.GetCeilInt64("provider_id")}, Map{"id": user.GetCeilInt64("company_id")})

		}()

		//如果有则直接返回用户信息到微信小程序里
		defer func() {
			if userId != 0 {
				user := that.Db.Get("user", "*", Map{"id": userId})
				if user == nil {
					that.Display(4, "获取个人信息失败")
					return
				}
				delete(user, "password")
				company := that.Db.Get("company", "id,name", Map{"id": user.GetCeilInt("company_id")})
				salesman := that.Db.Get("salesman", "id,name", Map{"id": user.GetCeilInt("salesman_id")})
				provider := that.Db.Get("provider", "id,name", Map{"id": user.GetCeilInt("provider_id")})

				user["company"] = company
				user["salesman"] = salesman
				user["provider"] = provider

				that.Display(0, user)
			}
		}()

		wechat := that.Db.Get("wechat", "*", Map{"AND": Map{"openid": userInfo.OpenID, "del_flag": 0}})

		if wechat != nil {
			//有用户直接返回
			if wechat.GetCeilInt("user_id") != 0 {

				that.Db.Update("user", Map{"login_time[#]": "now()"}, Map{"id": wechat.GetCeilInt("user_id")})
				that.Session("wechat_id", wechat.GetCeilInt("id"))
				that.Session("user_id", wechat.GetCeilInt("user_id"))

				userId = wechat.GetCeilInt("user_id")
				//that.Display(0, "登录成功")
				return
			}
			//没有用户继续查询数据库看是否有其他unionid
			wechat1 := that.Db.Get("wechat", "*", Map{"AND": Map{"unionid": userInfo.Unionid, "openid[!]": userInfo.OpenID, "user_id[>]": 0, "del_flag": 0}})

			//其他表有该数据,则更新当前表数据信息
			if wechat1 != nil {

				wechatInfo["user_id"] = wechat1.GetCeilInt("user_id")
				that.Db.Update("wechat", wechatInfo, Map{"id": wechat.GetCeilInt("id")})

				that.Db.Update("user", Map{"login_time[#]": "now()"}, Map{"id": wechat1.GetCeilInt("user_id")})
				that.Session("wechat_id", wechat.GetCeilInt("id"))
				that.Session("user_id", wechatInfo.GetCeilInt("user_id"))

				userId = wechatInfo.GetCeilInt("user_id")
				//that.Display(0, "登录成功")
				return
			}

			//其他表也没有当前信息，则生成user表，并更新当前用户,一般不会走到这里来
			user := Map{
				"nickname":       wechatInfo.GetString("nickname"),
				"avatar":         wechatInfo.GetString("avatar"),
				"create_time[#]": "now()",
				"modify_time[#]": "now()",
				"login_time[#]":  "now()",
				"del_flag":       0,
			}
			user["id"] = that.Db.Insert("user", user)
			//创建企业
			user["company_id"] = that.Db.Insert("company", Map{
				"user_id":        user["id"],
				"create_time[#]": "now()",
				"modify_time[#]": "now()",
				"del_flag":       0,
			})
			that.Db.Update("user", Map{"company_id": user["company_id"]}, Map{"id": user["id"]})

			wechatInfo["user_id"] = user.GetCeilInt("id")

			that.Db.Update("wechat", wechatInfo, Map{"id": wechat.GetCeilInt("id")})

			that.Session("wechat_id", wechat.GetCeilInt("id"))
			that.Session("user_id", wechatInfo.GetCeilInt("user_id"))

			userId = wechatInfo.GetCeilInt("user_id")
			//that.Display(0, "登录成功")
			return
		}
		user := Map{
			"nickname":       wechatInfo.GetString("nickname"),
			"avatar":         wechatInfo.GetString("avatar"),
			"create_time[#]": "now()",
			"modify_time[#]": "now()",
			"login_time[#]":  "now()",
			"del_flag":       0,
		}
		user["id"] = that.Db.Insert("user", user)
		user["company_id"] = that.Db.Insert("company", Map{
			"user_id":        user["id"],
			"create_time[#]": "now()",
			"modify_time[#]": "now()",
			"del_flag":       0,
		})

		that.Db.Update("user", Map{"company_id": user["company_id"]}, Map{"id": user["id"]})

		wechatInfo["user_id"] = user.GetCeilInt("id")

		wechatInfo["id"] = that.Db.Insert("wechat", wechatInfo)

		if wechatInfo.GetCeilInt("id") == 0 {
			that.Display(4, "创建用户失败")
			return
		}

		that.Session("wechat_id", wechatInfo.GetCeilInt("id"))
		that.Session("user_id", wechatInfo.GetCeilInt("user_id"))

		userId = wechatInfo.GetCeilInt("user_id")
		//that.Display(0, "登录成功")

	},
	//网页签名
	"sign": func(that *Context) {

		signUrl := that.Req.FormValue("url")
		if signUrl == "" {
			that.Display(3, "参数不足")
			return
		}

		cfg1, e := wechat.H5Program.GetSignUrl(signUrl)

		if e != nil {
			that.Display(4, e)
			return
		}

		sign := Map{
			"appId":     cfg1.AppID,
			"timestamp": cfg1.Timestamp,
			"nonceStr":  cfg1.NonceStr,
			"signature": cfg1.Signature,
		}

		that.Display(0, sign)

	},
}