From 8ae0c4406714cb86e9317049d2a662585c9fb8b1 Mon Sep 17 00:00:00 2001
From: Tobias Bieniek <tobias@bieniek.cloud>
Date: Sat, 1 Jun 2024 14:39:19 +0200
Subject: [PATCH] crates-io: Add support for `db-dump.zip` files

---
 .../modules/crates-io/compute-static/src/main.rs    |  2 ++
 terragrunt/modules/crates-io/s3-static.tf           | 13 ++++++++++++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/terragrunt/modules/crates-io/compute-static/src/main.rs b/terragrunt/modules/crates-io/compute-static/src/main.rs
index e893e8eb5..3753f7e17 100644
--- a/terragrunt/modules/crates-io/compute-static/src/main.rs
+++ b/terragrunt/modules/crates-io/compute-static/src/main.rs
@@ -117,6 +117,8 @@ fn handle_request(config: &Config, mut request: Request) -> Result<Response, Err
     // Database dump is too big to cache on Fastly
     if request.get_url_str().ends_with("db-dump.tar.gz") {
         redirect_to_cloudfront(config, "db-dump.tar.gz")
+    } else if request.get_url_str().ends_with("db-dump.zip") {
+        redirect_to_cloudfront(config, "db-dump.zip")
     } else {
         send_request_to_s3(config, &request)
     }
diff --git a/terragrunt/modules/crates-io/s3-static.tf b/terragrunt/modules/crates-io/s3-static.tf
index 6fcc0d1c6..f19fa8374 100644
--- a/terragrunt/modules/crates-io/s3-static.tf
+++ b/terragrunt/modules/crates-io/s3-static.tf
@@ -13,7 +13,7 @@ resource "aws_s3_bucket" "static" {
     max_age_seconds = 3000
   }
 
-  // Keep only the live db-dump.tar.gz and the previous day's version, removing
+  // Keep only the live db-dumps and the previous day's versions, removing
   // all the other ones. This is needed because we don't want this file to be
   // versioned, while all the other ones in the bucket should be versioned.
   lifecycle_rule {
@@ -27,6 +27,17 @@ resource "aws_s3_bucket" "static" {
     }
   }
 
+  lifecycle_rule {
+    id      = "purge-db-dump-zip"
+    enabled = true
+    prefix  = "db-dump.zip"
+
+    abort_incomplete_multipart_upload_days = 1
+    noncurrent_version_expiration {
+      days = 1
+    }
+  }
+
   lifecycle {
     ignore_changes = [
       replication_configuration,