moving files to root directory

cullancarey · May 8, 2022 · 231e62b · 231e62b
1 parent 841f8b1
commit 231e62b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/lambda_rotate_header.tf b/lambda_rotate_header.tf
@@ -32,7 +32,12 @@ resource "aws_lambda_function" "rotate_custom_header_lambda" {
   handler       = "rotate_custom_headers.lambda_handler"
 
   source_code_hash = "${data.archive_file.custom_header_lambda_zip.output_base64sha256}"
-
+  environment {
+    variables = {
+      primary_bucket = "${var.root_domain_name}"
+      backup_bucket = "www.${var.root_domain_name}"
+    }
+  }
   runtime = "python3.8"
   timeout = 300
   tags = {

diff --git a/...om_header_lambda/rotate_custom_headers.py → rotate_custom_headers.py b/...om_header_lambda/rotate_custom_headers.py → rotate_custom_headers.py
@@ -2,7 +2,7 @@
 import json
 import random
 import string
-
+import os
 
 def get_bucket_policy(client, bucket_name):
 	result = client.get_bucket_policy(Bucket=f"{bucket_name}")
@@ -35,9 +35,8 @@ def get_cloudfront_headers(client, value):
 
 def lambda_handler(event, context):
 	s3 = boto3.client('s3')
-	bucket = 'justinschoenegge.com'
-	backup_bucket = 'backup-justinschoenegge.com'
-	# header = random_pass()
+	bucket = os.environ['primary_bucket']
+	backup_bucket = os.environ['backup_bucket']
 	value = random_pass()
 	policy = get_bucket_policy(s3, bucket)
 	backup_policy = get_bucket_policy(s3, backup_bucket)