From b06f2be81c168f5944ad08ccd36f98d6fc4ab9ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Marti=CC=81n?= <danimartinprieto@gmail.com>
Date: Mon, 23 Dec 2024 18:07:52 +0100
Subject: [PATCH] Add variants to link to media

---
 .../legacy/lib/models/common/widgets/Link/Form.jsx   |  8 ++++----
 .../lib/models/common/widgets/LinkToMedia/Form.jsx   | 12 +++++++++---
 .../lib/models/common/widgets/LinkToMedia/index.ts   |  1 -
 3 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/packages/slice-machine/src/legacy/lib/models/common/widgets/Link/Form.jsx b/packages/slice-machine/src/legacy/lib/models/common/widgets/Link/Form.jsx
index 78b60d552..6e727a742 100644
--- a/packages/slice-machine/src/legacy/lib/models/common/widgets/Link/Form.jsx
+++ b/packages/slice-machine/src/legacy/lib/models/common/widgets/Link/Form.jsx
@@ -15,7 +15,7 @@ import { CheckBox } from "@/legacy/lib/forms/fields";
 import {
   DisplayTextCheckbox,
   RepeatableCheckbox,
-  Variants,
+  Variants as VariantsForm,
 } from "./components";
 
 const FormFields = {
@@ -82,7 +82,7 @@ const Form = (props) => {
           // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
           setFieldValue={setFieldValue}
         />
-        <VariantsWrapper variants={variants} setFieldValue={setFieldValue} />
+        <Variants variants={variants} setFieldValue={setFieldValue} />
       </Box>
     </>
   );
@@ -91,7 +91,7 @@ const Form = (props) => {
 export { FormFields };
 export default Form;
 
-function VariantsWrapper({ variants, setFieldValue }) {
+export function Variants({ variants, setFieldValue }) {
   const fieldKey = "config.variants";
 
   const onVariantsChange = (newVariants) =>
@@ -102,7 +102,7 @@ function VariantsWrapper({ variants, setFieldValue }) {
   const error = meta.error?.find((err) => err);
 
   return (
-    <Variants
+    <VariantsForm
       variants={variants}
       onVariantsChange={onVariantsChange}
       error={error}
diff --git a/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/Form.jsx b/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/Form.jsx
index 08dc86f5d..25c17950b 100644
--- a/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/Form.jsx
+++ b/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/Form.jsx
@@ -1,9 +1,14 @@
+/* eslint-disable @typescript-eslint/no-unsafe-assignment */
+
+import { Box } from "@prismicio/editor-ui";
+
 import { Col, Flex as FlexGrid } from "@/legacy/components/Flex";
 import WidgetFormField from "@/legacy/lib/builders/common/EditModal/Field";
 import { createFieldNameFromKey } from "@/legacy/lib/forms";
 import { DefaultFields } from "@/legacy/lib/forms/defaults";
 
 import { DisplayTextCheckbox } from "../Link/components";
+import { Variants } from "../Link/Form";
 
 const FormFields = {
   ...DefaultFields,
@@ -15,11 +20,11 @@ const Form = (props) => {
 
   // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
   const {
-    config: { allowText },
+    config: { allowText, variants },
   } = formValues;
 
   return (
-    <>
+    <Box flexDirection="column" gap={16}>
       <FlexGrid>
         {Object.entries(FormFields).map(([key, field]) => (
           <Col key={key}>
@@ -40,7 +45,8 @@ const Form = (props) => {
           setFieldValue={setFieldValue}
         />
       </FlexGrid>
-    </>
+      <Variants variants={variants} setFieldValue={setFieldValue} />
+    </Box>
   );
 };
 
diff --git a/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/index.ts b/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/index.ts
index 2a5ea315c..90146c140 100644
--- a/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/index.ts
+++ b/packages/slice-machine/src/legacy/lib/models/common/widgets/LinkToMedia/index.ts
@@ -41,7 +41,6 @@ export const LinkToMediaWidget: Widget<Link, typeof schema> = {
       placeholder: "",
       select: "media",
       allowText: true,
-      repeat: false,
     },
   }),
   TYPE_NAME: "Link",