func Register(w http.ResponseWriter, r *http.Request) {
if utils.GetCurrentUser(r) != nil {
http.Redirect(w, r, "/", http.StatusFound)
return
}
if r.Method == "POST" {
username := r.FormValue("username")
password := r.FormValue("password")
confirm := r.FormValue("password2")
var error string
if password != confirm {
error = "Passwords don't match"
}
if error != "" {
utils.RenderTemplate(w, r, "register.html", map[string]interface{}{
"error": error,
})
return
}
// We're good, let's make it
db_map := models.GetDbSession()
user := models.NewUser(username, password)
err := db_map.Insert(user)
if err != nil {
fmt.Printf("[error] Could not insert user (%s)\n", err.Error())
} else {
http.Redirect(w, r, "/login", http.StatusFound)
return
}
}
utils.RenderTemplate(w, r, "register.html", nil)
}
func Register(w http.ResponseWriter, r *http.Request) {
if utils.GetCurrentUser(r) != nil {
http.Redirect(w, r, "/", http.StatusFound)
return
}
if r.Method == "POST" {
username := r.FormValue("username")
password := r.FormValue("password")
confirm := r.FormValue("password2")
var error string
if password != confirm {
error = "Passwords don't match"
}
// See if a user with this name already exists
db := models.GetDbSession()
count, err := db.SelectInt("SELECT COUNT(*) FROM users WHERE username=$1", username)
if count > 0 || err != nil {
error = "This username is already taken."
}
if len(username) < 3 {
error = "Username must be greater than 3 characters."
}
if error != "" {
utils.RenderTemplate(w, r, "register.html", map[string]interface{}{
"error": error,
}, nil)
return
}
// We're good, let's make it
user := models.NewUser(username, password)
err = db.Insert(user)
if err != nil {
fmt.Printf("[error] Could not insert user (%s)\n", err.Error())
return;
}
// Adminify the first user
id, err := db.SelectInt("SELECT lastval()")
if err == nil && id == 1 {
user.GroupId = 2
count, err = db.Update(user)
if err != nil {
fmt.Printf("[error] Could not adminify user (%s)\n", err.Error())
return;
}
}
http.Redirect(w, r, "/login", http.StatusFound)
return
}
utils.RenderTemplate(w, r, "register.html", nil, nil)
}
func main() {
// Get the config file
var config_path string
flag.StringVar(&config_path, "config", "gobb.conf", "Specifies the location of a config file")
run_migrations := flag.Bool("migrate", false, "Runs database migrations")
ign_migrations := flag.Bool("ignore-migrations", false, "Ignores an out of date database and runs the server anyways")
serve := flag.Bool("serve", false, "run server")
useradd := flag.Bool("add-user", false, "add a user")
var name, password string
flag.StringVar(&name, "name", "", "username to add")
flag.StringVar(&password, "password", "", "password new user")
group := flag.Int64("group", 0, "group of new user (<0 is special group, 0 is default, 1 is mod, 2 is admin)")
flag.Parse()
config.GetConfig(config_path)
// Do we need to run migrations?
latest_db_version, migrations, err := utils.GetMigrationInfo()
if len(migrations) != 0 && *run_migrations {
fmt.Println("[notice] Running database migrations:\n")
err = utils.RunMigrations(latest_db_version)
if err != nil {
fmt.Printf("[error] Could not run migrations (%s)\n", err.Error())
return
}
fmt.Println("\n[notice] Database migration successful!")
} else if len(migrations) != 0 && !(*ign_migrations) {
fmt.Println("Your database appears to be out of date. Please run migrations with --migrate or ignore this message with --ignore-migrations")
return
}
db := models.GetDbSession()
if *useradd {
user := models.NewUser(name, password)
user.GroupId = *group
err = db.Insert(user)
}
if !*serve {
return
}
// URL Routing!
r := mux.NewRouter()
r.StrictSlash(true)
r.HandleFunc("/", controllers.Index)
r.HandleFunc("/register", controllers.Register)
r.HandleFunc("/login", controllers.Login)
r.HandleFunc("/logout", controllers.Logout)
r.HandleFunc("/admin", controllers.Admin)
r.HandleFunc("/admin/boards", controllers.AdminBoards)
r.HandleFunc("/admin/users/{id:[0-9]+}", controllers.AdminUser)
r.HandleFunc("/admin/users", controllers.AdminUsers)
r.HandleFunc("/action/stick", controllers.ActionStickThread)
r.HandleFunc("/action/lock", controllers.ActionLockThread)
r.HandleFunc("/action/delete", controllers.ActionDeleteThread)
r.HandleFunc("/action/move", controllers.ActionMoveThread)
r.HandleFunc("/action/mark_read", controllers.ActionMarkAllRead)
r.HandleFunc("/action/edit", controllers.PostEditor)
r.HandleFunc("/board/{id:[0-9]+}", controllers.Board)
r.HandleFunc("/board/{board_id:[0-9]+}/new", controllers.PostEditor)
r.HandleFunc("/board/{board_id:[0-9]+}/{post_id:[0-9]+}", controllers.Thread)
r.HandleFunc("/user/{id:[0-9]+}", controllers.User)
r.HandleFunc("/user/{id:[0-9]+}/settings", controllers.UserSettings)
// Handle static files
selected_template, _ := models.GetStringSetting("template")
base_path, _ := config.Config.GetString("gobb", "base_path")
if selected_template == "default" {
pkg, _ := build.Import("github.com/stevenleeg/gobb/gobb", ".", build.FindOnly)
static_path := filepath.Join(pkg.SrcRoot, pkg.ImportPath, "../templates")
r.PathPrefix("/static/").Handler(http.FileServer(http.Dir(static_path)))
} else {
static_path := filepath.Join(base_path, "templates", selected_template)
r.PathPrefix("/static/").Handler(http.FileServer(http.Dir(static_path)))
}
// User provided static files
static_path, err := config.Config.GetString("gobb", "base_path")
if err == nil {
r.PathPrefix("/assets/").Handler(http.FileServer(http.Dir(static_path)))
}
http.Handle("/", r)
port, err := config.Config.GetString("gobb", "port")
fmt.Println("[notice] Starting server on port " + port)
http.ListenAndServe(":"+port, nil)
}