// Tạo mã OTP và gửi qua email
function send_otp_to_user($user_email) {
    $otp = rand(100000, 999999);  // Tạo mã OTP ngẫu nhiên 6 chữ số
    update_user_meta(get_current_user_id(), 'user_otp', $otp);  // Lưu OTP vào user_meta

    // Nội dung email gửi OTP
    $subject = 'Mã OTP xác thực của bạn';
    $message = 'Mã OTP của bạn là: ' . $otp;

    wp_mail($user_email, $subject, $message);  // Gửi email
}

// Kiểm tra mã OTP
function check_user_otp($user_login, $user) {
    if (isset($_POST['otp'])) {
        $input_otp = $_POST['otp'];
        $stored_otp = get_user_meta($user->ID, 'user_otp', true);

        if ($input_otp != $stored_otp) {
            wp_logout();  // Nếu mã OTP sai, logout user
            wp_redirect(home_url());  // Quay lại trang chủ
            exit;
        }
    }
}
add_action('wp_login', 'check_user_otp', 10, 2);

// Thêm form nhập OTP vào trang đăng nhập
function otp_input_form() {
    if (!is_user_logged_in()) return;  // Chỉ hiển thị cho người dùng đã đăng nhập

    echo '<div class="otp-form">';
    echo '<form action="' . esc_url($_SERVER['REQUEST_URI']) . '" method="POST">';
    echo '<label for="otp">Nhập mã OTP đã gửi tới email của bạn:</label>';
    echo '<input type="text" name="otp" id="otp" required />';
    echo '<input type="submit" value="Xác nhận" />';
    echo '</form>';
    echo '</div>';
}
add_action('login_form', 'otp_input_form');

// Gửi mã OTP khi đăng nhập thành công
function send_otp_on_login($user_login, $user) {
    $user_email = $user->user_email;
    send_otp_to_user($user_email);  // Gửi mã OTP đến email
}
add_action('wp_login', 'send_otp_on_login', 10, 2);

// Kiểm tra mã OTP khi người dùng nhập
function check_user_otp($user_login, $user) {
    if (isset($_POST['otp'])) {
        $input_otp = $_POST['otp'];
        $stored_otp = get_user_meta($user->ID, 'user_otp', true);

        if ($input_otp != $stored_otp) {
            wp_logout();  // Nếu mã OTP sai, logout user
            wp_redirect(home_url());  // Quay lại trang chủ
            exit;
        } else {
            delete_user_meta($user->ID, 'user_otp');  // Xóa mã OTP sau khi xác thực thành công
        }
    }
}