feat: Add vpc configuration (#416)

Co-authored-by: Daniel Cottone <daniel.cottone@asurion.com>

README.md

@@ -146,6 +146,22 @@ provider:
   role: arn:aws:iam::123456789012:role/MyCustomRole
 ```
 
+#### vpc
+
+(Optional) VPC configuration for the log processor lambda to have.
+
+```yaml
+custom:
+  esLogs:
+    vpc: 
+      securityGroupIds:
+        - sg-123456789
+      subnetIds:
+        - subnet-123456789
+        - subnet-223456789
+        - subnet-323456789
+```
+
 #### xrayTracingPermissions
 
 (Optional) Adds AWS Xray writing permissions to the processor lambda. You will need these if you enable tracing for ApiGateway on your service. 

src/index.ts

@@ -260,7 +260,7 @@ class ServerlessEsLogsPlugin {
   }
 
   private addLogProcesser(): void {
-    const { index, endpoint, tags } = this.custom().esLogs;
+    const { index, endpoint, tags, vpc } = this.custom().esLogs;
     const tagsStringified = tags ? JSON.stringify(tags) : /* istanbul ignore next */ '';
     const dirPath = path.join(this.serverless.config.servicePath, this.logProcesserDir);
     const filePath = path.join(dirPath, 'index.js');
@@ -275,6 +275,7 @@ class ServerlessEsLogsPlugin {
         ES_INDEX_PREFIX: index,
         ES_TAGS: tagsStringified,
       },
+      vpc,
       events: [],
       handler,
       memorySize: 512,

src/interfaces.ts

@@ -1,6 +1,10 @@
 export interface IPluginOpts {
   domainArn?: string;
   index?: string;
+  vpc?: {
+      securityGroupIds: string[];
+      subnetIds: string[];
+  };
 }
 
 export interface IFormatterOpts {

test/unit/plugin.test.ts

@@ -133,6 +133,29 @@ describe('serverless-es-logs :: Plugin tests', () => {
           expect(serverless.service.functions).to.have.property('esLogsProcesser');
           expect(fs.existsSync(dirPath)).to.be.true;
         });
+
+        it('should add vpc config if specified', () => {
+          const vpc = {
+            securityGroupIds: ['securityGroup'],
+            subnetIds: ['subnet']
+          };
+          const opts = {
+            service: {
+              custom: {
+                esLogs: {
+                  endpoint: 'some_endpoint',
+                  index: 'some_index',
+                  vpc
+                },
+              },
+            },
+          };
+
+          serverless = new ServerlessBuilder(opts).build();
+          plugin = new ServerlessEsLogsPlugin(serverless, options);
+          plugin.hooks['after:package:initialize']();
+          expect(serverless.service.functions.esLogsProcesser).to.have.deep.property('vpc', vpc);
+        });
       });
     });