serverless/lib/plugins/invoke.js

import _ from 'lodash';
import cliCommandsSchema from '../cli/commands-schema.js';

class Invoke {
  constructor(serverless, options) {
    this.serverless = serverless;
    this.options = options || {};

    this.commands = {
      invoke: {
        ...cliCommandsSchema.get('invoke'),
        commands: {
          local: {
            ...cliCommandsSchema.get('invoke local'),
          },
        },
      },
    };

    this.hooks = {
      'initialize': () => {
        this.options = this.serverless.processedInput.options;
      },
      'invoke:local:loadEnvVars': async () => this.loadEnvVarsForLocal(),
      'after:invoke:invoke': async () => this.trackInvoke(),
      'after:invoke:local:invoke': async () => this.trackInvokeLocal(),
    };
  }

  trackInvoke() {
    return;
  }

  trackInvokeLocal() {
    return;
  }

  /**
   * Set environment variables for "invoke local" that are provider independent.
   */
  loadEnvVarsForLocal() {
    const defaultEnvVars = {
      IS_LOCAL: 'true',
    };

    _.merge(process.env, defaultEnvVars);

    // in some circumstances, setting these provider-independent environment variables is not enough
    // eg. in case of local 'docker' invocation, which relies on this module,
    // these provider-independent environment variables have to be propagated to the container
    this.serverless.service.provider.environment =
      this.serverless.service.provider.environment || {};
    const providerEnv = this.serverless.service.provider.environment;
    for (const [envVariableKey, envVariableValue] of Object.entries(defaultEnvVars)) {
      if (!Object.prototype.hasOwnProperty.call(providerEnv, envVariableKey)) {
        providerEnv[envVariableKey] = envVariableValue;
      }
    }

    // Turn zero or more --env options into an array
    //   ...then split --env NAME=value and put into process.env.
    [].concat(this.options.env || []).forEach((itm) => {
      const splitItm = itm.split(/=(.+)/);
      process.env[splitItm[0]] = splitItm[1] || '';
    });
  }
}

export default Invoke;