feat: store author in documents

jkklapp · May 27, 2022 · 2b8c14e · 2b8c14e
1 parent 309ca4e
commit 2b8c14e
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 12 deletions.
diff --git a/backend/src/posts/controller.spec.ts b/backend/src/posts/controller.spec.ts
@@ -18,24 +18,34 @@ describe('Controller', () => {
         {
           message: 'test',
           date: Timestamp.fromMillis(100000),
+          author: '1234',
         },
       ];
       jest
         .spyOn(s, 'findAll')
         .mockImplementation(() => Promise.resolve(result));
 
-      expect(await c.findAll()).toBe(result);
+      expect(await c.findAll({ user: { user_id: '1234' } })).toBe(result);
     });
   });
   describe('create', () => {
     it('should return a new post', async () => {
       const post = {
         message: 'test',
-        date: Timestamp.fromMillis(100000),
       };
-      jest.spyOn(s, 'create').mockImplementation(() => Promise.resolve(post));
+      jest.spyOn(s, 'create').mockImplementation(() =>
+        Promise.resolve({
+          ...post,
+          date: Timestamp.fromMillis(100000),
+          author: '1234',
+        }),
+      );
 
-      expect(await c.create(post)).toBe(post);
+      expect(await c.create({ user: { user_id: '1234' } }, post)).toEqual({
+        ...post,
+        date: Timestamp.fromMillis(100000),
+        author: '1234',
+      });
     });
   });
 });
diff --git a/backend/src/posts/controller.ts b/backend/src/posts/controller.ts
@@ -3,9 +3,10 @@ import {
   Controller as BaseController,
   Get,
   Post,
+  Req,
   UseGuards,
 } from '@nestjs/common';
-import { PostDocument } from './document';
+import { PostDocument, NewPostDocument } from './document';
 import { Service } from './service';
 import { FirebaseAuthGuard } from '../firebase/firebase-auth.guard';
 
@@ -15,12 +16,17 @@ export class Controller {
 
   @Get()
   @UseGuards(FirebaseAuthGuard)
-  findAll(): Promise<PostDocument[]> {
-    return this.service.findAll();
+  findAll(@Req() request: any): Promise<PostDocument[]> {
+    const userId = request.user.user_id;
+    return this.service.findAll(userId);
   }
   @Post()
   @UseGuards(FirebaseAuthGuard)
-  public create(@Body() post: PostDocument): Promise<PostDocument> {
-    return this.service.create(post);
+  public create(
+    @Req() request: any,
+    @Body() post: NewPostDocument,
+  ): Promise<PostDocument> {
+    const userId = request.user.user_id;
+    return this.service.create(post, userId);
   }
 }
diff --git a/backend/src/posts/document.ts b/backend/src/posts/document.ts
@@ -5,4 +5,9 @@ export class PostDocument {
 
   message: string;
   date: Timestamp;
+  author: string;
+}
+
+export class NewPostDocument {
+  message: string;
 }
diff --git a/backend/src/posts/service.ts b/backend/src/posts/service.ts
@@ -12,23 +12,28 @@ export class Service {
     private postsCollection: CollectionReference<PostDocument>,
   ) {}
 
-  async create({ message }): Promise<PostDocument> {
+  async create({ message }, userId): Promise<PostDocument> {
     const t = dayjs(new Date()).valueOf();
     const date = Timestamp.fromMillis(t);
     const docRef = this.postsCollection.doc(t.toString());
     await docRef.set({
       message,
       date,
+      author: userId,
     });
     const postDoc = await docRef.get();
     const post = postDoc.data();
     return post;
   }
 
-  async findAll(): Promise<PostDocument[]> {
+  async findAll(userId): Promise<PostDocument[]> {
     const snapshot = await this.postsCollection.get();
     const posts: PostDocument[] = [];
-    snapshot.forEach((doc) => posts.push(doc.data()));
+    snapshot.forEach((doc) => {
+      if (doc.data().author === userId) {
+        posts.push(doc.data());
+      }
+    });
     return posts;
   }
 }