add hacky command parser

1 files changed, 57 insertions, 147 deletions

diff --git a/conf/main.go b/conf/main.go
index d0785f8..a768bac 100644
--- a/conf/main.go
+++ b/conf/main.go
@@ -5,181 +5,91 @@ import (
 	"fmt"
 	"log"
 	"flag"
-	"strings"
-	"bufio"
+//	"strings"
+//	"bufio"
 	db "sanine.net/git/phlox/db"
 )
 
 
-var p *db.Phlox
-
-
-type Params struct {
-	DbFilename string
+var Params struct {
+	InitDb string
+	Db string
 }
-var params Params
 
 
-type Command struct {
-	Set bool
-	Value int
-	Flag string
-	Usage string
-	Invoke func()
-}
-
-
-func RegisterCommand(val int, cmd *Command) {
-	cmd.Value = val
-	flag.BoolVar(&(*cmd).Set, cmd.Flag, false, cmd.Usage)
-}
-
+var p *db.Phlox
 
-func OpenDb() {
-	err := p.Open(params.DbFilename)
-	if err != nil { log.Fatal(err) }
+type Command struct {
+	Name string
+	Help string
+	subcommands map[string]*Command
+	Branch bool
+	Execute func([]string)
 }
 
-
-func ReadLine(prompt string) string {
-	reader := bufio.NewReader(os.Stdin)
-
-	fmt.Print(prompt)
-	line, err := reader.ReadString('\n')
-	if err != nil {
-		log.Fatal(err)
+func (c *Command) errNoSubCommand() {
+	fmt.Println("error: a subcommand was required but not provided.")
+	fmt.Println("available options are:\n")
+	for name := range c.subcommands {
+		fmt.Printf("\t* %v\n", name)
 	}
-
-	return strings.TrimSpace(line)
+	fmt.Println()
 }
 
-
-func main() {
-	p = &db.Phlox{}
-
-	cmds := map[int]*Command{
-		1: Create(),
-		2: SchemaVersion(),
-
-		3: CreateUser(),
-		4: DeleteUser(),
-		5: UpdatePassword(),
-		6: ListUsers(),
-
-		7: CreateEndpoint(),
-		8: DeleteEndpoint(),
-		9: SetPath(),
-		10: SetAddress(),
-		11: ListEndpoints(),
-
-		12: SetHost(),
-		13: GetHost(),
-		14: SetLogin(),
-		15: GetLogin(),
-	}
-
-	for val, cmd := range cmds {
-		RegisterCommand(val, cmd)
-	}
-
-	flag.StringVar(&params.DbFilename, "db", "phlox.conf", "select the db file to operate on")
-	flag.Parse()
-
-	for _, cmd := range cmds {
-		if cmd.Set { cmd.Invoke(); os.Exit(0); }
+func (c *Command) Parse(args []string) {
+	if !c.Branch {
+		c.Execute(args)
+	} else {
+		if len(args) > 0 {
+			cn, ok := c.subcommands[args[0]]
+			if ok {
+				cn.Parse(args[1:])
+			} else {
+				c.errNoSubCommand()
+			}
+		} else {
+			c.errNoSubCommand()
+		}
 	}
-
-	log.Fatal("no command specified!")
 }
 
 
-func Create() *Command {
-	return &Command{
-		Flag: "create",
-		Usage: "create a new database",
-		Invoke: func() {
-			err := p.Create(params.DbFilename)
-			if err != nil {
-				log.Fatal(err)
-			}
-			fmt.Printf("created database %v\n", params.DbFilename)
-		},
+func (c *Command) AddCommand(cn *Command) {
+	if c.subcommands == nil {
+		c.subcommands = make(map[string]*Command)
 	}
+	c.subcommands[cn.Name] = cn
 }
 
 
-func SchemaVersion() *Command {
-	return &Command{
-		Flag: "schema-version",
-		Usage: "get the schema version",
-		Invoke: func() {
-			err := p.Open(params.DbFilename)
-			if err != nil { log.Fatal(err) }
-			version, err := p.GetSchemaVersion()
-			if err != nil { log.Fatal(err) }
-			fmt.Printf("schema version: %v\n", version)
-		},
-	}
-}
+var parser *Command
 
 
-func SetHost() *Command {
-	return &Command{
-		Flag: "set-host-address",
-		Usage: "set the address to serve the reverse proxy on",
-		Invoke: func() {
-			OpenDb()
-			defer p.Close()
-			address := ReadLine("Address: ")
-			err := p.SetHostAddress(address)
-			if err != nil { log.Fatal(err) }
-			fmt.Printf("set host address to %v\n", address)
-		},
-	}
-}
+func main() {
+	p = &db.Phlox{}
 
+	flag.StringVar(&Params.InitDb, "init", "", "initialize a new database at the given path")
+	flag.StringVar(&Params.Db, "db", "/etc/phlox/phlox.conf", "open the database at the given path for processing")
+	flag.Parse()
 
-func GetHost() *Command {
-	return &Command{
-		Flag: "get-host-address",
-		Usage: "get the address to serve the reverse proxy on",
-		Invoke: func() {
-			OpenDb()
-			defer p.Close()
-			address, err := p.GetHostAddress()
-			if err != nil { log.Fatal(err) }
-			fmt.Printf("host address is %v\n", address)
-		},
+	if Params.InitDb != "" {
+		// initialize a new db
+		err := p.Create(Params.InitDb)
+		if err != nil { log.Fatal(err) }
+		fmt.Printf("created new database at %v\n", Params.InitDb)
+		p.Close()
+		os.Exit(0)
 	}
-}
 
-
-func SetLogin() *Command {
-	return &Command{
-		Flag: "set-login-page",
-		Usage: "set the file to serve as the login page",
-		Invoke: func() {
-			OpenDb()
-			defer p.Close()
-			path := ReadLine("File Path (blank for internal default): ")
-			err := p.SetLoginPage(path)
-			if err != nil { log.Fatal(err) }
-			fmt.Printf("set login page to %v\n", path)
-		},
+	err := p.Open(Params.Db)
+	if err != nil {
+		log.Fatal(err)
 	}
-}
+	defer p.Close()
 
+	parser = &Command{Branch: true}
+	ListInit(parser)
+	CreateInit(parser)
 
-func GetLogin() *Command {
-	return &Command{
-		Flag: "get-login-page",
-		Usage: "get the file to serve the reverse proxy on",
-		Invoke: func() {
-			OpenDb()
-			defer p.Close()
-			path, err := p.GetLoginPage()
-			if err != nil { log.Fatal(err) }
-			fmt.Printf("login page is at %v\n", path)
-		},
-	}
+	parser.Parse(flag.Args())
 }