processPage.js

// /processPage.js

const extractor = require('./src/utils/event_extraction.util');
const referrer_detection = require('./src/utils/referrer_detection.util');
const { unmarshallNewImageEvent } = require('./src/utils/dynamo_stream.util');

const dynamoDBFactory = require('./src/dynamodb.factory');
const { SourceAttributionModel } = require('./src/models/SourceAttribution');

const dynamoDb = dynamoDBFactory();
const model = new SourceAttributionModel(dynamoDb);

module.exports.handler = async (event) => {
    const eventData = event.Records[0];

    if (eventData.eventName !== 'INSERT') {
        return;
    }

    const eventAddedUnmarshalled = unmarshallNewImageEvent(eventData);
    const extractedData = extractor(eventAddedUnmarshalled);
    const {referrer, href } = extractedData;

    const extraction = await referrer_detection(href, referrer);
    if (!extraction) {
        console.log('Skipping useless extraction', extractedData);
        return;
    }

    try {
        await model.store(eventAddedUnmarshalled, extraction);
    } catch (e) {
        console.log(e.message)
    }

    return eventData.eventName;
};