Rweb

src/packages/shared-types/action-types/withdrawPackage.ts

@@ -0,0 +1,21 @@
+import { z, ZodArray } from "zod";
+import { onemacAttachmentSchema } from "../onemac";
+
+export const withdrawPackageSchema = (attachmentArrayType: ZodArray<any>) =>
+  z.object({
+    id: z.string(),
+    additionalInformation: z
+      .string()
+      .max(4000, "This field may only be up to 4000 characters.")
+      .optional(),
+    attachments: z.object({
+      supportingDocumentation: attachmentArrayType.optional(),
+    }),
+  });
+
+export const withdrawPackageEventSchema = withdrawPackageSchema(
+  z.array(onemacAttachmentSchema)
+);
+export type WithdrawPackageEventSchema = z.infer<
+  typeof withdrawPackageEventSchema
+>;

src/packages/shared-types/actions.ts

@@ -1,5 +1,6 @@
 export enum Action {
   ENABLE_RAI_WITHDRAW = "enable-rai-withdraw",
+  WITHDRAW_PACKAGE = "withdraw-package",
   DISABLE_RAI_WITHDRAW = "disable-rai-withdraw",
   ISSUE_RAI = "issue-rai",
   RESPOND_TO_RAI = "respond-to-rai",

src/packages/shared-types/index.ts

@@ -9,3 +9,4 @@ export * from "./actions";
 export * from "./attachments";
 export * from "./authority";
 export * from "./action-types/withdraw-record";
+export * from "./action-types/withdrawPackage";

src/packages/shared-types/onemac.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import { s3ParseUrl } from "shared-utils/s3-url-parser";
 
-const onemacAttachmentSchema = z.object({
+export const onemacAttachmentSchema = z.object({
   s3Key: z.string().nullish(),
   filename: z.string(),
   title: z.string(),

src/services/api/handlers/action.ts

@@ -12,6 +12,7 @@ import {
   issueRai,
   respondToRai,
   toggleRaiResponseWithdraw,
+  withdrawPackage
 } from "./packageActions";
 
 export const handler = async (event: APIGatewayEvent) => {
@@ -53,6 +54,9 @@ export const handler = async (event: APIGatewayEvent) => {
 
     // Call package action
     switch (actionType) {
+      case Action.WITHDRAW_PACKAGE:
+        await withdrawPackage(body);
+        break;
       case Action.ISSUE_RAI:
         await issueRai(body);
         break;

src/services/api/handlers/getPackageActions.ts

@@ -1,7 +1,7 @@
 import { APIGatewayEvent } from "aws-lambda";
 import { Action, CognitoUserAttributes, ItemResult } from "shared-types";
 import {
   isCmsUser,
   getActiveRai,
   isCmsWriteUser,
   isStateUser,
@@ -53,6 +53,7 @@
       }
     }
   } else if (isStateUser(user)) {
+    actions.push(Action.WITHDRAW_PACKAGE);
     switch (result._source.seatoolStatus) {
       case SEATOOL_STATUS.PENDING_RAI:
         if (activeRai) {
@@ -67,7 +68,6 @@
 export const getPackageActions = async (event: APIGatewayEvent) => {
   const body = JSON.parse(event.body) as GetPackageActionsBody;
   try {
-    console.log(body);
     const result = await getPackage(body.id);
     const passedStateAuth = await isAuthorized(event, result._source.state);
     if (!passedStateAuth)
@@ -80,13 +80,11 @@
         statusCode: 404,
         body: { message: "No record found for the given id" },
       });
-
     const authDetails = getAuthDetails(event);
     const userAttr = await lookupUserAttributes(
       authDetails.userId,
       authDetails.poolId
     );
-
     return response({
       statusCode: 200,
       body: {
@@ -101,5 +99,4 @@
     });
   }
 };
-
 export const handler = getPackageActions;

src/services/api/handlers/packageActions.ts

@@ -12,7 +12,15 @@
   database: "SEA",
 };
 
-import { Action, raiSchema, RaiSchema } from "shared-types";
+import {
+  Action,
+  raiSchema,
+  RaiSchema,
+  OneMacSink,
+  WithdrawPackageEventSchema,
+  transformOnemac,
+  withdrawPackageEventSchema,
+} from "shared-types";
 import { produceMessage } from "../libs/kafka";
 import { response } from "../libs/handler";
 import { SEATOOL_STATUS } from "shared-types/statusHelper";
@@ -140,8 +148,53 @@
   console.log("heyo");
 }
 
-export async function withdrawPackage(id, timestamp) {
+export async function withdrawPackage(body: WithdrawPackageEventSchema) {
   console.log("State withdrawing a package.");
+  // Check incoming data
+  const result = withdrawPackageEventSchema.safeParse(body);
+  if (result.success === false) {
+    console.error(
+      "Withdraw Package event validation error. The following record failed to parse: ",
+      JSON.stringify(body),
+      "Because of the following Reason(s):",
+      result.error.message
+    );
+    return response({
+      statusCode: 400,
+      body: { message: "Withdraw Package event validation error" },
+    });
+  }
+  // Begin query (data is confirmed)
+  const pool = await sql.connect(config);
+  const transaction = new sql.Transaction(pool);
+  const query = `
+    UPDATE SEA.dbo.State_Plan
+      SET SPW_Status_ID = (Select SPW_Status_ID from SEA.dbo.SPW_Status where SPW_Status_DESC = '${SEATOOL_STATUS.WITHDRAWN}')
+      WHERE ID_Number = '${body.id}'
+  `;
+
+  try {
+    const txnResult = await transaction.request().query(query);
+    console.log(txnResult);
+    await produceMessage(
+      TOPIC_NAME,
+      body.id,
+      JSON.stringify({ ...result.data, actionType: Action.WITHDRAW_PACKAGE })
+    );
+    // Commit transaction
+    await transaction.commit();
+  } catch (err) {
+    // Rollback and log
+    await transaction.rollback();
+    console.error("Error executing query:", err);
+    return response({
+      statusCode: 500,
+      body: { message: err.message },
+    });
+  } finally {
+    // Close pool
+    await pool.close();
+  }
 }
 
 export async function toggleRaiResponseWithdraw(

src/services/api/handlers/search.ts

@@ -13,7 +13,6 @@ export const getSearchData = async (event: APIGatewayEvent) => {
     if (event.body) {
       query = JSON.parse(event.body);
     }
-
     query.query = query?.query || {};
     query.query.bool = query.query?.bool || {};
     query.query.bool.must = query.query.bool?.must || [];
@@ -30,7 +29,6 @@ export const getSearchData = async (event: APIGatewayEvent) => {
     console.log(JSON.stringify(query, null, 2));
 
     const results = await os.search(process.env.osDomain, "main", query);
-
     return response<unknown>({
       statusCode: 200,
       body: results,