serverless-ephemeral/src/lib/packager/Factory.js

Serverless Framework plugin to bundles stateless zipped dependencies into the deployable Lambda artifact

const TensorFlow = require('./tensorflow');
const CustomPackager = require('./custom');

class PackagerFactory {
    /**
     * Picks a packager depending on the options provided.
     * @param {Object} serverless - Serverless object
     * @param {Object} ephemeral - Ephemeral object
     * @param {Object} [options]
     * @param {string} [options.name] - The packager name
     * @return {Object} The packager class
     */
    static build (serverless, ephemeral, options = {}) {
        options.name = options.name || '__custom';

        switch (options.name) {
            case 'tensorflow':
                return new TensorFlow(serverless, ephemeral, options);
            case '__custom':
            default:
                return new CustomPackager(serverless, ephemeral, options);
        }
    }
}

module.exports = PackagerFactory;