Update checkWallet

Author: ekzyis

worker/wallet.js

@@ -12,9 +12,7 @@ import {
   paidActionCanceling
 } from './paidAction'
 import { payingActionConfirmed, payingActionFailed } from './payingAction'
-import { canReceive, getWalletByType } from '@/wallets/common'
 import { notifyNewStreak, notifyStreakLost } from '@/lib/webPush'
-import { hasVault, vaultPrismaFragments } from '@/wallets/vault'
 
 export async function subscribeToWallet (args) {
   await subscribeToDeposits(args)
@@ -292,19 +290,20 @@ export async function checkWallet ({ data: { userId }, models }) {
   const pushNotifications = []
 
   await models.$transaction(async tx => {
-    // TODO: use UserWallet instead of Wallet table
-    const wallets = await tx.wallet.findMany({
+    const wallets = await tx.userWallet.findMany({
       where: {
         userId,
         enabled: true
       },
-      include: vaultPrismaFragments.include()
+      include: {
+        protocols: true
+      }
     })
 
     const { hasRecvWallet: oldHasRecvWallet, hasSendWallet: oldHasSendWallet } = await tx.user.findUnique({ where: { id: userId } })
 
-    const newHasRecvWallet = wallets.some(({ type, wallet }) => canReceive({ def: getWalletByType(type), config: wallet }))
-    const newHasSendWallet = wallets.some(hasVault)
+    const newHasRecvWallet = wallets.some(({ protocols }) => protocols.some(proto => !proto.send))
+    const newHasSendWallet = wallets.some(({ protocols }) => protocols.some(proto => proto.send))
 
     await tx.user.update({
       where: { id: userId },