package auth import ( "fmt" "net/url" "nixcn-cms/data" "nixcn-cms/internal/exception" "nixcn-cms/pkgs/authcode" "nixcn-cms/utils" "github.com/gin-gonic/gin" "github.com/google/uuid" ) const () func Exchange(c *gin.Context) { var exchangeReq struct { ClientId string `json:"client_id"` RedirectUri string `json:"redirect_uri"` State string `json:"state"` } err := c.ShouldBindJSON(&exchangeReq) if err != nil { fmt.Println(err) errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusClient). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeCommon). SetOriginal(exception.CommonErrorInvalidInput). Build() utils.HttpResponse(c, 400, errorCode) return } userIdOrig, ok := c.Get("user_id") if !ok { errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusClient). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeCommon). SetOriginal(exception.CommonErrorUnauthorized). Build() utils.HttpResponse(c, 401, errorCode) return } userId, err := uuid.Parse(userIdOrig.(string)) if err != nil { errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusServer). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeCommon). SetOriginal(exception.CommonErrorUuidParseFailed). Build() utils.HttpResponse(c, 500, errorCode) return } userData := new(data.User) user, err := userData.GetByUserId(userId) if err != nil { errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusServer). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeSpecific). SetOriginal(exception.AuthExchangeGetUserIdFailed). Build() utils.HttpResponse(c, 500, errorCode) return } code, err := authcode.NewAuthCode(exchangeReq.ClientId, user.Email) if err != nil { errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusServer). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeSpecific). SetOriginal(exception.AuthExchangeCodeGenFailed). Build() utils.HttpResponse(c, 500, errorCode) return } url, err := url.Parse(exchangeReq.RedirectUri) if err != nil { errorCode := new(exception.Builder). SetStatus(exception.ErrorStatusClient). SetService(exception.AuthService). SetEndpoint(exception.AuthExchangeEndpoint). SetType(exception.ErrorTypeSpecific). SetOriginal(exception.AuthExchangeInvalidRedirectUri). Build() utils.HttpResponse(c, 400, errorCode) return } query := url.Query() query.Set("code", code) url.RawQuery = query.Encode() exchangeResp := struct { RedirectUri string `json:"redirect_uri"` }{url.String()} utils.HttpResponse(c, 200, "", "success", exchangeResp) }