Merge pull request #109 from 5am-code/feature/add-database-attributes

add missing database attributes and introduce traits for general attributes

Author: johguentner

src/Entities/Blocks/Block.php

@@ -59,13 +59,10 @@ protected function setResponseData(array $responseData): void
 
     protected function fillFromRaw(): void
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillType();
         $this->fillRawContent();
         $this->fillHasChildren();
-        $this->fillParentAttributes();
-        $this->fillArchivedAttributes();
-        $this->fillTimestampableAttributes();
     }
 
     private function fillType(): void

src/Entities/Database.php

@@ -99,17 +99,14 @@ protected function setResponseData(array $responseData): void
 
     private function fillFromRaw()
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillIcon();
         $this->fillCover();
         $this->fillTitle();
         $this->fillIsInline();
         $this->fillDescription();
         $this->fillProperties();
         $this->fillDatabaseUrl();
-        $this->fillParentAttributes();
-        $this->fillArchivedAttributes();
-        $this->fillTimestampableAttributes();
     }
 
     private function fillTitle(): void

src/Entities/Entity.php

@@ -72,10 +72,48 @@ protected function setResponseData(array $responseData): void
         $this->responseData = $responseData;
     }
 
-    protected function fillEntityBase(): void
+    protected function fillEssentials(): void
     {
         $this->fillId();
         $this->fillObjectType();
+        $this->fillTraitAttributes();
+    }
+
+    private function fillTraitAttributes(): void
+    {
+        $traitMapping = [
+            'FiveamCode\LaravelNotionApi\Traits\HasTimestamps' => function ($entity) {
+                $entity->fillTimestampableAttributes();
+            },
+            'FiveamCode\LaravelNotionApi\Traits\HasParent' => function ($entity) {
+                $entity->fillParentAttributes();
+            },
+            'FiveamCode\LaravelNotionApi\Traits\HasArchive' => function ($entity) {
+                $entity->fillArchivedAttributes();
+            },
+        ];
+
+        $traits = $this->class_uses_deep($this);
+        foreach ($traits as $trait) {
+            if (Arr::exists($traitMapping, $trait)) {
+                $traitMapping[$trait]($this);
+            }
+        }
+    }
+
+    private function class_uses_deep($class, $autoload = true)
+    {
+        $traits = [];
+
+        do {
+            $traits = array_merge(class_uses($class, $autoload), $traits);
+        } while ($class = get_parent_class($class));
+
+        foreach ($traits as $trait => $same) {
+            $traits = array_merge(class_uses($trait, $autoload), $traits);
+        }
+
+        return array_unique($traits);
     }
 
     private function fillId()

src/Entities/Page.php

@@ -111,15 +111,12 @@ protected function setResponseData(array $responseData): void
 
     private function fillFromRaw(): void
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillProperties();
         $this->fillTitle(); // This has to be called after fillProperties(), since title is provided by properties
         $this->fillPageUrl();
         $this->fillIcon();
         $this->fillCover();
-        $this->fillParentAttributes();
-        $this->fillArchivedAttributes();
-        $this->fillTimestampableAttributes();
     }
 
     /**

src/Entities/Properties/Property.php

@@ -62,7 +62,7 @@ protected function setResponseData(array $responseData): void
 
     protected function fillFromRaw(): void
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillType();
         $this->fillContent();
     }

src/Entities/PropertyItems/SelectItem.php

@@ -36,7 +36,7 @@ protected function setResponseData(array $responseData): void
 
     protected function fillFromRaw(): void
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillName();
         $this->fillColor();
     }

src/Entities/User.php

@@ -37,7 +37,7 @@ protected function setResponseData(array $responseData): void
 
     private function fillFromRaw(): void
     {
-        parent::fillEntityBase();
+        parent::fillEssentials();
         $this->fillName();
         $this->fillAvatarUrl();
     }