energylink/pkg/models/account.go

package models

import (
	"errors"
	"time"
)

type CollaboratorLink struct {
	AccountId  int
	Role       string
	ModifiedOn time.Time
	CreatedOn  time.Time
}

type UserAccount struct {
	Id              int `gorethink:"id"`
	UserName        string
	Login           string
	Email           string
	Password        string
	NextDashboardId int
	UsingAccountId  int
	Collaborators   []CollaboratorLink
	CreatedOn       time.Time
	ModifiedOn      time.Time
}

func (account *UserAccount) AddCollaborator(accountId int) error {
	for _, collaborator := range account.Collaborators {
		if collaborator.AccountId == accountId {
			return errors.New("Collaborator already exists")
		}
	}

	account.Collaborators = append(account.Collaborators, CollaboratorLink{
		AccountId:  accountId,
		Role:       "admin",
		CreatedOn:  time.Now(),
		ModifiedOn: time.Now(),
	})

	return nil
}