106 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			106 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package api
 | |
| 
 | |
| import (
 | |
| 	"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
 | |
| 	"git.solsynth.dev/hydrogen/passport/pkg/internal/http/exts"
 | |
| 	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
 | |
| 	"git.solsynth.dev/hydrogen/passport/pkg/internal/services"
 | |
| 	"github.com/gofiber/fiber/v2"
 | |
| 	"strconv"
 | |
| )
 | |
| 
 | |
| func listDailySignRecord(c *fiber.Ctx) error {
 | |
| 	take := c.QueryInt("take", 0)
 | |
| 	offset := c.QueryInt("offset", 0)
 | |
| 
 | |
| 	if err := exts.EnsureAuthenticated(c); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	user := c.Locals("user").(models.Account)
 | |
| 
 | |
| 	var count int64
 | |
| 	if err := database.C.
 | |
| 		Model(&models.SignRecord{}).
 | |
| 		Where("account_id = ?", user.ID).
 | |
| 		Count(&count).Error; err != nil {
 | |
| 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 | |
| 	}
 | |
| 
 | |
| 	var records []models.SignRecord
 | |
| 	if err := database.C.
 | |
| 		Where("account_id = ?", user.ID).
 | |
| 		Limit(take).Offset(offset).
 | |
| 		Order("created_at DESC").
 | |
| 		Find(&records).Error; err != nil {
 | |
| 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 | |
| 	}
 | |
| 
 | |
| 	return c.JSON(fiber.Map{
 | |
| 		"count": count,
 | |
| 		"data":  records,
 | |
| 	})
 | |
| }
 | |
| 
 | |
| func listOtherUserDailySignRecord(c *fiber.Ctx) error {
 | |
| 	take := c.QueryInt("take", 0)
 | |
| 	offset := c.QueryInt("offset", 0)
 | |
| 
 | |
| 	alias := c.Params("alias")
 | |
| 
 | |
| 	var account models.Account
 | |
| 	if err := database.C.
 | |
| 		Where(&models.Account{Name: alias}).
 | |
| 		First(&account).Error; err != nil {
 | |
| 		return fiber.NewError(fiber.StatusBadRequest, err.Error())
 | |
| 	}
 | |
| 
 | |
| 	var count int64
 | |
| 	if err := database.C.
 | |
| 		Model(&models.SignRecord{}).
 | |
| 		Where("account_id = ?", account.ID).
 | |
| 		Count(&count).Error; err != nil {
 | |
| 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 | |
| 	}
 | |
| 
 | |
| 	var records []models.SignRecord
 | |
| 	if err := database.C.
 | |
| 		Where("account_id = ?", account.ID).
 | |
| 		Limit(take).Offset(offset).
 | |
| 		Order("created_at DESC").
 | |
| 		Find(&records).Error; err != nil {
 | |
| 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())
 | |
| 	}
 | |
| 
 | |
| 	return c.JSON(fiber.Map{
 | |
| 		"count": count,
 | |
| 		"data":  records,
 | |
| 	})
 | |
| }
 | |
| 
 | |
| func getTodayDailySign(c *fiber.Ctx) error {
 | |
| 	if err := exts.EnsureAuthenticated(c); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	user := c.Locals("user").(models.Account)
 | |
| 
 | |
| 	if record, err := services.GetTodayDailySign(user); err != nil {
 | |
| 		return fiber.NewError(fiber.StatusNotFound, err.Error())
 | |
| 	} else {
 | |
| 		return c.JSON(record)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func doDailySign(c *fiber.Ctx) error {
 | |
| 	if err := exts.EnsureAuthenticated(c); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	user := c.Locals("user").(models.Account)
 | |
| 
 | |
| 	if record, err := services.DailySign(user); err != nil {
 | |
| 		return fiber.NewError(fiber.StatusBadRequest, err.Error())
 | |
| 	} else {
 | |
| 		services.AddEvent(user.ID, "dailySign", strconv.Itoa(int(record.ID)), c.IP(), c.Get(fiber.HeaderUserAgent))
 | |
| 		return c.JSON(record)
 | |
| 	}
 | |
| }
 |