Accept regex as stages

README.md

@@ -91,9 +91,10 @@ Docker setup:
 ```yml
 custom:
   dynamodb:
-  # If you only want to use DynamoDB Local in some stages, declare them here
+  # If you only want to use DynamoDB Local in some stages, declare them here as string or stringified regex
     stages:
       - dev
+      - 'dev-\\d+'
     start:
       docker: true
       port: 8000

index.js

@@ -5,6 +5,8 @@ const AWS = require("aws-sdk");
 const dynamodbLocal = require("dynamodb-localhost");
 const seeder = require("./src/seeder");
 const path = require('path');
+const matchStage = require('./src/util');
+const { match } = require("assert");
 
 class ServerlessDynamodbLocal {
     constructor(serverless, options) {
@@ -168,8 +170,8 @@ class ServerlessDynamodbLocal {
      * @return {Boolean} if the handler can run for the provided stage
      */
     shouldExecute() {
-      if (this.config.stages && this.config.stages.includes(this.stage)) {
-        return true;
+      if (this.config.stages) {
+        return this.config.stages.some((stage) => matchStage(this.stage, stage));
       }
       return false;
     }

src/util.js

@@ -0,0 +1,6 @@
+function matchStage(currentStage, stage) {
+  const matches = currentStage.match(new RegExp(stage));
+  return !!(matches && matches.length > 0);
+}
+
+module.exports = matchStage;

test/testUtil.js

@@ -0,0 +1,10 @@
+const { expect } = require('chai');
+const matchStage = require('../src/util');
+
+describe('Test matching stages', () => {
+  it('should match regex', () => {
+    expect(matchStage('dev-777', 'dev-\\d+')).to.eq(true);
+    expect(matchStage('dev', 'dev-\\d+')).to.eq(false);
+    expect(matchStage('dev', 'dev')).to.eq(true);
+  });  
+})