ictgj-voting/model_teams.go

package main

import (
	"errors"

	"github.com/pborman/uuid"
)

type Team struct {
	UUID    string
	Name    string
	Members []TeamMember
	Game    *Game
}

type TeamMember struct {
	UUID    string
	Name    string
	SlackId string
	Twitter string
	Email   string
}

func dbCreateNewTeam(nm string) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	// Generate a UUID
	uuid := uuid.New()
	teamPath := []string{"teams", uuid}

	if err := db.MkBucketPath(teamPath); err != nil {
		return err
	}
	if err := db.SetValue(teamPath, "name", nm); err != nil {
		return err
	}
	if err := db.MkBucketPath(append(teamPath, "members")); err != nil {
		return err
	}
	gamePath := append(teamPath, "game")
	if err := db.MkBucketPath(gamePath); err != nil {
		return err
	}
	if err := db.SetValue(append(gamePath), "name", ""); err != nil {
		return err
	}
	return db.MkBucketPath(append(gamePath, "screenshots"))
}

func dbIsValidTeam(id string) bool {
	var err error
	if err = openDatabase(); err != nil {
		return false
	}
	defer closeDatabase()

	teamPath := []string{"teams"}
	if teamUids, err := db.GetBucketList(teamPath); err == nil {
		for _, v := range teamUids {
			if v == id {
				return true
			}
		}
	}
	return false
}

func dbGetAllTeams() []Team {
	var ret []Team
	var err error
	if err = openDatabase(); err != nil {
		return ret
	}
	defer closeDatabase()

	teamPath := []string{"teams"}
	var teamUids []string
	if teamUids, err = db.GetBucketList(teamPath); err != nil {
		return ret
	}
	for _, v := range teamUids {
		if tm := dbGetTeam(v); tm != nil {
			ret = append(ret, *tm)
		}
	}
	return ret
}

func dbGetTeam(id string) *Team {
	var err error
	if err = openDatabase(); err != nil {
		return nil
	}
	defer closeDatabase()

	teamPath := []string{"teams", id}
	tm := new(Team)
	tm.UUID = id
	if tm.Name, err = db.GetValue(teamPath, "name"); err != nil {
		return nil
	}
	tm.Members, _ = dbGetTeamMembers(id)
	tm.Game = dbGetTeamGame(id)
	return tm
}

func dbGetTeamByName(nm string) *Team {
	var err error
	if err = openDatabase(); err != nil {
		return nil
	}
	defer closeDatabase()

	teamPath := []string{"teams"}
	var teamUids []string
	if teamUids, err = db.GetBucketList(teamPath); err != nil {
		for _, v := range teamUids {
			var name string
			if name, err = db.GetValue(append(teamPath, v), "name"); name == nm {
				return dbGetTeam(v)
			}
		}
	}
	return nil
}

func dbUpdateTeam(id string, tm *Team) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	teamPath := []string{"teams", id}
	return db.SetValue(teamPath, "name", tm.Name)
}

func dbDeleteTeam(id string) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	teamPath := []string{"teams"}
	return db.DeleteBucket(teamPath, id)
}

func dbEditTeamGame(teamid, name string) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	gamePath := []string{"teams", teamid, "game"}
	return db.SetValue(gamePath, "name", name)
}

func dbAddTeamMember(teamid, mbrName, mbrEmail, mbrSlack, mbrTwitter string) error {
	// First check if this member already exists on this team
	mbrs, _ := dbGetTeamMembers(teamid)
	if len(mbrs) > 0 {
		for i := range mbrs {
			if mbrs[i].Name == mbrName {
				return dbUpdateTeamMember(teamid, mbrs[i].UUID, mbrName, mbrEmail, mbrSlack, mbrTwitter)
			}
		}
	}
	// It's really an add
	mbrId := uuid.New()
	return dbUpdateTeamMember(teamid, mbrId, mbrName, mbrEmail, mbrSlack, mbrTwitter)
}

func dbUpdateTeamMember(teamid, mbrId, mbrName, mbrEmail, mbrSlack, mbrTwitter string) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	mbrPath := []string{"teams", teamid, "members", mbrId}
	if db.SetValue(mbrPath, "name", mbrName) != nil {
		return err
	}
	if db.SetValue(mbrPath, "slackid", mbrSlack) != nil {
		return err
	}
	if db.SetValue(mbrPath, "twitter", mbrTwitter) != nil {
		return err
	}
	if db.SetValue(mbrPath, "email", mbrEmail) != nil {
		return err
	}
	return nil
}

func dbGetTeamMembers(teamid string) ([]TeamMember, error) {
	var ret []TeamMember
	var err error
	if err = openDatabase(); err != nil {
		return ret, err
	}
	defer closeDatabase()

	teamPath := []string{"teams", teamid, "members"}
	var memberUuids []string
	if memberUuids, err = db.GetBucketList(teamPath); err == nil {
		for _, v := range memberUuids {
			var mbr *TeamMember
			if mbr, err = dbGetTeamMember(teamid, v); err == nil {
				ret = append(ret, *mbr)
			}
		}
	}
	return ret, err
}

func dbGetTeamMember(teamid, mbrid string) (*TeamMember, error) {
	var err error
	if err = openDatabase(); err != nil {
		return nil, err
	}
	defer closeDatabase()

	mbr := new(TeamMember)
	teamMbrPath := []string{"teams", teamid, "members", mbrid}
	mbr.UUID = mbrid
	if mbr.Name, err = db.GetValue(teamMbrPath, "name"); err != nil {
		return nil, err
	}
	if mbr.SlackId, err = db.GetValue(teamMbrPath, "slackid"); err != nil {
		return nil, err
	}
	if mbr.Twitter, err = db.GetValue(teamMbrPath, "twitter"); err != nil {
		return nil, err
	}
	if mbr.Email, err = db.GetValue(teamMbrPath, "email"); err != nil {
		return nil, err
	}
	return mbr, err
}

// This function returns the team for a specific member
func dbGetMembersTeam(mbrid string) (*Team, error) {
	var err error
	if err = openDatabase(); err != nil {
		return nil, err
	}
	defer closeDatabase()

	teams := dbGetAllTeams()
	for i := range teams {
		var tmMbrs []TeamMember
		tmMbrs, err = dbGetTeamMembers(teams[i].UUID)
		if err == nil {
			for j := range tmMbrs {
				if tmMbrs[j].UUID == mbrid {
					return &teams[i], nil
				}
			}
		}
	}
	return nil, errors.New("Unable to find team member")
}

// This function searches all teams for a member with the given name
func dbGetTeamMembersByName(mbrName string) ([]TeamMember, error) {
	var ret []TeamMember
	var err error
	if err = openDatabase(); err != nil {
		return ret, err
	}
	defer closeDatabase()

	teams := dbGetAllTeams()
	for i := range teams {
		var tmMbrs []TeamMember
		tmMbrs, err = dbGetTeamMembers(teams[i].UUID)
		if err == nil {
			for j := range tmMbrs {
				if tmMbrs[j].Name == mbrName {
					ret = append(ret, tmMbrs[j])
				}
			}
		}
	}
	if len(ret) == 0 {
		return nil, errors.New("Couldn't find any members with the requested name")
	}
	return ret, nil
}

func dbDeleteTeamMember(teamId, mbrId string) error {
	var err error
	if err = openDatabase(); err != nil {
		return err
	}
	defer closeDatabase()

	teamPath := []string{"teams", teamId, "members"}
	return db.DeleteBucket(teamPath, mbrId)
}