Fix: UserEMode user creation and relations (#33)

* fix: create user if not exists on eModeSwitch event

* feat: add transaction data on user e mode

* fix: make row of userEMode unique per protocol

* chore: update user e mode docs

Author: yvesfracari

relations.schema.ts

@@ -33,6 +33,7 @@ export const protocolRelations = relations(protocol, ({ one, many }) => ({
   delegatesUpdated: many(delegateUpdated),
   delegates: many(userDelegation),
   eModes: many(eMode),
+  userEModes: many(userEMode),
 }));
 
 export const actionPausedRelations = relations(actionPaused, ({ one }) => ({
@@ -323,6 +324,14 @@ export const userEModeRelations = relations(userEMode, ({ one }) => ({
     fields: [userEMode.userId],
     references: [user.id],
   }),
+  transaction: one(transaction, {
+    fields: [userEMode.transactionId],
+    references: [transaction.id],
+  }),
+  protocol: one(protocol, {
+    fields: [userEMode.protocolId],
+    references: [protocol.id],
+  }),
 }));
 
 export const priceSnapshotRelations = relations(priceSnapshot, ({ one }) => ({

src/riskEngine.ts

@@ -439,21 +439,28 @@ async function handleEModeSwitched({
   const protocolId = getAddressId(context.network.chainId, event.log.address);
   const eModeId = getEModeId(protocolId, event.args.newCategory);
   const chainId = BigInt(context.network.chainId);
-  const userEModeId = getUserEModeId(event.args.account, eModeId);
+  const transactionId = getTransactionId(event, context);
+  const userEModeId = getUserEModeId(event.args.account, protocolId);
 
   const params = {
     chainId,
     eModeId,
+    transactionId,
     userId: event.args.account,
+    protocolId,
   };
 
-  await context.db
-    .insert(userEMode)
-    .values({
-      id: userEModeId,
-      ...params,
-    })
-    .onConflictDoUpdate(params);
+  await Promise.all([
+    context.db
+      .insert(userEMode)
+      .values({
+        id: userEModeId,
+        ...params,
+      })
+      .onConflictDoUpdate(params),
+    createIfNotExistsTransaction(event, context),
+    createIfNotExistsUser(context, event.args.account),
+  ]);
 }
 
 async function handleNewCloseFactor({

src/utils/docs.ts

@@ -137,7 +137,7 @@ const docs = extendWithBaseDefinitions({
 
   ...generateTypeDocSet('userEMode', 'user e-mode settings', {
     'userEMode.id':
-      'Unique identifier for the user e-mode composed by <userAddress>-<eModeCategoryId>-<chainId>',
+      'Unique identifier for the user e-mode composed by <userAddress>-<riskEngineAddress>-<chainId>',
     'userEMode.chainId': 'The blockchain network identifier',
     'userEMode.userId': 'User identifier',
     'userEMode.eModeId': 'E-mode identifier',

src/utils/id.ts

@@ -36,6 +36,7 @@ export function getPTokenEModeId(pTokenId: string, eModeId: string) {
   return `${pTokenAddress}-${eModeId}`;
 }
 
-export function getUserEModeId(userAddress: string, eModeId: string) {
-  return `${userAddress}-${eModeId}`;
+export function getUserEModeId(userAddress: string, protocolId: string) {
+  // With that we only saves the current eMode for each protocol
+  return `${userAddress}-${protocolId}`;
 }

tables.schema.ts

@@ -152,6 +152,8 @@ export const userEMode = onchainTable(
     chainId: t.bigint().notNull(),
     userId: t.text().notNull(),
     eModeId: t.text().notNull(),
+    transactionId: t.text().notNull(),
+    protocolId: t.text().notNull(),
   }),
   table => ({
     userIdx: index().on(table.userId),