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 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, "allby": allByCmd, "fieldsall": fieldsAllCmd, "fieldsallby": fieldsAllByCmd, "find": findCmd, "findby": findByCmd, "fieldsfind": fieldsFindCmd, "fieldsfindby": fieldsFindByCmd, // 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, } var allCmd = &cobra.Command{ Use: "all", Short: "Generate a helper to select all records", } var allByCmd = &cobra.Command{ Use: "allby", Short: "Generate a helper to select all records with specific column values", } var fieldsAllCmd = &cobra.Command{ Use: "fieldsall", Short: "Generate a helper to select specific fields of all records", } var fieldsAllByCmd = &cobra.Command{ Use: "fieldsallby", 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 findByCmd = &cobra.Command{ Use: "findby", Short: "Generate a helper to select a single record that has specific column values", } var fieldsFindCmd = &cobra.Command{ Use: "fieldsfind", Short: "Generate a helper to select specific fields of records by ID", } var fieldsFindByCmd = &cobra.Command{ Use: "fieldsfindby", Short: "Generate a helper to select specific fields of a single record that has specific column values", } var insertCmd = &cobra.Command{ Use: "insert", Short: "Generate insert statement helpers from table definitions", } var deleteCmd = &cobra.Command{ Use: "delete", Short: "Generate delete statement helpers from table definitions", } var structCmd = &cobra.Command{ Use: "struct", Short: "Generate structs from table definitions", }