Allow disabling individual servers

Makes testing a bit easier

Author: swissspidy

src/AiCommand.php

@@ -148,19 +148,25 @@ public function get_sessions( bool $with_wp_server, bool $with_cli_server ): arr
 		$servers = array_values( ( new McpConfig() )->get_config() );
 
 		foreach ( $servers as $args ) {
-			if ( str_starts_with( $args, 'http://' ) || str_starts_with( $args, 'https://' ) ) {
+			if ( 'enabled' !== $args['status'] ) {
+				continue;
+			}
+
+			$server = $args['server'];
+
+			if ( str_starts_with( $server, 'http://' ) || str_starts_with( $server, 'https://' ) ) {
 				$sessions[] = ( new Client( new CliLogger() ) )->connect(
-					$args
+					$server
 				);
 				continue;
 			}
 
-			$args       = explode( ' ', $args );
-			$cmd_or_url = array_shift( $args );
+			$server     = explode( ' ', $server );
+			$cmd_or_url = array_shift( $server );
 
 			$sessions[] = ( new Client( new CliLogger() ) )->connect(
 				$cmd_or_url,
-				$args,
+				$server,
 			);
 		}
 

src/McpServerCommand.php

@@ -3,8 +3,10 @@
 namespace McpWp\AiCommand;
 
 use McpWp\AiCommand\Utils\McpConfig;
+use WP_CLI;
 use WP_CLI\Formatter;
 use WP_CLI_Command;
+use WP_CLI\Utils;
 
 /**
  * MCP Server command.
@@ -51,7 +53,8 @@ public function list_( $args, $assoc_args ): void {
 		foreach ( $config as $name => $server ) {
 			$servers[] = [
 				'name'   => $name,
-				'server' => $server,
+				'server' => $server['server'],
+				'status' => $server['status'],
 			];
 		}
 
@@ -86,26 +89,33 @@ public function add( $args ): void {
 		$config = $this->get_config()->get_config();
 
 		if ( isset( $config[ $args[0] ] ) ) {
-			\WP_CLI::error( 'Server already exists.' );
+			WP_CLI::error( 'Server already exists.' );
 		} else {
-			$config[ $args[0] ] = $args[1];
-			$result             = $this->get_config()->update_config( $config );
+			$config[ $args[0] ] = [
+				'server' => $args[1],
+				'status' => 'enabled',
+			];
+
+			$result = $this->get_config()->update_config( $config );
 
 			if ( ! $result ) {
-				\WP_CLI::error( 'Could not add server.' );
+				WP_CLI::error( 'Could not add server.' );
 			} else {
-				\WP_CLI::success( 'Server added.' );
+				WP_CLI::success( 'Server added.' );
 			}
 		}
 	}
 
 	/**
-	 * Remove a new MCP server from the list
+	 * Remove one or more MCP servers.
 	 *
 	 * ## OPTIONS
 	 *
-	 * <name>
-	 * : Name of the server to remove
+	 * [<name>...]
+	 * : One or more servers to remove
+	 *
+	 * [--all]
+	 * : Whether to remove all servers.
 	 *
 	 * ## EXAMPLES
 	 *
@@ -115,21 +125,81 @@ public function add( $args ): void {
 	 *
 	 * @param array $args Indexed array of positional arguments.
 	 */
-	public function remove( $args ): void {
+	public function remove( $args, $assoc_args ): void {
+		$all = Utils\get_flag_value( $assoc_args, 'all', false );
+
+		if ( ! $all && empty( $args ) ) {
+			WP_CLI::error( 'Please specify one or more servers, or use --all.' );
+		}
+
 		$config = $this->get_config()->get_config();
 
-		if ( ! array_key_exists( $args[0], $config ) ) {
-			\WP_CLI::error( 'Server not found.' );
-		} else {
-			unset( $config[ $args[0] ] );
-			$result = $this->get_config()->update_config( $config );
+		$successes = 0;
+		$errors    = 0;
+		$count     = count( $args );
 
-			if ( ! $result ) {
-				\WP_CLI::error( 'Could not remove server.' );
+		foreach ( $args as $server ) {
+			if ( ! array_key_exists( $server, $config ) ) {
+				WP_CLI::warning( "Server '$server' not found." );
+				++$errors;
 			} else {
-				\WP_CLI::success( 'Server removed.' );
+				unset( $config[ $args[0] ] );
+				++$successes;
+			}
+		}
+
+		$result = $this->get_config()->update_config( $config );
+
+		if ( ! $result ) {
+			$successes = 0;
+			$errors    = $count;
+		}
+
+		Utils\report_batch_operation_results( 'server', 'remove', $count, $successes, $errors );
+	}
+
+	/**
+	 * Update an MCP server.
+	 *
+	 * ## OPTIONS
+	 *
+	 * <name>
+	 * : Name of the server.
+	 *
+	 * --<field>=<value>
+	 * : One or more fields to update.
+	 *
+	 * ## EXAMPLES
+	 *
+	 *     # Remove server.
+	 *     $ wp mcp server update "server-filesystem" --status=disabled
+	 *     Success: Server updated.
+	 *
+	 * @param array $args Indexed array of positional arguments.
+	 */
+	public function update( $args, $assoc_args ): void {
+		$config = $this->get_config()->get_config();
+
+		if ( ! isset( $config[ $args[0] ] ) ) {
+			WP_CLI::error( "Server '$args[0]' not found." );
+		}
+
+		foreach ( $config[ $args[0] ] as $key => $value ) {
+			if ( isset( $assoc_args[ $key ] ) ) {
+				if ( 'status' === $key ) {
+					$value = 'disabled' === $value ? 'enabled' : 'disabled';
+				}
+				$config[ $args[0] ][ $key ] = $value;
 			}
 		}
+
+		$result = $this->get_config()->update_config( $config );
+
+		if ( ! $result ) {
+			WP_CLI::error( 'Could not update server.' );
+		} else {
+			WP_CLI::success( 'Server updated.' );
+		}
 	}
 
 	/**
@@ -144,6 +214,7 @@ protected function get_formatter( &$assoc_args ) {
 			[
 				'name',
 				'server',
+				'status',
 			]
 		);
 	}