toolcollection

Author: davidmosterd

ai-command.php

@@ -14,4 +14,11 @@
 	require_once $ai_command_autoloader;
 }
 
-WP_CLI::add_command( 'ai', AiCommand::class );
+WP_CLI::add_command( 'ai', function ( $args, $assoc_args ) {
+	$tools = new ToolCollection();
+
+	// TODO Register your tool here and add it to the collection
+
+	$ai_command = new AiCommand( $tools );
+	$ai_command( $args, $assoc_args );
+} );

src/AiCommand.php

@@ -23,6 +23,12 @@
  */
 class AiCommand extends WP_CLI_Command {
 
+	public function __construct(
+		private ToolCollection $tools,
+	) {
+		parent::__construct();
+	}
+
 	/**
 	 * Greets the world.
 	 *

src/Collection.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WP_CLI\AiCommand;
+
+use Countable;
+use Iterator;
+
+abstract class Collection implements Iterator, Countable
+{
+
+	protected array $data = [];
+
+	public function next(): void
+	{
+		next($this->data);
+	}
+
+	public function key(): int
+	{
+		return (int)key($this->data);
+	}
+
+	public function valid(): bool
+	{
+		return key($this->data) !== null;
+	}
+
+	public function rewind(): void
+	{
+		reset($this->data);
+	}
+
+	public function count(): int
+	{
+		return count($this->data);
+	}
+
+}

src/Entity/Tool.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+final class Tool
+{
+
+	public function __construct(
+		private array $data,
+		private array $tags = []
+	) {
+		$this->validate();
+	}
+
+	private function validate(): void
+	{
+		foreach ($this->tags as $tag) {
+			if ( ! preg_match('/^[a-z][a-z-]+$/', $tag)) {
+				throw new InvalidArgumentException('Tags can only contain [a-z] and -.');
+			}
+		}
+	}
+
+	public function get_name(): string
+	{
+		return $this->data['name'];
+	}
+
+	public function get_tags(): array
+	{
+		return $this->tags;
+	}
+
+	public function get_data(): array
+	{
+		return $this->data;
+	}
+
+}

src/ToolCollection.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WP_CLI\AiCommand;
+
+use Tool;
+
+final class ToolCollection extends Collection
+{
+
+	public function __construct(array $data = [])
+	{
+		foreach ($data as $tool) {
+			$this->add($tool);
+		}
+	}
+
+	public function add(Tool $tool): void
+	{
+		$this->data[] = $tool;
+	}
+
+	public function current(): Tool
+	{
+		return current($this->data);
+	}
+
+}

src/ToolRepository.php

@@ -0,0 +1,31 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WP_CLI\AiCommand;
+
+use Tool;
+
+interface ToolRepository
+{
+
+	public function find(string $name): ?Tool;
+
+	/**
+	 * Controls which tags are included or excluded from processing.
+	 *
+	 * Keys allowed:
+	 * - `include`: Defines which tags to include. Accepts:
+	 *   - `'all'` (default): includes all available tags.
+	 *   - an empty string or empty array: includes nothing.
+	 *   - a string of comma-separated tag names or an array of tag names: includes only those.
+	 *
+	 * - `exclude`: Removes tags from the resolved include list. Can be a string or array of tag names.
+	 *
+	 * Behavior:
+	 * - Inclusion is resolved first.
+	 * - Exclusion is applied afterward to filter out specific tags from the inclusion list.
+	 */
+	public function find_all(array $filters = []): ToolCollection;
+
+}

src/ToolRepository/CollectionToolRepository.php

@@ -0,0 +1,68 @@
+<?php
+
+declare(strict_types=1);
+
+use WP_CLI\AiCommand\ToolCollection;
+use WP_CLI\AiCommand\ToolRepository;
+
+class CollectionToolRepository implements ToolRepository
+{
+
+	public function __construct(
+		private ToolCollection $collection
+	) {
+	}
+
+	public function find(string $name): ?Tool
+	{
+		foreach ($this->find_all() as $tool) {
+			if ($tool->get_name() === $name) {
+				return $tool;
+			}
+		}
+
+		return null;
+	}
+
+	public function find_all(array $filters = []): ToolCollection
+	{
+		$defaults = [
+			'include' => 'all',
+			'exclude' => [],
+		];
+
+		$filters = array_merge($defaults, $filters);
+
+		$all = iterator_to_array($this->collection);
+		$filtered = [];
+
+		if ($filters['include'] !== 'all') {
+			foreach ($filters['include'] as $tag_to_include) {
+				foreach ($all as $tool) {
+					foreach ($tool->get_tags() as $tag_to_check) {
+						if ($tag_to_include === $tag_to_check) {
+							$filtered[$tool->get_name()] = $tool;
+
+							continue 2;
+						}
+					}
+				}
+			}
+		}
+
+		if ($filters['exclude']) {
+			foreach ($filters['exclude'] as $tag_to_exclude) {
+				foreach ($filtered as $tool) {
+					foreach ($tool->get_tags() as $tag_to_check) {
+						if ($tag_to_exclude === $tag_to_check) {
+							unset($filtered[$tool->get_name()]);
+						}
+					}
+				}
+			}
+		}
+
+		return new ToolCollection($filtered);
+	}
+
+}