add env var for path prefix

Author: zainasir

lfs-broker/main.go

@@ -79,6 +79,7 @@ var (
 	smClient    *secretsmanager.Client
 	bucketName  string
 	secretName  string
+	pathPrefix  string
 	apiKeyCache = &secretCache{}
 )
 
@@ -93,6 +94,11 @@ func init() {
 		log.Fatal("LFS_SECRET_NAME environment variable is required")
 	}
 
+	pathPrefix = os.Getenv("PATH_PREFIX")
+	if pathPrefix == "" {
+		pathPrefix = "lfs/objects"
+	}
+
 	cfg, err := config.LoadDefaultConfig(context.Background())
 	if err != nil {
 		log.Fatalf("failed to load AWS config: %v", err)
@@ -226,8 +232,8 @@ func handler(ctx context.Context, req events.LambdaFunctionURLRequest) (events.L
 
 func processObject(ctx context.Context, operation string, obj LFSObject) (LFSObjectResult, error) {
 	// S3 key derived from OID using standard LFS content-addressable layout
-	// e.g. oid abc123... -> lfs/objects/ab/c1/abc123...
-	key := fmt.Sprintf("lfs/objects/%s/%s/%s", obj.OID[:2], obj.OID[2:4], obj.OID)
+	// e.g. oid abc123... -> <pathPrefix>/ab/c1/abc123...
+	key := fmt.Sprintf("%s/%s/%s/%s", pathPrefix, obj.OID[:2], obj.OID[2:4], obj.OID)
 
 	var href string
 	var err error

lfs-snapshot/main.go

@@ -64,6 +64,7 @@ var (
 	lfsBucket      string
 	snapshotBucket string
 	snapshotPrefix string
+	pathPrefix     string
 	extraBuckets   []string
 	httpClient     = &http.Client{}
 )
@@ -92,6 +93,11 @@ func init() {
 		extraBuckets = strings.Split(raw, ",")
 	}
 
+	pathPrefix = os.Getenv("PATH_PREFIX")
+	if pathPrefix == "" {
+		pathPrefix = "lfs/objects"
+	}
+
 	cfg, err := config.LoadDefaultConfig(context.Background())
 	if err != nil {
 		log.Fatalf("failed to load AWS config: %v", err)
@@ -310,8 +316,8 @@ func streamLFSObjectToSnapshot(ctx context.Context, pointer *LFSPointer, destBuc
 	}
 
 	// LFS content-addressable key
-	srcKey := fmt.Sprintf("lfs/objects/%s/%s/%s",
-		pointer.OID[:2], pointer.OID[2:4], pointer.OID)
+	srcKey := fmt.Sprintf("%s/%s/%s/%s",
+		pathPrefix, pointer.OID[:2], pointer.OID[2:4], pointer.OID)
 
 	// Get the object from the LFS bucket as a stream
 	getResult, err := s3Client.GetObject(ctx, &s3.GetObjectInput{