WooCommerce: login e registrazione separati con shortcode personalizzati

// Shortcode per solo il login WooCommerce
add_shortcode('wc_login_form', function () {
    if (is_user_logged_in()) {
        $logout_url = wp_logout_url(home_url()); // Oppure dove vuoi reindirizzare l'utente
        ob_start();
        ?>
        <p>Sei già loggato.</p>
        <p><a href="<?php echo esc_url($logout_url); ?>" class="button woocommerce-Button">Logout</a></p>
        <?php
        return ob_get_clean();
    }
    ob_start();
    woocommerce_login_form([
        'redirect' => home_url(), // Cambia questo URL se vuoi reindirizzare altrove
        'hidden' => false,
    ]);
    return ob_get_clean();
});

add_shortcode('wc_register_form', function () {
    if (is_user_logged_in()) {
        return '<p>Sei già registrato e loggato.</p>';
    }
    if ('yes' !== get_option('woocommerce_enable_myaccount_registration')) {
        return '<p>La registrazione è disabilitata.</p>';
    }
    ob_start();
    ?>
    <form method="post" class="woocommerce-form woocommerce-form-register register">
        <?php do_action('woocommerce_register_form_start'); ?>

        <?php if ( 'no' === get_option( 'woocommerce_registration_generate_username' ) ) : ?>
            <p class="woocommerce-form-row woocommerce-form-row--wide form-row form-row-wide">
                <label for="reg_username"><?php esc_html_e( 'Username', 'woocommerce' ); ?> <span class="required">*</span></label>
                <input type="text" class="woocommerce-Input woocommerce-Input--text input-text" name="username" id="reg_username" autocomplete="username" value="<?php echo ( ! empty( $_POST['username'] ) ) ? esc_attr( wp_unslash( $_POST['username'] ) ) : ''; ?>" />
            </p>
        <?php endif; ?>

        <p class="woocommerce-form-row woocommerce-form-row--wide form-row form-row-wide">
            <label for="reg_email"><?php esc_html_e( 'Email address', 'woocommerce' ); ?> <span class="required">*</span></label>
            <input type="email" class="woocommerce-Input woocommerce-Input--text input-text" name="email" id="reg_email" autocomplete="email" value="<?php echo ( ! empty( $_POST['email'] ) ) ? esc_attr( wp_unslash( $_POST['email'] ) ) : ''; ?>" />
        </p>

        <?php if ( 'no' === get_option( 'woocommerce_registration_generate_password' ) ) : ?>
            <p class="woocommerce-form-row woocommerce-form-row--wide form-row form-row-wide">
                <label for="reg_password"><?php esc_html_e( 'Password', 'woocommerce' ); ?> <span class="required">*</span></label>
                <input type="password" class="woocommerce-Input woocommerce-Input--text input-text" name="password" id="reg_password" autocomplete="new-password" />
            </p>
        <?php else : ?>
            <p><?php esc_html_e( 'A link to set a new password will be sent to your email address.', 'woocommerce' ); ?></p>
        <?php endif; ?>

        <?php do_action('woocommerce_register_form'); ?>

        <p class="woocommerce-FormRow form-row">
            <?php wp_nonce_field('woocommerce-register', 'woocommerce-register-nonce'); ?>
            <button type="submit" class="woocommerce-Button button" name="register" value="<?php esc_attr_e('Register', 'woocommerce'); ?>"><?php esc_html_e('Register', 'woocommerce'); ?></button>
        </p>

        <?php do_action('woocommerce_register_form_end'); ?>
    </form>
    <?php
    return ob_get_clean();
});

add_action('wp', function() {
    // Slug delle pagine su cui caricare stili WooCommerce
    $custom_pages = ['login', 'registrazione']; // personalizza se usi slug diversi

    if ( is_page($custom_pages) ) {

        // Aggiungi le classi body necessarie per lo stile WooCommerce
        add_filter('body_class', function($classes) {
            $classes[] = 'woocommerce';
            $classes[] = 'woocommerce-page';
            return $classes;
        });

        // Forza il caricamento degli stili e script WooCommerce
        add_action('wp_enqueue_scripts', function() {
            if (class_exists('WooCommerce')) {
                wp_enqueue_style('woocommerce-general');
                wp_enqueue_style('woocommerce-layout');
                wp_enqueue_style('woocommerce-smallscreen');
                wp_enqueue_style('woocommerce-inline');

                wp_enqueue_script('wc-password-strength-meter');
                wp_enqueue_script('wc-country-select');
                wp_enqueue_script('woocommerce');
                wp_enqueue_script('wc-cart-fragments');
            }
        }, 20);
    }
});