fix(babel-plugin-transform-svg): support template that only return a …

…single node Closes #223
gregberge · Nov 8, 2018 · 80ac40f · 80ac40f
1 parent 00a2625
commit 80ac40f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/packages/babel-plugin-transform-svg-component/src/index.js b/packages/babel-plugin-transform-svg-component/src/index.js
@@ -16,13 +16,18 @@ const plugin = (api, opts) => ({
     Program(path) {
       const { types: t } = api
       const template = opts.template || defaultTemplate
-      path.node.body = template(api, opts, {
+      const body = template(api, opts, {
         componentName: t.identifier(opts.state.componentName),
         props: getProps(api, opts),
         imports: getImport(api, opts),
         exports: getExport(api, opts),
         jsx: path.node.body[0].expression,
       })
+      if (Array.isArray(body)) {
+        path.node.body = body
+      } else {
+        path.node.body = [body]
+      }
       path.replaceWith(path.node)
     },
   },

diff --git a/packages/babel-plugin-transform-svg-component/src/index.test.js b/packages/babel-plugin-transform-svg-component/src/index.test.js
@@ -45,6 +45,14 @@ export default MyComponent;"
 `)
   })
 
+  it('should handle template that does not return an array', () => {
+    const { code } = testPlugin('<svg><div /></svg>', {
+      template: ({ template }, opts, { jsx }) => template.ast`${jsx}`,
+      state: { componentName: 'SvgComponent' },
+    })
+    expect(code).toMatchInlineSnapshot(`"<svg><div /></svg>;"`)
+  })
+
   it('should work with ref', () => {
     const { code } = testPlugin('<svg><div /></svg>', {
       state: { componentName: 'SvgComponent' },