mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
583 lines
15 KiB
583 lines
15 KiB
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"unicode/utf8"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
|
"github.com/rocboss/paopao-ce/internal/model"
|
|
"github.com/rocboss/paopao-ce/internal/service"
|
|
"github.com/rocboss/paopao-ce/pkg/app"
|
|
"github.com/rocboss/paopao-ce/pkg/convert"
|
|
"github.com/rocboss/paopao-ce/pkg/errcode"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/smartwalle/alipay/v3"
|
|
)
|
|
|
|
// Login 用户登录
|
|
func Login(c *gin.Context) {
|
|
param := service.AuthRequest{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
user, err := service.DoLogin(c, ¶m)
|
|
if err != nil {
|
|
logrus.Errorf("service.DoLogin err: %v", err)
|
|
response.ToErrorResponse(err.(*errcode.Error))
|
|
return
|
|
}
|
|
|
|
token, err := app.GenerateToken(user)
|
|
if err != nil {
|
|
logrus.Errorf("app.GenerateToken err: %v", err)
|
|
response.ToErrorResponse(errcode.UnauthorizedTokenGenerate)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"token": token,
|
|
})
|
|
}
|
|
|
|
// Register 用户注册
|
|
func Register(c *gin.Context) {
|
|
|
|
param := service.RegisterRequest{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
// 用户名检查
|
|
err := service.ValidUsername(param.Username)
|
|
if err != nil {
|
|
logrus.Errorf("service.Register err: %v", err)
|
|
response.ToErrorResponse(err.(*errcode.Error))
|
|
return
|
|
}
|
|
|
|
// 密码检查
|
|
err = service.CheckPassword(param.Password)
|
|
if err != nil {
|
|
logrus.Errorf("service.Register err: %v", err)
|
|
response.ToErrorResponse(err.(*errcode.Error))
|
|
return
|
|
}
|
|
|
|
user, err := service.Register(
|
|
param.Username,
|
|
param.Password,
|
|
)
|
|
|
|
if err != nil {
|
|
logrus.Errorf("service.Register err: %v", err)
|
|
response.ToErrorResponse(errcode.UserRegisterFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"id": user.ID,
|
|
"username": user.Username,
|
|
})
|
|
}
|
|
|
|
// 获取用户基本信息
|
|
func GetUserInfo(c *gin.Context) {
|
|
param := service.AuthRequest{}
|
|
response := app.NewResponse(c)
|
|
|
|
if username, exists := c.Get("USERNAME"); exists {
|
|
param.Username = username.(string)
|
|
}
|
|
|
|
user, err := service.GetUserInfo(¶m)
|
|
|
|
if err != nil {
|
|
response.ToErrorResponse(errcode.UnauthorizedAuthNotExist)
|
|
return
|
|
}
|
|
phone := ""
|
|
if user.Phone != "" && len(user.Phone) == 11 {
|
|
phone = user.Phone[0:3] + "****" + user.Phone[7:]
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"id": user.ID,
|
|
"nickname": user.Nickname,
|
|
"username": user.Username,
|
|
"status": user.Status,
|
|
"avatar": user.Avatar,
|
|
"balance": user.Balance,
|
|
"phone": phone,
|
|
"is_admin": user.IsAdmin,
|
|
})
|
|
}
|
|
|
|
// 修改密码
|
|
func ChangeUserPassword(c *gin.Context) {
|
|
param := service.ChangePasswordReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
user := &model.User{}
|
|
if u, exists := c.Get("USER"); exists {
|
|
user = u.(*model.User)
|
|
}
|
|
|
|
// 密码检查
|
|
err := service.CheckPassword(param.Password)
|
|
if err != nil {
|
|
logrus.Errorf("service.Register err: %v", err)
|
|
response.ToErrorResponse(err.(*errcode.Error))
|
|
return
|
|
}
|
|
|
|
// 旧密码校验
|
|
if !service.ValidPassword(user.Password, param.OldPassword, user.Salt) {
|
|
response.ToErrorResponse(errcode.ErrorOldPassword)
|
|
return
|
|
}
|
|
|
|
// 更新入库
|
|
user.Password, user.Salt = service.EncryptPasswordAndSalt(param.Password)
|
|
service.UpdateUserInfo(user)
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
// 修改昵称
|
|
func ChangeNickname(c *gin.Context) {
|
|
param := service.ChangeNicknameReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
user := &model.User{}
|
|
if u, exists := c.Get("USER"); exists {
|
|
user = u.(*model.User)
|
|
}
|
|
|
|
if utf8.RuneCountInString(param.Nickname) < 2 || utf8.RuneCountInString(param.Nickname) > 12 {
|
|
response.ToErrorResponse(errcode.NicknameLengthLimit)
|
|
return
|
|
}
|
|
|
|
// 执行绑定
|
|
user.Nickname = param.Nickname
|
|
service.UpdateUserInfo(user)
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
// 修改头像
|
|
func ChangeAvatar(c *gin.Context) {
|
|
param := service.ChangeAvatarReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
user := &model.User{}
|
|
if u, exists := c.Get("USER"); exists {
|
|
user = u.(*model.User)
|
|
}
|
|
|
|
if err := attachmentChecker.Check(param.Avatar); err != nil {
|
|
response.ToErrorResponse(errcode.InvalidParams)
|
|
return
|
|
}
|
|
|
|
// 执行绑定
|
|
user.Avatar = param.Avatar
|
|
service.UpdateUserInfo(user)
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
// 用户绑定手机号
|
|
func BindUserPhone(c *gin.Context) {
|
|
param := service.UserPhoneBindReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
user := &model.User{}
|
|
if u, exists := c.Get("USER"); exists {
|
|
user = u.(*model.User)
|
|
}
|
|
|
|
// 手机重复性检查
|
|
if service.CheckPhoneExist(user.ID, param.Phone) {
|
|
response.ToErrorResponse(errcode.ExistedUserPhone)
|
|
return
|
|
}
|
|
|
|
if err := service.CheckPhoneCaptcha(param.Phone, param.Captcha); err != nil {
|
|
logrus.Errorf("service.CheckPhoneCaptcha err: %v\n", err)
|
|
response.ToErrorResponse(err)
|
|
return
|
|
}
|
|
|
|
// 执行绑定
|
|
user.Phone = param.Phone
|
|
service.UpdateUserInfo(user)
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
// 修改用户状态
|
|
func ChangeUserStatus(c *gin.Context) {
|
|
param := service.ChangeUserStatusReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
if param.Status != model.UserStatusNormal && param.Status != model.UserStatusClosed {
|
|
response.ToErrorResponse(errcode.InvalidParams)
|
|
return
|
|
}
|
|
|
|
user, err := service.GetUserByID(param.ID)
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserByID err: %v\n", err)
|
|
response.ToErrorResponse(errcode.NoExistUsername)
|
|
return
|
|
}
|
|
|
|
// 执行更新
|
|
user.Status = param.Status
|
|
service.UpdateUserInfo(user)
|
|
|
|
response.ToResponse(nil)
|
|
}
|
|
|
|
func GetUserProfile(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
username := c.Query("username")
|
|
|
|
user, err := service.GetUserByUsername(username)
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserByUsername err: %v\n", err)
|
|
response.ToErrorResponse(errcode.NoExistUsername)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"id": user.ID,
|
|
"nickname": user.Nickname,
|
|
"username": user.Username,
|
|
"status": user.Status,
|
|
"avatar": user.Avatar,
|
|
"is_admin": user.IsAdmin,
|
|
})
|
|
}
|
|
|
|
func GetUserPosts(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
username := c.Query("username")
|
|
|
|
user, err := service.GetUserByUsername(username)
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserByUsername err: %v\n", err)
|
|
response.ToErrorResponse(errcode.NoExistUsername)
|
|
return
|
|
}
|
|
|
|
visibilities := []model.PostVisibleT{model.PostVisitPublic}
|
|
if u, exists := c.Get("USER"); exists {
|
|
self := u.(*model.User)
|
|
if self.ID == user.ID || self.IsAdmin {
|
|
visibilities = append(visibilities, model.PostVisitPrivate, model.PostVisitFriend)
|
|
} else if service.IsFriend(user.ID, self.ID) {
|
|
visibilities = append(visibilities, model.PostVisitFriend)
|
|
}
|
|
}
|
|
conditions := model.ConditionsT{
|
|
"user_id": user.ID,
|
|
"visibility IN ?": visibilities,
|
|
"ORDER": "latest_replied_on DESC",
|
|
}
|
|
|
|
posts, err := service.GetPostList(&service.PostListReq{
|
|
Conditions: &conditions,
|
|
Offset: (app.GetPage(c) - 1) * app.GetPageSize(c),
|
|
Limit: app.GetPageSize(c),
|
|
})
|
|
if err != nil {
|
|
logrus.Errorf("service.GetPostList err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetPostsFailed)
|
|
return
|
|
}
|
|
totalRows, _ := service.GetPostCount(&conditions)
|
|
|
|
response.ToResponseList(posts, totalRows)
|
|
}
|
|
|
|
func GetUserCollections(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
userID, _ := c.Get("UID")
|
|
posts, totalRows, err := service.GetUserCollections(userID.(int64), (app.GetPage(c)-1)*app.GetPageSize(c), app.GetPageSize(c))
|
|
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserCollections err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetCollectionsFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponseList(posts, totalRows)
|
|
}
|
|
|
|
func GetUserStars(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
userID, _ := c.Get("UID")
|
|
posts, totalRows, err := service.GetUserStars(userID.(int64), (app.GetPage(c)-1)*app.GetPageSize(c), app.GetPageSize(c))
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserStars err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetCollectionsFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponseList(posts, totalRows)
|
|
}
|
|
|
|
func GetSuggestUsers(c *gin.Context) {
|
|
keyword := c.Query("k")
|
|
response := app.NewResponse(c)
|
|
|
|
usernames, err := service.GetSuggestUsers(keyword)
|
|
if err != nil {
|
|
logrus.Errorf("service.GetSuggestUsers err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetCollectionsFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(usernames)
|
|
}
|
|
|
|
func GetSuggestTags(c *gin.Context) {
|
|
keyword := c.Query("k")
|
|
response := app.NewResponse(c)
|
|
|
|
tags, err := service.GetSuggestTags(keyword)
|
|
if err != nil {
|
|
logrus.Errorf("service.GetSuggestTags err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetCollectionsFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(tags)
|
|
}
|
|
|
|
func GetUserRechargeLink(c *gin.Context) {
|
|
param := service.RechargeReq{}
|
|
response := app.NewResponse(c)
|
|
valid, errs := app.BindAndValid(c, ¶m)
|
|
if !valid {
|
|
logrus.Errorf("app.BindAndValid errs: %v", errs)
|
|
response.ToErrorResponse(errcode.InvalidParams.WithDetails(errs.Errors()...))
|
|
return
|
|
}
|
|
|
|
// 下单
|
|
userID, _ := c.Get("UID")
|
|
recharge, err := service.CreateRecharge(userID.(int64), param.Amount)
|
|
if err != nil {
|
|
logrus.Errorf("service.CreateRecharge err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
|
|
}
|
|
|
|
client, err := alipay.New(conf.AlipaySetting.AppID, conf.AlipaySetting.PrivateKey, true)
|
|
// 将 key 的验证调整到初始化阶段
|
|
if err != nil {
|
|
logrus.Errorf("alipay.New err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
err = client.LoadAppPublicCertFromFile("configs/alipayAppCertPublicKey.crt") // 加载应用公钥证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAppPublicCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
err = client.LoadAliPayRootCertFromFile("configs/alipayRootCert.crt") // 加载支付宝根证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAliPayRootCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
err = client.LoadAliPayPublicCertFromFile("configs/alipayCertPublicKey_RSA2.crt") // 加载支付宝公钥证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAliPayPublicCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
p := alipay.TradePreCreate{}
|
|
p.OutTradeNo = fmt.Sprintf("%d", recharge.ID)
|
|
p.Subject = "PaoPao用户钱包充值"
|
|
p.TotalAmount = fmt.Sprintf("%.2f", float64(recharge.Amount)/100.0)
|
|
p.NotifyURL = "https://" + c.Request.Host + "/alipay/notify"
|
|
|
|
rsp, err := client.TradePreCreate(p)
|
|
if err != nil {
|
|
logrus.Errorf("client.TradePreCreate err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
if rsp.Content.Code != alipay.CodeSuccess {
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"id": recharge.ID,
|
|
"pay": rsp.Content.QRCode,
|
|
})
|
|
}
|
|
|
|
func GetUserRechargeResult(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
id := c.Query("id")
|
|
userID, _ := c.Get("UID")
|
|
|
|
recharge, err := service.GetRechargeByID(convert.StrTo(id).MustInt64())
|
|
if err != nil {
|
|
response.ToErrorResponse(errcode.GetRechargeFailed)
|
|
return
|
|
}
|
|
|
|
if recharge.UserID != userID.(int64) {
|
|
response.ToErrorResponse(errcode.GetRechargeFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponse(gin.H{
|
|
"id": recharge.ID,
|
|
"status": recharge.TradeStatus,
|
|
})
|
|
}
|
|
|
|
func AlipayNotify(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
c.Request.ParseForm()
|
|
|
|
aliClient, err := alipay.New(conf.AlipaySetting.AppID, conf.AlipaySetting.PrivateKey, true)
|
|
// 将 key 的验证调整到初始化阶段
|
|
if err != nil {
|
|
logrus.Errorf("alipay.New err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
err = aliClient.LoadAppPublicCertFromFile("configs/alipayAppCertPublicKey.crt") // 加载应用公钥证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAppPublicCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
err = aliClient.LoadAliPayRootCertFromFile("configs/alipayRootCert.crt") // 加载支付宝根证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAliPayRootCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
err = aliClient.LoadAliPayPublicCertFromFile("configs/alipayCertPublicKey_RSA2.crt") // 加载支付宝公钥证书
|
|
if err != nil {
|
|
logrus.Errorf("client.LoadAliPayPublicCertFromFile err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeReqFail)
|
|
return
|
|
}
|
|
|
|
_, err = aliClient.GetTradeNotification(c.Request)
|
|
if err != nil {
|
|
logrus.Errorf("aliClient.GetTradeNotification err: %v\n", err)
|
|
logrus.Infoln(c.Request.Form)
|
|
response.ToErrorResponse(errcode.RechargeNotifyError)
|
|
return
|
|
}
|
|
|
|
id := c.Request.Form.Get("out_trade_no")
|
|
tradeNo := c.Request.Form.Get("trade_no")
|
|
tradeStatus := c.Request.Form.Get("trade_status")
|
|
if tradeStatus == "TRADE_SUCCESS" {
|
|
// 交易支付成功
|
|
err = service.FinishRecharge(c, convert.StrTo(id).MustInt64(), tradeNo)
|
|
if err != nil {
|
|
logrus.Errorf("service.FinishRecharge err: %v\n", err)
|
|
response.ToErrorResponse(errcode.RechargeNotifyError)
|
|
return
|
|
}
|
|
}
|
|
response.Ctx.String(http.StatusOK, "success")
|
|
}
|
|
|
|
func GetUserWalletBills(c *gin.Context) {
|
|
response := app.NewResponse(c)
|
|
|
|
userID, _ := c.Get("UID")
|
|
bills, totalRows, err := service.GetUserWalletBills(userID.(int64), (app.GetPage(c)-1)*app.GetPageSize(c), app.GetPageSize(c))
|
|
|
|
if err != nil {
|
|
logrus.Errorf("service.GetUserWalletBills err: %v\n", err)
|
|
response.ToErrorResponse(errcode.GetCollectionsFailed)
|
|
return
|
|
}
|
|
|
|
response.ToResponseList(bills, totalRows)
|
|
}
|
|
|
|
func userFrom(c *gin.Context) (*model.User, bool) {
|
|
if u, exists := c.Get("USER"); exists {
|
|
user, ok := u.(*model.User)
|
|
return user, ok
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func userIdFrom(c *gin.Context) (int64, bool) {
|
|
if u, exists := c.Get("UID"); exists {
|
|
uid, ok := u.(int64)
|
|
return uid, ok
|
|
}
|
|
return -1, false
|
|
}
|