checkWallet.js

require("dotenv").config();
const connectDB = require("./config/db.js");
connectDB();

const axios = require("axios");
const WalletModel = require("./models/walletModel.js");
const { getBalance, getSPLBalance } = require("./utils/transfer");
const { solNativeAddress } = require("./constant/constants.js");
const { getBalanceFromDB } = require("./utils/others.js");

const checkWallet = async () => {
  try {
    setInterval(async () => {
      const walletAccounts = await fetchWalletList();

      if (walletAccounts.length === 0) {
        return;
      }
      await checkWalletBalanceAndRequestDeposit(walletAccounts);
    }, process.env.CHECK_WALLET_INTERVAL * 1000);
  } catch (err) {
    console.log("CheckWallet Error: ", err);
  }
};

const fetchWalletList = async () => {
  try {
    const walletAccounts = await WalletModel.find({ archived: false });
    return walletAccounts;
  } catch (err) {
    console.log("fetchWalletList error: ", err);
  }
};

// SOL
const checkWalletBalanceAndRequestDeposit = async (walletAccounts) => {
  try {
    const promises = Promise.all(walletAccounts.map((d) => handleSingleWallet(d)));
    await promises.then();
  } catch (err) {
    console.log("checkWalletBalanceAndDeposit error: ", err);
  }
};

// SOL
const handleSingleWallet = async (walletAccount) => {
  try {
    const wallet = await WalletModel.findOne({ address: walletAccount.address });
    const userId = wallet.userId;

    const balance = await getBalance(walletAccount.address);
    const dbBalance = await getBalanceFromDB(userId, walletAccount.address, solNativeAddress);
    const depositAmount = (balance - dbBalance).toFixed(9);
    console.log(`deposit amount at ${walletAccount.address}`, depositAmount);
    if (balance > 0 && depositAmount > 0) {
      try {
        await axios
          .post(`${process.env.HOST_ENDPOINT}/user/deposit`, {
            userId,
            walletAddress: walletAccount.address,
            tokenAddress: solNativeAddress,
            amountIn: depositAmount,
          })
          .then((res) => {
            console.log(res.data);
          });
      } catch (err) {
        console.log("Deposit SOL Error: ", err);
      }
    }

    const splAccounts = await getSPLBalance(walletAccount.address);

    // Deposit SPL
    if (splAccounts.length !== 0) {
      try {
        splAccounts.forEach(async (account) => {
          console.log(
            `${account.tokenAddress} Balance of ${walletAccount.address} is ${account.balance}`
          );
          const dbBalance = await getBalanceFromDB(
            userId,
            walletAccount.address,
            account.tokenAddress
          );
          const depositAmount = (account.balance - dbBalance).toFixed(account.decimal);
          if(depositAmount !== 0) console.log(`deposit amount at ${walletAccount.address}`, depositAmount);
          if (account.balance > 0 && depositAmount > 0) {
            await axios
              .post(`${process.env.HOST_ENDPOINT}/user/deposit`, {
                userId,
                walletAddress: walletAccount.address,
                tokenAddress: account.tokenAddress,
                amountIn: depositAmount,
              })
              .then((res) => {
                console.log(res.data);
              });
          }
        });
      } catch (err) {
        console.log("Deposit SPL Error: ", err);
      }
    }

    console.log("End of a cycle");
  } catch (err) {
    console.log("handleSingleWallet error", err);
  }
};

checkWallet();