forked from jason/cpolis
		
	Handle first user differently from the rest
This commit is contained in:
		@@ -36,7 +36,7 @@ func HomePage(c *control.CliArgs, db *model.DB, s *control.CookieStore) http.Han
 | 
			
		||||
 | 
			
		||||
		files := []string{c.WebDir + "/templates/index.html"}
 | 
			
		||||
		if numRows == 0 {
 | 
			
		||||
			files = append(files, c.WebDir+"/templates/add-user.html")
 | 
			
		||||
			files = append(files, c.WebDir+"/templates/first-user.html")
 | 
			
		||||
			tmpl, err := template.ParseFiles(files...)
 | 
			
		||||
			template.Must(tmpl, err).Execute(w, nil)
 | 
			
		||||
		} else {
 | 
			
		||||
 
 | 
			
		||||
@@ -88,37 +88,13 @@ func AddUser(c *control.CliArgs, db *model.DB, s *control.CookieStore) http.Hand
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		htmlData.ID, err = db.AddUser(htmlData.User, pass)
 | 
			
		||||
		_, err = db.AddUser(htmlData.User, pass)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if htmlData.ID == 1 {
 | 
			
		||||
			htmlData.Role = model.Admin
 | 
			
		||||
 | 
			
		||||
			if err = db.UpdateAttributes(
 | 
			
		||||
				&model.Attribute{Table: "users", ID: id, AttName: "role", Value: htmlData.Role},
 | 
			
		||||
			); err != nil {
 | 
			
		||||
				log.Println(err)
 | 
			
		||||
				http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err := saveSession(w, r, s, htmlData.User); err != nil {
 | 
			
		||||
				log.Println(err)
 | 
			
		||||
				http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if _, err := db.AddIssue(); err != nil {
 | 
			
		||||
				log.Println(err)
 | 
			
		||||
				http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		tmpl, err := template.ParseFiles(c.WebDir + "/templates/hub.html")
 | 
			
		||||
		template.Must(tmpl, err).ExecuteTemplate(w, "page-content", 0)
 | 
			
		||||
	}
 | 
			
		||||
@@ -214,3 +190,72 @@ func UpdateUser(c *control.CliArgs, db *model.DB, s *control.CookieStore) http.H
 | 
			
		||||
		tmpl.ExecuteTemplate(w, "page-content", session.Values["role"].(int))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func AddFirstUser(c *control.CliArgs, db *model.DB, s *control.CookieStore) http.HandlerFunc {
 | 
			
		||||
	return func(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
		var err error
 | 
			
		||||
 | 
			
		||||
		htmlData := UserData{
 | 
			
		||||
			User: &model.User{
 | 
			
		||||
				UserName:  r.PostFormValue("username"),
 | 
			
		||||
				FirstName: r.PostFormValue("first-name"),
 | 
			
		||||
				LastName:  r.PostFormValue("last-name"),
 | 
			
		||||
				Role:      model.Admin,
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
		pass := r.PostFormValue("password")
 | 
			
		||||
		pass2 := r.PostFormValue("password2")
 | 
			
		||||
 | 
			
		||||
		if len(htmlData.UserName) == 0 || len(htmlData.FirstName) == 0 ||
 | 
			
		||||
			len(htmlData.LastName) == 0 || len(pass) == 0 || len(pass2) == 0 {
 | 
			
		||||
			htmlData.Msg = "Alle Felder müssen ausgefüllt werden."
 | 
			
		||||
			tmpl, err := template.ParseFiles(c.WebDir + "/templates/add-user.html")
 | 
			
		||||
			template.Must(tmpl, err).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		userString, stringLen, ok := checkUserStrings(htmlData.User)
 | 
			
		||||
		if !ok {
 | 
			
		||||
			htmlData.Msg = fmt.Sprint(userString, " ist zu lang. Maximal ",
 | 
			
		||||
				stringLen, " Zeichen erlaubt.")
 | 
			
		||||
			tmpl, err := template.ParseFiles(c.WebDir + "/templates/add-user.html")
 | 
			
		||||
			template.Must(tmpl, err).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		id, _ := db.GetID(htmlData.UserName)
 | 
			
		||||
		if id != 0 {
 | 
			
		||||
			htmlData.Msg = fmt.Sprint(htmlData.UserName,
 | 
			
		||||
				" ist bereits vergeben. Bitte anderen Benutzernamen wählen.")
 | 
			
		||||
			tmpl, err := template.ParseFiles(c.WebDir + "/templates/add-user.html")
 | 
			
		||||
			template.Must(tmpl, err).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		if pass != pass2 {
 | 
			
		||||
			htmlData.Msg = "Die Passwörter stimmen nicht überein."
 | 
			
		||||
			tmpl, err := template.ParseFiles(c.WebDir + "/templates/add-user.html")
 | 
			
		||||
			template.Must(tmpl, err).ExecuteTemplate(w, "page-content", htmlData)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		htmlData.ID, err = db.AddUser(htmlData.User, pass)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if err := saveSession(w, r, s, htmlData.User); err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if _, err := db.AddIssue(); err != nil {
 | 
			
		||||
			log.Println(err)
 | 
			
		||||
			http.Error(w, err.Error(), http.StatusInternalServerError)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		tmpl, err := template.ParseFiles(c.WebDir + "/templates/hub.html")
 | 
			
		||||
		template.Must(tmpl, err).ExecuteTemplate(w, "page-content", 0)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user