splitbee · tobiaslins · Jul 30, 2024 · Jul 30, 2024 · Jul 30, 2024 · Jul 30, 2024
diff --git a/.editorconfig b/.editorconfig
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,36 @@
-dist
-node_modules
-worker
-wrangler.toml
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+.yarn/install-state.gz
+
+# testing
+/coverage
+
+# next.js
+/.next/
+/out/
+
+# production
+/build
+
+# misc
+.DS_Store
+*.pem
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+
+# local env files
+.env*.local
+
+# vercel
+.vercel
+
+# typescript
+*.tsbuildinfo
+next-env.d.ts
diff --git a/.prettierrc b/.prettierrc
diff --git a/LICENSE.md b/LICENSE.md
diff --git a/app/layout.tsx b/app/layout.tsx
@@ -0,0 +1,12 @@
+import React from "react";
+export default function RootLayout({
+  children,
+}: Readonly<{
+  children: React.ReactNode;
+}>) {
+  return (
+    <html lang="en">
+      <body>{children}</body>
+    </html>
+  );
+}
diff --git a/src/api/notion.ts → app/notion-api/notion.ts b/src/api/notion.ts → app/notion-api/notion.ts
diff --git a/src/api/types.ts → app/notion-api/types.ts b/src/api/types.ts → app/notion-api/types.ts
@@ -1,4 +1,5 @@
-import { Params } from "tiny-request-router";
+import { Context } from "hono";
+
 
 type BoldFormatType = ["b"];
 type ItalicFormatType = ["i"];
@@ -201,9 +202,4 @@ export interface NotionSearchResultsType {
   total: number;
 }
 
-export interface HandlerRequest {
-  params: Params;
-  searchParams: URLSearchParams;
-  request: Request;
-  notionToken?: string;
-}
+export type HandlerRequest  = Context
diff --git a/src/api/utils.ts → app/notion-api/utils.ts b/src/api/utils.ts → app/notion-api/utils.ts
@@ -6,11 +6,8 @@ import {
   RowType,
 } from "./types";
 
