IMS Indicator on new version should update links to figures 282960

eea/dexterity/indicators/contentrules/actions.py

@@ -62,12 +62,15 @@ def __call__(self):
         elif oid.endswith(".1"):
             old_id = oid.replace(".1", "", 1)
             new_id = old_id + "-%d" % time()
-
         if not (old_id and new_id):
             return True
-
         try:
-            old_version = parent[old_id]
+            if old_id not in parent:
+                old_version = obj.relatedItems[0].to_object
+                obj.relatedItems = []
+                obj.reindexObject(idxs=["relatedItems"])
+            else:
+                old_version = parent[old_id]
             api.content.transition(
                 obj=old_version,
                 transition="markForDeletion",

eea/dexterity/indicators/contentrules/version.py

@@ -1,15 +1,21 @@
 """Copy action for content rules."""
+
+from urllib.parse import urlparse
 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:
@@ -19,11 +25,24 @@
 from Products.statusmessages.interfaces import IStatusMessage
 from ZODB.POSException import ConflictError
 from zope import schema
-from zope.component import adapter
+from zope.component import adapter, getUtility
 from zope.event import notify
-from zope.interface import implementer
-from zope.interface import Interface
-from zope.lifecycleevent import ObjectCopiedEvent
+from zope.interface import implementer, Interface
+from zope.lifecycleevent import ObjectCopiedEvent, modified
+from zope.intid.interfaces import IIntIds
+from z3c.relationfield.relation import RelationValue
+
+
+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 +94,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 +121,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
 
@@ -123,14 +144,33 @@ def __call__(self):
         obj.wl_clearLocks()
 
         obj._postCopy(target, op=0)
+        try:
+            intids = getUtility(IIntIds)
+            relation = RelationValue(intids.getId(orig_obj))
+            obj.relatedItems = [relation]
+            obj.reindexObject(idxs=["relatedItems"])
+        except Exception as e:
+            self.error(obj, str(e))
 
         OFS.subscribers.compatibilityCall("manage_afterClone", obj, obj)
 
         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):
+                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))
+                    block_data["url"] = url
+        modified(obj)
         return True
 
     def error(self, obj, error):
@@ -169,6 +209,7 @@ class CopyAddForm(ActionAddForm):
 
 class CopyAddFormView(ContentRuleFormWrapper):
     """A wrapper for the add form."""
+
     form = CopyAddForm
 
 
@@ -186,4 +227,5 @@ class CopyEditForm(ActionEditForm):
 
 class CopyEditFormView(ContentRuleFormWrapper):
     """A wrapper for the edit form."""
+
     form = CopyEditForm