From 97e75f9ff7a3e1dd74f1bfc441bb80e9473edf05 Mon Sep 17 00:00:00 2001
From: Minsoo Kim <minsoo@eazel.net>
Date: Wed, 29 Jul 2020 18:56:53 +0900
Subject: [PATCH] Allow multiple inlineStyles to be returned by customInlineFn
 #155

---
 .../src/__tests__/stateFromElement-test.js    | 35 +++++++++++++++++++
 .../src/stateFromElement.js                   |  2 +-
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/packages/draft-js-import-element/src/__tests__/stateFromElement-test.js b/packages/draft-js-import-element/src/__tests__/stateFromElement-test.js
index 079ac6fb..fbab2622 100644
--- a/packages/draft-js-import-element/src/__tests__/stateFromElement-test.js
+++ b/packages/draft-js-import-element/src/__tests__/stateFromElement-test.js
@@ -146,6 +146,41 @@ describe('stateFromElement', () => {
     });
   });
 
+  it('should return multiple style from customInlineFn', () => {
+    let element = new ElementNode('div', [], [
+      new ElementNode(
+        'span',
+        [{style: 'font-family', value: 'sans-serif'},{style: 'font-size', value: '12px'},],
+        [new TextNode('Hello')]
+      )
+    ]);
+    let options = {
+      customInlineFn(el, {Style}) {
+        if (el.tagName === 'SPAN') {
+          return Style(['CUSTOM_STYLE_1', 'CUSTOM_STYLE_2']);
+        }
+      },
+    };
+    let contentState = stateFromElement(element, options);
+    let rawContentState = removeBlockKeys(convertToRaw(contentState));
+    expect(rawContentState).toEqual({
+      entityMap: {},
+      blocks: [
+        {
+          text: 'Hello',
+          type: 'unstyled',
+          depth: 0,
+          inlineStyleRanges: [
+            {offset: 0, length: 5, style: 'CUSTOM_STYLE_1'},
+            {offset: 0, length: 5, style: 'CUSTOM_STYLE_2'}
+          ],
+          entityRanges: [],
+          data: {},
+        },
+      ],
+    });
+  });
+  
   it('should support option elementStyles', () => {
     let textNode = new TextNode('Superscript');
     let element = new ElementNode('sup', [], [textNode]);
diff --git a/packages/draft-js-import-element/src/stateFromElement.js b/packages/draft-js-import-element/src/stateFromElement.js
index 2f2b546c..0e4ea3fd 100644
--- a/packages/draft-js-import-element/src/stateFromElement.js
+++ b/packages/draft-js-import-element/src/stateFromElement.js
@@ -347,7 +347,7 @@ class ContentGenerator {
     if (customInline != null) {
       switch (customInline.type) {
         case 'STYLE': {
-          style = style.add(customInline.style);
+          [].concat(customInline.style).forEach(customStyle => style = style.add(customStyle));
           break;
         }
         case 'ENTITY': {