@aws-amplify/storage/src/providers/s3/utils/client/utils/parsePayload.ts

Storage category of aws-amplify

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

import {
	ErrorParser,
	HttpResponse,
	parseMetadata,
} from '@aws-amplify/core/internals/aws-client-utils';

import { parser } from '../runtime';

/**
 * Factory creating a parser that parses the JS Error object from the XML
 * response payload.
 *
 * @param input Input object
 * @param input.noErrorWrapping Whether the error code and message are located
 *   directly in the root XML element, or in a nested `<Error>` element.
 *   See: https://smithy.io/2.0/aws/protocols/aws-restxml-protocol.html#restxml-errors
 *
 *   Default to false.
 *
 * @internal
 */
export const createXmlErrorParser =
	({
		noErrorWrapping = false,
	}: { noErrorWrapping?: boolean } = {}): ErrorParser =>
	async (response?: HttpResponse) => {
		if (!response || response.statusCode < 300) {
			return;
		}
		const { statusCode } = response;
		const body = await parseXmlBody(response);
		const errorLocation = noErrorWrapping ? body : body.Error;
		const code = errorLocation?.Code
			? (errorLocation.Code as string)
			: statusCode === 404
				? 'NotFound'
				: statusCode.toString();
		const message = errorLocation?.message ?? errorLocation?.Message ?? code;
		const error = new Error(message);

		return Object.assign(error, {
			name: code,
			$metadata: parseMetadata(response),
		});
	};

export const parseXmlBody = async (response: HttpResponse): Promise<any> => {
	if (!response.body) {
		// S3 can return 200 without a body indicating failure.
		throw new Error('S3 aborted request.');
	}
	const data = await response.body.text();
	if (data?.length > 0) {
		try {
			return parser.parse(data);
		} catch (error) {
			throw new Error(`Failed to parse XML response: ${error}`);
		}
	}

	return {};
};