forked from goffee/goffee
initial commits 1
This commit is contained in:
parent
61645c1459
commit
c97afa7564
19 changed files with 1612 additions and 142 deletions
63
cmd/generate-model.go
Normal file
63
cmd/generate-model.go
Normal file
|
|
@ -0,0 +1,63 @@
|
|||
// Copyright © Harran Ali <harran.m@gmail.com>. All rights reserved.
|
||||
// Use of this source code is governed by MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var GenerateModelCmd = &cobra.Command{
|
||||
Use: "gen:model [ModelName]",
|
||||
Short: "Create a database model",
|
||||
Long: `Helps you generate a boilderplate code for database model
|
||||
example:
|
||||
goffee gen:model User
|
||||
|
||||
`,
|
||||
Args: cobra.ExactArgs(1),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
pwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
modelName := args[0]
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
mfn := camelCaseToSnake(modelName, "-") + ".go"
|
||||
mfnp := filepath.Join(pwd, "models/", mfn)
|
||||
mfs, err := os.Stat(mfnp)
|
||||
if err != nil && !os.IsNotExist(err) {
|
||||
fmt.Printf("problem creating the file: %v\n", mfnp)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if mfs != nil {
|
||||
fmt.Printf("file \"%v\" already exist\n", mfn)
|
||||
os.Exit(1)
|
||||
}
|
||||
ModelFile, err := os.Create(mfnp)
|
||||
if err != err {
|
||||
fmt.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
tableName := camelCaseToSnake(modelName, "_")
|
||||
tableName = singleToPlural(tableName)
|
||||
ModelFile.WriteString(prepareModelContent(modelName, tableName))
|
||||
ModelFile.Close()
|
||||
fmt.Println("model generated successfully")
|
||||
},
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(GenerateModelCmd)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue