insert user

Author: avoidik

cmd/web/handlers.go

@@ -9,6 +9,7 @@ import (
 	"strings"
 
 	"snippetbox.org/pkg/forms"
+	"snippetbox.org/pkg/models"
 )
 
 func (app *App) Home(w http.ResponseWriter, r *http.Request) {
@@ -138,7 +139,25 @@ func (app *App) CreateUser(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	fmt.Fprintf(w, "CreateUser")
+	err = app.database.InsertUser(form.Name, form.Email, form.Password)
+	if err == models.ErrDuplicateEmail {
+		form.Failures["Email"] = "Email already in use"
+		app.RenderHtml(w, r, "signup.page.html", &HtmlData{Form: form})
+		return
+	} else if err != nil {
+		app.ServerError(w, err)
+		return
+	}
+
+	msg := "Your signup was successful. Please log in using your credentials."
+	session := app.sessions.Load(r)
+	err = session.PutString(w, "flash", msg)
+	if err != nil {
+		app.ServerError(w, err)
+		return
+	}
+
+	http.Redirect(w, r, "/user/login", http.StatusSeeOther)
 }
 
 func (app *App) LoginUser(w http.ResponseWriter, r *http.Request) {

pkg/models/database.go

@@ -2,12 +2,18 @@ package models
 
 import (
 	"database/sql"
+	"errors"
+
+	"github.com/mattn/go-sqlite3"
+	"golang.org/x/crypto/bcrypt"
 )
 
 type Database struct {
 	*sql.DB
 }
 
+var ErrDuplicateEmail = errors.New("models: email address already in use")
+
 func (db *Database) GetSnippet(id int) (*Snippet, error) {
 	stmt := `SELECT id, title, content, created, expires FROM snippets
 	WHERE id = ? AND expires > datetime('now')`
@@ -128,3 +134,22 @@ func (db *Database) InitializeDb() error {
 
 	return nil
 }
+
+func (db *Database) InsertUser(name, email, password string) error {
+	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), 12)
+	if err != nil {
+		return err
+	}
+
+	stmt := `INSERT INTO users (name, email, password, created)
+	VALUES (?, ?, ?, datetime('now'))`
+
+	_, err = db.Exec(stmt, name, email, string(hashedPassword))
+	if err != nil {
+		dbErr := err.(sqlite3.Error)
+		if dbErr.Code == sqlite3.ErrConstraint && dbErr.ExtendedCode == sqlite3.ErrConstraintUnique {
+			return ErrDuplicateEmail
+		}
+	}
+	return err
+}