func (u *User) GetUser(r *http.Request, Id int64) (UserReturn, error) {
userModel := model.User{}
userData, err := userModel.GetUser(r, Id)
if err != nil {
return UserReturn{}, err
}
user := UserReturn{
Email: userData.Email,
Name: userData.Name,
Phone: userData.Phone,
Role: userData.Role,
}
return user, nil
}
func (u *User) Login(w http.ResponseWriter, r *http.Request) (LoginReturn, error) {
type Message struct {
Email string `json:"email"`
Password string `json:"password"`
}
var m Message
decoder := json.NewDecoder(r.Body)
err := decoder.Decode(&m)
if err != nil {
//handle err
log.Println(err)
}
userModel := model.User{}
userModel.Email = m.Email
//find user
user, err := userModel.GetLoginData(w, r)
if err != nil {
//handle err
return LoginReturn{}, err
}
//match passwords
err = bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(m.Password))
if err != nil {
//passwords don't match
return LoginReturn{}, err
}
//everything checks out!
var l LoginReturn
l = LoginReturn{
Id: user.Id,
Role: user.Role,
}
return l, nil
}
func (u *User) CreateAdmin(r *http.Request) (bool, error) {
//populate user data
userModel := model.User{}
userModel.Email = "*****@*****.**"
plainPassword := "******"
//encrypt password
password, err := bcrypt.GenerateFromPassword([]byte(plainPassword), 10)
if err != nil {
panic(err)
}
//populate model
userModel.Password = string(password)
userModel.Role = 2
//create new user
err = userModel.CreateUser(r)
if err != nil {
return false, err
}
return true, nil
}
func (u *User) CheckEmail(w http.ResponseWriter, r *http.Request) (bool, error) {
type Message struct {
Email string `json:"email"`
}
var m Message
decoder := json.NewDecoder(r.Body)
err := decoder.Decode(&m)
if err != nil {
//handle err
}
userModel := model.User{}
emailStatus, err := userModel.CheckEmail(w, r, m.Email)
if err != nil {
return false, err
}
return emailStatus, nil
}
func (u *User) CreateUser(w http.ResponseWriter, r *http.Request) (bool, error) {
type Message struct {
Email string `json:"email"`
Password string `json:"password"`
}
var m Message
decoder := json.NewDecoder(r.Body)
err := decoder.Decode(&m)
if err != nil {
//handle err
}
//populate user data
userModel := model.User{}
userModel.Email = m.Email
//encrypt password
password, err := bcrypt.GenerateFromPassword([]byte(m.Password), 10)
if err != nil {
panic(err)
}
//populate model
userModel.Password = string(password)
userModel.Role = 1
//make sure email doesn't exist
emailStatus, err := u.CheckEmail(w, r)
if emailStatus == true || err != nil {
return false, nil
}
//create new user
err = userModel.CreateUser(r)
if err != nil {
return false, err
}
return true, nil
}