Now uses stage in default&custom SSM key prefixes

Also uses a more appropriate lifecycle hook for updating SSM

Author: jlaitio

README.md

@@ -4,15 +4,15 @@ A simple plugin that, when deploying your API, updates SSM Parameter Store with
 
 ## Configuration
 
-The API git versions are updated into SSM using a specific key prefix, which by default is '/api-gateway/versions/'. If you want to supply a custom prefix, you can do so by putting the following configuration in your serverless config file:
+The API git versions are updated into SSM using a specific key prefix, which by default is '/api-gateway/<stage>/versions/'. If you want to supply a custom prefix, you can do so by putting the following configuration in your serverless config file:
 
 ```yaml
 custom:
   ssmApiGitVersion:
     ssmPrefix: '/my-custom/prefix/'
 ```
 
-Note that the plugin currently always uses the region supplied in the `provider` section of serverless configuration, regardless of the command line region parameter.
+The `<stage>` placeholder gets replaced with the stack stage.
 
 ## Usage
 

index.js

@@ -19,14 +19,15 @@ class ServerlessPlugin {
     this.commands = {};
 
     this.hooks = {
-      'deploy:finalize': this.updateGitDescriptionToSsm.bind(this)
+      'after:aws:deploy:deploy:updateStack': this.updateGitDescriptionToSsm.bind(this)
     };
   }
 
   updateGitDescriptionToSsm() {
     this.serverless.cli.log('SSM API git version: Acquiring git description...');
 
-    const region = this.serverless.service.provider.region;
+    const stage = this.options.stage;
+    const region = this.options.region;
 
     const SSM = new AWS.SSM({ region });
 
@@ -46,7 +47,11 @@ class ServerlessPlugin {
 
     return promisexec('git describe --tags')
       .then(value => {
-        const ssmPrefix = this.serverless.service.custom.ssmApiGitVersion.ssmPrefix || '/api-gateway/versions/';
+        const ssmPrefix = (this.serverless.service.custom
+        && this.serverless.service.custom.ssmApiGitVersion
+        && this.serverless.service.custom.ssmApiGitVersion.ssmPrefix)
+          ? this.serverless.service.custom.ssmApiGitVersion.ssmPrefix.replace(/<stage>/g, stage)
+          : `/api-gateway/${stage}/versions/`;
         const name = ssmPrefix + this.serverless.service.service;
 
         this.serverless.cli.log(`SSM API git version: Updating git description '${value}' to SSM with key '${name}' at region ${region}`);