'use strict';

const BbPromise = require('bluebird');
const fs = require('fs');
const SDK = require('../');
const validate = require('../lib/validate');

// The Package plugin which is used to zip the service
const Package = require('../../package');

class AwsDeployFunction {
  constructor(serverless, options) {
    this.serverless = serverless;
    this.options = options || {};
    this.provider = 'aws';
    this.sdk = new SDK(serverless);

    this.pkg = new Package(this.serverless, this.options);

    Object.assign(this, validate);

    this.hooks = {
      'deploy:function:deploy': () => BbPromise.bind(this)
        .then(this.validate)
        .then(this.checkIfFunctionExists)
        .then(this.zipFunction)
        .then(this.deployFunction)
        .then(this.cleanup),
    };
  }

  checkIfFunctionExists() {
    // check if the function exists in the service
    this.options.functionObj = this.serverless.service.getFunction(this.options.function);

    // check if function exists on AWS
    const params = {
      FunctionName: this.options.functionObj.name,
    };

    this.sdk.request(
      'Lambda',
      'getFunction',
      params,
      this.options.stage, this.options.region
    ).catch(() => {
      const errorMessage = [
        `The function "${this.options.function}" you want to update is not yet deployed.`,
        ' Please run "serverless deploy" to deploy your service.',
        ' After that you can redeploy your services functions with the',
        ' "serverless deploy function" command.',
      ].join('');
      throw new this.serverless.classes
        .Error(errorMessage);
    });

    return BbPromise.resolve();
  }

  zipFunction() {
    // TODO: Update to zip individual function if we are packaging functions individually
    return this.pkg.packageFunction(this.options.functionObj.name);
  }

  deployFunction() {
    const data = fs.readFileSync(this.serverless.service.package.artifact);

    const params = {
      FunctionName: this.options.functionObj.name,
      ZipFile: data,
    };

    this.sdk.request(
      'Lambda',
      'updateFunctionCode',
      params,
      this.options.stage, this.options.region
    );

    this.serverless.cli.log(`Successfully deployed function "${this.options.function}"`);

    return BbPromise.resolve();
  }

  cleanup() {
    return this.pkg.cleanup();
  }
}

module.exports = AwsDeployFunction;