Redirecting Traffic Based on Simple Conditions

redirect-based-on-conditions.js

function handler(event) {
    var request = event.request;
    var supportedCountries = ['de', 'it', 'fr'];

    if (request.uri.substr(3, 1) != '/') {
        var headers = request.headers;
        var nextUri;

        if (headers['cloudfront-viewer-country']) {
            var countryCode = headers['cloudfront-viewer-country'].value.toLowerCase();
            if (supportedCountries.includes(countryCode)) {
                nextUri = '/' + countryCode + request.uri;
            }
        }

        if (!nextUri) {
            nextUri = '/en' + request.uri; 
        }

        return {
            statusCode: 302,
            statusDescription: 'Found',
            headers: {
                location: { value: newUri }
            }
        }
    }

    return request;
}