110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package cmds
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"text/template"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
// sqlBoilerImports defines the list of default template imports.
|
|
var sqlBoilerImports = imports{
|
|
standard: importList{
|
|
`"errors"`,
|
|
`"fmt"`,
|
|
},
|
|
thirdparty: importList{
|
|
`"github.com/pobri19/sqlboiler/boil"`,
|
|
},
|
|
}
|
|
|
|
// sqlBoilerTestImports defines the list of default test template imports.
|
|
var sqlBoilerTestImports = imports{
|
|
standard: importList{
|
|
`"testing"`,
|
|
`"os"`,
|
|
`"os/exec"`,
|
|
`"fmt"`,
|
|
`"io/ioutil"`,
|
|
`"bytes"`,
|
|
`"errors"`,
|
|
},
|
|
thirdparty: importList{
|
|
`"github.com/BurntSushi/toml"`,
|
|
},
|
|
}
|
|
|
|
// sqlBoilerTypeImports imports are only included in the template output if the database
|
|
// requires one of the following special types. Check TranslateColumnType to see the type assignments.
|
|
var sqlBoilerTypeImports = map[string]imports{
|
|
"null.Int": imports{
|
|
thirdparty: importList{`"gopkg.in/guregu/null.v3"`},
|
|
},
|
|
"null.String": imports{
|
|
thirdparty: importList{`"gopkg.in/guregu/null.v3"`},
|
|
},
|
|
"null.Bool": imports{
|
|
thirdparty: importList{`"gopkg.in/guregu/null.v3"`},
|
|
},
|
|
"null.Float": imports{
|
|
thirdparty: importList{`"gopkg.in/guregu/null.v3"`},
|
|
},
|
|
"null.Time": imports{
|
|
thirdparty: importList{`"gopkg.in/guregu/null.v3"`},
|
|
},
|
|
"time.Time": imports{
|
|
standard: importList{`"time"`},
|
|
},
|
|
}
|
|
|
|
// sqlBoilerDriverTestImports defines the test template imports
|
|
// for the particular database interfaces
|
|
var sqlBoilerDriverTestImports = map[string]imports{
|
|
"postgres": imports{
|
|
standard: importList{`"database/sql"`},
|
|
thirdparty: importList{`_ "github.com/lib/pq"`},
|
|
},
|
|
}
|
|
|
|
// 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{
|
|
"singular": singular,
|
|
"plural": plural,
|
|
"titleCase": titleCase,
|
|
"titleCaseSingular": titleCaseSingular,
|
|
"titleCasePlural": titleCasePlural,
|
|
"camelCase": camelCase,
|
|
"camelCaseSingular": camelCaseSingular,
|
|
"camelCasePlural": camelCasePlural,
|
|
"makeDBName": makeDBName,
|
|
"selectParamNames": selectParamNames,
|
|
"insertParamNames": insertParamNames,
|
|
"insertParamFlags": insertParamFlags,
|
|
"insertParamVariables": insertParamVariables,
|
|
"scanParamNames": scanParamNames,
|
|
"hasPrimaryKey": hasPrimaryKey,
|
|
"getPrimaryKey": getPrimaryKey,
|
|
"updateParamNames": updateParamNames,
|
|
"updateParamVariables": updateParamVariables,
|
|
}
|
|
|
|
// LoadConfigFile loads the toml config file into the cfg object
|
|
func (cmdData *CmdData) LoadConfigFile(filename string) error {
|
|
cfg := &Config{}
|
|
|
|
_, err := toml.DecodeFile(filename, &cfg)
|
|
|
|
if os.IsNotExist(err) {
|
|
return fmt.Errorf("Failed to find the toml configuration file %s: %s", filename, err)
|
|
}
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to decode toml configuration file: %s", err)
|
|
}
|
|
|
|
cmdData.Config = cfg
|
|
return nil
|
|
}
|