-
Notifications
You must be signed in to change notification settings - Fork 1
/
user.go
83 lines (68 loc) · 1.75 KB
/
user.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package main
import (
"code.google.com/p/go-uuid/uuid"
"crypto/sha1"
"fmt"
"gopkg.in/gorp.v1"
"strings"
"time"
)
const getUsersQuery = "SELECT id, name, password_hash, created_at FROM users ORDER BY name"
const getSingleUserQuery = "SELECT id, name, password_hash, created_at FROM users WHERE name = ?"
type User struct {
Id int64 `db:"id"`
Name string `db:"name"`
Password string `db:"-"`
PasswordHash string `db:"password_hash"`
CreatedAt time.Time `db:"created_at"`
}
func NewUser(name string) *User {
user := User{Id: 0, Name: name}
user.Password = newPassword()
return &user
}
func GetUser(name string) (User, error) {
var user User
err := DbSelectOne(&user, getSingleUserQuery, name)
return user, err
}
func GetUsers() ([]User, error) {
var users []User
_, err := DbSelectAll(&users, getUsersQuery)
return users, err
}
func IsValidUser(name string, password string) bool {
var user User
if err := DbSelectOne(&user, getSingleUserQuery, name); err != nil {
return false
}
return hashString(password) == user.PasswordHash
}
func (self *User) isNew() bool {
return self.Id == 0
}
func hashString(plaintext string) string {
hash := sha1.New()
hash.Write([]byte(plaintext))
return fmt.Sprintf("%x", hash.Sum(nil))
}
func newPassword() string {
return strings.Replace(uuid.New(), "-", "", -1)
}
func (self *User) ResetPassword() error {
self.Password = newPassword()
self.PasswordHash = hashString(self.Password)
return self.Save()
}
func (self *User) Save() error {
if self.isNew() {
return DbInsert(self)
}
_, err := DbUpdate(self)
return err
}
func (self *User) PreInsert(sqlExec gorp.SqlExecutor) error {
self.CreatedAt = time.Now()
self.PasswordHash = hashString(self.Password)
return nil
}