Merge pull request #8 from lambda-feedback/7-evaluation-function-authentication

API key authentication

Author: m-messer

.mockery.yaml

@@ -26,5 +26,4 @@ packages:
       inpackage: True
       dir: "{{.InterfaceDir}}"
     interfaces:
-      Dispatcher:
-# testonly: True
+      Dispatcher:

README.md

@@ -48,6 +48,10 @@ GLOBAL OPTIONS:
    --log-level value   set the log level. Options: debug, info, warn, error, panic, fatal. [$LOG_LEVEL]
    --version           print the version
 
+   auth
+
+   --auth-key value, -k value  the authentication key to use for incoming requests. [$AUTH_KEY]
+
    function
 
    --arg value, -a value [ --arg value, -a value ]  additional arguments for to the worker process. [$FUNCTION_ARGS]

app/app.go

@@ -1,13 +1,14 @@
 package app
 
 import (
+	"github.com/urfave/cli/v2"
+	"go.uber.org/fx"
+
 	"github.com/lambda-feedback/shimmy/config"
 	"github.com/lambda-feedback/shimmy/internal/shell"
 	"github.com/lambda-feedback/shimmy/runtime"
 	"github.com/lambda-feedback/shimmy/util/conf"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"github.com/urfave/cli/v2"
-	"go.uber.org/fx"
 )
 
 func New(ctx *cli.Context) (*shell.Shell, error) {

app/lambda/handler.go

@@ -7,9 +7,10 @@ import (
 
 	"github.com/aws/aws-lambda-go/lambda"
 	"github.com/awslabs/aws-lambda-go-api-proxy/httpadapter"
-	"github.com/lambda-feedback/shimmy/internal/server"
 	"go.uber.org/fx"
 	"go.uber.org/zap"
+
+	"github.com/lambda-feedback/shimmy/internal/server"
 )
 
 // LambdaHandlerParams represents the parameters required for

app/lambda/module.go

@@ -1,9 +1,10 @@
 package lambda
 
 import (
+	"go.uber.org/fx"
+
 	"github.com/lambda-feedback/shimmy/handler"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"go.uber.org/fx"
 )
 
 func Module(config Config) fx.Option {

app/standalone/module.go

@@ -1,10 +1,11 @@
 package standalone
 
 import (
+	"go.uber.org/fx"
+
 	"github.com/lambda-feedback/shimmy/handler"
 	"github.com/lambda-feedback/shimmy/internal/server"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"go.uber.org/fx"
 )
 
 func Module(config Config) fx.Option {

cmd/lambda.go

@@ -1,11 +1,12 @@
 package cmd
 
 import (
+	"github.com/urfave/cli/v2"
+
 	"github.com/lambda-feedback/shimmy/app"
 	"github.com/lambda-feedback/shimmy/app/lambda"
 	"github.com/lambda-feedback/shimmy/util/conf"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"github.com/urfave/cli/v2"
 )
 
 var (

cmd/root.go

@@ -6,11 +6,12 @@ import (
 	"os"
 	"time"
 
+	"github.com/urfave/cli/v2"
+	"go.uber.org/zap"
+
 	"github.com/lambda-feedback/shimmy/config"
 	"github.com/lambda-feedback/shimmy/util/conf"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"github.com/urfave/cli/v2"
-	"go.uber.org/zap"
 )
 
 var (
@@ -35,6 +36,13 @@ functions on arbitrary, serverless platforms.`
 				Usage:   "set the log format. Options: production, development.",
 				EnvVars: []string{"LOG_FORMAT"},
 			},
+			// auth flags
+			&cli.StringFlag{
+				Name:     "auth-key",
+				Usage:    "the secret key for the application.",
+				Category: "auth",
+				EnvVars:  []string{"AUTH_KEY"},
+			},
 			// shim flags
 			&cli.StringFlag{
 				Name:     "interface",
@@ -241,6 +249,7 @@ func parseRootConfig(ctx *cli.Context) (config.Config, error) {
 
 	// map cli flags to config fields
 	cliMap := map[string]string{
+		"auth-key":                   "auth.key",
 		"max-workers":                "runtime.max_workers",
 		"command":                    "runtime.cmd",
 		"cwd":                        "runtime.cwd",

cmd/run.go

@@ -3,8 +3,9 @@ package cmd
 import (
 	"os"
 
-	"github.com/lambda-feedback/shimmy/util/logging"
 	"github.com/urfave/cli/v2"
+
+	"github.com/lambda-feedback/shimmy/util/logging"
 )
 
 var (

cmd/serve.go

@@ -1,11 +1,12 @@
 package cmd
 
 import (
+	"github.com/urfave/cli/v2"
+
 	"github.com/lambda-feedback/shimmy/app"
 	"github.com/lambda-feedback/shimmy/app/standalone"
 	"github.com/lambda-feedback/shimmy/util/conf"
 	"github.com/lambda-feedback/shimmy/util/logging"
-	"github.com/urfave/cli/v2"
 )
 
 var (