serverless/lib/plugins/slstats/slstats.js

Serverless Framework - Build web, mobile and IoT applications with serverless architectures using AWS Lambda, Azure Functions, Google CloudFunctions & more

'use strict';

const userStats = require('../../utils/userStats');
const config = require('../../utils/config');
const BbPromise = require('bluebird');

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

    this.commands = {
      slstats: {
        usage: 'Enable or disable stats',
        configDependent: true,
        lifecycleEvents: ['slstats'],
        options: {
          enable: {
            usage: 'Enable stats ("--enable")',
            shortcut: 'e',
          },
          disable: {
            usage: 'Disable stats ("--disable")',
            shortcut: 'd',
          },
        },
      },
    };

    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;
    const data = { force: true };
    if (enableStats) {
      return userStats
        .track('user_enabledTracking', data)
        .then(() => {
          // set .serverlessrc config
          config.set('trackingDisabled', false);
          this.serverless.cli.log('Stats successfully enabled');
        })
        .catch(error => {
          const message = error;
          return BbPromise.reject(`Enabling / Disabling of statistics failed: ${message}`);
        });
    } else if (disabledStats) {
      return userStats
        .track('user_disabledTracking', data)
        .then(() => {
          // set .serverlessrc config
          config.set('trackingDisabled', true);
          this.serverless.cli.log('Stats successfully disabled');
        })
        .catch(error => {
          const message = error;
          return BbPromise.reject(`Enabling / Disabling of statistics failed: ${message}`);
        });
    }
    return BbPromise.resolve();
  }
}

module.exports = SlStats;