feat: create_highlight

omnivoreql/omnivoreql.py

@@ -1,6 +1,7 @@
 import uuid
+import shortuuid
 import os
-from typing import List, Optional
+from typing import List, Optional, Literal
 from gql.transport.requests import RequestsHTTPTransport
 from gql import gql, Client
 from dataclasses import asdict
@@ -279,3 +280,24 @@ def set_page_labels_by_ids(self, page_id: str, label_ids: List[str]) -> dict:
                 }
             },
         )
+    def create_highlight(self, article_id: str, annotation: str, 
+                         highlight_type: Literal["HIGHLIGHT", "NOTE"]):
+        """
+        Create a new highlight.
+
+        :param article_id: The ID of the article to create the highlight for.
+        :param annotation: The annotation of the highlight.
+        :param highlight_type: The type of the highlight.
+        """
+        return self.client.execute(
+            self._get_query("CreateHighlight"),
+            variable_values={
+                "input": {
+                    "annotation": annotation,
+                    "articleId": article_id,
+                    "id": str(uuid.uuid4()),
+                    "shortId": str(shortuuid.ShortUUID().random(length=8)),
+                    "type": highlight_type
+                }
+            },
+        )

omnivoreql/queries/CreateHighlight.graphql

@@ -11,3 +11,27 @@ mutation CreateHighlight($input: CreateHighlightInput!) {
         }
     }
 }
+
+fragment HighlightFields on Highlight {
+    id
+    type
+    shortId
+    quote
+    prefix
+    suffix
+    patch
+    color
+    annotation
+    createdByMe
+    createdAt
+    updatedAt
+    sharedAt
+    highlightPositionPercent
+    highlightPositionAnchorIndex
+    labels {
+        id
+        name
+        color
+        createdAt
+    }
+}