gorig-cli 1.0.3 → 1.0.5

package/README.md

@@ -24,7 +24,7 @@ gorig-cli init my-new-project
 
 ### 创建新模块
 
-使用 `create` 命令创建一个新模块：
+在项目根目录下使用 `create` 命令创建一个新模块：
 
 ```sh
 gorig-cli create user

package/commands/create.js

@@ -120,6 +120,36 @@ const createModule = async (args) => {
     await fs.writeFile(routerGoFilePath, routerGoContent);
     console.log(chalk.green(`成功创建 ${chalk.bold('router.go')} 文件`));
 
+    // 检查并更新 domain/init.go 文件
+    const initFilePath = path.join(domainDir, 'init.go');
+    if (await fs.pathExists(initFilePath)) {
+      let initFileContent = await fs.readFile(initFilePath, 'utf-8');
+      
+      const importStatement = `import _ "${projectName}/domain/${moduleName}/api"`;
+
+      if (!initFileContent.includes(importStatement)) {
+        const initFuncIndex = initFileContent.indexOf('func init()');
+        
+        if (initFuncIndex !== -1) {
+          // 找到 init 函数，确保在 init 函数上方添加 import 语句
+          const insertPosition = initFuncIndex;
+          initFileContent = 
+            initFileContent.slice(0, insertPosition) + 
+            `${importStatement}\n\n` + 
+            initFileContent.slice(insertPosition);
+
+          await fs.writeFile(initFilePath, initFileContent);
+          console.log(chalk.green(`成功更新 ${chalk.bold('init.go')} 文件，添加 import 语句`));
+        } else {
+          console.log(chalk.yellow(`未找到 init() 函数，跳过 import 语句添加`));
+        }
+      } else {
+        console.log(chalk.yellow(`init.go 文件中已存在相应的 import 语句，无需重复添加`));
+      }
+    } else {
+      console.log(chalk.yellow(`未找到 ${chalk.bold('init.go')} 文件，跳过 import 语句添加`));
+    }
+
     console.log(chalk.blue(`\n模块 ${chalk.bold(moduleName)} 已成功创建在 ${chalk.bold(`domain/${moduleName}`)} 目录下`));
   } catch (error) {
     console.error(chalk.red('创建模块时出错:'), chalk.redBright(error.message));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gorig-cli",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "type": "module",
   "description": "gorig构建工具",
   "main": "bin/cli.js",

package/templates/config.go.ejs

@@ -5,9 +5,10 @@ import (
 )
 
 var (
-	SysName = ""
+	DBDef    = "def"
+	SysName  = ""
 )
 
 func init() {
-	SysName = configure.GetString("sys.name", "<%= projectNameUpper %>")
+	SysName = configure.GetString("sys.name", "TEST")
 }

package/templates/internal.go.ejs

@@ -2,14 +2,14 @@ package internal
 
 import (
 	"github.com/jom-io/gorig/domainx"
+    "github.com/jom-io/gorig/global/variable"
+    "<%= projectName %>/domain/<%= moduleName %>/api/req"
 	"<%= projectName %>/domain/<%= moduleName %>/model"
-	"<%= projectName %>/domain/<%= moduleName %>/api/req"
-	"<%= projectName %>/global/constants"
-	"<%= projectName %>/global/varb"
+	"<%= projectName %>/global"
 )
 
 func <%= ModuleName %>() *<%= ModuleName %>M {
-	return &<%= ModuleName %>M{Complex: domainx.UseComplex[model.<%= ModuleName %>D](domainx.Mysql, constants.Main, "<%= moduleName %>")}
+	return &<%= ModuleName %>M{Complex: domainx.UseComplex[model.<%= ModuleName %>D](domainx.Mysql, global.DBDef, "<%= moduleName %>")}
 }
 
 type <%= ModuleName %>M struct {
@@ -17,7 +17,7 @@ type <%= ModuleName %>M struct {
 }
 
 func (u *<%= ModuleName %>M) TableName() string {
-	return varb.GetApp().TableName("<%= moduleName %>")
+	return variable.TBPrefix +"<%= moduleName %>"
 }
 
 func init() {