From 58797eb5bccac46d3900d43eb59d50f847ab0e32 Mon Sep 17 00:00:00 2001
From: foyarash <hfoyart@premieroctet.com>
Date: Thu, 3 Dec 2020 15:33:48 +0100
Subject: [PATCH] Fix incorrect code panel content for icons (#109)

---
 src/utils/code.ts | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/utils/code.ts b/src/utils/code.ts
index 3255454905..e5bbcb9244 100644
--- a/src/utils/code.ts
+++ b/src/utils/code.ts
@@ -46,12 +46,21 @@ const buildBlock = ({
       const componentName = capitalize(childComponent.type)
       let propsContent = ''
 
-      const propsNames = Object.keys(childComponent.props)
+      const propsNames = Object.keys(childComponent.props).filter(propName => {
+        if (childComponent.type === 'Icon') {
+          return propName !== 'icon'
+        }
+
+        return true
+      })
 
       propsNames.forEach((propName: string) => {
         const propsValue = childComponent.props[propName]
 
-        if (propName.toLowerCase().includes('icon')) {
+        if (
+          propName.toLowerCase().includes('icon') &&
+          childComponent.type !== 'Icon'
+        ) {
           if (Object.keys(icons).includes(propsValue)) {
             let operand = `={<${propsValue} />}`
 
@@ -79,6 +88,8 @@ const buildBlock = ({
         childComponent.children.length === 0
       ) {
         content += `<${componentName} ${propsContent}>${childComponent.props.children}</${componentName}>`
+      } else if (childComponent.type === 'Icon') {
+        content += `<${childComponent.props.icon} ${propsContent} />`
       } else if (childComponent.children.length) {
         content += `<${componentName} ${propsContent}>
       ${buildBlock({ component: childComponent, components, forceBuildBlock })}
@@ -152,7 +163,7 @@ const getIconsImports = (components: IComponents) => {
 export const generateCode = async (components: IComponents) => {
   let code = buildBlock({ component: components.root, components })
   let componentsCodes = buildComponents(components)
-  const iconImports = getIconsImports(components)
+  const iconImports = [...new Set(getIconsImports(components))]
 
   const imports = [
     ...new Set(