package repositories

import (
	configs "astrology-api/configs"
	usermodel "astrology-api/models/usermodel"
)

func CreateOTP(otp *usermodel.UserOTP) error {
	return configs.DB.Create(otp).Error
}

func GetOTPByUserID(userID uint) (*usermodel.UserOTP, error) {

	var otp usermodel.UserOTP

	err := configs.DB.
		Where("user_id = ?", userID).
		First(&otp).Error

	if err != nil {
		return nil, err
	}

	return &otp, nil
}

func GetOTPByMobile(mobile string) (*usermodel.UserOTP, error) {

	var otp usermodel.UserOTP

	err := configs.DB.
		Where("mobile = ?", mobile).
		First(&otp).Error

	if err != nil {
		return nil, err
	}

	return &otp, nil
}

func UpdateOTP(otp *usermodel.UserOTP) error {
	return configs.DB.Save(otp).Error
}