114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
package recovery
|
||
|
||
import (
|
||
"codeword/internal/models"
|
||
"codeword/internal/services"
|
||
"encoding/base64"
|
||
"fmt"
|
||
"github.com/gofiber/fiber/v2"
|
||
"go.uber.org/zap"
|
||
"time"
|
||
)
|
||
|
||
type RecoveryController struct {
|
||
logger *zap.Logger
|
||
service *services.RecoveryService
|
||
defaultURL string
|
||
}
|
||
|
||
func NewRecoveryController(logger *zap.Logger, service *services.RecoveryService, defaultRedirectionURL string) *RecoveryController {
|
||
return &RecoveryController{
|
||
logger: logger,
|
||
service: service,
|
||
defaultURL: defaultRedirectionURL,
|
||
}
|
||
}
|
||
|
||
func (r *RecoveryController) HandlePingDB(c *fiber.Ctx) error {
|
||
startTime := time.Now()
|
||
if err := r.service.Ping(c.Context()); err != nil {
|
||
r.logger.Error("Failed to ping the database", zap.Error(err))
|
||
return c.Status(fiber.StatusServiceUnavailable).SendString("DB ping failed")
|
||
}
|
||
duration := time.Since(startTime)
|
||
|
||
durationMillis := duration.Milliseconds()
|
||
responseMessage := fmt.Sprintf("DB ping success - Time taken: %d ms", durationMillis)
|
||
|
||
return c.Status(fiber.StatusOK).SendString(responseMessage)
|
||
}
|
||
|
||
// todo update docs
|
||
func (r *RecoveryController) HandleRecoveryRequest(c *fiber.Ctx) error {
|
||
var req models.RecoveryRequest
|
||
if err := c.BodyParser(&req); err != nil {
|
||
r.logger.Error("Failed to parse recovery request", zap.Error(err))
|
||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Bad Request"})
|
||
}
|
||
|
||
referralURL := c.Get("Referrer")
|
||
|
||
if req.RedirectionURL == "" && referralURL != "" {
|
||
req.RedirectionURL = referralURL
|
||
} else if req.RedirectionURL == "" {
|
||
req.RedirectionURL = r.defaultURL
|
||
}
|
||
|
||
user, err := r.service.FindUserByEmail(c.Context(), req.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"})
|
||
}
|
||
|
||
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"})
|
||
}
|
||
|
||
signUrl := req.RedirectionURL + base64.URLEncoding.EncodeToString(key)
|
||
sign := base64.URLEncoding.EncodeToString(key)
|
||
|
||
id, err := r.service.StoreRecoveryRecord(c.Context(), models.StoreRecDeps{UserID: user.ID.Hex(), Email: user.Email, Key: sign, Url: signUrl})
|
||
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"})
|
||
}
|
||
|
||
signWithID := sign + id // подпись с id записи
|
||
|
||
err = r.service.RecoveryEmailTask(c.Context(), models.RecEmailDeps{UserID: user.ID.Hex(), Email: req.Email, SignWithID: signWithID, ID: id})
|
||
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{
|
||
"id": id,
|
||
})
|
||
}
|
||
|
||
func (r *RecoveryController) HandleRecoveryLink(c *fiber.Ctx) error {
|
||
sign := c.Params("sign")
|
||
|
||
record, err := r.service.GetRecoveryRecord(c.Context(), sign)
|
||
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"})
|
||
}
|
||
|
||
if time.Since(record.CreatedAt) > 15*time.Minute {
|
||
r.logger.Error("Recovery link expired", zap.String("signature", sign))
|
||
return c.Status(fiber.StatusNotAcceptable).JSON(fiber.Map{"error": "Recovery link expired"})
|
||
}
|
||
|
||
tokens, err := r.service.ExchangeForTokens(record.UserID, record.Sign)
|
||
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)
|
||
}
|