mirror of https://github.com/statping/statping
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
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 }
|