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