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

package service

import (
	"context"
	"log"

	pb "lunar-tear/server/gen/proto"
	"lunar-tear/server/internal/gametime"
	"lunar-tear/server/internal/model"
	"lunar-tear/server/internal/runtime"
	"lunar-tear/server/internal/store"
)

type CageOrnamentServiceServer struct {
	pb.UnimplementedCageOrnamentServiceServer
	users    store.UserRepository
	sessions store.SessionRepository
	holder   *runtime.Holder
}

func NewCageOrnamentServiceServer(users store.UserRepository, sessions store.SessionRepository, holder *runtime.Holder) *CageOrnamentServiceServer {
	return &CageOrnamentServiceServer{users: users, sessions: sessions, holder: holder}
}

func (s *CageOrnamentServiceServer) ReceiveReward(ctx context.Context, req *pb.ReceiveRewardRequest) (*pb.ReceiveRewardResponse, error) {
	log.Printf("[CageOrnamentService] ReceiveReward: cageOrnamentId=%d", req.CageOrnamentId)

	cat := s.holder.Get()
	reward, ok := cat.CageOrnament.LookupReward(req.CageOrnamentId)
	if !ok {
		log.Fatalf("[CageOrnamentService] ReceiveReward: no reward for cageOrnamentId=%d", req.CageOrnamentId)
	}
	granter := cat.QuestHandler.Granter

	userId := CurrentUserId(ctx, s.users, s.sessions)
	nowMillis := gametime.NowMillis()
	s.users.UpdateUser(userId, func(user *store.UserState) {
		user.CageOrnamentRewards[req.CageOrnamentId] = store.CageOrnamentRewardState{
			CageOrnamentId:      req.CageOrnamentId,
			AcquisitionDatetime: nowMillis,
			LatestVersion:       nowMillis,
		}
		granter.GrantFull(user, model.PossessionType(reward.PossessionType), reward.PossessionId, reward.Count, nowMillis)
	})

	return &pb.ReceiveRewardResponse{
		CageOrnamentReward: []*pb.CageOrnamentReward{
			{
				PossessionType: reward.PossessionType,
				PossessionId:   reward.PossessionId,
				Count:          reward.Count,
			},
		},
	}, nil
}

func (s *CageOrnamentServiceServer) RecordAccess(ctx context.Context, req *pb.RecordAccessRequest) (*pb.RecordAccessResponse, error) {
	log.Printf("[CageOrnamentService] RecordAccess: cageOrnamentId=%d", req.CageOrnamentId)

	userId := CurrentUserId(ctx, s.users, s.sessions)
	nowMillis := gametime.NowMillis()
	s.users.UpdateUser(userId, func(user *store.UserState) {
		if _, exists := user.CageOrnamentRewards[req.CageOrnamentId]; !exists {
			user.CageOrnamentRewards[req.CageOrnamentId] = store.CageOrnamentRewardState{
				CageOrnamentId:      req.CageOrnamentId,
				AcquisitionDatetime: nowMillis,
				LatestVersion:       nowMillis,
			}
		}
	})

	return &pb.RecordAccessResponse{}, nil
}