evening_detective_server/internal/services/team_service/service.go

package team_service

import (
	"context"
	"errors"
	"fmt"
	"html"
	"strings"
	"time"

	"evening_detective_server/internal/modules/email_sender"
	"evening_detective_server/internal/modules/password_generator"
	"evening_detective_server/internal/modules/team_qr_pdf"
	"evening_detective_server/internal/repos"
	"evening_detective_server/internal/repos/games_repo"
	"evening_detective_server/internal/repos/teams_repo"
	"evening_detective_server/internal/services/game_model"
)

type TeamService struct {
	gamesRepo         *games_repo.GamesRepo
	teamsRepo         *teams_repo.TeamsRepo
	passwordGenerator password_generator.IPasswordGenerator
	appBaseURL        string
	emailSender       email_sender.IEmailSender
}

func NewTeamService(
	gamesRepo *games_repo.GamesRepo,
	teamsRepo *teams_repo.TeamsRepo,
	passwordGenerator password_generator.IPasswordGenerator,
	appBaseURL string,
	emailSender email_sender.IEmailSender,
) *TeamService {
	return &TeamService{
		gamesRepo:         gamesRepo,
		teamsRepo:         teamsRepo,
		passwordGenerator: passwordGenerator,
		appBaseURL:        appBaseURL,
		emailSender:       emailSender,
	}
}

func (s *TeamService) AddTeam(
	ctx context.Context,
	creatorID int,
	gameID int,
	name string,
	captainEmail string,
) (*game_model.Team, error) {
	captainEmail = strings.TrimSpace(captainEmail)
	if captainEmail != "" && !validCaptainEmail(captainEmail) {
		return nil, ErrInvalidCaptainEmail
	}
	password, err := s.passwordGenerator.Generate()
	if err != nil {
		return nil, err
	}
	game, err := s.gamesRepo.GetGameByID(ctx, gameID)
	if err != nil {
		if errors.Is(err, games_repo.ErrGameNotFound) {
			return nil, ErrGameNotFound
		}
		return nil, err
	}
	id, err := s.teamsRepo.AddTeamWithParticipation(ctx, creatorID, gameID, name, password, captainEmail, !time.Now().Before(game.StartAt.Add(-24*time.Hour)))
	if err != nil {
		return nil, err
	}
	return &game_model.Team{
		ID:       id,
		Password: password,
	}, nil
}

func (s *TeamService) UpdateTeam(ctx context.Context, teamID int, name string) error {
	err := s.teamsRepo.UpdateTeam(ctx, teamID, name)
	if errors.Is(err, teams_repo.ErrTeamNotFound) {
		return ErrTeamNotFound
	}
	return err
}

func (s *TeamService) DeleteTeam(ctx context.Context, teamID int) error {
	err := s.teamsRepo.DeleteTeam(ctx, teamID)
	if errors.Is(err, teams_repo.ErrTeamNotFound) {
		return ErrTeamNotFound
	}
	return err
}

func (s *TeamService) RegisterTeam(ctx context.Context, gameID int, deliveryID string, registration Registration) (int, error) {
	if err := registration.Validate(); err != nil {
		return 0, err
	}
	if gameID <= 0 || deliveryID == "" || len(deliveryID) > 255 || strings.ContainsRune(deliveryID, 0) {
		return 0, ErrInvalidRegistration
	}
	password, err := s.passwordGenerator.Generate()
	if err != nil {
		return 0, err
	}
	game, err := s.gamesRepo.GetGameByID(ctx, gameID)
	if err != nil {
		return 0, err
	}
	autoConfirm := !time.Now().Before(game.StartAt.Add(-24 * time.Hour))
	id, err := s.teamsRepo.RegisterTeamWithParticipation(ctx, gameID, deliveryID, registration.Name, registration.Email, registration.ParticipantsCount, password, autoConfirm)
	if err != nil {
		if errors.Is(err, teams_repo.ErrRegistrationClosed) {
			return 0, ErrRegistrationClosed
		}
		return 0, err
	}
	if s.emailSender == nil {
		return id, nil
	}
	team, err := s.teamsRepo.GetTeamByID(ctx, id)
	if err != nil {
		return 0, err
	}
	if err := s.emailSender.Send(ctx, registrationEmail(team, game.Name, s.appBaseURL)); err != nil {
		return 0, fmt.Errorf("email send: %w", err)
	}
	return id, nil
}

func registrationEmail(team *repos.Team, gameName, baseURL string) email_sender.Message {
	link := strings.TrimRight(baseURL, "/") + "/team-story/" + fmt.Sprint(team.ID) + "?password=" + team.Password
	body := fmt.Sprintf("Регистрация команды «%s» в игре «%s» прошла успешно.\n\nУчастников: %d\n\nСсылка для входа в игру: %s", team.Name, gameName, team.ParticipantsCount, link)
	content := fmt.Sprintf("<p class=\"email-text\">Регистрация команды <strong>%s</strong> в игре <strong>%s</strong> прошла успешно.</p><p class=\"email-text\">Участников: %d</p>%s", html.EscapeString(team.Name), html.EscapeString(gameName), team.ParticipantsCount, email_sender.ButtonHTML(link, "Войти в игру"))
	markup := email_sender.WrapHTML(content, strings.TrimRight(baseURL, "/"))
	return email_sender.Message{To: team.CaptainEmail, Subject: "Регистрация команды в «Вечернем детективе»", Body: body, HTML: markup}
}

func (s *TeamService) DownloadTeamQR(ctx context.Context, id int) ([]byte, string, error) {
	game, err := s.gamesRepo.GetGameByID(ctx, id)
	if err != nil {
		if errors.Is(err, games_repo.ErrGameNotFound) {
			return nil, "", ErrGameNotFound
		}
		return nil, "", err
	}
	teams, err := s.teamsRepo.GetActiveTeamsByGameID(ctx, id)
	if err != nil {
		return nil, "", err
	}
	if len(teams) == 0 {
		return nil, "", ErrNoCards
	}
	cards, filename, err := teamQRData(s.appBaseURL, game, teams)
	if err != nil {
		return nil, "", err
	}
	data, err := team_qr_pdf.Generate(cards)
	return data, filename, err
}