Обрамление видео с YouTube и Vime в определённые теги

Всем доброго времени суток!

Подскажите пожалуйста, как реализовать в рамках этой функции обрамление ссылок вида https://www.youtube.com/watch?v=pQqyX4h8I1I и https://www.vime.com/watch?v=pQqyX4h8I1I в теги <figure><video>видео</video></<figure>>

function get_the_content_dzen() {
	    
		$post = get_post();
		$content = $post->post_content;
		
		$content = str_replace(']]>', ']]>', $content);
		$pattern = '/(<img([^>]*)>)/i';
        $replacement = '<figure>$1</figure>';
        $content = preg_replace( $pattern, $replacement, $content );
		
		add_filter( 'layf_content_feed', 'wpautop' );

		return apply_filters('layf_content_feed', $content);		
}

---------- Добавлено 26.04.2017 в 23:25 ----------

Предполагаю, что это можно решить как-то так, но как прицепить этот код к функции понять пока что не могу...

if ((strpos($url, "youtu")) or (strpos($url, "vime"))) {
$return = '<div class="video-wrapper">'.$html.'</div>';
} else
$return = $content;