initial commit

Author: L-a-r-t

assets/icon.png

@@ -2,30 +2,42 @@
   "name": "chatgpt-to-notion",
   "displayName": "Chatgpt to notion",
   "version": "0.0.0",
-  "description": "A basic Plasmo extension.",
-  "author": "lrtig",
+  "description": "Save your discussions with ChatGPT directly into Notion!",
+  "author": "Théo Lartigau",
   "scripts": {
     "dev": "plasmo dev",
     "build": "plasmo build",
     "package": "plasmo package"
   },
   "dependencies": {
+    "@headlessui/react": "^1.7.7",
+    "@notionhq/client": "^2.2.3",
+    "@plasmohq/storage": "^0.13.1",
+    "@tryfabric/martian": "^1.2.4",
+    "node-html-markdown": "^1.3.0",
     "plasmo": "0.60.2",
     "react": "18.2.0",
-    "react-dom": "18.2.0"
+    "react-dom": "18.2.0",
+    "shrink-string": "^3.1.2"
   },
   "devDependencies": {
     "@plasmohq/prettier-plugin-sort-imports": "3.6.0",
     "@types/chrome": "0.0.202",
     "@types/node": "18.11.9",
     "@types/react": "18.0.25",
     "@types/react-dom": "18.0.9",
+    "autoprefixer": "^10.4.13",
+    "postcss": "^8.4.20",
     "prettier": "2.7.1",
+    "tailwindcss": "^3.2.4",
     "typescript": "4.9.3"
   },
   "manifest": {
     "host_permissions": [
       "https://*/*"
+    ],
+    "permissions": [
+      "tabs"
     ]
   }
 }

assets/illustration.png

@@ -0,0 +1,6 @@
+module.exports = {
+  plugins: {
+    tailwindcss: {},
+    autoprefixer: {},
+  },
+}

package.json

@@ -0,0 +1,24 @@
+import { Storage } from "@plasmohq/storage"
+
+export const generateToken = async (code: string) => {
+  console.log(code)
+  const storage = new Storage({
+    area: "session"
+  })
+  const response = await fetch(
+    "https://chatgpt-to-notion.onrender.com/token/new",
+    {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        code,
+        grant_type: "authorization-code",
+        redirect_uri: "https://github.com/L-a-r-t/chatgpt-to-notion"
+      })
+    }
+  )
+  const { token } = await response.json()
+  return token
+}

popup.tsx

@@ -0,0 +1,19 @@
+export const getToken = async ({ workspace_id, user_id }: GetTokenParams) => {
+  const response = await fetch("https://chatgpt-to-notion.onrender.com/token", {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json"
+    },
+    body: JSON.stringify({
+      workspace_id,
+      user_id
+    })
+  })
+  const { token } = await response.json()
+  return token
+}
+
+type GetTokenParams = {
+  workspace_id: string
+  user_id: string
+}

postcss.config.js

@@ -0,0 +1,84 @@
+import { decompress } from "shrink-string"
+
+import notion from "~config/notion"
+import { generateBlocks } from "~utils/functions/notion"
+
+// save new page to notion database
+export const saveAnswer = async ({
+  prompt,
+  answer,
+  title,
+  url,
+  database_id
+}: SaveAnswerParams) => {
+  const decompressedAnswer = await decompress(answer)
+  const decompressedPrompt = await decompress(prompt)
+  const { answerBlocks, promptBlocks } = generateBlocks(
+    decompressedPrompt,
+    decompressedAnswer
+  )
+
+  const searchRes = await notion.databases.query({
+    database_id,
+    filter: {
+      property: "Name",
+      title: {
+        equals: title
+      }
+    }
+  })
+
+  if (searchRes.results.length > 0) {
+    const page = searchRes.results[0]
+    const block_id = page.id
+    const response = await notion.blocks.children.append({
+      block_id,
+      children: [...promptBlocks, ...answerBlocks]
+    })
+    return response
+  }
+
+  const response = await notion.pages.create({
+    parent: {
+      database_id: database_id
+    },
+    icon: {
+      type: "external",
+      external: {
+        url: "https://openai.com/content/images/2022/05/openai-avatar.png"
+      }
+    },
+    properties: {
+      Name: {
+        title: [
+          {
+            text: {
+              content: title
+            }
+          }
+        ]
+      },
+      URL: {
+        url
+      }
+    },
+    children: [
+      {
+        object: "block",
+        type: "table_of_contents",
+        table_of_contents: {}
+      },
+      ...promptBlocks,
+      ...answerBlocks
+    ]
+  })
+  return response
+}
+
+type SaveAnswerParams = {
+  prompt: string
+  answer: string
+  title: string
+  database_id: string
+  url: string
+}

