package irc import ( "errors" "fmt" "strings" ) // // simple types // // a string with wildcards type Mask string // add, remove, list modes type ModeOp rune func (op ModeOp) String() string { return string(op) } // user mode flags type UserMode rune func (mode UserMode) String() string { return fmt.Sprintf("%c", mode) } type Phase uint type ReplyCode interface { String() string } type StringCode string func (code StringCode) String() string { return string(code) } type NumericCode uint func (code NumericCode) String() string { return fmt.Sprintf("%03d", code) } // channel mode flags type ChannelMode rune func (mode ChannelMode) String() string { return fmt.Sprintf("%c", mode) } type ChannelNameMap map[string]*Channel func (channels ChannelNameMap) Add(channel *Channel) error { if channels[channel.name] != nil { return fmt.Errorf("%s: already set", channel.name) } channels[channel.name] = channel return nil } func (channels ChannelNameMap) Remove(channel *Channel) error { if channel != channels[channel.name] { return fmt.Errorf("%s: mismatch", channel.name) } delete(channels, channel.name) return nil } type ClientNameMap map[string]*Client var ( ErrNickMissing = errors.New("nick missing") ErrNicknameInUse = errors.New("nickname in use") ErrNicknameMismatch = errors.New("nickname mismatch") ) func (clients ClientNameMap) Get(nick string) *Client { return clients[strings.ToLower(nick)] } func (clients ClientNameMap) Add(client *Client) error { if !client.HasNick() { return ErrNickMissing } if clients.Get(client.nick) != nil { return ErrNicknameInUse } clients[strings.ToLower(client.nick)] = client return nil } func (clients ClientNameMap) Remove(client *Client) error { if !client.HasNick() { return ErrNickMissing } if clients.Get(client.nick) != client { return ErrNicknameMismatch } delete(clients, client.nick) return nil } type ChannelModeSet map[ChannelMode]bool type ClientSet map[*Client]bool func (clients ClientSet) Add(client *Client) { clients[client] = true } func (clients ClientSet) Remove(client *Client) { delete(clients, client) } func (clients ClientSet) Has(client *Client) bool { return clients[client] } type MemberSet map[*Client]ChannelModeSet func (members MemberSet) Add(member *Client) { members[member] = make(ChannelModeSet) } func (members MemberSet) Remove(member *Client) { delete(members, member) } func (members MemberSet) Has(member *Client) bool { _, ok := members[member] return ok } func (members MemberSet) HasMode(member *Client, mode ChannelMode) bool { modes, ok := members[member] if !ok { return false } return modes[mode] } type ChannelSet map[*Channel]bool func (channels ChannelSet) Add(channel *Channel) { channels[channel] = true } func (channels ChannelSet) Remove(channel *Channel) { delete(channels, channel) } func (channels ChannelSet) First() *Channel { for channel := range channels { return channel } return nil } // // interfaces // type Identifier interface { Id() string Nick() string } type Replier interface { Reply(Reply) } type Reply interface { Code() ReplyCode Format(*Client) []string Source() Identifier } type Command interface { Code() StringCode Client() *Client Source() Identifier Reply(Reply) } type ServerCommand interface { Command HandleServer(*Server) } type AuthServerCommand interface { Command HandleAuthServer(*Server) } type RegServerCommand interface { Command HandleRegServer(*Server) } // // structs // type UserMask struct { nickname Mask username Mask hostname Mask } func (mask *UserMask) String() string { return fmt.Sprintf("%s!%s@%s", mask.nickname, mask.username, mask.hostname) }