Implemented proper User struct
This commit is contained in:
		@@ -5,42 +5,54 @@ import (
 | 
			
		||||
	"html/template"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"streifling.com/jason/cpolis/cmd/data"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type AddUserData struct {
 | 
			
		||||
	User  string
 | 
			
		||||
	First string
 | 
			
		||||
	Last  string
 | 
			
		||||
	Role  string
 | 
			
		||||
	Msg   string
 | 
			
		||||
	data.User
 | 
			
		||||
	Msg string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func inputsEmpty(user, pass, pass2, first, last, role string) bool {
 | 
			
		||||
	return len(user) == 0 ||
 | 
			
		||||
func inputsEmpty(user data.User, pass, pass2 string) bool {
 | 
			
		||||
	return len(user.UserName) == 0 ||
 | 
			
		||||
		len(user.FirstName) == 0 ||
 | 
			
		||||
		len(user.LastName) == 0 ||
 | 
			
		||||
		len(pass) == 0 ||
 | 
			
		||||
		len(pass2) == 0 ||
 | 
			
		||||
		len(first) == 0 ||
 | 
			
		||||
		len(last) == 0 ||
 | 
			
		||||
		len(role) == 0
 | 
			
		||||
		len(pass2) == 0
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func checkUserStrings(user, first, last string) (string, int, bool) {
 | 
			
		||||
func checkUserStrings(user data.User) (string, int, bool) {
 | 
			
		||||
	userLen := 15
 | 
			
		||||
	nameLen := 50
 | 
			
		||||
 | 
			
		||||
	if len(user) > userLen {
 | 
			
		||||
	if len(user.UserName) > userLen {
 | 
			
		||||
		return "Benutzername", userLen, false
 | 
			
		||||
	} else if len(first) > nameLen {
 | 
			
		||||
	} else if len(user.FirstName) > nameLen {
 | 
			
		||||
		return "Vorname", nameLen, false
 | 
			
		||||
	} else if len(last) > nameLen {
 | 
			
		||||
	} else if len(user.LastName) > nameLen {
 | 
			
		||||
		return "Nachname", nameLen, false
 | 
			
		||||
	} else {
 | 
			
		||||
		return "", 0, true
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func HomePage(db *data.DB) http.HandlerFunc {
 | 
			
		||||
	return func(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
		numRows, err := db.CountEntries()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalln(err)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if numRows == 0 {
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/index.html", "web/templates/add-user.html")).Execute(w, nil)
 | 
			
		||||
		} else {
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/index.html", "web/templates/login.html")).Execute(w, nil)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func CreateUser() http.HandlerFunc {
 | 
			
		||||
	return func(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
		template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", nil)
 | 
			
		||||
@@ -49,31 +61,38 @@ func CreateUser() http.HandlerFunc {
 | 
			
		||||
 | 
			
		||||
func AddUser(db *data.DB) http.HandlerFunc {
 | 
			
		||||
	return func(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
		var writer, editor, admin bool
 | 
			
		||||
		role, err := strconv.Atoi(r.PostFormValue("role"))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		htmlData := AddUserData{
 | 
			
		||||
			User:  r.PostFormValue("username"),
 | 
			
		||||
			First: r.PostFormValue("first-name"),
 | 
			
		||||
			Last:  r.PostFormValue("last-name"),
 | 
			
		||||
			Role:  r.PostFormValue("role"),
 | 
			
		||||
			User: data.User{
 | 
			
		||||
				UserName:  r.PostFormValue("username"),
 | 
			
		||||
				FirstName: r.PostFormValue("first-name"),
 | 
			
		||||
				LastName:  r.PostFormValue("last-name"),
 | 
			
		||||
				Role:      data.Role(role),
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
		pass := r.PostFormValue("password")
 | 
			
		||||
		pass2 := r.PostFormValue("password2")
 | 
			
		||||
 | 
			
		||||
		if inputsEmpty(htmlData.User, pass, pass2, htmlData.First, htmlData.Last, htmlData.Role) {
 | 
			
		||||
		if inputsEmpty(htmlData.User, pass, pass2) {
 | 
			
		||||
			htmlData.Msg = "Alle Felder müssen ausgefüllt werden."
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		userString, stringLen, ok := checkUserStrings(htmlData.User, htmlData.First, htmlData.Last)
 | 
			
		||||
		userString, stringLen, ok := checkUserStrings(htmlData.User)
 | 
			
		||||
		if !ok {
 | 
			
		||||
			htmlData.Msg = fmt.Sprint(userString, " ist zu lang. Maximal ", stringLen, " Zeichen erlaubt.")
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		id, _ := db.GetID(htmlData.User)
 | 
			
		||||
		id, _ := db.GetID(htmlData.UserName)
 | 
			
		||||
		if id != 0 {
 | 
			
		||||
			htmlData.Msg = fmt.Sprint(htmlData.User, " ist bereits vergeben. Bitte anderen Benutzernamen wählen.")
 | 
			
		||||
			htmlData.Msg = fmt.Sprint(htmlData.UserName, " ist bereits vergeben. Bitte anderen Benutzernamen wählen.")
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
@@ -82,27 +101,8 @@ func AddUser(db *data.DB) http.HandlerFunc {
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		switch htmlData.Role {
 | 
			
		||||
		case "writer":
 | 
			
		||||
			writer = true
 | 
			
		||||
			editor = false
 | 
			
		||||
			admin = false
 | 
			
		||||
		case "editor":
 | 
			
		||||
			writer = false
 | 
			
		||||
			editor = true
 | 
			
		||||
			admin = false
 | 
			
		||||
		case "admin":
 | 
			
		||||
			writer = false
 | 
			
		||||
			editor = false
 | 
			
		||||
			admin = true
 | 
			
		||||
		default:
 | 
			
		||||
			log.Println("When setting up", htmlData.User, "the HTML was altered.")
 | 
			
		||||
			htmlData.Msg = "Das HTML der Seite zu verändern ist unzulässig. Dieser Vorfall wurde gespeichert."
 | 
			
		||||
			template.Must(template.ParseFiles("web/templates/add-user.html")).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if err := db.AddUser(htmlData.User, pass, htmlData.First, htmlData.Last, writer, editor, admin); err != nil {
 | 
			
		||||
		if err := db.AddUser(htmlData.User, pass); err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user