route information

Author: davidmosterd

src/MapRESTtoMCP.php

@@ -2,10 +2,12 @@
 
 namespace WP_CLI\AiCommand;
 
+use WP_CLI\AiCommand\RouteInformation;
 use WP_CLI;
 use WP_CLI\AiCommand\MCP\Server;
 use WP_REST_Request;
 
+
 class MapRESTtoMCP {
 
     public function __construct(
@@ -96,12 +98,20 @@ public function map_rest_to_mcp( Server $mcp_server ) {
 					continue; // This route is the block list.
 				}
 
-
 				foreach( $endpoint['methods'] as $method_name => $enabled ) {
+					$information = new RouteInformation(
+						$route,
+						$method_name,
+						$endpoint['callback'],
+					);
+
+					if ( ! $information->is_wp_rest_controller() ) {
+						continue;
+					}
 
 					$tool = [
-						'name' => $this->generate_tool_name($route, $method_name),
-						'description' => $this->generate_description( $route, $method_name, $endpoint ),
+						'name' => $information->get_sanitized_route_name(),
+						'description' => $this->generate_description( $information ),
 						'inputSchema' => $this->args_to_schema( $endpoint['args'] ),
 						'callable' => function ( $inputs ) use ( $route, $method_name, $server ){
 							return $this->rest_callable( $inputs, $route, $method_name, $server );
@@ -113,6 +123,7 @@ public function map_rest_to_mcp( Server $mcp_server ) {
 
 			}
 		}
+
 	}
 
 	protected function generate_tool_name($route, $method_name) {
@@ -130,36 +141,23 @@ protected function generate_tool_name($route, $method_name) {
 	 * Get a list of posts             GET /wp/v2/posts
 	 * Get post with id                GET /wp/v2/posts/(?P<id>[\d]+)
 	 */
-	protected function generate_description( $route, $method_name, $endpoint ) {
+	protected function generate_description( RouteInformation $information  )  : string {
 
-		 // TODO all validation + exception handling.
-		$verb = array(
+		$verb = match ($information->get_method()) {
 			'GET' => 'Get',
 			'POST' => 'Create',
-			'PUT' => 'Update',
-			'PATCH' => 'Update',
+			'PUT', 'PATCH'  => 'Update',
 			'DELETE' => 'Delete',
-		);
+		};
 
-		$controller = $endpoint['callback'][0];
-		if ( !isset($endpoint['callback']) || ! \is_object($endpoint['callback'][0])) {
-			throw new \Exception('Not an object: ' . $route);
-		}
-		if (! \method_exists($endpoint['callback'][0], 'get_public_item_schema')) {
-			throw new \Exception('missing method: ' . $route);
-		}
-
-		$schema = $controller->get_public_item_schema();
+		$schema = $information->get_wp_rest_controller()->get_public_item_schema();
 		$title = $schema['title'];
 
-		// is singular?
-		$singular = 'a';
-		if ( $method_name === 'GET' && ! \str_contains( $route, '(?P<' )) {
-			$singular = 'List of';
-
-		}
+		$determiner = $information->is_singular()
+			? 'a'
+			: 'list of';
 
-		return $verb[ $method_name ] . ' ' . $singular . ' ' . $title;
+		return $verb . ' ' . $determiner . ' ' . $title;
 	}
 
 	protected function rest_callable( $inputs, $route, $method_name, $server ) {

src/RouteInformation.php

@@ -2,6 +2,14 @@
 
 declare(strict_types=1);
 
+namespace WP_CLI\AiCommand;
+
+use BadMethodCallException;
+use WP_REST_Controller;
+use WP_REST_Posts_Controller;
+use WP_REST_Taxonomies_Controller;
+use WP_REST_Users_Controller;
+
 class RouteInformation
 {
 
@@ -12,29 +20,42 @@ public function __construct(
 	) {
 	}
 
+	public function get_sanitized_route_name(): string
+	{
+		$route = $this->route;
+
+		preg_match_all('/\(?P<(\w+)>/', $this->route, $matches);
+
+		foreach ($matches[1] as $match) {
+			$route = preg_replace('/(\(\?P<' . $match . '>.*\))/', 'p_' . $match, $route, 1);
+		}
+
+		return $this->method . '_' . sanitize_title( $route );
+	}
+
 	public function get_method(): string
 	{
 		return $this->method;
 	}
 
-	public function is_post(): bool
+	public function is_create(): bool
 	{
 		return $this->method === 'POST';
 	}
 
-	public function is_put(): bool
+	public function is_update(): bool
 	{
-		return $this->method === 'PUT';
+		return $this->method === 'PUT' || $this->method === 'PATCH';
 	}
 
 	public function is_delete(): bool
 	{
 		return $this->method === 'DELETE';
 	}
 
-	public function is_put(): bool
+	public function is_get(): bool
 	{
-		return $this->method === 'PUT';
+		return $this->method === 'GET';
 	}
 
 	public function is_singular(): bool
@@ -57,7 +78,7 @@ public function is_list(): bool
 		return ! $this->is_singular();
 	}
 
-	public function is_wp_rest_controller()
+	public function is_wp_rest_controller(): bool
 	{
 		$allowed = [
 			WP_REST_Posts_Controller::class,
@@ -74,4 +95,13 @@ public function is_wp_rest_controller()
 		return false;
 	}
 
+	public function get_wp_rest_controller(): WP_REST_Controller
+	{
+		if ( ! $this->is_wp_rest_controller()) {
+			throw new BadMethodCallException('The callback needs to be a WP_Rest_Controller');
+		}
+
+		return $this->callback[0];
+	}
+
 }