src/api/generateToken.ts

@@ -0,0 +1,84 @@
+import { decompress } from "shrink-string"
+
+import notion from "~config/notion"
+import { generateBlocks } from "~utils/functions/notion"
+
+// save new page to notion database
+export const saveChat = async ({
+  prompts,
+  answers,
+  title,
+  url,
+  database_id
+}: SaveChatParams) => {
+  const blocks = []
+  for (let i = 0; i < prompts.length; i++) {
+    const { answerBlocks, promptBlocks } = generateBlocks(
+      prompts[i],
+      answers[i]
+    )
+    blocks.push(...promptBlocks, ...answerBlocks)
+  }
+
+  const searchRes = await notion.databases.query({
+    database_id,
+    filter: {
+      property: "Name",
+      title: {
+        equals: title
+      }
+    }
+  })
+
+  if (searchRes.results.length > 0) {
+    const page = searchRes.results[0]
+    const page_id = page.id
+    await notion.pages.update({
+      page_id,
+      archived: true
+    })
+  }
+
+  const response = await notion.pages.create({
+    parent: {
+      database_id: database_id
+    },
+    icon: {
+      type: "external",
+      external: {
+        url: "https://openai.com/content/images/2022/05/openai-avatar.png"
+      }
+    },
+    properties: {
+      Name: {
+        title: [
+          {
+            text: {
+              content: title
+            }
+          }
+        ]
+      },
+      URL: {
+        url
+      }
+    },
+    children: [
+      {
+        object: "block",
+        type: "table_of_contents",
+        table_of_contents: {}
+      },
+      ...blocks
+    ]
+  })
+  return response
+}
+
+type SaveChatParams = {
+  prompts: string[]
+  answers: string[]
+  title: string
+  database_id: string
+  url: string
+}

src/api/getToken.ts

@@ -0,0 +1,16 @@
+import notion from "~config/notion"
+import { parseSearchResponse } from "~utils/functions/notion"
+
+// Search notion using the notion sdk
+export const searchNotion = async (query: string) => {
+  const response = await notion.search({
+    query,
+    page_size: 6,
+    filter: {
+      property: "object",
+      value: "database"
+    }
+  })
+  const parsed = parseSearchResponse(response.results)
+  return parsed
+}

src/api/saveAnswer.ts

@@ -0,0 +1,50 @@
+import { Storage } from "@plasmohq/storage"
+
+import { generateToken } from "~api/generateToken"
+import { getToken } from "~api/getToken"
+import { saveAnswer } from "~api/saveAnswer"
+import { searchNotion } from "~api/search"
+
+// API calls that can be made from content scripts transit trough the background script
+// This is done to prevent CORS errors
+chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
+  switch (message.type) {
+    case "search":
+      searchNotion(message.body.query).then((res) => {
+        sendResponse(res)
+      })
+      break
+    case "saveAnswer":
+      saveAnswer(message.body).then((res) => {
+        sendResponse(res)
+      })
+      break
+    case "generateToken":
+      generateToken(message.body.code).then((res) => {
+        sendResponse(res)
+      })
+      break
+    default:
+      return true
+  }
+  return true
+})
+
+const authenticate = async () => {
+  const storage = new Storage({
+    secretKeyList: ["token"]
+  })
+  const [workspace_id, user_id] = await Promise.all([
+    storage.get("workspace_id"),
+    storage.get("user_id")
+  ])
+  const token = await getToken({
+    workspace_id,
+    user_id
+  })
+  await storage.set("token", token)
+}
+
+authenticate()
+
+export default {}