IMS Indicator on new version should update links to figures 282960

eea/dexterity/indicators/contentrules/version.py

@@ -1,15 +1,23 @@
 """Copy action for content rules."""
+
+import copy
+from urllib.parse import urlparse
+import transaction
 from Acquisition import aq_base
 from OFS.event import ObjectClonedEvent
 from OFS.SimpleItem import SimpleItem
 import OFS.subscribers
+from plone.restapi.blocks import visit_blocks
 from plone.app.contentrules import PloneMessageFactory as _
 from plone.app.contentrules.actions import ActionAddForm
 from plone.app.contentrules.actions import ActionEditForm
 from plone.app.contentrules.browser.formhelper import ContentRuleFormWrapper
 from plone.app.vocabularies.catalog import CatalogSource
 from plone.contentrules.rule.interfaces import IExecutable
 from plone.contentrules.rule.interfaces import IRuleElementData
+from plone.restapi.serializer.utils import uid_to_url
+from plone.restapi.deserializer.utils import path2uid
+
 try:
     from plone.base.utils import pretty_title_or_id
 except ImportError:
@@ -23,7 +31,19 @@
 from zope.event import notify
 from zope.interface import implementer
 from zope.interface import Interface
-from zope.lifecycleevent import ObjectCopiedEvent
+from zope.lifecycleevent import ObjectCopiedEvent, modified
+
+
+def getLink(path):
+    """
+    Get link
+    """
+
+    URL = urlparse(path)
+
+    if URL.netloc.startswith("localhost") and URL.scheme:
+        return path.replace(URL.scheme + "://" + URL.netloc, "")
+    return path
 
 
 class ICopyAction(Interface):
@@ -75,11 +95,13 @@ def __init__(self, context, element, event):
         self.event = event
 
     def __call__(self):
+
         portal_url = getToolByName(self.context, "portal_url", None)
         if portal_url is None:
             return False
 
         obj = self.event.object
+        previous_obj_path = obj.absolute_url_path()
 
         path = self.element.target_folder
         change_note = self.element.change_note
@@ -100,7 +122,7 @@ def __call__(self):
 
         old_id = obj.getId()
         new_id = self.generate_id(target, old_id)
-        if not new_id.endswith('.1'):
+        if not new_id.endswith(".1"):
             # Version already exists, redirect to it - refs #279130
             return True
 
@@ -128,9 +150,26 @@ def __call__(self):
 
         notify(ObjectClonedEvent(obj))
 
-        pr = getToolByName(obj, 'portal_repository')
+        pr = getToolByName(obj, "portal_repository")
         pr.save(obj=obj, comment=change_note)
 
+        # CHANGE URL OF FIGURES TO THE NEW DRAFT VERSION
+        for block_data in visit_blocks(obj, obj.blocks):
+            if (block_data.get("@type") == "embed_content" and
+                "url" in block_data):
+                new_block = copy.deepcopy(block_data)
+                url = uid_to_url(block_data["url"])
+                if previous_obj_path in url:
+                    url = url.replace(
+                        previous_obj_path,
+                        previous_obj_path + ".1"
+                    )
+                    url = path2uid(context=self.context, link=getLink(url))
+                    new_block["url"] = url
+                block_data.clear()
+                block_data.update(new_block)
+        modified(obj)
+        transaction.commit()
         return True
 
     def error(self, obj, error):
@@ -169,6 +208,7 @@ class CopyAddForm(ActionAddForm):
 
 class CopyAddFormView(ContentRuleFormWrapper):
     """A wrapper for the add form."""
+
     form = CopyAddForm
 
 
@@ -186,4 +226,5 @@ class CopyEditForm(ActionEditForm):
 
 class CopyEditFormView(ContentRuleFormWrapper):
     """A wrapper for the edit form."""
+
     form = CopyEditForm