/**
 * Shortcode para exibir informações de assinaturas personalizadas com containers <div> e classes CSS.
 *
 * Exemplo de uso: [custom_membership_info]
 */
function wp24h_membership_info_shortcode() {
    // Verifica se o usuário está logado
    if ( ! is_user_logged_in() ) {
        return '<div class="custom-membership-info">Você precisa estar logado para ver esta informação.</div>';
    }

    // Obtém as informações do usuário
    $user_id = get_current_user_id();

    // Verifica se o usuário tem uma associação ativa
    if ( function_exists( 'pmpro_getMembershipLevelsForUser' ) ) {
        $mylevels = pmpro_getMembershipLevelsForUser( $user_id );

        if ( ! empty( $mylevels ) ) {
            // Loop personalizado para exibir as informações de associação
            ob_start();
            echo '<div class="custom-membership-info">';
            echo '<h3>Suas Assinaturas:</h3>';
            foreach ( $mylevels as $level ) {
                echo '<div class="membership-level">';
                echo '<div class="membership-field membership-name"><strong>Nível de Assinatura:</strong> ' . esc_html( $level->name ) . '</div>';
                echo '<div class="membership-field membership-description"><strong>Descrição:</strong> ' . esc_html( $level->description ) . '</div>';
                echo '<div class="membership-field membership-startdate"><strong>Início:</strong> ' . esc_html( date( 'd/m/Y', $level->startdate ) ) . '</div>';
                echo '<div class="membership-field membership-enddate"><strong>Expira em:</strong> ' . ( $level->enddate ? esc_html( date( 'd/m/Y', $level->enddate ) ) : 'Assinatura vitalícia' ) . '</div>';
                echo '</div>';
            }
            echo '</div>';
            return ob_get_clean();
        } else {
            return '<div class="custom-membership-info">Você não possui uma assinatura ativa.</div>';
        }
    } else {
        return '<div class="custom-membership-info">O plugin Paid Memberships Pro não está ativo.</div>';
    }
}
add_shortcode( 'custom_membership_info', 'wp24h_membership_info_shortcode' );