wpmind/includes/Providers/Image/ImageRouter.php

<?php
/**
 * 图像生成路由器
 *
 * 管理多个图像生成 Provider，提供智能路由
 *
 * @package WPMind
 * @since 2.4.0
 */

declare(strict_types=1);

namespace WPMind\Providers\Image;

defined( 'ABSPATH' ) || exit;

/**
 * 图像生成路由器
 */
class ImageRouter {

	/**
	 * 单例实例
	 *
	 * @var ImageRouter|null
	 */
	private static ?ImageRouter $instance = null;

	/**
	 * 已注册的 Provider
	 *
	 * @var array<string, ImageProviderInterface>
	 */
	private array $providers = [];

	/**
	 * Provider 配置
	 *
	 * @var array
	 */
	private array $config = [];

	/**
	 * 获取单例实例
	 *
	 * @return ImageRouter
	 */
	public static function instance(): ImageRouter {
		if ( null === self::$instance ) {
			self::$instance = new self();
		}
		return self::$instance;
	}

	/**
	 * 私有构造函数
	 */
	private function __construct() {
		$this->config = get_option( 'wpmind_image_endpoints', [] );
		$this->initProviders();
	}

	/**
	 * 初始化已启用的 Provider
	 */
	private function initProviders(): void {
		$provider_classes = [
			'openai_gpt_image'    => OpenAIImageProvider::class,
			'google_gemini_image' => GeminiImageProvider::class,
			'tencent_hunyuan'     => HunyuanImageProvider::class,
			'bytedance_reve'      => ReveImageProvider::class,
			'flux'                => FluxImageProvider::class,
			'bytedance_seedream'  => SeedreamImageProvider::class,
			'midjourney'          => MidjourneyImageProvider::class,
			'qwen_image'          => QwenImageProvider::class,
		];

		foreach ( $this->config as $key => $settings ) {
			if ( empty( $settings['enabled'] ) || empty( $settings['api_key'] ) ) {
				continue;
			}

			if ( ! isset( $provider_classes[ $key ] ) ) {
				continue;
			}

			$class = $provider_classes[ $key ];

			// 检查类是否存在
			if ( ! class_exists( $class ) ) {
				continue;
			}

			$custom_url              = $settings['custom_base_url'] ?? '';
			$this->providers[ $key ] = new $class( $settings['api_key'], $custom_url );
		}
	}

	/**
	 * 生成图像
	 *
	 * @param string $prompt 提示词
	 * @param array  $options 选项
	 * @return array
	 */
	public function generate( string $prompt, array $options = [] ): array {
		$provider_id = $options['provider'] ?? $this->getDefaultProvider();

		if ( ! isset( $this->providers[ $provider_id ] ) ) {
			// 尝试使用任何可用的 Provider
			$provider_id = array_key_first( $this->providers );
		}

		if ( ! $provider_id || ! isset( $this->providers[ $provider_id ] ) ) {
			return [
				'success' => false,
				'error'   => '没有可用的图像生成服务',
			];
		}

		$provider = $this->providers[ $provider_id ];

		try {
			$result = $provider->generate( $prompt, $options );

			// 记录用量
			if ( $result['success'] ) {
				$this->trackUsage( $provider_id, $prompt, $options );
			}

			return $result;
		} catch ( \Exception $e ) {
			return [
				'success' => false,
				'error'   => $e->getMessage(),
			];
		}
	}

	/**
	 * 获取默认 Provider
	 *
	 * @return string
	 */
	private function getDefaultProvider(): string {
		$default = get_option( 'wpmind_default_image_provider', '' );

		if ( ! empty( $default ) && isset( $this->providers[ $default ] ) ) {
			return $default;
		}

		return array_key_first( $this->providers ) ?? '';
	}

	/**
	 * 获取所有已启用的 Provider
	 *
	 * @return array
	 */
	public function getEnabledProviders(): array {
		return array_keys( $this->providers );
	}

	/**
	 * 检查是否有可用的 Provider
	 *
	 * @return bool
	 */
	public function hasAvailableProvider(): bool {
		return ! empty( $this->providers );
	}

	/**
	 * 测试 Provider 连接
	 *
	 * @param string $provider_id Provider ID
	 * @return array
	 */
	public function testConnection( string $provider_id ): array {
		if ( ! isset( $this->providers[ $provider_id ] ) ) {
			return [
				'success' => false,
				'message' => 'Provider 未找到或未启用',
			];
		}

		return $this->providers[ $provider_id ]->testConnection();
	}

	/**
	 * 记录用量
	 *
	 * @param string $provider_id Provider ID
	 * @param string $prompt 提示词
	 * @param array  $options 选项
	 */
	private function trackUsage( string $provider_id, string $prompt, array $options ): void {
		// TODO: 实现图像生成用量追踪
		do_action( 'wpmind_image_generated', $provider_id, $prompt, $options );
	}
}