wpmind/includes/Routing/RoutingHooks.php

<?php
/**
 * Routing Hooks - 路由钩子集成
 *
 * 将 IntelligentRouter 集成到 WordPress 过滤器系统
 *
 * @package WPMind
 * @since 3.2.0
 */

declare(strict_types=1);

namespace WPMind\Routing;

/**
 * 路由钩子类
 *
 * 负责将智能路由器连接到 wpmind_select_provider 过滤器
 */
class RoutingHooks {

	private static ?RoutingHooks $instance = null;

	/** @var bool 是否启用智能路由 */
	private bool $enabled = true;

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

	private function __construct() {
		$this->load_settings();
		$this->register_hooks();
	}

	/**
	 * 加载设置
	 */
	private function load_settings(): void {
		$settings      = get_option( 'wpmind_routing_settings', [] );
		$this->enabled = $settings['enabled'] ?? true;
	}

	/**
	 * 注册钩子
	 */
	private function register_hooks(): void {
		// 只有启用时才注册过滤器
		if ( $this->enabled ) {
			// 优先级 10，允许其他插件在之前或之后修改
			add_filter( 'wpmind_select_provider', [ $this, 'select_provider' ], 10, 2 );
		}

		// 注册设置变更钩子
		add_action( 'update_option_wpmind_routing_settings', [ $this, 'on_settings_update' ], 10, 2 );
	}

	/**
	 * 选择 Provider 过滤器回调
	 *
	 * @param string $provider 当前选择的 Provider
	 * @param string $context  请求上下文标识
	 * @return string 选择的 Provider ID
	 */
	public function select_provider( string $provider, string $context ): string {
		// 如果明确指定了 Provider（非 auto），尊重用户选择
		if ( $provider !== 'auto' && ! empty( $provider ) ) {
			// 但仍然检查该 Provider 是否可用
			$router         = IntelligentRouter::instance();
			$routingContext = $this->build_routing_context( $context, $provider );

			// 如果首选 Provider 可用，直接返回
			$selected = $router->route( $routingContext );
			if ( $selected === $provider ) {
				return $provider;
			}

			// 首选不可用时，记录日志并使用路由结果
			do_action( 'wpmind_routing_fallback', $provider, $selected, $context );
			return $selected ?? $provider;
		}

		// auto 模式：使用智能路由
		$router         = IntelligentRouter::instance();
		$routingContext = $this->build_routing_context( $context );

		$selected = $router->route( $routingContext );

		if ( $selected !== null ) {
			// 记录路由决策
			do_action( 'wpmind_routing_decision', $selected, $router->get_current_strategy(), $context );
			return $selected;
		}

		// 路由失败，返回默认 Provider
		return get_option( 'wpmind_default_provider', 'deepseek' );
	}

	/**
	 * 构建路由上下文
	 *
	 * @param string      $context           请求上下文标识
	 * @param string|null $preferredProvider 首选 Provider
	 * @return RoutingContext
	 */
	private function build_routing_context( string $context, ?string $preferredProvider = null ): RoutingContext {
		$routingContext = RoutingContext::create();

		// 设置首选 Provider
		if ( $preferredProvider !== null && $preferredProvider !== 'auto' ) {
			$routingContext->with_preferred_provider( $preferredProvider );
		}

		// 根据上下文设置模型类型
		$modelType = $this->infer_model_type( $context );
		if ( $modelType !== null ) {
			$routingContext->with_model_type( $modelType );
		}

		// 添加上下文元数据
		$routingContext->with_metadata( 'context', $context );
		$routingContext->with_metadata( 'timestamp', time() );

		return $routingContext;
	}

	/**
	 * 从上下文推断模型类型
	 *
	 * @param string $context 上下文标识
	 * @return string|null
	 */
	private function infer_model_type( string $context ): ?string {
		// 根据上下文关键词推断模型类型
		$contextLower = strtolower( $context );

		if ( str_contains( $contextLower, 'embed' ) || str_contains( $contextLower, 'vector' ) ) {
			return 'embedding';
		}

		if ( str_contains( $contextLower, 'image' ) || str_contains( $contextLower, 'vision' ) ) {
			return 'vision';
		}

		if ( str_contains( $contextLower, 'code' ) || str_contains( $contextLower, 'completion' ) ) {
			return 'completion';
		}

		// 默认为 chat
		return 'chat';
	}

	/**
	 * 设置更新回调
	 *
	 * @param mixed $old_value 旧值
	 * @param mixed $new_value 新值
	 */
	public function on_settings_update( $old_value, $new_value ): void {
		// 刷新路由器
		IntelligentRouter::instance()->refresh();

		// 更新启用状态
		$this->enabled = $new_value['enabled'] ?? true;
	}

	/**
	 * 检查智能路由是否启用
	 */
	public function is_enabled(): bool {
		return $this->enabled;
	}

	/**
	 * 启用智能路由
	 */
	public function enable(): void {
		$this->enabled       = true;
		$settings            = get_option( 'wpmind_routing_settings', [] );
		$settings['enabled'] = true;
		update_option( 'wpmind_routing_settings', $settings );

		// 重新注册过滤器
		if ( ! has_filter( 'wpmind_select_provider', [ $this, 'select_provider' ] ) ) {
			add_filter( 'wpmind_select_provider', [ $this, 'select_provider' ], 10, 2 );
		}
	}

	/**
	 * 禁用智能路由
	 */
	public function disable(): void {
		$this->enabled       = false;
		$settings            = get_option( 'wpmind_routing_settings', [] );
		$settings['enabled'] = false;
		update_option( 'wpmind_routing_settings', $settings );

		// 移除过滤器
		remove_filter( 'wpmind_select_provider', [ $this, 'select_provider' ], 10 );
	}
}