statping/users.go

package main

import (
	"golang.org/x/crypto/bcrypt"
	"net/http"
	"time"
)

type User struct {
	Id        int64     `db:"id,omitempty" json:"id"`
	Username  string    `db:"username" json:"username"`
	Password  string    `db:"password" json:"-"`
	Email     string    `db:"email" json:"-"`
	ApiKey    string    `db:"api_key" json:"api_key"`
	ApiSecret string    `db:"api_secret" json:"-"`
	Admin     bool      `db:"administrator" json:"admin"`
	CreatedAt time.Time `db:"created_at" json:"created_at"`
}

func SessionUser(r *http.Request) *User {
	session, _ := store.Get(r, cookieKey)
	if session == nil {
		return nil
	}
	uuid := session.Values["user_id"]
	var user *User
	col := dbSession.Collection("users")
	res := col.Find("id", uuid)
	res.One(&user)
	return user
}

func SelectUser(id int64) (*User, error) {
	var user User
	col := dbSession.Collection("users")
	res := col.Find("id", id)
	err := res.One(&user)
	return &user, err
}

func SelectUsername(username string) (*User, error) {
	var user User
	col := dbSession.Collection("users")
	res := col.Find("username", username)
	err := res.One(&user)
	return &user, err
}

func (u *User) Delete() error {
	col := dbSession.Collection("users")
	user := col.Find("id", u.Id)
	return user.Delete()
}

func (u *User) Create() (int64, error) {
	u.CreatedAt = time.Now()
	u.Password = HashPassword(u.Password)
	u.ApiKey = NewSHA1Hash(5)
	u.ApiSecret = NewSHA1Hash(10)
	col := dbSession.Collection("users")
	uuid, err := col.Insert(u)
	if uuid == nil {
		return 0, err
	}
	OnNewUser(u)
	return uuid.(int64), err
}

func SelectAllUsers() ([]User, error) {
	var users []User
	col := dbSession.Collection("users").Find()
	err := col.All(&users)
	return users, err
}

func AuthUser(username, password string) (*User, bool) {
	var auth bool
	user, err := SelectUsername(username)
	if err != nil {
		return nil, false
	}
	if CheckHash(password, user.Password) {
		auth = true
	}
	return user, auth
}

func CheckHash(password, hash string) bool {
	err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
	return err == nil
}