From a9d7358937bb54b11a29c4aa971698d7c038a597 Mon Sep 17 00:00:00 2001
From: Bernardo Vieira <bernardo@impactmarket.com>
Date: Mon, 6 Nov 2023 18:33:09 +0000
Subject: [PATCH] add tokens to levels rules

---
 .../models/learnAndEarn/learnAndEarnLevel.ts  |  1 +
 .../core/src/services/learnAndEarn/answer.ts  | 34 ++++++++++++++++---
 .../core/src/services/learnAndEarn/list.ts    |  7 +++-
 3 files changed, 36 insertions(+), 6 deletions(-)

diff --git a/packages/core/src/database/models/learnAndEarn/learnAndEarnLevel.ts b/packages/core/src/database/models/learnAndEarn/learnAndEarnLevel.ts
index 4fe1c32c4..8cf1b3d8a 100644
--- a/packages/core/src/database/models/learnAndEarn/learnAndEarnLevel.ts
+++ b/packages/core/src/database/models/learnAndEarn/learnAndEarnLevel.ts
@@ -17,6 +17,7 @@ export class LearnAndEarnLevelModel extends Model<LearnAndEarnLevel, LearnAndEar
     public rules!: {
         countries: string[];
         roles: string[];
+        tokens: { address: string; amount: number }[];
         limitUsers: number;
     };
 
diff --git a/packages/core/src/services/learnAndEarn/answer.ts b/packages/core/src/services/learnAndEarn/answer.ts
index 920fc699e..3d0b72cd0 100644
--- a/packages/core/src/services/learnAndEarn/answer.ts
+++ b/packages/core/src/services/learnAndEarn/answer.ts
@@ -8,6 +8,8 @@ import BigNumber from 'bignumber.js';
 
 import { BaseError } from '../../utils/baseError';
 import { cleanLearnAndEarnCache } from '../../utils/cache';
+import { contracts } from '../../../index';
+import { ethers } from 'ethers';
 import { getUserRoles } from '../../subgraph/queries/user';
 import { models, sequelize } from '../../database';
 import config from '../../config';
@@ -251,6 +253,32 @@ export async function answer(user: { userId: number; address: string }, answers:
         const availableLessons = await countAvailableLessons(lessonRegistry.levelId, user.userId);
 
         if (availableLessons === 0) {
+            // verify if user comply with rules
+            const level = await models.learnAndEarnLevel.findOne({
+                attributes: ['rules', 'rewardLimit'],
+                where: { id: lessonRegistry.levelId }
+            });
+
+            if (!level) {
+                throw new BaseError('LEVEL_NOT_FOUND', 'level not found');
+            }
+
+            // validate if user has the required tokens
+            if (level.rules.tokens) {
+                const provider = new ethers.providers.StaticJsonRpcProvider(config.chain.jsonRPCUrlCelo, {
+                    name: 'Celo',
+                    chainId: config.chain.isMainnet ? 42220 : 44787
+                });
+                for (const token of level.rules.tokens) {
+                    const tokenContract = new ethers.Contract(token.address, contracts.ERC20ABI, provider);
+                    const balance = new BigNumber((await tokenContract.balanceOf(user.address)).toString());
+                    const amount = new BigNumber(token.amount).multipliedBy(new BigNumber(10).pow(18));
+                    if (balance.lt(amount)) {
+                        throw new BaseError('INSUFFICIENT_TOKENS', 'user has insufficient tokens');
+                    }
+                }
+            }
+
             // if so, complete the level and make the payment
             await models.learnAndEarnUserLevel.update(
                 {
@@ -267,15 +295,11 @@ export async function answer(user: { userId: number; address: string }, answers:
             );
 
             // create signature
-            const level = await models.learnAndEarnLevel.findOne({
-                attributes: ['rewardLimit'],
-                where: { id: lessonRegistry.levelId }
-            });
             const amount = await calculateReward(user.userId, lessonRegistry.levelId, points);
             const signature = await signParams(user.address, lessonRegistry.levelId, amount);
 
             // check available reward
-            if (level?.rewardLimit) {
+            if (level.rewardLimit) {
                 const payments = await models.learnAndEarnPayment.sum('amount', {
                     where: {
                         levelId: lessonRegistry.levelId
diff --git a/packages/core/src/services/learnAndEarn/list.ts b/packages/core/src/services/learnAndEarn/list.ts
index 2f458cbc5..0a333e8f6 100644
--- a/packages/core/src/services/learnAndEarn/list.ts
+++ b/packages/core/src/services/learnAndEarn/list.ts
@@ -22,6 +22,10 @@ export async function listLevels(
         prismicId: string;
         totalReward: number;
         totalLessons: number;
+        asset: string;
+        // this is actually a json object
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        rules: any;
         status: string; // TODO: to be removed
     }[];
 }> {
@@ -45,7 +49,7 @@ export async function listLevels(
     }
 
     include.push({
-        attributes: ['id', 'totalReward', 'lessons', 'asset'],
+        attributes: ['id', 'totalReward', 'lessons', 'asset', 'rules'],
         model: learnAndEarnLevel,
         where: {
             [Op.and]: [
@@ -103,6 +107,7 @@ export async function listLevels(
             totalReward: level!.totalReward,
             totalLessons: level!.lessons,
             asset: level!.asset,
+            rules: level!.rules,
             status: level!.userLevel?.status || 'available'
         }))
     };