gosora/common/group.go

package common

import (
    "database/sql"
    "encoding/json"

    qgen "github.com/Azareal/Gosora/query_gen"
)

var blankGroup = Group{ID: 0, Name: ""}

type GroupAdmin struct {
    ID        int
    Name      string
    Rank      string
    RankClass string
    CanEdit   bool
    CanDelete bool
}

// ! Fix the data races in the fperms
type Group struct {
    ID              int
    Name            string
    IsMod           bool
    IsAdmin         bool
    IsBanned        bool
    Tag             string
    Perms           Perms
    PermissionsText []byte
    PluginPerms     map[string]bool // Custom permissions defined by plugins. What if two plugins declare the same permission, but they handle them in incompatible ways? Very unlikely, we probably don't need to worry about this, the plugin authors should be aware of each other to some extent
    PluginPermsText []byte
    CanSee          []int // The IDs of the forums this group can see
    UserCount       int   // ! Might be temporary as I might want to lean on the database instead for this
}

type GroupStmts struct {
    updateGroup      *sql.Stmt
    updateGroupRank  *sql.Stmt
    updateGroupPerms *sql.Stmt
}

var groupStmts GroupStmts

func init() {
    DbInits.Add(func(acc *qgen.Accumulator) error {
        set := func(s string) *sql.Stmt {
            return acc.Update("users_groups").Set(s).Where("gid=?").Prepare()
        }
        groupStmts = GroupStmts{
            updateGroup:      set("name=?,tag=?"),
            updateGroupRank:  set("is_admin=?,is_mod=?,is_banned=?"),
            updateGroupPerms: set("permissions=?"),
        }
        return acc.FirstError()
    })
}

func (g *Group) ChangeRank(isAdmin, isMod, isBanned bool) (err error) {
    _, err = groupStmts.updateGroupRank.Exec(isAdmin, isMod, isBanned, g.ID)
    if err != nil {
        return err
    }
    _ = Groups.Reload(g.ID)
    return nil
}

func (g *Group) Update(name, tag string) (err error) {
    _, err = groupStmts.updateGroup.Exec(name, tag, g.ID)
    if err != nil {
        return err
    }
    _ = Groups.Reload(g.ID)
    return nil
}

// Please don't pass arbitrary inputs to this method
func (g *Group) UpdatePerms(perms map[string]bool) (err error) {
    pjson, err := json.Marshal(perms)
    if err != nil {
        return err
    }
    _, err = groupStmts.updateGroupPerms.Exec(pjson, g.ID)
    if err != nil {
        return err
    }
    return Groups.Reload(g.ID)
}

// Copy gives you a non-pointer concurrency safe copy of the group
func (g *Group) Copy() Group {
    return *g
}

func (g *Group) CopyPtr() (co *Group) {
    co = new(Group)
    *co = *g
    return co
}

// TODO: Replace this sorting mechanism with something a lot more efficient
// ? - Use sort.Slice instead?
type SortGroup []*Group

func (sg SortGroup) Len() int {
    return len(sg)
}
func (sg SortGroup) Swap(i, j int) {
    sg[i], sg[j] = sg[j], sg[i]
}
func (sg SortGroup) Less(i, j int) bool {
    return sg[i].ID < sg[j].ID
}