From ed921d4a90b8e34bb41d49022629dfedcaa998d7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alex=20S=C3=A1nchez?= <sion333@gmail.com>
Date: Wed, 17 Apr 2024 12:51:16 +0200
Subject: [PATCH] structure for fills in text and improvements

---
 plugin-src/transformers/partials/index.ts     |  1 +
 .../partials/transformTextStyle.ts            | 16 ++++++++
 plugin-src/transformers/transformTextNode.ts  | 40 ++++---------------
 plugin-src/translators/index.ts               |  1 +
 plugin-src/translators/translateFills.ts      |  5 ++-
 .../translateStyledTextSegments.ts            | 20 ++++++++++
 6 files changed, 49 insertions(+), 34 deletions(-)
 create mode 100644 plugin-src/transformers/partials/transformTextStyle.ts
 create mode 100644 plugin-src/translators/translateStyledTextSegments.ts

diff --git a/plugin-src/transformers/partials/index.ts b/plugin-src/transformers/partials/index.ts
index c9441f99..0f58db1d 100644
--- a/plugin-src/transformers/partials/index.ts
+++ b/plugin-src/transformers/partials/index.ts
@@ -4,4 +4,5 @@ export * from './transformDimensionAndPosition';
 export * from './transformFills';
 export * from './transformSceneNode';
 export * from './transformStrokes';
+export * from './transformTextStyle';
 export * from './transformVectorPaths';
diff --git a/plugin-src/transformers/partials/transformTextStyle.ts b/plugin-src/transformers/partials/transformTextStyle.ts
new file mode 100644
index 00000000..85f8f3e2
--- /dev/null
+++ b/plugin-src/transformers/partials/transformTextStyle.ts
@@ -0,0 +1,16 @@
+import { translateTextDecoration, translateTextTransform } from '@plugin/translators';
+
+import { Paragraph, TextNode } from '@ui/lib/types/text/textContent';
+
+export const transformTextStyle = (
+  node: StyledTextSegment
+): Partial<Paragraph> | Partial<TextNode> => {
+  return {
+    fontFamily: node.fontName.family,
+    fontSize: node.fontSize.toString(),
+    fontStyle: node.fontName.style,
+    fontWeight: node.fontWeight.toString(),
+    textDecoration: translateTextDecoration(node),
+    textTransform: translateTextTransform(node)
+  };
+};
diff --git a/plugin-src/transformers/transformTextNode.ts b/plugin-src/transformers/transformTextNode.ts
index c4f83cd7..81b4584a 100644
--- a/plugin-src/transformers/transformTextNode.ts
+++ b/plugin-src/transformers/transformTextNode.ts
@@ -1,15 +1,12 @@
 import {
   transformBlend,
   transformDimensionAndPosition,
-  transformSceneNode
+  transformFills,
+  transformSceneNode,
+  transformTextStyle
 } from '@plugin/transformers/partials';
-import {
-  translateFills,
-  translateTextDecoration,
-  translateTextTransform
-} from '@plugin/translators';
+import { translateStyledTextSegments } from '@plugin/translators';
 
-import { TextNode as PenpotTextNode } from '@ui/lib/types/text/textContent';
 import { TextShape } from '@ui/lib/types/text/textShape';
 
 export const transformTextNode = (node: TextNode, baseX: number, baseY: number): TextShape => {
@@ -22,23 +19,7 @@ export const transformTextNode = (node: TextNode, baseX: number, baseY: number):
     'textCase',
     'textDecoration',
     'fills'
-  ]);
-
-  const children: PenpotTextNode[] = styledTextSegments.map(segment => {
-    figma.ui.postMessage({ type: 'FONT_NAME', data: segment.fontName.family });
-
-    return {
-      text: segment.characters,
-      fills: translateFills(segment.fills, node.width, node.height),
-      fontFamily: segment.fontName.family,
-      fontSize: segment.fontSize.toString(),
-      fontStyle: segment.fontName.style,
-      fontWeight: segment.fontWeight.toString(),
-      textDecoration: translateTextDecoration(segment),
-      textTransform: translateTextTransform(segment)
-    };
-  });
-
+  ]) as StyledTextSegment[];
   return {
     type: 'text',
     name: node.name,
@@ -50,14 +31,9 @@ export const transformTextNode = (node: TextNode, baseX: number, baseY: number):
           children: [
             {
               type: 'paragraph',
-              fills: translateFills(node.fills, node.width, node.height),
-              fontFamily: children[0].fontFamily,
-              fontSize: children[0].fontSize,
-              fontStyle: children[0].fontStyle,
-              fontWeight: children[0].fontWeight,
-              textDecoration: children[0].textDecoration,
-              textTransform: children[0].textTransform,
-              children: children
+              children: translateStyledTextSegments(styledTextSegments, node.width, node.height),
+              ...(styledTextSegments.length ? transformTextStyle(styledTextSegments[0]) : {}),
+              ...transformFills(node)
             }
           ]
         }
diff --git a/plugin-src/translators/index.ts b/plugin-src/translators/index.ts
index 3daac3e7..953fd238 100644
--- a/plugin-src/translators/index.ts
+++ b/plugin-src/translators/index.ts
@@ -1,6 +1,7 @@
 export * from './translateBlendMode';
 export * from './translateFills';
 export * from './translateStrokes';
+export * from './translateStyledTextSegments';
 export * from './translateTextDecoration';
 export * from './translateTextTransform';
 export * from './translateVectorPaths';
diff --git a/plugin-src/translators/translateFills.ts b/plugin-src/translators/translateFills.ts
index 3c6dab90..e582ef96 100644
--- a/plugin-src/translators/translateFills.ts
+++ b/plugin-src/translators/translateFills.ts
@@ -20,12 +20,13 @@ export const translateFills = (
   height: number
 ): Fill[] => {
   const figmaFills = fills === figma.mixed ? [] : fills;
-  let penpotFills: Fill[] = [];
+  const penpotFills: Fill[] = [];
 
   for (const fill of figmaFills) {
     const penpotFill = translateFill(fill, width, height);
     if (penpotFill) {
-      penpotFills = [penpotFill, ...penpotFills];
+      // colors are applied in reverse order in Figma, that's why we unshift
+      penpotFills.unshift(penpotFill);
     }
   }
 
diff --git a/plugin-src/translators/translateStyledTextSegments.ts b/plugin-src/translators/translateStyledTextSegments.ts
new file mode 100644
index 00000000..008fa729
--- /dev/null
+++ b/plugin-src/translators/translateStyledTextSegments.ts
@@ -0,0 +1,20 @@
+import { transformTextStyle } from '@plugin/transformers/partials';
+import { translateFills } from '@plugin/translators/translateFills';
+
+import { TextNode } from '@ui/lib/types/text/textContent';
+
+export const translateStyledTextSegments = (
+  segments: StyledTextSegment[],
+  width: number,
+  height: number
+): TextNode[] => {
+  return segments.map(segment => {
+    figma.ui.postMessage({ type: 'FONT_NAME', data: segment.fontName.family });
+
+    return {
+      fills: translateFills(segment.fills, width, height),
+      text: segment.characters,
+      ...transformTextStyle(segment)
+    };
+  });
+};