Fixed web export

Phara0h · Jul 8, 2020 · 8d8bbde · 8d8bbde
1 parent bc21ef7
commit 8d8bbde
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -388,6 +388,17 @@ parentName = collection.info.name;
 
 jsFile += genClass(collection.info.name, collection.item, '', collection.info)
 
+if(web) {
+jsFile += `
+  export default function(host){
+    if(host) {
+      hostUrl = host;
+    }
+    return {${allNewClasses.join()}};
+  }
+`
+}
+else {
 jsFile += `
   module.exports= function(host){
     if(host) {
@@ -396,6 +407,8 @@ jsFile += `
     return {${allNewClasses.join()}};
   }
 `
+}
+
 
 console.log(prettier.format(jsFile, {
   tabWidth: 2,