add user.go

2 files changed, 113 insertions, 0 deletions

diff --git a/conf/main.go b/conf/main.go
index 052f086..f3a321e 100644
--- a/conf/main.go
+++ b/conf/main.go
@@ -33,12 +33,20 @@ func RegisterCommand(val int, cmd *Command) {
 }
 
 
+func OpenDb() {
+	err := p.Open(params.DbFilename)
+	if err != nil { log.Fatal(err) }
+}
+
+
 func main() {
 	p = &db.Phlox{}
 
 	cmds := map[int]*Command{
 		1: Create(),
 		2: SchemaVersion(),
+
+		3: CreateUser(),
 	}
 
 	for val, cmd := range cmds {
diff --git a/conf/user.go b/conf/user.go
new file mode 100644
index 0000000..e4a0169
--- /dev/null
+++ b/conf/user.go
@@ -0,0 +1,105 @@
+package main
+
+import (
+	"os"
+	"fmt"
+	"log"
+	"strings"
+	"bufio"
+	"syscall"
+	"golang.org/x/term"
+	db "sanine.net/git/phlox/db"
+)
+
+
+func GetUsername(prompt string) (string, error) {
+	reader := bufio.NewReader(os.Stdin)
+
+	fmt.Print(prompt)
+	username, err := reader.ReadString('\n')
+	if err != nil {
+		return "", err
+	}
+
+	return strings.TrimSpace(username), nil
+}
+
+
+func GetPassword(prompt string) (string, error) {
+	fmt.Print(prompt)
+	bytes, err := term.ReadPassword(int(syscall.Stdin))
+	if err != nil {
+		return "", err
+	}
+
+	password := strings.TrimSpace(string(bytes))
+	return password, nil
+}
+
+
+func ConfirmPassword() string {
+	password, err := GetPassword("Enter Password: ")
+	if err != nil { log.Fatal(err) }
+	password1, err := GetPassword("Confirm Password: ")
+	if err != nil { log.Fatal(err) }
+	if password != password1 { log.Fatal("passwords do not match!") }
+	return password
+}
+
+
+func GetUser() User {
+	OpenDb()
+	username, err := GetUsername("Username: ")
+	if err != nil { log.Fatal(err) }
+	user, err := p.GetByUsername(username)
+	if err != nil { log.Fatal(err) }
+	return user
+}
+
+
+func CreateUser() *Command {
+	return &Command{
+		Flag: "create-user",
+		Usage: "create a new user",
+		Invoke: func() {
+			OpenDb()
+
+			username, err := GetUsername("Enter Username: ")
+			if err != nil { log.Fatal(err) }
+			password := ConfirmPassword()
+
+			_, err = p.CreateUser(username, password1)
+			if err != nil { log.Fatal(err) }
+			fmt.Printf("created user %v\n", username)
+		},
+	}
+}
+
+
+func DeleteUser() *Command {
+	return &Command{
+		Flag: "delete-user",
+		Usage: "delete an existing user",
+		Invoke: func() {
+			user := GetUser()
+			err := p.DeleteUser(user)
+			if err != nil { log.Fatal(err) }
+			fmt.Printf("deleted user %v\n", user.Name)
+		},
+	}
+}
+
+
+func UpdatePassword() *Command {
+	return &Command{
+		Flag: "update-password",
+		Usage: "update a user's password",
+		Invoke: func() {
+			user := GetUser()
+			password := ConfirmPassword()
+			err := SetPassword(user, password)
+			if err != nil { log.Fatal(err) }
+			fmt.Printf("updated password for user %v\n", user.Name)
+		},
+	}
+}