Вставка изображений по стандартам HTML5

Нашел вот такое решение, которое перезаписывает процесс вставки изображений согласно HTML5

function html5_insert_image($html, $id, $caption, $title, $align, $url, $size, $alt) {
  $src  = wp_get_attachment_image_src( $id, $size, false );
  $html5 = "<figure class='image width-$src[1] align-$align'>";
  $html5 .= "<img src='$src[0]' alt='$alt' />";
  if ($caption) {
    $html5 .= "<figcaption>$caption</figcaption>";
  }
  $html5 .= "</figure>";
  return $html5;
}
add_filter( 'image_send_to_editor', 'html5_insert_image', 10, 9 );

но проблема в том, что при этом вставляется сама картинка, игнорируется ссылка на большое изображение/атамчмент пейдж. Знает ли кто-тто, как можно допилить данную схему, чтобы выполнялась проверка, какую опцию вибрал пользователь и, если он выбрал обтекание картинки ссылкой, то чтобы она тоже добавлялась?

Спасибо