feat: add support for new transformations and layer modes in URL generation

Author: manu4543

src/imagekitio/lib/helper.py

@@ -81,6 +81,8 @@
     "sharpen": "e-sharpen",
     "unsharp_mask": "e-usm",
     "gradient": "e-gradient",
+    "color_replace": "cr",
+    "distort": "e-distort",
     # Other flags & finishing
     "progressive": "pr",
     "lossless": "lo",
@@ -315,6 +317,11 @@ def _process_overlay(overlay: Overlay) -> str:
 
         parsed_overlay.append(f"bg-{color}")
 
+    # Handle layer_mode property
+    layer_mode = overlay.get("layer_mode")
+    if layer_mode:
+        parsed_overlay.append(f"lm-{layer_mode}")
+
     # Handle position properties (x, y, focus)
     # Node.js uses if (x) which skips falsy values like 0, '', false, null, undefined
     x = position.get("x")

tests/custom/url_generation/test_advanced_url_generation.py

@@ -216,6 +216,17 @@ def test_should_handle_complex_transformation_combinations(self):
         expected = "https://ik.imagekit.io/test_url_endpoint/test_path1.jpg?tr=w-300,h-200,q-85,b-5_FF0000"
         assert url == expected
 
+    def test_should_handle_radius_with_complex_corner_values(self):
+        """Test radius transformation with complex corner-specific values."""
+        url = self.client.helper.build_url(
+            src="/test_path1.jpg",
+            url_endpoint="https://ik.imagekit.io/test_url_endpoint",
+            transformation_position="query",
+            transformation=[{"radius": "10_20_20_max"}],
+        )
+        expected = "https://ik.imagekit.io/test_url_endpoint/test_path1.jpg?tr=r-10_20_20_max"
+        assert url == expected
+
     def test_should_generate_the_correct_url_with_many_transformations_including_video_and_ai_transforms(self):
         """Test many transformations including video and AI."""
         url = self.client.helper.build_url(
@@ -271,11 +282,13 @@ def test_should_generate_the_correct_url_with_many_transformations_including_vid
                     "sharpen": 10,
                     "unsharp_mask": "2-2-0.8-0.024",
                     "gradient": "from-red_to-white",
+                    "color_replace": "FF0000_10_0000FF",
+                    "distort": "p-10_20_100_20_100_200_10_200",
                     "original": True,
                     "page": "2_4",
                     "raw": "h-200,w-300,l-image,i-logo.png,l-end",
                 }
             ],
         )
-        expected = "https://ik.imagekit.io/test_url_endpoint/test_path.jpg?tr=h-300,w-400,ar-4-3,q-40,c-force,cm-extract,fo-left,f-jpeg,r-50,bg-A94D34,b-5-A94D34,rt-90,bl-10,n-some_name,pr-true,lo-true,t-5,md-true,cp-true,di-folder@@file.jpg,dpr-3,x-10,y-20,xc-30,yc-40,fl-h,o-0.8,z-2,vc-h264,ac-aac,so-5,eo-15,du-10,sr-1440_1080,e-grayscale,e-upscale,e-retouch,e-genvar,e-dropshadow,e-changebg-prompt-car,e-edit-prompt-make it vintage,e-bgremove,e-contrast,e-shadow-bl-15_st-40_x-10_y-N5,e-sharpen-10,e-usm-2-2-0.8-0.024,e-gradient-from-red_to-white,orig-true,pg-2_4,h-200,w-300,l-image,i-logo.png,l-end"
+        expected = "https://ik.imagekit.io/test_url_endpoint/test_path.jpg?tr=h-300,w-400,ar-4-3,q-40,c-force,cm-extract,fo-left,f-jpeg,r-50,bg-A94D34,b-5-A94D34,rt-90,bl-10,n-some_name,pr-true,lo-true,t-5,md-true,cp-true,di-folder@@file.jpg,dpr-3,x-10,y-20,xc-30,yc-40,fl-h,o-0.8,z-2,vc-h264,ac-aac,so-5,eo-15,du-10,sr-1440_1080,e-grayscale,e-upscale,e-retouch,e-genvar,e-dropshadow,e-changebg-prompt-car,e-edit-prompt-make it vintage,e-bgremove,e-contrast,e-shadow-bl-15_st-40_x-10_y-N5,e-sharpen-10,e-usm-2-2-0.8-0.024,e-gradient-from-red_to-white,cr-FF0000_10_0000FF,e-distort-p-10_20_100_20_100_200_10_200,orig-true,pg-2_4,h-200,w-300,l-image,i-logo.png,l-end"
         assert url == expected

tests/custom/url_generation/test_overlay.py

@@ -403,3 +403,81 @@ def test_should_properly_encode_overlay_text_when_transformations_are_in_query_p
         )
         expected = "https://ik.imagekit.io/demo/sample.jpg?tr=l-text,i-Minimal%20Text,l-end"
         assert url == expected
+
+    # Layer mode tests
+    def test_should_handle_layer_mode_multiply_in_overlay(self):
+        """Test layer mode multiply in image overlay."""
+        url = self.client.helper.build_url(
+            src="/base-image.jpg",
+            url_endpoint="https://ik.imagekit.io/test_url_endpoint",
+            transformation_position="path",
+            transformation=[
+                {
+                    "overlay": {
+                        "type": "image",
+                        "input": "overlay.png",
+                        "layer_mode": "multiply",
+                    }
+                }
+            ],
+        )
+        expected = "https://ik.imagekit.io/test_url_endpoint/tr:l-image,i-overlay.png,lm-multiply,l-end/base-image.jpg"
+        assert url == expected
+
+    def test_should_handle_layer_mode_cutter_in_overlay(self):
+        """Test layer mode cutter in image overlay."""
+        url = self.client.helper.build_url(
+            src="/base-image.jpg",
+            url_endpoint="https://ik.imagekit.io/test_url_endpoint",
+            transformation_position="path",
+            transformation=[
+                {
+                    "overlay": {
+                        "type": "image",
+                        "input": "overlay.png",
+                        "layer_mode": "cutter",
+                    }
+                }
+            ],
+        )
+        expected = "https://ik.imagekit.io/test_url_endpoint/tr:l-image,i-overlay.png,lm-cutter,l-end/base-image.jpg"
+        assert url == expected
+
+    def test_should_handle_layer_mode_cutout_in_overlay(self):
+        """Test layer mode cutout in image overlay."""
+        url = self.client.helper.build_url(
+            src="/base-image.jpg",
+            url_endpoint="https://ik.imagekit.io/test_url_endpoint",
+            transformation_position="path",
+            transformation=[
+                {
+                    "overlay": {
+                        "type": "image",
+                        "input": "overlay.png",
+                        "layer_mode": "cutout",
+                    }
+                }
+            ],
+        )
+        expected = "https://ik.imagekit.io/test_url_endpoint/tr:l-image,i-overlay.png,lm-cutout,l-end/base-image.jpg"
+        assert url == expected
+
+    def test_should_handle_layer_mode_displace_in_overlay(self):
+        """Test layer mode displace in image overlay."""
+        url = self.client.helper.build_url(
+            src="/base-image.jpg",
+            url_endpoint="https://ik.imagekit.io/test_url_endpoint",
+            transformation_position="path",
+            transformation=[
+                {
+                    "overlay": {
+                        "type": "image",
+                        "input": "overlay.png",
+                        "layer_mode": "displace",
+                    }
+                }
+            ],
+        )
+        expected = "https://ik.imagekit.io/test_url_endpoint/tr:l-image,i-overlay.png,lm-displace,l-end/base-image.jpg"
+        assert url == expected
+