-export const idToUuid = (path: string) =>
-  `${path.slice(0, 8)}-${path.slice(8, 4)}-${path.slice(
-    12,
-    4
-  )}-${path.slice(16, 4)}-${path.slice(20)}`;
+export const idToUuid = (path: string): string =>
+  `${path.slice(0, 8)}-${path.slice(8, 12)}-${path.slice(12, 16)}-${path.slice(16, 20)}-${path.slice(20)}`;
 
 export const parsePageId = (id: string) => {
   if (id) {
@@ -84,3 +81,4 @@ export const getNotionValue = (
 const getTextContent = (text: DecorationType[]) => {
   return text.reduce((prev, current) => prev + current[0], "");
 };
+
diff --git a/src/routes/page.ts → app/routes/notion-page.ts b/src/routes/page.ts → app/routes/notion-page.ts
@@ -1,12 +1,15 @@
-import { fetchPageById, fetchBlocks } from "../api/notion";
-import { parsePageId } from "../api/utils";
-import { createResponse } from "../response";
+import { fetchPageById, fetchBlocks } from "../notion-api/notion";
+import { parsePageId } from "../notion-api/utils";
+import { BlockType, CollectionType, HandlerRequest } from "../notion-api/types";
 import { getTableData } from "./table";
-import { BlockType, CollectionType, HandlerRequest } from "../api/types";
+import { createResponse } from "../utils/response";
+import { getNotionToken } from "../utils";
 
-export async function pageRoute(req: HandlerRequest) {
-  const pageId = parsePageId(req.params.pageId);
-  const page = await fetchPageById(pageId!, req.notionToken);
+export async function pageRoute(c: HandlerRequest) {
+  const pageId = parsePageId(c.req.param("pageId"));
+  const notionToken = getNotionToken(c);
+
+  const page = await fetchPageById(pageId!, notionToken);
 
   const baseBlocks = page.recordMap.block;
 
@@ -34,7 +37,7 @@ export async function pageRoute(req: HandlerRequest) {
       break;
     }
 
-    const newBlocks = await fetchBlocks(pendingBlocks, req.notionToken).then(
+    const newBlocks = await fetchBlocks(pendingBlocks, notionToken).then(
       (res) => res.recordMap.block
     );
 
@@ -55,11 +58,13 @@ export async function pageRoute(req: HandlerRequest) {
     const pendingCollections = allBlockKeys.flatMap((blockId) => {
       const block = allBlocks[blockId];
 
-      return (block.value && block.value.type === "collection_view") ? [block.value.id] : [];
+      return block.value && block.value.type === "collection_view"
+        ? [block.value.id]
+        : [];
     });
 
     for (let b of pendingCollections) {
-      const collPage = await fetchPageById(b!, req.notionToken);
+      const collPage = await fetchPageById(b!, notionToken);
 
       const coll = Object.keys(collPage.recordMap.collection).map(
         (k) => collPage.recordMap.collection[k]
@@ -74,7 +79,7 @@ export async function pageRoute(req: HandlerRequest) {
       const { rows, schema } = await getTableData(
         coll,
         collView.value.id,
-        req.notionToken,
+        notionToken,
         true
       );
 
@@ -95,5 +100,7 @@ export async function pageRoute(req: HandlerRequest) {
     }
   }
 
-  return createResponse(allBlocks);
+  return createResponse(allBlocks, {
+    request: c,
+  });
 }
diff --git a/app/routes/search.ts b/app/routes/search.ts
@@ -0,0 +1,37 @@
+import { fetchNotionSearch } from "../notion-api/notion";
+import { HandlerRequest } from "../notion-api/types";
+import { parsePageId } from "../notion-api/utils";
+import { getNotionToken } from "../utils";
+import { createResponse } from "../utils/response";
+
+export async function searchRoute(c: HandlerRequest) {
+  const notionToken = getNotionToken(c);
+
+  const ancestorId = parsePageId(c.req.query("ancestorId") || "");
+  const query = c.req.query("query") || "";
+  const limit = Number(c.req.query("limit") || 20);
+
+  if (!ancestorId) {
+    return createResponse(
+      { error: 'missing required "ancestorId"' },
+      {
+        headers: { "Content-Type": "application/json" },
+        statusCode: 400,
+        request: c,
+      }
+    );
+  }
+
+  const results = await fetchNotionSearch(
+    {
+      ancestorId,
+      query,
+      limit,
+    },
+    notionToken
+  );
+
+  return createResponse(results, {
+    request: c,
+  });
+}
diff --git a/src/routes/table.ts → app/routes/table.ts b/src/routes/table.ts → app/routes/table.ts
@@ -1,12 +1,17 @@
-import { fetchPageById, fetchTableData, fetchNotionUsers } from "../api/notion";
-import { parsePageId, getNotionValue } from "../api/utils";
+import {
+  fetchPageById,
+  fetchTableData,
+  fetchNotionUsers,
+} from "../notion-api/notion";
+import { parsePageId, getNotionValue } from "../notion-api/utils";
 import {
   RowContentType,
   CollectionType,
   RowType,
   HandlerRequest,
-} from "../api/types";
-import { createResponse } from "../response";
+} from "../notion-api/types";
+import { createResponse } from "../utils/response";
+import { getNotionToken } from "../utils";
 
 export const getTableData = async (
   collection: CollectionType,
@@ -23,9 +28,10 @@ export const getTableData = async (
   const collectionRows = collection.value.schema;
   const collectionColKeys = Object.keys(collectionRows);
 
-  const tableArr: RowType[] = table.result.reducerResults.collection_group_results.blockIds.map(
-    (id: string) => table.recordMap.block[id]
-  );
+  const tableArr: RowType[] =
+    table.result.reducerResults.collection_group_results.blockIds.map(
+      (id: string) => table.recordMap.block[id]
+    );
 
   const tableData = tableArr.filter(
     (b) =>
@@ -56,15 +62,15 @@ export const getTableData = async (
   return { rows, schema: collectionRows };
 };
 
-export async function tableRoute(req: HandlerRequest) {
-  const pageId = parsePageId(req.params.pageId);
-  const page = await fetchPageById(pageId!, req.notionToken);
+export async function tableRoute(c: HandlerRequest) {
+  const pageId = parsePageId(c.req.param("pageId"));
+  const notionToken = getNotionToken(c);
+  const page = await fetchPageById(pageId!, notionToken);
 
   if (!page.recordMap.collection)
     return createResponse(
       JSON.stringify({ error: "No table found on Notion page: " + pageId }),
-      {},
-      401
+      { headers: {}, statusCode: 401, request: c }
     );
 
   const collection = Object.keys(page.recordMap.collection).map(
@@ -80,8 +86,8 @@ export async function tableRoute(req: HandlerRequest) {
   const { rows } = await getTableData(
     collection,
     collectionView.value.id,
-    req.notionToken
+    notionToken
   );
 
-  return createResponse(rows);
+  return createResponse(rows, { request: c });
 }
diff --git a/app/routes/user.ts b/app/routes/user.ts
@@ -0,0 +1,13 @@
+import { fetchNotionUsers } from "../notion-api/notion";
+import { HandlerRequest } from "../notion-api/types";
+import { getNotionToken } from "../utils";
+import { createResponse } from "../utils/response";
+
+export async function userRoute(c: HandlerRequest) {
+  const users = await fetchNotionUsers(
+    [c.req.param("userId")],
+    getNotionToken(c)
+  );
+
+  return createResponse(users[0], { request: c });
+}
diff --git a/app/utils/index.ts b/app/utils/index.ts
@@ -0,0 +1,6 @@
+import { HandlerRequest } from "../notion-api/types";
+
+export const getNotionToken = (c: HandlerRequest) => {
+  return process.env.NOTION_TOKEN || (c.req.header("Authorization") || "").split("Bearer ")[1] || undefined;
+
+}
diff --git a/app/utils/response.ts b/app/utils/response.ts
@@ -0,0 +1,26 @@
+import { HandlerRequest, JSONData } from "../notion-api/types";
+
+export const createResponse = (
+  body: JSONData | any,
+  {
+    headers,
+    statusCode,
+    request,
+  }: {
+    request: HandlerRequest;
+    headers?: { [key: string]: string };
+    statusCode?: number;
+  }
+) => {
+  const cacheControl = request.req.header("Cache-Control");
+  return new Response(JSON.stringify(body), {
+    status: statusCode || 200,
+    headers: {
+      "Access-Control-Allow-Origin": "*",
+      "Access-Control-Allow-Methods": "GET, OPTIONS",
+      "Content-Type": "application/json",
+      "Cache-Control": cacheControl || "public, max-age=3600",
+      ...headers,
+    },
+  });
+};
diff --git a/app/v1/[[...route]]/route.ts b/app/v1/[[...route]]/route.ts
@@ -0,0 +1,17 @@
+import { Hono } from "hono";
+import { handle } from "hono/vercel";
+
+import { pageRoute } from "../../routes/notion-page";
+import { tableRoute } from "../../routes/table";
+import { userRoute } from "../../routes/user";
+import { searchRoute } from "../../routes/search";
+
+const app = new Hono().basePath("/v1");
+
+app.get("/page/:pageId", pageRoute);
+app.get("/table/:pageId", tableRoute);
+app.get("/user/:userId", userRoute);
+app.get("/search", searchRoute);
+
+export const GET = handle(app);
+export const POST = handle(app);
diff --git a/next.config.mjs b/next.config.mjs
@@ -0,0 +1,4 @@
+/** @type {import('next').NextConfig} */
+const nextConfig = {};
+
+export default nextConfig;