feat: twitter utils

Author: transitive-bullshit

packages/notion-utils/src/get-page-tweet-ids.ts

@@ -0,0 +1,23 @@
+import type * as types from 'notion-types'
+
+import { getPageTweetUrls } from './get-page-tweet-urls'
+
+/**
+ * Gets the IDs of all tweets embedded on a page.
+ */
+export const getPageTweetIds = (
+  recordMap: types.ExtendedRecordMap
+): string[] => {
+  const tweetUrls = getPageTweetUrls(recordMap)
+  return tweetUrls
+    .map((url) => {
+      try {
+        const u = new URL(url)
+        const parts = u.pathname.split('/')
+        return parts.at(-1)
+      } catch {
+        return
+      }
+    })
+    .filter(Boolean)
+}

packages/notion-utils/src/get-page-tweet-urls.ts

@@ -0,0 +1,25 @@
+import type * as types from 'notion-types'
+
+/**
+ * Gets the URLs of all tweets embedded on a page.
+ */
+export const getPageTweetUrls = (
+  recordMap: types.ExtendedRecordMap
+): string[] => {
+  const blockIds = Object.keys(recordMap.block)
+  const tweetUrls: string[] = blockIds
+    .map((blockId) => {
+      const block = recordMap.block[blockId]?.value
+
+      if (block?.type === 'tweet') {
+        const tweetUrl = block.properties?.source?.[0]?.[0]
+
+        if (tweetUrl) {
+          return tweetUrl
+        }
+      }
+    })
+    .filter(Boolean)
+
+  return Array.from(new Set(tweetUrls))
+}

packages/notion-utils/src/get-page-tweets.ts

@@ -14,7 +14,8 @@ export * from './get-page-image-urls'
 export * from './get-page-property'
 export * from './get-page-table-of-contents'
 export * from './get-page-title'
-export * from './get-page-tweets'
+export * from './get-page-tweet-ids'
+export * from './get-page-tweet-urls'
 export * from './get-text-content'
 export * from './id-to-uuid'
 export * from './is-url'