WordPress | WordPressの投稿時にアイキャッチ画像を自動設定する方法（YouTube対応）

add_action('save_post', 'auto_set_featured_image_from_youtube');

add_action('save_post', 'auto_set_featured_image_from_youtube');

function auto_set_featured_image_from_youtube($post_id) {

    // 自動保存・リビジョン除外
    if (wp_is_post_autosave($post_id) || wp_is_post_revision($post_id)) {
        return;
    }

    // すでにアイキャッチがある場合は何もしない
    if (has_post_thumbnail($post_id)) {
        return;
    }

    $post = get_post($post_id);
    if (!$post) {
        return;
    }

    // 投稿本文
    $content = $post->post_content;

    // YouTube動画IDを取得
    $video_id = extract_youtube_video_id($content);
    if (!$video_id) {
        return;
    }

    // サムネイルURL（高画質）
    $thumbnail_url = "https://img.youtube.com/vi/{$video_id}/maxresdefault.jpg";

    // メディアに保存してアイキャッチ設定
    set_featured_image_from_url($thumbnail_url, $post_id);
}

function auto_set_featured_image_from_youtube($post_id) {

    // 自動保存・リビジョン除外
    if (wp_is_post_autosave($post_id) || wp_is_post_revision($post_id)) {
        return;
    }

    // すでにアイキャッチがある場合は何もしない
    if (has_post_thumbnail($post_id)) {
        return;
    }

    $post = get_post($post_id);
    if (!$post) {
        return;
    }

    // 投稿本文
    $content = $post->post_content;

    // YouTube動画IDを取得
    $video_id = extract_youtube_video_id($content);
    if (!$video_id) {
        return;
    }

    // サムネイルURL（高画質）
    $thumbnail_url = "https://img.youtube.com/vi/{$video_id}/maxresdefault.jpg";

    // メディアに保存してアイキャッチ設定
    set_featured_image_from_url($thumbnail_url, $post_id);
}

function extract_youtube_video_id($content) {

    $patterns = [
        '/youtube\.com\/watch\?v=([^\&\?\/]+)/',
        '/youtu\.be\/([^\&\?\/]+)/',
        '/youtube\.com\/embed\/([^\&\?\/]+)/'
    ];

    foreach ($patterns as $pattern) {
        if (preg_match($pattern, $content, $matches)) {
            return $matches[1];
        }
    }

    return false;
}

function extract_youtube_video_id($content) {

    $patterns = [
        '/youtube\.com\/watch\?v=([^\&\?\/]+)/',
        '/youtu\.be\/([^\&\?\/]+)/',
        '/youtube\.com\/embed\/([^\&\?\/]+)/'
    ];

    foreach ($patterns as $pattern) {
        if (preg_match($pattern, $content, $matches)) {
            return $matches[1];
        }
    }

    return false;
}

function set_featured_image_from_url($image_url, $post_id) {

    require_once ABSPATH . 'wp-admin/includes/file.php';
    require_once ABSPATH . 'wp-admin/includes/media.php';
    require_once ABSPATH . 'wp-admin/includes/image.php';

    // 画像を一時保存
    $tmp = download_url($image_url);
    if (is_wp_error($tmp)) {
        return;
    }

    $file = [
        'name'     => basename($image_url),
        'type'     => mime_content_type($tmp),
        'tmp_name' => $tmp,
        'size'     => filesize($tmp),
    ];

    // メディアに登録
    $attachment_id = media_handle_sideload($file, $post_id);

    // エラー時は削除
    if (is_wp_error($attachment_id)) {
        @unlink($tmp);
        return;
    }

    // アイキャッチ設定
    set_post_thumbnail($post_id, $attachment_id);
}

function set_featured_image_from_url($image_url, $post_id) {

    require_once ABSPATH . 'wp-admin/includes/file.php';
    require_once ABSPATH . 'wp-admin/includes/media.php';
    require_once ABSPATH . 'wp-admin/includes/image.php';

    // 画像を一時保存
    $tmp = download_url($image_url);
    if (is_wp_error($tmp)) {
        return;
    }

    $file = [
        'name'     => basename($image_url),
        'type'     => mime_content_type($tmp),
        'tmp_name' => $tmp,
        'size'     => filesize($tmp),
    ];

    // メディアに登録
    $attachment_id = media_handle_sideload($file, $post_id);

    // エラー時は削除
    if (is_wp_error($attachment_id)) {
        @unlink($tmp);
        return;
    }

    // アイキャッチ設定
    set_post_thumbnail($post_id, $attachment_id);
}

$thumbnail_url = "https://img.youtube.com/vi/{$video_id}/maxresdefault.jpg";

// 存在しない場合は hqdefault
$response = wp_remote_head($thumbnail_url);
if (is_wp_error($response) || wp_remote_retrieve_response_code($response) !== 200) {
    $thumbnail_url = "https://img.youtube.com/vi/{$video_id}/hqdefault.jpg";
}

$thumbnail_url = "https://img.youtube.com/vi/{$video_id}/maxresdefault.jpg";

// 存在しない場合は hqdefault
$response = wp_remote_head($thumbnail_url);
if (is_wp_error($response) || wp_remote_retrieve_response_code($response) !== 200) {
    $thumbnail_url = "https://img.youtube.com/vi/{$video_id}/hqdefault.jpg";
}

if ($post->post_type !== 'post') {
    return;
}

if ($post->post_type !== 'post') {
    return;
}