Add context for everything

Signed-off-by: Asai Neko <sugar@sne.moe>

service/auth/exchange.go

@@ -64,7 +64,7 @@ func Exchange(c *gin.Context) {
 	}
 
 	userData := new(data.User)
-	user, err := userData.GetByUserId(userId)
+	user, err := userData.GetByUserId(c, userId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).
@@ -78,7 +78,7 @@ func Exchange(c *gin.Context) {
 		return
 	}
 
-	code, err := authcode.NewAuthCode(exchangeReq.ClientId, user.Email)
+	code, err := authcode.NewAuthCode(c, exchangeReq.ClientId, user.Email)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/auth/magic.go

@@ -51,7 +51,7 @@ func Magic(c *gin.Context) {
 		return
 	}
 
-	code, err := authcode.NewAuthCode(req.ClientId, req.Email)
+	code, err := authcode.NewAuthCode(c, req.ClientId, req.Email)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/auth/redirect.go

@@ -55,7 +55,7 @@ func Redirect(c *gin.Context) {
 	code := c.Query("code")
 
 	// Verify email token
-	authCode, ok := authcode.VerifyAuthCode(code)
+	authCode, ok := authcode.VerifyAuthCode(c, code)
 	if !ok {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -70,7 +70,7 @@ func Redirect(c *gin.Context) {
 
 	// Verify if user exists
 	userData := new(data.User)
-	user, err := userData.GetByEmail(authCode.Email)
+	user, err := userData.GetByEmail(c, authCode.Email)
 
 	if err != nil {
 		if err == gorm.ErrRecordNotFound {
@@ -80,7 +80,7 @@ func Redirect(c *gin.Context) {
 			user.Email = authCode.Email
 			user.Username = user.UserId.String()
 			user.PermissionLevel = 10
-			if err := user.Create(); err != nil {
+			if err := user.Create(c); err != nil {
 				errorCode := new(exception.Builder).
 					SetStatus(exception.StatusServer).
 					SetService(exception.ServiceAuth).
@@ -107,7 +107,7 @@ func Redirect(c *gin.Context) {
 	}
 
 	clientData := new(data.Client)
-	client, err := clientData.GetClientByClientId(clientId)
+	client, err := clientData.GetClientByClientId(c, clientId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -135,7 +135,7 @@ func Redirect(c *gin.Context) {
 		return
 	}
 
-	newCode, err := authcode.NewAuthCode(clientId, authCode.Email)
+	newCode, err := authcode.NewAuthCode(c, clientId, authCode.Email)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/auth/refresh.go

@@ -31,7 +31,7 @@ func Refresh(c *gin.Context) {
 		Application: viper.GetString("server.application"),
 	}
 
-	accessToken, err := JwtTool.RefreshAccessToken(req.RefreshToken)
+	accessToken, err := JwtTool.RefreshAccessToken(c, req.RefreshToken)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -45,7 +45,7 @@ func Refresh(c *gin.Context) {
 		return
 	}
 
-	refreshToken, err := JwtTool.RenewRefreshToken(req.RefreshToken)
+	refreshToken, err := JwtTool.RenewRefreshToken(c, req.RefreshToken)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/auth/token.go

@@ -32,7 +32,7 @@ func Token(c *gin.Context) {
 		return
 	}
 
-	authCode, ok := authcode.VerifyAuthCode(req.Code)
+	authCode, ok := authcode.VerifyAuthCode(c, req.Code)
 	if !ok {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -46,7 +46,7 @@ func Token(c *gin.Context) {
 	}
 
 	userData := new(data.User)
-	user, err := userData.GetByEmail(authCode.Email)
+	user, err := userData.GetByEmail(c, authCode.Email)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).
@@ -64,7 +64,7 @@ func Token(c *gin.Context) {
 	JwtTool := authtoken.Token{
 		Application: viper.GetString("server.application"),
 	}
-	accessToken, refreshToken, err := JwtTool.IssueTokens(authCode.ClientId, user.UserId)
+	accessToken, refreshToken, err := JwtTool.IssueTokens(c, authCode.ClientId, user.UserId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/event/checkin.go

@@ -66,7 +66,7 @@ func Checkin(c *gin.Context) {
 		return
 	}
 	data.UserId = userId
-	code, err := data.GenCheckinCode(eventId)
+	code, err := data.GenCheckinCode(c, eventId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).
@@ -93,7 +93,7 @@ func CheckinSubmit(c *gin.Context) {
 	c.ShouldBindJSON(&req)
 
 	attendanceData := new(data.Attendance)
-	err := attendanceData.VerifyCheckinCode(req.ChekinCode)
+	err := attendanceData.VerifyCheckinCode(c, req.ChekinCode)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -164,7 +164,7 @@ func CheckinQuery(c *gin.Context) {
 	}
 
 	attendanceData := new(data.Attendance)
-	attendance, err := attendanceData.GetAttendance(userId, eventId)
+	attendance, err := attendanceData.GetAttendance(c, userId, eventId)
 
 	if err != nil {
 		errorCode := new(exception.Builder).

service/event/info.go

@@ -40,7 +40,7 @@ func Info(c *gin.Context) {
 		return
 	}
 
-	event, err := eventData.GetEventById(eventId)
+	event, err := eventData.GetEventById(c, eventId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).

service/user/full.go

@@ -36,7 +36,7 @@ func Full(c *gin.Context) {
 		return
 	}
 
-	userData, err := new(data.User).GetByUserId(userId)
+	userData, err := new(data.User).GetByUserId(c, userId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -50,7 +50,7 @@ func Full(c *gin.Context) {
 		return
 	}
 
-	users, err := userData.GetFullTable()
+	users, err := userData.GetFullTable(c)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/user/info.go

@@ -38,7 +38,7 @@ func Info(c *gin.Context) {
 	}
 
 	// Get user from database
-	user, err := userData.GetByUserId(userId)
+	user, err := userData.GetByUserId(c, userId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).

service/user/list.go

@@ -57,7 +57,7 @@ func List(c *gin.Context) {
 	}
 
 	// Get user list from search engine
-	list, err := new(data.User).FastListUsers(limitNum, offsetNum)
+	list, err := new(data.User).FastListUsers(c, limitNum, offsetNum)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusServer).

service/user/update.go

@@ -56,7 +56,7 @@ func Update(c *gin.Context) {
 	}
 
 	// Get user info
-	userData, err := new(data.User).GetByUserId(userId)
+	userData, err := new(data.User).GetByUserId(c, userId)
 	if err != nil {
 		errorCode := new(exception.Builder).
 			SetStatus(exception.StatusUser).
@@ -157,7 +157,7 @@ func Update(c *gin.Context) {
 	}
 
 	// Update user info
-	userData.UpdateByUserID(userId)
+	userData.UpdateByUserID(c, userId)
 
 	errorCode := new(exception.Builder).
 		SetStatus(exception.StatusUser).