anonircd/entity.go

158 lines
2.7 KiB
Go
Raw Permalink Normal View History

package main
import (
2016-09-16 01:25:52 +00:00
"strings"
"sync"
"time"
)
const entityClient = 0
const entityChannel = 1
const stateTerminating = 0
const stateNormal = 1
const clientModes = "cD"
const channelModes = "cDiklmprstz"
const channelModesWithArg = "kl"
type Entity struct {
entitytype int
identifier string
created int64
state int
modes *sync.Map
}
func (e *Entity) Initialize(etype int, identifier string) {
e.identifier = identifier
e.entitytype = etype
e.created = time.Now().Unix()
e.state = stateNormal
e.modes = new(sync.Map)
}
2017-12-20 09:06:42 +00:00
func (e *Entity) getMode(mode string) string {
if v, ok := e.modes.Load(mode); ok {
return v.(string)
}
return ""
}
func (e *Entity) getModes() map[string]string {
modes := make(map[string]string)
e.modes.Range(func(k, v interface{}) bool {
modes[k.(string)] = v.(string)
return true
})
return modes
}
func (e *Entity) hasMode(mode string) bool {
_, ok := e.modes.Load(mode)
return ok
}
func (e *Entity) addMode(mode string, value string) {
var allowedmodes string
if e.entitytype == entityClient {
allowedmodes = clientModes
} else {
allowedmodes = channelModes
}
2017-09-27 21:21:18 +00:00
if strings.Contains(allowedmodes, mode) && !e.hasMode(mode) {
e.modes.Store(mode, value)
}
}
2017-12-20 09:06:42 +00:00
func (e *Entity) addModes(newmodes []string) {
if len(newmodes) == 0 {
return
}
p := 1
for _, mode := range strings.Split(newmodes[0], "") {
if strings.ContainsAny(mode, channelModesWithArg) {
2017-12-20 09:06:42 +00:00
if len(newmodes) > p {
e.addMode(mode, newmodes[p])
p++
}
} else {
e.addMode(mode, "")
}
}
}
func (e *Entity) removeMode(mode string) {
if e.hasMode(mode) {
e.modes.Delete(mode)
}
}
func (e *Entity) removeModes(modes string) {
for _, mode := range strings.Split(modes, "") {
2017-09-27 21:21:18 +00:00
e.removeMode(mode)
}
}
func (e *Entity) diffModes(lastmodes map[string]string) (map[string]string, map[string]string) {
addedmodes := make(map[string]string)
if lastmodes != nil {
e.modes.Range(func(k, v interface{}) bool {
if _, ok := lastmodes[k.(string)]; !ok {
addedmodes[k.(string)] = lastmodes[k.(string)]
}
return true
})
}
removedmodes := make(map[string]string)
for mode := range lastmodes {
if _, ok := e.modes.Load(mode); !ok {
removedmodes[mode] = mode
}
}
return addedmodes, removedmodes
}
func (e *Entity) printModes(addedmodes map[string]string, removedmodes map[string]string) string {
if removedmodes == nil {
removedmodes = make(map[string]string)
}
m := ""
// Added modes
sentsign := false
for mode := range addedmodes {
if !sentsign {
m += "+"
sentsign = true
}
m += mode
}
// Removed modes
sentsign = false
for mode := range removedmodes {
if !sentsign {
m += "-"
sentsign = true
}
m += mode
}
if m == "" {
m = "+"
}
return m
}