You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.3 KiB
Go
79 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/urfave/cli"
|
|
|
|
"github.com/bradrydzewski/togo/parser"
|
|
"github.com/bradrydzewski/togo/template"
|
|
)
|
|
|
|
type sqlParams struct {
|
|
Package string
|
|
Dialect string
|
|
Statements []*parser.Statement
|
|
}
|
|
|
|
var sqlCommand = cli.Command{
|
|
Name: "sql",
|
|
Usage: "embed sql statements",
|
|
Action: sqlAction,
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "package",
|
|
Value: "sql",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "dialect",
|
|
Value: "sqlite3",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "input",
|
|
Value: "files/*.sql",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "output",
|
|
Value: "sql_gen.go",
|
|
},
|
|
},
|
|
}
|
|
|
|
func sqlAction(c *cli.Context) error {
|
|
pattern := c.Args().First()
|
|
if pattern == "" {
|
|
pattern = c.String("input")
|
|
}
|
|
|
|
matches, err := filepath.Glob(pattern)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
params := sqlParams{
|
|
Package: c.String("package"),
|
|
Dialect: c.String("dialect"),
|
|
}
|
|
|
|
parse := parser.New()
|
|
for _, match := range matches {
|
|
statements, perr := parse.ParseFile(match)
|
|
if perr != nil {
|
|
return perr
|
|
}
|
|
params.Statements = append(params.Statements, statements...)
|
|
}
|
|
|
|
wr := os.Stdout
|
|
if output := c.String("output"); output != "" {
|
|
wr, err = os.Create(output)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer wr.Close()
|
|
}
|
|
|
|
return template.Execute(wr, "sql.tmpl", params)
|
|
}
|