memberDB/handlers/members.go

package handlers

import (
	"errors"
	"fmt"

	"gitea.tecamino.com/paadi/memberDB/crypto"
	"gitea.tecamino.com/paadi/memberDB/models"
	"gitea.tecamino.com/paadi/memberDB/utils"
)

// AddNewMember adds a new member to memeber table at least fist, las name and birthday has to be entered
func (dh *DatabaseHandler) AddNewMember(members ...models.Member) error {
	if !dh.DatabaseOpened() {
		return errors.New("database not opened")
	}

	for _, member := range members {

		exists, err := dh.memberExists(member)
		if err != nil {
			return err
		} else if exists {
			return fmt.Errorf("member %s %s %s exists already", member.FirstName, member.LastName, member.Birthday)
		}

		//check correct birtday format
		if member.Birthday != "" && !utils.IsValidBirthday(member.Birthday) {
			return errors.New("incorrect birthday format")
		}

		//check correct email format
		if member.Email != "" && !utils.IsValidEmail(member.Email) {
			return errors.New("incorrect email format")
		}

		now := utils.GetCurrentTime(dh.timeLocation)
		member.FirstVisit = now
		member.LastVisit = now

		member.Encrypt(dh.token)

		err = dh.database.AddNewColum(&member)
		if err != nil {
			return err
		}
	}
	return nil
}

// DeleteMember removes members by given ids
func (dh *DatabaseHandler) DeleteMember(ids ...uint) error {
	if !dh.DatabaseOpened() {
		return errors.New("database not opened")
	}

	if len(ids) == 0 {
		return errors.New("no ids given to be deleted")
	}
	return dh.database.DeleteById(&models.Member{}, "", ids...)
}

// GetMember returns one member by given id
func (dh *DatabaseHandler) GetMember(id uint) (members []models.Member, err error) {
	if !dh.DatabaseOpened() {
		return members, errors.New("database not opened")
	}

	err = dh.database.GetById(&members, "", id)
	if err != nil {
		return
	}

	for i := range members {
		if err := members[i].Decrypt(dh.token); err != nil {
			return members, err
		}
	}
	return
}

// UpdateMember updates/overrides all information given meber id
func (dh *DatabaseHandler) UpdateMember(id int, member models.Member) (err error) {
	if !dh.DatabaseOpened() {
		return errors.New("database not opened")
	}

	if member.FirstName != "" {
		member.FirstNameHash = dh.hashField(member.FirstName)
		member.FirstName, err = crypto.Encrypt(member.FirstName, dh.token)
		if err != nil {
			return
		}
	}

	if member.LastName != "" {
		member.LastNameHash = dh.hashField(member.LastName)
		member.LastName, err = crypto.Encrypt(member.LastName, dh.token)
		if err != nil {
			return
		}
	}

	//check correct birtday format
	if member.Birthday != "" && utils.IsValidBirthday(member.Birthday) {
		member.BirthdayHash = dh.hashField(member.BirthdayHash)
		member.Birthday, err = crypto.Encrypt(member.Birthday, dh.token)
		if err != nil {
			return err
		}
	} else if member.Birthday != "" {
		return errors.New("incorrect birthday format")
	}

	if member.Address != "" {
		member.Address, err = crypto.Encrypt(member.Address, dh.token)
		if err != nil {
			return err
		}
	}

	if member.Zip != "" {
		member.Zip, err = crypto.Encrypt(member.Zip, dh.token)
		if err != nil {
			return err
		}
	}

	if member.Town != "" {
		member.Town, err = crypto.Encrypt(member.Town, dh.token)
		if err != nil {
			return err
		}
	}

	if member.Phone != "" {
		member.Phone, err = crypto.Encrypt(member.Phone, dh.token)
		if err != nil {
			return err
		}
	}

	//check correct email format
	if member.Email != "" && utils.IsValidEmail(member.Email) {
		member.Email, err = crypto.Encrypt(member.Email, dh.token)
		if err != nil {
			return err
		}
	} else if member.Email != "" {
		return errors.New("incorrect email format")
	}

	if member.FirstVisit != "" {
		member.FirstVisit, err = crypto.Encrypt(member.FirstVisit, dh.token)
		if err != nil {
			return err
		}
	}

	if member.LastVisit != "" {
		member.LastVisit, err = crypto.Encrypt(member.LastVisit, dh.token)
		if err != nil {
			return err
		}
	}

	if member.Comment != "" {
		member.Comment, err = crypto.Encrypt(member.Comment, dh.token)
		if err != nil {
			return err
		}
	}
	return dh.database.UpdateValuesById(&member, "", uint(member.Id))
}

// memberExists helper to check wheter member already exists
func (dh *DatabaseHandler) memberExists(checkMember models.Member) (bool, error) {
	if !dh.DatabaseOpened() {
		return false, errors.New("database not opened")
	}

	var members []models.Member
	if !dh.database.Exists(&members, "", "birthdayHash", dh.hashField(checkMember.Birthday), false) {
		return false, nil
	}

	for _, member := range members {
		if dh.hashField(checkMember.FirstName) == member.FirstNameHash && dh.hashField(checkMember.LastName) == member.LastNameHash && dh.hashField(checkMember.Birthday) == member.BirthdayHash {
			return true, nil
		}
	}

	return false, nil
}