package logic

import (
	"context"
	"errors"
	"time"

	"juwan-backend/app/shop/rpc/internal/svc"
	"juwan-backend/app/shop/rpc/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateShopPlayersLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUpdateShopPlayersLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateShopPlayersLogic {
	return &UpdateShopPlayersLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *UpdateShopPlayersLogic) UpdateShopPlayers(in *pb.UpdateShopPlayersReq) (*pb.UpdateShopPlayersResp, error) {
	if in.ShopId <= 0 || in.PlayerId <= 0 {
		return nil, errors.New("invalid shop_id or player_id")
	}

	// ent 的 Update().Where() 不支持复合主键表（shop_players 的主键是 (shop_id, player_id)），
	// 会报 "sql/sqlgraph: invalid composite id for update table"。
	// Create/Delete/Query 不受影响，仅 Update 需要走原生 SQL。
	query := `UPDATE shop_players SET is_primary = $1, left_at = $2 WHERE shop_id = $3 AND player_id = $4`

	var leftAt *time.Time
	if in.LeftAt > 0 {
		t := time.Unix(in.LeftAt, 0)
		leftAt = &t
	}

	result, err := l.svcCtx.DBRW.ExecContext(l.ctx, query, in.IsPrimary, leftAt, in.ShopId, in.PlayerId)
	if err != nil {
		logx.Errorf("update shop players failed, %s", err.Error())
		return nil, errors.New("update shop players failed")
	}

	affected, err := result.RowsAffected()
	if err != nil {
		logx.Errorf("get rows affected failed, %s", err.Error())
		return nil, errors.New("update shop players failed")
	}
	if affected == 0 {
		return nil, errors.New("shop player not found")
	}

	return &pb.UpdateShopPlayersResp{}, nil
}