forked from lug/matterbridge
		
	Compare commits
	
		
			117 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | daac3ebca2 | ||
|   | 639f9cf966 | ||
|   | 4fc48b5aa4 | ||
|   | 307ff77b42 | ||
|   | 9b500bc5f7 | ||
|   | e313154134 | ||
|   | 27e94c438d | ||
|   | 58392876df | ||
|   | 115c4b1aa7 | ||
|   | ba5649d259 | ||
|   | 1b30575510 | ||
|   | 7dbebd3ea7 | ||
|   | 6f18790352 | ||
|   | d1e04a2ece | ||
|   | bea0bbd0c2 | ||
|   | 0530503ef2 | ||
|   | d1e8ff814b | ||
|   | 4f8ae761a2 | ||
|   | b530e92834 | ||
|   | b2a6777995 | ||
|   | b461fc5e40 | ||
|   | b7a8c6b60f | ||
|   | 41aa8ad799 | ||
|   | 7973baedd0 | ||
|   | 299b71d982 | ||
|   | 76aafe1fa8 | ||
|   | 95a0229aaf | ||
|   | 915a8fbad7 | ||
|   | d4d7fef313 | ||
|   | 4e1dc9f885 | ||
|   | 155ae80d22 | ||
|   | c7e336efd9 | ||
|   | ac3c65a0cc | ||
|   | df74df475b | ||
|   | a61e2db7cb | ||
|   | 7aabe12acf | ||
|   | c4b75e5754 | ||
|   | 6a7adb20a8 | ||
|   | b49fb2b69c | ||
|   | 4bda29cb38 | ||
|   | 5f14141ec9 | ||
|   | c088e45d85 | ||
|   | d59c51a94b | ||
|   | 47b7fae61b | ||
|   | 1a40b0c1e9 | ||
|   | 27d886826c | ||
|   | 18981cb636 | ||
|   | ffa8f65aa8 | ||
|   | 82588b00c5 | ||
|   | 603449e850 | ||
|   | 248d88c849 | ||
|   | d19535fa21 | ||
|   | 49204cafcc | ||
|   | 812db2d267 | ||
|   | 14490bea9f | ||
|   | 0352970051 | ||
|   | ed01820722 | ||
|   | 90a61f15cc | ||
|   | 86cd7f1ba6 | ||
|   | d6ee55e35f | ||
|   | aef64eec32 | ||
|   | c4193d5ccd | ||
|   | 0c94186818 | ||
|   | 9039720013 | ||
|   | a3470f8aec | ||
|   | 01badde21d | ||
|   | a37b232dd9 | ||
|   | 579ee48385 | ||
|   | dd985d1dad | ||
|   | d2caea70a2 | ||
|   | 21143cf5ee | ||
|   | dc2aed698d | ||
|   | 37c350f19f | ||
|   | 9e03fcf162 | ||
|   | 8d4521c1df | ||
|   | 9226252336 | ||
|   | f4fb83e787 | ||
|   | e7fcb25107 | ||
|   | 5a85258f74 | ||
|   | 2f7df2df43 | ||
|   | ad3a753718 | ||
|   | e45c551880 | ||
|   | e59d338d4e | ||
|   | 7a86044f7a | ||
|   | 8b98f605bc | ||
|   | 7c773ebae0 | ||
|   | e84417430d | ||
|   | 5a8d7b5f6d | ||
|   | cfb8107138 | ||
|   | 43bd779fb7 | ||
|   | 7f9a400776 | ||
|   | ce1c5873ac | ||
|   | 85ff1995fd | ||
|   | b963f83c6a | ||
|   | f6297ebbb0 | ||
|   | a5259f56c5 | ||
|   | 3f75ed9c18 | ||
|   | 49ece51167 | ||
|   | e77c3eb20a | ||
|   | 59b2a5f8d0 | ||
|   | 28710d0bc7 | ||
|   | ad4d461606 | ||
|   | 67905089ba | ||
|   | f2483af561 | ||
|   | c28b87641e | ||
|   | f8e6a69d6e | ||
|   | 54216cec4b | ||
|   | 12989bbd99 | ||
|   | 38d09dba2e | ||
|   | fafd0c68e9 | ||
|   | 41195c8e48 | ||
|   | a97804548e | ||
|   | ba653c0841 | ||
|   | 5b191f78a0 | ||
|   | 83ef61287e | ||
|   | 3527e09bc5 | ||
|   | ddc5b3268f | 
| @@ -1,7 +1,7 @@ | |||||||
| language: go | language: go | ||||||
| go: | go: | ||||||
|     #- 1.7.x |     #- 1.7.x | ||||||
|     - 1.8.x |     - 1.9.x | ||||||
|       # - tip |       # - tip | ||||||
|  |  | ||||||
| # we have everything vendored | # we have everything vendored | ||||||
| @@ -35,7 +35,7 @@ before_script: | |||||||
| # set -e enabled in bash.  | # set -e enabled in bash.  | ||||||
| script: | script: | ||||||
|   - test -z $(gofmt -s -l $GO_FILES)  # Fail if a .go file hasn't been formatted with gofmt |   - test -z $(gofmt -s -l $GO_FILES)  # Fail if a .go file hasn't been formatted with gofmt | ||||||
|   #- go test -v -race $PKGS            # Run all the tests with the race detector enabled |   - go test -v -race $PKGS            # Run all the tests with the race detector enabled | ||||||
|   - go vet $PKGS                      # go vet is the official Go static analyzer |   - go vet $PKGS                      # go vet is the official Go static analyzer | ||||||
|   - megacheck $PKGS                   # "go vet on steroids" + linter |   - megacheck $PKGS                   # "go vet on steroids" + linter | ||||||
|   - /bin/bash ci/bintray.sh |   - /bin/bash ci/bintray.sh | ||||||
|   | |||||||
							
								
								
									
										17
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										17
									
								
								README.md
									
									
									
									
									
								
							| @@ -13,6 +13,7 @@ Has a REST API. | |||||||
| # Table of Contents | # Table of Contents | ||||||
|  * [Features](#features) |  * [Features](#features) | ||||||
|  * [Requirements](#requirements) |  * [Requirements](#requirements) | ||||||
|  |  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) | ||||||
|  * [Installing](#installing) |  * [Installing](#installing) | ||||||
|    * [Binaries](#binaries) |    * [Binaries](#binaries) | ||||||
|    * [Building](#building) |    * [Building](#building) | ||||||
| @@ -28,14 +29,15 @@ Has a REST API. | |||||||
| # Features | # Features | ||||||
| * Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam.  | * Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam.  | ||||||
|   Pick and mix. |   Pick and mix. | ||||||
| * Matterbridge can also work with private groups on your mattermost/slack. | * Support private groups on your mattermost/slack. | ||||||
| * Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts. | * Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts. | ||||||
| * The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways). | * The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways). | ||||||
|  | * Edits and delete messages across bridges that support it (mattermost,slack,discord,gitter,telegram) | ||||||
| * REST API to read/post messages to bridges (WIP). | * REST API to read/post messages to bridges (WIP). | ||||||
|  |  | ||||||
| # Requirements | # Requirements | ||||||
| Accounts to one of the supported bridges | Accounts to one of the supported bridges | ||||||
| * [Mattermost](https://github.com/mattermost/platform/) 3.5.x - 3.10.x, 4.0.x | * [Mattermost](https://github.com/mattermost/platform/) 3.8.x - 3.10.x, 4.x | ||||||
| * [IRC](http://www.mirc.com/servers.html) | * [IRC](http://www.mirc.com/servers.html) | ||||||
| * [XMPP](https://jabber.org) | * [XMPP](https://jabber.org) | ||||||
| * [Gitter](https://gitter.im) | * [Gitter](https://gitter.im) | ||||||
| @@ -47,13 +49,16 @@ Accounts to one of the supported bridges | |||||||
| * [Matrix](https://matrix.org) | * [Matrix](https://matrix.org) | ||||||
| * [Steam](https://store.steampowered.com/) | * [Steam](https://store.steampowered.com/) | ||||||
|  |  | ||||||
|  | # Screenshots | ||||||
|  | See https://github.com/42wim/matterbridge/wiki | ||||||
|  |  | ||||||
| # Installing | # Installing | ||||||
| ## Binaries | ## Binaries | ||||||
| * Latest stable release [v0.16.3](https://github.com/42wim/matterbridge/releases/latest) | * Latest stable release [v1.4.0](https://github.com/42wim/matterbridge/releases/latest) | ||||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)   | * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)   | ||||||
|  |  | ||||||
| ## Building | ## Building | ||||||
| Go 1.7+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH] (https://golang.org/doc/code.html#GOPATH) | Go 1.8+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH] (https://golang.org/doc/code.html#GOPATH) | ||||||
|  |  | ||||||
| ``` | ``` | ||||||
| cd $GOPATH | cd $GOPATH | ||||||
| @@ -160,6 +165,10 @@ See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.m | |||||||
|  |  | ||||||
| See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | ||||||
|  |  | ||||||
|  | Want to tip ?  | ||||||
|  | * eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f | ||||||
|  | * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | ||||||
|  |  | ||||||
| # Thanks | # Thanks | ||||||
| Matterbridge wouldn't exist without these libraries: | Matterbridge wouldn't exist without these libraries: | ||||||
| * discord - https://github.com/bwmarrin/discordgo | * discord - https://github.com/bwmarrin/discordgo | ||||||
|   | |||||||
| @@ -61,16 +61,20 @@ func (b *Api) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
| func (b *Api) JoinChannel(channel string) error { | func (b *Api) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) Send(msg config.Message) error { | func (b *Api) Send(msg config.Message) (string, error) { | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	b.Messages.Enqueue(&msg) | 	b.Messages.Enqueue(&msg) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) handlePostMessage(c echo.Context) error { | func (b *Api) handlePostMessage(c echo.Context) error { | ||||||
|   | |||||||
| @@ -19,9 +19,9 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bridger interface { | type Bridger interface { | ||||||
| 	Send(msg config.Message) error | 	Send(msg config.Message) (string, error) | ||||||
| 	Connect() error | 	Connect() error | ||||||
| 	JoinChannel(channel string) error | 	JoinChannel(channel config.ChannelInfo) error | ||||||
| 	Disconnect() error | 	Disconnect() error | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -92,16 +92,10 @@ func (b *Bridge) JoinChannels() error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | ||||||
| 	mychannel := "" |  | ||||||
| 	for ID, channel := range channels { | 	for ID, channel := range channels { | ||||||
| 		if !exists[ID] { | 		if !exists[ID] { | ||||||
| 			mychannel = channel.Name |  | ||||||
| 			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID) | 			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID) | ||||||
| 			if b.Protocol == "irc" && channel.Options.Key != "" { | 			err := b.JoinChannel(channel) | ||||||
| 				log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) |  | ||||||
| 				mychannel = mychannel + " " + channel.Options.Key |  | ||||||
| 			} |  | ||||||
| 			err := b.JoinChannel(mychannel) |  | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
|   | |||||||
| @@ -13,6 +13,8 @@ const ( | |||||||
| 	EVENT_JOIN_LEAVE      = "join_leave" | 	EVENT_JOIN_LEAVE      = "join_leave" | ||||||
| 	EVENT_FAILURE         = "failure" | 	EVENT_FAILURE         = "failure" | ||||||
| 	EVENT_REJOIN_CHANNELS = "rejoin_channels" | 	EVENT_REJOIN_CHANNELS = "rejoin_channels" | ||||||
|  | 	EVENT_USER_ACTION     = "user_action" | ||||||
|  | 	EVENT_MSG_DELETE      = "msg_delete" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Message struct { | type Message struct { | ||||||
| @@ -26,6 +28,13 @@ type Message struct { | |||||||
| 	Protocol  string    `json:"protocol"` | 	Protocol  string    `json:"protocol"` | ||||||
| 	Gateway   string    `json:"gateway"` | 	Gateway   string    `json:"gateway"` | ||||||
| 	Timestamp time.Time `json:"timestamp"` | 	Timestamp time.Time `json:"timestamp"` | ||||||
|  | 	ID        string    `json:"id"` | ||||||
|  | 	Extra     map[string][]interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type FileInfo struct { | ||||||
|  | 	Name string | ||||||
|  | 	Data *[]byte | ||||||
| } | } | ||||||
|  |  | ||||||
| type ChannelInfo struct { | type ChannelInfo struct { | ||||||
| @@ -33,7 +42,6 @@ type ChannelInfo struct { | |||||||
| 	Account     string | 	Account     string | ||||||
| 	Direction   string | 	Direction   string | ||||||
| 	ID          string | 	ID          string | ||||||
| 	GID         map[string]bool |  | ||||||
| 	SameChannel map[string]bool | 	SameChannel map[string]bool | ||||||
| 	Options     ChannelOptions | 	Options     ChannelOptions | ||||||
| } | } | ||||||
| @@ -42,6 +50,7 @@ type Protocol struct { | |||||||
| 	AuthCode               string // steam | 	AuthCode               string // steam | ||||||
| 	BindAddress            string // mattermost, slack // DEPRECATED | 	BindAddress            string // mattermost, slack // DEPRECATED | ||||||
| 	Buffer                 int    // api | 	Buffer                 int    // api | ||||||
|  | 	Charset                string // irc | ||||||
| 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||||
| 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||||
| 	IconURL                string // mattermost, slack | 	IconURL                string // mattermost, slack | ||||||
| @@ -54,6 +63,7 @@ type Protocol struct { | |||||||
| 	Nick                   string // all protocols | 	Nick                   string // all protocols | ||||||
| 	NickFormatter          string // mattermost, slack | 	NickFormatter          string // mattermost, slack | ||||||
| 	NickServNick           string // IRC | 	NickServNick           string // IRC | ||||||
|  | 	NickServUsername       string // IRC | ||||||
| 	NickServPassword       string // IRC | 	NickServPassword       string // IRC | ||||||
| 	NicksPerRow            int    // mattermost, slack | 	NicksPerRow            int    // mattermost, slack | ||||||
| 	NoHomeServerSuffix     bool   // matrix | 	NoHomeServerSuffix     bool   // matrix | ||||||
| @@ -70,6 +80,7 @@ type Protocol struct { | |||||||
| 	ShowJoinPart           bool   // all protocols | 	ShowJoinPart           bool   // all protocols | ||||||
| 	ShowEmbeds             bool   // discord | 	ShowEmbeds             bool   // discord | ||||||
| 	SkipTLSVerify          bool   // IRC, mattermost | 	SkipTLSVerify          bool   // IRC, mattermost | ||||||
|  | 	StripNick              bool   // all protocols | ||||||
| 	Team                   string // mattermost | 	Team                   string // mattermost | ||||||
| 	Token                  string // gitter, slack, discord, api | 	Token                  string // gitter, slack, discord, api | ||||||
| 	URL                    string // mattermost, slack // DEPRECATED | 	URL                    string // mattermost, slack // DEPRECATED | ||||||
| @@ -77,6 +88,7 @@ type Protocol struct { | |||||||
| 	UseSASL                bool   // IRC | 	UseSASL                bool   // IRC | ||||||
| 	UseTLS                 bool   // IRC | 	UseTLS                 bool   // IRC | ||||||
| 	UseFirstName           bool   // telegram | 	UseFirstName           bool   // telegram | ||||||
|  | 	UseUserName            bool   // discord | ||||||
| 	UseInsecureURL         bool   // telegram | 	UseInsecureURL         bool   // telegram | ||||||
| 	WebhookBindAddress     string // mattermost, slack | 	WebhookBindAddress     string // mattermost, slack | ||||||
| 	WebhookURL             string // mattermost, slack | 	WebhookURL             string // mattermost, slack | ||||||
| @@ -84,7 +96,8 @@ type Protocol struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| type ChannelOptions struct { | type ChannelOptions struct { | ||||||
| 	Key string // irc | 	Key        string // irc | ||||||
|  | 	WebhookURL string // discord | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bridge struct { | type Bridge struct { | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| package bdiscord | package bdiscord | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
| @@ -10,17 +11,18 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| type bdiscord struct { | type bdiscord struct { | ||||||
| 	c             *discordgo.Session | 	c              *discordgo.Session | ||||||
| 	Config        *config.Protocol | 	Config         *config.Protocol | ||||||
| 	Remote        chan config.Message | 	Remote         chan config.Message | ||||||
| 	Account       string | 	Account        string | ||||||
| 	Channels      []*discordgo.Channel | 	Channels       []*discordgo.Channel | ||||||
| 	Nick          string | 	Nick           string | ||||||
| 	UseChannelID  bool | 	UseChannelID   bool | ||||||
| 	userMemberMap map[string]*discordgo.Member | 	userMemberMap  map[string]*discordgo.Member | ||||||
| 	guildID       string | 	guildID        string | ||||||
| 	webhookID     string | 	webhookID      string | ||||||
| 	webhookToken  string | 	webhookToken   string | ||||||
|  | 	channelInfoMap map[string]*config.ChannelInfo | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -37,11 +39,10 @@ func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord { | |||||||
| 	b.Remote = c | 	b.Remote = c | ||||||
| 	b.Account = account | 	b.Account = account | ||||||
| 	b.userMemberMap = make(map[string]*discordgo.Member) | 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||||
|  | 	b.channelInfoMap = make(map[string]*config.ChannelInfo) | ||||||
| 	if b.Config.WebhookURL != "" { | 	if b.Config.WebhookURL != "" { | ||||||
| 		flog.Debug("Configuring Discord Incoming Webhook") | 		flog.Debug("Configuring Discord Incoming Webhook") | ||||||
| 		webhookURLSplit := strings.Split(b.Config.WebhookURL, "/") | 		b.webhookID, b.webhookToken = b.splitURL(b.Config.WebhookURL) | ||||||
| 		b.webhookToken = webhookURLSplit[len(webhookURLSplit)-1] |  | ||||||
| 		b.webhookID = webhookURLSplit[len(webhookURLSplit)-2] |  | ||||||
| 	} | 	} | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -66,6 +67,7 @@ func (b *bdiscord) Connect() error { | |||||||
| 	b.c.AddHandler(b.messageCreate) | 	b.c.AddHandler(b.messageCreate) | ||||||
| 	b.c.AddHandler(b.memberUpdate) | 	b.c.AddHandler(b.memberUpdate) | ||||||
| 	b.c.AddHandler(b.messageUpdate) | 	b.c.AddHandler(b.messageUpdate) | ||||||
|  | 	b.c.AddHandler(b.messageDelete) | ||||||
| 	err = b.c.Open() | 	err = b.c.Open() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| @@ -99,37 +101,92 @@ func (b *bdiscord) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) JoinChannel(channel string) error { | func (b *bdiscord) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	idcheck := strings.Split(channel, "ID:") | 	b.channelInfoMap[channel.ID] = &channel | ||||||
|  | 	idcheck := strings.Split(channel.Name, "ID:") | ||||||
| 	if len(idcheck) > 1 { | 	if len(idcheck) > 1 { | ||||||
| 		b.UseChannelID = true | 		b.UseChannelID = true | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) Send(msg config.Message) error { | func (b *bdiscord) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	channelID := b.getChannelID(msg.Channel) | 	channelID := b.getChannelID(msg.Channel) | ||||||
| 	if channelID == "" { | 	if channelID == "" { | ||||||
| 		flog.Errorf("Could not find channelID for %v", msg.Channel) | 		flog.Errorf("Could not find channelID for %v", msg.Channel) | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL == "" { | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "_" + msg.Text + "_" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wID := b.webhookID | ||||||
|  | 	wToken := b.webhookToken | ||||||
|  | 	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok { | ||||||
|  | 		if ci.Options.WebhookURL != "" { | ||||||
|  | 			wID, wToken = b.splitURL(ci.Options.WebhookURL) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if wID == "" { | ||||||
| 		flog.Debugf("Broadcasting using token (API)") | 		flog.Debugf("Broadcasting using token (API)") | ||||||
| 		b.c.ChannelMessageSend(channelID, msg.Username+msg.Text) | 		if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	} else { | 			if msg.ID == "" { | ||||||
| 		flog.Debugf("Broadcasting using Webhook") | 				return "", nil | ||||||
| 		b.c.WebhookExecute( | 			} | ||||||
| 			b.webhookID, | 			err := b.c.ChannelMessageDelete(channelID, msg.ID) | ||||||
| 			b.webhookToken, | 			return "", err | ||||||
| 			true, | 		} | ||||||
| 			&discordgo.WebhookParams{ | 		if msg.ID != "" { | ||||||
| 				Content:   msg.Text, | 			_, err := b.c.ChannelMessageEdit(channelID, msg.ID, msg.Username+msg.Text) | ||||||
| 				Username:  msg.Username, | 			return msg.ID, err | ||||||
| 				AvatarURL: msg.Avatar, | 		} | ||||||
| 			}) |  | ||||||
|  | 		if msg.Extra != nil { | ||||||
|  | 			// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 			if len(msg.Extra["file"]) > 0 { | ||||||
|  | 				var err error | ||||||
|  | 				for _, f := range msg.Extra["file"] { | ||||||
|  | 					fi := f.(config.FileInfo) | ||||||
|  | 					files := []*discordgo.File{} | ||||||
|  | 					files = append(files, &discordgo.File{fi.Name, "", bytes.NewReader(*fi.Data)}) | ||||||
|  | 					_, err = b.c.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{Content: msg.Text, Files: files}) | ||||||
|  | 					if err != nil { | ||||||
|  | 						flog.Errorf("file upload failed: %#v", err) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		res, err := b.c.ChannelMessageSend(channelID, msg.Username+msg.Text) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return res.ID, err | ||||||
| 	} | 	} | ||||||
| 	return nil | 	flog.Debugf("Broadcasting using Webhook") | ||||||
|  | 	err := b.c.WebhookExecute( | ||||||
|  | 		wID, | ||||||
|  | 		wToken, | ||||||
|  | 		true, | ||||||
|  | 		&discordgo.WebhookParams{ | ||||||
|  | 			Content:   msg.Text, | ||||||
|  | 			Username:  msg.Username, | ||||||
|  | 			AvatarURL: msg.Avatar, | ||||||
|  | 		}) | ||||||
|  | 	return "", err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { | ||||||
|  | 	rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EVENT_MSG_DELETE, Text: config.EVENT_MSG_DELETE} | ||||||
|  | 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||||
|  | 	if b.UseChannelID { | ||||||
|  | 		rmsg.Channel = "ID:" + m.ChannelID | ||||||
|  | 	} | ||||||
|  | 	flog.Debugf("Sending message from %s to gateway", b.Account) | ||||||
|  | 	flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 	b.Remote <- rmsg | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | ||||||
| @@ -150,7 +207,7 @@ func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// if using webhooks, do not relay if it's ours | 	// if using webhooks, do not relay if it's ours | ||||||
| 	if b.Config.WebhookURL != "" && m.Author.Bot && m.Author.ID == b.webhookID { | 	if b.useWebhook() && m.Author.Bot && b.isWebhookID(m.Author.ID) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -171,11 +228,19 @@ func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | |||||||
| 		text = m.ContentWithMentionsReplaced() | 		text = m.ContentWithMentionsReplaced() | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	channelName := b.getChannelName(m.ChannelID) | 	rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", | ||||||
|  | 		UserID: m.Author.ID, ID: m.ID} | ||||||
|  |  | ||||||
|  | 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||||
| 	if b.UseChannelID { | 	if b.UseChannelID { | ||||||
| 		channelName = "ID:" + m.ChannelID | 		rmsg.Channel = "ID:" + m.ChannelID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !b.Config.UseUserName { | ||||||
|  | 		rmsg.Username = b.getNick(m.Author) | ||||||
|  | 	} else { | ||||||
|  | 		rmsg.Username = m.Author.Username | ||||||
| 	} | 	} | ||||||
| 	username := b.getNick(m.Author) |  | ||||||
|  |  | ||||||
| 	if b.Config.ShowEmbeds && m.Message.Embeds != nil { | 	if b.Config.ShowEmbeds && m.Message.Embeds != nil { | ||||||
| 		for _, embed := range m.Message.Embeds { | 		for _, embed := range m.Message.Embeds { | ||||||
| @@ -188,10 +253,15 @@ func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	text, ok := b.replaceAction(text) | ||||||
|  | 	if ok { | ||||||
|  | 		rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rmsg.Text = text | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account) | 	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||||
| 	b.Remote <- config.Message{Username: username, Text: text, Channel: channelName, | 	flog.Debugf("Message is %#v", rmsg) | ||||||
| 		Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", | 	b.Remote <- rmsg | ||||||
| 		UserID: m.Author.ID} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | func (b *bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||||
| @@ -283,8 +353,53 @@ func (b *bdiscord) replaceChannelMentions(text string) string { | |||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||||
|  | 		return strings.Replace(text, "_", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|  |  | ||||||
| func (b *bdiscord) stripCustomoji(text string) string { | func (b *bdiscord) stripCustomoji(text string) string { | ||||||
| 	// <:doge:302803592035958784> | 	// <:doge:302803592035958784> | ||||||
| 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||||
| 	return re.ReplaceAllString(text, `$1`) | 	return re.ReplaceAllString(text, `$1`) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // splitURL splits a webhookURL and returns the id and token | ||||||
|  | func (b *bdiscord) splitURL(url string) (string, string) { | ||||||
|  | 	webhookURLSplit := strings.Split(url, "/") | ||||||
|  | 	return webhookURLSplit[len(webhookURLSplit)-2], webhookURLSplit[len(webhookURLSplit)-1] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // useWebhook returns true if we have a webhook defined somewhere | ||||||
|  | func (b *bdiscord) useWebhook() bool { | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	for _, channel := range b.channelInfoMap { | ||||||
|  | 		if channel.Options.WebhookURL != "" { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // isWebhookID returns true if the specified id is used in a defined webhook | ||||||
|  | func (b *bdiscord) isWebhookID(id string) bool { | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		wID, _ := b.splitURL(b.Config.WebhookURL) | ||||||
|  | 		if wID == id { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, channel := range b.channelInfoMap { | ||||||
|  | 		if channel.Options.WebhookURL != "" { | ||||||
|  | 			wID, _ := b.splitURL(channel.Options.WebhookURL) | ||||||
|  | 			if wID == id { | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -2,9 +2,9 @@ package bgitter | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"github.com/42wim/go-gitter" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/sromku/go-gitter" |  | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -13,6 +13,7 @@ type Bgitter struct { | |||||||
| 	Config  *config.Protocol | 	Config  *config.Protocol | ||||||
| 	Remote  chan config.Message | 	Remote  chan config.Message | ||||||
| 	Account string | 	Account string | ||||||
|  | 	User    *gitter.User | ||||||
| 	Users   []gitter.User | 	Users   []gitter.User | ||||||
| 	Rooms   []gitter.Room | 	Rooms   []gitter.Room | ||||||
| } | } | ||||||
| @@ -36,7 +37,7 @@ func (b *Bgitter) Connect() error { | |||||||
| 	var err error | 	var err error | ||||||
| 	flog.Info("Connecting") | 	flog.Info("Connecting") | ||||||
| 	b.c = gitter.New(b.Config.Token) | 	b.c = gitter.New(b.Config.Token) | ||||||
| 	_, err = b.c.GetUser() | 	b.User, err = b.c.GetUser() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| @@ -51,10 +52,10 @@ func (b *Bgitter) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) JoinChannel(channel string) error { | func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	roomID, err := b.c.GetRoomId(channel) | 	roomID, err := b.c.GetRoomId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel) | 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel.Name) | ||||||
| 	} | 	} | ||||||
| 	room, err := b.c.GetRoom(roomID) | 	room, err := b.c.GetRoom(roomID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -78,29 +79,57 @@ func (b *Bgitter) JoinChannel(channel string) error { | |||||||
| 		for event := range stream.Event { | 		for event := range stream.Event { | ||||||
| 			switch ev := event.Data.(type) { | 			switch ev := event.Data.(type) { | ||||||
| 			case *gitter.MessageReceived: | 			case *gitter.MessageReceived: | ||||||
| 				// check for ZWSP to see if it's not an echo | 				if ev.Message.From.ID != b.User.ID { | ||||||
| 				if !strings.HasSuffix(ev.Message.Text, "") { |  | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | 					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | ||||||
| 					b.Remote <- config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | 					rmsg := config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | ||||||
| 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID} | 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID, | ||||||
|  | 						ID: ev.Message.ID} | ||||||
|  | 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | ||||||
|  | 						rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 					b.Remote <- rmsg | ||||||
| 				} | 				} | ||||||
| 			case *gitter.GitterConnectionClosed: | 			case *gitter.GitterConnectionClosed: | ||||||
| 				flog.Errorf("connection with gitter closed for room %s", room) | 				flog.Errorf("connection with gitter closed for room %s", room) | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	}(stream, room.Name) | 	}(stream, room.URI) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) Send(msg config.Message) error { | func (b *Bgitter) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	roomID := b.getRoomID(msg.Channel) | 	roomID := b.getRoomID(msg.Channel) | ||||||
| 	if roomID == "" { | 	if roomID == "" { | ||||||
| 		flog.Errorf("Could not find roomID for %v", msg.Channel) | 		flog.Errorf("Could not find roomID for %v", msg.Channel) | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	// add ZWSP because gitter echoes our own messages | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	return b.c.SendMessage(roomID, msg.Username+msg.Text+" ") | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		// gitter has no delete message api | ||||||
|  | 		_, err := b.c.UpdateMessage(roomID, msg.ID, "") | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		flog.Debugf("updating message with id %s", msg.ID) | ||||||
|  | 		_, err := b.c.UpdateMessage(roomID, msg.ID, msg.Username+msg.Text) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	resp, err := b.c.SendMessage(roomID, msg.Username+msg.Text) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return resp.ID, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) getRoomID(channel string) string { | func (b *Bgitter) getRoomID(channel string) string { | ||||||
|   | |||||||
							
								
								
									
										28
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | |||||||
|  | package helper | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func DownloadFile(url string) (*[]byte, error) { | ||||||
|  | 	var buf bytes.Buffer | ||||||
|  | 	client := &http.Client{ | ||||||
|  | 		Timeout: time.Second * 5, | ||||||
|  | 	} | ||||||
|  | 	req, err := http.NewRequest("GET", url, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	resp, err := client.Do(req) | ||||||
|  | 	if err != nil { | ||||||
|  | 		resp.Body.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	io.Copy(&buf, resp.Body) | ||||||
|  | 	data := buf.Bytes() | ||||||
|  | 	resp.Body.Close() | ||||||
|  | 	return &data, nil | ||||||
|  | } | ||||||
| @@ -1,17 +1,18 @@ | |||||||
| package birc | package birc | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/lrstanley/girc" | ||||||
| 	"github.com/paulrosania/go-charset/charset" | 	"github.com/paulrosania/go-charset/charset" | ||||||
| 	_ "github.com/paulrosania/go-charset/data" | 	_ "github.com/paulrosania/go-charset/data" | ||||||
| 	"github.com/saintfish/chardet" | 	"github.com/saintfish/chardet" | ||||||
| 	ircm "github.com/sorcix/irc" |  | ||||||
| 	"github.com/thoj/go-ircevent" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
|  | 	"net" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"sort" | 	"sort" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| @@ -20,7 +21,7 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| type Birc struct { | type Birc struct { | ||||||
| 	i               *irc.Connection | 	i               *girc.Client | ||||||
| 	Nick            string | 	Nick            string | ||||||
| 	names           map[string][]string | 	names           map[string][]string | ||||||
| 	Config          *config.Protocol | 	Config          *config.Protocol | ||||||
| @@ -62,9 +63,9 @@ func New(cfg config.Protocol, account string, c chan config.Message) *Birc { | |||||||
| func (b *Birc) Command(msg *config.Message) string { | func (b *Birc) Command(msg *config.Message) string { | ||||||
| 	switch msg.Text { | 	switch msg.Text { | ||||||
| 	case "!users": | 	case "!users": | ||||||
| 		b.i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames) | 		b.i.Handlers.Add(girc.RPL_NAMREPLY, b.storeNames) | ||||||
| 		b.i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames) | 		b.i.Handlers.Add(girc.RPL_ENDOFNAMES, b.endNames) | ||||||
| 		b.i.SendRaw("NAMES " + msg.Channel) | 		b.i.Cmd.SendRaw("NAMES " + msg.Channel) | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
| @@ -72,25 +73,50 @@ func (b *Birc) Command(msg *config.Message) string { | |||||||
| func (b *Birc) Connect() error { | func (b *Birc) Connect() error { | ||||||
| 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | ||||||
| 	flog.Infof("Connecting %s", b.Config.Server) | 	flog.Infof("Connecting %s", b.Config.Server) | ||||||
| 	i := irc.IRC(b.Config.Nick, b.Config.Nick) | 	server, portstr, err := net.SplitHostPort(b.Config.Server) | ||||||
| 	if log.GetLevel() == log.DebugLevel { |  | ||||||
| 		i.Debug = true |  | ||||||
| 	} |  | ||||||
| 	i.UseTLS = b.Config.UseTLS |  | ||||||
| 	i.UseSASL = b.Config.UseSASL |  | ||||||
| 	i.SASLLogin = b.Config.NickServNick |  | ||||||
| 	i.SASLPassword = b.Config.NickServPassword |  | ||||||
| 	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify} |  | ||||||
| 	i.KeepAlive = time.Minute |  | ||||||
| 	i.PingFreq = time.Minute |  | ||||||
| 	if b.Config.Password != "" { |  | ||||||
| 		i.Password = b.Config.Password |  | ||||||
| 	} |  | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, b.handleNewConnection) |  | ||||||
| 	err := i.Connect(b.Config.Server) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	port, err := strconv.Atoi(portstr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	i := girc.New(girc.Config{ | ||||||
|  | 		Server:     server, | ||||||
|  | 		ServerPass: b.Config.Password, | ||||||
|  | 		Port:       port, | ||||||
|  | 		Nick:       b.Config.Nick, | ||||||
|  | 		User:       b.Config.Nick, | ||||||
|  | 		Name:       b.Config.Nick, | ||||||
|  | 		SSL:        b.Config.UseTLS, | ||||||
|  | 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, ServerName: server}, | ||||||
|  | 		PingDelay:  time.Minute, | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	if b.Config.UseSASL { | ||||||
|  | 		i.Config.SASL = &girc.SASLPlain{b.Config.NickServNick, b.Config.NickServPassword} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	i.Handlers.Add(girc.RPL_WELCOME, b.handleNewConnection) | ||||||
|  | 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||||
|  | 	i.Handlers.Add("*", b.handleOther) | ||||||
|  | 	go func() { | ||||||
|  | 		for { | ||||||
|  | 			if err := i.Connect(); err != nil { | ||||||
|  | 				flog.Errorf("error: %s", err) | ||||||
|  | 				flog.Info("reconnecting in 30 seconds...") | ||||||
|  | 				time.Sleep(30 * time.Second) | ||||||
|  | 				i.Handlers.Clear(girc.RPL_WELCOME) | ||||||
|  | 				i.Handlers.Add(girc.RPL_WELCOME, func(client *girc.Client, event girc.Event) { | ||||||
|  | 					b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
|  | 					// set our correct nick on reconnect if necessary | ||||||
|  | 					b.Nick = event.Source.Name | ||||||
|  | 				}) | ||||||
|  | 			} else { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
| 	b.i = i | 	b.i = i | ||||||
| 	select { | 	select { | ||||||
| 	case <-b.connected: | 	case <-b.connected: | ||||||
| @@ -98,15 +124,8 @@ func (b *Birc) Connect() error { | |||||||
| 	case <-time.After(time.Second * 30): | 	case <-time.After(time.Second * 30): | ||||||
| 		return fmt.Errorf("connection timed out") | 		return fmt.Errorf("connection timed out") | ||||||
| 	} | 	} | ||||||
| 	i.Debug = false | 	//i.Debug = false | ||||||
| 	// clear on reconnects | 	i.Handlers.Clear("*") | ||||||
| 	i.ClearCallback(ircm.RPL_WELCOME) |  | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, func(event *irc.Event) { |  | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} |  | ||||||
| 		// set our correct nick on reconnect if necessary |  | ||||||
| 		b.Nick = event.Nick |  | ||||||
| 	}) |  | ||||||
| 	go i.Loop() |  | ||||||
| 	go b.doSend() | 	go b.doSend() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -117,19 +136,38 @@ func (b *Birc) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) JoinChannel(channel string) error { | func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.i.Join(channel) | 	if channel.Options.Key != "" { | ||||||
|  | 		flog.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||||
|  | 		b.i.Cmd.JoinKey(channel.Name, channel.Options.Key) | ||||||
|  | 	} else { | ||||||
|  | 		b.i.Cmd.Join(channel.Name) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Send(msg config.Message) error { | func (b *Birc) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	// ignore delete messages | ||||||
| 	if msg.Account == b.Account { | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
|  | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	if strings.HasPrefix(msg.Text, "!") { | 	if strings.HasPrefix(msg.Text, "!") { | ||||||
| 		b.Command(&msg) | 		b.Command(&msg) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if b.Config.Charset != "" { | ||||||
|  | 		buf := new(bytes.Buffer) | ||||||
|  | 		w, err := charset.NewWriter(b.Config.Charset, buf) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("charset from utf-8 conversion failed: %s", err) | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		fmt.Fprintf(w, msg.Text) | ||||||
|  | 		w.Close() | ||||||
|  | 		msg.Text = buf.String() | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, text := range strings.Split(msg.Text, "\n") { | 	for _, text := range strings.Split(msg.Text, "\n") { | ||||||
| 		if len(text) > b.Config.MessageLength { | 		if len(text) > b.Config.MessageLength { | ||||||
| 			text = text[:b.Config.MessageLength] + " <message clipped>" | 			text = text[:b.Config.MessageLength] + " <message clipped>" | ||||||
| @@ -138,12 +176,12 @@ func (b *Birc) Send(msg config.Message) error { | |||||||
| 			if len(b.Local) == b.Config.MessageQueue-1 { | 			if len(b.Local) == b.Config.MessageQueue-1 { | ||||||
| 				text = text + " <message clipped>" | 				text = text + " <message clipped>" | ||||||
| 			} | 			} | ||||||
| 			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel} | 			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||||
| 		} else { | 		} else { | ||||||
| 			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | 			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) doSend() { | func (b *Birc) doSend() { | ||||||
| @@ -151,12 +189,16 @@ func (b *Birc) doSend() { | |||||||
| 	throttle := time.NewTicker(rate) | 	throttle := time.NewTicker(rate) | ||||||
| 	for msg := range b.Local { | 	for msg := range b.Local { | ||||||
| 		<-throttle.C | 		<-throttle.C | ||||||
| 		b.i.Privmsg(msg.Channel, msg.Username+msg.Text) | 		if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 			b.i.Cmd.Action(msg.Channel, msg.Username+msg.Text) | ||||||
|  | 		} else { | ||||||
|  | 			b.i.Cmd.Message(msg.Channel, msg.Username+msg.Text) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) endNames(event *irc.Event) { | func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[1] | 	channel := event.Params[1] | ||||||
| 	sort.Strings(b.names[channel]) | 	sort.Strings(b.names[channel]) | ||||||
| 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | ||||||
| 	continued := false | 	continued := false | ||||||
| @@ -169,108 +211,122 @@ func (b *Birc) endNames(event *irc.Event) { | |||||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | ||||||
| 		Channel: channel, Account: b.Account} | 		Channel: channel, Account: b.Account} | ||||||
| 	b.names[channel] = nil | 	b.names[channel] = nil | ||||||
| 	b.i.ClearCallback(ircm.RPL_NAMREPLY) | 	b.i.Handlers.Clear(girc.RPL_NAMREPLY) | ||||||
| 	b.i.ClearCallback(ircm.RPL_ENDOFNAMES) | 	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNewConnection(event *irc.Event) { | func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||||
| 	flog.Debug("Registering callbacks") | 	flog.Debug("Registering callbacks") | ||||||
| 	i := b.i | 	i := b.i | ||||||
| 	b.Nick = event.Arguments[0] | 	b.Nick = event.Params[0] | ||||||
| 	i.AddCallback("PRIVMSG", b.handlePrivMsg) |  | ||||||
| 	i.AddCallback("CTCP_ACTION", b.handlePrivMsg) | 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||||
| 	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||||
| 	i.AddCallback(ircm.NOTICE, b.handleNotice) | 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||||
| 	//i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) }) | 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||||
| 	i.AddCallback("PING", func(e *irc.Event) { | 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||||
| 		i.SendRaw("PONG :" + e.Message()) | 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||||
| 		flog.Debugf("PING/PONG") | 	i.Handlers.Add("PART", b.handleJoinPart) | ||||||
| 	}) | 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||||
| 	i.AddCallback("JOIN", b.handleJoinPart) | 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||||
| 	i.AddCallback("PART", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("QUIT", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("KICK", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("*", b.handleOther) |  | ||||||
| 	// we are now fully connected | 	// we are now fully connected | ||||||
| 	b.connected <- struct{}{} | 	b.connected <- struct{}{} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleJoinPart(event *irc.Event) { | func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[0] | 	if len(event.Params) == 0 { | ||||||
| 	if event.Code == "KICK" { | 		flog.Debugf("handleJoinPart: empty Params? %#v", event) | ||||||
| 		flog.Infof("Got kicked from %s by %s", channel, event.Nick) | 		return | ||||||
|  | 	} | ||||||
|  | 	channel := event.Params[0] | ||||||
|  | 	if event.Command == "KICK" { | ||||||
|  | 		flog.Infof("Got kicked from %s by %s", channel, event.Source.Name) | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if event.Code == "QUIT" { | 	if event.Command == "QUIT" { | ||||||
| 		if event.Nick == b.Nick && strings.Contains(event.Raw, "Ping timeout") { | 		if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") { | ||||||
| 			flog.Infof("%s reconnecting ..", b.Account) | 			flog.Infof("%s reconnecting ..", b.Account) | ||||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	if event.Nick != b.Nick { | 	if event.Source.Name != b.Nick { | ||||||
| 		flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | 		flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: event.Nick + " " + strings.ToLower(event.Code) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | 		b.Remote <- config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("handle %#v", event) | 	flog.Debugf("handle %#v", event) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNotice(event *irc.Event) { | func (b *Birc) handleNotice(client *girc.Client, event girc.Event) { | ||||||
| 	if strings.Contains(event.Message(), "This nickname is registered") && event.Nick == b.Config.NickServNick { | 	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.Config.NickServNick { | ||||||
| 		b.i.Privmsg(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword) | 		b.i.Cmd.Message(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword) | ||||||
| 	} else { | 	} else { | ||||||
| 		b.handlePrivMsg(event) | 		b.handlePrivMsg(client, event) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleOther(event *irc.Event) { | func (b *Birc) handleOther(client *girc.Client, event girc.Event) { | ||||||
| 	switch event.Code { | 	switch event.Command { | ||||||
| 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("%#v", event.Raw) | 	flog.Debugf("%#v", event.String()) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handlePrivMsg(event *irc.Event) { | func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) { | ||||||
|  | 	if strings.EqualFold(b.Config.NickServNick, "Q@CServe.quakenet.org") { | ||||||
|  | 		flog.Debugf("Authenticating %s against %s", b.Config.NickServUsername, b.Config.NickServNick) | ||||||
|  | 		b.i.Cmd.Message(b.Config.NickServNick, "AUTH "+b.Config.NickServUsername+" "+b.Config.NickServPassword) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||||
| 	b.Nick = b.i.GetNick() | 	b.Nick = b.i.GetNick() | ||||||
| 	// freenode doesn't send 001 as first reply | 	// freenode doesn't send 001 as first reply | ||||||
| 	if event.Code == "NOTICE" { | 	if event.Command == "NOTICE" { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// don't forward queries to the bot | 	// don't forward queries to the bot | ||||||
| 	if event.Arguments[0] == b.Nick { | 	if event.Params[0] == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// don't forward message from ourself | 	// don't forward message from ourself | ||||||
| 	if event.Nick == b.Nick { | 	if event.Source.Name == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("handlePrivMsg() %s %s %#v", event.Nick, event.Message(), event) | 	rmsg := config.Message{Username: event.Source.Name, Channel: event.Params[0], Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||||
|  | 	flog.Debugf("handlePrivMsg() %s %s %#v", event.Source.Name, event.Trailing, event) | ||||||
| 	msg := "" | 	msg := "" | ||||||
| 	if event.Code == "CTCP_ACTION" { | 	if event.Command == "CTCP_ACTION" { | ||||||
| 		msg = event.Nick + " " | 		//	msg = event.Source.Name + " " | ||||||
|  | 		rmsg.Event = config.EVENT_USER_ACTION | ||||||
| 	} | 	} | ||||||
| 	msg += event.Message() | 	msg += event.Trailing | ||||||
| 	// strip IRC colors | 	// strip IRC colors | ||||||
| 	re := regexp.MustCompile(`[[:cntrl:]](\d+,|)\d+`) | 	re := regexp.MustCompile(`[[:cntrl:]](?:\d{1,2}(?:,\d{1,2})?)?`) | ||||||
| 	msg = re.ReplaceAllString(msg, "") | 	msg = re.ReplaceAllString(msg, "") | ||||||
|  |  | ||||||
| 	// detect what were sending so that we convert it to utf-8 |  | ||||||
| 	detector := chardet.NewTextDetector() |  | ||||||
| 	result, err := detector.DetectBest([]byte(msg)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		flog.Infof("detection failed for msg: %#v", msg) |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) |  | ||||||
| 	var r io.Reader | 	var r io.Reader | ||||||
| 	r, err = charset.NewReader(result.Charset, strings.NewReader(msg)) | 	var err error | ||||||
| 	// if we're not sure, just pick ISO-8859-1 | 	mycharset := b.Config.Charset | ||||||
| 	if result.Confidence < 80 { | 	if mycharset == "" { | ||||||
| 		r, err = charset.NewReader("ISO-8859-1", strings.NewReader(msg)) | 		// detect what were sending so that we convert it to utf-8 | ||||||
|  | 		detector := chardet.NewTextDetector() | ||||||
|  | 		result, err := detector.DetectBest([]byte(msg)) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Infof("detection failed for msg: %#v", msg) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) | ||||||
|  | 		mycharset = result.Charset | ||||||
|  | 		// if we're not sure, just pick ISO-8859-1 | ||||||
|  | 		if result.Confidence < 80 { | ||||||
|  | 			mycharset = "ISO-8859-1" | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  | 	r, err = charset.NewReader(mycharset, strings.NewReader(msg)) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Errorf("charset to utf-8 conversion failed: %s", err) | 		flog.Errorf("charset to utf-8 conversion failed: %s", err) | ||||||
| 		return | 		return | ||||||
| @@ -278,48 +334,35 @@ func (b *Birc) handlePrivMsg(event *irc.Event) { | |||||||
| 	output, _ := ioutil.ReadAll(r) | 	output, _ := ioutil.ReadAll(r) | ||||||
| 	msg = string(output) | 	msg = string(output) | ||||||
|  |  | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", event.Arguments[0], b.Account) | 	flog.Debugf("Sending message from %s on %s to gateway", event.Params[0], b.Account) | ||||||
| 	b.Remote <- config.Message{Username: event.Nick, Text: msg, Channel: event.Arguments[0], Account: b.Account, UserID: event.User + "@" + event.Host} | 	rmsg.Text = msg | ||||||
|  | 	b.Remote <- rmsg | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleTopicWhoTime(event *irc.Event) { | func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) { | ||||||
| 	parts := strings.Split(event.Arguments[2], "!") | 	parts := strings.Split(event.Params[2], "!") | ||||||
| 	t, err := strconv.ParseInt(event.Arguments[3], 10, 64) | 	t, err := strconv.ParseInt(event.Params[3], 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Errorf("Invalid time stamp: %s", event.Arguments[3]) | 		flog.Errorf("Invalid time stamp: %s", event.Params[3]) | ||||||
| 	} | 	} | ||||||
| 	user := parts[0] | 	user := parts[0] | ||||||
| 	if len(parts) > 1 { | 	if len(parts) > 1 { | ||||||
| 		user += " [" + parts[1] + "]" | 		user += " [" + parts[1] + "]" | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0)) | 	flog.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0)) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) nicksPerRow() int { | func (b *Birc) nicksPerRow() int { | ||||||
| 	return 4 | 	return 4 | ||||||
| 	/* |  | ||||||
| 		if b.Config.Mattermost.NicksPerRow < 1 { |  | ||||||
| 			return 4 |  | ||||||
| 		} |  | ||||||
| 		return b.Config.Mattermost.NicksPerRow |  | ||||||
| 	*/ |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) storeNames(event *irc.Event) { | func (b *Birc) storeNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[2] | 	channel := event.Params[2] | ||||||
| 	b.names[channel] = append( | 	b.names[channel] = append( | ||||||
| 		b.names[channel], | 		b.names[channel], | ||||||
| 		strings.Split(strings.TrimSpace(event.Message()), " ")...) | 		strings.Split(strings.TrimSpace(event.Trailing), " ")...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) formatnicks(nicks []string, continued bool) string { | func (b *Birc) formatnicks(nicks []string, continued bool) string { | ||||||
| 	return plainformatter(nicks, b.nicksPerRow()) | 	return plainformatter(nicks, b.nicksPerRow()) | ||||||
| 	/* |  | ||||||
| 		switch b.Config.Mattermost.NickFormatter { |  | ||||||
| 		case "table": |  | ||||||
| 			return tableformatter(nicks, b.nicksPerRow(), continued) |  | ||||||
| 		default: |  | ||||||
| 			return plainformatter(nicks, b.nicksPerRow()) |  | ||||||
| 		} |  | ||||||
| 	*/ |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -63,23 +63,32 @@ func (b *Bmatrix) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) JoinChannel(channel string) error { | func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	resp, err := b.mc.JoinRoom(channel, "", nil) | 	resp, err := b.mc.JoinRoom(channel.Name, "", nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	b.RoomMap[resp.RoomID] = channel | 	b.RoomMap[resp.RoomID] = channel.Name | ||||||
| 	b.Unlock() | 	b.Unlock() | ||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) Send(msg config.Message) error { | func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	channel := b.getRoomID(msg.Channel) | 	channel := b.getRoomID(msg.Channel) | ||||||
| 	flog.Debugf("Sending to channel %s", channel) | 	flog.Debugf("Sending to channel %s", channel) | ||||||
|  | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		b.mc.SendMessageEvent(channel, "m.room.message", | ||||||
|  | 			matrix.TextMessage{"m.emote", msg.Username + msg.Text}) | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	b.mc.SendText(channel, msg.Username+msg.Text) | 	b.mc.SendText(channel, msg.Username+msg.Text) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) getRoomID(channel string) string { | func (b *Bmatrix) getRoomID(channel string) string { | ||||||
| @@ -95,7 +104,7 @@ func (b *Bmatrix) getRoomID(channel string) string { | |||||||
| func (b *Bmatrix) handlematrix() error { | func (b *Bmatrix) handlematrix() error { | ||||||
| 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||||
| 	syncer.OnEventType("m.room.message", func(ev *matrix.Event) { | 	syncer.OnEventType("m.room.message", func(ev *matrix.Event) { | ||||||
| 		if ev.Content["msgtype"].(string) == "m.text" && ev.Sender != b.UserID { | 		if (ev.Content["msgtype"].(string) == "m.text" || ev.Content["msgtype"].(string) == "m.notice" || ev.Content["msgtype"].(string) == "m.emote") && ev.Sender != b.UserID { | ||||||
| 			b.RLock() | 			b.RLock() | ||||||
| 			channel, ok := b.RoomMap[ev.RoomID] | 			channel, ok := b.RoomMap[ev.RoomID] | ||||||
| 			b.RUnlock() | 			b.RUnlock() | ||||||
| @@ -108,8 +117,12 @@ func (b *Bmatrix) handlematrix() error { | |||||||
| 				re := regexp.MustCompile("(.*?):.*") | 				re := regexp.MustCompile("(.*?):.*") | ||||||
| 				username = re.ReplaceAllString(username, `$1`) | 				username = re.ReplaceAllString(username, `$1`) | ||||||
| 			} | 			} | ||||||
|  | 			rmsg := config.Message{Username: username, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account, UserID: ev.Sender} | ||||||
|  | 			if ev.Content["msgtype"].(string) == "m.emote" { | ||||||
|  | 				rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 			} | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) | 			flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) | ||||||
| 			b.Remote <- config.Message{Username: username, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account, UserID: ev.Sender} | 			b.Remote <- rmsg | ||||||
| 		} | 		} | ||||||
| 		flog.Debugf("Received: %#v", ev) | 		flog.Debugf("Received: %#v", ev) | ||||||
| 	}) | 	}) | ||||||
|   | |||||||
| @@ -2,10 +2,12 @@ package bmattermost | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/matterclient" | 	"github.com/42wim/matterbridge/matterclient" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type MMhook struct { | type MMhook struct { | ||||||
| @@ -22,6 +24,9 @@ type MMMessage struct { | |||||||
| 	Channel  string | 	Channel  string | ||||||
| 	Username string | 	Username string | ||||||
| 	UserID   string | 	UserID   string | ||||||
|  | 	ID       string | ||||||
|  | 	Event    string | ||||||
|  | 	Extra    map[string][]interface{} | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bmattermost struct { | type Bmattermost struct { | ||||||
| @@ -60,6 +65,12 @@ func (b *Bmattermost) Connect() error { | |||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
|  | 		} else if b.Config.Token != "" { | ||||||
|  | 			flog.Info("Connecting using token (sending)") | ||||||
|  | 			err := b.apiLogin() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
| 		} else if b.Config.Login != "" { | 		} else if b.Config.Login != "" { | ||||||
| 			flog.Info("Connecting using login/password (sending)") | 			flog.Info("Connecting using login/password (sending)") | ||||||
| 			err := b.apiLogin() | 			err := b.apiLogin() | ||||||
| @@ -80,7 +91,14 @@ func (b *Bmattermost) Connect() error { | |||||||
| 		b.mh = matterhook.New(b.Config.WebhookURL, | 		b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
| 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
| 				DisableServer: true}) | 				DisableServer: true}) | ||||||
| 		if b.Config.Login != "" { | 		if b.Config.Token != "" { | ||||||
|  | 			flog.Info("Connecting using token (receiving)") | ||||||
|  | 			err := b.apiLogin() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			go b.handleMatter() | ||||||
|  | 		} else if b.Config.Login != "" { | ||||||
| 			flog.Info("Connecting using login/password (receiving)") | 			flog.Info("Connecting using login/password (receiving)") | ||||||
| 			err := b.apiLogin() | 			err := b.apiLogin() | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| @@ -89,6 +107,13 @@ func (b *Bmattermost) Connect() error { | |||||||
| 			go b.handleMatter() | 			go b.handleMatter() | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return nil | ||||||
|  | 	} else if b.Config.Token != "" { | ||||||
|  | 		flog.Info("Connecting using token (sending and receiving)") | ||||||
|  | 		err := b.apiLogin() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		go b.handleMatter() | ||||||
| 	} else if b.Config.Login != "" { | 	} else if b.Config.Login != "" { | ||||||
| 		flog.Info("Connecting using login/password (sending and receiving)") | 		flog.Info("Connecting using login/password (sending and receiving)") | ||||||
| 		err := b.apiLogin() | 		err := b.apiLogin() | ||||||
| @@ -97,8 +122,8 @@ func (b *Bmattermost) Connect() error { | |||||||
| 		} | 		} | ||||||
| 		go b.handleMatter() | 		go b.handleMatter() | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Login == "" { | 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Login == "" && b.Config.Token == "" { | ||||||
| 		return errors.New("No connection method found. See that you have WebhookBindAddress, WebhookURL or Login/Password/Server/Team configured.") | 		return errors.New("No connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured.") | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -107,16 +132,23 @@ func (b *Bmattermost) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) JoinChannel(channel string) error { | func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	// we can only join channels using the API | 	// we can only join channels using the API | ||||||
| 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | ||||||
| 		return b.mc.JoinChannel(b.mc.GetChannelId(channel, "")) | 		id := b.mc.GetChannelId(channel.Name, "") | ||||||
|  | 		if id == "" { | ||||||
|  | 			return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) | ||||||
|  | 		} | ||||||
|  | 		return b.mc.JoinChannel(id) | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Send(msg config.Message) error { | func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "*" + msg.Text + "*" | ||||||
|  | 	} | ||||||
| 	nick := msg.Username | 	nick := msg.Username | ||||||
| 	message := msg.Text | 	message := msg.Text | ||||||
| 	channel := msg.Channel | 	channel := msg.Channel | ||||||
| @@ -131,15 +163,46 @@ func (b *Bmattermost) Send(msg config.Message) error { | |||||||
| 		matterMessage.UserName = nick | 		matterMessage.UserName = nick | ||||||
| 		matterMessage.Type = "" | 		matterMessage.Type = "" | ||||||
| 		matterMessage.Text = message | 		matterMessage.Text = message | ||||||
|  | 		matterMessage.Text = message | ||||||
|  | 		matterMessage.Props = make(map[string]interface{}) | ||||||
|  | 		matterMessage.Props["matterbridge"] = true | ||||||
| 		err := b.mh.Send(matterMessage) | 		err := b.mh.Send(matterMessage) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			flog.Info(err) | 			flog.Info(err) | ||||||
| 			return err | 			return "", err | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message) | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	return nil | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		return msg.ID, b.mc.DeleteMessage(msg.ID) | ||||||
|  | 	} | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var err error | ||||||
|  | 			var res, id string | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				id, err = b.mc.UploadFile(*fi.Data, b.mc.GetChannelId(channel, ""), fi.Name) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Debugf("ERROR %#v", err) | ||||||
|  | 					return "", err | ||||||
|  | 				} | ||||||
|  | 				message = "uploaded a file: " + fi.Name | ||||||
|  | 				if b.Config.PrefixMessagesWithNick { | ||||||
|  | 					message = nick + "uploaded a file: " + fi.Name | ||||||
|  | 				} | ||||||
|  | 				res, err = b.mc.PostMessageWithFiles(b.mc.GetChannelId(channel, ""), message, []string{id}) | ||||||
|  | 			} | ||||||
|  | 			return res, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		return b.mc.EditMessage(msg.ID, message) | ||||||
|  | 	} | ||||||
|  | 	return b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) handleMatter() { | func (b *Bmattermost) handleMatter() { | ||||||
| @@ -148,12 +211,23 @@ func (b *Bmattermost) handleMatter() { | |||||||
| 		flog.Debugf("Choosing webhooks based receiving") | 		flog.Debugf("Choosing webhooks based receiving") | ||||||
| 		go b.handleMatterHook(mchan) | 		go b.handleMatterHook(mchan) | ||||||
| 	} else { | 	} else { | ||||||
| 		flog.Debugf("Choosing login/password based receiving") | 		if b.Config.Token != "" { | ||||||
|  | 			flog.Debugf("Choosing token based receiving") | ||||||
|  | 		} else { | ||||||
|  | 			flog.Debugf("Choosing login/password based receiving") | ||||||
|  | 		} | ||||||
| 		go b.handleMatterClient(mchan) | 		go b.handleMatterClient(mchan) | ||||||
| 	} | 	} | ||||||
| 	for message := range mchan { | 	for message := range mchan { | ||||||
|  | 		rmsg := config.Message{Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID, ID: message.ID, Event: message.Event, Extra: message.Extra} | ||||||
|  | 		text, ok := b.replaceAction(message.Text) | ||||||
|  | 		if ok { | ||||||
|  | 			rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 		} | ||||||
|  | 		rmsg.Text = text | ||||||
| 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | ||||||
| 		b.Remote <- config.Message{Text: message.Text, Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID} | 		flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 		b.Remote <- rmsg | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -170,19 +244,42 @@ func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) { | |||||||
| 		if (message.Raw.Event == "post_edited") && b.Config.EditDisable { | 		if (message.Raw.Event == "post_edited") && b.Config.EditDisable { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		m := &MMMessage{Extra: make(map[string][]interface{})} | ||||||
|  |  | ||||||
|  | 		props := message.Post.Props | ||||||
|  | 		if props != nil { | ||||||
|  | 			if _, ok := props["matterbridge"].(bool); ok { | ||||||
|  | 				flog.Debugf("sent by matterbridge, ignoring") | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if _, ok := props["override_username"].(string); ok { | ||||||
|  | 				message.Username = props["override_username"].(string) | ||||||
|  | 			} | ||||||
|  | 			if _, ok := props["attachments"].([]interface{}); ok { | ||||||
|  | 				m.Extra["attachments"] = props["attachments"].([]interface{}) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 		// do not post our own messages back to irc | 		// do not post our own messages back to irc | ||||||
| 		// only listen to message from our team | 		// only listen to message from our team | ||||||
| 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited") && | 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited" || message.Raw.Event == "post_deleted") && | ||||||
| 			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId { | 			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId { | ||||||
|  | 			// if the message has reactions don't repost it (for now, until we can correlate reaction with message) | ||||||
|  | 			if message.Post.HasReactions { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
| 			flog.Debugf("Receiving from matterclient %#v", message) | 			flog.Debugf("Receiving from matterclient %#v", message) | ||||||
| 			m := &MMMessage{} |  | ||||||
| 			m.UserID = message.UserID | 			m.UserID = message.UserID | ||||||
| 			m.Username = message.Username | 			m.Username = message.Username | ||||||
| 			m.Channel = message.Channel | 			m.Channel = message.Channel | ||||||
| 			m.Text = message.Text | 			m.Text = message.Text | ||||||
|  | 			m.ID = message.Post.Id | ||||||
| 			if message.Raw.Event == "post_edited" && !b.Config.EditDisable { | 			if message.Raw.Event == "post_edited" && !b.Config.EditDisable { | ||||||
| 				m.Text = message.Text + b.Config.EditSuffix | 				m.Text = message.Text + b.Config.EditSuffix | ||||||
| 			} | 			} | ||||||
|  | 			if message.Raw.Event == "post_deleted" { | ||||||
|  | 				m.Event = config.EVENT_MSG_DELETE | ||||||
|  | 			} | ||||||
| 			if len(message.Post.FileIds) > 0 { | 			if len(message.Post.FileIds) > 0 { | ||||||
| 				for _, link := range b.mc.GetFileLinks(message.Post.FileIds) { | 				for _, link := range b.mc.GetFileLinks(message.Post.FileIds) { | ||||||
| 					m.Text = m.Text + "\n" + link | 					m.Text = m.Text + "\n" + link | ||||||
| @@ -207,7 +304,12 @@ func (b *Bmattermost) handleMatterHook(mchan chan *MMMessage) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) apiLogin() error { | func (b *Bmattermost) apiLogin() error { | ||||||
| 	b.mc = matterclient.New(b.Config.Login, b.Config.Password, | 	password := b.Config.Password | ||||||
|  | 	if b.Config.Token != "" { | ||||||
|  | 		password = "MMAUTHTOKEN=" + b.Config.Token | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.mc = matterclient.New(b.Config.Login, password, | ||||||
| 		b.Config.Team, b.Config.Server) | 		b.Config.Team, b.Config.Server) | ||||||
| 	b.mc.SkipTLSVerify = b.Config.SkipTLSVerify | 	b.mc.SkipTLSVerify = b.Config.SkipTLSVerify | ||||||
| 	b.mc.NoTLS = b.Config.NoTLS | 	b.mc.NoTLS = b.Config.NoTLS | ||||||
| @@ -222,3 +324,10 @@ func (b *Bmattermost) apiLogin() error { | |||||||
| 	go b.mc.StatusLoop() | 	go b.mc.StatusLoop() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bmattermost) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | ||||||
|  | 		return strings.Replace(text, "*", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -53,11 +53,15 @@ func (b *Brocketchat) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) JoinChannel(channel string) error { | func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Send(msg config.Message) error { | func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | ||||||
| 	matterMessage.Channel = msg.Channel | 	matterMessage.Channel = msg.Channel | ||||||
| @@ -67,9 +71,9 @@ func (b *Brocketchat) Send(msg config.Message) error { | |||||||
| 	err := b.mh.Send(matterMessage) | 	err := b.mh.Send(matterMessage) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Info(err) | 		flog.Info(err) | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) handleRocketHook() { | func (b *Brocketchat) handleRocketHook() { | ||||||
|   | |||||||
| @@ -1,13 +1,16 @@ | |||||||
| package bslack | package bslack | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/nlopes/slack" | 	"github.com/matterbridge/slack" | ||||||
| 	"html" | 	"html" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -108,14 +111,14 @@ func (b *Bslack) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) JoinChannel(channel string) error { | func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	// we can only join channels using the API | 	// we can only join channels using the API | ||||||
| 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | ||||||
| 		if strings.HasPrefix(b.Config.Token, "xoxb") { | 		if strings.HasPrefix(b.Config.Token, "xoxb") { | ||||||
| 			// TODO check if bot has already joined channel | 			// TODO check if bot has already joined channel | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 		_, err := b.sc.JoinChannel(channel) | 		_, err := b.sc.JoinChannel(channel.Name) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			if err.Error() != "name_taken" { | 			if err.Error() != "name_taken" { | ||||||
| 				return err | 				return err | ||||||
| @@ -125,8 +128,11 @@ func (b *Bslack) JoinChannel(channel string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Send(msg config.Message) error { | func (b *Bslack) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "_" + msg.Text + "_" | ||||||
|  | 	} | ||||||
| 	nick := msg.Username | 	nick := msg.Username | ||||||
| 	message := msg.Text | 	message := msg.Text | ||||||
| 	channel := msg.Channel | 	channel := msg.Channel | ||||||
| @@ -142,13 +148,13 @@ func (b *Bslack) Send(msg config.Message) error { | |||||||
| 		err := b.mh.Send(matterMessage) | 		err := b.mh.Send(matterMessage) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			flog.Info(err) | 			flog.Info(err) | ||||||
| 			return err | 			return "", err | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	schannel, err := b.getChannelByName(channel) | 	schannel, err := b.getChannelByName(channel) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	np := slack.NewPostMessageParameters() | 	np := slack.NewPostMessageParameters() | ||||||
| 	if b.Config.PrefixMessagesWithNick { | 	if b.Config.PrefixMessagesWithNick { | ||||||
| @@ -160,14 +166,51 @@ func (b *Bslack) Send(msg config.Message) error { | |||||||
| 		np.IconURL = msg.Avatar | 		np.IconURL = msg.Avatar | ||||||
| 	} | 	} | ||||||
| 	np.Attachments = append(np.Attachments, slack.Attachment{CallbackID: "matterbridge"}) | 	np.Attachments = append(np.Attachments, slack.Attachment{CallbackID: "matterbridge"}) | ||||||
| 	b.sc.PostMessage(schannel.ID, message, np) | 	np.Attachments = append(np.Attachments, b.createAttach(msg.Extra)...) | ||||||
|  |  | ||||||
| 	/* | 	// replace mentions | ||||||
| 	   newmsg := b.rtm.NewOutgoingMessage(message, schannel.ID) | 	np.LinkNames = 1 | ||||||
| 	   b.rtm.SendMessage(newmsg) |  | ||||||
| 	*/ |  | ||||||
|  |  | ||||||
| 	return nil | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		// some protocols echo deletes, but with empty ID | ||||||
|  | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		// we get a "slack <ID>", split it | ||||||
|  | 		ts := strings.Fields(msg.ID) | ||||||
|  | 		b.sc.DeleteMessage(schannel.ID, ts[1]) | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	// if we have no ID it means we're creating a new message, not updating an existing one | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		ts := strings.Fields(msg.ID) | ||||||
|  | 		b.sc.UpdateMessage(schannel.ID, ts[1], message) | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var err error | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				_, err = b.sc.UploadFile(slack.FileUploadParameters{ | ||||||
|  | 					Reader:   bytes.NewReader(*fi.Data), | ||||||
|  | 					Filename: fi.Name, | ||||||
|  | 					Channels: []string{schannel.ID}, | ||||||
|  | 				}) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("uploadfile %#v", err) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, id, err := b.sc.PostMessage(schannel.ID, message, np) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return "slack " + id, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) getAvatar(user string) string { | func (b *Bslack) getAvatar(user string) string { | ||||||
| @@ -222,80 +265,117 @@ func (b *Bslack) handleSlack() { | |||||||
| 		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name { | 		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		if message.Text == "" || message.Username == "" { | 		if (message.Text == "" || message.Username == "") && message.Raw.SubType != "message_deleted" { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		texts := strings.Split(message.Text, "\n") | 		text := message.Text | ||||||
| 		for _, text := range texts { | 		text = b.replaceURL(text) | ||||||
| 			text = b.replaceURL(text) | 		text = html.UnescapeString(text) | ||||||
| 			text = html.UnescapeString(text) | 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | 		msg := config.Message{Text: text, Username: message.Username, Channel: message.Channel, Account: b.Account, Avatar: b.getAvatar(message.Username), UserID: message.UserID, ID: "slack " + message.Raw.Timestamp, Extra: make(map[string][]interface{})} | ||||||
| 			msg := config.Message{Text: text, Username: message.Username, Channel: message.Channel, Account: b.Account, Avatar: b.getAvatar(message.Username), UserID: message.UserID} | 		if message.Raw.SubType == "me_message" { | ||||||
| 			b.Remote <- msg | 			msg.Event = config.EVENT_USER_ACTION | ||||||
| 		} | 		} | ||||||
|  | 		if message.Raw.SubType == "channel_leave" || message.Raw.SubType == "channel_join" { | ||||||
|  | 			msg.Username = "system" | ||||||
|  | 			msg.Event = config.EVENT_JOIN_LEAVE | ||||||
|  | 		} | ||||||
|  | 		// edited messages have a submessage, use this timestamp | ||||||
|  | 		if message.Raw.SubMessage != nil { | ||||||
|  | 			msg.ID = "slack " + message.Raw.SubMessage.Timestamp | ||||||
|  | 		} | ||||||
|  | 		if message.Raw.SubType == "message_deleted" { | ||||||
|  | 			msg.Text = config.EVENT_MSG_DELETE | ||||||
|  | 			msg.Event = config.EVENT_MSG_DELETE | ||||||
|  | 			msg.ID = "slack " + message.Raw.DeletedTimestamp | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||||
|  | 		if message.Raw.File != nil { | ||||||
|  | 			// limit to 1MB for now | ||||||
|  | 			if message.Raw.File.Size <= 1000000 { | ||||||
|  | 				data, err := b.downloadFile(message.Raw.File.URLPrivateDownload) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("download %s failed %#v", message.Raw.File.URLPrivateDownload, err) | ||||||
|  | 				} else { | ||||||
|  | 					msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: message.Raw.File.Name, Data: data}) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("Message is %#v", msg) | ||||||
|  | 		b.Remote <- msg | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | ||||||
| 	count := 0 |  | ||||||
| 	for msg := range b.rtm.IncomingEvents { | 	for msg := range b.rtm.IncomingEvents { | ||||||
| 		switch ev := msg.Data.(type) { | 		switch ev := msg.Data.(type) { | ||||||
| 		case *slack.MessageEvent: | 		case *slack.MessageEvent: | ||||||
| 			// ignore first message | 			flog.Debugf("Receiving from slackclient %#v", ev) | ||||||
| 			if count > 0 { | 			if len(ev.Attachments) > 0 { | ||||||
| 				flog.Debugf("Receiving from slackclient %#v", ev) | 				// skip messages we made ourselves | ||||||
| 				if len(ev.Attachments) > 0 { | 				if ev.Attachments[0].CallbackID == "matterbridge" { | ||||||
| 					// skip messages we made ourselves | 					continue | ||||||
| 					if ev.Attachments[0].CallbackID == "matterbridge" { |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 				} | 				} | ||||||
| 				if !b.Config.EditDisable && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { | 			} | ||||||
| 					flog.Debugf("SubMessage %#v", ev.SubMessage) | 			if !b.Config.EditDisable && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { | ||||||
| 					ev.User = ev.SubMessage.User | 				flog.Debugf("SubMessage %#v", ev.SubMessage) | ||||||
| 					ev.Text = ev.SubMessage.Text + b.Config.EditSuffix | 				ev.User = ev.SubMessage.User | ||||||
|  | 				ev.Text = ev.SubMessage.Text + b.Config.EditSuffix | ||||||
|  |  | ||||||
|  | 				// it seems ev.SubMessage.Edited == nil when slack unfurls | ||||||
|  | 				// do not forward these messages #266 | ||||||
|  | 				if ev.SubMessage.Edited == nil { | ||||||
|  | 					continue | ||||||
| 				} | 				} | ||||||
| 				// use our own func because rtm.GetChannelInfo doesn't work for private channels | 			} | ||||||
| 				channel, err := b.getChannelByID(ev.Channel) | 			// use our own func because rtm.GetChannelInfo doesn't work for private channels | ||||||
|  | 			channel, err := b.getChannelByID(ev.Channel) | ||||||
|  | 			if err != nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			m := &MMMessage{} | ||||||
|  | 			if ev.BotID == "" && ev.SubType != "message_deleted" { | ||||||
|  | 				user, err := b.rtm.GetUserInfo(ev.User) | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| 					continue | 					continue | ||||||
| 				} | 				} | ||||||
| 				m := &MMMessage{} | 				m.UserID = user.ID | ||||||
| 				if ev.BotID == "" { | 				m.Username = user.Name | ||||||
| 					user, err := b.rtm.GetUserInfo(ev.User) | 				if user.Profile.DisplayName != "" { | ||||||
| 					if err != nil { | 					m.Username = user.Profile.DisplayName | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 					m.UserID = user.ID |  | ||||||
| 					m.Username = user.Name |  | ||||||
| 				} | 				} | ||||||
| 				m.Channel = channel.Name |  | ||||||
| 				m.Text = ev.Text |  | ||||||
| 				if m.Text == "" { |  | ||||||
| 					for _, attach := range ev.Attachments { |  | ||||||
| 						if attach.Text != "" { |  | ||||||
| 							m.Text = attach.Text |  | ||||||
| 						} else { |  | ||||||
| 							m.Text = attach.Fallback |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				m.Raw = ev |  | ||||||
| 				m.Text = b.replaceMention(m.Text) |  | ||||||
| 				// when using webhookURL we can't check if it's our webhook or not for now |  | ||||||
| 				if ev.BotID != "" && b.Config.WebhookURL == "" { |  | ||||||
| 					bot, err := b.rtm.GetBotInfo(ev.BotID) |  | ||||||
| 					if err != nil { |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 					if bot.Name != "" { |  | ||||||
| 						m.Username = bot.Name |  | ||||||
| 						m.UserID = bot.ID |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				mchan <- m |  | ||||||
| 			} | 			} | ||||||
| 			count++ | 			m.Channel = channel.Name | ||||||
|  | 			m.Text = ev.Text | ||||||
|  | 			if m.Text == "" { | ||||||
|  | 				for _, attach := range ev.Attachments { | ||||||
|  | 					if attach.Text != "" { | ||||||
|  | 						m.Text = attach.Text | ||||||
|  | 					} else { | ||||||
|  | 						m.Text = attach.Fallback | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			m.Raw = ev | ||||||
|  | 			m.Text = b.replaceMention(m.Text) | ||||||
|  | 			m.Text = b.replaceVariable(m.Text) | ||||||
|  | 			m.Text = b.replaceChannel(m.Text) | ||||||
|  | 			// when using webhookURL we can't check if it's our webhook or not for now | ||||||
|  | 			if ev.BotID != "" && b.Config.WebhookURL == "" { | ||||||
|  | 				bot, err := b.rtm.GetBotInfo(ev.BotID) | ||||||
|  | 				if err != nil { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 				if bot.Name != "" { | ||||||
|  | 					m.Username = bot.Name | ||||||
|  | 					if ev.Username != "" { | ||||||
|  | 						m.Username = ev.Username | ||||||
|  | 					} | ||||||
|  | 					m.UserID = bot.ID | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			mchan <- m | ||||||
| 		case *slack.OutgoingErrorEvent: | 		case *slack.OutgoingErrorEvent: | ||||||
| 			flog.Debugf("%#v", ev.Error()) | 			flog.Debugf("%#v", ev.Error()) | ||||||
| 		case *slack.ChannelJoinedEvent: | 		case *slack.ChannelJoinedEvent: | ||||||
| @@ -327,6 +407,8 @@ func (b *Bslack) handleMatterHook(mchan chan *MMMessage) { | |||||||
| 		m.Username = message.UserName | 		m.Username = message.UserName | ||||||
| 		m.Text = message.Text | 		m.Text = message.Text | ||||||
| 		m.Text = b.replaceMention(m.Text) | 		m.Text = b.replaceMention(m.Text) | ||||||
|  | 		m.Text = b.replaceVariable(m.Text) | ||||||
|  | 		m.Text = b.replaceChannel(m.Text) | ||||||
| 		m.Channel = message.ChannelName | 		m.Channel = message.ChannelName | ||||||
| 		if m.Username == "slackbot" { | 		if m.Username == "slackbot" { | ||||||
| 			continue | 			continue | ||||||
| @@ -338,25 +420,91 @@ func (b *Bslack) handleMatterHook(mchan chan *MMMessage) { | |||||||
| func (b *Bslack) userName(id string) string { | func (b *Bslack) userName(id string) string { | ||||||
| 	for _, u := range b.Users { | 	for _, u := range b.Users { | ||||||
| 		if u.ID == id { | 		if u.ID == id { | ||||||
|  | 			if u.Profile.DisplayName != "" { | ||||||
|  | 				return u.Profile.DisplayName | ||||||
|  | 			} | ||||||
| 			return u.Name | 			return u.Name | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
| func (b *Bslack) replaceMention(text string) string { | func (b *Bslack) replaceMention(text string) string { | ||||||
| 	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1) | 	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1) | ||||||
| 	for _, r := range results { | 	for _, r := range results { | ||||||
| 		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1) | 		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1) | ||||||
|  |  | ||||||
| 	} | 	} | ||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
|  | func (b *Bslack) replaceChannel(text string) string { | ||||||
|  | 	results := regexp.MustCompile(`<#[a-zA-Z0-9]+\|(.+?)>`).FindAllStringSubmatch(text, -1) | ||||||
|  | 	for _, r := range results { | ||||||
|  | 		text = strings.Replace(text, r[0], "#"+r[1], -1) | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#variables | ||||||
|  | func (b *Bslack) replaceVariable(text string) string { | ||||||
|  | 	results := regexp.MustCompile(`<!([a-zA-Z0-9]+)(\|.+?)?>`).FindAllStringSubmatch(text, -1) | ||||||
|  | 	for _, r := range results { | ||||||
|  | 		text = strings.Replace(text, r[0], "@"+r[1], -1) | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_urls | ||||||
| func (b *Bslack) replaceURL(text string) string { | func (b *Bslack) replaceURL(text string) string { | ||||||
| 	results := regexp.MustCompile(`<(.*?)\|.*?>`).FindAllStringSubmatch(text, -1) | 	results := regexp.MustCompile(`<(.*?)(\|.*?)?>`).FindAllStringSubmatch(text, -1) | ||||||
| 	for _, r := range results { | 	for _, r := range results { | ||||||
| 		text = strings.Replace(text, r[0], r[1], -1) | 		text = strings.Replace(text, r[0], r[1], -1) | ||||||
| 	} | 	} | ||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) createAttach(extra map[string][]interface{}) []slack.Attachment { | ||||||
|  | 	var attachs []slack.Attachment | ||||||
|  | 	for _, v := range extra["attachments"] { | ||||||
|  | 		entry := v.(map[string]interface{}) | ||||||
|  | 		s := slack.Attachment{} | ||||||
|  | 		s.Fallback = entry["fallback"].(string) | ||||||
|  | 		s.Color = entry["color"].(string) | ||||||
|  | 		s.Pretext = entry["pretext"].(string) | ||||||
|  | 		s.AuthorName = entry["author_name"].(string) | ||||||
|  | 		s.AuthorLink = entry["author_link"].(string) | ||||||
|  | 		s.AuthorIcon = entry["author_icon"].(string) | ||||||
|  | 		s.Title = entry["title"].(string) | ||||||
|  | 		s.TitleLink = entry["title_link"].(string) | ||||||
|  | 		s.Text = entry["text"].(string) | ||||||
|  | 		s.ImageURL = entry["image_url"].(string) | ||||||
|  | 		s.ThumbURL = entry["thumb_url"].(string) | ||||||
|  | 		s.Footer = entry["footer"].(string) | ||||||
|  | 		s.FooterIcon = entry["footer_icon"].(string) | ||||||
|  | 		attachs = append(attachs, s) | ||||||
|  | 	} | ||||||
|  | 	return attachs | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) downloadFile(url string) (*[]byte, error) { | ||||||
|  | 	var buf bytes.Buffer | ||||||
|  | 	client := &http.Client{ | ||||||
|  | 		Timeout: time.Second * 5, | ||||||
|  | 	} | ||||||
|  | 	req, err := http.NewRequest("GET", url, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	req.Header.Add("Authorization", "Bearer "+b.Config.Token) | ||||||
|  | 	resp, err := client.Do(req) | ||||||
|  | 	if err != nil { | ||||||
|  | 		resp.Body.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	io.Copy(&buf, resp.Body) | ||||||
|  | 	data := buf.Bytes() | ||||||
|  | 	resp.Body.Close() | ||||||
|  | 	return &data, nil | ||||||
|  | } | ||||||
|   | |||||||
| @@ -60,8 +60,8 @@ func (b *Bsteam) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) JoinChannel(channel string) error { | func (b *Bsteam) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	id, err := steamid.NewId(channel) | 	id, err := steamid.NewId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -69,13 +69,17 @@ func (b *Bsteam) JoinChannel(channel string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) Send(msg config.Message) error { | func (b *Bsteam) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	id, err := steamid.NewId(msg.Channel) | 	id, err := steamid.NewId(msg.Channel) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) getNick(id steamid.SteamId) string { | func (b *Bsteam) getNick(id steamid.SteamId) string { | ||||||
| @@ -101,8 +105,13 @@ func (b *Bsteam) handleEvents() { | |||||||
| 		case *steam.ChatMsgEvent: | 		case *steam.ChatMsgEvent: | ||||||
| 			flog.Debugf("Receiving ChatMsgEvent: %#v", e) | 			flog.Debugf("Receiving ChatMsgEvent: %#v", e) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | 			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | ||||||
| 			// for some reason we have to remove 0x18000000000000 | 			var channel int64 | ||||||
| 			channel := int64(e.ChatRoomId) - 0x18000000000000 | 			if e.ChatRoomId == 0 { | ||||||
|  | 				channel = int64(e.ChatterId) | ||||||
|  | 			} else { | ||||||
|  | 				// for some reason we have to remove 0x18000000000000 | ||||||
|  | 				channel = int64(e.ChatRoomId) - 0x18000000000000 | ||||||
|  | 			} | ||||||
| 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | ||||||
| 			b.Remote <- msg | 			b.Remote <- msg | ||||||
| 		case *steam.PersonaStateEvent: | 		case *steam.PersonaStateEvent: | ||||||
|   | |||||||
| @@ -1,9 +1,11 @@ | |||||||
| package btelegram | package btelegram | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"regexp" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||||
| ) | ) | ||||||
| @@ -53,34 +55,90 @@ func (b *Btelegram) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) JoinChannel(channel string) error { | func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Send(msg config.Message) error { | func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if b.Config.MessageFormat == "HTML" { | 	if b.Config.MessageFormat == "HTML" { | ||||||
| 		msg.Text = makeHTML(msg.Text) | 		msg.Text = makeHTML(msg.Text) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		msgid, err := strconv.Atoi(msg.ID) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		_, err = b.c.DeleteMessage(tgbotapi.DeleteMessageConfig{ChatID: chatid, MessageID: msgid}) | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// edit the message if we have a msg ID | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		msgid, err := strconv.Atoi(msg.ID) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text) | ||||||
|  | 		_, err = b.c.Send(m) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var c tgbotapi.Chattable | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				file := tgbotapi.FileBytes{fi.Name, *fi.Data} | ||||||
|  | 				re := regexp.MustCompile(".(jpg|png)$") | ||||||
|  | 				if re.MatchString(fi.Name) { | ||||||
|  | 					c = tgbotapi.NewPhotoUpload(chatid, file) | ||||||
|  | 				} else { | ||||||
|  | 					c = tgbotapi.NewDocumentUpload(chatid, file) | ||||||
|  | 				} | ||||||
|  | 				_, err := b.c.Send(c) | ||||||
|  | 				if err != nil { | ||||||
|  | 					log.Errorf("file upload failed: %#v", err) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text) | 	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text) | ||||||
| 	if b.Config.MessageFormat == "HTML" { | 	if b.Config.MessageFormat == "HTML" { | ||||||
| 		m.ParseMode = tgbotapi.ModeHTML | 		m.ParseMode = tgbotapi.ModeHTML | ||||||
| 	} | 	} | ||||||
| 	_, err = b.c.Send(m) | 	res, err := b.c.Send(m) | ||||||
| 	return err | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return strconv.Itoa(res.MessageID), nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||||
| 	for update := range updates { | 	for update := range updates { | ||||||
|  | 		flog.Debugf("Receiving from telegram: %#v", update.Message) | ||||||
| 		var message *tgbotapi.Message | 		var message *tgbotapi.Message | ||||||
| 		username := "" | 		username := "" | ||||||
| 		channel := "" | 		channel := "" | ||||||
| 		text := "" | 		text := "" | ||||||
|  |  | ||||||
|  | 		fmsg := config.Message{Extra: make(map[string][]interface{})} | ||||||
|  |  | ||||||
| 		// handle channels | 		// handle channels | ||||||
| 		if update.ChannelPost != nil { | 		if update.ChannelPost != nil { | ||||||
| 			message = update.ChannelPost | 			message = update.ChannelPost | ||||||
| @@ -114,23 +172,45 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | |||||||
| 		if username == "" { | 		if username == "" { | ||||||
| 			username = "unknown" | 			username = "unknown" | ||||||
| 		} | 		} | ||||||
| 		if message.Sticker != nil && b.Config.UseInsecureURL { | 		if message.Sticker != nil { | ||||||
| 			text = text + " " + b.getFileDirectURL(message.Sticker.FileID) | 			b.handleDownload(message.Sticker, &fmsg) | ||||||
| 		} | 		} | ||||||
| 		if message.Video != nil && b.Config.UseInsecureURL { | 		if message.Video != nil { | ||||||
| 			text = text + " " + b.getFileDirectURL(message.Video.FileID) | 			b.handleDownload(message.Video, &fmsg) | ||||||
| 		} | 		} | ||||||
| 		if message.Photo != nil && b.Config.UseInsecureURL { | 		if message.Photo != nil && b.Config.UseInsecureURL { | ||||||
| 			photos := *message.Photo | 			b.handleDownload(message.Photo, &fmsg) | ||||||
| 			// last photo is the biggest |  | ||||||
| 			text = text + " " + b.getFileDirectURL(photos[len(photos)-1].FileID) |  | ||||||
| 		} | 		} | ||||||
| 		if message.Document != nil && b.Config.UseInsecureURL { | 		if message.Document != nil && b.Config.UseInsecureURL { | ||||||
|  | 			b.handleDownload(message.Sticker, &fmsg) | ||||||
| 			text = text + " " + message.Document.FileName + " : " + b.getFileDirectURL(message.Document.FileID) | 			text = text + " " + message.Document.FileName + " : " + b.getFileDirectURL(message.Document.FileID) | ||||||
| 		} | 		} | ||||||
| 		if text != "" { |  | ||||||
|  | 		// quote the previous message | ||||||
|  | 		if message.ReplyToMessage != nil { | ||||||
|  | 			usernameReply := "" | ||||||
|  | 			if message.ReplyToMessage.From != nil { | ||||||
|  | 				if b.Config.UseFirstName { | ||||||
|  | 					usernameReply = message.ReplyToMessage.From.FirstName | ||||||
|  | 				} | ||||||
|  | 				if usernameReply == "" { | ||||||
|  | 					usernameReply = message.ReplyToMessage.From.UserName | ||||||
|  | 					if usernameReply == "" { | ||||||
|  | 						usernameReply = message.ReplyToMessage.From.FirstName | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if usernameReply == "" { | ||||||
|  | 				usernameReply = "unknown" | ||||||
|  | 			} | ||||||
|  | 			text = text + " (re @" + usernameReply + ":" + message.ReplyToMessage.Text + ")" | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if text != "" || len(fmsg.Extra) > 0 { | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account) | 			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account) | ||||||
| 			b.Remote <- config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID)} | 			msg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID), ID: strconv.Itoa(message.MessageID)} | ||||||
|  | 			flog.Debugf("Message is %#v", msg) | ||||||
|  | 			b.Remote <- msg | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -142,3 +222,47 @@ func (b *Btelegram) getFileDirectURL(id string) string { | |||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Btelegram) handleDownload(file interface{}, msg *config.Message) { | ||||||
|  | 	size := 0 | ||||||
|  | 	url := "" | ||||||
|  | 	name := "" | ||||||
|  | 	text := "" | ||||||
|  | 	switch v := file.(type) { | ||||||
|  | 	case *tgbotapi.Sticker: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		name = "sticker" | ||||||
|  | 		text = " " + url | ||||||
|  | 	case *tgbotapi.Video: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		name = "video" | ||||||
|  | 		text = " " + url | ||||||
|  | 	case *[]tgbotapi.PhotoSize: | ||||||
|  | 		photos := *v | ||||||
|  | 		size = photos[len(photos)-1].FileSize | ||||||
|  | 		url = b.getFileDirectURL(photos[len(photos)-1].FileID) | ||||||
|  | 		name = "photo" | ||||||
|  | 		text = " " + url | ||||||
|  | 	case *tgbotapi.Document: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		name = v.FileName | ||||||
|  | 		text = " " + v.FileName + " : " + url | ||||||
|  | 	} | ||||||
|  | 	if b.Config.UseInsecureURL { | ||||||
|  | 		msg.Text = text | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||||
|  | 	// limit to 1MB for now | ||||||
|  | 	if size <= 1000000 { | ||||||
|  | 		data, err := helper.DownloadFile(url) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("download %s failed %#v", url, err) | ||||||
|  | 		} else { | ||||||
|  | 			msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -74,15 +74,19 @@ func (b *Bxmpp) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) JoinChannel(channel string) error { | func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.xc.JoinMUCNoHistory(channel+"@"+b.Config.Muc, b.Config.Nick) | 	b.xc.JoinMUCNoHistory(channel.Name+"@"+b.Config.Muc, b.Config.Nick) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) Send(msg config.Message) error { | func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text}) | 	b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text}) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | ||||||
| @@ -133,6 +137,7 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) handleXmpp() error { | func (b *Bxmpp) handleXmpp() error { | ||||||
|  | 	var ok bool | ||||||
| 	done := b.xmppKeepAlive() | 	done := b.xmppKeepAlive() | ||||||
| 	defer close(done) | 	defer close(done) | ||||||
| 	nodelay := time.Time{} | 	nodelay := time.Time{} | ||||||
| @@ -154,8 +159,13 @@ func (b *Bxmpp) handleXmpp() error { | |||||||
| 					nick = s[1] | 					nick = s[1] | ||||||
| 				} | 				} | ||||||
| 				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" { | 				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" { | ||||||
|  | 					rmsg := config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote} | ||||||
|  | 					rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||||
|  | 					if ok { | ||||||
|  | 						rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 					} | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account) | 					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account) | ||||||
| 					b.Remote <- config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote} | 					b.Remote <- rmsg | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		case xmpp.Presence: | 		case xmpp.Presence: | ||||||
| @@ -163,3 +173,10 @@ func (b *Bxmpp) handleXmpp() error { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bxmpp) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "/me ") { | ||||||
|  | 		return strings.Replace(text, "/me ", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										105
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						
									
										105
									
								
								changelog.md
									
									
									
									
									
								
							| @@ -1,3 +1,108 @@ | |||||||
|  | # v1.4.0 | ||||||
|  | ## Breaking changes | ||||||
|  | * general: `[general]` settings don't override the specific bridge settings | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * irc: Replace sorcix/irc and go-ircevent with girc, this should be give better reconnects | ||||||
|  | * steam: Add support for bridging to individual steam chats. (steam) (#294) | ||||||
|  | * telegram: Download files from telegram and reupload to supported bridges (telegram). #278 | ||||||
|  | * slack: Add support to upload files to slack, from bridges with private urls like slack/mattermost/telegram. (slack) | ||||||
|  | * discord: Add support to upload files to discord, from bridges with private urls like slack/mattermost/telegram. (discord) | ||||||
|  | * general: Add systemd service file (#291) | ||||||
|  | * general: Add support for DEBUG=1 envvar to enable debug. Closes #283 | ||||||
|  | * general: Add StripNick option, only allow alphanumerical nicks. Closes #285 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * gitter: Use room.URI instead of room.Name. (gitter) (#293) | ||||||
|  | * slack: Allow slack messages with variables (eg. @here) to be formatted correctly. (slack) (#288) | ||||||
|  | * slack: Resolve slack channel to human-readable name. (slack) (#282) | ||||||
|  | * slack: Use DisplayName instead of deprecated username (slack). Closes #276 | ||||||
|  | * slack: Allowed Slack bridge to extract simpler link format. (#287) | ||||||
|  | * irc: Strip irc colors correct, strip also ctrl chars (irc) | ||||||
|  |  | ||||||
|  | # v1.3.1 | ||||||
|  | ## New features | ||||||
|  | * Support mattermost 4.3.0 and every other 4.x as api4 should be stable (mattermost) | ||||||
|  | ## Bugfix | ||||||
|  | * Use bot username if specified (slack). Closes #273 | ||||||
|  |  | ||||||
|  | # v1.3.0 | ||||||
|  | ## New features | ||||||
|  | * Relay slack_attachments from mattermost to slack (slack). Closes #260 | ||||||
|  | * Add support for quoting previous message when replying (telegram). #237 | ||||||
|  | * Add support for Quakenet auth (irc). Closes #263 | ||||||
|  | * Download files (max size 1MB) from slack and reupload to mattermost (slack/mattermost). Closes #255 | ||||||
|  |  | ||||||
|  | ## Enhancements | ||||||
|  | * Backoff for 60 seconds when reconnecting too fast (irc) #267 | ||||||
|  | * Use override username if specified (mattermost). #260 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * Try to not forward slack unfurls. Closes #266 | ||||||
|  |  | ||||||
|  | # v1.2.0 | ||||||
|  | ## Breaking changes | ||||||
|  | * If you're running a discord bridge, update to this release before 16 october otherwise | ||||||
|  | it will stop working. (see https://discordapp.com/developers/docs/reference) | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * general: Add delete support. (actually delete the messages on bridges that support it) | ||||||
|  |     (mattermost,discord,gitter,slack,telegram) | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * Do not break messages on newline (slack). Closes #258  | ||||||
|  | * Update telegram library | ||||||
|  | * Update discord library (supports v6 API now). Old API is deprecated on 16 October | ||||||
|  |  | ||||||
|  | # v1.1.2 | ||||||
|  | ## New features | ||||||
|  | * general: also build darwin binaries | ||||||
|  | * mattermost: add support for mattermost 4.2.x | ||||||
|  |  | ||||||
|  | ## Bugfix  | ||||||
|  | * mattermost: Send images when text is empty regression. (mattermost). Closes #254 | ||||||
|  | * slack: also send the first messsage after connect. #252 | ||||||
|  |  | ||||||
|  | # v1.1.1 | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: fix public links | ||||||
|  |  | ||||||
|  | # v1.1.0 | ||||||
|  | ## New features | ||||||
|  | * general: Add better editing support. (actually edit the messages on bridges that support it) | ||||||
|  | 	(mattermost,discord,gitter,slack,telegram) | ||||||
|  | * mattermost: use API v4 (removes support for mattermost < 3.8) | ||||||
|  | * mattermost: add support for personal access tokens (since mattermost 4.1) | ||||||
|  | 	Use ```Token="yourtoken"``` in mattermost config | ||||||
|  | 	See https://docs.mattermost.com/developer/personal-access-tokens.html for more info | ||||||
|  | * matrix: Relay notices (matrix). Closes #243 | ||||||
|  | * irc: Add a charset option. Closes #247 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Handle leave/join events (slack). Closes #246 | ||||||
|  | * slack: Replace mentions from other bridges. (slack). Closes #233 | ||||||
|  | * gitter: remove ZWSP after messages | ||||||
|  |  | ||||||
|  | # v1.0.1 | ||||||
|  | ## New features | ||||||
|  | * mattermost: add support for mattermost 4.1.x | ||||||
|  | * discord: allow a webhookURL per channel #239 | ||||||
|  |  | ||||||
|  | # v1.0.0 | ||||||
|  | ## New features | ||||||
|  | * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | ||||||
|  | * discord: Shows the username instead of the server nickname #234 | ||||||
|  |  | ||||||
|  | # v1.0.0-rc1 | ||||||
|  | ## New features | ||||||
|  | * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * general: Handle same account in multiple gateways better | ||||||
|  | * mattermost: ignore edited messages with reactions | ||||||
|  | * mattermost: Fix double posting of edited messages by using lru cache | ||||||
|  | * irc: update vendor | ||||||
|  |  | ||||||
| # v0.16.3 | # v0.16.3 | ||||||
| ## Bugfix | ## Bugfix | ||||||
| * general: Fix in/out logic. Closes #224  | * general: Fix in/out logic. Closes #224  | ||||||
|   | |||||||
| @@ -1,10 +1,11 @@ | |||||||
| #!/bin/bash | #!/bin/bash | ||||||
| go version |grep go1.8 || exit | go version |grep go1.9 || exit | ||||||
| VERSION=$(git describe --tags) | VERSION=$(git describe --tags) | ||||||
| mkdir ci/binaries | mkdir ci/binaries | ||||||
| GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-win64.exe | GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-windows-amd64.exe | ||||||
| GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux64 | GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-amd64 | ||||||
| GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | ||||||
|  | GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-darwin-amd64 | ||||||
| cd ci | cd ci | ||||||
| cat > deploy.json <<EOF | cat > deploy.json <<EOF | ||||||
| { | { | ||||||
|   | |||||||
							
								
								
									
										11
									
								
								contrib/matterbridge.service
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								contrib/matterbridge.service
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | |||||||
|  | [Unit] | ||||||
|  | Description=matterbridge | ||||||
|  | After=network.target | ||||||
|  |  | ||||||
|  | [Service] | ||||||
|  | ExecStart=/usr/bin/matterbridge -conf /etc/matterbridge/bridge.toml | ||||||
|  | User=matterbridge | ||||||
|  | Group=matterbridge | ||||||
|  |  | ||||||
|  | [Install] | ||||||
|  | WantedBy=multi-user.target | ||||||
| @@ -5,8 +5,9 @@ import ( | |||||||
| 	"github.com/42wim/matterbridge/bridge" | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/peterhellberg/emojilib" |  | ||||||
| 	//	"github.com/davecgh/go-spew/spew" | 	//	"github.com/davecgh/go-spew/spew" | ||||||
|  | 	"github.com/hashicorp/golang-lru" | ||||||
|  | 	"github.com/peterhellberg/emojilib" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -14,62 +15,41 @@ import ( | |||||||
|  |  | ||||||
| type Gateway struct { | type Gateway struct { | ||||||
| 	*config.Config | 	*config.Config | ||||||
| 	MyConfig        *config.Gateway | 	Router         *Router | ||||||
| 	Bridges         map[string]*bridge.Bridge | 	MyConfig       *config.Gateway | ||||||
| 	Channels        map[string]*config.ChannelInfo | 	Bridges        map[string]*bridge.Bridge | ||||||
| 	ChannelOptions  map[string]config.ChannelOptions | 	Channels       map[string]*config.ChannelInfo | ||||||
| 	Names           map[string]bool | 	ChannelOptions map[string]config.ChannelOptions | ||||||
| 	Name            string | 	Message        chan config.Message | ||||||
| 	Message         chan config.Message | 	Name           string | ||||||
| 	DestChannelFunc func(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo | 	Messages       *lru.Cache | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg *config.Config) *Gateway { | type BrMsgID struct { | ||||||
| 	gw := &Gateway{} | 	br *bridge.Bridge | ||||||
| 	gw.Config = cfg | 	ID string | ||||||
| 	gw.Channels = make(map[string]*config.ChannelInfo) | } | ||||||
| 	gw.Message = make(chan config.Message) |  | ||||||
| 	gw.Bridges = make(map[string]*bridge.Bridge) | func New(cfg config.Gateway, r *Router) *Gateway { | ||||||
| 	gw.Names = make(map[string]bool) | 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | ||||||
| 	gw.DestChannelFunc = gw.getDestChannel | 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | ||||||
|  | 	cache, _ := lru.New(5000) | ||||||
|  | 	gw.Messages = cache | ||||||
|  | 	gw.AddConfig(&cfg) | ||||||
| 	return gw | 	return gw | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||||
| 	for _, br := range gw.Bridges { | 	br := gw.Router.getBridge(cfg.Account) | ||||||
| 		if br.Account == cfg.Account { | 	if br == nil { | ||||||
| 			gw.mapChannelsToBridge(br) | 		br = bridge.New(gw.Config, cfg, gw.Message) | ||||||
| 			err := br.JoinChannels() |  | ||||||
| 			if err != nil { |  | ||||||
| 				return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) |  | ||||||
| 			} |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| 	log.Infof("Starting bridge: %s ", cfg.Account) |  | ||||||
| 	br := bridge.New(gw.Config, cfg, gw.Message) |  | ||||||
| 	gw.mapChannelsToBridge(br) | 	gw.mapChannelsToBridge(br) | ||||||
| 	gw.Bridges[cfg.Account] = br | 	gw.Bridges[cfg.Account] = br | ||||||
| 	err := br.Connect() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) |  | ||||||
| 	} |  | ||||||
| 	err = br.JoinChannels() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) |  | ||||||
| 	} |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | ||||||
| 	if gw.Names[cfg.Name] { |  | ||||||
| 		return fmt.Errorf("Gateway with name %s already exists", cfg.Name) |  | ||||||
| 	} |  | ||||||
| 	if cfg.Name == "" { |  | ||||||
| 		return fmt.Errorf("%s", "Gateway without name found") |  | ||||||
| 	} |  | ||||||
| 	log.Infof("Starting gateway: %s", cfg.Name) |  | ||||||
| 	gw.Names[cfg.Name] = true |  | ||||||
| 	gw.Name = cfg.Name | 	gw.Name = cfg.Name | ||||||
| 	gw.MyConfig = cfg | 	gw.MyConfig = cfg | ||||||
| 	gw.mapChannels() | 	gw.mapChannels() | ||||||
| @@ -90,39 +70,6 @@ func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) Start() error { |  | ||||||
| 	go gw.handleReceive() |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (gw *Gateway) handleReceive() { |  | ||||||
| 	for msg := range gw.Message { |  | ||||||
| 		if msg.Event == config.EVENT_FAILURE { |  | ||||||
| 			for _, br := range gw.Bridges { |  | ||||||
| 				if msg.Account == br.Account { |  | ||||||
| 					go gw.reconnectBridge(br) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if msg.Event == config.EVENT_REJOIN_CHANNELS { |  | ||||||
| 			for _, br := range gw.Bridges { |  | ||||||
| 				if msg.Account == br.Account { |  | ||||||
| 					br.Joined = make(map[string]bool) |  | ||||||
| 					br.JoinChannels() |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		if !gw.ignoreMessage(&msg) { |  | ||||||
| 			msg.Timestamp = time.Now() |  | ||||||
| 			gw.modifyMessage(&msg) |  | ||||||
| 			for _, br := range gw.Bridges { |  | ||||||
| 				gw.handleMessage(msg, br) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | ||||||
| 	br.Disconnect() | 	br.Disconnect() | ||||||
| 	time.Sleep(time.Second * 5) | 	time.Sleep(time.Second * 5) | ||||||
| @@ -146,8 +93,7 @@ func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | |||||||
| 		ID := br.Channel + br.Account | 		ID := br.Channel + br.Account | ||||||
| 		if _, ok := gw.Channels[ID]; !ok { | 		if _, ok := gw.Channels[ID]; !ok { | ||||||
| 			channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account, | 			channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account, | ||||||
| 				GID: make(map[string]bool), SameChannel: make(map[string]bool)} | 				SameChannel: make(map[string]bool)} | ||||||
| 			channel.GID[gw.Name] = true |  | ||||||
| 			channel.SameChannel[gw.Name] = br.SameChannel | 			channel.SameChannel[gw.Name] = br.SameChannel | ||||||
| 			gw.Channels[channel.ID] = channel | 			gw.Channels[channel.ID] = channel | ||||||
| 		} else { | 		} else { | ||||||
| @@ -156,10 +102,10 @@ func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | |||||||
| 				gw.Channels[ID].Direction = "inout" | 				gw.Channels[ID].Direction = "inout" | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		gw.Channels[ID].GID[gw.Name] = true |  | ||||||
| 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) mapChannels() error { | func (gw *Gateway) mapChannels() error { | ||||||
| 	gw.mapChannelConfig(gw.MyConfig.In, "in") | 	gw.mapChannelConfig(gw.MyConfig.In, "in") | ||||||
| 	gw.mapChannelConfig(gw.MyConfig.Out, "out") | 	gw.mapChannelConfig(gw.MyConfig.Out, "out") | ||||||
| @@ -184,8 +130,6 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con | |||||||
| 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		// add gateway to message |  | ||||||
| 		gw.validGatewayDest(msg, channel) |  | ||||||
|  |  | ||||||
| 		// do samechannelgateway logic | 		// do samechannelgateway logic | ||||||
| 		if channel.SameChannel[msg.Gateway] { | 		if channel.SameChannel[msg.Gateway] { | ||||||
| @@ -201,19 +145,32 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con | |||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) { | func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||||
|  | 	var brMsgIDs []*BrMsgID | ||||||
|  |  | ||||||
|  | 	// TODO refactor | ||||||
|  | 	// only slack now, check will have to be done in the different bridges. | ||||||
|  | 	// we need to check if we can't use fallback or text in other bridges | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		if dest.Protocol != "slack" { | ||||||
|  | 			if msg.Text == "" { | ||||||
|  | 				return brMsgIDs | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
| 	// only relay join/part when configged | 	// only relay join/part when configged | ||||||
| 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart { | 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart { | ||||||
| 		return | 		return brMsgIDs | ||||||
| 	} | 	} | ||||||
| 	// broadcast to every out channel (irc QUIT) | 	// broadcast to every out channel (irc QUIT) | ||||||
| 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | ||||||
| 		log.Debug("empty channel") | 		log.Debug("empty channel") | ||||||
| 		return | 		return brMsgIDs | ||||||
| 	} | 	} | ||||||
| 	originchannel := msg.Channel | 	originchannel := msg.Channel | ||||||
| 	origmsg := msg | 	origmsg := msg | ||||||
| 	for _, channel := range gw.DestChannelFunc(&msg, *dest) { | 	channels := gw.getDestChannel(&msg, *dest) | ||||||
|  | 	for _, channel := range channels { | ||||||
| 		// do not send to ourself | 		// do not send to ourself | ||||||
| 		if channel.ID == getChannelID(origmsg) { | 		if channel.ID == getChannelID(origmsg) { | ||||||
| 			continue | 			continue | ||||||
| @@ -222,19 +179,41 @@ func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) { | |||||||
| 		msg.Channel = channel.Name | 		msg.Channel = channel.Name | ||||||
| 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||||
| 		msg.Username = gw.modifyUsername(origmsg, dest) | 		msg.Username = gw.modifyUsername(origmsg, dest) | ||||||
|  | 		msg.ID = "" | ||||||
|  | 		if res, ok := gw.Messages.Get(origmsg.ID); ok { | ||||||
|  | 			IDs := res.([]*BrMsgID) | ||||||
|  | 			for _, id := range IDs { | ||||||
|  | 				if dest.Protocol == id.br.Protocol { | ||||||
|  | 					msg.ID = id.ID | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 		// for api we need originchannel as channel | 		// for api we need originchannel as channel | ||||||
| 		if dest.Protocol == "api" { | 		if dest.Protocol == "api" { | ||||||
| 			msg.Channel = originchannel | 			msg.Channel = originchannel | ||||||
| 		} | 		} | ||||||
| 		err := dest.Send(msg) | 		mID, err := dest.Send(msg) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			fmt.Println(err) | 			fmt.Println(err) | ||||||
| 		} | 		} | ||||||
|  | 		// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||||
|  | 		if mID != "" { | ||||||
|  | 			brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID}) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  | 	return brMsgIDs | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||||
|  | 	// if we don't have the bridge, ignore it | ||||||
|  | 	if _, ok := gw.Bridges[msg.Account]; !ok { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
| 	if msg.Text == "" { | 	if msg.Text == "" { | ||||||
|  | 		// we have an attachment | ||||||
|  | 		if msg.Extra != nil && msg.Extra["attachments"] != nil { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
| 		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | 		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| @@ -264,9 +243,13 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | |||||||
| func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string { | func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string { | ||||||
| 	br := gw.Bridges[msg.Account] | 	br := gw.Bridges[msg.Account] | ||||||
| 	msg.Protocol = br.Protocol | 	msg.Protocol = br.Protocol | ||||||
| 	nick := gw.Config.General.RemoteNickFormat | 	if gw.Config.General.StripNick || dest.Config.StripNick { | ||||||
|  | 		re := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||||
|  | 		msg.Username = re.ReplaceAllString(msg.Username, "") | ||||||
|  | 	} | ||||||
|  | 	nick := dest.Config.RemoteNickFormat | ||||||
| 	if nick == "" { | 	if nick == "" { | ||||||
| 		nick = dest.Config.RemoteNickFormat | 		nick = gw.Config.General.RemoteNickFormat | ||||||
| 	} | 	} | ||||||
| 	if len(msg.Username) > 0 { | 	if len(msg.Username) > 0 { | ||||||
| 		// fix utf-8 issue #193 | 		// fix utf-8 issue #193 | ||||||
| @@ -301,6 +284,7 @@ func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string | |||||||
| func (gw *Gateway) modifyMessage(msg *config.Message) { | func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||||
| 	// replace :emoji: to unicode | 	// replace :emoji: to unicode | ||||||
| 	msg.Text = emojilib.Replace(msg.Text) | 	msg.Text = emojilib.Replace(msg.Text) | ||||||
|  | 	msg.Gateway = gw.Name | ||||||
| } | } | ||||||
|  |  | ||||||
| func getChannelID(msg config.Message) string { | func getChannelID(msg config.Message) string { | ||||||
| @@ -308,35 +292,7 @@ func getChannelID(msg config.Message) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | ||||||
| 	GIDmap := gw.Channels[getChannelID(*msg)].GID | 	return msg.Gateway == gw.Name | ||||||
|  |  | ||||||
| 	// gateway is specified in message (probably from api) |  | ||||||
| 	if msg.Gateway != "" { |  | ||||||
| 		return channel.GID[msg.Gateway] |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// check if we are running a samechannelgateway. |  | ||||||
| 	// if it is and the channel name matches it's ok, otherwise we shouldn't use this channel. |  | ||||||
| 	for k := range GIDmap { |  | ||||||
| 		if channel.SameChannel[k] { |  | ||||||
| 			if msg.Channel == channel.Name { |  | ||||||
| 				// add the gateway to our message |  | ||||||
| 				msg.Gateway = k |  | ||||||
| 				return true |  | ||||||
| 			} else { |  | ||||||
| 				return false |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	// check if we are in the correct gateway |  | ||||||
| 	for k := range GIDmap { |  | ||||||
| 		if channel.GID[k] { |  | ||||||
| 			// add the gateway to our message |  | ||||||
| 			msg.Gateway = k |  | ||||||
| 			return true |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func isApi(account string) bool { | func isApi(account string) bool { | ||||||
|   | |||||||
							
								
								
									
										288
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										288
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,288 @@ | |||||||
|  | package gateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/BurntSushi/toml" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  | 	"strconv" | ||||||
|  |  | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var testconfig = ` | ||||||
|  | [irc.freenode] | ||||||
|  | [mattermost.test] | ||||||
|  | [gitter.42wim] | ||||||
|  | [discord.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge1" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting" | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |     #channel="matterbridge/Lobby" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general" | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="slack.test" | ||||||
|  |     channel="testing" | ||||||
|  | 	` | ||||||
|  |  | ||||||
|  | var testconfig2 = ` | ||||||
|  | [irc.freenode] | ||||||
|  | [mattermost.test] | ||||||
|  | [gitter.42wim] | ||||||
|  | [discord.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge1" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting" | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general" | ||||||
|  |      | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="slack.test" | ||||||
|  |     channel="testing" | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge2" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting2" | ||||||
|  |      | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general2" | ||||||
|  | 	` | ||||||
|  | var testconfig3 = ` | ||||||
|  | [irc.zzz] | ||||||
|  | [telegram.zzz] | ||||||
|  | [slack.zzz] | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main"		 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="-1111111111111" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="slack.zzz" | ||||||
|  |     channel="irc"	 | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="announcements" | ||||||
|  | enable=true | ||||||
|  | 	 | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="-2222222222222"	 | ||||||
|  | 	 | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main"		 | ||||||
|  | 	 | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-help"	 | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--333333333333"	 | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="slack.zzz" | ||||||
|  |     channel="general"		 | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge2" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-help"	 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--444444444444"	 | ||||||
|  |  | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge3" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-telegram"	 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--333333333333" | ||||||
|  | ` | ||||||
|  |  | ||||||
|  | func maketestRouter(input string) *Router { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(input, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	r, err := NewRouter(cfg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	return r | ||||||
|  | } | ||||||
|  | func TestNewRouter(t *testing.T) { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	r, err := NewRouter(cfg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	assert.Equal(t, 1, len(r.Gateways)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
|  |  | ||||||
|  | 	r = maketestRouter(testconfig2) | ||||||
|  | 	assert.Equal(t, 2, len(r.Gateways)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | ||||||
|  | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
|  | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "out", | ||||||
|  | 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||||
|  | 		SameChannel: map[string]bool{"bridge2": false}}, | ||||||
|  | 		r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "in", | ||||||
|  | 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||||
|  | 		SameChannel: map[string]bool{"bridge1": false}}, | ||||||
|  | 		r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "general", Direction: "inout", | ||||||
|  | 		ID: "generaldiscord.test", Account: "discord.test", | ||||||
|  | 		SameChannel: map[string]bool{"bridge1": false}}, | ||||||
|  | 		r.Gateways["bridge1"].Channels["generaldiscord.test"]) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestGetDestChannel(t *testing.T) { | ||||||
|  | 	r := maketestRouter(testconfig2) | ||||||
|  | 	msg := &config.Message{Text: "test", Channel: "general", Account: "discord.test", Gateway: "bridge1", Protocol: "discord", Username: "test"} | ||||||
|  | 	for _, br := range r.Gateways["bridge1"].Bridges { | ||||||
|  | 		switch br.Account { | ||||||
|  | 		case "discord.test": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "discord.test", Direction: "inout", ID: "generaldiscord.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||||
|  | 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "slack.test": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo{{Name: "testing", Account: "slack.test", Direction: "out", ID: "testingslack.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||||
|  | 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "gitter.42wim": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "irc.freenode": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestGetDestChannelAdvanced(t *testing.T) { | ||||||
|  | 	r := maketestRouter(testconfig3) | ||||||
|  | 	var msgs []*config.Message | ||||||
|  | 	i := 0 | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		for _, channel := range gw.Channels { | ||||||
|  | 			msgs = append(msgs, &config.Message{Text: "text" + strconv.Itoa(i), Channel: channel.Name, Account: channel.Account, Gateway: gw.Name, Username: "user" + strconv.Itoa(i)}) | ||||||
|  | 			i++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	hits := make(map[string]int) | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		for _, br := range gw.Bridges { | ||||||
|  | 			for _, msg := range msgs { | ||||||
|  | 				channels := gw.getDestChannel(msg, *br) | ||||||
|  | 				if gw.Name != msg.Gateway { | ||||||
|  | 					assert.Equal(t, []config.ChannelInfo(nil), channels) | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 				switch gw.Name { | ||||||
|  | 				case "bridge": | ||||||
|  | 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz" || msg.Account == "slack.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "inout", ID: "#mainirc.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "-1111111111111", Account: "telegram.zzz", Direction: "inout", ID: "-1111111111111telegram.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "slack.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "irc", Account: "slack.zzz", Direction: "inout", ID: "ircslack.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "bridge2": | ||||||
|  | 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-help", Account: "irc.zzz", Direction: "inout", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "--444444444444", Account: "telegram.zzz", Direction: "inout", ID: "--444444444444telegram.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "bridge3": | ||||||
|  | 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-telegram", Account: "irc.zzz", Direction: "inout", ID: "#main-telegramirc.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "inout", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "announcements": | ||||||
|  | 					if msg.Channel != "-2222222222222" && msg.Account != "telegram" { | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo(nil), channels) | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					hits[gw.Name]++ | ||||||
|  | 					switch br.Account { | ||||||
|  | 					case "irc.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "out", ID: "#mainirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}, {Name: "#main-help", Account: "irc.zzz", Direction: "out", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					case "slack.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "slack.zzz", Direction: "out", ID: "generalslack.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					case "telegram.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "out", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits) | ||||||
|  | } | ||||||
							
								
								
									
										112
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										112
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,112 @@ | |||||||
|  | package gateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/gateway/samechannel" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	//	"github.com/davecgh/go-spew/spew" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Router struct { | ||||||
|  | 	Gateways map[string]*Gateway | ||||||
|  | 	Message  chan config.Message | ||||||
|  | 	*config.Config | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewRouter(cfg *config.Config) (*Router, error) { | ||||||
|  | 	r := &Router{} | ||||||
|  | 	r.Config = cfg | ||||||
|  | 	r.Message = make(chan config.Message) | ||||||
|  | 	r.Gateways = make(map[string]*Gateway) | ||||||
|  | 	sgw := samechannelgateway.New(cfg) | ||||||
|  | 	gwconfigs := sgw.GetConfig() | ||||||
|  |  | ||||||
|  | 	for _, entry := range append(gwconfigs, cfg.Gateway...) { | ||||||
|  | 		if !entry.Enable { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if entry.Name == "" { | ||||||
|  | 			return nil, fmt.Errorf("%s", "Gateway without name found") | ||||||
|  | 		} | ||||||
|  | 		if _, ok := r.Gateways[entry.Name]; ok { | ||||||
|  | 			return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name) | ||||||
|  | 		} | ||||||
|  | 		r.Gateways[entry.Name] = New(entry, r) | ||||||
|  | 	} | ||||||
|  | 	return r, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) Start() error { | ||||||
|  | 	m := make(map[string]*bridge.Bridge) | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		for _, br := range gw.Bridges { | ||||||
|  | 			m[br.Account] = br | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, br := range m { | ||||||
|  | 		log.Infof("Starting bridge: %s ", br.Account) | ||||||
|  | 		err := br.Connect() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||||
|  | 		} | ||||||
|  | 		err = br.JoinChannels() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	go r.handleReceive() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) getBridge(account string) *bridge.Bridge { | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		if br, ok := gw.Bridges[account]; ok { | ||||||
|  | 			return br | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) handleReceive() { | ||||||
|  | 	for msg := range r.Message { | ||||||
|  | 		if msg.Event == config.EVENT_FAILURE { | ||||||
|  | 		Loop: | ||||||
|  | 			for _, gw := range r.Gateways { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						go gw.reconnectBridge(br) | ||||||
|  | 						break Loop | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if msg.Event == config.EVENT_REJOIN_CHANNELS { | ||||||
|  | 			for _, gw := range r.Gateways { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						br.Joined = make(map[string]bool) | ||||||
|  | 						br.JoinChannels() | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for _, gw := range r.Gateways { | ||||||
|  | 			// record all the message ID's of the different bridges | ||||||
|  | 			var msgIDs []*BrMsgID | ||||||
|  | 			if !gw.ignoreMessage(&msg) { | ||||||
|  | 				msg.Timestamp = time.Now() | ||||||
|  | 				gw.modifyMessage(&msg) | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					msgIDs = append(msgIDs, gw.handleMessage(msg, br)...) | ||||||
|  | 				} | ||||||
|  | 				// only add the message ID if it doesn't already exists | ||||||
|  | 				if _, ok := gw.Messages.Get(msg.ID); !ok && msg.ID != "" { | ||||||
|  | 					gw.Messages.Add(msg.ID, msgIDs) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										31
									
								
								gateway/samechannel/samechannel_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								gateway/samechannel/samechannel_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | package samechannelgateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/BurntSushi/toml" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  |  | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var testconfig = ` | ||||||
|  | [mattermost.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[samechannelgateway]] | ||||||
|  |    enable = true | ||||||
|  |    name = "blah" | ||||||
|  |       accounts = [ "mattermost.test","slack.test" ] | ||||||
|  |       channels = [ "testing","testing2","testing10"] | ||||||
|  | ` | ||||||
|  |  | ||||||
|  | func TestGetConfig(t *testing.T) { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	sgw := New(cfg) | ||||||
|  | 	configs := sgw.GetConfig() | ||||||
|  | 	assert.Equal(t, []config.Gateway{{Name: "blah", Enable: true, In: []config.Bridge(nil), Out: []config.Bridge(nil), InOut: []config.Bridge{{Account: "mattermost.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}}}}, configs) | ||||||
|  | } | ||||||
| @@ -5,14 +5,14 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/gateway" | 	"github.com/42wim/matterbridge/gateway" | ||||||
| 	"github.com/42wim/matterbridge/gateway/samechannel" |  | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
| 	"github.com/google/gops/agent" | 	"github.com/google/gops/agent" | ||||||
|  | 	"os" | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	version = "0.16.3" | 	version = "1.4.0" | ||||||
| 	githash string | 	githash string | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -34,7 +34,7 @@ func main() { | |||||||
| 		fmt.Printf("version: %s %s\n", version, githash) | 		fmt.Printf("version: %s %s\n", version, githash) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if *flagDebug { | 	if *flagDebug || os.Getenv("DEBUG") == "1" { | ||||||
| 		log.Info("Enabling debug") | 		log.Info("Enabling debug") | ||||||
| 		log.SetLevel(log.DebugLevel) | 		log.SetLevel(log.DebugLevel) | ||||||
| 	} | 	} | ||||||
| @@ -43,20 +43,11 @@ func main() { | |||||||
| 		log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | 		log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||||
| 	} | 	} | ||||||
| 	cfg := config.NewConfig(*flagConfig) | 	cfg := config.NewConfig(*flagConfig) | ||||||
|  | 	r, err := gateway.NewRouter(cfg) | ||||||
| 	g := gateway.New(cfg) | 	if err != nil { | ||||||
| 	sgw := samechannelgateway.New(cfg) | 		log.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	gwconfigs := sgw.GetConfig() |  | ||||||
| 	for _, gw := range append(gwconfigs, cfg.Gateway...) { |  | ||||||
| 		if !gw.Enable { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		err := g.AddConfig(&gw) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("Starting gateway failed: %s", err) |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| 	err := g.Start() | 	err = r.Start() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Fatalf("Starting gateway failed: %s", err) | 		log.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -32,16 +32,38 @@ UseSASL=false | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| SkipTLSVerify=true | SkipTLSVerify=true | ||||||
|  |  | ||||||
|  | #If you know your charset, you can specify it manually.  | ||||||
|  | #Otherwise it tries to detect this automatically. Select one below | ||||||
|  | # "iso-8859-2:1987", "iso-8859-9:1989", "866", "latin9", "iso-8859-10:1992", "iso-ir-109", "hebrew",  | ||||||
|  | # "cp932", "iso-8859-15", "cp437", "utf-16be", "iso-8859-3:1988", "windows-1251", "utf16", "latin6",  | ||||||
|  | # "latin3", "iso-8859-1:1987", "iso-8859-9", "utf-16le", "big5", "cp819", "asmo-708", "utf-8",  | ||||||
|  | # "ibm437", "iso-ir-157", "iso-ir-144", "latin4", "850", "iso-8859-5", "iso-8859-5:1988", "l3",  | ||||||
|  | # "windows-31j", "utf8", "iso-8859-3", "437", "greek", "iso-8859-8", "l6", "l9-iso-8859-15",  | ||||||
|  | # "iso-8859-2", "latin2", "iso-ir-100", "iso-8859-6", "arabic", "iso-ir-148", "us-ascii", "x-sjis",  | ||||||
|  | # "utf16be", "iso-8859-8:1988", "utf16le", "l4", "utf-16", "iso-ir-138", "iso-8859-7", "iso-8859-7:1987",  | ||||||
|  | # "windows-1252", "l2", "koi8-r", "iso8859-1", "latin1", "ecma-114", "iso-ir-110", "elot-928",  | ||||||
|  | # "iso-ir-126", "iso-8859-1", "iso-ir-127", "cp850", "cyrillic", "greek8", "windows-1250", "iso-latin-1",  | ||||||
|  | # "l5", "ibm866", "cp866", "ms-kanji", "ibm850", "ecma-118", "iso-ir-101", "ibm819", "l1", "iso-8859-6:1987",  | ||||||
|  | # "latin5", "ascii", "sjis", "iso-8859-10", "iso-8859-4", "iso-8859-4:1988", "shift-jis | ||||||
|  | # The select charset will be converted to utf-8 when sent to other bridges. | ||||||
|  | #OPTIONAL (default "") | ||||||
|  | Charset="" | ||||||
|  |  | ||||||
| #Your nick on irc.  | #Your nick on irc.  | ||||||
| #REQUIRED | #REQUIRED | ||||||
| Nick="matterbot" | Nick="matterbot" | ||||||
|  |  | ||||||
| #If you registered your bot with a service like Nickserv on freenode.  | #If you registered your bot with a service like Nickserv on freenode.  | ||||||
| #Also being used when UseSASL=true | #Also being used when UseSASL=true | ||||||
|  | # | ||||||
|  | #Note: if you want do to quakenet auth, set NickServNick="Q@CServe.quakenet.org" | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| NickServNick="nickserv" | NickServNick="nickserv" | ||||||
| NickServPassword="secret" | NickServPassword="secret" | ||||||
|  |  | ||||||
|  | #OPTIONAL only used for quakenet auth | ||||||
|  | NickServUsername="username" | ||||||
|  |  | ||||||
| #Flood control | #Flood control | ||||||
| #Delay in milliseconds between each message send to the IRC server | #Delay in milliseconds between each message send to the IRC server | ||||||
| #OPTIONAL (default 1300) | #OPTIONAL (default 1300) | ||||||
| @@ -82,6 +104,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #XMPP section | #XMPP section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -139,6 +166,10 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #hipchat section | #hipchat section | ||||||
| @@ -189,6 +220,10 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #mattermost section | #mattermost section | ||||||
| @@ -213,6 +248,11 @@ Team="yourteam" | |||||||
| Login="yourlogin" | Login="yourlogin" | ||||||
| Password="yourpass" | Password="yourpass" | ||||||
|  |  | ||||||
|  | #personal access token of the bot. | ||||||
|  | #new feature since mattermost 4.1. See https://docs.mattermost.com/developer/personal-access-tokens.html | ||||||
|  | #OPTIONAL (you can use token instead of login/password) | ||||||
|  | #Token="abcdefghijklm" | ||||||
|  |  | ||||||
| #Enable this to make a http connection (instead of https) to your mattermost.  | #Enable this to make a http connection (instead of https) to your mattermost.  | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| NoTLS=false | NoTLS=false | ||||||
| @@ -294,6 +334,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #Gitter section | #Gitter section | ||||||
| #Best to make a dedicated gitter account for the bot. | #Best to make a dedicated gitter account for the bot. | ||||||
| @@ -333,6 +378,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #slack section | #slack section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -361,7 +411,6 @@ WebhookURL="https://hooks.slack.com/services/yourhook" | |||||||
| #AND DEDICATED BOT USER WHEN POSSIBLE! | #AND DEDICATED BOT USER WHEN POSSIBLE! | ||||||
| #Address to listen on for outgoing webhook requests from slack | #Address to listen on for outgoing webhook requests from slack | ||||||
| #See account settings - integrations - outgoing webhooks on slack | #See account settings - integrations - outgoing webhooks on slack | ||||||
| #This setting will not be used when useAPI is eanbled |  | ||||||
| #webhooks | #webhooks | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| WebhookBindAddress="0.0.0.0:9999" | WebhookBindAddress="0.0.0.0:9999" | ||||||
| @@ -420,6 +469,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #discord section | #discord section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -443,7 +497,12 @@ Server="yourservername" | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowEmbeds=false | ShowEmbeds=false | ||||||
|  |  | ||||||
|  | #Shows the username (minus the discriminator) instead of the server nickname | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | UseUserName=false | ||||||
|  |  | ||||||
| #Specify WebhookURL. If given, will relay messages using the Webhook, which gives a better look to messages. | #Specify WebhookURL. If given, will relay messages using the Webhook, which gives a better look to messages. | ||||||
|  | #This only works if you have one discord channel, if you have multiple discord channels you'll have to specify it in the gateway config | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| WebhookURL="Yourwebhooktokenhere" | WebhookURL="Yourwebhooktokenhere" | ||||||
|  |  | ||||||
| @@ -478,6 +537,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #telegram section | #telegram section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -540,6 +604,10 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #rocketchat section | #rocketchat section | ||||||
| @@ -604,6 +672,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #matrix section | #matrix section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -659,6 +732,11 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #steam section | #steam section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -708,6 +786,10 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #API | #API | ||||||
| @@ -742,7 +824,7 @@ RemoteNickFormat="{NICK}" | |||||||
| ################################################################### | ################################################################### | ||||||
| #General configuration | #General configuration | ||||||
| ################################################################### | ################################################################### | ||||||
| #Settings here override specific settings for each protocol | # Settings here are defaults that each protocol can override | ||||||
| [general] | [general] | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| @@ -751,6 +833,11 @@ RemoteNickFormat="{NICK}" | |||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #Gateway configuration | #Gateway configuration | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -786,7 +873,7 @@ enable=true | |||||||
|     #mattermost - channel (the channel name as seen in the URL, not the displayname) |     #mattermost - channel (the channel name as seen in the URL, not the displayname) | ||||||
|     #gitter     - username/room  |     #gitter     - username/room  | ||||||
|     #xmpp       - channel |     #xmpp       - channel | ||||||
|     #slack      - channel (the channel name as seen in the URL, not the displayname) |     #slack      - channel (without the #) | ||||||
|     #discord    - channel (without the #) |     #discord    - channel (without the #) | ||||||
|     #           - ID:123456789 (where 123456789 is the channel ID)  |     #           - ID:123456789 (where 123456789 is the channel ID)  | ||||||
|     #               (https://github.com/42wim/matterbridge/issues/57) |     #               (https://github.com/42wim/matterbridge/issues/57) | ||||||
| @@ -829,6 +916,14 @@ enable=true | |||||||
|         #OPTIONAL - your irc channel key |         #OPTIONAL - your irc channel key | ||||||
|         key="yourkey" |         key="yourkey" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="discord.game" | ||||||
|  |     channel="mygreatgame" | ||||||
|  |  | ||||||
|  |         #OPTIONAL - webhookurl only works for discord (it needs a different URL for each cahnnel) | ||||||
|  |         [gateway.inout.options] | ||||||
|  |         webhookurl=""https://discordapp.com/api/webhooks/123456789123456789/C9WPqExYWONPDZabcdef-def1434FGFjstasJX9pYht73y" | ||||||
|  |  | ||||||
|     #API example |     #API example | ||||||
|     #[[gateway.inout]] |     #[[gateway.inout]] | ||||||
|     #account="api.local" |     #account="api.local" | ||||||
|   | |||||||
| @@ -6,7 +6,6 @@ | |||||||
|  |  | ||||||
| [mattermost] | [mattermost] | ||||||
|     [mattermost.work] |     [mattermost.work] | ||||||
|     useAPI=true |  | ||||||
|     #do not prefix it wit http:// or https:// |     #do not prefix it wit http:// or https:// | ||||||
|     Server="yourmattermostserver.domain"  |     Server="yourmattermostserver.domain"  | ||||||
|     Team="yourteam" |     Team="yourteam" | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| package matterclient | package matterclient | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"crypto/md5" | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"errors" | 	"errors" | ||||||
| @@ -8,7 +9,6 @@ import ( | |||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"net/http/cookiejar" | 	"net/http/cookiejar" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strconv" |  | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -16,6 +16,7 @@ import ( | |||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
|  |  | ||||||
| 	"github.com/gorilla/websocket" | 	"github.com/gorilla/websocket" | ||||||
|  | 	"github.com/hashicorp/golang-lru" | ||||||
| 	"github.com/jpillora/backoff" | 	"github.com/jpillora/backoff" | ||||||
| 	"github.com/mattermost/platform/model" | 	"github.com/mattermost/platform/model" | ||||||
| ) | ) | ||||||
| @@ -43,8 +44,8 @@ type Message struct { | |||||||
| type Team struct { | type Team struct { | ||||||
| 	Team         *model.Team | 	Team         *model.Team | ||||||
| 	Id           string | 	Id           string | ||||||
| 	Channels     *model.ChannelList | 	Channels     []*model.Channel | ||||||
| 	MoreChannels *model.ChannelList | 	MoreChannels []*model.Channel | ||||||
| 	Users        map[string]*model.User | 	Users        map[string]*model.User | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -53,7 +54,7 @@ type MMClient struct { | |||||||
| 	*Credentials | 	*Credentials | ||||||
| 	Team          *Team | 	Team          *Team | ||||||
| 	OtherTeams    []*Team | 	OtherTeams    []*Team | ||||||
| 	Client        *model.Client | 	Client        *model.Client4 | ||||||
| 	User          *model.User | 	User          *model.User | ||||||
| 	Users         map[string]*model.User | 	Users         map[string]*model.User | ||||||
| 	MessageChan   chan *Message | 	MessageChan   chan *Message | ||||||
| @@ -66,6 +67,7 @@ type MMClient struct { | |||||||
| 	WsPingChan    chan *model.WebSocketResponse | 	WsPingChan    chan *model.WebSocketResponse | ||||||
| 	ServerVersion string | 	ServerVersion string | ||||||
| 	OnWsConnect   func() | 	OnWsConnect   func() | ||||||
|  | 	lruCache      *lru.Cache | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(login, pass, team, server string) *MMClient { | func New(login, pass, team, server string) *MMClient { | ||||||
| @@ -73,6 +75,7 @@ func New(login, pass, team, server string) *MMClient { | |||||||
| 	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | 	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | ||||||
| 	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"}) | 	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"}) | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | ||||||
|  | 	mmclient.lruCache, _ = lru.New(500) | ||||||
| 	return mmclient | 	return mmclient | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -105,21 +108,21 @@ func (m *MMClient) Login() error { | |||||||
| 		uriScheme = "http://" | 		uriScheme = "http://" | ||||||
| 	} | 	} | ||||||
| 	// login to mattermost | 	// login to mattermost | ||||||
| 	m.Client = model.NewClient(uriScheme + m.Credentials.Server) | 	m.Client = model.NewAPIv4Client(uriScheme + m.Credentials.Server) | ||||||
| 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | ||||||
| 	m.Client.HttpClient.Timeout = time.Second * 10 | 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
| 		d := b.Duration() | 		d := b.Duration() | ||||||
| 		// bogus call to get the serverversion | 		// bogus call to get the serverversion | ||||||
| 		_, err := m.Client.GetClientProperties() | 		_, resp := m.Client.Logout() | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return fmt.Errorf("%#v", err.Error()) | 			return fmt.Errorf("%#v", resp.Error.Error()) | ||||||
| 		} | 		} | ||||||
| 		if firstConnection && !supportedVersion(m.Client.ServerVersion) { | 		if firstConnection && !supportedVersion(resp.ServerVersion) { | ||||||
| 			return fmt.Errorf("unsupported mattermost version: %s", m.Client.ServerVersion) | 			return fmt.Errorf("unsupported mattermost version: %s", resp.ServerVersion) | ||||||
| 		} | 		} | ||||||
| 		m.ServerVersion = m.Client.ServerVersion | 		m.ServerVersion = resp.ServerVersion | ||||||
| 		if m.ServerVersion == "" { | 		if m.ServerVersion == "" { | ||||||
| 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||||
| 			time.Sleep(d) | 			time.Sleep(d) | ||||||
| @@ -130,30 +133,33 @@ func (m *MMClient) Login() error { | |||||||
| 	} | 	} | ||||||
| 	b.Reset() | 	b.Reset() | ||||||
|  |  | ||||||
| 	var myinfo *model.Result | 	var resp *model.Response | ||||||
|  | 	//var myinfo *model.Result | ||||||
| 	var appErr *model.AppError | 	var appErr *model.AppError | ||||||
| 	var logmsg = "trying login" | 	var logmsg = "trying login" | ||||||
| 	for { | 	for { | ||||||
| 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | ||||||
| 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||||
| 			m.log.Debugf(logmsg+" with %s", model.SESSION_COOKIE_TOKEN) | 			m.log.Debugf(logmsg + " with token") | ||||||
| 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||||
| 			if len(token) != 2 { | 			if len(token) != 2 { | ||||||
| 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||||
| 			} | 			} | ||||||
| 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | ||||||
| 			m.Client.MockSession(token[1]) | 			m.Client.AuthToken = token[1] | ||||||
| 			myinfo, appErr = m.Client.GetMe("") | 			m.Client.AuthType = model.HEADER_BEARER | ||||||
| 			if appErr != nil { | 			m.User, resp = m.Client.GetMe("") | ||||||
| 				return errors.New(appErr.DetailedError) | 			if resp.Error != nil { | ||||||
|  | 				return resp.Error | ||||||
| 			} | 			} | ||||||
| 			if myinfo.Data.(*model.User) == nil { | 			if m.User == nil { | ||||||
| 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | ||||||
| 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | ||||||
| 			} | 			} | ||||||
| 		} else { | 		} else { | ||||||
| 			_, appErr = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | 			m.User, resp = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | ||||||
| 		} | 		} | ||||||
|  | 		appErr = resp.Error | ||||||
| 		if appErr != nil { | 		if appErr != nil { | ||||||
| 			d := b.Duration() | 			d := b.Duration() | ||||||
| 			m.log.Debug(appErr.DetailedError) | 			m.log.Debug(appErr.DetailedError) | ||||||
| @@ -181,8 +187,6 @@ func (m *MMClient) Login() error { | |||||||
| 	if m.Team == nil { | 	if m.Team == nil { | ||||||
| 		return errors.New("team not found") | 		return errors.New("team not found") | ||||||
| 	} | 	} | ||||||
| 	// set our team id as default route |  | ||||||
| 	m.Client.SetTeamId(m.Team.Id) |  | ||||||
|  |  | ||||||
| 	m.wsConnect() | 	m.wsConnect() | ||||||
|  |  | ||||||
| @@ -203,7 +207,7 @@ func (m *MMClient) wsConnect() { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// setup websocket connection | 	// setup websocket connection | ||||||
| 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V3 + "/users/websocket" | 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V4 + "/websocket" | ||||||
| 	header := http.Header{} | 	header := http.Header{} | ||||||
| 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||||
|  |  | ||||||
| @@ -237,9 +241,9 @@ func (m *MMClient) Logout() error { | |||||||
| 		m.log.Debug("Not invalidating session in logout, credential is a token") | 		m.log.Debug("Not invalidating session in logout, credential is a token") | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	_, err := m.Client.Logout() | 	_, resp := m.Client.Logout() | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return err | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -270,7 +274,17 @@ func (m *MMClient) WsReceiver() { | |||||||
| 			m.log.Debugf("WsReceiver event: %#v", event) | 			m.log.Debugf("WsReceiver event: %#v", event) | ||||||
| 			msg := &Message{Raw: &event, Team: m.Credentials.Team} | 			msg := &Message{Raw: &event, Team: m.Credentials.Team} | ||||||
| 			m.parseMessage(msg) | 			m.parseMessage(msg) | ||||||
| 			m.MessageChan <- msg | 			// check if we didn't empty the message | ||||||
|  | 			if msg.Text != "" { | ||||||
|  | 				m.MessageChan <- msg | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// if we have file attached but the message is empty, also send it | ||||||
|  | 			if msg.Post != nil { | ||||||
|  | 				if msg.Text != "" || len(msg.Post.FileIds) > 0 || msg.Post.Type == "slack_attachment" { | ||||||
|  | 					m.MessageChan <- msg | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| @@ -285,7 +299,7 @@ func (m *MMClient) WsReceiver() { | |||||||
|  |  | ||||||
| func (m *MMClient) parseMessage(rmsg *Message) { | func (m *MMClient) parseMessage(rmsg *Message) { | ||||||
| 	switch rmsg.Raw.Event { | 	switch rmsg.Raw.Event { | ||||||
| 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED: | 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED, model.WEBSOCKET_EVENT_POST_DELETED: | ||||||
| 		m.parseActionPost(rmsg) | 		m.parseActionPost(rmsg) | ||||||
| 		/* | 		/* | ||||||
| 			case model.ACTION_USER_REMOVED: | 			case model.ACTION_USER_REMOVED: | ||||||
| @@ -306,6 +320,13 @@ func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) parseActionPost(rmsg *Message) { | func (m *MMClient) parseActionPost(rmsg *Message) { | ||||||
|  | 	// add post to cache, if it already exists don't relay this again. | ||||||
|  | 	// this should fix reposts | ||||||
|  | 	if ok, _ := m.lruCache.ContainsOrAdd(digestString(rmsg.Raw.Data["post"].(string)), true); ok { | ||||||
|  | 		m.log.Debugf("message %#v in cache, not processing again", rmsg.Raw.Data["post"].(string)) | ||||||
|  | 		rmsg.Text = "" | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
| 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | ||||||
| 	// we don't have the user, refresh the userlist | 	// we don't have the user, refresh the userlist | ||||||
| 	if m.GetUser(data.UserId) == nil { | 	if m.GetUser(data.UserId) == nil { | ||||||
| @@ -335,33 +356,34 @@ func (m *MMClient) parseActionPost(rmsg *Message) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateUsers() error { | func (m *MMClient) UpdateUsers() error { | ||||||
| 	mmusers, err := m.Client.GetProfiles(0, 50000, "") | 	mmusers, resp := m.Client.GetUsers(0, 50000, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Users = mmusers.Data.(map[string]*model.User) | 	for _, user := range mmusers { | ||||||
|  | 		m.Users[user.Id] = user | ||||||
|  | 	} | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannels() error { | func (m *MMClient) UpdateChannels() error { | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} |  | ||||||
| 	var mmchannels2 *model.Result |  | ||||||
| 	if m.mmVersion() >= 3.08 { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannelsPage(0, 5000) |  | ||||||
| 	} else { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannels("") |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return errors.New(err.DetailedError) |  | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) | 	m.Team.Channels = mmchannels | ||||||
| 	m.Team.MoreChannels = mmchannels2.Data.(*model.ChannelList) | 	m.Unlock() | ||||||
|  |  | ||||||
|  | 	mmchannels, resp = m.Client.GetPublicChannelsForTeam(m.Team.Id, 0, 5000, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return errors.New(resp.Error.DetailedError) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	m.Team.MoreChannels = mmchannels | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -374,14 +396,14 @@ func (m *MMClient) GetChannelName(channelId string) string { | |||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		if t.Channels != nil { | 		if t.Channels != nil { | ||||||
| 			for _, channel := range *t.Channels { | 			for _, channel := range t.Channels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			for _, channel := range *t.MoreChannels { | 			for _, channel := range t.MoreChannels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| @@ -399,7 +421,7 @@ func (m *MMClient) GetChannelId(name string, teamId string) string { | |||||||
| 	} | 	} | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id == teamId { | 		if t.Id == teamId { | ||||||
| 			for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 				if channel.Name == name { | 				if channel.Name == name { | ||||||
| 					return channel.Id | 					return channel.Id | ||||||
| 				} | 				} | ||||||
| @@ -413,7 +435,7 @@ func (m *MMClient) GetChannelTeamId(id string) string { | |||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range append(m.OtherTeams, m.Team) { | 	for _, t := range append(m.OtherTeams, m.Team) { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == id { | 			if channel.Id == id { | ||||||
| 				return channel.TeamId | 				return channel.TeamId | ||||||
| 			} | 			} | ||||||
| @@ -426,7 +448,7 @@ func (m *MMClient) GetChannelHeader(channelId string) string { | |||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == channelId { | 			if channel.Id == channelId { | ||||||
| 				return channel.Header | 				return channel.Header | ||||||
| 			} | 			} | ||||||
| @@ -436,55 +458,85 @@ func (m *MMClient) GetChannelHeader(channelId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) PostMessage(channelId string, text string) { | func (m *MMClient) PostMessage(channelId string, text string) (string, error) { | ||||||
| 	post := &model.Post{ChannelId: channelId, Message: text} | 	post := &model.Post{ChannelId: channelId, Message: text} | ||||||
| 	m.Client.CreatePost(post) | 	res, resp := m.Client.CreatePost(post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) PostMessageWithFiles(channelId string, text string, fileIds []string) (string, error) { | ||||||
|  | 	post := &model.Post{ChannelId: channelId, Message: text, FileIds: fileIds} | ||||||
|  | 	res, resp := m.Client.CreatePost(post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) EditMessage(postId string, text string) (string, error) { | ||||||
|  | 	post := &model.Post{Message: text} | ||||||
|  | 	res, resp := m.Client.UpdatePost(postId, post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) DeleteMessage(postId string) error { | ||||||
|  | 	_, resp := m.Client.DeletePost(postId) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) JoinChannel(channelId string) error { | func (m *MMClient) JoinChannel(channelId string) error { | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, c := range *m.Team.Channels { | 	for _, c := range m.Team.Channels { | ||||||
| 		if c.Id == channelId { | 		if c.Id == channelId { | ||||||
| 			m.log.Debug("Not joining ", channelId, " already joined.") | 			m.log.Debug("Not joining ", channelId, " already joined.") | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	m.log.Debug("Joining ", channelId) | 	m.log.Debug("Joining ", channelId) | ||||||
| 	_, err := m.Client.JoinChannel(channelId) | 	_, resp := m.Client.AddChannelMember(channelId, m.User.Id) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New("failed to join") | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | ||||||
| 	res, err := m.Client.GetPostsSince(channelId, time) | 	res, resp := m.Client.GetPostsSince(channelId, time) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) SearchPosts(query string) *model.PostList { | func (m *MMClient) SearchPosts(query string) *model.PostList { | ||||||
| 	res, err := m.Client.SearchPosts(query, false) | 	res, resp := m.Client.SearchPosts(m.Team.Id, query, false) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | ||||||
| 	res, err := m.Client.GetPosts(channelId, 0, limit, "") | 	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPublicLink(filename string) string { | func (m *MMClient) GetPublicLink(filename string) string { | ||||||
| 	res, err := m.Client.GetPublicLink(filename) | 	res, resp := m.Client.GetFileLink(filename) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| @@ -493,8 +545,8 @@ func (m *MMClient) GetPublicLink(filename string) string { | |||||||
| func (m *MMClient) GetPublicLinks(filenames []string) []string { | func (m *MMClient) GetPublicLinks(filenames []string) []string { | ||||||
| 	var output []string | 	var output []string | ||||||
| 	for _, f := range filenames { | 	for _, f := range filenames { | ||||||
| 		res, err := m.Client.GetPublicLink(f) | 		res, resp := m.Client.GetFileLink(f) | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		output = append(output, res) | 		output = append(output, res) | ||||||
| @@ -510,8 +562,8 @@ func (m *MMClient) GetFileLinks(filenames []string) []string { | |||||||
|  |  | ||||||
| 	var output []string | 	var output []string | ||||||
| 	for _, f := range filenames { | 	for _, f := range filenames { | ||||||
| 		res, err := m.Client.GetPublicLink(f) | 		res, resp := m.Client.GetFileLink(f) | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			// public links is probably disabled, create the link ourselves | 			// public links is probably disabled, create the link ourselves | ||||||
| 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V3+"/files/"+f+"/get") | 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V3+"/files/"+f+"/get") | ||||||
| 			continue | 			continue | ||||||
| @@ -522,42 +574,43 @@ func (m *MMClient) GetFileLinks(filenames []string) []string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | ||||||
| 	data := make(map[string]string) | 	channel := &model.Channel{Id: channelId, Header: header} | ||||||
| 	data["channel_id"] = channelId |  | ||||||
| 	data["channel_header"] = header |  | ||||||
| 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||||
| 	_, err := m.Client.UpdateChannelHeader(data) | 	_, resp := m.Client.UpdateChannel(channel) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		log.Error(err) | 		log.Error(resp.Error) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateLastViewed(channelId string) { | func (m *MMClient) UpdateLastViewed(channelId string) { | ||||||
| 	m.log.Debugf("posting lastview %#v", channelId) | 	m.log.Debugf("posting lastview %#v", channelId) | ||||||
| 	if m.mmVersion() >= 3.08 { | 	view := &model.ChannelView{ChannelId: channelId} | ||||||
| 		view := model.ChannelView{ChannelId: channelId} | 	res, _ := m.Client.ViewChannel(m.User.Id, view) | ||||||
| 		res, _ := m.Client.ViewChannel(view) | 	if !res { | ||||||
| 		if !res { | 		m.log.Errorf("ChannelView update for %s failed", channelId) | ||||||
| 			m.log.Errorf("ChannelView update for %s failed", channelId) |  | ||||||
| 		} |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	_, err := m.Client.UpdateLastViewedAt(channelId, true) |  | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Error(err) |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UpdateUserNick(nick string) error { | ||||||
|  | 	user := m.User | ||||||
|  | 	user.Nickname = nick | ||||||
|  | 	_, resp := m.Client.UpdateUser(user) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) UsernamesInChannel(channelId string) []string { | func (m *MMClient) UsernamesInChannel(channelId string) []string { | ||||||
| 	res, err := m.Client.GetProfilesInChannel(channelId, 0, 50000, "") | 	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, err) | 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error) | ||||||
| 		return []string{} | 		return []string{} | ||||||
| 	} | 	} | ||||||
| 	members := res.Data.(map[string]*model.User) | 	allusers := m.GetUsers() | ||||||
| 	result := []string{} | 	result := []string{} | ||||||
| 	for _, member := range members { | 	for _, member := range *res { | ||||||
| 		result = append(result, member.Nickname) | 		result = append(result, allusers[member.UserId].Nickname) | ||||||
| 	} | 	} | ||||||
| 	return result | 	return result | ||||||
| } | } | ||||||
| @@ -581,22 +634,15 @@ func (m *MMClient) createCookieJar(token string) *cookiejar.Jar { | |||||||
| func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | ||||||
| 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||||
| 	// create DM channel (only happens on first message) | 	// create DM channel (only happens on first message) | ||||||
| 	_, err := m.Client.CreateDirectChannel(toUserId) | 	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err) | 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||||
|  |  | ||||||
| 	// update our channels | 	// update our channels | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	m.UpdateChannels() | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Debug("SendDirectMessage: Couldn't update channels") |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	// build & send the message | 	// build & send the message | ||||||
| 	msg = strings.Replace(msg, "\r", "", -1) | 	msg = strings.Replace(msg, "\r", "", -1) | ||||||
| @@ -622,10 +668,10 @@ func (m *MMClient) GetChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	// our primary team channels first | 	// our primary team channels first | ||||||
| 	channels = append(channels, *m.Team.Channels...) | 	channels = append(channels, m.Team.Channels...) | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id != m.Team.Id { | 		if t.Id != m.Team.Id { | ||||||
| 			channels = append(channels, *t.Channels...) | 			channels = append(channels, t.Channels...) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| @@ -637,7 +683,7 @@ func (m *MMClient) GetMoreChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.MoreChannels...) | 		channels = append(channels, t.MoreChannels...) | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
| @@ -648,9 +694,9 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.Channels...) | 		channels = append(channels, t.Channels...) | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			channels = append(channels, *t.MoreChannels...) | 			channels = append(channels, t.MoreChannels...) | ||||||
| 		} | 		} | ||||||
| 		for _, c := range channels { | 		for _, c := range channels { | ||||||
| 			if c.Id == channelId { | 			if c.Id == channelId { | ||||||
| @@ -664,12 +710,11 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string { | |||||||
| func (m *MMClient) GetLastViewedAt(channelId string) int64 { | func (m *MMClient) GetLastViewedAt(channelId string) int64 { | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	res, err := m.Client.GetChannel(channelId, "") | 	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return model.GetMillis() | 		return model.GetMillis() | ||||||
| 	} | 	} | ||||||
| 	data := res.Data.(*model.ChannelData) | 	return res.LastViewedAt | ||||||
| 	return data.Member.LastViewedAt |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetUsers() map[string]*model.User { | func (m *MMClient) GetUsers() map[string]*model.User { | ||||||
| @@ -687,12 +732,11 @@ func (m *MMClient) GetUser(userId string) *model.User { | |||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	_, ok := m.Users[userId] | 	_, ok := m.Users[userId] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		res, err := m.Client.GetProfilesByIds([]string{userId}) | 		res, resp := m.Client.GetUser(userId, "") | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 		u := res.Data.(map[string]*model.User)[userId] | 		m.Users[userId] = res | ||||||
| 		m.Users[userId] = u |  | ||||||
| 	} | 	} | ||||||
| 	return m.Users[userId] | 	return m.Users[userId] | ||||||
| } | } | ||||||
| @@ -706,36 +750,36 @@ func (m *MMClient) GetUserName(userId string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetStatus(userId string) string { | func (m *MMClient) GetStatus(userId string) string { | ||||||
| 	res, err := m.Client.GetStatusesByIds([]string{userId}) | 	res, resp := m.Client.GetUserStatus(userId, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	status := res.Data.(map[string]string) | 	if res.Status == model.STATUS_AWAY { | ||||||
| 	if status[userId] == model.STATUS_AWAY { |  | ||||||
| 		return "away" | 		return "away" | ||||||
| 	} | 	} | ||||||
| 	if status[userId] == model.STATUS_ONLINE { | 	if res.Status == model.STATUS_ONLINE { | ||||||
| 		return "online" | 		return "online" | ||||||
| 	} | 	} | ||||||
| 	return "offline" | 	return "offline" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetStatuses() map[string]string { | func (m *MMClient) GetStatuses() map[string]string { | ||||||
| 	var ok bool | 	var ids []string | ||||||
| 	statuses := make(map[string]string) | 	statuses := make(map[string]string) | ||||||
| 	res, err := m.Client.GetStatuses() | 	for id := range m.Users { | ||||||
| 	if err != nil { | 		ids = append(ids, id) | ||||||
|  | 	} | ||||||
|  | 	res, resp := m.Client.GetUsersStatusesByIds(ids) | ||||||
|  | 	if resp.Error != nil { | ||||||
| 		return statuses | 		return statuses | ||||||
| 	} | 	} | ||||||
| 	if statuses, ok = res.Data.(map[string]string); ok { | 	for _, status := range res { | ||||||
| 		for userId, status := range statuses { | 		statuses[status.UserId] = "offline" | ||||||
| 			statuses[userId] = "offline" | 		if status.Status == model.STATUS_AWAY { | ||||||
| 			if status == model.STATUS_AWAY { | 			statuses[status.UserId] = "away" | ||||||
| 				statuses[userId] = "away" | 		} | ||||||
| 			} | 		if status.Status == model.STATUS_ONLINE { | ||||||
| 			if status == model.STATUS_ONLINE { | 			statuses[status.UserId] = "online" | ||||||
| 				statuses[userId] = "online" |  | ||||||
| 			} |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return statuses | 	return statuses | ||||||
| @@ -745,6 +789,14 @@ func (m *MMClient) GetTeamId() string { | |||||||
| 	return m.Team.Id | 	return m.Team.Id | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { | ||||||
|  | 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return f.FileInfos[0].Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) StatusLoop() { | func (m *MMClient) StatusLoop() { | ||||||
| 	retries := 0 | 	retries := 0 | ||||||
| 	backoff := time.Second * 60 | 	backoff := time.Second * 60 | ||||||
| @@ -786,40 +838,39 @@ func (m *MMClient) StatusLoop() { | |||||||
| func (m *MMClient) initUser() error { | func (m *MMClient) initUser() error { | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	initLoad, err := m.Client.GetInitialLoad() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	initData := initLoad.Data.(*model.InitialLoad) |  | ||||||
| 	m.User = initData.User |  | ||||||
| 	// we only load all team data on initial login. | 	// we only load all team data on initial login. | ||||||
| 	// all other updates are for channels from our (primary) team only. | 	// all other updates are for channels from our (primary) team only. | ||||||
| 	//m.log.Debug("initUser(): loading all team data") | 	//m.log.Debug("initUser(): loading all team data") | ||||||
| 	for _, v := range initData.Teams { | 	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "") | ||||||
| 		m.Client.SetTeamId(v.Id) | 	if resp.Error != nil { | ||||||
| 		mmusers, err := m.Client.GetProfiles(0, 50000, "") | 		return resp.Error | ||||||
| 		if err != nil { | 	} | ||||||
| 			return errors.New(err.DetailedError) | 	for _, team := range teams { | ||||||
|  | 		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return errors.New(resp.Error.DetailedError) | ||||||
| 		} | 		} | ||||||
| 		t := &Team{Team: v, Users: mmusers.Data.(map[string]*model.User), Id: v.Id} | 		usermap := make(map[string]*model.User) | ||||||
| 		mmchannels, err := m.Client.GetChannels("") | 		for _, user := range mmusers { | ||||||
| 		if err != nil { | 			usermap[user.Id] = user | ||||||
| 			return errors.New(err.DetailedError) |  | ||||||
| 		} | 		} | ||||||
| 		t.Channels = mmchannels.Data.(*model.ChannelList) |  | ||||||
| 		if m.mmVersion() >= 3.08 { | 		t := &Team{Team: team, Users: usermap, Id: team.Id} | ||||||
| 			mmchannels, err = m.Client.GetMoreChannelsPage(0, 5000) |  | ||||||
| 		} else { | 		mmchannels, resp := m.Client.GetChannelsForTeamForUser(team.Id, m.User.Id, "") | ||||||
| 			mmchannels, err = m.Client.GetMoreChannels("") | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
| 		} | 		} | ||||||
| 		if err != nil { | 		t.Channels = mmchannels | ||||||
| 			return errors.New(err.DetailedError) | 		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
| 		} | 		} | ||||||
| 		t.MoreChannels = mmchannels.Data.(*model.ChannelList) | 		t.MoreChannels = mmchannels | ||||||
| 		m.OtherTeams = append(m.OtherTeams, t) | 		m.OtherTeams = append(m.OtherTeams, t) | ||||||
| 		if v.Name == m.Credentials.Team { | 		if team.Name == m.Credentials.Team { | ||||||
| 			m.Team = t | 			m.Team = t | ||||||
| 			m.log.Debugf("initUser(): found our team %s (id: %s)", v.Name, v.Id) | 			m.log.Debugf("initUser(): found our team %s (id: %s)", team.Name, team.Id) | ||||||
| 		} | 		} | ||||||
| 		// add all users | 		// add all users | ||||||
| 		for k, v := range t.Users { | 		for k, v := range t.Users { | ||||||
| @@ -840,23 +891,16 @@ func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) err | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) mmVersion() float64 { |  | ||||||
| 	v, _ := strconv.ParseFloat(string(m.ServerVersion[0:2])+"0"+string(m.ServerVersion[2]), 64) |  | ||||||
| 	if string(m.ServerVersion[4]) == "." { |  | ||||||
| 		v, _ = strconv.ParseFloat(m.ServerVersion[0:4], 64) |  | ||||||
| 	} |  | ||||||
| 	return v |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func supportedVersion(version string) bool { | func supportedVersion(version string) bool { | ||||||
| 	if strings.HasPrefix(version, "3.5.0") || | 	if strings.HasPrefix(version, "3.8.0") || | ||||||
| 		strings.HasPrefix(version, "3.6.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.7.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.8.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.9.0") || | 		strings.HasPrefix(version, "3.9.0") || | ||||||
| 		strings.HasPrefix(version, "3.10.0") || | 		strings.HasPrefix(version, "3.10.0") || | ||||||
| 		strings.HasPrefix(version, "4.0") { | 		strings.HasPrefix(version, "4.") { | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| 	return false | 	return false | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func digestString(s string) string { | ||||||
|  | 	return fmt.Sprintf("%x", md5.Sum([]byte(s))) | ||||||
|  | } | ||||||
|   | |||||||
| @@ -17,13 +17,14 @@ import ( | |||||||
|  |  | ||||||
| // OMessage for mattermost incoming webhook. (send to mattermost) | // OMessage for mattermost incoming webhook. (send to mattermost) | ||||||
| type OMessage struct { | type OMessage struct { | ||||||
| 	Channel     string      `json:"channel,omitempty"` | 	Channel     string                 `json:"channel,omitempty"` | ||||||
| 	IconURL     string      `json:"icon_url,omitempty"` | 	IconURL     string                 `json:"icon_url,omitempty"` | ||||||
| 	IconEmoji   string      `json:"icon_emoji,omitempty"` | 	IconEmoji   string                 `json:"icon_emoji,omitempty"` | ||||||
| 	UserName    string      `json:"username,omitempty"` | 	UserName    string                 `json:"username,omitempty"` | ||||||
| 	Text        string      `json:"text"` | 	Text        string                 `json:"text"` | ||||||
| 	Attachments interface{} `json:"attachments,omitempty"` | 	Attachments interface{}            `json:"attachments,omitempty"` | ||||||
| 	Type        string      `json:"type,omitempty"` | 	Type        string                 `json:"type,omitempty"` | ||||||
|  | 	Props       map[string]interface{} `json:"props"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // IMessage for mattermost outgoing webhook. (received from mattermost) | // IMessage for mattermost outgoing webhook. (received from mattermost) | ||||||
| @@ -43,6 +44,7 @@ type IMessage struct { | |||||||
| 	ServiceId   string `schema:"service_id"` | 	ServiceId   string `schema:"service_id"` | ||||||
| 	Text        string `schema:"text"` | 	Text        string `schema:"text"` | ||||||
| 	TriggerWord string `schema:"trigger_word"` | 	TriggerWord string `schema:"trigger_word"` | ||||||
|  | 	FileIDs     string `schema:"file_ids"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Client for Mattermost. | // Client for Mattermost. | ||||||
|   | |||||||
| @@ -205,17 +205,43 @@ func (gitter *Gitter) GetMessage(roomID, messageID string) (*Message, error) { | |||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // SendMessage sends a message to a room | // SendMessage sends a message to a room | ||||||
| func (gitter *Gitter) SendMessage(roomID, text string) error { | func (gitter *Gitter) SendMessage(roomID, text string) (*Message, error) { | ||||||
| 
 | 
 | ||||||
| 	message := Message{Text: text} | 	message := Message{Text: text} | ||||||
| 	body, _ := json.Marshal(message) | 	body, _ := json.Marshal(message) | ||||||
| 	_, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | 	response, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		gitter.log(err) | 		gitter.log(err) | ||||||
| 		return err | 		return nil, err | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	return nil | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // UpdateMessage updates a message in a room | ||||||
|  | func (gitter *Gitter) UpdateMessage(roomID, msgID, text string) (*Message, error) { | ||||||
|  | 
 | ||||||
|  | 	message := Message{Text: text} | ||||||
|  | 	body, _ := json.Marshal(message) | ||||||
|  | 	response, err := gitter.put(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages/"+msgID, body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // JoinRoom joins a room | // JoinRoom joins a room | ||||||
| @@ -265,7 +291,7 @@ func (gitter *Gitter) SearchRooms(room string) ([]Room, error) { | |||||||
| 		Results []Room `json:"results"` | 		Results []Room `json:"results"` | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room ) | 	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room) | ||||||
| 
 | 
 | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		gitter.log(err) | 		gitter.log(err) | ||||||
| @@ -414,6 +440,39 @@ func (gitter *Gitter) post(url string, body []byte) ([]byte, error) { | |||||||
| 	return result, nil | 	return result, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | func (gitter *Gitter) put(url string, body []byte) ([]byte, error) { | ||||||
|  | 	r, err := http.NewRequest("PUT", url, bytes.NewBuffer(body)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	r.Header.Set("Content-Type", "application/json") | ||||||
|  | 	r.Header.Set("Accept", "application/json") | ||||||
|  | 	r.Header.Set("Authorization", "Bearer "+gitter.config.token) | ||||||
|  | 
 | ||||||
|  | 	resp, err := gitter.config.client.Do(r) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer resp.Body.Close() | ||||||
|  | 
 | ||||||
|  | 	if resp.StatusCode != http.StatusOK { | ||||||
|  | 		err = APIError{What: fmt.Sprintf("Status code: %v", resp.StatusCode)} | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	result, err := ioutil.ReadAll(resp.Body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return result, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| func (gitter *Gitter) delete(url string) ([]byte, error) { | func (gitter *Gitter) delete(url string) ([]byte, error) { | ||||||
| 	r, err := http.NewRequest("delete", url, nil) | 	r, err := http.NewRequest("delete", url, nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
							
								
								
									
										117
									
								
								vendor/github.com/thoj/go-ircevent/irc.go → vendor/github.com/42wim/go-ircevent/irc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										117
									
								
								vendor/github.com/thoj/go-ircevent/irc.go → vendor/github.com/42wim/go-ircevent/irc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -87,6 +87,17 @@ func (irc *Connection) readLoop() { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // Unescape tag values as defined in the IRCv3.2 message tags spec | ||||||
|  | // http://ircv3.net/specs/core/message-tags-3.2.html | ||||||
|  | func unescapeTagValue(value string) string { | ||||||
|  | 	value = strings.Replace(value, "\\:", ";", -1) | ||||||
|  | 	value = strings.Replace(value, "\\s", " ", -1) | ||||||
|  | 	value = strings.Replace(value, "\\\\", "\\", -1) | ||||||
|  | 	value = strings.Replace(value, "\\r", "\r", -1) | ||||||
|  | 	value = strings.Replace(value, "\\n", "\n", -1) | ||||||
|  | 	return value | ||||||
|  | } | ||||||
|  | 
 | ||||||
| //Parse raw irc messages | //Parse raw irc messages | ||||||
| func parseToEvent(msg string) (*Event, error) { | func parseToEvent(msg string) (*Event, error) { | ||||||
| 	msg = strings.TrimSuffix(msg, "\n") //Remove \r\n | 	msg = strings.TrimSuffix(msg, "\n") //Remove \r\n | ||||||
| @@ -95,6 +106,26 @@ func parseToEvent(msg string) (*Event, error) { | |||||||
| 	if len(msg) < 5 { | 	if len(msg) < 5 { | ||||||
| 		return nil, errors.New("Malformed msg from server") | 		return nil, errors.New("Malformed msg from server") | ||||||
| 	} | 	} | ||||||
|  | 
 | ||||||
|  | 	if msg[0] == '@' { | ||||||
|  | 		// IRCv3 Message Tags | ||||||
|  | 		if i := strings.Index(msg, " "); i > -1 { | ||||||
|  | 			event.Tags = make(map[string]string) | ||||||
|  | 			tags := strings.Split(msg[1:i], ";") | ||||||
|  | 			for _, data := range tags { | ||||||
|  | 				parts := strings.SplitN(data, "=", 2) | ||||||
|  | 				if len(parts) == 1 { | ||||||
|  | 					event.Tags[parts[0]] = "" | ||||||
|  | 				} else { | ||||||
|  | 					event.Tags[parts[0]] = unescapeTagValue(parts[1]) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			msg = msg[i+1 : len(msg)] | ||||||
|  | 		} else { | ||||||
|  | 			return nil, errors.New("Malformed msg from server") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	if msg[0] == ':' { | 	if msg[0] == ':' { | ||||||
| 		if i := strings.Index(msg, " "); i > -1 { | 		if i := strings.Index(msg, " "); i > -1 { | ||||||
| 			event.Source = msg[1:i] | 			event.Source = msg[1:i] | ||||||
| @@ -196,12 +227,17 @@ func (irc *Connection) isQuitting() bool { | |||||||
| // Main loop to control the connection. | // Main loop to control the connection. | ||||||
| func (irc *Connection) Loop() { | func (irc *Connection) Loop() { | ||||||
| 	errChan := irc.ErrorChan() | 	errChan := irc.ErrorChan() | ||||||
|  | 	connTime := time.Now() | ||||||
| 	for !irc.isQuitting() { | 	for !irc.isQuitting() { | ||||||
| 		err := <-errChan | 		err := <-errChan | ||||||
| 		close(irc.end) | 		close(irc.end) | ||||||
| 		irc.Wait() | 		irc.Wait() | ||||||
| 		for !irc.isQuitting() { | 		for !irc.isQuitting() { | ||||||
| 			irc.Log.Printf("Error, disconnected: %s\n", err) | 			irc.Log.Printf("Error, disconnected: %s\n", err) | ||||||
|  | 			if time.Now().Sub(connTime) < time.Second*5 { | ||||||
|  | 				irc.Log.Println("Rreconnecting too fast, sleeping 60 seconds") | ||||||
|  | 				time.Sleep(60 * time.Second) | ||||||
|  | 			} | ||||||
| 			if err = irc.Reconnect(); err != nil { | 			if err = irc.Reconnect(); err != nil { | ||||||
| 				irc.Log.Printf("Error while reconnecting: %s\n", err) | 				irc.Log.Printf("Error while reconnecting: %s\n", err) | ||||||
| 				time.Sleep(60 * time.Second) | 				time.Sleep(60 * time.Second) | ||||||
| @@ -210,6 +246,7 @@ func (irc *Connection) Loop() { | |||||||
| 				break | 				break | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | 		connTime = time.Now() | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| @@ -430,26 +467,84 @@ func (irc *Connection) Connect(server string) error { | |||||||
| 		irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password) | 		irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	resChan := make(chan *SASLResult) | 	err = irc.negotiateCaps() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("NICK %s\r\n", irc.nick) | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Negotiate IRCv3 capabilities | ||||||
|  | func (irc *Connection) negotiateCaps() error { | ||||||
|  | 	saslResChan := make(chan *SASLResult) | ||||||
|  | 	if irc.UseSASL { | ||||||
|  | 		irc.RequestCaps = append(irc.RequestCaps, "sasl") | ||||||
|  | 		irc.setupSASLCallbacks(saslResChan) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if len(irc.RequestCaps) == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	cap_chan := make(chan bool, len(irc.RequestCaps)) | ||||||
|  | 	irc.AddCallback("CAP", func(e *Event) { | ||||||
|  | 		if len(e.Arguments) != 3 { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		command := e.Arguments[1] | ||||||
|  | 
 | ||||||
|  | 		if command == "LS" { | ||||||
|  | 			missing_caps := len(irc.RequestCaps) | ||||||
|  | 			for _, cap_name := range strings.Split(e.Arguments[2], " ") { | ||||||
|  | 				for _, req_cap := range irc.RequestCaps { | ||||||
|  | 					if cap_name == req_cap { | ||||||
|  | 						irc.pwrite <- fmt.Sprintf("CAP REQ :%s\r\n", cap_name) | ||||||
|  | 						missing_caps-- | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			for i := 0; i < missing_caps; i++ { | ||||||
|  | 				cap_chan <- true | ||||||
|  | 			} | ||||||
|  | 		} else if command == "ACK" || command == "NAK" { | ||||||
|  | 			for _, cap_name := range strings.Split(strings.TrimSpace(e.Arguments[2]), " ") { | ||||||
|  | 				if cap_name == "" { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 
 | ||||||
|  | 				if command == "ACK" { | ||||||
|  | 					irc.AcknowledgedCaps = append(irc.AcknowledgedCaps, cap_name) | ||||||
|  | 				} | ||||||
|  | 				cap_chan <- true | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	irc.pwrite <- "CAP LS\r\n" | ||||||
|  | 
 | ||||||
| 	if irc.UseSASL { | 	if irc.UseSASL { | ||||||
| 		irc.setupSASLCallbacks(resChan) |  | ||||||
| 		irc.pwrite <- fmt.Sprintf("CAP LS\r\n") |  | ||||||
| 		// request SASL |  | ||||||
| 		irc.pwrite <- fmt.Sprintf("CAP REQ :sasl\r\n") |  | ||||||
| 		// if sasl request doesn't complete in 15 seconds, close chan and timeout |  | ||||||
| 		select { | 		select { | ||||||
| 		case res := <-resChan: | 		case res := <-saslResChan: | ||||||
| 			if res.Failed { | 			if res.Failed { | ||||||
| 				close(resChan) | 				close(saslResChan) | ||||||
| 				return res.Err | 				return res.Err | ||||||
| 			} | 			} | ||||||
| 		case <-time.After(time.Second * 15): | 		case <-time.After(time.Second * 15): | ||||||
| 			close(resChan) | 			close(saslResChan) | ||||||
| 			return errors.New("SASL setup timed out. This shouldn't happen.") | 			return errors.New("SASL setup timed out. This shouldn't happen.") | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	irc.pwrite <- fmt.Sprintf("NICK %s\r\n", irc.nick) | 
 | ||||||
| 	irc.pwrite <- fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user) | 	// Wait for all capabilities to be ACKed or NAKed before ending negotiation | ||||||
|  | 	for i := 0; i < len(irc.RequestCaps); i++ { | ||||||
|  | 		<-cap_chan | ||||||
|  | 	} | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("CAP END\r\n") | ||||||
|  | 
 | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| @@ -43,7 +43,6 @@ func (irc *Connection) setupSASLCallbacks(result chan<- *SASLResult) { | |||||||
| 		result <- &SASLResult{true, errors.New(e.Arguments[1])} | 		result <- &SASLResult{true, errors.New(e.Arguments[1])} | ||||||
| 	}) | 	}) | ||||||
| 	irc.AddCallback("903", func(e *Event) { | 	irc.AddCallback("903", func(e *Event) { | ||||||
| 		irc.SendRaw("CAP END") |  | ||||||
| 		result <- &SASLResult{false, nil} | 		result <- &SASLResult{false, nil} | ||||||
| 	}) | 	}) | ||||||
| 	irc.AddCallback("904", func(e *Event) { | 	irc.AddCallback("904", func(e *Event) { | ||||||
| @@ -15,20 +15,22 @@ import ( | |||||||
| type Connection struct { | type Connection struct { | ||||||
| 	sync.Mutex | 	sync.Mutex | ||||||
| 	sync.WaitGroup | 	sync.WaitGroup | ||||||
| 	Debug        bool | 	Debug            bool | ||||||
| 	Error        chan error | 	Error            chan error | ||||||
| 	Password     string | 	Password         string | ||||||
| 	UseTLS       bool | 	UseTLS           bool | ||||||
| 	UseSASL      bool | 	UseSASL          bool | ||||||
| 	SASLLogin    string | 	RequestCaps      []string | ||||||
| 	SASLPassword string | 	AcknowledgedCaps []string | ||||||
| 	SASLMech     string | 	SASLLogin        string | ||||||
| 	TLSConfig    *tls.Config | 	SASLPassword     string | ||||||
| 	Version      string | 	SASLMech         string | ||||||
| 	Timeout      time.Duration | 	TLSConfig        *tls.Config | ||||||
| 	PingFreq     time.Duration | 	Version          string | ||||||
| 	KeepAlive    time.Duration | 	Timeout          time.Duration | ||||||
| 	Server       string | 	PingFreq         time.Duration | ||||||
|  | 	KeepAlive        time.Duration | ||||||
|  | 	Server           string | ||||||
| 
 | 
 | ||||||
| 	socket net.Conn | 	socket net.Conn | ||||||
| 	pwrite chan string | 	pwrite chan string | ||||||
| @@ -59,6 +61,7 @@ type Event struct { | |||||||
| 	Source     string //<host> | 	Source     string //<host> | ||||||
| 	User       string //<usr> | 	User       string //<usr> | ||||||
| 	Arguments  []string | 	Arguments  []string | ||||||
|  | 	Tags       map[string]string | ||||||
| 	Connection *Connection | 	Connection *Connection | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
							
								
								
									
										3
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -21,7 +21,7 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| // VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/) | // VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/) | ||||||
| const VERSION = "0.16.0" | const VERSION = "0.17.0" | ||||||
|  |  | ||||||
| // ErrMFA will be risen by New when the user has 2FA. | // ErrMFA will be risen by New when the user has 2FA. | ||||||
| var ErrMFA = errors.New("account has 2FA enabled") | var ErrMFA = errors.New("account has 2FA enabled") | ||||||
| @@ -59,6 +59,7 @@ func New(args ...interface{}) (s *Session, err error) { | |||||||
| 		MaxRestRetries:         3, | 		MaxRestRetries:         3, | ||||||
| 		Client:                 &http.Client{Timeout: (20 * time.Second)}, | 		Client:                 &http.Client{Timeout: (20 * time.Second)}, | ||||||
| 		sequence:               new(int64), | 		sequence:               new(int64), | ||||||
|  | 		LastHeartbeatAck:       time.Now().UTC(), | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// If no arguments are passed return the empty Session interface. | 	// If no arguments are passed return the empty Session interface. | ||||||
|   | |||||||
							
								
								
									
										42
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										42
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -11,6 +11,9 @@ | |||||||
|  |  | ||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
|  | // APIVersion is the Discord API version used for the REST and Websocket API. | ||||||
|  | var APIVersion = "6" | ||||||
|  |  | ||||||
| // Known Discord API Endpoints. | // Known Discord API Endpoints. | ||||||
| var ( | var ( | ||||||
| 	EndpointStatus     = "https://status.discordapp.com/api/v2/" | 	EndpointStatus     = "https://status.discordapp.com/api/v2/" | ||||||
| @@ -18,13 +21,14 @@ var ( | |||||||
| 	EndpointSmActive   = EndpointSm + "active.json" | 	EndpointSmActive   = EndpointSm + "active.json" | ||||||
| 	EndpointSmUpcoming = EndpointSm + "upcoming.json" | 	EndpointSmUpcoming = EndpointSm + "upcoming.json" | ||||||
|  |  | ||||||
| 	EndpointDiscord  = "https://discordapp.com/" | 	EndpointDiscord    = "https://discordapp.com/" | ||||||
| 	EndpointAPI      = EndpointDiscord + "api/" | 	EndpointAPI        = EndpointDiscord + "api/v" + APIVersion + "/" | ||||||
| 	EndpointGuilds   = EndpointAPI + "guilds/" | 	EndpointGuilds     = EndpointAPI + "guilds/" | ||||||
| 	EndpointChannels = EndpointAPI + "channels/" | 	EndpointChannels   = EndpointAPI + "channels/" | ||||||
| 	EndpointUsers    = EndpointAPI + "users/" | 	EndpointUsers      = EndpointAPI + "users/" | ||||||
| 	EndpointGateway  = EndpointAPI + "gateway" | 	EndpointGateway    = EndpointAPI + "gateway" | ||||||
| 	EndpointWebhooks = EndpointAPI + "webhooks/" | 	EndpointGatewayBot = EndpointGateway + "/bot" | ||||||
|  | 	EndpointWebhooks   = EndpointAPI + "webhooks/" | ||||||
|  |  | ||||||
| 	EndpointCDN             = "https://cdn.discordapp.com/" | 	EndpointCDN             = "https://cdn.discordapp.com/" | ||||||
| 	EndpointCDNAttachments  = EndpointCDN + "attachments/" | 	EndpointCDNAttachments  = EndpointCDN + "attachments/" | ||||||
| @@ -54,16 +58,17 @@ var ( | |||||||
| 	EndpointReport       = EndpointAPI + "report" | 	EndpointReport       = EndpointAPI + "report" | ||||||
| 	EndpointIntegrations = EndpointAPI + "integrations" | 	EndpointIntegrations = EndpointAPI + "integrations" | ||||||
|  |  | ||||||
| 	EndpointUser              = func(uID string) string { return EndpointUsers + uID } | 	EndpointUser               = func(uID string) string { return EndpointUsers + uID } | ||||||
| 	EndpointUserAvatar        = func(uID, aID string) string { return EndpointCDNAvatars + uID + "/" + aID + ".png" } | 	EndpointUserAvatar         = func(uID, aID string) string { return EndpointCDNAvatars + uID + "/" + aID + ".png" } | ||||||
| 	EndpointUserSettings      = func(uID string) string { return EndpointUsers + uID + "/settings" } | 	EndpointUserAvatarAnimated = func(uID, aID string) string { return EndpointCDNAvatars + uID + "/" + aID + ".gif" } | ||||||
| 	EndpointUserGuilds        = func(uID string) string { return EndpointUsers + uID + "/guilds" } | 	EndpointUserSettings       = func(uID string) string { return EndpointUsers + uID + "/settings" } | ||||||
| 	EndpointUserGuild         = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID } | 	EndpointUserGuilds         = func(uID string) string { return EndpointUsers + uID + "/guilds" } | ||||||
| 	EndpointUserGuildSettings = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID + "/settings" } | 	EndpointUserGuild          = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID } | ||||||
| 	EndpointUserChannels      = func(uID string) string { return EndpointUsers + uID + "/channels" } | 	EndpointUserGuildSettings  = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID + "/settings" } | ||||||
| 	EndpointUserDevices       = func(uID string) string { return EndpointUsers + uID + "/devices" } | 	EndpointUserChannels       = func(uID string) string { return EndpointUsers + uID + "/channels" } | ||||||
| 	EndpointUserConnections   = func(uID string) string { return EndpointUsers + uID + "/connections" } | 	EndpointUserDevices        = func(uID string) string { return EndpointUsers + uID + "/devices" } | ||||||
| 	EndpointUserNotes         = func(uID string) string { return EndpointUsers + "@me/notes/" + uID } | 	EndpointUserConnections    = func(uID string) string { return EndpointUsers + uID + "/connections" } | ||||||
|  | 	EndpointUserNotes          = func(uID string) string { return EndpointUsers + "@me/notes/" + uID } | ||||||
|  |  | ||||||
| 	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID } | 	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID } | ||||||
| 	EndpointGuildInivtes         = func(gID string) string { return EndpointGuilds + gID + "/invites" } | 	EndpointGuildInivtes         = func(gID string) string { return EndpointGuilds + gID + "/invites" } | ||||||
| @@ -103,6 +108,9 @@ var ( | |||||||
| 	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID } | 	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID } | ||||||
| 	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token } | 	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token } | ||||||
|  |  | ||||||
|  | 	EndpointMessageReactionsAll = func(cID, mID string) string { | ||||||
|  | 		return EndpointChannelMessage(cID, mID) + "/reactions" | ||||||
|  | 	} | ||||||
| 	EndpointMessageReactions = func(cID, mID, eID string) string { | 	EndpointMessageReactions = func(cID, mID, eID string) string { | ||||||
| 		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID | 		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										14
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -156,12 +156,20 @@ func (s *Session) removeEventHandlerInstance(t string, ehi *eventHandlerInstance | |||||||
| // Handles calling permanent and once handlers for an event type. | // Handles calling permanent and once handlers for an event type. | ||||||
| func (s *Session) handle(t string, i interface{}) { | func (s *Session) handle(t string, i interface{}) { | ||||||
| 	for _, eh := range s.handlers[t] { | 	for _, eh := range s.handlers[t] { | ||||||
| 		go eh.eventHandler.Handle(s, i) | 		if s.SyncEvents { | ||||||
|  | 			eh.eventHandler.Handle(s, i) | ||||||
|  | 		} else { | ||||||
|  | 			go eh.eventHandler.Handle(s, i) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if len(s.onceHandlers[t]) > 0 { | 	if len(s.onceHandlers[t]) > 0 { | ||||||
| 		for _, eh := range s.onceHandlers[t] { | 		for _, eh := range s.onceHandlers[t] { | ||||||
| 			go eh.eventHandler.Handle(s, i) | 			if s.SyncEvents { | ||||||
|  | 				eh.eventHandler.Handle(s, i) | ||||||
|  | 			} else { | ||||||
|  | 				go eh.eventHandler.Handle(s, i) | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 		s.onceHandlers[t] = nil | 		s.onceHandlers[t] = nil | ||||||
| 	} | 	} | ||||||
| @@ -216,7 +224,7 @@ func (s *Session) onInterface(i interface{}) { | |||||||
| 	case *VoiceStateUpdate: | 	case *VoiceStateUpdate: | ||||||
| 		go s.onVoiceStateUpdate(t) | 		go s.onVoiceStateUpdate(t) | ||||||
| 	} | 	} | ||||||
| 	err := s.State.onInterface(s, i) | 	err := s.State.OnInterface(s, i) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		s.log(LogDebug, "error dispatching internal event, %s", err) | 		s.log(LogDebug, "error dispatching internal event, %s", err) | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										94
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										94
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -10,9 +10,24 @@ | |||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"regexp" | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // MessageType is the type of Message | ||||||
|  | type MessageType int | ||||||
|  |  | ||||||
|  | // Block contains the valid known MessageType values | ||||||
|  | const ( | ||||||
|  | 	MessageTypeDefault MessageType = iota | ||||||
|  | 	MessageTypeRecipientAdd | ||||||
|  | 	MessageTypeRecipientRemove | ||||||
|  | 	MessageTypeCall | ||||||
|  | 	MessageTypeChannelNameChange | ||||||
|  | 	MessageTypeChannelIconChange | ||||||
|  | 	MessageTypeChannelPinnedMessage | ||||||
|  | 	MessageTypeGuildMemberJoin | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // A Message stores all data related to a specific Discord message. | // A Message stores all data related to a specific Discord message. | ||||||
| @@ -30,12 +45,14 @@ type Message struct { | |||||||
| 	Embeds          []*MessageEmbed      `json:"embeds"` | 	Embeds          []*MessageEmbed      `json:"embeds"` | ||||||
| 	Mentions        []*User              `json:"mentions"` | 	Mentions        []*User              `json:"mentions"` | ||||||
| 	Reactions       []*MessageReactions  `json:"reactions"` | 	Reactions       []*MessageReactions  `json:"reactions"` | ||||||
|  | 	Type            MessageType          `json:"type"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // File stores info about files you e.g. send in messages. | // File stores info about files you e.g. send in messages. | ||||||
| type File struct { | type File struct { | ||||||
| 	Name   string | 	Name        string | ||||||
| 	Reader io.Reader | 	ContentType string | ||||||
|  | 	Reader      io.Reader | ||||||
| } | } | ||||||
|  |  | ||||||
| // MessageSend stores all parameters you can send with ChannelMessageSendComplex. | // MessageSend stores all parameters you can send with ChannelMessageSendComplex. | ||||||
| @@ -43,7 +60,10 @@ type MessageSend struct { | |||||||
| 	Content string        `json:"content,omitempty"` | 	Content string        `json:"content,omitempty"` | ||||||
| 	Embed   *MessageEmbed `json:"embed,omitempty"` | 	Embed   *MessageEmbed `json:"embed,omitempty"` | ||||||
| 	Tts     bool          `json:"tts"` | 	Tts     bool          `json:"tts"` | ||||||
| 	File    *File         `json:"file"` | 	Files   []*File       `json:"-"` | ||||||
|  |  | ||||||
|  | 	// TODO: Remove this when compatibility is not required. | ||||||
|  | 	File *File `json:"-"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // MessageEdit is used to chain parameters via ChannelMessageEditComplex, which | // MessageEdit is used to chain parameters via ChannelMessageEditComplex, which | ||||||
| @@ -168,13 +188,65 @@ type MessageReactions struct { | |||||||
|  |  | ||||||
| // ContentWithMentionsReplaced will replace all @<id> mentions with the | // ContentWithMentionsReplaced will replace all @<id> mentions with the | ||||||
| // username of the mention. | // username of the mention. | ||||||
| func (m *Message) ContentWithMentionsReplaced() string { | func (m *Message) ContentWithMentionsReplaced() (content string) { | ||||||
| 	if m.Mentions == nil { | 	content = m.Content | ||||||
| 		return m.Content |  | ||||||
| 	} |  | ||||||
| 	content := m.Content |  | ||||||
| 	for _, user := range m.Mentions { | 	for _, user := range m.Mentions { | ||||||
| 		content = regexp.MustCompile(fmt.Sprintf("<@!?(%s)>", user.ID)).ReplaceAllString(content, "@"+user.Username) | 		content = strings.NewReplacer( | ||||||
|  | 			"<@"+user.ID+">", "@"+user.Username, | ||||||
|  | 			"<@!"+user.ID+">", "@"+user.Username, | ||||||
|  | 		).Replace(content) | ||||||
| 	} | 	} | ||||||
| 	return content | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var patternChannels = regexp.MustCompile("<#[^>]*>") | ||||||
|  |  | ||||||
|  | // ContentWithMoreMentionsReplaced will replace all @<id> mentions with the | ||||||
|  | // username of the mention, but also role IDs and more. | ||||||
|  | func (m *Message) ContentWithMoreMentionsReplaced(s *Session) (content string, err error) { | ||||||
|  | 	content = m.Content | ||||||
|  |  | ||||||
|  | 	if !s.StateEnabled { | ||||||
|  | 		content = m.ContentWithMentionsReplaced() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel, err := s.State.Channel(m.ChannelID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		content = m.ContentWithMentionsReplaced() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, user := range m.Mentions { | ||||||
|  | 		nick := user.Username | ||||||
|  |  | ||||||
|  | 		member, err := s.State.Member(channel.GuildID, user.ID) | ||||||
|  | 		if err == nil && member.Nick != "" { | ||||||
|  | 			nick = member.Nick | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		content = strings.NewReplacer( | ||||||
|  | 			"<@"+user.ID+">", "@"+user.Username, | ||||||
|  | 			"<@!"+user.ID+">", "@"+nick, | ||||||
|  | 		).Replace(content) | ||||||
|  | 	} | ||||||
|  | 	for _, roleID := range m.MentionRoles { | ||||||
|  | 		role, err := s.State.Role(channel.GuildID, roleID) | ||||||
|  | 		if err != nil || !role.Mentionable { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		content = strings.Replace(content, "<&"+role.ID+">", "@"+role.Name, -1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	content = patternChannels.ReplaceAllStringFunc(content, func(mention string) string { | ||||||
|  | 		channel, err := s.State.Channel(mention[2 : len(mention)-1]) | ||||||
|  | 		if err != nil || channel.Type == ChannelTypeGuildVoice { | ||||||
|  | 			return mention | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return "#" + channel.Name | ||||||
|  | 	}) | ||||||
|  | 	return | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										47
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										47
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,17 +3,26 @@ package discordgo | |||||||
| import ( | import ( | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // customRateLimit holds information for defining a custom rate limit | ||||||
|  | type customRateLimit struct { | ||||||
|  | 	suffix   string | ||||||
|  | 	requests int | ||||||
|  | 	reset    time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
| // RateLimiter holds all ratelimit buckets | // RateLimiter holds all ratelimit buckets | ||||||
| type RateLimiter struct { | type RateLimiter struct { | ||||||
| 	sync.Mutex | 	sync.Mutex | ||||||
| 	global          *int64 | 	global           *int64 | ||||||
| 	buckets         map[string]*Bucket | 	buckets          map[string]*Bucket | ||||||
| 	globalRateLimit time.Duration | 	globalRateLimit  time.Duration | ||||||
|  | 	customRateLimits []*customRateLimit | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewRatelimiter returns a new RateLimiter | // NewRatelimiter returns a new RateLimiter | ||||||
| @@ -22,6 +31,13 @@ func NewRatelimiter() *RateLimiter { | |||||||
| 	return &RateLimiter{ | 	return &RateLimiter{ | ||||||
| 		buckets: make(map[string]*Bucket), | 		buckets: make(map[string]*Bucket), | ||||||
| 		global:  new(int64), | 		global:  new(int64), | ||||||
|  | 		customRateLimits: []*customRateLimit{ | ||||||
|  | 			&customRateLimit{ | ||||||
|  | 				suffix:   "//reactions//", | ||||||
|  | 				requests: 1, | ||||||
|  | 				reset:    200 * time.Millisecond, | ||||||
|  | 			}, | ||||||
|  | 		}, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -40,6 +56,14 @@ func (r *RateLimiter) getBucket(key string) *Bucket { | |||||||
| 		global:    r.global, | 		global:    r.global, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Check if there is a custom ratelimit set for this bucket ID. | ||||||
|  | 	for _, rl := range r.customRateLimits { | ||||||
|  | 		if strings.HasSuffix(b.Key, rl.suffix) { | ||||||
|  | 			b.customRateLimit = rl | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	r.buckets[key] = b | 	r.buckets[key] = b | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -76,13 +100,28 @@ type Bucket struct { | |||||||
| 	limit     int | 	limit     int | ||||||
| 	reset     time.Time | 	reset     time.Time | ||||||
| 	global    *int64 | 	global    *int64 | ||||||
|  |  | ||||||
|  | 	lastReset       time.Time | ||||||
|  | 	customRateLimit *customRateLimit | ||||||
| } | } | ||||||
|  |  | ||||||
| // Release unlocks the bucket and reads the headers to update the buckets ratelimit info | // Release unlocks the bucket and reads the headers to update the buckets ratelimit info | ||||||
| // and locks up the whole thing in case if there's a global ratelimit. | // and locks up the whole thing in case if there's a global ratelimit. | ||||||
| func (b *Bucket) Release(headers http.Header) error { | func (b *Bucket) Release(headers http.Header) error { | ||||||
|  |  | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the bucket uses a custom ratelimiter | ||||||
|  | 	if rl := b.customRateLimit; rl != nil { | ||||||
|  | 		if time.Now().Sub(b.lastReset) >= rl.reset { | ||||||
|  | 			b.remaining = rl.requests - 1 | ||||||
|  | 			b.lastReset = time.Now() | ||||||
|  | 		} | ||||||
|  | 		if b.remaining < 1 { | ||||||
|  | 			b.reset = time.Now().Add(rl.reset) | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if headers == nil { | 	if headers == nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										138
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										138
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -23,6 +23,7 @@ import ( | |||||||
| 	"log" | 	"log" | ||||||
| 	"mime/multipart" | 	"mime/multipart" | ||||||
| 	"net/http" | 	"net/http" | ||||||
|  | 	"net/textproto" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
| @@ -309,8 +310,8 @@ func (s *Session) UserUpdate(email, password, username, avatar, newPassword stri | |||||||
| 	// If left blank, avatar will be set to null/blank | 	// If left blank, avatar will be set to null/blank | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| 		Email       string `json:"email"` | 		Email       string `json:"email,omitempty"` | ||||||
| 		Password    string `json:"password"` | 		Password    string `json:"password,omitempty"` | ||||||
| 		Username    string `json:"username,omitempty"` | 		Username    string `json:"username,omitempty"` | ||||||
| 		Avatar      string `json:"avatar,omitempty"` | 		Avatar      string `json:"avatar,omitempty"` | ||||||
| 		NewPassword string `json:"new_password,omitempty"` | 		NewPassword string `json:"new_password,omitempty"` | ||||||
| @@ -763,7 +764,21 @@ func (s *Session) GuildMember(guildID, userID string) (st *Member, err error) { | |||||||
| // userID    : The ID of a User | // userID    : The ID of a User | ||||||
| func (s *Session) GuildMemberDelete(guildID, userID string) (err error) { | func (s *Session) GuildMemberDelete(guildID, userID string) (err error) { | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("DELETE", EndpointGuildMember(guildID, userID), nil, EndpointGuildMember(guildID, "")) | 	return s.GuildMemberDeleteWithReason(guildID, userID, "") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildMemberDeleteWithReason removes the given user from the given guild. | ||||||
|  | // guildID   : The ID of a Guild. | ||||||
|  | // userID    : The ID of a User | ||||||
|  | // reason    : The reason for the kick | ||||||
|  | func (s *Session) GuildMemberDeleteWithReason(guildID, userID, reason string) (err error) { | ||||||
|  |  | ||||||
|  | 	uri := EndpointGuildMember(guildID, userID) | ||||||
|  | 	if reason != "" { | ||||||
|  | 		uri += "?reason=" + url.QueryEscape(reason) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, err = s.RequestWithBucketID("DELETE", uri, nil, EndpointGuildMember(guildID, "")) | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -1316,6 +1331,8 @@ func (s *Session) ChannelMessageSend(channelID string, content string) (*Message | |||||||
| 	}) | 	}) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"") | ||||||
|  |  | ||||||
| // ChannelMessageSendComplex sends a message to the given channel. | // ChannelMessageSendComplex sends a message to the given channel. | ||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // data      : The message struct to send. | // data      : The message struct to send. | ||||||
| @@ -1326,48 +1343,62 @@ func (s *Session) ChannelMessageSendComplex(channelID string, data *MessageSend) | |||||||
|  |  | ||||||
| 	endpoint := EndpointChannelMessages(channelID) | 	endpoint := EndpointChannelMessages(channelID) | ||||||
|  |  | ||||||
| 	var response []byte | 	// TODO: Remove this when compatibility is not required. | ||||||
|  | 	files := data.Files | ||||||
| 	if data.File != nil { | 	if data.File != nil { | ||||||
|  | 		if files == nil { | ||||||
|  | 			files = []*File{data.File} | ||||||
|  | 		} else { | ||||||
|  | 			err = fmt.Errorf("cannot specify both File and Files") | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var response []byte | ||||||
|  | 	if len(files) > 0 { | ||||||
| 		body := &bytes.Buffer{} | 		body := &bytes.Buffer{} | ||||||
| 		bodywriter := multipart.NewWriter(body) | 		bodywriter := multipart.NewWriter(body) | ||||||
|  |  | ||||||
| 		// What's a better way of doing this? Reflect? Generator? I'm open to suggestions | 		var payload []byte | ||||||
|  | 		payload, err = json.Marshal(data) | ||||||
| 		if data.Content != "" { |  | ||||||
| 			if err = bodywriter.WriteField("content", data.Content); err != nil { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if data.Embed != nil { |  | ||||||
| 			var embed []byte |  | ||||||
| 			embed, err = json.Marshal(data.Embed) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 			err = bodywriter.WriteField("embed", string(embed)) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if data.Tts { |  | ||||||
| 			if err = bodywriter.WriteField("tts", "true"); err != nil { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		var writer io.Writer |  | ||||||
| 		writer, err = bodywriter.CreateFormFile("file", data.File.Name) |  | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		_, err = io.Copy(writer, data.File.Reader) | 		var p io.Writer | ||||||
|  |  | ||||||
|  | 		h := make(textproto.MIMEHeader) | ||||||
|  | 		h.Set("Content-Disposition", `form-data; name="payload_json"`) | ||||||
|  | 		h.Set("Content-Type", "application/json") | ||||||
|  |  | ||||||
|  | 		p, err = bodywriter.CreatePart(h) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		if _, err = p.Write(payload); err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i, file := range files { | ||||||
|  | 			h := make(textproto.MIMEHeader) | ||||||
|  | 			h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file%d"; filename="%s"`, i, quoteEscaper.Replace(file.Name))) | ||||||
|  | 			contentType := file.ContentType | ||||||
|  | 			if contentType == "" { | ||||||
|  | 				contentType = "application/octet-stream" | ||||||
|  | 			} | ||||||
|  | 			h.Set("Content-Type", contentType) | ||||||
|  |  | ||||||
|  | 			p, err = bodywriter.CreatePart(h) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if _, err = io.Copy(p, file.Reader); err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		err = bodywriter.Close() | 		err = bodywriter.Close() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return | 			return | ||||||
| @@ -1685,6 +1716,28 @@ func (s *Session) Gateway() (gateway string, err error) { | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // GatewayBot returns the websocket Gateway address and the recommended number of shards | ||||||
|  | func (s *Session) GatewayBot() (st *GatewayBotResponse, err error) { | ||||||
|  |  | ||||||
|  | 	response, err := s.RequestWithBucketID("GET", EndpointGatewayBot, nil, EndpointGatewayBot) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	err = unmarshal(response, &st) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Ensure the gateway always has a trailing slash. | ||||||
|  | 	// MacOS will fail to connect if we add query params without a trailing slash on the base domain. | ||||||
|  | 	if !strings.HasSuffix(st.URL, "/") { | ||||||
|  | 		st.URL += "/" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
| // Functions specific to Webhooks | // Functions specific to Webhooks | ||||||
|  |  | ||||||
| // WebhookCreate returns a new Webhook. | // WebhookCreate returns a new Webhook. | ||||||
| @@ -1810,14 +1863,9 @@ func (s *Session) WebhookEditWithToken(webhookID, token, name, avatar string) (s | |||||||
|  |  | ||||||
| // WebhookDelete deletes a webhook for a given ID | // WebhookDelete deletes a webhook for a given ID | ||||||
| // webhookID: The ID of a webhook. | // webhookID: The ID of a webhook. | ||||||
| func (s *Session) WebhookDelete(webhookID string) (st *Webhook, err error) { | func (s *Session) WebhookDelete(webhookID string) (err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("DELETE", EndpointWebhook(webhookID), nil, EndpointWebhooks) | 	_, err = s.RequestWithBucketID("DELETE", EndpointWebhook(webhookID), nil, EndpointWebhooks) | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = unmarshal(body, &st) |  | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -1875,6 +1923,16 @@ func (s *Session) MessageReactionRemove(channelID, messageID, emojiID, userID st | |||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MessageReactionsRemoveAll deletes all reactions from a message | ||||||
|  | // channelID : The channel ID | ||||||
|  | // messageID : The message ID. | ||||||
|  | func (s *Session) MessageReactionsRemoveAll(channelID, messageID string) error { | ||||||
|  |  | ||||||
|  | 	_, err := s.RequestWithBucketID("DELETE", EndpointMessageReactionsAll(channelID, messageID), nil, EndpointMessageReactionsAll(channelID, messageID)) | ||||||
|  |  | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
| // MessageReactions gets all the users reactions for a specific emoji. | // MessageReactions gets all the users reactions for a specific emoji. | ||||||
| // channelID : The channel ID. | // channelID : The channel ID. | ||||||
| // messageID : The message ID. | // messageID : The message ID. | ||||||
|   | |||||||
							
								
								
									
										72
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										72
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -42,6 +42,7 @@ type State struct { | |||||||
|  |  | ||||||
| 	guildMap   map[string]*Guild | 	guildMap   map[string]*Guild | ||||||
| 	channelMap map[string]*Channel | 	channelMap map[string]*Channel | ||||||
|  | 	memberMap  map[string]map[string]*Member | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewState creates an empty state. | // NewState creates an empty state. | ||||||
| @@ -59,9 +60,18 @@ func NewState() *State { | |||||||
| 		TrackPresences: true, | 		TrackPresences: true, | ||||||
| 		guildMap:       make(map[string]*Guild), | 		guildMap:       make(map[string]*Guild), | ||||||
| 		channelMap:     make(map[string]*Channel), | 		channelMap:     make(map[string]*Channel), | ||||||
|  | 		memberMap:      make(map[string]map[string]*Member), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (s *State) createMemberMap(guild *Guild) { | ||||||
|  | 	members := make(map[string]*Member) | ||||||
|  | 	for _, m := range guild.Members { | ||||||
|  | 		members[m.User.ID] = m | ||||||
|  | 	} | ||||||
|  | 	s.memberMap[guild.ID] = members | ||||||
|  | } | ||||||
|  |  | ||||||
| // GuildAdd adds a guild to the current world state, or | // GuildAdd adds a guild to the current world state, or | ||||||
| // updates it if it already exists. | // updates it if it already exists. | ||||||
| func (s *State) GuildAdd(guild *Guild) error { | func (s *State) GuildAdd(guild *Guild) error { | ||||||
| @@ -77,6 +87,14 @@ func (s *State) GuildAdd(guild *Guild) error { | |||||||
| 		s.channelMap[c.ID] = c | 		s.channelMap[c.ID] = c | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// If this guild contains a new member slice, we must regenerate the member map so the pointers stay valid | ||||||
|  | 	if guild.Members != nil { | ||||||
|  | 		s.createMemberMap(guild) | ||||||
|  | 	} else if _, ok := s.memberMap[guild.ID]; !ok { | ||||||
|  | 		// Even if we have no new member slice, we still initialize the member map for this guild if it doesn't exist | ||||||
|  | 		s.memberMap[guild.ID] = make(map[string]*Member) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if g, ok := s.guildMap[guild.ID]; ok { | 	if g, ok := s.guildMap[guild.ID]; ok { | ||||||
| 		// We are about to replace `g` in the state with `guild`, but first we need to | 		// We are about to replace `g` in the state with `guild`, but first we need to | ||||||
| 		// make sure we preserve any fields that the `guild` doesn't contain from `g`. | 		// make sure we preserve any fields that the `guild` doesn't contain from `g`. | ||||||
| @@ -271,14 +289,19 @@ func (s *State) MemberAdd(member *Member) error { | |||||||
| 	s.Lock() | 	s.Lock() | ||||||
| 	defer s.Unlock() | 	defer s.Unlock() | ||||||
|  |  | ||||||
| 	for i, m := range guild.Members { | 	members, ok := s.memberMap[member.GuildID] | ||||||
| 		if m.User.ID == member.User.ID { | 	if !ok { | ||||||
| 			guild.Members[i] = member | 		return ErrStateNotFound | ||||||
| 			return nil | 	} | ||||||
| 		} |  | ||||||
|  | 	m, ok := members[member.User.ID] | ||||||
|  | 	if !ok { | ||||||
|  | 		members[member.User.ID] = member | ||||||
|  | 		guild.Members = append(guild.Members, member) | ||||||
|  | 	} else { | ||||||
|  | 		*m = *member // Update the actual data, which will also update the member pointer in the slice | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	guild.Members = append(guild.Members, member) |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -296,6 +319,17 @@ func (s *State) MemberRemove(member *Member) error { | |||||||
| 	s.Lock() | 	s.Lock() | ||||||
| 	defer s.Unlock() | 	defer s.Unlock() | ||||||
|  |  | ||||||
|  | 	members, ok := s.memberMap[member.GuildID] | ||||||
|  | 	if !ok { | ||||||
|  | 		return ErrStateNotFound | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, ok = members[member.User.ID] | ||||||
|  | 	if !ok { | ||||||
|  | 		return ErrStateNotFound | ||||||
|  | 	} | ||||||
|  | 	delete(members, member.User.ID) | ||||||
|  |  | ||||||
| 	for i, m := range guild.Members { | 	for i, m := range guild.Members { | ||||||
| 		if m.User.ID == member.User.ID { | 		if m.User.ID == member.User.ID { | ||||||
| 			guild.Members = append(guild.Members[:i], guild.Members[i+1:]...) | 			guild.Members = append(guild.Members[:i], guild.Members[i+1:]...) | ||||||
| @@ -312,18 +346,17 @@ func (s *State) Member(guildID, userID string) (*Member, error) { | |||||||
| 		return nil, ErrNilState | 		return nil, ErrNilState | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	guild, err := s.Guild(guildID) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	s.RLock() | 	s.RLock() | ||||||
| 	defer s.RUnlock() | 	defer s.RUnlock() | ||||||
|  |  | ||||||
| 	for _, m := range guild.Members { | 	members, ok := s.memberMap[guildID] | ||||||
| 		if m.User.ID == userID { | 	if !ok { | ||||||
| 			return m, nil | 		return nil, ErrStateNotFound | ||||||
| 		} | 	} | ||||||
|  |  | ||||||
|  | 	m, ok := members[userID] | ||||||
|  | 	if ok { | ||||||
|  | 		return m, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, ErrStateNotFound | 	return nil, ErrStateNotFound | ||||||
| @@ -427,7 +460,7 @@ func (s *State) ChannelAdd(channel *Channel) error { | |||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if channel.IsPrivate { | 	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM { | ||||||
| 		s.PrivateChannels = append(s.PrivateChannels, channel) | 		s.PrivateChannels = append(s.PrivateChannels, channel) | ||||||
| 	} else { | 	} else { | ||||||
| 		guild, ok := s.guildMap[channel.GuildID] | 		guild, ok := s.guildMap[channel.GuildID] | ||||||
| @@ -454,7 +487,7 @@ func (s *State) ChannelRemove(channel *Channel) error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if channel.IsPrivate { | 	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM { | ||||||
| 		s.Lock() | 		s.Lock() | ||||||
| 		defer s.Unlock() | 		defer s.Unlock() | ||||||
|  |  | ||||||
| @@ -735,6 +768,7 @@ func (s *State) onReady(se *Session, r *Ready) (err error) { | |||||||
|  |  | ||||||
| 	for _, g := range s.Guilds { | 	for _, g := range s.Guilds { | ||||||
| 		s.guildMap[g.ID] = g | 		s.guildMap[g.ID] = g | ||||||
|  | 		s.createMemberMap(g) | ||||||
|  |  | ||||||
| 		for _, c := range g.Channels { | 		for _, c := range g.Channels { | ||||||
| 			s.channelMap[c.ID] = c | 			s.channelMap[c.ID] = c | ||||||
| @@ -748,8 +782,8 @@ func (s *State) onReady(se *Session, r *Ready) (err error) { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // onInterface handles all events related to states. | // OnInterface handles all events related to states. | ||||||
| func (s *State) onInterface(se *Session, i interface{}) (err error) { | func (s *State) OnInterface(se *Session, i interface{}) (err error) { | ||||||
| 	if s == nil { | 	if s == nil { | ||||||
| 		return ErrNilState | 		return ErrNilState | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										87
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										87
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -50,6 +50,10 @@ type Session struct { | |||||||
| 	// active guilds and the members of the guilds. | 	// active guilds and the members of the guilds. | ||||||
| 	StateEnabled bool | 	StateEnabled bool | ||||||
|  |  | ||||||
|  | 	// Whether or not to call event handlers synchronously. | ||||||
|  | 	// e.g false = launch event handlers in their own goroutines. | ||||||
|  | 	SyncEvents bool | ||||||
|  |  | ||||||
| 	// Exposed but should not be modified by User. | 	// Exposed but should not be modified by User. | ||||||
|  |  | ||||||
| 	// Whether the Data Websocket is ready | 	// Whether the Data Websocket is ready | ||||||
| @@ -78,6 +82,9 @@ type Session struct { | |||||||
| 	// The http client used for REST requests | 	// The http client used for REST requests | ||||||
| 	Client *http.Client | 	Client *http.Client | ||||||
|  |  | ||||||
|  | 	// Stores the last HeartbeatAck that was recieved (in UTC) | ||||||
|  | 	LastHeartbeatAck time.Time | ||||||
|  |  | ||||||
| 	// Event handlers | 	// Event handlers | ||||||
| 	handlersMu   sync.RWMutex | 	handlersMu   sync.RWMutex | ||||||
| 	handlers     map[string][]*eventHandlerInstance | 	handlers     map[string][]*eventHandlerInstance | ||||||
| @@ -141,18 +148,30 @@ type Invite struct { | |||||||
| 	Temporary bool      `json:"temporary"` | 	Temporary bool      `json:"temporary"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ChannelType is the type of a Channel | ||||||
|  | type ChannelType int | ||||||
|  |  | ||||||
|  | // Block contains known ChannelType values | ||||||
|  | const ( | ||||||
|  | 	ChannelTypeGuildText ChannelType = iota | ||||||
|  | 	ChannelTypeDM | ||||||
|  | 	ChannelTypeGuildVoice | ||||||
|  | 	ChannelTypeGroupDM | ||||||
|  | 	ChannelTypeGuildCategory | ||||||
|  | ) | ||||||
|  |  | ||||||
| // A Channel holds all data related to an individual Discord channel. | // A Channel holds all data related to an individual Discord channel. | ||||||
| type Channel struct { | type Channel struct { | ||||||
| 	ID                   string                 `json:"id"` | 	ID                   string                 `json:"id"` | ||||||
| 	GuildID              string                 `json:"guild_id"` | 	GuildID              string                 `json:"guild_id"` | ||||||
| 	Name                 string                 `json:"name"` | 	Name                 string                 `json:"name"` | ||||||
| 	Topic                string                 `json:"topic"` | 	Topic                string                 `json:"topic"` | ||||||
| 	Type                 string                 `json:"type"` | 	Type                 ChannelType            `json:"type"` | ||||||
| 	LastMessageID        string                 `json:"last_message_id"` | 	LastMessageID        string                 `json:"last_message_id"` | ||||||
|  | 	NSFW                 bool                   `json:"nsfw"` | ||||||
| 	Position             int                    `json:"position"` | 	Position             int                    `json:"position"` | ||||||
| 	Bitrate              int                    `json:"bitrate"` | 	Bitrate              int                    `json:"bitrate"` | ||||||
| 	IsPrivate            bool                   `json:"is_private"` | 	Recipients           []*User                `json:"recipient"` | ||||||
| 	Recipient            *User                  `json:"recipient"` |  | ||||||
| 	Messages             []*Message             `json:"-"` | 	Messages             []*Message             `json:"-"` | ||||||
| 	PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"` | 	PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"` | ||||||
| } | } | ||||||
| @@ -292,13 +311,14 @@ type Presence struct { | |||||||
| 	Game   *Game    `json:"game"` | 	Game   *Game    `json:"game"` | ||||||
| 	Nick   string   `json:"nick"` | 	Nick   string   `json:"nick"` | ||||||
| 	Roles  []string `json:"roles"` | 	Roles  []string `json:"roles"` | ||||||
|  | 	Since  *int     `json:"since"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // A Game struct holds the name of the "playing .." game for a user | // A Game struct holds the name of the "playing .." game for a user | ||||||
| type Game struct { | type Game struct { | ||||||
| 	Name string `json:"name"` | 	Name string `json:"name"` | ||||||
| 	Type int    `json:"type"` | 	Type int    `json:"type"` | ||||||
| 	URL  string `json:"url"` | 	URL  string `json:"url,omitempty"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // UnmarshalJSON unmarshals json to Game struct | // UnmarshalJSON unmarshals json to Game struct | ||||||
| @@ -509,6 +529,12 @@ type MessageReaction struct { | |||||||
| 	ChannelID string `json:"channel_id"` | 	ChannelID string `json:"channel_id"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // GatewayBotResponse stores the data for the gateway/bot response | ||||||
|  | type GatewayBotResponse struct { | ||||||
|  | 	URL    string `json:"url"` | ||||||
|  | 	Shards int    `json:"shards"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Constants for the different bit offsets of text channel permissions | // Constants for the different bit offsets of text channel permissions | ||||||
| const ( | const ( | ||||||
| 	PermissionReadMessages = 1 << (iota + 10) | 	PermissionReadMessages = 1 << (iota + 10) | ||||||
| @@ -579,3 +605,56 @@ const ( | |||||||
| 		PermissionManageServer | | 		PermissionManageServer | | ||||||
| 		PermissionAdministrator | 		PermissionAdministrator | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Block contains Discord JSON Error Response codes | ||||||
|  | const ( | ||||||
|  | 	ErrCodeUnknownAccount     = 10001 | ||||||
|  | 	ErrCodeUnknownApplication = 10002 | ||||||
|  | 	ErrCodeUnknownChannel     = 10003 | ||||||
|  | 	ErrCodeUnknownGuild       = 10004 | ||||||
|  | 	ErrCodeUnknownIntegration = 10005 | ||||||
|  | 	ErrCodeUnknownInvite      = 10006 | ||||||
|  | 	ErrCodeUnknownMember      = 10007 | ||||||
|  | 	ErrCodeUnknownMessage     = 10008 | ||||||
|  | 	ErrCodeUnknownOverwrite   = 10009 | ||||||
|  | 	ErrCodeUnknownProvider    = 10010 | ||||||
|  | 	ErrCodeUnknownRole        = 10011 | ||||||
|  | 	ErrCodeUnknownToken       = 10012 | ||||||
|  | 	ErrCodeUnknownUser        = 10013 | ||||||
|  | 	ErrCodeUnknownEmoji       = 10014 | ||||||
|  |  | ||||||
|  | 	ErrCodeBotsCannotUseEndpoint  = 20001 | ||||||
|  | 	ErrCodeOnlyBotsCanUseEndpoint = 20002 | ||||||
|  |  | ||||||
|  | 	ErrCodeMaximumGuildsReached     = 30001 | ||||||
|  | 	ErrCodeMaximumFriendsReached    = 30002 | ||||||
|  | 	ErrCodeMaximumPinsReached       = 30003 | ||||||
|  | 	ErrCodeMaximumGuildRolesReached = 30005 | ||||||
|  | 	ErrCodeTooManyReactions         = 30010 | ||||||
|  |  | ||||||
|  | 	ErrCodeUnauthorized = 40001 | ||||||
|  |  | ||||||
|  | 	ErrCodeMissingAccess                             = 50001 | ||||||
|  | 	ErrCodeInvalidAccountType                        = 50002 | ||||||
|  | 	ErrCodeCannotExecuteActionOnDMChannel            = 50003 | ||||||
|  | 	ErrCodeEmbedCisabled                             = 50004 | ||||||
|  | 	ErrCodeCannotEditFromAnotherUser                 = 50005 | ||||||
|  | 	ErrCodeCannotSendEmptyMessage                    = 50006 | ||||||
|  | 	ErrCodeCannotSendMessagesToThisUser              = 50007 | ||||||
|  | 	ErrCodeCannotSendMessagesInVoiceChannel          = 50008 | ||||||
|  | 	ErrCodeChannelVerificationLevelTooHigh           = 50009 | ||||||
|  | 	ErrCodeOAuth2ApplicationDoesNotHaveBot           = 50010 | ||||||
|  | 	ErrCodeOAuth2ApplicationLimitReached             = 50011 | ||||||
|  | 	ErrCodeInvalidOAuthState                         = 50012 | ||||||
|  | 	ErrCodeMissingPermissions                        = 50013 | ||||||
|  | 	ErrCodeInvalidAuthenticationToken                = 50014 | ||||||
|  | 	ErrCodeNoteTooLong                               = 50015 | ||||||
|  | 	ErrCodeTooFewOrTooManyMessagesToDelete           = 50016 | ||||||
|  | 	ErrCodeCanOnlyPinMessageToOriginatingChannel     = 50019 | ||||||
|  | 	ErrCodeCannotExecuteActionOnSystemMessage        = 50021 | ||||||
|  | 	ErrCodeMessageProvidedTooOldForBulkDelete        = 50034 | ||||||
|  | 	ErrCodeInvalidFormBody                           = 50035 | ||||||
|  | 	ErrCodeInviteAcceptedToGuildApplicationsBotNotIn = 50036 | ||||||
|  |  | ||||||
|  | 	ErrCodeReactionBlocked = 90001 | ||||||
|  | ) | ||||||
|   | |||||||
							
								
								
									
										18
									
								
								vendor/github.com/bwmarrin/discordgo/user.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/bwmarrin/discordgo/user.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,6 +1,9 @@ | |||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
| import "fmt" | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
| // A User stores all data for an individual Discord user. | // A User stores all data for an individual Discord user. | ||||||
| type User struct { | type User struct { | ||||||
| @@ -24,3 +27,16 @@ func (u *User) String() string { | |||||||
| func (u *User) Mention() string { | func (u *User) Mention() string { | ||||||
| 	return fmt.Sprintf("<@%s>", u.ID) | 	return fmt.Sprintf("<@%s>", u.ID) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // AvatarURL returns a URL to the user's avatar. | ||||||
|  | //		size:     The size of the user's avatar as a power of two | ||||||
|  | func (u *User) AvatarURL(size string) string { | ||||||
|  | 	var URL string | ||||||
|  | 	if strings.HasPrefix(u.Avatar, "a_") { | ||||||
|  | 		URL = EndpointUserAvatarAnimated(u.ID, u.Avatar) | ||||||
|  | 	} else { | ||||||
|  | 		URL = EndpointUserAvatar(u.ID, u.Avatar) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return URL + "?size=" + size | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										13
									
								
								vendor/github.com/bwmarrin/discordgo/voice.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/bwmarrin/discordgo/voice.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -796,7 +796,7 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// For now, skip anything except audio. | 		// For now, skip anything except audio. | ||||||
| 		if rlen < 12 || recvbuf[0] != 0x80 { | 		if rlen < 12 || (recvbuf[0] != 0x80 && recvbuf[0] != 0x90) { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| @@ -810,8 +810,17 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct | |||||||
| 		copy(nonce[:], recvbuf[0:12]) | 		copy(nonce[:], recvbuf[0:12]) | ||||||
| 		p.Opus, _ = secretbox.Open(nil, recvbuf[12:rlen], &nonce, &v.op4.SecretKey) | 		p.Opus, _ = secretbox.Open(nil, recvbuf[12:rlen], &nonce, &v.op4.SecretKey) | ||||||
|  |  | ||||||
|  | 		if len(p.Opus) > 8 && recvbuf[0] == 0x90 { | ||||||
|  | 			// Extension bit is set, first 8 bytes is the extended header | ||||||
|  | 			p.Opus = p.Opus[8:] | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		if c != nil { | 		if c != nil { | ||||||
| 			c <- &p | 			select { | ||||||
|  | 			case c <- &p: | ||||||
|  | 			case <-close: | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										58
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										58
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -15,7 +15,6 @@ import ( | |||||||
| 	"compress/zlib" | 	"compress/zlib" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"runtime" | 	"runtime" | ||||||
| @@ -87,7 +86,7 @@ func (s *Session) Open() (err error) { | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// Add the version and encoding to the URL | 		// Add the version and encoding to the URL | ||||||
| 		s.gateway = fmt.Sprintf("%s?v=5&encoding=json", s.gateway) | 		s.gateway = s.gateway + "?v=" + APIVersion + "&encoding=json" | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	header := http.Header{} | 	header := http.Header{} | ||||||
| @@ -131,6 +130,7 @@ func (s *Session) Open() (err error) { | |||||||
| 	// lock. | 	// lock. | ||||||
| 	s.listening = make(chan interface{}) | 	s.listening = make(chan interface{}) | ||||||
| 	go s.listen(s.wsConn, s.listening) | 	go s.listen(s.wsConn, s.listening) | ||||||
|  | 	s.LastHeartbeatAck = time.Now().UTC() | ||||||
|  |  | ||||||
| 	s.Unlock() | 	s.Unlock() | ||||||
|  |  | ||||||
| @@ -199,10 +199,13 @@ type helloOp struct { | |||||||
| 	Trace             []string      `json:"_trace"` | 	Trace             []string      `json:"_trace"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // FailedHeartbeatAcks is the Number of heartbeat intervals to wait until forcing a connection restart. | ||||||
|  | const FailedHeartbeatAcks time.Duration = 5 * time.Millisecond | ||||||
|  |  | ||||||
| // heartbeat sends regular heartbeats to Discord so it knows the client | // heartbeat sends regular heartbeats to Discord so it knows the client | ||||||
| // is still connected.  If you do not send these heartbeats Discord will | // is still connected.  If you do not send these heartbeats Discord will | ||||||
| // disconnect the websocket connection after a few seconds. | // disconnect the websocket connection after a few seconds. | ||||||
| func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}, i time.Duration) { | func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}, heartbeatIntervalMsec time.Duration) { | ||||||
|  |  | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
| @@ -211,20 +214,26 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{} | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var err error | 	var err error | ||||||
| 	ticker := time.NewTicker(i * time.Millisecond) | 	ticker := time.NewTicker(heartbeatIntervalMsec * time.Millisecond) | ||||||
| 	defer ticker.Stop() | 	defer ticker.Stop() | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
|  | 		s.RLock() | ||||||
|  | 		last := s.LastHeartbeatAck | ||||||
|  | 		s.RUnlock() | ||||||
| 		sequence := atomic.LoadInt64(s.sequence) | 		sequence := atomic.LoadInt64(s.sequence) | ||||||
| 		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", sequence) | 		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", sequence) | ||||||
| 		s.wsMutex.Lock() | 		s.wsMutex.Lock() | ||||||
| 		err = wsConn.WriteJSON(heartbeatOp{1, sequence}) | 		err = wsConn.WriteJSON(heartbeatOp{1, sequence}) | ||||||
| 		s.wsMutex.Unlock() | 		s.wsMutex.Unlock() | ||||||
| 		if err != nil { | 		if err != nil || time.Now().UTC().Sub(last) > (heartbeatIntervalMsec*FailedHeartbeatAcks) { | ||||||
| 			s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err) | 			if err != nil { | ||||||
| 			s.Lock() | 				s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err) | ||||||
| 			s.DataReady = false | 			} else { | ||||||
| 			s.Unlock() | 				s.log(LogError, "haven't gotten a heartbeat ACK in %v, triggering a reconnection", time.Now().UTC().Sub(last)) | ||||||
|  | 			} | ||||||
|  | 			s.Close() | ||||||
|  | 			s.reconnect() | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		s.Lock() | 		s.Lock() | ||||||
| @@ -241,8 +250,10 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{} | |||||||
| } | } | ||||||
|  |  | ||||||
| type updateStatusData struct { | type updateStatusData struct { | ||||||
| 	IdleSince *int  `json:"idle_since"` | 	IdleSince *int   `json:"since"` | ||||||
| 	Game      *Game `json:"game"` | 	Game      *Game  `json:"game"` | ||||||
|  | 	AFK       bool   `json:"afk"` | ||||||
|  | 	Status    string `json:"status"` | ||||||
| } | } | ||||||
|  |  | ||||||
| type updateStatusOp struct { | type updateStatusOp struct { | ||||||
| @@ -265,7 +276,10 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err | |||||||
| 		return ErrWSNotFound | 		return ErrWSNotFound | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var usd updateStatusData | 	usd := updateStatusData{ | ||||||
|  | 		Status: "online", | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if idle > 0 { | 	if idle > 0 { | ||||||
| 		usd.IdleSince = &idle | 		usd.IdleSince = &idle | ||||||
| 	} | 	} | ||||||
| @@ -398,7 +412,10 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 	// Reconnect | 	// Reconnect | ||||||
| 	// Must immediately disconnect from gateway and reconnect to new gateway. | 	// Must immediately disconnect from gateway and reconnect to new gateway. | ||||||
| 	if e.Operation == 7 { | 	if e.Operation == 7 { | ||||||
| 		// TODO | 		s.log(LogInformational, "Closing and reconnecting in response to Op7") | ||||||
|  | 		s.Close() | ||||||
|  | 		s.reconnect() | ||||||
|  | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Invalid Session | 	// Invalid Session | ||||||
| @@ -426,6 +443,14 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if e.Operation == 11 { | ||||||
|  | 		s.Lock() | ||||||
|  | 		s.LastHeartbeatAck = time.Now().UTC() | ||||||
|  | 		s.Unlock() | ||||||
|  | 		s.log(LogInformational, "got heartbeat ACK") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// Do not try to Dispatch a non-Dispatch Message | 	// Do not try to Dispatch a non-Dispatch Message | ||||||
| 	if e.Operation != 0 { | 	if e.Operation != 0 { | ||||||
| 		// But we probably should be doing something with them. | 		// But we probably should be doing something with them. | ||||||
| @@ -688,6 +713,13 @@ func (s *Session) reconnect() { | |||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
|  | 			// Certain race conditions can call reconnect() twice. If this happens, we | ||||||
|  | 			// just break out of the reconnect loop | ||||||
|  | 			if err == ErrWSAlreadyOpen { | ||||||
|  | 				s.log(LogInformational, "Websocket already exists, no need to reconnect") | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
| 			s.log(LogError, "error reconnecting to gateway, %s", err) | 			s.log(LogError, "error reconnecting to gateway, %s", err) | ||||||
|  |  | ||||||
| 			<-time.After(wait * time.Second) | 			<-time.After(wait * time.Second) | ||||||
|   | |||||||
							
								
								
									
										197
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										197
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -191,7 +191,11 @@ func (bot *BotAPI) UploadFile(endpoint string, params map[string]string, fieldna | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var apiResp APIResponse | 	var apiResp APIResponse | ||||||
| 	json.Unmarshal(bytes, &apiResp) |  | ||||||
|  | 	err = json.Unmarshal(bytes, &apiResp) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if !apiResp.Ok { | 	if !apiResp.Ok { | ||||||
| 		return APIResponse{}, errors.New(apiResp.Description) | 		return APIResponse{}, errors.New(apiResp.Description) | ||||||
| @@ -438,14 +442,7 @@ func (bot *BotAPI) SetWebhook(config WebhookConfig) (APIResponse, error) { | |||||||
| 		return APIResponse{}, err | 		return APIResponse{}, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var apiResp APIResponse | 	return resp, nil | ||||||
| 	json.Unmarshal(resp.Result, &apiResp) |  | ||||||
|  |  | ||||||
| 	if bot.Debug { |  | ||||||
| 		log.Printf("setWebhook resp: %+v\n", apiResp) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return apiResp, nil |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // GetWebhookInfo allows you to fetch information about a webhook and if | // GetWebhookInfo allows you to fetch information about a webhook and if | ||||||
| @@ -550,7 +547,7 @@ func (bot *BotAPI) AnswerCallbackQuery(config CallbackConfig) (APIResponse, erro | |||||||
| // KickChatMember kicks a user from a chat. Note that this only will work | // KickChatMember kicks a user from a chat. Note that this only will work | ||||||
| // in supergroups, and requires the bot to be an admin. Also note they | // in supergroups, and requires the bot to be an admin. Also note they | ||||||
| // will be unable to rejoin until they are unbanned. | // will be unable to rejoin until they are unbanned. | ||||||
| func (bot *BotAPI) KickChatMember(config ChatMemberConfig) (APIResponse, error) { | func (bot *BotAPI) KickChatMember(config KickChatMemberConfig) (APIResponse, error) { | ||||||
| 	v := url.Values{} | 	v := url.Values{} | ||||||
|  |  | ||||||
| 	if config.SuperGroupUsername == "" { | 	if config.SuperGroupUsername == "" { | ||||||
| @@ -560,6 +557,10 @@ func (bot *BotAPI) KickChatMember(config ChatMemberConfig) (APIResponse, error) | |||||||
| 	} | 	} | ||||||
| 	v.Add("user_id", strconv.Itoa(config.UserID)) | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if config.UntilDate != 0 { | ||||||
|  | 		v.Add("until_date", strconv.FormatInt(config.UntilDate, 10)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	bot.debugLog("kickChatMember", v, nil) | 	bot.debugLog("kickChatMember", v, nil) | ||||||
|  |  | ||||||
| 	return bot.MakeRequest("kickChatMember", v) | 	return bot.MakeRequest("kickChatMember", v) | ||||||
| @@ -677,14 +678,16 @@ func (bot *BotAPI) GetChatMember(config ChatConfigWithUser) (ChatMember, error) | |||||||
| } | } | ||||||
|  |  | ||||||
| // UnbanChatMember unbans a user from a chat. Note that this only will work | // UnbanChatMember unbans a user from a chat. Note that this only will work | ||||||
| // in supergroups, and requires the bot to be an admin. | // in supergroups and channels, and requires the bot to be an admin. | ||||||
| func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) { | func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) { | ||||||
| 	v := url.Values{} | 	v := url.Values{} | ||||||
|  |  | ||||||
| 	if config.SuperGroupUsername == "" { | 	if config.SuperGroupUsername != "" { | ||||||
| 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) |  | ||||||
| 	} else { |  | ||||||
| 		v.Add("chat_id", config.SuperGroupUsername) | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
| 	} | 	} | ||||||
| 	v.Add("user_id", strconv.Itoa(config.UserID)) | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
| @@ -693,6 +696,82 @@ func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) | |||||||
| 	return bot.MakeRequest("unbanChatMember", v) | 	return bot.MakeRequest("unbanChatMember", v) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // RestrictChatMember to restrict a user in a supergroup. The bot must be an | ||||||
|  | //administrator in the supergroup for this to work and must have the | ||||||
|  | //appropriate admin rights. Pass True for all boolean parameters to lift | ||||||
|  | //restrictions from a user. Returns True on success. | ||||||
|  | func (bot *BotAPI) RestrictChatMember(config RestrictChatMemberConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} | ||||||
|  | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if &config.CanSendMessages != nil { | ||||||
|  | 		v.Add("can_send_messages", strconv.FormatBool(*config.CanSendMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanSendMediaMessages != nil { | ||||||
|  | 		v.Add("can_send_media_messages", strconv.FormatBool(*config.CanSendMediaMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanSendOtherMessages != nil { | ||||||
|  | 		v.Add("can_send_other_messages", strconv.FormatBool(*config.CanSendOtherMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanAddWebPagePreviews != nil { | ||||||
|  | 		v.Add("can_add_web_page_previews", strconv.FormatBool(*config.CanAddWebPagePreviews)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("restrictChatMember", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("restrictChatMember", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (bot *BotAPI) PromoteChatMember(config PromoteChatMemberConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} | ||||||
|  | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if &config.CanChangeInfo != nil { | ||||||
|  | 		v.Add("can_change_info", strconv.FormatBool(*config.CanChangeInfo)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPostMessages != nil { | ||||||
|  | 		v.Add("can_post_messages", strconv.FormatBool(*config.CanPostMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanEditMessages != nil { | ||||||
|  | 		v.Add("can_edit_messages", strconv.FormatBool(*config.CanEditMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanDeleteMessages != nil { | ||||||
|  | 		v.Add("can_delete_messages", strconv.FormatBool(*config.CanDeleteMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanInviteUsers != nil { | ||||||
|  | 		v.Add("can_invite_users", strconv.FormatBool(*config.CanInviteUsers)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanRestrictMembers != nil { | ||||||
|  | 		v.Add("can_restrict_members", strconv.FormatBool(*config.CanRestrictMembers)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPinMessages != nil { | ||||||
|  | 		v.Add("can_pin_messages", strconv.FormatBool(*config.CanPinMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPromoteMembers != nil { | ||||||
|  | 		v.Add("can_promote_members", strconv.FormatBool(*config.CanPromoteMembers)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("promoteChatMember", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("promoteChatMember", v) | ||||||
|  | } | ||||||
|  |  | ||||||
| // GetGameHighScores allows you to get the high scores for a game. | // GetGameHighScores allows you to get the high scores for a game. | ||||||
| func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHighScore, error) { | func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHighScore, error) { | ||||||
| 	v, _ := config.values() | 	v, _ := config.values() | ||||||
| @@ -707,3 +786,93 @@ func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHigh | |||||||
|  |  | ||||||
| 	return highScores, err | 	return highScores, err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // AnswerShippingQuery allows you to reply to Update with shipping_query parameter. | ||||||
|  | func (bot *BotAPI) AnswerShippingQuery(config ShippingConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("shipping_query_id", config.ShippingQueryID) | ||||||
|  | 	v.Add("ok", strconv.FormatBool(config.OK)) | ||||||
|  | 	if config.OK == true { | ||||||
|  | 		data, err := json.Marshal(config.ShippingOptions) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return APIResponse{}, err | ||||||
|  | 		} | ||||||
|  | 		v.Add("shipping_options", string(data)) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("error_message", config.ErrorMessage) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("answerShippingQuery", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("answerShippingQuery", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AnswerPreCheckoutQuery allows you to reply to Update with pre_checkout_query. | ||||||
|  | func (bot *BotAPI) AnswerPreCheckoutQuery(config PreCheckoutConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("pre_checkout_query_id", config.PreCheckoutQueryID) | ||||||
|  | 	v.Add("ok", strconv.FormatBool(config.OK)) | ||||||
|  | 	if config.OK != true { | ||||||
|  | 		v.Add("error", config.ErrorMessage) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("answerPreCheckoutQuery", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("answerPreCheckoutQuery", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DeleteMessage deletes a message in a chat | ||||||
|  | func (bot *BotAPI) DeleteMessage(config DeleteMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetInviteLink get InviteLink for a chat | ||||||
|  | func (bot *BotAPI) GetInviteLink(config ChatConfig) (string, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername == "" { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	resp, err := bot.MakeRequest("exportChatInviteLink", v) | ||||||
|  |  | ||||||
|  | 	var inviteLink string | ||||||
|  | 	err = json.Unmarshal(resp.Result, &inviteLink) | ||||||
|  |  | ||||||
|  | 	return inviteLink, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pin message in supergroup | ||||||
|  | func (bot *BotAPI) PinChatMessage(config PinChatMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Unpin message in supergroup | ||||||
|  | func (bot *BotAPI) UnpinChatMessage(config UnpinChatMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
							
								
								
									
										243
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										243
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -349,6 +349,7 @@ func (config AudioConfig) method() string { | |||||||
| // DocumentConfig contains information about a SendDocument request. | // DocumentConfig contains information about a SendDocument request. | ||||||
| type DocumentConfig struct { | type DocumentConfig struct { | ||||||
| 	BaseFile | 	BaseFile | ||||||
|  | 	Caption string | ||||||
| } | } | ||||||
|  |  | ||||||
| // values returns a url.Values representation of DocumentConfig. | // values returns a url.Values representation of DocumentConfig. | ||||||
| @@ -359,6 +360,9 @@ func (config DocumentConfig) values() (url.Values, error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	v.Add(config.name(), config.FileID) | 	v.Add(config.name(), config.FileID) | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		v.Add("caption", config.Caption) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return v, nil | 	return v, nil | ||||||
| } | } | ||||||
| @@ -367,6 +371,10 @@ func (config DocumentConfig) values() (url.Values, error) { | |||||||
| func (config DocumentConfig) params() (map[string]string, error) { | func (config DocumentConfig) params() (map[string]string, error) { | ||||||
| 	params, _ := config.BaseFile.params() | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -443,6 +451,10 @@ func (config VideoConfig) values() (url.Values, error) { | |||||||
| func (config VideoConfig) params() (map[string]string, error) { | func (config VideoConfig) params() (map[string]string, error) { | ||||||
| 	params, _ := config.BaseFile.params() | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -456,6 +468,57 @@ func (config VideoConfig) method() string { | |||||||
| 	return "sendVideo" | 	return "sendVideo" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // VideoNoteConfig contains information about a SendVideoNote request. | ||||||
|  | type VideoNoteConfig struct { | ||||||
|  | 	BaseFile | ||||||
|  | 	Duration int | ||||||
|  | 	Length   int | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // values returns a url.Values representation of VideoNoteConfig. | ||||||
|  | func (config VideoNoteConfig) values() (url.Values, error) { | ||||||
|  | 	v, err := config.BaseChat.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	v.Add(config.name(), config.FileID) | ||||||
|  | 	if config.Duration != 0 { | ||||||
|  | 		v.Add("duration", strconv.Itoa(config.Duration)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Telegram API seems to have a bug, if no length is provided or it is 0, it will send an error response | ||||||
|  | 	if config.Length != 0 { | ||||||
|  | 		v.Add("length", strconv.Itoa(config.Length)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // params returns a map[string]string representation of VideoNoteConfig. | ||||||
|  | func (config VideoNoteConfig) params() (map[string]string, error) { | ||||||
|  | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Length != 0 { | ||||||
|  | 		params["length"] = strconv.Itoa(config.Length) | ||||||
|  | 	} | ||||||
|  | 	if config.Duration != 0 { | ||||||
|  | 		params["duration"] = strconv.Itoa(config.Duration) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return params, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // name returns the field name for the VideoNote. | ||||||
|  | func (config VideoNoteConfig) name() string { | ||||||
|  | 	return "video_note" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // method returns Telegram API method name for sending VideoNote. | ||||||
|  | func (config VideoNoteConfig) method() string { | ||||||
|  | 	return "sendVideoNote" | ||||||
|  | } | ||||||
|  |  | ||||||
| // VoiceConfig contains information about a SendVoice request. | // VoiceConfig contains information about a SendVoice request. | ||||||
| type VoiceConfig struct { | type VoiceConfig struct { | ||||||
| 	BaseFile | 	BaseFile | ||||||
| @@ -474,6 +537,9 @@ func (config VoiceConfig) values() (url.Values, error) { | |||||||
| 	if config.Duration != 0 { | 	if config.Duration != 0 { | ||||||
| 		v.Add("duration", strconv.Itoa(config.Duration)) | 		v.Add("duration", strconv.Itoa(config.Duration)) | ||||||
| 	} | 	} | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		v.Add("caption", config.Caption) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return v, nil | 	return v, nil | ||||||
| } | } | ||||||
| @@ -485,6 +551,9 @@ func (config VoiceConfig) params() (map[string]string, error) { | |||||||
| 	if config.Duration != 0 { | 	if config.Duration != 0 { | ||||||
| 		params["duration"] = strconv.Itoa(config.Duration) | 		params["duration"] = strconv.Itoa(config.Duration) | ||||||
| 	} | 	} | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
| @@ -814,9 +883,39 @@ type CallbackConfig struct { | |||||||
| type ChatMemberConfig struct { | type ChatMemberConfig struct { | ||||||
| 	ChatID             int64 | 	ChatID             int64 | ||||||
| 	SuperGroupUsername string | 	SuperGroupUsername string | ||||||
|  | 	ChannelUsername    string | ||||||
| 	UserID             int | 	UserID             int | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // KickChatMemberConfig contains extra fields to kick user | ||||||
|  | type KickChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	UntilDate int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RestrictChatMemberConfig contains fields to restrict members of chat | ||||||
|  | type RestrictChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	UntilDate             int64 | ||||||
|  | 	CanSendMessages       *bool | ||||||
|  | 	CanSendMediaMessages  *bool | ||||||
|  | 	CanSendOtherMessages  *bool | ||||||
|  | 	CanAddWebPagePreviews *bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PromoteChatMemberConfig contains fields to promote members of chat | ||||||
|  | type PromoteChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	CanChangeInfo      *bool | ||||||
|  | 	CanPostMessages    *bool | ||||||
|  | 	CanEditMessages    *bool | ||||||
|  | 	CanDeleteMessages  *bool | ||||||
|  | 	CanInviteUsers     *bool | ||||||
|  | 	CanRestrictMembers *bool | ||||||
|  | 	CanPinMessages     *bool | ||||||
|  | 	CanPromoteMembers  *bool | ||||||
|  | } | ||||||
|  |  | ||||||
| // ChatConfig contains information about getting information on a chat. | // ChatConfig contains information about getting information on a chat. | ||||||
| type ChatConfig struct { | type ChatConfig struct { | ||||||
| 	ChatID             int64 | 	ChatID             int64 | ||||||
| @@ -830,3 +929,147 @@ type ChatConfigWithUser struct { | |||||||
| 	SuperGroupUsername string | 	SuperGroupUsername string | ||||||
| 	UserID             int | 	UserID             int | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // InvoiceConfig contains information for sendInvoice request. | ||||||
|  | type InvoiceConfig struct { | ||||||
|  | 	BaseChat | ||||||
|  | 	Title               string          // required | ||||||
|  | 	Description         string          // required | ||||||
|  | 	Payload             string          // required | ||||||
|  | 	ProviderToken       string          // required | ||||||
|  | 	StartParameter      string          // required | ||||||
|  | 	Currency            string          // required | ||||||
|  | 	Prices              *[]LabeledPrice // required | ||||||
|  | 	PhotoURL            string | ||||||
|  | 	PhotoSize           int | ||||||
|  | 	PhotoWidth          int | ||||||
|  | 	PhotoHeight         int | ||||||
|  | 	NeedName            bool | ||||||
|  | 	NeedPhoneNumber     bool | ||||||
|  | 	NeedEmail           bool | ||||||
|  | 	NeedShippingAddress bool | ||||||
|  | 	IsFlexible          bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config InvoiceConfig) values() (url.Values, error) { | ||||||
|  | 	v, err := config.BaseChat.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  | 	v.Add("title", config.Title) | ||||||
|  | 	v.Add("description", config.Description) | ||||||
|  | 	v.Add("payload", config.Payload) | ||||||
|  | 	v.Add("provider_token", config.ProviderToken) | ||||||
|  | 	v.Add("start_parameter", config.StartParameter) | ||||||
|  | 	v.Add("currency", config.Currency) | ||||||
|  | 	data, err := json.Marshal(config.Prices) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  | 	v.Add("prices", string(data)) | ||||||
|  | 	if config.PhotoURL != "" { | ||||||
|  | 		v.Add("photo_url", config.PhotoURL) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoSize != 0 { | ||||||
|  | 		v.Add("photo_size", strconv.Itoa(config.PhotoSize)) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoWidth != 0 { | ||||||
|  | 		v.Add("photo_width", strconv.Itoa(config.PhotoWidth)) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoHeight != 0 { | ||||||
|  | 		v.Add("photo_height", strconv.Itoa(config.PhotoHeight)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedName != false { | ||||||
|  | 		v.Add("need_name", strconv.FormatBool(config.NeedName)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedPhoneNumber != false { | ||||||
|  | 		v.Add("need_phone_number", strconv.FormatBool(config.NeedPhoneNumber)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedEmail != false { | ||||||
|  | 		v.Add("need_email", strconv.FormatBool(config.NeedEmail)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedShippingAddress != false { | ||||||
|  | 		v.Add("need_shipping_address", strconv.FormatBool(config.NeedShippingAddress)) | ||||||
|  | 	} | ||||||
|  | 	if config.IsFlexible != false { | ||||||
|  | 		v.Add("is_flexible", strconv.FormatBool(config.IsFlexible)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config InvoiceConfig) method() string { | ||||||
|  | 	return "sendInvoice" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingConfig contains information for answerShippingQuery request. | ||||||
|  | type ShippingConfig struct { | ||||||
|  | 	ShippingQueryID string // required | ||||||
|  | 	OK              bool   // required | ||||||
|  | 	ShippingOptions *[]ShippingOption | ||||||
|  | 	ErrorMessage    string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PreCheckoutConfig conatins information for answerPreCheckoutQuery request. | ||||||
|  | type PreCheckoutConfig struct { | ||||||
|  | 	PreCheckoutQueryID string // required | ||||||
|  | 	OK                 bool   // required | ||||||
|  | 	ErrorMessage       string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DeleteMessageConfig contains information of a message in a chat to delete. | ||||||
|  | type DeleteMessageConfig struct { | ||||||
|  | 	ChatID    int64 | ||||||
|  | 	MessageID int | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config DeleteMessageConfig) method() string { | ||||||
|  | 	return "deleteMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config DeleteMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	v.Add("message_id", strconv.Itoa(config.MessageID)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PinChatMessageConfig contains information of a message in a chat to pin. | ||||||
|  | type PinChatMessageConfig struct { | ||||||
|  | 	ChatID int64 | ||||||
|  | 	MessageID int | ||||||
|  | 	DisableNotification bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config PinChatMessageConfig) method() string { | ||||||
|  | 	return "pinChatMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config PinChatMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	v.Add("message_id", strconv.Itoa(config.MessageID)) | ||||||
|  | 	v.Add("disable_notification", strconv.FormatBool(config.DisableNotification)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnpinChatMessageConfig contains information of chat to unpin. | ||||||
|  | type UnpinChatMessageConfig struct { | ||||||
|  | 	ChatID int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config UnpinChatMessageConfig) method() string { | ||||||
|  | 	return "unpinChatMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config UnpinChatMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
							
								
								
									
										44
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										44
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -194,6 +194,37 @@ func NewVideoShare(chatID int64, fileID string) VideoConfig { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // NewVideoNoteUpload creates a new video note uploader. | ||||||
|  | // | ||||||
|  | // chatID is where to send it, file is a string path to the file, | ||||||
|  | // FileReader, or FileBytes. | ||||||
|  | func NewVideoNoteUpload(chatID int64, length int, file interface{}) VideoNoteConfig { | ||||||
|  | 	return VideoNoteConfig{ | ||||||
|  | 		BaseFile: BaseFile{ | ||||||
|  | 			BaseChat:    BaseChat{ChatID: chatID}, | ||||||
|  | 			File:        file, | ||||||
|  | 			UseExisting: false, | ||||||
|  | 		}, | ||||||
|  | 		Length: length, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewVideoNoteShare shares an existing video. | ||||||
|  | // You may use this to reshare an existing video without reuploading it. | ||||||
|  | // | ||||||
|  | // chatID is where to send it, fileID is the ID of the video | ||||||
|  | // already uploaded. | ||||||
|  | func NewVideoNoteShare(chatID int64, length int, fileID string) VideoNoteConfig { | ||||||
|  | 	return VideoNoteConfig{ | ||||||
|  | 		BaseFile: BaseFile{ | ||||||
|  | 			BaseChat:    BaseChat{ChatID: chatID}, | ||||||
|  | 			FileID:      fileID, | ||||||
|  | 			UseExisting: true, | ||||||
|  | 		}, | ||||||
|  | 		Length: length, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewVoiceUpload creates a new voice uploader. | // NewVoiceUpload creates a new voice uploader. | ||||||
| // | // | ||||||
| // chatID is where to send it, file is a string path to the file, | // chatID is where to send it, file is a string path to the file, | ||||||
| @@ -609,3 +640,16 @@ func NewCallbackWithAlert(id, text string) CallbackConfig { | |||||||
| 		ShowAlert:       true, | 		ShowAlert:       true, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // NewInvoice created a new Invoice request to the user. | ||||||
|  | func NewInvoice(chatID int64, title, description, payload, providerToken, startParameter, currency string, prices *[]LabeledPrice) InvoiceConfig { | ||||||
|  | 	return InvoiceConfig{ | ||||||
|  | 		BaseChat:       BaseChat{ChatID: chatID}, | ||||||
|  | 		Title:          title, | ||||||
|  | 		Description:    description, | ||||||
|  | 		Payload:        payload, | ||||||
|  | 		ProviderToken:  providerToken, | ||||||
|  | 		StartParameter: startParameter, | ||||||
|  | 		Currency:       currency, | ||||||
|  | 		Prices:         prices} | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										205
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										205
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -35,6 +35,8 @@ type Update struct { | |||||||
| 	InlineQuery        *InlineQuery        `json:"inline_query"` | 	InlineQuery        *InlineQuery        `json:"inline_query"` | ||||||
| 	ChosenInlineResult *ChosenInlineResult `json:"chosen_inline_result"` | 	ChosenInlineResult *ChosenInlineResult `json:"chosen_inline_result"` | ||||||
| 	CallbackQuery      *CallbackQuery      `json:"callback_query"` | 	CallbackQuery      *CallbackQuery      `json:"callback_query"` | ||||||
|  | 	ShippingQuery      *ShippingQuery      `json:"shipping_query"` | ||||||
|  | 	PreCheckoutQuery   *PreCheckoutQuery   `json:"pre_checkout_query"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // UpdatesChannel is the channel for getting updates. | // UpdatesChannel is the channel for getting updates. | ||||||
| @@ -49,10 +51,11 @@ func (ch UpdatesChannel) Clear() { | |||||||
|  |  | ||||||
| // User is a user on Telegram. | // User is a user on Telegram. | ||||||
| type User struct { | type User struct { | ||||||
| 	ID        int    `json:"id"` | 	ID           int    `json:"id"` | ||||||
| 	FirstName string `json:"first_name"` | 	FirstName    string `json:"first_name"` | ||||||
| 	LastName  string `json:"last_name"` // optional | 	LastName     string `json:"last_name"`     // optional | ||||||
| 	UserName  string `json:"username"`  // optional | 	UserName     string `json:"username"`      // optional | ||||||
|  | 	LanguageCode string `json:"language_code"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // String displays a simple text version of a user. | // String displays a simple text version of a user. | ||||||
| @@ -78,15 +81,24 @@ type GroupChat struct { | |||||||
| 	Title string `json:"title"` | 	Title string `json:"title"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ChatPhoto represents a chat photo. | ||||||
|  | type ChatPhoto struct { | ||||||
|  | 	SmallFileID string `json:"small_file_id"` | ||||||
|  | 	BigFileID   string `json:"big_file_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Chat contains information about the place a message was sent. | // Chat contains information about the place a message was sent. | ||||||
| type Chat struct { | type Chat struct { | ||||||
| 	ID                  int64  `json:"id"` | 	ID                  int64      `json:"id"` | ||||||
| 	Type                string `json:"type"` | 	Type                string     `json:"type"` | ||||||
| 	Title               string `json:"title"`                          // optional | 	Title               string     `json:"title"`                          // optional | ||||||
| 	UserName            string `json:"username"`                       // optional | 	UserName            string     `json:"username"`                       // optional | ||||||
| 	FirstName           string `json:"first_name"`                     // optional | 	FirstName           string     `json:"first_name"`                     // optional | ||||||
| 	LastName            string `json:"last_name"`                      // optional | 	LastName            string     `json:"last_name"`                      // optional | ||||||
| 	AllMembersAreAdmins bool   `json:"all_members_are_administrators"` // optional | 	AllMembersAreAdmins bool       `json:"all_members_are_administrators"` // optional | ||||||
|  | 	Photo               *ChatPhoto `json:"photo"` | ||||||
|  | 	Description         string     `json:"description,omitempty"` // optional | ||||||
|  | 	InviteLink          string     `json:"invite_link,omitempty"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // IsPrivate returns if the Chat is a private conversation. | // IsPrivate returns if the Chat is a private conversation. | ||||||
| @@ -117,40 +129,43 @@ func (c Chat) ChatConfig() ChatConfig { | |||||||
| // Message is returned by almost every request, and contains data about | // Message is returned by almost every request, and contains data about | ||||||
| // almost anything. | // almost anything. | ||||||
| type Message struct { | type Message struct { | ||||||
| 	MessageID             int              `json:"message_id"` | 	MessageID             int                `json:"message_id"` | ||||||
| 	From                  *User            `json:"from"` // optional | 	From                  *User              `json:"from"` // optional | ||||||
| 	Date                  int              `json:"date"` | 	Date                  int                `json:"date"` | ||||||
| 	Chat                  *Chat            `json:"chat"` | 	Chat                  *Chat              `json:"chat"` | ||||||
| 	ForwardFrom           *User            `json:"forward_from"`            // optional | 	ForwardFrom           *User              `json:"forward_from"`            // optional | ||||||
| 	ForwardFromChat       *Chat            `json:"forward_from_chat"`       // optional | 	ForwardFromChat       *Chat              `json:"forward_from_chat"`       // optional | ||||||
| 	ForwardFromMessageID  int              `json:"forward_from_message_id"` // optional | 	ForwardFromMessageID  int                `json:"forward_from_message_id"` // optional | ||||||
| 	ForwardDate           int              `json:"forward_date"`            // optional | 	ForwardDate           int                `json:"forward_date"`            // optional | ||||||
| 	ReplyToMessage        *Message         `json:"reply_to_message"`        // optional | 	ReplyToMessage        *Message           `json:"reply_to_message"`        // optional | ||||||
| 	EditDate              int              `json:"edit_date"`               // optional | 	EditDate              int                `json:"edit_date"`               // optional | ||||||
| 	Text                  string           `json:"text"`                    // optional | 	Text                  string             `json:"text"`                    // optional | ||||||
| 	Entities              *[]MessageEntity `json:"entities"`                // optional | 	Entities              *[]MessageEntity   `json:"entities"`                // optional | ||||||
| 	Audio                 *Audio           `json:"audio"`                   // optional | 	Audio                 *Audio             `json:"audio"`                   // optional | ||||||
| 	Document              *Document        `json:"document"`                // optional | 	Document              *Document          `json:"document"`                // optional | ||||||
| 	Game                  *Game            `json:"game"`                    // optional | 	Game                  *Game              `json:"game"`                    // optional | ||||||
| 	Photo                 *[]PhotoSize     `json:"photo"`                   // optional | 	Photo                 *[]PhotoSize       `json:"photo"`                   // optional | ||||||
| 	Sticker               *Sticker         `json:"sticker"`                 // optional | 	Sticker               *Sticker           `json:"sticker"`                 // optional | ||||||
| 	Video                 *Video           `json:"video"`                   // optional | 	Video                 *Video             `json:"video"`                   // optional | ||||||
| 	Voice                 *Voice           `json:"voice"`                   // optional | 	VideoNote             *VideoNote         `json:"video_note"`              // optional | ||||||
| 	Caption               string           `json:"caption"`                 // optional | 	Voice                 *Voice             `json:"voice"`                   // optional | ||||||
| 	Contact               *Contact         `json:"contact"`                 // optional | 	Caption               string             `json:"caption"`                 // optional | ||||||
| 	Location              *Location        `json:"location"`                // optional | 	Contact               *Contact           `json:"contact"`                 // optional | ||||||
| 	Venue                 *Venue           `json:"venue"`                   // optional | 	Location              *Location          `json:"location"`                // optional | ||||||
| 	NewChatMember         *User            `json:"new_chat_member"`         // optional | 	Venue                 *Venue             `json:"venue"`                   // optional | ||||||
| 	LeftChatMember        *User            `json:"left_chat_member"`        // optional | 	NewChatMembers        *[]User            `json:"new_chat_members"`        // optional | ||||||
| 	NewChatTitle          string           `json:"new_chat_title"`          // optional | 	LeftChatMember        *User              `json:"left_chat_member"`        // optional | ||||||
| 	NewChatPhoto          *[]PhotoSize     `json:"new_chat_photo"`          // optional | 	NewChatTitle          string             `json:"new_chat_title"`          // optional | ||||||
| 	DeleteChatPhoto       bool             `json:"delete_chat_photo"`       // optional | 	NewChatPhoto          *[]PhotoSize       `json:"new_chat_photo"`          // optional | ||||||
| 	GroupChatCreated      bool             `json:"group_chat_created"`      // optional | 	DeleteChatPhoto       bool               `json:"delete_chat_photo"`       // optional | ||||||
| 	SuperGroupChatCreated bool             `json:"supergroup_chat_created"` // optional | 	GroupChatCreated      bool               `json:"group_chat_created"`      // optional | ||||||
| 	ChannelChatCreated    bool             `json:"channel_chat_created"`    // optional | 	SuperGroupChatCreated bool               `json:"supergroup_chat_created"` // optional | ||||||
| 	MigrateToChatID       int64            `json:"migrate_to_chat_id"`      // optional | 	ChannelChatCreated    bool               `json:"channel_chat_created"`    // optional | ||||||
| 	MigrateFromChatID     int64            `json:"migrate_from_chat_id"`    // optional | 	MigrateToChatID       int64              `json:"migrate_to_chat_id"`      // optional | ||||||
| 	PinnedMessage         *Message         `json:"pinned_message"`          // optional | 	MigrateFromChatID     int64              `json:"migrate_from_chat_id"`    // optional | ||||||
|  | 	PinnedMessage         *Message           `json:"pinned_message"`          // optional | ||||||
|  | 	Invoice               *Invoice           `json:"invoice"`                 // optional | ||||||
|  | 	SuccessfulPayment     *SuccessfulPayment `json:"successful_payment"`      // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // Time converts the message timestamp into a Time. | // Time converts the message timestamp into a Time. | ||||||
| @@ -263,6 +278,15 @@ type Video struct { | |||||||
| 	FileSize  int        `json:"file_size"` // optional | 	FileSize  int        `json:"file_size"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // VideoNote contains information about a video. | ||||||
|  | type VideoNote struct { | ||||||
|  | 	FileID    string     `json:"file_id"` | ||||||
|  | 	Length    int        `json:"length"` | ||||||
|  | 	Duration  int        `json:"duration"` | ||||||
|  | 	Thumbnail *PhotoSize `json:"thumb"`     // optional | ||||||
|  | 	FileSize  int        `json:"file_size"` // optional | ||||||
|  | } | ||||||
|  |  | ||||||
| // Voice contains information about a voice. | // Voice contains information about a voice. | ||||||
| type Voice struct { | type Voice struct { | ||||||
| 	FileID   string `json:"file_id"` | 	FileID   string `json:"file_id"` | ||||||
| @@ -361,6 +385,7 @@ type InlineKeyboardButton struct { | |||||||
| 	SwitchInlineQuery            *string       `json:"switch_inline_query,omitempty"`              // optional | 	SwitchInlineQuery            *string       `json:"switch_inline_query,omitempty"`              // optional | ||||||
| 	SwitchInlineQueryCurrentChat *string       `json:"switch_inline_query_current_chat,omitempty"` // optional | 	SwitchInlineQueryCurrentChat *string       `json:"switch_inline_query_current_chat,omitempty"` // optional | ||||||
| 	CallbackGame                 *CallbackGame `json:"callback_game,omitempty"`                    // optional | 	CallbackGame                 *CallbackGame `json:"callback_game,omitempty"`                    // optional | ||||||
|  | 	Pay                          bool          `json:"pay,omitempty"`			       // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // CallbackQuery is data sent when a keyboard button with callback data | // CallbackQuery is data sent when a keyboard button with callback data | ||||||
| @@ -384,8 +409,22 @@ type ForceReply struct { | |||||||
|  |  | ||||||
| // ChatMember is information about a member in a chat. | // ChatMember is information about a member in a chat. | ||||||
| type ChatMember struct { | type ChatMember struct { | ||||||
| 	User   *User  `json:"user"` | 	User                  *User  `json:"user"` | ||||||
| 	Status string `json:"status"` | 	Status                string `json:"status"` | ||||||
|  | 	UntilDate             int64  `json:"until_date,omitempty"`                // optional | ||||||
|  | 	CanBeEdited           bool   `json:"can_be_edited,omitempty"`             // optional | ||||||
|  | 	CanChangeInfo         bool   `json:"can_change_info,omitempty"`           // optional | ||||||
|  | 	CanPostMessages       bool   `json:"can_post_messages,omitempty"`         // optional | ||||||
|  | 	CanEditMessages       bool   `json:"can_edit_messages,omitempty"`         // optional | ||||||
|  | 	CanDeleteMessages     bool   `json:"can_delete_messages,omitempty"`       // optional | ||||||
|  | 	CanInviteUsers        bool   `json:"can_invite_users,omitempty"`          // optional | ||||||
|  | 	CanRestrictMembers    bool   `json:"can_restrict_members,omitempty"`      // optional | ||||||
|  | 	CanPinMessages        bool   `json:"can_pin_messages,omitempty"`          // optional | ||||||
|  | 	CanPromoteMembers     bool   `json:"can_promote_members,omitempty"`       // optional | ||||||
|  | 	CanSendMessages       bool   `json:"can_send_messages,omitempty"`         // optional | ||||||
|  | 	CanSendMediaMessages  bool   `json:"can_send_media_messages,omitempty"`   // optional | ||||||
|  | 	CanSendOtherMessages  bool   `json:"can_send_other_messages,omitempty"`   // optional | ||||||
|  | 	CanAddWebPagePreviews bool   `json:"can_add_web_page_previews,omitempty"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // IsCreator returns if the ChatMember was the creator of the chat. | // IsCreator returns if the ChatMember was the creator of the chat. | ||||||
| @@ -493,6 +532,7 @@ type InlineQueryResultGIF struct { | |||||||
| 	URL                 string                `json:"gif_url"` // required | 	URL                 string                `json:"gif_url"` // required | ||||||
| 	Width               int                   `json:"gif_width"` | 	Width               int                   `json:"gif_width"` | ||||||
| 	Height              int                   `json:"gif_height"` | 	Height              int                   `json:"gif_height"` | ||||||
|  | 	Duration            int                   `json:"gif_duration"` | ||||||
| 	ThumbURL            string                `json:"thumb_url"` | 	ThumbURL            string                `json:"thumb_url"` | ||||||
| 	Title               string                `json:"title"` | 	Title               string                `json:"title"` | ||||||
| 	Caption             string                `json:"caption"` | 	Caption             string                `json:"caption"` | ||||||
| @@ -507,6 +547,7 @@ type InlineQueryResultMPEG4GIF struct { | |||||||
| 	URL                 string                `json:"mpeg4_url"` // required | 	URL                 string                `json:"mpeg4_url"` // required | ||||||
| 	Width               int                   `json:"mpeg4_width"` | 	Width               int                   `json:"mpeg4_width"` | ||||||
| 	Height              int                   `json:"mpeg4_height"` | 	Height              int                   `json:"mpeg4_height"` | ||||||
|  | 	Duration            int                   `json:"mpeg4_duration"` | ||||||
| 	ThumbURL            string                `json:"thumb_url"` | 	ThumbURL            string                `json:"thumb_url"` | ||||||
| 	Title               string                `json:"title"` | 	Title               string                `json:"title"` | ||||||
| 	Caption             string                `json:"caption"` | 	Caption             string                `json:"caption"` | ||||||
| @@ -635,3 +676,73 @@ type InputContactMessageContent struct { | |||||||
| 	FirstName   string `json:"first_name"` | 	FirstName   string `json:"first_name"` | ||||||
| 	LastName    string `json:"last_name"` | 	LastName    string `json:"last_name"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Invoice contains basic information about an invoice. | ||||||
|  | type Invoice struct { | ||||||
|  | 	Title          string `json:"title"` | ||||||
|  | 	Description    string `json:"description"` | ||||||
|  | 	StartParameter string `json:"start_parameter"` | ||||||
|  | 	Currency       string `json:"currency"` | ||||||
|  | 	TotalAmount    int    `json:"total_amount"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LabeledPrice represents a portion of the price for goods or services. | ||||||
|  | type LabeledPrice struct { | ||||||
|  | 	Label  string `json:"label"` | ||||||
|  | 	Amount int    `json:"amount"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingAddress represents a shipping address. | ||||||
|  | type ShippingAddress struct { | ||||||
|  | 	CountryCode string `json:"country_code"` | ||||||
|  | 	State       string `json:"state"` | ||||||
|  | 	City        string `json:"city"` | ||||||
|  | 	StreetLine1 string `json:"street_line1"` | ||||||
|  | 	StreetLine2 string `json:"street_line2"` | ||||||
|  | 	PostCode    string `json:"post_code"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // OrderInfo represents information about an order. | ||||||
|  | type OrderInfo struct { | ||||||
|  | 	Name            string           `json:"name,omitempty"` | ||||||
|  | 	PhoneNumber     string           `json:"phone_number,omitempty"` | ||||||
|  | 	Email           string           `json:"email,omitempty"` | ||||||
|  | 	ShippingAddress *ShippingAddress `json:"shipping_address,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingOption represents one shipping option. | ||||||
|  | type ShippingOption struct { | ||||||
|  | 	ID     string          `json:"id"` | ||||||
|  | 	Title  string          `json:"title"` | ||||||
|  | 	Prices *[]LabeledPrice `json:"prices"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SuccessfulPayment contains basic information about a successful payment. | ||||||
|  | type SuccessfulPayment struct { | ||||||
|  | 	Currency                string     `json:"currency"` | ||||||
|  | 	TotalAmount             int        `json:"total_amount"` | ||||||
|  | 	InvoicePayload          string     `json:"invoice_payload"` | ||||||
|  | 	ShippingOptionID        string     `json:"shipping_option_id,omitempty"` | ||||||
|  | 	OrderInfo               *OrderInfo `json:"order_info,omitempty"` | ||||||
|  | 	TelegramPaymentChargeID string     `json:"telegram_payment_charge_id"` | ||||||
|  | 	ProviderPaymentChargeID string     `json:"provider_payment_charge_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingQuery contains information about an incoming shipping query. | ||||||
|  | type ShippingQuery struct { | ||||||
|  | 	ID              string           `json:"id"` | ||||||
|  | 	From            *User            `json:"from"` | ||||||
|  | 	InvoicePayload  string           `json:"invoice_payload"` | ||||||
|  | 	ShippingAddress *ShippingAddress `json:"shipping_address"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PreCheckoutQuery contains information about an incoming pre-checkout query. | ||||||
|  | type PreCheckoutQuery struct { | ||||||
|  | 	ID               string     `json:"id"` | ||||||
|  | 	From             *User      `json:"from"` | ||||||
|  | 	Currency         string     `json:"currency"` | ||||||
|  | 	TotalAmount      int        `json:"total_amount"` | ||||||
|  | 	InvoicePayload   string     `json:"invoice_payload"` | ||||||
|  | 	ShippingOptionID string     `json:"shipping_option_id,omitempty"` | ||||||
|  | 	OrderInfo        *OrderInfo `json:"order_info,omitempty"` | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										212
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										212
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,212 @@ | |||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	// Default2QRecentRatio is the ratio of the 2Q cache dedicated | ||||||
|  | 	// to recently added entries that have only been accessed once. | ||||||
|  | 	Default2QRecentRatio = 0.25 | ||||||
|  |  | ||||||
|  | 	// Default2QGhostEntries is the default ratio of ghost | ||||||
|  | 	// entries kept to track entries recently evicted | ||||||
|  | 	Default2QGhostEntries = 0.50 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // TwoQueueCache is a thread-safe fixed size 2Q cache. | ||||||
|  | // 2Q is an enhancement over the standard LRU cache | ||||||
|  | // in that it tracks both frequently and recently used | ||||||
|  | // entries separately. This avoids a burst in access to new | ||||||
|  | // entries from evicting frequently used entries. It adds some | ||||||
|  | // additional tracking overhead to the standard LRU cache, and is | ||||||
|  | // computationally about 2x the cost, and adds some metadata over | ||||||
|  | // head. The ARCCache is similar, but does not require setting any | ||||||
|  | // parameters. | ||||||
|  | type TwoQueueCache struct { | ||||||
|  | 	size       int | ||||||
|  | 	recentSize int | ||||||
|  |  | ||||||
|  | 	recent      *simplelru.LRU | ||||||
|  | 	frequent    *simplelru.LRU | ||||||
|  | 	recentEvict *simplelru.LRU | ||||||
|  | 	lock        sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New2Q creates a new TwoQueueCache using the default | ||||||
|  | // values for the parameters. | ||||||
|  | func New2Q(size int) (*TwoQueueCache, error) { | ||||||
|  | 	return New2QParams(size, Default2QRecentRatio, Default2QGhostEntries) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New2QParams creates a new TwoQueueCache using the provided | ||||||
|  | // parameter values. | ||||||
|  | func New2QParams(size int, recentRatio float64, ghostRatio float64) (*TwoQueueCache, error) { | ||||||
|  | 	if size <= 0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid size") | ||||||
|  | 	} | ||||||
|  | 	if recentRatio < 0.0 || recentRatio > 1.0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid recent ratio") | ||||||
|  | 	} | ||||||
|  | 	if ghostRatio < 0.0 || ghostRatio > 1.0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid ghost ratio") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Determine the sub-sizes | ||||||
|  | 	recentSize := int(float64(size) * recentRatio) | ||||||
|  | 	evictSize := int(float64(size) * ghostRatio) | ||||||
|  |  | ||||||
|  | 	// Allocate the LRUs | ||||||
|  | 	recent, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	frequent, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	recentEvict, err := simplelru.NewLRU(evictSize, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Initialize the cache | ||||||
|  | 	c := &TwoQueueCache{ | ||||||
|  | 		size:        size, | ||||||
|  | 		recentSize:  recentSize, | ||||||
|  | 		recent:      recent, | ||||||
|  | 		frequent:    frequent, | ||||||
|  | 		recentEvict: recentEvict, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if this is a frequent value | ||||||
|  | 	if val, ok := c.frequent.Get(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the value is contained in recent, then we | ||||||
|  | 	// promote it to frequent | ||||||
|  | 	if val, ok := c.recent.Peek(key); ok { | ||||||
|  | 		c.recent.Remove(key) | ||||||
|  | 		c.frequent.Add(key, val) | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// No hit | ||||||
|  | 	return nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Add(key, value interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the value is frequently used already, | ||||||
|  | 	// and just update the value | ||||||
|  | 	if c.frequent.Contains(key) { | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is recently used, and promote | ||||||
|  | 	// the value into the frequent list | ||||||
|  | 	if c.recent.Contains(key) { | ||||||
|  | 		c.recent.Remove(key) | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the value was recently evicted, add it to the | ||||||
|  | 	// frequently used list | ||||||
|  | 	if c.recentEvict.Contains(key) { | ||||||
|  | 		c.ensureSpace(true) | ||||||
|  | 		c.recentEvict.Remove(key) | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add to the recently seen list | ||||||
|  | 	c.ensureSpace(false) | ||||||
|  | 	c.recent.Add(key, value) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ensureSpace is used to ensure we have space in the cache | ||||||
|  | func (c *TwoQueueCache) ensureSpace(recentEvict bool) { | ||||||
|  | 	// If we have space, nothing to do | ||||||
|  | 	recentLen := c.recent.Len() | ||||||
|  | 	freqLen := c.frequent.Len() | ||||||
|  | 	if recentLen+freqLen < c.size { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the recent buffer is larger than | ||||||
|  | 	// the target, evict from there | ||||||
|  | 	if recentLen > 0 && (recentLen > c.recentSize || (recentLen == c.recentSize && !recentEvict)) { | ||||||
|  | 		k, _, _ := c.recent.RemoveOldest() | ||||||
|  | 		c.recentEvict.Add(k, nil) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Remove from the frequent list otherwise | ||||||
|  | 	c.frequent.RemoveOldest() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.recent.Len() + c.frequent.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	k1 := c.frequent.Keys() | ||||||
|  | 	k2 := c.recent.Keys() | ||||||
|  | 	return append(k1, k2...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	if c.frequent.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.recent.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.recentEvict.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	c.recent.Purge() | ||||||
|  | 	c.frequent.Purge() | ||||||
|  | 	c.recentEvict.Purge() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.frequent.Contains(key) || c.recent.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	if val, ok := c.frequent.Peek(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  | 	return c.recent.Peek(key) | ||||||
|  | } | ||||||
							
								
								
									
										362
									
								
								vendor/github.com/hashicorp/golang-lru/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										362
									
								
								vendor/github.com/hashicorp/golang-lru/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,362 @@ | |||||||
|  | Mozilla Public License, version 2.0 | ||||||
|  |  | ||||||
|  | 1. Definitions | ||||||
|  |  | ||||||
|  | 1.1. "Contributor" | ||||||
|  |  | ||||||
|  |      means each individual or legal entity that creates, contributes to the | ||||||
|  |      creation of, or owns Covered Software. | ||||||
|  |  | ||||||
|  | 1.2. "Contributor Version" | ||||||
|  |  | ||||||
|  |      means the combination of the Contributions of others (if any) used by a | ||||||
|  |      Contributor and that particular Contributor's Contribution. | ||||||
|  |  | ||||||
|  | 1.3. "Contribution" | ||||||
|  |  | ||||||
|  |      means Covered Software of a particular Contributor. | ||||||
|  |  | ||||||
|  | 1.4. "Covered Software" | ||||||
|  |  | ||||||
|  |      means Source Code Form to which the initial Contributor has attached the | ||||||
|  |      notice in Exhibit A, the Executable Form of such Source Code Form, and | ||||||
|  |      Modifications of such Source Code Form, in each case including portions | ||||||
|  |      thereof. | ||||||
|  |  | ||||||
|  | 1.5. "Incompatible With Secondary Licenses" | ||||||
|  |      means | ||||||
|  |  | ||||||
|  |      a. that the initial Contributor has attached the notice described in | ||||||
|  |         Exhibit B to the Covered Software; or | ||||||
|  |  | ||||||
|  |      b. that the Covered Software was made available under the terms of | ||||||
|  |         version 1.1 or earlier of the License, but not also under the terms of | ||||||
|  |         a Secondary License. | ||||||
|  |  | ||||||
|  | 1.6. "Executable Form" | ||||||
|  |  | ||||||
|  |      means any form of the work other than Source Code Form. | ||||||
|  |  | ||||||
|  | 1.7. "Larger Work" | ||||||
|  |  | ||||||
|  |      means a work that combines Covered Software with other material, in a | ||||||
|  |      separate file or files, that is not Covered Software. | ||||||
|  |  | ||||||
|  | 1.8. "License" | ||||||
|  |  | ||||||
|  |      means this document. | ||||||
|  |  | ||||||
|  | 1.9. "Licensable" | ||||||
|  |  | ||||||
|  |      means having the right to grant, to the maximum extent possible, whether | ||||||
|  |      at the time of the initial grant or subsequently, any and all of the | ||||||
|  |      rights conveyed by this License. | ||||||
|  |  | ||||||
|  | 1.10. "Modifications" | ||||||
|  |  | ||||||
|  |      means any of the following: | ||||||
|  |  | ||||||
|  |      a. any file in Source Code Form that results from an addition to, | ||||||
|  |         deletion from, or modification of the contents of Covered Software; or | ||||||
|  |  | ||||||
|  |      b. any new file in Source Code Form that contains any Covered Software. | ||||||
|  |  | ||||||
|  | 1.11. "Patent Claims" of a Contributor | ||||||
|  |  | ||||||
|  |       means any patent claim(s), including without limitation, method, | ||||||
|  |       process, and apparatus claims, in any patent Licensable by such | ||||||
|  |       Contributor that would be infringed, but for the grant of the License, | ||||||
|  |       by the making, using, selling, offering for sale, having made, import, | ||||||
|  |       or transfer of either its Contributions or its Contributor Version. | ||||||
|  |  | ||||||
|  | 1.12. "Secondary License" | ||||||
|  |  | ||||||
|  |       means either the GNU General Public License, Version 2.0, the GNU Lesser | ||||||
|  |       General Public License, Version 2.1, the GNU Affero General Public | ||||||
|  |       License, Version 3.0, or any later versions of those licenses. | ||||||
|  |  | ||||||
|  | 1.13. "Source Code Form" | ||||||
|  |  | ||||||
|  |       means the form of the work preferred for making modifications. | ||||||
|  |  | ||||||
|  | 1.14. "You" (or "Your") | ||||||
|  |  | ||||||
|  |       means an individual or a legal entity exercising rights under this | ||||||
|  |       License. For legal entities, "You" includes any entity that controls, is | ||||||
|  |       controlled by, or is under common control with You. For purposes of this | ||||||
|  |       definition, "control" means (a) the power, direct or indirect, to cause | ||||||
|  |       the direction or management of such entity, whether by contract or | ||||||
|  |       otherwise, or (b) ownership of more than fifty percent (50%) of the | ||||||
|  |       outstanding shares or beneficial ownership of such entity. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 2. License Grants and Conditions | ||||||
|  |  | ||||||
|  | 2.1. Grants | ||||||
|  |  | ||||||
|  |      Each Contributor hereby grants You a world-wide, royalty-free, | ||||||
|  |      non-exclusive license: | ||||||
|  |  | ||||||
|  |      a. under intellectual property rights (other than patent or trademark) | ||||||
|  |         Licensable by such Contributor to use, reproduce, make available, | ||||||
|  |         modify, display, perform, distribute, and otherwise exploit its | ||||||
|  |         Contributions, either on an unmodified basis, with Modifications, or | ||||||
|  |         as part of a Larger Work; and | ||||||
|  |  | ||||||
|  |      b. under Patent Claims of such Contributor to make, use, sell, offer for | ||||||
|  |         sale, have made, import, and otherwise transfer either its | ||||||
|  |         Contributions or its Contributor Version. | ||||||
|  |  | ||||||
|  | 2.2. Effective Date | ||||||
|  |  | ||||||
|  |      The licenses granted in Section 2.1 with respect to any Contribution | ||||||
|  |      become effective for each Contribution on the date the Contributor first | ||||||
|  |      distributes such Contribution. | ||||||
|  |  | ||||||
|  | 2.3. Limitations on Grant Scope | ||||||
|  |  | ||||||
|  |      The licenses granted in this Section 2 are the only rights granted under | ||||||
|  |      this License. No additional rights or licenses will be implied from the | ||||||
|  |      distribution or licensing of Covered Software under this License. | ||||||
|  |      Notwithstanding Section 2.1(b) above, no patent license is granted by a | ||||||
|  |      Contributor: | ||||||
|  |  | ||||||
|  |      a. for any code that a Contributor has removed from Covered Software; or | ||||||
|  |  | ||||||
|  |      b. for infringements caused by: (i) Your and any other third party's | ||||||
|  |         modifications of Covered Software, or (ii) the combination of its | ||||||
|  |         Contributions with other software (except as part of its Contributor | ||||||
|  |         Version); or | ||||||
|  |  | ||||||
|  |      c. under Patent Claims infringed by Covered Software in the absence of | ||||||
|  |         its Contributions. | ||||||
|  |  | ||||||
|  |      This License does not grant any rights in the trademarks, service marks, | ||||||
|  |      or logos of any Contributor (except as may be necessary to comply with | ||||||
|  |      the notice requirements in Section 3.4). | ||||||
|  |  | ||||||
|  | 2.4. Subsequent Licenses | ||||||
|  |  | ||||||
|  |      No Contributor makes additional grants as a result of Your choice to | ||||||
|  |      distribute the Covered Software under a subsequent version of this | ||||||
|  |      License (see Section 10.2) or under the terms of a Secondary License (if | ||||||
|  |      permitted under the terms of Section 3.3). | ||||||
|  |  | ||||||
|  | 2.5. Representation | ||||||
|  |  | ||||||
|  |      Each Contributor represents that the Contributor believes its | ||||||
|  |      Contributions are its original creation(s) or it has sufficient rights to | ||||||
|  |      grant the rights to its Contributions conveyed by this License. | ||||||
|  |  | ||||||
|  | 2.6. Fair Use | ||||||
|  |  | ||||||
|  |      This License is not intended to limit any rights You have under | ||||||
|  |      applicable copyright doctrines of fair use, fair dealing, or other | ||||||
|  |      equivalents. | ||||||
|  |  | ||||||
|  | 2.7. Conditions | ||||||
|  |  | ||||||
|  |      Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted in | ||||||
|  |      Section 2.1. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 3. Responsibilities | ||||||
|  |  | ||||||
|  | 3.1. Distribution of Source Form | ||||||
|  |  | ||||||
|  |      All distribution of Covered Software in Source Code Form, including any | ||||||
|  |      Modifications that You create or to which You contribute, must be under | ||||||
|  |      the terms of this License. You must inform recipients that the Source | ||||||
|  |      Code Form of the Covered Software is governed by the terms of this | ||||||
|  |      License, and how they can obtain a copy of this License. You may not | ||||||
|  |      attempt to alter or restrict the recipients' rights in the Source Code | ||||||
|  |      Form. | ||||||
|  |  | ||||||
|  | 3.2. Distribution of Executable Form | ||||||
|  |  | ||||||
|  |      If You distribute Covered Software in Executable Form then: | ||||||
|  |  | ||||||
|  |      a. such Covered Software must also be made available in Source Code Form, | ||||||
|  |         as described in Section 3.1, and You must inform recipients of the | ||||||
|  |         Executable Form how they can obtain a copy of such Source Code Form by | ||||||
|  |         reasonable means in a timely manner, at a charge no more than the cost | ||||||
|  |         of distribution to the recipient; and | ||||||
|  |  | ||||||
|  |      b. You may distribute such Executable Form under the terms of this | ||||||
|  |         License, or sublicense it under different terms, provided that the | ||||||
|  |         license for the Executable Form does not attempt to limit or alter the | ||||||
|  |         recipients' rights in the Source Code Form under this License. | ||||||
|  |  | ||||||
|  | 3.3. Distribution of a Larger Work | ||||||
|  |  | ||||||
|  |      You may create and distribute a Larger Work under terms of Your choice, | ||||||
|  |      provided that You also comply with the requirements of this License for | ||||||
|  |      the Covered Software. If the Larger Work is a combination of Covered | ||||||
|  |      Software with a work governed by one or more Secondary Licenses, and the | ||||||
|  |      Covered Software is not Incompatible With Secondary Licenses, this | ||||||
|  |      License permits You to additionally distribute such Covered Software | ||||||
|  |      under the terms of such Secondary License(s), so that the recipient of | ||||||
|  |      the Larger Work may, at their option, further distribute the Covered | ||||||
|  |      Software under the terms of either this License or such Secondary | ||||||
|  |      License(s). | ||||||
|  |  | ||||||
|  | 3.4. Notices | ||||||
|  |  | ||||||
|  |      You may not remove or alter the substance of any license notices | ||||||
|  |      (including copyright notices, patent notices, disclaimers of warranty, or | ||||||
|  |      limitations of liability) contained within the Source Code Form of the | ||||||
|  |      Covered Software, except that You may alter any license notices to the | ||||||
|  |      extent required to remedy known factual inaccuracies. | ||||||
|  |  | ||||||
|  | 3.5. Application of Additional Terms | ||||||
|  |  | ||||||
|  |      You may choose to offer, and to charge a fee for, warranty, support, | ||||||
|  |      indemnity or liability obligations to one or more recipients of Covered | ||||||
|  |      Software. However, You may do so only on Your own behalf, and not on | ||||||
|  |      behalf of any Contributor. You must make it absolutely clear that any | ||||||
|  |      such warranty, support, indemnity, or liability obligation is offered by | ||||||
|  |      You alone, and You hereby agree to indemnify every Contributor for any | ||||||
|  |      liability incurred by such Contributor as a result of warranty, support, | ||||||
|  |      indemnity or liability terms You offer. You may include additional | ||||||
|  |      disclaimers of warranty and limitations of liability specific to any | ||||||
|  |      jurisdiction. | ||||||
|  |  | ||||||
|  | 4. Inability to Comply Due to Statute or Regulation | ||||||
|  |  | ||||||
|  |    If it is impossible for You to comply with any of the terms of this License | ||||||
|  |    with respect to some or all of the Covered Software due to statute, | ||||||
|  |    judicial order, or regulation then You must: (a) comply with the terms of | ||||||
|  |    this License to the maximum extent possible; and (b) describe the | ||||||
|  |    limitations and the code they affect. Such description must be placed in a | ||||||
|  |    text file included with all distributions of the Covered Software under | ||||||
|  |    this License. Except to the extent prohibited by statute or regulation, | ||||||
|  |    such description must be sufficiently detailed for a recipient of ordinary | ||||||
|  |    skill to be able to understand it. | ||||||
|  |  | ||||||
|  | 5. Termination | ||||||
|  |  | ||||||
|  | 5.1. The rights granted under this License will terminate automatically if You | ||||||
|  |      fail to comply with any of its terms. However, if You become compliant, | ||||||
|  |      then the rights granted under this License from a particular Contributor | ||||||
|  |      are reinstated (a) provisionally, unless and until such Contributor | ||||||
|  |      explicitly and finally terminates Your grants, and (b) on an ongoing | ||||||
|  |      basis, if such Contributor fails to notify You of the non-compliance by | ||||||
|  |      some reasonable means prior to 60 days after You have come back into | ||||||
|  |      compliance. Moreover, Your grants from a particular Contributor are | ||||||
|  |      reinstated on an ongoing basis if such Contributor notifies You of the | ||||||
|  |      non-compliance by some reasonable means, this is the first time You have | ||||||
|  |      received notice of non-compliance with this License from such | ||||||
|  |      Contributor, and You become compliant prior to 30 days after Your receipt | ||||||
|  |      of the notice. | ||||||
|  |  | ||||||
|  | 5.2. If You initiate litigation against any entity by asserting a patent | ||||||
|  |      infringement claim (excluding declaratory judgment actions, | ||||||
|  |      counter-claims, and cross-claims) alleging that a Contributor Version | ||||||
|  |      directly or indirectly infringes any patent, then the rights granted to | ||||||
|  |      You by any and all Contributors for the Covered Software under Section | ||||||
|  |      2.1 of this License shall terminate. | ||||||
|  |  | ||||||
|  | 5.3. In the event of termination under Sections 5.1 or 5.2 above, all end user | ||||||
|  |      license agreements (excluding distributors and resellers) which have been | ||||||
|  |      validly granted by You or Your distributors under this License prior to | ||||||
|  |      termination shall survive termination. | ||||||
|  |  | ||||||
|  | 6. Disclaimer of Warranty | ||||||
|  |  | ||||||
|  |    Covered Software is provided under this License on an "as is" basis, | ||||||
|  |    without warranty of any kind, either expressed, implied, or statutory, | ||||||
|  |    including, without limitation, warranties that the Covered Software is free | ||||||
|  |    of defects, merchantable, fit for a particular purpose or non-infringing. | ||||||
|  |    The entire risk as to the quality and performance of the Covered Software | ||||||
|  |    is with You. Should any Covered Software prove defective in any respect, | ||||||
|  |    You (not any Contributor) assume the cost of any necessary servicing, | ||||||
|  |    repair, or correction. This disclaimer of warranty constitutes an essential | ||||||
|  |    part of this License. No use of  any Covered Software is authorized under | ||||||
|  |    this License except under this disclaimer. | ||||||
|  |  | ||||||
|  | 7. Limitation of Liability | ||||||
|  |  | ||||||
|  |    Under no circumstances and under no legal theory, whether tort (including | ||||||
|  |    negligence), contract, or otherwise, shall any Contributor, or anyone who | ||||||
|  |    distributes Covered Software as permitted above, be liable to You for any | ||||||
|  |    direct, indirect, special, incidental, or consequential damages of any | ||||||
|  |    character including, without limitation, damages for lost profits, loss of | ||||||
|  |    goodwill, work stoppage, computer failure or malfunction, or any and all | ||||||
|  |    other commercial damages or losses, even if such party shall have been | ||||||
|  |    informed of the possibility of such damages. This limitation of liability | ||||||
|  |    shall not apply to liability for death or personal injury resulting from | ||||||
|  |    such party's negligence to the extent applicable law prohibits such | ||||||
|  |    limitation. Some jurisdictions do not allow the exclusion or limitation of | ||||||
|  |    incidental or consequential damages, so this exclusion and limitation may | ||||||
|  |    not apply to You. | ||||||
|  |  | ||||||
|  | 8. Litigation | ||||||
|  |  | ||||||
|  |    Any litigation relating to this License may be brought only in the courts | ||||||
|  |    of a jurisdiction where the defendant maintains its principal place of | ||||||
|  |    business and such litigation shall be governed by laws of that | ||||||
|  |    jurisdiction, without reference to its conflict-of-law provisions. Nothing | ||||||
|  |    in this Section shall prevent a party's ability to bring cross-claims or | ||||||
|  |    counter-claims. | ||||||
|  |  | ||||||
|  | 9. Miscellaneous | ||||||
|  |  | ||||||
|  |    This License represents the complete agreement concerning the subject | ||||||
|  |    matter hereof. If any provision of this License is held to be | ||||||
|  |    unenforceable, such provision shall be reformed only to the extent | ||||||
|  |    necessary to make it enforceable. Any law or regulation which provides that | ||||||
|  |    the language of a contract shall be construed against the drafter shall not | ||||||
|  |    be used to construe this License against a Contributor. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 10. Versions of the License | ||||||
|  |  | ||||||
|  | 10.1. New Versions | ||||||
|  |  | ||||||
|  |       Mozilla Foundation is the license steward. Except as provided in Section | ||||||
|  |       10.3, no one other than the license steward has the right to modify or | ||||||
|  |       publish new versions of this License. Each version will be given a | ||||||
|  |       distinguishing version number. | ||||||
|  |  | ||||||
|  | 10.2. Effect of New Versions | ||||||
|  |  | ||||||
|  |       You may distribute the Covered Software under the terms of the version | ||||||
|  |       of the License under which You originally received the Covered Software, | ||||||
|  |       or under the terms of any subsequent version published by the license | ||||||
|  |       steward. | ||||||
|  |  | ||||||
|  | 10.3. Modified Versions | ||||||
|  |  | ||||||
|  |       If you create software not governed by this License, and you want to | ||||||
|  |       create a new license for such software, you may create and use a | ||||||
|  |       modified version of this License if you rename the license and remove | ||||||
|  |       any references to the name of the license steward (except to note that | ||||||
|  |       such modified license differs from this License). | ||||||
|  |  | ||||||
|  | 10.4. Distributing Source Code Form that is Incompatible With Secondary | ||||||
|  |       Licenses If You choose to distribute Source Code Form that is | ||||||
|  |       Incompatible With Secondary Licenses under the terms of this version of | ||||||
|  |       the License, the notice described in Exhibit B of this License must be | ||||||
|  |       attached. | ||||||
|  |  | ||||||
|  | Exhibit A - Source Code Form License Notice | ||||||
|  |  | ||||||
|  |       This Source Code Form is subject to the | ||||||
|  |       terms of the Mozilla Public License, v. | ||||||
|  |       2.0. If a copy of the MPL was not | ||||||
|  |       distributed with this file, You can | ||||||
|  |       obtain one at | ||||||
|  |       http://mozilla.org/MPL/2.0/. | ||||||
|  |  | ||||||
|  | If it is not possible or desirable to put the notice in a particular file, | ||||||
|  | then You may include the notice in a location (such as a LICENSE file in a | ||||||
|  | relevant directory) where a recipient would be likely to look for such a | ||||||
|  | notice. | ||||||
|  |  | ||||||
|  | You may add additional accurate notices of copyright ownership. | ||||||
|  |  | ||||||
|  | Exhibit B - "Incompatible With Secondary Licenses" Notice | ||||||
|  |  | ||||||
|  |       This Source Code Form is "Incompatible | ||||||
|  |       With Secondary Licenses", as defined by | ||||||
|  |       the Mozilla Public License, v. 2.0. | ||||||
							
								
								
									
										257
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										257
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,257 @@ | |||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // ARCCache is a thread-safe fixed size Adaptive Replacement Cache (ARC). | ||||||
|  | // ARC is an enhancement over the standard LRU cache in that tracks both | ||||||
|  | // frequency and recency of use. This avoids a burst in access to new | ||||||
|  | // entries from evicting the frequently used older entries. It adds some | ||||||
|  | // additional tracking overhead to a standard LRU cache, computationally | ||||||
|  | // it is roughly 2x the cost, and the extra memory overhead is linear | ||||||
|  | // with the size of the cache. ARC has been patented by IBM, but is | ||||||
|  | // similar to the TwoQueueCache (2Q) which requires setting parameters. | ||||||
|  | type ARCCache struct { | ||||||
|  | 	size int // Size is the total capacity of the cache | ||||||
|  | 	p    int // P is the dynamic preference towards T1 or T2 | ||||||
|  |  | ||||||
|  | 	t1 *simplelru.LRU // T1 is the LRU for recently accessed items | ||||||
|  | 	b1 *simplelru.LRU // B1 is the LRU for evictions from t1 | ||||||
|  |  | ||||||
|  | 	t2 *simplelru.LRU // T2 is the LRU for frequently accessed items | ||||||
|  | 	b2 *simplelru.LRU // B2 is the LRU for evictions from t2 | ||||||
|  |  | ||||||
|  | 	lock sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewARC creates an ARC of the given size | ||||||
|  | func NewARC(size int) (*ARCCache, error) { | ||||||
|  | 	// Create the sub LRUs | ||||||
|  | 	b1, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	b2, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	t1, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	t2, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Initialize the ARC | ||||||
|  | 	c := &ARCCache{ | ||||||
|  | 		size: size, | ||||||
|  | 		p:    0, | ||||||
|  | 		t1:   t1, | ||||||
|  | 		b1:   b1, | ||||||
|  | 		t2:   t2, | ||||||
|  | 		b2:   b2, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *ARCCache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Ff the value is contained in T1 (recent), then | ||||||
|  | 	// promote it to T2 (frequent) | ||||||
|  | 	if val, ok := c.t1.Peek(key); ok { | ||||||
|  | 		c.t1.Remove(key) | ||||||
|  | 		c.t2.Add(key, val) | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is contained in T2 (frequent) | ||||||
|  | 	if val, ok := c.t2.Get(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// No hit | ||||||
|  | 	return nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache. | ||||||
|  | func (c *ARCCache) Add(key, value interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the value is contained in T1 (recent), and potentially | ||||||
|  | 	// promote it to frequent T2 | ||||||
|  | 	if c.t1.Contains(key) { | ||||||
|  | 		c.t1.Remove(key) | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is already in T2 (frequent) and update it | ||||||
|  | 	if c.t2.Contains(key) { | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if this value was recently evicted as part of the | ||||||
|  | 	// recently used list | ||||||
|  | 	if c.b1.Contains(key) { | ||||||
|  | 		// T1 set is too small, increase P appropriately | ||||||
|  | 		delta := 1 | ||||||
|  | 		b1Len := c.b1.Len() | ||||||
|  | 		b2Len := c.b2.Len() | ||||||
|  | 		if b2Len > b1Len { | ||||||
|  | 			delta = b2Len / b1Len | ||||||
|  | 		} | ||||||
|  | 		if c.p+delta >= c.size { | ||||||
|  | 			c.p = c.size | ||||||
|  | 		} else { | ||||||
|  | 			c.p += delta | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Potentially need to make room in the cache | ||||||
|  | 		if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 			c.replace(false) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remove from B1 | ||||||
|  | 		c.b1.Remove(key) | ||||||
|  |  | ||||||
|  | 		// Add the key to the frequently used list | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if this value was recently evicted as part of the | ||||||
|  | 	// frequently used list | ||||||
|  | 	if c.b2.Contains(key) { | ||||||
|  | 		// T2 set is too small, decrease P appropriately | ||||||
|  | 		delta := 1 | ||||||
|  | 		b1Len := c.b1.Len() | ||||||
|  | 		b2Len := c.b2.Len() | ||||||
|  | 		if b1Len > b2Len { | ||||||
|  | 			delta = b1Len / b2Len | ||||||
|  | 		} | ||||||
|  | 		if delta >= c.p { | ||||||
|  | 			c.p = 0 | ||||||
|  | 		} else { | ||||||
|  | 			c.p -= delta | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Potentially need to make room in the cache | ||||||
|  | 		if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 			c.replace(true) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remove from B2 | ||||||
|  | 		c.b2.Remove(key) | ||||||
|  |  | ||||||
|  | 		// Add the key to the frequntly used list | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Potentially need to make room in the cache | ||||||
|  | 	if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 		c.replace(false) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Keep the size of the ghost buffers trim | ||||||
|  | 	if c.b1.Len() > c.size-c.p { | ||||||
|  | 		c.b1.RemoveOldest() | ||||||
|  | 	} | ||||||
|  | 	if c.b2.Len() > c.p { | ||||||
|  | 		c.b2.RemoveOldest() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add to the recently seen list | ||||||
|  | 	c.t1.Add(key, value) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // replace is used to adaptively evict from either T1 or T2 | ||||||
|  | // based on the current learned value of P | ||||||
|  | func (c *ARCCache) replace(b2ContainsKey bool) { | ||||||
|  | 	t1Len := c.t1.Len() | ||||||
|  | 	if t1Len > 0 && (t1Len > c.p || (t1Len == c.p && b2ContainsKey)) { | ||||||
|  | 		k, _, ok := c.t1.RemoveOldest() | ||||||
|  | 		if ok { | ||||||
|  | 			c.b1.Add(k, nil) | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		k, _, ok := c.t2.RemoveOldest() | ||||||
|  | 		if ok { | ||||||
|  | 			c.b2.Add(k, nil) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of cached entries | ||||||
|  | func (c *ARCCache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.t1.Len() + c.t2.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns all the cached keys | ||||||
|  | func (c *ARCCache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	k1 := c.t1.Keys() | ||||||
|  | 	k2 := c.t2.Keys() | ||||||
|  | 	return append(k1, k2...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove is used to purge a key from the cache | ||||||
|  | func (c *ARCCache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	if c.t1.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.t2.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.b1.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.b2.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to clear the cache | ||||||
|  | func (c *ARCCache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	c.t1.Purge() | ||||||
|  | 	c.t2.Purge() | ||||||
|  | 	c.b1.Purge() | ||||||
|  | 	c.b2.Purge() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Contains is used to check if the cache contains a key | ||||||
|  | // without updating recency or frequency. | ||||||
|  | func (c *ARCCache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.t1.Contains(key) || c.t2.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Peek is used to inspect the cache value of a key | ||||||
|  | // without updating recency or frequency. | ||||||
|  | func (c *ARCCache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	if val, ok := c.t1.Peek(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  | 	return c.t2.Peek(key) | ||||||
|  | } | ||||||
							
								
								
									
										114
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,114 @@ | |||||||
|  | // This package provides a simple LRU cache. It is based on the | ||||||
|  | // LRU implementation in groupcache: | ||||||
|  | // https://github.com/golang/groupcache/tree/master/lru | ||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Cache is a thread-safe fixed size LRU cache. | ||||||
|  | type Cache struct { | ||||||
|  | 	lru  *simplelru.LRU | ||||||
|  | 	lock sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New creates an LRU of the given size | ||||||
|  | func New(size int) (*Cache, error) { | ||||||
|  | 	return NewWithEvict(size, nil) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewWithEvict constructs a fixed size cache with the given eviction | ||||||
|  | // callback. | ||||||
|  | func NewWithEvict(size int, onEvicted func(key interface{}, value interface{})) (*Cache, error) { | ||||||
|  | 	lru, err := simplelru.NewLRU(size, simplelru.EvictCallback(onEvicted)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	c := &Cache{ | ||||||
|  | 		lru: lru, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to completely clear the cache | ||||||
|  | func (c *Cache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.Purge() | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||||
|  | func (c *Cache) Add(key, value interface{}) bool { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	return c.lru.Add(key, value) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *Cache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	return c.lru.Get(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Check if a key is in the cache, without updating the recent-ness | ||||||
|  | // or deleting it for being stale. | ||||||
|  | func (c *Cache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Returns the key value (or undefined if not found) without updating | ||||||
|  | // the "recently used"-ness of the key. | ||||||
|  | func (c *Cache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Peek(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ContainsOrAdd checks if a key is in the cache  without updating the | ||||||
|  | // recent-ness or deleting it for being stale,  and if not, adds the value. | ||||||
|  | // Returns whether found and whether an eviction occurred. | ||||||
|  | func (c *Cache) ContainsOrAdd(key, value interface{}) (ok, evict bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	if c.lru.Contains(key) { | ||||||
|  | 		return true, false | ||||||
|  | 	} else { | ||||||
|  | 		evict := c.lru.Add(key, value) | ||||||
|  | 		return false, evict | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove removes the provided key from the cache. | ||||||
|  | func (c *Cache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.Remove(key) | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RemoveOldest removes the oldest item from the cache. | ||||||
|  | func (c *Cache) RemoveOldest() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.RemoveOldest() | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns a slice of the keys in the cache, from oldest to newest. | ||||||
|  | func (c *Cache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Keys() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of items in the cache. | ||||||
|  | func (c *Cache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Len() | ||||||
|  | } | ||||||
							
								
								
									
										160
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										160
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,160 @@ | |||||||
|  | package simplelru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"container/list" | ||||||
|  | 	"errors" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // EvictCallback is used to get a callback when a cache entry is evicted | ||||||
|  | type EvictCallback func(key interface{}, value interface{}) | ||||||
|  |  | ||||||
|  | // LRU implements a non-thread safe fixed size LRU cache | ||||||
|  | type LRU struct { | ||||||
|  | 	size      int | ||||||
|  | 	evictList *list.List | ||||||
|  | 	items     map[interface{}]*list.Element | ||||||
|  | 	onEvict   EvictCallback | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // entry is used to hold a value in the evictList | ||||||
|  | type entry struct { | ||||||
|  | 	key   interface{} | ||||||
|  | 	value interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewLRU constructs an LRU of the given size | ||||||
|  | func NewLRU(size int, onEvict EvictCallback) (*LRU, error) { | ||||||
|  | 	if size <= 0 { | ||||||
|  | 		return nil, errors.New("Must provide a positive size") | ||||||
|  | 	} | ||||||
|  | 	c := &LRU{ | ||||||
|  | 		size:      size, | ||||||
|  | 		evictList: list.New(), | ||||||
|  | 		items:     make(map[interface{}]*list.Element), | ||||||
|  | 		onEvict:   onEvict, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to completely clear the cache | ||||||
|  | func (c *LRU) Purge() { | ||||||
|  | 	for k, v := range c.items { | ||||||
|  | 		if c.onEvict != nil { | ||||||
|  | 			c.onEvict(k, v.Value.(*entry).value) | ||||||
|  | 		} | ||||||
|  | 		delete(c.items, k) | ||||||
|  | 	} | ||||||
|  | 	c.evictList.Init() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||||
|  | func (c *LRU) Add(key, value interface{}) bool { | ||||||
|  | 	// Check for existing item | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.evictList.MoveToFront(ent) | ||||||
|  | 		ent.Value.(*entry).value = value | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add new item | ||||||
|  | 	ent := &entry{key, value} | ||||||
|  | 	entry := c.evictList.PushFront(ent) | ||||||
|  | 	c.items[key] = entry | ||||||
|  |  | ||||||
|  | 	evict := c.evictList.Len() > c.size | ||||||
|  | 	// Verify size not exceeded | ||||||
|  | 	if evict { | ||||||
|  | 		c.removeOldest() | ||||||
|  | 	} | ||||||
|  | 	return evict | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *LRU) Get(key interface{}) (value interface{}, ok bool) { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.evictList.MoveToFront(ent) | ||||||
|  | 		return ent.Value.(*entry).value, true | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Check if a key is in the cache, without updating the recent-ness | ||||||
|  | // or deleting it for being stale. | ||||||
|  | func (c *LRU) Contains(key interface{}) (ok bool) { | ||||||
|  | 	_, ok = c.items[key] | ||||||
|  | 	return ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Returns the key value (or undefined if not found) without updating | ||||||
|  | // the "recently used"-ness of the key. | ||||||
|  | func (c *LRU) Peek(key interface{}) (value interface{}, ok bool) { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		return ent.Value.(*entry).value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove removes the provided key from the cache, returning if the | ||||||
|  | // key was contained. | ||||||
|  | func (c *LRU) Remove(key interface{}) bool { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RemoveOldest removes the oldest item from the cache. | ||||||
|  | func (c *LRU) RemoveOldest() (interface{}, interface{}, bool) { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 		kv := ent.Value.(*entry) | ||||||
|  | 		return kv.key, kv.value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetOldest returns the oldest entry | ||||||
|  | func (c *LRU) GetOldest() (interface{}, interface{}, bool) { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		kv := ent.Value.(*entry) | ||||||
|  | 		return kv.key, kv.value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns a slice of the keys in the cache, from oldest to newest. | ||||||
|  | func (c *LRU) Keys() []interface{} { | ||||||
|  | 	keys := make([]interface{}, len(c.items)) | ||||||
|  | 	i := 0 | ||||||
|  | 	for ent := c.evictList.Back(); ent != nil; ent = ent.Prev() { | ||||||
|  | 		keys[i] = ent.Value.(*entry).key | ||||||
|  | 		i++ | ||||||
|  | 	} | ||||||
|  | 	return keys | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of items in the cache. | ||||||
|  | func (c *LRU) Len() int { | ||||||
|  | 	return c.evictList.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // removeOldest removes the oldest item from the cache. | ||||||
|  | func (c *LRU) removeOldest() { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // removeElement is used to remove a given list element from the cache | ||||||
|  | func (c *LRU) removeElement(e *list.Element) { | ||||||
|  | 	c.evictList.Remove(e) | ||||||
|  | 	kv := e.Value.(*entry) | ||||||
|  | 	delete(c.items, kv.key) | ||||||
|  | 	if c.onEvict != nil { | ||||||
|  | 		c.onEvict(kv.key, kv.value) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										21
									
								
								vendor/github.com/lrstanley/girc/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/lrstanley/girc/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,21 @@ | |||||||
|  | MIT License | ||||||
|  |  | ||||||
|  | Copyright (c) 2016 Liam Stanley <me@liamstanley.io> | ||||||
|  |  | ||||||
|  | Permission is hereby granted, free of charge, to any person obtaining a copy | ||||||
|  | of this software and associated documentation files (the "Software"), to deal | ||||||
|  | in the Software without restriction, including without limitation the rights | ||||||
|  | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||||||
|  | copies of the Software, and to permit persons to whom the Software is | ||||||
|  | furnished to do so, subject to the following conditions: | ||||||
|  |  | ||||||
|  | The above copyright notice and this permission notice shall be included in all | ||||||
|  | copies or substantial portions of the Software. | ||||||
|  |  | ||||||
|  | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||||||
|  | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||||||
|  | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||||||
|  | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||||||
|  | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||||||
|  | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | ||||||
|  | SOFTWARE. | ||||||
							
								
								
									
										518
									
								
								vendor/github.com/lrstanley/girc/builtin.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										518
									
								
								vendor/github.com/lrstanley/girc/builtin.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,518 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // registerBuiltin sets up built-in handlers, based on client | ||||||
|  | // configuration. | ||||||
|  | func (c *Client) registerBuiltins() { | ||||||
|  | 	c.debug.Print("registering built-in handlers") | ||||||
|  | 	c.Handlers.mu.Lock() | ||||||
|  |  | ||||||
|  | 	// Built-in things that should always be supported. | ||||||
|  | 	c.Handlers.register(true, RPL_WELCOME, HandlerFunc(func(c *Client, e Event) { | ||||||
|  | 		go handleConnect(c, e) | ||||||
|  | 	})) | ||||||
|  | 	c.Handlers.register(true, PING, HandlerFunc(handlePING)) | ||||||
|  | 	c.Handlers.register(true, PONG, HandlerFunc(handlePONG)) | ||||||
|  |  | ||||||
|  | 	if !c.Config.disableTracking { | ||||||
|  | 		// Joins/parts/anything that may add/remove/rename users. | ||||||
|  | 		c.Handlers.register(true, JOIN, HandlerFunc(handleJOIN)) | ||||||
|  | 		c.Handlers.register(true, PART, HandlerFunc(handlePART)) | ||||||
|  | 		c.Handlers.register(true, KICK, HandlerFunc(handleKICK)) | ||||||
|  | 		c.Handlers.register(true, QUIT, HandlerFunc(handleQUIT)) | ||||||
|  | 		c.Handlers.register(true, NICK, HandlerFunc(handleNICK)) | ||||||
|  | 		c.Handlers.register(true, RPL_NAMREPLY, HandlerFunc(handleNAMES)) | ||||||
|  |  | ||||||
|  | 		// Modes. | ||||||
|  | 		c.Handlers.register(true, MODE, HandlerFunc(handleMODE)) | ||||||
|  | 		c.Handlers.register(true, RPL_CHANNELMODEIS, HandlerFunc(handleMODE)) | ||||||
|  |  | ||||||
|  | 		// WHO/WHOX responses. | ||||||
|  | 		c.Handlers.register(true, RPL_WHOREPLY, HandlerFunc(handleWHO)) | ||||||
|  | 		c.Handlers.register(true, RPL_WHOSPCRPL, HandlerFunc(handleWHO)) | ||||||
|  |  | ||||||
|  | 		// Other misc. useful stuff. | ||||||
|  | 		c.Handlers.register(true, TOPIC, HandlerFunc(handleTOPIC)) | ||||||
|  | 		c.Handlers.register(true, RPL_TOPIC, HandlerFunc(handleTOPIC)) | ||||||
|  | 		c.Handlers.register(true, RPL_MYINFO, HandlerFunc(handleMYINFO)) | ||||||
|  | 		c.Handlers.register(true, RPL_ISUPPORT, HandlerFunc(handleISUPPORT)) | ||||||
|  | 		c.Handlers.register(true, RPL_MOTDSTART, HandlerFunc(handleMOTD)) | ||||||
|  | 		c.Handlers.register(true, RPL_MOTD, HandlerFunc(handleMOTD)) | ||||||
|  |  | ||||||
|  | 		// Keep users lastactive times up to date. | ||||||
|  | 		c.Handlers.register(true, PRIVMSG, HandlerFunc(updateLastActive)) | ||||||
|  | 		c.Handlers.register(true, NOTICE, HandlerFunc(updateLastActive)) | ||||||
|  | 		c.Handlers.register(true, TOPIC, HandlerFunc(updateLastActive)) | ||||||
|  | 		c.Handlers.register(true, KICK, HandlerFunc(updateLastActive)) | ||||||
|  |  | ||||||
|  | 		// CAP IRCv3-specific tracking and functionality. | ||||||
|  | 		c.Handlers.register(true, CAP, HandlerFunc(handleCAP)) | ||||||
|  | 		c.Handlers.register(true, CAP_CHGHOST, HandlerFunc(handleCHGHOST)) | ||||||
|  | 		c.Handlers.register(true, CAP_AWAY, HandlerFunc(handleAWAY)) | ||||||
|  | 		c.Handlers.register(true, CAP_ACCOUNT, HandlerFunc(handleACCOUNT)) | ||||||
|  | 		c.Handlers.register(true, ALL_EVENTS, HandlerFunc(handleTags)) | ||||||
|  |  | ||||||
|  | 		// SASL IRCv3 support. | ||||||
|  | 		c.Handlers.register(true, AUTHENTICATE, HandlerFunc(handleSASL)) | ||||||
|  | 		c.Handlers.register(true, RPL_SASLSUCCESS, HandlerFunc(handleSASL)) | ||||||
|  | 		c.Handlers.register(true, RPL_NICKLOCKED, HandlerFunc(handleSASLError)) | ||||||
|  | 		c.Handlers.register(true, ERR_SASLFAIL, HandlerFunc(handleSASLError)) | ||||||
|  | 		c.Handlers.register(true, ERR_SASLTOOLONG, HandlerFunc(handleSASLError)) | ||||||
|  | 		c.Handlers.register(true, ERR_SASLABORTED, HandlerFunc(handleSASLError)) | ||||||
|  | 		c.Handlers.register(true, RPL_SASLMECHS, HandlerFunc(handleSASLError)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Nickname collisions. | ||||||
|  | 	c.Handlers.register(true, ERR_NICKNAMEINUSE, HandlerFunc(nickCollisionHandler)) | ||||||
|  | 	c.Handlers.register(true, ERR_NICKCOLLISION, HandlerFunc(nickCollisionHandler)) | ||||||
|  | 	c.Handlers.register(true, ERR_UNAVAILRESOURCE, HandlerFunc(nickCollisionHandler)) | ||||||
|  |  | ||||||
|  | 	c.Handlers.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleConnect is a helper function which lets the client know that enough | ||||||
|  | // time has passed and now they can send commands. | ||||||
|  | // | ||||||
|  | // Should always run in separate thread due to blocking delay. | ||||||
|  | func handleConnect(c *Client, e Event) { | ||||||
|  | 	// This should be the nick that the server gives us. 99% of the time, it's | ||||||
|  | 	// the one we supplied during connection, but some networks will rename | ||||||
|  | 	// users on connect. | ||||||
|  | 	if len(e.Params) > 0 { | ||||||
|  | 		c.state.Lock() | ||||||
|  | 		c.state.nick = e.Params[0] | ||||||
|  | 		c.state.Unlock() | ||||||
|  |  | ||||||
|  | 		c.state.notify(c, UPDATE_GENERAL) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	time.Sleep(2 * time.Second) | ||||||
|  | 	c.RunHandlers(&Event{Command: CONNECTED, Trailing: c.Server()}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // nickCollisionHandler helps prevent the client from having conflicting | ||||||
|  | // nicknames with another bot, user, etc. | ||||||
|  | func nickCollisionHandler(c *Client, e Event) { | ||||||
|  | 	if c.Config.HandleNickCollide == nil { | ||||||
|  | 		c.Cmd.Nick(c.GetNick() + "_") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.Cmd.Nick(c.Config.HandleNickCollide(c.GetNick())) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handlePING helps respond to ping requests from the server. | ||||||
|  | func handlePING(c *Client, e Event) { | ||||||
|  | 	c.Cmd.Pong(e.Trailing) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func handlePONG(c *Client, e Event) { | ||||||
|  | 	c.conn.lastPong = time.Now() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleJOIN ensures that the state has updated users and channels. | ||||||
|  | func handleJOIN(c *Client, e Event) { | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var channelName string | ||||||
|  | 	if len(e.Params) > 0 { | ||||||
|  | 		channelName = e.Params[0] | ||||||
|  | 	} else { | ||||||
|  | 		channelName = e.Trailing | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  |  | ||||||
|  | 	channel := c.state.lookupChannel(channelName) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		if ok := c.state.createChannel(channelName); !ok { | ||||||
|  | 			c.state.Unlock() | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		channel = c.state.lookupChannel(channelName) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user == nil { | ||||||
|  | 		if ok := c.state.createUser(e.Source.Name); !ok { | ||||||
|  | 			c.state.Unlock() | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		user = c.state.lookupUser(e.Source.Name) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	defer c.state.notify(c, UPDATE_STATE) | ||||||
|  |  | ||||||
|  | 	channel.addUser(user.Nick) | ||||||
|  | 	user.addChannel(channel.Name) | ||||||
|  |  | ||||||
|  | 	// Assume extended-join (ircv3). | ||||||
|  | 	if len(e.Params) == 2 { | ||||||
|  | 		if e.Params[1] != "*" { | ||||||
|  | 			user.Extras.Account = e.Params[1] | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(e.Trailing) > 0 { | ||||||
|  | 			user.Extras.Name = e.Trailing | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  |  | ||||||
|  | 	if e.Source.Name == c.GetNick() { | ||||||
|  | 		// If it's us, don't just add our user to the list. Run a WHO which | ||||||
|  | 		// will tell us who exactly is in the entire channel. | ||||||
|  | 		c.Send(&Event{Command: WHO, Params: []string{channelName, "%tacuhnr,1"}}) | ||||||
|  |  | ||||||
|  | 		// Also send a MODE to obtain the list of channel modes. | ||||||
|  | 		c.Send(&Event{Command: MODE, Params: []string{channelName}}) | ||||||
|  |  | ||||||
|  | 		// Update our ident and host too, in state -- since there is no | ||||||
|  | 		// cleaner method to do this. | ||||||
|  | 		c.state.Lock() | ||||||
|  | 		c.state.ident = e.Source.Ident | ||||||
|  | 		c.state.host = e.Source.Host | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Only WHO the user, which is more efficient. | ||||||
|  | 	c.Send(&Event{Command: WHO, Params: []string{e.Source.Name, "%tacuhnr,1"}}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handlePART ensures that the state is clean of old user and channel entries. | ||||||
|  | func handlePART(c *Client, e Event) { | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var channel string | ||||||
|  | 	if len(e.Params) > 0 { | ||||||
|  | 		channel = e.Params[0] | ||||||
|  | 	} else { | ||||||
|  | 		channel = e.Trailing | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if channel == "" { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	defer c.state.notify(c, UPDATE_STATE) | ||||||
|  |  | ||||||
|  | 	if e.Source.Name == c.GetNick() { | ||||||
|  | 		c.state.Lock() | ||||||
|  | 		c.state.deleteChannel(channel) | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	c.state.deleteUser(channel, e.Source.Name) | ||||||
|  | 	c.state.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleTOPIC handles incoming TOPIC events and keeps channel tracking info | ||||||
|  | // updated with the latest channel topic. | ||||||
|  | func handleTOPIC(c *Client, e Event) { | ||||||
|  | 	var name string | ||||||
|  | 	switch len(e.Params) { | ||||||
|  | 	case 0: | ||||||
|  | 		return | ||||||
|  | 	case 1: | ||||||
|  | 		name = e.Params[0] | ||||||
|  | 	default: | ||||||
|  | 		name = e.Params[len(e.Params)-1] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	channel := c.state.lookupChannel(name) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel.Topic = e.Trailing | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handlWHO updates our internal tracking of users/channels with WHO/WHOX | ||||||
|  | // information. | ||||||
|  | func handleWHO(c *Client, e Event) { | ||||||
|  | 	var ident, host, nick, account, realname string | ||||||
|  |  | ||||||
|  | 	// Assume WHOX related. | ||||||
|  | 	if e.Command == RPL_WHOSPCRPL { | ||||||
|  | 		if len(e.Params) != 7 { | ||||||
|  | 			// Assume there was some form of error or invalid WHOX response. | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if e.Params[1] != "1" { | ||||||
|  | 			// We should always be sending 1, and we should receive 1. If this | ||||||
|  | 			// is anything but, then we didn't send the request and we can | ||||||
|  | 			// ignore it. | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		ident, host, nick, account = e.Params[3], e.Params[4], e.Params[5], e.Params[6] | ||||||
|  | 		realname = e.Trailing | ||||||
|  | 	} else { | ||||||
|  | 		// Assume RPL_WHOREPLY. | ||||||
|  | 		ident, host, nick = e.Params[2], e.Params[3], e.Params[5] | ||||||
|  | 		if len(e.Trailing) > 2 { | ||||||
|  | 			realname = e.Trailing[2:] | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	user := c.state.lookupUser(nick) | ||||||
|  | 	if user == nil { | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user.Host = host | ||||||
|  | 	user.Ident = ident | ||||||
|  | 	user.Extras.Name = realname | ||||||
|  |  | ||||||
|  | 	if account != "0" { | ||||||
|  | 		user.Extras.Account = account | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleKICK ensures that users are cleaned up after being kicked from the | ||||||
|  | // channel | ||||||
|  | func handleKICK(c *Client, e Event) { | ||||||
|  | 	if len(e.Params) < 2 { | ||||||
|  | 		// Needs at least channel and user. | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	defer c.state.notify(c, UPDATE_STATE) | ||||||
|  |  | ||||||
|  | 	if e.Params[1] == c.GetNick() { | ||||||
|  | 		c.state.Lock() | ||||||
|  | 		c.state.deleteChannel(e.Params[0]) | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Assume it's just another user. | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	c.state.deleteUser(e.Params[0], e.Params[1]) | ||||||
|  | 	c.state.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleNICK ensures that users are renamed in state, or the client name is | ||||||
|  | // up to date. | ||||||
|  | func handleNICK(c *Client, e Event) { | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	// renameUser updates the LastActive time automatically. | ||||||
|  | 	if len(e.Params) == 1 { | ||||||
|  | 		c.state.renameUser(e.Source.Name, e.Params[0]) | ||||||
|  | 	} else if len(e.Trailing) > 0 { | ||||||
|  | 		c.state.renameUser(e.Source.Name, e.Trailing) | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleQUIT handles users that are quitting from the network. | ||||||
|  | func handleQUIT(c *Client, e Event) { | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Source.Name == c.GetNick() { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	c.state.deleteUser("", e.Source.Name) | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleMYINFO handles incoming MYINFO events -- these are commonly used | ||||||
|  | // to tell us what the server name is, what version of software is being used | ||||||
|  | // as well as what channel and user modes are being used on the server. | ||||||
|  | func handleMYINFO(c *Client, e Event) { | ||||||
|  | 	// Malformed or odd output. As this can differ strongly between networks, | ||||||
|  | 	// just skip it. | ||||||
|  | 	if len(e.Params) < 3 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	c.state.serverOptions["SERVER"] = e.Params[1] | ||||||
|  | 	c.state.serverOptions["VERSION"] = e.Params[2] | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_GENERAL) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleISUPPORT handles incoming RPL_ISUPPORT (also known as RPL_PROTOCTL) | ||||||
|  | // events. These commonly contain the server capabilities and limitations. | ||||||
|  | // For example, things like max channel name length, or nickname length. | ||||||
|  | func handleISUPPORT(c *Client, e Event) { | ||||||
|  | 	// Must be a ISUPPORT-based message. 005 is also used for server bounce | ||||||
|  | 	// related things, so this handler may be triggered during other | ||||||
|  | 	// situations. | ||||||
|  |  | ||||||
|  | 	// Also known as RPL_PROTOCTL. | ||||||
|  | 	if !strings.HasSuffix(e.Trailing, "this server") { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Must have at least one configuration. | ||||||
|  | 	if len(e.Params) < 2 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	// Skip the first parameter, as it's our nickname. | ||||||
|  | 	for i := 1; i < len(e.Params); i++ { | ||||||
|  | 		j := strings.IndexByte(e.Params[i], 0x3D) // = | ||||||
|  |  | ||||||
|  | 		if j < 1 || (j+1) == len(e.Params[i]) { | ||||||
|  | 			c.state.serverOptions[e.Params[i]] = "" | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		name := e.Params[i][0:j] | ||||||
|  | 		val := e.Params[i][j+1:] | ||||||
|  | 		c.state.serverOptions[name] = val | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_GENERAL) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleMOTD handles incoming MOTD messages and buffers them up for use with | ||||||
|  | // Client.ServerMOTD(). | ||||||
|  | func handleMOTD(c *Client, e Event) { | ||||||
|  | 	c.state.Lock() | ||||||
|  |  | ||||||
|  | 	defer c.state.notify(c, UPDATE_GENERAL) | ||||||
|  |  | ||||||
|  | 	// Beginning of the MOTD. | ||||||
|  | 	if e.Command == RPL_MOTDSTART { | ||||||
|  | 		c.state.motd = "" | ||||||
|  |  | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Otherwise, assume we're getting sent the MOTD line-by-line. | ||||||
|  | 	if len(c.state.motd) != 0 { | ||||||
|  | 		e.Trailing = "\n" + e.Trailing | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.motd += e.Trailing | ||||||
|  | 	c.state.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleNAMES handles incoming NAMES queries, of which lists all users in | ||||||
|  | // a given channel. Optionally also obtains ident/host values, as well as | ||||||
|  | // permissions for each user, depending on what capabilities are enabled. | ||||||
|  | func handleNAMES(c *Client, e Event) { | ||||||
|  | 	if len(e.Params) < 1 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel := c.state.lookupChannel(e.Params[len(e.Params)-1]) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	parts := strings.Split(e.Trailing, " ") | ||||||
|  |  | ||||||
|  | 	var host, ident, modes, nick string | ||||||
|  | 	var ok bool | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	for i := 0; i < len(parts); i++ { | ||||||
|  | 		modes, nick, ok = parseUserPrefix(parts[i]) | ||||||
|  | 		if !ok { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// If userhost-in-names. | ||||||
|  | 		if strings.Contains(nick, "@") { | ||||||
|  | 			s := ParseSource(nick) | ||||||
|  | 			if s == nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			host = s.Host | ||||||
|  | 			nick = s.Name | ||||||
|  | 			ident = s.Ident | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if !IsValidNick(nick) { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		c.state.createUser(nick) | ||||||
|  | 		user := c.state.lookupUser(nick) | ||||||
|  | 		if user == nil { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		user.addChannel(channel.Name) | ||||||
|  | 		channel.addUser(nick) | ||||||
|  |  | ||||||
|  | 		// Add necessary userhost-in-names data into the user. | ||||||
|  | 		if host != "" { | ||||||
|  | 			user.Host = host | ||||||
|  | 		} | ||||||
|  | 		if ident != "" { | ||||||
|  | 			user.Ident = ident | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Don't append modes, overwrite them. | ||||||
|  | 		perms, _ := user.Perms.Lookup(channel.Name) | ||||||
|  | 		perms.set(modes, false) | ||||||
|  | 		user.Perms.set(channel.Name, perms) | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // updateLastActive is a wrapper for any event which the source author | ||||||
|  | // should have it's LastActive time updated. This is useful for things like | ||||||
|  | // a KICK where we know they are active, as they just kicked another user, | ||||||
|  | // even though they may not be talking. | ||||||
|  | func updateLastActive(c *Client, e Event) { | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  |  | ||||||
|  | 	// Update the users last active time, if they exist. | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user == nil { | ||||||
|  | 		c.state.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user.LastActive = time.Now() | ||||||
|  | 	c.state.Unlock() | ||||||
|  | } | ||||||
							
								
								
									
										639
									
								
								vendor/github.com/lrstanley/girc/cap.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										639
									
								
								vendor/github.com/lrstanley/girc/cap.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,639 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"encoding/base64" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io" | ||||||
|  | 	"sort" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var possibleCap = map[string][]string{ | ||||||
|  | 	"account-notify":    nil, | ||||||
|  | 	"account-tag":       nil, | ||||||
|  | 	"away-notify":       nil, | ||||||
|  | 	"batch":             nil, | ||||||
|  | 	"cap-notify":        nil, | ||||||
|  | 	"chghost":           nil, | ||||||
|  | 	"extended-join":     nil, | ||||||
|  | 	"invite-notify":     nil, | ||||||
|  | 	"message-tags":      nil, | ||||||
|  | 	"multi-prefix":      nil, | ||||||
|  | 	"userhost-in-names": nil, | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Client) listCAP() { | ||||||
|  | 	if !c.Config.disableTracking { | ||||||
|  | 		c.write(&Event{Command: CAP, Params: []string{CAP_LS, "302"}}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func possibleCapList(c *Client) map[string][]string { | ||||||
|  | 	out := make(map[string][]string) | ||||||
|  |  | ||||||
|  | 	if c.Config.SASL != nil { | ||||||
|  | 		out["sasl"] = nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for k := range c.Config.SupportedCaps { | ||||||
|  | 		out[k] = c.Config.SupportedCaps[k] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for k := range possibleCap { | ||||||
|  | 		out[k] = possibleCap[k] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func parseCap(raw string) map[string][]string { | ||||||
|  | 	out := make(map[string][]string) | ||||||
|  | 	parts := strings.Split(raw, " ") | ||||||
|  |  | ||||||
|  | 	var val int | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(parts); i++ { | ||||||
|  | 		val = strings.IndexByte(parts[i], prefixTagValue) // = | ||||||
|  |  | ||||||
|  | 		// No value splitter, or has splitter but no trailing value. | ||||||
|  | 		if val < 1 || len(parts[i]) < val+1 { | ||||||
|  | 			// The capability doesn't contain a value. | ||||||
|  | 			out[parts[i]] = nil | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		out[parts[i][:val]] = strings.Split(parts[i][val+1:], ",") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCAP attempts to find out what IRCv3 capabilities the server supports. | ||||||
|  | // This will lock further registration until we have acknowledged the | ||||||
|  | // capabilities. | ||||||
|  | func handleCAP(c *Client, e Event) { | ||||||
|  | 	if len(e.Params) >= 2 && (e.Params[1] == CAP_NEW || e.Params[1] == CAP_DEL) { | ||||||
|  | 		c.listCAP() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// We can assume there was a failure attempting to enable a capability. | ||||||
|  | 	if len(e.Params) == 2 && e.Params[1] == CAP_NAK { | ||||||
|  | 		// Let the server know that we're done. | ||||||
|  | 		c.write(&Event{Command: CAP, Params: []string{CAP_END}}) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	possible := possibleCapList(c) | ||||||
|  |  | ||||||
|  | 	if len(e.Params) >= 2 && len(e.Trailing) > 1 && e.Params[1] == CAP_LS { | ||||||
|  | 		c.state.Lock() | ||||||
|  |  | ||||||
|  | 		caps := parseCap(e.Trailing) | ||||||
|  |  | ||||||
|  | 		for k := range caps { | ||||||
|  | 			if _, ok := possible[k]; !ok { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if len(possible[k]) == 0 || len(caps[k]) == 0 { | ||||||
|  | 				c.state.tmpCap = append(c.state.tmpCap, k) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			var contains bool | ||||||
|  | 			for i := 0; i < len(caps[k]); i++ { | ||||||
|  | 				for j := 0; j < len(possible[k]); j++ { | ||||||
|  | 					if caps[k][i] == possible[k][j] { | ||||||
|  | 						// Assume we have a matching split value. | ||||||
|  | 						contains = true | ||||||
|  | 						goto checkcontains | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 		checkcontains: | ||||||
|  | 			if !contains { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			c.state.tmpCap = append(c.state.tmpCap, k) | ||||||
|  | 		} | ||||||
|  | 		c.state.Unlock() | ||||||
|  |  | ||||||
|  | 		// Indicates if this is a multi-line LS. (2 args means it's the | ||||||
|  | 		// last LS). | ||||||
|  | 		if len(e.Params) == 2 { | ||||||
|  | 			// If we support no caps, just ack the CAP message and END. | ||||||
|  | 			if len(c.state.tmpCap) == 0 { | ||||||
|  | 				c.write(&Event{Command: CAP, Params: []string{CAP_END}}) | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Let them know which ones we'd like to enable. | ||||||
|  | 			c.write(&Event{Command: CAP, Params: []string{CAP_REQ}, Trailing: strings.Join(c.state.tmpCap, " ")}) | ||||||
|  |  | ||||||
|  | 			// Re-initialize the tmpCap, so if we get multiple 'CAP LS' requests | ||||||
|  | 			// due to cap-notify, we can re-evaluate what we can support. | ||||||
|  | 			c.state.Lock() | ||||||
|  | 			c.state.tmpCap = []string{} | ||||||
|  | 			c.state.Unlock() | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(e.Params) == 2 && len(e.Trailing) > 1 && e.Params[1] == CAP_ACK { | ||||||
|  | 		c.state.Lock() | ||||||
|  | 		c.state.enabledCap = strings.Split(e.Trailing, " ") | ||||||
|  |  | ||||||
|  | 		// Do we need to do sasl auth? | ||||||
|  | 		wantsSASL := false | ||||||
|  | 		for i := 0; i < len(c.state.enabledCap); i++ { | ||||||
|  | 			if c.state.enabledCap[i] == "sasl" { | ||||||
|  | 				wantsSASL = true | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		c.state.Unlock() | ||||||
|  |  | ||||||
|  | 		if wantsSASL { | ||||||
|  | 			c.write(&Event{Command: AUTHENTICATE, Params: []string{c.Config.SASL.Method()}}) | ||||||
|  | 			// Don't "CAP END", since we want to authenticate. | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Let the server know that we're done. | ||||||
|  | 		c.write(&Event{Command: CAP, Params: []string{CAP_END}}) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SASLMech is an representation of what a SASL mechanism should support. | ||||||
|  | // See SASLExternal and SASLPlain for implementations of this. | ||||||
|  | type SASLMech interface { | ||||||
|  | 	// Method returns the uppercase version of the SASL mechanism name. | ||||||
|  | 	Method() string | ||||||
|  | 	// Encode returns the response that the SASL mechanism wants to use. If | ||||||
|  | 	// the returned string is empty (e.g. the mechanism gives up), the handler | ||||||
|  | 	// will attempt to panic, as expectation is that if SASL authentication | ||||||
|  | 	// fails, the client will disconnect. | ||||||
|  | 	Encode(params []string) (output string) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SASLExternal implements the "EXTERNAL" SASL type. | ||||||
|  | type SASLExternal struct { | ||||||
|  | 	// Identity is an optional field which allows the client to specify | ||||||
|  | 	// pre-authentication identification. This means that EXTERNAL will | ||||||
|  | 	// supply this in the initial response. This usually isn't needed (e.g. | ||||||
|  | 	// CertFP). | ||||||
|  | 	Identity string `json:"identity"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Method identifies what type of SASL this implements. | ||||||
|  | func (sasl *SASLExternal) Method() string { | ||||||
|  | 	return "EXTERNAL" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Encode for external SALS authentication should really only return a "+", | ||||||
|  | // unless the user has specified pre-authentication or identification data. | ||||||
|  | // See https://tools.ietf.org/html/rfc4422#appendix-A for more info. | ||||||
|  | func (sasl *SASLExternal) Encode(params []string) string { | ||||||
|  | 	if len(params) != 1 || params[0] != "+" { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if sasl.Identity != "" { | ||||||
|  | 		return sasl.Identity | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return "+" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SASLPlain contains the user and password needed for PLAIN SASL authentication. | ||||||
|  | type SASLPlain struct { | ||||||
|  | 	User string `json:"user"` // User is the username for SASL. | ||||||
|  | 	Pass string `json:"pass"` // Pass is the password for SASL. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Method identifies what type of SASL this implements. | ||||||
|  | func (sasl *SASLPlain) Method() string { | ||||||
|  | 	return "PLAIN" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Encode encodes the plain user+password into a SASL PLAIN implementation. | ||||||
|  | // See https://tools.ietf.org/rfc/rfc4422.txt for more info. | ||||||
|  | func (sasl *SASLPlain) Encode(params []string) string { | ||||||
|  | 	if len(params) != 1 || params[0] != "+" { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	in := []byte(sasl.User) | ||||||
|  |  | ||||||
|  | 	in = append(in, 0x0) | ||||||
|  | 	in = append(in, []byte(sasl.User)...) | ||||||
|  | 	in = append(in, 0x0) | ||||||
|  | 	in = append(in, []byte(sasl.Pass)...) | ||||||
|  |  | ||||||
|  | 	return base64.StdEncoding.EncodeToString(in) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const saslChunkSize = 400 | ||||||
|  |  | ||||||
|  | func handleSASL(c *Client, e Event) { | ||||||
|  | 	if e.Command == RPL_SASLSUCCESS || e.Command == ERR_SASLALREADY { | ||||||
|  | 		// Let the server know that we're done. | ||||||
|  | 		c.write(&Event{Command: CAP, Params: []string{CAP_END}}) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Assume they want us to handle sending auth. | ||||||
|  | 	auth := c.Config.SASL.Encode(e.Params) | ||||||
|  |  | ||||||
|  | 	if auth == "" { | ||||||
|  | 		// Assume the SASL authentication method doesn't want to respond for | ||||||
|  | 		// some reason. The SASL spec and IRCv3 spec do not define a clear | ||||||
|  | 		// way to abort a SASL exchange, other than to disconnect, or proceed | ||||||
|  | 		// with CAP END. | ||||||
|  | 		c.rx <- &Event{Command: ERROR, Trailing: fmt.Sprintf( | ||||||
|  | 			"closing connection: invalid %s SASL configuration provided: %s", | ||||||
|  | 			c.Config.SASL.Method(), e.Trailing, | ||||||
|  | 		)} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Send in "saslChunkSize"-length byte chunks. If the last chuck is | ||||||
|  | 	// exactly "saslChunkSize" bytes, send a "AUTHENTICATE +" 0-byte | ||||||
|  | 	// acknowledgement response to let the server know that we're done. | ||||||
|  | 	for { | ||||||
|  | 		if len(auth) > saslChunkSize { | ||||||
|  | 			c.write(&Event{Command: AUTHENTICATE, Params: []string{auth[0 : saslChunkSize-1]}, Sensitive: true}) | ||||||
|  | 			auth = auth[saslChunkSize:] | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(auth) <= saslChunkSize { | ||||||
|  | 			c.write(&Event{Command: AUTHENTICATE, Params: []string{auth}, Sensitive: true}) | ||||||
|  |  | ||||||
|  | 			if len(auth) == 400 { | ||||||
|  | 				c.write(&Event{Command: AUTHENTICATE, Params: []string{"+"}}) | ||||||
|  | 			} | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func handleSASLError(c *Client, e Event) { | ||||||
|  | 	if c.Config.SASL == nil { | ||||||
|  | 		c.write(&Event{Command: CAP, Params: []string{CAP_END}}) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Authentication failed. The SASL spec and IRCv3 spec do not define a | ||||||
|  | 	// clear way to abort a SASL exchange, other than to disconnect, or | ||||||
|  | 	// proceed with CAP END. | ||||||
|  | 	c.rx <- &Event{Command: ERROR, Trailing: "closing connection: " + e.Trailing} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCHGHOST handles incoming IRCv3 hostname change events. CHGHOST is | ||||||
|  | // what occurs (when enabled) when a servers services change the hostname of | ||||||
|  | // a user. Traditionally, this was simply resolved with a quick QUIT and JOIN, | ||||||
|  | // however CHGHOST resolves this in a much cleaner fashion. | ||||||
|  | func handleCHGHOST(c *Client, e Event) { | ||||||
|  | 	if len(e.Params) != 2 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user != nil { | ||||||
|  | 		user.Ident = e.Params[0] | ||||||
|  | 		user.Host = e.Params[1] | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleAWAY handles incoming IRCv3 AWAY events, for which are sent both | ||||||
|  | // when users are no longer away, or when they are away. | ||||||
|  | func handleAWAY(c *Client, e Event) { | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user != nil { | ||||||
|  | 		user.Extras.Away = e.Trailing | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleACCOUNT handles incoming IRCv3 ACCOUNT events. ACCOUNT is sent when | ||||||
|  | // a user logs into an account, logs out of their account, or logs into a | ||||||
|  | // different account. The account backend is handled server-side, so this | ||||||
|  | // could be NickServ, X (undernet?), etc. | ||||||
|  | func handleACCOUNT(c *Client, e Event) { | ||||||
|  | 	if len(e.Params) != 1 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	account := e.Params[0] | ||||||
|  | 	if account == "*" { | ||||||
|  | 		account = "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user != nil { | ||||||
|  | 		user.Extras.Account = account | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleTags handles any messages that have tags that will affect state. (e.g. | ||||||
|  | // 'account' tags.) | ||||||
|  | func handleTags(c *Client, e Event) { | ||||||
|  | 	if len(e.Tags) == 0 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	account, ok := e.Tags.Get("account") | ||||||
|  | 	if !ok { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	user := c.state.lookupUser(e.Source.Name) | ||||||
|  | 	if user != nil { | ||||||
|  | 		user.Extras.Account = account | ||||||
|  | 	} | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	prefixTag      byte = 0x40 // @ | ||||||
|  | 	prefixTagValue byte = 0x3D // = | ||||||
|  | 	prefixUserTag  byte = 0x2B // + | ||||||
|  | 	tagSeparator   byte = 0x3B // ; | ||||||
|  | 	maxTagLength   int  = 511  // 510 + @ and " " (space), though space usually not included. | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Tags represents the key-value pairs in IRCv3 message tags. The map contains | ||||||
|  | // the encoded message-tag values. If the tag is present, it may still be | ||||||
|  | // empty. See Tags.Get() and Tags.Set() for use with getting/setting | ||||||
|  | // information within the tags. | ||||||
|  | // | ||||||
|  | // Note that retrieving and setting tags are not concurrent safe. If this is | ||||||
|  | // necessary, you will need to implement it yourself. | ||||||
|  | type Tags map[string]string | ||||||
|  |  | ||||||
|  | // ParseTags parses out the key-value map of tags. raw should only be the tag | ||||||
|  | // data, not a full message. For example: | ||||||
|  | //   @aaa=bbb;ccc;example.com/ddd=eee | ||||||
|  | // NOT: | ||||||
|  | //   @aaa=bbb;ccc;example.com/ddd=eee :nick!ident@host.com PRIVMSG me :Hello | ||||||
|  | func ParseTags(raw string) (t Tags) { | ||||||
|  | 	t = make(Tags) | ||||||
|  |  | ||||||
|  | 	if len(raw) > 0 && raw[0] == prefixTag { | ||||||
|  | 		raw = raw[1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	parts := strings.Split(raw, string(tagSeparator)) | ||||||
|  | 	var hasValue int | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(parts); i++ { | ||||||
|  | 		hasValue = strings.IndexByte(parts[i], prefixTagValue) | ||||||
|  |  | ||||||
|  | 		// The tag doesn't contain a value or has a splitter with no value. | ||||||
|  | 		if hasValue < 1 || len(parts[i]) < hasValue+1 { | ||||||
|  | 			if !validTag(parts[i]) { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			t[parts[i]] = "" | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Check if tag key or decoded value are invalid. | ||||||
|  | 		if !validTag(parts[i][:hasValue]) || !validTagValue(tagDecoder.Replace(parts[i][hasValue+1:])) { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		t[parts[i][:hasValue]] = parts[i][hasValue+1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return t | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len determines the length of the bytes representation of this tag map. This | ||||||
|  | // does not include the trailing space required when creating an event, but | ||||||
|  | // does include the tag prefix ("@"). | ||||||
|  | func (t Tags) Len() (length int) { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return len(t.Bytes()) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Count finds how many total tags that there are. | ||||||
|  | func (t Tags) Count() int { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return len(t) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Bytes returns a []byte representation of this tag map, including the tag | ||||||
|  | // prefix ("@"). Note that this will return the tags sorted, regardless of | ||||||
|  | // the order of how they were originally parsed. | ||||||
|  | func (t Tags) Bytes() []byte { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return []byte{} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	max := len(t) | ||||||
|  | 	if max == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	buffer := new(bytes.Buffer) | ||||||
|  | 	buffer.WriteByte(prefixTag) | ||||||
|  |  | ||||||
|  | 	var current int | ||||||
|  |  | ||||||
|  | 	// Sort the writing of tags so we can at least guarantee that they will | ||||||
|  | 	// be in order, and testable. | ||||||
|  | 	var names []string | ||||||
|  | 	for tagName := range t { | ||||||
|  | 		names = append(names, tagName) | ||||||
|  | 	} | ||||||
|  | 	sort.Strings(names) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(names); i++ { | ||||||
|  | 		// Trim at max allowed chars. | ||||||
|  | 		if (buffer.Len() + len(names[i]) + len(t[names[i]]) + 2) > maxTagLength { | ||||||
|  | 			return buffer.Bytes() | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		buffer.WriteString(names[i]) | ||||||
|  |  | ||||||
|  | 		// Write the value as necessary. | ||||||
|  | 		if len(t[names[i]]) > 0 { | ||||||
|  | 			buffer.WriteByte(prefixTagValue) | ||||||
|  | 			buffer.WriteString(t[names[i]]) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// add the separator ";" between tags. | ||||||
|  | 		if current < max-1 { | ||||||
|  | 			buffer.WriteByte(tagSeparator) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		current++ | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return buffer.Bytes() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a string representation of this tag map. | ||||||
|  | func (t Tags) String() string { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return string(t.Bytes()) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // writeTo writes the necessary tag bytes to an io.Writer, including a trailing | ||||||
|  | // space-separator. | ||||||
|  | func (t Tags) writeTo(w io.Writer) (n int, err error) { | ||||||
|  | 	b := t.Bytes() | ||||||
|  | 	if len(b) == 0 { | ||||||
|  | 		return n, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	n, err = w.Write(b) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return n, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var j int | ||||||
|  | 	j, err = w.Write([]byte{eventSpace}) | ||||||
|  | 	n += j | ||||||
|  |  | ||||||
|  | 	return n, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // tagDecode are encoded -> decoded pairs for replacement to decode. | ||||||
|  | var tagDecode = []string{ | ||||||
|  | 	"\\:", ";", | ||||||
|  | 	"\\s", " ", | ||||||
|  | 	"\\\\", "\\", | ||||||
|  | 	"\\r", "\r", | ||||||
|  | 	"\\n", "\n", | ||||||
|  | } | ||||||
|  | var tagDecoder = strings.NewReplacer(tagDecode...) | ||||||
|  |  | ||||||
|  | // tagEncode are decoded -> encoded pairs for replacement to decode. | ||||||
|  | var tagEncode = []string{ | ||||||
|  | 	";", "\\:", | ||||||
|  | 	" ", "\\s", | ||||||
|  | 	"\\", "\\\\", | ||||||
|  | 	"\r", "\\r", | ||||||
|  | 	"\n", "\\n", | ||||||
|  | } | ||||||
|  | var tagEncoder = strings.NewReplacer(tagEncode...) | ||||||
|  |  | ||||||
|  | // Get returns the unescaped value of given tag key. Note that this is not | ||||||
|  | // concurrent safe. | ||||||
|  | func (t Tags) Get(key string) (tag string, success bool) { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return "", false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, ok := t[key]; ok { | ||||||
|  | 		tag = tagDecoder.Replace(t[key]) | ||||||
|  | 		success = true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return tag, success | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Set escapes given value and saves it as the value for given key. Note that | ||||||
|  | // this is not concurrent safe. | ||||||
|  | func (t Tags) Set(key, value string) error { | ||||||
|  | 	if t == nil { | ||||||
|  | 		t = make(Tags) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !validTag(key) { | ||||||
|  | 		return fmt.Errorf("tag key %q is invalid", key) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	value = tagEncoder.Replace(value) | ||||||
|  |  | ||||||
|  | 	if len(value) > 0 && !validTagValue(value) { | ||||||
|  | 		return fmt.Errorf("tag value %q of key %q is invalid", value, key) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check to make sure it's not too long here. | ||||||
|  | 	if (t.Len() + len(key) + len(value) + 2) > maxTagLength { | ||||||
|  | 		return fmt.Errorf("unable to set tag %q [value %q]: tags too long for message", key, value) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	t[key] = value | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove deletes the tag frwom the tag map. | ||||||
|  | func (t Tags) Remove(key string) (success bool) { | ||||||
|  | 	if t == nil { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, success = t[key]; success { | ||||||
|  | 		delete(t, key) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return success | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // validTag validates an IRC tag. | ||||||
|  | func validTag(name string) bool { | ||||||
|  | 	if len(name) < 1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Allow user tags to be passed to validTag. | ||||||
|  | 	if len(name) >= 2 && name[0] == prefixUserTag { | ||||||
|  | 		name = name[1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(name); i++ { | ||||||
|  | 		// A-Z, a-z, 0-9, -/._ | ||||||
|  | 		if (name[i] < 0x41 || name[i] > 0x5A) && (name[i] < 0x61 || name[i] > 0x7A) && (name[i] < 0x2D || name[i] > 0x39) && name[i] != 0x5F { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // validTagValue valids a decoded IRC tag value. If the value is not decoded | ||||||
|  | // with tagDecoder first, it may be seen as invalid. | ||||||
|  | func validTagValue(value string) bool { | ||||||
|  | 	for i := 0; i < len(value); i++ { | ||||||
|  | 		// Don't allow any invisible chars within the tag, or semicolons. | ||||||
|  | 		if value[i] < 0x21 || value[i] > 0x7E || value[i] == 0x3B { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return true | ||||||
|  | } | ||||||
							
								
								
									
										615
									
								
								vendor/github.com/lrstanley/girc/client.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										615
									
								
								vendor/github.com/lrstanley/girc/client.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,615 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"log" | ||||||
|  | 	"runtime" | ||||||
|  | 	"sort" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Client contains all of the information necessary to run a single IRC | ||||||
|  | // client. | ||||||
|  | type Client struct { | ||||||
|  | 	// Config represents the configuration. Please take extra caution in that | ||||||
|  | 	// entries in this are not edited while the client is connected, to prevent | ||||||
|  | 	// data races. This is NOT concurrent safe to update. | ||||||
|  | 	Config Config | ||||||
|  | 	// rx is a buffer of events waiting to be processed. | ||||||
|  | 	rx chan *Event | ||||||
|  | 	// tx is a buffer of events waiting to be sent. | ||||||
|  | 	tx chan *Event | ||||||
|  | 	// state represents the throw-away state for the irc session. | ||||||
|  | 	state *state | ||||||
|  | 	// initTime represents the creation time of the client. | ||||||
|  | 	initTime time.Time | ||||||
|  | 	// Handlers is a handler which manages internal and external handlers. | ||||||
|  | 	Handlers *Caller | ||||||
|  | 	// CTCP is a handler which manages internal and external CTCP handlers. | ||||||
|  | 	CTCP *CTCP | ||||||
|  | 	// Cmd contains various helper methods to interact with the server. | ||||||
|  | 	Cmd *Commands | ||||||
|  | 	// mu is the mux used for connections/disconnections from the server, | ||||||
|  | 	// so multiple threads aren't trying to connect at the same time, and | ||||||
|  | 	// vice versa. | ||||||
|  | 	mu sync.RWMutex | ||||||
|  | 	// stop is used to communicate with Connect(), letting it know that the | ||||||
|  | 	// client wishes to cancel/close. | ||||||
|  | 	stop context.CancelFunc | ||||||
|  | 	// conn is a net.Conn reference to the IRC server. If this is nil, it is | ||||||
|  | 	// safe to assume that we're not connected. If this is not nil, this | ||||||
|  | 	// means we're either connected, connecting, or cleaning up. This should | ||||||
|  | 	// be guarded with Client.mu. | ||||||
|  | 	conn *ircConn | ||||||
|  | 	// debug is used if a writer is supplied for Client.Config.Debugger. | ||||||
|  | 	debug *log.Logger | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Config contains configuration options for an IRC client | ||||||
|  | type Config struct { | ||||||
|  | 	// Server is a host/ip of the server you want to connect to. This only | ||||||
|  | 	// has an affect during the dial process | ||||||
|  | 	Server string | ||||||
|  | 	// ServerPass is the server password used to authenticate. This only has | ||||||
|  | 	// an affect during the dial process. | ||||||
|  | 	ServerPass string | ||||||
|  | 	// Port is the port that will be used during server connection. This only | ||||||
|  | 	// has an affect during the dial process. | ||||||
|  | 	Port int | ||||||
|  | 	// Nick is an rfc-valid nickname used during connection. This only has an | ||||||
|  | 	// affect during the dial process. | ||||||
|  | 	Nick string | ||||||
|  | 	// User is the username/ident to use on connect. Ignored if an identd | ||||||
|  | 	// server is used. This only has an affect during the dial process. | ||||||
|  | 	User string | ||||||
|  | 	// Name is the "realname" that's used during connection. This only has an | ||||||
|  | 	// affect during the dial process. | ||||||
|  | 	Name string | ||||||
|  | 	// SASL contains the necessary authentication data to authenticate | ||||||
|  | 	// with SASL. See the documentation for SASLMech for what is currently | ||||||
|  | 	// supported. Capability tracking must be enabled for this to work, as | ||||||
|  | 	// this requires IRCv3 CAP handling. | ||||||
|  | 	SASL SASLMech | ||||||
|  | 	// Bind is used to bind to a specific host or ip during the dial process | ||||||
|  | 	// when connecting to the server. This can be a hostname, however it must | ||||||
|  | 	// resolve to an IPv4/IPv6 address bindable on your system. Otherwise, | ||||||
|  | 	// you can simply use a IPv4/IPv6 address directly. This only has an | ||||||
|  | 	// affect during the dial process and will not work with DialerConnect(). | ||||||
|  | 	Bind string | ||||||
|  | 	// SSL allows dialing via TLS. See TLSConfig to set your own TLS | ||||||
|  | 	// configuration (e.g. to not force hostname checking). This only has an | ||||||
|  | 	// affect during the dial process. | ||||||
|  | 	SSL bool | ||||||
|  | 	// TLSConfig is an optional user-supplied tls configuration, used during | ||||||
|  | 	// socket creation to the server. SSL must be enabled for this to be used. | ||||||
|  | 	// This only has an affect during the dial process. | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  | 	// AllowFlood allows the client to bypass the rate limit of outbound | ||||||
|  | 	// messages. | ||||||
|  | 	AllowFlood bool | ||||||
|  | 	// GlobalFormat enables passing through all events which have trailing | ||||||
|  | 	// text through the color Fmt() function, so you don't have to wrap | ||||||
|  | 	// every response in the Fmt() method. | ||||||
|  | 	// | ||||||
|  | 	// Note that this only actually applies to PRIVMSG, NOTICE and TOPIC | ||||||
|  | 	// events, to ensure it doesn't clobber unwanted events. | ||||||
|  | 	GlobalFormat bool | ||||||
|  | 	// Debug is an optional, user supplied location to log the raw lines | ||||||
|  | 	// sent from the server, or other useful debug logs. Defaults to | ||||||
|  | 	// ioutil.Discard. For quick debugging, this could be set to os.Stdout. | ||||||
|  | 	Debug io.Writer | ||||||
|  | 	// Out is used to write out a prettified version of incoming events. For | ||||||
|  | 	// example, channel JOIN/PART, PRIVMSG/NOTICE, KICk, etc. Useful to get | ||||||
|  | 	// a brief output of the activity of the client. If you are looking to | ||||||
|  | 	// log raw messages, look at a handler and girc.ALLEVENTS and the relevant | ||||||
|  | 	// Event.Bytes() or Event.String() methods. | ||||||
|  | 	Out io.Writer | ||||||
|  | 	// RecoverFunc is called when a handler throws a panic. If RecoverFunc is | ||||||
|  | 	// set, the panic will be considered recovered, otherwise the client will | ||||||
|  | 	// panic. Set this to DefaultRecoverHandler if you don't want the client | ||||||
|  | 	// to panic, however you don't want to handle the panic yourself. | ||||||
|  | 	// DefaultRecoverHandler will log the panic to Debug or os.Stdout if | ||||||
|  | 	// Debug is unset. | ||||||
|  | 	RecoverFunc func(c *Client, e *HandlerError) | ||||||
|  | 	// SupportedCaps are the IRCv3 capabilities you would like the client to | ||||||
|  | 	// support on top of the ones which the client already supports (see | ||||||
|  | 	// cap.go for which ones the client enables by default). Only use this | ||||||
|  | 	// if you have not called DisableTracking(). The keys value gets passed | ||||||
|  | 	// to the server if supported. | ||||||
|  | 	SupportedCaps map[string][]string | ||||||
|  | 	// Version is the application version information that will be used in | ||||||
|  | 	// response to a CTCP VERSION, if default CTCP replies have not been | ||||||
|  | 	// overwritten or a VERSION handler was already supplied. | ||||||
|  | 	Version string | ||||||
|  | 	// PingDelay is the frequency between when the client sends a keep-alive | ||||||
|  | 	// PING to the server, and awaits a response (and times out if the server | ||||||
|  | 	// doesn't respond in time). This should be between 20-600 seconds. See | ||||||
|  | 	// Client.Lag() if you want to determine the delay between the server | ||||||
|  | 	// and the client. If this is set to -1, the client will not attempt to | ||||||
|  | 	// send client -> server PING requests. | ||||||
|  | 	PingDelay time.Duration | ||||||
|  |  | ||||||
|  | 	// disableTracking disables all channel and user-level tracking. Useful | ||||||
|  | 	// for highly embedded scripts with single purposes. This has an exported | ||||||
|  | 	// method which enables this and ensures prop cleanup, see | ||||||
|  | 	// Client.DisableTracking(). | ||||||
|  | 	disableTracking bool | ||||||
|  | 	// HandleNickCollide when set, allows the client to handle nick collisions | ||||||
|  | 	// in a custom way. If unset, the client will attempt to append a | ||||||
|  | 	// underscore to the end of the nickname, in order to bypass using | ||||||
|  | 	// an invalid nickname. For example, if "test" is already in use, or is | ||||||
|  | 	// blocked by the network/a service, the client will try and use "test_", | ||||||
|  | 	// then it will attempt "test__", "test___", and so on. | ||||||
|  | 	HandleNickCollide func(oldNick string) (newNick string) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrInvalidConfig is returned when the configuration passed to the client | ||||||
|  | // is invalid. | ||||||
|  | type ErrInvalidConfig struct { | ||||||
|  | 	Conf Config // Conf is the configuration that was not valid. | ||||||
|  | 	err  error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e ErrInvalidConfig) Error() string { return "invalid configuration: " + e.err.Error() } | ||||||
|  |  | ||||||
|  | // isValid checks some basic settings to ensure the config is valid. | ||||||
|  | func (conf *Config) isValid() error { | ||||||
|  | 	if conf.Server == "" { | ||||||
|  | 		return &ErrInvalidConfig{Conf: *conf, err: errors.New("empty server")} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Default port to 6667 (the standard IRC port). | ||||||
|  | 	if conf.Port == 0 { | ||||||
|  | 		conf.Port = 6667 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if conf.Port < 21 || conf.Port > 65535 { | ||||||
|  | 		return &ErrInvalidConfig{Conf: *conf, err: errors.New("port outside valid range (21-65535)")} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !IsValidNick(conf.Nick) { | ||||||
|  | 		return &ErrInvalidConfig{Conf: *conf, err: errors.New("bad nickname specified")} | ||||||
|  | 	} | ||||||
|  | 	if !IsValidUser(conf.User) { | ||||||
|  | 		return &ErrInvalidConfig{Conf: *conf, err: errors.New("bad user/ident specified")} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrNotConnected is returned if a method is used when the client isn't | ||||||
|  | // connected. | ||||||
|  | var ErrNotConnected = errors.New("client is not connected to server") | ||||||
|  |  | ||||||
|  | // ErrDisconnected is called when Config.Retries is less than 1, and we | ||||||
|  | // non-intentionally disconnected from the server. | ||||||
|  | var ErrDisconnected = errors.New("unexpectedly disconnected") | ||||||
|  |  | ||||||
|  | // ErrInvalidTarget should be returned if the target which you are | ||||||
|  | // attempting to send an event to is invalid or doesn't match RFC spec. | ||||||
|  | type ErrInvalidTarget struct { | ||||||
|  | 	Target string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *ErrInvalidTarget) Error() string { return "invalid target: " + e.Target } | ||||||
|  |  | ||||||
|  | // New creates a new IRC client with the specified server, name and config. | ||||||
|  | func New(config Config) *Client { | ||||||
|  | 	c := &Client{ | ||||||
|  | 		Config:   config, | ||||||
|  | 		rx:       make(chan *Event, 25), | ||||||
|  | 		tx:       make(chan *Event, 25), | ||||||
|  | 		CTCP:     newCTCP(), | ||||||
|  | 		initTime: time.Now(), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.Cmd = &Commands{c: c} | ||||||
|  |  | ||||||
|  | 	if c.Config.PingDelay >= 0 && c.Config.PingDelay < (20*time.Second) { | ||||||
|  | 		c.Config.PingDelay = 20 * time.Second | ||||||
|  | 	} else if c.Config.PingDelay > (600 * time.Second) { | ||||||
|  | 		c.Config.PingDelay = 600 * time.Second | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if c.Config.Debug == nil { | ||||||
|  | 		c.debug = log.New(ioutil.Discard, "", 0) | ||||||
|  | 	} else { | ||||||
|  | 		c.debug = log.New(c.Config.Debug, "debug:", log.Ltime|log.Lshortfile) | ||||||
|  | 		c.debug.Print("initializing debugging") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Setup the caller. | ||||||
|  | 	c.Handlers = newCaller(c.debug) | ||||||
|  |  | ||||||
|  | 	// Give ourselves a new state. | ||||||
|  | 	c.state = &state{} | ||||||
|  | 	c.state.reset() | ||||||
|  |  | ||||||
|  | 	// Register builtin handlers. | ||||||
|  | 	c.registerBuiltins() | ||||||
|  |  | ||||||
|  | 	// Register default CTCP responses. | ||||||
|  | 	c.CTCP.addDefaultHandlers() | ||||||
|  |  | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a brief description of the current client state. | ||||||
|  | func (c *Client) String() string { | ||||||
|  | 	connected := c.IsConnected() | ||||||
|  |  | ||||||
|  | 	return fmt.Sprintf( | ||||||
|  | 		"<Client init:%q handlers:%d connected:%t>", c.initTime.String(), c.Handlers.Len(), connected, | ||||||
|  | 	) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Close closes the network connection to the server, and sends a STOPPED | ||||||
|  | // event. This should cause Connect() to return with nil. This should be | ||||||
|  | // safe to call multiple times. See Connect()'s documentation on how | ||||||
|  | // handlers and goroutines are handled when disconnected from the server. | ||||||
|  | func (c *Client) Close() { | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	if c.stop != nil { | ||||||
|  | 		c.debug.Print("requesting client to stop") | ||||||
|  | 		c.stop() | ||||||
|  | 	} | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrEvent is an error returned when the server (or library) sends an ERROR | ||||||
|  | // message response. The string returned contains the trailing text from the | ||||||
|  | // message. | ||||||
|  | type ErrEvent struct { | ||||||
|  | 	Event *Event | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *ErrEvent) Error() string { | ||||||
|  | 	if e.Event == nil { | ||||||
|  | 		return "unknown error occurred" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return e.Event.Trailing | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Client) execLoop(ctx context.Context, errs chan error, wg *sync.WaitGroup) { | ||||||
|  | 	c.debug.Print("starting execLoop") | ||||||
|  | 	defer c.debug.Print("closing execLoop") | ||||||
|  |  | ||||||
|  | 	var event *Event | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-ctx.Done(): | ||||||
|  | 			// We've been told to exit, however we shouldn't bail on the | ||||||
|  | 			// current events in the queue that should be processed, as one | ||||||
|  | 			// may want to handle an ERROR, QUIT, etc. | ||||||
|  | 			c.debug.Printf("received signal to close, flushing %d events and executing", len(c.rx)) | ||||||
|  | 			for { | ||||||
|  | 				select { | ||||||
|  | 				case event = <-c.rx: | ||||||
|  | 					c.RunHandlers(event) | ||||||
|  | 				default: | ||||||
|  | 					goto done | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 		done: | ||||||
|  | 			wg.Done() | ||||||
|  | 			return | ||||||
|  | 		case event = <-c.rx: | ||||||
|  | 			if event != nil && event.Command == ERROR { | ||||||
|  | 				// Handles incoming ERROR responses. These are only ever sent | ||||||
|  | 				// by the server (with the exception that this library may use | ||||||
|  | 				// them as a lower level way of signalling to disconnect due | ||||||
|  | 				// to some other client-choosen error), and should always be | ||||||
|  | 				// followed up by the server disconnecting the client. If for | ||||||
|  | 				// some reason the server doesn't disconnect the client, or | ||||||
|  | 				// if this library is the source of the error, this should | ||||||
|  | 				// signal back up to the main connect loop, to disconnect. | ||||||
|  | 				errs <- &ErrEvent{Event: event} | ||||||
|  |  | ||||||
|  | 				// Make sure to not actually exit, so we can let any handlers | ||||||
|  | 				// actually handle the ERROR event. | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			c.RunHandlers(event) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DisableTracking disables all channel/user-level/CAP tracking, and clears | ||||||
|  | // all internal handlers. Useful for highly embedded scripts with single | ||||||
|  | // purposes. This cannot be un-done on a client. | ||||||
|  | func (c *Client) DisableTracking() { | ||||||
|  | 	c.debug.Print("disabling tracking") | ||||||
|  | 	c.Config.disableTracking = true | ||||||
|  | 	c.Handlers.clearInternal() | ||||||
|  |  | ||||||
|  | 	c.state.Lock() | ||||||
|  | 	c.state.channels = nil | ||||||
|  | 	c.state.Unlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  |  | ||||||
|  | 	c.registerBuiltins() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Server returns the string representation of host+port pair for net.Conn. | ||||||
|  | func (c *Client) Server() string { | ||||||
|  | 	return fmt.Sprintf("%s:%d", c.Config.Server, c.Config.Port) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Lifetime returns the amount of time that has passed since the client was | ||||||
|  | // created. | ||||||
|  | func (c *Client) Lifetime() time.Duration { | ||||||
|  | 	return time.Since(c.initTime) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Uptime is the time at which the client successfully connected to the | ||||||
|  | // server. | ||||||
|  | func (c *Client) Uptime() (up *time.Time, err error) { | ||||||
|  | 	if !c.IsConnected() { | ||||||
|  | 		return nil, ErrNotConnected | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	c.conn.mu.RLock() | ||||||
|  | 	up = c.conn.connTime | ||||||
|  | 	c.conn.mu.RUnlock() | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return up, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ConnSince is the duration that has past since the client successfully | ||||||
|  | // connected to the server. | ||||||
|  | func (c *Client) ConnSince() (since *time.Duration, err error) { | ||||||
|  | 	if !c.IsConnected() { | ||||||
|  | 		return nil, ErrNotConnected | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	c.conn.mu.RLock() | ||||||
|  | 	timeSince := time.Since(*c.conn.connTime) | ||||||
|  | 	c.conn.mu.RUnlock() | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return &timeSince, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsConnected returns true if the client is connected to the server. | ||||||
|  | func (c *Client) IsConnected() (connected bool) { | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	if c.conn == nil { | ||||||
|  | 		c.mu.RUnlock() | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.conn.mu.RLock() | ||||||
|  | 	connected = c.conn.connected | ||||||
|  | 	c.conn.mu.RUnlock() | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return connected | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetNick returns the current nickname of the active connection. Panics if | ||||||
|  | // tracking is disabled. | ||||||
|  | func (c *Client) GetNick() string { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	defer c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	if c.state.nick == "" { | ||||||
|  | 		return c.Config.Nick | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c.state.nick | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetIdent returns the current ident of the active connection. Panics if | ||||||
|  | // tracking is disabled. May be empty, as this is obtained from when we join | ||||||
|  | // a channel, as there is no other more efficient method to return this info. | ||||||
|  | func (c *Client) GetIdent() string { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	defer c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	if c.state.ident == "" { | ||||||
|  | 		return c.Config.User | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c.state.ident | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetHost returns the current host of the active connection. Panics if | ||||||
|  | // tracking is disabled. May be empty, as this is obtained from when we join | ||||||
|  | // a channel, as there is no other more efficient method to return this info. | ||||||
|  | func (c *Client) GetHost() string { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	defer c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return c.state.host | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Channels returns the active list of channels that the client is in. | ||||||
|  | // Panics if tracking is disabled. | ||||||
|  | func (c *Client) Channels() []string { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	channels := make([]string, len(c.state.channels)) | ||||||
|  | 	var i int | ||||||
|  | 	for channel := range c.state.channels { | ||||||
|  | 		channels[i] = c.state.channels[channel].Name | ||||||
|  | 		i++ | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  | 	sort.Strings(channels) | ||||||
|  |  | ||||||
|  | 	return channels | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Users returns the active list of users that the client is tracking across | ||||||
|  | // all files. Panics if tracking is disabled. | ||||||
|  | func (c *Client) Users() []string { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	users := make([]string, len(c.state.users)) | ||||||
|  | 	var i int | ||||||
|  | 	for user := range c.state.users { | ||||||
|  | 		users[i] = c.state.users[user].Nick | ||||||
|  | 		i++ | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  | 	sort.Strings(users) | ||||||
|  |  | ||||||
|  | 	return users | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LookupChannel looks up a given channel in state. If the channel doesn't | ||||||
|  | // exist, nil is returned. Panics if tracking is disabled. | ||||||
|  | func (c *Client) LookupChannel(name string) *Channel { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  | 	if name == "" { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	defer c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	channel := c.state.lookupChannel(name) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return channel.Copy() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LookupUser looks up a given user in state. If the user doesn't exist, nil | ||||||
|  | // is returned. Panics if tracking is disabled. | ||||||
|  | func (c *Client) LookupUser(nick string) *User { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  | 	if nick == "" { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	defer c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	user := c.state.lookupUser(nick) | ||||||
|  | 	if user == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return user.Copy() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsInChannel returns true if the client is in channel. Panics if tracking | ||||||
|  | // is disabled. | ||||||
|  | func (c *Client) IsInChannel(channel string) bool { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	_, inChannel := c.state.channels[ToRFC1459(channel)] | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return inChannel | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetServerOption retrieves a server capability setting that was retrieved | ||||||
|  | // during client connection. This is also known as ISUPPORT (or RPL_PROTOCTL). | ||||||
|  | // Will panic if used when tracking has been disabled. Examples of usage: | ||||||
|  | // | ||||||
|  | //   nickLen, success := GetServerOption("MAXNICKLEN") | ||||||
|  | // | ||||||
|  | func (c *Client) GetServerOption(key string) (result string, ok bool) { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	result, ok = c.state.serverOptions[key] | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return result, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NetworkName returns the network identifier. E.g. "EsperNet", "ByteIRC". | ||||||
|  | // May be empty if the server does not support RPL_ISUPPORT (or RPL_PROTOCTL). | ||||||
|  | // Will panic if used when tracking has been disabled. | ||||||
|  | func (c *Client) NetworkName() (name string) { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	name, _ = c.GetServerOption("NETWORK") | ||||||
|  |  | ||||||
|  | 	return name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ServerVersion returns the server software version, if the server has | ||||||
|  | // supplied this information during connection. May be empty if the server | ||||||
|  | // does not support RPL_MYINFO. Will panic if used when tracking has been | ||||||
|  | // disabled. | ||||||
|  | func (c *Client) ServerVersion() (version string) { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	version, _ = c.GetServerOption("VERSION") | ||||||
|  |  | ||||||
|  | 	return version | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ServerMOTD returns the servers message of the day, if the server has sent | ||||||
|  | // it upon connect. Will panic if used when tracking has been disabled. | ||||||
|  | func (c *Client) ServerMOTD() (motd string) { | ||||||
|  | 	c.panicIfNotTracking() | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	motd = c.state.motd | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return motd | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Lag is the latency between the server and the client. This is measured by | ||||||
|  | // determining the difference in time between when we ping the server, and | ||||||
|  | // when we receive a pong. | ||||||
|  | func (c *Client) Lag() time.Duration { | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	c.conn.mu.RLock() | ||||||
|  | 	delta := c.conn.lastPong.Sub(c.conn.lastPing) | ||||||
|  | 	c.conn.mu.RUnlock() | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	if delta < 0 { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return delta | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // panicIfNotTracking will throw a panic when it's called, and tracking is | ||||||
|  | // disabled. Adds useful info like what function specifically, and where it | ||||||
|  | // was called from. | ||||||
|  | func (c *Client) panicIfNotTracking() { | ||||||
|  | 	if !c.Config.disableTracking { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	pc, _, _, _ := runtime.Caller(1) | ||||||
|  | 	fn := runtime.FuncForPC(pc) | ||||||
|  | 	_, file, line, _ := runtime.Caller(2) | ||||||
|  |  | ||||||
|  | 	panic(fmt.Sprintf("%s used when tracking is disabled (caller %s:%d)", fn.Name(), file, line)) | ||||||
|  | } | ||||||
							
								
								
									
										197
									
								
								vendor/github.com/lrstanley/girc/cmdhandler/cmd.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										197
									
								
								vendor/github.com/lrstanley/girc/cmdhandler/cmd.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,197 @@ | |||||||
|  | package cmdhandler | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/lrstanley/girc" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Input is a wrapper for events, based around private messages. | ||||||
|  | type Input struct { | ||||||
|  | 	Origin *girc.Event | ||||||
|  | 	Args   []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Command is an IRC command, supporting aliases, help documentation and easy | ||||||
|  | // wrapping for message inputs. | ||||||
|  | type Command struct { | ||||||
|  | 	// Name of command, e.g. "search" or "ping". | ||||||
|  | 	Name string | ||||||
|  | 	// Aliases for the above command, e.g. "s" for search, or "p" for "ping". | ||||||
|  | 	Aliases []string | ||||||
|  | 	// Help documentation. Should be in the format "<arg> <arg> [arg] -- | ||||||
|  | 	// something useful here" | ||||||
|  | 	Help string | ||||||
|  | 	// MinArgs is the minimum required arguments for the command. Defaults to | ||||||
|  | 	// 0, which means multiple, or no arguments can be supplied. If set | ||||||
|  | 	// above 0, this means that the command handler will throw an error asking | ||||||
|  | 	// the person to check "<prefix>help <command>" for more info. | ||||||
|  | 	MinArgs int | ||||||
|  | 	// Fn is the function which is executed when the command is ran from a | ||||||
|  | 	// private message, or channel. | ||||||
|  | 	Fn func(*girc.Client, *Input) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Command) genHelp(prefix string) string { | ||||||
|  | 	out := "{b}" + prefix + c.Name + "{b}" | ||||||
|  |  | ||||||
|  | 	if c.Aliases != nil && len(c.Aliases) > 0 { | ||||||
|  | 		out += " ({b}" + prefix + strings.Join(c.Aliases, "{b}, {b}"+prefix) + "{b})" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out += " :: " + c.Help | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // CmdHandler is an irc command parser and execution format which you could | ||||||
|  | // use as an example for building your own version/bot. | ||||||
|  | // | ||||||
|  | // An example of how you would register this with girc: | ||||||
|  | // | ||||||
|  | //	ch, err := cmdhandler.New("!") | ||||||
|  | //	if err != nil { | ||||||
|  | //		panic(err) | ||||||
|  | //	} | ||||||
|  | // | ||||||
|  | //	ch.Add(&cmdhandler.Command{ | ||||||
|  | //		Name:    "ping", | ||||||
|  | //		Help:    "Sends a pong reply back to the original user.", | ||||||
|  | //		Fn: func(c *girc.Client, input *cmdhandler.Input) { | ||||||
|  | //			c.Commands.ReplyTo(*input.Origin, "pong!") | ||||||
|  | //		}, | ||||||
|  | //	}) | ||||||
|  | // | ||||||
|  | //	client.Handlers.AddHandler(girc.PRIVMSG, ch) | ||||||
|  | type CmdHandler struct { | ||||||
|  | 	prefix string | ||||||
|  | 	re     *regexp.Regexp | ||||||
|  |  | ||||||
|  | 	mu   sync.Mutex | ||||||
|  | 	cmds map[string]*Command | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var cmdMatch = `^%s([a-z0-9-_]{1,20})(?: (.*))?$` | ||||||
|  |  | ||||||
|  | // New returns a new CmdHandler based on the specified command prefix. A good | ||||||
|  | // prefix is a single character, and easy to remember/use. E.g. "!", or ".". | ||||||
|  | func New(prefix string) (*CmdHandler, error) { | ||||||
|  | 	re, err := regexp.Compile(fmt.Sprintf(cmdMatch, regexp.QuoteMeta(prefix))) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return &CmdHandler{prefix: prefix, re: re, cmds: make(map[string]*Command)}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var validName = regexp.MustCompile(`^[a-z0-9-_]{1,20}$`) | ||||||
|  |  | ||||||
|  | // Add registers a new command to the handler. Note that you cannot remove | ||||||
|  | // commands once added, unless you add another CmdHandler to the client. | ||||||
|  | func (ch *CmdHandler) Add(cmd *Command) error { | ||||||
|  | 	if cmd == nil { | ||||||
|  | 		return errors.New("nil command provided to CmdHandler") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.Name = strings.ToLower(cmd.Name) | ||||||
|  | 	if !validName.MatchString(cmd.Name) { | ||||||
|  | 		return fmt.Errorf("invalid command name: %q (req: %q)", cmd.Name, validName.String()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if cmd.Aliases != nil { | ||||||
|  | 		for i := 0; i < len(cmd.Aliases); i++ { | ||||||
|  | 			cmd.Aliases[i] = strings.ToLower(cmd.Aliases[i]) | ||||||
|  | 			if !validName.MatchString(cmd.Aliases[i]) { | ||||||
|  | 				return fmt.Errorf("invalid command name: %q (req: %q)", cmd.Aliases[i], validName.String()) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if cmd.MinArgs < 0 { | ||||||
|  | 		cmd.MinArgs = 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ch.mu.Lock() | ||||||
|  | 	defer ch.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	if _, ok := ch.cmds[cmd.Name]; ok { | ||||||
|  | 		return fmt.Errorf("command already registered: %s", cmd.Name) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ch.cmds[cmd.Name] = cmd | ||||||
|  |  | ||||||
|  | 	// Since we'd be storing pointers, duplicates do not matter. | ||||||
|  | 	for i := 0; i < len(cmd.Aliases); i++ { | ||||||
|  | 		if _, ok := ch.cmds[cmd.Aliases[i]]; ok { | ||||||
|  | 			return fmt.Errorf("alias already registered: %s", cmd.Aliases[i]) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		ch.cmds[cmd.Aliases[i]] = cmd | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Execute satisfies the girc.Handler interface. | ||||||
|  | func (ch *CmdHandler) Execute(client *girc.Client, event girc.Event) { | ||||||
|  | 	if event.Source == nil || event.Command != girc.PRIVMSG { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	parsed := ch.re.FindStringSubmatch(event.Trailing) | ||||||
|  | 	if len(parsed) != 3 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	invCmd := strings.ToLower(parsed[1]) | ||||||
|  | 	args := strings.Split(parsed[2], " ") | ||||||
|  | 	if len(args) == 1 && args[0] == "" { | ||||||
|  | 		args = []string{} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ch.mu.Lock() | ||||||
|  | 	defer ch.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	if invCmd == "help" { | ||||||
|  | 		if len(args) == 0 { | ||||||
|  | 			client.Cmd.ReplyTo(event, girc.Fmt("type '{b}!help {blue}<command>{c}{b}' to optionally get more info about a specific command.")) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		args[0] = strings.ToLower(args[0]) | ||||||
|  |  | ||||||
|  | 		if _, ok := ch.cmds[args[0]]; !ok { | ||||||
|  | 			client.Cmd.ReplyTof(event, girc.Fmt("unknown command {b}%q{b}."), args[0]) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if ch.cmds[args[0]].Help == "" { | ||||||
|  | 			client.Cmd.ReplyTof(event, girc.Fmt("there is no help documentation for {b}%q{b}"), args[0]) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		client.Cmd.ReplyTo(event, girc.Fmt(ch.cmds[args[0]].genHelp(ch.prefix))) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd, ok := ch.cmds[invCmd] | ||||||
|  | 	if !ok { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(args) < cmd.MinArgs { | ||||||
|  | 		client.Cmd.ReplyTof(event, girc.Fmt("not enough arguments supplied for {b}%q{b}. try '{b}%shelp %s{b}'?"), invCmd, ch.prefix, invCmd) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	in := &Input{ | ||||||
|  | 		Origin: &event, | ||||||
|  | 		Args:   args, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	go cmd.Fn(client, in) | ||||||
|  | } | ||||||
							
								
								
									
										398
									
								
								vendor/github.com/lrstanley/girc/commands.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										398
									
								
								vendor/github.com/lrstanley/girc/commands.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,398 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Commands holds a large list of useful methods to interact with the server, | ||||||
|  | // and wrappers for common events. | ||||||
|  | type Commands struct { | ||||||
|  | 	c *Client | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Nick changes the client nickname. | ||||||
|  | func (cmd *Commands) Nick(name string) error { | ||||||
|  | 	if !IsValidNick(name) { | ||||||
|  | 		return &ErrInvalidTarget{Target: name} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: NICK, Params: []string{name}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Join attempts to enter a list of IRC channels, at bulk if possible to | ||||||
|  | // prevent sending extensive JOIN commands. | ||||||
|  | func (cmd *Commands) Join(channels ...string) error { | ||||||
|  | 	// We can join multiple channels at once, however we need to ensure that | ||||||
|  | 	// we are not exceeding the line length. (see maxLength) | ||||||
|  | 	max := maxLength - len(JOIN) - 1 | ||||||
|  |  | ||||||
|  | 	var buffer string | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(channels); i++ { | ||||||
|  | 		if !IsValidChannel(channels[i]) { | ||||||
|  | 			return &ErrInvalidTarget{Target: channels[i]} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(buffer+","+channels[i]) > max { | ||||||
|  | 			cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}}) | ||||||
|  | 			buffer = "" | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(buffer) == 0 { | ||||||
|  | 			buffer = channels[i] | ||||||
|  | 		} else { | ||||||
|  | 			buffer += "," + channels[i] | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if i == len(channels)-1 { | ||||||
|  | 			cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}}) | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // JoinKey attempts to enter an IRC channel with a password. | ||||||
|  | func (cmd *Commands) JoinKey(channel, password string) error { | ||||||
|  | 	if !IsValidChannel(channel) { | ||||||
|  | 		return &ErrInvalidTarget{Target: channel} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: JOIN, Params: []string{channel, password}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Part leaves an IRC channel. | ||||||
|  | func (cmd *Commands) Part(channel, message string) error { | ||||||
|  | 	if !IsValidChannel(channel) { | ||||||
|  | 		return &ErrInvalidTarget{Target: channel} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: JOIN, Params: []string{channel}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PartMessage leaves an IRC channel with a specified leave message. | ||||||
|  | func (cmd *Commands) PartMessage(channel, message string) error { | ||||||
|  | 	if !IsValidChannel(channel) { | ||||||
|  | 		return &ErrInvalidTarget{Target: channel} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: JOIN, Params: []string{channel}, Trailing: message}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendCTCP sends a CTCP request to target. Note that this method uses | ||||||
|  | // PRIVMSG specifically. | ||||||
|  | func (cmd *Commands) SendCTCP(target, ctcpType, message string) error { | ||||||
|  | 	out := encodeCTCPRaw(ctcpType, message) | ||||||
|  | 	if out == "" { | ||||||
|  | 		return errors.New("invalid CTCP") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd.Message(target, out) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendCTCPf sends a CTCP request to target using a specific format. Note that | ||||||
|  | // this method uses PRIVMSG specifically. | ||||||
|  | func (cmd *Commands) SendCTCPf(target, ctcpType, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.SendCTCP(target, ctcpType, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendCTCPReplyf sends a CTCP response to target using a specific format. | ||||||
|  | // Note that this method uses NOTICE specifically. | ||||||
|  | func (cmd *Commands) SendCTCPReplyf(target, ctcpType, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.SendCTCPReply(target, ctcpType, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendCTCPReply sends a CTCP response to target. Note that this method uses | ||||||
|  | // NOTICE specifically. | ||||||
|  | func (cmd *Commands) SendCTCPReply(target, ctcpType, message string) error { | ||||||
|  | 	out := encodeCTCPRaw(ctcpType, message) | ||||||
|  | 	if out == "" { | ||||||
|  | 		return errors.New("invalid CTCP") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd.Notice(target, out) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Message sends a PRIVMSG to target (either channel, service, or user). | ||||||
|  | func (cmd *Commands) Message(target, message string) error { | ||||||
|  | 	if !IsValidNick(target) && !IsValidChannel(target) { | ||||||
|  | 		return &ErrInvalidTarget{Target: target} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: PRIVMSG, Params: []string{target}, Trailing: message}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Messagef sends a formated PRIVMSG to target (either channel, service, or | ||||||
|  | // user). | ||||||
|  | func (cmd *Commands) Messagef(target, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.Message(target, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrInvalidSource is returned when a method needs to know the origin of an | ||||||
|  | // event, however Event.Source is unknown (e.g. sent by the user, not the | ||||||
|  | // server.) | ||||||
|  | var ErrInvalidSource = errors.New("event has nil or invalid source address") | ||||||
|  |  | ||||||
|  | // Reply sends a reply to channel or user, based on where the supplied event | ||||||
|  | // originated from. See also ReplyTo(). | ||||||
|  | func (cmd *Commands) Reply(event Event, message string) error { | ||||||
|  | 	if event.Source == nil { | ||||||
|  | 		return ErrInvalidSource | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(event.Params) > 0 && IsValidChannel(event.Params[0]) { | ||||||
|  | 		return cmd.Message(event.Params[0], message) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd.Message(event.Source.Name, message) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Replyf sends a reply to channel or user with a format string, based on | ||||||
|  | // where the supplied event originated from. See also ReplyTof(). | ||||||
|  | func (cmd *Commands) Replyf(event Event, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.Reply(event, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ReplyTo sends a reply to a channel or user, based on where the supplied | ||||||
|  | // event originated from. ReplyTo(), when originating from a channel will | ||||||
|  | // default to replying with "<user>, <message>". See also Reply(). | ||||||
|  | func (cmd *Commands) ReplyTo(event Event, message string) error { | ||||||
|  | 	if event.Source == nil { | ||||||
|  | 		return ErrInvalidSource | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(event.Params) > 0 && IsValidChannel(event.Params[0]) { | ||||||
|  | 		return cmd.Message(event.Params[0], event.Source.Name+", "+message) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd.Message(event.Source.Name, message) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ReplyTof sends a reply to a channel or user with a format string, based | ||||||
|  | // on where the supplied event originated from. ReplyTo(), when originating | ||||||
|  | // from a channel will default to replying with "<user>, <message>". See | ||||||
|  | // also Replyf(). | ||||||
|  | func (cmd *Commands) ReplyTof(event Event, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.ReplyTo(event, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Action sends a PRIVMSG ACTION (/me) to target (either channel, service, | ||||||
|  | // or user). | ||||||
|  | func (cmd *Commands) Action(target, message string) error { | ||||||
|  | 	if !IsValidNick(target) && !IsValidChannel(target) { | ||||||
|  | 		return &ErrInvalidTarget{Target: target} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{ | ||||||
|  | 		Command:  PRIVMSG, | ||||||
|  | 		Params:   []string{target}, | ||||||
|  | 		Trailing: fmt.Sprintf("\001ACTION %s\001", message), | ||||||
|  | 	}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Actionf sends a formated PRIVMSG ACTION (/me) to target (either channel, | ||||||
|  | // service, or user). | ||||||
|  | func (cmd *Commands) Actionf(target, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.Action(target, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Notice sends a NOTICE to target (either channel, service, or user). | ||||||
|  | func (cmd *Commands) Notice(target, message string) error { | ||||||
|  | 	if !IsValidNick(target) && !IsValidChannel(target) { | ||||||
|  | 		return &ErrInvalidTarget{Target: target} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: NOTICE, Params: []string{target}, Trailing: message}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Noticef sends a formated NOTICE to target (either channel, service, or | ||||||
|  | // user). | ||||||
|  | func (cmd *Commands) Noticef(target, format string, a ...interface{}) error { | ||||||
|  | 	return cmd.Notice(target, fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendRaw sends a raw string back to the server, without carriage returns | ||||||
|  | // or newlines. | ||||||
|  | func (cmd *Commands) SendRaw(raw string) error { | ||||||
|  | 	e := ParseEvent(raw) | ||||||
|  | 	if e == nil { | ||||||
|  | 		return errors.New("invalid event: " + raw) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(e) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SendRawf sends a formated string back to the server, without carriage | ||||||
|  | // returns or newlines. | ||||||
|  | func (cmd *Commands) SendRawf(format string, a ...interface{}) error { | ||||||
|  | 	return cmd.SendRaw(fmt.Sprintf(format, a...)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Topic sets the topic of channel to message. Does not verify the length | ||||||
|  | // of the topic. | ||||||
|  | func (cmd *Commands) Topic(channel, message string) { | ||||||
|  | 	cmd.c.Send(&Event{Command: TOPIC, Params: []string{channel}, Trailing: message}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Who sends a WHO query to the server, which will attempt WHOX by default. | ||||||
|  | // See http://faerion.sourceforge.net/doc/irc/whox.var for more details. This | ||||||
|  | // sends "%tcuhnr,2" per default. Do not use "1" as this will conflict with | ||||||
|  | // girc's builtin tracking functionality. | ||||||
|  | func (cmd *Commands) Who(target string) error { | ||||||
|  | 	if !IsValidNick(target) && !IsValidChannel(target) && !IsValidUser(target) { | ||||||
|  | 		return &ErrInvalidTarget{Target: target} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: WHO, Params: []string{target, "%tcuhnr,2"}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Whois sends a WHOIS query to the server, targeted at a specific user. | ||||||
|  | // as WHOIS is a bit slower, you may want to use WHO for brief user info. | ||||||
|  | func (cmd *Commands) Whois(nick string) error { | ||||||
|  | 	if !IsValidNick(nick) { | ||||||
|  | 		return &ErrInvalidTarget{Target: nick} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: WHOIS, Params: []string{nick}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Ping sends a PING query to the server, with a specific identifier that | ||||||
|  | // the server should respond with. | ||||||
|  | func (cmd *Commands) Ping(id string) { | ||||||
|  | 	cmd.c.write(&Event{Command: PING, Params: []string{id}}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pong sends a PONG query to the server, with an identifier which was | ||||||
|  | // received from a previous PING query received by the client. | ||||||
|  | func (cmd *Commands) Pong(id string) { | ||||||
|  | 	cmd.c.write(&Event{Command: PONG, Params: []string{id}}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Oper sends a OPER authentication query to the server, with a username | ||||||
|  | // and password. | ||||||
|  | func (cmd *Commands) Oper(user, pass string) { | ||||||
|  | 	cmd.c.Send(&Event{Command: OPER, Params: []string{user, pass}, Sensitive: true}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Kick sends a KICK query to the server, attempting to kick nick from | ||||||
|  | // channel, with reason. If reason is blank, one will not be sent to the | ||||||
|  | // server. | ||||||
|  | func (cmd *Commands) Kick(channel, nick, reason string) error { | ||||||
|  | 	if !IsValidChannel(channel) { | ||||||
|  | 		return &ErrInvalidTarget{Target: channel} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !IsValidNick(nick) { | ||||||
|  | 		return &ErrInvalidTarget{Target: nick} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if reason != "" { | ||||||
|  | 		cmd.c.Send(&Event{Command: KICK, Params: []string{channel, nick}, Trailing: reason}) | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: KICK, Params: []string{channel, nick}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Invite sends a INVITE query to the server, to invite nick to channel. | ||||||
|  | func (cmd *Commands) Invite(channel, nick string) error { | ||||||
|  | 	if !IsValidChannel(channel) { | ||||||
|  | 		return &ErrInvalidTarget{Target: channel} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !IsValidNick(nick) { | ||||||
|  | 		return &ErrInvalidTarget{Target: nick} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: INVITE, Params: []string{nick, channel}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Away sends a AWAY query to the server, suggesting that the client is no | ||||||
|  | // longer active. If reason is blank, Client.Back() is called. Also see | ||||||
|  | // Client.Back(). | ||||||
|  | func (cmd *Commands) Away(reason string) { | ||||||
|  | 	if reason == "" { | ||||||
|  | 		cmd.Back() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: AWAY, Params: []string{reason}}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Back sends a AWAY query to the server, however the query is blank, | ||||||
|  | // suggesting that the client is active once again. Also see Client.Away(). | ||||||
|  | func (cmd *Commands) Back() { | ||||||
|  | 	cmd.c.Send(&Event{Command: AWAY}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // List sends a LIST query to the server, which will list channels and topics. | ||||||
|  | // Supports multiple channels at once, in hopes it will reduce extensive | ||||||
|  | // LIST queries to the server. Supply no channels to run a list against the | ||||||
|  | // entire server (warning, that may mean LOTS of channels!) | ||||||
|  | func (cmd *Commands) List(channels ...string) error { | ||||||
|  | 	if len(channels) == 0 { | ||||||
|  | 		cmd.c.Send(&Event{Command: LIST}) | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// We can LIST multiple channels at once, however we need to ensure that | ||||||
|  | 	// we are not exceeding the line length. (see maxLength) | ||||||
|  | 	max := maxLength - len(JOIN) - 1 | ||||||
|  |  | ||||||
|  | 	var buffer string | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(channels); i++ { | ||||||
|  | 		if !IsValidChannel(channels[i]) { | ||||||
|  | 			return &ErrInvalidTarget{Target: channels[i]} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(buffer+","+channels[i]) > max { | ||||||
|  | 			cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}}) | ||||||
|  | 			buffer = "" | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(buffer) == 0 { | ||||||
|  | 			buffer = channels[i] | ||||||
|  | 		} else { | ||||||
|  | 			buffer += "," + channels[i] | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if i == len(channels)-1 { | ||||||
|  | 			cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}}) | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Whowas sends a WHOWAS query to the server. amount is the amount of results | ||||||
|  | // you want back. | ||||||
|  | func (cmd *Commands) Whowas(nick string, amount int) error { | ||||||
|  | 	if !IsValidNick(nick) { | ||||||
|  | 		return &ErrInvalidTarget{Target: nick} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd.c.Send(&Event{Command: WHOWAS, Params: []string{nick, string(amount)}}) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										566
									
								
								vendor/github.com/lrstanley/girc/conn.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										566
									
								
								vendor/github.com/lrstanley/girc/conn.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,566 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bufio" | ||||||
|  | 	"context" | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Messages are delimited with CR and LF line endings, we're using the last | ||||||
|  | // one to split the stream. Both are removed during parsing of the message. | ||||||
|  | const delim byte = '\n' | ||||||
|  |  | ||||||
|  | var endline = []byte("\r\n") | ||||||
|  |  | ||||||
|  | // ircConn represents an IRC network protocol connection, it consists of an | ||||||
|  | // Encoder and Decoder to manage i/o. | ||||||
|  | type ircConn struct { | ||||||
|  | 	io   *bufio.ReadWriter | ||||||
|  | 	sock net.Conn | ||||||
|  |  | ||||||
|  | 	mu sync.RWMutex | ||||||
|  | 	// lastWrite is used to keep track of when we last wrote to the server. | ||||||
|  | 	lastWrite time.Time | ||||||
|  | 	// lastActive is the last time the client was interacting with the server, | ||||||
|  | 	// excluding a few background commands (PING, PONG, WHO, etc). | ||||||
|  | 	lastActive time.Time | ||||||
|  | 	// writeDelay is used to keep track of rate limiting of events sent to | ||||||
|  | 	// the server. | ||||||
|  | 	writeDelay time.Duration | ||||||
|  | 	// connected is true if we're actively connected to a server. | ||||||
|  | 	connected bool | ||||||
|  | 	// connTime is the time at which the client has connected to a server. | ||||||
|  | 	connTime *time.Time | ||||||
|  | 	// lastPing is the last time that we pinged the server. | ||||||
|  | 	lastPing time.Time | ||||||
|  | 	// lastPong is the last successful time that we pinged the server and | ||||||
|  | 	// received a successful pong back. | ||||||
|  | 	lastPong  time.Time | ||||||
|  | 	pingDelay time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Dialer is an interface implementation of net.Dialer. Use this if you would | ||||||
|  | // like to implement your own dialer which the client will use when connecting. | ||||||
|  | type Dialer interface { | ||||||
|  | 	// Dial takes two arguments. Network, which should be similar to "tcp", | ||||||
|  | 	// "tdp6", "udp", etc -- as well as address, which is the hostname or ip | ||||||
|  | 	// of the network. Note that network can be ignored if your transport | ||||||
|  | 	// doesn't take advantage of network types. | ||||||
|  | 	Dial(network, address string) (net.Conn, error) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // newConn sets up and returns a new connection to the server. | ||||||
|  | func newConn(conf Config, dialer Dialer, addr string) (*ircConn, error) { | ||||||
|  | 	if err := conf.isValid(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var conn net.Conn | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	if dialer == nil { | ||||||
|  | 		netDialer := &net.Dialer{Timeout: 5 * time.Second} | ||||||
|  |  | ||||||
|  | 		if conf.Bind != "" { | ||||||
|  | 			var local *net.TCPAddr | ||||||
|  | 			local, err = net.ResolveTCPAddr("tcp", conf.Bind+":0") | ||||||
|  | 			if err != nil { | ||||||
|  | 				return nil, err | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			netDialer.LocalAddr = local | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		dialer = netDialer | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if conn, err = dialer.Dial("tcp", addr); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if conf.SSL { | ||||||
|  | 		var tlsConn net.Conn | ||||||
|  | 		tlsConn, err = tlsHandshake(conn, conf.TLSConfig, conf.Server, true) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		conn = tlsConn | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ctime := time.Now() | ||||||
|  |  | ||||||
|  | 	c := &ircConn{ | ||||||
|  | 		sock:      conn, | ||||||
|  | 		connTime:  &ctime, | ||||||
|  | 		connected: true, | ||||||
|  | 	} | ||||||
|  | 	c.newReadWriter() | ||||||
|  |  | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func newMockConn(conn net.Conn) *ircConn { | ||||||
|  | 	ctime := time.Now() | ||||||
|  | 	c := &ircConn{ | ||||||
|  | 		sock:      conn, | ||||||
|  | 		connTime:  &ctime, | ||||||
|  | 		connected: true, | ||||||
|  | 	} | ||||||
|  | 	c.newReadWriter() | ||||||
|  |  | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrParseEvent is returned when an event cannot be parsed with ParseEvent(). | ||||||
|  | type ErrParseEvent struct { | ||||||
|  | 	Line string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e ErrParseEvent) Error() string { return "unable to parse event: " + e.Line } | ||||||
|  |  | ||||||
|  | func (c *ircConn) decode() (event *Event, err error) { | ||||||
|  | 	line, err := c.io.ReadString(delim) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if event = ParseEvent(line); event == nil { | ||||||
|  | 		return nil, ErrParseEvent{line} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return event, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *ircConn) encode(event *Event) error { | ||||||
|  | 	if _, err := c.io.Write(event.Bytes()); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if _, err := c.io.Write(endline); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c.io.Flush() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *ircConn) newReadWriter() { | ||||||
|  | 	c.io = bufio.NewReadWriter(bufio.NewReader(c.sock), bufio.NewWriter(c.sock)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func tlsHandshake(conn net.Conn, conf *tls.Config, server string, validate bool) (net.Conn, error) { | ||||||
|  | 	if conf == nil { | ||||||
|  | 		conf = &tls.Config{ServerName: server, InsecureSkipVerify: !validate} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	tlsConn := tls.Client(conn, conf) | ||||||
|  | 	return net.Conn(tlsConn), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Close closes the underlying socket. | ||||||
|  | func (c *ircConn) Close() error { | ||||||
|  | 	return c.sock.Close() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Connect attempts to connect to the given IRC server. Returns only when | ||||||
|  | // an error has occurred, or a disconnect was requested with Close(). Connect | ||||||
|  | // will only return once all client-based goroutines have been closed to | ||||||
|  | // ensure there are no long-running routines becoming backed up. | ||||||
|  | // | ||||||
|  | // Connect will wait for all non-goroutine handlers to complete on error/quit, | ||||||
|  | // however it will not wait for goroutine-based handlers. | ||||||
|  | // | ||||||
|  | // If this returns nil, this means that the client requested to be closed | ||||||
|  | // (e.g. Client.Close()). Connect will panic if called when the last call has | ||||||
|  | // not completed. | ||||||
|  | func (c *Client) Connect() error { | ||||||
|  | 	return c.internalConnect(nil, nil) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialerConnect allows you to specify your own custom dialer which implements | ||||||
|  | // the Dialer interface. | ||||||
|  | // | ||||||
|  | // An example of using this library would be to take advantage of the | ||||||
|  | // golang.org/x/net/proxy library: | ||||||
|  | // | ||||||
|  | //	proxyUrl, _ := proxyURI, err = url.Parse("socks5://1.2.3.4:8888") | ||||||
|  | //	dialer, _ := proxy.FromURL(proxyURI, &net.Dialer{Timeout: 5 * time.Second}) | ||||||
|  | //	_ := girc.DialerConnect(dialer) | ||||||
|  | func (c *Client) DialerConnect(dialer Dialer) error { | ||||||
|  | 	return c.internalConnect(nil, dialer) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MockConnect is used to implement mocking with an IRC server. Supply a net.Conn | ||||||
|  | // that will be used to spoof the server. A useful way to do this is to so | ||||||
|  | // net.Pipe(), pass one end into MockConnect(), and the other end into | ||||||
|  | // bufio.NewReader(). | ||||||
|  | // | ||||||
|  | // For example: | ||||||
|  | // | ||||||
|  | //	client := girc.New(girc.Config{ | ||||||
|  | //		Server: "dummy.int", | ||||||
|  | //		Port:   6667, | ||||||
|  | //		Nick:   "test", | ||||||
|  | //		User:   "test", | ||||||
|  | //		Name:   "Testing123", | ||||||
|  | //	}) | ||||||
|  | // | ||||||
|  | //	in, out := net.Pipe() | ||||||
|  | //	defer in.Close() | ||||||
|  | //	defer out.Close() | ||||||
|  | //	b := bufio.NewReader(in) | ||||||
|  | // | ||||||
|  | //	go func() { | ||||||
|  | //		if err := client.MockConnect(out); err != nil { | ||||||
|  | //			panic(err) | ||||||
|  | //		} | ||||||
|  | //	}() | ||||||
|  | // | ||||||
|  | //	defer client.Close(false) | ||||||
|  | // | ||||||
|  | //	for { | ||||||
|  | //		in.SetReadDeadline(time.Now().Add(300 * time.Second)) | ||||||
|  | //		line, err := b.ReadString(byte('\n')) | ||||||
|  | //		if err != nil { | ||||||
|  | //			panic(err) | ||||||
|  | //		} | ||||||
|  | // | ||||||
|  | //		event := girc.ParseEvent(line) | ||||||
|  | // | ||||||
|  | //		if event == nil { | ||||||
|  | //	 		continue | ||||||
|  | //	 	} | ||||||
|  | // | ||||||
|  | //	 	// Do stuff with event here. | ||||||
|  | //	 } | ||||||
|  | func (c *Client) MockConnect(conn net.Conn) error { | ||||||
|  | 	return c.internalConnect(conn, nil) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Client) internalConnect(mock net.Conn, dialer Dialer) error { | ||||||
|  | 	// We want to be the only one handling connects/disconnects right now. | ||||||
|  | 	c.mu.Lock() | ||||||
|  |  | ||||||
|  | 	if c.conn != nil { | ||||||
|  | 		panic("use of connect more than once") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Reset the state. | ||||||
|  | 	c.state.reset() | ||||||
|  |  | ||||||
|  | 	if mock == nil { | ||||||
|  | 		// Validate info, and actually make the connection. | ||||||
|  | 		c.debug.Printf("connecting to %s...", c.Server()) | ||||||
|  | 		conn, err := newConn(c.Config, dialer, c.Server()) | ||||||
|  | 		if err != nil { | ||||||
|  | 			c.mu.Unlock() | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		c.conn = conn | ||||||
|  | 	} else { | ||||||
|  | 		c.conn = newMockConn(mock) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var ctx context.Context | ||||||
|  | 	ctx, c.stop = context.WithCancel(context.Background()) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	errs := make(chan error, 4) | ||||||
|  | 	var wg sync.WaitGroup | ||||||
|  | 	// 4 being the number of goroutines we need to finish when this function | ||||||
|  | 	// returns. | ||||||
|  | 	wg.Add(4) | ||||||
|  | 	go c.execLoop(ctx, errs, &wg) | ||||||
|  | 	go c.readLoop(ctx, errs, &wg) | ||||||
|  | 	go c.sendLoop(ctx, errs, &wg) | ||||||
|  | 	go c.pingLoop(ctx, errs, &wg) | ||||||
|  |  | ||||||
|  | 	// Passwords first. | ||||||
|  | 	if c.Config.ServerPass != "" { | ||||||
|  | 		c.write(&Event{Command: PASS, Params: []string{c.Config.ServerPass}, Sensitive: true}) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// List the IRCv3 capabilities, specifically with the max protocol we | ||||||
|  | 	// support. The IRCv3 specification doesn't directly state if this should | ||||||
|  | 	// be called directly before registration, or if it should be called | ||||||
|  | 	// after NICK/USER requests. It looks like non-supporting networks | ||||||
|  | 	// should ignore this, and some IRCv3 capable networks require this to | ||||||
|  | 	// occur before NICK/USER registration. | ||||||
|  | 	c.listCAP() | ||||||
|  |  | ||||||
|  | 	// Then nickname. | ||||||
|  | 	c.write(&Event{Command: NICK, Params: []string{c.Config.Nick}}) | ||||||
|  |  | ||||||
|  | 	// Then username and realname. | ||||||
|  | 	if c.Config.Name == "" { | ||||||
|  | 		c.Config.Name = c.Config.User | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.write(&Event{Command: USER, Params: []string{c.Config.User, "*", "*"}, Trailing: c.Config.Name}) | ||||||
|  |  | ||||||
|  | 	// Send a virtual event allowing hooks for successful socket connection. | ||||||
|  | 	c.RunHandlers(&Event{Command: INITIALIZED, Trailing: c.Server()}) | ||||||
|  |  | ||||||
|  | 	// Wait for the first error. | ||||||
|  | 	var result error | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		c.debug.Print("received request to close, beginning clean up") | ||||||
|  | 		c.RunHandlers(&Event{Command: STOPPED, Trailing: c.Server()}) | ||||||
|  | 	case err := <-errs: | ||||||
|  | 		c.debug.Print("received error, beginning clean up") | ||||||
|  | 		result = err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Make sure that the connection is closed if not already. | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	if c.stop != nil { | ||||||
|  | 		c.stop() | ||||||
|  | 	} | ||||||
|  | 	c.conn.mu.Lock() | ||||||
|  | 	c.conn.connected = false | ||||||
|  | 	_ = c.conn.Close() | ||||||
|  | 	c.conn.mu.Unlock() | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	// Once we have our error/result, let all other functions know we're done. | ||||||
|  | 	c.debug.Print("waiting for all routines to finish") | ||||||
|  |  | ||||||
|  | 	// Wait for all goroutines to finish. | ||||||
|  | 	wg.Wait() | ||||||
|  | 	close(errs) | ||||||
|  |  | ||||||
|  | 	// This helps ensure that the end user isn't improperly using the client | ||||||
|  | 	// more than once. If they want to do this, they should be using multiple | ||||||
|  | 	// clients, not multiple instances of Connect(). | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	c.conn = nil | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	return result | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // readLoop sets a timeout of 300 seconds, and then attempts to read from the | ||||||
|  | // IRC server. If there is an error, it calls Reconnect. | ||||||
|  | func (c *Client) readLoop(ctx context.Context, errs chan error, wg *sync.WaitGroup) { | ||||||
|  | 	c.debug.Print("starting readLoop") | ||||||
|  | 	defer c.debug.Print("closing readLoop") | ||||||
|  |  | ||||||
|  | 	var event *Event | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-ctx.Done(): | ||||||
|  | 			wg.Done() | ||||||
|  | 			return | ||||||
|  | 		default: | ||||||
|  | 			_ = c.conn.sock.SetReadDeadline(time.Now().Add(300 * time.Second)) | ||||||
|  | 			event, err = c.conn.decode() | ||||||
|  | 			if err != nil { | ||||||
|  | 				errs <- err | ||||||
|  | 				wg.Done() | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			c.rx <- event | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Send sends an event to the server. Use Client.RunHandlers() if you are | ||||||
|  | // simply looking to trigger handlers with an event. | ||||||
|  | func (c *Client) Send(event *Event) { | ||||||
|  | 	if !c.Config.AllowFlood { | ||||||
|  | 		<-time.After(c.conn.rate(event.Len())) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if c.Config.GlobalFormat && event.Trailing != "" && | ||||||
|  | 		(event.Command == PRIVMSG || event.Command == TOPIC || event.Command == NOTICE) { | ||||||
|  | 		event.Trailing = Fmt(event.Trailing) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.write(event) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // write is the lower level function to write an event. It does not have a | ||||||
|  | // write-delay when sending events. | ||||||
|  | func (c *Client) write(event *Event) { | ||||||
|  | 	c.tx <- event | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // rate allows limiting events based on how frequent the event is being sent, | ||||||
|  | // as well as how many characters each event has. | ||||||
|  | func (c *ircConn) rate(chars int) time.Duration { | ||||||
|  | 	_time := time.Second + ((time.Duration(chars) * time.Second) / 100) | ||||||
|  |  | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	if c.writeDelay += _time - time.Now().Sub(c.lastWrite); c.writeDelay < 0 { | ||||||
|  | 		c.writeDelay = 0 | ||||||
|  | 	} | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	defer c.mu.RUnlock() | ||||||
|  | 	if c.writeDelay > (8 * time.Second) { | ||||||
|  | 		return _time | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return 0 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Client) sendLoop(ctx context.Context, errs chan error, wg *sync.WaitGroup) { | ||||||
|  | 	c.debug.Print("starting sendLoop") | ||||||
|  | 	defer c.debug.Print("closing sendLoop") | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case event := <-c.tx: | ||||||
|  | 			// Check if tags exist on the event. If they do, and message-tags | ||||||
|  | 			// isn't a supported capability, remove them from the event. | ||||||
|  | 			if event.Tags != nil { | ||||||
|  | 				c.state.RLock() | ||||||
|  | 				var in bool | ||||||
|  | 				for i := 0; i < len(c.state.enabledCap); i++ { | ||||||
|  | 					if c.state.enabledCap[i] == "message-tags" { | ||||||
|  | 						in = true | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 				if !in { | ||||||
|  | 					event.Tags = Tags{} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Log the event. | ||||||
|  | 			if event.Sensitive { | ||||||
|  | 				c.debug.Printf("> %s ***redacted***", event.Command) | ||||||
|  | 			} else { | ||||||
|  | 				c.debug.Print("> ", StripRaw(event.String())) | ||||||
|  | 			} | ||||||
|  | 			if c.Config.Out != nil { | ||||||
|  | 				if pretty, ok := event.Pretty(); ok { | ||||||
|  | 					fmt.Fprintln(c.Config.Out, StripRaw(pretty)) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			c.conn.mu.Lock() | ||||||
|  | 			c.conn.lastWrite = time.Now() | ||||||
|  |  | ||||||
|  | 			if event.Command != PING && event.Command != PONG && event.Command != WHO { | ||||||
|  | 				c.conn.lastActive = c.conn.lastWrite | ||||||
|  | 			} | ||||||
|  | 			c.conn.mu.Unlock() | ||||||
|  |  | ||||||
|  | 			// Write the raw line. | ||||||
|  | 			_, err = c.conn.io.Write(event.Bytes()) | ||||||
|  | 			if err == nil { | ||||||
|  | 				// And the \r\n. | ||||||
|  | 				_, err = c.conn.io.Write(endline) | ||||||
|  | 				if err == nil { | ||||||
|  | 					// Lastly, flush everything to the socket. | ||||||
|  | 					err = c.conn.io.Flush() | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if err != nil { | ||||||
|  | 				errs <- err | ||||||
|  | 				wg.Done() | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		case <-ctx.Done(): | ||||||
|  | 			wg.Done() | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrTimedOut is returned when we attempt to ping the server, and timed out | ||||||
|  | // before receiving a PONG back. | ||||||
|  | type ErrTimedOut struct { | ||||||
|  | 	// TimeSinceSuccess is how long ago we received a successful pong. | ||||||
|  | 	TimeSinceSuccess time.Duration | ||||||
|  | 	// LastPong is the time we received our last successful pong. | ||||||
|  | 	LastPong time.Time | ||||||
|  | 	// LastPong is the last time we sent a pong request. | ||||||
|  | 	LastPing time.Time | ||||||
|  | 	// Delay is the configured delay between how often we send a ping request. | ||||||
|  | 	Delay time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ErrTimedOut) Error() string { return "timed out during ping to server" } | ||||||
|  |  | ||||||
|  | func (c *Client) pingLoop(ctx context.Context, errs chan error, wg *sync.WaitGroup) { | ||||||
|  | 	// Don't run the pingLoop if they want to disable it. | ||||||
|  | 	if c.Config.PingDelay <= 0 { | ||||||
|  | 		wg.Done() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.debug.Print("starting pingLoop") | ||||||
|  | 	defer c.debug.Print("closing pingLoop") | ||||||
|  |  | ||||||
|  | 	c.conn.mu.Lock() | ||||||
|  | 	c.conn.lastPing = time.Now() | ||||||
|  | 	c.conn.lastPong = time.Now() | ||||||
|  | 	c.conn.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	tick := time.NewTicker(c.Config.PingDelay) | ||||||
|  | 	defer tick.Stop() | ||||||
|  |  | ||||||
|  | 	started := time.Now() | ||||||
|  | 	past := false | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-tick.C: | ||||||
|  | 			// Delay during connect to wait for the client to register, otherwise | ||||||
|  | 			// some ircd's will not respond (e.g. during SASL negotiation). | ||||||
|  | 			if !past { | ||||||
|  | 				if time.Since(started) < 30*time.Second { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				past = true | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			c.conn.mu.RLock() | ||||||
|  | 			if time.Since(c.conn.lastPong) > c.Config.PingDelay+(60*time.Second) { | ||||||
|  | 				// It's 60 seconds over what out ping delay is, connection | ||||||
|  | 				// has probably dropped. | ||||||
|  | 				errs <- ErrTimedOut{ | ||||||
|  | 					TimeSinceSuccess: time.Since(c.conn.lastPong), | ||||||
|  | 					LastPong:         c.conn.lastPong, | ||||||
|  | 					LastPing:         c.conn.lastPing, | ||||||
|  | 					Delay:            c.Config.PingDelay, | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				wg.Done() | ||||||
|  | 				c.conn.mu.RUnlock() | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			c.conn.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 			c.conn.mu.Lock() | ||||||
|  | 			c.conn.lastPing = time.Now() | ||||||
|  | 			c.conn.mu.Unlock() | ||||||
|  |  | ||||||
|  | 			c.Cmd.Ping(fmt.Sprintf("%d", time.Now().UnixNano())) | ||||||
|  | 		case <-ctx.Done(): | ||||||
|  | 			wg.Done() | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										338
									
								
								vendor/github.com/lrstanley/girc/contants.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										338
									
								
								vendor/github.com/lrstanley/girc/contants.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,338 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | // Standard CTCP based constants. | ||||||
|  | const ( | ||||||
|  | 	CTCP_PING       = "PING" | ||||||
|  | 	CTCP_PONG       = "PONG" | ||||||
|  | 	CTCP_VERSION    = "VERSION" | ||||||
|  | 	CTCP_USERINFO   = "USERINFO" | ||||||
|  | 	CTCP_CLIENTINFO = "CLIENTINFO" | ||||||
|  | 	CTCP_SOURCE     = "SOURCE" | ||||||
|  | 	CTCP_TIME       = "TIME" | ||||||
|  | 	CTCP_FINGER     = "FINGER" | ||||||
|  | 	CTCP_ERRMSG     = "ERRMSG" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Emulated event commands used to allow easier hooks into the changing | ||||||
|  | // state of the client. | ||||||
|  | const ( | ||||||
|  | 	UPDATE_STATE   = "CLIENT_STATE_UPDATED"   // when channel/user state is updated. | ||||||
|  | 	UPDATE_GENERAL = "CLIENT_GENERAL_UPDATED" // when general state (client nick, server name, etc) is updated. | ||||||
|  | 	ALL_EVENTS     = "*"                      // trigger on all events | ||||||
|  | 	CONNECTED      = "CLIENT_CONNECTED"       // when it's safe to send arbitrary commands (joins, list, who, etc), trailing is host:port | ||||||
|  | 	INITIALIZED    = "CLIENT_INIT"            // verifies successful socket connection, trailing is host:port | ||||||
|  | 	DISCONNECTED   = "CLIENT_DISCONNECTED"    // occurs when we're disconnected from the server (user-requested or not) | ||||||
|  | 	STOPPED        = "CLIENT_STOPPED"         // occurs when Client.Stop() has been called | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // User/channel prefixes :: RFC1459. | ||||||
|  | const ( | ||||||
|  | 	DefaultPrefixes = "(ov)@+" // the most common default prefixes | ||||||
|  | 	ModeAddPrefix   = "+"      // modes are being added | ||||||
|  | 	ModeDelPrefix   = "-"      // modes are being removed | ||||||
|  |  | ||||||
|  | 	ChannelPrefix      = "#" // regular channel | ||||||
|  | 	DistributedPrefix  = "&" // distributed channel | ||||||
|  | 	OwnerPrefix        = "~" // user owner +q (non-rfc) | ||||||
|  | 	AdminPrefix        = "&" // user admin +a (non-rfc) | ||||||
|  | 	HalfOperatorPrefix = "%" // user half operator +h (non-rfc) | ||||||
|  | 	OperatorPrefix     = "@" // user operator +o | ||||||
|  | 	VoicePrefix        = "+" // user has voice +v | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // User modes :: RFC1459; section 4.2.3.2. | ||||||
|  | const ( | ||||||
|  | 	UserModeInvisible     = "i" // invisible | ||||||
|  | 	UserModeOperator      = "o" // server operator | ||||||
|  | 	UserModeServerNotices = "s" // user wants to receive server notices | ||||||
|  | 	UserModeWallops       = "w" // user wants to receive wallops | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Channel modes :: RFC1459; section 4.2.3.1. | ||||||
|  | const ( | ||||||
|  | 	ModeDefaults = "beI,k,l,imnpst" // the most common default modes | ||||||
|  |  | ||||||
|  | 	ModeInviteOnly = "i" // only join with an invite | ||||||
|  | 	ModeKey        = "k" // channel password | ||||||
|  | 	ModeLimit      = "l" // user limit | ||||||
|  | 	ModeModerated  = "m" // only voiced users and operators can talk | ||||||
|  | 	ModeOperator   = "o" // operator | ||||||
|  | 	ModePrivate    = "p" // private | ||||||
|  | 	ModeSecret     = "s" // secret | ||||||
|  | 	ModeTopic      = "t" // must be op to set topic | ||||||
|  | 	ModeVoice      = "v" // speak during moderation mode | ||||||
|  |  | ||||||
|  | 	ModeOwner        = "q" // owner privileges (non-rfc) | ||||||
|  | 	ModeAdmin        = "a" // admin privileges (non-rfc) | ||||||
|  | 	ModeHalfOperator = "h" // half-operator privileges (non-rfc) | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // IRC commands :: RFC2812; section 3 :: RFC2813; section 4. | ||||||
|  | const ( | ||||||
|  | 	ADMIN    = "ADMIN" | ||||||
|  | 	AWAY     = "AWAY" | ||||||
|  | 	CONNECT  = "CONNECT" | ||||||
|  | 	DIE      = "DIE" | ||||||
|  | 	ERROR    = "ERROR" | ||||||
|  | 	INFO     = "INFO" | ||||||
|  | 	INVITE   = "INVITE" | ||||||
|  | 	ISON     = "ISON" | ||||||
|  | 	JOIN     = "JOIN" | ||||||
|  | 	KICK     = "KICK" | ||||||
|  | 	KILL     = "KILL" | ||||||
|  | 	LINKS    = "LINKS" | ||||||
|  | 	LIST     = "LIST" | ||||||
|  | 	LUSERS   = "LUSERS" | ||||||
|  | 	MODE     = "MODE" | ||||||
|  | 	MOTD     = "MOTD" | ||||||
|  | 	NAMES    = "NAMES" | ||||||
|  | 	NICK     = "NICK" | ||||||
|  | 	NJOIN    = "NJOIN" | ||||||
|  | 	NOTICE   = "NOTICE" | ||||||
|  | 	OPER     = "OPER" | ||||||
|  | 	PART     = "PART" | ||||||
|  | 	PASS     = "PASS" | ||||||
|  | 	PING     = "PING" | ||||||
|  | 	PONG     = "PONG" | ||||||
|  | 	PRIVMSG  = "PRIVMSG" | ||||||
|  | 	QUIT     = "QUIT" | ||||||
|  | 	REHASH   = "REHASH" | ||||||
|  | 	RESTART  = "RESTART" | ||||||
|  | 	SERVER   = "SERVER" | ||||||
|  | 	SERVICE  = "SERVICE" | ||||||
|  | 	SERVLIST = "SERVLIST" | ||||||
|  | 	SQUERY   = "SQUERY" | ||||||
|  | 	SQUIT    = "SQUIT" | ||||||
|  | 	STATS    = "STATS" | ||||||
|  | 	SUMMON   = "SUMMON" | ||||||
|  | 	TIME     = "TIME" | ||||||
|  | 	TOPIC    = "TOPIC" | ||||||
|  | 	TRACE    = "TRACE" | ||||||
|  | 	USER     = "USER" | ||||||
|  | 	USERHOST = "USERHOST" | ||||||
|  | 	USERS    = "USERS" | ||||||
|  | 	VERSION  = "VERSION" | ||||||
|  | 	WALLOPS  = "WALLOPS" | ||||||
|  | 	WHO      = "WHO" | ||||||
|  | 	WHOIS    = "WHOIS" | ||||||
|  | 	WHOWAS   = "WHOWAS" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Numeric IRC reply mapping :: RFC2812; section 5. | ||||||
|  | const ( | ||||||
|  | 	RPL_WELCOME           = "001" | ||||||
|  | 	RPL_YOURHOST          = "002" | ||||||
|  | 	RPL_CREATED           = "003" | ||||||
|  | 	RPL_MYINFO            = "004" | ||||||
|  | 	RPL_BOUNCE            = "005" | ||||||
|  | 	RPL_ISUPPORT          = "005" | ||||||
|  | 	RPL_USERHOST          = "302" | ||||||
|  | 	RPL_ISON              = "303" | ||||||
|  | 	RPL_AWAY              = "301" | ||||||
|  | 	RPL_UNAWAY            = "305" | ||||||
|  | 	RPL_NOWAWAY           = "306" | ||||||
|  | 	RPL_WHOISUSER         = "311" | ||||||
|  | 	RPL_WHOISSERVER       = "312" | ||||||
|  | 	RPL_WHOISOPERATOR     = "313" | ||||||
|  | 	RPL_WHOISIDLE         = "317" | ||||||
|  | 	RPL_ENDOFWHOIS        = "318" | ||||||
|  | 	RPL_WHOISCHANNELS     = "319" | ||||||
|  | 	RPL_WHOWASUSER        = "314" | ||||||
|  | 	RPL_ENDOFWHOWAS       = "369" | ||||||
|  | 	RPL_LISTSTART         = "321" | ||||||
|  | 	RPL_LIST              = "322" | ||||||
|  | 	RPL_LISTEND           = "323" | ||||||
|  | 	RPL_UNIQOPIS          = "325" | ||||||
|  | 	RPL_CHANNELMODEIS     = "324" | ||||||
|  | 	RPL_NOTOPIC           = "331" | ||||||
|  | 	RPL_TOPIC             = "332" | ||||||
|  | 	RPL_INVITING          = "341" | ||||||
|  | 	RPL_SUMMONING         = "342" | ||||||
|  | 	RPL_INVITELIST        = "346" | ||||||
|  | 	RPL_ENDOFINVITELIST   = "347" | ||||||
|  | 	RPL_EXCEPTLIST        = "348" | ||||||
|  | 	RPL_ENDOFEXCEPTLIST   = "349" | ||||||
|  | 	RPL_VERSION           = "351" | ||||||
|  | 	RPL_WHOREPLY          = "352" | ||||||
|  | 	RPL_ENDOFWHO          = "315" | ||||||
|  | 	RPL_NAMREPLY          = "353" | ||||||
|  | 	RPL_ENDOFNAMES        = "366" | ||||||
|  | 	RPL_LINKS             = "364" | ||||||
|  | 	RPL_ENDOFLINKS        = "365" | ||||||
|  | 	RPL_BANLIST           = "367" | ||||||
|  | 	RPL_ENDOFBANLIST      = "368" | ||||||
|  | 	RPL_INFO              = "371" | ||||||
|  | 	RPL_ENDOFINFO         = "374" | ||||||
|  | 	RPL_MOTDSTART         = "375" | ||||||
|  | 	RPL_MOTD              = "372" | ||||||
|  | 	RPL_ENDOFMOTD         = "376" | ||||||
|  | 	RPL_YOUREOPER         = "381" | ||||||
|  | 	RPL_REHASHING         = "382" | ||||||
|  | 	RPL_YOURESERVICE      = "383" | ||||||
|  | 	RPL_TIME              = "391" | ||||||
|  | 	RPL_USERSSTART        = "392" | ||||||
|  | 	RPL_USERS             = "393" | ||||||
|  | 	RPL_ENDOFUSERS        = "394" | ||||||
|  | 	RPL_NOUSERS           = "395" | ||||||
|  | 	RPL_TRACELINK         = "200" | ||||||
|  | 	RPL_TRACECONNECTING   = "201" | ||||||
|  | 	RPL_TRACEHANDSHAKE    = "202" | ||||||
|  | 	RPL_TRACEUNKNOWN      = "203" | ||||||
|  | 	RPL_TRACEOPERATOR     = "204" | ||||||
|  | 	RPL_TRACEUSER         = "205" | ||||||
|  | 	RPL_TRACESERVER       = "206" | ||||||
|  | 	RPL_TRACESERVICE      = "207" | ||||||
|  | 	RPL_TRACENEWTYPE      = "208" | ||||||
|  | 	RPL_TRACECLASS        = "209" | ||||||
|  | 	RPL_TRACERECONNECT    = "210" | ||||||
|  | 	RPL_TRACELOG          = "261" | ||||||
|  | 	RPL_TRACEEND          = "262" | ||||||
|  | 	RPL_STATSLINKINFO     = "211" | ||||||
|  | 	RPL_STATSCOMMANDS     = "212" | ||||||
|  | 	RPL_ENDOFSTATS        = "219" | ||||||
|  | 	RPL_STATSUPTIME       = "242" | ||||||
|  | 	RPL_STATSOLINE        = "243" | ||||||
|  | 	RPL_UMODEIS           = "221" | ||||||
|  | 	RPL_SERVLIST          = "234" | ||||||
|  | 	RPL_SERVLISTEND       = "235" | ||||||
|  | 	RPL_LUSERCLIENT       = "251" | ||||||
|  | 	RPL_LUSEROP           = "252" | ||||||
|  | 	RPL_LUSERUNKNOWN      = "253" | ||||||
|  | 	RPL_LUSERCHANNELS     = "254" | ||||||
|  | 	RPL_LUSERME           = "255" | ||||||
|  | 	RPL_ADMINME           = "256" | ||||||
|  | 	RPL_ADMINLOC1         = "257" | ||||||
|  | 	RPL_ADMINLOC2         = "258" | ||||||
|  | 	RPL_ADMINEMAIL        = "259" | ||||||
|  | 	RPL_TRYAGAIN          = "263" | ||||||
|  | 	ERR_NOSUCHNICK        = "401" | ||||||
|  | 	ERR_NOSUCHSERVER      = "402" | ||||||
|  | 	ERR_NOSUCHCHANNEL     = "403" | ||||||
|  | 	ERR_CANNOTSENDTOCHAN  = "404" | ||||||
|  | 	ERR_TOOMANYCHANNELS   = "405" | ||||||
|  | 	ERR_WASNOSUCHNICK     = "406" | ||||||
|  | 	ERR_TOOMANYTARGETS    = "407" | ||||||
|  | 	ERR_NOSUCHSERVICE     = "408" | ||||||
|  | 	ERR_NOORIGIN          = "409" | ||||||
|  | 	ERR_NORECIPIENT       = "411" | ||||||
|  | 	ERR_NOTEXTTOSEND      = "412" | ||||||
|  | 	ERR_NOTOPLEVEL        = "413" | ||||||
|  | 	ERR_WILDTOPLEVEL      = "414" | ||||||
|  | 	ERR_BADMASK           = "415" | ||||||
|  | 	ERR_UNKNOWNCOMMAND    = "421" | ||||||
|  | 	ERR_NOMOTD            = "422" | ||||||
|  | 	ERR_NOADMININFO       = "423" | ||||||
|  | 	ERR_FILEERROR         = "424" | ||||||
|  | 	ERR_NONICKNAMEGIVEN   = "431" | ||||||
|  | 	ERR_ERRONEUSNICKNAME  = "432" | ||||||
|  | 	ERR_NICKNAMEINUSE     = "433" | ||||||
|  | 	ERR_NICKCOLLISION     = "436" | ||||||
|  | 	ERR_UNAVAILRESOURCE   = "437" | ||||||
|  | 	ERR_USERNOTINCHANNEL  = "441" | ||||||
|  | 	ERR_NOTONCHANNEL      = "442" | ||||||
|  | 	ERR_USERONCHANNEL     = "443" | ||||||
|  | 	ERR_NOLOGIN           = "444" | ||||||
|  | 	ERR_SUMMONDISABLED    = "445" | ||||||
|  | 	ERR_USERSDISABLED     = "446" | ||||||
|  | 	ERR_NOTREGISTERED     = "451" | ||||||
|  | 	ERR_NEEDMOREPARAMS    = "461" | ||||||
|  | 	ERR_ALREADYREGISTRED  = "462" | ||||||
|  | 	ERR_NOPERMFORHOST     = "463" | ||||||
|  | 	ERR_PASSWDMISMATCH    = "464" | ||||||
|  | 	ERR_YOUREBANNEDCREEP  = "465" | ||||||
|  | 	ERR_YOUWILLBEBANNED   = "466" | ||||||
|  | 	ERR_KEYSET            = "467" | ||||||
|  | 	ERR_CHANNELISFULL     = "471" | ||||||
|  | 	ERR_UNKNOWNMODE       = "472" | ||||||
|  | 	ERR_INVITEONLYCHAN    = "473" | ||||||
|  | 	ERR_BANNEDFROMCHAN    = "474" | ||||||
|  | 	ERR_BADCHANNELKEY     = "475" | ||||||
|  | 	ERR_BADCHANMASK       = "476" | ||||||
|  | 	ERR_NOCHANMODES       = "477" | ||||||
|  | 	ERR_BANLISTFULL       = "478" | ||||||
|  | 	ERR_NOPRIVILEGES      = "481" | ||||||
|  | 	ERR_CHANOPRIVSNEEDED  = "482" | ||||||
|  | 	ERR_CANTKILLSERVER    = "483" | ||||||
|  | 	ERR_RESTRICTED        = "484" | ||||||
|  | 	ERR_UNIQOPPRIVSNEEDED = "485" | ||||||
|  | 	ERR_NOOPERHOST        = "491" | ||||||
|  | 	ERR_UMODEUNKNOWNFLAG  = "501" | ||||||
|  | 	ERR_USERSDONTMATCH    = "502" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // IRCv3 commands and extensions :: http://ircv3.net/irc/. | ||||||
|  | const ( | ||||||
|  | 	AUTHENTICATE = "AUTHENTICATE" | ||||||
|  | 	STARTTLS     = "STARTTLS" | ||||||
|  |  | ||||||
|  | 	CAP       = "CAP" | ||||||
|  | 	CAP_ACK   = "ACK" | ||||||
|  | 	CAP_CLEAR = "CLEAR" | ||||||
|  | 	CAP_END   = "END" | ||||||
|  | 	CAP_LIST  = "LIST" | ||||||
|  | 	CAP_LS    = "LS" | ||||||
|  | 	CAP_NAK   = "NAK" | ||||||
|  | 	CAP_REQ   = "REQ" | ||||||
|  | 	CAP_NEW   = "NEW" | ||||||
|  | 	CAP_DEL   = "DEL" | ||||||
|  |  | ||||||
|  | 	CAP_CHGHOST = "CHGHOST" | ||||||
|  | 	CAP_AWAY    = "AWAY" | ||||||
|  | 	CAP_ACCOUNT = "ACCOUNT" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Numeric IRC reply mapping for ircv3 :: http://ircv3.net/irc/. | ||||||
|  | const ( | ||||||
|  | 	RPL_LOGGEDIN    = "900" | ||||||
|  | 	RPL_LOGGEDOUT   = "901" | ||||||
|  | 	RPL_NICKLOCKED  = "902" | ||||||
|  | 	RPL_SASLSUCCESS = "903" | ||||||
|  | 	ERR_SASLFAIL    = "904" | ||||||
|  | 	ERR_SASLTOOLONG = "905" | ||||||
|  | 	ERR_SASLABORTED = "906" | ||||||
|  | 	ERR_SASLALREADY = "907" | ||||||
|  | 	RPL_SASLMECHS   = "908" | ||||||
|  | 	RPL_STARTTLS    = "670" | ||||||
|  | 	ERR_STARTTLS    = "691" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Numeric IRC event mapping :: RFC2812; section 5.3. | ||||||
|  | const ( | ||||||
|  | 	RPL_STATSCLINE    = "213" | ||||||
|  | 	RPL_STATSNLINE    = "214" | ||||||
|  | 	RPL_STATSILINE    = "215" | ||||||
|  | 	RPL_STATSKLINE    = "216" | ||||||
|  | 	RPL_STATSQLINE    = "217" | ||||||
|  | 	RPL_STATSYLINE    = "218" | ||||||
|  | 	RPL_SERVICEINFO   = "231" | ||||||
|  | 	RPL_ENDOFSERVICES = "232" | ||||||
|  | 	RPL_SERVICE       = "233" | ||||||
|  | 	RPL_STATSVLINE    = "240" | ||||||
|  | 	RPL_STATSLLINE    = "241" | ||||||
|  | 	RPL_STATSHLINE    = "244" | ||||||
|  | 	RPL_STATSSLINE    = "245" | ||||||
|  | 	RPL_STATSPING     = "246" | ||||||
|  | 	RPL_STATSBLINE    = "247" | ||||||
|  | 	RPL_STATSDLINE    = "250" | ||||||
|  | 	RPL_NONE          = "300" | ||||||
|  | 	RPL_WHOISCHANOP   = "316" | ||||||
|  | 	RPL_KILLDONE      = "361" | ||||||
|  | 	RPL_CLOSING       = "362" | ||||||
|  | 	RPL_CLOSEEND      = "363" | ||||||
|  | 	RPL_INFOSTART     = "373" | ||||||
|  | 	RPL_MYPORTIS      = "384" | ||||||
|  | 	ERR_NOSERVICEHOST = "492" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Misc. | ||||||
|  | const ( | ||||||
|  | 	ERR_TOOMANYMATCHES = "416" // IRCNet. | ||||||
|  | 	RPL_GLOBALUSERS    = "266" // aircd/hybrid/bahamut, used on freenode. | ||||||
|  | 	RPL_LOCALUSERS     = "265" // aircd/hybrid/bahamut, used on freenode. | ||||||
|  | 	RPL_TOPICWHOTIME   = "333" // ircu, used on freenode. | ||||||
|  | 	RPL_WHOSPCRPL      = "354" // ircu, used on networks with WHOX support. | ||||||
|  | ) | ||||||
							
								
								
									
										288
									
								
								vendor/github.com/lrstanley/girc/ctcp.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										288
									
								
								vendor/github.com/lrstanley/girc/ctcp.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,288 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // ctcpDelim if the delimiter used for CTCP formatted events/messages. | ||||||
|  | const ctcpDelim byte = 0x01 // Prefix and suffix for CTCP messages. | ||||||
|  |  | ||||||
|  | // CTCPEvent is the necessary information from an IRC message. | ||||||
|  | type CTCPEvent struct { | ||||||
|  | 	// Origin is the original event that the CTCP event was decoded from. | ||||||
|  | 	Origin *Event `json:"origin"` | ||||||
|  | 	// Source is the author of the CTCP event. | ||||||
|  | 	Source *Source `json:"source"` | ||||||
|  | 	// Command is the type of CTCP event. E.g. PING, TIME, VERSION. | ||||||
|  | 	Command string `json:"command"` | ||||||
|  | 	// Text is the raw arguments following the command. | ||||||
|  | 	Text string `json:"text"` | ||||||
|  | 	// Reply is true if the CTCP event is intended to be a reply to a | ||||||
|  | 	// previous CTCP (e.g, if we sent one). | ||||||
|  | 	Reply bool `json:"reply"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // decodeCTCP decodes an incoming CTCP event, if it is CTCP. nil is returned | ||||||
|  | // if the incoming event does not match a valid CTCP. | ||||||
|  | func decodeCTCP(e *Event) *CTCPEvent { | ||||||
|  | 	// http://www.irchelp.org/protocol/ctcpspec.html | ||||||
|  |  | ||||||
|  | 	// Must be targeting a user/channel, AND trailing must have | ||||||
|  | 	// DELIM+TAG+DELIM minimum (at least 3 chars). | ||||||
|  | 	if len(e.Params) != 1 || len(e.Trailing) < 3 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if (e.Command != PRIVMSG && e.Command != NOTICE) || !IsValidNick(e.Params[0]) { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Trailing[0] != ctcpDelim || e.Trailing[len(e.Trailing)-1] != ctcpDelim { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Strip delimiters. | ||||||
|  | 	text := e.Trailing[1 : len(e.Trailing)-1] | ||||||
|  |  | ||||||
|  | 	s := strings.IndexByte(text, eventSpace) | ||||||
|  |  | ||||||
|  | 	// Check to see if it only contains a tag. | ||||||
|  | 	if s < 0 { | ||||||
|  | 		for i := 0; i < len(text); i++ { | ||||||
|  | 			// Check for A-Z, 0-9. | ||||||
|  | 			if (text[i] < 0x41 || text[i] > 0x5A) && (text[i] < 0x30 || text[i] > 0x39) { | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return &CTCPEvent{ | ||||||
|  | 			Origin:  e, | ||||||
|  | 			Source:  e.Source, | ||||||
|  | 			Command: text, | ||||||
|  | 			Reply:   e.Command == NOTICE, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Loop through checking the tag first. | ||||||
|  | 	for i := 0; i < s; i++ { | ||||||
|  | 		// Check for A-Z, 0-9. | ||||||
|  | 		if (text[i] < 0x41 || text[i] > 0x5A) && (text[i] < 0x30 || text[i] > 0x39) { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return &CTCPEvent{ | ||||||
|  | 		Origin:  e, | ||||||
|  | 		Source:  e.Source, | ||||||
|  | 		Command: text[0:s], | ||||||
|  | 		Text:    text[s+1:], | ||||||
|  | 		Reply:   e.Command == NOTICE, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // encodeCTCP encodes a CTCP event into a string, including delimiters. | ||||||
|  | func encodeCTCP(ctcp *CTCPEvent) (out string) { | ||||||
|  | 	if ctcp == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return encodeCTCPRaw(ctcp.Command, ctcp.Text) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // encodeCTCPRaw is much like encodeCTCP, however accepts a raw command and | ||||||
|  | // string as input. | ||||||
|  | func encodeCTCPRaw(cmd, text string) (out string) { | ||||||
|  | 	if len(cmd) <= 0 { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out = string(ctcpDelim) + cmd | ||||||
|  |  | ||||||
|  | 	if len(text) > 0 { | ||||||
|  | 		out += string(eventSpace) + text | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out + string(ctcpDelim) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // CTCP handles the storage and execution of CTCP handlers against incoming | ||||||
|  | // CTCP events. | ||||||
|  | type CTCP struct { | ||||||
|  | 	// mu is the mutex that should be used when accessing any ctcp handlers. | ||||||
|  | 	mu sync.RWMutex | ||||||
|  | 	// handlers is a map of CTCP message -> functions. | ||||||
|  | 	handlers map[string]CTCPHandler | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // newCTCP returns a new clean CTCP handler. | ||||||
|  | func newCTCP() *CTCP { | ||||||
|  | 	return &CTCP{handlers: map[string]CTCPHandler{}} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // call executes the necessary CTCP handler for the incoming event/CTCP | ||||||
|  | // command. | ||||||
|  | func (c *CTCP) call(client *Client, event *CTCPEvent) { | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	defer c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	// If they want to catch any panics, add to defer stack. | ||||||
|  | 	if client.Config.RecoverFunc != nil && event.Origin != nil { | ||||||
|  | 		defer recoverHandlerPanic(client, event.Origin, "ctcp-"+strings.ToLower(event.Command), 3) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Support wildcard CTCP event handling. Gets executed first before | ||||||
|  | 	// regular event handlers. | ||||||
|  | 	if _, ok := c.handlers["*"]; ok { | ||||||
|  | 		c.handlers["*"](client, *event) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, ok := c.handlers[event.Command]; !ok { | ||||||
|  | 		// Send a ERRMSG reply, if we know who sent it. | ||||||
|  | 		if event.Source != nil && IsValidNick(event.Source.Name) { | ||||||
|  | 			client.Cmd.SendCTCPReply(event.Source.Name, CTCP_ERRMSG, "that is an unknown CTCP query") | ||||||
|  | 		} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.handlers[event.Command](client, *event) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // parseCMD parses a CTCP command/tag, ensuring it's valid. If not, an empty | ||||||
|  | // string is returned. | ||||||
|  | func (c *CTCP) parseCMD(cmd string) string { | ||||||
|  | 	// TODO: Needs proper testing. | ||||||
|  | 	// Check if wildcard. | ||||||
|  | 	if cmd == "*" { | ||||||
|  | 		return "*" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	cmd = strings.ToUpper(cmd) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(cmd); i++ { | ||||||
|  | 		// Check for A-Z, 0-9. | ||||||
|  | 		if (cmd[i] < 0x41 || cmd[i] > 0x5A) && (cmd[i] < 0x30 || cmd[i] > 0x39) { | ||||||
|  | 			return "" | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Set saves handler for execution upon a matching incoming CTCP event. | ||||||
|  | // Use SetBg if the handler may take an extended period of time to execute. | ||||||
|  | // If you would like to have a handler which will catch ALL CTCP requests, | ||||||
|  | // simply use "*" in place of the command. | ||||||
|  | func (c *CTCP) Set(cmd string, handler func(client *Client, ctcp CTCPEvent)) { | ||||||
|  | 	if cmd = c.parseCMD(cmd); cmd == "" { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	c.handlers[cmd] = CTCPHandler(handler) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetBg is much like Set, however the handler is executed in the background, | ||||||
|  | // ensuring that event handling isn't hung during long running tasks. See Set | ||||||
|  | // for more information. | ||||||
|  | func (c *CTCP) SetBg(cmd string, handler func(client *Client, ctcp CTCPEvent)) { | ||||||
|  | 	c.Set(cmd, func(client *Client, ctcp CTCPEvent) { | ||||||
|  | 		go handler(client, ctcp) | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Clear removes currently setup handler for cmd, if one is set. | ||||||
|  | func (c *CTCP) Clear(cmd string) { | ||||||
|  | 	if cmd = c.parseCMD(cmd); cmd == "" { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	delete(c.handlers, cmd) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ClearAll removes all currently setup and re-sets the default handlers. | ||||||
|  | func (c *CTCP) ClearAll() { | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	c.handlers = map[string]CTCPHandler{} | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	// Register necessary handlers. | ||||||
|  | 	c.addDefaultHandlers() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // CTCPHandler is a type that represents the function necessary to | ||||||
|  | // implement a CTCP handler. | ||||||
|  | type CTCPHandler func(client *Client, ctcp CTCPEvent) | ||||||
|  |  | ||||||
|  | // addDefaultHandlers adds some useful default CTCP response handlers. | ||||||
|  | func (c *CTCP) addDefaultHandlers() { | ||||||
|  | 	c.SetBg(CTCP_PING, handleCTCPPing) | ||||||
|  | 	c.SetBg(CTCP_PONG, handleCTCPPong) | ||||||
|  | 	c.SetBg(CTCP_VERSION, handleCTCPVersion) | ||||||
|  | 	c.SetBg(CTCP_SOURCE, handleCTCPSource) | ||||||
|  | 	c.SetBg(CTCP_TIME, handleCTCPTime) | ||||||
|  | 	c.SetBg(CTCP_FINGER, handleCTCPFinger) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPPing replies with a ping and whatever was originally requested. | ||||||
|  | func handleCTCPPing(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	if ctcp.Reply { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_PING, ctcp.Text) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPPong replies with a pong. | ||||||
|  | func handleCTCPPong(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	if ctcp.Reply { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_PONG, "") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPVersion replies with the name of the client, Go version, as well | ||||||
|  | // as the os type (darwin, linux, windows, etc) and architecture type (x86, | ||||||
|  | // arm, etc). | ||||||
|  | func handleCTCPVersion(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	if client.Config.Version != "" { | ||||||
|  | 		client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_VERSION, client.Config.Version) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	client.Cmd.SendCTCPReplyf( | ||||||
|  | 		ctcp.Source.Name, CTCP_VERSION, | ||||||
|  | 		"girc (github.com/lrstanley/girc) using %s (%s, %s)", | ||||||
|  | 		runtime.Version(), runtime.GOOS, runtime.GOARCH, | ||||||
|  | 	) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPSource replies with the public git location of this library. | ||||||
|  | func handleCTCPSource(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_SOURCE, "https://github.com/lrstanley/girc") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPTime replies with a RFC 1123 (Z) formatted version of Go's | ||||||
|  | // local time. | ||||||
|  | func handleCTCPTime(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_TIME, ":"+time.Now().Format(time.RFC1123Z)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleCTCPFinger replies with the realname and idle time of the user. This | ||||||
|  | // is obsoleted by improvements to the IRC protocol, however still supported. | ||||||
|  | func handleCTCPFinger(client *Client, ctcp CTCPEvent) { | ||||||
|  | 	client.conn.mu.RLock() | ||||||
|  | 	active := client.conn.lastActive | ||||||
|  | 	client.conn.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	client.Cmd.SendCTCPReply(ctcp.Source.Name, CTCP_FINGER, fmt.Sprintf("%s -- idle %s", client.Config.Name, time.Since(active))) | ||||||
|  | } | ||||||
							
								
								
									
										12
									
								
								vendor/github.com/lrstanley/girc/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								vendor/github.com/lrstanley/girc/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | |||||||
|  | // Package girc provides a high level, yet flexible IRC library for use with | ||||||
|  | // interacting with IRC servers. girc has support for user/channel tracking, | ||||||
|  | // as well as a few other neat features (like auto-reconnect). | ||||||
|  | // | ||||||
|  | // Much of what girc can do, can also be disabled. The goal is to provide a | ||||||
|  | // solid API that you don't necessarily have to work with out of the box if | ||||||
|  | // you don't want to. | ||||||
|  | // | ||||||
|  | // See the examples below for a few brief and useful snippets taking | ||||||
|  | // advantage of girc, which should give you a general idea of how the API | ||||||
|  | // works. | ||||||
|  | package girc | ||||||
							
								
								
									
										550
									
								
								vendor/github.com/lrstanley/girc/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										550
									
								
								vendor/github.com/lrstanley/girc/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,550 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	eventSpace byte = 0x20 // Separator. | ||||||
|  | 	maxLength       = 510  // Maximum length is 510 (2 for line endings). | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // cutCRFunc is used to trim CR characters from prefixes/messages. | ||||||
|  | func cutCRFunc(r rune) bool { | ||||||
|  | 	return r == '\r' || r == '\n' | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Event represents an IRC protocol message, see RFC1459 section 2.3.1 | ||||||
|  | // | ||||||
|  | //    <message>  :: [':' <prefix> <SPACE>] <command> <params> <crlf> | ||||||
|  | //    <prefix>   :: <servername> | <nick> ['!' <user>] ['@' <host>] | ||||||
|  | //    <command>  :: <letter>{<letter>} | <number> <number> <number> | ||||||
|  | //    <SPACE>    :: ' '{' '} | ||||||
|  | //    <params>   :: <SPACE> [':' <trailing> | <middle> <params>] | ||||||
|  | //    <middle>   :: <Any *non-empty* sequence of octets not including SPACE or NUL | ||||||
|  | //                   or CR or LF, the first of which may not be ':'> | ||||||
|  | //    <trailing> :: <Any, possibly empty, sequence of octets not including NUL or | ||||||
|  | //                   CR or LF> | ||||||
|  | //    <crlf>     :: CR LF | ||||||
|  | type Event struct { | ||||||
|  | 	Source        *Source  `json:"source"`         // The source of the event. | ||||||
|  | 	Tags          Tags     `json:"tags"`           // IRCv3 style message tags. Only use if network supported. | ||||||
|  | 	Command       string   `json:"command"`        // the IRC command, e.g. JOIN, PRIVMSG, KILL. | ||||||
|  | 	Params        []string `json:"params"`         // parameters to the command. Commonly nickname, channel, etc. | ||||||
|  | 	Trailing      string   `json:"trailing"`       // any trailing data. e.g. with a PRIVMSG, this is the message text. | ||||||
|  | 	EmptyTrailing bool     `json:"empty_trailing"` // if true, trailing prefix (:) will be added even if Event.Trailing is empty. | ||||||
|  | 	Sensitive     bool     `json:"sensitive"`      // if the message is sensitive (e.g. and should not be logged). | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ParseEvent takes a string and attempts to create a Event struct. | ||||||
|  | // | ||||||
|  | // Returns nil if the Event is invalid. | ||||||
|  | func ParseEvent(raw string) (e *Event) { | ||||||
|  | 	// Ignore empty events. | ||||||
|  | 	if raw = strings.TrimFunc(raw, cutCRFunc); len(raw) < 2 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var i, j int | ||||||
|  | 	e = &Event{} | ||||||
|  |  | ||||||
|  | 	if raw[0] == prefixTag { | ||||||
|  | 		// Tags end with a space. | ||||||
|  | 		i = strings.IndexByte(raw, eventSpace) | ||||||
|  |  | ||||||
|  | 		if i < 2 { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		e.Tags = ParseTags(raw[1:i]) | ||||||
|  | 		raw = raw[i+1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if raw[0] == messagePrefix { | ||||||
|  | 		// Prefix ends with a space. | ||||||
|  | 		i = strings.IndexByte(raw, eventSpace) | ||||||
|  |  | ||||||
|  | 		// Prefix string must not be empty if the indicator is present. | ||||||
|  | 		if i < 2 { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		e.Source = ParseSource(raw[1:i]) | ||||||
|  |  | ||||||
|  | 		// Skip space at the end of the prefix. | ||||||
|  | 		i++ | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Find end of command. | ||||||
|  | 	j = i + strings.IndexByte(raw[i:], eventSpace) | ||||||
|  |  | ||||||
|  | 	if j < i { | ||||||
|  | 		// If there are no proceeding spaces, it's the only thing specified. | ||||||
|  | 		e.Command = strings.ToUpper(raw[i:]) | ||||||
|  | 		return e | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	e.Command = strings.ToUpper(raw[i:j]) | ||||||
|  |  | ||||||
|  | 	// Skip the space after the command. | ||||||
|  | 	j++ | ||||||
|  |  | ||||||
|  | 	// Check if and where the trailing text is within the incoming line. | ||||||
|  | 	var lastIndex, trailerIndex int | ||||||
|  | 	for { | ||||||
|  | 		// We must loop through, as it's possible that the first message | ||||||
|  | 		// prefix is not actually what we want. (e.g, colons are commonly | ||||||
|  | 		// used within ISUPPORT to delegate things like CHANLIMIT or TARGMAX.) | ||||||
|  | 		lastIndex = trailerIndex | ||||||
|  | 		trailerIndex = strings.IndexByte(raw[j+lastIndex:], messagePrefix) | ||||||
|  |  | ||||||
|  | 		if trailerIndex == -1 { | ||||||
|  | 			// No trailing argument found, assume the rest is just params. | ||||||
|  | 			e.Params = strings.Split(raw[j:], string(eventSpace)) | ||||||
|  | 			return e | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// This means we found a prefix that was proceeded by a space, and | ||||||
|  | 		// it's good to assume this is the start of trailing text to the line. | ||||||
|  | 		if raw[j+lastIndex+trailerIndex-1] == eventSpace { | ||||||
|  | 			i = lastIndex + trailerIndex | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Keep looping through until we either can't find any more prefixes, | ||||||
|  | 		// or we find the one we want. | ||||||
|  | 		trailerIndex += lastIndex + 1 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Set i to that of the substring we were using before, and where the | ||||||
|  | 	// trailing prefix is. | ||||||
|  | 	i = j + i | ||||||
|  |  | ||||||
|  | 	// Check if we need to parse arguments. If so, take everything after the | ||||||
|  | 	// command, and right before the trailing prefix, and cut it up. | ||||||
|  | 	if i > j { | ||||||
|  | 		e.Params = strings.Split(raw[j:i-1], string(eventSpace)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	e.Trailing = raw[i+1:] | ||||||
|  |  | ||||||
|  | 	// We need to re-encode the trailing argument even if it was empty. | ||||||
|  | 	if len(e.Trailing) <= 0 { | ||||||
|  | 		e.EmptyTrailing = true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return e | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy makes a deep copy of a given event, for use with allowing untrusted | ||||||
|  | // functions/handlers edit the event without causing potential issues with | ||||||
|  | // other handlers. | ||||||
|  | func (e *Event) Copy() *Event { | ||||||
|  | 	if e == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	newEvent := &Event{ | ||||||
|  | 		Command:       e.Command, | ||||||
|  | 		Trailing:      e.Trailing, | ||||||
|  | 		EmptyTrailing: e.EmptyTrailing, | ||||||
|  | 		Sensitive:     e.Sensitive, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Copy Source field, as it's a pointer and needs to be dereferenced. | ||||||
|  | 	if e.Source != nil { | ||||||
|  | 		newEvent.Source = e.Source.Copy() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Copy Params in order to dereference as well. | ||||||
|  | 	if e.Params != nil { | ||||||
|  | 		newEvent.Params = make([]string, len(e.Params)) | ||||||
|  | 		copy(newEvent.Params, e.Params) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Copy tags as necessary. | ||||||
|  | 	if e.Tags != nil { | ||||||
|  | 		newEvent.Tags = Tags{} | ||||||
|  | 		for k, v := range e.Tags { | ||||||
|  | 			newEvent.Tags[k] = v | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return newEvent | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len calculates the length of the string representation of event. Note that | ||||||
|  | // this will return the true length (even if longer than what IRC supports), | ||||||
|  | // which may be useful if you are trying to check and see if a message is | ||||||
|  | // too long, to trim it down yourself. | ||||||
|  | func (e *Event) Len() (length int) { | ||||||
|  | 	if e.Tags != nil { | ||||||
|  | 		// Include tags and trailing space. | ||||||
|  | 		length = e.Tags.Len() + 1 | ||||||
|  | 	} | ||||||
|  | 	if e.Source != nil { | ||||||
|  | 		// Include prefix and trailing space. | ||||||
|  | 		length += e.Source.Len() + 2 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	length += len(e.Command) | ||||||
|  |  | ||||||
|  | 	if len(e.Params) > 0 { | ||||||
|  | 		length += len(e.Params) | ||||||
|  |  | ||||||
|  | 		for i := 0; i < len(e.Params); i++ { | ||||||
|  | 			length += len(e.Params[i]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(e.Trailing) > 0 || e.EmptyTrailing { | ||||||
|  | 		// Include prefix and space. | ||||||
|  | 		length += len(e.Trailing) + 2 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Bytes returns a []byte representation of event. Strips all newlines and | ||||||
|  | // carriage returns. | ||||||
|  | // | ||||||
|  | // Per RFC2812 section 2.3, messages should not exceed 512 characters in | ||||||
|  | // length. This method forces that limit by discarding any characters | ||||||
|  | // exceeding the length limit. | ||||||
|  | func (e *Event) Bytes() []byte { | ||||||
|  | 	buffer := new(bytes.Buffer) | ||||||
|  |  | ||||||
|  | 	// Tags. | ||||||
|  | 	if e.Tags != nil { | ||||||
|  | 		e.Tags.writeTo(buffer) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Event prefix. | ||||||
|  | 	if e.Source != nil { | ||||||
|  | 		buffer.WriteByte(messagePrefix) | ||||||
|  | 		e.Source.writeTo(buffer) | ||||||
|  | 		buffer.WriteByte(eventSpace) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Command is required. | ||||||
|  | 	buffer.WriteString(e.Command) | ||||||
|  |  | ||||||
|  | 	// Space separated list of arguments. | ||||||
|  | 	if len(e.Params) > 0 { | ||||||
|  | 		buffer.WriteByte(eventSpace) | ||||||
|  | 		buffer.WriteString(strings.Join(e.Params, string(eventSpace))) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(e.Trailing) > 0 || e.EmptyTrailing { | ||||||
|  | 		buffer.WriteByte(eventSpace) | ||||||
|  | 		buffer.WriteByte(messagePrefix) | ||||||
|  | 		buffer.WriteString(e.Trailing) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// We need the limit the buffer length. | ||||||
|  | 	if buffer.Len() > (maxLength) { | ||||||
|  | 		buffer.Truncate(maxLength) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out := buffer.Bytes() | ||||||
|  |  | ||||||
|  | 	// Strip newlines and carriage returns. | ||||||
|  | 	for i := 0; i < len(out); i++ { | ||||||
|  | 		if out[i] == 0x0A || out[i] == 0x0D { | ||||||
|  | 			out = append(out[:i], out[i+1:]...) | ||||||
|  | 			i-- // Decrease the index so we can pick up where we left off. | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a string representation of this event. Strips all newlines | ||||||
|  | // and carriage returns. | ||||||
|  | func (e *Event) String() string { | ||||||
|  | 	return string(e.Bytes()) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pretty returns a prettified string of the event. If the event doesn't | ||||||
|  | // support prettification, ok is false. Pretty is not just useful to make | ||||||
|  | // an event prettier, but also to filter out events that most don't visually | ||||||
|  | // see in normal IRC clients. e.g. most clients don't show WHO queries. | ||||||
|  | func (e *Event) Pretty() (out string, ok bool) { | ||||||
|  | 	if e.Sensitive { | ||||||
|  | 		return "", false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == ERROR { | ||||||
|  | 		return fmt.Sprintf("[*] an error occurred: %s", e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Source == nil { | ||||||
|  | 		if e.Command != PRIVMSG && e.Command != NOTICE { | ||||||
|  | 			return "", false | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if len(e.Params) > 0 && len(e.Trailing) > 0 { | ||||||
|  | 			return fmt.Sprintf("[>] writing %s [%s]: %s", strings.ToLower(e.Command), strings.Join(e.Params, ", "), e.Trailing), true | ||||||
|  | 		} else if len(e.Params) > 0 { | ||||||
|  | 			return fmt.Sprintf("[>] writing %s [%s]", strings.ToLower(e.Command), strings.Join(e.Params, ", ")), true | ||||||
|  | 		} else if len(e.Trailing) > 0 { | ||||||
|  | 			return fmt.Sprintf("[>] writing %s: %s", strings.ToLower(e.Command), e.Trailing), true | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return "", false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == INITIALIZED { | ||||||
|  | 		return fmt.Sprintf("[*] connection to %s initialized", e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == CONNECTED { | ||||||
|  | 		return fmt.Sprintf("[*] successfully connected to %s", e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if (e.Command == PRIVMSG || e.Command == NOTICE) && len(e.Params) > 0 { | ||||||
|  | 		if ctcp := decodeCTCP(e); ctcp != nil { | ||||||
|  | 			if ctcp.Reply { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			return fmt.Sprintf("[*] CTCP query from %s: %s%s", ctcp.Source.Name, ctcp.Command, " "+ctcp.Text), true | ||||||
|  | 		} | ||||||
|  | 		return fmt.Sprintf("[%s] (%s) %s", strings.Join(e.Params, ","), e.Source.Name, e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == RPL_MOTD || e.Command == RPL_MOTDSTART || | ||||||
|  | 		e.Command == RPL_WELCOME || e.Command == RPL_YOURHOST || | ||||||
|  | 		e.Command == RPL_CREATED || e.Command == RPL_LUSERCLIENT { | ||||||
|  | 		return "[*] " + e.Trailing, true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == JOIN && len(e.Params) > 0 { | ||||||
|  | 		return fmt.Sprintf("[*] %s (%s) has joined %s", e.Source.Name, e.Source.Host, e.Params[0]), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == PART && len(e.Params) > 0 { | ||||||
|  | 		return fmt.Sprintf("[*] %s (%s) has left %s (%s)", e.Source.Name, e.Source.Host, e.Params[0], e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == QUIT { | ||||||
|  | 		return fmt.Sprintf("[*] %s has quit (%s)", e.Source.Name, e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == KICK && len(e.Params) == 2 { | ||||||
|  | 		return fmt.Sprintf("[%s] *** %s has kicked %s: %s", e.Params[0], e.Source.Name, e.Params[1], e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == NICK && len(e.Params) == 1 { | ||||||
|  | 		return fmt.Sprintf("[*] %s is now known as %s", e.Source.Name, e.Params[0]), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == TOPIC && len(e.Params) > 0 { | ||||||
|  | 		return fmt.Sprintf("[%s] *** %s has set the topic to: %s", e.Params[len(e.Params)-1], e.Source.Name, e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == MODE && len(e.Params) > 2 { | ||||||
|  | 		return fmt.Sprintf("[%s] *** %s set modes: %s", e.Params[0], e.Source.Name, strings.Join(e.Params[1:], " ")), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == CAP_AWAY { | ||||||
|  | 		if len(e.Trailing) > 0 { | ||||||
|  | 			return fmt.Sprintf("[*] %s is now away: %s", e.Source.Name, e.Trailing), true | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return fmt.Sprintf("[*] %s is no longer away", e.Source.Name), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == CAP_CHGHOST && len(e.Params) == 2 { | ||||||
|  | 		return fmt.Sprintf("[*] %s has changed their host to %s (was %s)", e.Source.Name, e.Params[1], e.Source.Host), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == CAP_ACCOUNT && len(e.Params) == 1 { | ||||||
|  | 		if e.Params[0] == "*" { | ||||||
|  | 			return fmt.Sprintf("[*] %s has become un-authenticated", e.Source.Name), true | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return fmt.Sprintf("[*] %s has authenticated for account: %s", e.Source.Name, e.Params[0]), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Command == RPL_TOPIC && len(e.Params) > 0 && len(e.Trailing) > 0 { | ||||||
|  | 		return fmt.Sprintf("[*] topic for %s is: %s", e.Params[len(e.Params)-1], e.Trailing), true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return "", false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsAction checks to see if the event is a PRIVMSG, and is an ACTION (/me). | ||||||
|  | func (e *Event) IsAction() bool { | ||||||
|  | 	if e.Source == nil || e.Command != PRIVMSG || len(e.Trailing) < 9 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !strings.HasPrefix(e.Trailing, "\001ACTION") || e.Trailing[len(e.Trailing)-1] != ctcpDelim { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsFromChannel checks to see if a message was from a channel (rather than | ||||||
|  | // a private message). | ||||||
|  | func (e *Event) IsFromChannel() bool { | ||||||
|  | 	if e.Source == nil || e.Command != PRIVMSG || len(e.Params) < 1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !IsValidChannel(e.Params[0]) { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsFromUser checks to see if a message was from a user (rather than a | ||||||
|  | // channel). | ||||||
|  | func (e *Event) IsFromUser() bool { | ||||||
|  | 	if e.Source == nil || e.Command != PRIVMSG || len(e.Params) < 1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !IsValidNick(e.Params[0]) { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // StripAction returns the stripped version of the action encoding from a | ||||||
|  | // PRIVMSG ACTION (/me). | ||||||
|  | func (e *Event) StripAction() string { | ||||||
|  | 	if !e.IsAction() { | ||||||
|  | 		return e.Trailing | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return e.Trailing[8 : len(e.Trailing)-1] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	messagePrefix byte = 0x3A // ":" -- prefix or last argument | ||||||
|  | 	prefixIdent   byte = 0x21 // "!" -- username | ||||||
|  | 	prefixHost    byte = 0x40 // "@" -- hostname | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Source represents the sender of an IRC event, see RFC1459 section 2.3.1. | ||||||
|  | // <servername> | <nick> [ '!' <user> ] [ '@' <host> ] | ||||||
|  | type Source struct { | ||||||
|  | 	// Name is the nickname, server name, or service name. | ||||||
|  | 	Name string `json:"name"` | ||||||
|  | 	// Ident is commonly known as the "user". | ||||||
|  | 	Ident string `json:"ident"` | ||||||
|  | 	// Host is the hostname or IP address of the user/service. Is not accurate | ||||||
|  | 	// due to how IRC servers can spoof hostnames. | ||||||
|  | 	Host string `json:"host"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a deep copy of Source. | ||||||
|  | func (s *Source) Copy() *Source { | ||||||
|  | 	if s == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	newSource := &Source{ | ||||||
|  | 		Name:  s.Name, | ||||||
|  | 		Ident: s.Ident, | ||||||
|  | 		Host:  s.Host, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return newSource | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ParseSource takes a string and attempts to create a Source struct. | ||||||
|  | func ParseSource(raw string) (src *Source) { | ||||||
|  | 	src = new(Source) | ||||||
|  |  | ||||||
|  | 	user := strings.IndexByte(raw, prefixIdent) | ||||||
|  | 	host := strings.IndexByte(raw, prefixHost) | ||||||
|  |  | ||||||
|  | 	switch { | ||||||
|  | 	case user > 0 && host > user: | ||||||
|  | 		src.Name = raw[:user] | ||||||
|  | 		src.Ident = raw[user+1 : host] | ||||||
|  | 		src.Host = raw[host+1:] | ||||||
|  | 	case user > 0: | ||||||
|  | 		src.Name = raw[:user] | ||||||
|  | 		src.Ident = raw[user+1:] | ||||||
|  | 	case host > 0: | ||||||
|  | 		src.Name = raw[:host] | ||||||
|  | 		src.Host = raw[host+1:] | ||||||
|  | 	default: | ||||||
|  | 		src.Name = raw | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return src | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len calculates the length of the string representation of prefix | ||||||
|  | func (s *Source) Len() (length int) { | ||||||
|  | 	length = len(s.Name) | ||||||
|  | 	if len(s.Ident) > 0 { | ||||||
|  | 		length = 1 + length + len(s.Ident) | ||||||
|  | 	} | ||||||
|  | 	if len(s.Host) > 0 { | ||||||
|  | 		length = 1 + length + len(s.Host) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Bytes returns a []byte representation of source. | ||||||
|  | func (s *Source) Bytes() []byte { | ||||||
|  | 	buffer := new(bytes.Buffer) | ||||||
|  | 	s.writeTo(buffer) | ||||||
|  |  | ||||||
|  | 	return buffer.Bytes() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a string representation of source. | ||||||
|  | func (s *Source) String() (out string) { | ||||||
|  | 	out = s.Name | ||||||
|  | 	if len(s.Ident) > 0 { | ||||||
|  | 		out = out + string(prefixIdent) + s.Ident | ||||||
|  | 	} | ||||||
|  | 	if len(s.Host) > 0 { | ||||||
|  | 		out = out + string(prefixHost) + s.Host | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsHostmask returns true if source looks like a user hostmask. | ||||||
|  | func (s *Source) IsHostmask() bool { | ||||||
|  | 	return len(s.Ident) > 0 && len(s.Host) > 0 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsServer returns true if this source looks like a server name. | ||||||
|  | func (s *Source) IsServer() bool { | ||||||
|  | 	return len(s.Ident) <= 0 && len(s.Host) <= 0 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // writeTo is an utility function to write the source to the bytes.Buffer | ||||||
|  | // in Event.String(). | ||||||
|  | func (s *Source) writeTo(buffer *bytes.Buffer) { | ||||||
|  | 	buffer.WriteString(s.Name) | ||||||
|  | 	if len(s.Ident) > 0 { | ||||||
|  | 		buffer.WriteByte(prefixIdent) | ||||||
|  | 		buffer.WriteString(s.Ident) | ||||||
|  | 	} | ||||||
|  | 	if len(s.Host) > 0 { | ||||||
|  | 		buffer.WriteByte(prefixHost) | ||||||
|  | 		buffer.WriteString(s.Host) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
							
								
								
									
										350
									
								
								vendor/github.com/lrstanley/girc/format.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										350
									
								
								vendor/github.com/lrstanley/girc/format.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,350 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	fmtOpenChar  = 0x7B // { | ||||||
|  | 	fmtCloseChar = 0x7D // } | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var fmtColors = map[string]int{ | ||||||
|  | 	"white":       0, | ||||||
|  | 	"black":       1, | ||||||
|  | 	"blue":        2, | ||||||
|  | 	"navy":        2, | ||||||
|  | 	"green":       3, | ||||||
|  | 	"red":         4, | ||||||
|  | 	"brown":       5, | ||||||
|  | 	"maroon":      5, | ||||||
|  | 	"purple":      6, | ||||||
|  | 	"gold":        7, | ||||||
|  | 	"olive":       7, | ||||||
|  | 	"orange":      7, | ||||||
|  | 	"yellow":      8, | ||||||
|  | 	"lightgreen":  9, | ||||||
|  | 	"lime":        9, | ||||||
|  | 	"teal":        10, | ||||||
|  | 	"cyan":        11, | ||||||
|  | 	"lightblue":   12, | ||||||
|  | 	"royal":       12, | ||||||
|  | 	"fuchsia":     13, | ||||||
|  | 	"lightpurple": 13, | ||||||
|  | 	"pink":        13, | ||||||
|  | 	"gray":        14, | ||||||
|  | 	"grey":        14, | ||||||
|  | 	"lightgrey":   15, | ||||||
|  | 	"silver":      15, | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var fmtCodes = map[string]string{ | ||||||
|  | 	"bold":      "\x02", | ||||||
|  | 	"b":         "\x02", | ||||||
|  | 	"italic":    "\x1d", | ||||||
|  | 	"i":         "\x1d", | ||||||
|  | 	"reset":     "\x0f", | ||||||
|  | 	"r":         "\x0f", | ||||||
|  | 	"clear":     "\x03", | ||||||
|  | 	"c":         "\x03", // Clears formatting. | ||||||
|  | 	"reverse":   "\x16", | ||||||
|  | 	"underline": "\x1f", | ||||||
|  | 	"ul":        "\x1f", | ||||||
|  | 	"ctcp":      "\x01", // CTCP/ACTION delimiter. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Fmt takes format strings like "{red}" or "{red,blue}" (for background | ||||||
|  | // colors) and turns them into the resulting ASCII format/color codes for IRC. | ||||||
|  | // See format.go for the list of supported format codes allowed. | ||||||
|  | // | ||||||
|  | // For example: | ||||||
|  | // | ||||||
|  | //   client.Message("#channel", Fmt("{red}{b}Hello {red,blue}World{c}")) | ||||||
|  | func Fmt(text string) string { | ||||||
|  | 	var last = -1 | ||||||
|  | 	for i := 0; i < len(text); i++ { | ||||||
|  | 		if text[i] == fmtOpenChar { | ||||||
|  | 			last = i | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if text[i] == fmtCloseChar && last > -1 { | ||||||
|  | 			code := strings.ToLower(text[last+1 : i]) | ||||||
|  |  | ||||||
|  | 			// Check to see if they're passing in a second (background) color | ||||||
|  | 			// as {fgcolor,bgcolor}. | ||||||
|  | 			var secondary string | ||||||
|  | 			if com := strings.Index(code, ","); com > -1 { | ||||||
|  | 				secondary = code[com+1:] | ||||||
|  | 				code = code[:com] | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			var repl string | ||||||
|  |  | ||||||
|  | 			if color, ok := fmtColors[code]; ok { | ||||||
|  | 				repl = fmt.Sprintf("\x03%02d", color) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if repl != "" && secondary != "" { | ||||||
|  | 				if color, ok := fmtColors[secondary]; ok { | ||||||
|  | 					repl += fmt.Sprintf(",%02d", color) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if repl == "" { | ||||||
|  | 				if fmtCode, ok := fmtCodes[code]; ok { | ||||||
|  | 					repl = fmtCode | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			next := len(text[:last]+repl) - 1 | ||||||
|  | 			text = text[:last] + repl + text[i+1:] | ||||||
|  | 			last = -1 | ||||||
|  | 			i = next | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if last > -1 { | ||||||
|  | 			// A-Z, a-z, and "," | ||||||
|  | 			if text[i] != 0x2c && (text[i] <= 0x41 || text[i] >= 0x5a) && (text[i] <= 0x61 || text[i] >= 0x7a) { | ||||||
|  | 				last = -1 | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // TrimFmt strips all "{fmt}" formatting strings from the input text. | ||||||
|  | // See Fmt() for more information. | ||||||
|  | func TrimFmt(text string) string { | ||||||
|  | 	for color := range fmtColors { | ||||||
|  | 		text = strings.Replace(text, "{"+color+"}", "", -1) | ||||||
|  | 	} | ||||||
|  | 	for code := range fmtCodes { | ||||||
|  | 		text = strings.Replace(text, "{"+code+"}", "", -1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // This is really the only fastest way of doing this (marginably better than | ||||||
|  | // actually trying to parse it manually.) | ||||||
|  | var reStripColor = regexp.MustCompile(`\x03([019]?[0-9](,[019]?[0-9])?)?`) | ||||||
|  |  | ||||||
|  | // StripRaw tries to strip all ASCII format codes that are used for IRC. | ||||||
|  | // Primarily, foreground/background colors, and other control bytes like | ||||||
|  | // reset, bold, italic, reverse, etc. This also is done in a specific way | ||||||
|  | // in order to ensure no truncation of other non-irc formatting. | ||||||
|  | func StripRaw(text string) string { | ||||||
|  | 	text = reStripColor.ReplaceAllString(text, "") | ||||||
|  |  | ||||||
|  | 	for _, code := range fmtCodes { | ||||||
|  | 		text = strings.Replace(text, code, "", -1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsValidChannel validates if channel is an RFC complaint channel or not. | ||||||
|  | // | ||||||
|  | // NOTE: If you are using this to validate a channel that contains a channel | ||||||
|  | // ID, (!<channelid>NAME), this only supports the standard 5 character length. | ||||||
|  | // | ||||||
|  | // NOTE: If you do not need to validate against servers that support unicode, | ||||||
|  | // you may want to ensure that all channel chars are within the range of | ||||||
|  | // all ASCII printable chars. This function will NOT do that for | ||||||
|  | // compatibility reasons. | ||||||
|  | // | ||||||
|  | //   channel    =  ( "#" / "+" / ( "!" channelid ) / "&" ) chanstring | ||||||
|  | //                 [ ":" chanstring ] | ||||||
|  | //   chanstring =  0x01-0x07 / 0x08-0x09 / 0x0B-0x0C / 0x0E-0x1F / 0x21-0x2B | ||||||
|  | //   chanstring =  / 0x2D-0x39 / 0x3B-0xFF | ||||||
|  | //                   ; any octet except NUL, BELL, CR, LF, " ", "," and ":" | ||||||
|  | //   channelid  = 5( 0x41-0x5A / digit )   ; 5( A-Z / 0-9 ) | ||||||
|  | func IsValidChannel(channel string) bool { | ||||||
|  | 	if len(channel) <= 1 || len(channel) > 50 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// #, +, !<channelid>, or & | ||||||
|  | 	// Including "*" in the prefix list, as this is commonly used (e.g. ZNC) | ||||||
|  | 	if bytes.IndexByte([]byte{0x21, 0x23, 0x26, 0x2A, 0x2B}, channel[0]) == -1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// !<channelid> -- not very commonly supported, but we'll check it anyway. | ||||||
|  | 	// The ID must be 5 chars. This means min-channel size should be: | ||||||
|  | 	//   1 (prefix) + 5 (id) + 1 (+, channel name) | ||||||
|  | 	// On some networks, this may be extended with ISUPPORT capabilities, | ||||||
|  | 	// however this is extremely uncommon. | ||||||
|  | 	if channel[0] == 0x21 { | ||||||
|  | 		if len(channel) < 7 { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// check for valid ID | ||||||
|  | 		for i := 1; i < 6; i++ { | ||||||
|  | 			if (channel[i] < 0x30 || channel[i] > 0x39) && (channel[i] < 0x41 || channel[i] > 0x5A) { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check for invalid octets here. | ||||||
|  | 	bad := []byte{0x00, 0x07, 0x0D, 0x0A, 0x20, 0x2C, 0x3A} | ||||||
|  | 	for i := 1; i < len(channel); i++ { | ||||||
|  | 		if bytes.IndexByte(bad, channel[i]) != -1 { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsValidNick validates an IRC nickame. Note that this does not validate | ||||||
|  | // IRC nickname length. | ||||||
|  | // | ||||||
|  | //   nickname =  ( letter / special ) *8( letter / digit / special / "-" ) | ||||||
|  | //   letter   =  0x41-0x5A / 0x61-0x7A | ||||||
|  | //   digit    =  0x30-0x39 | ||||||
|  | //   special  =  0x5B-0x60 / 0x7B-0x7D | ||||||
|  | func IsValidNick(nick string) bool { | ||||||
|  | 	if len(nick) <= 0 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	nick = ToRFC1459(nick) | ||||||
|  |  | ||||||
|  | 	// Check the first index. Some characters aren't allowed for the first | ||||||
|  | 	// index of an IRC nickname. | ||||||
|  | 	if nick[0] < 0x41 || nick[0] > 0x7D { | ||||||
|  | 		// a-z, A-Z, and _\[]{}^| | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 1; i < len(nick); i++ { | ||||||
|  | 		if (nick[i] < 0x41 || nick[i] > 0x7D) && (nick[i] < 0x30 || nick[i] > 0x39) && nick[i] != 0x2D { | ||||||
|  | 			// a-z, A-Z, 0-9, -, and _\[]{}^| | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsValidUser validates an IRC ident/username. Note that this does not | ||||||
|  | // validate IRC ident length. | ||||||
|  | // | ||||||
|  | // The validation checks are much like what characters are allowed with an | ||||||
|  | // IRC nickname (see IsValidNick()), however an ident/username can: | ||||||
|  | // | ||||||
|  | // 1. Must either start with alphanumberic char, or "~" then alphanumberic | ||||||
|  | // char. | ||||||
|  | // | ||||||
|  | // 2. Contain a "." (period), for use with "first.last". Though, this may | ||||||
|  | // not be supported on all networks. Some limit this to only a single period. | ||||||
|  | // | ||||||
|  | // Per RFC: | ||||||
|  | //   user =  1*( %x01-09 / %x0B-0C / %x0E-1F / %x21-3F / %x41-FF ) | ||||||
|  | //           ; any octet except NUL, CR, LF, " " and "@" | ||||||
|  | func IsValidUser(name string) bool { | ||||||
|  | 	if len(name) <= 0 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	name = ToRFC1459(name) | ||||||
|  |  | ||||||
|  | 	// "~" is prepended (commonly) if there was no ident server response. | ||||||
|  | 	if name[0] == 0x7E { | ||||||
|  | 		// Means name only contained "~". | ||||||
|  | 		if len(name) < 2 { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		name = name[1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check to see if the first index is alphanumeric. | ||||||
|  | 	if (name[0] < 0x41 || name[0] > 0x4A) && (name[0] < 0x61 || name[0] > 0x7A) && (name[0] < 0x30 || name[0] > 0x39) { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 1; i < len(name); i++ { | ||||||
|  | 		if (name[i] < 0x41 || name[i] > 0x7D) && (name[i] < 0x30 || name[i] > 0x39) && name[i] != 0x2D && name[i] != 0x2E { | ||||||
|  | 			// a-z, A-Z, 0-9, -, and _\[]{}^| | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ToRFC1459 converts a string to the stripped down conversion within RFC | ||||||
|  | // 1459. This will do things like replace an "A" with an "a", "[]" with "{}", | ||||||
|  | // and so forth. Useful to compare two nicknames or channels. | ||||||
|  | func ToRFC1459(input string) (out string) { | ||||||
|  | 	for i := 0; i < len(input); i++ { | ||||||
|  | 		if input[i] >= 65 && input[i] <= 94 { | ||||||
|  | 			out += string(rune(input[i]) + 32) | ||||||
|  | 		} else { | ||||||
|  | 			out += string(input[i]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const globChar = "*" | ||||||
|  |  | ||||||
|  | // Glob will test a string pattern, potentially containing globs, against a | ||||||
|  | // string. The glob character is *. | ||||||
|  | func Glob(input, match string) bool { | ||||||
|  | 	// Empty pattern. | ||||||
|  | 	if match == "" { | ||||||
|  | 		return input == match | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If a glob, match all. | ||||||
|  | 	if match == globChar { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	parts := strings.Split(match, globChar) | ||||||
|  |  | ||||||
|  | 	if len(parts) == 1 { | ||||||
|  | 		// No globs, test for equality. | ||||||
|  | 		return input == match | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	leadingGlob, trailingGlob := strings.HasPrefix(match, globChar), strings.HasSuffix(match, globChar) | ||||||
|  | 	last := len(parts) - 1 | ||||||
|  |  | ||||||
|  | 	// Check prefix first. | ||||||
|  | 	if !leadingGlob && !strings.HasPrefix(input, parts[0]) { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check middle section. | ||||||
|  | 	for i := 1; i < last; i++ { | ||||||
|  | 		if !strings.Contains(input, parts[i]) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Trim already-evaluated text from input during loop over match | ||||||
|  | 		// text. | ||||||
|  | 		idx := strings.Index(input, parts[i]) + len(parts[i]) | ||||||
|  | 		input = input[idx:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check suffix last. | ||||||
|  | 	return trailingGlob || strings.HasSuffix(input, parts[last]) | ||||||
|  | } | ||||||
							
								
								
									
										484
									
								
								vendor/github.com/lrstanley/girc/handler.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										484
									
								
								vendor/github.com/lrstanley/girc/handler.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,484 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"log" | ||||||
|  | 	"math/rand" | ||||||
|  | 	"runtime" | ||||||
|  | 	"runtime/debug" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // RunHandlers manually runs handlers for a given event. | ||||||
|  | func (c *Client) RunHandlers(event *Event) { | ||||||
|  | 	if event == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Log the event. | ||||||
|  | 	c.debug.Print("< " + StripRaw(event.String())) | ||||||
|  | 	if c.Config.Out != nil { | ||||||
|  | 		if pretty, ok := event.Pretty(); ok { | ||||||
|  | 			fmt.Fprintln(c.Config.Out, StripRaw(pretty)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Regular wildcard handlers. | ||||||
|  | 	c.Handlers.exec(ALL_EVENTS, c, event.Copy()) | ||||||
|  |  | ||||||
|  | 	// Then regular handlers. | ||||||
|  | 	c.Handlers.exec(event.Command, c, event.Copy()) | ||||||
|  |  | ||||||
|  | 	// Check if it's a CTCP. | ||||||
|  | 	if ctcp := decodeCTCP(event.Copy()); ctcp != nil { | ||||||
|  | 		// Execute it. | ||||||
|  | 		c.CTCP.call(c, ctcp) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handler is lower level implementation of a handler. See | ||||||
|  | // Caller.AddHandler() | ||||||
|  | type Handler interface { | ||||||
|  | 	Execute(*Client, Event) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // HandlerFunc is a type that represents the function necessary to | ||||||
|  | // implement Handler. | ||||||
|  | type HandlerFunc func(client *Client, event Event) | ||||||
|  |  | ||||||
|  | // Execute calls the HandlerFunc with the sender and irc message. | ||||||
|  | func (f HandlerFunc) Execute(client *Client, event Event) { | ||||||
|  | 	f(client, event) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Caller manages internal and external (user facing) handlers. | ||||||
|  | type Caller struct { | ||||||
|  | 	// mu is the mutex that should be used when accessing handlers. | ||||||
|  | 	mu sync.RWMutex | ||||||
|  |  | ||||||
|  | 	// external/internal keys are of structure: | ||||||
|  | 	//   map[COMMAND][CUID]Handler | ||||||
|  | 	// Also of note: "COMMAND" should always be uppercase for normalization. | ||||||
|  |  | ||||||
|  | 	// external is a map of user facing handlers. | ||||||
|  | 	external map[string]map[string]Handler | ||||||
|  | 	// internal is a map of internally used handlers for the client. | ||||||
|  | 	internal map[string]map[string]Handler | ||||||
|  | 	// debug is the clients logger used for debugging. | ||||||
|  | 	debug *log.Logger | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // newCaller creates and initializes a new handler. | ||||||
|  | func newCaller(debugOut *log.Logger) *Caller { | ||||||
|  | 	c := &Caller{ | ||||||
|  | 		external: map[string]map[string]Handler{}, | ||||||
|  | 		internal: map[string]map[string]Handler{}, | ||||||
|  | 		debug:    debugOut, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the total amount of user-entered registered handlers. | ||||||
|  | func (c *Caller) Len() int { | ||||||
|  | 	var total int | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	for command := range c.external { | ||||||
|  | 		total += len(c.external[command]) | ||||||
|  | 	} | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return total | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Count is much like Caller.Len(), however it counts the number of | ||||||
|  | // registered handlers for a given command. | ||||||
|  | func (c *Caller) Count(cmd string) int { | ||||||
|  | 	var total int | ||||||
|  |  | ||||||
|  | 	cmd = strings.ToUpper(cmd) | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	for command := range c.external { | ||||||
|  | 		if command == cmd { | ||||||
|  | 			total += len(c.external[command]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return total | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *Caller) String() string { | ||||||
|  | 	var total int | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	for cmd := range c.internal { | ||||||
|  | 		total += len(c.internal[cmd]) | ||||||
|  | 	} | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return fmt.Sprintf("<Caller external:%d internal:%d>", c.Len(), total) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" | ||||||
|  |  | ||||||
|  | // cuid generates a unique UID string for each handler for ease of removal. | ||||||
|  | func (c *Caller) cuid(cmd string, n int) (cuid, uid string) { | ||||||
|  | 	b := make([]byte, n) | ||||||
|  |  | ||||||
|  | 	for i := range b { | ||||||
|  | 		b[i] = letterBytes[rand.Int63()%int64(len(letterBytes))] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cmd + ":" + string(b), string(b) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // cuidToID allows easy mapping between a generated cuid and the caller | ||||||
|  | // external/internal handler maps. | ||||||
|  | func (c *Caller) cuidToID(input string) (cmd, uid string) { | ||||||
|  | 	i := strings.IndexByte(input, 0x3A) | ||||||
|  | 	if i < 0 { | ||||||
|  | 		return "", "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return input[:i], input[i+1:] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type execStack struct { | ||||||
|  | 	Handler | ||||||
|  | 	cuid string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // exec executes all handlers pertaining to specified event. Internal first, | ||||||
|  | // then external. | ||||||
|  | // | ||||||
|  | // Please note that there is no specific order/priority for which the | ||||||
|  | // handler types themselves or the handlers are executed. | ||||||
|  | func (c *Caller) exec(command string, client *Client, event *Event) { | ||||||
|  | 	// Build a stack of handlers which can be executed concurrently. | ||||||
|  | 	var stack []execStack | ||||||
|  |  | ||||||
|  | 	c.mu.RLock() | ||||||
|  | 	// Get internal handlers first. | ||||||
|  | 	if _, ok := c.internal[command]; ok { | ||||||
|  | 		for cuid := range c.internal[command] { | ||||||
|  | 			stack = append(stack, execStack{c.internal[command][cuid], cuid}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Aaand then external handlers. | ||||||
|  | 	if _, ok := c.external[command]; ok { | ||||||
|  | 		for cuid := range c.external[command] { | ||||||
|  | 			stack = append(stack, execStack{c.external[command][cuid], cuid}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	// Run all handlers concurrently across the same event. This should | ||||||
|  | 	// still help prevent mis-ordered events, while speeding up the | ||||||
|  | 	// execution speed. | ||||||
|  | 	var wg sync.WaitGroup | ||||||
|  | 	wg.Add(len(stack)) | ||||||
|  | 	for i := 0; i < len(stack); i++ { | ||||||
|  | 		go func(index int) { | ||||||
|  | 			c.debug.Printf("executing handler %s for event %s (%d of %d)", stack[index].cuid, command, index+1, len(stack)) | ||||||
|  | 			start := time.Now() | ||||||
|  |  | ||||||
|  | 			// If they want to catch any panics, add to defer stack. | ||||||
|  | 			if client.Config.RecoverFunc != nil { | ||||||
|  | 				defer recoverHandlerPanic(client, event, stack[index].cuid, 3) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			stack[index].Execute(client, *event) | ||||||
|  |  | ||||||
|  | 			c.debug.Printf("execution of %s took %s (%d of %d)", stack[index].cuid, time.Since(start), index+1, len(stack)) | ||||||
|  | 			wg.Done() | ||||||
|  | 		}(i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Wait for all of the handlers to complete. Not doing this may cause | ||||||
|  | 	// new events from becoming ahead of older handlers. | ||||||
|  | 	wg.Wait() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ClearAll clears all external handlers currently setup within the client. | ||||||
|  | // This ignores internal handlers. | ||||||
|  | func (c *Caller) ClearAll() { | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	c.external = map[string]map[string]Handler{} | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	c.debug.Print("cleared all external handlers") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // clearInternal clears all internal handlers currently setup within the | ||||||
|  | // client. | ||||||
|  | func (c *Caller) clearInternal() { | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	c.internal = map[string]map[string]Handler{} | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	c.debug.Print("cleared all internal handlers") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Clear clears all of the handlers for the given event. | ||||||
|  | // This ignores internal handlers. | ||||||
|  | func (c *Caller) Clear(cmd string) { | ||||||
|  | 	cmd = strings.ToUpper(cmd) | ||||||
|  |  | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	if _, ok := c.external[cmd]; ok { | ||||||
|  | 		delete(c.external, cmd) | ||||||
|  | 	} | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	c.debug.Printf("cleared external handlers for %s", cmd) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove removes the handler with cuid from the handler stack. success | ||||||
|  | // indicates that it existed, and has been removed. If not success, it | ||||||
|  | // wasn't a registered handler. | ||||||
|  | func (c *Caller) Remove(cuid string) (success bool) { | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	success = c.remove(cuid) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	return success | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // remove is much like Remove, however is NOT concurrency safe. Lock Caller.mu | ||||||
|  | // on your own. | ||||||
|  | func (c *Caller) remove(cuid string) (success bool) { | ||||||
|  | 	cmd, uid := c.cuidToID(cuid) | ||||||
|  | 	if len(cmd) == 0 || len(uid) == 0 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the irc command/event has any handlers on it. | ||||||
|  | 	if _, ok := c.external[cmd]; !ok { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check to see if it's actually a registered handler. | ||||||
|  | 	if _, ok := c.external[cmd][uid]; !ok { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	delete(c.external[cmd], uid) | ||||||
|  | 	c.debug.Printf("removed handler %s", cuid) | ||||||
|  |  | ||||||
|  | 	// Assume success. | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // sregister is much like Caller.register(), except that it safely locks | ||||||
|  | // the Caller mutex. | ||||||
|  | func (c *Caller) sregister(internal bool, cmd string, handler Handler) (cuid string) { | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	cuid = c.register(internal, cmd, handler) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	return cuid | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // register will register a handler in the internal tracker. Unsafe (you | ||||||
|  | // must lock c.mu yourself!) | ||||||
|  | func (c *Caller) register(internal bool, cmd string, handler Handler) (cuid string) { | ||||||
|  | 	var uid string | ||||||
|  |  | ||||||
|  | 	cmd = strings.ToUpper(cmd) | ||||||
|  |  | ||||||
|  | 	if internal { | ||||||
|  | 		if _, ok := c.internal[cmd]; !ok { | ||||||
|  | 			c.internal[cmd] = map[string]Handler{} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		cuid, uid = c.cuid(cmd, 20) | ||||||
|  | 		c.internal[cmd][uid] = handler | ||||||
|  | 	} else { | ||||||
|  | 		if _, ok := c.external[cmd]; !ok { | ||||||
|  | 			c.external[cmd] = map[string]Handler{} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		cuid, uid = c.cuid(cmd, 20) | ||||||
|  | 		c.external[cmd][uid] = handler | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, file, line, _ := runtime.Caller(3) | ||||||
|  |  | ||||||
|  | 	c.debug.Printf("registering handler for %q with cuid %q (internal: %t) from: %s:%d", cmd, cuid, internal, file, line) | ||||||
|  |  | ||||||
|  | 	return cuid | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddHandler registers a handler (matching the handler interface) for the | ||||||
|  | // given event. cuid is the handler uid which can be used to remove the | ||||||
|  | // handler with Caller.Remove(). | ||||||
|  | func (c *Caller) AddHandler(cmd string, handler Handler) (cuid string) { | ||||||
|  | 	return c.sregister(false, cmd, handler) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add registers the handler function for the given event. cuid is the | ||||||
|  | // handler uid which can be used to remove the handler with Caller.Remove(). | ||||||
|  | func (c *Caller) Add(cmd string, handler func(client *Client, event Event)) (cuid string) { | ||||||
|  | 	return c.sregister(false, cmd, HandlerFunc(handler)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddBg registers the handler function for the given event and executes it | ||||||
|  | // in a go-routine. cuid is the handler uid which can be used to remove the | ||||||
|  | // handler with Caller.Remove(). | ||||||
|  | func (c *Caller) AddBg(cmd string, handler func(client *Client, event Event)) (cuid string) { | ||||||
|  | 	return c.sregister(false, cmd, HandlerFunc(func(client *Client, event Event) { | ||||||
|  | 		// Setting up background-based handlers this way allows us to get | ||||||
|  | 		// clean call stacks for use with panic recovery. | ||||||
|  | 		go func() { | ||||||
|  | 			// If they want to catch any panics, add to defer stack. | ||||||
|  | 			if client.Config.RecoverFunc != nil { | ||||||
|  | 				defer recoverHandlerPanic(client, &event, "goroutine", 3) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			handler(client, event) | ||||||
|  | 		}() | ||||||
|  | 	})) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddTmp adds a "temporary" handler, which is good for one-time or few-time | ||||||
|  | // uses. This supports a deadline and/or manual removal, as this differs | ||||||
|  | // much from how normal handlers work. An example of a good use for this | ||||||
|  | // would be to capture the entire output of a multi-response query to the | ||||||
|  | // server. (e.g. LIST, WHOIS, etc) | ||||||
|  | // | ||||||
|  | // The supplied handler is able to return a boolean, which if true, will | ||||||
|  | // remove the handler from the handler stack. | ||||||
|  | // | ||||||
|  | // Additionally, AddTmp has a useful option, deadline. When set to greater | ||||||
|  | // than 0, deadline will be the amount of time that passes before the handler | ||||||
|  | // is removed from the stack, regardless if the handler returns true or not. | ||||||
|  | // This is useful in that it ensures that the handler is cleaned up if the | ||||||
|  | // server does not respond appropriately, or takes too long to respond. | ||||||
|  | // | ||||||
|  | // Note that handlers supplied with AddTmp are executed in a goroutine to | ||||||
|  | // ensure that they are not blocking other handlers. Additionally, use cuid | ||||||
|  | // with Caller.Remove() to prematurely remove the handler from the stack, | ||||||
|  | // bypassing the timeout or waiting for the handler to return that it wants | ||||||
|  | // to be removed from the stack. | ||||||
|  | func (c *Caller) AddTmp(cmd string, deadline time.Duration, handler func(client *Client, event Event) bool) (cuid string, done chan struct{}) { | ||||||
|  | 	var uid string | ||||||
|  | 	cuid, uid = c.cuid(cmd, 20) | ||||||
|  |  | ||||||
|  | 	done = make(chan struct{}) | ||||||
|  |  | ||||||
|  | 	c.mu.Lock() | ||||||
|  | 	if _, ok := c.external[cmd]; !ok { | ||||||
|  | 		c.external[cmd] = map[string]Handler{} | ||||||
|  | 	} | ||||||
|  | 	c.external[cmd][uid] = HandlerFunc(func(client *Client, event Event) { | ||||||
|  | 		// Setting up background-based handlers this way allows us to get | ||||||
|  | 		// clean call stacks for use with panic recovery. | ||||||
|  | 		go func() { | ||||||
|  | 			// If they want to catch any panics, add to defer stack. | ||||||
|  | 			if client.Config.RecoverFunc != nil { | ||||||
|  | 				defer recoverHandlerPanic(client, &event, "tmp-goroutine", 3) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			remove := handler(client, event) | ||||||
|  | 			if remove { | ||||||
|  | 				if ok := c.Remove(cuid); ok { | ||||||
|  | 					close(done) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  | 	}) | ||||||
|  | 	c.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	if deadline > 0 { | ||||||
|  | 		go func() { | ||||||
|  | 			<-time.After(deadline) | ||||||
|  | 			if ok := c.Remove(cuid); ok { | ||||||
|  | 				close(done) | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cuid, done | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // recoverHandlerPanic is used to catch all handler panics, and re-route | ||||||
|  | // them if necessary. | ||||||
|  | func recoverHandlerPanic(client *Client, event *Event, id string, skip int) { | ||||||
|  | 	perr := recover() | ||||||
|  | 	if perr == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var file string | ||||||
|  | 	var line int | ||||||
|  | 	var ok bool | ||||||
|  |  | ||||||
|  | 	_, file, line, ok = runtime.Caller(skip) | ||||||
|  |  | ||||||
|  | 	err := &HandlerError{ | ||||||
|  | 		Event:  *event, | ||||||
|  | 		ID:     id, | ||||||
|  | 		File:   file, | ||||||
|  | 		Line:   line, | ||||||
|  | 		Panic:  perr, | ||||||
|  | 		Stack:  debug.Stack(), | ||||||
|  | 		callOk: ok, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	client.Config.RecoverFunc(client, err) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // HandlerError is the error returned when a panic is intentionally recovered | ||||||
|  | // from. It contains useful information like the handler identifier (if | ||||||
|  | // applicable), filename, line in file where panic occurred, the call | ||||||
|  | // trace, and original event. | ||||||
|  | type HandlerError struct { | ||||||
|  | 	Event  Event       // Event is the event that caused the error. | ||||||
|  | 	ID     string      // ID is the CUID of the handler. | ||||||
|  | 	File   string      // File is the file from where the panic originated. | ||||||
|  | 	Line   int         // Line number where panic originated. | ||||||
|  | 	Panic  interface{} // Panic is the error that was passed to panic(). | ||||||
|  | 	Stack  []byte      // Stack is the call stack. Note you may have to skip 1 or 2 due to debug functions. | ||||||
|  | 	callOk bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Error returns a prettified version of HandlerError, containing ID, file, | ||||||
|  | // line, and basic error string. | ||||||
|  | func (e *HandlerError) Error() string { | ||||||
|  | 	if e.callOk { | ||||||
|  | 		return fmt.Sprintf("panic during handler [%s] execution in %s:%d: %s", e.ID, e.File, e.Line, e.Panic) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return fmt.Sprintf("panic during handler [%s] execution in unknown: %s", e.ID, e.Panic) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns the error that panic returned, as well as the entire call | ||||||
|  | // trace of where it originated. | ||||||
|  | func (e *HandlerError) String() string { | ||||||
|  | 	return fmt.Sprintf("panic: %s\n\n%s", e.Panic, string(e.Stack)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DefaultRecoverHandler can be used with Config.RecoverFunc as a default | ||||||
|  | // catch-all for panics. This will log the error, and the call trace to the | ||||||
|  | // debug log (see Config.Debug), or os.Stdout if Config.Debug is unset. | ||||||
|  | func DefaultRecoverHandler(client *Client, err *HandlerError) { | ||||||
|  | 	if client.Config.Debug == nil { | ||||||
|  | 		fmt.Println(err.Error()) | ||||||
|  | 		fmt.Println(err.String()) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	client.debug.Println(err.Error()) | ||||||
|  | 	client.debug.Println(err.String()) | ||||||
|  | } | ||||||
							
								
								
									
										550
									
								
								vendor/github.com/lrstanley/girc/modes.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										550
									
								
								vendor/github.com/lrstanley/girc/modes.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,550 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // CMode represents a single step of a given mode change. | ||||||
|  | type CMode struct { | ||||||
|  | 	add     bool   // if it's a +, or -. | ||||||
|  | 	name    byte   // character representation of the given mode. | ||||||
|  | 	setting bool   // if it's a setting (should be stored) or temporary (op/voice/etc). | ||||||
|  | 	args    string // arguments to the mode, if arguments are supported. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Short returns a short representation of a mode without arguments. E.g. "+a", | ||||||
|  | // or "-b". | ||||||
|  | func (c *CMode) Short() string { | ||||||
|  | 	var status string | ||||||
|  | 	if c.add { | ||||||
|  | 		status = "+" | ||||||
|  | 	} else { | ||||||
|  | 		status = "-" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return status + string(c.name) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a string representation of a mode, including optional | ||||||
|  | // arguments. E.g. "+b user*!ident@host.*.com" | ||||||
|  | func (c *CMode) String() string { | ||||||
|  | 	if len(c.args) == 0 { | ||||||
|  | 		return c.Short() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c.Short() + " " + c.args | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // CModes is a representation of a set of modes. This may be the given state | ||||||
|  | // of a channel/user, or the given state changes to a given channel/user. | ||||||
|  | type CModes struct { | ||||||
|  | 	raw           string // raw supported modes. | ||||||
|  | 	modesListArgs string // modes that add/remove users from lists and support args. | ||||||
|  | 	modesArgs     string // modes that support args. | ||||||
|  | 	modesSetArgs  string // modes that support args ONLY when set. | ||||||
|  | 	modesNoArgs   string // modes that do not support args. | ||||||
|  |  | ||||||
|  | 	prefixes string  // user permission prefixes. these aren't a CMode.setting. | ||||||
|  | 	modes    []CMode // the list of modes for this given state. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a deep copy of CModes. | ||||||
|  | func (c *CModes) Copy() (nc CModes) { | ||||||
|  | 	nc = CModes{} | ||||||
|  | 	nc = *c | ||||||
|  |  | ||||||
|  | 	nc.modes = make([]CMode, len(c.modes)) | ||||||
|  |  | ||||||
|  | 	// Copy modes. | ||||||
|  | 	for i := 0; i < len(c.modes); i++ { | ||||||
|  | 		nc.modes[i] = c.modes[i] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nc | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a complete set of modes for this given state (change?). For | ||||||
|  | // example, "+a-b+cde some-arg". | ||||||
|  | func (c *CModes) String() string { | ||||||
|  | 	var out string | ||||||
|  | 	var args string | ||||||
|  |  | ||||||
|  | 	if len(c.modes) > 0 { | ||||||
|  | 		out += "+" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(c.modes); i++ { | ||||||
|  | 		out += string(c.modes[i].name) | ||||||
|  |  | ||||||
|  | 		if len(c.modes[i].args) > 0 { | ||||||
|  | 			args += " " + c.modes[i].args | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out + args | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // HasMode checks if the CModes state has a given mode. E.g. "m", or "I". | ||||||
|  | func (c *CModes) HasMode(mode string) bool { | ||||||
|  | 	for i := 0; i < len(c.modes); i++ { | ||||||
|  | 		if string(c.modes[i].name) == mode { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get returns the arguments for a given mode within this session, if it | ||||||
|  | // supports args. | ||||||
|  | func (c *CModes) Get(mode string) (args string, ok bool) { | ||||||
|  | 	for i := 0; i < len(c.modes); i++ { | ||||||
|  | 		if string(c.modes[i].name) == mode { | ||||||
|  | 			if len(c.modes[i].args) == 0 { | ||||||
|  | 				return "", false | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			return c.modes[i].args, true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return "", false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // hasArg checks to see if the mode supports arguments. What ones support this?: | ||||||
|  | //   A = Mode that adds or removes a nick or address to a list. Always has a parameter. | ||||||
|  | //   B = Mode that changes a setting and always has a parameter. | ||||||
|  | //   C = Mode that changes a setting and only has a parameter when set. | ||||||
|  | //   D = Mode that changes a setting and never has a parameter. | ||||||
|  | //   Note: Modes of type A return the list when there is no parameter present. | ||||||
|  | //   Note: Some clients assumes that any mode not listed is of type D. | ||||||
|  | //   Note: Modes in PREFIX are not listed but could be considered type B. | ||||||
|  | func (c *CModes) hasArg(set bool, mode byte) (hasArgs, isSetting bool) { | ||||||
|  | 	if len(c.raw) < 1 { | ||||||
|  | 		return false, true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if strings.IndexByte(c.modesListArgs, mode) > -1 { | ||||||
|  | 		return true, false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if strings.IndexByte(c.modesArgs, mode) > -1 { | ||||||
|  | 		return true, true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if strings.IndexByte(c.modesSetArgs, mode) > -1 { | ||||||
|  | 		if set { | ||||||
|  | 			return true, true | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return false, true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if strings.IndexByte(c.prefixes, mode) > -1 { | ||||||
|  | 		return true, false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false, true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Apply merges two state changes, or one state change into a state of modes. | ||||||
|  | // For example, the latter would mean applying an incoming MODE with the modes | ||||||
|  | // stored for a channel. | ||||||
|  | func (c *CModes) Apply(modes []CMode) { | ||||||
|  | 	var new []CMode | ||||||
|  |  | ||||||
|  | 	for j := 0; j < len(c.modes); j++ { | ||||||
|  | 		isin := false | ||||||
|  | 		for i := 0; i < len(modes); i++ { | ||||||
|  | 			if !modes[i].setting { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if c.modes[j].name == modes[i].name && modes[i].add { | ||||||
|  | 				new = append(new, modes[i]) | ||||||
|  | 				isin = true | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if !isin { | ||||||
|  | 			new = append(new, c.modes[j]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(modes); i++ { | ||||||
|  | 		if !modes[i].setting || !modes[i].add { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		isin := false | ||||||
|  | 		for j := 0; j < len(new); j++ { | ||||||
|  | 			if modes[i].name == new[j].name { | ||||||
|  | 				isin = true | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if !isin { | ||||||
|  | 			new = append(new, modes[i]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.modes = new | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Parse parses a set of flags and args, returning the necessary list of | ||||||
|  | // mappings for the mode flags. | ||||||
|  | func (c *CModes) Parse(flags string, args []string) (out []CMode) { | ||||||
|  | 	// add is the mode state we're currently in. Adding, or removing modes. | ||||||
|  | 	add := true | ||||||
|  | 	var argCount int | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(flags); i++ { | ||||||
|  | 		if flags[i] == 0x2B { | ||||||
|  | 			add = true | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if flags[i] == 0x2D { | ||||||
|  | 			add = false | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		mode := CMode{ | ||||||
|  | 			name: flags[i], | ||||||
|  | 			add:  add, | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		hasArgs, isSetting := c.hasArg(add, flags[i]) | ||||||
|  | 		if hasArgs && len(args) >= argCount+1 { | ||||||
|  | 			mode.args = args[argCount] | ||||||
|  | 			argCount++ | ||||||
|  | 		} | ||||||
|  | 		mode.setting = isSetting | ||||||
|  |  | ||||||
|  | 		out = append(out, mode) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewCModes returns a new CModes reference. channelModes and userPrefixes | ||||||
|  | // would be something you see from the server's "CHANMODES" and "PREFIX" | ||||||
|  | // ISUPPORT capability messages (alternatively, fall back to the standard) | ||||||
|  | // DefaultPrefixes and ModeDefaults. | ||||||
|  | func NewCModes(channelModes, userPrefixes string) CModes { | ||||||
|  | 	split := strings.SplitN(channelModes, ",", 4) | ||||||
|  | 	if len(split) != 4 { | ||||||
|  | 		for i := len(split); i < 4; i++ { | ||||||
|  | 			split = append(split, "") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return CModes{ | ||||||
|  | 		raw:           channelModes, | ||||||
|  | 		modesListArgs: split[0], | ||||||
|  | 		modesArgs:     split[1], | ||||||
|  | 		modesSetArgs:  split[2], | ||||||
|  | 		modesNoArgs:   split[3], | ||||||
|  |  | ||||||
|  | 		prefixes: userPrefixes, | ||||||
|  | 		modes:    []CMode{}, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsValidChannelMode validates a channel mode (CHANMODES). | ||||||
|  | func IsValidChannelMode(raw string) bool { | ||||||
|  | 	if len(raw) < 1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(raw); i++ { | ||||||
|  | 		// Allowed are: ",", A-Z and a-z. | ||||||
|  | 		if raw[i] != 0x2C && (raw[i] < 0x41 || raw[i] > 0x5A) && (raw[i] < 0x61 || raw[i] > 0x7A) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // isValidUserPrefix validates a list of ISUPPORT-style user prefixes (PREFIX). | ||||||
|  | func isValidUserPrefix(raw string) bool { | ||||||
|  | 	if len(raw) < 1 { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if raw[0] != 0x28 { // (. | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var keys, rep int | ||||||
|  | 	var passedKeys bool | ||||||
|  |  | ||||||
|  | 	// Skip the first one as we know it's (. | ||||||
|  | 	for i := 1; i < len(raw); i++ { | ||||||
|  | 		if raw[i] == 0x29 { // ). | ||||||
|  | 			passedKeys = true | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if passedKeys { | ||||||
|  | 			rep++ | ||||||
|  | 		} else { | ||||||
|  | 			keys++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return keys == rep | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // parsePrefixes parses the mode character mappings from the symbols of a | ||||||
|  | // ISUPPORT-style user prefixes list (PREFIX). | ||||||
|  | func parsePrefixes(raw string) (modes, prefixes string) { | ||||||
|  | 	if !isValidUserPrefix(raw) { | ||||||
|  | 		return modes, prefixes | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	i := strings.Index(raw, ")") | ||||||
|  | 	if i < 1 { | ||||||
|  | 		return modes, prefixes | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return raw[1:i], raw[i+1:] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleMODE handles incoming MODE messages, and updates the tracking | ||||||
|  | // information for each channel, as well as if any of the modes affect user | ||||||
|  | // permissions. | ||||||
|  | func handleMODE(c *Client, e Event) { | ||||||
|  | 	// Check if it's a RPL_CHANNELMODEIS. | ||||||
|  | 	if e.Command == RPL_CHANNELMODEIS && len(e.Params) > 2 { | ||||||
|  | 		// RPL_CHANNELMODEIS sends the user as the first param, skip it. | ||||||
|  | 		e.Params = e.Params[1:] | ||||||
|  | 	} | ||||||
|  | 	// Should be at least MODE <target> <flags>, to be useful. As well, only | ||||||
|  | 	// tracking channel modes at the moment. | ||||||
|  | 	if len(e.Params) < 2 || !IsValidChannel(e.Params[0]) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	channel := c.state.lookupChannel(e.Params[0]) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		c.state.RUnlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	flags := e.Params[1] | ||||||
|  | 	var args []string | ||||||
|  | 	if len(e.Params) > 2 { | ||||||
|  | 		args = append(args, e.Params[2:]...) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	modes := channel.Modes.Parse(flags, args) | ||||||
|  | 	channel.Modes.Apply(modes) | ||||||
|  |  | ||||||
|  | 	// Loop through and update users modes as necessary. | ||||||
|  | 	for i := 0; i < len(modes); i++ { | ||||||
|  | 		if modes[i].setting || len(modes[i].args) == 0 { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		user := c.state.lookupUser(modes[i].args) | ||||||
|  | 		if user != nil { | ||||||
|  | 			perms, _ := user.Perms.Lookup(channel.Name) | ||||||
|  | 			perms.setFromMode(modes[i]) | ||||||
|  | 			user.Perms.set(channel.Name, perms) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c.state.RUnlock() | ||||||
|  | 	c.state.notify(c, UPDATE_STATE) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // chanModes returns the ISUPPORT list of server-supported channel modes, | ||||||
|  | // alternatively falling back to ModeDefaults. | ||||||
|  | func (s *state) chanModes() string { | ||||||
|  | 	if modes, ok := s.serverOptions["CHANMODES"]; ok && IsValidChannelMode(modes) { | ||||||
|  | 		return modes | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ModeDefaults | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // userPrefixes returns the ISUPPORT list of server-supported user prefixes. | ||||||
|  | // This includes mode characters, as well as user prefix symbols. Falls back | ||||||
|  | // to DefaultPrefixes if not server-supported. | ||||||
|  | func (s *state) userPrefixes() string { | ||||||
|  | 	if prefix, ok := s.serverOptions["PREFIX"]; ok && isValidUserPrefix(prefix) { | ||||||
|  | 		return prefix | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return DefaultPrefixes | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UserPerms contains all of the permissions for each channel the user is | ||||||
|  | // in. | ||||||
|  | type UserPerms struct { | ||||||
|  | 	mu       sync.RWMutex | ||||||
|  | 	channels map[string]Perms | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a deep copy of the channel permissions. | ||||||
|  | func (p *UserPerms) Copy() (perms *UserPerms) { | ||||||
|  | 	np := &UserPerms{ | ||||||
|  | 		channels: make(map[string]Perms), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	p.mu.RLock() | ||||||
|  | 	for key := range p.channels { | ||||||
|  | 		np.channels[key] = p.channels[key] | ||||||
|  | 	} | ||||||
|  | 	p.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return np | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MarshalJSON implements json.Marshaler. | ||||||
|  | func (p *UserPerms) MarshalJSON() ([]byte, error) { | ||||||
|  | 	p.mu.Lock() | ||||||
|  | 	out, err := json.Marshal(&p.channels) | ||||||
|  | 	p.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	return out, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Lookup looks up the users permissions for a given channel. ok is false | ||||||
|  | // if the user is not in the given channel. | ||||||
|  | func (p *UserPerms) Lookup(channel string) (perms Perms, ok bool) { | ||||||
|  | 	p.mu.RLock() | ||||||
|  | 	perms, ok = p.channels[ToRFC1459(channel)] | ||||||
|  | 	p.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	return perms, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (p *UserPerms) set(channel string, perms Perms) { | ||||||
|  | 	p.mu.Lock() | ||||||
|  | 	p.channels[ToRFC1459(channel)] = perms | ||||||
|  | 	p.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (p *UserPerms) remove(channel string) { | ||||||
|  | 	p.mu.Lock() | ||||||
|  | 	delete(p.channels, ToRFC1459(channel)) | ||||||
|  | 	p.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Perms contains all channel-based user permissions. The minimum op, and | ||||||
|  | // voice should be supported on all networks. This also supports non-rfc | ||||||
|  | // Owner, Admin, and HalfOp, if the network has support for it. | ||||||
|  | type Perms struct { | ||||||
|  | 	// Owner (non-rfc) indicates that the user has full permissions to the | ||||||
|  | 	// channel. More than one user can have owner permission. | ||||||
|  | 	Owner bool `json:"owner"` | ||||||
|  | 	// Admin (non-rfc) is commonly given to users that are trusted enough | ||||||
|  | 	// to manage channel permissions, as well as higher level service settings. | ||||||
|  | 	Admin bool `json:"admin"` | ||||||
|  | 	// Op is commonly given to trusted users who can manage a given channel | ||||||
|  | 	// by kicking, and banning users. | ||||||
|  | 	Op bool `json:"op"` | ||||||
|  | 	// HalfOp (non-rfc) is commonly used to give users permissions like the | ||||||
|  | 	// ability to kick, without giving them greater abilities to ban all users. | ||||||
|  | 	HalfOp bool `json:"half_op"` | ||||||
|  | 	// Voice indicates the user has voice permissions, commonly given to known | ||||||
|  | 	// users, with very light trust, or to indicate a user is active. | ||||||
|  | 	Voice bool `json:"voice"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsAdmin indicates that the user has banning abilities, and are likely a | ||||||
|  | // very trustable user (e.g. op+). | ||||||
|  | func (m Perms) IsAdmin() bool { | ||||||
|  | 	if m.Owner || m.Admin || m.Op { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsTrusted indicates that the user at least has modes set upon them, higher | ||||||
|  | // than a regular joining user. | ||||||
|  | func (m Perms) IsTrusted() bool { | ||||||
|  | 	if m.IsAdmin() || m.HalfOp || m.Voice { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // reset resets the modes of a user. | ||||||
|  | func (m *Perms) reset() { | ||||||
|  | 	m.Owner = false | ||||||
|  | 	m.Admin = false | ||||||
|  | 	m.Op = false | ||||||
|  | 	m.HalfOp = false | ||||||
|  | 	m.Voice = false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // set translates raw prefix characters into proper permissions. Only | ||||||
|  | // use this function when you have a session lock. | ||||||
|  | func (m *Perms) set(prefix string, append bool) { | ||||||
|  | 	if !append { | ||||||
|  | 		m.reset() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(prefix); i++ { | ||||||
|  | 		switch string(prefix[i]) { | ||||||
|  | 		case OwnerPrefix: | ||||||
|  | 			m.Owner = true | ||||||
|  | 		case AdminPrefix: | ||||||
|  | 			m.Admin = true | ||||||
|  | 		case OperatorPrefix: | ||||||
|  | 			m.Op = true | ||||||
|  | 		case HalfOperatorPrefix: | ||||||
|  | 			m.HalfOp = true | ||||||
|  | 		case VoicePrefix: | ||||||
|  | 			m.Voice = true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // setFromMode sets user-permissions based on channel user mode chars. E.g. | ||||||
|  | // "o" being oper, "v" being voice, etc. | ||||||
|  | func (m *Perms) setFromMode(mode CMode) { | ||||||
|  | 	switch string(mode.name) { | ||||||
|  | 	case ModeOwner: | ||||||
|  | 		m.Owner = mode.add | ||||||
|  | 	case ModeAdmin: | ||||||
|  | 		m.Admin = mode.add | ||||||
|  | 	case ModeOperator: | ||||||
|  | 		m.Op = mode.add | ||||||
|  | 	case ModeHalfOperator: | ||||||
|  | 		m.HalfOp = mode.add | ||||||
|  | 	case ModeVoice: | ||||||
|  | 		m.Voice = mode.add | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // parseUserPrefix parses a raw mode line, like "@user" or "@+user". | ||||||
|  | func parseUserPrefix(raw string) (modes, nick string, success bool) { | ||||||
|  | 	for i := 0; i < len(raw); i++ { | ||||||
|  | 		char := string(raw[i]) | ||||||
|  |  | ||||||
|  | 		if char == OwnerPrefix || char == AdminPrefix || char == HalfOperatorPrefix || | ||||||
|  | 			char == OperatorPrefix || char == VoicePrefix { | ||||||
|  | 			modes += char | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Assume we've gotten to the nickname part. | ||||||
|  | 		return modes, raw[i:], true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
							
								
								
									
										489
									
								
								vendor/github.com/lrstanley/girc/state.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										489
									
								
								vendor/github.com/lrstanley/girc/state.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,489 @@ | |||||||
|  | // Copyright (c) Liam Stanley <me@liamstanley.io>. All rights reserved. Use | ||||||
|  | // of this source code is governed by the MIT license that can be found in | ||||||
|  | // the LICENSE file. | ||||||
|  |  | ||||||
|  | package girc | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"sort" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // state represents the actively-changing variables within the client | ||||||
|  | // runtime. Note that everything within the state should be guarded by the | ||||||
|  | // embedded sync.RWMutex. | ||||||
|  | type state struct { | ||||||
|  | 	sync.RWMutex | ||||||
|  | 	// nick, ident, and host are the internal trackers for our user. | ||||||
|  | 	nick, ident, host string | ||||||
|  | 	// channels represents all channels we're active in. | ||||||
|  | 	channels map[string]*Channel | ||||||
|  | 	// users represents all of users that we're tracking. | ||||||
|  | 	users map[string]*User | ||||||
|  | 	// enabledCap are the capabilities which are enabled for this connection. | ||||||
|  | 	enabledCap []string | ||||||
|  | 	// tmpCap are the capabilties which we share with the server during the | ||||||
|  | 	// last capability check. These will get sent once we have received the | ||||||
|  | 	// last capability list command from the server. | ||||||
|  | 	tmpCap []string | ||||||
|  | 	// serverOptions are the standard capabilities and configurations | ||||||
|  | 	// supported by the server at connection time. This also includes | ||||||
|  | 	// RPL_ISUPPORT entries. | ||||||
|  | 	serverOptions map[string]string | ||||||
|  | 	// motd is the servers message of the day. | ||||||
|  | 	motd string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // notify sends state change notifications so users can update their refs | ||||||
|  | // when state changes. | ||||||
|  | func (s *state) notify(c *Client, ntype string) { | ||||||
|  | 	c.RunHandlers(&Event{Command: ntype}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // reset resets the state back to it's original form. | ||||||
|  | func (s *state) reset() { | ||||||
|  | 	s.Lock() | ||||||
|  | 	s.nick = "" | ||||||
|  | 	s.ident = "" | ||||||
|  | 	s.host = "" | ||||||
|  | 	s.channels = make(map[string]*Channel) | ||||||
|  | 	s.users = make(map[string]*User) | ||||||
|  | 	s.serverOptions = make(map[string]string) | ||||||
|  | 	s.enabledCap = []string{} | ||||||
|  | 	s.motd = "" | ||||||
|  | 	s.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // User represents an IRC user and the state attached to them. | ||||||
|  | type User struct { | ||||||
|  | 	// Nick is the users current nickname. rfc1459 compliant. | ||||||
|  | 	Nick string `json:"nick"` | ||||||
|  | 	// Ident is the users username/ident. Ident is commonly prefixed with a | ||||||
|  | 	// "~", which indicates that they do not have a identd server setup for | ||||||
|  | 	// authentication. | ||||||
|  | 	Ident string `json:"ident"` | ||||||
|  | 	// Host is the visible host of the users connection that the server has | ||||||
|  | 	// provided to us for their connection. May not always be accurate due to | ||||||
|  | 	// many networks spoofing/hiding parts of the hostname for privacy | ||||||
|  | 	// reasons. | ||||||
|  | 	Host string `json:"host"` | ||||||
|  |  | ||||||
|  | 	// ChannelList is a sorted list of all channels that we are currently | ||||||
|  | 	// tracking the user in. Each channel name is rfc1459 compliant. See | ||||||
|  | 	// User.Channels() for a shorthand if you're looking for the *Channel | ||||||
|  | 	// version of the channel list. | ||||||
|  | 	ChannelList []string `json:"channels"` | ||||||
|  |  | ||||||
|  | 	// FirstSeen represents the first time that the user was seen by the | ||||||
|  | 	// client for the given channel. Only usable if from state, not in past. | ||||||
|  | 	FirstSeen time.Time `json:"first_seen"` | ||||||
|  | 	// LastActive represents the last time that we saw the user active, | ||||||
|  | 	// which could be during nickname change, message, channel join, etc. | ||||||
|  | 	// Only usable if from state, not in past. | ||||||
|  | 	LastActive time.Time `json:"last_active"` | ||||||
|  |  | ||||||
|  | 	// Perms are the user permissions applied to this user that affect the given | ||||||
|  | 	// channel. This supports non-rfc style modes like Admin, Owner, and HalfOp. | ||||||
|  | 	Perms *UserPerms `json:"perms"` | ||||||
|  |  | ||||||
|  | 	// Extras are things added on by additional tracking methods, which may | ||||||
|  | 	// or may not work on the IRC server in mention. | ||||||
|  | 	Extras struct { | ||||||
|  | 		// Name is the users "realname" or full name. Commonly contains links | ||||||
|  | 		// to the IRC client being used, or something of non-importance. May | ||||||
|  | 		// also be empty if unsupported by the server/tracking is disabled. | ||||||
|  | 		Name string `json:"name"` | ||||||
|  | 		// Account refers to the account which the user is authenticated as. | ||||||
|  | 		// This differs between each network (e.g. usually Nickserv, but | ||||||
|  | 		// could also be something like Undernet). May also be empty if | ||||||
|  | 		// unsupported by the server/tracking is disabled. | ||||||
|  | 		Account string `json:"account"` | ||||||
|  | 		// Away refers to the away status of the user. An empty string | ||||||
|  | 		// indicates that they are active, otherwise the string is what they | ||||||
|  | 		// set as their away message. May also be empty if unsupported by the | ||||||
|  | 		// server/tracking is disabled. | ||||||
|  | 		Away string `json:"away"` | ||||||
|  | 	} `json:"extras"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Channels returns a reference of *Channels that the client knows the user | ||||||
|  | // is in. If you're just looking for the namme of the channels, use | ||||||
|  | // User.ChannelList. | ||||||
|  | func (u User) Channels(c *Client) []*Channel { | ||||||
|  | 	if c == nil { | ||||||
|  | 		panic("nil Client provided") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channels := []*Channel{} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	for i := 0; i < len(u.ChannelList); i++ { | ||||||
|  | 		ch := c.state.lookupChannel(u.ChannelList[i]) | ||||||
|  | 		if ch != nil { | ||||||
|  | 			channels = append(channels, ch) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return channels | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a deep copy of the user which can be modified without making | ||||||
|  | // changes to the actual state. | ||||||
|  | func (u *User) Copy() *User { | ||||||
|  | 	nu := &User{} | ||||||
|  | 	*nu = *u | ||||||
|  |  | ||||||
|  | 	nu.Perms = u.Perms.Copy() | ||||||
|  | 	_ = copy(nu.ChannelList, u.ChannelList) | ||||||
|  |  | ||||||
|  | 	return nu | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // addChannel adds the channel to the users channel list. | ||||||
|  | func (u *User) addChannel(name string) { | ||||||
|  | 	if u.InChannel(name) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	u.ChannelList = append(u.ChannelList, ToRFC1459(name)) | ||||||
|  | 	sort.StringsAreSorted(u.ChannelList) | ||||||
|  |  | ||||||
|  | 	u.Perms.set(name, Perms{}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // deleteChannel removes an existing channel from the users channel list. | ||||||
|  | func (u *User) deleteChannel(name string) { | ||||||
|  | 	name = ToRFC1459(name) | ||||||
|  |  | ||||||
|  | 	j := -1 | ||||||
|  | 	for i := 0; i < len(u.ChannelList); i++ { | ||||||
|  | 		if u.ChannelList[i] == name { | ||||||
|  | 			j = i | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if j != -1 { | ||||||
|  | 		u.ChannelList = append(u.ChannelList[:j], u.ChannelList[j+1:]...) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	u.Perms.remove(name) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // InChannel checks to see if a user is in the given channel. | ||||||
|  | func (u *User) InChannel(name string) bool { | ||||||
|  | 	name = ToRFC1459(name) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(u.ChannelList); i++ { | ||||||
|  | 		if u.ChannelList[i] == name { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Lifetime represents the amount of time that has passed since we have first | ||||||
|  | // seen the user. | ||||||
|  | func (u *User) Lifetime() time.Duration { | ||||||
|  | 	return time.Since(u.FirstSeen) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Active represents the the amount of time that has passed since we have | ||||||
|  | // last seen the user. | ||||||
|  | func (u *User) Active() time.Duration { | ||||||
|  | 	return time.Since(u.LastActive) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsActive returns true if they were active within the last 30 minutes. | ||||||
|  | func (u *User) IsActive() bool { | ||||||
|  | 	return u.Active() < (time.Minute * 30) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Channel represents an IRC channel and the state attached to it. | ||||||
|  | type Channel struct { | ||||||
|  | 	// Name of the channel. Must be rfc1459 compliant. | ||||||
|  | 	Name string `json:"name"` | ||||||
|  | 	// Topic of the channel. | ||||||
|  | 	Topic string `json:"topic"` | ||||||
|  |  | ||||||
|  | 	// UserList is a sorted list of all users we are currently tracking within | ||||||
|  | 	// the channel. Each is the nickname, and is rfc1459 compliant. | ||||||
|  | 	UserList []string `json:"user_list"` | ||||||
|  | 	// Joined represents the first time that the client joined the channel. | ||||||
|  | 	Joined time.Time `json:"joined"` | ||||||
|  | 	// Modes are the known channel modes that the bot has captured. | ||||||
|  | 	Modes CModes `json:"modes"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Users returns a reference of *Users that the client knows the channel has | ||||||
|  | // If you're just looking for just the name of the users, use Channnel.UserList. | ||||||
|  | func (ch Channel) Users(c *Client) []*User { | ||||||
|  | 	if c == nil { | ||||||
|  | 		panic("nil Client provided") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	users := []*User{} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	for i := 0; i < len(ch.UserList); i++ { | ||||||
|  | 		user := c.state.lookupUser(ch.UserList[i]) | ||||||
|  | 		if user != nil { | ||||||
|  | 			users = append(users, user) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return users | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Trusted returns a list of users which have voice or greater in the given | ||||||
|  | // channel. See Perms.IsTrusted() for more information. | ||||||
|  | func (ch Channel) Trusted(c *Client) []*User { | ||||||
|  | 	if c == nil { | ||||||
|  | 		panic("nil Client provided") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	users := []*User{} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	for i := 0; i < len(ch.UserList); i++ { | ||||||
|  | 		user := c.state.lookupUser(ch.UserList[i]) | ||||||
|  | 		if user == nil { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		perms, ok := user.Perms.Lookup(ch.Name) | ||||||
|  | 		if ok && perms.IsTrusted() { | ||||||
|  | 			users = append(users, user) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return users | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Admins returns a list of users which have half-op (if supported), or | ||||||
|  | // greater permissions (op, admin, owner, etc) in the given channel. See | ||||||
|  | // Perms.IsAdmin() for more information. | ||||||
|  | func (ch Channel) Admins(c *Client) []*User { | ||||||
|  | 	if c == nil { | ||||||
|  | 		panic("nil Client provided") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	users := []*User{} | ||||||
|  |  | ||||||
|  | 	c.state.RLock() | ||||||
|  | 	for i := 0; i < len(ch.UserList); i++ { | ||||||
|  | 		user := c.state.lookupUser(ch.UserList[i]) | ||||||
|  | 		if user == nil { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		perms, ok := user.Perms.Lookup(ch.Name) | ||||||
|  | 		if ok && perms.IsAdmin() { | ||||||
|  | 			users = append(users, user) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	c.state.RUnlock() | ||||||
|  |  | ||||||
|  | 	return users | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // addUser adds a user to the users list. | ||||||
|  | func (ch *Channel) addUser(nick string) { | ||||||
|  | 	if ch.UserIn(nick) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ch.UserList = append(ch.UserList, ToRFC1459(nick)) | ||||||
|  | 	sort.Strings(ch.UserList) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // deleteUser removes an existing user from the users list. | ||||||
|  | func (ch *Channel) deleteUser(nick string) { | ||||||
|  | 	nick = ToRFC1459(nick) | ||||||
|  |  | ||||||
|  | 	j := -1 | ||||||
|  | 	for i := 0; i < len(ch.UserList); i++ { | ||||||
|  | 		if ch.UserList[i] == nick { | ||||||
|  | 			j = i | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if j != -1 { | ||||||
|  | 		ch.UserList = append(ch.UserList[:j], ch.UserList[j+1:]...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a deep copy of a given channel. | ||||||
|  | func (ch *Channel) Copy() *Channel { | ||||||
|  | 	nc := &Channel{} | ||||||
|  | 	*nc = *ch | ||||||
|  |  | ||||||
|  | 	_ = copy(nc.UserList, ch.UserList) | ||||||
|  |  | ||||||
|  | 	// And modes. | ||||||
|  | 	nc.Modes = ch.Modes.Copy() | ||||||
|  |  | ||||||
|  | 	return nc | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the count of users in a given channel. | ||||||
|  | func (ch *Channel) Len() int { | ||||||
|  | 	return len(ch.UserList) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UserIn checks to see if a given user is in a channel. | ||||||
|  | func (ch *Channel) UserIn(name string) bool { | ||||||
|  | 	name = ToRFC1459(name) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(ch.UserList); i++ { | ||||||
|  | 		if ch.UserList[i] == name { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Lifetime represents the amount of time that has passed since we have first | ||||||
|  | // joined the channel. | ||||||
|  | func (ch *Channel) Lifetime() time.Duration { | ||||||
|  | 	return time.Since(ch.Joined) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // createChannel creates the channel in state, if not already done. | ||||||
|  | func (s *state) createChannel(name string) (ok bool) { | ||||||
|  | 	supported := s.chanModes() | ||||||
|  | 	prefixes, _ := parsePrefixes(s.userPrefixes()) | ||||||
|  |  | ||||||
|  | 	if _, ok := s.channels[ToRFC1459(name)]; ok { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	s.channels[ToRFC1459(name)] = &Channel{ | ||||||
|  | 		Name:     name, | ||||||
|  | 		UserList: []string{}, | ||||||
|  | 		Joined:   time.Now(), | ||||||
|  | 		Modes:    NewCModes(supported, prefixes), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // deleteChannel removes the channel from state, if not already done. | ||||||
|  | func (s *state) deleteChannel(name string) { | ||||||
|  | 	name = ToRFC1459(name) | ||||||
|  |  | ||||||
|  | 	_, ok := s.channels[name] | ||||||
|  | 	if !ok { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, user := range s.channels[name].UserList { | ||||||
|  | 		s.users[user].deleteChannel(name) | ||||||
|  |  | ||||||
|  | 		if len(s.users[user].ChannelList) == 0 { | ||||||
|  | 			// Assume we were only tracking them in this channel, and they | ||||||
|  | 			// should be removed from state. | ||||||
|  |  | ||||||
|  | 			delete(s.users, user) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	delete(s.channels, name) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lookupChannel returns a reference to a channel, nil returned if no results | ||||||
|  | // found. | ||||||
|  | func (s *state) lookupChannel(name string) *Channel { | ||||||
|  | 	return s.channels[ToRFC1459(name)] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lookupUser returns a reference to a user, nil returned if no results | ||||||
|  | // found. | ||||||
|  | func (s *state) lookupUser(name string) *User { | ||||||
|  | 	return s.users[ToRFC1459(name)] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // createUser creates the user in state, if not already done. | ||||||
|  | func (s *state) createUser(nick string) (ok bool) { | ||||||
|  | 	if _, ok := s.users[ToRFC1459(nick)]; ok { | ||||||
|  | 		// User already exists. | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	s.users[ToRFC1459(nick)] = &User{ | ||||||
|  | 		Nick:       nick, | ||||||
|  | 		FirstSeen:  time.Now(), | ||||||
|  | 		LastActive: time.Now(), | ||||||
|  | 		Perms:      &UserPerms{channels: make(map[string]Perms)}, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // deleteUser removes the user from channel state. | ||||||
|  | func (s *state) deleteUser(channelName, nick string) { | ||||||
|  | 	user := s.lookupUser(nick) | ||||||
|  | 	if user == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if channelName == "" { | ||||||
|  | 		for i := 0; i < len(user.ChannelList); i++ { | ||||||
|  | 			s.channels[user.ChannelList[i]].deleteUser(nick) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		delete(s.users, ToRFC1459(nick)) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel := s.lookupChannel(channelName) | ||||||
|  | 	if channel == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user.deleteChannel(channelName) | ||||||
|  | 	channel.deleteUser(nick) | ||||||
|  |  | ||||||
|  | 	if len(user.ChannelList) == 0 { | ||||||
|  | 		// This means they are no longer in any channels we track, delete | ||||||
|  | 		// them from state. | ||||||
|  |  | ||||||
|  | 		delete(s.users, ToRFC1459(nick)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // renameUser renames the user in state, in all locations where relevant. | ||||||
|  | func (s *state) renameUser(from, to string) { | ||||||
|  | 	from = ToRFC1459(from) | ||||||
|  |  | ||||||
|  | 	// Update our nickname. | ||||||
|  | 	if from == ToRFC1459(s.nick) { | ||||||
|  | 		s.nick = to | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user := s.lookupUser(from) | ||||||
|  | 	if user == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	delete(s.users, from) | ||||||
|  |  | ||||||
|  | 	user.Nick = to | ||||||
|  | 	user.LastActive = time.Now() | ||||||
|  | 	s.users[ToRFC1459(to)] = user | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(user.ChannelList); i++ { | ||||||
|  | 		for j := 0; j < len(s.channels[user.ChannelList[i]].UserList); j++ { | ||||||
|  | 			if s.channels[user.ChannelList[i]].UserList[j] == from { | ||||||
|  | 				s.channels[user.ChannelList[i]].UserList[j] = ToRFC1459(to) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										0
									
								
								vendor/github.com/nlopes/slack/dnd.go → vendor/github.com/matterbridge/slack/dnd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										0
									
								
								vendor/github.com/nlopes/slack/dnd.go → vendor/github.com/matterbridge/slack/dnd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
								
								
									
										0
									
								
								vendor/github.com/nlopes/slack/im.go → vendor/github.com/matterbridge/slack/im.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										0
									
								
								vendor/github.com/nlopes/slack/im.go → vendor/github.com/matterbridge/slack/im.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
								
								
									
										0
									
								
								vendor/github.com/nlopes/slack/rtm.go → vendor/github.com/matterbridge/slack/rtm.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										0
									
								
								vendor/github.com/nlopes/slack/rtm.go → vendor/github.com/matterbridge/slack/rtm.go
									
									
									
										generated
									
									
										vendored
									
									
								
							Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user