All files / lib/attachment-archive rebuild-queue.ts

33.33% Statements 3/9
71.42% Branches 5/7
25% Functions 1/4
33.33% Lines 3/9

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42          1x           8x 8x                                                          
import { SendMessageCommand, SQSClient } from "@aws-sdk/client-sqs";
import { createHash } from "crypto";
 
import { AttachmentArchiveRebuildMessage } from "./types";
 
const sqsClient = new SQSClient({ region: process.env.region || process.env.AWS_REGION });
 
export function buildAttachmentArchiveMessageGroupId(
  packageId: string,
  options: { preferDraft?: boolean } = {},
): string {
  const hashInput = options.preferDraft ? `${packageId}:draft` : packageId;
  return `package-${createHash("sha256").update(hashInput).digest("hex")}`;
}
 
export function hasAttachmentArchiveRebuildQueueConfigured(): boolean {
  return Boolean(process.env.ATTACHMENT_ARCHIVE_REBUILD_QUEUE_URL);
}
 
export function getAttachmentArchiveRebuildQueueUrl(): string {
  const queueUrl = process.env.ATTACHMENT_ARCHIVE_REBUILD_QUEUE_URL;
  if (!queueUrl) {
    throw new Error("ATTACHMENT_ARCHIVE_REBUILD_QUEUE_URL must be defined");
  }
 
  return queueUrl;
}
 
export async function sendAttachmentArchiveRebuildRequest(
  message: AttachmentArchiveRebuildMessage,
) {
  await sqsClient.send(
    new SendMessageCommand({
      QueueUrl: getAttachmentArchiveRebuildQueueUrl(),
      MessageBody: JSON.stringify(message),
      MessageGroupId: buildAttachmentArchiveMessageGroupId(message.packageId, {
        preferDraft: message.preferDraft,
      }),
    }),
  );
}