From d081639b885c229ff2051fb18dfc4ba8ae1675e3 Mon Sep 17 00:00:00 2001 From: Matthew Dillon Date: Mon, 5 Jan 2015 13:27:37 -0900 Subject: [PATCH] WIP --- cmd/bactdb/bactdb.go | 39 ++++++++++++++++++++++++++++++++++----- 1 file changed, 34 insertions(+), 5 deletions(-) diff --git a/cmd/bactdb/bactdb.go b/cmd/bactdb/bactdb.go index c9c3ba6..1f46b06 100644 --- a/cmd/bactdb/bactdb.go +++ b/cmd/bactdb/bactdb.go @@ -1,6 +1,7 @@ package main import ( + "fmt" "log" "net/http" "os" @@ -21,21 +22,38 @@ func main() { ShortName: "s", Usage: "Start web server", Flags: []cli.Flag{ - cli.StringFlag{ - Name: "http", - Value: ":8901", - Usage: "HTTP service address", + cli.IntFlag{ + Name: "port", + Value: 8901, + Usage: "HTTP service port", }, }, Action: cmdServe, }, + { + Name: "createdb", + ShortName: "c", + Usage: "create the database schema", + Flags: []cli.Flag{ + cli.BoolFlag{ + Name: "drop", + Usage: "drop DB before creating", + }, + cli.StringFlag{ + Name: "migration_path", + Usage: "path to migrations", + Value: "./datastore/migrations", + }, + }, + Action: cmdCreateDB, + }, } app.Run(os.Args) } func cmdServe(c *cli.Context) { - httpAddr := c.String("http") + httpAddr := fmt.Sprintf(":%v", c.Int("port")) datastore.Connect() @@ -47,3 +65,14 @@ func cmdServe(c *cli.Context) { log.Fatal("ListenAndServe: ", err) } } + +func cmdCreateDB(c *cli.Context) { + migrationsPath := c.String("migration_path") + + datastore.Connect() + + if c.Bool("drop") { + datastore.Drop(migrationsPath) + } + datastore.Create(migrationsPath) +}