fix: response of api

aws-samples · Oct 14, 2024 · d6f4f1c · d6f4f1c
1 parent f61aec0
commit d6f4f1c
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 16 deletions.
diff --git a/packages/cdk/lambda/createUseCase.ts b/packages/cdk/lambda/createUseCase.ts
@@ -11,17 +11,15 @@ export const handler = async (
       event.requestContext.authorizer!.claims['cognito:username'];
     const title = req.title;
     const promptTemplate = req.promptTemplate;
-    const useCaseId = await createUseCase(userId, title, promptTemplate);
+    const useCaseIdRes = await createUseCase(userId, title, promptTemplate);
 
     return {
       statusCode: 200,
       headers: {
         'Content-Type': 'application/json',
         'Access-Control-Allow-Origin': '*',
       },
-      body: JSON.stringify({
-        useCaseId,
-      }),
+      body: JSON.stringify(useCaseIdRes),
     };
   } catch (error) {
     console.log(error);

diff --git a/packages/cdk/lambda/listFavoriteUseCases.ts b/packages/cdk/lambda/listFavoriteUseCases.ts
@@ -7,15 +7,15 @@ export const handler = async (
   try {
     const userId: string =
       event.requestContext.authorizer!.claims['cognito:username'];
-    const favoriteUseCases = await listFavoriteUseCases(userId);
+    const favoriteUseCasesRes = await listFavoriteUseCases(userId);
 
     return {
       statusCode: 200,
       headers: {
         'Content-Type': 'application/json',
         'Access-Control-Allow-Origin': '*',
       },
-      body: JSON.stringify({ favoriteUseCases }),
+      body: JSON.stringify(favoriteUseCasesRes),
     };
   } catch (error) {
     console.log(error);

diff --git a/packages/cdk/lambda/listUseCases.ts b/packages/cdk/lambda/listUseCases.ts
@@ -7,15 +7,15 @@ export const handler = async (
   try {
     const userId: string =
       event.requestContext.authorizer!.claims['cognito:username'];
-    const useCases = await listUseCases(userId);
+    const useCasesRes = await listUseCases(userId);
 
     return {
       statusCode: 200,
       headers: {
         'Content-Type': 'application/json',
         'Access-Control-Allow-Origin': '*',
       },
-      body: JSON.stringify({ useCases }),
+      body: JSON.stringify(useCasesRes),
     };
   } catch (error) {
     console.log(error);

diff --git a/packages/cdk/lambda/repository.ts b/packages/cdk/lambda/repository.ts
@@ -726,9 +726,7 @@ export const toggleFavorite = async (
         },
       })
     );
-    return {
-      isFavorite: false,
-    };
+    return { isFavorite: false };
   } else {
     await dynamoDbDocument.send(
       new PutCommand({

diff --git a/packages/cdk/lambda/toggleFavorite.ts b/packages/cdk/lambda/toggleFavorite.ts
@@ -1,5 +1,5 @@
 import { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
-import { ToggleFavoriteRequest } from 'generative-ai-use-cases-jp';
+import { IsFavorite, ToggleFavoriteRequest } from 'generative-ai-use-cases-jp';
 import { toggleFavorite } from './repository';
 
 export const handler = async (
@@ -12,15 +12,19 @@ export const handler = async (
     const req: ToggleFavoriteRequest = JSON.parse(event.body!);
     const ownerUserId = req.ownerUserId;
 
-    await toggleFavorite(userId, useCaseId, ownerUserId);
+    const isFavoriteRes: IsFavorite = await toggleFavorite(
+      userId,
+      useCaseId,
+      ownerUserId
+    );
 
     return {
       statusCode: 200,
       headers: {
         'Content-Type': 'application/json',
         'Access-Control-Allow-Origin': '*',
       },
-      body: '',
+      body: JSON.stringify(isFavoriteRes),
     };
   } catch (error) {
     console.log(error);

diff --git a/packages/cdk/lambda/toggleShared.ts b/packages/cdk/lambda/toggleShared.ts
@@ -24,7 +24,7 @@ export const handler = async (
     }
 
     const hasShared = useCase.hasShared;
-    const newHasShared: HasShared = await toggleShared(
+    const newHasSharedRes: HasShared = await toggleShared(
       userId,
       useCaseId,
       hasShared
@@ -36,7 +36,7 @@ export const handler = async (
         'Content-Type': 'application/json',
         'Access-Control-Allow-Origin': '*',
       },
-      body: JSON.stringify({ newHasShared }),
+      body: JSON.stringify(newHasSharedRes),
     };
   } catch (error) {
     console.log(error);