Add VPC support

README.md

@@ -1,3 +1,3 @@
 # Terraform Lambda Shell
 
-This module creates a Lambda function that can run arbitrary shell commands. It is intended to be used with the [Invicton-Labs/lambda-shell-data/aws](https://registry.terraform.io/modules/Invicton-Labs/lambda-shell-data/aws/latest) and [Invicton-Labs/lambda-shell-resource/aws](https://registry.terraform.io/modules/Invicton-Labs/lambda-shell-resource/aws/latest) modules.
+This module creates a Lambda function that can run arbitrary shell commands. It is intended to be used with the [Invicton-Labs/lambda-shell-data/aws](https://registry.terraform.io/modules/Invicton-Labs/lambda-shell-data/aws/latest) and [Invicton-Labs/lambda-shell-resource/aws](https://registry.terraform.io/modules/Invicton-Labs/lambda-shell-resource/aws/latest) modules. See those module pages for documentation on how to use the system.

main.tf

@@ -29,6 +29,7 @@ module "shell_lambda" {
       "ModuleAuthor" = "InvictonLabs"
       "ModuleUrl"    = "https://registry.terraform.io/modules/Invicton-Labs/lambda-shell/aws"
     }
+    vpc_config = var.lambda_vpc_config
   }
   role_policies                 = var.lambda_role_policies_json
   role_policy_arns              = var.lambda_role_policy_arns

variables.tf

@@ -51,6 +51,15 @@ variable "lambda_logs_non_lambda_subscriptions" {
   default = []
 }
 
+variable "lambda_vpc_config" {
+  description = "The VPC configuration for the Lambda."
+  type = object({
+    subnet_ids         = list(string)
+    security_group_ids = list(string)
+  })
+  default = null
+}
+
 data "aws_caller_identity" "current" {}
 
 data "aws_arn" "role" {