A few more api call basics

web/pages/api/v1/bookmarks.js

@@ -1,27 +1,45 @@
-import db from "../../../lib/db";
 import Ajv from "ajv";
 import addFormats from "ajv-formats";
-import { withErrorHandling } from "../../../lib/apiHelpers";
 
-const handler = withErrorHandling(async (req, res) => {
-  const { method } = req;
+import db from "../../../lib/db";
+import { withErrorHandling, withSession } from "../../../lib/apiHelpers";
 
-  switch (method) {
-    case "GET":
-      await getBookmark(req, res);
-      break;
-    case "POST":
-      await createBookmark(req, res);
-      break;
-    default:
-      res.setHeader("Allow", ["GET", "POST"]);
-      res.status(405).end(`Method ${method} Not Allowed`);
-  }
-});
+const handler = withErrorHandling(
+  withSession(async (req, res, session) => {
+    const { method } = req;
+    const { user } = session;
+
+    switch (method) {
+      case "GET":
+        await getBookmarks(req, res, user);
+        break;
+      case "POST":
+        await createBookmark(req, res, user);
+        break;
+      default:
+        res.setHeader("Allow", ["GET", "POST"]);
+        res.status(405).end(`Method ${method} Not Allowed`);
+    }
+  })
+);
 
 const ajv = new Ajv();
 addFormats(ajv);
 
+const getBookmarks = async (req, res, user) => {
+  const params = {
+    TableName: process.env.FMN_DYNAMODB_TABLENAME,
+    KeyConditionExpression: "hk = :hk AND begins_with(sk, :sk)",
+    ExpressionAttributeValues: {
+      ":hk": `user#${user.sub}`,
+      ":sk": "bookmark#",
+    },
+  };
+
+  let bookmarks = await db.query(params).promise();
+  res.status(200).json(bookmarks.Items);
+};
+
 const createBookmarkSchema = {
   type: "object",
   properties: {
@@ -33,18 +51,7 @@ const createBookmarkSchema = {
 
 const validateCreateBookmarkSchema = ajv.compile(createBookmarkSchema);
 
-const getBookmark = async (req, res) => {
-  const params = {
-    TableName: process.env.FMN_DYNAMODB_TABLENAME,
-    KeyConditionExpression: "hk = :hk",
-    ExpressionAttributeValues: { ":hk": "user#123" },
-  };
-
-  let bookmark = await db.query(params).promise();
-  res.status(200).json(bookmark);
-};
-
-const createBookmark = async (req, res) => {
+const createBookmark = async (req, res, user) => {
   if (!validateCreateBookmarkSchema(req.body)) {
     console.warn(validateCreateBookmarkSchema.errors);
     res.status(400).json(validateCreateBookmarkSchema.errors);
@@ -54,7 +61,7 @@ const createBookmark = async (req, res) => {
   const params = {
     TableName: process.env.FMN_DYNAMODB_TABLENAME,
     Item: {
-      hk: `user#${123}`,
+      hk: `user#${user.sub}`,
       sk: `bookmark#${"abc"}`,
       title: req.body.title,
       url: req.body.url,
@@ -62,6 +69,7 @@ const createBookmark = async (req, res) => {
   };
 
   await db.put(params).promise();
+  res.status(200).json(params.Item);
 };
 
 export default handler;