3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-11-30 07:59:24 +01:00
ergo/irc/types.go

73 lines
1.6 KiB
Go
Raw Normal View History

// Copyright (c) 2012-2014 Jeremy Latt
// Copyright (c) 2014-2015 Edmund Huber
2017-03-27 14:15:02 +02:00
// Copyright (c) 2016 Daniel Oaks <daniel@danieloaks.net>
// released under the MIT license
package irc
import "github.com/oragono/oragono/irc/modes"
type empty struct{}
2017-10-05 15:29:34 +02:00
// ClientSet is a set of clients.
type ClientSet map[*Client]empty
2014-02-09 08:15:05 +01:00
2017-10-05 15:29:34 +02:00
// Add adds the given client to this set.
2014-02-09 08:15:05 +01:00
func (clients ClientSet) Add(client *Client) {
clients[client] = empty{}
2014-02-09 08:15:05 +01:00
}
2017-10-05 15:29:34 +02:00
// Remove removes the given client from this set.
2014-02-09 08:15:05 +01:00
func (clients ClientSet) Remove(client *Client) {
delete(clients, client)
}
2017-10-05 15:29:34 +02:00
// Has returns true if the given client is in this set.
func (clients ClientSet) Has(client *Client) bool {
_, ok := clients[client]
return ok
}
2019-12-17 19:21:26 +01:00
type StringSet map[string]empty
func (s StringSet) Has(str string) bool {
_, ok := s[str]
return ok
}
func (s StringSet) Add(str string) {
s[str] = empty{}
}
2017-10-05 15:29:34 +02:00
// MemberSet is a set of members with modes.
2018-04-23 00:47:10 +02:00
type MemberSet map[*Client]*modes.ModeSet
2017-10-05 15:29:34 +02:00
// Add adds the given client to this set.
func (members MemberSet) Add(member *Client) {
2018-04-23 00:47:10 +02:00
members[member] = modes.NewModeSet()
}
2017-10-05 15:29:34 +02:00
// Remove removes the given client from this set.
func (members MemberSet) Remove(member *Client) {
delete(members, member)
}
2017-10-05 15:29:34 +02:00
// Has returns true if the given client is in this set.
func (members MemberSet) Has(member *Client) bool {
_, ok := members[member]
return ok
}
2017-10-05 15:29:34 +02:00
// AnyHasMode returns true if any of our clients has the given mode.
func (members MemberSet) AnyHasMode(mode modes.Mode) bool {
for _, modes := range members {
2018-04-23 00:47:10 +02:00
if modes.HasMode(mode) {
return true
}
}
return false
}
2017-10-05 15:29:34 +02:00
// ChannelSet is a set of channels.
2020-07-01 01:24:56 +02:00
type ChannelSet map[*Channel]empty