Add the account to watch to DynamoDB

Author: yoanntlm

cdk/functions/receiver.ts

@@ -1,10 +1,19 @@
+import {
+  DynamoDBClient,
+  GetItemCommand,
+  PutItemCommand,
+} from "@aws-sdk/client-dynamodb";
+import { unmarshall } from "@aws-sdk/util-dynamodb";
 import axios from "axios";
 import { getFromEnv } from "../lib/cdk-stack";
 
+const ddb = new DynamoDBClient({});
+
 export const handler = async (event: any) => {
   const body = JSON.parse(event.body) as Body;
   const chatId = body.message.chat.id;
   const text = body.message.text;
+  const user = body.message.chat.username;
   console.log(text);
 
   const addressRegex = new RegExp(/^.*(0x[a-zA-Z0-9]{64}$)/g);
@@ -14,7 +23,7 @@ export const handler = async (event: any) => {
     await sendErrorMessagetoUser(chatId);
   } else {
     await sendConfirmationMessagetoUser(chatId, match[1]);
-    await addAddresstoDb(match[1]);
+    await addAddresstoDb(chatId, match[1], user);
 
     const addressToWatch = match[1];
     console.log(addressToWatch);
@@ -28,16 +37,7 @@ export const handler = async (event: any) => {
   };
 };
 
-interface Body {
-  message: {
-    chat: {
-      id: string;
-    };
-    text: string;
-  };
-}
-
-async function sendErrorMessagetoUser(chatId: string) {
+async function sendErrorMessagetoUser(chatId: number) {
   const errorMessage =
     "Address not found. Please include your Starknet address following the command /watch";
   const BOT_API_KEY = getFromEnv("BOT_API_KEY");
@@ -46,7 +46,7 @@ async function sendErrorMessagetoUser(chatId: string) {
   console.log(res);
 }
 
-async function sendConfirmationMessagetoUser(chatId: string, match: string) {
+async function sendConfirmationMessagetoUser(chatId: number, match: string) {
   const addressWatched = match;
   const confirmationMessage = `Watching address ${addressWatched} ...`;
   const BOT_API_KEY = getFromEnv("BOT_API_KEY");
@@ -55,7 +55,28 @@ async function sendConfirmationMessagetoUser(chatId: string, match: string) {
   console.log(res);
 }
 
-async function addAddresstoDb(match: string) {
-  const addressWatched = match;
-  return;
+async function addAddresstoDb(chatId: number, addressWatched: string, username: string) {
+
+  const params = {
+    TableName: "CdkStack-AccountsToWatch0A702DCE-Z6XSJDP9YT94",
+    Item: {
+      accountAddress: { S: addressWatched },
+      chatId: { N: chatId.toString() },
+      username: { S: username },
+    },
+  };
+
+  const res = await ddb.send(new PutItemCommand(params));
+
+  return null;
+}
+
+interface Body {
+  message: {
+    chat: {
+      id: number;
+      username: string;
+    };
+    text: string;
+  };
 }

cdk/lib/cdk-stack.ts

@@ -12,21 +12,25 @@ import * as apigateway from 'aws-cdk-lib/aws-apigateway';
 
 
 export class CdkStack extends cdk.Stack {
-  table: cdk.aws_dynamodb.Table;
   fetchLambda: cdk.aws_lambda_nodejs.NodejsFunction;
   sendLambda: cdk.aws_lambda_nodejs.NodejsFunction;
   receiveLambda: cdk.aws_lambda_nodejs.NodejsFunction;
   api: cdk.aws_apigateway.RestApi;
+  transactionStatusTable: cdk.aws_dynamodb.Table;
+  accountTable: cdk.aws_dynamodb.Table;
 
   constructor(scope: Construct, id: string, props?: cdk.StackProps) {
     super(scope, id, props);
 
-    this.setupTable();
+    this.setupTransactionTable();
+    this.setupAccountTable();
     this.setupFetcher();
     this.setupSender();
     this.setupReceiver();
-    this.table.grantFullAccess(this.fetchLambda);
     this.setupApiGateway();
+    this.transactionStatusTable.grantFullAccess(this.fetchLambda);
+    this.accountTable.grantFullAccess(this.receiveLambda);
+    this.accountTable.grantFullAccess(this.fetchLambda);
   }
 
   setupApiGateway() {
@@ -60,8 +64,8 @@ export class CdkStack extends cdk.Stack {
 
 
 
-  setupTable() {
-    this.table = new Table(this, "TransactionStatus", {
+  setupTransactionTable() {
+    this.transactionStatusTable = new Table(this, "TransactionStatus", {
       partitionKey: {
         name: "transactionHash",
         type: AttributeType.STRING,
@@ -71,6 +75,17 @@ export class CdkStack extends cdk.Stack {
     });
   }
 
+  setupAccountTable() {
+    this.accountTable = new Table(this, "AccountsToWatch", {
+      partitionKey: {
+        name: "accountAddress",
+        type: AttributeType.STRING,
+      },
+      billingMode: BillingMode.PAY_PER_REQUEST,
+      removalPolicy: RemovalPolicy.RETAIN,
+    });
+  }
+
   setupFetcher() {
     this.fetchLambda = new NodejsFunction(this, "Fetcher", {
       timeout: Duration.seconds(40),