forked from golang/hotime
278 lines
7.5 KiB
Go
278 lines
7.5 KiB
Go
package app
|
||
|
||
import (
|
||
. "code.hoteas.com/golang/hotime"
|
||
. "code.hoteas.com/golang/hotime/common"
|
||
"code.hoteas.com/golang/hotime/dri/download"
|
||
"fmt"
|
||
"github.com/silenceper/wechat/v2/cache"
|
||
"github.com/silenceper/wechat/v2/miniprogram/config"
|
||
"time"
|
||
|
||
//"fmt"
|
||
"github.com/silenceper/wechat/v2"
|
||
//"github.com/silenceper/wechat/v2/cache"
|
||
)
|
||
|
||
var Wechat = Ctr{
|
||
"login": func(that *Context) {
|
||
nickname := that.Req.FormValue("nickname")
|
||
avatar := that.Req.FormValue("avatarUrl")
|
||
encryptedData := that.Req.FormValue("encryptedData")
|
||
iv := that.Req.FormValue("iv")
|
||
|
||
if that.Session("wechat_id").Data == nil {
|
||
that.Display(2, "还没有获取token")
|
||
return
|
||
}
|
||
|
||
if nickname == "" || avatar == "" || encryptedData == "" || iv == "" {
|
||
that.Display(3, "参数不足")
|
||
return
|
||
}
|
||
|
||
if weixin == nil {
|
||
weixin = wechat.NewWechat()
|
||
memory := cache.NewMemory()
|
||
cfg := &config.Config{
|
||
AppID: that.Config.GetString("wechatAppID"),
|
||
AppSecret: that.Config.GetString("wechatAppSecret"),
|
||
//Token: "xxx",
|
||
//EncodingAESKey: "xxxx",
|
||
Cache: memory,
|
||
}
|
||
|
||
miniProgram = weixin.GetMiniProgram(cfg)
|
||
}
|
||
|
||
wechatId := that.Session("wechat_id").ToCeilInt()
|
||
|
||
wchat := that.Db.Get("wechat", "*", Map{"id": wechatId})
|
||
|
||
if wchat == nil {
|
||
that.Display(4, "找不到该用户")
|
||
return
|
||
}
|
||
|
||
eny := miniProgram.GetEncryptor()
|
||
re, e := eny.Decrypt(wchat.GetString("retoken"), encryptedData, iv)
|
||
|
||
if e != nil {
|
||
that.Display(4, e)
|
||
return
|
||
}
|
||
|
||
admin := that.Db.Get("admin", "*", Map{"phone": re.PhoneNumber})
|
||
if admin == nil {
|
||
that.Display(5, "你不是本系统用户，无法使用本系统")
|
||
return
|
||
}
|
||
|
||
t := time.Now().Unix()
|
||
path := time.Now().Format(that.Config.GetString("imgPath"))
|
||
filename := Md5(ObjToStr(t)) + ".jpg"
|
||
down := download.Down(avatar, that.Config.GetString("tpt")+"/"+path, filename)
|
||
if down {
|
||
avatar = path + filename
|
||
}
|
||
|
||
that.Db.Update("wechat", Map{"admin_id": admin.GetCeilInt("id"), "modify_time": t, "avatar": avatar, "nickname": nickname}, Map{"id": wchat.GetCeilInt64("id")})
|
||
that.Db.Update("admin", Map{"avatar_img": avatar, "modify_time": t}, Map{"id": admin.GetCeilInt64("id")})
|
||
|
||
that.Session("admin_id", admin.GetCeilInt64("id"))
|
||
that.Display(0, "成功！")
|
||
},
|
||
"code": func(that *Context) {
|
||
|
||
that.Session("admin_id", 1)
|
||
that.Display(0, Map{"status": 1, "token": that.SessionId})
|
||
return
|
||
|
||
code := that.Req.FormValue("code")
|
||
if code == "" {
|
||
that.Display(3, "缺少code")
|
||
return
|
||
}
|
||
if weixin == nil {
|
||
weixin = wechat.NewWechat()
|
||
memory := cache.NewMemory()
|
||
cfg := &config.Config{
|
||
AppID: that.Config.GetString("wechatAppID"),
|
||
AppSecret: that.Config.GetString("wechatAppSecret"),
|
||
//Token: "xxx",
|
||
//EncodingAESKey: "xxxx",
|
||
Cache: memory,
|
||
}
|
||
|
||
miniProgram = weixin.GetMiniProgram(cfg)
|
||
}
|
||
//mini := weixin.GetMiniProgram(cfg)
|
||
a := miniProgram.GetAuth()
|
||
re, e := a.Code2Session(code)
|
||
fmt.Println(re)
|
||
if e != nil {
|
||
that.Display(4, e)
|
||
return
|
||
}
|
||
wchat := that.Db.Get("wechat", "*", Map{"openid": re.OpenID})
|
||
|
||
if wchat == nil {
|
||
wchat = Map{"openid": re.OpenID,
|
||
"appid": a.AppID,
|
||
"state": 0,
|
||
"retoken": re.SessionKey,
|
||
"modify_time": time.Now().Unix(),
|
||
"create_time": time.Now().Unix(),
|
||
"unionid": re.UnionID,
|
||
}
|
||
wchat["id"] = that.Db.Insert("wechat", wchat)
|
||
|
||
} else {
|
||
wchat["modify_time"] = time.Now().Unix()
|
||
wchat["retoken"] = re.SessionKey
|
||
wchat["unionid"] = re.UnionID
|
||
that.Db.Update("wechat", wchat, Map{"id": wchat.GetCeilInt("id")})
|
||
}
|
||
|
||
that.Session("wechat_id", wchat.GetCeilInt("id"))
|
||
|
||
if wchat.GetCeilInt("admin_id") == 0 {
|
||
that.Display(0, Map{"status": 0, "token": that.SessionId})
|
||
return
|
||
}
|
||
|
||
that.Session("admin_id", wchat.GetCeilInt("admin_id"))
|
||
that.Display(0, Map{"status": 1, "token": that.SessionId})
|
||
return
|
||
|
||
},
|
||
////微信注册，0已经完整的注册了，1还没有注册
|
||
//"codebase": func(that *Context) {
|
||
// wx := Weixin(that)
|
||
// auth := wx.GetOauth()
|
||
// resToken, err := auth.GetUserAccessToken(that.Req.FormValue("code"))
|
||
// if err != nil {
|
||
// that.Display(6, "code错误")
|
||
// return
|
||
// }
|
||
//
|
||
// //判断用户是否已经注册
|
||
// user := that.Db.Get("wechat", Map{"[><]user": "wechat.user_id=user.id"}, "user.id,user.state", Map{"openid": resToken.OpenID})
|
||
// if user != nil {
|
||
// that.Session("user_id", user.Get("id"))
|
||
// that.Display(0, 0)
|
||
// return
|
||
// }
|
||
//
|
||
// user_id := that.Db.Insert("user", Map{"time": time.Now().Unix(), "state": 2})
|
||
// if user_id == 0 {
|
||
// that.Display(4, "创建用户失败")
|
||
// return
|
||
// }
|
||
//
|
||
// wid := that.Db.Insert("wechat", Map{"openid": resToken.OpenID, "appid": that.Config.GetString("wechatAppID"), "state": 1, "user_id": user_id})
|
||
// if wid == 0 {
|
||
// that.Display(4, "关联微信失败！")
|
||
// return
|
||
// }
|
||
//
|
||
// that.Session("user_id", user.Get("id"))
|
||
//
|
||
// that.Display(0, 1)
|
||
//},
|
||
////微信注册，0已经完整的注册了，1还没有注册
|
||
//"code": func(that *Context) {
|
||
//
|
||
// orgId := ObjToInt(that.Req.FormValue("org_id"))
|
||
// //if orgId==0{
|
||
// // that.Display(3, "缺少组织id")
|
||
// // return
|
||
// //}
|
||
//
|
||
// wx := Weixin(that)
|
||
// auth := wx.GetOauth()
|
||
// resToken, err := auth.GetUserAccessToken(that.Req.FormValue("code"))
|
||
// if err != nil {
|
||
// that.Display(5, "code错误")
|
||
// return
|
||
// }
|
||
//
|
||
// //判断用户是否已经注册
|
||
// user := that.Db.Get("wechat", Map{"[><]user": "wechat.user_id=user.id"}, "user.id,user.state", Map{"openid": resToken.OpenID})
|
||
// if user != nil && user.GetCeilInt("id") != 0 && user.GetCeilInt("state") == 0 {
|
||
// that.Session("user_id", user.Get("id"))
|
||
// that.Display(0, Map{"type": 0, "token": that.SessionId})
|
||
// return
|
||
// }
|
||
// //getUserInfo
|
||
// userInfo, err := auth.GetUserInfo(resToken.AccessToken, resToken.OpenID)
|
||
// if err != nil {
|
||
// that.Display(6, "微信个人信息无法获取")
|
||
// return
|
||
// }
|
||
//
|
||
// //wechatInfo := ObjToMap(userInfo)
|
||
// t := time.Now().Unix()
|
||
// wechatInfo := Map{
|
||
// "openid": userInfo.OpenID,
|
||
// "acttoken": resToken.AccessToken,
|
||
// "retoken": resToken.RefreshToken,
|
||
// "appid": that.Config.GetString("wechatAppID"),
|
||
// "unionid": userInfo.Unionid,
|
||
// "nickname": userInfo.Nickname,
|
||
// "avatar": userInfo.HeadImgURL,
|
||
// }
|
||
// if orgId != 0 {
|
||
// wechatInfo["org_id"] = orgId
|
||
// wechatInfo["status"] = 0
|
||
// }
|
||
//
|
||
// wechatDb := that.Db.Get("wechat", "*", Map{"openid": wechatInfo.GetString("openid")})
|
||
//
|
||
// if wechatDb != nil {
|
||
//
|
||
// that.Db.Update("wechat", wechatInfo, Map{"id": wechatDb.GetCeilInt("id")})
|
||
// //userInfo["wid"]=wechatDb.GetCeilInt("wid")
|
||
// } else {
|
||
// wechatInfo["create_time"] = t
|
||
// wechatInfo["id"] = that.Db.Insert("wechat", wechatInfo)
|
||
// }
|
||
// wechatDb = that.Db.Get("wechat", "*", Map{"openid": wechatInfo.GetString("openid")})
|
||
//
|
||
// that.Session("wechatInfo", wechatDb)
|
||
// fmt.Println(wechatDb)
|
||
// fmt.Println(that.Session("wechatInfo"))
|
||
// //that.Display(0, 1)
|
||
// that.Display(0, Map{"type": 1, "token": that.SessionId})
|
||
//},
|
||
//网页签名
|
||
//"sign": func(that *Context) {
|
||
//
|
||
// if that.Req.FormValue("sign_url") == "" {
|
||
// that.Display(2, "参数不足")
|
||
// return
|
||
// }
|
||
// signUrl := that.Req.FormValue("sign_url")
|
||
// signUrl = "https://hycb.hoteas.com/wxapp"
|
||
//
|
||
// js := officialAccount.GetJs()
|
||
// //js.GetConfig()
|
||
// //js := weixin.GetJs()
|
||
// cfg1, e := js.GetConfig(signUrl)
|
||
// if e != nil {
|
||
// that.Display(7, e)
|
||
// return
|
||
// }
|
||
//
|
||
// sign := Map{
|
||
// "appId": cfg1.AppID,
|
||
// "timestamp": cfg1.Timestamp,
|
||
// "nonceStr": cfg1.NonceStr,
|
||
// "signature": cfg1.Signature,
|
||
// }
|
||
//
|
||
// that.Display(0, sign)
|
||
//
|
||
//},
|
||
}
|