From fcfdab0498477b06ddf7bafb5df45c631fc4ebc0 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Sun, 13 Feb 2022 09:50:58 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=B3=E9=97=AD=E7=94=9F=E4=BA=A7=E7=8E=AF?=
 =?UTF-8?q?=E5=A2=83=20=20migrate?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/release.yml | 1 -
 cmd/migrate.go                | 9 +++++++++
 cmd/root.go                   | 4 +++-
 share/const.go                | 2 +-
 4 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index ca871a30..60801fca 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -116,7 +116,6 @@ jobs:
       - name: Debug Query
         run: |
           ls -l ./dist/release/
-          coscmd upload -h
 
       - name: Upload To SV
         run: |
diff --git a/cmd/migrate.go b/cmd/migrate.go
index 308bf63a..a6ac9f73 100644
--- a/cmd/migrate.go
+++ b/cmd/migrate.go
@@ -10,9 +10,11 @@ import (
 	"github.com/yaoapp/kun/exception"
 	"github.com/yaoapp/yao/config"
 	"github.com/yaoapp/yao/engine"
+	"github.com/yaoapp/yao/share"
 )
 
 var name string
+var force bool = false
 var migrateCmd = &cobra.Command{
 	Use:   "migrate",
 	Short: L("Update database schema"),
@@ -26,6 +28,12 @@ var migrateCmd = &cobra.Command{
 		}()
 
 		Boot()
+
+		if !force && config.Conf.Mode == "production" {
+			fmt.Println(color.WhiteString(L("TRY:")), color.GreenString("%s migrate --force", share.BUILDNAME))
+			exception.New(L("Migrate is not allowed on production mode."), 403).Throw()
+		}
+
 		// 加载数据模型
 		err := engine.Load(config.Conf)
 		if err != nil {
@@ -53,4 +61,5 @@ var migrateCmd = &cobra.Command{
 
 func init() {
 	migrateCmd.PersistentFlags().StringVarP(&name, "name", "n", "", L("Model name"))
+	migrateCmd.PersistentFlags().BoolVarP(&force, "force", "", false, L("Force migrate"))
 }
diff --git a/cmd/root.go b/cmd/root.go
index b5f1296c..6850188c 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -22,7 +22,7 @@ var langs = map[string]string{
 	"Environment file":                      "指定环境变量文件",
 	"Help for yao":                          "显示命令帮助文档",
 	"Show app configure":                    "显示应用配置信息",
-	"Update database schema":                "更新数据库结构",
+	"Update database schema":                "更新数据表结构",
 	"Execute process":                       "运行处理器",
 	"Show version":                          "显示当前版本号",
 	"Development mode":                      "使用开发模式启动",
@@ -46,6 +46,8 @@ var langs = map[string]string{
 	"Listening":                             "    监听",
 	"✨LISTENING✨":                           "✨服务正在运行✨",
 	"SessionPort":                           "会话服务端口",
+	"Force migrate":                         "强制更新数据表结构",
+	"Migrate is not allowed on production mode.": "Migrate 不能再生产环境下使用",
 }
 
 // L 多语言切换
diff --git a/share/const.go b/share/const.go
index 1cccdebd..17576eca 100644
--- a/share/const.go
+++ b/share/const.go
@@ -4,7 +4,7 @@ package share
 // session -> freeport fix bug
 // 更新制品目录 -> ui/downloads
 // Arm64 制品 -> arm build test
-// MacOS 制品 -> debug 13
+// MacOS 制品 -> debug 14
 
 // VERSION 版本号
 const VERSION = "0.9.1"