fix authorizer, add long ttl and change schema name

jotacemarin · Sep 8, 2024 · dd1b401 · dd1b401
1 parent baba12c
commit dd1b401
Show file tree

Hide file tree

Showing 9 changed files with 33 additions and 14 deletions.
diff --git a/src/functions/telegram_authorizer/index.ts b/src/functions/telegram_authorizer/index.ts
@@ -33,6 +33,7 @@ const extractToken = (authorizationToken: string) => {
   const [_, token] = authorizationToken?.split(" ");
   const cleanToken = atob(token);
   const [clientId, clientSecret] = cleanToken?.split(":");
+  console.log("extractToken", `${clientId}, ${clientSecret}`);
   return { clientId, clientSecret };
 };
 

diff --git a/src/functions/telegram_get_messages/function.yml b/src/functions/telegram_get_messages/function.yml
@@ -8,4 +8,6 @@ events:
       path: /${self:custom.secrets.service.name}-${self:custom.secrets.provider.stage}/telegram/get-messages
       method: GET
       cors: true
-      authorizer: telegramAuthorizer
+      authorizer:
+        name: telegramAuthorizer
+        resultTtlInSeconds: 0
diff --git a/src/functions/telegram_get_messages/index.ts b/src/functions/telegram_get_messages/index.ts
@@ -1,10 +1,16 @@
 import { APIGatewayEvent, Callback, Context } from "aws-lambda";
-import { OK } from "http-status";
+import { BAD_REQUEST, OK } from "http-status";
 import { ChatMessageDao } from "../../lib/dao";
 
-const execute = async (): Promise<any> => {
+const execute = async (event: APIGatewayEvent): Promise<any> => {
+  if (!event?.queryStringParameters?.chat_id) {
+    return { statusCode: BAD_REQUEST };
+  }
+
+  const chatId = Number(event.queryStringParameters.chat_id);
+
   await ChatMessageDao.initInstance();
-  const messages = await ChatMessageDao.getAll();
+  const messages = await ChatMessageDao.getAll(chatId);
 
   return { statusCode: OK, body: JSON.stringify(messages) };
 };
@@ -16,7 +22,7 @@ export const telegramGetMessages = async (
 ): Promise<void> => {
   context.callbackWaitsForEmptyEventLoop = false;
 
-  const response = await execute();
+  const response = await execute(event);
 
   return callback(null, response);
 };
diff --git a/src/functions/telegram_send_message/function.yml b/src/functions/telegram_send_message/function.yml
@@ -8,4 +8,6 @@ events:
       path: /${self:custom.secrets.service.name}-${self:custom.secrets.provider.stage}/telegram/send-message
       method: POST
       cors: true
-      authorizer: telegramAuthorizer
+      authorizer:
+        name: telegramAuthorizer
+        resultTtlInSeconds: 0
diff --git a/src/functions/telegram_send_photo/function.yml b/src/functions/telegram_send_photo/function.yml
@@ -8,4 +8,6 @@ events:
       path: /${self:custom.secrets.service.name}-${self:custom.secrets.provider.stage}/telegram/send-photo
       method: POST
       cors: true
-      authorizer: telegramAuthorizer
+      authorizer:
+        name: telegramAuthorizer
+        resultTtlInSeconds: 0
diff --git a/src/functions/telegram_send_video/function.yml b/src/functions/telegram_send_video/function.yml
@@ -8,4 +8,6 @@ events:
       path: /${self:custom.secrets.service.name}-${self:custom.secrets.provider.stage}/telegram/send-video
       method: POST
       cors: true
-      authorizer: telegramAuthorizer
+      authorizer:
+        name: telegramAuthorizer
+        resultTtlInSeconds: 0
diff --git a/src/functions/telegram_set_webhook/function.yml b/src/functions/telegram_set_webhook/function.yml
@@ -8,4 +8,6 @@ events:
       path: /${self:custom.secrets.service.name}-${self:custom.secrets.provider.stage}/telegram/set-webhook
       method: POST
       cors: true
-      authorizer: telegramAuthorizer
+      authorizer:
+        name: telegramAuthorizer
+        resultTtlInSeconds: 0
diff --git a/src/lib/dao/chatMessageDao.ts b/src/lib/dao/chatMessageDao.ts
@@ -3,7 +3,7 @@ import { ChatMessage } from "../models";
 import { MongodbService } from "../services";
 
 export class ChatMessageDao {
-  private static schemaName: string = "chatMessage";
+  private static schemaName: string = "chat_message";
   private static chatMessageSchema: Schema<ChatMessage>;
   public static chatMessageModel: Model<ChatMessage>;
 
@@ -15,7 +15,7 @@ export class ChatMessageDao {
     if (!ChatMessageDao.chatMessageSchema) {
       ChatMessageDao.chatMessageSchema = new Schema(
         {
-          message: { type: Schema.Types.Mixed, required: true },
+          telegramMessage: { type: Schema.Types.Mixed, required: true },
           expireAt: { type: Schema.Types.Date, expires: 43200 },
         },
         {
@@ -39,8 +39,10 @@ export class ChatMessageDao {
     return ChatMessageDao.chatMessageModel.create(chatMessage);
   }
 
-  public static async getAll(): Promise<Array<ChatMessage>> {
-    const chatMessages = await ChatMessageDao.chatMessageModel.find({}).exec();
+  public static async getAll(chatId: Number): Promise<Array<ChatMessage>> {
+    const chatMessages = await ChatMessageDao.chatMessageModel
+      .find({ telegramMessage: { message: { chat: { id: chatId } } } })
+      .exec();
     if (!chatMessages || !chatMessages?.length) {
       return [];
     }

diff --git a/src/lib/models/botnorrea.ts b/src/lib/models/botnorrea.ts
@@ -39,7 +39,7 @@ export interface Command {
 
 export interface ChatMessage {
   _id?: ID | string;
-  message: UpdateTg;
+  telegramMessage: UpdateTg;
   expireAt?: Date | AtedAt | string;
   createdAt?: AtedAt | string;
   updatedAt?: AtedAt | string;