codeword/internal/controller/recovery/recovery_controller.go

package controller

import (
	"codeword/internal/services"
	"github.com/gofiber/fiber/v2"
	"go.uber.org/zap"
	"time"
)

type RecoveryController struct {
	logger  *zap.Logger
	service *services.RecoveryService
}

func NewRecoveryController(logger *zap.Logger, service *services.RecoveryService) *RecoveryController {
	return &RecoveryController{
		logger:  logger,
		service: service,
	}
}

func (r *RecoveryController) HandlePingDB(c *fiber.Ctx) error {
	return r.service.Ping(c.Context())
}

// HandleRecoveryRequest обрабатывает запрос на восстановление пароля
func (r *RecoveryController) HandleRecoveryRequest(c *fiber.Ctx) error {
	email := c.FormValue("email")

	key, err := r.service.GenerateKey()
	if err != nil {
		r.logger.Error("Failed to generate key", zap.Error(err))
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	user, err := r.service.FindUserByEmail(c.Context(), email)
	if err != nil || user == nil {
		r.logger.Error("Failed to find user by email", zap.Error(err))
		return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
	}

	err = r.service.StoreRecoveryRecord(c.Context(), user.ID.Hex(), user.Email, key)
	if err != nil {
		r.logger.Error("Failed to store recovery record", zap.Error(err))
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	err = r.service.RecoveryEmailTask(c.Context(), user.ID.Hex(), email, key)
	if err != nil {
		r.logger.Error("Failed to send recovery email", zap.Error(err))
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	return c.Status(fiber.StatusOK).JSON(fiber.Map{"message": "Recovery email sent successfully"})
}

// HandleRecoveryLink обрабатывает ссылку восстановления и обменивает ее на токены
func (r *RecoveryController) HandleRecoveryLink(c *fiber.Ctx) error {
	key := c.Params("sign")
	// тут получается
	record, err := r.service.GetRecoveryRecord(c.Context(), key)
	if err != nil {
		r.logger.Error("Failed to get recovery record", zap.Error(err))
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	// проверка на более чем 15 минут
	if time.Since(record.CreatedAt) > 15*time.Minute {
		r.logger.Error("Recovery link expired", zap.String("signature", key))
		return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Recovery link expired"})
	}

	tokens, err := r.service.ExchangeForTokens(record.UserID)
	if err != nil {
		r.logger.Error("Failed to exchange recovery link for tokens", zap.Error(err))
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Internal Server Error"})
	}

	return c.Status(fiber.StatusOK).JSON(tokens)
}