strapi-dz-component-duplicator 0.1.2 → 0.1.4

package/admin/src/components/DynamicZoneActionInjector.mjs

@@ -1,10 +1,22 @@
 import React from 'react';
-import { useNotification } from '@strapi/admin/strapi-admin';
-import { unstable_useContentManagerContext as useContentManagerContext } from '@strapi/content-manager/strapi-admin';
+import { useFetchClient, useNotification, useQueryParams } from '@strapi/admin/strapi-admin';
+import {
+  buildValidParams,
+  unstable_useContentManagerContext as useContentManagerContext,
+} from '@strapi/content-manager/strapi-admin';
 import { useIntl } from 'react-intl';
 
 const DUPLICATE_CONTAINER_ATTR = 'data-dz-component-duplicator-action';
 const INDEX_SEGMENT_REGEX = /^\d+$/;
+const COLLECTION_TYPES = 'collection-types';
+const SINGLE_TYPES = 'single-types';
+const ONE_WAY_RELATIONS = new Set([
+  'oneWay',
+  'oneToOne',
+  'manyToOne',
+  'oneToManyMorph',
+  'oneToOneMorph',
+]);
 const DUPLICATE_ICON_PATH =
   'M27 4H11a1 1 0 0 0-1 1v5H5a1 1 0 0 0-1 1v16a1 1 0 0 0 1 1h16a1 1 0 0 0 1-1v-5h5a1 1 0 0 0 1-1V5a1 1 0 0 0-1-1m-1 16h-4v-9a1 1 0 0 0-1-1h-9V6h14z';
 
@@ -44,6 +56,10 @@ const isDynamicZoneItem = (value) => {
   return isPlainObject(value) && typeof value.__component === 'string';
 };
 
