package grpc

import (
	"context"

	"git.techease.ru/Smart-search/smart-search-back/pkg/errors"
	pb "git.techease.ru/Smart-search/smart-search-back/pkg/pb/user"
)

func (h *UserHandler) GetInfo(ctx context.Context, req *pb.GetInfoRequest) (*pb.GetInfoResponse, error) {
	user, err := h.userService.GetInfo(ctx, int(req.UserId))
	if err != nil {
		return nil, errors.ToGRPCError(err, h.logger, "UserService.GetInfo")
	}

	return &pb.GetInfoResponse{
		Email:         user.Email,
		Name:          user.Name,
		Phone:         user.Phone,
		CompanyName:   user.CompanyName,
		PaymentStatus: user.PaymentStatus,
	}, nil
}

func (h *UserHandler) GetBalance(ctx context.Context, req *pb.GetBalanceRequest) (*pb.GetBalanceResponse, error) {
	balance, err := h.userService.GetBalance(ctx, int(req.UserId))
	if err != nil {
		return nil, errors.ToGRPCError(err, h.logger, "UserService.GetBalance")
	}

	return &pb.GetBalanceResponse{
		Balance: balance,
	}, nil
}

func (h *UserHandler) GetStatistics(ctx context.Context, req *pb.GetStatisticsRequest) (*pb.GetStatisticsResponse, error) {
	stats, err := h.userService.GetStatistics(ctx, int(req.UserId))
	if err != nil {
		return nil, errors.ToGRPCError(err, h.logger, "UserService.GetStatistics")
	}

	return &pb.GetStatisticsResponse{
		TotalRequests:      int32(stats.RequestsCount),
		SuccessfulRequests: int32(stats.SuppliersCount),
		FailedRequests:     0,
		TotalSpent:         0,
	}, nil
}

func (h *UserHandler) GetBalanceStatistics(ctx context.Context, req *pb.GetBalanceStatisticsRequest) (*pb.GetBalanceStatisticsResponse, error) {
	balance, err := h.userService.GetBalance(ctx, int(req.UserId))
	if err != nil {
		return nil, errors.ToGRPCError(err, h.logger, "UserService.GetBalanceStatistics")
	}

	stats, err := h.userService.GetStatistics(ctx, int(req.UserId))
	if err != nil {
		return nil, errors.ToGRPCError(err, h.logger, "UserService.GetBalanceStatistics")
	}

	return &pb.GetBalanceStatisticsResponse{
		Balance:       balance,
		TotalRequests: int32(stats.RequestsCount),
		TotalSpent:    0,
	}, nil
}