Ensure WA exists for lock and unlock calls

interledger · Oct 2, 2024 · b0de9ce · b0de9ce
1 parent fcfc76b
commit b0de9ce
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/packages/wallet/backend/src/card/controller.ts b/packages/wallet/backend/src/card/controller.ts
@@ -132,12 +132,14 @@ export class CardController implements ICardController {
 
   public lock = async (req: Request, res: Response, next: NextFunction) => {
     try {
+      const userId = req.session.user.id
       const { params, query, body } = await validate(lockCardSchema, req)
       const { cardId } = params
       const { reasonCode } = query
       const requestBody: ICardLockRequest = body
 
       const result = await this.cardService.lock(
+        userId,
         cardId,
         reasonCode,
         requestBody
@@ -151,11 +153,12 @@ export class CardController implements ICardController {
 
   public unlock = async (req: Request, res: Response, next: NextFunction) => {
     try {
+      const userId = req.session.user.id
       const { params, body } = await validate(unlockCardSchema, req)
       const { cardId } = params
       const requestBody: ICardUnlockRequest = body
 
-      const result = await this.cardService.unlock(cardId, requestBody)
+      const result = await this.cardService.unlock(userId, cardId, requestBody)
 
       res.status(200).json(toSuccessResponse(result))
     } catch (error) {

diff --git a/packages/wallet/backend/src/card/service.ts b/packages/wallet/backend/src/card/service.ts
@@ -64,17 +64,23 @@ export class CardService {
   }
 
   async lock(
+    userId: string,
     cardId: string,
     reasonCode: LockReasonCode,
     requestBody: ICardLockRequest
   ): Promise<ICardResponse> {
+    await this.ensureWalletAddressExists(userId, cardId)
+
     return this.gateHubClient.lockCard(cardId, reasonCode, requestBody)
   }
 
   async unlock(
+    userId: string,
     cardId: string,
     requestBody: ICardUnlockRequest
   ): Promise<ICardResponse> {
+    await this.ensureWalletAddressExists(userId, cardId)
+
     return this.gateHubClient.unlockCard(cardId, requestBody)
   }