123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408 |
- // Copyright (c) 2012-2014 Jeremy Latt
- // Copyright (c) 2014-2015 Edmund Huber
- // Copyright (c) 2016-2017 Daniel Oaks <daniel@danieloaks.net>
- // released under the MIT license
-
- package modes
-
- import (
- "strings"
- "sync/atomic"
-
- "github.com/oragono/oragono/irc/utils"
- )
-
- var (
- // SupportedUserModes are the user modes that we actually support (modifying).
- SupportedUserModes = Modes{
- Away, Bot, Invisible, Operator, RegisteredOnly, ServerNotice, UserRoleplaying,
- }
-
- // SupportedChannelModes are the channel modes that we support.
- SupportedChannelModes = Modes{
- BanMask, ChanRoleplaying, ExceptMask, InviteMask, InviteOnly, Key,
- Moderated, NoOutside, OpOnlyTopic, RegisteredOnly, Secret, UserLimit,
- }
- )
-
- // ModeOp is an operation performed with modes
- type ModeOp rune
-
- func (op ModeOp) String() string {
- return string(op)
- }
-
- const (
- // Add is used when adding the given key.
- Add ModeOp = '+'
- // List is used when listing modes (for instance, listing the current bans on a channel).
- List ModeOp = '='
- // Remove is used when taking away the given key.
- Remove ModeOp = '-'
- )
-
- // Mode represents a user/channel/server mode
- type Mode rune
-
- func (mode Mode) String() string {
- return string(mode)
- }
-
- // ModeChange is a single mode changing
- type ModeChange struct {
- Mode Mode
- Op ModeOp
- Arg string
- }
-
- func (change *ModeChange) String() (str string) {
- if (change.Op == Add) || (change.Op == Remove) {
- str = change.Op.String()
- }
- str += change.Mode.String()
- if change.Arg != "" {
- str += " " + change.Arg
- }
- return
- }
-
- // ModeChanges are a collection of 'ModeChange's
- type ModeChanges []ModeChange
-
- func (changes ModeChanges) String() string {
- if len(changes) == 0 {
- return ""
- }
-
- op := changes[0].Op
- str := changes[0].Op.String()
-
- for _, change := range changes {
- if change.Op != op {
- op = change.Op
- str += change.Op.String()
- }
- str += change.Mode.String()
- }
-
- for _, change := range changes {
- if change.Arg == "" {
- continue
- }
- str += " " + change.Arg
- }
- return str
- }
-
- // Modes is just a raw list of modes
- type Modes []Mode
-
- func (modes Modes) String() string {
- strs := make([]string, len(modes))
- for index, mode := range modes {
- strs[index] = mode.String()
- }
- return strings.Join(strs, "")
- }
-
- // User Modes
- const (
- Away Mode = 'a'
- Bot Mode = 'B'
- Invisible Mode = 'i'
- LocalOperator Mode = 'O'
- Operator Mode = 'o'
- Restricted Mode = 'r'
- RegisteredOnly Mode = 'R'
- ServerNotice Mode = 's'
- TLS Mode = 'Z'
- UserRoleplaying Mode = 'E'
- WallOps Mode = 'w'
- )
-
- // Channel Modes
- const (
- BanMask Mode = 'b' // arg
- ChanRoleplaying Mode = 'E' // flag
- ExceptMask Mode = 'e' // arg
- InviteMask Mode = 'I' // arg
- InviteOnly Mode = 'i' // flag
- Key Mode = 'k' // flag arg
- Moderated Mode = 'm' // flag
- NoOutside Mode = 'n' // flag
- OpOnlyTopic Mode = 't' // flag
- // RegisteredOnly mode is reused here from umode definition
- Secret Mode = 's' // flag
- UserLimit Mode = 'l' // flag arg
- )
-
- var (
- ChannelFounder Mode = 'q' // arg
- ChannelAdmin Mode = 'a' // arg
- ChannelOperator Mode = 'o' // arg
- Halfop Mode = 'h' // arg
- Voice Mode = 'v' // arg
-
- // ChannelPrivModes holds the list of modes that are privileged, ie founder/op/halfop, in order.
- // voice is not in this list because it cannot perform channel operator actions.
- ChannelPrivModes = Modes{
- ChannelFounder, ChannelAdmin, ChannelOperator, Halfop,
- }
-
- // ChannelUserModes holds the list of all modes that can be applied to a user in a channel,
- // including Voice, in descending order of precedence
- ChannelUserModes = Modes{
- ChannelFounder, ChannelAdmin, ChannelOperator, Halfop, Voice,
- }
-
- ChannelModePrefixes = map[Mode]string{
- ChannelFounder: "~",
- ChannelAdmin: "&",
- ChannelOperator: "@",
- Halfop: "%",
- Voice: "+",
- }
- )
-
- //
- // channel membership prefixes
- //
-
- // SplitChannelMembershipPrefixes takes a target and returns the prefixes on it, then the name.
- func SplitChannelMembershipPrefixes(target string) (prefixes string, name string) {
- name = target
- for {
- if len(name) > 0 && strings.Contains("~&@%+", string(name[0])) {
- prefixes += string(name[0])
- name = name[1:]
- } else {
- break
- }
- }
-
- return prefixes, name
- }
-
- // GetLowestChannelModePrefix returns the lowest channel prefix mode out of the given prefixes.
- func GetLowestChannelModePrefix(prefixes string) (lowest *Mode) {
- for i, mode := range ChannelUserModes {
- if strings.Contains(prefixes, ChannelModePrefixes[mode]) {
- lowest = &ChannelPrivModes[i]
- }
- }
- return
- }
-
- //
- // commands
- //
-
- // ParseUserModeChanges returns the valid changes, and the list of unknown chars.
- func ParseUserModeChanges(params ...string) (ModeChanges, map[rune]bool) {
- changes := make(ModeChanges, 0)
- unknown := make(map[rune]bool)
-
- op := List
-
- if 0 < len(params) {
- modeArg := params[0]
- skipArgs := 1
-
- for _, mode := range modeArg {
- if mode == '-' || mode == '+' {
- op = ModeOp(mode)
- continue
- }
- change := ModeChange{
- Mode: Mode(mode),
- Op: op,
- }
-
- // put arg into modechange if needed
- switch Mode(mode) {
- case ServerNotice:
- // always require arg
- if len(params) > skipArgs {
- change.Arg = params[skipArgs]
- skipArgs++
- } else {
- continue
- }
- }
-
- var isKnown bool
- for _, supportedMode := range SupportedUserModes {
- if rune(supportedMode) == mode {
- isKnown = true
- break
- }
- }
- if !isKnown {
- unknown[mode] = true
- continue
- }
-
- changes = append(changes, change)
- }
- }
-
- return changes, unknown
- }
-
- // ParseChannelModeChanges returns the valid changes, and the list of unknown chars.
- func ParseChannelModeChanges(params ...string) (ModeChanges, map[rune]bool) {
- changes := make(ModeChanges, 0)
- unknown := make(map[rune]bool)
-
- op := List
-
- if 0 < len(params) {
- modeArg := params[0]
- skipArgs := 1
-
- for _, mode := range modeArg {
- if mode == '-' || mode == '+' {
- op = ModeOp(mode)
- continue
- }
- change := ModeChange{
- Mode: Mode(mode),
- Op: op,
- }
-
- // put arg into modechange if needed
- switch Mode(mode) {
- case BanMask, ExceptMask, InviteMask:
- if len(params) > skipArgs {
- change.Arg = params[skipArgs]
- skipArgs++
- } else {
- change.Op = List
- }
- case ChannelFounder, ChannelAdmin, ChannelOperator, Halfop, Voice:
- if len(params) > skipArgs {
- change.Arg = params[skipArgs]
- skipArgs++
- } else {
- continue
- }
- case Key, UserLimit:
- // don't require value when removing
- if change.Op == Add {
- if len(params) > skipArgs {
- change.Arg = params[skipArgs]
- skipArgs++
- } else {
- continue
- }
- }
- }
-
- var isKnown bool
- for _, supportedMode := range SupportedChannelModes {
- if rune(supportedMode) == mode {
- isKnown = true
- break
- }
- }
- for _, supportedMode := range ChannelUserModes {
- if rune(supportedMode) == mode {
- isKnown = true
- break
- }
- }
- if !isKnown {
- unknown[mode] = true
- continue
- }
-
- changes = append(changes, change)
- }
- }
-
- return changes, unknown
- }
-
- // ModeSet holds a set of modes.
- type ModeSet [1]uint64
-
- // valid modes go from 65 ('A') to 122 ('z'), making at most 58 possible values;
- // subtract 65 from the mode value and use that bit of the uint64 to represent it
- const (
- minMode = 65 // 'A'
- )
-
- // returns a pointer to a new ModeSet
- func NewModeSet() *ModeSet {
- var set ModeSet
- utils.BitsetInitialize(set[:])
- return &set
- }
-
- // test whether `mode` is set
- func (set *ModeSet) HasMode(mode Mode) bool {
- if set == nil {
- return false
- }
-
- return utils.BitsetGet(set[:], uint(mode)-minMode)
- }
-
- // set `mode` to be on or off, return whether the value actually changed
- func (set *ModeSet) SetMode(mode Mode, on bool) (applied bool) {
- return utils.BitsetSet(set[:], uint(mode)-minMode, on)
- }
-
- // copy the contents of another modeset on top of this one
- func (set *ModeSet) Copy(other *ModeSet) {
- utils.BitsetCopy(set[:], other[:])
- }
-
- // return the modes in the set as a slice
- func (set *ModeSet) AllModes() (result []Mode) {
- if set == nil {
- return
- }
-
- block := atomic.LoadUint64(&set[0])
- var i uint
- for i = 0; i < 64; i++ {
- if block&(1<<i) != 0 {
- result = append(result, Mode(minMode+i))
- }
- }
- return
- }
-
- // String returns the modes in this set.
- func (set *ModeSet) String() (result string) {
- if set == nil {
- return
- }
-
- var buf strings.Builder
- for _, mode := range set.AllModes() {
- buf.WriteRune(rune(mode))
- }
- return buf.String()
- }
-
- // Prefixes returns a list of prefixes for the given set of channel modes.
- func (set *ModeSet) Prefixes(isMultiPrefix bool) (prefixes string) {
- if set == nil {
- return
- }
-
- // add prefixes in order from highest to lowest privs
- for _, mode := range ChannelUserModes {
- if set.HasMode(mode) {
- prefixes += ChannelModePrefixes[mode]
- }
- }
-
- if !isMultiPrefix && len(prefixes) > 1 {
- prefixes = string(prefixes[0])
- }
-
- return prefixes
- }
|