Add hasColumn method

Author: oleksandrb

src/Schema/Builder.php

@@ -5,6 +5,7 @@
 namespace MongoDB\Laravel\Schema;
 
 use Closure;
+use Illuminate\Support\Facades\DB;
 use MongoDB\Model\CollectionInfo;
 
 use function count;
@@ -13,15 +14,37 @@
 
 class Builder extends \Illuminate\Database\Schema\Builder
 {
-    /** @inheritdoc */
-    public function hasColumn($table, $column)
+    /**
+     * Check if column exists in the collection schema.
+     *
+     * @param $table
+     * @param $column
+     * @return bool
+     */
+    public function hasColumn($table, $column): bool
     {
-        return true;
+        $collection = $this->connection->table($table);
+
+        return $collection->where($column, 'exists', true)
+            ->project(['_id' => 1])
+            ->exists();
     }
 
-    /** @inheritdoc */
-    public function hasColumns($table, array $columns)
+    /**
+     * Check if columns exists in the collection schema.
+     *
+     * @param       $table
+     * @param array $columns
+     *
+     * @return bool
+     */
+    public function hasColumns($table, array $columns): bool
     {
+        foreach ($columns as $column) {
+            if (!$this->hasColumn($table, $column)) {
+                return false;
+            }
+        }
         return true;
     }
 

tests/SchemaTest.php

@@ -377,6 +377,23 @@ public function testRenameColumn(): void
         $this->assertSame($check[2]['column'], $check2[2]['column']);
     }
 
+    public function testHasColumn(): void
+    {
+        DB::connection()->collection('newcollection')->insert(['column1' => 'value']);
+
+        $this->assertTrue(Schema::hasColumn('newcollection', 'column1'));
+        $this->assertFalse(Schema::hasColumn('newcollection', 'column2'));
+    }
+
+    public function testHasColumns(): void
+    {
+        DB::connection()->collection('newcollection')->insert(['column1' => 'value']);
+        DB::connection()->collection('newcollection')->insert(['column2' => 'value']);
+
+        $this->assertTrue(Schema::hasColumns('newcollection', ['column1', 'column2']));
+        $this->assertFalse(Schema::hasColumns('newcollection', ['column1', 'column3']));
+    }
+
     protected function getIndex(string $collection, string $name)
     {
         $collection = DB::getCollection($collection);