Added ability to add user

cmd/data/db.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	"github.com/go-sql-driver/mysql"
+	"golang.org/x/crypto/bcrypt"
 )
 
 type DB struct {
@@ -32,3 +33,28 @@ func OpenDB(dbName string) (*DB, error) {
 
 	return &db, nil
 }
+
+func (db *DB) AddUser(user, pass, first, last string, writer, editor, admin bool) error {
+	hashedPass, err := bcrypt.GenerateFromPassword([]byte(pass), bcrypt.DefaultCost)
+	if err != nil {
+		return fmt.Errorf("error creating password hash: %v", err)
+	}
+
+	if !permissionsOK(writer, editor, admin) {
+		return fmt.Errorf("error with mutually exclusive user permissions: writer = %v, editor = %v, admin = %v",
+			writer, editor, admin)
+	}
+
+	query := `
+    INSERT INTO users
+        (username, password, first_name, last_name, writer, editor, admin)
+    VALUES
+        (?, ?, ?, ?, ?, ?)
+    `
+	_, err = db.Exec(query, user, hashedPass, first, last, writer, editor, admin)
+	if err != nil {
+		return fmt.Errorf("error inserting user into DB: %v", err)
+	}
+
+	return nil
+}

cmd/data/helpers.go

@@ -50,3 +50,9 @@ func getCredentials() (string, string, error) {
 
 	return user, pass, nil
 }
+
+func permissionsOK(writer, editor, admin bool) bool {
+	return writer && !editor && !admin ||
+		!writer && editor && !admin ||
+		!writer && !editor && admin
+}

go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/gorilla/feeds v1.1.2
 	github.com/microcosm-cc/bluemonday v1.0.26
 	github.com/yuin/goldmark v1.7.0
+	golang.org/x/crypto v0.14.0
 	golang.org/x/term v0.17.0
 )
 

go.sum

@@ -16,6 +16,8 @@ github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZV
 github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
 github.com/yuin/goldmark v1.7.0 h1:EfOIvIMZIzHdB/R/zVrikYLPPwJlfMcNczJFMs1m6sA=
 github.com/yuin/goldmark v1.7.0/go.mod h1:uzxRWxtg69N339t3louHJ7+O03ezfj6PlliRlaOzY1E=
+golang.org/x/crypto v0.14.0 h1:wBqGXzWJW6m1XrIKlAH0Hs1JJ7+9KBwnIO8v66Q9cHc=
+golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=
 golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
 golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
 golang.org/x/sys v0.17.0 h1:25cE3gD+tdBA7lp7QfhuV+rJiE9YXTcS3VG1SqssI/Y=