feat: auto connect for layouts

Author: ascoders

package.json

@@ -1,6 +1,6 @@
 {
   "name": "pri",
-  "version": "0.0.46",
+  "version": "0.0.47",
   "types": "src/index.ts",
   "main": "built/index.js",
   "scripts": {
@@ -79,4 +79,4 @@
     "walk": "^2.3.9",
     "yargs": "^11.0.0"
   }
-}
+}

src/utils/create-entry.ts

@@ -184,8 +184,19 @@ export async function createEntry(info: IProjectInfo, projectRootPath: string, e
   // Set layout
   if (info.layout) {
     const layoutPath = path.parse(info.layout.filePath)
+    let layoutImportCode = ""
+
+    if (info.stores.length === 0) {
+      layoutImportCode = `import LayoutComponent from "${path.join(layoutPath.dir, layoutPath.name)}"`
+    } else {
+      layoutImportCode = `
+        import LayoutComponentOrigin from "${path.join(layoutPath.dir, layoutPath.name)}"
+        const LayoutComponent = Connect()(LayoutComponentOrigin)
+      `
+    }
+
     entryText.layoutImporter = `
-      import LayoutComponent from "${path.join(layoutPath.dir, layoutPath.name)}"
+      ${layoutImportCode}
 
       const LayoutRoute = ({ component: Component, ...rest }) => {
         return (