Add iopipe wrapper to package excludes (#10)

Closes #9

example/handler.py

@@ -1,5 +1,8 @@
 def handler(event, context):
-    if not hasattr(context, "iopipe") or not hasattr(context.iopipe, "mark"):
+    if not hasattr(context, "iopipe"):
+        raise Exception("No iopipe.")
+
+    if not hasattr(context.iopipe, "mark"):
         raise Exception("No plugins")
 
     return 200

example/serverless.yml

@@ -16,20 +16,45 @@ custom:
 functions:
   layer-nodejs610:
     handler: handler.handler
+    package:
+      exclude:
+        - ./**
+      include:
+        - handler.js
     runtime: nodejs6.10
 
   layer-nodejs810:
     handler: handler.handler
+    package:
+      exclude:
+        - ./**
+      include:
+        - handler.js
     runtime: nodejs8.10
 
   layer-python27:
     handler: handler.handler
+    package:
+      exclude:
+        - ./**
+      include:
+        - handler.py
     runtime: python2.7
 
   layer-python36:
     handler: handler.handler
+    package:
+      exclude:
+        - ./**
+      include:
+        - handler.py
     runtime: python3.6
 
   layer-python37:
     handler: handler.handler
+    package:
+      exclude:
+        - ./**
+      include:
+        - handler.py
     runtime: python3.7

src/index.ts

@@ -19,7 +19,9 @@ export default class IOpipeLayerPlugin {
     this.options = options;
 
     this.hooks = {
+      "after:deploy:function:packageFunction": this.cleanup.bind(this),
       "after:package:createDeploymentArtifacts": this.cleanup.bind(this),
+      "before:deploy:function:packageFunction": this.run.bind(this),
       "before:package:createDeploymentArtifacts": this.run.bind(this)
     };
   }
@@ -64,7 +66,8 @@ export default class IOpipeLayerPlugin {
       environment = {},
       handler,
       runtime = _.get(this.serverless.service, "provider.runtime"),
-      layers = []
+      layers = [],
+      package: pkg = {}
     } = funcDef;
 
     if (!this.config.token && !environment.IOPIPE_TOKEN) {
@@ -115,6 +118,7 @@ export default class IOpipeLayerPlugin {
     funcDef.environment = environment;
 
     funcDef.handler = this.getHandlerWrapper(runtime, handler);
+    funcDef.package = this.updatePackageExcludes(runtime, pkg);
   }
 
   private getLayerArn(runtime: string, region: string) {
@@ -160,6 +164,18 @@ export default class IOpipeLayerPlugin {
       fs.removeSync(helperPath);
     }
   }
+
+  private updatePackageExcludes(runtime: string, pkg: any) {
+    if (!runtime.match("nodejs")) {
+      return pkg;
+    }
+
+    const { exclude = [] } = pkg;
+    exclude.push("!iopipe_wrapper.js");
+    pkg.exclude = exclude;
+
+    return pkg;
+  }
 }
 
 module.exports = IOpipeLayerPlugin;