|
@@ -8,19 +8,26 @@ import (
|
|
|
"golang.org/x/crypto/bcrypt"
|
|
|
)
|
|
|
|
|
|
+// An User represents a user of the application.
|
|
|
type User struct {
|
|
|
ID string `sql:"uid"`
|
|
|
- UserName string `sql:"username"`
|
|
|
+ Name string `sql:"username"`
|
|
|
Email string `sql:"email"`
|
|
|
DateCreated string `sql:"date_created"`
|
|
|
PasswordHash string `sql:"password"`
|
|
|
}
|
|
|
|
|
|
+func (u User) String() string {
|
|
|
+ return fmt.Sprintf("{User: %s <%s>}", u.Name, u.Email)
|
|
|
+}
|
|
|
+
|
|
|
+// DB represents the application database. It provides high-level methods to get and manipulate objects.
|
|
|
type DB struct {
|
|
|
PgDB *pg.DB
|
|
|
Schema string
|
|
|
}
|
|
|
|
|
|
+// Setup will create schemas and tables as appropriate
|
|
|
func (db *DB) Setup() error {
|
|
|
_, err := db.PgDB.Exec(`CREATE SCHEMA IF NOT EXISTS users`)
|
|
|
if err != nil {
|
|
@@ -42,6 +49,7 @@ func (db *DB) Setup() error {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
+// UserAdd will add a user to the database
|
|
|
func (db *DB) UserAdd(username, email, pwd string) error {
|
|
|
stmt, err := db.PgDB.Prepare("INSERT INTO users.users (username, email, password) VALUES ($1, $2, $3)")
|
|
|
if err != nil {
|
|
@@ -58,6 +66,7 @@ func (db *DB) UserAdd(username, email, pwd string) error {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
+// GetUsers returns an array of Users
|
|
|
func (db *DB) GetUsers() ([]User, error) {
|
|
|
var users []User
|
|
|
_, err := db.PgDB.Query(&users, "SELECT * FROM users.users")
|
|
@@ -66,3 +75,33 @@ func (db *DB) GetUsers() ([]User, error) {
|
|
|
}
|
|
|
return users, err
|
|
|
}
|
|
|
+
|
|
|
+// GetUserByName Search for exact username
|
|
|
+func (db *DB) GetUserByName(username string) (User, error) {
|
|
|
+ var user User
|
|
|
+ _, err := db.PgDB.QueryOne(&user, `SELECT * FROM users.users WHERE username = ?`, username)
|
|
|
+ if err != nil {
|
|
|
+ return user, errors.Wrap(err, "User fetch failed")
|
|
|
+ }
|
|
|
+ return user, nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetUserByEmail searches for exact email
|
|
|
+func (db *DB) GetUserByEmail(email string) (User, error) {
|
|
|
+ var user User
|
|
|
+ _, err := db.PgDB.QueryOne(&user, "SELECT * FROM users.users WHERE email = ?", email)
|
|
|
+ if err != nil {
|
|
|
+ return user, errors.Wrap(err, "User fetch failed")
|
|
|
+ }
|
|
|
+ return user, nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetUserByID search for exact uid
|
|
|
+func (db *DB) GetUserByID(uid string) (User, error) {
|
|
|
+ var user User
|
|
|
+ _, err := db.PgDB.QueryOne(&user, "SELECT * FROM users.users WHERE uid=$1", uid)
|
|
|
+ if err != nil {
|
|
|
+ return user, errors.Wrap(err, "User fetch failed")
|
|
|
+ }
|
|
|
+ return user, nil
|
|
|
+}
|