index.js

'use strict'
const _ = require('lodash')
const path = require('path')
const createServer = require('./lib/server')

class ServerlessAppSyncPlugin {
  constructor (serverless, options) {
    this.serverless = serverless
    this.servicePath = serverless.config.servicePath
    this.serverlessLog = (log) => serverless.cli.log(log, 'appSync')
    this.options = options

    this.commands = {
      'appsync-offline': {
        usage: 'Simulates local AppSync',
        commands: {
          start: {
            lifecycleEvents: ['startHandler'],
            usage: 'Simulates local AppSync',
            options: {
              port: {
                shortcut: 'p',
                usage:
                  'The port number that appsync will use to provide the graphql access point. If you do not specify this option, the default port is dynamic'
              },
              elasticEndpoint: {
                shortcut: 'e',
                usage:
                  'The endpoint for Elasticseaarch. If you do not specify this option, the default endpoint is http://localhost:9200'
              },
              dynamoDbPort: {
                shortcut: 'd',
                usage:
                  'The port number that DynamoDB will use to communicate with your application. If you do not specify this option, the default port is dynamic'
              },
              inMemory: {
                shortcut: 'i',
                usage:
                  'DynamoDB; will run in memory, instead of using a database file. When you stop DynamoDB;, none of the data will be saved. Note that you cannot specify both -dbPath and -inMemory at once.'
              },
              dbPath: {
                shortcut: 'b',
                usage:
                  'The directory where DynamoDB will write its database file. If you do not specify this option, the file will be written to the current directory. Note that you cannot specify both -dbPath and -inMemory at once. For the path, current working directory is <projectroot>/node_modules/serverless-dynamodb-local/dynamob. For example to create <projectroot>/node_modules/serverless-dynamodb-local/dynamob/<mypath> you should specify -d <mypath>/ or --dbPath <mypath>/ with a forwardslash at the end.'
              },
              sharedDb: {
                shortcut: 'h',
                usage:
                  'DynamoDB will use a single database file, instead of using separate files for each credential and region. If you specify -sharedDb, all DynamoDB clients will interact with the same set of tables regardless of their region and credential configuration.'
              },
              delayTransientStatuses: {
                shortcut: 't',
                usage:
                  'Causes DynamoDB to introduce delays for certain operations. DynamoDB can perform some tasks almost instantaneously, such as create/update/delete operations on tables and indexes; however, the actual DynamoDB service requires more time for these tasks. Setting this parameter helps DynamoDB simulate the behavior of the Amazon DynamoDB web service more closely. (Currently, this parameter introduces delays only for global secondary indexes that are in either CREATING or DELETING status.'
              },
              optimizeDbBeforeStartup: {
                shortcut: 'o',
                usage:
                  'Optimizes the underlying database tables before starting up DynamoDB on your computer. You must also specify -dbPath when you use this parameter.'
              },
              preferredSchema: {
                shortcut: 'a',
                usage: 'Schema name to use for multiple schema handling.'
              },
              bootstrapScript: {
                shortcut: 'x',
                usage:
                  'Path to bootstrap script - useful for launching mocks for example.'
              }
            }
          }
        }
      }
    }

    this.hooks = {
      'appsync-offline:start:startHandler': this.startStandaloneHandler.bind(
        this
      ),
      'before:offline:start:init': this.startHandler.bind(this),
      'before:offline:start:end': this.endHandler.bind(this)
    }
  }

  get port () {
    const config = (this.service.custom && this.service.custom.dynamodb) || {}
    const port = _.get(config, 'start.port', null)
    return port
  }

