123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524 |
- package irc
-
- import (
- "log"
- "strconv"
- )
-
- type Channel struct {
- flags ChannelModeSet
- lists map[ChannelMode]*UserMaskSet
- key Text
- members MemberSet
- name Name
- server *Server
- topic Text
- userLimit uint64
- }
-
- // NewChannel creates a new channel from a `Server` and a `name`
- // string, which must be unique on the server.
- func NewChannel(s *Server, name Name) *Channel {
- channel := &Channel{
- flags: make(ChannelModeSet),
- lists: map[ChannelMode]*UserMaskSet{
- BanMask: NewUserMaskSet(),
- ExceptMask: NewUserMaskSet(),
- InviteMask: NewUserMaskSet(),
- },
- members: make(MemberSet),
- name: name,
- server: s,
- }
-
- s.channels.Add(channel)
-
- return channel
- }
-
- func (channel *Channel) IsEmpty() bool {
- return len(channel.members) == 0
- }
-
- func (channel *Channel) Names(client *Client) {
- client.RplNamReply(channel)
- client.RplEndOfNames(channel)
- }
-
- func (channel *Channel) ClientIsOperator(client *Client) bool {
- return client.flags[Operator] || channel.members.HasMode(client, ChannelOperator)
- }
-
- func (channel *Channel) Nicks(target *Client) []string {
- isMultiPrefix := (target != nil) && target.capabilities[MultiPrefix]
- nicks := make([]string, len(channel.members))
- i := 0
- for client, modes := range channel.members {
- if isMultiPrefix {
- if modes[ChannelOperator] {
- nicks[i] += "@"
- }
- if modes[Voice] {
- nicks[i] += "+"
- }
- } else {
- if modes[ChannelOperator] {
- nicks[i] += "@"
- } else if modes[Voice] {
- nicks[i] += "+"
- }
- }
- nicks[i] += client.Nick().String()
- i += 1
- }
- return nicks
- }
-
- func (channel *Channel) Id() Name {
- return channel.name
- }
-
- func (channel *Channel) Nick() Name {
- return channel.name
- }
-
- func (channel *Channel) String() string {
- return channel.Id().String()
- }
-
- // <mode> <mode params>
- func (channel *Channel) ModeString(client *Client) (str string) {
- isMember := client.flags[Operator] || channel.members.Has(client)
- showKey := isMember && (channel.key != "")
- showUserLimit := channel.userLimit > 0
-
- // flags with args
- if showKey {
- str += Key.String()
- }
- if showUserLimit {
- str += UserLimit.String()
- }
-
- // flags
- for mode := range channel.flags {
- str += mode.String()
- }
-
- if len(str) > 0 {
- str = "+" + str
- }
-
- // args for flags with args: The order must match above to keep
- // positional arguments in place.
- if showKey {
- str += " " + channel.key.String()
- }
- if showUserLimit {
- str += " " + strconv.FormatUint(channel.userLimit, 10)
- }
-
- return
- }
-
- func (channel *Channel) IsFull() bool {
- return (channel.userLimit > 0) &&
- (uint64(len(channel.members)) >= channel.userLimit)
- }
-
- func (channel *Channel) CheckKey(key Text) bool {
- return (channel.key == "") || (channel.key == key)
- }
-
- func (channel *Channel) Join(client *Client, key Text) {
- if channel.members.Has(client) {
- // already joined, no message?
- return
- }
-
- if channel.IsFull() {
- client.ErrChannelIsFull(channel)
- return
- }
-
- if !channel.CheckKey(key) {
- client.ErrBadChannelKey(channel)
- return
- }
-
- isInvited := channel.lists[InviteMask].Match(client.UserHost())
- if channel.flags[InviteOnly] && !isInvited {
- client.ErrInviteOnlyChan(channel)
- return
- }
-
- if channel.lists[BanMask].Match(client.UserHost()) &&
- !isInvited &&
- !channel.lists[ExceptMask].Match(client.UserHost()) {
- client.ErrBannedFromChan(channel)
- return
- }
-
- client.channels.Add(channel)
- channel.members.Add(client)
- if !channel.flags[Persistent] && (len(channel.members) == 1) {
- channel.members[client][ChannelCreator] = true
- channel.members[client][ChannelOperator] = true
- }
-
- reply := RplJoin(client, channel)
- for member := range channel.members {
- member.Reply(reply)
- }
- channel.GetTopic(client)
- channel.Names(client)
- }
-
- func (channel *Channel) Part(client *Client, message Text) {
- if !channel.members.Has(client) {
- client.ErrNotOnChannel(channel)
- return
- }
-
- reply := RplPart(client, channel, message)
- for member := range channel.members {
- member.Reply(reply)
- }
- channel.Quit(client)
- }
-
- func (channel *Channel) GetTopic(client *Client) {
- if !channel.members.Has(client) {
- client.ErrNotOnChannel(channel)
- return
- }
-
- if channel.topic == "" {
- // clients appear not to expect this
- //replier.Reply(RplNoTopic(channel))
- return
- }
-
- client.RplTopic(channel)
- }
-
- func (channel *Channel) SetTopic(client *Client, topic Text) {
- if !(client.flags[Operator] || channel.members.Has(client)) {
- client.ErrNotOnChannel(channel)
- return
- }
-
- if channel.flags[OpOnlyTopic] && !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return
- }
-
- channel.topic = topic
-
- reply := RplTopicMsg(client, channel)
- for member := range channel.members {
- member.Reply(reply)
- }
-
- if err := channel.Persist(); err != nil {
- log.Println("Channel.Persist:", channel, err)
- }
- }
-
- func (channel *Channel) CanSpeak(client *Client) bool {
- if client.flags[Operator] {
- return true
- }
- if channel.flags[NoOutside] && !channel.members.Has(client) {
- return false
- }
- if channel.flags[Moderated] && !(channel.members.HasMode(client, Voice) ||
- channel.members.HasMode(client, ChannelOperator)) {
- return false
- }
- return true
- }
-
- func (channel *Channel) PrivMsg(client *Client, message Text) {
- if !channel.CanSpeak(client) {
- client.ErrCannotSendToChan(channel)
- return
- }
- reply := RplPrivMsg(client, channel, message)
- for member := range channel.members {
- if member == client {
- continue
- }
- member.Reply(reply)
- }
- }
-
- func (channel *Channel) applyModeFlag(client *Client, mode ChannelMode,
- op ModeOp) bool {
- if !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return false
- }
-
- switch op {
- case Add:
- if channel.flags[mode] {
- return false
- }
- channel.flags[mode] = true
- return true
-
- case Remove:
- if !channel.flags[mode] {
- return false
- }
- delete(channel.flags, mode)
- return true
- }
- return false
- }
-
- func (channel *Channel) applyModeMember(client *Client, mode ChannelMode,
- op ModeOp, nick Name) bool {
- if !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return false
- }
-
- if nick == "" {
- client.ErrNeedMoreParams("MODE")
- return false
- }
-
- target := channel.server.clients.Get(nick)
- if target == nil {
- client.ErrNoSuchNick(nick)
- return false
- }
-
- if !channel.members.Has(target) {
- client.ErrUserNotInChannel(channel, target)
- return false
- }
-
- switch op {
- case Add:
- if channel.members[target][mode] {
- return false
- }
- channel.members[target][mode] = true
- return true
-
- case Remove:
- if !channel.members[target][mode] {
- return false
- }
- channel.members[target][mode] = false
- return true
- }
- return false
- }
-
- func (channel *Channel) ShowMaskList(client *Client, mode ChannelMode) {
- for lmask := range channel.lists[mode].masks {
- client.RplMaskList(mode, channel, lmask)
- }
- client.RplEndOfMaskList(mode, channel)
- }
-
- func (channel *Channel) applyModeMask(client *Client, mode ChannelMode, op ModeOp,
- mask Name) bool {
- list := channel.lists[mode]
- if list == nil {
- // This should never happen, but better safe than panicky.
- return false
- }
-
- if (op == List) || (mask == "") {
- channel.ShowMaskList(client, mode)
- return false
- }
-
- if !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return false
- }
-
- if op == Add {
- return list.Add(mask)
- }
-
- if op == Remove {
- return list.Remove(mask)
- }
-
- return false
- }
-
- func (channel *Channel) applyMode(client *Client, change *ChannelModeChange) bool {
- switch change.mode {
- case BanMask, ExceptMask, InviteMask:
- return channel.applyModeMask(client, change.mode, change.op,
- NewName(change.arg))
-
- case InviteOnly, Moderated, NoOutside, OpOnlyTopic, Persistent, Private:
- return channel.applyModeFlag(client, change.mode, change.op)
-
- case Key:
- if !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return false
- }
-
- switch change.op {
- case Add:
- if change.arg == "" {
- client.ErrNeedMoreParams("MODE")
- return false
- }
- key := NewText(change.arg)
- if key == channel.key {
- return false
- }
-
- channel.key = key
- return true
-
- case Remove:
- channel.key = ""
- return true
- }
-
- case UserLimit:
- limit, err := strconv.ParseUint(change.arg, 10, 64)
- if err != nil {
- client.ErrNeedMoreParams("MODE")
- return false
- }
- if (limit == 0) || (limit == channel.userLimit) {
- return false
- }
-
- channel.userLimit = limit
- return true
-
- case ChannelOperator, Voice:
- return channel.applyModeMember(client, change.mode, change.op,
- NewName(change.arg))
-
- default:
- client.ErrUnknownMode(change.mode, channel)
- }
- return false
- }
-
- func (channel *Channel) Mode(client *Client, changes ChannelModeChanges) {
- if len(changes) == 0 {
- client.RplChannelModeIs(channel)
- return
- }
-
- applied := make(ChannelModeChanges, 0)
- for _, change := range changes {
- if channel.applyMode(client, change) {
- applied = append(applied, change)
- }
- }
-
- if len(applied) > 0 {
- reply := RplChannelMode(client, channel, applied)
- for member := range channel.members {
- member.Reply(reply)
- }
-
- if err := channel.Persist(); err != nil {
- log.Println("Channel.Persist:", channel, err)
- }
- }
- }
-
- func (channel *Channel) Persist() (err error) {
- if channel.flags[Persistent] {
- _, err = channel.server.db.Exec(`
- INSERT OR REPLACE INTO channel
- (name, flags, key, topic, user_limit, ban_list, except_list,
- invite_list)
- VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
- channel.name.String(), channel.flags.String(), channel.key.String(),
- channel.topic.String(), channel.userLimit, channel.lists[BanMask].String(),
- channel.lists[ExceptMask].String(), channel.lists[InviteMask].String())
- } else {
- _, err = channel.server.db.Exec(`
- DELETE FROM channel WHERE name = ?`, channel.name.String())
- }
- return
- }
-
- func (channel *Channel) Notice(client *Client, message Text) {
- if !channel.CanSpeak(client) {
- client.ErrCannotSendToChan(channel)
- return
- }
- reply := RplNotice(client, channel, message)
- for member := range channel.members {
- if member == client {
- continue
- }
- member.Reply(reply)
- }
- }
-
- func (channel *Channel) Quit(client *Client) {
- channel.members.Remove(client)
- client.channels.Remove(channel)
-
- if !channel.flags[Persistent] && channel.IsEmpty() {
- channel.server.channels.Remove(channel)
- }
- }
-
- func (channel *Channel) Kick(client *Client, target *Client, comment Text) {
- if !(client.flags[Operator] || channel.members.Has(client)) {
- client.ErrNotOnChannel(channel)
- return
- }
- if !channel.ClientIsOperator(client) {
- client.ErrChanOPrivIsNeeded(channel)
- return
- }
- if !channel.members.Has(target) {
- client.ErrUserNotInChannel(channel, target)
- return
- }
-
- reply := RplKick(channel, client, target, comment)
- for member := range channel.members {
- member.Reply(reply)
- }
- channel.Quit(target)
- }
-
- func (channel *Channel) Invite(invitee *Client, inviter *Client) {
- if channel.flags[InviteOnly] && !channel.ClientIsOperator(inviter) {
- inviter.ErrChanOPrivIsNeeded(channel)
- return
- }
-
- if !channel.members.Has(inviter) {
- inviter.ErrNotOnChannel(channel)
- return
- }
-
- if channel.flags[InviteOnly] {
- channel.lists[InviteMask].Add(invitee.UserHost())
- if err := channel.Persist(); err != nil {
- log.Println("Channel.Persist:", channel, err)
- }
- }
-
- inviter.RplInviting(invitee, channel.name)
- invitee.Reply(RplInviteMsg(inviter, invitee, channel.name))
- if invitee.flags[Away] {
- inviter.RplAway(invitee)
- }
- }
|