914 [Feature] - "Add remaining CRUD endpoints to SST"

serverless/packages/functions/src/constants.ts

@@ -0,0 +1,9 @@
+
+export enum TrainStatus {
+    QUEUED,
+    STARTING,
+    UPLOADING,
+    TRAINING,
+    SUCCESS,
+    ERROR
+}

serverless/packages/functions/src/datasets/user/delete_userdata.ts

@@ -0,0 +1,10 @@
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import { S3Client, ListObjectsV2Command } from "@aws-sdk/client-s3";
+import parseJwt from "@dlp-sst-app/core/parseJwt";
+
+export const handler: APIGatewayProxyHandlerV2 = async (event) => {
+    return {
+        statusCode: 404,
+        body: JSON.stringify({ message: "Not Found" }),
+    };
+};

serverless/packages/functions/src/trainspace/create_trainspace.ts

@@ -0,0 +1,38 @@
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import parseJwt from "@dlp-sst-app/core/parseJwt";
+import TrainspaceData from './trainspace-data';
+import { v4 as uuidv4 } from 'uuid';
+import { DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { PutCommand, DynamoDBDocumentClient } from '@aws-sdk/lib-dynamodb';
+
+export const handler: APIGatewayProxyHandlerV2 = async (event) => {
+    if (
+        event &&
+        event.pathParameters &&
+        event.pathParameters.type &&
+        event.pathParameters.filename
+      ) {
+
+        const uid: string = parseJwt(event.headers.authorization ?? "")["user_id"];
+        const trainspace_id: string = uuidv4();
+        const trainspaceData: TrainspaceData = new TrainspaceData(trainspace_id, uid);
+
+        const client: DynamoDBClient = new DynamoDBClient({});
+        const docClient: DynamoDBDocumentClient = new DynamoDBDocumentClient.from(client);
+
+        const command: PutCommand = new PutCommand({
+            TableName: "trainspace",
+            Item: trainspaceData
+        });
+
+        const response = await docClient.send(command);
+        return {
+            statusCode: 200,
+            body: JSON.stringify({ message: "Successfully created a new trainspace."})
+        };
+      }
+    return {
+        statusCode: 404,
+        body: JSON.stringify({ message: "Not Found" }),
+    };
+};

serverless/packages/functions/src/trainspace/delete_trainspace.ts

@@ -0,0 +1,10 @@
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import { S3Client, ListObjectsV2Command } from "@aws-sdk/client-s3";
+import parseJwt from "@dlp-sst-app/core/parseJwt";
+
+export const handler: APIGatewayProxyHandlerV2 = async (event) => {
+    return {
+        statusCode: 404,
+        body: JSON.stringify({ message: "Not Found" }),
+    };
+};

serverless/packages/functions/src/trainspace/get_all_trainspace.ts

@@ -0,0 +1,10 @@
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import { S3Client, ListObjectsV2Command } from "@aws-sdk/client-s3";
+import parseJwt from "@dlp-sst-app/core/parseJwt";
+
+export const handler: APIGatewayProxyHandlerV2 = async (event) => {
+    return {
+        statusCode: 404,
+        body: JSON.stringify({ message: "Not Found" }),
+    };
+};

serverless/packages/functions/src/trainspace/get_trainspace.ts

@@ -0,0 +1,10 @@
+import { APIGatewayProxyHandlerV2 } from "aws-lambda";
+import { S3Client, ListObjectsV2Command } from "@aws-sdk/client-s3";
+import parseJwt from "@dlp-sst-app/core/parseJwt";
+
+export const handler: APIGatewayProxyHandlerV2 = async (event) => {
+    return {
+        statusCode: 404,
+        body: JSON.stringify({ message: "Not Found" }),
+    };
+};

serverless/packages/functions/src/trainspace/trainspace-data.ts

@@ -0,0 +1,20 @@
+import { TrainStatus } from '../constants';
+
+export default class TrainspaceData
+{
+    trainspace_id: string;
+    uid: string;
+    created: string = "";
+    data_source: string = "";
+    dataset_data: object = null;
+    name: string = "";
+    parameters_data: object = null;
+    review_data: string = "";
+    status: TrainStatus = TrainStatus.QUEUED;
+
+    constructor(trainspace_id: string, uid: string) {
+        this.trainspace_id = trainspace_id;
+        this.uid = uid;
+    }
+
+}

serverless/stacks/AppStack.ts

@@ -35,6 +35,16 @@ export function AppStack({ stack }: StackContext) {
         "packages/functions/src/datasets/user/all_of_type.handler",
       "GET /datasets/user/{type}/{filename}/columns":
         "packages/functions/src/datasets/user/columns.handler",
+      "DELETE /dataset/user/{type}/{filename}" :
+        "packages/functions/src/datasets/user/delete_userdata.handler",
+      "POST /trainspace":
+        "packages/functions/src/trainspace/create_trainspace.handler",
+      "GET /trainspace/{id}":
+        "packages/functions/src/trainspace/get_trainspace.handler",
+      "GET /trainspace":
+        "packages/functions/src/trainspace/get_all_trainspace.handler",
+      "DELETE /trainspace/{id}":
+        "packages/functions/src/trainspace/delete_trainspace.handler"
     },
   });