wpmind/includes/Providers/Image/QwenImageProvider.php

<?php
/**
 * 通义万相图像生成 Provider
 *
 * @package WPMind
 * @since 2.4.0
 */

declare(strict_types=1);

namespace WPMind\Providers\Image;

defined( 'ABSPATH' ) || exit;

/**
 * 通义万相 Provider
 */
class QwenImageProvider extends AbstractImageProvider {

    /**
     * {@inheritdoc}
     */
    protected string $id = 'qwen_image';

    /**
     * {@inheritdoc}
     */
    protected string $name = '通义万相';

    /**
     * {@inheritdoc}
     */
    protected string $base_url = 'https://dashscope.aliyuncs.com/api/v1/';

    /**
     * {@inheritdoc}
     */
    protected array $models = [ 'wanx-v1', 'wanx2.1-t2i-turbo' ];

    /**
     * {@inheritdoc}
     */
    protected function getHeaders(): array {
        return [
            'Content-Type'  => 'application/json',
            'Authorization' => 'Bearer ' . $this->api_key,
            'X-DashScope-Async' => 'enable', // 异步模式
        ];
    }

    /**
     * {@inheritdoc}
     */
    public function generate( string $prompt, array $options = [] ): array {
        $model = $options['model'] ?? 'wanx-v1';
        $size  = $options['size'] ?? '1024*1024';
        $n     = $options['n'] ?? 1;

        // 创建任务
        $response = $this->request( 'services/aigc/text2image/image-synthesis', [
            'model' => $model,
            'input' => [
                'prompt' => $prompt,
            ],
            'parameters' => [
                'size' => $size,
                'n'    => $n,
            ],
        ] );

        if ( is_wp_error( $response ) ) {
            return [
                'success' => false,
                'error'   => $response->get_error_message(),
            ];
        }

        // 获取任务 ID
        $task_id = $response['output']['task_id'] ?? null;
        if ( ! $task_id ) {
            return [
                'success' => false,
                'error'   => '无法获取任务 ID',
            ];
        }

        // 轮询任务状态
        $max_attempts = 60;
        $attempt = 0;
        
        while ( $attempt < $max_attempts ) {
            sleep( 2 );
            $attempt++;
            
            $status_response = $this->request( 'tasks/' . $task_id, [], 'GET' );
            
            if ( is_wp_error( $status_response ) ) {
                continue;
            }
            
            $status = $status_response['output']['task_status'] ?? '';
            
            if ( $status === 'SUCCEEDED' ) {
                $results = $status_response['output']['results'] ?? [];
                $urls = array_column( $results, 'url' );
                
                return [
                    'success' => true,
                    'urls'    => $urls,
                    'url'     => $urls[0] ?? '',
                ];
            }
            
            if ( $status === 'FAILED' ) {
                return [
                    'success' => false,
                    'error'   => $status_response['output']['message'] ?? '生成失败',
                ];
            }
        }

        return [
            'success' => false,
            'error'   => '任务超时',
        ];
    }

    /**
     * {@inheritdoc}
     */
    public function testConnection(): array {
        // 简单的模型列表请求来测试连接
        $response = wp_remote_get( $this->base_url . 'models', [
            'timeout' => 30,
            'headers' => [
                'Authorization' => 'Bearer ' . $this->api_key,
            ],
        ] );

        if ( is_wp_error( $response ) ) {
            return [
                'success' => false,
                'message' => $response->get_error_message(),
            ];
        }

        $status_code = wp_remote_retrieve_response_code( $response );
        
        if ( $status_code === 200 || $status_code === 401 ) {
            // 401 也说明连接成功，只是认证失败
            if ( $status_code === 401 ) {
                return [
                    'success' => false,
                    'message' => 'API Key 无效',
                ];
            }
            
            return [
                'success' => true,
                'message' => '连接成功',
            ];
        }

        return [
            'success' => false,
            'message' => '连接失败 (HTTP ' . $status_code . ')',
        ];
    }
}