// Generate generates migration templates.
func (g *migrationGenerator) Generate() {
name := g.flag.Arg(0)
if name == "" {
kocha.PanicOnError(g, "abort: no NAME given")
}
tx := kocha.TxTypeMap[g.txType]
if tx == nil {
kocha.PanicOnError(g, "abort: unsupported transaction type: `%v`", g.txType)
}
now := Now().Format("20060102150405")
data := map[string]interface{}{
"Name": kocha.ToCamelCase(name),
"TimeStamp": now,
"ImportPath": tx.ImportPath(),
"TxType": reflect.TypeOf(tx.TransactionType()).String(),
}
kocha.CopyTemplate(g,
filepath.Join(SkeletonDir("migration"), "migration.go.template"),
filepath.Join("db", "migrations", fmt.Sprintf("%v_%v.go", now, kocha.ToSnakeCase(name))), data)
initPath := filepath.Join("db", "migrations", "init.go")
if _, err := os.Stat(initPath); os.IsNotExist(err) {
appDir, err := kocha.FindAppDir()
if err != nil {
panic(err)
}
kocha.CopyTemplate(g,
filepath.Join(SkeletonDir("migration"), "init.go.template"),
initPath, map[string]interface{}{
"typeName": g.txType,
"tx": strings.TrimSpace(kocha.GoString(tx)),
"dbImportPath": path.Join(appDir, "db"),
})
}
}
func (c *migrateCommand) Run() {
direction := c.flag.Arg(0)
switch direction {
case "up", "down":
// do nothing.
default:
kocha.PanicOnError(c, `abort: no "up" or "down" specified`)
}
tmpDir, err := filepath.Abs("tmp")
if err != nil {
panic(err)
}
if err := os.MkdirAll(tmpDir, 0755); err != nil && !os.IsExist(err) {
kocha.PanicOnError(c, "abort: failed to create directory: %v", err)
}
_, filename, _, _ := runtime.Caller(0)
skeletonDir := filepath.Join(filepath.Dir(filename), "skeleton", "migrate")
t := template.Must(template.ParseFiles(filepath.Join(skeletonDir, direction+".go.template")))
mainFilePath := filepath.ToSlash(filepath.Join(tmpDir, "migrate.go"))
file, err := os.Create(mainFilePath)
if err != nil {
kocha.PanicOnError(c, "abort: failed to create file: %v", err)
}
defer file.Close()
appDir, err := kocha.FindAppDir()
if err != nil {
panic(err)
}
data := map[string]interface{}{
"dbImportPath": c.Package(path.Join(appDir, "db")).ImportPath,
"migrationsImportPath": c.Package(path.Join(appDir, "db", "migrations")).ImportPath,
"dbconf": c.dbconf,
"limit": c.limit,
}
if err := t.Execute(file, data); err != nil {
kocha.PanicOnError(c, "abort: failed to write file: %v", err)
}
c.execCmd("go", "run", mainFilePath)
if err := os.RemoveAll(tmpDir); err != nil {
panic(err)
}
kocha.PrintGreen("All migrations are successful!\n")
}
// Run execute the process for `build` command.
func (c *buildCommand) Run() {
dir, err := os.Getwd()
if err != nil {
panic(err)
}
appDir, err := kocha.FindAppDir()
if err != nil {
panic(err)
}
appName := filepath.Base(dir)
configPkg, err := c.Package(path.Join(appDir, "config"))
if err != nil {
kocha.PanicOnError(c, "abort: cannot import `%s`: %v", path.Join(appDir, "config"), err)
}
var dbImportPath string
dbPkg, err := c.Package(path.Join(appDir, "db"))
if err == nil {
dbImportPath = dbPkg.ImportPath
}
var migrationsImportPath string
migrationsPkg, err := c.Package(path.Join(appDir, "db", "migrations"))
if err == nil {
migrationsImportPath = migrationsPkg.ImportPath
}
tmpDir, err := filepath.Abs("tmp")
if err != nil {
panic(err)
}
if err := os.Mkdir(tmpDir, 0755); err != nil && !os.IsExist(err) {
kocha.PanicOnError(c, "abort: failed to create directory: %v", err)
}
_, filename, _, _ := runtime.Caller(0)
baseDir := filepath.Dir(filename)
skeletonDir := filepath.Join(baseDir, "skeleton", "build")
mainTemplate, err := ioutil.ReadFile(filepath.Join(skeletonDir, "main.go.template"))
if err != nil {
panic(err)
}
mainFilePath := filepath.ToSlash(filepath.Join(tmpDir, "main.go"))
builderFilePath := filepath.ToSlash(filepath.Join(tmpDir, "builder.go"))
file, err := os.Create(builderFilePath)
if err != nil {
kocha.PanicOnError(c, "abort: failed to create file: %v", err)
}
defer file.Close()
builderTemplatePath := filepath.ToSlash(filepath.Join(skeletonDir, "builder.go.template"))
t := template.Must(template.ParseFiles(builderTemplatePath))
var resources map[string]string
if c.all {
resources = c.collectResourcePaths(filepath.Join(dir, kocha.StaticDir))
}
data := map[string]interface{}{
"configImportPath": configPkg.ImportPath,
"dbImportPath": dbImportPath,
"migrationsImportPath": migrationsImportPath,
"mainTemplate": string(mainTemplate),
"mainFilePath": mainFilePath,
"resources": resources,
"version": c.detectVersionTag(),
}
if err := t.Execute(file, data); err != nil {
kocha.PanicOnError(c, "abort: failed to write file: %v", err)
}
execName := appName
if runtime.GOOS == "windows" {
execName += ".exe"
}
c.execCmd("go", "run", builderFilePath)
c.execCmd("go", "build", "-o", execName, mainFilePath)
if err := os.RemoveAll(tmpDir); err != nil {
panic(err)
}
printSettingEnv()
fmt.Printf("build all-in-one binary to %v\n", filepath.Join(dir, execName))
kocha.PrintGreen("Build successful!\n")
}