func (this *LoginRouter) Get() {
// Set const
this.Data["App_Name"] = App_Name
// xsrf
this.Data["xsrf"] = template.HTML(this.XsrfFormHtml())
// set token in case twice submit
Token := models.Token()
this.SetSession("Token", Token)
// set cookie not bot in case bots
Cookies := models.RandString(20)
this.SetSession("Cookie", Cookies)
this.Data["Token"] = Token
this.Data["Cookie"] = Cookies
// Get referral
Refer := this.Input().Get("username")
if models.UserExist(Refer) {
this.Data["Refer"] = Refer
} else {
models.Log(models.Log_Struct{"error", "Login:"******"No such referer")})
}
// Get User Session
var user Session_User
u := this.GetSession("_User")
if u != nil {
user = u.(Session_User)
this.Data["User"] = user
} else {
this.Data["User"] = false
}
// Show the login page or Error page
var showLogin bool
v := this.GetSession("ShowLogin")
if v != nil {
showLogin = v.(bool)
this.DelSession("ShowLogin")
} else {
showLogin = true
}
this.Data["ShowLogin"] = showLogin
// Errors
if !showLogin {
e := this.GetSession("Error")
if e != nil {
this.Data["Error"] = GetError(e)
this.DelSession("Error")
}
}
this.TplNames = "login.html"
}
func (this *RegisterRouter) Post() {
// Get user inputs
inputs := this.Input()
// Check token in case twice submit
var token string
token_sess := this.GetSession("Token")
if token_sess != nil {
this.DelSession("Token")
token = fmt.Sprintf("%d", token_sess.(int64))
}
if token != inputs.Get("token") {
this.SetSession("Error", ERROR_TWICESUBMIT)
models.Log(models.Log_Struct{"error", "Register:", errors.New("Submit twice")})
this.fail()
return
}
// Check cookie in case bots
cookie_sess := this.GetSession("Cookie")
if cookie_sess != nil {
this.DelSession("Cookie")
cookie := cookie_sess.(string)
if cookie != this.Ctx.GetCookie("nobot") {
this.SetSession("Error", ERROR_CAPTCHA)
models.Log(models.Log_Struct{"error", "Register:", errors.New("No bot is allowed")})
this.fail()
return
}
}
// Validate user inputs
username := strings.TrimSpace(inputs.Get("username"))
password := strings.TrimSpace(inputs.Get("password"))
re_password := strings.TrimSpace(inputs.Get("re-password"))
fundpass := strings.TrimSpace(inputs.Get("fundpassword"))
re_fundpass := strings.TrimSpace(inputs.Get("re-fundpassword"))
email := strings.TrimSpace(inputs.Get("email"))
dateofbirth := strings.TrimSpace(inputs.Get("birth"))
// Check referral
refer := strings.TrimSpace(inputs.Get("refer"))
if refer != "" && models.ValidString(refer) {
if !models.UserExist(refer) {
models.Log(models.Log_Struct{"info", "Register:", errors.New("Referral user not exist.")})
this.SetSession("Error", ERROR_REFERNOTEXIST)
this.fail()
return
}
}
// Validate user inputs, set sessions and redirect
if models.ValidString(username) && models.ValidString(password) && models.ValidString(re_password) && models.ValidString(fundpass) && models.ValidString(re_fundpass) && password == re_password && fundpass == re_fundpass && models.ValidEmail(email) && models.ValidBirth(dateofbirth) {
// Check if user exist
if models.UserExist(username) {
models.Log(models.Log_Struct{"info", "Register:", errors.New("User already exist.")})
this.SetSession("Error", ERROR_USEREXIST)
this.fail()
return
}
// Check if email exist
if models.EmailExist(email) {
models.Log(models.Log_Struct{"info", "Register:", errors.New("Email already exist.")})
this.SetSession("Error", ERROR_EMAILEXIST)
this.fail()
return
}
// Generate new address for new user
address, err := models.NewAddress(username)
if err != nil {
models.Log(models.Log_Struct{"info", "Register:", err})
this.fail()
this.SetSession("Error", ERROR_ADDRESS)
return
}
// Insert new user to DB
ok := models.NewUser(models.User{Username: username, Password: models.EncodePass(password), Fundpassword: models.EncodePass(fundpass), Email: email, Btcaddress: address, Birth: dateofbirth, Referral: refer})
if !ok {
models.Log(models.Log_Struct{"info", "Register:", errors.New("Unable to insert user, need to delete from wallet.")})
err = models.Archive(address)
if err != nil {
models.Log(models.Log_Struct{"warn", "Register:", errors.New("Unable to archive.")})
} else {
models.Log(models.Log_Struct{"info", "Register:", errors.New("Succeed archiving address.")})
}
this.fail()
this.SetSession("Error", ERROR_DB)
return
}
this.succ()
return
}
models.Log(models.Log_Struct{"info", "Register:", errors.New("Failed, invalid data")})
this.SetSession("Error", ERROR_INVALIDINPUT)
this.fail()
}
func (this *ForgetRouter) Post() {
// Get inputs and validate
inputs := this.Input()
// Check token in case twice submit
var token string
token_sess := this.GetSession("Token")
if token_sess != nil {
this.DelSession("Token")
token = fmt.Sprintf("%d", token_sess.(int64))
}
if token != inputs.Get("token") {
this.SetSession("Error", ERROR_TWICESUBMIT)
models.Log(models.Log_Struct{"error", "Forget:", errors.New("Submit twice")})
this.fail()
return
}
// Check cookie in case bots
cookie_sess := this.GetSession("Cookie")
if cookie_sess != nil {
this.DelSession("Cookie")
cookie := cookie_sess.(string)
if cookie != this.Ctx.GetCookie("nobot") {
this.SetSession("Error", ERROR_CAPTCHA)
models.Log(models.Log_Struct{"error", "Forget:", errors.New("No bot is allowed")})
this.fail()
return
}
}
// Validate inputs
username := strings.TrimSpace(inputs.Get("username"))
dateofbirth := strings.TrimSpace(inputs.Get("birth"))
email := strings.TrimSpace(inputs.Get("email"))
if models.ValidString(username) && models.ValidEmail(email) && models.ValidBirth(dateofbirth) {
// Check if user exist
if !models.UserExist(username) {
models.Log(models.Log_Struct{"info", "Forget:", errors.New("User not exist.")})
this.SetSession("Error", ERROR_USERNOTEXIST)
this.fail()
return
}
// Check if birth matches
if !models.BirthMatch(username, dateofbirth) {
models.Log(models.Log_Struct{"info", "Forget:", errors.New("Birth not match.")})
this.SetSession("Error", ERROR_BIRTHNOTMATCH)
this.fail()
return
}
// Check if email mathces
if !models.EmailMatch(username, email) {
models.Log(models.Log_Struct{"info", "Forget:", errors.New("Email not match.")})
this.SetSession("Error", ERROR_EMAILNOTMATCH)
this.fail()
return
}
// Send Email to authenticate
authen := models.RandString(8)
if !models.SendEmail(email, "重设密码", username+": 请复制验证码，以完成重设密码操作---->", authen) {
models.Log(models.Log_Struct{"warn", "Forget:", errors.New("Cant send email to authen password reset.")})
this.SetSession("Error", ERROR_EMAILNOTSENT)
this.fail()
return
}
this.SetSession("Username", username)
this.SetSession("Authen", authen)
this.succ()
return
}
models.Log(models.Log_Struct{"info", "Forget:", errors.New("Failed, invalid data.")})
this.SetSession("Error", ERROR_INVALIDINPUT)
this.fail()
}
func (this *LoginRouter) Post() {
//fmt.Println(this.CheckXsrfCookie())
// Get inputs and validate
inputs := this.Input()
// Check token in case twice submit
var token string
token_sess := this.GetSession("Token")
if token_sess != nil {
this.DelSession("Token")
token = fmt.Sprintf("%d", token_sess.(int64))
}
if token != inputs.Get("token") {
this.SetSession("Error", ERROR_TWICESUBMIT)
models.Log(models.Log_Struct{"error", "Login:"******"Submit twice")})
this.fail()
return
}
// Check cookie in case bots
cookie_sess := this.GetSession("Cookie")
if cookie_sess != nil {
this.DelSession("Cookie")
cookie := cookie_sess.(string)
if cookie != this.Ctx.GetCookie("nobot") {
this.SetSession("Error", ERROR_CAPTCHA)
models.Log(models.Log_Struct{"error", "Login:"******"No bot is allowed")})
this.fail()
return
}
}
// Validate user inputs
username := strings.TrimSpace(inputs.Get("username"))
password := strings.TrimSpace(inputs.Get("password"))
if models.ValidString(username) && models.ValidString(password) {
// Check if user exist
if !models.UserExist(username) {
models.Log(models.Log_Struct{"info", "Login:"******"User not exist")})
this.SetSession("Error", ERROR_USERNOTEXIST)
this.fail()
return
}
// Check if password correct
if !models.PassMatch(username, password) {
models.Log(models.Log_Struct{"info", "Login:"******"Password incorrect")})
this.SetSession("Error", ERROR_PASSINCORRECT)
this.fail()
return
}
// Update IP
ip := strings.Split(this.Ctx.Request.RemoteAddr, ":")[0]
models.UpdateIP(username, ip)
// Get Uid
id := models.UidByUsername(username)
this.SetSession("_User", Session_User{Uid: id, Username: username, Ip: ip})
this.succ()
return
}
models.Log(models.Log_Struct{"info", "Login:"******"Failed, invalid data")})
this.SetSession("Error", ERROR_INVALIDINPUT)
this.fail()
}