AWS Serverless
Intake Automation
Platform

import {
  TextractClient,
  StartDocumentTextDetectionCommand
} from "@aws-sdk/client-textract";

const textract = new TextractClient({});

export const handler = async (event) => {
  console.log("Lambda #1 triggered:", JSON.stringify(event));

  const record = event.Records[0];
  const bucket = record.s3.bucket.name;
  const key = decodeURIComponent(record.s3.object.key.replace(/\+/g, " "));

  const params = {
    DocumentLocation: {
      S3Object: {
        Bucket: bucket,
        Name: key
      }
    },
    NotificationChannel: {
      SNSTopicArn: process.env.SNS_TOPIC_ARN,
      RoleArn: process.env.TEXTRACT_ROLE_ARN
    },
    JobTag: key
  };

  try {
    const response = await textract.send(
      new StartDocumentTextDetectionCommand(params)
    );
    console.log("Textract job started. JobId:", response.JobId);
    return { statusCode: 200, body: "Job started successfully" };
  } catch (err) {
    console.error("Error starting Textract job:", err);
    throw err;
  }
};

import {
  TextractClient,
  GetDocumentTextDetectionCommand
} from "@aws-sdk/client-textract";

import {
  DynamoDBClient,
  PutItemCommand
} from "@aws-sdk/client-dynamodb";

const textract = new TextractClient({});
const dynamodb = new DynamoDBClient({});

const getAllBlocks = async (jobId) => {
  let blocks = [];
  let nextToken = null;

  do {
    const params = { JobId: jobId };
    if (nextToken) params.NextToken = nextToken;

    const response = await textract.send(
      new GetDocumentTextDetectionCommand(params)
    );

    blocks = blocks.concat(response.Blocks || []);
    nextToken = response.NextToken || null;
  } while (nextToken);

  return blocks;
};

export const handler = async (event) => {
  console.log("Lambda #2 triggered:", JSON.stringify(event));

  const snsMessage = JSON.parse(event.Records[0].Sns.Message);
  const jobId = snsMessage.JobId;
  const status = snsMessage.Status;
  const fileName = snsMessage.JobTag;

  console.log(`Job ID: ${jobId} | Status: ${status} | File: ${fileName}`);

  if (status !== "SUCCEEDED") {
    console.error("Textract job did not succeed. Status:", status);
    return { statusCode: 400, body: "Textract job failed" };
  }

  try {
    const blocks = await getAllBlocks(jobId);

    const lines = blocks
      .filter(b => b.BlockType === "LINE")
      .map(b => b.Text)
      .join("\n");

    console.log(`Extracted ${lines.split("\n").length} lines of text`);

    await dynamodb.send(new PutItemCommand({
      TableName: process.env.DYNAMODB_TABLE,
      Item: {
        FileName: { S: fileName },
        ExtractedText: { S: lines },
        JobId: { S: jobId },
        ProcessedAt: { S: new Date().toISOString() }
      }
    }));

    console.log("Successfully saved to DynamoDB");
    return { statusCode: 200, body: "Processing complete" };
  } catch (err) {
    console.error("Error processing results:", err);
    throw err;
  }
};