WordPress 投稿時にアイキャッチ画像を自動設定

/**
 * 投稿の保存時にコンテンツ内の最初の画像をアイキャッチ画像として保存します。
 *
 * @param int $post_id
 * @param post $post
 * @filter save_post
 */
function save_post_featured_image_lite( $post_id, $post ) {

	if ( wp_is_post_revision( $post_id ) )
		return;

	// disable_featured_image カスタム フィールドが設定されている場合は、アイキャッチ画像を設定しない。
	if ( get_post_meta( $post_id, 'disable_featured_image', true ) )
		return;

	$attachment_id = get_post_meta( $post_id, '_thumbnail_id', true );
	$content = $post->post_content;

	if ( ! $attachment_id ) {
		$attachment_id = 0;
		$matches = array();
		preg_match_all( '/<\s*img .*?src\s*=\s*[\"|\'](.*?)[\"|\'].*?>/i', $content, $matches );
		if ( count( $matches ) ) {
			// wp-image-{$id} クラスより、ID を取得する。
			foreach ( $matches[0] as $key => $img ) {
				$class_matches = array();
				if ( preg_match( '/class\s*=\s*[\"|\'].*?wp-image-([0-9]*).*?[\"|\']/i', $img, $class_matches ) ) {
					$attachment_id = $class_matches[1];
					break;
				}
			}
			if ( $attachment_id ) {
				update_post_meta( $post_id, '_thumbnail_id', $attachment_id );
			}
		}
	}
}

add_action( 'save_post', 'save_post_featured_image_lite', 10, 2 );

/**
 * 添付ファイルの URL より ID を取得します。
 *
 * @global wpdb $wpdb
 * @param string $url 添付ファイルの URL。
 * @param bool $is_full_size オプション。フルサイズかどうかを示す値。フルサイズの場合は true、別サイズの場合は false。デフォルトは true。
 * @return int 成功した場合は ID（1 以上）を、存在しない場合は 0 を返します。
 */
function get_attachment_id_by_url( $url, $is_full_size = true ) {
	global $wpdb;

	$attachment_id = 0;
	$full_size_url = $url;

	if ( ! $is_full_size ) {
		// URL から、サイズ表記（-999x999）を除去して、フルサイズの URL を取得する。
		$full_size_url = preg_replace( '/(-[0-9]+x[0-9]+)(\.[^.]+){0,1}$/i', '${2}', $url );
		if ( $url === $full_size_url ) {
			// 別サイズではないので中止。
			return $attachment_id;
		}
	}

	$uploads = wp_upload_dir();
	$base_url = $uploads['baseurl'];
	if ( strpos( $full_size_url, $base_url ) === 0 ) {
		//$attached_file = ltrim( $full_size_url, $base_url . '/' );
		$attached_file = str_replace( $base_url . '/', '', $full_size_url );
		$attachment_id = $wpdb->get_var( $wpdb->prepare(
			"SELECT post_id FROM $wpdb->postmeta WHERE meta_key='_wp_attached_file' AND meta_value='%s' LIMIT 1;",
			$attached_file
		) );
	}

	return (int)$attachment_id;
}

/**
 * 投稿の保存時にコンテンツ内の最初の画像をアイキャッチ画像として保存します。
 *
 * @param int $post_id
 * @param post $post
 * @filter save_post
 */
function save_post_featured_image( $post_id, $post ) {

	if ( wp_is_post_revision( $post_id ) )
		return;

	// disable_featured_image カスタム フィールドが設定されている場合は、アイキャッチ画像を設定しない。
	if ( get_post_meta( $post_id, 'disable_featured_image', true ) )
		return;

	$attachment_id = get_post_meta( $post_id, '_thumbnail_id', true );
	$content = $post->post_content;

	if ( ! $attachment_id ) {
		$attachment_id = 0;
		$matches = array();
		preg_match_all( '/<\s*img .*?src\s*=\s*[\"|\'](.*?)[\"|\'].*?>/i', $content, $matches );
		if ( count( $matches ) ) {
			// wp-image-{$id} クラスより、ID を取得する。
			foreach ( $matches[0] as $key => $img ) {
				$class_matches = array();
				if ( preg_match( '/class\s*=\s*[\"|\'].*?wp-image-([0-9]*).*?[\"|\']/i', $img, $class_matches ) ) {
					$attachment_id = $class_matches[1];
					break;
				}
			}
			// URL より ID を取得する。
			if ( ! $attachment_id ) {
				foreach ( $matches[0] as $key => $img ) {
					$url = $matches[1][$key];
					$attachment_id = get_attachment_id_by_url( $url );
					if ( ! $attachment_id ) {
						$attachment_id = get_attachment_id_by_url( $url, false );
					}
					if ( $attachment_id ) {
						break;
					}
				}
			}
			if ( $attachment_id ) {
				update_post_meta( $post_id, '_thumbnail_id', $attachment_id );
			}
		}
	}
}

add_action( 'save_post', 'save_post_featured_image', 10, 2 );