1 files changed, 71 insertions, 8 deletions
diff --git a/indoteknik_api/controllers/controller.py b/indoteknik_api/controllers/controller.py
index 4d6716b2..50e86b68 100644
--- a/indoteknik_api/controllers/controller.py
+++ b/indoteknik_api/controllers/controller.py
@@ -1,14 +1,17 @@
-from array import array
-import datetime
 import base64
+import datetime
+import functools
+import io
 import json
+from array import array
 
+import jwt
 from odoo import http
 from odoo.http import request
+from odoo.modules import get_module_resource
 from odoo.tools.config import config
+from PIL import Image
 from pytz import timezone
-import jwt
-import functools
 
 
 class Controller(http.Controller):
@@ -182,8 +185,68 @@ class Controller(http.Controller):
         return self.response(address)
 
     @http.route('/api/image/<model>/<field>/<id>', auth='public', methods=['GET'])
-    def get_image(self, model, field, id):
+    def get_image(self, model, field, id, **kw):
+        model_name = model
         model = request.env[model].sudo().search([('id', '=', id)], limit=1)
-        image = model[field] if model[field] else ''
-        request.env['user.activity.log'].record_activity()
-        return request.make_response(base64.b64decode(image), [('Content-Type', 'image/jpg')])
+        image = model[field] if field in model else ''
+        
+        if model_name in ['product.template']:
+            version = '1' if field in ['image_256', 'image_512', 'image_1024', 'image_1920'] else '2'
+            ratio = kw.get('ratio', '')
+            image = model['image_256'] or ''
+            image = self.add_watermark_to_image(image, ratio, version)
+        
+        response_headers = [
+            ('Content-Type', 'image/jpg'), 
+            ('Cache-Control', 'public, max-age=3600')
+        ]
+        
+        return request.make_response(
+            base64.b64decode(image), 
+            response_headers
+        )
+        
+        
+    def add_watermark_to_image(self, image, ratio, version = '1'):
+        if not image: return ''
+        
+        logo_path = get_module_resource('indoteknik_api', 'static', 'src', 'images', 'logo-indoteknik-gray.png')
+        logo_img = Image.open(logo_path).convert('RGBA')
+        
+        img_data = io.BytesIO(base64.b64decode(image))
+        img = Image.open(img_data).convert('RGBA')
+        
+        img_width, img_height = img.size
+        longest_wh = max(img_height, img_width)
+        
+        # Resize logo image
+        logo_img_w = img_width // 2.2
+        logo_img_h = img_height // 2.2
+        
+        new_img = img
+        
+        if ratio == 'square':
+            new_img = Image.new('RGBA', (longest_wh, longest_wh), (255, 255, 255, 255))
+        
+            paste_x = (longest_wh - img_width) // 2
+            paste_y = (longest_wh - img_height) // 2
+            
+            new_img.paste(img, (paste_x, paste_y), img)
+        
+        if version == '2':
+            logo__footer_path = get_module_resource('indoteknik_api', 'static', 'src', 'images', 'logo-indoteknik-footer.png')
+            logo_footer_img = Image.open(logo__footer_path).convert('RGBA')
+            logo_footer_img.thumbnail((img_width, img_height // 1))
+            logo_footer_w, logo_footer_h = logo_footer_img.size
+            new_img.paste(logo_footer_img, (0, img_height - logo_footer_h - 20), logo_footer_img)
+            
+        logo_img.thumbnail((logo_img_w, logo_img_h))
+        
+        if version == '1':
+            # Add watermark
+            new_img.paste(logo_img, (12, 10), logo_img)
+        
+        buffered = io.BytesIO()
+        new_img.save(buffered, format="PNG")
+        return base64.b64encode(buffered.getvalue()).decode('utf-8')
+