WordPress在发布或更新文章设置内容中的第一张图片为缩略图

// 自动将文章内容中的第一张图片设置为缩略图
function auto_set_featured_image($post_id) {
    // 检查是否为自动保存
    if (defined('DOING_AUTOSAVE') && DOING_AUTOSAVE) {
        return;
    }
    
    // 检查用户权限
    if (!current_user_can('edit_post', $post_id)) {
        return;
    }
    
    // 如果已经设置了缩略图，则不执行
    if (has_post_thumbnail($post_id)) {
        return;
    }
    
    // 获取文章内容
    $post_content = get_post_field('post_content', $post_id);
    
    // 从内容中提取第一张图片
    preg_match('/<img[^>]+src=[\'"]([^\'"]+)[\'"]/i', $post_content, $matches);
    
    if (!empty($matches[1])) {
        $image_url = $matches[1];
        
        // 处理相对路径
        if (strpos($image_url, site_url()) === false) {
            $image_url = site_url() . $image_url;
        }
        
        // 尝试获取图片附件ID
        $attachment_id = attachment_url_to_postid($image_url);
        
        // 如果图片已存在于媒体库中
        if ($attachment_id) {
            set_post_thumbnail($post_id, $attachment_id);
        } else {
            // 如果图片不在媒体库中，尝试下载并添加
            require_once(ABSPATH . 'wp-admin/includes/media.php');
            require_once(ABSPATH . 'wp-admin/includes/file.php');
            require_once(ABSPATH . 'wp-admin/includes/image.php');
            
            // 下载图片并添加到媒体库
            $attachment_id = media_sideload_image($image_url, $post_id, '', 'id');
            
            // 如果成功添加，则设置为缩略图
            if (!is_wp_error($attachment_id)) {
                set_post_thumbnail($post_id, $attachment_id);
            }
        }
    }
}
add_action('save_post', 'auto_set_featured_image');

// 只对post类型生效
if (get_post_type($post_id) !== 'post') {
    return;
}