WP_AI_Client_Ability_Function_Resolver

25class WP_AI_Client_Ability_Function_Resolver {26 27	/**28	 * Prefix used to identify ability function calls.29	 *30	 * @since 7.0.031	 * @var string32	 */33	private const ABILITY_PREFIX = 'wpab__';34 35	/**36	 * Map of allowed ability names for this instance.37	 *38	 * Keys are ability name strings, values are `true` for O(1) lookup.39	 *40	 * @since 7.0.041	 * @var array<string, true>42	 */43	private array $allowed_abilities;44 45	/**46	 * Constructor.47	 *48	 * @since 7.0.049	 *50	 * @param WP_Ability|string ...$abilities The abilities that this resolver is allowed to execute.51	 */52	public function __construct( ...$abilities ) {53		$this->allowed_abilities = array();54 55		foreach ( $abilities as $ability ) {56			if ( $ability instanceof WP_Ability ) {57				$this->allowed_abilities[ $ability->get_name() ] = true;58			} elseif ( is_string( $ability ) ) {59				$this->allowed_abilities[ $ability ] = true;60			}61		}62	}63 64	/**65	 * Checks if a function call is an ability call.66	 *67	 * @since 7.0.068	 *69	 * @param FunctionCall $call The function call to check.70	 * @return bool True if the function call is an ability call, false otherwise.71	 */72	public function is_ability_call( FunctionCall $call ): bool {73		$name = $call->getName();74		if ( null === $name ) {75			return false;76		}77 78		return str_starts_with( $name, self::ABILITY_PREFIX );79	}80 81	/**82	 * Executes a WordPress ability from a function call.83	 *84	 * Only abilities that were specified in the constructor are allowed to be85	 * executed. If the ability is not in the allowed list, an error response86	 * with code `ability_not_allowed` is returned.87	 *88	 * @since 7.0.089	 *90	 * @param FunctionCall $call The function call to execute.91	 * @return FunctionResponse The response from executing the ability.92	 */93	public function execute_ability( FunctionCall $call ): FunctionResponse {94		$function_name = $call->getName() ?? 'unknown';95		$function_id   = $call->getId() ?? 'unknown';96 97		if ( ! $this->is_ability_call( $call ) ) {98			return new FunctionResponse(99				$function_id,100				$function_name,101				array(102					'error' => __( 'Not an ability function call' ),103					'code'  => 'invalid_ability_call',104				)