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 }