@aws-lambda-powertools/parser/lib/esm/envelopes/kinesis.js

The parser package for the Powertools for AWS Lambda (TypeScript) library.

import { ZodError } from 'zod';
import { ParseError } from '../errors.js';
import { KinesisDataStreamSchema } from '../schemas/kinesis.js';
import { envelopeDiscriminator } from './envelope.js';
/**
 * Kinesis Data Stream Envelope to extract array of Records
 *
 * The record's data parameter is a base64 encoded string which is parsed into a bytes array,
 * though it can also be a JSON encoded string.
 * Regardless of its type it'll be parsed into a BaseModel object.
 *
 * Note: Records will be parsed the same way so if model is str,
 * all items in the list will be parsed as str and not as JSON (and vice versa)
 */
export const KinesisEnvelope = {
    /**
     * This is a discriminator to differentiate whether an envelope returns an array or an object
     * @hidden
     */
    [envelopeDiscriminator]: 'array',
    parse(data, schema) {
        let parsedEnvelope;
        try {
            parsedEnvelope = KinesisDataStreamSchema.parse(data);
        }
        catch (error) {
            throw new ParseError('Failed to parse Kinesis Data Stream envelope', {
                cause: error,
            });
        }
        return parsedEnvelope.Records.map((record, recordIndex) => {
            let parsedRecord;
            try {
                parsedRecord = schema.parse(record.kinesis.data);
            }
            catch (error) {
                throw new ParseError(`Failed to parse Kinesis Data Stream record at index ${recordIndex}`, {
                    cause: new ZodError(error.issues.map((issue) => ({
                        ...issue,
                        path: [
                            'Records',
                            recordIndex,
                            'kinesis',
                            'data',
                            ...issue.path,
                        ],
                    }))),
                });
            }
            return parsedRecord;
        });
    },
    safeParse(data, schema) {
        const parsedEnvelope = KinesisDataStreamSchema.safeParse(data);
        if (!parsedEnvelope.success) {
            return {
                success: false,
                error: new ParseError('Failed to parse Kinesis Data Stream envelope', {
                    cause: parsedEnvelope.error,
                }),
                originalEvent: data,
            };
        }
        const result = parsedEnvelope.data.Records.reduce((acc, record, index) => {
            const parsedRecord = schema.safeParse(record.kinesis.data);
            if (!parsedRecord.success) {
                const issues = parsedRecord.error.issues.map((issue) => ({
                    ...issue,
                    path: ['Records', index, 'kinesis', 'data', ...issue.path],
                }));
                acc.success = false;
                acc.errors[index] = { issues };
                return acc;
            }
            acc.records.push(parsedRecord.data);
            return acc;
        }, { success: true, records: [], errors: {} });
        if (result.success) {
            return { success: true, data: result.records };
        }
        const errorMessage = Object.keys(result.errors).length > 1
            ? `Failed to parse Kinesis Data Stream records at indexes ${Object.keys(result.errors).join(', ')}`
            : `Failed to parse Kinesis Data Stream record at index ${Object.keys(result.errors)[0]}`;
        return {
            success: false,
            error: new ParseError(errorMessage, {
                cause: new ZodError(Object.values(result.errors).flatMap((error) => error.issues)),
            }),
            originalEvent: data,
        };
    },
};