implement extensions on existing objects

1 files changed, 24 insertions, 0 deletions

diff --git a/Digitigrade/Model/Note.php b/Digitigrade/Model/Note.php
index 01bc861..0164e7d 100644
--- a/Digitigrade/Model/Note.php
+++ b/Digitigrade/Model/Note.php
@@ -81,10 +81,12 @@ class Note extends PushableModel implements TimelineIncludeable, Notifyable {
         $this->formattedContent = $this->findFormattedContents();
         $this->mentions = $this->findMentions();
         $this->attachments = NoteAttachment::findAllWhere('note_id = ?', [$this->id]);
+        $this->extensions = $this->findExtensions();
     }
 
     protected function dehydrate() {
         $this->saveMentions();
+        $this->saveExtensions();
     }
 
     protected function validate(): bool {
@@ -109,6 +111,17 @@ class Note extends PushableModel implements TimelineIncludeable, Notifyable {
         return $actors;
     }
 
+    private function findExtensions(): array {
+        $pdo = Db::getInstance()->getPdo();
+        $stmt = $pdo->prepare('SELECT uri, data FROM note_extension WHERE note_id = ?');
+        $stmt->execute([$this->id]);
+        $items = [];
+        while ($row = $stmt->fetch()) {
+            $items[$row['uri']] = json_decode($row['data']);
+        }
+        return $items;
+    }
+
     private function saveMentions() {
         $pdo = Db::getInstance()->getPdo();
         foreach ($this->mentions as $mention) {
@@ -117,6 +130,17 @@ class Note extends PushableModel implements TimelineIncludeable, Notifyable {
         }
     }
 
+    private function saveExtensions() {
+        $pdo = Db::getInstance()->getPdo();
+        foreach ($this->extensions as $uri => $obj) {
+            $stmt = $pdo->prepare(
+                'INSERT INTO note_extension(note_id, uri, data) VALUES (?, ?, ?) ' .
+                'ON CONFLICT (note_id, uri) DO UPDATE SET data = EXCLUDED.data'
+            );
+            $stmt->execute([$this->id, $uri, json_encode($obj)]);
+        }
+    }
+
     public static function findAllWithAuthor(Actor $author, ?int $limit = null, int $offset = 0, bool $includeDeleted = false): array {
         $where = 'author = ?';
         if (!$includeDeleted) {