🐛 update existing recovery words (#99)

packages/tom-server/src/vault-api/controllers/vault.test.ts

@@ -14,7 +14,8 @@ describe('Vault controllers', () => {
   const dbManager: Partial<TwakeDB> = {
     get: jest.fn(),
     insert: jest.fn(),
-    deleteWhere: jest.fn()
+    deleteWhere: jest.fn(),
+    update: jest.fn()
   }
   let mockRequest: ITestRequest
   let mockResponse: Partial<Response>
@@ -69,6 +70,7 @@ describe('Vault controllers', () => {
   // Testing saveRecoveryWords
   it('should return response with status code 201 on save success', async () => {
     jest.spyOn(dbManager, 'insert').mockResolvedValue([{ words }])
+    jest.spyOn(dbManager, 'get').mockResolvedValue([])
     const handler: expressAppHandler = saveRecoveryWords(dbManager as TwakeDB)
     handler(mockRequest as Request, mockResponse as Response, nextFunction)
     await new Promise(process.nextTick)
@@ -78,12 +80,26 @@ describe('Vault controllers', () => {
   it('should call next function to throw error on saving failed', async () => {
     const errorMsg = 'Insert failed'
     jest.spyOn(dbManager, 'insert').mockRejectedValue(new Error(errorMsg))
+    jest.spyOn(dbManager, 'get').mockResolvedValue([])
     const handler: expressAppHandler = saveRecoveryWords(dbManager as TwakeDB)
     handler(mockRequest as Request, mockResponse as Response, nextFunction)
     await new Promise(process.nextTick)
     expect(nextFunction).toHaveBeenCalledWith(new Error(errorMsg))
   })
 
+  it('should update the recoverywords when it already exists', async () => {
+    jest
+      .spyOn(dbManager, 'get')
+      .mockResolvedValue([{ words: 'Another sentence for the same user' }])
+    jest.spyOn(dbManager, 'update').mockResolvedValue([{ words }])
+    const handler: expressAppHandler = saveRecoveryWords(dbManager as TwakeDB)
+    handler(mockRequest as Request, mockResponse as Response, nextFunction)
+    await new Promise(process.nextTick)
+    expect(mockResponse.statusCode).toEqual(200)
+    expect(dbManager.update).toHaveBeenCalled()
+    expect(dbManager.insert).not.toHaveBeenCalled()
+  })
+
   // Testing getRecoveryWords
 
   it('should return response with status code 200 on get success', async () => {

packages/tom-server/src/vault-api/controllers/vault.ts

@@ -1,4 +1,6 @@
+/* eslint-disable no-useless-return */
 import { type TwakeDB } from '../../db'
+import type { Collections } from '../../types'
 import { VaultAPIError, type expressAppHandler } from '../utils'
 
 export type VaultController = (db: TwakeDB) => expressAppHandler
@@ -9,14 +11,34 @@ export const methodNotAllowed: expressAppHandler = (req, res, next) => {
 
 export const saveRecoveryWords = (db: TwakeDB): expressAppHandler => {
   return (req, res, next) => {
-    const data: Record<string, string> = {
-      userId: req.token.content.sub,
-      words: req.body.words
-    }
-    // @ts-expect-error 'recoveryWords' isn't declared in Collection
-    db.insert('recoveryWords', data)
-      .then((_) => {
-        res.status(201).json({ message: 'Saved recovery words sucessfully' })
+    const { words } = req.body
+    const userId = req.token.content.sub
+
+    db.get('recoveryWords' as Collections, ['words'], { userId })
+      .then((data) => {
+        if (data.length > 0) {
+          db.update('recoveryWords' as Collections, { words }, 'userId', userId)
+            .then((_) => {
+              res
+                .status(200)
+                .json({ message: 'Updated recovery words sucessfully' })
+            })
+            .catch((err) => {
+              next(err)
+              return
+            })
+        } else {
+          db.insert('recoveryWords' as Collections, { userId, words })
+            .then((_) => {
+              res
+                .status(201)
+                .json({ message: 'Saved recovery words sucessfully' })
+            })
+            .catch((err) => {
+              next(err)
+              return
+            })
+        }
       })
       .catch((err) => {
         next(err)