124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package cmds
|
|
|
|
import (
|
|
"text/template"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// imports defines the optional standard imports and
|
|
// thirdparty imports (from github for example)
|
|
type imports struct {
|
|
standard []string
|
|
thirdparty []string
|
|
}
|
|
|
|
// sqlBoilerDefaultImports defines the list of default template imports.
|
|
// Imports that are defined
|
|
var sqlBoilerDefaultImports = imports{
|
|
standard: []string{
|
|
`"errors"`,
|
|
`"fmt"`,
|
|
},
|
|
thirdparty: []string{
|
|
`"github.com/pobri19/sqlboiler/boil"`,
|
|
`"gopkg.in/guregu/null.v3"`,
|
|
},
|
|
}
|
|
|
|
var sqlBoilerCustomImports map[string]imports
|
|
|
|
// sqlBoilerCommands points each command to its cobra.Command variable.
|
|
//
|
|
// If you would like to add your own custom command, add it to this
|
|
// map, and point it to your <commandName>Cmd variable.
|
|
//
|
|
// Command names should match the template file name (without the file extension).
|
|
var sqlBoilerCommands = map[string]*cobra.Command{
|
|
// Command to generate all commands
|
|
"boil": boilCmd,
|
|
// Struct commands
|
|
"struct": structCmd,
|
|
// Insert commands
|
|
"insert": insertCmd,
|
|
// Select commands
|
|
"all": allCmd,
|
|
"where": whereCmd,
|
|
"select": selectCmd,
|
|
"selectwhere": selectWhereCmd,
|
|
"find": findCmd,
|
|
"findselect": findSelectCmd,
|
|
// Delete commands
|
|
"delete": deleteCmd,
|
|
}
|
|
|
|
// sqlBoilerCommandRuns points each command to its custom run function.
|
|
// If a run function is not defined here, it will use the defaultRun() default run function.
|
|
var sqlBoilerCommandRuns = map[string]CobraRunFunc{
|
|
"boil": boilRun,
|
|
}
|
|
|
|
// sqlBoilerTemplateFuncs is a map of all the functions that get passed into the templates.
|
|
// If you wish to pass a new function into your own template, add a pointer to it here.
|
|
var sqlBoilerTemplateFuncs = template.FuncMap{
|
|
"titleCase": titleCase,
|
|
"camelCase": camelCase,
|
|
"makeDBName": makeDBName,
|
|
"selectParamNames": selectParamNames,
|
|
"insertParamNames": insertParamNames,
|
|
"insertParamFlags": insertParamFlags,
|
|
"scanParamNames": scanParamNames,
|
|
}
|
|
|
|
/* Struct commands */
|
|
|
|
var structCmd = &cobra.Command{
|
|
Use: "struct",
|
|
Short: "Generate structs from table definitions",
|
|
}
|
|
|
|
/* Insert commands */
|
|
|
|
var insertCmd = &cobra.Command{
|
|
Use: "insert",
|
|
Short: "Generate insert statement helpers from table definitions",
|
|
}
|
|
|
|
/* Select commands */
|
|
|
|
var allCmd = &cobra.Command{
|
|
Use: "all",
|
|
Short: "Generate a helper to select all records",
|
|
}
|
|
|
|
var whereCmd = &cobra.Command{
|
|
Use: "where",
|
|
Short: "Generate a helper to select all records with specific column values",
|
|
}
|
|
|
|
var selectCmd = &cobra.Command{
|
|
Use: "select",
|
|
Short: "Generate a helper to select specific fields of all records",
|
|
}
|
|
|
|
var selectWhereCmd = &cobra.Command{
|
|
Use: "selectwhere",
|
|
Short: "Generate a helper to select specific fields of records with specific column values",
|
|
}
|
|
|
|
var findCmd = &cobra.Command{
|
|
Use: "find",
|
|
Short: "Generate a helper to select a single record by ID",
|
|
}
|
|
|
|
var findSelectCmd = &cobra.Command{
|
|
Use: "findselect",
|
|
Short: "Generate a helper to select specific fields of a record by ID",
|
|
}
|
|
|
|
/* Delete commands */
|
|
|
|
var deleteCmd = &cobra.Command{
|
|
Use: "delete",
|
|
Short: "Generate delete statement helpers from table definitions",
|
|
}
|