serverless/lib/plugins/slstats.js

'use strict';

const config = require('@serverless/utils/config');
const cliCommandsSchema = require('../cli/commands-schema');

class SlStats {
  constructor(serverless, options) {
    this.serverless = serverless;
    this.options = options;

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

    this.hooks = {
      'slstats:slstats': this.toggleStats.bind(this),
    };
  }

  toggleStats() {
    const enableStats = this.options.enable && !this.options.disable;
    const disabledStats = this.options.disable && !this.options.enable;
    if (enableStats) {
      try {
        // set .serverlessrc config
        config.set('trackingDisabled', false);
        this.serverless.cli.log('Stats successfully enabled');
      } catch (error) {
        const message = error;
        throw new Error(`Enabling / Disabling of statistics failed: ${message}`);
      }
    } else if (disabledStats) {
      try {
        // set .serverlessrc config
        config.set('trackingDisabled', true);
        this.serverless.cli.log('Stats successfully disabled');
      } catch (error) {
        const message = error;
        throw new Error(`Enabling / Disabling of statistics failed: ${message}`);
      }
    }
  }
}

module.exports = SlStats;