statping/core/groups.go

package core

import (
	"github.com/hunterlong/statping/database"
	"github.com/hunterlong/statping/types"
	"sort"
)

type Group struct {
	database.Grouper
}

// SelectGroups returns all groups
func SelectGroups(includeAll bool, auth bool) []database.Grouper {
	var validGroups []database.Grouper

	groups := database.AllGroups()

	for _, g := range groups {
		if !g.Model().Public.Bool {
			if auth {
				validGroups = append(validGroups, g)
			}
		} else {
			validGroups = append(validGroups, g)
		}
	}
	sort.Sort(GroupOrder(validGroups))
	if includeAll {
		emptyGroup := &Group{}
		validGroups = append(validGroups, emptyGroup)
	}
	return validGroups
}

// SelectGroup returns a *core.Group
func SelectGroup(id int64) *types.Group {
	for _, g := range SelectGroups(true, true) {
		if g.Model().Id == id {
			return g.Model()
		}
	}
	return nil
}

// GroupOrder will reorder the groups based on 'order_id' (Order)
type GroupOrder []database.Grouper

// Sort interface for resorting the Groups in order
func (c GroupOrder) Len() int           { return len(c) }
func (c GroupOrder) Swap(i, j int)      { c[i], c[j] = c[j], c[i] }
func (c GroupOrder) Less(i, j int) bool { return c[i].Model().Order < c[j].Model().Order }