Import sources from Kontor

pkg/admin/user_dao.go

@@ -0,0 +1,156 @@
+package admin
+
+import (
+	"gitlab.ingenieurbuero-peetz.de/tpeetz/kalorienmanager.git/pkg/dao"
+	"log"
+
+	"github.com/gin-gonic/gin"
+
+	"golang.org/x/crypto/bcrypt"
+	mgo "gopkg.in/mgo.v2"
+	"gopkg.in/mgo.v2/bson"
+)
+
+// UserDAO extends the type BaseDAO.
+type UserDAO struct {
+	Db dao.BaseDAO
+}
+
+const (
+	// USERCOLLECTION defines the collection name for storing application user data.
+	USERCOLLECTION = "user"
+	// USERMODEL defines the name of the user data model.
+	USERMODEL = "kalorienmanager.admin.user"
+)
+
+// HashPassword returns the encrypted password from password string.
+func HashPassword(password string) (string, error) {
+	bytes, err := bcrypt.GenerateFromPassword([]byte(password), 14)
+	return string(bytes), err
+}
+
+// CheckPasswordHash returns if password correlates with pasword hash.
+func CheckPasswordHash(password, hash string) bool {
+	err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
+	return err == nil
+}
+
+// FindAll retrieves the list of users from the database.
+func (m *UserDAO) FindAll() ([]User, error) {
+	m.Db.Connect()
+	var users []User
+	err := m.Db.MongoDb.C(USERCOLLECTION).Find(bson.M{"model": USERMODEL}).All(&users)
+	return users, err
+}
+
+// FindByID returns a user with given id or returns the error.
+func (m *UserDAO) FindByID(id string) (User, error) {
+	m.Db.Connect()
+	var user User
+	err := m.Db.MongoDb.C(USERCOLLECTION).FindId(bson.ObjectIdHex(id)).One(&user)
+	return user, err
+}
+
+// FindByUsername returns a user with given name or returns the error.
+func (m *UserDAO) FindByUsername(username string) (User, error) {
+	m.Db.Connect()
+	var user User
+	err := m.Db.MongoDb.C(USERCOLLECTION).Find(bson.M{"username": username, "model": USERMODEL}).One(&user)
+	return user, err
+}
+
+// Insert a user into database.
+func (m *UserDAO) Insert(user User) error {
+	m.Db.Connect()
+	user.Model = USERMODEL
+	err := m.Db.MongoDb.C(USERCOLLECTION).Insert(&user)
+	return err
+}
+
+// Upsert a user into database.
+func (m *UserDAO) Upsert(user User) (*mgo.ChangeInfo, error) {
+	m.Db.Connect()
+	user.Model = USERMODEL
+	info, err := m.Db.MongoDb.C(USERCOLLECTION).Upsert(bson.M{"username": user.Username}, &user)
+	return info, err
+}
+
+// Update an existing user.
+func (m *UserDAO) Update(user User) error {
+	m.Db.Connect()
+	err := m.Db.MongoDb.C(USERCOLLECTION).UpdateId(user.ID, &user)
+	return err
+}
+
+// Delete an existing user.
+func (m *UserDAO) Delete(user User) error {
+	m.Db.Connect()
+	err := m.Db.MongoDb.C(USERCOLLECTION).Remove(&user)
+	return err
+}
+
+// IsUserValid checks if the username and password combination is valid
+func (m *UserDAO) IsUserValid(username, password string) bool {
+	if gin.IsDebugging() {
+		log.Printf("UserDAO.IsUserValid(%s)", username)
+	}
+	user, err := m.FindByUsername(username)
+	if gin.IsDebugging() {
+		log.Printf("UserDAO.IsUserValid: %v, %v", user, err)
+	}
+	if &user == nil || err != nil {
+		return false
+	}
+	return CheckPasswordHash(password, user.Password)
+}
+
+// IsUserAdmin checks if user identified by name has admin rights.
+func (m *UserDAO) IsUserAdmin(username string) bool {
+	user, err := m.FindByUsername(username)
+	if &user == nil || err != nil {
+		return false
+	}
+	return user.IsAdmin
+}
+
+// IsUsernameAvailable checks if the supplied username is available.
+func (m *UserDAO) IsUsernameAvailable(username string) bool {
+	user, err := m.FindByUsername(username)
+	if &user == nil || err != nil {
+		return true
+	}
+	return false
+}
+
+// adduser adds user in database. TODO is method necessary?
+func addUser(username, password, firstname, lastname string, isAdmin bool) (*User, error) {
+	//	passwordHash, _ := HashPassword(password)
+	//	var user = User{Username: username, Password: passwordHash, Firstname: firstname, Lastname: lastname, IsAdmin: isAdmin, Model: USERMODEL}
+	//	conn, err := util.GetCollection(USERCOLLECTION)
+	//	err = conn.Insert(&user)
+	//	if err != nil {
+	//		return nil, err
+	//	}
+	//	return &user, nil
+	return nil, nil
+}
+
+// changeuser changes user in database. TODO is method necessary?
+func changeUser(userid, username, password, firstname, lastname string, isAdmin bool) (*User, error) {
+	//	conn, err := util.GetCollection(USERCOLLECTION)
+	//	var user *User
+	//	err = conn.Find(bson.M{"_id": bson.ObjectIdHex(userid)}).One(&user)
+	//	if err != nil {
+	//		return nil, err
+	//	}
+	//	var change bson.M
+	//	if password != "" {
+	//		passwordHash, _ := HashPassword(password)
+	//		change = bson.M{"$set": bson.M{"username": username, "password": passwordHash, "firstname": firstname, "lastname": lastname, "is_admin": isAdmin}}
+	//	} else {
+	//		change = bson.M{"$set": bson.M{"username": username, "firstname": firstname, "lastname": lastname, "is_admin": isAdmin}}
+	//	}
+	//	err = conn.Update(bson.M{"_id": bson.ObjectIdHex(userid)}, change)
+	//	return user, nil
+	return nil, nil
+}