36 lines
781 B
Go
36 lines
781 B
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"trojan/core"
|
|
"trojan/trojan"
|
|
)
|
|
|
|
// upgradeCmd represents the update command
|
|
var upgradeCmd = &cobra.Command{
|
|
Use: "upgrade",
|
|
Short: "升级数据库和trojan配置文件",
|
|
}
|
|
|
|
func upgradeConfig() {
|
|
domain, _ := core.GetValue("domain")
|
|
if domain == "" {
|
|
return
|
|
}
|
|
core.WriteDomain(domain)
|
|
trojan.Restart()
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(upgradeCmd)
|
|
upgradeCmd.AddCommand(&cobra.Command{Use: "db", Short: "升级数据库", Run: func(cmd *cobra.Command, args []string) {
|
|
if err := core.GetMysql().UpgradeDB(); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}})
|
|
upgradeCmd.AddCommand(&cobra.Command{Use: "config", Short: "升级配置文件", Run: func(cmd *cobra.Command, args []string) {
|
|
upgradeConfig()
|
|
}})
|
|
}
|