namespace App\Repositories;

use PDO;

class TwoFactorAuthRepository extends BaseRepository
{
    protected $table_name = "two_factor_auth";

    public function __construct(PDO $db)
    {
        parent::__construct($db);
    }

    public function findByUserId($userId)
    {
        $query = "SELECT * FROM " . $this->table_name . " WHERE user_id = ?";
        $stmt = $this->db->prepare($query);
        $stmt->bindParam(1, $userId);
        $stmt->execute();
        return $stmt->fetch(PDO::FETCH_ASSOC);
    }

    public function update2FAStatus($userId, $status)
    {
        $query = "UPDATE " . $this->table_name . " SET is_enabled = ? WHERE user_id = ?";
        $stmt = $this->db->prepare($query);
        $stmt->bindParam(1, $status, PDO::PARAM_BOOL);
        $stmt->bindParam(2, $userId);
        return $stmt->execute();
    }
}