user.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. package handler
  2. import (
  3. "github.com/gin-gonic/gin"
  4. "github.com/go-nunu/nunu-layout-advanced/internal/pkg/request"
  5. "github.com/go-nunu/nunu-layout-advanced/internal/pkg/response"
  6. "github.com/go-nunu/nunu-layout-advanced/internal/service"
  7. "net/http"
  8. )
  9. type UserHandler interface {
  10. Register(ctx *gin.Context)
  11. Login(ctx *gin.Context)
  12. GetProfile(ctx *gin.Context)
  13. UpdateProfile(ctx *gin.Context)
  14. }
  15. func NewUserHandler(handler *Handler, userService service.UserService) UserHandler {
  16. return &userHandler{
  17. Handler: handler,
  18. userService: userService,
  19. }
  20. }
  21. type userHandler struct {
  22. *Handler
  23. userService service.UserService
  24. }
  25. // Register godoc
  26. // @Summary ping example
  27. // @Schemes
  28. // @Description do ping
  29. // @Tags example
  30. // @Accept json
  31. // @Produce json
  32. // @Success 200 {string} Helloworld
  33. // @Router /example/helloworld [get]
  34. func (h *userHandler) Register(ctx *gin.Context) {
  35. req := new(request.RegisterRequest)
  36. if err := ctx.ShouldBindJSON(req); err != nil {
  37. response.HandleError(ctx, http.StatusBadRequest, response.ErrBadRequest, nil)
  38. return
  39. }
  40. if err := h.userService.Register(ctx, req); err != nil {
  41. response.HandleError(ctx, http.StatusBadRequest, response.ErrBadRequest, nil)
  42. return
  43. }
  44. response.HandleSuccess(ctx, nil)
  45. }
  46. func (h *userHandler) Login(ctx *gin.Context) {
  47. var req request.LoginRequest
  48. if err := ctx.ShouldBindJSON(&req); err != nil {
  49. response.HandleError(ctx, http.StatusBadRequest, response.ErrBadRequest, nil)
  50. return
  51. }
  52. token, err := h.userService.Login(ctx, &req)
  53. if err != nil {
  54. response.HandleError(ctx, http.StatusUnauthorized, response.ErrUnauthorized, nil)
  55. return
  56. }
  57. response.HandleSuccess(ctx, gin.H{
  58. "accessToken": token,
  59. })
  60. }
  61. func (h *userHandler) GetProfile(ctx *gin.Context) {
  62. userId := GetUserIdFromCtx(ctx)
  63. if userId == "" {
  64. response.HandleError(ctx, http.StatusUnauthorized, response.ErrUnauthorized, nil)
  65. return
  66. }
  67. user, err := h.userService.GetProfile(ctx, userId)
  68. if err != nil {
  69. response.HandleError(ctx, http.StatusBadRequest, response.ErrBadRequest, nil)
  70. return
  71. }
  72. response.HandleSuccess(ctx, user)
  73. }
  74. func (h *userHandler) UpdateProfile(ctx *gin.Context) {
  75. userId := GetUserIdFromCtx(ctx)
  76. var req request.UpdateProfileRequest
  77. if err := ctx.ShouldBindJSON(&req); err != nil {
  78. response.HandleError(ctx, http.StatusBadRequest, response.ErrBadRequest, nil)
  79. return
  80. }
  81. if err := h.userService.UpdateProfile(ctx, userId, &req); err != nil {
  82. response.HandleError(ctx, http.StatusInternalServerError, response.ErrInternalServerError, nil)
  83. return
  84. }
  85. response.HandleSuccess(ctx, nil)
  86. }