Auto create index

Author: pdphilip

src/Commands/IndicesCommand.php

@@ -24,7 +24,7 @@ public function handle(): int
         $showAll = $this->option('all');
 
         $this->newLine();
-        $this->omni->roundedBox('Elasticsearch Indices', 'text-cyan-500', 'text-cyan-500');
+        $this->omni->titleBar('Elasticsearch Indices', 'teal');
         $this->newLine();
 
         try {

src/Commands/ShowCommand.php

@@ -25,7 +25,7 @@ public function handle(): int
         $connectionName = $this->option('connection');
 
         $this->newLine();
-        $this->omni->roundedBox('Index: '.$indexName, 'text-cyan-500', 'text-cyan-500');
+        $this->omni->titleBar('Index: '.$indexName, 'sky');
         $this->newLine();
 
         try {

src/Commands/StatusCommand.php

@@ -23,7 +23,7 @@ public function handle(): int
         $connectionName = $this->option('connection');
 
         $this->newLine();
-        $this->omni->roundedBox('Elasticsearch Status', 'text-cyan-500', 'text-cyan-500');
+        $this->omni->titleBar('Elasticsearch Status', 'cyan');
         $this->newLine();
 
         try {
@@ -58,8 +58,7 @@ public function handle(): int
         $this->omni->tableRow('Index Prefix', $prefix);
         $this->newLine();
 
-        $this->omni->newLoader('dots');
-        $result = $this->omni->runTask('Connecting to Elasticsearch', function () use ($connection) {
+        $result = $this->omni->task('Connecting to Elasticsearch', function () use ($connection) {
             try {
                 $info = $connection->getClientInfo();
 
@@ -84,11 +83,11 @@ public function handle(): int
             }
         });
 
-        if (empty($result['data'])) {
+        if (! $result || $result->isError()) {
             $this->newLine();
             $this->omni->statusError(
                 'CONNECTION FAILED',
-                $result['details'] ?? 'Unable to connect',
+                $result ? $result->details : 'Unable to connect',
                 [
                     'Check that Elasticsearch is running',
                     'Verify hosts, credentials, and auth_type in config/database.php',
@@ -99,8 +98,8 @@ public function handle(): int
             return self::FAILURE;
         }
 
-        $info = $result['data']['info'];
-        $license = $result['data']['license'];
+        $info = $result->data['info'];
+        $license = $result->data['license'];
 
         $this->newLine();
         $this->omni->tableHeader('Cluster', 'Value');

src/Connection.php

@@ -128,6 +128,7 @@ private function sanitizeConfig(): void
                     'meta_header' => null,
                     'default_limit' => null,
                     'allow_id_sort' => false,
+                    'auto_create_index' => true,
                 ],
             ],
             $this->config
@@ -711,12 +712,44 @@ protected function addClientParams(array $params): array
     protected function runQueryCallback($query, $bindings, Closure $callback)
     {
         try {
-            $result = $callback($query, $bindings);
+            return $callback($query, $bindings);
+        } catch (ClientResponseException $e) {
+            if ($this->shouldAutoCreateIndex($e, $query)) {
+                return $callback($query, $bindings);
+            }
+            throw new QueryException($e, $query);
         } catch (Exception $e) {
             throw new QueryException($e, $query);
         }
+    }
 
-        return $result;
+    private function shouldAutoCreateIndex(ClientResponseException $e, mixed $query): bool
+    {
+        if (! ($this->config['options']['auto_create_index'] ?? true)) {
+            return false;
+        }
+
+        if ($e->getCode() !== 404) {
+            return false;
+        }
+
+        $body = json_decode((string) $e->getResponse()->getBody(), true);
+        if (($body['error']['type'] ?? '') !== 'index_not_found_exception') {
+            return false;
+        }
+
+        $index = is_array($query) ? ($query['index'] ?? null) : null;
+        if (! $index) {
+            return false;
+        }
+
+        try {
+            $this->connection->createIndex($index, []);
+
+            return true;
+        } catch (Exception) {
+            return false;
+        }
     }
 
     /**

tests/CommandTest.php

@@ -2,6 +2,8 @@
 
 declare(strict_types=1);
 
+use PDPhilip\Elasticsearch\Tests\Models\User;
+
 it('runs elastic:status successfully', function () {
     $this->artisan('elastic:status')
         ->assertSuccessful();
@@ -19,7 +21,7 @@
 
 it('runs elastic:show on an existing index', function () {
     // Ensure the users index exists
-    \PDPhilip\Elasticsearch\Tests\Models\User::executeSchema();
+    User::executeSchema();
 
     $this->artisan('elastic:show', ['index' => 'users'])
         ->assertSuccessful();