add new method to check if table exists

effectra · Sep 27, 2023 · 21609ec · 21609ec
1 parent fbf31e0
commit 21609ec
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 6 deletions.
diff --git a/src/Build/InfoQueryBuilder.php b/src/Build/InfoQueryBuilder.php
@@ -33,13 +33,11 @@ public function __construct(protected array $attributes, protected Syntax $synta
      */
     public function dbName(): string
     {
-      return (string) match ($this->syntax->getDriver()) {
+        return (string) match ($this->syntax->getDriver()) {
             Driver::MySQL =>  $this->syntax->getCommand('select', 1) . $this->syntax->getCommand('database') . "()",
             Driver::PostgreSQL => $this->syntax->getCommand('select', 1) . 'current_database()',
             Driver::SQLite => $this->syntax->getCommand('pragma', 1) . 'database_list',
         };
-
-
     }
 
     /**
@@ -55,7 +53,7 @@ public function listDatabase(): string
             Driver::SQLite => '',
         };
 
-        if(empty($query)){
+        if (empty($query)) {
             throw new \Exception("Error Processing Query, driver '{$this->syntax->getDriver()}' doesn't has statement for list database tables");
         }
         return $query;
@@ -75,7 +73,7 @@ public function listTables(): string
         };
     }
 
-     /**
+    /**
      * Get the SQL query to list columns of a table.
      *
      * @return string The SQL query to list columns.
@@ -94,7 +92,27 @@ public function listCols(): string
         };
     }
 
-     /**
+    public function tableExists(): string
+    {
+        return (string) match ($this->syntax->getDriver()) {
+            Driver::MySQL => sprintf(
+                "%s %s %s '%s'",
+                $this->syntax->getCommand('show', 1),
+                $this->syntax->getCommand('tables', 1),
+                $this->syntax->getCommand('like', 1),
+                $this->getAttribute('table_name')
+            ),
+            Driver::PostgreSQL => sprintf(
+                '%s %s (%s)',
+                $this->syntax->getCommand('select', 1),
+                $this->syntax->getCommand('exists', 1),
+                (new Select('information_schema.tables'))->columns([1])->where(['table_name' => $this->getAttribute('table_name')]),
+            ),
+            Driver::SQLite => (new Select('sqlite_master'))->columns(['name'])->where(['type' => 'table', 'name' => $this->getAttribute('table_name')])
+        };
+    }
+
+    /**
      * Get the SQL query to retrieve the schema of a table.
      *
      * @return string The SQL query to retrieve table schema.
@@ -141,6 +159,7 @@ public function build(): string
             'list_cols' => $this->listCols(),
             'table_schema' => $this->tableSchema(),
             'table_indexes' => $this->tableIndexes(),
+            'table_exists' => $this->tableExists()
         };
     }
 

diff --git a/src/Operations/Info.php b/src/Operations/Info.php
@@ -62,6 +62,13 @@ public function listColumns(string $table_name): self
         return $this;
     }
 
+    public function tableExists(string $table_name): self
+    {
+        $this->setAttribute('table_name', $table_name);
+        $this->setAttribute('info', 'table_exists');
+        return $this;
+    }
+
     /**
      * Specify that the INFO query should retrieve the schema of a table.
      *