Adding ability to embed PDFs in Notion using Zotero Web API

content/notero-item.ts

@@ -1,4 +1,5 @@
 import Notion from './notion';
+import { NoteroPref, getNoteroPref } from './notero-pref';
 
 const APA_STYLE = 'bibliography=http://www.zotero.org/styles/apa';
 
@@ -134,6 +135,26 @@ export default class NoteroItem {
     return Zotero.URI.getItemURI(this.zoteroItem);
   }
 
+  public getPDFURL(): string {
+    const zoteroAPIKey = getNoteroPref(NoteroPref.zoteroAPIKey);
+    const zoteroUserID = getNoteroPref(NoteroPref.zoteroUserID);
+    const attachmentIDs = this.zoteroItem
+      .getAttachments(false)
+      .slice()
+      // Sort to get largest ID first
+      .sort((a, b) => b - a);
+
+    for (const id of attachmentIDs) {
+      const attachment = Zotero.Items.get(id);
+      if (attachment.attachmentContentType == 'application/pdf') {
+        const pdfItemID = Zotero.URI.getItemURI(attachment).split('/').pop();
+        return `https://api.zotero.org/users/${zoteroUserID}/items/${pdfItemID}/file/view?key=${zoteroAPIKey}`;
+      }
+    }
+    return 'https://zotero.org';
+  }
+
+
   public getNotionLinkAttachments(): Zotero.Item[] {
     const attachmentIDs = this.zoteroItem
       .getAttachments(false)

content/notero-pref.ts

@@ -4,13 +4,17 @@ export enum NoteroPref {
   notionDatabaseID = 'notionDatabaseID',
   notionToken = 'notionToken',
   syncOnModifyItems = 'syncOnModifyItems',
+  zoteroUserID = 'zoteroUserID',
+  zoteroAPIKey = 'zoteroAPIKey',
 }
 
 type NoteroPrefValue = Partial<{
   [NoteroPref.collectionName]: string;
   [NoteroPref.collectionSyncConfigs]: string;
   [NoteroPref.notionDatabaseID]: string;
   [NoteroPref.notionToken]: string;
+  [NoteroPref.zoteroAPIKey]: string;
+  [NoteroPref.zoteroUserID]: string;
   [NoteroPref.syncOnModifyItems]: boolean;
 }>;
 
@@ -35,6 +39,8 @@ export function getNoteroPref<P extends NoteroPref>(
     [NoteroPref.collectionSyncConfigs]: stringPref,
     [NoteroPref.notionDatabaseID]: stringPref,
     [NoteroPref.notionToken]: stringPref,
+    [NoteroPref.zoteroAPIKey]: stringPref,
+    [NoteroPref.zoteroUserID]: stringPref,
     [NoteroPref.syncOnModifyItems]: booleanPref,
   }[pref];
 }

content/notero.ts

@@ -287,6 +287,8 @@ class Notero {
 
     if ('url' in response) {
       await noteroItem.saveNotionLinkAttachment(response.url);
+      const pageID = Notion.getPageIDFromURL(Notion.convertWebURLToLocal(response.url));
+      const response_block = await notion.addEmbedToPage(noteroItem, pageID);
     }
   }
 }

content/notion.ts

@@ -10,6 +10,9 @@ import {
   CreatePageResponse,
   GetDatabaseResponse,
   UpdatePageResponse,
+  AppendBlockChildrenParameters,
+  AppendBlockChildrenResponse,
+  UpdateBlockResponse
 } from '@notionhq/client/build/src/api-endpoints';
 import 'core-js/stable/object/from-entries';
 import NoteroItem from './notero-item';
@@ -238,6 +241,11 @@ export default class Notion {
         type: 'url',
         buildRequest: () => item.getZoteroURI(),
       },
+      {
+        name: 'File URL',
+        type: 'url',
+        buildRequest: () => item.getPDFURL(),
+      },
     ];
 
     const validPropertyDefinitions =
@@ -254,4 +262,22 @@ export default class Notion {
 
     return itemProperties;
   }
+
+  public async addEmbedToPage(
+    item: NoteroItem,
+    pageID: string
+  ): Promise<AppendBlockChildrenResponse> {
+    const children = [
+        {
+            "object": "block",
+            "type": "embed",
+            "embed": {"url": item.getPDFURL()},
+        },
+      ];
+
+    return await this.client.blocks.children.append({ block_id: pageID, children });
+
+
+  }
+
 }

content/preferences.xul

@@ -42,6 +42,16 @@
              value="&notero.preferences.notionDatabaseID;"
              control="notero-notionDatabaseID" />
       <textbox id="notero-notionDatabaseID" preference="pref-notionDatabaseID" />
+      <separator />
+      <label id="notero-zoteroAPIKey-label"
+             value="&notero.preferences.zoteroAPIKey;"
+             control="notero-zoteroAPIKey" />
+      <textbox id="notero-zoteroAPIKey" preference="pref-zoteroAPIKey" />
+      <separator />
+      <label id="notero-zoteroUserID-label"
+             value="&notero.preferences.zoteroUserID;"
+             control="notero-zoteroUserID" />
+      <textbox id="notero-zoteroUserID" preference="pref-zoteroUserID" />
     </groupbox>
 
     <separator />

defaults/preferences/notero.js

@@ -4,3 +4,5 @@ pref('extensions.notero.collectionSyncConfigs', '');
 pref('extensions.notero.notionToken', '');
 pref('extensions.notero.notionDatabaseID', '');
 pref('extensions.notero.syncOnModifyItems', true);
+pref('extensions.notero.zoteroAPIKey', '');
+pref('extensions.notero.zoteroUserID', '');

locale/en-US/notero.dtd

@@ -10,6 +10,7 @@
 <!ENTITY notero.preferences.readme "README">
 <!ENTITY notero.preferences.notionToken "Integration Token">
 <!ENTITY notero.preferences.notionDatabaseID "Database ID">
+<!ENTITY notero.preferences.zoteroAPIKey "Zotero API Key">
 
 <!ENTITY notero.preferences.syncGroupboxCaption "Sync Preferences">
 <!ENTITY notero.preferences.syncGroupboxDescription1 "Notero will monitor the collections enabled below.">

locale/en-US/notero.properties

@@ -1,3 +1,4 @@
 notero.preferences.collectionName   = Collection Name
 notero.preferences.notionToken      = Notion Integration Token
 notero.preferences.notionDatabaseID = Notion Database ID
+notero.preferences.zoteroAPIKey     = Zotero API Key