  async startHandler (isStandalone = false) {
    this._setOptions()
    let dynamodb = null

    try {
      if (!this.options.dynamodb.client.endpoint) { throw new Error('Provide a DynamoDB endpoint') }

      const { DynamoDB } = require('aws-sdk')
      dynamodb = new DynamoDB(this.options.dynamodb.client)
      this.serverlessLog('Using existing DynamoDB instance')
      // } else {
      //   // start the dynamodb emulator
      //   const dynamoEmulator = require('@conduitvc/dynamodb-emulator')
      //   this.emulator = await dynamoEmulator.launch(
      //     this.options.dynamodb.server
      //   )
      //   dynamodb = dynamoEmulator.getClient(
      //     this.emulator,
      //     this.options.dynamodb.client
      //   )
      //   this.serverlessLog('dynamoDB started: ' + dynamodb.endpoint.href)
      //   //this.serverlessLog(JSON.stringify( dynamodb))
      // }

      const port = this.options.port

      const server = await createServer({
        serverless: this.serverless,
        selectApi: this.options.schema,
        bootstrapScript: this.options.bootstrap,
        port,
        dynamodb,
        elastic: this.options.elastic || {}
      })

      this.serverlessLog('AppSync started: ' + server.url)
      if (!isStandalone) this._listenForTermination()
      return server
    } catch (err) {
      console.log(err)
      this.serverlessLog('ERROR: ' + err)
    }
  }

  async startStandaloneHandler () {
    this.serverlessLog('AppSync Standalone')
    return Promise.resolve(this.startHandler(true)).then(() =>
      this._listenForTermination()
    )
  }

  endHandler () {
    if (this.emulator) {
      // DynamoDB only needs stopping if we actually started it. If an external
      // connection was specified then this.emulator will be undefined.
      this.serverlessLog('AppSync offline - stopping graphql and dynamoDB')
      this.emulator.terminate().then(() => {
        process.exit(0)
      })
    } else {
      this.serverlessLog('AppSync offline - stopping graphql')
      process.exit(0)
    }
  }

  _setOptions () {
    // Merge the different sources of values for this.options
    // Precedence is: command line options, YAML options, defaults.
    // this.serverlessLog(JSON.stringify(this.options));
    const defaultOpts = {
      port: null,
      elastic: {
        endpoint: 'http://localhost:9200'
      },
      dynamodb: {
        client: {
          region: 'localhost',
          accessKeyId: 'MOCK_ACCESS_KEY_ID',
          secretAccessKey: 'MOCK_SECRET_ACCESS_KEY'
        },
        server: {
          port: null,
          dbPath: path.join(this.serverless.config.servicePath, '.dynamodb'),
          inMemory: false
        }
      }
    }

    const appSyncOfflineOptions = (this.serverless.service.custom || {}).appSyncOffline

    this.options = _.merge(
      {},
      defaultOpts,
      appSyncOfflineOptions,
      {
        bootstrap:
          this.options.bootstrapScript || appSyncOfflineOptions.bootstrap
      },
      {
        schema: this.options.preferredSchema || appSyncOfflineOptions.schema
      },
      {
        port: this.options.port,
        elastic: {
          endpoint: this.options.elasticEndpoint
        },
        dynamodb: {
          server: {
            port: this.options.dynamoDbPort,
            dbPath: this.options.dbPath,
            inMemory: this.options.inMemory,
            sharedDb: this.options.sharedDb,
            delayTransientStatuses: this.options.delayTransientStatuses,
            optimizeDbBeforeStartup: this.options.optimizeDbBeforeStartup
          }
        }
      }
    )
  }

  _listenForTermination () {
    // SIGINT will be usually sent when user presses ctrl+c
    const waitForSigInt = new Promise(resolve => {
      process.on('SIGINT', () => resolve('SIGINT'))
    })

    // SIGTERM is a default termination signal in many cases,
    // for example when "killing" a subprocess spawned in node
    // with child_process methods
    const waitForSigTerm = new Promise(resolve => {
      process.on('SIGTERM', () => resolve('SIGTERM'))
    })

    return Promise.race([waitForSigInt, waitForSigTerm]).then(command => {
      this.serverlessLog(
        `Got ${command} signal. Serverless Appsync Offline Halting...`
      )
      // _ensure_ we do not leave java processes lying around.
      this.endHandler()
    })
  }
}

module.exports = ServerlessAppSyncPlugin