mirror of
https://github.com/mindoc-org/mindoc.git
synced 2025-04-05 04:14:55 +08:00
08d0e1613d
* go mod update * feat: change to new wxwork sso login * fix: can't log in by workwx browser * fix: workwx auto regist * fix: change app.conf.example * fix: workwx account can't be disabled * fix: workwx account delete * fix: workwx bind error * feat: optimize wecom login * feat: rewrite dingtalk login * feat: rewrite dingtalk login * feat: optimize auth2 login
1451 lines
44 KiB
Go
1451 lines
44 KiB
Go
package controllers
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"github.com/mindoc-org/mindoc/cache"
|
||
"github.com/mindoc-org/mindoc/utils/auth2"
|
||
"github.com/mindoc-org/mindoc/utils/auth2/dingtalk"
|
||
"github.com/mindoc-org/mindoc/utils/auth2/wecom"
|
||
"html/template"
|
||
"math/rand"
|
||
"net/http"
|
||
"net/url"
|
||
"regexp"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/beego/beego/v2/client/orm"
|
||
"github.com/beego/beego/v2/core/logs"
|
||
"github.com/beego/beego/v2/server/web"
|
||
"github.com/beego/i18n"
|
||
"github.com/lifei6671/gocaptcha"
|
||
"github.com/mindoc-org/mindoc/conf"
|
||
"github.com/mindoc-org/mindoc/mail"
|
||
"github.com/mindoc-org/mindoc/models"
|
||
"github.com/mindoc-org/mindoc/utils"
|
||
)
|
||
|
||
const (
|
||
SessionUserInfoKey = "session-user-info-key"
|
||
AccessTokenCacheKey = "access-token-cache-key"
|
||
)
|
||
|
||
var src = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
|
||
// AccountController 用户登录与注册
|
||
type AccountController struct {
|
||
BaseController
|
||
}
|
||
|
||
func (c *AccountController) referer() string {
|
||
u, _ := url.PathUnescape(c.GetString("url"))
|
||
if u == "" {
|
||
u = conf.URLFor("HomeController.Index")
|
||
}
|
||
return u
|
||
}
|
||
|
||
func (c *AccountController) IsInWorkWeixin() bool {
|
||
ua := c.Ctx.Input.UserAgent()
|
||
var wechatRule = regexp.MustCompile(`\bMicroMessenger\/\d+(\.\d+)*\b`)
|
||
var wxworkRule = regexp.MustCompile(`\bwxwork\/\d+(\.\d+)*\b`)
|
||
return wechatRule.MatchString(ua) && wxworkRule.MatchString(ua)
|
||
}
|
||
|
||
func (c *AccountController) Prepare() {
|
||
c.BaseController.Prepare()
|
||
c.EnableXSRF = web.AppConfig.DefaultBool("enablexsrf", true)
|
||
|
||
c.Data["xsrfdata"] = template.HTML(c.XSRFFormHTML())
|
||
c.Data["CanLoginWorkWeixin"] = len(web.AppConfig.DefaultString("workweixin_corpid", "")) > 0
|
||
c.Data["CanLoginDingTalk"] = len(web.AppConfig.DefaultString("dingtalk_app_key", "")) > 0
|
||
|
||
if !c.EnableXSRF {
|
||
return
|
||
}
|
||
if c.Ctx.Input.IsPost() {
|
||
token := c.Ctx.Input.Query("_xsrf")
|
||
if token == "" {
|
||
token = c.Ctx.Request.Header.Get("X-Xsrftoken")
|
||
}
|
||
if token == "" {
|
||
token = c.Ctx.Request.Header.Get("X-Csrftoken")
|
||
}
|
||
if token == "" {
|
||
if c.IsAjax() {
|
||
c.JsonResult(403, i18n.Tr(c.Lang, "message.illegal_request"))
|
||
} else {
|
||
c.ShowErrorPage(403, i18n.Tr(c.Lang, "message.illegal_request"))
|
||
}
|
||
}
|
||
xsrfToken := c.XSRFToken()
|
||
if xsrfToken != token {
|
||
if c.IsAjax() {
|
||
c.JsonResult(403, i18n.Tr(c.Lang, "message.illegal_request"))
|
||
} else {
|
||
c.ShowErrorPage(403, i18n.Tr(c.Lang, "message.illegal_request"))
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// Login 用户登录
|
||
func (c *AccountController) Login() {
|
||
c.TplName = "account/login.tpl"
|
||
|
||
if member, ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0 {
|
||
u := c.GetString("url")
|
||
if u == "" {
|
||
u = c.Ctx.Request.Header.Get("Referer")
|
||
}
|
||
if u == "" {
|
||
u = conf.URLFor("HomeController.Index")
|
||
}
|
||
c.Redirect(u, 302)
|
||
}
|
||
var remember CookieRemember
|
||
// 如果 Cookie 中存在登录信息
|
||
if cookie, ok := c.GetSecureCookie(conf.GetAppKey(), "login"); ok {
|
||
if err := utils.Decode(cookie, &remember); err == nil {
|
||
if member, err := models.NewMember().Find(remember.MemberId); err == nil {
|
||
c.SetMember(*member)
|
||
c.LoggedIn(false)
|
||
c.StopRun()
|
||
}
|
||
}
|
||
}
|
||
|
||
if c.Ctx.Input.IsPost() {
|
||
account := c.GetString("account")
|
||
password := c.GetString("password")
|
||
captcha := c.GetString("code")
|
||
isRemember := c.GetString("is_remember")
|
||
|
||
// 如果开启了验证码
|
||
if v, ok := c.Option["ENABLED_CAPTCHA"]; ok && strings.EqualFold(v, "true") {
|
||
v, ok := c.GetSession(conf.CaptchaSessionName).(string)
|
||
if !ok || !strings.EqualFold(v, captcha) {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.captcha_wrong"))
|
||
}
|
||
}
|
||
|
||
if account == "" || password == "" {
|
||
c.JsonResult(6002, i18n.Tr(c.Lang, "message.account_or_password_empty"))
|
||
}
|
||
|
||
member, err := models.NewMember().Login(account, password)
|
||
if err == nil {
|
||
member.LastLoginTime = time.Now()
|
||
_ = member.Update("last_login_time")
|
||
|
||
c.SetMember(*member)
|
||
|
||
if strings.EqualFold(isRemember, "yes") {
|
||
remember.MemberId = member.MemberId
|
||
remember.Account = member.Account
|
||
remember.Time = time.Now()
|
||
v, err := utils.Encode(remember)
|
||
if err == nil {
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30).Unix())
|
||
}
|
||
}
|
||
|
||
c.JsonResult(0, "ok", c.referer())
|
||
} else {
|
||
logs.Error("用户登录 ->", err)
|
||
c.JsonResult(500, i18n.Tr(c.Lang, "message.wrong_account_password"), nil)
|
||
}
|
||
return
|
||
}
|
||
|
||
referer := c.referer()
|
||
u := c.GetString("url")
|
||
if u == "" {
|
||
u = referer
|
||
if u == "" {
|
||
u = conf.BaseUrl
|
||
}
|
||
} else {
|
||
var schemaRule = regexp.MustCompile(`^https?\:\/\/`)
|
||
if !schemaRule.MatchString(u) {
|
||
u = conf.BaseUrl + u
|
||
}
|
||
}
|
||
c.Data["url"] = referer
|
||
|
||
auth2Redirect := "AccountController.Auth2Redirect"
|
||
if can, _ := c.Data["CanLoginWorkWeixin"].(bool); can {
|
||
c.Data["workweixin_login_url"] = conf.URLFor(auth2Redirect, ":app", wecom.AppName, "url", url.PathEscape(u))
|
||
}
|
||
|
||
if can, _ := c.Data["CanLoginDingTalk"].(bool); can {
|
||
c.Data["dingtalk_login_url"] = conf.URLFor(auth2Redirect, ":app", dingtalk.AppName, "url", url.PathEscape(u))
|
||
|
||
}
|
||
return
|
||
}
|
||
|
||
/*
|
||
Auth2.0 第三方对接思路:
|
||
1. Auth2Redirect: 点击相应第三方接口，路由重定向至第三方提供的Auth2.0地址
|
||
2. Auth2Callback: 第三方回调处理，接收回调的授权码，并获取用户信息
|
||
已绑定: 则读取用户信息，直接登录
|
||
未绑定: 则弹窗提示（需要敏感信息）
|
||
a) Auth2BindAccount: 绑定已有账户（用户名+密码）
|
||
b) Auth2AutoAccount: 自动创建账户，以第三方用户ID作为用户名，密码123456。
|
||
用该方式创建的账户，无法使用账号密码登录，需要修改一次密码后才可以进行账号密码登录。
|
||
*/
|
||
|
||
func (c *AccountController) getAuth2Client() (auth2.Client, error) {
|
||
app := c.Ctx.Input.Param(":app")
|
||
var client auth2.Client
|
||
tokenKey := AccessTokenCacheKey + "-" + app
|
||
|
||
switch app {
|
||
case wecom.AppName:
|
||
if can, _ := c.Data["CanLoginWorkWeixin"].(bool); !can {
|
||
return nil, errors.New("auth2.client.wecom.disabled")
|
||
}
|
||
corpId, _ := web.AppConfig.String("workweixin_corpid")
|
||
agentId, _ := web.AppConfig.String("workweixin_agentid")
|
||
secret, _ := web.AppConfig.String("workweixin_secret")
|
||
client = wecom.NewClient(corpId, agentId, secret)
|
||
|
||
case dingtalk.AppName:
|
||
if can, _ := c.Data["CanLoginDingTalk"].(bool); !can {
|
||
return nil, errors.New("auth2.client.dingtalk.disabled")
|
||
}
|
||
|
||
appKey, _ := web.AppConfig.String("dingtalk_app_key")
|
||
appSecret, _ := web.AppConfig.String("dingtalk_app_secret")
|
||
client = dingtalk.NewClient(appSecret, appKey)
|
||
|
||
default:
|
||
return nil, errors.New("auth2.client.notsupported")
|
||
}
|
||
|
||
var tokenCache auth2.AccessTokenCache
|
||
err := cache.Get(tokenKey, &tokenCache)
|
||
if err != nil {
|
||
logs.Info("AccessToken从缓存读取失败")
|
||
token, err := client.GetAccessToken(context.Background())
|
||
if err != nil {
|
||
return client, nil
|
||
}
|
||
tokenCache = auth2.NewAccessToken(token)
|
||
cache.Put(tokenKey, tokenCache, tokenCache.GetExpireIn())
|
||
}
|
||
|
||
// 处理过期Token
|
||
if tokenCache.IsExpired() {
|
||
token, err := client.GetAccessToken(context.Background())
|
||
if err != nil {
|
||
return client, nil
|
||
}
|
||
tokenCache = auth2.NewAccessToken(token)
|
||
cache.Put(tokenKey, tokenCache, tokenCache.GetExpireIn())
|
||
}
|
||
|
||
client.SetAccessToken(tokenCache)
|
||
return client, nil
|
||
}
|
||
|
||
func (c *AccountController) parseAuth2CallbackParam() (code, state string) {
|
||
switch c.Ctx.Input.Param(":app") {
|
||
case wecom.AppName:
|
||
code = c.GetString("code")
|
||
state = c.GetString("state")
|
||
case dingtalk.AppName:
|
||
code = c.GetString("authCode")
|
||
state = c.GetString("state")
|
||
}
|
||
|
||
logs.Debug("code: ", code)
|
||
logs.Debug("state: ", state)
|
||
return
|
||
}
|
||
|
||
func (c *AccountController) getAuth2Account() (models.Auth2Account, error) {
|
||
switch c.Ctx.Input.Param(":app") {
|
||
case wecom.AppName:
|
||
return models.NewWorkWeixinAccount(), nil
|
||
|
||
case dingtalk.AppName:
|
||
return models.NewDingTalkAccount(), nil
|
||
}
|
||
|
||
return nil, errors.New("auth2.account.notsupported")
|
||
}
|
||
|
||
// Auth2Redirect 第三方auth2.0登录: 钉钉、企业微信
|
||
func (c *AccountController) Auth2Redirect() {
|
||
client, err := c.getAuth2Client()
|
||
if err != nil {
|
||
c.DelSession(conf.LoginSessionName)
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
c.StopRun()
|
||
return
|
||
}
|
||
|
||
app := c.Ctx.Input.Param(":app")
|
||
var isAppBrowser bool
|
||
switch app {
|
||
case wecom.AppName:
|
||
isAppBrowser = c.IsInWorkWeixin()
|
||
}
|
||
|
||
var callback string
|
||
u := c.GetString("url")
|
||
if u == "" {
|
||
u = c.referer()
|
||
callback = conf.URLFor("AccountController.Auth2Callback", ":app", app)
|
||
}
|
||
if u != "" {
|
||
var schemaRule = regexp.MustCompile(`^https?\:\/\/`)
|
||
if !schemaRule.MatchString(u) {
|
||
u = strings.TrimRight(conf.BaseUrl, "/") + strings.TrimLeft(u, "/")
|
||
}
|
||
callback = conf.URLFor("AccountController.Auth2Callback", ":app", app, "url", url.PathEscape(u))
|
||
}
|
||
|
||
logs.Debug("callback: ", callback) // debug
|
||
c.Redirect(client.BuildURL(callback, isAppBrowser), http.StatusFound)
|
||
}
|
||
|
||
// Auth2Callback 第三方auth2.0回调
|
||
func (c *AccountController) Auth2Callback() {
|
||
client, err := c.getAuth2Client()
|
||
if err != nil {
|
||
c.DelSession(conf.LoginSessionName)
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
c.StopRun()
|
||
logs.Error(err)
|
||
return
|
||
}
|
||
|
||
if member, ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0 {
|
||
u := c.GetString("url")
|
||
if u == "" {
|
||
u = c.Ctx.Request.Header.Get("Referer")
|
||
}
|
||
if u == "" {
|
||
u = conf.URLFor("HomeController.Index")
|
||
}
|
||
member, err := models.NewMember().Find(member.MemberId)
|
||
if err != nil {
|
||
c.DelSession(conf.LoginSessionName)
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
} else {
|
||
c.SetMember(*member)
|
||
}
|
||
c.Redirect(u, 302)
|
||
}
|
||
|
||
var remember CookieRemember
|
||
// 如果 Cookie 中存在登录信息
|
||
if cookie, ok := c.GetSecureCookie(conf.GetAppKey(), "login"); ok {
|
||
if err := utils.Decode(cookie, &remember); err == nil {
|
||
if member, err := models.NewMember().Find(remember.MemberId); err == nil {
|
||
c.SetMember(*member)
|
||
c.LoggedIn(false)
|
||
c.StopRun()
|
||
}
|
||
}
|
||
}
|
||
|
||
c.TplName = "account/auth2_callback.tpl"
|
||
bindExisted := "false"
|
||
errMsg := ""
|
||
userInfoJson := "{}"
|
||
defer func() {
|
||
c.Data["bind_existed"] = template.JS(bindExisted)
|
||
logs.Debug("bind_existed: ", bindExisted)
|
||
c.Data["error_msg"] = template.JS(errMsg)
|
||
c.Data["user_info_json"] = template.JS(userInfoJson)
|
||
c.Data["app"] = template.JS(c.Ctx.Input.Param(":app"))
|
||
}()
|
||
|
||
// 请求参数获取
|
||
code, state := c.parseAuth2CallbackParam()
|
||
if err := client.ValidateCallback(state); err != nil {
|
||
c.DelSession(conf.LoginSessionName)
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
errMsg = err.Error()
|
||
logs.Error(err)
|
||
return
|
||
}
|
||
|
||
userInfo, err := client.GetUserInfo(context.Background(), code)
|
||
if err != nil {
|
||
c.DelSession(conf.LoginSessionName)
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
errMsg = err.Error()
|
||
logs.Error(err)
|
||
return
|
||
}
|
||
|
||
account, err := c.getAuth2Account()
|
||
if err != nil {
|
||
logs.Error("获取Auth2用户失败 ->", err)
|
||
c.JsonResult(500, "不支持的第三方用户", nil)
|
||
return
|
||
}
|
||
|
||
member, err := account.ExistedMember(userInfo.UserId)
|
||
if err != nil {
|
||
if err == orm.ErrNoRows {
|
||
if userInfo.Mobile == "" {
|
||
errMsg = "请到应用浏览器中登录，并授权获取敏感信息。"
|
||
} else {
|
||
jsonInfo, _ := json.Marshal(userInfo)
|
||
userInfoJson = string(jsonInfo)
|
||
errMsg = ""
|
||
c.SetSession(SessionUserInfoKey, userInfo)
|
||
}
|
||
} else {
|
||
logs.Error("Error: ", err)
|
||
errMsg = "登录错误: " + err.Error()
|
||
}
|
||
return
|
||
}
|
||
|
||
bindExisted = "true"
|
||
errMsg = ""
|
||
|
||
member.LastLoginTime = time.Now()
|
||
_ = member.Update("last_login_time")
|
||
|
||
c.SetMember(*member)
|
||
remember.MemberId = member.MemberId
|
||
remember.Account = member.Account
|
||
remember.Time = time.Now()
|
||
v, err := utils.Encode(remember)
|
||
if err == nil {
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
}
|
||
u := c.GetString("url")
|
||
if u == "" {
|
||
u = conf.URLFor("HomeController.Index")
|
||
}
|
||
c.Redirect(u, 302)
|
||
|
||
}
|
||
|
||
// Auth2BindAccount 第三方auth2.0绑定已有账号
|
||
func (c *AccountController) Auth2BindAccount() {
|
||
userInfo, ok := c.GetSession(SessionUserInfoKey).(auth2.UserInfo)
|
||
if !ok || len(userInfo.UserId) <= 0 {
|
||
c.DelSession(SessionUserInfoKey)
|
||
c.JsonResult(400, "请求错误, 请从首页重新登录")
|
||
return
|
||
}
|
||
|
||
account := c.GetString("account")
|
||
password := c.GetString("password")
|
||
if account == "" || password == "" {
|
||
c.JsonResult(400, "账号或密码不能为空")
|
||
return
|
||
}
|
||
|
||
member, err := models.NewMember().Login(account, password)
|
||
if err != nil {
|
||
logs.Error("用户登录 ->", err)
|
||
c.JsonResult(500, "账号或密码错误", nil)
|
||
return
|
||
}
|
||
|
||
bindAccount, err := c.getAuth2Account()
|
||
if err != nil {
|
||
logs.Error("获取Auth2用户失败 ->", err)
|
||
c.JsonResult(500, "不支持的第三方用户", nil)
|
||
return
|
||
}
|
||
|
||
member.CreateAt = 0
|
||
ormer := orm.NewOrm()
|
||
o, err := ormer.Begin()
|
||
if err != nil {
|
||
logs.Error("开启事务时出错 -> ", err)
|
||
c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
return
|
||
}
|
||
if err := bindAccount.AddBind(ormer, userInfo, member); err != nil {
|
||
logs.Error(err)
|
||
o.Rollback()
|
||
c.JsonResult(500, "绑定失败，数据库错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
// 绑定成功之后修改用户信息
|
||
member.LastLoginTime = time.Now()
|
||
//member.RealName = user_info.Name
|
||
//member.Avatar = user_info.Avatar
|
||
if len(member.Avatar) < 1 {
|
||
member.Avatar = conf.GetDefaultAvatar()
|
||
}
|
||
//member.Email = user_info.Email
|
||
//member.Phone = user_info.Mobile
|
||
if _, err := ormer.Update(member, "last_login_time", "real_name", "avatar", "email", "phone"); err != nil {
|
||
o.Rollback()
|
||
logs.Error("保存用户信息失败=>", err)
|
||
c.JsonResult(500, "绑定失败，现有账户信息更新失败: "+err.Error())
|
||
return
|
||
|
||
}
|
||
|
||
if err := o.Commit(); err != nil {
|
||
logs.Error("开启事务时出错 -> ", err)
|
||
c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
return
|
||
}
|
||
|
||
c.DelSession(SessionUserInfoKey)
|
||
c.SetMember(*member)
|
||
|
||
var remember CookieRemember
|
||
remember.MemberId = member.MemberId
|
||
remember.Account = member.Account
|
||
remember.Time = time.Now()
|
||
v, err := utils.Encode(remember)
|
||
if err != nil {
|
||
c.JsonResult(500, "绑定成功, 但自动登录失败, 请返回首页重新登录", nil)
|
||
return
|
||
}
|
||
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
c.JsonResult(0, "绑定成功", nil)
|
||
}
|
||
|
||
// Auth2AutoAccount auth2.0自动创建账号
|
||
func (c *AccountController) Auth2AutoAccount() {
|
||
app := c.Ctx.Input.Param(":app")
|
||
logs.Debug("app: ", app)
|
||
|
||
userInfo, ok := c.GetSession(SessionUserInfoKey).(auth2.UserInfo)
|
||
if !ok || len(userInfo.UserId) <= 0 {
|
||
c.DelSession(SessionUserInfoKey)
|
||
c.JsonResult(400, "请求错误, 请从首页重新登录")
|
||
return
|
||
}
|
||
|
||
c.DelSession(SessionUserInfoKey)
|
||
member := models.NewMember()
|
||
|
||
if _, err := member.FindByAccount(userInfo.UserId); err == nil && member.MemberId > 0 {
|
||
c.JsonResult(400, "账号已存在")
|
||
return
|
||
}
|
||
|
||
ormer := orm.NewOrm()
|
||
o, err := ormer.Begin()
|
||
if err != nil {
|
||
logs.Error("开启事务时出错 -> ", err)
|
||
c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
return
|
||
}
|
||
|
||
member.Account = userInfo.UserId
|
||
member.RealName = userInfo.Name
|
||
member.Password = "123456" // 强制设置默认密码，需修改一次密码后，才可以进行账号密码登录
|
||
hash, err := utils.PasswordHash(member.Password)
|
||
|
||
if err != nil {
|
||
logs.Error("加密用户密码失败 =>", err)
|
||
c.JsonResult(500, "加密用户密码失败"+err.Error())
|
||
return
|
||
}
|
||
|
||
logs.Debug("member.Password: ", member.Password)
|
||
logs.Debug("hash: ", hash)
|
||
member.Password = hash
|
||
|
||
member.Role = conf.MemberGeneralRole
|
||
member.Avatar = userInfo.Avatar
|
||
if len(member.Avatar) < 1 {
|
||
member.Avatar = conf.GetDefaultAvatar()
|
||
}
|
||
member.CreateAt = 0
|
||
member.Email = userInfo.Mail
|
||
member.Phone = userInfo.Mobile
|
||
member.Status = 0
|
||
if _, err = ormer.Insert(member); err != nil {
|
||
o.Rollback()
|
||
c.JsonResult(500, "注册失败，数据库错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
account, err := c.getAuth2Account()
|
||
if err != nil {
|
||
logs.Error("获取Auth2用户失败 ->", err)
|
||
c.JsonResult(500, "不支持的第三方用户", nil)
|
||
return
|
||
}
|
||
|
||
member.CreateAt = 0
|
||
if err := account.AddBind(ormer, userInfo, member); err != nil {
|
||
logs.Error(err)
|
||
o.Rollback()
|
||
c.JsonResult(500, "注册失败，数据库错误: "+err.Error())
|
||
return
|
||
}
|
||
|
||
if err := o.Commit(); err != nil {
|
||
logs.Error("提交事务时出错 -> ", err)
|
||
c.JsonResult(500, "提交事务时出错: ", err.Error())
|
||
return
|
||
}
|
||
|
||
member.LastLoginTime = time.Now()
|
||
_ = member.Update("last_login_time")
|
||
|
||
c.SetMember(*member)
|
||
|
||
var remember CookieRemember
|
||
remember.MemberId = member.MemberId
|
||
remember.Account = member.Account
|
||
remember.Time = time.Now()
|
||
v, err := utils.Encode(remember)
|
||
if err != nil {
|
||
c.JsonResult(500, "绑定成功, 但自动登录失败, 请返回首页重新登录", nil)
|
||
return
|
||
}
|
||
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
c.JsonResult(0, "绑定成功", nil)
|
||
}
|
||
|
||
// 钉钉登录
|
||
//func (c *AccountController) DingTalkLogin() {
|
||
// code := c.GetString("dingtalk_code")
|
||
// if code == "" {
|
||
// c.JsonResult(500, i18n.Tr(c.Lang, "message.failed_obtain_user_info"), nil)
|
||
// }
|
||
//
|
||
// appKey, _ := web.AppConfig.String("dingtalk_app_key")
|
||
// appSecret, _ := web.AppConfig.String("dingtalk_app_secret")
|
||
// tmpReader, _ := web.AppConfig.String("dingtalk_tmp_reader")
|
||
//
|
||
// if appKey == "" || appSecret == "" || tmpReader == "" {
|
||
// c.JsonResult(500, i18n.Tr(c.Lang, "message.dingtalk_auto_login_not_enable"), nil)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// dingtalkAgent := dingtalk.NewDingTalkAgent(appSecret, appKey)
|
||
// err := dingtalkAgent.GetAccesstoken()
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉临时Token失败 ->", err)
|
||
// c.JsonResult(500, i18n.Tr(c.Lang, "message.failed_auto_login"), nil)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// userid, err := dingtalkAgent.GetUserIDByCode(code)
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉用户ID失败 ->", err)
|
||
// c.JsonResult(500, i18n.Tr(c.Lang, "message.failed_auto_login"), nil)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// username, avatar, err := dingtalkAgent.GetUserNameAndAvatarByUserID(userid)
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉用户信息失败 ->", err)
|
||
// c.JsonResult(500, i18n.Tr(c.Lang, "message.failed_auto_login"), nil)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// member, err := models.NewMember().TmpLogin(tmpReader)
|
||
// if err == nil {
|
||
// member.LastLoginTime = time.Now()
|
||
// _ = member.Update("last_login_time")
|
||
// member.Account = username
|
||
// if avatar != "" {
|
||
// member.Avatar = avatar
|
||
// }
|
||
//
|
||
// c.SetMember(*member)
|
||
// }
|
||
// c.JsonResult(0, "ok", username)
|
||
//}
|
||
|
||
// WorkWeixinLogin 用户企业微信登录
|
||
//func (c *AccountController) WorkWeixinLogin() {
|
||
// logs.Info("UserAgent: ", c.Ctx.Input.UserAgent()) // debug
|
||
//
|
||
// if member, ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0 {
|
||
// u := c.GetString("url")
|
||
// if u == "" {
|
||
// u = c.Ctx.Request.Header.Get("Referer")
|
||
// if u == "" {
|
||
// u = conf.URLFor("HomeController.Index")
|
||
// }
|
||
// }
|
||
// // session自动登录时刷新session内容
|
||
// member, err := models.NewMember().Find(member.MemberId)
|
||
// if err != nil {
|
||
// c.DelSession(conf.LoginSessionName)
|
||
// c.SetMember(models.Member{})
|
||
// c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
// } else {
|
||
// c.SetMember(*member)
|
||
// }
|
||
// c.Redirect(u, 302)
|
||
// }
|
||
// var remember CookieRemember
|
||
// // 如果 Cookie 中存在登录信息
|
||
// if cookie, ok := c.GetSecureCookie(conf.GetAppKey(), "login"); ok {
|
||
// if err := utils.Decode(cookie, &remember); err == nil {
|
||
// if member, err := models.NewMember().Find(remember.MemberId); err == nil {
|
||
// c.SetMember(*member)
|
||
// c.LoggedIn(false)
|
||
// c.StopRun()
|
||
// }
|
||
// }
|
||
// }
|
||
//
|
||
// if c.Ctx.Input.IsPost() {
|
||
// // account := c.GetString("account")
|
||
// // password := c.GetString("password")
|
||
// // captcha := c.GetString("code")
|
||
// // isRemember := c.GetString("is_remember")
|
||
// c.JsonResult(400, "request method not allowed", nil)
|
||
// } else {
|
||
// var callback_u string
|
||
// u := c.GetString("url")
|
||
// if u == "" {
|
||
// u = c.referer()
|
||
// }
|
||
// if u != "" {
|
||
// var schemaRule = regexp.MustCompile(`^https?\:\/\/`)
|
||
// if !schemaRule.MatchString(u) {
|
||
// u = strings.TrimRight(conf.BaseUrl, "/") + strings.TrimLeft(u, "/")
|
||
// }
|
||
// }
|
||
// if u == "" {
|
||
// callback_u = conf.URLFor("AccountController.WorkWeixinLoginCallback")
|
||
// } else {
|
||
// callback_u = conf.URLFor("AccountController.WorkWeixinLoginCallback", "url", url.PathEscape(u))
|
||
// }
|
||
// logs.Info("callback_u: ", callback_u) // debug
|
||
//
|
||
// state := "mindoc"
|
||
// workweixinConf := conf.GetWorkWeixinConfig()
|
||
// appid := workweixinConf.CorpId
|
||
// agentid := workweixinConf.AgentId
|
||
// var redirect_uri string
|
||
//
|
||
// isInWorkWeixin := c.IsInWorkWeixin()
|
||
// c.Data["IsInWorkWeixin"] = isInWorkWeixin
|
||
// if isInWorkWeixin {
|
||
// // 企业微信内-网页授权登录
|
||
// urlFmt := "%s?appid=%s&agentid=%s&redirect_uri=%s&response_type=code&scope=snsapi_privateinfo&state=%s#wechat_redirect"
|
||
// redirect_uri = fmt.Sprintf(urlFmt, WorkWeixin_AuthorizeUrlBase, appid, agentid, url.PathEscape(callback_u), state)
|
||
// } else {
|
||
// // 浏览器内-扫码授权登录
|
||
// urlFmt := "%s?login_type=CorpApp&appid=%s&agentid=%s&redirect_uri=%s&state=%s"
|
||
// redirect_uri = fmt.Sprintf(urlFmt, WorkWeixin_QRConnectUrlBase, appid, agentid, url.PathEscape(callback_u), state)
|
||
// }
|
||
// logs.Info("redirect_uri: ", redirect_uri) // debug
|
||
// c.Redirect(redirect_uri, 302)
|
||
// }
|
||
//}
|
||
|
||
/*
|
||
思路:
|
||
1. 浏览器打开
|
||
用户名+密码 登录 与企业微信没有交集
|
||
手机企业微信登录->扫码页面->扫码后获取用户信息, 判断是否绑定了企业微信
|
||
已绑定，则读取用户信息，直接登录
|
||
未绑定，则弹窗提示[未绑定企业微信，请先在企业微信中打开，完成绑定]
|
||
2. 企业微信打开->自动登录->判断是否绑定了企业微信
|
||
已绑定，则读取用户信息，直接登录
|
||
未绑定，则弹窗提示
|
||
是否已有账户(用户名+密码方式)
|
||
有: 弹窗输入[用户名+密码+验证码]校验
|
||
无: 直接以企业UserId作为用户名(小写)，创建随机密码
|
||
*/
|
||
|
||
// WorkWeixinLoginCallback 用户企业微信登录-回调
|
||
//func (c *AccountController) WorkWeixinLoginCallback() {
|
||
// c.TplName = "account/auth2_callback.tpl"
|
||
//
|
||
// if member, ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0 {
|
||
// u := c.GetString("url")
|
||
// if u == "" {
|
||
// u = c.Ctx.Request.Header.Get("Referer")
|
||
// }
|
||
// if u == "" {
|
||
// u = conf.URLFor("HomeController.Index")
|
||
// }
|
||
// member, err := models.NewMember().Find(member.MemberId)
|
||
// if err != nil {
|
||
// c.DelSession(conf.LoginSessionName)
|
||
// c.SetMember(models.Member{})
|
||
// c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
// } else {
|
||
// c.SetMember(*member)
|
||
// }
|
||
// c.Redirect(u, 302)
|
||
// }
|
||
//
|
||
// var remember CookieRemember
|
||
// // 如果 Cookie 中存在登录信息
|
||
// if cookie, ok := c.GetSecureCookie(conf.GetAppKey(), "login"); ok {
|
||
// if err := utils.Decode(cookie, &remember); err == nil {
|
||
// if member, err := models.NewMember().Find(remember.MemberId); err == nil {
|
||
// c.SetMember(*member)
|
||
// c.LoggedIn(false)
|
||
// c.StopRun()
|
||
// }
|
||
// }
|
||
// }
|
||
//
|
||
// // 请求参数获取
|
||
// req_code := c.GetString("code")
|
||
// logs.Warning("req_code: ", req_code)
|
||
// req_state := c.GetString("state")
|
||
// logs.Warning("req_state: ", req_state)
|
||
// var user_info_json string
|
||
// var error_msg string
|
||
// var bind_existed string
|
||
// if len(req_code) > 0 && req_state == "mindoc" {
|
||
// // 获取当前应用的access_token
|
||
// access_token, ok := workweixin.GetAccessToken()
|
||
// if ok {
|
||
// logs.Warning("access_token: ", access_token)
|
||
// // 获取当前请求的userid
|
||
// user_id, ticket, ok := workweixin.RequestUserId(access_token, req_code)
|
||
// if ok {
|
||
// logs.Warning("user_id: ", user_id)
|
||
// // 查询系统现有数据，是否绑定了当前请求用户的企业微信
|
||
// member, err := models.NewWorkWeixinAccount().ExistedMember(user_id)
|
||
// if err == nil {
|
||
// member.LastLoginTime = time.Now()
|
||
// _ = member.Update("last_login_time")
|
||
//
|
||
// c.SetMember(*member)
|
||
//
|
||
// var remember CookieRemember
|
||
// remember.MemberId = member.MemberId
|
||
// remember.Account = member.Account
|
||
// remember.Time = time.Now()
|
||
// v, err := utils.Encode(remember)
|
||
// if err == nil {
|
||
// c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
// }
|
||
// bind_existed = "true"
|
||
// error_msg = ""
|
||
// u := c.GetString("url")
|
||
// if u == "" {
|
||
// u = conf.URLFor("HomeController.Index")
|
||
// }
|
||
// c.Redirect(u, 302)
|
||
// } else if err == orm.ErrNoRows {
|
||
// bind_existed = "false"
|
||
// if ticket == "" {
|
||
// error_msg = "请到企业微信中登录，并授权获取敏感信息。"
|
||
// } else {
|
||
// user_info, err := workweixin.RequestUserPrivateInfo(access_token, user_id, ticket)
|
||
// if err != nil {
|
||
// error_msg = "获取敏感信息错误: " + err.Error()
|
||
// } else {
|
||
// json_info, _ := json.Marshal(user_info)
|
||
// user_info_json = string(json_info)
|
||
// error_msg = ""
|
||
// c.SetSession(SessionUserInfoKey, user_info)
|
||
// }
|
||
// }
|
||
// } else {
|
||
// logs.Error("Error: ", err)
|
||
// error_msg = "登录错误: " + err.Error()
|
||
// }
|
||
// } else {
|
||
// error_msg = "获取用户Id失败: " + user_id
|
||
// }
|
||
// } else {
|
||
// error_msg = "应用凭据获取失败: " + access_token
|
||
// }
|
||
// } else {
|
||
// error_msg = "参数错误"
|
||
// }
|
||
// if user_info_json == "" {
|
||
// user_info_json = "{}"
|
||
// }
|
||
// if bind_existed == "" {
|
||
// bind_existed = "null"
|
||
// }
|
||
// // refer & doc:
|
||
// // - https://golang.org/pkg/html/template/#HTML
|
||
// // - https://stackoverflow.com/questions/24411880/go-html-templates-can-i-stop-the-templates-package-inserting-quotes-around-stri
|
||
// // - https://stackoverflow.com/questions/38035176/insert-javascript-snippet-inside-template-with-beego-golang
|
||
// c.Data["bind_existed"] = template.JS(bind_existed)
|
||
// logs.Debug("bind_existed: ", bind_existed)
|
||
// c.Data["error_msg"] = template.JS(error_msg)
|
||
// c.Data["user_info_json"] = template.JS(user_info_json)
|
||
// /*
|
||
// // 调试: 显示源码
|
||
// result, err := c.RenderString()
|
||
// if err != nil {
|
||
// logs.Error(err)
|
||
// } else {
|
||
// logs.Warning(result)
|
||
// }
|
||
// */
|
||
//}
|
||
|
||
// WorkWeixinLoginBind 用户企业微信登录-绑定
|
||
//func (c *AccountController) WorkWeixinLoginBind() {
|
||
// if user_info, ok := c.GetSession(SessionUserInfoKey).(workweixin.WorkWeixinUserPrivateInfo); ok && len(user_info.UserId) > 0 {
|
||
// req_account := c.GetString("account")
|
||
// req_password := c.GetString("password")
|
||
// if req_account == "" || req_password == "" {
|
||
// c.JsonResult(400, "账号或密码不能为空")
|
||
// } else {
|
||
// member, err := models.NewMember().Login(req_account, req_password)
|
||
// if err == nil {
|
||
// account := models.NewWorkWeixinAccount()
|
||
// account.MemberId = member.MemberId
|
||
// account.WorkWeixin_UserId = user_info.UserId
|
||
// member.CreateAt = 0
|
||
// ormer := orm.NewOrm()
|
||
// o, err := ormer.Begin()
|
||
// if err != nil {
|
||
// logs.Error("开启事务时出错 -> ", err)
|
||
// c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
// }
|
||
// if err := account.AddBind(ormer); err != nil {
|
||
// o.Rollback()
|
||
// c.JsonResult(500, "绑定失败，数据库错误: "+err.Error())
|
||
// } else {
|
||
// // 绑定成功之后修改用户信息
|
||
// member.LastLoginTime = time.Now()
|
||
// //member.RealName = user_info.Name
|
||
// //member.Avatar = user_info.Avatar
|
||
// if len(member.Avatar) < 1 {
|
||
// member.Avatar = conf.GetDefaultAvatar()
|
||
// }
|
||
// //member.Email = user_info.Email
|
||
// //member.Phone = user_info.Mobile
|
||
// if _, err := ormer.Update(member, "last_login_time", "real_name", "avatar", "email", "phone"); err != nil {
|
||
// o.Rollback()
|
||
// logs.Error("保存用户信息失败=>", err)
|
||
// c.JsonResult(500, "绑定失败，现有账户信息更新失败: "+err.Error())
|
||
// } else {
|
||
// if err := o.Commit(); err != nil {
|
||
// logs.Error("开启事务时出错 -> ", err)
|
||
// c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
// } else {
|
||
// c.DelSession(SessionUserInfoKey)
|
||
// c.SetMember(*member)
|
||
//
|
||
// var remember CookieRemember
|
||
// remember.MemberId = member.MemberId
|
||
// remember.Account = member.Account
|
||
// remember.Time = time.Now()
|
||
// v, err := utils.Encode(remember)
|
||
// if err == nil {
|
||
// c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
// c.JsonResult(0, "绑定成功", nil)
|
||
// } else {
|
||
// c.JsonResult(500, "绑定成功, 但自动登录失败, 请返回首页重新登录", nil)
|
||
// }
|
||
// }
|
||
// }
|
||
//
|
||
// }
|
||
//
|
||
// } else {
|
||
// logs.Error("用户登录 ->", err)
|
||
// c.JsonResult(500, "账号或密码错误", nil)
|
||
// }
|
||
// c.JsonResult(500, "TODO: 绑定以后账号功能开发中")
|
||
// }
|
||
// } else {
|
||
// if ok {
|
||
// c.DelSession(SessionUserInfoKey)
|
||
// }
|
||
// c.JsonResult(400, "请求错误, 请从首页重新登录")
|
||
// }
|
||
//
|
||
//}
|
||
|
||
// WorkWeixinLoginIgnore 用户企业微信登录-忽略
|
||
//func (c *AccountController) WorkWeixinLoginIgnore() {
|
||
// if user_info, ok := c.GetSession(SessionUserInfoKey).(workweixin.WorkWeixinUserPrivateInfo); ok && len(user_info.UserId) > 0 {
|
||
// c.DelSession(SessionUserInfoKey)
|
||
// member := models.NewMember()
|
||
//
|
||
// if _, err := member.FindByAccount(user_info.UserId); err == nil && member.MemberId > 0 {
|
||
// c.JsonResult(400, "账号已存在")
|
||
// }
|
||
//
|
||
// ormer := orm.NewOrm()
|
||
// o, err := ormer.Begin()
|
||
// if err != nil {
|
||
// logs.Error("开启事务时出错 -> ", err)
|
||
// c.JsonResult(500, "开启事务时出错: ", err.Error())
|
||
// }
|
||
//
|
||
// member.Account = user_info.UserId
|
||
// member.RealName = user_info.Name
|
||
// var rnd = rand.New(src)
|
||
// // fmt.Sprintf("%x", rnd.Uint64())
|
||
// // strconv.FormatUint(rnd.Uint64(), 16)
|
||
// member.Password = user_info.UserId + strconv.FormatUint(rnd.Uint64(), 16)
|
||
// member.Password = "123456" // 强制设置默认密码，需修改一次密码后，才可以进行账号密码登录
|
||
// hash, err := utils.PasswordHash(member.Password)
|
||
// if err != nil {
|
||
// logs.Error("加密用户密码失败 =>", err)
|
||
// c.JsonResult(500, "加密用户密码失败"+err.Error())
|
||
// } else {
|
||
// logs.Error("member.Password: ", member.Password)
|
||
// logs.Error("hash: ", hash)
|
||
// member.Password = hash
|
||
// }
|
||
// member.Role = conf.MemberGeneralRole
|
||
// member.Avatar = user_info.Avatar
|
||
// if len(member.Avatar) < 1 {
|
||
// member.Avatar = conf.GetDefaultAvatar()
|
||
// }
|
||
// member.CreateAt = 0
|
||
// member.Email = user_info.BizMail
|
||
// member.Phone = user_info.Mobile
|
||
// member.Status = 0
|
||
// if _, err = ormer.Insert(member); err != nil {
|
||
// o.Rollback()
|
||
// c.JsonResult(500, "注册失败，数据库错误: "+err.Error())
|
||
// } else {
|
||
// account := models.NewWorkWeixinAccount()
|
||
// account.MemberId = member.MemberId
|
||
// account.WorkWeixin_UserId = user_info.UserId
|
||
// member.CreateAt = 0
|
||
// if err := account.AddBind(ormer); err != nil {
|
||
// o.Rollback()
|
||
// c.JsonResult(500, "注册失败，数据库错误: "+err.Error())
|
||
// } else {
|
||
// if err := o.Commit(); err != nil {
|
||
// logs.Error("提交事务时出错 -> ", err)
|
||
// c.JsonResult(500, "提交事务时出错: ", err.Error())
|
||
// } else {
|
||
// member.LastLoginTime = time.Now()
|
||
// _ = member.Update("last_login_time")
|
||
//
|
||
// c.SetMember(*member)
|
||
//
|
||
// var remember CookieRemember
|
||
// remember.MemberId = member.MemberId
|
||
// remember.Account = member.Account
|
||
// remember.Time = time.Now()
|
||
// v, err := utils.Encode(remember)
|
||
// if err == nil {
|
||
// c.SetSecureCookie(conf.GetAppKey(), "login", v, time.Now().Add(time.Hour*24*30*5).Unix())
|
||
// c.JsonResult(0, "绑定成功", nil)
|
||
// } else {
|
||
// c.JsonResult(500, "绑定成功, 但自动登录失败, 请返回首页重新登录", nil)
|
||
// }
|
||
// }
|
||
// }
|
||
// }
|
||
// } else {
|
||
// if ok {
|
||
// c.DelSession(SessionUserInfoKey)
|
||
// }
|
||
// c.JsonResult(400, "请求错误, 请从首页重新登录")
|
||
// }
|
||
//}
|
||
|
||
// QR二维码登录
|
||
//func (c *AccountController) QRLogin() {
|
||
// appName := c.Ctx.Input.Param(":app")
|
||
//
|
||
// switch appName {
|
||
// // 钉钉扫码登录
|
||
// case "dingtalk":
|
||
// code := c.GetString("code")
|
||
// state := c.GetString("state")
|
||
// if state != "1" || code == "" {
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
// appKey, _ := web.AppConfig.String("dingtalk_qr_key")
|
||
// appSecret, _ := web.AppConfig.String("dingtalk_qr_secret")
|
||
//
|
||
// qrDingtalk := dingtalk.NewDingtalkQRLogin(appSecret, appKey)
|
||
// unionID, err := qrDingtalk.GetUnionIDByCode(code)
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉临时UnionID失败 ->", err)
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// appKey, _ = web.AppConfig.String("dingtalk_app_key")
|
||
// appSecret, _ = web.AppConfig.String("dingtalk_app_secret")
|
||
// tmpReader, _ := web.AppConfig.String("dingtalk_tmp_reader")
|
||
//
|
||
// dingtalkAgent := dingtalk.NewDingTalkAgent(appSecret, appKey)
|
||
// err = dingtalkAgent.GetAccesstoken()
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉临时Token失败 ->", err)
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// userid, err := dingtalkAgent.GetUserIDByUnionID(unionID)
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉用户ID失败 ->", err)
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// username, avatar, err := dingtalkAgent.GetUserNameAndAvatarByUserID(userid)
|
||
// if err != nil {
|
||
// logs.Warn("获取钉钉用户信息失败 ->", err)
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
//
|
||
// member, err := models.NewMember().TmpLogin(tmpReader)
|
||
// if err == nil {
|
||
// member.LastLoginTime = time.Now()
|
||
// _ = member.Update("last_login_time")
|
||
// member.Account = username
|
||
// if avatar != "" {
|
||
// member.Avatar = avatar
|
||
// }
|
||
//
|
||
// c.SetMember(*member)
|
||
// c.LoggedIn(false)
|
||
// c.StopRun()
|
||
// }
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
//
|
||
// // 企业微信扫码登录
|
||
// case "workweixin":
|
||
// //
|
||
//
|
||
// default:
|
||
// c.Redirect(conf.URLFor("AccountController.Login"), 302)
|
||
// c.StopRun()
|
||
// }
|
||
//}
|
||
|
||
// 登录成功后的操作，如重定向到原始请求页面
|
||
func (c *AccountController) LoggedIn(isPost bool) interface{} {
|
||
|
||
turl := c.referer()
|
||
|
||
if !isPost {
|
||
c.Redirect(turl, 302)
|
||
return nil
|
||
} else {
|
||
var data struct {
|
||
TURL string `json:"url"`
|
||
}
|
||
data.TURL = turl
|
||
return data
|
||
}
|
||
}
|
||
|
||
// 用户注册
|
||
func (c *AccountController) Register() {
|
||
c.TplName = "account/register.tpl"
|
||
|
||
//如果用户登录了，则跳转到网站首页
|
||
if member, ok := c.GetSession(conf.LoginSessionName).(models.Member); ok && member.MemberId > 0 {
|
||
c.Redirect(conf.URLFor("HomeController.Index"), 302)
|
||
}
|
||
// 如果没有开启用户注册
|
||
if v, ok := c.Option["ENABLED_REGISTER"]; ok && !strings.EqualFold(v, "true") {
|
||
c.Abort("404")
|
||
}
|
||
|
||
if c.Ctx.Input.IsPost() {
|
||
account := c.GetString("account")
|
||
password1 := c.GetString("password1")
|
||
password2 := c.GetString("password2")
|
||
email := c.GetString("email")
|
||
captcha := c.GetString("code")
|
||
|
||
if ok, err := regexp.MatchString(conf.RegexpAccount, account); account == "" || !ok || err != nil {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.username_invalid_format"))
|
||
}
|
||
if l := strings.Count(password1, ""); password1 == "" || l > 50 || l < 6 {
|
||
c.JsonResult(6002, i18n.Tr(c.Lang, "message.password_length_invalid"))
|
||
}
|
||
if password1 != password2 {
|
||
c.JsonResult(6003, i18n.Tr(c.Lang, "message.incorrect_confirm_password"))
|
||
}
|
||
if ok, err := regexp.MatchString(conf.RegexpEmail, email); !ok || err != nil || email == "" {
|
||
c.JsonResult(6004, i18n.Tr(c.Lang, "message.email_invalid_format"))
|
||
}
|
||
// 如果开启了验证码
|
||
if v, ok := c.Option["ENABLED_CAPTCHA"]; ok && strings.EqualFold(v, "true") {
|
||
v, ok := c.GetSession(conf.CaptchaSessionName).(string)
|
||
if !ok || !strings.EqualFold(v, captcha) {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.captcha_wrong"))
|
||
}
|
||
}
|
||
|
||
member := models.NewMember()
|
||
|
||
if _, err := member.FindByAccount(account); err == nil && member.MemberId > 0 {
|
||
c.JsonResult(6005, i18n.Tr(c.Lang, "message.account_existed"))
|
||
}
|
||
|
||
member.Account = account
|
||
member.Password = password1
|
||
member.Role = conf.MemberGeneralRole
|
||
member.Avatar = conf.GetDefaultAvatar()
|
||
member.CreateAt = 0
|
||
member.Email = email
|
||
member.Status = 0
|
||
if err := member.Add(); err != nil {
|
||
c.JsonResult(6006, i18n.Tr(c.Lang, "message.failed_register"))
|
||
}
|
||
|
||
c.JsonResult(0, "ok", member)
|
||
}
|
||
}
|
||
|
||
// 找回密码
|
||
func (c *AccountController) FindPassword() {
|
||
c.TplName = "account/find_password_setp1.tpl"
|
||
mailConf := conf.GetMailConfig()
|
||
|
||
if c.Ctx.Input.IsPost() {
|
||
|
||
email := c.GetString("email")
|
||
captcha := c.GetString("code")
|
||
|
||
if email == "" {
|
||
c.JsonResult(6005, i18n.Tr(c.Lang, "message.email_empty"))
|
||
}
|
||
if !mailConf.EnableMail {
|
||
c.JsonResult(6004, i18n.Tr(c.Lang, "message.mail_service_not_enable"))
|
||
}
|
||
|
||
// 如果开启了验证码
|
||
if v, ok := c.Option["ENABLED_CAPTCHA"]; ok && strings.EqualFold(v, "true") {
|
||
v, ok := c.GetSession(conf.CaptchaSessionName).(string)
|
||
if !ok || !strings.EqualFold(v, captcha) {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.captcha_wrong"))
|
||
}
|
||
}
|
||
|
||
member, err := models.NewMember().FindByFieldFirst("email", email)
|
||
if err != nil {
|
||
c.JsonResult(6006, i18n.Tr(c.Lang, "message.email_not_exist"))
|
||
}
|
||
if member == nil || member.Status != 0 {
|
||
c.JsonResult(6007, i18n.Tr(c.Lang, "message.account_disable"))
|
||
}
|
||
if member == nil || member.AuthMethod == conf.AuthMethodLDAP {
|
||
c.JsonResult(6011, i18n.Tr(c.Lang, "message.account_not_support_retrieval"))
|
||
}
|
||
|
||
count, err := models.NewMemberToken().FindSendCount(email, time.Now().Add(-1*time.Hour), time.Now())
|
||
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6008, i18n.Tr(c.Lang, "message.failed_send_mail"))
|
||
}
|
||
if count > mailConf.MailNumber {
|
||
c.JsonResult(6008, i18n.Tr(c.Lang, "message.sent_too_many_times"))
|
||
}
|
||
|
||
memberToken := models.NewMemberToken()
|
||
|
||
memberToken.Token = string(utils.Krand(32, utils.KC_RAND_KIND_ALL))
|
||
memberToken.Email = email
|
||
memberToken.MemberId = member.MemberId
|
||
memberToken.IsValid = false
|
||
if _, err := memberToken.InsertOrUpdate(); err != nil {
|
||
c.JsonResult(6009, i18n.Tr(c.Lang, "message.failed_send_mail"))
|
||
}
|
||
|
||
data := map[string]interface{}{
|
||
"SITE_NAME": c.Option["SITE_NAME"],
|
||
"url": conf.URLFor("AccountController.FindPassword", "token", memberToken.Token, "mail", email),
|
||
"BaseUrl": c.BaseUrl(),
|
||
}
|
||
|
||
body, err := c.ExecuteViewPathTemplate("account/mail_template.tpl", data)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6003, i18n.Tr(c.Lang, "message.failed_send_mail"))
|
||
}
|
||
|
||
go func(mailConf *conf.SmtpConf, email string, body string) {
|
||
|
||
mailConfig := &mail.SMTPConfig{
|
||
Username: mailConf.SmtpUserName,
|
||
Password: mailConf.SmtpPassword,
|
||
Host: mailConf.SmtpHost,
|
||
Port: mailConf.SmtpPort,
|
||
Secure: mailConf.Secure,
|
||
Identity: "",
|
||
}
|
||
logs.Info(mailConfig)
|
||
|
||
c := mail.NewSMTPClient(mailConfig)
|
||
m := mail.NewMail()
|
||
|
||
m.AddFrom(mailConf.FormUserName)
|
||
m.AddFromName(mailConf.FormUserName)
|
||
m.AddSubject("找回密码")
|
||
m.AddHTML(body)
|
||
m.AddTo(email)
|
||
|
||
if e := c.Send(m); e != nil {
|
||
logs.Error("发送邮件失败：" + e.Error())
|
||
} else {
|
||
logs.Info("邮件发送成功：" + email)
|
||
}
|
||
//auth := smtp.PlainAuth(
|
||
// "",
|
||
// mail_conf.SmtpUserName,
|
||
// mail_conf.SmtpPassword,
|
||
// mail_conf.SmtpHost,
|
||
//)
|
||
//
|
||
//mime := "MIME-version: 1.0;\nContent-Type: text/html; charset=\"UTF-8\";\n\n"
|
||
//subject := "Subject: 找回密码!\n"
|
||
//
|
||
//err = smtp.SendMail(
|
||
// mail_conf.SmtpHost+":"+strconv.Itoa(mail_conf.SmtpPort),
|
||
// auth,
|
||
// mail_conf.FormUserName,
|
||
// []string{email},
|
||
// []byte(subject+mime+"\n"+body),
|
||
//)
|
||
//if err != nil {
|
||
// logs.Error("邮件发送失败 => ", email, err)
|
||
//}
|
||
}(mailConf, email, body)
|
||
|
||
c.JsonResult(0, "ok", conf.URLFor("AccountController.Login"))
|
||
}
|
||
|
||
token := c.GetString("token")
|
||
email := c.GetString("mail")
|
||
|
||
if token != "" && email != "" {
|
||
memberToken, err := models.NewMemberToken().FindByFieldFirst("token", token)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.Data["ErrorMessage"] = i18n.Tr(c.Lang, "message.mail_expired")
|
||
c.TplName = "errors/error.tpl"
|
||
return
|
||
}
|
||
subTime := time.Until(memberToken.SendTime)
|
||
|
||
if !strings.EqualFold(memberToken.Email, email) || subTime.Minutes() > float64(mailConf.MailExpired) || !memberToken.ValidTime.IsZero() {
|
||
c.Data["ErrorMessage"] = i18n.Tr(c.Lang, "message.captcha_expired")
|
||
c.TplName = "errors/error.tpl"
|
||
return
|
||
}
|
||
c.Data["Email"] = memberToken.Email
|
||
c.Data["Token"] = memberToken.Token
|
||
c.TplName = "account/find_password_setp2.tpl"
|
||
|
||
}
|
||
}
|
||
|
||
// 校验邮件并修改密码
|
||
func (c *AccountController) ValidEmail() {
|
||
password1 := c.GetString("password1")
|
||
password2 := c.GetString("password2")
|
||
captcha := c.GetString("code")
|
||
token := c.GetString("token")
|
||
email := c.GetString("mail")
|
||
|
||
if password1 == "" {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.password_empty"))
|
||
}
|
||
if l := strings.Count(password1, ""); l < 6 || l > 50 {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.password_length_invalid"))
|
||
}
|
||
if password2 == "" {
|
||
c.JsonResult(6002, i18n.Tr(c.Lang, "message.confirm_password_empty"))
|
||
}
|
||
if password1 != password2 {
|
||
c.JsonResult(6003, i18n.Tr(c.Lang, "message.incorrect_confirm_password"))
|
||
}
|
||
if captcha == "" {
|
||
c.JsonResult(6004, i18n.Tr(c.Lang, "message.captcha_empty"))
|
||
}
|
||
v, ok := c.GetSession(conf.CaptchaSessionName).(string)
|
||
if !ok || !strings.EqualFold(v, captcha) {
|
||
c.JsonResult(6001, i18n.Tr(c.Lang, "message.captcha_wrong"))
|
||
}
|
||
|
||
mailConf := conf.GetMailConfig()
|
||
memberToken, err := models.NewMemberToken().FindByFieldFirst("token", token)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6007, i18n.Tr(c.Lang, "message.mail_expired"))
|
||
}
|
||
subTime := time.Until(memberToken.SendTime)
|
||
|
||
if !strings.EqualFold(memberToken.Email, email) || subTime.Minutes() > float64(mailConf.MailExpired) || !memberToken.ValidTime.IsZero() {
|
||
|
||
c.JsonResult(6008, i18n.Tr(c.Lang, "message.captcha_expired"))
|
||
}
|
||
member, err := models.NewMember().Find(memberToken.MemberId)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6005, i18n.Tr(c.Lang, "message.user_not_existed"))
|
||
}
|
||
hash, err := utils.PasswordHash(password1)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6006, i18n.Tr(c.Lang, "message.failed_save_password"))
|
||
}
|
||
|
||
member.Password = hash
|
||
|
||
err = member.Update("password")
|
||
memberToken.ValidTime = time.Now()
|
||
memberToken.IsValid = true
|
||
memberToken.InsertOrUpdate()
|
||
|
||
if err != nil {
|
||
logs.Error(err)
|
||
c.JsonResult(6006, i18n.Tr(c.Lang, "message.failed_save_password"))
|
||
}
|
||
c.JsonResult(0, "ok", conf.URLFor("AccountController.Login"))
|
||
}
|
||
|
||
// Logout 退出登录
|
||
func (c *AccountController) Logout() {
|
||
c.SetMember(models.Member{})
|
||
c.SetSecureCookie(conf.GetAppKey(), "login", "", -3600)
|
||
u := c.Ctx.Request.Header.Get("Referer")
|
||
c.Redirect(conf.URLFor("AccountController.Login", "url", u), 302)
|
||
}
|
||
|
||
// 验证码
|
||
func (c *AccountController) Captcha() {
|
||
captchaImage := gocaptcha.NewCaptchaImage(140, 40, gocaptcha.RandLightColor())
|
||
|
||
captchaImage.DrawNoise(gocaptcha.CaptchaComplexLower)
|
||
|
||
// captchaImage.DrawTextNoise(gocaptcha.CaptchaComplexHigh)
|
||
txt := gocaptcha.RandText(4)
|
||
|
||
c.SetSession(conf.CaptchaSessionName, txt)
|
||
|
||
captchaImage.DrawText(txt)
|
||
// captchaImage.Drawline(3);
|
||
captchaImage.DrawBorder(gocaptcha.ColorToRGB(0x17A7A7A))
|
||
// captchaImage.DrawHollowLine()
|
||
|
||
captchaImage.SaveImage(c.Ctx.ResponseWriter, gocaptcha.ImageFormatJpeg)
|
||
c.StopRun()
|
||
}
|