lunar-tear/server/internal/service/gift.go

package service

import (
	"context"
	"fmt"
	"log"
	"slices"
	"sort"
	"time"

	pb "lunar-tear/server/gen/proto"
	"lunar-tear/server/internal/gametime"
	"lunar-tear/server/internal/store"
	"lunar-tear/server/internal/userdata"

	emptypb "google.golang.org/protobuf/types/known/emptypb"
	"google.golang.org/protobuf/types/known/timestamppb"
)

type GiftServiceServer struct {
	pb.UnimplementedGiftServiceServer
	users    store.UserRepository
	sessions store.SessionRepository
}

func NewGiftServiceServer(users store.UserRepository, sessions store.SessionRepository) *GiftServiceServer {
	return &GiftServiceServer{users: users, sessions: sessions}
}

func (s *GiftServiceServer) ReceiveGift(ctx context.Context, req *pb.ReceiveGiftRequest) (*pb.ReceiveGiftResponse, error) {
	log.Printf("[GiftService] ReceiveGift: giftUuids=%d", len(req.UserGiftUuid))

	userId := currentUserId(ctx, s.users, s.sessions)
	received := make([]string, 0, len(req.UserGiftUuid))
	_, err := s.users.UpdateUser(userId, func(user *store.UserState) {
		nowMillis := gametime.NowMillis()
		remaining := make([]store.NotReceivedGiftState, 0, len(user.Gifts.NotReceived))
		for _, gift := range user.Gifts.NotReceived {
			if slices.Contains(req.UserGiftUuid, gift.UserGiftUuid) {
				received = append(received, gift.UserGiftUuid)
				user.Gifts.Received = append(user.Gifts.Received, store.ReceivedGiftState{
					GiftCommon:       gift.GiftCommon,
					ReceivedDatetime: nowMillis,
				})
				continue
			}
			remaining = append(remaining, gift)
		}
		user.Gifts.NotReceived = remaining
		user.Notifications.GiftNotReceiveCount = int32(len(user.Gifts.NotReceived))
	})
	if err != nil {
		return &pb.ReceiveGiftResponse{
			ReceivedGiftUuid: []string{},
			ExpiredGiftUuid:  []string{},
			OverflowGiftUuid: []string{},
			DiffUserData:     userdata.EmptyDiff(),
		}, nil
	}

	return &pb.ReceiveGiftResponse{
		ReceivedGiftUuid: received,
		ExpiredGiftUuid:  []string{},
		OverflowGiftUuid: []string{},
		DiffUserData:     userdata.EmptyDiff(),
	}, nil
}

func (s *GiftServiceServer) GetGiftList(ctx context.Context, req *pb.GetGiftListRequest) (*pb.GetGiftListResponse, error) {
	log.Printf("[GiftService] GetGiftList: rewardKinds=%v expirationType=%d ascending=%v nextCursor=%d previousCursor=%d getCount=%d",
		req.RewardKindType, req.ExpirationType, req.IsAscendingSort, req.NextCursor, req.PreviousCursor, req.GetCount)

	userId := currentUserId(ctx, s.users, s.sessions)
	user, err := s.users.SnapshotUser(userId)
	if err != nil {
		return nil, fmt.Errorf("snapshot user: %w", err)
	}

	gifts := append([]store.NotReceivedGiftState(nil), user.Gifts.NotReceived...)
	sort.Slice(gifts, func(i, j int) bool {
		if req.IsAscendingSort {
			return gifts[i].ExpirationDatetime < gifts[j].ExpirationDatetime
		}
		return gifts[i].ExpirationDatetime > gifts[j].ExpirationDatetime
	})
	if req.GetCount > 0 && len(gifts) > int(req.GetCount) {
		gifts = gifts[:req.GetCount]
	}

	items := make([]*pb.NotReceivedGift, 0, len(gifts))
	for _, gift := range gifts {
		items = append(items, &pb.NotReceivedGift{
			GiftCommon:         toProtoGiftCommon(gift.GiftCommon),
			ExpirationDatetime: timestampOrNilGift(gift.ExpirationDatetime),
			UserGiftUuid:       gift.UserGiftUuid,
		})
	}

	return &pb.GetGiftListResponse{
		Gift:           items,
		TotalPageCount: pageCount(len(user.Gifts.NotReceived), int(req.GetCount)),
		NextCursor:     0,
		PreviousCursor: 0,
		DiffUserData:   userdata.EmptyDiff(),
	}, nil
}

func (s *GiftServiceServer) GetGiftReceiveHistoryList(ctx context.Context, req *emptypb.Empty) (*pb.GetGiftReceiveHistoryListResponse, error) {
	log.Printf("[GiftService] GetGiftReceiveHistoryList")
	userId := currentUserId(ctx, s.users, s.sessions)
	user, err := s.users.SnapshotUser(userId)
	if err != nil {
		return nil, fmt.Errorf("snapshot user: %w", err)
	}

	items := make([]*pb.ReceivedGift, 0, len(user.Gifts.Received))
	for _, gift := range user.Gifts.Received {
		items = append(items, &pb.ReceivedGift{
			GiftCommon:       toProtoGiftCommon(gift.GiftCommon),
			ReceivedDatetime: timestampOrNilGift(gift.ReceivedDatetime),
		})
	}
	return &pb.GetGiftReceiveHistoryListResponse{
		Gift:         items,
		DiffUserData: userdata.EmptyDiff(),
	}, nil
}

func toProtoGiftCommon(gift store.GiftCommonState) *pb.GiftCommon {
	return &pb.GiftCommon{
		PossessionType:        gift.PossessionType,
		PossessionId:          gift.PossessionId,
		Count:                 gift.Count,
		GrantDatetime:         timestampOrNilGift(gift.GrantDatetime),
		DescriptionGiftTextId: gift.DescriptionGiftTextId,
		EquipmentData:         gift.EquipmentData,
	}
}

func timestampOrNilGift(unixMillis int64) *timestamppb.Timestamp {
	if unixMillis == 0 {
		return nil
	}
	return timestamppb.New(time.UnixMilli(unixMillis))
}

func pageCount(total, pageSize int) int32 {
	if total == 0 {
		return 0
	}
	if pageSize <= 0 {
		return 1
	}
	pages := total / pageSize
	if total%pageSize != 0 {
		pages++
	}
	return int32(pages)
}