+const isMediaObject = (value) => {
+  return isPlainObject(value) && typeof value.mime === 'string';
+};
+
 const collectDynamicZonePaths = (value, currentPath = '', acc = []) => {
   if (Array.isArray(value)) {
     if (currentPath && value.length > 0 && value.every(isDynamicZoneItem)) {
@@ -78,36 +94,14 @@ const cloneValue = (value) => {
   return JSON.parse(JSON.stringify(value));
 };
 
-const isMediaObject = (value) => {
-  return isPlainObject(value) && typeof value.mime === 'string';
-};
-
-const stripTransientKeys = (value) => {
-  if (Array.isArray(value)) {
-    return value.map(stripTransientKeys);
-  }
-
-  if (!isPlainObject(value)) {
-    return value;
-  }
-
-  // Media references must keep their id/documentId so Strapi can
-  // maintain the relation on save.
-  if (isMediaObject(value)) {
-    return value;
-  }
-
-  const next = {};
-
-  for (const [key, nested] of Object.entries(value)) {
-    if (key === 'id' || key === 'documentId' || key === '__temp_key__') {
-      continue;
-    }
-
-    next[key] = stripTransientKeys(nested);
-  }
+const createTempKeyFactory = () => {
+  let counter = 0;
+  const seed =
+    typeof crypto !== 'undefined' && typeof crypto.randomUUID === 'function'
+      ? crypto.randomUUID()
+      : `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 10)}`;
 
-  return next;
+  return () => `${String(counter++).padStart(4, '0')}-${seed}`;
 };
 
 const getActionAnchor = (listItem) => {
@@ -117,10 +111,6 @@ const getActionAnchor = (listItem) => {
     return null;
   }
 
-  // Dynamic zone headers include a drag handle, a delete button, and a
-  // more-actions menu.  We look for the drag button by its aria-label and
-  // fall back to verifying that at least three buttons exist (the minimum
-  // for a dynamic zone header as of Strapi 5).
   const dragButton =
     header.querySelector('button[aria-label="Drag"]') ||
     header.querySelector('button[aria-label="drag"]');
@@ -258,36 +248,424 @@ const createDuplicateButton = (anchor, label, onClick, signal) => {
 
   const textTemplate = anchor.querySelector('span');
   const text = document.createElement('span');
+
   if (textTemplate?.className) {
     text.className = textTemplate.className;
   }
   text.textContent = label;
 
   button.append(icon, text);
-  button.addEventListener('click', (event) => {
-    event.preventDefault();
-    event.stopPropagation();
-    onClick();
-  }, { signal });
+  button.addEventListener(
+    'click',
+    (event) => {
+      event.preventDefault();
+      event.stopPropagation();
+      void onClick();
+    },
+    { signal }
+  );
 
   return button;
 };
 
+const getRelationIdentity = (relation) => {
+  if (!isPlainObject(relation)) {
+    return null;
+  }
+
+  const documentId = relation.documentId ?? relation.apiData?.documentId;
+  const locale = relation.locale ?? relation.apiData?.locale ?? '';
+  const id = relation.id ?? relation.apiData?.id;
+
+  if (documentId) {
+    return `${documentId}::${locale}`;
+  }
+
+  if (id !== null && id !== undefined) {
+    return `id:${id}`;
+  }
+
+  return null;
+};
+
+const getRelationDisplayValue = (relation) => {
+  const candidateKeys = [
+    'label',
+    'title',
+    'name',
+    'displayName',
+    'question',
+    'heading',
+    'slug',
+    'documentId',
+    'id',
+  ];
+
+  for (const key of candidateKeys) {
+    const value = relation?.[key];
+
+    if (typeof value === 'string' && value.trim()) {
+      return value;
+    }
+
+    if (typeof value === 'number') {
+      return String(value);
+    }
+  }
+
+  return '';
+};
+
+const getRelationCollectionType = (targetModel, contentTypes) => {
+  const targetSchema = Array.isArray(contentTypes)
+    ? contentTypes.find((schema) => schema?.uid === targetModel)
+    : null;
+
+  return targetSchema?.kind === 'singleType' ? SINGLE_TYPES : COLLECTION_TYPES;
+};
+
+const getRelationHref = (targetModel, contentTypes, documentId, locale) => {
+  if (!targetModel || !documentId) {
+    return undefined;
+  }
+
+  const collectionType = getRelationCollectionType(targetModel, contentTypes);
+  const basePath =
+    collectionType === SINGLE_TYPES
+      ? `../${SINGLE_TYPES}/${targetModel}`
+      : `../${COLLECTION_TYPES}/${targetModel}/${documentId}`;
+
+  return locale ? `${basePath}?plugins[i18n][locale]=${locale}` : basePath;
+};
+
+const normalizeFetchedRelations = (value) => {
+  if (Array.isArray(value)) {
+    return value.filter(isPlainObject);
+  }
+
+  if (isPlainObject(value) && Array.isArray(value.results)) {
+    return value.results.filter(isPlainObject);
+  }
+
+  if (isPlainObject(value)) {
+    return [value];
+  }
+
+  return [];
+};
+
+const toRelationConnectEntry = (relation, attribute, contentTypes, createTempKey) => {
+  if (!isPlainObject(relation)) {
+    return null;
+  }
+
+  const id = relation.id ?? relation.apiData?.id;
+  const documentId = relation.documentId ?? relation.apiData?.documentId;
+  const locale = relation.locale ?? relation.apiData?.locale ?? null;
+  const label = relation.label ?? getRelationDisplayValue(relation);
+  const href = relation.href ?? getRelationHref(attribute.target, contentTypes, documentId, locale);
+  const next = {
+    id,
+    documentId,
+    locale,
+    href,
+    label: label || documentId || (id !== undefined && id !== null ? String(id) : ''),
+    __temp_key__: createTempKey(),
+    apiData: {
+      id,
+      locale,
+    },
+  };
+
+  if (relation.status !== undefined) {
+    next.status = relation.status;
+  } else if (typeof relation.publishedAt === 'string') {
+    next.status = 'published';
+  }
+
+  return next;
+};
+
+const sanitizeRelationValue = async ({
+  attribute,
+  value,
+  sourceContext,
+  fieldName,
+  contentTypes,
+  createTempKey,
+  fetchRelationItems,
+}) => {
+  const currentConnect = Array.isArray(value?.connect) ? value.connect : [];
+  const currentDisconnect = Array.isArray(value?.disconnect) ? value.disconnect : [];
+  const fetchedRelations =
+    sourceContext?.id && sourceContext?.model
+      ? await fetchRelationItems(sourceContext.model, sourceContext.id, fieldName)
+      : [];
+
+  const connectedByIdentity = new Map();
+
+  for (const relation of currentConnect) {
+    const entry = toRelationConnectEntry(relation, attribute, contentTypes, createTempKey);
+    const identity = getRelationIdentity(entry);
+
+    if (entry && identity) {
+      connectedByIdentity.set(identity, entry);
+    }
+  }
+
+  const disconnectedIdentities = new Set(
+    currentDisconnect.map(getRelationIdentity).filter(Boolean)
+  );
+
+  const effectiveRelations = [];
+
+  for (const relation of fetchedRelations) {
+    const entry = toRelationConnectEntry(relation, attribute, contentTypes, createTempKey);
+    const identity = getRelationIdentity(entry);
+
+    if (!entry || !identity || disconnectedIdentities.has(identity)) {
+      continue;
+    }
+
+    if (connectedByIdentity.has(identity)) {
+      effectiveRelations.push(connectedByIdentity.get(identity));
+      connectedByIdentity.delete(identity);
+      continue;
+    }
+
+    effectiveRelations.push(entry);
+  }
+
+  for (const relation of connectedByIdentity.values()) {
+    effectiveRelations.push(relation);
+  }
+
+  const connect = ONE_WAY_RELATIONS.has(attribute.relation)
+    ? effectiveRelations.slice(-1)
+    : effectiveRelations;
+
+  return {
+    connect,
+    disconnect: [],
+  };
+};
+
+const sanitizeComponentValue = async ({
+  value,
+  componentUid,
+  components,
+  contentTypes,
+  createTempKey,
+  fetchRelationItems,
+  sourceContext,
+}) => {
+  if (!isPlainObject(value)) {
+    return value;
+  }
+
+  const attributes = components?.[componentUid]?.attributes ?? {};
+  const next = {};
+
+  for (const [fieldName, attribute] of Object.entries(attributes)) {
+    if (fieldName === 'id' || !(fieldName in value)) {
+      continue;
+    }
+
+    next[fieldName] = await sanitizeAttributeValue({
+      attribute,
+      fieldName,
+      value: value[fieldName],
+      components,
+      contentTypes,
+      createTempKey,
+      fetchRelationItems,
+      sourceContext: {
+        model: componentUid,
+        id: sourceContext?.id ?? value?.id,
+      },
+    });
+  }
+
+  return next;
+};
+
+const sanitizeDynamicZoneValue = async ({
+  value,
+  components,
+  contentTypes,
+  createTempKey,
+  fetchRelationItems,
+}) => {
+  if (!isDynamicZoneItem(value)) {
+    return value;
+  }
+
+  const componentUid = value.__component;
+  const next = await sanitizeComponentValue({
+    value,
+    componentUid,
+    components,
+    contentTypes,
+    createTempKey,
+    fetchRelationItems,
+    sourceContext: {
+      model: componentUid,
+      id: value?.id,
+    },
+  });
+
+  return {
+    __component: componentUid,
+    ...next,
+  };
+};
+
+const sanitizeAttributeValue = async ({
+  attribute,
+  fieldName,
+  value,
+  components,
+  contentTypes,
+  createTempKey,
+  fetchRelationItems,
+  sourceContext,
+}) => {
+  if (value === null || value === undefined) {
+    return value;
+  }
+
+  if (attribute.type === 'component') {
+    if (attribute.repeatable) {
+      if (!Array.isArray(value)) {
+        return [];
+      }
+
+      const next = await Promise.all(
+        value.map((item) =>
+          sanitizeComponentValue({
+            value: item,
+            componentUid: attribute.component,
+            components,
+            contentTypes,
+            createTempKey,
+            fetchRelationItems,
+            sourceContext: {
+              model: attribute.component,
+              id: item?.id,
+            },
+          })
+        )
+      );
+
+      return next.map((item) =>
+        isPlainObject(item)
+          ? {
+              ...item,
+              __temp_key__: createTempKey(),
+            }
+          : item
+      );
+    }
+
+    return sanitizeComponentValue({
+      value,
+      componentUid: attribute.component,
+      components,
+      contentTypes,
+      createTempKey,
+      fetchRelationItems,
+      sourceContext: {
+        model: attribute.component,
+        id: value?.id,
+      },
+    });
+  }
+
+  if (attribute.type === 'dynamiczone') {
+    if (!Array.isArray(value)) {
+      return [];
+    }
+
+    const next = await Promise.all(
+      value.map((item) =>
+        sanitizeDynamicZoneValue({
+          value: item,
+          components,
+          contentTypes,
+          createTempKey,
+          fetchRelationItems,
+        })
+      )
+    );
+
+    return next.map((item) =>
+      isPlainObject(item)
+        ? {
+            ...item,
+            __temp_key__: createTempKey(),
+          }
+        : item
+    );
+  }
+
+  if (attribute.type === 'relation') {
+    return sanitizeRelationValue({
+      attribute,
+      value,
+      sourceContext,
+      fieldName,
+      contentTypes,
+      createTempKey,
+      fetchRelationItems,
+    });
+  }
+
+  if (attribute.type === 'media') {
+    return cloneValue(value);
+  }
+
+  if (Array.isArray(value)) {
+    return value.map((item) => cloneValue(item));
+  }
+
+  if (isPlainObject(value)) {
+    if (isMediaObject(value)) {
+      return cloneValue(value);
+    }
+
+    return cloneValue(value);
+  }
+
+  return value;
+};
+
 const DynamicZoneActionInjector = () => {
   const { formatMessage } = useIntl();
   const { toggleNotification } = useNotification();
-  const { form, isLoading, components } = useContentManagerContext();
+  const { get } = useFetchClient();
+  const [{ query }] = useQueryParams();
+  const { form, isLoading, components, contentTypes, model, collectionType, id } =
+    useContentManagerContext();
 
   const isBrowser = typeof document !== 'undefined';
-
   const values = form?.values;
   const valuesRef = React.useRef(values);
   const observerRef = React.useRef(null);
   const frameRef = React.useRef(0);
   const abortRef = React.useRef(null);
+  const relationCacheRef = React.useRef(new Map());
 
   valuesRef.current = values;
 
+  const relationQueryParams = React.useMemo(() => {
+    const params = buildValidParams(query ?? {});
+
+    return {
+      locale: params?.locale,
+      status: params?.status,
+    };
+  }, [query]);
+
   const duplicateLabel = formatMessage({
     id: 'strapi-dz-component-duplicator.action.duplicate',
     defaultMessage: 'Duplicate component',
@@ -298,6 +676,71 @@ const DynamicZoneActionInjector = () => {
     defaultMessage: 'Could not duplicate this component.',
   });
 
+  const fetchRelationItems = React.useCallback(
+    async (relationModel, relationId, fieldName) => {
+      if (!relationModel || !relationId || !fieldName) {
+        return [];
+      }
+
+      const cacheKey = JSON.stringify({
+        relationModel,
+        relationId,
+        fieldName,
+        relationQueryParams,
+      });
+
+      const cachedPromise = relationCacheRef.current.get(cacheKey);
+
+      if (cachedPromise) {
+        return cachedPromise;
+      }
+
+      const request = (async () => {
+        let page = 1;
+        let totalPages = 1;
+        const relations = [];
+
+        while (page <= totalPages) {
+          const response = await get(
+            `/content-manager/relations/${relationModel}/${relationId}/${fieldName}`,
+            {
+              params: {
+                ...relationQueryParams,
+                page,
+                pageSize: 100,
+              },
+            }
+          );
+          const payload = response?.data ?? {};
+          const pageResults = normalizeFetchedRelations(payload.results).reverse();
+
+          relations.push(...pageResults);
+
+          const pagination = payload?.pagination;
+          const pageCount =
+            typeof pagination?.pageCount === 'number'
+              ? pagination.pageCount
+              : Math.max(1, Math.ceil((pagination?.total ?? pageResults.length) / 100));
+
+          totalPages = pageCount;
+          page += 1;
+        }
+
+        return relations;
+      })();
+
+      relationCacheRef.current.set(cacheKey, request);
+
+      try {
+        return await request;
+      } catch (error) {
+        relationCacheRef.current.delete(cacheKey);
+        throw error;
+      }
+    },
+    [get, relationQueryParams]
+  );
+
   const cleanupInjectedButtons = React.useCallback(() => {
     if (abortRef.current) {
       abortRef.current.abort();
@@ -311,7 +754,7 @@ const DynamicZoneActionInjector = () => {
   }, []);
 
   const handleDuplicate = React.useCallback(
-    (dynamicZonePath, index) => {
+    async (dynamicZonePath, index) => {
       if (!form || typeof form.addFieldRow !== 'function') {
         return;
       }
@@ -323,7 +766,20 @@ const DynamicZoneActionInjector = () => {
       }
 
       try {
-        const cloned = stripTransientKeys(cloneValue(item));
+        relationCacheRef.current.clear();
+        const createTempKey = createTempKeyFactory();
+        const cloned = await sanitizeDynamicZoneValue({
+          value: item,
+          components,
+          contentTypes,
+          createTempKey,
+          fetchRelationItems,
+        });
+
+        if (!isDynamicZoneItem(cloned)) {
+          throw new Error('Invalid dynamic zone clone');
+        }
+
         form.addFieldRow(dynamicZonePath, cloned, index + 1);
       } catch {
         toggleNotification({
@@ -332,7 +788,14 @@ const DynamicZoneActionInjector = () => {
         });
       }
     },
-    [form, toggleNotification, duplicateErrorLabel]
+    [
+      components,
+      contentTypes,
+      duplicateErrorLabel,
+      fetchRelationItems,
+      form,
+      toggleNotification,
+    ]
   );
 
   const injectButtons = React.useCallback(() => {
@@ -379,13 +842,13 @@ const DynamicZoneActionInjector = () => {
         continue;
       }
 
-      const duplicateButton = createDuplicateButton(anchor, duplicateLabel, () => {
-        handleDuplicate(location.dynamicZonePath, location.index);
-      }, controller.signal);
-      anchor.parentElement.insertBefore(
-        duplicateButton,
-        anchor
+      const duplicateButton = createDuplicateButton(
+        anchor,
+        duplicateLabel,
+        () => handleDuplicate(location.dynamicZonePath, location.index),
+        controller.signal
       );
+      anchor.parentElement.insertBefore(duplicateButton, anchor);
     }
 
     if (observer) {
@@ -396,6 +859,10 @@ const DynamicZoneActionInjector = () => {
     }
   }, [cleanupInjectedButtons, components, duplicateLabel, handleDuplicate, isBrowser, isLoading]);
 
+  React.useEffect(() => {
+    relationCacheRef.current.clear();
+  }, [id, model, collectionType, relationQueryParams]);
+
   React.useEffect(() => {
     if (!isBrowser) {
       return undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "strapi-dz-component-duplicator",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Duplicate dynamic zone components directly from Strapi 5 edit view.",
   "author": "Djordje Savanovic",
   "keywords": [