Códigos cortos de WordPress en el lugar equivocado

Estoy creando un complemento de wordpress y quiero usar códigos cortos para insertar un código bastante grande dentro de la publicación. Obtuve este código simple que simula mi problema.

function shortcode_fn( $attributes ) {
    wanted();
    return "unwanted";
}
add_shortcode( 'simplenote', 'shortcode_fn');

function wanted(){
    echo "wanted";
}

y publicar con este contenido

start
[simplenote]
end

que da este resultado:

wanted
start
unwanted
end

y quiero que inserte el texto “buscado” entre el inicio y el final. Sé que la solución más fácil sería simplemente devolver “buscado” en buscado(), pero ya tengo todas estas funciones y son bastante grandes. ¿Hay una solución fácil sin escribir todo desde cero?

@edit: ¿tal vez hay alguna forma de almacenar todos los ecos de la función en una cadena sin imprimirlos?

Una solución fácil es usar Control de salida funciones:

function shortcode_fn( $attributes ) {
    ob_start(); // start a buffer
    wanted(); // everything is echoed into a buffer
    $wanted = ob_get_clean(); // get the buffer contents and clean it
    return $wanted;
}

Siga este enlace http://codex.wordpress.org/Shortcode_API#Resumen

Cuando se muestra the_content, la API de shortcode analizará cualquier shortcode registrado como “[myshortcode]”, separe y analice los atributos y el contenido, si los hay, y páseles la función de controlador de shortcode correspondiente. Cualquier cadena devuelta (sin eco) por el controlador de shortcode se insertará en el cuerpo de la publicación en lugar del propio shortcode.

Por lo tanto, todas las funciones de shortcode deben devolverse, puede cambiar su función “quería” a esto:

function wanted(){
    return "wanted";
}

La forma estándar de WordPress sería proporcionar dos funciones. Una, get_wanted() haría devolver la cadena buscada; el otro, wanted()es solo:

function wanted() {
    echo get_wanted();
}

Entonces puede hacer cualquiera de las dos cosas, pero solo tiene la mayor parte del código en un solo lugar. Este es un patrón común en WordPress, por ejemplo the_title() versus get_the_title().

Como menciona @Ruslan, la forma más estándar sin reescribir su función existente para construir una cadena sería usar el control de salida.

Usando la respuesta de Ruslan Bes, el código se ve así:

function shortcode_fn( $attributes ) {
    ob_start(); // start a buffer
    wanted(); // everything is echoed into a buffer
    $wanted = ob_get_clean(); // get the buffer contents and clean it
    return $wanted;
}

add_shortcode( 'simplenote', 'shortcode_fn');

function wanted(){
    echo "wanted";
}