Compare commits
	
		
			234 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					e9105003b0 | ||
| 
						 | 
					587bb06558 | ||
| 
						 | 
					53e9664cde | ||
| 
						 | 
					482fbac68f | ||
| 
						 | 
					dcccd43427 | ||
| 
						 | 
					397b8ff892 | ||
| 
						 | 
					38a4cf315a | ||
| 
						 | 
					5f8b24e32c | ||
| 
						 | 
					678a7ceb4e | ||
| 
						 | 
					077d494c7b | ||
| 
						 | 
					09b243d8c2 | ||
| 
						 | 
					991183e514 | ||
| 
						 | 
					9bf10e4b58 | ||
| 
						 | 
					884599d27d | ||
| 
						 | 
					f8a6e65bfd | ||
| 
						 | 
					6df6c5d615 | ||
| 
						 | 
					93114b7682 | ||
| 
						 | 
					9987ac3f13 | ||
| 
						 | 
					01a32b2154 | ||
| 
						 | 
					b3c3142bb2 | ||
| 
						 | 
					77f1a959c3 | ||
| 
						 | 
					e3dda0e812 | ||
| 
						 | 
					38103d36b4 | ||
| 
						 | 
					7685fe1724 | ||
| 
						 | 
					01afe03a3f | ||
| 
						 | 
					7fbbf89c58 | ||
| 
						 | 
					84d259d8b3 | ||
| 
						 | 
					8b47670a74 | ||
| 
						 | 
					7f5dc1d461 | ||
| 
						 | 
					43e765f4f9 | ||
| 
						 | 
					adec73f542 | ||
| 
						 | 
					fee159541f | ||
| 
						 | 
					d81e6bf6ce | ||
| 
						 | 
					70c93d970c | ||
| 
						 | 
					4960273832 | ||
| 
						 | 
					6c018ee6fe | ||
| 
						 | 
					4ef32103ca | ||
| 
						 | 
					e4ec27c5e2 | ||
| 
						 | 
					20c04f7977 | ||
| 
						 | 
					571f50d734 | ||
| 
						 | 
					780ea6f7c0 | ||
| 
						 | 
					4279906f6e | ||
| 
						 | 
					2e54b97fc2 | ||
| 
						 | 
					e1641b2c2e | ||
| 
						 | 
					e0e1e4be80 | ||
| 
						 | 
					d5845ce900 | ||
| 
						 | 
					85f2cde4c3 | ||
| 
						 | 
					cef64e01b3 | ||
| 
						 | 
					94ea775232 | ||
| 
						 | 
					2e4b7fac11 | ||
| 
						 | 
					2867ec459a | ||
| 
						 | 
					cd18d89894 | ||
| 
						 | 
					449ed31e25 | ||
| 
						 | 
					1f36904588 | ||
| 
						 | 
					f7495dd0c3 | ||
| 
						 | 
					a11f77835d | ||
| 
						 | 
					af1ad82c8e | ||
| 
						 | 
					4976338677 | ||
| 
						 | 
					99d130d1ed | ||
| 
						 | 
					4fb0544b0e | ||
| 
						 | 
					0b4ac61435 | ||
| 
						 | 
					1d5cd1d7c4 | ||
| 
						 | 
					08ebee6b4f | ||
| 
						 | 
					14830d9f1c | ||
| 
						 | 
					a3dd0f1345 | ||
| 
						 | 
					37873acfcd | ||
| 
						 | 
					2dbe0eb557 | ||
| 
						 | 
					50a0df4279 | ||
| 
						 | 
					c3a8b7a997 | ||
| 
						 | 
					95fac548bb | ||
| 
						 | 
					581847f415 | ||
| 
						 | 
					1b15897135 | ||
| 
						 | 
					8e606e3cef | ||
| 
						 | 
					be513622ac | ||
| 
						 | 
					6f309f2108 | ||
| 
						 | 
					92d9db5a2d | ||
| 
						 | 
					96620a3c2c | ||
| 
						 | 
					5249568b8e | ||
| 
						 | 
					4a336a6bba | ||
| 
						 | 
					60223d7f63 | ||
| 
						 | 
					5131253191 | ||
| 
						 | 
					035dc042a1 | ||
| 
						 | 
					dfc513530b | ||
| 
						 | 
					721e0a2dcd | ||
| 
						 | 
					8452eb12da | ||
| 
						 | 
					475bed5e19 | ||
| 
						 | 
					40a967523c | ||
| 
						 | 
					d3a34af073 | ||
| 
						 | 
					e7107cf782 | ||
| 
						 | 
					b7c918a195 | ||
| 
						 | 
					61e4c9b28c | ||
| 
						 | 
					e93847a95e | ||
| 
						 | 
					545377742c | ||
| 
						 | 
					47d38192b2 | ||
| 
						 | 
					ac80c47036 | ||
| 
						 | 
					1e84afbd90 | ||
| 
						 | 
					d31e641bac | ||
| 
						 | 
					4380c48b4b | ||
| 
						 | 
					db0e4ba8c5 | ||
| 
						 | 
					2d6ed51d94 | ||
| 
						 | 
					9ca4fe7a5e | ||
| 
						 | 
					e52b040b9c | ||
| 
						 | 
					1accee1653 | ||
| 
						 | 
					fff6f08cb6 | ||
| 
						 | 
					0e527a4252 | ||
| 
						 | 
					f10251a1a3 | ||
| 
						 | 
					0d4bad16a3 | ||
| 
						 | 
					8c6be434ac | ||
| 
						 | 
					3ca4309e8a | ||
| 
						 | 
					e8a2e1af63 | ||
| 
						 | 
					1d240140c9 | ||
| 
						 | 
					272eef544f | ||
| 
						 | 
					fd756c5332 | ||
| 
						 | 
					dce600ad51 | ||
| 
						 | 
					d02a737e0c | ||
| 
						 | 
					98ff59c716 | ||
| 
						 | 
					0e96e9f9be | ||
| 
						 | 
					e8c7898583 | ||
| 
						 | 
					11f4a6897a | ||
| 
						 | 
					002c5fd0d1 | ||
| 
						 | 
					18504ec08d | ||
| 
						 | 
					4737442185 | ||
| 
						 | 
					596096d6da | ||
| 
						 | 
					6af82401fc | ||
| 
						 | 
					a0b84beb9b | ||
| 
						 | 
					0816e96831 | ||
| 
						 | 
					7baf386ede | ||
| 
						 | 
					6e410b096e | ||
| 
						 | 
					f9e5994348 | ||
| 
						 | 
					ee77272cfd | ||
| 
						 | 
					16ed2aca6a | ||
| 
						 | 
					0f530e7902 | ||
| 
						 | 
					4ed66ce20e | ||
| 
						 | 
					b30e85836e | ||
| 
						 | 
					e449a97bd0 | ||
| 
						 | 
					39043f3fa4 | ||
| 
						 | 
					12389d602e | ||
| 
						 | 
					44144587a0 | ||
| 
						 | 
					d0a30e354b | ||
| 
						 | 
					c261dc89d5 | ||
| 
						 | 
					c2c135bca2 | ||
| 
						 | 
					eb20cb237d | ||
| 
						 | 
					106404d32f | ||
| 
						 | 
					e06efbad9f | ||
| 
						 | 
					3311c7f923 | ||
| 
						 | 
					3a6c655dfb | ||
| 
						 | 
					e11d786775 | ||
| 
						 | 
					889b6debc4 | ||
| 
						 | 
					9cb3413d9c | ||
| 
						 | 
					131826e1d1 | ||
| 
						 | 
					96e21dd051 | ||
| 
						 | 
					32e5f396e7 | ||
| 
						 | 
					6c6000dbbd | ||
| 
						 | 
					24defcb970 | ||
| 
						 | 
					a1a11a88b3 | ||
| 
						 | 
					a997ae29ad | ||
| 
						 | 
					ff94796700 | ||
| 
						 | 
					1f72ca4c4e | ||
| 
						 | 
					46faad8b57 | ||
| 
						 | 
					30f30364d5 | ||
| 
						 | 
					073d90da88 | ||
| 
						 | 
					c769e23a9a | ||
| 
						 | 
					9db48f4794 | ||
| 
						 | 
					911c597377 | ||
| 
						 | 
					28244ffd9a | ||
| 
						 | 
					3e38c7945c | ||
| 
						 | 
					79ffb76f6e | ||
| 
						 | 
					5fe4b749cf | ||
| 
						 | 
					6991d85da9 | ||
| 
						 | 
					c1c187a1ab | ||
| 
						 | 
					055d12e3ef | ||
| 
						 | 
					b49429d722 | ||
| 
						 | 
					815c7f8d64 | ||
| 
						 | 
					c879f79456 | ||
| 
						 | 
					3bc25f4707 | ||
| 
						 | 
					300cfe044a | ||
| 
						 | 
					fb586f4a96 | ||
| 
						 | 
					ced371bece | ||
| 
						 | 
					a87cac1982 | ||
| 
						 | 
					8fb5c7afa6 | ||
| 
						 | 
					aceb830378 | ||
| 
						 | 
					0f2976c5ce | ||
| 
						 | 
					78b17977c5 | ||
| 
						 | 
					6ec77e06ea | ||
| 
						 | 
					e48db67649 | ||
| 
						 | 
					e03f331f55 | ||
| 
						 | 
					ff5aeeb1e1 | ||
| 
						 | 
					33844fa60c | ||
| 
						 | 
					85faa43145 | ||
| 
						 | 
					59e6abcc11 | ||
| 
						 | 
					38e3bbe5c9 | ||
| 
						 | 
					51265d5464 | ||
| 
						 | 
					de4c780410 | ||
| 
						 | 
					6b18257185 | ||
| 
						 | 
					4b1ebaf7d5 | ||
| 
						 | 
					93db74e7e1 | ||
| 
						 | 
					0e6fe4070a | ||
| 
						 | 
					69b534ee99 | ||
| 
						 | 
					71a504945b | ||
| 
						 | 
					99ac7dc114 | ||
| 
						 | 
					4984473c1b | ||
| 
						 | 
					3fcce2d8a0 | ||
| 
						 | 
					a53e699112 | ||
| 
						 | 
					f29822db02 | ||
| 
						 | 
					a63433e41b | ||
| 
						 | 
					e0379ca5af | ||
| 
						 | 
					4759ee6132 | ||
| 
						 | 
					5ec94fdb43 | ||
| 
						 | 
					a64deb1238 | ||
| 
						 | 
					f914695801 | ||
| 
						 | 
					304dc2e25f | ||
| 
						 | 
					fd74dca175 | ||
| 
						 | 
					c7ace91bf6 | ||
| 
						 | 
					9f07a2cfd5 | ||
| 
						 | 
					0dc5e042d2 | ||
| 
						 | 
					f0a5d2396f | ||
| 
						 | 
					bdac03f725 | ||
| 
						 | 
					c1f80383f7 | ||
| 
						 | 
					bd7c1e3e3c | ||
| 
						 | 
					5c1b02c7a3 | ||
| 
						 | 
					38fce68609 | ||
| 
						 | 
					90f276863b | ||
| 
						 | 
					5282cdaccd | ||
| 
						 | 
					008ea94b53 | ||
| 
						 | 
					693f1946b7 | ||
| 
						 | 
					8b6a00d1c5 | ||
| 
						 | 
					43738dbc89 | ||
| 
						 | 
					6feccd4c6c | ||
| 
						 | 
					25d72a7e31 | ||
| 
						 | 
					523f6ffb80 | ||
| 
						 | 
					b346ac868b | ||
| 
						 | 
					d0cda03478 | ||
| 
						 | 
					19b3145bd1 | ||
| 
						 | 
					a0d1fc0d6a | 
							
								
								
									
										20
									
								
								.github/ISSUE_TEMPLATE.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								.github/ISSUE_TEMPLATE.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,20 @@
 | 
			
		||||
Please answer the following questions. 
 | 
			
		||||
 | 
			
		||||
### Which version of matterbridge are you using?
 | 
			
		||||
run ```matterbridge -version```
 | 
			
		||||
 | 
			
		||||
### If you're having problems with mattermost please specify mattermost version. 
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Please describe the expected behavior.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Please describe the actual behavior. 
 | 
			
		||||
#### Use logs from running ```matterbridge -debug``` if possible.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Any steps to reproduce the behavior?
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Please add your configuration file 
 | 
			
		||||
#### (be sure to exclude or anonymize private data (tokens/passwords))
 | 
			
		||||
							
								
								
									
										11
									
								
								Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								Dockerfile
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,11 @@
 | 
			
		||||
FROM alpine:edge
 | 
			
		||||
ENTRYPOINT ["/bin/matterbridge"]
 | 
			
		||||
 | 
			
		||||
COPY . /go/src/github.com/42wim/matterbridge
 | 
			
		||||
RUN apk update && apk add go git gcc musl-dev ca-certificates \
 | 
			
		||||
        && cd /go/src/github.com/42wim/matterbridge \
 | 
			
		||||
        && export GOPATH=/go \
 | 
			
		||||
        && go get \
 | 
			
		||||
        && go build -o /bin/matterbridge \
 | 
			
		||||
        && rm -rf /go \
 | 
			
		||||
        && apk del --purge git go gcc musl-dev
 | 
			
		||||
							
								
								
									
										115
									
								
								README-0.6.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								README-0.6.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,115 @@
 | 
			
		||||
# matterbridge
 | 
			
		||||
 | 
			
		||||
Simple bridge between mattermost, IRC, XMPP, Gitter and Slack
 | 
			
		||||
 | 
			
		||||
* Relays public channel messages between mattermost, IRC, XMPP, Gitter and Slack. Pick and mix.
 | 
			
		||||
* Supports multiple channels.
 | 
			
		||||
* Matterbridge can also work with private groups on your mattermost.
 | 
			
		||||
 | 
			
		||||
Look at [matterbridge.conf.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.conf.sample) for documentation and an example.
 | 
			
		||||
 | 
			
		||||
## Changelog
 | 
			
		||||
Since v0.6.1 support for XMPP, Gitter and Slack is added. More details in [changelog.md] (https://github.com/42wim/matterbridge/blob/master/changelog.md)
 | 
			
		||||
 | 
			
		||||
## Requirements:
 | 
			
		||||
Accounts to one of the supported bridges
 | 
			
		||||
* [Mattermost] (https://github.com/mattermost/platform/)
 | 
			
		||||
* [IRC] (http://www.mirc.com/servers.html)
 | 
			
		||||
* [XMPP] (https://jabber.org)
 | 
			
		||||
* [Gitter] (https://gitter.im)
 | 
			
		||||
* [Slack] (https://www.slack.com)
 | 
			
		||||
 | 
			
		||||
## binaries
 | 
			
		||||
Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/)
 | 
			
		||||
* For use with mattermost 3.3.0+ [v0.6.1](https://github.com/42wim/matterircd/releases/tag/v0.6.1)
 | 
			
		||||
* For use with mattermost 3.0.0-3.2.0 [v0.5.0](https://github.com/42wim/matterircd/releases/tag/v0.5.0)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
## Docker
 | 
			
		||||
Create your matterbridge.conf file locally eg in ```/tmp/matterbridge.conf```
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
docker run -ti -v /tmp/matterbridge.conf:/matterbridge.conf 42wim/matterbridge:0.6.1
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Compatibility
 | 
			
		||||
### Mattermost 
 | 
			
		||||
* Matterbridge v0.6.1 works with mattermost 3.3.0 and higher [3.3.0 release](https://github.com/mattermost/platform/releases/tag/v3.3.0)
 | 
			
		||||
* Matterbridge v0.5.0 works with mattermost 3.0.0 - 3.2.0 [3.2.0 release](https://github.com/mattermost/platform/releases/tag/v3.2.0)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#### Webhooks version
 | 
			
		||||
* Configured incoming/outgoing [webhooks](https://www.mattermost.org/webhooks/) on your mattermost instance.
 | 
			
		||||
 | 
			
		||||
#### Plus (API) version
 | 
			
		||||
* A dedicated user(bot) on your mattermost instance.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
## building
 | 
			
		||||
Go 1.6+ 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
 | 
			
		||||
go get github.com/42wim/matterbridge
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
You should now have matterbridge binary in the bin directory:
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
$ ls bin/
 | 
			
		||||
matterbridge
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## running
 | 
			
		||||
1) Copy the matterbridge.conf.sample to matterbridge.conf in the same directory as the matterbridge binary.  
 | 
			
		||||
2) Edit matterbridge.conf with the settings for your environment. See below for more config information.  
 | 
			
		||||
3) Now you can run matterbridge. 
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
Usage of ./matterbridge:
 | 
			
		||||
  -conf string
 | 
			
		||||
        config file (default "matterbridge.conf")
 | 
			
		||||
  -debug
 | 
			
		||||
        enable debug
 | 
			
		||||
  -plus
 | 
			
		||||
        running using API instead of webhooks (deprecated, set Plus flag in [general] config)
 | 
			
		||||
  -version
 | 
			
		||||
        show version
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## config
 | 
			
		||||
### matterbridge
 | 
			
		||||
matterbridge looks for matterbridge.conf in current directory. (use -conf to specify another file)
 | 
			
		||||
 | 
			
		||||
Look at [matterbridge.conf.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.conf.sample) for an example.
 | 
			
		||||
 | 
			
		||||
### mattermost
 | 
			
		||||
#### webhooks version
 | 
			
		||||
You'll have to configure the incoming and outgoing webhooks. 
 | 
			
		||||
 | 
			
		||||
* incoming webhooks
 | 
			
		||||
Go to "account settings" - integrations - "incoming webhooks".  
 | 
			
		||||
Choose a channel at "Add a new incoming webhook", this will create a webhook URL right below.  
 | 
			
		||||
This URL should be set in the matterbridge.conf in the [mattermost] section (see above)  
 | 
			
		||||
 | 
			
		||||
* outgoing webhooks
 | 
			
		||||
Go to "account settings" - integrations - "outgoing webhooks".  
 | 
			
		||||
Choose a channel (the same as the one from incoming webhooks) and fill in the address and port of the server matterbridge will run on.  
 | 
			
		||||
 | 
			
		||||
e.g. http://192.168.1.1:9999 (192.168.1.1:9999 is the BindAddress specified in [mattermost] section of matterbridge.conf)
 | 
			
		||||
 | 
			
		||||
#### plus version
 | 
			
		||||
You'll have to create a new dedicated user on your mattermost instance.
 | 
			
		||||
Specify the login and password in [mattermost] section of matterbridge.conf
 | 
			
		||||
 | 
			
		||||
## FAQ
 | 
			
		||||
Please look at [matterbridge.conf.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.conf.sample) for more information first. 
 | 
			
		||||
### Mattermost doesn't show the IRC nicks
 | 
			
		||||
If you're running the webhooks version, this can be fixed by either:
 | 
			
		||||
* enabling "override usernames". See [mattermost documentation](http://docs.mattermost.com/developer/webhooks-incoming.html#enabling-incoming-webhooks)
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.conf.
 | 
			
		||||
 | 
			
		||||
If you're running the plus version you'll need to:
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.conf.
 | 
			
		||||
 | 
			
		||||
Also look at the ```RemoteNickFormat``` setting.
 | 
			
		||||
							
								
								
									
										108
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										108
									
								
								README.md
									
									
									
									
									
								
							@@ -1,16 +1,62 @@
 | 
			
		||||
# matterbridge
 | 
			
		||||

 | 
			
		||||
 | 
			
		||||
Simple bridge between mattermost and IRC. Uses the in/outgoing webhooks.  
 | 
			
		||||
Relays public channel messages between mattermost and IRC.  
 | 
			
		||||
Simple bridge between mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat and Hipchat(via xmpp).
 | 
			
		||||
 | 
			
		||||
* Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat and Hipchat (via xmpp). Pick and mix.
 | 
			
		||||
* Supports multiple channels.
 | 
			
		||||
* Matterbridge can also work with private groups on your mattermost.
 | 
			
		||||
* 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).
 | 
			
		||||
 | 
			
		||||
Look at [matterbridge.toml.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example.
 | 
			
		||||
Look at [matterbridge.toml.simple] (https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.simple) for a simple example.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
## Changelog
 | 
			
		||||
Since v0.7.0 the configuration has changed. More details in [changelog.md] (https://github.com/42wim/matterbridge/blob/master/changelog.md)
 | 
			
		||||
 | 
			
		||||
## Requirements
 | 
			
		||||
Accounts to one of the supported bridges
 | 
			
		||||
* [Mattermost] (https://github.com/mattermost/platform/)
 | 
			
		||||
* [IRC] (http://www.mirc.com/servers.html)
 | 
			
		||||
* [XMPP] (https://jabber.org)
 | 
			
		||||
* [Gitter] (https://gitter.im)
 | 
			
		||||
* [Slack] (https://slack.com)
 | 
			
		||||
* [Discord] (https://discordapp.com)
 | 
			
		||||
* [Telegram] (https://telegram.org)
 | 
			
		||||
* [Hipchat] (https://www.hipchat.com)
 | 
			
		||||
* [Rocket.chat] (https://rocket.chat)
 | 
			
		||||
 | 
			
		||||
## Docker
 | 
			
		||||
Create your matterbridge.toml file locally eg in ```/tmp/matterbridge.toml```
 | 
			
		||||
```
 | 
			
		||||
docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## binaries
 | 
			
		||||
Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/)
 | 
			
		||||
* For use with mattermost 3.5.x - 3.6.0 [v0.9.2](https://github.com/42wim/matterircd/releases/tag/v0.9.2)
 | 
			
		||||
* For use with mattermost 3.3.0 - 3.4.0 [v0.7.1](https://github.com/42wim/matterircd/releases/tag/v0.7.1)
 | 
			
		||||
 | 
			
		||||
## Compatibility
 | 
			
		||||
### Mattermost 
 | 
			
		||||
* Matterbridge v0.9.2 works with mattermost 3.5.x - 3.6.0 [3.6.0 release](https://github.com/mattermost/platform/releases/tag/v3.6.0)
 | 
			
		||||
* Matterbridge v0.7.1 works with mattermost 3.3.0 - 3.4.0 [3.4.0 release](https://github.com/mattermost/platform/releases/tag/v3.4.0)
 | 
			
		||||
 | 
			
		||||
#### Webhooks version
 | 
			
		||||
* Configured incoming/outgoing [webhooks](https://www.mattermost.org/webhooks/) on your mattermost instance.
 | 
			
		||||
 | 
			
		||||
#### API version
 | 
			
		||||
* A dedicated user(bot) on your mattermost instance.
 | 
			
		||||
 | 
			
		||||
Work in progress. 
 | 
			
		||||
 | 
			
		||||
## building
 | 
			
		||||
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.6+ 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
 | 
			
		||||
go get https://github.com/42wim/matterbridge
 | 
			
		||||
go get github.com/42wim/matterbridge
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
You should now have matterbridge binary in the bin directory:
 | 
			
		||||
@@ -23,38 +69,27 @@ matterbridge
 | 
			
		||||
## running
 | 
			
		||||
1) Copy the matterbridge.conf.sample to matterbridge.conf in the same directory as the matterbridge binary.  
 | 
			
		||||
2) Edit matterbridge.conf with the settings for your environment. See below for more config information.  
 | 
			
		||||
3) Now you can run matterbridge.  
 | 
			
		||||
3) Now you can run matterbridge. 
 | 
			
		||||
 | 
			
		||||
Matterbridge will:
 | 
			
		||||
* start a webserver listening on the port specified in the configuration.
 | 
			
		||||
* connect to specified irc server and channel.
 | 
			
		||||
* send messages from mattermost to irc and vice versa, messages in mattermost will appear with irc-nick
 | 
			
		||||
```
 | 
			
		||||
Usage of ./matterbridge:
 | 
			
		||||
  -conf string
 | 
			
		||||
        config file (default "matterbridge.toml")
 | 
			
		||||
  -debug
 | 
			
		||||
        enable debug
 | 
			
		||||
  -version
 | 
			
		||||
        show version
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## config
 | 
			
		||||
### matterbridge
 | 
			
		||||
matterbridge looks for matterbridge.conf in current directory.
 | 
			
		||||
matterbridge looks for matterbridge.toml in current directory. (use -conf to specify another file)
 | 
			
		||||
 | 
			
		||||
Look at matterbridge.conf.sample for an example
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
[IRC]
 | 
			
		||||
server="irc.freenode.net"
 | 
			
		||||
port=6667
 | 
			
		||||
UseTLS=false
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
nick="matterbot"
 | 
			
		||||
channel="#matterbridge"
 | 
			
		||||
 | 
			
		||||
[mattermost]
 | 
			
		||||
#url is your incoming webhook url (account settings - integrations - incoming webhooks)
 | 
			
		||||
url="http://mattermost.yourdomain.com/hooks/incomingwebhookkey"  
 | 
			
		||||
#port the bridge webserver will listen on
 | 
			
		||||
port=9999
 | 
			
		||||
```
 | 
			
		||||
Look at [matterbridge.toml.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for an example.
 | 
			
		||||
 | 
			
		||||
### mattermost
 | 
			
		||||
You'll have to configure the incoming en outgoing webhooks. 
 | 
			
		||||
#### webhooks version
 | 
			
		||||
You'll have to configure the incoming and outgoing webhooks. 
 | 
			
		||||
 | 
			
		||||
* incoming webhooks
 | 
			
		||||
Go to "account settings" - integrations - "incoming webhooks".  
 | 
			
		||||
@@ -65,5 +100,16 @@ This URL should be set in the matterbridge.conf in the [mattermost] section (see
 | 
			
		||||
Go to "account settings" - integrations - "outgoing webhooks".  
 | 
			
		||||
Choose a channel (the same as the one from incoming webhooks) and fill in the address and port of the server matterbridge will run on.  
 | 
			
		||||
 | 
			
		||||
e.g. http://192.168.1.1:9999 (9999 is the port specified in [mattermost] section of matterbridge.conf)
 | 
			
		||||
e.g. http://192.168.1.1:9999 (192.168.1.1:9999 is the BindAddress specified in [mattermost] section of matterbridge.conf)
 | 
			
		||||
 | 
			
		||||
## FAQ
 | 
			
		||||
Please look at [matterbridge.toml.sample] (https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for more information first. 
 | 
			
		||||
### Mattermost doesn't show the IRC nicks
 | 
			
		||||
If you're running the webhooks version, this can be fixed by either:
 | 
			
		||||
* enabling "override usernames". See [mattermost documentation](http://docs.mattermost.com/developer/webhooks-incoming.html#enabling-incoming-webhooks)
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.toml.
 | 
			
		||||
 | 
			
		||||
If you're running the plus version you'll need to:
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.toml.
 | 
			
		||||
 | 
			
		||||
Also look at the ```RemoteNickFormat``` setting.
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										68
									
								
								bridge/bridge.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										68
									
								
								bridge/bridge.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,68 @@
 | 
			
		||||
package bridge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/discord"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/irc"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/mattermost"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/rocketchat"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/slack"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/telegram"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/xmpp"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bridger interface {
 | 
			
		||||
	Send(msg config.Message) error
 | 
			
		||||
	Connect() error
 | 
			
		||||
	JoinChannel(channel string) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
	Config config.Protocol
 | 
			
		||||
	Bridger
 | 
			
		||||
	Name     string
 | 
			
		||||
	Account  string
 | 
			
		||||
	Protocol string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge {
 | 
			
		||||
	b := new(Bridge)
 | 
			
		||||
	accInfo := strings.Split(bridge.Account, ".")
 | 
			
		||||
	protocol := accInfo[0]
 | 
			
		||||
	name := accInfo[1]
 | 
			
		||||
	b.Name = name
 | 
			
		||||
	b.Protocol = protocol
 | 
			
		||||
	b.Account = bridge.Account
 | 
			
		||||
 | 
			
		||||
	// override config from environment
 | 
			
		||||
	config.OverrideCfgFromEnv(cfg, protocol, name)
 | 
			
		||||
	switch protocol {
 | 
			
		||||
	case "mattermost":
 | 
			
		||||
		b.Config = cfg.Mattermost[name]
 | 
			
		||||
		b.Bridger = bmattermost.New(cfg.Mattermost[name], bridge.Account, c)
 | 
			
		||||
	case "irc":
 | 
			
		||||
		b.Config = cfg.IRC[name]
 | 
			
		||||
		b.Bridger = birc.New(cfg.IRC[name], bridge.Account, c)
 | 
			
		||||
	case "gitter":
 | 
			
		||||
		b.Config = cfg.Gitter[name]
 | 
			
		||||
		b.Bridger = bgitter.New(cfg.Gitter[name], bridge.Account, c)
 | 
			
		||||
	case "slack":
 | 
			
		||||
		b.Config = cfg.Slack[name]
 | 
			
		||||
		b.Bridger = bslack.New(cfg.Slack[name], bridge.Account, c)
 | 
			
		||||
	case "xmpp":
 | 
			
		||||
		b.Config = cfg.Xmpp[name]
 | 
			
		||||
		b.Bridger = bxmpp.New(cfg.Xmpp[name], bridge.Account, c)
 | 
			
		||||
	case "discord":
 | 
			
		||||
		b.Config = cfg.Discord[name]
 | 
			
		||||
		b.Bridger = bdiscord.New(cfg.Discord[name], bridge.Account, c)
 | 
			
		||||
	case "telegram":
 | 
			
		||||
		b.Config = cfg.Telegram[name]
 | 
			
		||||
		b.Bridger = btelegram.New(cfg.Telegram[name], bridge.Account, c)
 | 
			
		||||
	case "rocketchat":
 | 
			
		||||
		b.Config = cfg.Rocketchat[name]
 | 
			
		||||
		b.Bridger = brocketchat.New(cfg.Rocketchat[name], bridge.Account, c)
 | 
			
		||||
	}
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										148
									
								
								bridge/config/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										148
									
								
								bridge/config/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,148 @@
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/BurntSushi/toml"
 | 
			
		||||
	"log"
 | 
			
		||||
	"os"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	EVENT_JOIN_LEAVE = "join_leave"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	Avatar   string
 | 
			
		||||
	Account  string
 | 
			
		||||
	Event    string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Protocol struct {
 | 
			
		||||
	BindAddress            string // mattermost, slack
 | 
			
		||||
	IconURL                string // mattermost, slack
 | 
			
		||||
	IgnoreNicks            string // all protocols
 | 
			
		||||
	Jid                    string // xmpp
 | 
			
		||||
	Login                  string // mattermost
 | 
			
		||||
	Muc                    string // xmpp
 | 
			
		||||
	Name                   string // all protocols
 | 
			
		||||
	Nick                   string // all protocols
 | 
			
		||||
	NickFormatter          string // mattermost, slack
 | 
			
		||||
	NickServNick           string // IRC
 | 
			
		||||
	NickServPassword       string // IRC
 | 
			
		||||
	NicksPerRow            int    // mattermost, slack
 | 
			
		||||
	NoTLS                  bool   // mattermost
 | 
			
		||||
	Password               string // IRC,mattermost,XMPP
 | 
			
		||||
	PrefixMessagesWithNick bool   // mattemost, slack
 | 
			
		||||
	Protocol               string //all protocols
 | 
			
		||||
	MessageQueue           int    // IRC, size of message queue for flood control
 | 
			
		||||
	MessageDelay           int    // IRC, time in millisecond to wait between messages
 | 
			
		||||
	RemoteNickFormat       string // all protocols
 | 
			
		||||
	Server                 string // IRC,mattermost,XMPP,discord
 | 
			
		||||
	ShowJoinPart           bool   // all protocols
 | 
			
		||||
	SkipTLSVerify          bool   // IRC, mattermost
 | 
			
		||||
	Team                   string // mattermost
 | 
			
		||||
	Token                  string // gitter, slack, discord
 | 
			
		||||
	URL                    string // mattermost, slack
 | 
			
		||||
	UseAPI                 bool   // mattermost, slack
 | 
			
		||||
	UseSASL                bool   // IRC
 | 
			
		||||
	UseTLS                 bool   // IRC
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type ChannelOptions struct {
 | 
			
		||||
	Key string // irc
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
	Account string
 | 
			
		||||
	Channel string
 | 
			
		||||
	Options ChannelOptions
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Gateway struct {
 | 
			
		||||
	Name   string
 | 
			
		||||
	Enable bool
 | 
			
		||||
	In     []Bridge
 | 
			
		||||
	Out    []Bridge
 | 
			
		||||
	InOut  []Bridge
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type SameChannelGateway struct {
 | 
			
		||||
	Name     string
 | 
			
		||||
	Enable   bool
 | 
			
		||||
	Channels []string
 | 
			
		||||
	Accounts []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	IRC                map[string]Protocol
 | 
			
		||||
	Mattermost         map[string]Protocol
 | 
			
		||||
	Slack              map[string]Protocol
 | 
			
		||||
	Gitter             map[string]Protocol
 | 
			
		||||
	Xmpp               map[string]Protocol
 | 
			
		||||
	Discord            map[string]Protocol
 | 
			
		||||
	Telegram           map[string]Protocol
 | 
			
		||||
	Rocketchat         map[string]Protocol
 | 
			
		||||
	General            Protocol
 | 
			
		||||
	Gateway            []Gateway
 | 
			
		||||
	SameChannelGateway []SameChannelGateway
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewConfig(cfgfile string) *Config {
 | 
			
		||||
	var cfg Config
 | 
			
		||||
	if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	return &cfg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func OverrideCfgFromEnv(cfg *Config, protocol string, account string) {
 | 
			
		||||
	var protoCfg Protocol
 | 
			
		||||
	val := reflect.ValueOf(cfg).Elem()
 | 
			
		||||
	// loop over the Config struct
 | 
			
		||||
	for i := 0; i < val.NumField(); i++ {
 | 
			
		||||
		typeField := val.Type().Field(i)
 | 
			
		||||
		// look for the protocol map (both lowercase)
 | 
			
		||||
		if strings.ToLower(typeField.Name) == protocol {
 | 
			
		||||
			// get the Protocol struct from the map
 | 
			
		||||
			data := val.Field(i).MapIndex(reflect.ValueOf(account))
 | 
			
		||||
			protoCfg = data.Interface().(Protocol)
 | 
			
		||||
			protoStruct := reflect.ValueOf(&protoCfg).Elem()
 | 
			
		||||
			// loop over the found protocol struct
 | 
			
		||||
			for i := 0; i < protoStruct.NumField(); i++ {
 | 
			
		||||
				typeField := protoStruct.Type().Field(i)
 | 
			
		||||
				// build our environment key (eg MATTERBRIDGE_MATTERMOST_WORK_LOGIN)
 | 
			
		||||
				key := "matterbridge_" + protocol + "_" + account + "_" + typeField.Name
 | 
			
		||||
				key = strings.ToUpper(key)
 | 
			
		||||
				// search the environment
 | 
			
		||||
				res := os.Getenv(key)
 | 
			
		||||
				// if it exists and the current field is a string
 | 
			
		||||
				// then update the current field
 | 
			
		||||
				if res != "" {
 | 
			
		||||
					fieldVal := protoStruct.Field(i)
 | 
			
		||||
					if fieldVal.Kind() == reflect.String {
 | 
			
		||||
						log.Printf("config: overriding %s from env with %s\n", key, res)
 | 
			
		||||
						fieldVal.Set(reflect.ValueOf(res))
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			// update the map with the modified Protocol (cfg.Protocol[account] = Protocol)
 | 
			
		||||
			val.Field(i).SetMapIndex(reflect.ValueOf(account), reflect.ValueOf(protoCfg))
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetIconURL(msg *Message, cfg *Protocol) string {
 | 
			
		||||
	iconURL := cfg.IconURL
 | 
			
		||||
	info := strings.Split(msg.Account, ".")
 | 
			
		||||
	protocol := info[0]
 | 
			
		||||
	name := info[1]
 | 
			
		||||
	iconURL = strings.Replace(iconURL, "{NICK}", msg.Username, -1)
 | 
			
		||||
	iconURL = strings.Replace(iconURL, "{BRIDGE}", name, -1)
 | 
			
		||||
	iconURL = strings.Replace(iconURL, "{PROTOCOL}", protocol, -1)
 | 
			
		||||
	return iconURL
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										137
									
								
								bridge/discord/discord.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										137
									
								
								bridge/discord/discord.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,137 @@
 | 
			
		||||
package bdiscord
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type bdiscord struct {
 | 
			
		||||
	c            *discordgo.Session
 | 
			
		||||
	Config       *config.Protocol
 | 
			
		||||
	Remote       chan config.Message
 | 
			
		||||
	Account      string
 | 
			
		||||
	Channels     []*discordgo.Channel
 | 
			
		||||
	Nick         string
 | 
			
		||||
	UseChannelID bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "discord"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord {
 | 
			
		||||
	b := &bdiscord{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	if !strings.HasPrefix(b.Config.Token, "Bot ") {
 | 
			
		||||
		b.Config.Token = "Bot " + b.Config.Token
 | 
			
		||||
	}
 | 
			
		||||
	b.c, err = discordgo.New(b.Config.Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.c.AddHandler(b.messageCreate)
 | 
			
		||||
	err = b.c.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	guilds, err := b.c.UserGuilds()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	userinfo, err := b.c.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Nick = userinfo.Username
 | 
			
		||||
	for _, guild := range guilds {
 | 
			
		||||
		if guild.Name == b.Config.Server {
 | 
			
		||||
			b.Channels, err = b.c.GuildChannels(guild.ID)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				flog.Debugf("%#v", err)
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) JoinChannel(channel string) error {
 | 
			
		||||
	idcheck := strings.Split(channel, "ID:")
 | 
			
		||||
	if len(idcheck) > 1 {
 | 
			
		||||
		b.UseChannelID = true
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	channelID := b.getChannelID(msg.Channel)
 | 
			
		||||
	if channelID == "" {
 | 
			
		||||
		flog.Errorf("Could not find channelID for %v", msg.Channel)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	b.c.ChannelMessageSend(channelID, msg.Username+msg.Text)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
	// not relay our own messages
 | 
			
		||||
	if m.Author.Username == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if len(m.Attachments) > 0 {
 | 
			
		||||
		for _, attach := range m.Attachments {
 | 
			
		||||
			m.Content = m.Content + "\n" + attach.URL
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if m.Content == "" {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account)
 | 
			
		||||
	channelName := b.getChannelName(m.ChannelID)
 | 
			
		||||
	if b.UseChannelID {
 | 
			
		||||
		channelName = "ID:" + m.ChannelID
 | 
			
		||||
	}
 | 
			
		||||
	b.Remote <- config.Message{Username: m.Author.Username, Text: m.ContentWithMentionsReplaced(), Channel: channelName,
 | 
			
		||||
		Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg"}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) getChannelID(name string) string {
 | 
			
		||||
	idcheck := strings.Split(name, "ID:")
 | 
			
		||||
	if len(idcheck) > 1 {
 | 
			
		||||
		return idcheck[1]
 | 
			
		||||
	}
 | 
			
		||||
	for _, channel := range b.Channels {
 | 
			
		||||
		if channel.Name == name {
 | 
			
		||||
			return channel.ID
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) getChannelName(id string) string {
 | 
			
		||||
	for _, channel := range b.Channels {
 | 
			
		||||
		if channel.ID == id {
 | 
			
		||||
			return channel.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										115
									
								
								bridge/gitter/gitter.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								bridge/gitter/gitter.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,115 @@
 | 
			
		||||
package bgitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/go-gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bgitter struct {
 | 
			
		||||
	c       *gitter.Gitter
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
	Users   []gitter.User
 | 
			
		||||
	Rooms   []gitter.Room
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "gitter"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bgitter {
 | 
			
		||||
	b := &Bgitter{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	b.c = gitter.New(b.Config.Token)
 | 
			
		||||
	_, err = b.c.GetUser()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.Rooms, _ = b.c.GetRooms()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) JoinChannel(channel string) error {
 | 
			
		||||
	room := channel
 | 
			
		||||
	roomID := b.getRoomID(room)
 | 
			
		||||
	if roomID == "" {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	user, err := b.c.GetUser()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	_, err = b.c.JoinRoom(roomID, user.ID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	users, _ := b.c.GetUsersInRoom(roomID)
 | 
			
		||||
	b.Users = append(b.Users, users...)
 | 
			
		||||
	stream := b.c.Stream(roomID)
 | 
			
		||||
	go b.c.Listen(stream)
 | 
			
		||||
 | 
			
		||||
	go func(stream *gitter.Stream, room string) {
 | 
			
		||||
		for event := range stream.Event {
 | 
			
		||||
			switch ev := event.Data.(type) {
 | 
			
		||||
			case *gitter.MessageReceived:
 | 
			
		||||
				// check for ZWSP to see if it's not an echo
 | 
			
		||||
				if !strings.HasSuffix(ev.Message.Text, "") {
 | 
			
		||||
					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,
 | 
			
		||||
						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username)}
 | 
			
		||||
				}
 | 
			
		||||
			case *gitter.GitterConnectionClosed:
 | 
			
		||||
				flog.Errorf("connection with gitter closed for room %s", room)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}(stream, room)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	roomID := b.getRoomID(msg.Channel)
 | 
			
		||||
	if roomID == "" {
 | 
			
		||||
		flog.Errorf("Could not find roomID for %v", msg.Channel)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	// add ZWSP because gitter echoes our own messages
 | 
			
		||||
	return b.c.SendMessage(roomID, msg.Username+msg.Text+" ")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) getRoomID(channel string) string {
 | 
			
		||||
	for _, v := range b.Rooms {
 | 
			
		||||
		if v.URI == channel {
 | 
			
		||||
			return v.ID
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) getAvatar(user string) string {
 | 
			
		||||
	var avatar string
 | 
			
		||||
	if b.Users != nil {
 | 
			
		||||
		for _, u := range b.Users {
 | 
			
		||||
			if user == u.Username {
 | 
			
		||||
				return u.AvatarURLSmall
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return avatar
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										59
									
								
								bridge/irc/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								bridge/irc/helper.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,59 @@
 | 
			
		||||
package birc
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func tableformatter(nicks []string, nicksPerRow int, continued bool) string {
 | 
			
		||||
	result := "|IRC users"
 | 
			
		||||
	if continued {
 | 
			
		||||
		result = "|(continued)"
 | 
			
		||||
	}
 | 
			
		||||
	for i := 0; i < 2; i++ {
 | 
			
		||||
		for j := 1; j <= nicksPerRow && j <= len(nicks); j++ {
 | 
			
		||||
			if i == 0 {
 | 
			
		||||
				result += "|"
 | 
			
		||||
			} else {
 | 
			
		||||
				result += ":-|"
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		result += "\r\n|"
 | 
			
		||||
	}
 | 
			
		||||
	result += nicks[0] + "|"
 | 
			
		||||
	for i := 1; i < len(nicks); i++ {
 | 
			
		||||
		if i%nicksPerRow == 0 {
 | 
			
		||||
			result += "\r\n|" + nicks[i] + "|"
 | 
			
		||||
		} else {
 | 
			
		||||
			result += nicks[i] + "|"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func plainformatter(nicks []string, nicksPerRow int) string {
 | 
			
		||||
	return strings.Join(nicks, ", ") + " currently on IRC"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func IsMarkup(message string) bool {
 | 
			
		||||
	switch message[0] {
 | 
			
		||||
	case '|':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '#':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '_':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '*':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '~':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '-':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case ':':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '>':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '=':
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										257
									
								
								bridge/irc/irc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										257
									
								
								bridge/irc/irc.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,257 @@
 | 
			
		||||
package birc
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	ircm "github.com/sorcix/irc"
 | 
			
		||||
	"github.com/thoj/go-ircevent"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Birc struct {
 | 
			
		||||
	i         *irc.Connection
 | 
			
		||||
	Nick      string
 | 
			
		||||
	names     map[string][]string
 | 
			
		||||
	Config    *config.Protocol
 | 
			
		||||
	Remote    chan config.Message
 | 
			
		||||
	connected chan struct{}
 | 
			
		||||
	Local     chan config.Message // local queue for flood control
 | 
			
		||||
	Account   string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "irc"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Birc {
 | 
			
		||||
	b := &Birc{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Nick = b.Config.Nick
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.names = make(map[string][]string)
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.connected = make(chan struct{})
 | 
			
		||||
	if b.Config.MessageDelay == 0 {
 | 
			
		||||
		b.Config.MessageDelay = 1300
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.MessageQueue == 0 {
 | 
			
		||||
		b.Config.MessageQueue = 30
 | 
			
		||||
	}
 | 
			
		||||
	b.Local = make(chan config.Message, b.Config.MessageQueue+10)
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Command(msg *config.Message) string {
 | 
			
		||||
	switch msg.Text {
 | 
			
		||||
	case "!users":
 | 
			
		||||
		b.i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames)
 | 
			
		||||
		b.i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames)
 | 
			
		||||
		b.i.SendRaw("NAMES " + msg.Channel)
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Connect() error {
 | 
			
		||||
	flog.Infof("Connecting %s", b.Config.Server)
 | 
			
		||||
	i := irc.IRC(b.Config.Nick, b.Config.Nick)
 | 
			
		||||
	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}
 | 
			
		||||
	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 {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.i = i
 | 
			
		||||
	select {
 | 
			
		||||
	case <-b.connected:
 | 
			
		||||
		flog.Info("Connection succeeded")
 | 
			
		||||
	case <-time.After(time.Second * 30):
 | 
			
		||||
		return fmt.Errorf("connection timed out")
 | 
			
		||||
	}
 | 
			
		||||
	i.Debug = false
 | 
			
		||||
	go b.doSend()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) JoinChannel(channel string) error {
 | 
			
		||||
	b.i.Join(channel)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	if msg.Account == b.Account {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	if strings.HasPrefix(msg.Text, "!") {
 | 
			
		||||
		b.Command(&msg)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	for _, text := range strings.Split(msg.Text, "\n") {
 | 
			
		||||
		if len(b.Local) < b.Config.MessageQueue {
 | 
			
		||||
			if len(b.Local) == b.Config.MessageQueue-1 {
 | 
			
		||||
				text = text + " <message clipped>"
 | 
			
		||||
			}
 | 
			
		||||
			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel}
 | 
			
		||||
		} else {
 | 
			
		||||
			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local))
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) doSend() {
 | 
			
		||||
	rate := time.Millisecond * time.Duration(b.Config.MessageDelay)
 | 
			
		||||
	throttle := time.Tick(rate)
 | 
			
		||||
	for msg := range b.Local {
 | 
			
		||||
		<-throttle
 | 
			
		||||
		b.i.Privmsg(msg.Channel, msg.Username+msg.Text)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) endNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[1]
 | 
			
		||||
	sort.Strings(b.names[channel])
 | 
			
		||||
	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
 | 
			
		||||
	continued := false
 | 
			
		||||
	for len(b.names[channel]) > maxNamesPerPost {
 | 
			
		||||
		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost], continued),
 | 
			
		||||
			Channel: channel, Account: b.Account}
 | 
			
		||||
		b.names[channel] = b.names[channel][maxNamesPerPost:]
 | 
			
		||||
		continued = true
 | 
			
		||||
	}
 | 
			
		||||
	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued),
 | 
			
		||||
		Channel: channel, Account: b.Account}
 | 
			
		||||
	b.names[channel] = nil
 | 
			
		||||
	b.i.ClearCallback(ircm.RPL_NAMREPLY)
 | 
			
		||||
	b.i.ClearCallback(ircm.RPL_ENDOFNAMES)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleNewConnection(event *irc.Event) {
 | 
			
		||||
	flog.Debug("Registering callbacks")
 | 
			
		||||
	i := b.i
 | 
			
		||||
	b.Nick = event.Arguments[0]
 | 
			
		||||
	i.AddCallback("PRIVMSG", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback("CTCP_ACTION", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
 | 
			
		||||
	i.AddCallback(ircm.NOTICE, b.handleNotice)
 | 
			
		||||
	//i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) })
 | 
			
		||||
	i.AddCallback("PING", func(e *irc.Event) {
 | 
			
		||||
		i.SendRaw("PONG :" + e.Message())
 | 
			
		||||
		flog.Debugf("PING/PONG")
 | 
			
		||||
	})
 | 
			
		||||
	i.AddCallback("JOIN", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("PART", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("QUIT", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("*", b.handleOther)
 | 
			
		||||
	// we are now fully connected
 | 
			
		||||
	b.connected <- struct{}{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleJoinPart(event *irc.Event) {
 | 
			
		||||
	flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account)
 | 
			
		||||
	channel := event.Arguments[0]
 | 
			
		||||
	if event.Code == "QUIT" {
 | 
			
		||||
		channel = ""
 | 
			
		||||
	}
 | 
			
		||||
	b.Remote <- config.Message{Username: "system", Text: event.Nick + " " + strings.ToLower(event.Code) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE}
 | 
			
		||||
	flog.Debugf("handle %#v", event)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleNotice(event *irc.Event) {
 | 
			
		||||
	if strings.Contains(event.Message(), "This nickname is registered") && event.Nick == b.Config.NickServNick {
 | 
			
		||||
		b.i.Privmsg(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword)
 | 
			
		||||
	} else {
 | 
			
		||||
		b.handlePrivMsg(event)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleOther(event *irc.Event) {
 | 
			
		||||
	switch event.Code {
 | 
			
		||||
	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005":
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("%#v", event.Raw)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handlePrivMsg(event *irc.Event) {
 | 
			
		||||
	// don't forward queries to the bot
 | 
			
		||||
	if event.Arguments[0] == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// don't forward message from ourself
 | 
			
		||||
	if event.Nick == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("handlePrivMsg() %s %s %#v", event.Nick, event.Message(), event)
 | 
			
		||||
	msg := ""
 | 
			
		||||
	if event.Code == "CTCP_ACTION" {
 | 
			
		||||
		msg = event.Nick + " "
 | 
			
		||||
	}
 | 
			
		||||
	msg += event.Message()
 | 
			
		||||
	// strip IRC colors
 | 
			
		||||
	re := regexp.MustCompile(`[[:cntrl:]](\d+,|)\d+`)
 | 
			
		||||
	msg = re.ReplaceAllString(msg, "")
 | 
			
		||||
	flog.Debugf("Sending message from %s on %s to gateway", event.Arguments[0], b.Account)
 | 
			
		||||
	b.Remote <- config.Message{Username: event.Nick, Text: msg, Channel: event.Arguments[0], Account: b.Account}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleTopicWhoTime(event *irc.Event) {
 | 
			
		||||
	parts := strings.Split(event.Arguments[2], "!")
 | 
			
		||||
	t, err := strconv.ParseInt(event.Arguments[3], 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Errorf("Invalid time stamp: %s", event.Arguments[3])
 | 
			
		||||
	}
 | 
			
		||||
	user := parts[0]
 | 
			
		||||
	if len(parts) > 1 {
 | 
			
		||||
		user += " [" + parts[1] + "]"
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) nicksPerRow() int {
 | 
			
		||||
	return 4
 | 
			
		||||
	/*
 | 
			
		||||
		if b.Config.Mattermost.NicksPerRow < 1 {
 | 
			
		||||
			return 4
 | 
			
		||||
		}
 | 
			
		||||
		return b.Config.Mattermost.NicksPerRow
 | 
			
		||||
	*/
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) storeNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[2]
 | 
			
		||||
	b.names[channel] = append(
 | 
			
		||||
		b.names[channel],
 | 
			
		||||
		strings.Split(strings.TrimSpace(event.Message()), " ")...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) formatnicks(nicks []string, continued bool) string {
 | 
			
		||||
	return plainformatter(nicks, b.nicksPerRow())
 | 
			
		||||
	/*
 | 
			
		||||
		switch b.Config.Mattermost.NickFormatter {
 | 
			
		||||
		case "table":
 | 
			
		||||
			return tableformatter(nicks, b.nicksPerRow(), continued)
 | 
			
		||||
		default:
 | 
			
		||||
			return plainformatter(nicks, b.nicksPerRow())
 | 
			
		||||
		}
 | 
			
		||||
	*/
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										163
									
								
								bridge/mattermost/mattermost.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										163
									
								
								bridge/mattermost/mattermost.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,163 @@
 | 
			
		||||
package bmattermost
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterclient"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMhook struct {
 | 
			
		||||
	mh *matterhook.Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMapi struct {
 | 
			
		||||
	mc            *matterclient.MMClient
 | 
			
		||||
	mmMap         map[string]string
 | 
			
		||||
	mmIgnoreNicks []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMMessage struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bmattermost struct {
 | 
			
		||||
	MMhook
 | 
			
		||||
	MMapi
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	name    string
 | 
			
		||||
	TeamId  string
 | 
			
		||||
	Account string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "mattermost"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bmattermost {
 | 
			
		||||
	b := &Bmattermost{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.mmMap = make(map[string]string)
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) Command(cmd string) string {
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) Connect() error {
 | 
			
		||||
	if !b.Config.UseAPI {
 | 
			
		||||
		flog.Info("Connecting webhooks")
 | 
			
		||||
		b.mh = matterhook.New(b.Config.URL,
 | 
			
		||||
			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify,
 | 
			
		||||
				BindAddress: b.Config.BindAddress})
 | 
			
		||||
	} else {
 | 
			
		||||
		b.mc = matterclient.New(b.Config.Login, b.Config.Password,
 | 
			
		||||
			b.Config.Team, b.Config.Server)
 | 
			
		||||
		b.mc.SkipTLSVerify = b.Config.SkipTLSVerify
 | 
			
		||||
		b.mc.NoTLS = b.Config.NoTLS
 | 
			
		||||
		flog.Infof("Connecting %s (team: %s) on %s", b.Config.Login, b.Config.Team, b.Config.Server)
 | 
			
		||||
		err := b.mc.Login()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		flog.Info("Connection succeeded")
 | 
			
		||||
		b.TeamId = b.mc.GetTeamId()
 | 
			
		||||
		go b.mc.WsReceiver()
 | 
			
		||||
	}
 | 
			
		||||
	go b.handleMatter()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) JoinChannel(channel string) error {
 | 
			
		||||
	// we can only join channels using the API
 | 
			
		||||
	if b.Config.UseAPI {
 | 
			
		||||
		return b.mc.JoinChannel(b.mc.GetChannelId(channel, ""))
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	nick := msg.Username
 | 
			
		||||
	message := msg.Text
 | 
			
		||||
	channel := msg.Channel
 | 
			
		||||
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick {
 | 
			
		||||
		/*if IsMarkup(message) {
 | 
			
		||||
			message = nick + "\n\n" + message
 | 
			
		||||
		} else {
 | 
			
		||||
		*/
 | 
			
		||||
		message = nick + " " + message
 | 
			
		||||
		//}
 | 
			
		||||
	}
 | 
			
		||||
	if !b.Config.UseAPI {
 | 
			
		||||
		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
		matterMessage.Channel = channel
 | 
			
		||||
		matterMessage.UserName = nick
 | 
			
		||||
		matterMessage.Type = ""
 | 
			
		||||
		matterMessage.Text = message
 | 
			
		||||
		err := b.mh.Send(matterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.Info(err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatter() {
 | 
			
		||||
	flog.Debugf("Choosing API based Mattermost connection: %t", b.Config.UseAPI)
 | 
			
		||||
	mchan := make(chan *MMMessage)
 | 
			
		||||
	if b.Config.UseAPI {
 | 
			
		||||
		go b.handleMatterClient(mchan)
 | 
			
		||||
	} else {
 | 
			
		||||
		go b.handleMatterHook(mchan)
 | 
			
		||||
	}
 | 
			
		||||
	for message := range mchan {
 | 
			
		||||
		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}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) {
 | 
			
		||||
	for message := range b.mc.MessageChan {
 | 
			
		||||
		// do not post our own messages back to irc
 | 
			
		||||
		// only listen to message from our team
 | 
			
		||||
		if message.Raw.Event == "posted" && b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId {
 | 
			
		||||
			flog.Debugf("Receiving from matterclient %#v", message)
 | 
			
		||||
			m := &MMMessage{}
 | 
			
		||||
			m.Username = message.Username
 | 
			
		||||
			m.Channel = message.Channel
 | 
			
		||||
			m.Text = message.Text
 | 
			
		||||
			if len(message.Post.FileIds) > 0 {
 | 
			
		||||
				for _, link := range b.mc.GetPublicLinks(message.Post.FileIds) {
 | 
			
		||||
					m.Text = m.Text + "\n" + link
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			mchan <- m
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatterHook(mchan chan *MMMessage) {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.mh.Receive()
 | 
			
		||||
		flog.Debugf("Receiving from matterhook %#v", message)
 | 
			
		||||
		m := &MMMessage{}
 | 
			
		||||
		m.Username = message.UserName
 | 
			
		||||
		m.Text = message.Text
 | 
			
		||||
		m.Channel = message.ChannelName
 | 
			
		||||
		mchan <- m
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										82
									
								
								bridge/rocketchat/rocketchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								bridge/rocketchat/rocketchat.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,82 @@
 | 
			
		||||
package brocketchat
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/hook/rockethook"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMhook struct {
 | 
			
		||||
	mh *matterhook.Client
 | 
			
		||||
	rh *rockethook.Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Brocketchat struct {
 | 
			
		||||
	MMhook
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	name    string
 | 
			
		||||
	Account string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "rocketchat"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Brocketchat {
 | 
			
		||||
	b := &Brocketchat{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Command(cmd string) string {
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Connect() error {
 | 
			
		||||
	flog.Info("Connecting webhooks")
 | 
			
		||||
	b.mh = matterhook.New(b.Config.URL,
 | 
			
		||||
		matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify,
 | 
			
		||||
			DisableServer: true})
 | 
			
		||||
	b.rh = rockethook.New(b.Config.URL, rockethook.Config{BindAddress: b.Config.BindAddress})
 | 
			
		||||
	go b.handleRocketHook()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) JoinChannel(channel string) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
	matterMessage.Channel = msg.Channel
 | 
			
		||||
	matterMessage.UserName = msg.Username
 | 
			
		||||
	matterMessage.Type = ""
 | 
			
		||||
	matterMessage.Text = msg.Text
 | 
			
		||||
	err := b.mh.Send(matterMessage)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Info(err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) handleRocketHook() {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.rh.Receive()
 | 
			
		||||
		flog.Debugf("Receiving from rockethook %#v", message)
 | 
			
		||||
		// do not loop
 | 
			
		||||
		if message.UserName == b.Config.Nick {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		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.ChannelName, Account: b.Account}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										251
									
								
								bridge/slack/slack.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										251
									
								
								bridge/slack/slack.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,251 @@
 | 
			
		||||
package bslack
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/nlopes/slack"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMMessage struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	Raw      *slack.MessageEvent
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bslack struct {
 | 
			
		||||
	mh       *matterhook.Client
 | 
			
		||||
	sc       *slack.Client
 | 
			
		||||
	Config   *config.Protocol
 | 
			
		||||
	rtm      *slack.RTM
 | 
			
		||||
	Plus     bool
 | 
			
		||||
	Remote   chan config.Message
 | 
			
		||||
	Users    []slack.User
 | 
			
		||||
	Account  string
 | 
			
		||||
	si       *slack.Info
 | 
			
		||||
	channels []slack.Channel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "slack"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bslack {
 | 
			
		||||
	b := &Bslack{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Command(cmd string) string {
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Connect() error {
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	if !b.Config.UseAPI {
 | 
			
		||||
		b.mh = matterhook.New(b.Config.URL,
 | 
			
		||||
			matterhook.Config{BindAddress: b.Config.BindAddress})
 | 
			
		||||
	} else {
 | 
			
		||||
		b.sc = slack.New(b.Config.Token)
 | 
			
		||||
		b.rtm = b.sc.NewRTM()
 | 
			
		||||
		go b.rtm.ManageConnection()
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	go b.handleSlack()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) JoinChannel(channel string) error {
 | 
			
		||||
	// we can only join channels using the API
 | 
			
		||||
	if b.Config.UseAPI {
 | 
			
		||||
		_, err := b.sc.JoinChannel(channel)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	if msg.Account == b.Account {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	nick := msg.Username
 | 
			
		||||
	message := msg.Text
 | 
			
		||||
	channel := msg.Channel
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick {
 | 
			
		||||
		message = nick + " " + message
 | 
			
		||||
	}
 | 
			
		||||
	if !b.Config.UseAPI {
 | 
			
		||||
		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
		matterMessage.Channel = channel
 | 
			
		||||
		matterMessage.UserName = nick
 | 
			
		||||
		matterMessage.Type = ""
 | 
			
		||||
		matterMessage.Text = message
 | 
			
		||||
		err := b.mh.Send(matterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.Info(err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	schannel, err := b.getChannelByName(channel)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	np := slack.NewPostMessageParameters()
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick == true {
 | 
			
		||||
		np.AsUser = true
 | 
			
		||||
	}
 | 
			
		||||
	np.Username = nick
 | 
			
		||||
	np.IconURL = config.GetIconURL(&msg, b.Config)
 | 
			
		||||
	if msg.Avatar != "" {
 | 
			
		||||
		np.IconURL = msg.Avatar
 | 
			
		||||
	}
 | 
			
		||||
	b.sc.PostMessage(schannel.ID, message, np)
 | 
			
		||||
 | 
			
		||||
	/*
 | 
			
		||||
	   newmsg := b.rtm.NewOutgoingMessage(message, schannel.ID)
 | 
			
		||||
	   b.rtm.SendMessage(newmsg)
 | 
			
		||||
	*/
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) getAvatar(user string) string {
 | 
			
		||||
	var avatar string
 | 
			
		||||
	if b.Users != nil {
 | 
			
		||||
		for _, u := range b.Users {
 | 
			
		||||
			if user == u.Name {
 | 
			
		||||
				return u.Profile.Image48
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return avatar
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) getChannelByName(name string) (*slack.Channel, error) {
 | 
			
		||||
	if b.channels == nil {
 | 
			
		||||
		return nil, fmt.Errorf("%s: channel %s not found (no channels found)", b.Account, name)
 | 
			
		||||
	}
 | 
			
		||||
	for _, channel := range b.channels {
 | 
			
		||||
		if channel.Name == name {
 | 
			
		||||
			return &channel, nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil, fmt.Errorf("%s: channel %s not found", b.Account, name)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleSlack() {
 | 
			
		||||
	flog.Debugf("Choosing API based slack connection: %t", b.Config.UseAPI)
 | 
			
		||||
	mchan := make(chan *MMMessage)
 | 
			
		||||
	if b.Config.UseAPI {
 | 
			
		||||
		go b.handleSlackClient(mchan)
 | 
			
		||||
	} else {
 | 
			
		||||
		go b.handleMatterHook(mchan)
 | 
			
		||||
	}
 | 
			
		||||
	time.Sleep(time.Second)
 | 
			
		||||
	flog.Debug("Start listening for Slack messages")
 | 
			
		||||
	for message := range mchan {
 | 
			
		||||
		// do not send messages from ourself
 | 
			
		||||
		if b.Config.UseAPI && message.Username == b.si.User.Name {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		texts := strings.Split(message.Text, "\n")
 | 
			
		||||
		for _, text := range texts {
 | 
			
		||||
			flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account)
 | 
			
		||||
			b.Remote <- config.Message{Text: text, Username: message.Username, Channel: message.Channel, Account: b.Account, Avatar: b.getAvatar(message.Username)}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
 | 
			
		||||
	count := 0
 | 
			
		||||
	for msg := range b.rtm.IncomingEvents {
 | 
			
		||||
		switch ev := msg.Data.(type) {
 | 
			
		||||
		case *slack.MessageEvent:
 | 
			
		||||
			// ignore first message
 | 
			
		||||
			if count > 0 {
 | 
			
		||||
				flog.Debugf("Receiving from slackclient %#v", ev)
 | 
			
		||||
				//ev.ReplyTo
 | 
			
		||||
				channel, err := b.rtm.GetChannelInfo(ev.Channel)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				user, err := b.rtm.GetUserInfo(ev.User)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				m := &MMMessage{}
 | 
			
		||||
				m.Username = user.Name
 | 
			
		||||
				m.Channel = channel.Name
 | 
			
		||||
				m.Text = ev.Text
 | 
			
		||||
				m.Raw = ev
 | 
			
		||||
				m.Text = b.replaceMention(m.Text)
 | 
			
		||||
				mchan <- m
 | 
			
		||||
			}
 | 
			
		||||
			count++
 | 
			
		||||
		case *slack.OutgoingErrorEvent:
 | 
			
		||||
			flog.Debugf("%#v", ev.Error())
 | 
			
		||||
		case *slack.ChannelJoinedEvent:
 | 
			
		||||
			b.Users, _ = b.sc.GetUsers()
 | 
			
		||||
		case *slack.ConnectedEvent:
 | 
			
		||||
			b.channels = ev.Info.Channels
 | 
			
		||||
			b.si = ev.Info
 | 
			
		||||
			b.Users, _ = b.sc.GetUsers()
 | 
			
		||||
			// add private channels
 | 
			
		||||
			groups, _ := b.sc.GetGroups(true)
 | 
			
		||||
			for _, g := range groups {
 | 
			
		||||
				channel := new(slack.Channel)
 | 
			
		||||
				channel.ID = g.ID
 | 
			
		||||
				channel.Name = g.Name
 | 
			
		||||
				b.channels = append(b.channels, *channel)
 | 
			
		||||
			}
 | 
			
		||||
		case *slack.InvalidAuthEvent:
 | 
			
		||||
			flog.Fatalf("Invalid Token %#v", ev)
 | 
			
		||||
		default:
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleMatterHook(mchan chan *MMMessage) {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.mh.Receive()
 | 
			
		||||
		flog.Debugf("receiving from matterhook (slack) %#v", message)
 | 
			
		||||
		m := &MMMessage{}
 | 
			
		||||
		m.Username = message.UserName
 | 
			
		||||
		m.Text = message.Text
 | 
			
		||||
		m.Text = b.replaceMention(m.Text)
 | 
			
		||||
		m.Channel = message.ChannelName
 | 
			
		||||
		mchan <- m
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) userName(id string) string {
 | 
			
		||||
	for _, u := range b.Users {
 | 
			
		||||
		if u.ID == id {
 | 
			
		||||
			return u.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) replaceMention(text string) string {
 | 
			
		||||
	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1)
 | 
			
		||||
	for _, r := range results {
 | 
			
		||||
		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1)
 | 
			
		||||
 | 
			
		||||
	}
 | 
			
		||||
	return text
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										161
									
								
								bridge/telegram/telegram.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										161
									
								
								bridge/telegram/telegram.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,161 @@
 | 
			
		||||
package btelegram
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"html"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/go-telegram-bot-api/telegram-bot-api"
 | 
			
		||||
	"github.com/russross/blackfriday"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Btelegram struct {
 | 
			
		||||
	c       *tgbotapi.BotAPI
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "telegram"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Btelegram {
 | 
			
		||||
	b := &Btelegram{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	b.c, err = tgbotapi.NewBotAPI(b.Config.Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	updates, err := b.c.GetUpdatesChan(tgbotapi.NewUpdate(0))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	go b.handleRecv(updates)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) JoinChannel(channel string) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type customHtml struct {
 | 
			
		||||
	blackfriday.Renderer
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) Paragraph(out *bytes.Buffer, text func() bool) {
 | 
			
		||||
	marker := out.Len()
 | 
			
		||||
 | 
			
		||||
	if !text() {
 | 
			
		||||
		out.Truncate(marker)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	out.WriteString("\n")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) BlockCode(out *bytes.Buffer, text []byte, lang string) {
 | 
			
		||||
	out.WriteString("<pre>")
 | 
			
		||||
 | 
			
		||||
	out.WriteString(html.EscapeString(string(text)))
 | 
			
		||||
	out.WriteString("</pre>\n")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) Header(out *bytes.Buffer, text func() bool, level int, id string) {
 | 
			
		||||
	options.Paragraph(out, text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) HRule(out *bytes.Buffer) {
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) BlockQuote(out *bytes.Buffer, text []byte) {
 | 
			
		||||
	out.WriteString("> ")
 | 
			
		||||
	out.Write(text)
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) List(out *bytes.Buffer, text func() bool, flags int) {
 | 
			
		||||
	options.Paragraph(out, text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) ListItem(out *bytes.Buffer, text []byte, flags int) {
 | 
			
		||||
	out.WriteString("- ")
 | 
			
		||||
	out.Write(text)
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	chatid, err := strconv.ParseInt(msg.Channel, 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	parsed := blackfriday.Markdown([]byte(msg.Text),
 | 
			
		||||
		&customHtml{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")},
 | 
			
		||||
		blackfriday.EXTENSION_NO_INTRA_EMPHASIS|
 | 
			
		||||
			blackfriday.EXTENSION_FENCED_CODE|
 | 
			
		||||
			blackfriday.EXTENSION_AUTOLINK|
 | 
			
		||||
			blackfriday.EXTENSION_SPACE_HEADERS|
 | 
			
		||||
			blackfriday.EXTENSION_HEADER_IDS|
 | 
			
		||||
			blackfriday.EXTENSION_BACKSLASH_LINE_BREAK|
 | 
			
		||||
			blackfriday.EXTENSION_DEFINITION_LISTS)
 | 
			
		||||
 | 
			
		||||
	m := tgbotapi.NewMessage(chatid, html.EscapeString(msg.Username)+string(parsed))
 | 
			
		||||
	m.ParseMode = "HTML"
 | 
			
		||||
	_, err = b.c.Send(m)
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
 | 
			
		||||
	username := ""
 | 
			
		||||
	text := ""
 | 
			
		||||
	channel := ""
 | 
			
		||||
	for update := range updates {
 | 
			
		||||
		// handle channels
 | 
			
		||||
		if update.ChannelPost != nil {
 | 
			
		||||
			if update.ChannelPost.From != nil {
 | 
			
		||||
				username = update.ChannelPost.From.FirstName
 | 
			
		||||
				if username == "" {
 | 
			
		||||
					username = update.ChannelPost.From.UserName
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			text = update.ChannelPost.Text
 | 
			
		||||
			channel = strconv.FormatInt(update.ChannelPost.Chat.ID, 10)
 | 
			
		||||
		}
 | 
			
		||||
		// handle groups
 | 
			
		||||
		if update.Message != nil {
 | 
			
		||||
			if update.Message.From != nil {
 | 
			
		||||
				username = update.Message.From.FirstName
 | 
			
		||||
				if username == "" {
 | 
			
		||||
					username = update.Message.From.UserName
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			text = update.Message.Text
 | 
			
		||||
			channel = strconv.FormatInt(update.Message.Chat.ID, 10)
 | 
			
		||||
		}
 | 
			
		||||
		if username == "" {
 | 
			
		||||
			username = "unknown"
 | 
			
		||||
		}
 | 
			
		||||
		if text != "" {
 | 
			
		||||
			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}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										134
									
								
								bridge/xmpp/xmpp.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										134
									
								
								bridge/xmpp/xmpp.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,134 @@
 | 
			
		||||
package bxmpp
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/mattn/go-xmpp"
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bxmpp struct {
 | 
			
		||||
	xc      *xmpp.Client
 | 
			
		||||
	xmppMap map[string]string
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "xmpp"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bxmpp {
 | 
			
		||||
	b := &Bxmpp{}
 | 
			
		||||
	b.xmppMap = make(map[string]string)
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Infof("Connecting %s", b.Config.Server)
 | 
			
		||||
	b.xc, err = b.createXMPP()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	go b.handleXmpp()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) JoinChannel(channel string) error {
 | 
			
		||||
	b.xc.JoinMUCNoHistory(channel+"@"+b.Config.Muc, b.Config.Nick)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text})
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) createXMPP() (*xmpp.Client, error) {
 | 
			
		||||
	tc := new(tls.Config)
 | 
			
		||||
	tc.InsecureSkipVerify = b.Config.SkipTLSVerify
 | 
			
		||||
	tc.ServerName = strings.Split(b.Config.Server, ":")[0]
 | 
			
		||||
	options := xmpp.Options{
 | 
			
		||||
		Host:     b.Config.Server,
 | 
			
		||||
		User:     b.Config.Jid,
 | 
			
		||||
		Password: b.Config.Password,
 | 
			
		||||
		NoTLS:    true,
 | 
			
		||||
		StartTLS: true,
 | 
			
		||||
		TLSConfig: tc,
 | 
			
		||||
 | 
			
		||||
		//StartTLS:      false,
 | 
			
		||||
		Debug:                        true,
 | 
			
		||||
		Session:                      true,
 | 
			
		||||
		Status:                       "",
 | 
			
		||||
		StatusMessage:                "",
 | 
			
		||||
		Resource:                     "",
 | 
			
		||||
		InsecureAllowUnencryptedAuth: false,
 | 
			
		||||
		//InsecureAllowUnencryptedAuth: true,
 | 
			
		||||
	}
 | 
			
		||||
	var err error
 | 
			
		||||
	b.xc, err = options.NewClient()
 | 
			
		||||
	return b.xc, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) xmppKeepAlive() chan bool {
 | 
			
		||||
	done := make(chan bool)
 | 
			
		||||
	go func() {
 | 
			
		||||
		ticker := time.NewTicker(90 * time.Second)
 | 
			
		||||
		defer ticker.Stop()
 | 
			
		||||
		for {
 | 
			
		||||
			select {
 | 
			
		||||
			case <-ticker.C:
 | 
			
		||||
				b.xc.PingC2S("", "")
 | 
			
		||||
			case <-done:
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
	return done
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) handleXmpp() error {
 | 
			
		||||
	done := b.xmppKeepAlive()
 | 
			
		||||
	defer close(done)
 | 
			
		||||
	nodelay := time.Time{}
 | 
			
		||||
	for {
 | 
			
		||||
		m, err := b.xc.Recv()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		switch v := m.(type) {
 | 
			
		||||
		case xmpp.Chat:
 | 
			
		||||
			var channel, nick string
 | 
			
		||||
			if v.Type == "groupchat" {
 | 
			
		||||
				s := strings.Split(v.Remote, "@")
 | 
			
		||||
				if len(s) == 2 {
 | 
			
		||||
					channel = s[0]
 | 
			
		||||
				}
 | 
			
		||||
				s = strings.Split(s[1], "/")
 | 
			
		||||
				if len(s) == 2 {
 | 
			
		||||
					nick = s[1]
 | 
			
		||||
				}
 | 
			
		||||
				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" {
 | 
			
		||||
					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}
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		case xmpp.Presence:
 | 
			
		||||
			// do nothing
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										175
									
								
								changelog.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										175
									
								
								changelog.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,175 @@
 | 
			
		||||
# v0.9.2
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: make ignorenicks work again #115
 | 
			
		||||
* telegram: fix receiving from channels and groups #112
 | 
			
		||||
* telegram: use html for username
 | 
			
		||||
* telegram: use ```unknown``` as username when username is not visible.
 | 
			
		||||
* irc: update vendor (fixes some crashes) #117
 | 
			
		||||
* xmpp: fix tls by setting ServerName #114
 | 
			
		||||
 | 
			
		||||
## New features
 | 
			
		||||
* slack: support private channels #118
 | 
			
		||||
 | 
			
		||||
# v0.9.1
 | 
			
		||||
## New features
 | 
			
		||||
* Rocket.Chat: New protocol support added (https://rocket.chat)
 | 
			
		||||
* irc: add channel key support #27 (see matterbrige.toml.sample for example)
 | 
			
		||||
* xmpp: add SkipTLSVerify #106
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: Exit when a bridge fails to start
 | 
			
		||||
* mattermost: Check errors only on first connect. Keep retrying after first connection succeeds. #95
 | 
			
		||||
* telegram: fix missing username #102
 | 
			
		||||
* slack: do not use API functions in webhook (slack) #110
 | 
			
		||||
 | 
			
		||||
# v0.9.0
 | 
			
		||||
## New features
 | 
			
		||||
* Telegram: New protocol support added (https://telegram.org)
 | 
			
		||||
* Hipchat: Add sample config to connect to hipchat via xmpp
 | 
			
		||||
* discord: add "Bot " tag to discord tokens automatically
 | 
			
		||||
* slack: Add support for dynamic Iconurl #43
 | 
			
		||||
* general: Add ```gateway.inout``` config option for bidirectional bridges #85
 | 
			
		||||
* general: Add ```[general]``` section so that ```RemoteNickFormat``` can be set globally
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: when using samechannelgateway NickFormat get doubled by the NICK #77
 | 
			
		||||
* general: fix ShowJoinPart for messages from irc bridge #72
 | 
			
		||||
* gitter: fix high cpu usage #89
 | 
			
		||||
* irc: fix !users command #78
 | 
			
		||||
* xmpp: fix keepalive
 | 
			
		||||
* xmpp: do not relay delayed/empty messages
 | 
			
		||||
* slack: Replace id-mentions to usernames #86 
 | 
			
		||||
* mattermost: fix public links not working (API changes)
 | 
			
		||||
 | 
			
		||||
# v0.8.1
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: when using samechannelgateway NickFormat get doubled by the NICK #77
 | 
			
		||||
* irc: fix !users command #78
 | 
			
		||||
 | 
			
		||||
# v0.8.0
 | 
			
		||||
Release because of breaking mattermost API changes
 | 
			
		||||
## New features
 | 
			
		||||
* Supports mattermost v3.5.0
 | 
			
		||||
 | 
			
		||||
# v0.7.1
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: when using samechannelgateway NickFormat get doubled by the NICK #77
 | 
			
		||||
* irc: fix !users command #78
 | 
			
		||||
 | 
			
		||||
# v0.7.0
 | 
			
		||||
## Breaking config changes from 0.6 to 0.7
 | 
			
		||||
Matterbridge now uses TOML configuration (https://github.com/toml-lang/toml)
 | 
			
		||||
See matterbridge.toml.sample for an example
 | 
			
		||||
 | 
			
		||||
## New features
 | 
			
		||||
### General
 | 
			
		||||
* Allow for bridging the same type of bridge, which means you can eg bridge between multiple mattermosts.
 | 
			
		||||
* The bridge is now actually a gateway which has support multiple in and out bridges. (and supports multiple gateways).
 | 
			
		||||
* Discord support added. See matterbridge.toml.sample for more information.
 | 
			
		||||
* Samechannelgateway support added, easier configuration for 1:1 mapping of protocols with same channel names. #35
 | 
			
		||||
* Support for override from environment variables. #50
 | 
			
		||||
* Better debugging output.
 | 
			
		||||
* discord: New protocol support added. (http://www.discordapp.com)
 | 
			
		||||
* mattermost: Support attachments.
 | 
			
		||||
* irc: Strip colors. #33
 | 
			
		||||
* irc: Anti-flooding support. #40
 | 
			
		||||
* irc: Forward channel notices.
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* irc: Split newlines. #37
 | 
			
		||||
* irc: Only respond to nick related notices from nickserv.
 | 
			
		||||
* irc: Ignore queries send to the bot.
 | 
			
		||||
* irc: Ignore messages from ourself.
 | 
			
		||||
* irc: Only output the "users on irc information" when asked with "!users".
 | 
			
		||||
* irc: Actually wait until connection is complete before saying it is.
 | 
			
		||||
* mattermost: Fix mattermost channel joins.
 | 
			
		||||
* mattermost: Drop messages not from our team.
 | 
			
		||||
* slack: Do not panic on non-existing channels.
 | 
			
		||||
* general: Exit when a bridge fails to start.
 | 
			
		||||
 | 
			
		||||
# v0.6.1
 | 
			
		||||
## New features
 | 
			
		||||
* Slack support added.  See matterbridge.conf.sample for more information
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* Fix 100% CPU bug on incorrect closed connections
 | 
			
		||||
 | 
			
		||||
# v0.6.0-beta2
 | 
			
		||||
## New features
 | 
			
		||||
* Gitter support added.  See matterbridge.conf.sample for more information
 | 
			
		||||
 | 
			
		||||
# v0.6.0-beta1
 | 
			
		||||
## Breaking changes from 0.5 to 0.6
 | 
			
		||||
### commandline
 | 
			
		||||
* -plus switch deprecated. Use ```Plus=true``` or ```Plus``` in ```[general]``` section
 | 
			
		||||
 | 
			
		||||
### IRC section
 | 
			
		||||
* ```Enabled``` added (default false)  
 | 
			
		||||
Add ```Enabled=true``` or ```Enabled``` to the ```[IRC]``` section if you want to enable the IRC bridge
 | 
			
		||||
 | 
			
		||||
### Mattermost section
 | 
			
		||||
* ```Enabled``` added (default false)  
 | 
			
		||||
Add ```Enabled=true``` or ```Enabled``` to the ```[mattermost]``` section if you want to enable the mattermost bridge
 | 
			
		||||
 | 
			
		||||
### General section
 | 
			
		||||
* Use ```Plus=true``` or ```Plus``` in ```[general]``` section to enable the API version of matterbridge
 | 
			
		||||
 | 
			
		||||
## New features
 | 
			
		||||
* Matterbridge now bridges between any specified protocol (not only mattermost anymore) 
 | 
			
		||||
* XMPP support added.  See matterbridge.conf.sample for more information
 | 
			
		||||
* RemoteNickFormat {BRIDGE} variable added  
 | 
			
		||||
You can now add the originating bridge to ```RemoteNickFormat```  
 | 
			
		||||
eg ```RemoteNickFormat="[{BRIDGE}] <{NICK}> "```
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# v0.5.0
 | 
			
		||||
## Breaking changes from 0.4 to 0.5 for matterbridge (webhooks version)
 | 
			
		||||
### IRC section
 | 
			
		||||
#### Server
 | 
			
		||||
Port removed, added to server
 | 
			
		||||
```
 | 
			
		||||
server="irc.freenode.net"
 | 
			
		||||
port=6667
 | 
			
		||||
```
 | 
			
		||||
changed to
 | 
			
		||||
```
 | 
			
		||||
server="irc.freenode.net:6667"
 | 
			
		||||
```
 | 
			
		||||
#### Channel
 | 
			
		||||
Removed see Channels section below
 | 
			
		||||
 | 
			
		||||
#### UseSlackCircumfix=true
 | 
			
		||||
Removed, can be done by using ```RemoteNickFormat="<{NICK}> "```
 | 
			
		||||
 | 
			
		||||
### Mattermost section
 | 
			
		||||
#### BindAddress
 | 
			
		||||
Port removed, added to BindAddress
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
BindAddress="0.0.0.0"
 | 
			
		||||
port=9999
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
changed to
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
BindAddress="0.0.0.0:9999"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
#### Token
 | 
			
		||||
Removed
 | 
			
		||||
 | 
			
		||||
### Channels section
 | 
			
		||||
```
 | 
			
		||||
[Token "outgoingwebhooktoken1"] 
 | 
			
		||||
IRCChannel="#off-topic"
 | 
			
		||||
MMChannel="off-topic"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
changed to
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
[Channel "channelnameofchoice"] 
 | 
			
		||||
IRC="#off-topic"
 | 
			
		||||
Mattermost="off-topic"
 | 
			
		||||
```
 | 
			
		||||
							
								
								
									
										35
									
								
								config.go
									
									
									
									
									
								
							
							
						
						
									
										35
									
								
								config.go
									
									
									
									
									
								
							@@ -1,35 +0,0 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"gopkg.in/gcfg.v1"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	IRC struct {
 | 
			
		||||
		UseTLS        bool
 | 
			
		||||
		SkipTLSVerify bool
 | 
			
		||||
		Server        string
 | 
			
		||||
		Port          int
 | 
			
		||||
		Nick          string
 | 
			
		||||
		Channel       string
 | 
			
		||||
	}
 | 
			
		||||
	Mattermost struct {
 | 
			
		||||
		URL  string
 | 
			
		||||
		Port int
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewConfig(cfgfile string) *Config {
 | 
			
		||||
	var cfg Config
 | 
			
		||||
	content, err := ioutil.ReadFile(cfgfile)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	err = gcfg.ReadStringInto(&cfg, string(content))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal("Failed to parse "+cfgfile+":", err)
 | 
			
		||||
	}
 | 
			
		||||
	return &cfg
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										187
									
								
								gateway/gateway.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										187
									
								
								gateway/gateway.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,187 @@
 | 
			
		||||
package gateway
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Gateway struct {
 | 
			
		||||
	*config.Config
 | 
			
		||||
	MyConfig *config.Gateway
 | 
			
		||||
	//Bridges     []*bridge.Bridge
 | 
			
		||||
	Bridges        map[string]*bridge.Bridge
 | 
			
		||||
	ChannelsOut    map[string][]string
 | 
			
		||||
	ChannelsIn     map[string][]string
 | 
			
		||||
	ChannelOptions map[string]config.ChannelOptions
 | 
			
		||||
	Name           string
 | 
			
		||||
	Message        chan config.Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *config.Config, gateway *config.Gateway) *Gateway {
 | 
			
		||||
	gw := &Gateway{}
 | 
			
		||||
	gw.Name = gateway.Name
 | 
			
		||||
	gw.Config = cfg
 | 
			
		||||
	gw.MyConfig = gateway
 | 
			
		||||
	gw.Message = make(chan config.Message)
 | 
			
		||||
	gw.Bridges = make(map[string]*bridge.Bridge)
 | 
			
		||||
	return gw
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) AddBridge(cfg *config.Bridge) error {
 | 
			
		||||
	for _, br := range gw.Bridges {
 | 
			
		||||
		if br.Account == cfg.Account {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	log.Infof("Starting bridge: %s ", cfg.Account)
 | 
			
		||||
	br := bridge.New(gw.Config, cfg, gw.Message)
 | 
			
		||||
	gw.Bridges[cfg.Account] = br
 | 
			
		||||
	err := br.Connect()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
 | 
			
		||||
	}
 | 
			
		||||
	exists := make(map[string]bool)
 | 
			
		||||
	for _, channel := range append(gw.ChannelsOut[br.Account], gw.ChannelsIn[br.Account]...) {
 | 
			
		||||
		if !exists[br.Account+channel] {
 | 
			
		||||
			mychannel := channel
 | 
			
		||||
			log.Infof("%s: joining %s", br.Account, channel)
 | 
			
		||||
			if br.Protocol == "irc" && gw.ChannelOptions[br.Account+channel].Key != "" {
 | 
			
		||||
				log.Debugf("using key %s for channel %s", gw.ChannelOptions[br.Account+channel].Key, channel)
 | 
			
		||||
				mychannel = mychannel + " " + gw.ChannelOptions[br.Account+channel].Key
 | 
			
		||||
			}
 | 
			
		||||
			br.JoinChannel(mychannel)
 | 
			
		||||
			exists[br.Account+channel] = true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) Start() error {
 | 
			
		||||
	gw.mapChannels()
 | 
			
		||||
	for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) {
 | 
			
		||||
		err := gw.AddBridge(&br)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	go gw.handleReceive()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) handleReceive() {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case msg := <-gw.Message:
 | 
			
		||||
			if !gw.ignoreMessage(&msg) {
 | 
			
		||||
				for _, br := range gw.Bridges {
 | 
			
		||||
					gw.handleMessage(msg, br)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) mapChannels() error {
 | 
			
		||||
	options := make(map[string]config.ChannelOptions)
 | 
			
		||||
	m := make(map[string][]string)
 | 
			
		||||
	for _, br := range gw.MyConfig.Out {
 | 
			
		||||
		m[br.Account] = append(m[br.Account], br.Channel)
 | 
			
		||||
		options[br.Account+br.Channel] = br.Options
 | 
			
		||||
	}
 | 
			
		||||
	gw.ChannelsOut = m
 | 
			
		||||
	m = nil
 | 
			
		||||
	m = make(map[string][]string)
 | 
			
		||||
	for _, br := range gw.MyConfig.In {
 | 
			
		||||
		m[br.Account] = append(m[br.Account], br.Channel)
 | 
			
		||||
		options[br.Account+br.Channel] = br.Options
 | 
			
		||||
	}
 | 
			
		||||
	gw.ChannelsIn = m
 | 
			
		||||
	for _, br := range gw.MyConfig.InOut {
 | 
			
		||||
		gw.ChannelsIn[br.Account] = append(gw.ChannelsIn[br.Account], br.Channel)
 | 
			
		||||
		gw.ChannelsOut[br.Account] = append(gw.ChannelsOut[br.Account], br.Channel)
 | 
			
		||||
		options[br.Account+br.Channel] = br.Options
 | 
			
		||||
	}
 | 
			
		||||
	gw.ChannelOptions = options
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) getDestChannel(msg *config.Message, dest string) []string {
 | 
			
		||||
	channels := gw.ChannelsIn[msg.Account]
 | 
			
		||||
	// broadcast to every out channel (irc QUIT)
 | 
			
		||||
	if msg.Event == config.EVENT_JOIN_LEAVE && msg.Channel == "" {
 | 
			
		||||
		return gw.ChannelsOut[dest]
 | 
			
		||||
	}
 | 
			
		||||
	for _, channel := range channels {
 | 
			
		||||
		if channel == msg.Channel {
 | 
			
		||||
			return gw.ChannelsOut[dest]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return []string{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	// only relay join/part when configged
 | 
			
		||||
	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	originchannel := msg.Channel
 | 
			
		||||
	channels := gw.getDestChannel(&msg, dest.Account)
 | 
			
		||||
	for _, channel := range channels {
 | 
			
		||||
		// do not send the message to the bridge we come from if also the channel is the same
 | 
			
		||||
		if msg.Account == dest.Account && channel == originchannel {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		msg.Channel = channel
 | 
			
		||||
		if msg.Channel == "" {
 | 
			
		||||
			log.Debug("empty channel")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		log.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel)
 | 
			
		||||
		gw.modifyUsername(&msg, dest)
 | 
			
		||||
		err := dest.Send(msg)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println(err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) {
 | 
			
		||||
		if msg.Username == entry {
 | 
			
		||||
			log.Debugf("ignoring %s from %s", msg.Username, msg.Account)
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) modifyMessage(msg *config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	val := reflect.ValueOf(gw.Config).Elem()
 | 
			
		||||
	for i := 0; i < val.NumField(); i++ {
 | 
			
		||||
		typeField := val.Type().Field(i)
 | 
			
		||||
		// look for the protocol map (both lowercase)
 | 
			
		||||
		if strings.ToLower(typeField.Name) == dest.Protocol {
 | 
			
		||||
			// get the Protocol struct from the map
 | 
			
		||||
			protoCfg := val.Field(i).MapIndex(reflect.ValueOf(dest.Name))
 | 
			
		||||
			//config.SetNickFormat(msg, protoCfg.Interface().(config.Protocol))
 | 
			
		||||
			val.Field(i).SetMapIndex(reflect.ValueOf(dest.Name), protoCfg)
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) modifyUsername(msg *config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	br := gw.Bridges[msg.Account]
 | 
			
		||||
	nick := gw.Config.General.RemoteNickFormat
 | 
			
		||||
	if nick == "" {
 | 
			
		||||
		nick = dest.Config.RemoteNickFormat
 | 
			
		||||
	}
 | 
			
		||||
	nick = strings.Replace(nick, "{NICK}", msg.Username, -1)
 | 
			
		||||
	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
 | 
			
		||||
	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
 | 
			
		||||
	msg.Username = nick
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										105
									
								
								gateway/samechannel/samechannel.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										105
									
								
								gateway/samechannel/samechannel.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,105 @@
 | 
			
		||||
package samechannelgateway
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type SameChannelGateway struct {
 | 
			
		||||
	*config.Config
 | 
			
		||||
	MyConfig    *config.SameChannelGateway
 | 
			
		||||
	Bridges     map[string]*bridge.Bridge
 | 
			
		||||
	Channels    []string
 | 
			
		||||
	ignoreNicks map[string][]string
 | 
			
		||||
	Name        string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *config.Config, gateway *config.SameChannelGateway) error {
 | 
			
		||||
	c := make(chan config.Message)
 | 
			
		||||
	gw := &SameChannelGateway{}
 | 
			
		||||
	gw.Bridges = make(map[string]*bridge.Bridge)
 | 
			
		||||
	gw.Name = gateway.Name
 | 
			
		||||
	gw.Config = cfg
 | 
			
		||||
	gw.MyConfig = gateway
 | 
			
		||||
	gw.Channels = gateway.Channels
 | 
			
		||||
	for _, account := range gateway.Accounts {
 | 
			
		||||
		br := config.Bridge{Account: account}
 | 
			
		||||
		log.Infof("Starting bridge: %s", account)
 | 
			
		||||
		gw.Bridges[account] = bridge.New(cfg, &br, c)
 | 
			
		||||
	}
 | 
			
		||||
	for _, br := range gw.Bridges {
 | 
			
		||||
		err := br.Connect()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Bridge %s failed to start: %v", br.Account, err)
 | 
			
		||||
		}
 | 
			
		||||
		for _, channel := range gw.Channels {
 | 
			
		||||
			log.Infof("%s: joining %s", br.Account, channel)
 | 
			
		||||
			br.JoinChannel(channel)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	gw.handleReceive(c)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *SameChannelGateway) handleReceive(c chan config.Message) {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case msg := <-c:
 | 
			
		||||
			if !gw.ignoreMessage(&msg) {
 | 
			
		||||
				for _, br := range gw.Bridges {
 | 
			
		||||
					gw.handleMessage(msg, br)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *SameChannelGateway) handleMessage(msg config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	// is this a configured channel
 | 
			
		||||
	if !gw.validChannel(msg.Channel) {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// do not send the message to the bridge we come from if also the channel is the same
 | 
			
		||||
	if msg.Account == dest.Account {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	gw.modifyUsername(&msg, dest)
 | 
			
		||||
	log.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, msg.Channel, dest.Account, msg.Channel)
 | 
			
		||||
	err := dest.Send(msg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *SameChannelGateway) ignoreMessage(msg *config.Message) bool {
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) {
 | 
			
		||||
		if msg.Username == entry {
 | 
			
		||||
			log.Debugf("ignoring %s from %s", msg.Username, msg.Account)
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *SameChannelGateway) modifyUsername(msg *config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	br := gw.Bridges[msg.Account]
 | 
			
		||||
	nick := gw.Config.General.RemoteNickFormat
 | 
			
		||||
	if nick == "" {
 | 
			
		||||
		nick = dest.Config.RemoteNickFormat
 | 
			
		||||
	}
 | 
			
		||||
	nick = strings.Replace(nick, "{NICK}", msg.Username, -1)
 | 
			
		||||
	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
 | 
			
		||||
	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
 | 
			
		||||
	msg.Username = nick
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *SameChannelGateway) validChannel(channel string) bool {
 | 
			
		||||
	for _, c := range gw.Channels {
 | 
			
		||||
		if c == channel {
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										108
									
								
								hook/rockethook/rockethook.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										108
									
								
								hook/rockethook/rockethook.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,108 @@
 | 
			
		||||
package rockethook
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Message for rocketchat outgoing webhook.
 | 
			
		||||
type Message struct {
 | 
			
		||||
	Token       string `json:"token"`
 | 
			
		||||
	ChannelID   string `json:"channel_id"`
 | 
			
		||||
	ChannelName string `json:"channel_name"`
 | 
			
		||||
	Timestamp   string `json:"timestamp"`
 | 
			
		||||
	UserID      string `json:"user_id"`
 | 
			
		||||
	UserName    string `json:"user_name"`
 | 
			
		||||
	Text        string `json:"text"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Client for Rocketchat.
 | 
			
		||||
type Client struct {
 | 
			
		||||
	In         chan Message
 | 
			
		||||
	httpclient *http.Client
 | 
			
		||||
	Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Config for client.
 | 
			
		||||
type Config struct {
 | 
			
		||||
	BindAddress        string // Address to listen on
 | 
			
		||||
	Token              string // Only allow this token from Rocketchat. (Allow everything when empty)
 | 
			
		||||
	InsecureSkipVerify bool   // disable certificate checking
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New Rocketchat client.
 | 
			
		||||
func New(url string, config Config) *Client {
 | 
			
		||||
	c := &Client{In: make(chan Message), Config: config}
 | 
			
		||||
	tr := &http.Transport{
 | 
			
		||||
		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify},
 | 
			
		||||
	}
 | 
			
		||||
	c.httpclient = &http.Client{Transport: tr}
 | 
			
		||||
	_, _, err := net.SplitHostPort(c.BindAddress)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatalf("incorrect bindaddress %s", c.BindAddress)
 | 
			
		||||
	}
 | 
			
		||||
	go c.StartServer()
 | 
			
		||||
	return c
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// StartServer starts a webserver listening for incoming mattermost POSTS.
 | 
			
		||||
func (c *Client) StartServer() {
 | 
			
		||||
	mux := http.NewServeMux()
 | 
			
		||||
	mux.Handle("/", c)
 | 
			
		||||
	log.Printf("Listening on http://%v...\n", c.BindAddress)
 | 
			
		||||
	if err := http.ListenAndServe(c.BindAddress, mux); err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ServeHTTP implementation.
 | 
			
		||||
func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
	if r.Method != "POST" {
 | 
			
		||||
		log.Println("invalid " + r.Method + " connection from " + r.RemoteAddr)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	msg := Message{}
 | 
			
		||||
	body, err := ioutil.ReadAll(r.Body)
 | 
			
		||||
	log.Println(string(body))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Println(err)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	defer r.Body.Close()
 | 
			
		||||
	err = json.Unmarshal(body, &msg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Println(err)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if msg.Token == "" {
 | 
			
		||||
		log.Println("no token from " + r.RemoteAddr)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	msg.ChannelName = "#" + msg.ChannelName
 | 
			
		||||
	if c.Token != "" {
 | 
			
		||||
		if msg.Token != c.Token {
 | 
			
		||||
			log.Println("invalid token " + msg.Token + " from " + r.RemoteAddr)
 | 
			
		||||
			http.NotFound(w, r)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	c.In <- msg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Receive returns an incoming message from mattermost outgoing webhooks URL.
 | 
			
		||||
func (c *Client) Receive() Message {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case msg := <-c.In:
 | 
			
		||||
			return msg
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@@ -1,11 +0,0 @@
 | 
			
		||||
[IRC]
 | 
			
		||||
server="irc.freenode.net"
 | 
			
		||||
port=6667
 | 
			
		||||
UseTLS=false
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
nick="matterbot"
 | 
			
		||||
channel="#matterbridge"
 | 
			
		||||
 | 
			
		||||
[mattermost]
 | 
			
		||||
url="http://yourdomain/hooks/yourhookkey"
 | 
			
		||||
port=9999
 | 
			
		||||
@@ -1,56 +1,59 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	"github.com/thoj/go-ircevent"
 | 
			
		||||
	"log"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"time"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/gateway"
 | 
			
		||||
	"github.com/42wim/matterbridge/gateway/samechannel"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
	i *irc.Connection
 | 
			
		||||
	m *matterhook.Client
 | 
			
		||||
	*Config
 | 
			
		||||
}
 | 
			
		||||
var version = "0.9.2"
 | 
			
		||||
 | 
			
		||||
func NewBridge(name string, config *Config) *Bridge {
 | 
			
		||||
	b := &Bridge{}
 | 
			
		||||
	b.Config = config
 | 
			
		||||
	b.m = matterhook.New(b.Config.Mattermost.URL, matterhook.Config{Port: b.Config.Mattermost.Port})
 | 
			
		||||
	b.i = b.createIRC(name)
 | 
			
		||||
	go b.handleMatter()
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) createIRC(name string) *irc.Connection {
 | 
			
		||||
	i := irc.IRC(b.Config.IRC.Nick, b.Config.IRC.Nick)
 | 
			
		||||
	i.UseTLS = b.Config.IRC.UseTLS
 | 
			
		||||
	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.IRC.SkipTLSVerify}
 | 
			
		||||
	i.Connect(b.Config.IRC.Server + ":" + strconv.Itoa(b.Config.IRC.Port))
 | 
			
		||||
	time.Sleep(time.Second)
 | 
			
		||||
	log.Println("Joining", b.Config.IRC.Channel, "as", b.Config.IRC.Nick)
 | 
			
		||||
	i.Join(b.Config.IRC.Channel)
 | 
			
		||||
	i.AddCallback("PRIVMSG", b.handlePrivMsg)
 | 
			
		||||
	return i
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handlePrivMsg(event *irc.Event) {
 | 
			
		||||
	matterMessage := matterhook.OMessage{}
 | 
			
		||||
	matterMessage.Text = event.Message()
 | 
			
		||||
	matterMessage.UserName = "irc-" + event.Nick
 | 
			
		||||
	b.m.Send(matterMessage)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleMatter() {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.m.Receive()
 | 
			
		||||
		b.i.Privmsg(b.Config.IRC.Channel, message.UserName+": "+message.Text)
 | 
			
		||||
	}
 | 
			
		||||
func init() {
 | 
			
		||||
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	NewBridge("matterbot", NewConfig("matterbridge.conf"))
 | 
			
		||||
	flagConfig := flag.String("conf", "matterbridge.toml", "config file")
 | 
			
		||||
	flagDebug := flag.Bool("debug", false, "enable debug")
 | 
			
		||||
	flagVersion := flag.Bool("version", false, "show version")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
	if *flagVersion {
 | 
			
		||||
		fmt.Println("version:", version)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
	if *flagDebug {
 | 
			
		||||
		log.Info("enabling debug")
 | 
			
		||||
		log.SetLevel(log.DebugLevel)
 | 
			
		||||
	}
 | 
			
		||||
	fmt.Println("running version", version)
 | 
			
		||||
	cfg := config.NewConfig(*flagConfig)
 | 
			
		||||
	for _, gw := range cfg.SameChannelGateway {
 | 
			
		||||
		if !gw.Enable {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		fmt.Printf("starting samechannel gateway %#v\n", gw.Name)
 | 
			
		||||
		go func(gw config.SameChannelGateway) {
 | 
			
		||||
			err := samechannelgateway.New(cfg, &gw)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				log.Fatalf("starting gateway failed %#v", err)
 | 
			
		||||
			}
 | 
			
		||||
		}(gw)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, gw := range cfg.Gateway {
 | 
			
		||||
		if !gw.Enable {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		fmt.Printf("starting gateway %#v\n", gw.Name)
 | 
			
		||||
		g := gateway.New(cfg, &gw)
 | 
			
		||||
		err := g.Start()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("starting gateway failed %#v", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	select {}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										579
									
								
								matterbridge.toml.sample
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										579
									
								
								matterbridge.toml.sample
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,579 @@
 | 
			
		||||
#This is configuration for matterbridge.
 | 
			
		||||
###################################################################
 | 
			
		||||
#IRC section
 | 
			
		||||
###################################################################
 | 
			
		||||
#REQUIRED to start IRC section
 | 
			
		||||
[irc]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[irc.name]" or "[irc.name2]"
 | 
			
		||||
#In this example we use [irc.freenode]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[irc.freenode]
 | 
			
		||||
#irc server to connect to. 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Server="irc.freenode.net:6667"
 | 
			
		||||
 | 
			
		||||
#Password for irc server (if necessary)
 | 
			
		||||
#OPTIONAL (default "")
 | 
			
		||||
Password=""
 | 
			
		||||
 | 
			
		||||
#Enable to use TLS connection to your irc server. 
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
UseTLS=false
 | 
			
		||||
 | 
			
		||||
#Enable SASL (PLAIN) authentication. (freenode requires this from eg AWS hosts)
 | 
			
		||||
#It uses NickServNick and NickServPassword as login and password
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
UseSASL=false
 | 
			
		||||
 | 
			
		||||
#Enable to not verify the certificate on your irc server. i
 | 
			
		||||
#e.g. when using selfsigned certificates
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
#Your nick on irc. 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="matterbot"
 | 
			
		||||
 | 
			
		||||
#If you registered your bot with a service like Nickserv on freenode. 
 | 
			
		||||
#Also being used when UseSASL=true
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
NickServNick="nickserv"
 | 
			
		||||
NickServPassword="secret"
 | 
			
		||||
 | 
			
		||||
#Flood control
 | 
			
		||||
#Delay in milliseconds between each message send to the IRC server
 | 
			
		||||
#OPTIONAL (default 1300)
 | 
			
		||||
MessageDelay=1300
 | 
			
		||||
 | 
			
		||||
#Maximum amount of messages to hold in queue. If queue is full 
 | 
			
		||||
#messages will be dropped. 
 | 
			
		||||
#<clipped> will be add to the message that fills the queue.
 | 
			
		||||
#OPTIONAL (default 30)
 | 
			
		||||
MessageQueue=30
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#XMPP section
 | 
			
		||||
###################################################################
 | 
			
		||||
[xmpp]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[xmpp.name]" or "[xmpp.name2]"
 | 
			
		||||
#In this example we use [xmpp.jabber]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[xmpp.jabber]
 | 
			
		||||
#xmpp server to connect to. 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Server="jabber.example.com:5222"
 | 
			
		||||
 | 
			
		||||
#Jid
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Jid="user@example.com"
 | 
			
		||||
 | 
			
		||||
#Password
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Password="yourpass"
 | 
			
		||||
 | 
			
		||||
#MUC
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Muc="conference.jabber.example.com"
 | 
			
		||||
 | 
			
		||||
#Your nick in the rooms
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="xmppbot"
 | 
			
		||||
 | 
			
		||||
#Enable to not verify the certificate on your xmpp server.
 | 
			
		||||
#e.g. when using selfsigned certificates
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#hipchat section
 | 
			
		||||
###################################################################
 | 
			
		||||
#Go to https://www.hipchat.com/account/xmpp this will show you the necessary data
 | 
			
		||||
#to fill in the section below
 | 
			
		||||
[xmpp.hipchat]
 | 
			
		||||
#xmpp server to connect to. 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Server="chat.hipchat.com:5222"
 | 
			
		||||
 | 
			
		||||
#Jabber ID
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Jid="12345_12345@chat.hipchat.com"
 | 
			
		||||
 | 
			
		||||
#Password (your hipchat password)
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Password="yourpass"
 | 
			
		||||
 | 
			
		||||
#Conference (MUC) domain
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Muc="conf.hipchat.com"
 | 
			
		||||
 | 
			
		||||
#Room nickname
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="yourlogin"
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#mattermost section
 | 
			
		||||
###################################################################
 | 
			
		||||
[mattermost]
 | 
			
		||||
#You can configure multiple servers "[mattermost.name]" or "[mattermost.name2]"
 | 
			
		||||
#In this example we use [mattermost.work]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
 | 
			
		||||
[mattermost.work]
 | 
			
		||||
#### Settings for webhook matterbridge.
 | 
			
		||||
#### These settings will not be used when useAPI is enabled
 | 
			
		||||
 | 
			
		||||
#Url is your incoming webhook url as specified in mattermost. 
 | 
			
		||||
#See account settings - integrations - incoming webhooks on mattermost.
 | 
			
		||||
#REQUIRED (unless useAPI=true)
 | 
			
		||||
URL="https://yourdomain/hooks/yourhookkey"
 | 
			
		||||
 | 
			
		||||
#Address to listen on for outgoing webhook requests from mattermost.
 | 
			
		||||
#See account settings - integrations - outgoing webhooks on mattermost.
 | 
			
		||||
#This setting will not be used when using -plus switch which doesn't use 
 | 
			
		||||
#webhooks
 | 
			
		||||
#REQUIRED (unless useAPI=true)
 | 
			
		||||
BindAddress="0.0.0.0:9999"
 | 
			
		||||
 | 
			
		||||
#Icon that will be showed in mattermost. 
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IconURL="http://youricon.png"
 | 
			
		||||
 | 
			
		||||
#### Settings for matterbridge -plus
 | 
			
		||||
#### Thse settings will only be used when using the -plus switch.
 | 
			
		||||
 | 
			
		||||
#### Settings for using matterbridge API
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
useAPI=false
 | 
			
		||||
 | 
			
		||||
#The mattermost hostname. 
 | 
			
		||||
#REQUIRED (when useAPI=true)
 | 
			
		||||
Server="yourmattermostserver.domain"
 | 
			
		||||
 | 
			
		||||
#Your team on mattermost. 
 | 
			
		||||
#REQUIRED (when useAPI=true)
 | 
			
		||||
Team="yourteam"
 | 
			
		||||
 | 
			
		||||
#login/pass of your bot. 
 | 
			
		||||
#Use a dedicated user for this and not your own! 
 | 
			
		||||
#REQUIRED (when useAPI=true)
 | 
			
		||||
Login="yourlogin"
 | 
			
		||||
Password="yourpass"
 | 
			
		||||
 | 
			
		||||
#Enable this to make a http connection (instead of https) to your mattermost. 
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoTLS=false
 | 
			
		||||
 | 
			
		||||
#### Shared settings for matterbridge and -plus
 | 
			
		||||
 | 
			
		||||
#Enable to not verify the certificate on your mattermost server. 
 | 
			
		||||
#e.g. when using selfsigned certificates
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
#how to format the list of IRC nicks when displayed in mattermost. 
 | 
			
		||||
#Possible options are "table" and "plain"
 | 
			
		||||
#OPTIONAL (default plain)
 | 
			
		||||
NickFormatter="plain"
 | 
			
		||||
#How many nicks to list per row for formatters that support this. 
 | 
			
		||||
#OPTIONAL (default 4)
 | 
			
		||||
NicksPerRow=4
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to mattermost with the sender's nick. 
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#mattermost server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
#from bridge to Mattermost will by default be prefixed by "bridge-" + nick. You can, 
 | 
			
		||||
#however, modify how the messages appear, by setting (and modifying) RemoteNickFormat 
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
PrefixMessagesWithNick=false
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#Gitter section
 | 
			
		||||
#Best to make a dedicated gitter account for the bot.
 | 
			
		||||
###################################################################
 | 
			
		||||
 | 
			
		||||
[gitter]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[gitter.name]" or "[gitter.name2]"
 | 
			
		||||
#In this example we use [gitter.myproject]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[gitter.myproject]
 | 
			
		||||
#Token to connect with Gitter API
 | 
			
		||||
#You can get your token by going to https://developer.gitter.im/docs/welcome and SIGN IN
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#slack section
 | 
			
		||||
###################################################################
 | 
			
		||||
[slack]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[slack.name]" or "[slack.name2]"
 | 
			
		||||
#In this example we use [slack.hobby]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[slack.hobby]
 | 
			
		||||
#### Settings for webhook matterbridge.
 | 
			
		||||
#### These settings will not be used when useAPI is enabled
 | 
			
		||||
 | 
			
		||||
#Url is your incoming webhook url as specified in slack
 | 
			
		||||
#See account settings - integrations - incoming webhooks on slack
 | 
			
		||||
#REQUIRED (unless useAPI=true)
 | 
			
		||||
URL="https://hooks.slack.com/services/yourhook"
 | 
			
		||||
 | 
			
		||||
#Address to listen on for outgoing webhook requests from slack
 | 
			
		||||
#See account settings - integrations - outgoing webhooks on slack
 | 
			
		||||
#This setting will not be used when useAPI is eanbled
 | 
			
		||||
#webhooks
 | 
			
		||||
#REQUIRED (unless useAPI=true)
 | 
			
		||||
BindAddress="0.0.0.0:9999"
 | 
			
		||||
 | 
			
		||||
#### Settings for using slack API
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
useAPI=false
 | 
			
		||||
 | 
			
		||||
#Token to connect with the Slack API
 | 
			
		||||
#You'll have to use a test/api-token using a dedicated user and not a bot token.
 | 
			
		||||
#See https://github.com/42wim/matterbridge/issues/75 for more info.
 | 
			
		||||
#REQUIRED (when useAPI=true)
 | 
			
		||||
Token="yourslacktoken"
 | 
			
		||||
 | 
			
		||||
#### Shared settings for webhooks and API
 | 
			
		||||
 | 
			
		||||
#Icon that will be showed in slack
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IconURL="https://robohash.org/{NICK}.png?size=48x48"
 | 
			
		||||
 | 
			
		||||
#how to format the list of IRC nicks when displayed in slack
 | 
			
		||||
#Possible options are "table" and "plain"
 | 
			
		||||
#OPTIONAL (default plain)
 | 
			
		||||
NickFormatter="plain"
 | 
			
		||||
#How many nicks to list per row for formatters that support this. 
 | 
			
		||||
#OPTIONAL (default 4)
 | 
			
		||||
NicksPerRow=4
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to mattermost with RemoteNickFormat
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#slack server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
#from bridge to Slack will by default be prefixed by "bridge-" + nick. You can, 
 | 
			
		||||
#however, modify how the messages appear, by setting (and modifying) RemoteNickFormat 
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
PrefixMessagesWithNick=false
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#discord section
 | 
			
		||||
###################################################################
 | 
			
		||||
[discord]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[discord.name]" or "[discord.name2]"
 | 
			
		||||
#In this example we use [discord.game]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[discord.game]
 | 
			
		||||
#Token to connect with Discord API
 | 
			
		||||
#You can get your token by following the instructions on 
 | 
			
		||||
#https://github.com/reactiflux/discord-irc/wiki/Creating-a-discord-bot-&-getting-a-token
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Server="yourservername"
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#telegram section
 | 
			
		||||
###################################################################
 | 
			
		||||
[telegram]
 | 
			
		||||
 | 
			
		||||
#You can configure multiple servers "[telegram.name]" or "[telegram.name2]"
 | 
			
		||||
#In this example we use [telegram.secure]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
[telegram.secure]
 | 
			
		||||
#Token to connect with telegram API
 | 
			
		||||
#See https://core.telegram.org/bots#6-botfather and https://www.linkedin.com/pulse/telegram-bots-beginners-marco-frau
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#rocketchat section
 | 
			
		||||
###################################################################
 | 
			
		||||
[rocketchat]
 | 
			
		||||
#You can configure multiple servers "[rocketchat.name]" or "[rocketchat.name2]"
 | 
			
		||||
#In this example we use [rocketchat.work]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
 | 
			
		||||
[rocketchat.rockme]
 | 
			
		||||
#Url is your incoming webhook url as specified in rocketchat
 | 
			
		||||
#Read #https://rocket.chat/docs/administrator-guides/integrations/#how-to-create-a-new-incoming-webhook
 | 
			
		||||
#See administration - integrations - new integration - incoming webhook
 | 
			
		||||
#REQUIRED
 | 
			
		||||
URL="https://yourdomain/hooks/yourhookkey"
 | 
			
		||||
 | 
			
		||||
#Address to listen on for outgoing webhook requests from rocketchat.
 | 
			
		||||
#See administration - integrations - new integration - outgoing webhook
 | 
			
		||||
#REQUIRED 
 | 
			
		||||
BindAddress="0.0.0.0:9999"
 | 
			
		||||
 | 
			
		||||
#Your nick/username as specified in your incoming webhook "Post as" setting
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="matterbot"
 | 
			
		||||
 | 
			
		||||
#Enable this to make a http connection (instead of https) to your rocketchat
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoTLS=false
 | 
			
		||||
 | 
			
		||||
#Enable to not verify the certificate on your rocketchat server. 
 | 
			
		||||
#e.g. when using selfsigned certificates
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to rocketchat with the sender's nick. 
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#rocketchat server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
#from bridge to rocketchat will by default be prefixed by the RemoteNickFormat setting. i
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
PrefixMessagesWithNick=false
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges (only from irc-bridge at the moment)
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#General configuration
 | 
			
		||||
###################################################################
 | 
			
		||||
#Settings here override specific settings for each protocol
 | 
			
		||||
[general]
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#Gateway configuration
 | 
			
		||||
###################################################################
 | 
			
		||||
 | 
			
		||||
#You can specify multiple gateways using [[gateway]]
 | 
			
		||||
#Each gateway has a [[gateway.in]] and a [[gateway.out]]
 | 
			
		||||
#[[gateway.in]] specifies the account and channels we will receive messages from.
 | 
			
		||||
#[[gateway.out]] specifies the account and channels we will send the messages
 | 
			
		||||
#from [[gateway.in]] to.
 | 
			
		||||
#
 | 
			
		||||
#Most of the time [[gateway.in]] and [[gateway.out]] are the same if you 
 | 
			
		||||
#want bidirectional bridging. You can then use [[gateway.inout]]
 | 
			
		||||
#
 | 
			
		||||
 | 
			
		||||
[[gateway]]
 | 
			
		||||
#OPTIONAL (not used for now)
 | 
			
		||||
name="gateway1"
 | 
			
		||||
#Enable enables this gateway
 | 
			
		||||
##OPTIONAL (default false)
 | 
			
		||||
enable=true
 | 
			
		||||
 | 
			
		||||
    #[[gateway.in]] specifies the account and channels we will receive messages from.
 | 
			
		||||
    #The following example bridges between mattermost and irc
 | 
			
		||||
    [[gateway.in]]
 | 
			
		||||
 | 
			
		||||
    #account specified above
 | 
			
		||||
    #REQUIRED
 | 
			
		||||
    account="irc.freenode"
 | 
			
		||||
    #channel to connect on that account
 | 
			
		||||
    #How to specify them for the different bridges:
 | 
			
		||||
    #
 | 
			
		||||
    #irc        - #channel (# is required)
 | 
			
		||||
    #mattermost - channel (the channel name as seen in the URL, not the displayname)
 | 
			
		||||
    #gitter     - username/room 
 | 
			
		||||
    #xmpp       - channel
 | 
			
		||||
    #slack      - channel (the channel name as seen in the URL, not the displayname)
 | 
			
		||||
    #discord    - channel (without the #)
 | 
			
		||||
    #           - ID:123456789 (where 123456789 is the channel ID) 
 | 
			
		||||
    #               (https://github.com/42wim/matterbridge/issues/57)
 | 
			
		||||
    #telegram   - chatid (a large negative number, eg -123456789)
 | 
			
		||||
    #             see (https://www.linkedin.com/pulse/telegram-bots-beginners-marco-frau)
 | 
			
		||||
    #hipchat    - id_channel (see https://www.hipchat.com/account/xmpp for the correct channel)
 | 
			
		||||
    #rocketchat - #channel (# is required)
 | 
			
		||||
    #REQUIRED
 | 
			
		||||
    channel="#testing"
 | 
			
		||||
 | 
			
		||||
        #OPTIONAL - only used for IRC protocol at the moment
 | 
			
		||||
        [gateway.in.options]
 | 
			
		||||
        #OPTIONAL - your irc channel key
 | 
			
		||||
        key="yourkey"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    #[[gateway.out]] specifies the account and channels we will sent messages to.
 | 
			
		||||
    [[gateway.out]]
 | 
			
		||||
    account="irc.freenode"
 | 
			
		||||
    channel="#testing"
 | 
			
		||||
 | 
			
		||||
        #OPTIONAL - only used for IRC protocol at the moment
 | 
			
		||||
        [gateway.out.options]
 | 
			
		||||
        #OPTIONAL - your irc channel key
 | 
			
		||||
        key="yourkey"
 | 
			
		||||
 | 
			
		||||
    #[[gateway.inout]] can be used when then channel will be used to receive from 
 | 
			
		||||
    #and send messages to
 | 
			
		||||
    [[gateway.inout]]
 | 
			
		||||
    account="mattermost.work"
 | 
			
		||||
    channel="off-topic"
 | 
			
		||||
 | 
			
		||||
        #OPTIONAL - only used for IRC protocol at the moment
 | 
			
		||||
        [gateway.inout.options]
 | 
			
		||||
        #OPTIONAL - your irc channel key
 | 
			
		||||
        key="yourkey"
 | 
			
		||||
 | 
			
		||||
#If you want to do a 1:1 mapping between protocols where the channelnames are the same
 | 
			
		||||
#e.g. slack and mattermost you can use the samechannelgateway configuration
 | 
			
		||||
#the example configuration below send messages from channel testing on mattermost to
 | 
			
		||||
#channel testing on slack and vice versa. (and for the channel testing2 and testing3)
 | 
			
		||||
 | 
			
		||||
[[samechannelgateway]]
 | 
			
		||||
   enable = false
 | 
			
		||||
   accounts = [ "mattermost.work","slack.hobby" ]
 | 
			
		||||
   channels = [ "testing","testing2","testing3"]
 | 
			
		||||
							
								
								
									
										32
									
								
								matterbridge.toml.simple
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								matterbridge.toml.simple
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,32 @@
 | 
			
		||||
[irc]
 | 
			
		||||
    [irc.freenode]
 | 
			
		||||
    Server="irc.freenode.net:6667"
 | 
			
		||||
    Nick="matterbot"
 | 
			
		||||
 | 
			
		||||
[mattermost]
 | 
			
		||||
    [mattermost.work]
 | 
			
		||||
    useAPI=true
 | 
			
		||||
    Server="yourmattermostserver.domain"
 | 
			
		||||
    Team="yourteam"
 | 
			
		||||
    Login="yourlogin"
 | 
			
		||||
    Password="yourpass"
 | 
			
		||||
    PrefixMessagesWithNick=true
 | 
			
		||||
 | 
			
		||||
[[gateway]]
 | 
			
		||||
name="gateway1"
 | 
			
		||||
enable=true
 | 
			
		||||
    [[gateway.in]]
 | 
			
		||||
    account="irc.freenode"
 | 
			
		||||
    channel="#testing"
 | 
			
		||||
 | 
			
		||||
    [[gateway.out]]
 | 
			
		||||
    account="irc.freenode"
 | 
			
		||||
    channel="#testing"
 | 
			
		||||
 | 
			
		||||
    [[gateway.in]]
 | 
			
		||||
    account="mattermost.work"
 | 
			
		||||
    channel="off-topic"
 | 
			
		||||
    
 | 
			
		||||
    [[gateway.out]]
 | 
			
		||||
    account="mattermost.work"
 | 
			
		||||
    channel="off-topic"
 | 
			
		||||
							
								
								
									
										683
									
								
								matterclient/matterclient.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										683
									
								
								matterclient/matterclient.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,683 @@
 | 
			
		||||
package matterclient
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/cookiejar"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
 | 
			
		||||
	"github.com/gorilla/websocket"
 | 
			
		||||
	"github.com/jpillora/backoff"
 | 
			
		||||
	"github.com/mattermost/platform/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Credentials struct {
 | 
			
		||||
	Login         string
 | 
			
		||||
	Team          string
 | 
			
		||||
	Pass          string
 | 
			
		||||
	Server        string
 | 
			
		||||
	NoTLS         bool
 | 
			
		||||
	SkipTLSVerify bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
	Raw      *model.WebSocketEvent
 | 
			
		||||
	Post     *model.Post
 | 
			
		||||
	Team     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	Text     string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Team struct {
 | 
			
		||||
	Team         *model.Team
 | 
			
		||||
	Id           string
 | 
			
		||||
	Channels     *model.ChannelList
 | 
			
		||||
	MoreChannels *model.ChannelList
 | 
			
		||||
	Users        map[string]*model.User
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMClient struct {
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	*Credentials
 | 
			
		||||
	Team        *Team
 | 
			
		||||
	OtherTeams  []*Team
 | 
			
		||||
	Client      *model.Client
 | 
			
		||||
	User        *model.User
 | 
			
		||||
	Users       map[string]*model.User
 | 
			
		||||
	MessageChan chan *Message
 | 
			
		||||
	log         *log.Entry
 | 
			
		||||
	WsClient    *websocket.Conn
 | 
			
		||||
	WsQuit      bool
 | 
			
		||||
	WsAway      bool
 | 
			
		||||
	WsConnected bool
 | 
			
		||||
	WsSequence  int64
 | 
			
		||||
	WsPingChan  chan *model.WebSocketResponse
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(login, pass, team, server string) *MMClient {
 | 
			
		||||
	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server}
 | 
			
		||||
	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)}
 | 
			
		||||
	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"})
 | 
			
		||||
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
 | 
			
		||||
	return mmclient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SetLogLevel(level string) {
 | 
			
		||||
	l, err := log.ParseLevel(level)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.SetLevel(log.InfoLevel)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	log.SetLevel(l)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) Login() error {
 | 
			
		||||
	// check if this is a first connect or a reconnection
 | 
			
		||||
	firstConnection := true
 | 
			
		||||
	if m.WsConnected == true {
 | 
			
		||||
		firstConnection = false
 | 
			
		||||
	}
 | 
			
		||||
	m.WsConnected = false
 | 
			
		||||
	if m.WsQuit {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	b := &backoff.Backoff{
 | 
			
		||||
		Min:    time.Second,
 | 
			
		||||
		Max:    5 * time.Minute,
 | 
			
		||||
		Jitter: true,
 | 
			
		||||
	}
 | 
			
		||||
	uriScheme := "https://"
 | 
			
		||||
	wsScheme := "wss://"
 | 
			
		||||
	if m.NoTLS {
 | 
			
		||||
		uriScheme = "http://"
 | 
			
		||||
		wsScheme = "ws://"
 | 
			
		||||
	}
 | 
			
		||||
	// login to mattermost
 | 
			
		||||
	m.Client = model.NewClient(uriScheme + m.Credentials.Server)
 | 
			
		||||
	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}}
 | 
			
		||||
	var myinfo *model.Result
 | 
			
		||||
	var appErr *model.AppError
 | 
			
		||||
	var logmsg = "trying login"
 | 
			
		||||
	for {
 | 
			
		||||
		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) {
 | 
			
		||||
			m.log.Debugf(logmsg+" with %s", model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=")
 | 
			
		||||
			if len(token) != 2 {
 | 
			
		||||
				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken")
 | 
			
		||||
			}
 | 
			
		||||
			m.Client.HttpClient.Jar = m.createCookieJar(token[1])
 | 
			
		||||
			m.Client.MockSession(token[1])
 | 
			
		||||
			myinfo, appErr = m.Client.GetMe("")
 | 
			
		||||
			if appErr != nil {
 | 
			
		||||
				return errors.New(appErr.DetailedError)
 | 
			
		||||
			}
 | 
			
		||||
			if myinfo.Data.(*model.User) == nil {
 | 
			
		||||
				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass)
 | 
			
		||||
				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			}
 | 
			
		||||
		} else {
 | 
			
		||||
			myinfo, appErr = m.Client.Login(m.Credentials.Login, m.Credentials.Pass)
 | 
			
		||||
		}
 | 
			
		||||
		if appErr != nil {
 | 
			
		||||
			d := b.Duration()
 | 
			
		||||
			m.log.Debug(appErr.DetailedError)
 | 
			
		||||
			if firstConnection {
 | 
			
		||||
				if appErr.Message == "" {
 | 
			
		||||
					return errors.New(appErr.DetailedError)
 | 
			
		||||
				}
 | 
			
		||||
				return errors.New(appErr.Message)
 | 
			
		||||
			}
 | 
			
		||||
			m.log.Debugf("LOGIN: %s, reconnecting in %s", appErr, d)
 | 
			
		||||
			time.Sleep(d)
 | 
			
		||||
			logmsg = "retrying login"
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		break
 | 
			
		||||
	}
 | 
			
		||||
	// reset timer
 | 
			
		||||
	b.Reset()
 | 
			
		||||
 | 
			
		||||
	err := m.initUser()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if m.Team == nil {
 | 
			
		||||
		return errors.New("team not found")
 | 
			
		||||
	}
 | 
			
		||||
	// set our team id as default route
 | 
			
		||||
	m.Client.SetTeamId(m.Team.Id)
 | 
			
		||||
 | 
			
		||||
	// setup websocket connection
 | 
			
		||||
	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX + "/users/websocket"
 | 
			
		||||
	header := http.Header{}
 | 
			
		||||
	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken)
 | 
			
		||||
 | 
			
		||||
	m.log.Debug("WsClient: making connection")
 | 
			
		||||
	for {
 | 
			
		||||
		wsDialer := &websocket.Dialer{Proxy: http.ProxyFromEnvironment, TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}}
 | 
			
		||||
		m.WsClient, _, err = wsDialer.Dial(wsurl, header)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			d := b.Duration()
 | 
			
		||||
			m.log.Debugf("WSS: %s, reconnecting in %s", err, d)
 | 
			
		||||
			time.Sleep(d)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		break
 | 
			
		||||
	}
 | 
			
		||||
	b.Reset()
 | 
			
		||||
 | 
			
		||||
	m.WsSequence = 1
 | 
			
		||||
	m.WsPingChan = make(chan *model.WebSocketResponse)
 | 
			
		||||
	// only start to parse WS messages when login is completely done
 | 
			
		||||
	m.WsConnected = true
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) Logout() error {
 | 
			
		||||
	m.log.Debugf("logout as %s (team: %s) on %s", m.Credentials.Login, m.Credentials.Team, m.Credentials.Server)
 | 
			
		||||
	m.WsQuit = true
 | 
			
		||||
	m.WsClient.Close()
 | 
			
		||||
	m.WsClient.UnderlyingConn().Close()
 | 
			
		||||
	_, err := m.Client.Logout()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) WsReceiver() {
 | 
			
		||||
	for {
 | 
			
		||||
		var rawMsg json.RawMessage
 | 
			
		||||
		var err error
 | 
			
		||||
 | 
			
		||||
		if m.WsQuit {
 | 
			
		||||
			m.log.Debug("exiting WsReceiver")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if !m.WsConnected {
 | 
			
		||||
			time.Sleep(time.Millisecond * 100)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if _, rawMsg, err = m.WsClient.ReadMessage(); err != nil {
 | 
			
		||||
			m.log.Error("error:", err)
 | 
			
		||||
			// reconnect
 | 
			
		||||
			m.Login()
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		var event model.WebSocketEvent
 | 
			
		||||
		if err := json.Unmarshal(rawMsg, &event); err == nil && event.IsValid() {
 | 
			
		||||
			m.log.Debugf("WsReceiver: %#v", event)
 | 
			
		||||
			msg := &Message{Raw: &event, Team: m.Credentials.Team}
 | 
			
		||||
			m.parseMessage(msg)
 | 
			
		||||
			m.MessageChan <- msg
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		var response model.WebSocketResponse
 | 
			
		||||
		if err := json.Unmarshal(rawMsg, &response); err == nil && response.IsValid() {
 | 
			
		||||
			m.log.Debugf("WsReceiver: %#v", response)
 | 
			
		||||
			m.parseResponse(response)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseMessage(rmsg *Message) {
 | 
			
		||||
	switch rmsg.Raw.Event {
 | 
			
		||||
	case model.WEBSOCKET_EVENT_POSTED:
 | 
			
		||||
		m.parseActionPost(rmsg)
 | 
			
		||||
		/*
 | 
			
		||||
			case model.ACTION_USER_REMOVED:
 | 
			
		||||
				m.handleWsActionUserRemoved(&rmsg)
 | 
			
		||||
			case model.ACTION_USER_ADDED:
 | 
			
		||||
				m.handleWsActionUserAdded(&rmsg)
 | 
			
		||||
		*/
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) {
 | 
			
		||||
	if rmsg.Data != nil {
 | 
			
		||||
		// ping reply
 | 
			
		||||
		if rmsg.Data["text"].(string) == "pong" {
 | 
			
		||||
			m.WsPingChan <- &rmsg
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseActionPost(rmsg *Message) {
 | 
			
		||||
	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string)))
 | 
			
		||||
	// we don't have the user, refresh the userlist
 | 
			
		||||
	if m.GetUser(data.UserId) == nil {
 | 
			
		||||
		m.UpdateUsers()
 | 
			
		||||
	}
 | 
			
		||||
	rmsg.Username = m.GetUser(data.UserId).Username
 | 
			
		||||
	rmsg.Channel = m.GetChannelName(data.ChannelId)
 | 
			
		||||
	rmsg.Team = m.GetTeamName(rmsg.Raw.Data["team_id"].(string))
 | 
			
		||||
	// direct message
 | 
			
		||||
	if rmsg.Raw.Data["channel_type"] == "D" {
 | 
			
		||||
		rmsg.Channel = m.GetUser(data.UserId).Username
 | 
			
		||||
	}
 | 
			
		||||
	rmsg.Text = data.Message
 | 
			
		||||
	rmsg.Post = data
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateUsers() error {
 | 
			
		||||
	mmusers, err := m.Client.GetProfiles(0, 50000, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Users = mmusers.Data.(map[string]*model.User)
 | 
			
		||||
	m.Unlock()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannels() error {
 | 
			
		||||
	mmchannels, err := m.Client.GetChannels("")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	mmchannels2, err := m.Client.GetMoreChannels("")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Team.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	m.Team.MoreChannels = mmchannels2.Data.(*model.ChannelList)
 | 
			
		||||
	m.Unlock()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelName(channelId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		for _, channel := range append(*t.Channels, *t.MoreChannels...) {
 | 
			
		||||
			if channel.Id == channelId {
 | 
			
		||||
				return channel.Name
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelId(name string, teamId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	if teamId == "" {
 | 
			
		||||
		teamId = m.Team.Id
 | 
			
		||||
	}
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		if t.Id == teamId {
 | 
			
		||||
			for _, channel := range append(*t.Channels, *t.MoreChannels...) {
 | 
			
		||||
				if channel.Name == name {
 | 
			
		||||
					return channel.Id
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelHeader(channelId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		for _, channel := range append(*t.Channels, *t.MoreChannels...) {
 | 
			
		||||
			if channel.Id == channelId {
 | 
			
		||||
				return channel.Header
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) PostMessage(channelId string, text string) {
 | 
			
		||||
	post := &model.Post{ChannelId: channelId, Message: text}
 | 
			
		||||
	m.Client.CreatePost(post)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) JoinChannel(channelId string) error {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, c := range *m.Team.Channels {
 | 
			
		||||
		if c.Id == channelId {
 | 
			
		||||
			m.log.Debug("Not joining ", channelId, " already joined.")
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	m.log.Debug("Joining ", channelId)
 | 
			
		||||
	_, err := m.Client.JoinChannel(channelId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New("failed to join")
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPostsSince(channelId, time)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SearchPosts(query string) *model.PostList {
 | 
			
		||||
	res, err := m.Client.SearchPosts(query, false)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPosts(channelId, 0, limit, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPublicLink(filename string) string {
 | 
			
		||||
	res, err := m.Client.GetPublicLink(filename)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	return res
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPublicLinks(filenames []string) []string {
 | 
			
		||||
	var output []string
 | 
			
		||||
	for _, f := range filenames {
 | 
			
		||||
		res, err := m.Client.GetPublicLink(f)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		output = append(output, res)
 | 
			
		||||
	}
 | 
			
		||||
	return output
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannelHeader(channelId string, header string) {
 | 
			
		||||
	data := make(map[string]string)
 | 
			
		||||
	data["channel_id"] = channelId
 | 
			
		||||
	data["channel_header"] = header
 | 
			
		||||
	m.log.Debugf("updating channelheader %#v, %#v", channelId, header)
 | 
			
		||||
	_, err := m.Client.UpdateChannelHeader(data)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateLastViewed(channelId string) {
 | 
			
		||||
	m.log.Debugf("posting lastview %#v", channelId)
 | 
			
		||||
	_, err := m.Client.UpdateLastViewedAt(channelId, true)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Error(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UsernamesInChannel(channelId string) []string {
 | 
			
		||||
	res, err := m.Client.GetMyChannelMembers()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, err)
 | 
			
		||||
		return []string{}
 | 
			
		||||
	}
 | 
			
		||||
	members := res.Data.(*model.ChannelMembers)
 | 
			
		||||
	result := []string{}
 | 
			
		||||
	for _, channel := range *members {
 | 
			
		||||
		if channel.ChannelId == channelId {
 | 
			
		||||
			result = append(result, m.GetUser(channel.UserId).Username)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) createCookieJar(token string) *cookiejar.Jar {
 | 
			
		||||
	var cookies []*http.Cookie
 | 
			
		||||
	jar, _ := cookiejar.New(nil)
 | 
			
		||||
	firstCookie := &http.Cookie{
 | 
			
		||||
		Name:   "MMAUTHTOKEN",
 | 
			
		||||
		Value:  token,
 | 
			
		||||
		Path:   "/",
 | 
			
		||||
		Domain: m.Credentials.Server,
 | 
			
		||||
	}
 | 
			
		||||
	cookies = append(cookies, firstCookie)
 | 
			
		||||
	cookieURL, _ := url.Parse("https://" + m.Credentials.Server)
 | 
			
		||||
	jar.SetCookies(cookieURL, cookies)
 | 
			
		||||
	return jar
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SendDirectMessage sends a direct message to specified user
 | 
			
		||||
func (m *MMClient) SendDirectMessage(toUserId string, msg string) {
 | 
			
		||||
	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg)
 | 
			
		||||
	// create DM channel (only happens on first message)
 | 
			
		||||
	_, err := m.Client.CreateDirectChannel(toUserId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err)
 | 
			
		||||
	}
 | 
			
		||||
	channelName := model.GetDMNameFromIds(toUserId, m.User.Id)
 | 
			
		||||
 | 
			
		||||
	// update our channels
 | 
			
		||||
	mmchannels, _ := m.Client.GetChannels("")
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Team.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	m.Unlock()
 | 
			
		||||
 | 
			
		||||
	// build & send the message
 | 
			
		||||
	msg = strings.Replace(msg, "\r", "", -1)
 | 
			
		||||
	post := &model.Post{ChannelId: m.GetChannelId(channelName, ""), Message: msg}
 | 
			
		||||
	m.Client.CreatePost(post)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetTeamName returns the name of the specified teamId
 | 
			
		||||
func (m *MMClient) GetTeamName(teamId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		if t.Id == teamId {
 | 
			
		||||
			return t.Team.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetChannels returns all channels we're members off
 | 
			
		||||
func (m *MMClient) GetChannels() []*model.Channel {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	// our primary team channels first
 | 
			
		||||
	channels = append(channels, *m.Team.Channels...)
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		if t.Id != m.Team.Id {
 | 
			
		||||
			channels = append(channels, *t.Channels...)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return channels
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetMoreChannels returns existing channels where we're not a member off.
 | 
			
		||||
func (m *MMClient) GetMoreChannels() []*model.Channel {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		channels = append(channels, *t.MoreChannels...)
 | 
			
		||||
	}
 | 
			
		||||
	return channels
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetTeamFromChannel returns teamId belonging to channel (DM channels have no teamId).
 | 
			
		||||
func (m *MMClient) GetTeamFromChannel(channelId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		channels = append(channels, *t.Channels...)
 | 
			
		||||
		channels = append(channels, *t.MoreChannels...)
 | 
			
		||||
		for _, c := range channels {
 | 
			
		||||
			if c.Id == channelId {
 | 
			
		||||
				return t.Id
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetLastViewedAt(channelId string) int64 {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	res, err := m.Client.GetChannel(channelId, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return model.GetMillis()
 | 
			
		||||
	}
 | 
			
		||||
	data := res.Data.(*model.ChannelData)
 | 
			
		||||
	return data.Member.LastViewedAt
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetUsers() map[string]*model.User {
 | 
			
		||||
	users := make(map[string]*model.User)
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for k, v := range m.Users {
 | 
			
		||||
		users[k] = v
 | 
			
		||||
	}
 | 
			
		||||
	return users
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetUser(userId string) *model.User {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	return m.Users[userId]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetStatus(userId string) string {
 | 
			
		||||
	res, err := m.Client.GetStatuses()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	status := res.Data.(map[string]string)
 | 
			
		||||
	if status[userId] == model.STATUS_AWAY {
 | 
			
		||||
		return "away"
 | 
			
		||||
	}
 | 
			
		||||
	if status[userId] == model.STATUS_ONLINE {
 | 
			
		||||
		return "online"
 | 
			
		||||
	}
 | 
			
		||||
	return "offline"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetStatuses() map[string]string {
 | 
			
		||||
	var ok bool
 | 
			
		||||
	statuses := make(map[string]string)
 | 
			
		||||
	res, err := m.Client.GetStatuses()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return statuses
 | 
			
		||||
	}
 | 
			
		||||
	if statuses, ok = res.Data.(map[string]string); ok {
 | 
			
		||||
		for userId, status := range statuses {
 | 
			
		||||
			statuses[userId] = "offline"
 | 
			
		||||
			if status == model.STATUS_AWAY {
 | 
			
		||||
				statuses[userId] = "away"
 | 
			
		||||
			}
 | 
			
		||||
			if status == model.STATUS_ONLINE {
 | 
			
		||||
				statuses[userId] = "online"
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return statuses
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetTeamId() string {
 | 
			
		||||
	return m.Team.Id
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) StatusLoop() {
 | 
			
		||||
	for {
 | 
			
		||||
		if m.WsQuit {
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		if m.WsConnected {
 | 
			
		||||
			m.log.Debug("WS PING")
 | 
			
		||||
			m.sendWSRequest("ping", nil)
 | 
			
		||||
			select {
 | 
			
		||||
			case <-m.WsPingChan:
 | 
			
		||||
				m.log.Debug("WS PONG received")
 | 
			
		||||
			case <-time.After(time.Second * 5):
 | 
			
		||||
				m.Logout()
 | 
			
		||||
				m.WsQuit = false
 | 
			
		||||
				m.Login()
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * 60)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// initialize user and teams
 | 
			
		||||
func (m *MMClient) initUser() error {
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	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.
 | 
			
		||||
	// all other updates are for channels from our (primary) team only.
 | 
			
		||||
	//m.log.Debug("initUser(): loading all team data")
 | 
			
		||||
	for _, v := range initData.Teams {
 | 
			
		||||
		m.Client.SetTeamId(v.Id)
 | 
			
		||||
		mmusers, err := m.Client.GetProfiles(0, 50000, "")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
		}
 | 
			
		||||
		t := &Team{Team: v, Users: mmusers.Data.(map[string]*model.User), Id: v.Id}
 | 
			
		||||
		mmchannels, err := m.Client.GetChannels("")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
		}
 | 
			
		||||
		t.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
		mmchannels, err = m.Client.GetMoreChannels("")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
		}
 | 
			
		||||
		t.MoreChannels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
		m.OtherTeams = append(m.OtherTeams, t)
 | 
			
		||||
		if v.Name == m.Credentials.Team {
 | 
			
		||||
			m.Team = t
 | 
			
		||||
			m.log.Debugf("initUser(): found our team %s (id: %s)", v.Name, v.Id)
 | 
			
		||||
		}
 | 
			
		||||
		// add all users
 | 
			
		||||
		for k, v := range t.Users {
 | 
			
		||||
			m.Users[k] = v
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error {
 | 
			
		||||
	req := &model.WebSocketRequest{}
 | 
			
		||||
	req.Seq = m.WsSequence
 | 
			
		||||
	req.Action = action
 | 
			
		||||
	req.Data = data
 | 
			
		||||
	m.WsSequence++
 | 
			
		||||
	m.log.Debugf("sendWsRequest %#v", req)
 | 
			
		||||
	m.WsClient.WriteJSON(req)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -1,60 +1,80 @@
 | 
			
		||||
//Package matterhook provides interaction with mattermost incoming/outgoing webhooks
 | 
			
		||||
package matterhook
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gorilla/schema"
 | 
			
		||||
	"io"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// OMessage for mattermost incoming webhook. (send to mattermost)
 | 
			
		||||
type OMessage struct {
 | 
			
		||||
	Channel   string `json:"channel,omitempty"`
 | 
			
		||||
	IconURL   string `json:"icon_url,omitempty"`
 | 
			
		||||
	IconEmoji string `json:"icon_emoji,omitempty"`
 | 
			
		||||
	UserName  string `json:"username,omitempty"`
 | 
			
		||||
	Text      string `json:"text"`
 | 
			
		||||
	Channel     string      `json:"channel,omitempty"`
 | 
			
		||||
	IconURL     string      `json:"icon_url,omitempty"`
 | 
			
		||||
	IconEmoji   string      `json:"icon_emoji,omitempty"`
 | 
			
		||||
	UserName    string      `json:"username,omitempty"`
 | 
			
		||||
	Text        string      `json:"text"`
 | 
			
		||||
	Attachments interface{} `json:"attachments,omitempty"`
 | 
			
		||||
	Type        string      `json:"type,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// IMessage for mattermost outgoing webhook. (received from mattermost)
 | 
			
		||||
type IMessage struct {
 | 
			
		||||
	BotID       string `schema:"bot_id"`
 | 
			
		||||
	BotName     string `schema:"bot_name"`
 | 
			
		||||
	Token       string `schema:"token"`
 | 
			
		||||
	TeamID      string `schema:"team_id"`
 | 
			
		||||
	TeamDomain  string `schema:"team_domain"`
 | 
			
		||||
	ChannelID   string `schema:"channel_id"`
 | 
			
		||||
	ServiceID   string `schema:"service_id"`
 | 
			
		||||
	ChannelName string `schema:"channel_name"`
 | 
			
		||||
	Timestamp   string `schema:"timestamp"`
 | 
			
		||||
	UserID      string `schema:"user_id"`
 | 
			
		||||
	UserName    string `schema:"user_name"`
 | 
			
		||||
	PostId      string `schema:"post_id"`
 | 
			
		||||
	RawText     string `schema:"raw_text"`
 | 
			
		||||
	ServiceId   string `schema:"service_id"`
 | 
			
		||||
	Text        string `schema:"text"`
 | 
			
		||||
	TriggerWord string `schema:"trigger_word"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Client for Mattermost.
 | 
			
		||||
type Client struct {
 | 
			
		||||
	url string
 | 
			
		||||
	In  chan IMessage
 | 
			
		||||
	Out chan OMessage
 | 
			
		||||
	Url        string // URL for incoming webhooks on mattermost.
 | 
			
		||||
	In         chan IMessage
 | 
			
		||||
	Out        chan OMessage
 | 
			
		||||
	httpclient *http.Client
 | 
			
		||||
	Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Config for client.
 | 
			
		||||
type Config struct {
 | 
			
		||||
	Port int
 | 
			
		||||
	BindAddress        string // Address to listen on
 | 
			
		||||
	Token              string // Only allow this token from Mattermost. (Allow everything when empty)
 | 
			
		||||
	InsecureSkipVerify bool   // disable certificate checking
 | 
			
		||||
	DisableServer      bool   // Do not start server for outgoing webhooks from Mattermost.
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New Mattermost client.
 | 
			
		||||
func New(url string, config Config) *Client {
 | 
			
		||||
	c := &Client{url: url, In: make(chan IMessage), Out: make(chan OMessage), Config: config}
 | 
			
		||||
	if c.Port == 0 {
 | 
			
		||||
		c.Port = 9999
 | 
			
		||||
	c := &Client{Url: url, In: make(chan IMessage), Out: make(chan OMessage), Config: config}
 | 
			
		||||
	tr := &http.Transport{
 | 
			
		||||
		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify},
 | 
			
		||||
	}
 | 
			
		||||
	c.httpclient = &http.Client{Transport: tr}
 | 
			
		||||
	if !c.DisableServer {
 | 
			
		||||
		_, _, err := net.SplitHostPort(c.BindAddress)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("incorrect bindaddress %s", c.BindAddress)
 | 
			
		||||
		}
 | 
			
		||||
		go c.StartServer()
 | 
			
		||||
	}
 | 
			
		||||
	go c.StartServer()
 | 
			
		||||
	return c
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -62,14 +82,19 @@ func New(url string, config Config) *Client {
 | 
			
		||||
func (c *Client) StartServer() {
 | 
			
		||||
	mux := http.NewServeMux()
 | 
			
		||||
	mux.Handle("/", c)
 | 
			
		||||
	log.Printf("Listening on http://0.0.0.0:%v...\n", c.Port)
 | 
			
		||||
	if err := http.ListenAndServe((":" + strconv.Itoa(c.Port)), mux); err != nil {
 | 
			
		||||
	log.Printf("Listening on http://%v...\n", c.BindAddress)
 | 
			
		||||
	if err := http.ListenAndServe(c.BindAddress, mux); err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ServeHTTP implementation.
 | 
			
		||||
func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
	if r.Method != "POST" {
 | 
			
		||||
		log.Println("invalid " + r.Method + " connection from " + r.RemoteAddr)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	msg := IMessage{}
 | 
			
		||||
	err := r.ParseForm()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -85,6 +110,18 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if msg.Token == "" {
 | 
			
		||||
		log.Println("no token from " + r.RemoteAddr)
 | 
			
		||||
		http.NotFound(w, r)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if c.Token != "" {
 | 
			
		||||
		if msg.Token != c.Token {
 | 
			
		||||
			log.Println("invalid token " + msg.Token + " from " + r.RemoteAddr)
 | 
			
		||||
			http.NotFound(w, r)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	c.In <- msg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -104,7 +141,7 @@ func (c *Client) Send(msg OMessage) error {
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp, err := http.Post(c.url, "application/json", bytes.NewReader(buf))
 | 
			
		||||
	resp, err := c.httpclient.Post(c.Url, "application/json", bytes.NewReader(buf))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										50
									
								
								migration.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								migration.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,50 @@
 | 
			
		||||
# Breaking changes from 0.4 to 0.5 for matterbridge (webhooks version)
 | 
			
		||||
## IRC section
 | 
			
		||||
### Server
 | 
			
		||||
Port removed, added to server
 | 
			
		||||
```
 | 
			
		||||
server="irc.freenode.net"
 | 
			
		||||
port=6667
 | 
			
		||||
```
 | 
			
		||||
changed to
 | 
			
		||||
```
 | 
			
		||||
server="irc.freenode.net:6667"
 | 
			
		||||
```
 | 
			
		||||
### Channel
 | 
			
		||||
Removed see Channels section below
 | 
			
		||||
 | 
			
		||||
### UseSlackCircumfix=true
 | 
			
		||||
Removed, can be done by using ```RemoteNickFormat="<{NICK}> "```
 | 
			
		||||
 | 
			
		||||
## Mattermost section
 | 
			
		||||
### BindAddress
 | 
			
		||||
Port removed, added to BindAddress
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
BindAddress="0.0.0.0"
 | 
			
		||||
port=9999
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
changed to
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
BindAddress="0.0.0.0:9999"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Token
 | 
			
		||||
Removed
 | 
			
		||||
 | 
			
		||||
## Channels section
 | 
			
		||||
```
 | 
			
		||||
[Token "outgoingwebhooktoken1"] 
 | 
			
		||||
IRCChannel="#off-topic"
 | 
			
		||||
MMChannel="off-topic"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
changed to
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
[Channel "channelnameofchoice"] 
 | 
			
		||||
IRC="#off-topic"
 | 
			
		||||
Mattermost="off-topic"
 | 
			
		||||
```
 | 
			
		||||
							
								
								
									
										201
									
								
								vendor/github.com/42wim/go-gitter/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										201
									
								
								vendor/github.com/42wim/go-gitter/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,201 @@
 | 
			
		||||
                                 Apache License
 | 
			
		||||
                           Version 2.0, January 2004
 | 
			
		||||
                        http://www.apache.org/licenses/
 | 
			
		||||
 | 
			
		||||
   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
   1. Definitions.
 | 
			
		||||
 | 
			
		||||
      "License" shall mean the terms and conditions for use, reproduction,
 | 
			
		||||
      and distribution as defined by Sections 1 through 9 of this document.
 | 
			
		||||
 | 
			
		||||
      "Licensor" shall mean the copyright owner or entity authorized by
 | 
			
		||||
      the copyright owner that is granting the License.
 | 
			
		||||
 | 
			
		||||
      "Legal Entity" shall mean the union of the acting entity and all
 | 
			
		||||
      other entities that control, are controlled by, or are under common
 | 
			
		||||
      control with that entity. For the purposes of this definition,
 | 
			
		||||
      "control" means (i) the power, direct or indirect, to cause the
 | 
			
		||||
      direction or management of such entity, whether by contract or
 | 
			
		||||
      otherwise, or (ii) ownership of fifty percent (50%) or more of the
 | 
			
		||||
      outstanding shares, or (iii) beneficial ownership of such entity.
 | 
			
		||||
 | 
			
		||||
      "You" (or "Your") shall mean an individual or Legal Entity
 | 
			
		||||
      exercising permissions granted by this License.
 | 
			
		||||
 | 
			
		||||
      "Source" form shall mean the preferred form for making modifications,
 | 
			
		||||
      including but not limited to software source code, documentation
 | 
			
		||||
      source, and configuration files.
 | 
			
		||||
 | 
			
		||||
      "Object" form shall mean any form resulting from mechanical
 | 
			
		||||
      transformation or translation of a Source form, including but
 | 
			
		||||
      not limited to compiled object code, generated documentation,
 | 
			
		||||
      and conversions to other media types.
 | 
			
		||||
 | 
			
		||||
      "Work" shall mean the work of authorship, whether in Source or
 | 
			
		||||
      Object form, made available under the License, as indicated by a
 | 
			
		||||
      copyright notice that is included in or attached to the work
 | 
			
		||||
      (an example is provided in the Appendix below).
 | 
			
		||||
 | 
			
		||||
      "Derivative Works" shall mean any work, whether in Source or Object
 | 
			
		||||
      form, that is based on (or derived from) the Work and for which the
 | 
			
		||||
      editorial revisions, annotations, elaborations, or other modifications
 | 
			
		||||
      represent, as a whole, an original work of authorship. For the purposes
 | 
			
		||||
      of this License, Derivative Works shall not include works that remain
 | 
			
		||||
      separable from, or merely link (or bind by name) to the interfaces of,
 | 
			
		||||
      the Work and Derivative Works thereof.
 | 
			
		||||
 | 
			
		||||
      "Contribution" shall mean any work of authorship, including
 | 
			
		||||
      the original version of the Work and any modifications or additions
 | 
			
		||||
      to that Work or Derivative Works thereof, that is intentionally
 | 
			
		||||
      submitted to Licensor for inclusion in the Work by the copyright owner
 | 
			
		||||
      or by an individual or Legal Entity authorized to submit on behalf of
 | 
			
		||||
      the copyright owner. For the purposes of this definition, "submitted"
 | 
			
		||||
      means any form of electronic, verbal, or written communication sent
 | 
			
		||||
      to the Licensor or its representatives, including but not limited to
 | 
			
		||||
      communication on electronic mailing lists, source code control systems,
 | 
			
		||||
      and issue tracking systems that are managed by, or on behalf of, the
 | 
			
		||||
      Licensor for the purpose of discussing and improving the Work, but
 | 
			
		||||
      excluding communication that is conspicuously marked or otherwise
 | 
			
		||||
      designated in writing by the copyright owner as "Not a Contribution."
 | 
			
		||||
 | 
			
		||||
      "Contributor" shall mean Licensor and any individual or Legal Entity
 | 
			
		||||
      on behalf of whom a Contribution has been received by Licensor and
 | 
			
		||||
      subsequently incorporated within the Work.
 | 
			
		||||
 | 
			
		||||
   2. Grant of Copyright License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      copyright license to reproduce, prepare Derivative Works of,
 | 
			
		||||
      publicly display, publicly perform, sublicense, and distribute the
 | 
			
		||||
      Work and such Derivative Works in Source or Object form.
 | 
			
		||||
 | 
			
		||||
   3. Grant of Patent License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      (except as stated in this section) patent license to make, have made,
 | 
			
		||||
      use, offer to sell, sell, import, and otherwise transfer the Work,
 | 
			
		||||
      where such license applies only to those patent claims licensable
 | 
			
		||||
      by such Contributor that are necessarily infringed by their
 | 
			
		||||
      Contribution(s) alone or by combination of their Contribution(s)
 | 
			
		||||
      with the Work to which such Contribution(s) was submitted. If You
 | 
			
		||||
      institute patent litigation against any entity (including a
 | 
			
		||||
      cross-claim or counterclaim in a lawsuit) alleging that the Work
 | 
			
		||||
      or a Contribution incorporated within the Work constitutes direct
 | 
			
		||||
      or contributory patent infringement, then any patent licenses
 | 
			
		||||
      granted to You under this License for that Work shall terminate
 | 
			
		||||
      as of the date such litigation is filed.
 | 
			
		||||
 | 
			
		||||
   4. Redistribution. You may reproduce and distribute copies of the
 | 
			
		||||
      Work or Derivative Works thereof in any medium, with or without
 | 
			
		||||
      modifications, and in Source or Object form, provided that You
 | 
			
		||||
      meet the following conditions:
 | 
			
		||||
 | 
			
		||||
      (a) You must give any other recipients of the Work or
 | 
			
		||||
          Derivative Works a copy of this License; and
 | 
			
		||||
 | 
			
		||||
      (b) You must cause any modified files to carry prominent notices
 | 
			
		||||
          stating that You changed the files; and
 | 
			
		||||
 | 
			
		||||
      (c) You must retain, in the Source form of any Derivative Works
 | 
			
		||||
          that You distribute, all copyright, patent, trademark, and
 | 
			
		||||
          attribution notices from the Source form of the Work,
 | 
			
		||||
          excluding those notices that do not pertain to any part of
 | 
			
		||||
          the Derivative Works; and
 | 
			
		||||
 | 
			
		||||
      (d) If the Work includes a "NOTICE" text file as part of its
 | 
			
		||||
          distribution, then any Derivative Works that You distribute must
 | 
			
		||||
          include a readable copy of the attribution notices contained
 | 
			
		||||
          within such NOTICE file, excluding those notices that do not
 | 
			
		||||
          pertain to any part of the Derivative Works, in at least one
 | 
			
		||||
          of the following places: within a NOTICE text file distributed
 | 
			
		||||
          as part of the Derivative Works; within the Source form or
 | 
			
		||||
          documentation, if provided along with the Derivative Works; or,
 | 
			
		||||
          within a display generated by the Derivative Works, if and
 | 
			
		||||
          wherever such third-party notices normally appear. The contents
 | 
			
		||||
          of the NOTICE file are for informational purposes only and
 | 
			
		||||
          do not modify the License. You may add Your own attribution
 | 
			
		||||
          notices within Derivative Works that You distribute, alongside
 | 
			
		||||
          or as an addendum to the NOTICE text from the Work, provided
 | 
			
		||||
          that such additional attribution notices cannot be construed
 | 
			
		||||
          as modifying the License.
 | 
			
		||||
 | 
			
		||||
      You may add Your own copyright statement to Your modifications and
 | 
			
		||||
      may provide additional or different license terms and conditions
 | 
			
		||||
      for use, reproduction, or distribution of Your modifications, or
 | 
			
		||||
      for any such Derivative Works as a whole, provided Your use,
 | 
			
		||||
      reproduction, and distribution of the Work otherwise complies with
 | 
			
		||||
      the conditions stated in this License.
 | 
			
		||||
 | 
			
		||||
   5. Submission of Contributions. Unless You explicitly state otherwise,
 | 
			
		||||
      any Contribution intentionally submitted for inclusion in the Work
 | 
			
		||||
      by You to the Licensor shall be under the terms and conditions of
 | 
			
		||||
      this License, without any additional terms or conditions.
 | 
			
		||||
      Notwithstanding the above, nothing herein shall supersede or modify
 | 
			
		||||
      the terms of any separate license agreement you may have executed
 | 
			
		||||
      with Licensor regarding such Contributions.
 | 
			
		||||
 | 
			
		||||
   6. Trademarks. This License does not grant permission to use the trade
 | 
			
		||||
      names, trademarks, service marks, or product names of the Licensor,
 | 
			
		||||
      except as required for reasonable and customary use in describing the
 | 
			
		||||
      origin of the Work and reproducing the content of the NOTICE file.
 | 
			
		||||
 | 
			
		||||
   7. Disclaimer of Warranty. Unless required by applicable law or
 | 
			
		||||
      agreed to in writing, Licensor provides the Work (and each
 | 
			
		||||
      Contributor provides its Contributions) on an "AS IS" BASIS,
 | 
			
		||||
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 | 
			
		||||
      implied, including, without limitation, any warranties or conditions
 | 
			
		||||
      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
 | 
			
		||||
      PARTICULAR PURPOSE. You are solely responsible for determining the
 | 
			
		||||
      appropriateness of using or redistributing the Work and assume any
 | 
			
		||||
      risks associated with Your exercise of permissions under this License.
 | 
			
		||||
 | 
			
		||||
   8. Limitation of Liability. In no event and under no legal theory,
 | 
			
		||||
      whether in tort (including negligence), contract, or otherwise,
 | 
			
		||||
      unless required by applicable law (such as deliberate and grossly
 | 
			
		||||
      negligent acts) or agreed to in writing, shall any Contributor be
 | 
			
		||||
      liable to You for damages, including any direct, indirect, special,
 | 
			
		||||
      incidental, or consequential damages of any character arising as a
 | 
			
		||||
      result of this License or out of the use or inability to use the
 | 
			
		||||
      Work (including but not limited to damages for loss of goodwill,
 | 
			
		||||
      work stoppage, computer failure or malfunction, or any and all
 | 
			
		||||
      other commercial damages or losses), even if such Contributor
 | 
			
		||||
      has been advised of the possibility of such damages.
 | 
			
		||||
 | 
			
		||||
   9. Accepting Warranty or Additional Liability. While redistributing
 | 
			
		||||
      the Work or Derivative Works thereof, You may choose to offer,
 | 
			
		||||
      and charge a fee for, acceptance of support, warranty, indemnity,
 | 
			
		||||
      or other liability obligations and/or rights consistent with this
 | 
			
		||||
      License. However, in accepting such obligations, You may act only
 | 
			
		||||
      on Your own behalf and on Your sole responsibility, not on behalf
 | 
			
		||||
      of any other Contributor, and only if You agree to indemnify,
 | 
			
		||||
      defend, and hold each Contributor harmless for any liability
 | 
			
		||||
      incurred by, or claims asserted against, such Contributor by reason
 | 
			
		||||
      of your accepting any such warranty or additional liability.
 | 
			
		||||
 | 
			
		||||
   END OF TERMS AND CONDITIONS
 | 
			
		||||
 | 
			
		||||
   APPENDIX: How to apply the Apache License to your work.
 | 
			
		||||
 | 
			
		||||
      To apply the Apache License to your work, attach the following
 | 
			
		||||
      boilerplate notice, with the fields enclosed by brackets "{}"
 | 
			
		||||
      replaced with your own identifying information. (Don't include
 | 
			
		||||
      the brackets!)  The text should be enclosed in the appropriate
 | 
			
		||||
      comment syntax for the file format. We also recommend that a
 | 
			
		||||
      file or class name and description of purpose be included on the
 | 
			
		||||
      same "printed page" as the copyright notice for easier
 | 
			
		||||
      identification within third-party archives.
 | 
			
		||||
 | 
			
		||||
   Copyright {yyyy} {name of copyright owner}
 | 
			
		||||
 | 
			
		||||
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
   you may not use this file except in compliance with the License.
 | 
			
		||||
   You may obtain a copy of the License at
 | 
			
		||||
 | 
			
		||||
       http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 | 
			
		||||
   Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
   See the License for the specific language governing permissions and
 | 
			
		||||
   limitations under the License.
 | 
			
		||||
							
								
								
									
										70
									
								
								vendor/github.com/42wim/go-gitter/faye.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								vendor/github.com/42wim/go-gitter/faye.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,70 @@
 | 
			
		||||
package gitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/mrexodia/wray"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Faye struct {
 | 
			
		||||
	endpoint string
 | 
			
		||||
	Event    chan Event
 | 
			
		||||
	client   *wray.FayeClient
 | 
			
		||||
	gitter   *Gitter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gitter *Gitter) Faye(roomID string) *Faye {
 | 
			
		||||
	wray.RegisterTransports([]wray.Transport{
 | 
			
		||||
		&wray.HttpTransport{
 | 
			
		||||
			SendHook: func(data map[string]interface{}) {
 | 
			
		||||
				if channel, ok := data["channel"]; ok && channel == "/meta/handshake" {
 | 
			
		||||
					data["ext"] = map[string]interface{}{"token": gitter.config.token}
 | 
			
		||||
				}
 | 
			
		||||
			},
 | 
			
		||||
		},
 | 
			
		||||
	})
 | 
			
		||||
	return &Faye{
 | 
			
		||||
		endpoint: "/api/v1/rooms/" + roomID + "/chatMessages",
 | 
			
		||||
		Event:    make(chan Event),
 | 
			
		||||
		client:   wray.NewFayeClient(fayeBaseURL),
 | 
			
		||||
		gitter:   gitter,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (faye *Faye) Listen() {
 | 
			
		||||
	defer faye.destroy()
 | 
			
		||||
 | 
			
		||||
	faye.client.Subscribe(faye.endpoint, false, func(message wray.Message) {
 | 
			
		||||
		dataBytes, err := json.Marshal(message.Data["model"])
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Printf("JSON Marshal error: %v\n", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		var gitterMessage Message
 | 
			
		||||
		err = json.Unmarshal(dataBytes, &gitterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Printf("JSON Unmarshal error: %v\n", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		faye.Event <- Event{
 | 
			
		||||
			Data: &MessageReceived{
 | 
			
		||||
				Message: gitterMessage,
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	//TODO: this might be needed in the future
 | 
			
		||||
	/*go func() {
 | 
			
		||||
		for {
 | 
			
		||||
			faye.client.Publish("/api/v1/ping2", map[string]interface{}{"reason": "ping"})
 | 
			
		||||
			time.Sleep(60 * time.Second)
 | 
			
		||||
		}
 | 
			
		||||
	}()*/
 | 
			
		||||
 | 
			
		||||
	faye.client.Listen()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (faye *Faye) destroy() {
 | 
			
		||||
	close(faye.Event)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										430
									
								
								vendor/github.com/42wim/go-gitter/gitter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										430
									
								
								vendor/github.com/42wim/go-gitter/gitter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,430 @@
 | 
			
		||||
// Package gitter is a Go client library for the Gitter API.
 | 
			
		||||
//
 | 
			
		||||
// Author: sromku
 | 
			
		||||
package gitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/mreiferson/go-httpclient"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	apiBaseURL    = "https://api.gitter.im/v1/"
 | 
			
		||||
	streamBaseURL = "https://stream.gitter.im/v1/"
 | 
			
		||||
	fayeBaseURL   = "https://ws.gitter.im/faye"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Gitter struct {
 | 
			
		||||
	config struct {
 | 
			
		||||
		apiBaseURL    string
 | 
			
		||||
		streamBaseURL string
 | 
			
		||||
		token         string
 | 
			
		||||
		client        *http.Client
 | 
			
		||||
	}
 | 
			
		||||
	debug     bool
 | 
			
		||||
	logWriter io.Writer
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New initializes the Gitter API client
 | 
			
		||||
//
 | 
			
		||||
// For example:
 | 
			
		||||
//  api := gitter.New("YOUR_ACCESS_TOKEN")
 | 
			
		||||
func New(token string) *Gitter {
 | 
			
		||||
 | 
			
		||||
	transport := &httpclient.Transport{
 | 
			
		||||
		ConnectTimeout:   5 * time.Second,
 | 
			
		||||
		ReadWriteTimeout: 40 * time.Second,
 | 
			
		||||
	}
 | 
			
		||||
	defer transport.Close()
 | 
			
		||||
 | 
			
		||||
	s := &Gitter{}
 | 
			
		||||
	s.config.apiBaseURL = apiBaseURL
 | 
			
		||||
	s.config.streamBaseURL = streamBaseURL
 | 
			
		||||
	s.config.token = token
 | 
			
		||||
	s.config.client = &http.Client{
 | 
			
		||||
		Transport: transport,
 | 
			
		||||
	}
 | 
			
		||||
	return s
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetClient sets a custom http client. Can be useful in App Engine case.
 | 
			
		||||
func (gitter *Gitter) SetClient(client *http.Client) {
 | 
			
		||||
	gitter.config.client = client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetUser returns the current user
 | 
			
		||||
func (gitter *Gitter) GetUser() (*User, error) {
 | 
			
		||||
 | 
			
		||||
	var users []User
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "user")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &users)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(users) > 0 {
 | 
			
		||||
		return &users[0], nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = APIError{What: "Failed to retrieve current user"}
 | 
			
		||||
	gitter.log(err)
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetUserRooms returns a list of Rooms the user is part of
 | 
			
		||||
func (gitter *Gitter) GetUserRooms(userID string) ([]Room, error) {
 | 
			
		||||
 | 
			
		||||
	var rooms []Room
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "user/" + userID + "/rooms")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &rooms)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return rooms, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetRooms returns a list of rooms the current user is in
 | 
			
		||||
func (gitter *Gitter) GetRooms() ([]Room, error) {
 | 
			
		||||
 | 
			
		||||
	var rooms []Room
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &rooms)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return rooms, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetUsersInRoom returns the users in the room with the passed id
 | 
			
		||||
func (gitter *Gitter) GetUsersInRoom(roomID string) ([]User, error) {
 | 
			
		||||
 | 
			
		||||
	var users []User
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms/" + roomID + "/users")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &users)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return users, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetRoom returns a room with the passed id
 | 
			
		||||
func (gitter *Gitter) GetRoom(roomID string) (*Room, error) {
 | 
			
		||||
 | 
			
		||||
	var room Room
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms/" + roomID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &room)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return &room, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetMessages returns a list of messages in a room.
 | 
			
		||||
// Pagination is optional. You can pass nil or specific pagination params.
 | 
			
		||||
func (gitter *Gitter) GetMessages(roomID string, params *Pagination) ([]Message, error) {
 | 
			
		||||
 | 
			
		||||
	var messages []Message
 | 
			
		||||
	url := gitter.config.apiBaseURL + "rooms/" + roomID + "/chatMessages"
 | 
			
		||||
	if params != nil {
 | 
			
		||||
		url += "?" + params.encode()
 | 
			
		||||
	}
 | 
			
		||||
	response, err := gitter.get(url)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = json.Unmarshal(response, &messages)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return messages, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetMessage returns a message in a room.
 | 
			
		||||
func (gitter *Gitter) GetMessage(roomID, messageID string) (*Message, error) {
 | 
			
		||||
 | 
			
		||||
	var message Message
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms/" + roomID + "/chatMessages/" + messageID)
 | 
			
		||||
	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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SendMessage sends a message to a room
 | 
			
		||||
func (gitter *Gitter) SendMessage(roomID, text string) error {
 | 
			
		||||
 | 
			
		||||
	message := Message{Text: text}
 | 
			
		||||
	body, _ := json.Marshal(message)
 | 
			
		||||
	_, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// JoinRoom joins a room
 | 
			
		||||
func (gitter *Gitter) JoinRoom(roomID, userID string) (*Room, error) {
 | 
			
		||||
 | 
			
		||||
	message := Room{ID: roomID}
 | 
			
		||||
	body, _ := json.Marshal(message)
 | 
			
		||||
	response, err := gitter.post(gitter.config.apiBaseURL+"user/"+userID+"/rooms", body)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var room Room
 | 
			
		||||
	err = json.Unmarshal(response, &room)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return &room, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// LeaveRoom removes a user from the room
 | 
			
		||||
func (gitter *Gitter) LeaveRoom(roomID, userID string) error {
 | 
			
		||||
 | 
			
		||||
	_, err := gitter.delete(gitter.config.apiBaseURL + "rooms/" + roomID + "/users/" + userID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetDebug traces errors if it's set to true.
 | 
			
		||||
func (gitter *Gitter) SetDebug(debug bool, logWriter io.Writer) {
 | 
			
		||||
	gitter.debug = debug
 | 
			
		||||
	gitter.logWriter = logWriter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Pagination params
 | 
			
		||||
type Pagination struct {
 | 
			
		||||
 | 
			
		||||
	// Skip n messages
 | 
			
		||||
	Skip int
 | 
			
		||||
 | 
			
		||||
	// Get messages before beforeId
 | 
			
		||||
	BeforeID string
 | 
			
		||||
 | 
			
		||||
	// Get messages after afterId
 | 
			
		||||
	AfterID string
 | 
			
		||||
 | 
			
		||||
	// Maximum number of messages to return
 | 
			
		||||
	Limit int
 | 
			
		||||
 | 
			
		||||
	// Search query
 | 
			
		||||
	Query string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (messageParams *Pagination) encode() string {
 | 
			
		||||
	values := url.Values{}
 | 
			
		||||
 | 
			
		||||
	if messageParams.AfterID != "" {
 | 
			
		||||
		values.Add("afterId", messageParams.AfterID)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if messageParams.BeforeID != "" {
 | 
			
		||||
		values.Add("beforeId", messageParams.BeforeID)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if messageParams.Skip > 0 {
 | 
			
		||||
		values.Add("skip", strconv.Itoa(messageParams.Skip))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if messageParams.Limit > 0 {
 | 
			
		||||
		values.Add("limit", strconv.Itoa(messageParams.Limit))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return values.Encode()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gitter *Gitter) getResponse(url string, stream *Stream) (*http.Response, error) {
 | 
			
		||||
	r, err := http.NewRequest("GET", url, nil)
 | 
			
		||||
	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)
 | 
			
		||||
	if stream != nil {
 | 
			
		||||
		stream.streamConnection.request = r
 | 
			
		||||
	}
 | 
			
		||||
	response, err := gitter.config.client.Do(r)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return response, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gitter *Gitter) get(url string) ([]byte, error) {
 | 
			
		||||
	resp, err := gitter.getResponse(url, nil)
 | 
			
		||||
	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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	body, err := ioutil.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return body, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gitter *Gitter) post(url string, body []byte) ([]byte, error) {
 | 
			
		||||
	r, err := http.NewRequest("POST", 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) {
 | 
			
		||||
	r, err := http.NewRequest("delete", url, nil)
 | 
			
		||||
	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) log(a interface{}) {
 | 
			
		||||
	if gitter.debug {
 | 
			
		||||
		log.Println(a)
 | 
			
		||||
		if gitter.logWriter != nil {
 | 
			
		||||
			timestamp := time.Now().Format(time.RFC3339)
 | 
			
		||||
			msg := fmt.Sprintf("%v: %v", timestamp, a)
 | 
			
		||||
			fmt.Fprintln(gitter.logWriter, msg)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// APIError holds data of errors returned from the API.
 | 
			
		||||
type APIError struct {
 | 
			
		||||
	What string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e APIError) Error() string {
 | 
			
		||||
	return fmt.Sprintf("%v", e.What)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										142
									
								
								vendor/github.com/42wim/go-gitter/model.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										142
									
								
								vendor/github.com/42wim/go-gitter/model.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,142 @@
 | 
			
		||||
package gitter
 | 
			
		||||
 | 
			
		||||
import "time"
 | 
			
		||||
 | 
			
		||||
// A Room in Gitter can represent a GitHub Organization, a GitHub Repository, a Gitter Channel or a One-to-one conversation.
 | 
			
		||||
// In the case of the Organizations and Repositories, the access control policies are inherited from GitHub.
 | 
			
		||||
type Room struct {
 | 
			
		||||
 | 
			
		||||
	// Room ID
 | 
			
		||||
	ID string `json:"id"`
 | 
			
		||||
 | 
			
		||||
	// Room name
 | 
			
		||||
	Name string `json:"name"`
 | 
			
		||||
 | 
			
		||||
	// Room topic. (default: GitHub repo description)
 | 
			
		||||
	Topic string `json:"topic"`
 | 
			
		||||
 | 
			
		||||
	// Room URI on Gitter
 | 
			
		||||
	URI string `json:"uri"`
 | 
			
		||||
 | 
			
		||||
	// Indicates if the room is a one-to-one chat
 | 
			
		||||
	OneToOne bool `json:"oneToOne"`
 | 
			
		||||
 | 
			
		||||
	// Count of users in the room
 | 
			
		||||
	UserCount int `json:"userCount"`
 | 
			
		||||
 | 
			
		||||
	// Number of unread messages for the current user
 | 
			
		||||
	UnreadItems int `json:"unreadItems"`
 | 
			
		||||
 | 
			
		||||
	// Number of unread mentions for the current user
 | 
			
		||||
	Mentions int `json:"mentions"`
 | 
			
		||||
 | 
			
		||||
	// Last time the current user accessed the room in ISO format
 | 
			
		||||
	LastAccessTime time.Time `json:"lastAccessTime"`
 | 
			
		||||
 | 
			
		||||
	// Indicates if the current user has disabled notifications
 | 
			
		||||
	Lurk bool `json:"lurk"`
 | 
			
		||||
 | 
			
		||||
	// Path to the room on gitter
 | 
			
		||||
	URL string `json:"url"`
 | 
			
		||||
 | 
			
		||||
	// Type of the room
 | 
			
		||||
	// - ORG: A room that represents a GitHub Organization.
 | 
			
		||||
	// - REPO: A room that represents a GitHub Repository.
 | 
			
		||||
	// - ONETOONE: A one-to-one chat.
 | 
			
		||||
	// - ORG_CHANNEL: A Gitter channel nested under a GitHub Organization.
 | 
			
		||||
	// - REPO_CHANNEL A Gitter channel nested under a GitHub Repository.
 | 
			
		||||
	// - USER_CHANNEL A Gitter channel nested under a GitHub User.
 | 
			
		||||
	GithubType string `json:"githubType"`
 | 
			
		||||
 | 
			
		||||
	// Tags that define the room
 | 
			
		||||
	Tags []string `json:"tags"`
 | 
			
		||||
 | 
			
		||||
	RoomMember bool `json:"roomMember"`
 | 
			
		||||
 | 
			
		||||
	// Room version.
 | 
			
		||||
	Version int `json:"v"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type User struct {
 | 
			
		||||
 | 
			
		||||
	// Gitter User ID
 | 
			
		||||
	ID string `json:"id"`
 | 
			
		||||
 | 
			
		||||
	// Gitter/GitHub username
 | 
			
		||||
	Username string `json:"username"`
 | 
			
		||||
 | 
			
		||||
	// Gitter/GitHub user real name
 | 
			
		||||
	DisplayName string `json:"displayName"`
 | 
			
		||||
 | 
			
		||||
	// Path to the user on Gitter
 | 
			
		||||
	URL string `json:"url"`
 | 
			
		||||
 | 
			
		||||
	// User avatar URI (small)
 | 
			
		||||
	AvatarURLSmall string `json:"avatarUrlSmall"`
 | 
			
		||||
 | 
			
		||||
	// User avatar URI (medium)
 | 
			
		||||
	AvatarURLMedium string `json:"avatarUrlMedium"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
 | 
			
		||||
	// ID of the message
 | 
			
		||||
	ID string `json:"id"`
 | 
			
		||||
 | 
			
		||||
	// Original message in plain-text/markdown
 | 
			
		||||
	Text string `json:"text"`
 | 
			
		||||
 | 
			
		||||
	// HTML formatted message
 | 
			
		||||
	HTML string `json:"html"`
 | 
			
		||||
 | 
			
		||||
	// ISO formatted date of the message
 | 
			
		||||
	Sent time.Time `json:"sent"`
 | 
			
		||||
 | 
			
		||||
	// ISO formatted date of the message if edited
 | 
			
		||||
	EditedAt time.Time `json:"editedAt"`
 | 
			
		||||
 | 
			
		||||
	// User that sent the message
 | 
			
		||||
	From User `json:"fromUser"`
 | 
			
		||||
 | 
			
		||||
	// Boolean that indicates if the current user has read the message.
 | 
			
		||||
	Unread bool `json:"unread"`
 | 
			
		||||
 | 
			
		||||
	// Number of users that have read the message
 | 
			
		||||
	ReadBy int `json:"readBy"`
 | 
			
		||||
 | 
			
		||||
	// List of URLs present in the message
 | 
			
		||||
	Urls []URL `json:"urls"`
 | 
			
		||||
 | 
			
		||||
	// List of @Mentions in the message
 | 
			
		||||
	Mentions []Mention `json:"mentions"`
 | 
			
		||||
 | 
			
		||||
	// List of #Issues referenced in the message
 | 
			
		||||
	Issues []Issue `json:"issues"`
 | 
			
		||||
 | 
			
		||||
	// Version
 | 
			
		||||
	Version int `json:"v"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Mention holds data about mentioned user in the message
 | 
			
		||||
type Mention struct {
 | 
			
		||||
 | 
			
		||||
	// User's username
 | 
			
		||||
	ScreenName string `json:"screenName"`
 | 
			
		||||
 | 
			
		||||
	// Gitter User ID
 | 
			
		||||
	UserID string `json:"userID"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Issue references issue in the message
 | 
			
		||||
type Issue struct {
 | 
			
		||||
 | 
			
		||||
	// Issue number
 | 
			
		||||
	Number string `json:"number"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// URL presented in the message
 | 
			
		||||
type URL struct {
 | 
			
		||||
 | 
			
		||||
	// URL
 | 
			
		||||
	URL string `json:"url"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										221
									
								
								vendor/github.com/42wim/go-gitter/stream.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										221
									
								
								vendor/github.com/42wim/go-gitter/stream.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,221 @@
 | 
			
		||||
package gitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/mreiferson/go-httpclient"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var defaultConnectionWaitTime time.Duration = 3000 // millis
 | 
			
		||||
var defaultConnectionMaxRetries = 5
 | 
			
		||||
 | 
			
		||||
// Stream initialize stream
 | 
			
		||||
func (gitter *Gitter) Stream(roomID string) *Stream {
 | 
			
		||||
	return &Stream{
 | 
			
		||||
		url:    streamBaseURL + "rooms/" + roomID + "/chatMessages",
 | 
			
		||||
		Event:  make(chan Event),
 | 
			
		||||
		gitter: gitter,
 | 
			
		||||
		streamConnection: gitter.newStreamConnection(
 | 
			
		||||
			defaultConnectionWaitTime,
 | 
			
		||||
			defaultConnectionMaxRetries),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Implemented to conform with https://developer.gitter.im/docs/streaming-api
 | 
			
		||||
func (gitter *Gitter) Listen(stream *Stream) {
 | 
			
		||||
 | 
			
		||||
	defer stream.destroy()
 | 
			
		||||
 | 
			
		||||
	var reader *bufio.Reader
 | 
			
		||||
	var gitterMessage Message
 | 
			
		||||
	lastKeepalive := time.Now().Unix()
 | 
			
		||||
 | 
			
		||||
	// connect
 | 
			
		||||
	stream.connect()
 | 
			
		||||
 | 
			
		||||
Loop:
 | 
			
		||||
	for {
 | 
			
		||||
 | 
			
		||||
		// if closed then stop trying
 | 
			
		||||
		if stream.isClosed() {
 | 
			
		||||
			stream.Event <- Event{
 | 
			
		||||
				Data: &GitterConnectionClosed{},
 | 
			
		||||
			}
 | 
			
		||||
			break Loop
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		resp := stream.getResponse()
 | 
			
		||||
		if resp.StatusCode != 200 {
 | 
			
		||||
			gitter.log(fmt.Sprintf("Unexpected response code %v", resp.StatusCode))
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		//"The JSON stream returns messages as JSON objects that are delimited by carriage return (\r)" <- Not true crap it's (\n) only
 | 
			
		||||
		reader = bufio.NewReader(resp.Body)
 | 
			
		||||
		line, err := reader.ReadBytes('\n')
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			gitter.log("ReadBytes error: " + err.Error())
 | 
			
		||||
			stream.connect()
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		//Check if the line only consists of whitespace
 | 
			
		||||
		onlyWhitespace := true
 | 
			
		||||
		for _, b := range line {
 | 
			
		||||
			if b != ' ' && b != '\t' && b != '\r' && b != '\n' {
 | 
			
		||||
				onlyWhitespace = false
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if onlyWhitespace {
 | 
			
		||||
			//"Parsers must be tolerant of occasional extra newline characters placed between messages."
 | 
			
		||||
			currentKeepalive := time.Now().Unix() //interesting behavior of 100+ keepalives per seconds was observed
 | 
			
		||||
			if currentKeepalive-lastKeepalive > 10 {
 | 
			
		||||
				lastKeepalive = currentKeepalive
 | 
			
		||||
				gitter.log("Keepalive was received")
 | 
			
		||||
			}
 | 
			
		||||
			continue
 | 
			
		||||
		} else if stream.isClosed() {
 | 
			
		||||
			gitter.log("Stream closed")
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// unmarshal the streamed data
 | 
			
		||||
		err = json.Unmarshal(line, &gitterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			gitter.log("JSON Unmarshal error: " + err.Error())
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// we are here, then we got the good message. pipe it forward.
 | 
			
		||||
		stream.Event <- Event{
 | 
			
		||||
			Data: &MessageReceived{
 | 
			
		||||
				Message: gitterMessage,
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	gitter.log("Listening was completed")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Stream holds stream data.
 | 
			
		||||
type Stream struct {
 | 
			
		||||
	url              string
 | 
			
		||||
	Event            chan Event
 | 
			
		||||
	streamConnection *streamConnection
 | 
			
		||||
	gitter           *Gitter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (stream *Stream) destroy() {
 | 
			
		||||
	close(stream.Event)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Event struct {
 | 
			
		||||
	Data interface{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type GitterConnectionClosed struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MessageReceived struct {
 | 
			
		||||
	Message Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// connect and try to reconnect with
 | 
			
		||||
func (stream *Stream) connect() {
 | 
			
		||||
 | 
			
		||||
	if stream.streamConnection.retries == stream.streamConnection.currentRetries {
 | 
			
		||||
		stream.Close()
 | 
			
		||||
		stream.gitter.log("Number of retries exceeded the max retries number, we are done here")
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	res, err := stream.gitter.getResponse(stream.url, stream)
 | 
			
		||||
	if stream.streamConnection.canceled {
 | 
			
		||||
		// do nothing
 | 
			
		||||
	} else if err != nil || res.StatusCode != 200 {
 | 
			
		||||
		stream.gitter.log("Failed to get response, trying reconnect ")
 | 
			
		||||
		stream.gitter.log(err)
 | 
			
		||||
 | 
			
		||||
		// sleep and wait
 | 
			
		||||
		stream.streamConnection.currentRetries++
 | 
			
		||||
		time.Sleep(time.Millisecond * stream.streamConnection.wait * time.Duration(stream.streamConnection.currentRetries))
 | 
			
		||||
 | 
			
		||||
		// connect again
 | 
			
		||||
		stream.Close()
 | 
			
		||||
		stream.connect()
 | 
			
		||||
	} else {
 | 
			
		||||
		stream.gitter.log("Response was received")
 | 
			
		||||
		stream.streamConnection.currentRetries = 0
 | 
			
		||||
		stream.streamConnection.closed = false
 | 
			
		||||
		stream.streamConnection.response = res
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type streamConnection struct {
 | 
			
		||||
 | 
			
		||||
	// connection was closed
 | 
			
		||||
	closed bool
 | 
			
		||||
 | 
			
		||||
	// canceled
 | 
			
		||||
	canceled bool
 | 
			
		||||
 | 
			
		||||
	// wait time till next try
 | 
			
		||||
	wait time.Duration
 | 
			
		||||
 | 
			
		||||
	// max tries to recover
 | 
			
		||||
	retries int
 | 
			
		||||
 | 
			
		||||
	// current streamed response
 | 
			
		||||
	response *http.Response
 | 
			
		||||
 | 
			
		||||
	// current request
 | 
			
		||||
	request *http.Request
 | 
			
		||||
 | 
			
		||||
	// current status
 | 
			
		||||
	currentRetries int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Close the stream connection and stop receiving streamed data
 | 
			
		||||
func (stream *Stream) Close() {
 | 
			
		||||
	conn := stream.streamConnection
 | 
			
		||||
	conn.closed = true
 | 
			
		||||
	if conn.response != nil {
 | 
			
		||||
		stream.gitter.log("Stream connection close response")
 | 
			
		||||
		defer conn.response.Body.Close()
 | 
			
		||||
	}
 | 
			
		||||
	if conn.request != nil {
 | 
			
		||||
		stream.gitter.log("Stream connection close request")
 | 
			
		||||
		switch transport := stream.gitter.config.client.Transport.(type) {
 | 
			
		||||
		case *httpclient.Transport:
 | 
			
		||||
			stream.streamConnection.canceled = true
 | 
			
		||||
			transport.CancelRequest(conn.request)
 | 
			
		||||
		default:
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
	}
 | 
			
		||||
	conn.currentRetries = 0
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (stream *Stream) isClosed() bool {
 | 
			
		||||
	return stream.streamConnection.closed
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (stream *Stream) getResponse() *http.Response {
 | 
			
		||||
	return stream.streamConnection.response
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Optional, set stream connection properties
 | 
			
		||||
// wait - time in milliseconds of waiting between reconnections. Will grow exponentially.
 | 
			
		||||
// retries - number of reconnections retries before dropping the stream.
 | 
			
		||||
func (gitter *Gitter) newStreamConnection(wait time.Duration, retries int) *streamConnection {
 | 
			
		||||
	return &streamConnection{
 | 
			
		||||
		closed:  true,
 | 
			
		||||
		wait:    wait,
 | 
			
		||||
		retries: retries,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										30
									
								
								vendor/github.com/42wim/go-gitter/test_utils.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								vendor/github.com/42wim/go-gitter/test_utils.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,30 @@
 | 
			
		||||
package gitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/httptest"
 | 
			
		||||
	"net/url"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	mux    *http.ServeMux
 | 
			
		||||
	gitter *Gitter
 | 
			
		||||
	server *httptest.Server
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func setup() {
 | 
			
		||||
	mux = http.NewServeMux()
 | 
			
		||||
	server = httptest.NewServer(mux)
 | 
			
		||||
 | 
			
		||||
	gitter = New("abc")
 | 
			
		||||
 | 
			
		||||
	// Fake the API and Stream base URLs by using the test
 | 
			
		||||
	// server URL instead.
 | 
			
		||||
	url, _ := url.Parse(server.URL)
 | 
			
		||||
	gitter.config.apiBaseURL = url.String() + "/"
 | 
			
		||||
	gitter.config.streamBaseURL = url.String() + "/"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func teardown() {
 | 
			
		||||
	server.Close()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										202
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										202
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,202 @@
 | 
			
		||||
                                 Apache License
 | 
			
		||||
                           Version 2.0, January 2004
 | 
			
		||||
                        http://www.apache.org/licenses/
 | 
			
		||||
 | 
			
		||||
   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
   1. Definitions.
 | 
			
		||||
 | 
			
		||||
      "License" shall mean the terms and conditions for use, reproduction,
 | 
			
		||||
      and distribution as defined by Sections 1 through 9 of this document.
 | 
			
		||||
 | 
			
		||||
      "Licensor" shall mean the copyright owner or entity authorized by
 | 
			
		||||
      the copyright owner that is granting the License.
 | 
			
		||||
 | 
			
		||||
      "Legal Entity" shall mean the union of the acting entity and all
 | 
			
		||||
      other entities that control, are controlled by, or are under common
 | 
			
		||||
      control with that entity. For the purposes of this definition,
 | 
			
		||||
      "control" means (i) the power, direct or indirect, to cause the
 | 
			
		||||
      direction or management of such entity, whether by contract or
 | 
			
		||||
      otherwise, or (ii) ownership of fifty percent (50%) or more of the
 | 
			
		||||
      outstanding shares, or (iii) beneficial ownership of such entity.
 | 
			
		||||
 | 
			
		||||
      "You" (or "Your") shall mean an individual or Legal Entity
 | 
			
		||||
      exercising permissions granted by this License.
 | 
			
		||||
 | 
			
		||||
      "Source" form shall mean the preferred form for making modifications,
 | 
			
		||||
      including but not limited to software source code, documentation
 | 
			
		||||
      source, and configuration files.
 | 
			
		||||
 | 
			
		||||
      "Object" form shall mean any form resulting from mechanical
 | 
			
		||||
      transformation or translation of a Source form, including but
 | 
			
		||||
      not limited to compiled object code, generated documentation,
 | 
			
		||||
      and conversions to other media types.
 | 
			
		||||
 | 
			
		||||
      "Work" shall mean the work of authorship, whether in Source or
 | 
			
		||||
      Object form, made available under the License, as indicated by a
 | 
			
		||||
      copyright notice that is included in or attached to the work
 | 
			
		||||
      (an example is provided in the Appendix below).
 | 
			
		||||
 | 
			
		||||
      "Derivative Works" shall mean any work, whether in Source or Object
 | 
			
		||||
      form, that is based on (or derived from) the Work and for which the
 | 
			
		||||
      editorial revisions, annotations, elaborations, or other modifications
 | 
			
		||||
      represent, as a whole, an original work of authorship. For the purposes
 | 
			
		||||
      of this License, Derivative Works shall not include works that remain
 | 
			
		||||
      separable from, or merely link (or bind by name) to the interfaces of,
 | 
			
		||||
      the Work and Derivative Works thereof.
 | 
			
		||||
 | 
			
		||||
      "Contribution" shall mean any work of authorship, including
 | 
			
		||||
      the original version of the Work and any modifications or additions
 | 
			
		||||
      to that Work or Derivative Works thereof, that is intentionally
 | 
			
		||||
      submitted to Licensor for inclusion in the Work by the copyright owner
 | 
			
		||||
      or by an individual or Legal Entity authorized to submit on behalf of
 | 
			
		||||
      the copyright owner. For the purposes of this definition, "submitted"
 | 
			
		||||
      means any form of electronic, verbal, or written communication sent
 | 
			
		||||
      to the Licensor or its representatives, including but not limited to
 | 
			
		||||
      communication on electronic mailing lists, source code control systems,
 | 
			
		||||
      and issue tracking systems that are managed by, or on behalf of, the
 | 
			
		||||
      Licensor for the purpose of discussing and improving the Work, but
 | 
			
		||||
      excluding communication that is conspicuously marked or otherwise
 | 
			
		||||
      designated in writing by the copyright owner as "Not a Contribution."
 | 
			
		||||
 | 
			
		||||
      "Contributor" shall mean Licensor and any individual or Legal Entity
 | 
			
		||||
      on behalf of whom a Contribution has been received by Licensor and
 | 
			
		||||
      subsequently incorporated within the Work.
 | 
			
		||||
 | 
			
		||||
   2. Grant of Copyright License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      copyright license to reproduce, prepare Derivative Works of,
 | 
			
		||||
      publicly display, publicly perform, sublicense, and distribute the
 | 
			
		||||
      Work and such Derivative Works in Source or Object form.
 | 
			
		||||
 | 
			
		||||
   3. Grant of Patent License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      (except as stated in this section) patent license to make, have made,
 | 
			
		||||
      use, offer to sell, sell, import, and otherwise transfer the Work,
 | 
			
		||||
      where such license applies only to those patent claims licensable
 | 
			
		||||
      by such Contributor that are necessarily infringed by their
 | 
			
		||||
      Contribution(s) alone or by combination of their Contribution(s)
 | 
			
		||||
      with the Work to which such Contribution(s) was submitted. If You
 | 
			
		||||
      institute patent litigation against any entity (including a
 | 
			
		||||
      cross-claim or counterclaim in a lawsuit) alleging that the Work
 | 
			
		||||
      or a Contribution incorporated within the Work constitutes direct
 | 
			
		||||
      or contributory patent infringement, then any patent licenses
 | 
			
		||||
      granted to You under this License for that Work shall terminate
 | 
			
		||||
      as of the date such litigation is filed.
 | 
			
		||||
 | 
			
		||||
   4. Redistribution. You may reproduce and distribute copies of the
 | 
			
		||||
      Work or Derivative Works thereof in any medium, with or without
 | 
			
		||||
      modifications, and in Source or Object form, provided that You
 | 
			
		||||
      meet the following conditions:
 | 
			
		||||
 | 
			
		||||
      (a) You must give any other recipients of the Work or
 | 
			
		||||
          Derivative Works a copy of this License; and
 | 
			
		||||
 | 
			
		||||
      (b) You must cause any modified files to carry prominent notices
 | 
			
		||||
          stating that You changed the files; and
 | 
			
		||||
 | 
			
		||||
      (c) You must retain, in the Source form of any Derivative Works
 | 
			
		||||
          that You distribute, all copyright, patent, trademark, and
 | 
			
		||||
          attribution notices from the Source form of the Work,
 | 
			
		||||
          excluding those notices that do not pertain to any part of
 | 
			
		||||
          the Derivative Works; and
 | 
			
		||||
 | 
			
		||||
      (d) If the Work includes a "NOTICE" text file as part of its
 | 
			
		||||
          distribution, then any Derivative Works that You distribute must
 | 
			
		||||
          include a readable copy of the attribution notices contained
 | 
			
		||||
          within such NOTICE file, excluding those notices that do not
 | 
			
		||||
          pertain to any part of the Derivative Works, in at least one
 | 
			
		||||
          of the following places: within a NOTICE text file distributed
 | 
			
		||||
          as part of the Derivative Works; within the Source form or
 | 
			
		||||
          documentation, if provided along with the Derivative Works; or,
 | 
			
		||||
          within a display generated by the Derivative Works, if and
 | 
			
		||||
          wherever such third-party notices normally appear. The contents
 | 
			
		||||
          of the NOTICE file are for informational purposes only and
 | 
			
		||||
          do not modify the License. You may add Your own attribution
 | 
			
		||||
          notices within Derivative Works that You distribute, alongside
 | 
			
		||||
          or as an addendum to the NOTICE text from the Work, provided
 | 
			
		||||
          that such additional attribution notices cannot be construed
 | 
			
		||||
          as modifying the License.
 | 
			
		||||
 | 
			
		||||
      You may add Your own copyright statement to Your modifications and
 | 
			
		||||
      may provide additional or different license terms and conditions
 | 
			
		||||
      for use, reproduction, or distribution of Your modifications, or
 | 
			
		||||
      for any such Derivative Works as a whole, provided Your use,
 | 
			
		||||
      reproduction, and distribution of the Work otherwise complies with
 | 
			
		||||
      the conditions stated in this License.
 | 
			
		||||
 | 
			
		||||
   5. Submission of Contributions. Unless You explicitly state otherwise,
 | 
			
		||||
      any Contribution intentionally submitted for inclusion in the Work
 | 
			
		||||
      by You to the Licensor shall be under the terms and conditions of
 | 
			
		||||
      this License, without any additional terms or conditions.
 | 
			
		||||
      Notwithstanding the above, nothing herein shall supersede or modify
 | 
			
		||||
      the terms of any separate license agreement you may have executed
 | 
			
		||||
      with Licensor regarding such Contributions.
 | 
			
		||||
 | 
			
		||||
   6. Trademarks. This License does not grant permission to use the trade
 | 
			
		||||
      names, trademarks, service marks, or product names of the Licensor,
 | 
			
		||||
      except as required for reasonable and customary use in describing the
 | 
			
		||||
      origin of the Work and reproducing the content of the NOTICE file.
 | 
			
		||||
 | 
			
		||||
   7. Disclaimer of Warranty. Unless required by applicable law or
 | 
			
		||||
      agreed to in writing, Licensor provides the Work (and each
 | 
			
		||||
      Contributor provides its Contributions) on an "AS IS" BASIS,
 | 
			
		||||
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 | 
			
		||||
      implied, including, without limitation, any warranties or conditions
 | 
			
		||||
      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
 | 
			
		||||
      PARTICULAR PURPOSE. You are solely responsible for determining the
 | 
			
		||||
      appropriateness of using or redistributing the Work and assume any
 | 
			
		||||
      risks associated with Your exercise of permissions under this License.
 | 
			
		||||
 | 
			
		||||
   8. Limitation of Liability. In no event and under no legal theory,
 | 
			
		||||
      whether in tort (including negligence), contract, or otherwise,
 | 
			
		||||
      unless required by applicable law (such as deliberate and grossly
 | 
			
		||||
      negligent acts) or agreed to in writing, shall any Contributor be
 | 
			
		||||
      liable to You for damages, including any direct, indirect, special,
 | 
			
		||||
      incidental, or consequential damages of any character arising as a
 | 
			
		||||
      result of this License or out of the use or inability to use the
 | 
			
		||||
      Work (including but not limited to damages for loss of goodwill,
 | 
			
		||||
      work stoppage, computer failure or malfunction, or any and all
 | 
			
		||||
      other commercial damages or losses), even if such Contributor
 | 
			
		||||
      has been advised of the possibility of such damages.
 | 
			
		||||
 | 
			
		||||
   9. Accepting Warranty or Additional Liability. While redistributing
 | 
			
		||||
      the Work or Derivative Works thereof, You may choose to offer,
 | 
			
		||||
      and charge a fee for, acceptance of support, warranty, indemnity,
 | 
			
		||||
      or other liability obligations and/or rights consistent with this
 | 
			
		||||
      License. However, in accepting such obligations, You may act only
 | 
			
		||||
      on Your own behalf and on Your sole responsibility, not on behalf
 | 
			
		||||
      of any other Contributor, and only if You agree to indemnify,
 | 
			
		||||
      defend, and hold each Contributor harmless for any liability
 | 
			
		||||
      incurred by, or claims asserted against, such Contributor by reason
 | 
			
		||||
      of your accepting any such warranty or additional liability.
 | 
			
		||||
 | 
			
		||||
   END OF TERMS AND CONDITIONS
 | 
			
		||||
 | 
			
		||||
   APPENDIX: How to apply the Apache License to your work.
 | 
			
		||||
 | 
			
		||||
      To apply the Apache License to your work, attach the following
 | 
			
		||||
      boilerplate notice, with the fields enclosed by brackets "{}"
 | 
			
		||||
      replaced with your own identifying information. (Don't include
 | 
			
		||||
      the brackets!)  The text should be enclosed in the appropriate
 | 
			
		||||
      comment syntax for the file format. We also recommend that a
 | 
			
		||||
      file or class name and description of purpose be included on the
 | 
			
		||||
      same "printed page" as the copyright notice for easier
 | 
			
		||||
      identification within third-party archives.
 | 
			
		||||
 | 
			
		||||
   Copyright {yyyy} {name of copyright owner}
 | 
			
		||||
 | 
			
		||||
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
   you may not use this file except in compliance with the License.
 | 
			
		||||
   You may obtain a copy of the License at
 | 
			
		||||
 | 
			
		||||
       http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 | 
			
		||||
   Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
   See the License for the specific language governing permissions and
 | 
			
		||||
   limitations under the License.
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										434
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										434
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,434 @@
 | 
			
		||||
package bridge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"github.com/42wim/matterbridge-plus/matterclient"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/peterhellberg/giphy"
 | 
			
		||||
	ircm "github.com/sorcix/irc"
 | 
			
		||||
	"github.com/thoj/go-ircevent"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
//type Bridge struct {
 | 
			
		||||
type MMhook struct {
 | 
			
		||||
	mh *matterhook.Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMapi struct {
 | 
			
		||||
	mc            *matterclient.MMClient
 | 
			
		||||
	mmMap         map[string]string
 | 
			
		||||
	mmIgnoreNicks []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMirc struct {
 | 
			
		||||
	i              *irc.Connection
 | 
			
		||||
	ircNick        string
 | 
			
		||||
	ircMap         map[string]string
 | 
			
		||||
	names          map[string][]string
 | 
			
		||||
	ircIgnoreNicks []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMMessage struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
	MMhook
 | 
			
		||||
	MMapi
 | 
			
		||||
	MMirc
 | 
			
		||||
	*Config
 | 
			
		||||
	kind string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type FancyLog struct {
 | 
			
		||||
	irc *log.Entry
 | 
			
		||||
	mm  *log.Entry
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog FancyLog
 | 
			
		||||
 | 
			
		||||
const Legacy = "legacy"
 | 
			
		||||
 | 
			
		||||
func initFLog() {
 | 
			
		||||
	flog.irc = log.WithFields(log.Fields{"module": "irc"})
 | 
			
		||||
	flog.mm = log.WithFields(log.Fields{"module": "mattermost"})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewBridge(name string, config *Config, kind string) *Bridge {
 | 
			
		||||
	initFLog()
 | 
			
		||||
	b := &Bridge{}
 | 
			
		||||
	b.Config = config
 | 
			
		||||
	b.kind = kind
 | 
			
		||||
	b.ircNick = b.Config.IRC.Nick
 | 
			
		||||
	b.ircMap = make(map[string]string)
 | 
			
		||||
	b.MMirc.names = make(map[string][]string)
 | 
			
		||||
	b.ircIgnoreNicks = strings.Fields(b.Config.IRC.IgnoreNicks)
 | 
			
		||||
	b.mmIgnoreNicks = strings.Fields(b.Config.Mattermost.IgnoreNicks)
 | 
			
		||||
	if kind == Legacy {
 | 
			
		||||
		if len(b.Config.Token) > 0 {
 | 
			
		||||
			for _, val := range b.Config.Token {
 | 
			
		||||
				b.ircMap[val.IRCChannel] = val.MMChannel
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		b.mh = matterhook.New(b.Config.Mattermost.URL,
 | 
			
		||||
			matterhook.Config{Port: b.Config.Mattermost.Port, Token: b.Config.Mattermost.Token,
 | 
			
		||||
				InsecureSkipVerify: b.Config.Mattermost.SkipTLSVerify,
 | 
			
		||||
				BindAddress:        b.Config.Mattermost.BindAddress})
 | 
			
		||||
	} else {
 | 
			
		||||
		b.mmMap = make(map[string]string)
 | 
			
		||||
		if len(b.Config.Channel) > 0 {
 | 
			
		||||
			for _, val := range b.Config.Channel {
 | 
			
		||||
				b.ircMap[val.IRC] = val.Mattermost
 | 
			
		||||
				b.mmMap[val.Mattermost] = val.IRC
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		b.mc = matterclient.New(b.Config.Mattermost.Login, b.Config.Mattermost.Password,
 | 
			
		||||
			b.Config.Mattermost.Team, b.Config.Mattermost.Server)
 | 
			
		||||
		b.mc.SkipTLSVerify = b.Config.Mattermost.SkipTLSVerify
 | 
			
		||||
		b.mc.NoTLS = b.Config.Mattermost.NoTLS
 | 
			
		||||
		flog.mm.Infof("Trying login %s (team: %s) on %s", b.Config.Mattermost.Login, b.Config.Mattermost.Team, b.Config.Mattermost.Server)
 | 
			
		||||
		err := b.mc.Login()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.mm.Fatal("Can not connect", err)
 | 
			
		||||
		}
 | 
			
		||||
		flog.mm.Info("Login ok")
 | 
			
		||||
		b.mc.JoinChannel(b.Config.Mattermost.Channel)
 | 
			
		||||
		if len(b.Config.Channel) > 0 {
 | 
			
		||||
			for _, val := range b.Config.Channel {
 | 
			
		||||
				b.mc.JoinChannel(val.Mattermost)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		go b.mc.WsReceiver()
 | 
			
		||||
	}
 | 
			
		||||
	flog.irc.Info("Trying IRC connection")
 | 
			
		||||
	b.i = b.createIRC(name)
 | 
			
		||||
	flog.irc.Info("Connection succeeded")
 | 
			
		||||
	go b.handleMatter()
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) createIRC(name string) *irc.Connection {
 | 
			
		||||
	i := irc.IRC(b.Config.IRC.Nick, b.Config.IRC.Nick)
 | 
			
		||||
	i.UseTLS = b.Config.IRC.UseTLS
 | 
			
		||||
	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.IRC.SkipTLSVerify}
 | 
			
		||||
	if b.Config.IRC.Password != "" {
 | 
			
		||||
		i.Password = b.Config.IRC.Password
 | 
			
		||||
	}
 | 
			
		||||
	i.AddCallback(ircm.RPL_WELCOME, b.handleNewConnection)
 | 
			
		||||
	i.Connect(b.Config.IRC.Server + ":" + strconv.Itoa(b.Config.IRC.Port))
 | 
			
		||||
	return i
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleNewConnection(event *irc.Event) {
 | 
			
		||||
	flog.irc.Info("Registering callbacks")
 | 
			
		||||
	i := b.i
 | 
			
		||||
	b.ircNick = event.Arguments[0]
 | 
			
		||||
	i.AddCallback("PRIVMSG", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback("CTCP_ACTION", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames)
 | 
			
		||||
	i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames)
 | 
			
		||||
	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
 | 
			
		||||
	i.AddCallback(ircm.NOTICE, b.handleNotice)
 | 
			
		||||
	i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.irc.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) })
 | 
			
		||||
	i.AddCallback("PING", func(e *irc.Event) {
 | 
			
		||||
		i.SendRaw("PONG :" + e.Message())
 | 
			
		||||
		flog.irc.Debugf("PING/PONG")
 | 
			
		||||
	})
 | 
			
		||||
	if b.Config.Mattermost.ShowJoinPart {
 | 
			
		||||
		i.AddCallback("JOIN", b.handleJoinPart)
 | 
			
		||||
		i.AddCallback("PART", b.handleJoinPart)
 | 
			
		||||
	}
 | 
			
		||||
	i.AddCallback("*", b.handleOther)
 | 
			
		||||
	b.setupChannels()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) setupChannels() {
 | 
			
		||||
	i := b.i
 | 
			
		||||
	if b.Config.IRC.Channel != "" {
 | 
			
		||||
		flog.irc.Infof("Joining %s as %s", b.Config.IRC.Channel, b.ircNick)
 | 
			
		||||
		i.Join(b.Config.IRC.Channel)
 | 
			
		||||
	}
 | 
			
		||||
	if b.kind == Legacy {
 | 
			
		||||
		for _, val := range b.Config.Token {
 | 
			
		||||
			flog.irc.Infof("Joining %s as %s", val.IRCChannel, b.ircNick)
 | 
			
		||||
			i.Join(val.IRCChannel)
 | 
			
		||||
		}
 | 
			
		||||
	} else {
 | 
			
		||||
		for _, val := range b.Config.Channel {
 | 
			
		||||
			flog.irc.Infof("Joining %s as %s", val.IRC, b.ircNick)
 | 
			
		||||
			i.Join(val.IRC)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleIrcBotCommand(event *irc.Event) bool {
 | 
			
		||||
	parts := strings.Fields(event.Message())
 | 
			
		||||
	exp, _ := regexp.Compile("[:,]+$")
 | 
			
		||||
	channel := event.Arguments[0]
 | 
			
		||||
	command := ""
 | 
			
		||||
	if len(parts) == 2 {
 | 
			
		||||
		command = parts[1]
 | 
			
		||||
	}
 | 
			
		||||
	if exp.ReplaceAllString(parts[0], "") == b.ircNick {
 | 
			
		||||
		switch command {
 | 
			
		||||
		case "users":
 | 
			
		||||
			usernames := b.mc.UsernamesInChannel(b.getMMChannel(channel))
 | 
			
		||||
			sort.Strings(usernames)
 | 
			
		||||
			b.i.Privmsg(channel, "Users on Mattermost: "+strings.Join(usernames, ", "))
 | 
			
		||||
		default:
 | 
			
		||||
			b.i.Privmsg(channel, "Valid commands are: [users, help]")
 | 
			
		||||
		}
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) ircNickFormat(nick string) string {
 | 
			
		||||
	if nick == b.ircNick {
 | 
			
		||||
		return nick
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.Mattermost.RemoteNickFormat == nil {
 | 
			
		||||
		return "irc-" + nick
 | 
			
		||||
	}
 | 
			
		||||
	return strings.Replace(*b.Config.Mattermost.RemoteNickFormat, "{NICK}", nick, -1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handlePrivMsg(event *irc.Event) {
 | 
			
		||||
	if b.ignoreMessage(event.Nick, event.Message(), "irc") {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if b.handleIrcBotCommand(event) {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	msg := ""
 | 
			
		||||
	if event.Code == "CTCP_ACTION" {
 | 
			
		||||
		msg = event.Nick + " "
 | 
			
		||||
	}
 | 
			
		||||
	msg += event.Message()
 | 
			
		||||
	b.Send(b.ircNickFormat(event.Nick), msg, b.getMMChannel(event.Arguments[0]))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleJoinPart(event *irc.Event) {
 | 
			
		||||
	b.Send(b.ircNick, b.ircNickFormat(event.Nick)+" "+strings.ToLower(event.Code)+"s "+event.Message(), b.getMMChannel(event.Arguments[0]))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleNotice(event *irc.Event) {
 | 
			
		||||
	if strings.Contains(event.Message(), "This nickname is registered") {
 | 
			
		||||
		b.i.Privmsg(b.Config.IRC.NickServNick, "IDENTIFY "+b.Config.IRC.NickServPassword)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) nicksPerRow() int {
 | 
			
		||||
	if b.Config.Mattermost.NicksPerRow < 1 {
 | 
			
		||||
		return 4
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.Mattermost.NicksPerRow
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) formatnicks(nicks []string, continued bool) string {
 | 
			
		||||
	switch b.Config.Mattermost.NickFormatter {
 | 
			
		||||
	case "table":
 | 
			
		||||
		return tableformatter(nicks, b.nicksPerRow(), continued)
 | 
			
		||||
	default:
 | 
			
		||||
		return plainformatter(nicks, b.nicksPerRow())
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) storeNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[2]
 | 
			
		||||
	b.MMirc.names[channel] = append(
 | 
			
		||||
		b.MMirc.names[channel],
 | 
			
		||||
		strings.Split(strings.TrimSpace(event.Message()), " ")...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) endNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[1]
 | 
			
		||||
	sort.Strings(b.MMirc.names[channel])
 | 
			
		||||
	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
 | 
			
		||||
	continued := false
 | 
			
		||||
	for len(b.MMirc.names[channel]) > maxNamesPerPost {
 | 
			
		||||
		b.Send(
 | 
			
		||||
			b.ircNick,
 | 
			
		||||
			b.formatnicks(b.MMirc.names[channel][0:maxNamesPerPost], continued),
 | 
			
		||||
			b.getMMChannel(channel))
 | 
			
		||||
		b.MMirc.names[channel] = b.MMirc.names[channel][maxNamesPerPost:]
 | 
			
		||||
		continued = true
 | 
			
		||||
	}
 | 
			
		||||
	b.Send(b.ircNick, b.formatnicks(b.MMirc.names[channel], continued), b.getMMChannel(channel))
 | 
			
		||||
	b.MMirc.names[channel] = nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleTopicWhoTime(event *irc.Event) {
 | 
			
		||||
	parts := strings.Split(event.Arguments[2], "!")
 | 
			
		||||
	t, err := strconv.ParseInt(event.Arguments[3], 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.irc.Errorf("Invalid time stamp: %s", event.Arguments[3])
 | 
			
		||||
	}
 | 
			
		||||
	user := parts[0]
 | 
			
		||||
	if len(parts) > 1 {
 | 
			
		||||
		user += " [" + parts[1] + "]"
 | 
			
		||||
	}
 | 
			
		||||
	flog.irc.Infof("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleOther(event *irc.Event) {
 | 
			
		||||
	flog.irc.Debugf("%#v", event)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) Send(nick string, message string, channel string) error {
 | 
			
		||||
	return b.SendType(nick, message, channel, "")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) SendType(nick string, message string, channel string, mtype string) error {
 | 
			
		||||
	if b.Config.Mattermost.PrefixMessagesWithNick {
 | 
			
		||||
		if IsMarkup(message) {
 | 
			
		||||
			message = nick + "\n\n" + message
 | 
			
		||||
		} else {
 | 
			
		||||
			message = nick + " " + message
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if b.kind == Legacy {
 | 
			
		||||
		matterMessage := matterhook.OMessage{IconURL: b.Config.Mattermost.IconURL}
 | 
			
		||||
		matterMessage.Channel = channel
 | 
			
		||||
		matterMessage.UserName = nick
 | 
			
		||||
		matterMessage.Type = mtype
 | 
			
		||||
		matterMessage.Text = message
 | 
			
		||||
		err := b.mh.Send(matterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.mm.Info(err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	flog.mm.Debug("->mattermost channel: ", channel, " ", message)
 | 
			
		||||
	b.mc.PostMessage(channel, message)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleMatterHook(mchan chan *MMMessage) {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.mh.Receive()
 | 
			
		||||
		m := &MMMessage{}
 | 
			
		||||
		m.Username = message.UserName
 | 
			
		||||
		m.Text = message.Text
 | 
			
		||||
		m.Channel = message.Token
 | 
			
		||||
		mchan <- m
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleMatterClient(mchan chan *MMMessage) {
 | 
			
		||||
	for message := range b.mc.MessageChan {
 | 
			
		||||
		// do not post our own messages back to irc
 | 
			
		||||
		if message.Raw.Action == "posted" && b.mc.User.Username != message.Username {
 | 
			
		||||
			m := &MMMessage{}
 | 
			
		||||
			m.Username = message.Username
 | 
			
		||||
			m.Channel = message.Channel
 | 
			
		||||
			m.Text = message.Text
 | 
			
		||||
			flog.mm.Debugf("<-mattermost channel: %s %#v %#v", message.Channel, message.Post, message.Raw)
 | 
			
		||||
			mchan <- m
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) handleMatter() {
 | 
			
		||||
	flog.mm.Infof("Choosing Mattermost connection type %s", b.kind)
 | 
			
		||||
	mchan := make(chan *MMMessage)
 | 
			
		||||
	if b.kind == Legacy {
 | 
			
		||||
		go b.handleMatterHook(mchan)
 | 
			
		||||
	} else {
 | 
			
		||||
		go b.handleMatterClient(mchan)
 | 
			
		||||
	}
 | 
			
		||||
	flog.mm.Info("Start listening for Mattermost messages")
 | 
			
		||||
	for message := range mchan {
 | 
			
		||||
		var username string
 | 
			
		||||
		if b.ignoreMessage(message.Username, message.Text, "mattermost") {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		username = message.Username + ": "
 | 
			
		||||
		if b.Config.IRC.RemoteNickFormat != "" {
 | 
			
		||||
			username = strings.Replace(b.Config.IRC.RemoteNickFormat, "{NICK}", message.Username, -1)
 | 
			
		||||
		} else if b.Config.IRC.UseSlackCircumfix {
 | 
			
		||||
			username = "<" + message.Username + "> "
 | 
			
		||||
		}
 | 
			
		||||
		cmds := strings.Fields(message.Text)
 | 
			
		||||
		// empty message
 | 
			
		||||
		if len(cmds) == 0 {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		cmd := cmds[0]
 | 
			
		||||
		switch cmd {
 | 
			
		||||
		case "!users":
 | 
			
		||||
			flog.mm.Info("Received !users from ", message.Username)
 | 
			
		||||
			b.i.SendRaw("NAMES " + b.getIRCChannel(message.Channel))
 | 
			
		||||
			continue
 | 
			
		||||
		case "!gif":
 | 
			
		||||
			message.Text = b.giphyRandom(strings.Fields(strings.Replace(message.Text, "!gif ", "", 1)))
 | 
			
		||||
			b.Send(b.ircNick, message.Text, b.getIRCChannel(message.Channel))
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		texts := strings.Split(message.Text, "\n")
 | 
			
		||||
		for _, text := range texts {
 | 
			
		||||
			flog.mm.Debug("Sending message from " + message.Username + " to " + message.Channel)
 | 
			
		||||
			b.i.Privmsg(b.getIRCChannel(message.Channel), username+text)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) giphyRandom(query []string) string {
 | 
			
		||||
	g := giphy.DefaultClient
 | 
			
		||||
	if b.Config.General.GiphyAPIKey != "" {
 | 
			
		||||
		g.APIKey = b.Config.General.GiphyAPIKey
 | 
			
		||||
	}
 | 
			
		||||
	res, err := g.Random(query)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "error"
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.FixedHeightDownsampledURL
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) getMMChannel(ircChannel string) string {
 | 
			
		||||
	mmchannel, ok := b.ircMap[ircChannel]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		mmchannel = b.Config.Mattermost.Channel
 | 
			
		||||
	}
 | 
			
		||||
	return mmchannel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) getIRCChannel(channel string) string {
 | 
			
		||||
	if b.kind == Legacy {
 | 
			
		||||
		ircchannel := b.Config.IRC.Channel
 | 
			
		||||
		_, ok := b.Config.Token[channel]
 | 
			
		||||
		if ok {
 | 
			
		||||
			ircchannel = b.Config.Token[channel].IRCChannel
 | 
			
		||||
		}
 | 
			
		||||
		return ircchannel
 | 
			
		||||
	}
 | 
			
		||||
	ircchannel, ok := b.mmMap[channel]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		ircchannel = b.Config.IRC.Channel
 | 
			
		||||
	}
 | 
			
		||||
	return ircchannel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) ignoreMessage(nick string, message string, protocol string) bool {
 | 
			
		||||
	var ignoreNicks = b.mmIgnoreNicks
 | 
			
		||||
	if protocol == "irc" {
 | 
			
		||||
		ignoreNicks = b.ircIgnoreNicks
 | 
			
		||||
	}
 | 
			
		||||
	// should we discard messages ?
 | 
			
		||||
	for _, entry := range ignoreNicks {
 | 
			
		||||
		if nick == entry {
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										68
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										68
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,68 @@
 | 
			
		||||
package bridge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"gopkg.in/gcfg.v1"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	IRC struct {
 | 
			
		||||
		UseTLS            bool
 | 
			
		||||
		SkipTLSVerify     bool
 | 
			
		||||
		Server            string
 | 
			
		||||
		Port              int
 | 
			
		||||
		Nick              string
 | 
			
		||||
		Password          string
 | 
			
		||||
		Channel           string
 | 
			
		||||
		UseSlackCircumfix bool
 | 
			
		||||
		NickServNick      string
 | 
			
		||||
		NickServPassword  string
 | 
			
		||||
		RemoteNickFormat  string
 | 
			
		||||
		IgnoreNicks       string
 | 
			
		||||
	}
 | 
			
		||||
	Mattermost struct {
 | 
			
		||||
		URL                    string
 | 
			
		||||
		Port                   int
 | 
			
		||||
		ShowJoinPart           bool
 | 
			
		||||
		Token                  string
 | 
			
		||||
		IconURL                string
 | 
			
		||||
		SkipTLSVerify          bool
 | 
			
		||||
		BindAddress            string
 | 
			
		||||
		Channel                string
 | 
			
		||||
		PrefixMessagesWithNick bool
 | 
			
		||||
		NicksPerRow            int
 | 
			
		||||
		NickFormatter          string
 | 
			
		||||
		Server                 string
 | 
			
		||||
		Team                   string
 | 
			
		||||
		Login                  string
 | 
			
		||||
		Password               string
 | 
			
		||||
		RemoteNickFormat       *string
 | 
			
		||||
		IgnoreNicks            string
 | 
			
		||||
		NoTLS                  bool
 | 
			
		||||
	}
 | 
			
		||||
	Token map[string]*struct {
 | 
			
		||||
		IRCChannel string
 | 
			
		||||
		MMChannel  string
 | 
			
		||||
	}
 | 
			
		||||
	Channel map[string]*struct {
 | 
			
		||||
		IRC        string
 | 
			
		||||
		Mattermost string
 | 
			
		||||
	}
 | 
			
		||||
	General struct {
 | 
			
		||||
		GiphyAPIKey string
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewConfig(cfgfile string) *Config {
 | 
			
		||||
	var cfg Config
 | 
			
		||||
	content, err := ioutil.ReadFile(cfgfile)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	err = gcfg.ReadStringInto(&cfg, string(content))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal("Failed to parse "+cfgfile+":", err)
 | 
			
		||||
	}
 | 
			
		||||
	return &cfg
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										59
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/helper.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/helper.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,59 @@
 | 
			
		||||
package bridge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func tableformatter(nicks []string, nicksPerRow int, continued bool) string {
 | 
			
		||||
	result := "|IRC users"
 | 
			
		||||
	if continued {
 | 
			
		||||
		result = "|(continued)"
 | 
			
		||||
	}
 | 
			
		||||
	for i := 0; i < 2; i++ {
 | 
			
		||||
		for j := 1; j <= nicksPerRow && j <= len(nicks); j++ {
 | 
			
		||||
			if i == 0 {
 | 
			
		||||
				result += "|"
 | 
			
		||||
			} else {
 | 
			
		||||
				result += ":-|"
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		result += "\r\n|"
 | 
			
		||||
	}
 | 
			
		||||
	result += nicks[0] + "|"
 | 
			
		||||
	for i := 1; i < len(nicks); i++ {
 | 
			
		||||
		if i%nicksPerRow == 0 {
 | 
			
		||||
			result += "\r\n|" + nicks[i] + "|"
 | 
			
		||||
		} else {
 | 
			
		||||
			result += nicks[i] + "|"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func plainformatter(nicks []string, nicksPerRow int) string {
 | 
			
		||||
	return strings.Join(nicks, ", ") + " currently on IRC"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func IsMarkup(message string) bool {
 | 
			
		||||
	switch message[0] {
 | 
			
		||||
	case '|':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '#':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '_':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '*':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '~':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '-':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case ':':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '>':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '=':
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										202
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										202
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,202 @@
 | 
			
		||||
                                 Apache License
 | 
			
		||||
                           Version 2.0, January 2004
 | 
			
		||||
                        http://www.apache.org/licenses/
 | 
			
		||||
 | 
			
		||||
   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
   1. Definitions.
 | 
			
		||||
 | 
			
		||||
      "License" shall mean the terms and conditions for use, reproduction,
 | 
			
		||||
      and distribution as defined by Sections 1 through 9 of this document.
 | 
			
		||||
 | 
			
		||||
      "Licensor" shall mean the copyright owner or entity authorized by
 | 
			
		||||
      the copyright owner that is granting the License.
 | 
			
		||||
 | 
			
		||||
      "Legal Entity" shall mean the union of the acting entity and all
 | 
			
		||||
      other entities that control, are controlled by, or are under common
 | 
			
		||||
      control with that entity. For the purposes of this definition,
 | 
			
		||||
      "control" means (i) the power, direct or indirect, to cause the
 | 
			
		||||
      direction or management of such entity, whether by contract or
 | 
			
		||||
      otherwise, or (ii) ownership of fifty percent (50%) or more of the
 | 
			
		||||
      outstanding shares, or (iii) beneficial ownership of such entity.
 | 
			
		||||
 | 
			
		||||
      "You" (or "Your") shall mean an individual or Legal Entity
 | 
			
		||||
      exercising permissions granted by this License.
 | 
			
		||||
 | 
			
		||||
      "Source" form shall mean the preferred form for making modifications,
 | 
			
		||||
      including but not limited to software source code, documentation
 | 
			
		||||
      source, and configuration files.
 | 
			
		||||
 | 
			
		||||
      "Object" form shall mean any form resulting from mechanical
 | 
			
		||||
      transformation or translation of a Source form, including but
 | 
			
		||||
      not limited to compiled object code, generated documentation,
 | 
			
		||||
      and conversions to other media types.
 | 
			
		||||
 | 
			
		||||
      "Work" shall mean the work of authorship, whether in Source or
 | 
			
		||||
      Object form, made available under the License, as indicated by a
 | 
			
		||||
      copyright notice that is included in or attached to the work
 | 
			
		||||
      (an example is provided in the Appendix below).
 | 
			
		||||
 | 
			
		||||
      "Derivative Works" shall mean any work, whether in Source or Object
 | 
			
		||||
      form, that is based on (or derived from) the Work and for which the
 | 
			
		||||
      editorial revisions, annotations, elaborations, or other modifications
 | 
			
		||||
      represent, as a whole, an original work of authorship. For the purposes
 | 
			
		||||
      of this License, Derivative Works shall not include works that remain
 | 
			
		||||
      separable from, or merely link (or bind by name) to the interfaces of,
 | 
			
		||||
      the Work and Derivative Works thereof.
 | 
			
		||||
 | 
			
		||||
      "Contribution" shall mean any work of authorship, including
 | 
			
		||||
      the original version of the Work and any modifications or additions
 | 
			
		||||
      to that Work or Derivative Works thereof, that is intentionally
 | 
			
		||||
      submitted to Licensor for inclusion in the Work by the copyright owner
 | 
			
		||||
      or by an individual or Legal Entity authorized to submit on behalf of
 | 
			
		||||
      the copyright owner. For the purposes of this definition, "submitted"
 | 
			
		||||
      means any form of electronic, verbal, or written communication sent
 | 
			
		||||
      to the Licensor or its representatives, including but not limited to
 | 
			
		||||
      communication on electronic mailing lists, source code control systems,
 | 
			
		||||
      and issue tracking systems that are managed by, or on behalf of, the
 | 
			
		||||
      Licensor for the purpose of discussing and improving the Work, but
 | 
			
		||||
      excluding communication that is conspicuously marked or otherwise
 | 
			
		||||
      designated in writing by the copyright owner as "Not a Contribution."
 | 
			
		||||
 | 
			
		||||
      "Contributor" shall mean Licensor and any individual or Legal Entity
 | 
			
		||||
      on behalf of whom a Contribution has been received by Licensor and
 | 
			
		||||
      subsequently incorporated within the Work.
 | 
			
		||||
 | 
			
		||||
   2. Grant of Copyright License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      copyright license to reproduce, prepare Derivative Works of,
 | 
			
		||||
      publicly display, publicly perform, sublicense, and distribute the
 | 
			
		||||
      Work and such Derivative Works in Source or Object form.
 | 
			
		||||
 | 
			
		||||
   3. Grant of Patent License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      (except as stated in this section) patent license to make, have made,
 | 
			
		||||
      use, offer to sell, sell, import, and otherwise transfer the Work,
 | 
			
		||||
      where such license applies only to those patent claims licensable
 | 
			
		||||
      by such Contributor that are necessarily infringed by their
 | 
			
		||||
      Contribution(s) alone or by combination of their Contribution(s)
 | 
			
		||||
      with the Work to which such Contribution(s) was submitted. If You
 | 
			
		||||
      institute patent litigation against any entity (including a
 | 
			
		||||
      cross-claim or counterclaim in a lawsuit) alleging that the Work
 | 
			
		||||
      or a Contribution incorporated within the Work constitutes direct
 | 
			
		||||
      or contributory patent infringement, then any patent licenses
 | 
			
		||||
      granted to You under this License for that Work shall terminate
 | 
			
		||||
      as of the date such litigation is filed.
 | 
			
		||||
 | 
			
		||||
   4. Redistribution. You may reproduce and distribute copies of the
 | 
			
		||||
      Work or Derivative Works thereof in any medium, with or without
 | 
			
		||||
      modifications, and in Source or Object form, provided that You
 | 
			
		||||
      meet the following conditions:
 | 
			
		||||
 | 
			
		||||
      (a) You must give any other recipients of the Work or
 | 
			
		||||
          Derivative Works a copy of this License; and
 | 
			
		||||
 | 
			
		||||
      (b) You must cause any modified files to carry prominent notices
 | 
			
		||||
          stating that You changed the files; and
 | 
			
		||||
 | 
			
		||||
      (c) You must retain, in the Source form of any Derivative Works
 | 
			
		||||
          that You distribute, all copyright, patent, trademark, and
 | 
			
		||||
          attribution notices from the Source form of the Work,
 | 
			
		||||
          excluding those notices that do not pertain to any part of
 | 
			
		||||
          the Derivative Works; and
 | 
			
		||||
 | 
			
		||||
      (d) If the Work includes a "NOTICE" text file as part of its
 | 
			
		||||
          distribution, then any Derivative Works that You distribute must
 | 
			
		||||
          include a readable copy of the attribution notices contained
 | 
			
		||||
          within such NOTICE file, excluding those notices that do not
 | 
			
		||||
          pertain to any part of the Derivative Works, in at least one
 | 
			
		||||
          of the following places: within a NOTICE text file distributed
 | 
			
		||||
          as part of the Derivative Works; within the Source form or
 | 
			
		||||
          documentation, if provided along with the Derivative Works; or,
 | 
			
		||||
          within a display generated by the Derivative Works, if and
 | 
			
		||||
          wherever such third-party notices normally appear. The contents
 | 
			
		||||
          of the NOTICE file are for informational purposes only and
 | 
			
		||||
          do not modify the License. You may add Your own attribution
 | 
			
		||||
          notices within Derivative Works that You distribute, alongside
 | 
			
		||||
          or as an addendum to the NOTICE text from the Work, provided
 | 
			
		||||
          that such additional attribution notices cannot be construed
 | 
			
		||||
          as modifying the License.
 | 
			
		||||
 | 
			
		||||
      You may add Your own copyright statement to Your modifications and
 | 
			
		||||
      may provide additional or different license terms and conditions
 | 
			
		||||
      for use, reproduction, or distribution of Your modifications, or
 | 
			
		||||
      for any such Derivative Works as a whole, provided Your use,
 | 
			
		||||
      reproduction, and distribution of the Work otherwise complies with
 | 
			
		||||
      the conditions stated in this License.
 | 
			
		||||
 | 
			
		||||
   5. Submission of Contributions. Unless You explicitly state otherwise,
 | 
			
		||||
      any Contribution intentionally submitted for inclusion in the Work
 | 
			
		||||
      by You to the Licensor shall be under the terms and conditions of
 | 
			
		||||
      this License, without any additional terms or conditions.
 | 
			
		||||
      Notwithstanding the above, nothing herein shall supersede or modify
 | 
			
		||||
      the terms of any separate license agreement you may have executed
 | 
			
		||||
      with Licensor regarding such Contributions.
 | 
			
		||||
 | 
			
		||||
   6. Trademarks. This License does not grant permission to use the trade
 | 
			
		||||
      names, trademarks, service marks, or product names of the Licensor,
 | 
			
		||||
      except as required for reasonable and customary use in describing the
 | 
			
		||||
      origin of the Work and reproducing the content of the NOTICE file.
 | 
			
		||||
 | 
			
		||||
   7. Disclaimer of Warranty. Unless required by applicable law or
 | 
			
		||||
      agreed to in writing, Licensor provides the Work (and each
 | 
			
		||||
      Contributor provides its Contributions) on an "AS IS" BASIS,
 | 
			
		||||
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 | 
			
		||||
      implied, including, without limitation, any warranties or conditions
 | 
			
		||||
      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
 | 
			
		||||
      PARTICULAR PURPOSE. You are solely responsible for determining the
 | 
			
		||||
      appropriateness of using or redistributing the Work and assume any
 | 
			
		||||
      risks associated with Your exercise of permissions under this License.
 | 
			
		||||
 | 
			
		||||
   8. Limitation of Liability. In no event and under no legal theory,
 | 
			
		||||
      whether in tort (including negligence), contract, or otherwise,
 | 
			
		||||
      unless required by applicable law (such as deliberate and grossly
 | 
			
		||||
      negligent acts) or agreed to in writing, shall any Contributor be
 | 
			
		||||
      liable to You for damages, including any direct, indirect, special,
 | 
			
		||||
      incidental, or consequential damages of any character arising as a
 | 
			
		||||
      result of this License or out of the use or inability to use the
 | 
			
		||||
      Work (including but not limited to damages for loss of goodwill,
 | 
			
		||||
      work stoppage, computer failure or malfunction, or any and all
 | 
			
		||||
      other commercial damages or losses), even if such Contributor
 | 
			
		||||
      has been advised of the possibility of such damages.
 | 
			
		||||
 | 
			
		||||
   9. Accepting Warranty or Additional Liability. While redistributing
 | 
			
		||||
      the Work or Derivative Works thereof, You may choose to offer,
 | 
			
		||||
      and charge a fee for, acceptance of support, warranty, indemnity,
 | 
			
		||||
      or other liability obligations and/or rights consistent with this
 | 
			
		||||
      License. However, in accepting such obligations, You may act only
 | 
			
		||||
      on Your own behalf and on Your sole responsibility, not on behalf
 | 
			
		||||
      of any other Contributor, and only if You agree to indemnify,
 | 
			
		||||
      defend, and hold each Contributor harmless for any liability
 | 
			
		||||
      incurred by, or claims asserted against, such Contributor by reason
 | 
			
		||||
      of your accepting any such warranty or additional liability.
 | 
			
		||||
 | 
			
		||||
   END OF TERMS AND CONDITIONS
 | 
			
		||||
 | 
			
		||||
   APPENDIX: How to apply the Apache License to your work.
 | 
			
		||||
 | 
			
		||||
      To apply the Apache License to your work, attach the following
 | 
			
		||||
      boilerplate notice, with the fields enclosed by brackets "{}"
 | 
			
		||||
      replaced with your own identifying information. (Don't include
 | 
			
		||||
      the brackets!)  The text should be enclosed in the appropriate
 | 
			
		||||
      comment syntax for the file format. We also recommend that a
 | 
			
		||||
      file or class name and description of purpose be included on the
 | 
			
		||||
      same "printed page" as the copyright notice for easier
 | 
			
		||||
      identification within third-party archives.
 | 
			
		||||
 | 
			
		||||
   Copyright {yyyy} {name of copyright owner}
 | 
			
		||||
 | 
			
		||||
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
   you may not use this file except in compliance with the License.
 | 
			
		||||
   You may obtain a copy of the License at
 | 
			
		||||
 | 
			
		||||
       http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 | 
			
		||||
   Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
   See the License for the specific language governing permissions and
 | 
			
		||||
   limitations under the License.
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										441
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/matterclient.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										441
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/matterclient.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,441 @@
 | 
			
		||||
package matterclient
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"errors"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/cookiejar"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/gorilla/websocket"
 | 
			
		||||
	"github.com/jpillora/backoff"
 | 
			
		||||
	"github.com/mattermost/platform/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Credentials struct {
 | 
			
		||||
	Login         string
 | 
			
		||||
	Team          string
 | 
			
		||||
	Pass          string
 | 
			
		||||
	Server        string
 | 
			
		||||
	NoTLS         bool
 | 
			
		||||
	SkipTLSVerify bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
	Raw      *model.Message
 | 
			
		||||
	Post     *model.Post
 | 
			
		||||
	Team     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	Text     string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMClient struct {
 | 
			
		||||
	*Credentials
 | 
			
		||||
	Client       *model.Client
 | 
			
		||||
	WsClient     *websocket.Conn
 | 
			
		||||
	WsQuit       bool
 | 
			
		||||
	WsAway       bool
 | 
			
		||||
	Channels     *model.ChannelList
 | 
			
		||||
	MoreChannels *model.ChannelList
 | 
			
		||||
	User         *model.User
 | 
			
		||||
	Users        map[string]*model.User
 | 
			
		||||
	MessageChan  chan *Message
 | 
			
		||||
	Team         *model.Team
 | 
			
		||||
	log          *log.Entry
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(login, pass, team, server string) *MMClient {
 | 
			
		||||
	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server}
 | 
			
		||||
	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100)}
 | 
			
		||||
	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"})
 | 
			
		||||
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
 | 
			
		||||
	return mmclient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SetLogLevel(level string) {
 | 
			
		||||
	l, err := log.ParseLevel(level)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.SetLevel(log.InfoLevel)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	log.SetLevel(l)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) Login() error {
 | 
			
		||||
	if m.WsQuit {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	b := &backoff.Backoff{
 | 
			
		||||
		Min:    time.Second,
 | 
			
		||||
		Max:    5 * time.Minute,
 | 
			
		||||
		Jitter: true,
 | 
			
		||||
	}
 | 
			
		||||
	uriScheme := "https://"
 | 
			
		||||
	wsScheme := "wss://"
 | 
			
		||||
	if m.NoTLS {
 | 
			
		||||
		uriScheme = "http://"
 | 
			
		||||
		wsScheme = "ws://"
 | 
			
		||||
	}
 | 
			
		||||
	// login to mattermost
 | 
			
		||||
	m.Client = model.NewClient(uriScheme + m.Credentials.Server)
 | 
			
		||||
	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}}
 | 
			
		||||
	var myinfo *model.Result
 | 
			
		||||
	var appErr *model.AppError
 | 
			
		||||
	var logmsg = "trying login"
 | 
			
		||||
	for {
 | 
			
		||||
		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) {
 | 
			
		||||
			m.log.Debugf(logmsg+" with ", model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=")
 | 
			
		||||
			m.Client.HttpClient.Jar = m.createCookieJar(token[1])
 | 
			
		||||
			m.Client.MockSession(token[1])
 | 
			
		||||
			myinfo, appErr = m.Client.GetMe("")
 | 
			
		||||
			if myinfo.Data.(*model.User) == nil {
 | 
			
		||||
				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass)
 | 
			
		||||
				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			}
 | 
			
		||||
		} else {
 | 
			
		||||
			myinfo, appErr = m.Client.Login(m.Credentials.Login, m.Credentials.Pass)
 | 
			
		||||
		}
 | 
			
		||||
		if appErr != nil {
 | 
			
		||||
			d := b.Duration()
 | 
			
		||||
			m.log.Debug(appErr.DetailedError)
 | 
			
		||||
			if !strings.Contains(appErr.DetailedError, "connection refused") &&
 | 
			
		||||
				!strings.Contains(appErr.DetailedError, "invalid character") {
 | 
			
		||||
				if appErr.Message == "" {
 | 
			
		||||
					return errors.New(appErr.DetailedError)
 | 
			
		||||
				}
 | 
			
		||||
				return errors.New(appErr.Message)
 | 
			
		||||
			}
 | 
			
		||||
			m.log.Debugf("LOGIN: %s, reconnecting in %s", appErr, d)
 | 
			
		||||
			time.Sleep(d)
 | 
			
		||||
			logmsg = "retrying login"
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		break
 | 
			
		||||
	}
 | 
			
		||||
	// reset timer
 | 
			
		||||
	b.Reset()
 | 
			
		||||
 | 
			
		||||
	initLoad, _ := m.Client.GetInitialLoad()
 | 
			
		||||
	initData := initLoad.Data.(*model.InitialLoad)
 | 
			
		||||
	m.User = initData.User
 | 
			
		||||
	for _, v := range initData.Teams {
 | 
			
		||||
		m.log.Debugf("trying %s (id: %s)", v.Name, v.Id)
 | 
			
		||||
		if v.Name == m.Credentials.Team {
 | 
			
		||||
			m.Client.SetTeamId(v.Id)
 | 
			
		||||
			m.Team = v
 | 
			
		||||
			m.log.Debugf("GetallTeamListings: found id %s for team %s", v.Id, v.Name)
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if m.Team == nil {
 | 
			
		||||
		return errors.New("team not found")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// setup websocket connection
 | 
			
		||||
	wsurl := wsScheme + m.Credentials.Server + "/api/v3/users/websocket"
 | 
			
		||||
	header := http.Header{}
 | 
			
		||||
	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken)
 | 
			
		||||
 | 
			
		||||
	m.log.Debug("WsClient: making connection")
 | 
			
		||||
	var err error
 | 
			
		||||
	for {
 | 
			
		||||
		wsDialer := &websocket.Dialer{Proxy: http.ProxyFromEnvironment, TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}}
 | 
			
		||||
		m.WsClient, _, err = wsDialer.Dial(wsurl, header)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			d := b.Duration()
 | 
			
		||||
			m.log.Debugf("WSS: %s, reconnecting in %s", err, d)
 | 
			
		||||
			time.Sleep(d)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		break
 | 
			
		||||
	}
 | 
			
		||||
	b.Reset()
 | 
			
		||||
 | 
			
		||||
	// populating users
 | 
			
		||||
	m.UpdateUsers()
 | 
			
		||||
 | 
			
		||||
	// populating channels
 | 
			
		||||
	m.UpdateChannels()
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) WsReceiver() {
 | 
			
		||||
	var rmsg model.Message
 | 
			
		||||
	for {
 | 
			
		||||
		if m.WsQuit {
 | 
			
		||||
			m.log.Debug("exiting WsReceiver")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		if err := m.WsClient.ReadJSON(&rmsg); err != nil {
 | 
			
		||||
			m.log.Error("error:", err)
 | 
			
		||||
			// reconnect
 | 
			
		||||
			m.Login()
 | 
			
		||||
		}
 | 
			
		||||
		if rmsg.Action == "ping" {
 | 
			
		||||
			m.handleWsPing()
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		msg := &Message{Raw: &rmsg, Team: m.Credentials.Team}
 | 
			
		||||
		m.parseMessage(msg)
 | 
			
		||||
		m.MessageChan <- msg
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) handleWsPing() {
 | 
			
		||||
	m.log.Debug("Ws PING")
 | 
			
		||||
	if !m.WsQuit && !m.WsAway {
 | 
			
		||||
		m.log.Debug("Ws PONG")
 | 
			
		||||
		m.WsClient.WriteMessage(websocket.PongMessage, []byte{})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseMessage(rmsg *Message) {
 | 
			
		||||
	switch rmsg.Raw.Action {
 | 
			
		||||
	case model.ACTION_POSTED:
 | 
			
		||||
		m.parseActionPost(rmsg)
 | 
			
		||||
		/*
 | 
			
		||||
			case model.ACTION_USER_REMOVED:
 | 
			
		||||
				m.handleWsActionUserRemoved(&rmsg)
 | 
			
		||||
			case model.ACTION_USER_ADDED:
 | 
			
		||||
				m.handleWsActionUserAdded(&rmsg)
 | 
			
		||||
		*/
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseActionPost(rmsg *Message) {
 | 
			
		||||
	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Props["post"]))
 | 
			
		||||
	//	log.Println("receiving userid", data.UserId)
 | 
			
		||||
	// we don't have the user, refresh the userlist
 | 
			
		||||
	if m.Users[data.UserId] == nil {
 | 
			
		||||
		m.UpdateUsers()
 | 
			
		||||
	}
 | 
			
		||||
	rmsg.Username = m.Users[data.UserId].Username
 | 
			
		||||
	rmsg.Channel = m.GetChannelName(data.ChannelId)
 | 
			
		||||
	// direct message
 | 
			
		||||
	if strings.Contains(rmsg.Channel, "__") {
 | 
			
		||||
		//log.Println("direct message")
 | 
			
		||||
		rcvusers := strings.Split(rmsg.Channel, "__")
 | 
			
		||||
		if rcvusers[0] != m.User.Id {
 | 
			
		||||
			rmsg.Channel = m.Users[rcvusers[0]].Username
 | 
			
		||||
		} else {
 | 
			
		||||
			rmsg.Channel = m.Users[rcvusers[1]].Username
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	rmsg.Text = data.Message
 | 
			
		||||
	rmsg.Post = data
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateUsers() error {
 | 
			
		||||
	mmusers, _ := m.Client.GetProfilesForDirectMessageList(m.Team.Id)
 | 
			
		||||
	m.Users = mmusers.Data.(map[string]*model.User)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannels() error {
 | 
			
		||||
	mmchannels, _ := m.Client.GetChannels("")
 | 
			
		||||
	m.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	mmchannels, _ = m.Client.GetMoreChannels("")
 | 
			
		||||
	m.MoreChannels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelName(id string) string {
 | 
			
		||||
	for _, channel := range append(m.Channels.Channels, m.MoreChannels.Channels...) {
 | 
			
		||||
		if channel.Id == id {
 | 
			
		||||
			return channel.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	// not found? could be a new direct message from mattermost. Try to update and check again
 | 
			
		||||
	m.UpdateChannels()
 | 
			
		||||
	for _, channel := range append(m.Channels.Channels, m.MoreChannels.Channels...) {
 | 
			
		||||
		if channel.Id == id {
 | 
			
		||||
			return channel.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelId(name string) string {
 | 
			
		||||
	for _, channel := range append(m.Channels.Channels, m.MoreChannels.Channels...) {
 | 
			
		||||
		if channel.Name == name {
 | 
			
		||||
			return channel.Id
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetChannelHeader(id string) string {
 | 
			
		||||
	for _, channel := range append(m.Channels.Channels, m.MoreChannels.Channels...) {
 | 
			
		||||
		if channel.Id == id {
 | 
			
		||||
			return channel.Header
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) PostMessage(channel string, text string) {
 | 
			
		||||
	post := &model.Post{ChannelId: m.GetChannelId(channel), Message: text}
 | 
			
		||||
	m.Client.CreatePost(post)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) JoinChannel(channel string) error {
 | 
			
		||||
	cleanChan := strings.Replace(channel, "#", "", 1)
 | 
			
		||||
	if m.GetChannelId(cleanChan) == "" {
 | 
			
		||||
		return errors.New("failed to join")
 | 
			
		||||
	}
 | 
			
		||||
	for _, c := range m.Channels.Channels {
 | 
			
		||||
		if c.Name == cleanChan {
 | 
			
		||||
			m.log.Debug("Not joining ", cleanChan, " already joined.")
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	m.log.Debug("Joining ", cleanChan)
 | 
			
		||||
	_, err := m.Client.JoinChannel(m.GetChannelId(cleanChan))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New("failed to join")
 | 
			
		||||
	}
 | 
			
		||||
	//	m.SyncChannel(m.getMMChannelId(strings.Replace(channel, "#", "", 1)), strings.Replace(channel, "#", "", 1))
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPostsSince(channelId, time)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SearchPosts(query string) *model.PostList {
 | 
			
		||||
	res, err := m.Client.SearchPosts(query, false)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPosts(channelId, 0, limit, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPublicLink(filename string) string {
 | 
			
		||||
	res, err := m.Client.GetPublicLink(filename)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(string)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPublicLinks(filenames []string) []string {
 | 
			
		||||
	var output []string
 | 
			
		||||
	for _, f := range filenames {
 | 
			
		||||
		res, err := m.Client.GetPublicLink(f)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		output = append(output, res.Data.(string))
 | 
			
		||||
	}
 | 
			
		||||
	return output
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannelHeader(channelId string, header string) {
 | 
			
		||||
	data := make(map[string]string)
 | 
			
		||||
	data["channel_id"] = channelId
 | 
			
		||||
	data["channel_header"] = header
 | 
			
		||||
	m.log.Debugf("updating channelheader %#v, %#v", channelId, header)
 | 
			
		||||
	_, err := m.Client.UpdateChannelHeader(data)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateLastViewed(channelId string) {
 | 
			
		||||
	m.log.Debugf("posting lastview %#v", channelId)
 | 
			
		||||
	_, err := m.Client.UpdateLastViewedAt(channelId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Error(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UsernamesInChannel(channelName string) []string {
 | 
			
		||||
	ceiRes, err := m.Client.GetChannelExtraInfo(m.GetChannelId(channelName), 5000, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelName, err)
 | 
			
		||||
		return []string{}
 | 
			
		||||
	}
 | 
			
		||||
	extra := ceiRes.Data.(*model.ChannelExtra)
 | 
			
		||||
	result := []string{}
 | 
			
		||||
	for _, member := range extra.Members {
 | 
			
		||||
		result = append(result, member.Username)
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) createCookieJar(token string) *cookiejar.Jar {
 | 
			
		||||
	var cookies []*http.Cookie
 | 
			
		||||
	jar, _ := cookiejar.New(nil)
 | 
			
		||||
	firstCookie := &http.Cookie{
 | 
			
		||||
		Name:   "MMAUTHTOKEN",
 | 
			
		||||
		Value:  token,
 | 
			
		||||
		Path:   "/",
 | 
			
		||||
		Domain: m.Credentials.Server,
 | 
			
		||||
	}
 | 
			
		||||
	cookies = append(cookies, firstCookie)
 | 
			
		||||
	cookieURL, _ := url.Parse("https://" + m.Credentials.Server)
 | 
			
		||||
	jar.SetCookies(cookieURL, cookies)
 | 
			
		||||
	return jar
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetOtherUserDM(channel string) *model.User {
 | 
			
		||||
	m.UpdateUsers()
 | 
			
		||||
	var rcvuser *model.User
 | 
			
		||||
	if strings.Contains(channel, "__") {
 | 
			
		||||
		rcvusers := strings.Split(channel, "__")
 | 
			
		||||
		if rcvusers[0] != m.User.Id {
 | 
			
		||||
			rcvuser = m.Users[rcvusers[0]]
 | 
			
		||||
		} else {
 | 
			
		||||
			rcvuser = m.Users[rcvusers[1]]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return rcvuser
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SendDirectMessage(toUserId string, msg string) {
 | 
			
		||||
	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg)
 | 
			
		||||
	var channel string
 | 
			
		||||
	// We don't have a DM with this user yet.
 | 
			
		||||
	if m.GetChannelId(toUserId+"__"+m.User.Id) == "" && m.GetChannelId(m.User.Id+"__"+toUserId) == "" {
 | 
			
		||||
		// create DM channel
 | 
			
		||||
		_, err := m.Client.CreateDirectChannel(toUserId)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err)
 | 
			
		||||
		}
 | 
			
		||||
		// update our channels
 | 
			
		||||
		mmchannels, _ := m.Client.GetChannels("")
 | 
			
		||||
		m.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// build the channel name
 | 
			
		||||
	if toUserId > m.User.Id {
 | 
			
		||||
		channel = m.User.Id + "__" + toUserId
 | 
			
		||||
	} else {
 | 
			
		||||
		channel = toUserId + "__" + m.User.Id
 | 
			
		||||
	}
 | 
			
		||||
	// build & send the message
 | 
			
		||||
	msg = strings.Replace(msg, "\r", "", -1)
 | 
			
		||||
	post := &model.Post{ChannelId: m.GetChannelId(channel), Message: msg}
 | 
			
		||||
	m.Client.CreatePost(post)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										14
									
								
								vendor/github.com/BurntSushi/toml/COPYING
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/BurntSushi/toml/COPYING
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
 | 
			
		||||
                    Version 2, December 2004
 | 
			
		||||
 | 
			
		||||
 Copyright (C) 2004 Sam Hocevar <sam@hocevar.net>
 | 
			
		||||
 | 
			
		||||
 Everyone is permitted to copy and distribute verbatim or modified
 | 
			
		||||
 copies of this license document, and changing it is allowed as long
 | 
			
		||||
 as the name is changed.
 | 
			
		||||
 | 
			
		||||
            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
 | 
			
		||||
   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
 | 
			
		||||
 | 
			
		||||
  0. You just DO WHAT THE FUCK YOU WANT TO.
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										90
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-decoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										90
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-decoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,90 @@
 | 
			
		||||
// Command toml-test-decoder satisfies the toml-test interface for testing
 | 
			
		||||
// TOML decoders. Namely, it accepts TOML on stdin and outputs JSON on stdout.
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"log"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/BurntSushi/toml"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.SetFlags(0)
 | 
			
		||||
 | 
			
		||||
	flag.Usage = usage
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func usage() {
 | 
			
		||||
	log.Printf("Usage: %s < toml-file\n", path.Base(os.Args[0]))
 | 
			
		||||
	flag.PrintDefaults()
 | 
			
		||||
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	if flag.NArg() != 0 {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var tmp interface{}
 | 
			
		||||
	if _, err := toml.DecodeReader(os.Stdin, &tmp); err != nil {
 | 
			
		||||
		log.Fatalf("Error decoding TOML: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	typedTmp := translate(tmp)
 | 
			
		||||
	if err := json.NewEncoder(os.Stdout).Encode(typedTmp); err != nil {
 | 
			
		||||
		log.Fatalf("Error encoding JSON: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func translate(tomlData interface{}) interface{} {
 | 
			
		||||
	switch orig := tomlData.(type) {
 | 
			
		||||
	case map[string]interface{}:
 | 
			
		||||
		typed := make(map[string]interface{}, len(orig))
 | 
			
		||||
		for k, v := range orig {
 | 
			
		||||
			typed[k] = translate(v)
 | 
			
		||||
		}
 | 
			
		||||
		return typed
 | 
			
		||||
	case []map[string]interface{}:
 | 
			
		||||
		typed := make([]map[string]interface{}, len(orig))
 | 
			
		||||
		for i, v := range orig {
 | 
			
		||||
			typed[i] = translate(v).(map[string]interface{})
 | 
			
		||||
		}
 | 
			
		||||
		return typed
 | 
			
		||||
	case []interface{}:
 | 
			
		||||
		typed := make([]interface{}, len(orig))
 | 
			
		||||
		for i, v := range orig {
 | 
			
		||||
			typed[i] = translate(v)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// We don't really need to tag arrays, but let's be future proof.
 | 
			
		||||
		// (If TOML ever supports tuples, we'll need this.)
 | 
			
		||||
		return tag("array", typed)
 | 
			
		||||
	case time.Time:
 | 
			
		||||
		return tag("datetime", orig.Format("2006-01-02T15:04:05Z"))
 | 
			
		||||
	case bool:
 | 
			
		||||
		return tag("bool", fmt.Sprintf("%v", orig))
 | 
			
		||||
	case int64:
 | 
			
		||||
		return tag("integer", fmt.Sprintf("%d", orig))
 | 
			
		||||
	case float64:
 | 
			
		||||
		return tag("float", fmt.Sprintf("%v", orig))
 | 
			
		||||
	case string:
 | 
			
		||||
		return tag("string", orig)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	panic(fmt.Sprintf("Unknown type: %T", tomlData))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func tag(typeName string, data interface{}) map[string]interface{} {
 | 
			
		||||
	return map[string]interface{}{
 | 
			
		||||
		"type":  typeName,
 | 
			
		||||
		"value": data,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										131
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-encoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										131
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-encoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,131 @@
 | 
			
		||||
// Command toml-test-encoder satisfies the toml-test interface for testing
 | 
			
		||||
// TOML encoders. Namely, it accepts JSON on stdin and outputs TOML on stdout.
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"log"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/BurntSushi/toml"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.SetFlags(0)
 | 
			
		||||
 | 
			
		||||
	flag.Usage = usage
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func usage() {
 | 
			
		||||
	log.Printf("Usage: %s < json-file\n", path.Base(os.Args[0]))
 | 
			
		||||
	flag.PrintDefaults()
 | 
			
		||||
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	if flag.NArg() != 0 {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var tmp interface{}
 | 
			
		||||
	if err := json.NewDecoder(os.Stdin).Decode(&tmp); err != nil {
 | 
			
		||||
		log.Fatalf("Error decoding JSON: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	tomlData := translate(tmp)
 | 
			
		||||
	if err := toml.NewEncoder(os.Stdout).Encode(tomlData); err != nil {
 | 
			
		||||
		log.Fatalf("Error encoding TOML: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func translate(typedJson interface{}) interface{} {
 | 
			
		||||
	switch v := typedJson.(type) {
 | 
			
		||||
	case map[string]interface{}:
 | 
			
		||||
		if len(v) == 2 && in("type", v) && in("value", v) {
 | 
			
		||||
			return untag(v)
 | 
			
		||||
		}
 | 
			
		||||
		m := make(map[string]interface{}, len(v))
 | 
			
		||||
		for k, v2 := range v {
 | 
			
		||||
			m[k] = translate(v2)
 | 
			
		||||
		}
 | 
			
		||||
		return m
 | 
			
		||||
	case []interface{}:
 | 
			
		||||
		tabArray := make([]map[string]interface{}, len(v))
 | 
			
		||||
		for i := range v {
 | 
			
		||||
			if m, ok := translate(v[i]).(map[string]interface{}); ok {
 | 
			
		||||
				tabArray[i] = m
 | 
			
		||||
			} else {
 | 
			
		||||
				log.Fatalf("JSON arrays may only contain objects. This " +
 | 
			
		||||
					"corresponds to only tables being allowed in " +
 | 
			
		||||
					"TOML table arrays.")
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return tabArray
 | 
			
		||||
	}
 | 
			
		||||
	log.Fatalf("Unrecognized JSON format '%T'.", typedJson)
 | 
			
		||||
	panic("unreachable")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func untag(typed map[string]interface{}) interface{} {
 | 
			
		||||
	t := typed["type"].(string)
 | 
			
		||||
	v := typed["value"]
 | 
			
		||||
	switch t {
 | 
			
		||||
	case "string":
 | 
			
		||||
		return v.(string)
 | 
			
		||||
	case "integer":
 | 
			
		||||
		v := v.(string)
 | 
			
		||||
		n, err := strconv.Atoi(v)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Could not parse '%s' as integer: %s", v, err)
 | 
			
		||||
		}
 | 
			
		||||
		return n
 | 
			
		||||
	case "float":
 | 
			
		||||
		v := v.(string)
 | 
			
		||||
		f, err := strconv.ParseFloat(v, 64)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Could not parse '%s' as float64: %s", v, err)
 | 
			
		||||
		}
 | 
			
		||||
		return f
 | 
			
		||||
	case "datetime":
 | 
			
		||||
		v := v.(string)
 | 
			
		||||
		t, err := time.Parse("2006-01-02T15:04:05Z", v)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Could not parse '%s' as a datetime: %s", v, err)
 | 
			
		||||
		}
 | 
			
		||||
		return t
 | 
			
		||||
	case "bool":
 | 
			
		||||
		v := v.(string)
 | 
			
		||||
		switch v {
 | 
			
		||||
		case "true":
 | 
			
		||||
			return true
 | 
			
		||||
		case "false":
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
		log.Fatalf("Could not parse '%s' as a boolean.", v)
 | 
			
		||||
	case "array":
 | 
			
		||||
		v := v.([]interface{})
 | 
			
		||||
		array := make([]interface{}, len(v))
 | 
			
		||||
		for i := range v {
 | 
			
		||||
			if m, ok := v[i].(map[string]interface{}); ok {
 | 
			
		||||
				array[i] = untag(m)
 | 
			
		||||
			} else {
 | 
			
		||||
				log.Fatalf("Arrays may only contain other arrays or "+
 | 
			
		||||
					"primitive values, but found a '%T'.", m)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return array
 | 
			
		||||
	}
 | 
			
		||||
	log.Fatalf("Unrecognized tag type '%s'.", t)
 | 
			
		||||
	panic("unreachable")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func in(key string, m map[string]interface{}) bool {
 | 
			
		||||
	_, ok := m[key]
 | 
			
		||||
	return ok
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										61
									
								
								vendor/github.com/BurntSushi/toml/cmd/tomlv/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										61
									
								
								vendor/github.com/BurntSushi/toml/cmd/tomlv/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,61 @@
 | 
			
		||||
// Command tomlv validates TOML documents and prints each key's type.
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"log"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"text/tabwriter"
 | 
			
		||||
 | 
			
		||||
	"github.com/BurntSushi/toml"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	flagTypes = false
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.SetFlags(0)
 | 
			
		||||
 | 
			
		||||
	flag.BoolVar(&flagTypes, "types", flagTypes,
 | 
			
		||||
		"When set, the types of every defined key will be shown.")
 | 
			
		||||
 | 
			
		||||
	flag.Usage = usage
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func usage() {
 | 
			
		||||
	log.Printf("Usage: %s toml-file [ toml-file ... ]\n",
 | 
			
		||||
		path.Base(os.Args[0]))
 | 
			
		||||
	flag.PrintDefaults()
 | 
			
		||||
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	if flag.NArg() < 1 {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
	}
 | 
			
		||||
	for _, f := range flag.Args() {
 | 
			
		||||
		var tmp interface{}
 | 
			
		||||
		md, err := toml.DecodeFile(f, &tmp)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Error in '%s': %s", f, err)
 | 
			
		||||
		}
 | 
			
		||||
		if flagTypes {
 | 
			
		||||
			printTypes(md)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func printTypes(md toml.MetaData) {
 | 
			
		||||
	tabw := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
 | 
			
		||||
	for _, key := range md.Keys() {
 | 
			
		||||
		fmt.Fprintf(tabw, "%s%s\t%s\n",
 | 
			
		||||
			strings.Repeat("    ", len(key)-1), key, md.Type(key...))
 | 
			
		||||
	}
 | 
			
		||||
	tabw.Flush()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										509
									
								
								vendor/github.com/BurntSushi/toml/decode.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										509
									
								
								vendor/github.com/BurntSushi/toml/decode.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,509 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"math"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func e(format string, args ...interface{}) error {
 | 
			
		||||
	return fmt.Errorf("toml: "+format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Unmarshaler is the interface implemented by objects that can unmarshal a
 | 
			
		||||
// TOML description of themselves.
 | 
			
		||||
type Unmarshaler interface {
 | 
			
		||||
	UnmarshalTOML(interface{}) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Unmarshal decodes the contents of `p` in TOML format into a pointer `v`.
 | 
			
		||||
func Unmarshal(p []byte, v interface{}) error {
 | 
			
		||||
	_, err := Decode(string(p), v)
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Primitive is a TOML value that hasn't been decoded into a Go value.
 | 
			
		||||
// When using the various `Decode*` functions, the type `Primitive` may
 | 
			
		||||
// be given to any value, and its decoding will be delayed.
 | 
			
		||||
//
 | 
			
		||||
// A `Primitive` value can be decoded using the `PrimitiveDecode` function.
 | 
			
		||||
//
 | 
			
		||||
// The underlying representation of a `Primitive` value is subject to change.
 | 
			
		||||
// Do not rely on it.
 | 
			
		||||
//
 | 
			
		||||
// N.B. Primitive values are still parsed, so using them will only avoid
 | 
			
		||||
// the overhead of reflection. They can be useful when you don't know the
 | 
			
		||||
// exact type of TOML data until run time.
 | 
			
		||||
type Primitive struct {
 | 
			
		||||
	undecoded interface{}
 | 
			
		||||
	context   Key
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// DEPRECATED!
 | 
			
		||||
//
 | 
			
		||||
// Use MetaData.PrimitiveDecode instead.
 | 
			
		||||
func PrimitiveDecode(primValue Primitive, v interface{}) error {
 | 
			
		||||
	md := MetaData{decoded: make(map[string]bool)}
 | 
			
		||||
	return md.unify(primValue.undecoded, rvalue(v))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// PrimitiveDecode is just like the other `Decode*` functions, except it
 | 
			
		||||
// decodes a TOML value that has already been parsed. Valid primitive values
 | 
			
		||||
// can *only* be obtained from values filled by the decoder functions,
 | 
			
		||||
// including this method. (i.e., `v` may contain more `Primitive`
 | 
			
		||||
// values.)
 | 
			
		||||
//
 | 
			
		||||
// Meta data for primitive values is included in the meta data returned by
 | 
			
		||||
// the `Decode*` functions with one exception: keys returned by the Undecoded
 | 
			
		||||
// method will only reflect keys that were decoded. Namely, any keys hidden
 | 
			
		||||
// behind a Primitive will be considered undecoded. Executing this method will
 | 
			
		||||
// update the undecoded keys in the meta data. (See the example.)
 | 
			
		||||
func (md *MetaData) PrimitiveDecode(primValue Primitive, v interface{}) error {
 | 
			
		||||
	md.context = primValue.context
 | 
			
		||||
	defer func() { md.context = nil }()
 | 
			
		||||
	return md.unify(primValue.undecoded, rvalue(v))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Decode will decode the contents of `data` in TOML format into a pointer
 | 
			
		||||
// `v`.
 | 
			
		||||
//
 | 
			
		||||
// TOML hashes correspond to Go structs or maps. (Dealer's choice. They can be
 | 
			
		||||
// used interchangeably.)
 | 
			
		||||
//
 | 
			
		||||
// TOML arrays of tables correspond to either a slice of structs or a slice
 | 
			
		||||
// of maps.
 | 
			
		||||
//
 | 
			
		||||
// TOML datetimes correspond to Go `time.Time` values.
 | 
			
		||||
//
 | 
			
		||||
// All other TOML types (float, string, int, bool and array) correspond
 | 
			
		||||
// to the obvious Go types.
 | 
			
		||||
//
 | 
			
		||||
// An exception to the above rules is if a type implements the
 | 
			
		||||
// encoding.TextUnmarshaler interface. In this case, any primitive TOML value
 | 
			
		||||
// (floats, strings, integers, booleans and datetimes) will be converted to
 | 
			
		||||
// a byte string and given to the value's UnmarshalText method. See the
 | 
			
		||||
// Unmarshaler example for a demonstration with time duration strings.
 | 
			
		||||
//
 | 
			
		||||
// Key mapping
 | 
			
		||||
//
 | 
			
		||||
// TOML keys can map to either keys in a Go map or field names in a Go
 | 
			
		||||
// struct. The special `toml` struct tag may be used to map TOML keys to
 | 
			
		||||
// struct fields that don't match the key name exactly. (See the example.)
 | 
			
		||||
// A case insensitive match to struct names will be tried if an exact match
 | 
			
		||||
// can't be found.
 | 
			
		||||
//
 | 
			
		||||
// The mapping between TOML values and Go values is loose. That is, there
 | 
			
		||||
// may exist TOML values that cannot be placed into your representation, and
 | 
			
		||||
// there may be parts of your representation that do not correspond to
 | 
			
		||||
// TOML values. This loose mapping can be made stricter by using the IsDefined
 | 
			
		||||
// and/or Undecoded methods on the MetaData returned.
 | 
			
		||||
//
 | 
			
		||||
// This decoder will not handle cyclic types. If a cyclic type is passed,
 | 
			
		||||
// `Decode` will not terminate.
 | 
			
		||||
func Decode(data string, v interface{}) (MetaData, error) {
 | 
			
		||||
	rv := reflect.ValueOf(v)
 | 
			
		||||
	if rv.Kind() != reflect.Ptr {
 | 
			
		||||
		return MetaData{}, e("Decode of non-pointer %s", reflect.TypeOf(v))
 | 
			
		||||
	}
 | 
			
		||||
	if rv.IsNil() {
 | 
			
		||||
		return MetaData{}, e("Decode of nil %s", reflect.TypeOf(v))
 | 
			
		||||
	}
 | 
			
		||||
	p, err := parse(data)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return MetaData{}, err
 | 
			
		||||
	}
 | 
			
		||||
	md := MetaData{
 | 
			
		||||
		p.mapping, p.types, p.ordered,
 | 
			
		||||
		make(map[string]bool, len(p.ordered)), nil,
 | 
			
		||||
	}
 | 
			
		||||
	return md, md.unify(p.mapping, indirect(rv))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// DecodeFile is just like Decode, except it will automatically read the
 | 
			
		||||
// contents of the file at `fpath` and decode it for you.
 | 
			
		||||
func DecodeFile(fpath string, v interface{}) (MetaData, error) {
 | 
			
		||||
	bs, err := ioutil.ReadFile(fpath)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return MetaData{}, err
 | 
			
		||||
	}
 | 
			
		||||
	return Decode(string(bs), v)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// DecodeReader is just like Decode, except it will consume all bytes
 | 
			
		||||
// from the reader and decode it for you.
 | 
			
		||||
func DecodeReader(r io.Reader, v interface{}) (MetaData, error) {
 | 
			
		||||
	bs, err := ioutil.ReadAll(r)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return MetaData{}, err
 | 
			
		||||
	}
 | 
			
		||||
	return Decode(string(bs), v)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// unify performs a sort of type unification based on the structure of `rv`,
 | 
			
		||||
// which is the client representation.
 | 
			
		||||
//
 | 
			
		||||
// Any type mismatch produces an error. Finding a type that we don't know
 | 
			
		||||
// how to handle produces an unsupported type error.
 | 
			
		||||
func (md *MetaData) unify(data interface{}, rv reflect.Value) error {
 | 
			
		||||
 | 
			
		||||
	// Special case. Look for a `Primitive` value.
 | 
			
		||||
	if rv.Type() == reflect.TypeOf((*Primitive)(nil)).Elem() {
 | 
			
		||||
		// Save the undecoded data and the key context into the primitive
 | 
			
		||||
		// value.
 | 
			
		||||
		context := make(Key, len(md.context))
 | 
			
		||||
		copy(context, md.context)
 | 
			
		||||
		rv.Set(reflect.ValueOf(Primitive{
 | 
			
		||||
			undecoded: data,
 | 
			
		||||
			context:   context,
 | 
			
		||||
		}))
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Special case. Unmarshaler Interface support.
 | 
			
		||||
	if rv.CanAddr() {
 | 
			
		||||
		if v, ok := rv.Addr().Interface().(Unmarshaler); ok {
 | 
			
		||||
			return v.UnmarshalTOML(data)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Special case. Handle time.Time values specifically.
 | 
			
		||||
	// TODO: Remove this code when we decide to drop support for Go 1.1.
 | 
			
		||||
	// This isn't necessary in Go 1.2 because time.Time satisfies the encoding
 | 
			
		||||
	// interfaces.
 | 
			
		||||
	if rv.Type().AssignableTo(rvalue(time.Time{}).Type()) {
 | 
			
		||||
		return md.unifyDatetime(data, rv)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Special case. Look for a value satisfying the TextUnmarshaler interface.
 | 
			
		||||
	if v, ok := rv.Interface().(TextUnmarshaler); ok {
 | 
			
		||||
		return md.unifyText(data, v)
 | 
			
		||||
	}
 | 
			
		||||
	// BUG(burntsushi)
 | 
			
		||||
	// The behavior here is incorrect whenever a Go type satisfies the
 | 
			
		||||
	// encoding.TextUnmarshaler interface but also corresponds to a TOML
 | 
			
		||||
	// hash or array. In particular, the unmarshaler should only be applied
 | 
			
		||||
	// to primitive TOML values. But at this point, it will be applied to
 | 
			
		||||
	// all kinds of values and produce an incorrect error whenever those values
 | 
			
		||||
	// are hashes or arrays (including arrays of tables).
 | 
			
		||||
 | 
			
		||||
	k := rv.Kind()
 | 
			
		||||
 | 
			
		||||
	// laziness
 | 
			
		||||
	if k >= reflect.Int && k <= reflect.Uint64 {
 | 
			
		||||
		return md.unifyInt(data, rv)
 | 
			
		||||
	}
 | 
			
		||||
	switch k {
 | 
			
		||||
	case reflect.Ptr:
 | 
			
		||||
		elem := reflect.New(rv.Type().Elem())
 | 
			
		||||
		err := md.unify(data, reflect.Indirect(elem))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		rv.Set(elem)
 | 
			
		||||
		return nil
 | 
			
		||||
	case reflect.Struct:
 | 
			
		||||
		return md.unifyStruct(data, rv)
 | 
			
		||||
	case reflect.Map:
 | 
			
		||||
		return md.unifyMap(data, rv)
 | 
			
		||||
	case reflect.Array:
 | 
			
		||||
		return md.unifyArray(data, rv)
 | 
			
		||||
	case reflect.Slice:
 | 
			
		||||
		return md.unifySlice(data, rv)
 | 
			
		||||
	case reflect.String:
 | 
			
		||||
		return md.unifyString(data, rv)
 | 
			
		||||
	case reflect.Bool:
 | 
			
		||||
		return md.unifyBool(data, rv)
 | 
			
		||||
	case reflect.Interface:
 | 
			
		||||
		// we only support empty interfaces.
 | 
			
		||||
		if rv.NumMethod() > 0 {
 | 
			
		||||
			return e("unsupported type %s", rv.Type())
 | 
			
		||||
		}
 | 
			
		||||
		return md.unifyAnything(data, rv)
 | 
			
		||||
	case reflect.Float32:
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case reflect.Float64:
 | 
			
		||||
		return md.unifyFloat64(data, rv)
 | 
			
		||||
	}
 | 
			
		||||
	return e("unsupported type %s", rv.Kind())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyStruct(mapping interface{}, rv reflect.Value) error {
 | 
			
		||||
	tmap, ok := mapping.(map[string]interface{})
 | 
			
		||||
	if !ok {
 | 
			
		||||
		if mapping == nil {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		return e("type mismatch for %s: expected table but found %T",
 | 
			
		||||
			rv.Type().String(), mapping)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for key, datum := range tmap {
 | 
			
		||||
		var f *field
 | 
			
		||||
		fields := cachedTypeFields(rv.Type())
 | 
			
		||||
		for i := range fields {
 | 
			
		||||
			ff := &fields[i]
 | 
			
		||||
			if ff.name == key {
 | 
			
		||||
				f = ff
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
			if f == nil && strings.EqualFold(ff.name, key) {
 | 
			
		||||
				f = ff
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if f != nil {
 | 
			
		||||
			subv := rv
 | 
			
		||||
			for _, i := range f.index {
 | 
			
		||||
				subv = indirect(subv.Field(i))
 | 
			
		||||
			}
 | 
			
		||||
			if isUnifiable(subv) {
 | 
			
		||||
				md.decoded[md.context.add(key).String()] = true
 | 
			
		||||
				md.context = append(md.context, key)
 | 
			
		||||
				if err := md.unify(datum, subv); err != nil {
 | 
			
		||||
					return err
 | 
			
		||||
				}
 | 
			
		||||
				md.context = md.context[0 : len(md.context)-1]
 | 
			
		||||
			} else if f.name != "" {
 | 
			
		||||
				// Bad user! No soup for you!
 | 
			
		||||
				return e("cannot write unexported field %s.%s",
 | 
			
		||||
					rv.Type().String(), f.name)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyMap(mapping interface{}, rv reflect.Value) error {
 | 
			
		||||
	tmap, ok := mapping.(map[string]interface{})
 | 
			
		||||
	if !ok {
 | 
			
		||||
		if tmap == nil {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		return badtype("map", mapping)
 | 
			
		||||
	}
 | 
			
		||||
	if rv.IsNil() {
 | 
			
		||||
		rv.Set(reflect.MakeMap(rv.Type()))
 | 
			
		||||
	}
 | 
			
		||||
	for k, v := range tmap {
 | 
			
		||||
		md.decoded[md.context.add(k).String()] = true
 | 
			
		||||
		md.context = append(md.context, k)
 | 
			
		||||
 | 
			
		||||
		rvkey := indirect(reflect.New(rv.Type().Key()))
 | 
			
		||||
		rvval := reflect.Indirect(reflect.New(rv.Type().Elem()))
 | 
			
		||||
		if err := md.unify(v, rvval); err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		md.context = md.context[0 : len(md.context)-1]
 | 
			
		||||
 | 
			
		||||
		rvkey.SetString(k)
 | 
			
		||||
		rv.SetMapIndex(rvkey, rvval)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyArray(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	datav := reflect.ValueOf(data)
 | 
			
		||||
	if datav.Kind() != reflect.Slice {
 | 
			
		||||
		if !datav.IsValid() {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		return badtype("slice", data)
 | 
			
		||||
	}
 | 
			
		||||
	sliceLen := datav.Len()
 | 
			
		||||
	if sliceLen != rv.Len() {
 | 
			
		||||
		return e("expected array length %d; got TOML array of length %d",
 | 
			
		||||
			rv.Len(), sliceLen)
 | 
			
		||||
	}
 | 
			
		||||
	return md.unifySliceArray(datav, rv)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifySlice(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	datav := reflect.ValueOf(data)
 | 
			
		||||
	if datav.Kind() != reflect.Slice {
 | 
			
		||||
		if !datav.IsValid() {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		return badtype("slice", data)
 | 
			
		||||
	}
 | 
			
		||||
	n := datav.Len()
 | 
			
		||||
	if rv.IsNil() || rv.Cap() < n {
 | 
			
		||||
		rv.Set(reflect.MakeSlice(rv.Type(), n, n))
 | 
			
		||||
	}
 | 
			
		||||
	rv.SetLen(n)
 | 
			
		||||
	return md.unifySliceArray(datav, rv)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifySliceArray(data, rv reflect.Value) error {
 | 
			
		||||
	sliceLen := data.Len()
 | 
			
		||||
	for i := 0; i < sliceLen; i++ {
 | 
			
		||||
		v := data.Index(i).Interface()
 | 
			
		||||
		sliceval := indirect(rv.Index(i))
 | 
			
		||||
		if err := md.unify(v, sliceval); err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyDatetime(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	if _, ok := data.(time.Time); ok {
 | 
			
		||||
		rv.Set(reflect.ValueOf(data))
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return badtype("time.Time", data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyString(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	if s, ok := data.(string); ok {
 | 
			
		||||
		rv.SetString(s)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return badtype("string", data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyFloat64(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	if num, ok := data.(float64); ok {
 | 
			
		||||
		switch rv.Kind() {
 | 
			
		||||
		case reflect.Float32:
 | 
			
		||||
			fallthrough
 | 
			
		||||
		case reflect.Float64:
 | 
			
		||||
			rv.SetFloat(num)
 | 
			
		||||
		default:
 | 
			
		||||
			panic("bug")
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return badtype("float", data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyInt(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	if num, ok := data.(int64); ok {
 | 
			
		||||
		if rv.Kind() >= reflect.Int && rv.Kind() <= reflect.Int64 {
 | 
			
		||||
			switch rv.Kind() {
 | 
			
		||||
			case reflect.Int, reflect.Int64:
 | 
			
		||||
				// No bounds checking necessary.
 | 
			
		||||
			case reflect.Int8:
 | 
			
		||||
				if num < math.MinInt8 || num > math.MaxInt8 {
 | 
			
		||||
					return e("value %d is out of range for int8", num)
 | 
			
		||||
				}
 | 
			
		||||
			case reflect.Int16:
 | 
			
		||||
				if num < math.MinInt16 || num > math.MaxInt16 {
 | 
			
		||||
					return e("value %d is out of range for int16", num)
 | 
			
		||||
				}
 | 
			
		||||
			case reflect.Int32:
 | 
			
		||||
				if num < math.MinInt32 || num > math.MaxInt32 {
 | 
			
		||||
					return e("value %d is out of range for int32", num)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			rv.SetInt(num)
 | 
			
		||||
		} else if rv.Kind() >= reflect.Uint && rv.Kind() <= reflect.Uint64 {
 | 
			
		||||
			unum := uint64(num)
 | 
			
		||||
			switch rv.Kind() {
 | 
			
		||||
			case reflect.Uint, reflect.Uint64:
 | 
			
		||||
				// No bounds checking necessary.
 | 
			
		||||
			case reflect.Uint8:
 | 
			
		||||
				if num < 0 || unum > math.MaxUint8 {
 | 
			
		||||
					return e("value %d is out of range for uint8", num)
 | 
			
		||||
				}
 | 
			
		||||
			case reflect.Uint16:
 | 
			
		||||
				if num < 0 || unum > math.MaxUint16 {
 | 
			
		||||
					return e("value %d is out of range for uint16", num)
 | 
			
		||||
				}
 | 
			
		||||
			case reflect.Uint32:
 | 
			
		||||
				if num < 0 || unum > math.MaxUint32 {
 | 
			
		||||
					return e("value %d is out of range for uint32", num)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			rv.SetUint(unum)
 | 
			
		||||
		} else {
 | 
			
		||||
			panic("unreachable")
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return badtype("integer", data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyBool(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	if b, ok := data.(bool); ok {
 | 
			
		||||
		rv.SetBool(b)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return badtype("boolean", data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyAnything(data interface{}, rv reflect.Value) error {
 | 
			
		||||
	rv.Set(reflect.ValueOf(data))
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (md *MetaData) unifyText(data interface{}, v TextUnmarshaler) error {
 | 
			
		||||
	var s string
 | 
			
		||||
	switch sdata := data.(type) {
 | 
			
		||||
	case TextMarshaler:
 | 
			
		||||
		text, err := sdata.MarshalText()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		s = string(text)
 | 
			
		||||
	case fmt.Stringer:
 | 
			
		||||
		s = sdata.String()
 | 
			
		||||
	case string:
 | 
			
		||||
		s = sdata
 | 
			
		||||
	case bool:
 | 
			
		||||
		s = fmt.Sprintf("%v", sdata)
 | 
			
		||||
	case int64:
 | 
			
		||||
		s = fmt.Sprintf("%d", sdata)
 | 
			
		||||
	case float64:
 | 
			
		||||
		s = fmt.Sprintf("%f", sdata)
 | 
			
		||||
	default:
 | 
			
		||||
		return badtype("primitive (string-like)", data)
 | 
			
		||||
	}
 | 
			
		||||
	if err := v.UnmarshalText([]byte(s)); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// rvalue returns a reflect.Value of `v`. All pointers are resolved.
 | 
			
		||||
func rvalue(v interface{}) reflect.Value {
 | 
			
		||||
	return indirect(reflect.ValueOf(v))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// indirect returns the value pointed to by a pointer.
 | 
			
		||||
// Pointers are followed until the value is not a pointer.
 | 
			
		||||
// New values are allocated for each nil pointer.
 | 
			
		||||
//
 | 
			
		||||
// An exception to this rule is if the value satisfies an interface of
 | 
			
		||||
// interest to us (like encoding.TextUnmarshaler).
 | 
			
		||||
func indirect(v reflect.Value) reflect.Value {
 | 
			
		||||
	if v.Kind() != reflect.Ptr {
 | 
			
		||||
		if v.CanSet() {
 | 
			
		||||
			pv := v.Addr()
 | 
			
		||||
			if _, ok := pv.Interface().(TextUnmarshaler); ok {
 | 
			
		||||
				return pv
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return v
 | 
			
		||||
	}
 | 
			
		||||
	if v.IsNil() {
 | 
			
		||||
		v.Set(reflect.New(v.Type().Elem()))
 | 
			
		||||
	}
 | 
			
		||||
	return indirect(reflect.Indirect(v))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isUnifiable(rv reflect.Value) bool {
 | 
			
		||||
	if rv.CanSet() {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	if _, ok := rv.Interface().(TextUnmarshaler); ok {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func badtype(expected string, data interface{}) error {
 | 
			
		||||
	return e("cannot load TOML value of type %T into a Go %s", data, expected)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										121
									
								
								vendor/github.com/BurntSushi/toml/decode_meta.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										121
									
								
								vendor/github.com/BurntSushi/toml/decode_meta.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,121 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
import "strings"
 | 
			
		||||
 | 
			
		||||
// MetaData allows access to meta information about TOML data that may not
 | 
			
		||||
// be inferrable via reflection. In particular, whether a key has been defined
 | 
			
		||||
// and the TOML type of a key.
 | 
			
		||||
type MetaData struct {
 | 
			
		||||
	mapping map[string]interface{}
 | 
			
		||||
	types   map[string]tomlType
 | 
			
		||||
	keys    []Key
 | 
			
		||||
	decoded map[string]bool
 | 
			
		||||
	context Key // Used only during decoding.
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// IsDefined returns true if the key given exists in the TOML data. The key
 | 
			
		||||
// should be specified hierarchially. e.g.,
 | 
			
		||||
//
 | 
			
		||||
//	// access the TOML key 'a.b.c'
 | 
			
		||||
//	IsDefined("a", "b", "c")
 | 
			
		||||
//
 | 
			
		||||
// IsDefined will return false if an empty key given. Keys are case sensitive.
 | 
			
		||||
func (md *MetaData) IsDefined(key ...string) bool {
 | 
			
		||||
	if len(key) == 0 {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var hash map[string]interface{}
 | 
			
		||||
	var ok bool
 | 
			
		||||
	var hashOrVal interface{} = md.mapping
 | 
			
		||||
	for _, k := range key {
 | 
			
		||||
		if hash, ok = hashOrVal.(map[string]interface{}); !ok {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
		if hashOrVal, ok = hash[k]; !ok {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Type returns a string representation of the type of the key specified.
 | 
			
		||||
//
 | 
			
		||||
// Type will return the empty string if given an empty key or a key that
 | 
			
		||||
// does not exist. Keys are case sensitive.
 | 
			
		||||
func (md *MetaData) Type(key ...string) string {
 | 
			
		||||
	fullkey := strings.Join(key, ".")
 | 
			
		||||
	if typ, ok := md.types[fullkey]; ok {
 | 
			
		||||
		return typ.typeString()
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Key is the type of any TOML key, including key groups. Use (MetaData).Keys
 | 
			
		||||
// to get values of this type.
 | 
			
		||||
type Key []string
 | 
			
		||||
 | 
			
		||||
func (k Key) String() string {
 | 
			
		||||
	return strings.Join(k, ".")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k Key) maybeQuotedAll() string {
 | 
			
		||||
	var ss []string
 | 
			
		||||
	for i := range k {
 | 
			
		||||
		ss = append(ss, k.maybeQuoted(i))
 | 
			
		||||
	}
 | 
			
		||||
	return strings.Join(ss, ".")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k Key) maybeQuoted(i int) string {
 | 
			
		||||
	quote := false
 | 
			
		||||
	for _, c := range k[i] {
 | 
			
		||||
		if !isBareKeyChar(c) {
 | 
			
		||||
			quote = true
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if quote {
 | 
			
		||||
		return "\"" + strings.Replace(k[i], "\"", "\\\"", -1) + "\""
 | 
			
		||||
	}
 | 
			
		||||
	return k[i]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k Key) add(piece string) Key {
 | 
			
		||||
	newKey := make(Key, len(k)+1)
 | 
			
		||||
	copy(newKey, k)
 | 
			
		||||
	newKey[len(k)] = piece
 | 
			
		||||
	return newKey
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Keys returns a slice of every key in the TOML data, including key groups.
 | 
			
		||||
// Each key is itself a slice, where the first element is the top of the
 | 
			
		||||
// hierarchy and the last is the most specific.
 | 
			
		||||
//
 | 
			
		||||
// The list will have the same order as the keys appeared in the TOML data.
 | 
			
		||||
//
 | 
			
		||||
// All keys returned are non-empty.
 | 
			
		||||
func (md *MetaData) Keys() []Key {
 | 
			
		||||
	return md.keys
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Undecoded returns all keys that have not been decoded in the order in which
 | 
			
		||||
// they appear in the original TOML document.
 | 
			
		||||
//
 | 
			
		||||
// This includes keys that haven't been decoded because of a Primitive value.
 | 
			
		||||
// Once the Primitive value is decoded, the keys will be considered decoded.
 | 
			
		||||
//
 | 
			
		||||
// Also note that decoding into an empty interface will result in no decoding,
 | 
			
		||||
// and so no keys will be considered decoded.
 | 
			
		||||
//
 | 
			
		||||
// In this sense, the Undecoded keys correspond to keys in the TOML document
 | 
			
		||||
// that do not have a concrete type in your representation.
 | 
			
		||||
func (md *MetaData) Undecoded() []Key {
 | 
			
		||||
	undecoded := make([]Key, 0, len(md.keys))
 | 
			
		||||
	for _, key := range md.keys {
 | 
			
		||||
		if !md.decoded[key.String()] {
 | 
			
		||||
			undecoded = append(undecoded, key)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return undecoded
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										27
									
								
								vendor/github.com/BurntSushi/toml/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								vendor/github.com/BurntSushi/toml/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,27 @@
 | 
			
		||||
/*
 | 
			
		||||
Package toml provides facilities for decoding and encoding TOML configuration
 | 
			
		||||
files via reflection. There is also support for delaying decoding with
 | 
			
		||||
the Primitive type, and querying the set of keys in a TOML document with the
 | 
			
		||||
MetaData type.
 | 
			
		||||
 | 
			
		||||
The specification implemented: https://github.com/mojombo/toml
 | 
			
		||||
 | 
			
		||||
The sub-command github.com/BurntSushi/toml/cmd/tomlv can be used to verify
 | 
			
		||||
whether a file is a valid TOML document. It can also be used to print the
 | 
			
		||||
type of each key in a TOML document.
 | 
			
		||||
 | 
			
		||||
Testing
 | 
			
		||||
 | 
			
		||||
There are two important types of tests used for this package. The first is
 | 
			
		||||
contained inside '*_test.go' files and uses the standard Go unit testing
 | 
			
		||||
framework. These tests are primarily devoted to holistically testing the
 | 
			
		||||
decoder and encoder.
 | 
			
		||||
 | 
			
		||||
The second type of testing is used to verify the implementation's adherence
 | 
			
		||||
to the TOML specification. These tests have been factored into their own
 | 
			
		||||
project: https://github.com/BurntSushi/toml-test
 | 
			
		||||
 | 
			
		||||
The reason the tests are in a separate project is so that they can be used by
 | 
			
		||||
any implementation of TOML. Namely, it is language agnostic.
 | 
			
		||||
*/
 | 
			
		||||
package toml
 | 
			
		||||
							
								
								
									
										568
									
								
								vendor/github.com/BurntSushi/toml/encode.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										568
									
								
								vendor/github.com/BurntSushi/toml/encode.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,568 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type tomlEncodeError struct{ error }
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	errArrayMixedElementTypes = errors.New(
 | 
			
		||||
		"toml: cannot encode array with mixed element types")
 | 
			
		||||
	errArrayNilElement = errors.New(
 | 
			
		||||
		"toml: cannot encode array with nil element")
 | 
			
		||||
	errNonString = errors.New(
 | 
			
		||||
		"toml: cannot encode a map with non-string key type")
 | 
			
		||||
	errAnonNonStruct = errors.New(
 | 
			
		||||
		"toml: cannot encode an anonymous field that is not a struct")
 | 
			
		||||
	errArrayNoTable = errors.New(
 | 
			
		||||
		"toml: TOML array element cannot contain a table")
 | 
			
		||||
	errNoKey = errors.New(
 | 
			
		||||
		"toml: top-level values must be Go maps or structs")
 | 
			
		||||
	errAnything = errors.New("") // used in testing
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var quotedReplacer = strings.NewReplacer(
 | 
			
		||||
	"\t", "\\t",
 | 
			
		||||
	"\n", "\\n",
 | 
			
		||||
	"\r", "\\r",
 | 
			
		||||
	"\"", "\\\"",
 | 
			
		||||
	"\\", "\\\\",
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Encoder controls the encoding of Go values to a TOML document to some
 | 
			
		||||
// io.Writer.
 | 
			
		||||
//
 | 
			
		||||
// The indentation level can be controlled with the Indent field.
 | 
			
		||||
type Encoder struct {
 | 
			
		||||
	// A single indentation level. By default it is two spaces.
 | 
			
		||||
	Indent string
 | 
			
		||||
 | 
			
		||||
	// hasWritten is whether we have written any output to w yet.
 | 
			
		||||
	hasWritten bool
 | 
			
		||||
	w          *bufio.Writer
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewEncoder returns a TOML encoder that encodes Go values to the io.Writer
 | 
			
		||||
// given. By default, a single indentation level is 2 spaces.
 | 
			
		||||
func NewEncoder(w io.Writer) *Encoder {
 | 
			
		||||
	return &Encoder{
 | 
			
		||||
		w:      bufio.NewWriter(w),
 | 
			
		||||
		Indent: "  ",
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Encode writes a TOML representation of the Go value to the underlying
 | 
			
		||||
// io.Writer. If the value given cannot be encoded to a valid TOML document,
 | 
			
		||||
// then an error is returned.
 | 
			
		||||
//
 | 
			
		||||
// The mapping between Go values and TOML values should be precisely the same
 | 
			
		||||
// as for the Decode* functions. Similarly, the TextMarshaler interface is
 | 
			
		||||
// supported by encoding the resulting bytes as strings. (If you want to write
 | 
			
		||||
// arbitrary binary data then you will need to use something like base64 since
 | 
			
		||||
// TOML does not have any binary types.)
 | 
			
		||||
//
 | 
			
		||||
// When encoding TOML hashes (i.e., Go maps or structs), keys without any
 | 
			
		||||
// sub-hashes are encoded first.
 | 
			
		||||
//
 | 
			
		||||
// If a Go map is encoded, then its keys are sorted alphabetically for
 | 
			
		||||
// deterministic output. More control over this behavior may be provided if
 | 
			
		||||
// there is demand for it.
 | 
			
		||||
//
 | 
			
		||||
// Encoding Go values without a corresponding TOML representation---like map
 | 
			
		||||
// types with non-string keys---will cause an error to be returned. Similarly
 | 
			
		||||
// for mixed arrays/slices, arrays/slices with nil elements, embedded
 | 
			
		||||
// non-struct types and nested slices containing maps or structs.
 | 
			
		||||
// (e.g., [][]map[string]string is not allowed but []map[string]string is OK
 | 
			
		||||
// and so is []map[string][]string.)
 | 
			
		||||
func (enc *Encoder) Encode(v interface{}) error {
 | 
			
		||||
	rv := eindirect(reflect.ValueOf(v))
 | 
			
		||||
	if err := enc.safeEncode(Key([]string{}), rv); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return enc.w.Flush()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) safeEncode(key Key, rv reflect.Value) (err error) {
 | 
			
		||||
	defer func() {
 | 
			
		||||
		if r := recover(); r != nil {
 | 
			
		||||
			if terr, ok := r.(tomlEncodeError); ok {
 | 
			
		||||
				err = terr.error
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			panic(r)
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
	enc.encode(key, rv)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) encode(key Key, rv reflect.Value) {
 | 
			
		||||
	// Special case. Time needs to be in ISO8601 format.
 | 
			
		||||
	// Special case. If we can marshal the type to text, then we used that.
 | 
			
		||||
	// Basically, this prevents the encoder for handling these types as
 | 
			
		||||
	// generic structs (or whatever the underlying type of a TextMarshaler is).
 | 
			
		||||
	switch rv.Interface().(type) {
 | 
			
		||||
	case time.Time, TextMarshaler:
 | 
			
		||||
		enc.keyEqElement(key, rv)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	k := rv.Kind()
 | 
			
		||||
	switch k {
 | 
			
		||||
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32,
 | 
			
		||||
		reflect.Int64,
 | 
			
		||||
		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32,
 | 
			
		||||
		reflect.Uint64,
 | 
			
		||||
		reflect.Float32, reflect.Float64, reflect.String, reflect.Bool:
 | 
			
		||||
		enc.keyEqElement(key, rv)
 | 
			
		||||
	case reflect.Array, reflect.Slice:
 | 
			
		||||
		if typeEqual(tomlArrayHash, tomlTypeOfGo(rv)) {
 | 
			
		||||
			enc.eArrayOfTables(key, rv)
 | 
			
		||||
		} else {
 | 
			
		||||
			enc.keyEqElement(key, rv)
 | 
			
		||||
		}
 | 
			
		||||
	case reflect.Interface:
 | 
			
		||||
		if rv.IsNil() {
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		enc.encode(key, rv.Elem())
 | 
			
		||||
	case reflect.Map:
 | 
			
		||||
		if rv.IsNil() {
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		enc.eTable(key, rv)
 | 
			
		||||
	case reflect.Ptr:
 | 
			
		||||
		if rv.IsNil() {
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		enc.encode(key, rv.Elem())
 | 
			
		||||
	case reflect.Struct:
 | 
			
		||||
		enc.eTable(key, rv)
 | 
			
		||||
	default:
 | 
			
		||||
		panic(e("unsupported type for key '%s': %s", key, k))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// eElement encodes any value that can be an array element (primitives and
 | 
			
		||||
// arrays).
 | 
			
		||||
func (enc *Encoder) eElement(rv reflect.Value) {
 | 
			
		||||
	switch v := rv.Interface().(type) {
 | 
			
		||||
	case time.Time:
 | 
			
		||||
		// Special case time.Time as a primitive. Has to come before
 | 
			
		||||
		// TextMarshaler below because time.Time implements
 | 
			
		||||
		// encoding.TextMarshaler, but we need to always use UTC.
 | 
			
		||||
		enc.wf(v.UTC().Format("2006-01-02T15:04:05Z"))
 | 
			
		||||
		return
 | 
			
		||||
	case TextMarshaler:
 | 
			
		||||
		// Special case. Use text marshaler if it's available for this value.
 | 
			
		||||
		if s, err := v.MarshalText(); err != nil {
 | 
			
		||||
			encPanic(err)
 | 
			
		||||
		} else {
 | 
			
		||||
			enc.writeQuoted(string(s))
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	switch rv.Kind() {
 | 
			
		||||
	case reflect.Bool:
 | 
			
		||||
		enc.wf(strconv.FormatBool(rv.Bool()))
 | 
			
		||||
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32,
 | 
			
		||||
		reflect.Int64:
 | 
			
		||||
		enc.wf(strconv.FormatInt(rv.Int(), 10))
 | 
			
		||||
	case reflect.Uint, reflect.Uint8, reflect.Uint16,
 | 
			
		||||
		reflect.Uint32, reflect.Uint64:
 | 
			
		||||
		enc.wf(strconv.FormatUint(rv.Uint(), 10))
 | 
			
		||||
	case reflect.Float32:
 | 
			
		||||
		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 32)))
 | 
			
		||||
	case reflect.Float64:
 | 
			
		||||
		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 64)))
 | 
			
		||||
	case reflect.Array, reflect.Slice:
 | 
			
		||||
		enc.eArrayOrSliceElement(rv)
 | 
			
		||||
	case reflect.Interface:
 | 
			
		||||
		enc.eElement(rv.Elem())
 | 
			
		||||
	case reflect.String:
 | 
			
		||||
		enc.writeQuoted(rv.String())
 | 
			
		||||
	default:
 | 
			
		||||
		panic(e("unexpected primitive type: %s", rv.Kind()))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// By the TOML spec, all floats must have a decimal with at least one
 | 
			
		||||
// number on either side.
 | 
			
		||||
func floatAddDecimal(fstr string) string {
 | 
			
		||||
	if !strings.Contains(fstr, ".") {
 | 
			
		||||
		return fstr + ".0"
 | 
			
		||||
	}
 | 
			
		||||
	return fstr
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) writeQuoted(s string) {
 | 
			
		||||
	enc.wf("\"%s\"", quotedReplacer.Replace(s))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eArrayOrSliceElement(rv reflect.Value) {
 | 
			
		||||
	length := rv.Len()
 | 
			
		||||
	enc.wf("[")
 | 
			
		||||
	for i := 0; i < length; i++ {
 | 
			
		||||
		elem := rv.Index(i)
 | 
			
		||||
		enc.eElement(elem)
 | 
			
		||||
		if i != length-1 {
 | 
			
		||||
			enc.wf(", ")
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	enc.wf("]")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eArrayOfTables(key Key, rv reflect.Value) {
 | 
			
		||||
	if len(key) == 0 {
 | 
			
		||||
		encPanic(errNoKey)
 | 
			
		||||
	}
 | 
			
		||||
	for i := 0; i < rv.Len(); i++ {
 | 
			
		||||
		trv := rv.Index(i)
 | 
			
		||||
		if isNil(trv) {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		panicIfInvalidKey(key)
 | 
			
		||||
		enc.newline()
 | 
			
		||||
		enc.wf("%s[[%s]]", enc.indentStr(key), key.maybeQuotedAll())
 | 
			
		||||
		enc.newline()
 | 
			
		||||
		enc.eMapOrStruct(key, trv)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eTable(key Key, rv reflect.Value) {
 | 
			
		||||
	panicIfInvalidKey(key)
 | 
			
		||||
	if len(key) == 1 {
 | 
			
		||||
		// Output an extra new line between top-level tables.
 | 
			
		||||
		// (The newline isn't written if nothing else has been written though.)
 | 
			
		||||
		enc.newline()
 | 
			
		||||
	}
 | 
			
		||||
	if len(key) > 0 {
 | 
			
		||||
		enc.wf("%s[%s]", enc.indentStr(key), key.maybeQuotedAll())
 | 
			
		||||
		enc.newline()
 | 
			
		||||
	}
 | 
			
		||||
	enc.eMapOrStruct(key, rv)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eMapOrStruct(key Key, rv reflect.Value) {
 | 
			
		||||
	switch rv := eindirect(rv); rv.Kind() {
 | 
			
		||||
	case reflect.Map:
 | 
			
		||||
		enc.eMap(key, rv)
 | 
			
		||||
	case reflect.Struct:
 | 
			
		||||
		enc.eStruct(key, rv)
 | 
			
		||||
	default:
 | 
			
		||||
		panic("eTable: unhandled reflect.Value Kind: " + rv.Kind().String())
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eMap(key Key, rv reflect.Value) {
 | 
			
		||||
	rt := rv.Type()
 | 
			
		||||
	if rt.Key().Kind() != reflect.String {
 | 
			
		||||
		encPanic(errNonString)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Sort keys so that we have deterministic output. And write keys directly
 | 
			
		||||
	// underneath this key first, before writing sub-structs or sub-maps.
 | 
			
		||||
	var mapKeysDirect, mapKeysSub []string
 | 
			
		||||
	for _, mapKey := range rv.MapKeys() {
 | 
			
		||||
		k := mapKey.String()
 | 
			
		||||
		if typeIsHash(tomlTypeOfGo(rv.MapIndex(mapKey))) {
 | 
			
		||||
			mapKeysSub = append(mapKeysSub, k)
 | 
			
		||||
		} else {
 | 
			
		||||
			mapKeysDirect = append(mapKeysDirect, k)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var writeMapKeys = func(mapKeys []string) {
 | 
			
		||||
		sort.Strings(mapKeys)
 | 
			
		||||
		for _, mapKey := range mapKeys {
 | 
			
		||||
			mrv := rv.MapIndex(reflect.ValueOf(mapKey))
 | 
			
		||||
			if isNil(mrv) {
 | 
			
		||||
				// Don't write anything for nil fields.
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			enc.encode(key.add(mapKey), mrv)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	writeMapKeys(mapKeysDirect)
 | 
			
		||||
	writeMapKeys(mapKeysSub)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) eStruct(key Key, rv reflect.Value) {
 | 
			
		||||
	// Write keys for fields directly under this key first, because if we write
 | 
			
		||||
	// a field that creates a new table, then all keys under it will be in that
 | 
			
		||||
	// table (not the one we're writing here).
 | 
			
		||||
	rt := rv.Type()
 | 
			
		||||
	var fieldsDirect, fieldsSub [][]int
 | 
			
		||||
	var addFields func(rt reflect.Type, rv reflect.Value, start []int)
 | 
			
		||||
	addFields = func(rt reflect.Type, rv reflect.Value, start []int) {
 | 
			
		||||
		for i := 0; i < rt.NumField(); i++ {
 | 
			
		||||
			f := rt.Field(i)
 | 
			
		||||
			// skip unexported fields
 | 
			
		||||
			if f.PkgPath != "" && !f.Anonymous {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			frv := rv.Field(i)
 | 
			
		||||
			if f.Anonymous {
 | 
			
		||||
				t := f.Type
 | 
			
		||||
				switch t.Kind() {
 | 
			
		||||
				case reflect.Struct:
 | 
			
		||||
					// Treat anonymous struct fields with
 | 
			
		||||
					// tag names as though they are not
 | 
			
		||||
					// anonymous, like encoding/json does.
 | 
			
		||||
					if getOptions(f.Tag).name == "" {
 | 
			
		||||
						addFields(t, frv, f.Index)
 | 
			
		||||
						continue
 | 
			
		||||
					}
 | 
			
		||||
				case reflect.Ptr:
 | 
			
		||||
					if t.Elem().Kind() == reflect.Struct &&
 | 
			
		||||
						getOptions(f.Tag).name == "" {
 | 
			
		||||
						if !frv.IsNil() {
 | 
			
		||||
							addFields(t.Elem(), frv.Elem(), f.Index)
 | 
			
		||||
						}
 | 
			
		||||
						continue
 | 
			
		||||
					}
 | 
			
		||||
					// Fall through to the normal field encoding logic below
 | 
			
		||||
					// for non-struct anonymous fields.
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if typeIsHash(tomlTypeOfGo(frv)) {
 | 
			
		||||
				fieldsSub = append(fieldsSub, append(start, f.Index...))
 | 
			
		||||
			} else {
 | 
			
		||||
				fieldsDirect = append(fieldsDirect, append(start, f.Index...))
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	addFields(rt, rv, nil)
 | 
			
		||||
 | 
			
		||||
	var writeFields = func(fields [][]int) {
 | 
			
		||||
		for _, fieldIndex := range fields {
 | 
			
		||||
			sft := rt.FieldByIndex(fieldIndex)
 | 
			
		||||
			sf := rv.FieldByIndex(fieldIndex)
 | 
			
		||||
			if isNil(sf) {
 | 
			
		||||
				// Don't write anything for nil fields.
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			opts := getOptions(sft.Tag)
 | 
			
		||||
			if opts.skip {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			keyName := sft.Name
 | 
			
		||||
			if opts.name != "" {
 | 
			
		||||
				keyName = opts.name
 | 
			
		||||
			}
 | 
			
		||||
			if opts.omitempty && isEmpty(sf) {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			if opts.omitzero && isZero(sf) {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			enc.encode(key.add(keyName), sf)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	writeFields(fieldsDirect)
 | 
			
		||||
	writeFields(fieldsSub)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// tomlTypeName returns the TOML type name of the Go value's type. It is
 | 
			
		||||
// used to determine whether the types of array elements are mixed (which is
 | 
			
		||||
// forbidden). If the Go value is nil, then it is illegal for it to be an array
 | 
			
		||||
// element, and valueIsNil is returned as true.
 | 
			
		||||
 | 
			
		||||
// Returns the TOML type of a Go value. The type may be `nil`, which means
 | 
			
		||||
// no concrete TOML type could be found.
 | 
			
		||||
func tomlTypeOfGo(rv reflect.Value) tomlType {
 | 
			
		||||
	if isNil(rv) || !rv.IsValid() {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	switch rv.Kind() {
 | 
			
		||||
	case reflect.Bool:
 | 
			
		||||
		return tomlBool
 | 
			
		||||
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32,
 | 
			
		||||
		reflect.Int64,
 | 
			
		||||
		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32,
 | 
			
		||||
		reflect.Uint64:
 | 
			
		||||
		return tomlInteger
 | 
			
		||||
	case reflect.Float32, reflect.Float64:
 | 
			
		||||
		return tomlFloat
 | 
			
		||||
	case reflect.Array, reflect.Slice:
 | 
			
		||||
		if typeEqual(tomlHash, tomlArrayType(rv)) {
 | 
			
		||||
			return tomlArrayHash
 | 
			
		||||
		}
 | 
			
		||||
		return tomlArray
 | 
			
		||||
	case reflect.Ptr, reflect.Interface:
 | 
			
		||||
		return tomlTypeOfGo(rv.Elem())
 | 
			
		||||
	case reflect.String:
 | 
			
		||||
		return tomlString
 | 
			
		||||
	case reflect.Map:
 | 
			
		||||
		return tomlHash
 | 
			
		||||
	case reflect.Struct:
 | 
			
		||||
		switch rv.Interface().(type) {
 | 
			
		||||
		case time.Time:
 | 
			
		||||
			return tomlDatetime
 | 
			
		||||
		case TextMarshaler:
 | 
			
		||||
			return tomlString
 | 
			
		||||
		default:
 | 
			
		||||
			return tomlHash
 | 
			
		||||
		}
 | 
			
		||||
	default:
 | 
			
		||||
		panic("unexpected reflect.Kind: " + rv.Kind().String())
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// tomlArrayType returns the element type of a TOML array. The type returned
 | 
			
		||||
// may be nil if it cannot be determined (e.g., a nil slice or a zero length
 | 
			
		||||
// slize). This function may also panic if it finds a type that cannot be
 | 
			
		||||
// expressed in TOML (such as nil elements, heterogeneous arrays or directly
 | 
			
		||||
// nested arrays of tables).
 | 
			
		||||
func tomlArrayType(rv reflect.Value) tomlType {
 | 
			
		||||
	if isNil(rv) || !rv.IsValid() || rv.Len() == 0 {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	firstType := tomlTypeOfGo(rv.Index(0))
 | 
			
		||||
	if firstType == nil {
 | 
			
		||||
		encPanic(errArrayNilElement)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rvlen := rv.Len()
 | 
			
		||||
	for i := 1; i < rvlen; i++ {
 | 
			
		||||
		elem := rv.Index(i)
 | 
			
		||||
		switch elemType := tomlTypeOfGo(elem); {
 | 
			
		||||
		case elemType == nil:
 | 
			
		||||
			encPanic(errArrayNilElement)
 | 
			
		||||
		case !typeEqual(firstType, elemType):
 | 
			
		||||
			encPanic(errArrayMixedElementTypes)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	// If we have a nested array, then we must make sure that the nested
 | 
			
		||||
	// array contains ONLY primitives.
 | 
			
		||||
	// This checks arbitrarily nested arrays.
 | 
			
		||||
	if typeEqual(firstType, tomlArray) || typeEqual(firstType, tomlArrayHash) {
 | 
			
		||||
		nest := tomlArrayType(eindirect(rv.Index(0)))
 | 
			
		||||
		if typeEqual(nest, tomlHash) || typeEqual(nest, tomlArrayHash) {
 | 
			
		||||
			encPanic(errArrayNoTable)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return firstType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type tagOptions struct {
 | 
			
		||||
	skip      bool // "-"
 | 
			
		||||
	name      string
 | 
			
		||||
	omitempty bool
 | 
			
		||||
	omitzero  bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getOptions(tag reflect.StructTag) tagOptions {
 | 
			
		||||
	t := tag.Get("toml")
 | 
			
		||||
	if t == "-" {
 | 
			
		||||
		return tagOptions{skip: true}
 | 
			
		||||
	}
 | 
			
		||||
	var opts tagOptions
 | 
			
		||||
	parts := strings.Split(t, ",")
 | 
			
		||||
	opts.name = parts[0]
 | 
			
		||||
	for _, s := range parts[1:] {
 | 
			
		||||
		switch s {
 | 
			
		||||
		case "omitempty":
 | 
			
		||||
			opts.omitempty = true
 | 
			
		||||
		case "omitzero":
 | 
			
		||||
			opts.omitzero = true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return opts
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isZero(rv reflect.Value) bool {
 | 
			
		||||
	switch rv.Kind() {
 | 
			
		||||
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
 | 
			
		||||
		return rv.Int() == 0
 | 
			
		||||
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
 | 
			
		||||
		return rv.Uint() == 0
 | 
			
		||||
	case reflect.Float32, reflect.Float64:
 | 
			
		||||
		return rv.Float() == 0.0
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isEmpty(rv reflect.Value) bool {
 | 
			
		||||
	switch rv.Kind() {
 | 
			
		||||
	case reflect.Array, reflect.Slice, reflect.Map, reflect.String:
 | 
			
		||||
		return rv.Len() == 0
 | 
			
		||||
	case reflect.Bool:
 | 
			
		||||
		return !rv.Bool()
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) newline() {
 | 
			
		||||
	if enc.hasWritten {
 | 
			
		||||
		enc.wf("\n")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) keyEqElement(key Key, val reflect.Value) {
 | 
			
		||||
	if len(key) == 0 {
 | 
			
		||||
		encPanic(errNoKey)
 | 
			
		||||
	}
 | 
			
		||||
	panicIfInvalidKey(key)
 | 
			
		||||
	enc.wf("%s%s = ", enc.indentStr(key), key.maybeQuoted(len(key)-1))
 | 
			
		||||
	enc.eElement(val)
 | 
			
		||||
	enc.newline()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) wf(format string, v ...interface{}) {
 | 
			
		||||
	if _, err := fmt.Fprintf(enc.w, format, v...); err != nil {
 | 
			
		||||
		encPanic(err)
 | 
			
		||||
	}
 | 
			
		||||
	enc.hasWritten = true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (enc *Encoder) indentStr(key Key) string {
 | 
			
		||||
	return strings.Repeat(enc.Indent, len(key)-1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func encPanic(err error) {
 | 
			
		||||
	panic(tomlEncodeError{err})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func eindirect(v reflect.Value) reflect.Value {
 | 
			
		||||
	switch v.Kind() {
 | 
			
		||||
	case reflect.Ptr, reflect.Interface:
 | 
			
		||||
		return eindirect(v.Elem())
 | 
			
		||||
	default:
 | 
			
		||||
		return v
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isNil(rv reflect.Value) bool {
 | 
			
		||||
	switch rv.Kind() {
 | 
			
		||||
	case reflect.Interface, reflect.Map, reflect.Ptr, reflect.Slice:
 | 
			
		||||
		return rv.IsNil()
 | 
			
		||||
	default:
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func panicIfInvalidKey(key Key) {
 | 
			
		||||
	for _, k := range key {
 | 
			
		||||
		if len(k) == 0 {
 | 
			
		||||
			encPanic(e("Key '%s' is not a valid table name. Key names "+
 | 
			
		||||
				"cannot be empty.", key.maybeQuotedAll()))
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isValidKeyName(s string) bool {
 | 
			
		||||
	return len(s) != 0
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										19
									
								
								vendor/github.com/BurntSushi/toml/encoding_types.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								vendor/github.com/BurntSushi/toml/encoding_types.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,19 @@
 | 
			
		||||
// +build go1.2
 | 
			
		||||
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
// In order to support Go 1.1, we define our own TextMarshaler and
 | 
			
		||||
// TextUnmarshaler types. For Go 1.2+, we just alias them with the
 | 
			
		||||
// standard library interfaces.
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here
 | 
			
		||||
// so that Go 1.1 can be supported.
 | 
			
		||||
type TextMarshaler encoding.TextMarshaler
 | 
			
		||||
 | 
			
		||||
// TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined
 | 
			
		||||
// here so that Go 1.1 can be supported.
 | 
			
		||||
type TextUnmarshaler encoding.TextUnmarshaler
 | 
			
		||||
							
								
								
									
										18
									
								
								vendor/github.com/BurntSushi/toml/encoding_types_1.1.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/BurntSushi/toml/encoding_types_1.1.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
			
		||||
// +build !go1.2
 | 
			
		||||
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
// These interfaces were introduced in Go 1.2, so we add them manually when
 | 
			
		||||
// compiling for Go 1.1.
 | 
			
		||||
 | 
			
		||||
// TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here
 | 
			
		||||
// so that Go 1.1 can be supported.
 | 
			
		||||
type TextMarshaler interface {
 | 
			
		||||
	MarshalText() (text []byte, err error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined
 | 
			
		||||
// here so that Go 1.1 can be supported.
 | 
			
		||||
type TextUnmarshaler interface {
 | 
			
		||||
	UnmarshalText(text []byte) error
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										858
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										858
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,858 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"unicode"
 | 
			
		||||
	"unicode/utf8"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type itemType int
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	itemError itemType = iota
 | 
			
		||||
	itemNIL            // used in the parser to indicate no type
 | 
			
		||||
	itemEOF
 | 
			
		||||
	itemText
 | 
			
		||||
	itemString
 | 
			
		||||
	itemRawString
 | 
			
		||||
	itemMultilineString
 | 
			
		||||
	itemRawMultilineString
 | 
			
		||||
	itemBool
 | 
			
		||||
	itemInteger
 | 
			
		||||
	itemFloat
 | 
			
		||||
	itemDatetime
 | 
			
		||||
	itemArray // the start of an array
 | 
			
		||||
	itemArrayEnd
 | 
			
		||||
	itemTableStart
 | 
			
		||||
	itemTableEnd
 | 
			
		||||
	itemArrayTableStart
 | 
			
		||||
	itemArrayTableEnd
 | 
			
		||||
	itemKeyStart
 | 
			
		||||
	itemCommentStart
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	eof             = 0
 | 
			
		||||
	tableStart      = '['
 | 
			
		||||
	tableEnd        = ']'
 | 
			
		||||
	arrayTableStart = '['
 | 
			
		||||
	arrayTableEnd   = ']'
 | 
			
		||||
	tableSep        = '.'
 | 
			
		||||
	keySep          = '='
 | 
			
		||||
	arrayStart      = '['
 | 
			
		||||
	arrayEnd        = ']'
 | 
			
		||||
	arrayValTerm    = ','
 | 
			
		||||
	commentStart    = '#'
 | 
			
		||||
	stringStart     = '"'
 | 
			
		||||
	stringEnd       = '"'
 | 
			
		||||
	rawStringStart  = '\''
 | 
			
		||||
	rawStringEnd    = '\''
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type stateFn func(lx *lexer) stateFn
 | 
			
		||||
 | 
			
		||||
type lexer struct {
 | 
			
		||||
	input string
 | 
			
		||||
	start int
 | 
			
		||||
	pos   int
 | 
			
		||||
	width int
 | 
			
		||||
	line  int
 | 
			
		||||
	state stateFn
 | 
			
		||||
	items chan item
 | 
			
		||||
 | 
			
		||||
	// A stack of state functions used to maintain context.
 | 
			
		||||
	// The idea is to reuse parts of the state machine in various places.
 | 
			
		||||
	// For example, values can appear at the top level or within arbitrarily
 | 
			
		||||
	// nested arrays. The last state on the stack is used after a value has
 | 
			
		||||
	// been lexed. Similarly for comments.
 | 
			
		||||
	stack []stateFn
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type item struct {
 | 
			
		||||
	typ  itemType
 | 
			
		||||
	val  string
 | 
			
		||||
	line int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) nextItem() item {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case item := <-lx.items:
 | 
			
		||||
			return item
 | 
			
		||||
		default:
 | 
			
		||||
			lx.state = lx.state(lx)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lex(input string) *lexer {
 | 
			
		||||
	lx := &lexer{
 | 
			
		||||
		input: input + "\n",
 | 
			
		||||
		state: lexTop,
 | 
			
		||||
		line:  1,
 | 
			
		||||
		items: make(chan item, 10),
 | 
			
		||||
		stack: make([]stateFn, 0, 10),
 | 
			
		||||
	}
 | 
			
		||||
	return lx
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) push(state stateFn) {
 | 
			
		||||
	lx.stack = append(lx.stack, state)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) pop() stateFn {
 | 
			
		||||
	if len(lx.stack) == 0 {
 | 
			
		||||
		return lx.errorf("BUG in lexer: no states to pop.")
 | 
			
		||||
	}
 | 
			
		||||
	last := lx.stack[len(lx.stack)-1]
 | 
			
		||||
	lx.stack = lx.stack[0 : len(lx.stack)-1]
 | 
			
		||||
	return last
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) current() string {
 | 
			
		||||
	return lx.input[lx.start:lx.pos]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) emit(typ itemType) {
 | 
			
		||||
	lx.items <- item{typ, lx.current(), lx.line}
 | 
			
		||||
	lx.start = lx.pos
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) emitTrim(typ itemType) {
 | 
			
		||||
	lx.items <- item{typ, strings.TrimSpace(lx.current()), lx.line}
 | 
			
		||||
	lx.start = lx.pos
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (lx *lexer) next() (r rune) {
 | 
			
		||||
	if lx.pos >= len(lx.input) {
 | 
			
		||||
		lx.width = 0
 | 
			
		||||
		return eof
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if lx.input[lx.pos] == '\n' {
 | 
			
		||||
		lx.line++
 | 
			
		||||
	}
 | 
			
		||||
	r, lx.width = utf8.DecodeRuneInString(lx.input[lx.pos:])
 | 
			
		||||
	lx.pos += lx.width
 | 
			
		||||
	return r
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ignore skips over the pending input before this point.
 | 
			
		||||
func (lx *lexer) ignore() {
 | 
			
		||||
	lx.start = lx.pos
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// backup steps back one rune. Can be called only once per call of next.
 | 
			
		||||
func (lx *lexer) backup() {
 | 
			
		||||
	lx.pos -= lx.width
 | 
			
		||||
	if lx.pos < len(lx.input) && lx.input[lx.pos] == '\n' {
 | 
			
		||||
		lx.line--
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// accept consumes the next rune if it's equal to `valid`.
 | 
			
		||||
func (lx *lexer) accept(valid rune) bool {
 | 
			
		||||
	if lx.next() == valid {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// peek returns but does not consume the next rune in the input.
 | 
			
		||||
func (lx *lexer) peek() rune {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	return r
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// skip ignores all input that matches the given predicate.
 | 
			
		||||
func (lx *lexer) skip(pred func(rune) bool) {
 | 
			
		||||
	for {
 | 
			
		||||
		r := lx.next()
 | 
			
		||||
		if pred(r) {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// errorf stops all lexing by emitting an error and returning `nil`.
 | 
			
		||||
// Note that any value that is a character is escaped if it's a special
 | 
			
		||||
// character (new lines, tabs, etc.).
 | 
			
		||||
func (lx *lexer) errorf(format string, values ...interface{}) stateFn {
 | 
			
		||||
	lx.items <- item{
 | 
			
		||||
		itemError,
 | 
			
		||||
		fmt.Sprintf(format, values...),
 | 
			
		||||
		lx.line,
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexTop consumes elements at the top level of TOML data.
 | 
			
		||||
func lexTop(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isWhitespace(r) || isNL(r) {
 | 
			
		||||
		return lexSkip(lx, lexTop)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	switch r {
 | 
			
		||||
	case commentStart:
 | 
			
		||||
		lx.push(lexTop)
 | 
			
		||||
		return lexCommentStart
 | 
			
		||||
	case tableStart:
 | 
			
		||||
		return lexTableStart
 | 
			
		||||
	case eof:
 | 
			
		||||
		if lx.pos > lx.start {
 | 
			
		||||
			return lx.errorf("Unexpected EOF.")
 | 
			
		||||
		}
 | 
			
		||||
		lx.emit(itemEOF)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// At this point, the only valid item can be a key, so we back up
 | 
			
		||||
	// and let the key lexer do the rest.
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.push(lexTopEnd)
 | 
			
		||||
	return lexKeyStart
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexTopEnd is entered whenever a top-level item has been consumed. (A value
 | 
			
		||||
// or a table.) It must see only whitespace, and will turn back to lexTop
 | 
			
		||||
// upon a new line. If it sees EOF, it will quit the lexer successfully.
 | 
			
		||||
func lexTopEnd(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case r == commentStart:
 | 
			
		||||
		// a comment will read to a new line for us.
 | 
			
		||||
		lx.push(lexTop)
 | 
			
		||||
		return lexCommentStart
 | 
			
		||||
	case isWhitespace(r):
 | 
			
		||||
		return lexTopEnd
 | 
			
		||||
	case isNL(r):
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lexTop
 | 
			
		||||
	case r == eof:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lexTop
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Expected a top-level item to end with a new line, "+
 | 
			
		||||
		"comment or EOF, but got %q instead.", r)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexTable lexes the beginning of a table. Namely, it makes sure that
 | 
			
		||||
// it starts with a character other than '.' and ']'.
 | 
			
		||||
// It assumes that '[' has already been consumed.
 | 
			
		||||
// It also handles the case that this is an item in an array of tables.
 | 
			
		||||
// e.g., '[[name]]'.
 | 
			
		||||
func lexTableStart(lx *lexer) stateFn {
 | 
			
		||||
	if lx.peek() == arrayTableStart {
 | 
			
		||||
		lx.next()
 | 
			
		||||
		lx.emit(itemArrayTableStart)
 | 
			
		||||
		lx.push(lexArrayTableEnd)
 | 
			
		||||
	} else {
 | 
			
		||||
		lx.emit(itemTableStart)
 | 
			
		||||
		lx.push(lexTableEnd)
 | 
			
		||||
	}
 | 
			
		||||
	return lexTableNameStart
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexTableEnd(lx *lexer) stateFn {
 | 
			
		||||
	lx.emit(itemTableEnd)
 | 
			
		||||
	return lexTopEnd
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexArrayTableEnd(lx *lexer) stateFn {
 | 
			
		||||
	if r := lx.next(); r != arrayTableEnd {
 | 
			
		||||
		return lx.errorf("Expected end of table array name delimiter %q, "+
 | 
			
		||||
			"but got %q instead.", arrayTableEnd, r)
 | 
			
		||||
	}
 | 
			
		||||
	lx.emit(itemArrayTableEnd)
 | 
			
		||||
	return lexTopEnd
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexTableNameStart(lx *lexer) stateFn {
 | 
			
		||||
	lx.skip(isWhitespace)
 | 
			
		||||
	switch r := lx.peek(); {
 | 
			
		||||
	case r == tableEnd || r == eof:
 | 
			
		||||
		return lx.errorf("Unexpected end of table name. (Table names cannot " +
 | 
			
		||||
			"be empty.)")
 | 
			
		||||
	case r == tableSep:
 | 
			
		||||
		return lx.errorf("Unexpected table separator. (Table names cannot " +
 | 
			
		||||
			"be empty.)")
 | 
			
		||||
	case r == stringStart || r == rawStringStart:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		lx.push(lexTableNameEnd)
 | 
			
		||||
		return lexValue // reuse string lexing
 | 
			
		||||
	default:
 | 
			
		||||
		return lexBareTableName
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexBareTableName lexes the name of a table. It assumes that at least one
 | 
			
		||||
// valid character for the table has already been read.
 | 
			
		||||
func lexBareTableName(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isBareKeyChar(r) {
 | 
			
		||||
		return lexBareTableName
 | 
			
		||||
	}
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.emit(itemText)
 | 
			
		||||
	return lexTableNameEnd
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexTableNameEnd reads the end of a piece of a table name, optionally
 | 
			
		||||
// consuming whitespace.
 | 
			
		||||
func lexTableNameEnd(lx *lexer) stateFn {
 | 
			
		||||
	lx.skip(isWhitespace)
 | 
			
		||||
	switch r := lx.next(); {
 | 
			
		||||
	case isWhitespace(r):
 | 
			
		||||
		return lexTableNameEnd
 | 
			
		||||
	case r == tableSep:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lexTableNameStart
 | 
			
		||||
	case r == tableEnd:
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	default:
 | 
			
		||||
		return lx.errorf("Expected '.' or ']' to end table name, but got %q "+
 | 
			
		||||
			"instead.", r)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexKeyStart consumes a key name up until the first non-whitespace character.
 | 
			
		||||
// lexKeyStart will ignore whitespace.
 | 
			
		||||
func lexKeyStart(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.peek()
 | 
			
		||||
	switch {
 | 
			
		||||
	case r == keySep:
 | 
			
		||||
		return lx.errorf("Unexpected key separator %q.", keySep)
 | 
			
		||||
	case isWhitespace(r) || isNL(r):
 | 
			
		||||
		lx.next()
 | 
			
		||||
		return lexSkip(lx, lexKeyStart)
 | 
			
		||||
	case r == stringStart || r == rawStringStart:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		lx.emit(itemKeyStart)
 | 
			
		||||
		lx.push(lexKeyEnd)
 | 
			
		||||
		return lexValue // reuse string lexing
 | 
			
		||||
	default:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		lx.emit(itemKeyStart)
 | 
			
		||||
		return lexBareKey
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexBareKey consumes the text of a bare key. Assumes that the first character
 | 
			
		||||
// (which is not whitespace) has not yet been consumed.
 | 
			
		||||
func lexBareKey(lx *lexer) stateFn {
 | 
			
		||||
	switch r := lx.next(); {
 | 
			
		||||
	case isBareKeyChar(r):
 | 
			
		||||
		return lexBareKey
 | 
			
		||||
	case isWhitespace(r):
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		lx.emit(itemText)
 | 
			
		||||
		return lexKeyEnd
 | 
			
		||||
	case r == keySep:
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		lx.emit(itemText)
 | 
			
		||||
		return lexKeyEnd
 | 
			
		||||
	default:
 | 
			
		||||
		return lx.errorf("Bare keys cannot contain %q.", r)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexKeyEnd consumes the end of a key and trims whitespace (up to the key
 | 
			
		||||
// separator).
 | 
			
		||||
func lexKeyEnd(lx *lexer) stateFn {
 | 
			
		||||
	switch r := lx.next(); {
 | 
			
		||||
	case r == keySep:
 | 
			
		||||
		return lexSkip(lx, lexValue)
 | 
			
		||||
	case isWhitespace(r):
 | 
			
		||||
		return lexSkip(lx, lexKeyEnd)
 | 
			
		||||
	default:
 | 
			
		||||
		return lx.errorf("Expected key separator %q, but got %q instead.",
 | 
			
		||||
			keySep, r)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexValue starts the consumption of a value anywhere a value is expected.
 | 
			
		||||
// lexValue will ignore whitespace.
 | 
			
		||||
// After a value is lexed, the last state on the next is popped and returned.
 | 
			
		||||
func lexValue(lx *lexer) stateFn {
 | 
			
		||||
	// We allow whitespace to precede a value, but NOT new lines.
 | 
			
		||||
	// In array syntax, the array states are responsible for ignoring new
 | 
			
		||||
	// lines.
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case isWhitespace(r):
 | 
			
		||||
		return lexSkip(lx, lexValue)
 | 
			
		||||
	case isDigit(r):
 | 
			
		||||
		lx.backup() // avoid an extra state and use the same as above
 | 
			
		||||
		return lexNumberOrDateStart
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case arrayStart:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		lx.emit(itemArray)
 | 
			
		||||
		return lexArrayValue
 | 
			
		||||
	case stringStart:
 | 
			
		||||
		if lx.accept(stringStart) {
 | 
			
		||||
			if lx.accept(stringStart) {
 | 
			
		||||
				lx.ignore() // Ignore """
 | 
			
		||||
				return lexMultilineString
 | 
			
		||||
			}
 | 
			
		||||
			lx.backup()
 | 
			
		||||
		}
 | 
			
		||||
		lx.ignore() // ignore the '"'
 | 
			
		||||
		return lexString
 | 
			
		||||
	case rawStringStart:
 | 
			
		||||
		if lx.accept(rawStringStart) {
 | 
			
		||||
			if lx.accept(rawStringStart) {
 | 
			
		||||
				lx.ignore() // Ignore """
 | 
			
		||||
				return lexMultilineRawString
 | 
			
		||||
			}
 | 
			
		||||
			lx.backup()
 | 
			
		||||
		}
 | 
			
		||||
		lx.ignore() // ignore the "'"
 | 
			
		||||
		return lexRawString
 | 
			
		||||
	case '+', '-':
 | 
			
		||||
		return lexNumberStart
 | 
			
		||||
	case '.': // special error case, be kind to users
 | 
			
		||||
		return lx.errorf("Floats must start with a digit, not '.'.")
 | 
			
		||||
	}
 | 
			
		||||
	if unicode.IsLetter(r) {
 | 
			
		||||
		// Be permissive here; lexBool will give a nice error if the
 | 
			
		||||
		// user wrote something like
 | 
			
		||||
		//   x = foo
 | 
			
		||||
		// (i.e. not 'true' or 'false' but is something else word-like.)
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		return lexBool
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Expected value but found %q instead.", r)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexArrayValue consumes one value in an array. It assumes that '[' or ','
 | 
			
		||||
// have already been consumed. All whitespace and new lines are ignored.
 | 
			
		||||
func lexArrayValue(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case isWhitespace(r) || isNL(r):
 | 
			
		||||
		return lexSkip(lx, lexArrayValue)
 | 
			
		||||
	case r == commentStart:
 | 
			
		||||
		lx.push(lexArrayValue)
 | 
			
		||||
		return lexCommentStart
 | 
			
		||||
	case r == arrayValTerm:
 | 
			
		||||
		return lx.errorf("Unexpected array value terminator %q.",
 | 
			
		||||
			arrayValTerm)
 | 
			
		||||
	case r == arrayEnd:
 | 
			
		||||
		return lexArrayEnd
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.push(lexArrayValueEnd)
 | 
			
		||||
	return lexValue
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexArrayValueEnd consumes the cruft between values of an array. Namely,
 | 
			
		||||
// it ignores whitespace and expects either a ',' or a ']'.
 | 
			
		||||
func lexArrayValueEnd(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case isWhitespace(r) || isNL(r):
 | 
			
		||||
		return lexSkip(lx, lexArrayValueEnd)
 | 
			
		||||
	case r == commentStart:
 | 
			
		||||
		lx.push(lexArrayValueEnd)
 | 
			
		||||
		return lexCommentStart
 | 
			
		||||
	case r == arrayValTerm:
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lexArrayValue // move on to the next value
 | 
			
		||||
	case r == arrayEnd:
 | 
			
		||||
		return lexArrayEnd
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Expected an array value terminator %q or an array "+
 | 
			
		||||
		"terminator %q, but got %q instead.", arrayValTerm, arrayEnd, r)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexArrayEnd finishes the lexing of an array. It assumes that a ']' has
 | 
			
		||||
// just been consumed.
 | 
			
		||||
func lexArrayEnd(lx *lexer) stateFn {
 | 
			
		||||
	lx.ignore()
 | 
			
		||||
	lx.emit(itemArrayEnd)
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexString consumes the inner contents of a string. It assumes that the
 | 
			
		||||
// beginning '"' has already been consumed and ignored.
 | 
			
		||||
func lexString(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case isNL(r):
 | 
			
		||||
		return lx.errorf("Strings cannot contain new lines.")
 | 
			
		||||
	case r == '\\':
 | 
			
		||||
		lx.push(lexString)
 | 
			
		||||
		return lexStringEscape
 | 
			
		||||
	case r == stringEnd:
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		lx.emit(itemString)
 | 
			
		||||
		lx.next()
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	}
 | 
			
		||||
	return lexString
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexMultilineString consumes the inner contents of a string. It assumes that
 | 
			
		||||
// the beginning '"""' has already been consumed and ignored.
 | 
			
		||||
func lexMultilineString(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case r == '\\':
 | 
			
		||||
		return lexMultilineStringEscape
 | 
			
		||||
	case r == stringEnd:
 | 
			
		||||
		if lx.accept(stringEnd) {
 | 
			
		||||
			if lx.accept(stringEnd) {
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.emit(itemMultilineString)
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.ignore()
 | 
			
		||||
				return lx.pop()
 | 
			
		||||
			}
 | 
			
		||||
			lx.backup()
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return lexMultilineString
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexRawString consumes a raw string. Nothing can be escaped in such a string.
 | 
			
		||||
// It assumes that the beginning "'" has already been consumed and ignored.
 | 
			
		||||
func lexRawString(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case isNL(r):
 | 
			
		||||
		return lx.errorf("Strings cannot contain new lines.")
 | 
			
		||||
	case r == rawStringEnd:
 | 
			
		||||
		lx.backup()
 | 
			
		||||
		lx.emit(itemRawString)
 | 
			
		||||
		lx.next()
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	}
 | 
			
		||||
	return lexRawString
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexMultilineRawString consumes a raw string. Nothing can be escaped in such
 | 
			
		||||
// a string. It assumes that the beginning "'" has already been consumed and
 | 
			
		||||
// ignored.
 | 
			
		||||
func lexMultilineRawString(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch {
 | 
			
		||||
	case r == rawStringEnd:
 | 
			
		||||
		if lx.accept(rawStringEnd) {
 | 
			
		||||
			if lx.accept(rawStringEnd) {
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.backup()
 | 
			
		||||
				lx.emit(itemRawMultilineString)
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.next()
 | 
			
		||||
				lx.ignore()
 | 
			
		||||
				return lx.pop()
 | 
			
		||||
			}
 | 
			
		||||
			lx.backup()
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return lexMultilineRawString
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexMultilineStringEscape consumes an escaped character. It assumes that the
 | 
			
		||||
// preceding '\\' has already been consumed.
 | 
			
		||||
func lexMultilineStringEscape(lx *lexer) stateFn {
 | 
			
		||||
	// Handle the special case first:
 | 
			
		||||
	if isNL(lx.next()) {
 | 
			
		||||
		return lexMultilineString
 | 
			
		||||
	}
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.push(lexMultilineString)
 | 
			
		||||
	return lexStringEscape(lx)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexStringEscape(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	switch r {
 | 
			
		||||
	case 'b':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case 't':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case 'n':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case 'f':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case 'r':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '"':
 | 
			
		||||
		fallthrough
 | 
			
		||||
	case '\\':
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	case 'u':
 | 
			
		||||
		return lexShortUnicodeEscape
 | 
			
		||||
	case 'U':
 | 
			
		||||
		return lexLongUnicodeEscape
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Invalid escape character %q. Only the following "+
 | 
			
		||||
		"escape characters are allowed: "+
 | 
			
		||||
		"\\b, \\t, \\n, \\f, \\r, \\\", \\/, \\\\, "+
 | 
			
		||||
		"\\uXXXX and \\UXXXXXXXX.", r)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexShortUnicodeEscape(lx *lexer) stateFn {
 | 
			
		||||
	var r rune
 | 
			
		||||
	for i := 0; i < 4; i++ {
 | 
			
		||||
		r = lx.next()
 | 
			
		||||
		if !isHexadecimal(r) {
 | 
			
		||||
			return lx.errorf("Expected four hexadecimal digits after '\\u', "+
 | 
			
		||||
				"but got '%s' instead.", lx.current())
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func lexLongUnicodeEscape(lx *lexer) stateFn {
 | 
			
		||||
	var r rune
 | 
			
		||||
	for i := 0; i < 8; i++ {
 | 
			
		||||
		r = lx.next()
 | 
			
		||||
		if !isHexadecimal(r) {
 | 
			
		||||
			return lx.errorf("Expected eight hexadecimal digits after '\\U', "+
 | 
			
		||||
				"but got '%s' instead.", lx.current())
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexNumberOrDateStart consumes either an integer, a float, or datetime.
 | 
			
		||||
func lexNumberOrDateStart(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isDigit(r) {
 | 
			
		||||
		return lexNumberOrDate
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case '_':
 | 
			
		||||
		return lexNumber
 | 
			
		||||
	case 'e', 'E':
 | 
			
		||||
		return lexFloat
 | 
			
		||||
	case '.':
 | 
			
		||||
		return lx.errorf("Floats must start with a digit, not '.'.")
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Expected a digit but got %q.", r)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexNumberOrDate consumes either an integer, float or datetime.
 | 
			
		||||
func lexNumberOrDate(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isDigit(r) {
 | 
			
		||||
		return lexNumberOrDate
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case '-':
 | 
			
		||||
		return lexDatetime
 | 
			
		||||
	case '_':
 | 
			
		||||
		return lexNumber
 | 
			
		||||
	case '.', 'e', 'E':
 | 
			
		||||
		return lexFloat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.emit(itemInteger)
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexDatetime consumes a Datetime, to a first approximation.
 | 
			
		||||
// The parser validates that it matches one of the accepted formats.
 | 
			
		||||
func lexDatetime(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isDigit(r) {
 | 
			
		||||
		return lexDatetime
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case '-', 'T', ':', '.', 'Z':
 | 
			
		||||
		return lexDatetime
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.emit(itemDatetime)
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexNumberStart consumes either an integer or a float. It assumes that a sign
 | 
			
		||||
// has already been read, but that *no* digits have been consumed.
 | 
			
		||||
// lexNumberStart will move to the appropriate integer or float states.
 | 
			
		||||
func lexNumberStart(lx *lexer) stateFn {
 | 
			
		||||
	// We MUST see a digit. Even floats have to start with a digit.
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if !isDigit(r) {
 | 
			
		||||
		if r == '.' {
 | 
			
		||||
			return lx.errorf("Floats must start with a digit, not '.'.")
 | 
			
		||||
		}
 | 
			
		||||
		return lx.errorf("Expected a digit but got %q.", r)
 | 
			
		||||
	}
 | 
			
		||||
	return lexNumber
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexNumber consumes an integer or a float after seeing the first digit.
 | 
			
		||||
func lexNumber(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isDigit(r) {
 | 
			
		||||
		return lexNumber
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case '_':
 | 
			
		||||
		return lexNumber
 | 
			
		||||
	case '.', 'e', 'E':
 | 
			
		||||
		return lexFloat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.emit(itemInteger)
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexFloat consumes the elements of a float. It allows any sequence of
 | 
			
		||||
// float-like characters, so floats emitted by the lexer are only a first
 | 
			
		||||
// approximation and must be validated by the parser.
 | 
			
		||||
func lexFloat(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.next()
 | 
			
		||||
	if isDigit(r) {
 | 
			
		||||
		return lexFloat
 | 
			
		||||
	}
 | 
			
		||||
	switch r {
 | 
			
		||||
	case '_', '.', '-', '+', 'e', 'E':
 | 
			
		||||
		return lexFloat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lx.backup()
 | 
			
		||||
	lx.emit(itemFloat)
 | 
			
		||||
	return lx.pop()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexBool consumes a bool string: 'true' or 'false.
 | 
			
		||||
func lexBool(lx *lexer) stateFn {
 | 
			
		||||
	var rs []rune
 | 
			
		||||
	for {
 | 
			
		||||
		r := lx.next()
 | 
			
		||||
		if r == eof || isWhitespace(r) || isNL(r) {
 | 
			
		||||
			lx.backup()
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		rs = append(rs, r)
 | 
			
		||||
	}
 | 
			
		||||
	s := string(rs)
 | 
			
		||||
	switch s {
 | 
			
		||||
	case "true", "false":
 | 
			
		||||
		lx.emit(itemBool)
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	}
 | 
			
		||||
	return lx.errorf("Expected value but found %q instead.", s)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexCommentStart begins the lexing of a comment. It will emit
 | 
			
		||||
// itemCommentStart and consume no characters, passing control to lexComment.
 | 
			
		||||
func lexCommentStart(lx *lexer) stateFn {
 | 
			
		||||
	lx.ignore()
 | 
			
		||||
	lx.emit(itemCommentStart)
 | 
			
		||||
	return lexComment
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexComment lexes an entire comment. It assumes that '#' has been consumed.
 | 
			
		||||
// It will consume *up to* the first new line character, and pass control
 | 
			
		||||
// back to the last state on the stack.
 | 
			
		||||
func lexComment(lx *lexer) stateFn {
 | 
			
		||||
	r := lx.peek()
 | 
			
		||||
	if isNL(r) || r == eof {
 | 
			
		||||
		lx.emit(itemText)
 | 
			
		||||
		return lx.pop()
 | 
			
		||||
	}
 | 
			
		||||
	lx.next()
 | 
			
		||||
	return lexComment
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// lexSkip ignores all slurped input and moves on to the next state.
 | 
			
		||||
func lexSkip(lx *lexer, nextState stateFn) stateFn {
 | 
			
		||||
	return func(lx *lexer) stateFn {
 | 
			
		||||
		lx.ignore()
 | 
			
		||||
		return nextState
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// isWhitespace returns true if `r` is a whitespace character according
 | 
			
		||||
// to the spec.
 | 
			
		||||
func isWhitespace(r rune) bool {
 | 
			
		||||
	return r == '\t' || r == ' '
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isNL(r rune) bool {
 | 
			
		||||
	return r == '\n' || r == '\r'
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isDigit(r rune) bool {
 | 
			
		||||
	return r >= '0' && r <= '9'
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isHexadecimal(r rune) bool {
 | 
			
		||||
	return (r >= '0' && r <= '9') ||
 | 
			
		||||
		(r >= 'a' && r <= 'f') ||
 | 
			
		||||
		(r >= 'A' && r <= 'F')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isBareKeyChar(r rune) bool {
 | 
			
		||||
	return (r >= 'A' && r <= 'Z') ||
 | 
			
		||||
		(r >= 'a' && r <= 'z') ||
 | 
			
		||||
		(r >= '0' && r <= '9') ||
 | 
			
		||||
		r == '_' ||
 | 
			
		||||
		r == '-'
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (itype itemType) String() string {
 | 
			
		||||
	switch itype {
 | 
			
		||||
	case itemError:
 | 
			
		||||
		return "Error"
 | 
			
		||||
	case itemNIL:
 | 
			
		||||
		return "NIL"
 | 
			
		||||
	case itemEOF:
 | 
			
		||||
		return "EOF"
 | 
			
		||||
	case itemText:
 | 
			
		||||
		return "Text"
 | 
			
		||||
	case itemString, itemRawString, itemMultilineString, itemRawMultilineString:
 | 
			
		||||
		return "String"
 | 
			
		||||
	case itemBool:
 | 
			
		||||
		return "Bool"
 | 
			
		||||
	case itemInteger:
 | 
			
		||||
		return "Integer"
 | 
			
		||||
	case itemFloat:
 | 
			
		||||
		return "Float"
 | 
			
		||||
	case itemDatetime:
 | 
			
		||||
		return "DateTime"
 | 
			
		||||
	case itemTableStart:
 | 
			
		||||
		return "TableStart"
 | 
			
		||||
	case itemTableEnd:
 | 
			
		||||
		return "TableEnd"
 | 
			
		||||
	case itemKeyStart:
 | 
			
		||||
		return "KeyStart"
 | 
			
		||||
	case itemArray:
 | 
			
		||||
		return "Array"
 | 
			
		||||
	case itemArrayEnd:
 | 
			
		||||
		return "ArrayEnd"
 | 
			
		||||
	case itemCommentStart:
 | 
			
		||||
		return "CommentStart"
 | 
			
		||||
	}
 | 
			
		||||
	panic(fmt.Sprintf("BUG: Unknown type '%d'.", int(itype)))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (item item) String() string {
 | 
			
		||||
	return fmt.Sprintf("(%s, %s)", item.typ.String(), item.val)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										557
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										557
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,557 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
	"unicode"
 | 
			
		||||
	"unicode/utf8"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type parser struct {
 | 
			
		||||
	mapping map[string]interface{}
 | 
			
		||||
	types   map[string]tomlType
 | 
			
		||||
	lx      *lexer
 | 
			
		||||
 | 
			
		||||
	// A list of keys in the order that they appear in the TOML data.
 | 
			
		||||
	ordered []Key
 | 
			
		||||
 | 
			
		||||
	// the full key for the current hash in scope
 | 
			
		||||
	context Key
 | 
			
		||||
 | 
			
		||||
	// the base key name for everything except hashes
 | 
			
		||||
	currentKey string
 | 
			
		||||
 | 
			
		||||
	// rough approximation of line number
 | 
			
		||||
	approxLine int
 | 
			
		||||
 | 
			
		||||
	// A map of 'key.group.names' to whether they were created implicitly.
 | 
			
		||||
	implicits map[string]bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type parseError string
 | 
			
		||||
 | 
			
		||||
func (pe parseError) Error() string {
 | 
			
		||||
	return string(pe)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func parse(data string) (p *parser, err error) {
 | 
			
		||||
	defer func() {
 | 
			
		||||
		if r := recover(); r != nil {
 | 
			
		||||
			var ok bool
 | 
			
		||||
			if err, ok = r.(parseError); ok {
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			panic(r)
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	p = &parser{
 | 
			
		||||
		mapping:   make(map[string]interface{}),
 | 
			
		||||
		types:     make(map[string]tomlType),
 | 
			
		||||
		lx:        lex(data),
 | 
			
		||||
		ordered:   make([]Key, 0),
 | 
			
		||||
		implicits: make(map[string]bool),
 | 
			
		||||
	}
 | 
			
		||||
	for {
 | 
			
		||||
		item := p.next()
 | 
			
		||||
		if item.typ == itemEOF {
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		p.topLevel(item)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return p, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) panicf(format string, v ...interface{}) {
 | 
			
		||||
	msg := fmt.Sprintf("Near line %d (last key parsed '%s'): %s",
 | 
			
		||||
		p.approxLine, p.current(), fmt.Sprintf(format, v...))
 | 
			
		||||
	panic(parseError(msg))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) next() item {
 | 
			
		||||
	it := p.lx.nextItem()
 | 
			
		||||
	if it.typ == itemError {
 | 
			
		||||
		p.panicf("%s", it.val)
 | 
			
		||||
	}
 | 
			
		||||
	return it
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) bug(format string, v ...interface{}) {
 | 
			
		||||
	panic(fmt.Sprintf("BUG: "+format+"\n\n", v...))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) expect(typ itemType) item {
 | 
			
		||||
	it := p.next()
 | 
			
		||||
	p.assertEqual(typ, it.typ)
 | 
			
		||||
	return it
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) assertEqual(expected, got itemType) {
 | 
			
		||||
	if expected != got {
 | 
			
		||||
		p.bug("Expected '%s' but got '%s'.", expected, got)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) topLevel(item item) {
 | 
			
		||||
	switch item.typ {
 | 
			
		||||
	case itemCommentStart:
 | 
			
		||||
		p.approxLine = item.line
 | 
			
		||||
		p.expect(itemText)
 | 
			
		||||
	case itemTableStart:
 | 
			
		||||
		kg := p.next()
 | 
			
		||||
		p.approxLine = kg.line
 | 
			
		||||
 | 
			
		||||
		var key Key
 | 
			
		||||
		for ; kg.typ != itemTableEnd && kg.typ != itemEOF; kg = p.next() {
 | 
			
		||||
			key = append(key, p.keyString(kg))
 | 
			
		||||
		}
 | 
			
		||||
		p.assertEqual(itemTableEnd, kg.typ)
 | 
			
		||||
 | 
			
		||||
		p.establishContext(key, false)
 | 
			
		||||
		p.setType("", tomlHash)
 | 
			
		||||
		p.ordered = append(p.ordered, key)
 | 
			
		||||
	case itemArrayTableStart:
 | 
			
		||||
		kg := p.next()
 | 
			
		||||
		p.approxLine = kg.line
 | 
			
		||||
 | 
			
		||||
		var key Key
 | 
			
		||||
		for ; kg.typ != itemArrayTableEnd && kg.typ != itemEOF; kg = p.next() {
 | 
			
		||||
			key = append(key, p.keyString(kg))
 | 
			
		||||
		}
 | 
			
		||||
		p.assertEqual(itemArrayTableEnd, kg.typ)
 | 
			
		||||
 | 
			
		||||
		p.establishContext(key, true)
 | 
			
		||||
		p.setType("", tomlArrayHash)
 | 
			
		||||
		p.ordered = append(p.ordered, key)
 | 
			
		||||
	case itemKeyStart:
 | 
			
		||||
		kname := p.next()
 | 
			
		||||
		p.approxLine = kname.line
 | 
			
		||||
		p.currentKey = p.keyString(kname)
 | 
			
		||||
 | 
			
		||||
		val, typ := p.value(p.next())
 | 
			
		||||
		p.setValue(p.currentKey, val)
 | 
			
		||||
		p.setType(p.currentKey, typ)
 | 
			
		||||
		p.ordered = append(p.ordered, p.context.add(p.currentKey))
 | 
			
		||||
		p.currentKey = ""
 | 
			
		||||
	default:
 | 
			
		||||
		p.bug("Unexpected type at top level: %s", item.typ)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Gets a string for a key (or part of a key in a table name).
 | 
			
		||||
func (p *parser) keyString(it item) string {
 | 
			
		||||
	switch it.typ {
 | 
			
		||||
	case itemText:
 | 
			
		||||
		return it.val
 | 
			
		||||
	case itemString, itemMultilineString,
 | 
			
		||||
		itemRawString, itemRawMultilineString:
 | 
			
		||||
		s, _ := p.value(it)
 | 
			
		||||
		return s.(string)
 | 
			
		||||
	default:
 | 
			
		||||
		p.bug("Unexpected key type: %s", it.typ)
 | 
			
		||||
		panic("unreachable")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// value translates an expected value from the lexer into a Go value wrapped
 | 
			
		||||
// as an empty interface.
 | 
			
		||||
func (p *parser) value(it item) (interface{}, tomlType) {
 | 
			
		||||
	switch it.typ {
 | 
			
		||||
	case itemString:
 | 
			
		||||
		return p.replaceEscapes(it.val), p.typeOfPrimitive(it)
 | 
			
		||||
	case itemMultilineString:
 | 
			
		||||
		trimmed := stripFirstNewline(stripEscapedWhitespace(it.val))
 | 
			
		||||
		return p.replaceEscapes(trimmed), p.typeOfPrimitive(it)
 | 
			
		||||
	case itemRawString:
 | 
			
		||||
		return it.val, p.typeOfPrimitive(it)
 | 
			
		||||
	case itemRawMultilineString:
 | 
			
		||||
		return stripFirstNewline(it.val), p.typeOfPrimitive(it)
 | 
			
		||||
	case itemBool:
 | 
			
		||||
		switch it.val {
 | 
			
		||||
		case "true":
 | 
			
		||||
			return true, p.typeOfPrimitive(it)
 | 
			
		||||
		case "false":
 | 
			
		||||
			return false, p.typeOfPrimitive(it)
 | 
			
		||||
		}
 | 
			
		||||
		p.bug("Expected boolean value, but got '%s'.", it.val)
 | 
			
		||||
	case itemInteger:
 | 
			
		||||
		if !numUnderscoresOK(it.val) {
 | 
			
		||||
			p.panicf("Invalid integer %q: underscores must be surrounded by digits",
 | 
			
		||||
				it.val)
 | 
			
		||||
		}
 | 
			
		||||
		val := strings.Replace(it.val, "_", "", -1)
 | 
			
		||||
		num, err := strconv.ParseInt(val, 10, 64)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			// Distinguish integer values. Normally, it'd be a bug if the lexer
 | 
			
		||||
			// provides an invalid integer, but it's possible that the number is
 | 
			
		||||
			// out of range of valid values (which the lexer cannot determine).
 | 
			
		||||
			// So mark the former as a bug but the latter as a legitimate user
 | 
			
		||||
			// error.
 | 
			
		||||
			if e, ok := err.(*strconv.NumError); ok &&
 | 
			
		||||
				e.Err == strconv.ErrRange {
 | 
			
		||||
 | 
			
		||||
				p.panicf("Integer '%s' is out of the range of 64-bit "+
 | 
			
		||||
					"signed integers.", it.val)
 | 
			
		||||
			} else {
 | 
			
		||||
				p.bug("Expected integer value, but got '%s'.", it.val)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return num, p.typeOfPrimitive(it)
 | 
			
		||||
	case itemFloat:
 | 
			
		||||
		parts := strings.FieldsFunc(it.val, func(r rune) bool {
 | 
			
		||||
			switch r {
 | 
			
		||||
			case '.', 'e', 'E':
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			return false
 | 
			
		||||
		})
 | 
			
		||||
		for _, part := range parts {
 | 
			
		||||
			if !numUnderscoresOK(part) {
 | 
			
		||||
				p.panicf("Invalid float %q: underscores must be "+
 | 
			
		||||
					"surrounded by digits", it.val)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if !numPeriodsOK(it.val) {
 | 
			
		||||
			// As a special case, numbers like '123.' or '1.e2',
 | 
			
		||||
			// which are valid as far as Go/strconv are concerned,
 | 
			
		||||
			// must be rejected because TOML says that a fractional
 | 
			
		||||
			// part consists of '.' followed by 1+ digits.
 | 
			
		||||
			p.panicf("Invalid float %q: '.' must be followed "+
 | 
			
		||||
				"by one or more digits", it.val)
 | 
			
		||||
		}
 | 
			
		||||
		val := strings.Replace(it.val, "_", "", -1)
 | 
			
		||||
		num, err := strconv.ParseFloat(val, 64)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			if e, ok := err.(*strconv.NumError); ok &&
 | 
			
		||||
				e.Err == strconv.ErrRange {
 | 
			
		||||
 | 
			
		||||
				p.panicf("Float '%s' is out of the range of 64-bit "+
 | 
			
		||||
					"IEEE-754 floating-point numbers.", it.val)
 | 
			
		||||
			} else {
 | 
			
		||||
				p.panicf("Invalid float value: %q", it.val)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return num, p.typeOfPrimitive(it)
 | 
			
		||||
	case itemDatetime:
 | 
			
		||||
		var t time.Time
 | 
			
		||||
		var ok bool
 | 
			
		||||
		var err error
 | 
			
		||||
		for _, format := range []string{
 | 
			
		||||
			"2006-01-02T15:04:05Z07:00",
 | 
			
		||||
			"2006-01-02T15:04:05",
 | 
			
		||||
			"2006-01-02",
 | 
			
		||||
		} {
 | 
			
		||||
			t, err = time.ParseInLocation(format, it.val, time.Local)
 | 
			
		||||
			if err == nil {
 | 
			
		||||
				ok = true
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if !ok {
 | 
			
		||||
			p.panicf("Invalid TOML Datetime: %q.", it.val)
 | 
			
		||||
		}
 | 
			
		||||
		return t, p.typeOfPrimitive(it)
 | 
			
		||||
	case itemArray:
 | 
			
		||||
		array := make([]interface{}, 0)
 | 
			
		||||
		types := make([]tomlType, 0)
 | 
			
		||||
 | 
			
		||||
		for it = p.next(); it.typ != itemArrayEnd; it = p.next() {
 | 
			
		||||
			if it.typ == itemCommentStart {
 | 
			
		||||
				p.expect(itemText)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			val, typ := p.value(it)
 | 
			
		||||
			array = append(array, val)
 | 
			
		||||
			types = append(types, typ)
 | 
			
		||||
		}
 | 
			
		||||
		return array, p.typeOfArray(types)
 | 
			
		||||
	}
 | 
			
		||||
	p.bug("Unexpected value type: %s", it.typ)
 | 
			
		||||
	panic("unreachable")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// numUnderscoresOK checks whether each underscore in s is surrounded by
 | 
			
		||||
// characters that are not underscores.
 | 
			
		||||
func numUnderscoresOK(s string) bool {
 | 
			
		||||
	accept := false
 | 
			
		||||
	for _, r := range s {
 | 
			
		||||
		if r == '_' {
 | 
			
		||||
			if !accept {
 | 
			
		||||
				return false
 | 
			
		||||
			}
 | 
			
		||||
			accept = false
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		accept = true
 | 
			
		||||
	}
 | 
			
		||||
	return accept
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// numPeriodsOK checks whether every period in s is followed by a digit.
 | 
			
		||||
func numPeriodsOK(s string) bool {
 | 
			
		||||
	period := false
 | 
			
		||||
	for _, r := range s {
 | 
			
		||||
		if period && !isDigit(r) {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
		period = r == '.'
 | 
			
		||||
	}
 | 
			
		||||
	return !period
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// establishContext sets the current context of the parser,
 | 
			
		||||
// where the context is either a hash or an array of hashes. Which one is
 | 
			
		||||
// set depends on the value of the `array` parameter.
 | 
			
		||||
//
 | 
			
		||||
// Establishing the context also makes sure that the key isn't a duplicate, and
 | 
			
		||||
// will create implicit hashes automatically.
 | 
			
		||||
func (p *parser) establishContext(key Key, array bool) {
 | 
			
		||||
	var ok bool
 | 
			
		||||
 | 
			
		||||
	// Always start at the top level and drill down for our context.
 | 
			
		||||
	hashContext := p.mapping
 | 
			
		||||
	keyContext := make(Key, 0)
 | 
			
		||||
 | 
			
		||||
	// We only need implicit hashes for key[0:-1]
 | 
			
		||||
	for _, k := range key[0 : len(key)-1] {
 | 
			
		||||
		_, ok = hashContext[k]
 | 
			
		||||
		keyContext = append(keyContext, k)
 | 
			
		||||
 | 
			
		||||
		// No key? Make an implicit hash and move on.
 | 
			
		||||
		if !ok {
 | 
			
		||||
			p.addImplicit(keyContext)
 | 
			
		||||
			hashContext[k] = make(map[string]interface{})
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// If the hash context is actually an array of tables, then set
 | 
			
		||||
		// the hash context to the last element in that array.
 | 
			
		||||
		//
 | 
			
		||||
		// Otherwise, it better be a table, since this MUST be a key group (by
 | 
			
		||||
		// virtue of it not being the last element in a key).
 | 
			
		||||
		switch t := hashContext[k].(type) {
 | 
			
		||||
		case []map[string]interface{}:
 | 
			
		||||
			hashContext = t[len(t)-1]
 | 
			
		||||
		case map[string]interface{}:
 | 
			
		||||
			hashContext = t
 | 
			
		||||
		default:
 | 
			
		||||
			p.panicf("Key '%s' was already created as a hash.", keyContext)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	p.context = keyContext
 | 
			
		||||
	if array {
 | 
			
		||||
		// If this is the first element for this array, then allocate a new
 | 
			
		||||
		// list of tables for it.
 | 
			
		||||
		k := key[len(key)-1]
 | 
			
		||||
		if _, ok := hashContext[k]; !ok {
 | 
			
		||||
			hashContext[k] = make([]map[string]interface{}, 0, 5)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Add a new table. But make sure the key hasn't already been used
 | 
			
		||||
		// for something else.
 | 
			
		||||
		if hash, ok := hashContext[k].([]map[string]interface{}); ok {
 | 
			
		||||
			hashContext[k] = append(hash, make(map[string]interface{}))
 | 
			
		||||
		} else {
 | 
			
		||||
			p.panicf("Key '%s' was already created and cannot be used as "+
 | 
			
		||||
				"an array.", keyContext)
 | 
			
		||||
		}
 | 
			
		||||
	} else {
 | 
			
		||||
		p.setValue(key[len(key)-1], make(map[string]interface{}))
 | 
			
		||||
	}
 | 
			
		||||
	p.context = append(p.context, key[len(key)-1])
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// setValue sets the given key to the given value in the current context.
 | 
			
		||||
// It will make sure that the key hasn't already been defined, account for
 | 
			
		||||
// implicit key groups.
 | 
			
		||||
func (p *parser) setValue(key string, value interface{}) {
 | 
			
		||||
	var tmpHash interface{}
 | 
			
		||||
	var ok bool
 | 
			
		||||
 | 
			
		||||
	hash := p.mapping
 | 
			
		||||
	keyContext := make(Key, 0)
 | 
			
		||||
	for _, k := range p.context {
 | 
			
		||||
		keyContext = append(keyContext, k)
 | 
			
		||||
		if tmpHash, ok = hash[k]; !ok {
 | 
			
		||||
			p.bug("Context for key '%s' has not been established.", keyContext)
 | 
			
		||||
		}
 | 
			
		||||
		switch t := tmpHash.(type) {
 | 
			
		||||
		case []map[string]interface{}:
 | 
			
		||||
			// The context is a table of hashes. Pick the most recent table
 | 
			
		||||
			// defined as the current hash.
 | 
			
		||||
			hash = t[len(t)-1]
 | 
			
		||||
		case map[string]interface{}:
 | 
			
		||||
			hash = t
 | 
			
		||||
		default:
 | 
			
		||||
			p.bug("Expected hash to have type 'map[string]interface{}', but "+
 | 
			
		||||
				"it has '%T' instead.", tmpHash)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	keyContext = append(keyContext, key)
 | 
			
		||||
 | 
			
		||||
	if _, ok := hash[key]; ok {
 | 
			
		||||
		// Typically, if the given key has already been set, then we have
 | 
			
		||||
		// to raise an error since duplicate keys are disallowed. However,
 | 
			
		||||
		// it's possible that a key was previously defined implicitly. In this
 | 
			
		||||
		// case, it is allowed to be redefined concretely. (See the
 | 
			
		||||
		// `tests/valid/implicit-and-explicit-after.toml` test in `toml-test`.)
 | 
			
		||||
		//
 | 
			
		||||
		// But we have to make sure to stop marking it as an implicit. (So that
 | 
			
		||||
		// another redefinition provokes an error.)
 | 
			
		||||
		//
 | 
			
		||||
		// Note that since it has already been defined (as a hash), we don't
 | 
			
		||||
		// want to overwrite it. So our business is done.
 | 
			
		||||
		if p.isImplicit(keyContext) {
 | 
			
		||||
			p.removeImplicit(keyContext)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Otherwise, we have a concrete key trying to override a previous
 | 
			
		||||
		// key, which is *always* wrong.
 | 
			
		||||
		p.panicf("Key '%s' has already been defined.", keyContext)
 | 
			
		||||
	}
 | 
			
		||||
	hash[key] = value
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// setType sets the type of a particular value at a given key.
 | 
			
		||||
// It should be called immediately AFTER setValue.
 | 
			
		||||
//
 | 
			
		||||
// Note that if `key` is empty, then the type given will be applied to the
 | 
			
		||||
// current context (which is either a table or an array of tables).
 | 
			
		||||
func (p *parser) setType(key string, typ tomlType) {
 | 
			
		||||
	keyContext := make(Key, 0, len(p.context)+1)
 | 
			
		||||
	for _, k := range p.context {
 | 
			
		||||
		keyContext = append(keyContext, k)
 | 
			
		||||
	}
 | 
			
		||||
	if len(key) > 0 { // allow type setting for hashes
 | 
			
		||||
		keyContext = append(keyContext, key)
 | 
			
		||||
	}
 | 
			
		||||
	p.types[keyContext.String()] = typ
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// addImplicit sets the given Key as having been created implicitly.
 | 
			
		||||
func (p *parser) addImplicit(key Key) {
 | 
			
		||||
	p.implicits[key.String()] = true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// removeImplicit stops tagging the given key as having been implicitly
 | 
			
		||||
// created.
 | 
			
		||||
func (p *parser) removeImplicit(key Key) {
 | 
			
		||||
	p.implicits[key.String()] = false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// isImplicit returns true if the key group pointed to by the key was created
 | 
			
		||||
// implicitly.
 | 
			
		||||
func (p *parser) isImplicit(key Key) bool {
 | 
			
		||||
	return p.implicits[key.String()]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// current returns the full key name of the current context.
 | 
			
		||||
func (p *parser) current() string {
 | 
			
		||||
	if len(p.currentKey) == 0 {
 | 
			
		||||
		return p.context.String()
 | 
			
		||||
	}
 | 
			
		||||
	if len(p.context) == 0 {
 | 
			
		||||
		return p.currentKey
 | 
			
		||||
	}
 | 
			
		||||
	return fmt.Sprintf("%s.%s", p.context, p.currentKey)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func stripFirstNewline(s string) string {
 | 
			
		||||
	if len(s) == 0 || s[0] != '\n' {
 | 
			
		||||
		return s
 | 
			
		||||
	}
 | 
			
		||||
	return s[1:]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func stripEscapedWhitespace(s string) string {
 | 
			
		||||
	esc := strings.Split(s, "\\\n")
 | 
			
		||||
	if len(esc) > 1 {
 | 
			
		||||
		for i := 1; i < len(esc); i++ {
 | 
			
		||||
			esc[i] = strings.TrimLeftFunc(esc[i], unicode.IsSpace)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return strings.Join(esc, "")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) replaceEscapes(str string) string {
 | 
			
		||||
	var replaced []rune
 | 
			
		||||
	s := []byte(str)
 | 
			
		||||
	r := 0
 | 
			
		||||
	for r < len(s) {
 | 
			
		||||
		if s[r] != '\\' {
 | 
			
		||||
			c, size := utf8.DecodeRune(s[r:])
 | 
			
		||||
			r += size
 | 
			
		||||
			replaced = append(replaced, c)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		r += 1
 | 
			
		||||
		if r >= len(s) {
 | 
			
		||||
			p.bug("Escape sequence at end of string.")
 | 
			
		||||
			return ""
 | 
			
		||||
		}
 | 
			
		||||
		switch s[r] {
 | 
			
		||||
		default:
 | 
			
		||||
			p.bug("Expected valid escape code after \\, but got %q.", s[r])
 | 
			
		||||
			return ""
 | 
			
		||||
		case 'b':
 | 
			
		||||
			replaced = append(replaced, rune(0x0008))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case 't':
 | 
			
		||||
			replaced = append(replaced, rune(0x0009))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case 'n':
 | 
			
		||||
			replaced = append(replaced, rune(0x000A))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case 'f':
 | 
			
		||||
			replaced = append(replaced, rune(0x000C))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case 'r':
 | 
			
		||||
			replaced = append(replaced, rune(0x000D))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case '"':
 | 
			
		||||
			replaced = append(replaced, rune(0x0022))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case '\\':
 | 
			
		||||
			replaced = append(replaced, rune(0x005C))
 | 
			
		||||
			r += 1
 | 
			
		||||
		case 'u':
 | 
			
		||||
			// At this point, we know we have a Unicode escape of the form
 | 
			
		||||
			// `uXXXX` at [r, r+5). (Because the lexer guarantees this
 | 
			
		||||
			// for us.)
 | 
			
		||||
			escaped := p.asciiEscapeToUnicode(s[r+1 : r+5])
 | 
			
		||||
			replaced = append(replaced, escaped)
 | 
			
		||||
			r += 5
 | 
			
		||||
		case 'U':
 | 
			
		||||
			// At this point, we know we have a Unicode escape of the form
 | 
			
		||||
			// `uXXXX` at [r, r+9). (Because the lexer guarantees this
 | 
			
		||||
			// for us.)
 | 
			
		||||
			escaped := p.asciiEscapeToUnicode(s[r+1 : r+9])
 | 
			
		||||
			replaced = append(replaced, escaped)
 | 
			
		||||
			r += 9
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return string(replaced)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *parser) asciiEscapeToUnicode(bs []byte) rune {
 | 
			
		||||
	s := string(bs)
 | 
			
		||||
	hex, err := strconv.ParseUint(strings.ToLower(s), 16, 32)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		p.bug("Could not parse '%s' as a hexadecimal number, but the "+
 | 
			
		||||
			"lexer claims it's OK: %s", s, err)
 | 
			
		||||
	}
 | 
			
		||||
	if !utf8.ValidRune(rune(hex)) {
 | 
			
		||||
		p.panicf("Escaped character '\\u%s' is not valid UTF-8.", s)
 | 
			
		||||
	}
 | 
			
		||||
	return rune(hex)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isStringType(ty itemType) bool {
 | 
			
		||||
	return ty == itemString || ty == itemMultilineString ||
 | 
			
		||||
		ty == itemRawString || ty == itemRawMultilineString
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										91
									
								
								vendor/github.com/BurntSushi/toml/type_check.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										91
									
								
								vendor/github.com/BurntSushi/toml/type_check.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,91 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
// tomlType represents any Go type that corresponds to a TOML type.
 | 
			
		||||
// While the first draft of the TOML spec has a simplistic type system that
 | 
			
		||||
// probably doesn't need this level of sophistication, we seem to be militating
 | 
			
		||||
// toward adding real composite types.
 | 
			
		||||
type tomlType interface {
 | 
			
		||||
	typeString() string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// typeEqual accepts any two types and returns true if they are equal.
 | 
			
		||||
func typeEqual(t1, t2 tomlType) bool {
 | 
			
		||||
	if t1 == nil || t2 == nil {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	return t1.typeString() == t2.typeString()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func typeIsHash(t tomlType) bool {
 | 
			
		||||
	return typeEqual(t, tomlHash) || typeEqual(t, tomlArrayHash)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type tomlBaseType string
 | 
			
		||||
 | 
			
		||||
func (btype tomlBaseType) typeString() string {
 | 
			
		||||
	return string(btype)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (btype tomlBaseType) String() string {
 | 
			
		||||
	return btype.typeString()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	tomlInteger   tomlBaseType = "Integer"
 | 
			
		||||
	tomlFloat     tomlBaseType = "Float"
 | 
			
		||||
	tomlDatetime  tomlBaseType = "Datetime"
 | 
			
		||||
	tomlString    tomlBaseType = "String"
 | 
			
		||||
	tomlBool      tomlBaseType = "Bool"
 | 
			
		||||
	tomlArray     tomlBaseType = "Array"
 | 
			
		||||
	tomlHash      tomlBaseType = "Hash"
 | 
			
		||||
	tomlArrayHash tomlBaseType = "ArrayHash"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// typeOfPrimitive returns a tomlType of any primitive value in TOML.
 | 
			
		||||
// Primitive values are: Integer, Float, Datetime, String and Bool.
 | 
			
		||||
//
 | 
			
		||||
// Passing a lexer item other than the following will cause a BUG message
 | 
			
		||||
// to occur: itemString, itemBool, itemInteger, itemFloat, itemDatetime.
 | 
			
		||||
func (p *parser) typeOfPrimitive(lexItem item) tomlType {
 | 
			
		||||
	switch lexItem.typ {
 | 
			
		||||
	case itemInteger:
 | 
			
		||||
		return tomlInteger
 | 
			
		||||
	case itemFloat:
 | 
			
		||||
		return tomlFloat
 | 
			
		||||
	case itemDatetime:
 | 
			
		||||
		return tomlDatetime
 | 
			
		||||
	case itemString:
 | 
			
		||||
		return tomlString
 | 
			
		||||
	case itemMultilineString:
 | 
			
		||||
		return tomlString
 | 
			
		||||
	case itemRawString:
 | 
			
		||||
		return tomlString
 | 
			
		||||
	case itemRawMultilineString:
 | 
			
		||||
		return tomlString
 | 
			
		||||
	case itemBool:
 | 
			
		||||
		return tomlBool
 | 
			
		||||
	}
 | 
			
		||||
	p.bug("Cannot infer primitive type of lex item '%s'.", lexItem)
 | 
			
		||||
	panic("unreachable")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// typeOfArray returns a tomlType for an array given a list of types of its
 | 
			
		||||
// values.
 | 
			
		||||
//
 | 
			
		||||
// In the current spec, if an array is homogeneous, then its type is always
 | 
			
		||||
// "Array". If the array is not homogeneous, an error is generated.
 | 
			
		||||
func (p *parser) typeOfArray(types []tomlType) tomlType {
 | 
			
		||||
	// Empty arrays are cool.
 | 
			
		||||
	if len(types) == 0 {
 | 
			
		||||
		return tomlArray
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	theType := types[0]
 | 
			
		||||
	for _, t := range types[1:] {
 | 
			
		||||
		if !typeEqual(theType, t) {
 | 
			
		||||
			p.panicf("Array contains values of type '%s' and '%s', but "+
 | 
			
		||||
				"arrays must be homogeneous.", theType, t)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return tomlArray
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										242
									
								
								vendor/github.com/BurntSushi/toml/type_fields.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										242
									
								
								vendor/github.com/BurntSushi/toml/type_fields.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,242 @@
 | 
			
		||||
package toml
 | 
			
		||||
 | 
			
		||||
// Struct field handling is adapted from code in encoding/json:
 | 
			
		||||
//
 | 
			
		||||
// Copyright 2010 The Go Authors.  All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the Go distribution.
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// A field represents a single field found in a struct.
 | 
			
		||||
type field struct {
 | 
			
		||||
	name  string       // the name of the field (`toml` tag included)
 | 
			
		||||
	tag   bool         // whether field has a `toml` tag
 | 
			
		||||
	index []int        // represents the depth of an anonymous field
 | 
			
		||||
	typ   reflect.Type // the type of the field
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// byName sorts field by name, breaking ties with depth,
 | 
			
		||||
// then breaking ties with "name came from toml tag", then
 | 
			
		||||
// breaking ties with index sequence.
 | 
			
		||||
type byName []field
 | 
			
		||||
 | 
			
		||||
func (x byName) Len() int { return len(x) }
 | 
			
		||||
 | 
			
		||||
func (x byName) Swap(i, j int) { x[i], x[j] = x[j], x[i] }
 | 
			
		||||
 | 
			
		||||
func (x byName) Less(i, j int) bool {
 | 
			
		||||
	if x[i].name != x[j].name {
 | 
			
		||||
		return x[i].name < x[j].name
 | 
			
		||||
	}
 | 
			
		||||
	if len(x[i].index) != len(x[j].index) {
 | 
			
		||||
		return len(x[i].index) < len(x[j].index)
 | 
			
		||||
	}
 | 
			
		||||
	if x[i].tag != x[j].tag {
 | 
			
		||||
		return x[i].tag
 | 
			
		||||
	}
 | 
			
		||||
	return byIndex(x).Less(i, j)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// byIndex sorts field by index sequence.
 | 
			
		||||
type byIndex []field
 | 
			
		||||
 | 
			
		||||
func (x byIndex) Len() int { return len(x) }
 | 
			
		||||
 | 
			
		||||
func (x byIndex) Swap(i, j int) { x[i], x[j] = x[j], x[i] }
 | 
			
		||||
 | 
			
		||||
func (x byIndex) Less(i, j int) bool {
 | 
			
		||||
	for k, xik := range x[i].index {
 | 
			
		||||
		if k >= len(x[j].index) {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
		if xik != x[j].index[k] {
 | 
			
		||||
			return xik < x[j].index[k]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return len(x[i].index) < len(x[j].index)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// typeFields returns a list of fields that TOML should recognize for the given
 | 
			
		||||
// type. The algorithm is breadth-first search over the set of structs to
 | 
			
		||||
// include - the top struct and then any reachable anonymous structs.
 | 
			
		||||
func typeFields(t reflect.Type) []field {
 | 
			
		||||
	// Anonymous fields to explore at the current level and the next.
 | 
			
		||||
	current := []field{}
 | 
			
		||||
	next := []field{{typ: t}}
 | 
			
		||||
 | 
			
		||||
	// Count of queued names for current level and the next.
 | 
			
		||||
	count := map[reflect.Type]int{}
 | 
			
		||||
	nextCount := map[reflect.Type]int{}
 | 
			
		||||
 | 
			
		||||
	// Types already visited at an earlier level.
 | 
			
		||||
	visited := map[reflect.Type]bool{}
 | 
			
		||||
 | 
			
		||||
	// Fields found.
 | 
			
		||||
	var fields []field
 | 
			
		||||
 | 
			
		||||
	for len(next) > 0 {
 | 
			
		||||
		current, next = next, current[:0]
 | 
			
		||||
		count, nextCount = nextCount, map[reflect.Type]int{}
 | 
			
		||||
 | 
			
		||||
		for _, f := range current {
 | 
			
		||||
			if visited[f.typ] {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			visited[f.typ] = true
 | 
			
		||||
 | 
			
		||||
			// Scan f.typ for fields to include.
 | 
			
		||||
			for i := 0; i < f.typ.NumField(); i++ {
 | 
			
		||||
				sf := f.typ.Field(i)
 | 
			
		||||
				if sf.PkgPath != "" && !sf.Anonymous { // unexported
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				opts := getOptions(sf.Tag)
 | 
			
		||||
				if opts.skip {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				index := make([]int, len(f.index)+1)
 | 
			
		||||
				copy(index, f.index)
 | 
			
		||||
				index[len(f.index)] = i
 | 
			
		||||
 | 
			
		||||
				ft := sf.Type
 | 
			
		||||
				if ft.Name() == "" && ft.Kind() == reflect.Ptr {
 | 
			
		||||
					// Follow pointer.
 | 
			
		||||
					ft = ft.Elem()
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				// Record found field and index sequence.
 | 
			
		||||
				if opts.name != "" || !sf.Anonymous || ft.Kind() != reflect.Struct {
 | 
			
		||||
					tagged := opts.name != ""
 | 
			
		||||
					name := opts.name
 | 
			
		||||
					if name == "" {
 | 
			
		||||
						name = sf.Name
 | 
			
		||||
					}
 | 
			
		||||
					fields = append(fields, field{name, tagged, index, ft})
 | 
			
		||||
					if count[f.typ] > 1 {
 | 
			
		||||
						// If there were multiple instances, add a second,
 | 
			
		||||
						// so that the annihilation code will see a duplicate.
 | 
			
		||||
						// It only cares about the distinction between 1 or 2,
 | 
			
		||||
						// so don't bother generating any more copies.
 | 
			
		||||
						fields = append(fields, fields[len(fields)-1])
 | 
			
		||||
					}
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				// Record new anonymous struct to explore in next round.
 | 
			
		||||
				nextCount[ft]++
 | 
			
		||||
				if nextCount[ft] == 1 {
 | 
			
		||||
					f := field{name: ft.Name(), index: index, typ: ft}
 | 
			
		||||
					next = append(next, f)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sort.Sort(byName(fields))
 | 
			
		||||
 | 
			
		||||
	// Delete all fields that are hidden by the Go rules for embedded fields,
 | 
			
		||||
	// except that fields with TOML tags are promoted.
 | 
			
		||||
 | 
			
		||||
	// The fields are sorted in primary order of name, secondary order
 | 
			
		||||
	// of field index length. Loop over names; for each name, delete
 | 
			
		||||
	// hidden fields by choosing the one dominant field that survives.
 | 
			
		||||
	out := fields[:0]
 | 
			
		||||
	for advance, i := 0, 0; i < len(fields); i += advance {
 | 
			
		||||
		// One iteration per name.
 | 
			
		||||
		// Find the sequence of fields with the name of this first field.
 | 
			
		||||
		fi := fields[i]
 | 
			
		||||
		name := fi.name
 | 
			
		||||
		for advance = 1; i+advance < len(fields); advance++ {
 | 
			
		||||
			fj := fields[i+advance]
 | 
			
		||||
			if fj.name != name {
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if advance == 1 { // Only one field with this name
 | 
			
		||||
			out = append(out, fi)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		dominant, ok := dominantField(fields[i : i+advance])
 | 
			
		||||
		if ok {
 | 
			
		||||
			out = append(out, dominant)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fields = out
 | 
			
		||||
	sort.Sort(byIndex(fields))
 | 
			
		||||
 | 
			
		||||
	return fields
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// dominantField looks through the fields, all of which are known to
 | 
			
		||||
// have the same name, to find the single field that dominates the
 | 
			
		||||
// others using Go's embedding rules, modified by the presence of
 | 
			
		||||
// TOML tags. If there are multiple top-level fields, the boolean
 | 
			
		||||
// will be false: This condition is an error in Go and we skip all
 | 
			
		||||
// the fields.
 | 
			
		||||
func dominantField(fields []field) (field, bool) {
 | 
			
		||||
	// The fields are sorted in increasing index-length order. The winner
 | 
			
		||||
	// must therefore be one with the shortest index length. Drop all
 | 
			
		||||
	// longer entries, which is easy: just truncate the slice.
 | 
			
		||||
	length := len(fields[0].index)
 | 
			
		||||
	tagged := -1 // Index of first tagged field.
 | 
			
		||||
	for i, f := range fields {
 | 
			
		||||
		if len(f.index) > length {
 | 
			
		||||
			fields = fields[:i]
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		if f.tag {
 | 
			
		||||
			if tagged >= 0 {
 | 
			
		||||
				// Multiple tagged fields at the same level: conflict.
 | 
			
		||||
				// Return no field.
 | 
			
		||||
				return field{}, false
 | 
			
		||||
			}
 | 
			
		||||
			tagged = i
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if tagged >= 0 {
 | 
			
		||||
		return fields[tagged], true
 | 
			
		||||
	}
 | 
			
		||||
	// All remaining fields have the same length. If there's more than one,
 | 
			
		||||
	// we have a conflict (two fields named "X" at the same level) and we
 | 
			
		||||
	// return no field.
 | 
			
		||||
	if len(fields) > 1 {
 | 
			
		||||
		return field{}, false
 | 
			
		||||
	}
 | 
			
		||||
	return fields[0], true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var fieldCache struct {
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	m map[reflect.Type][]field
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// cachedTypeFields is like typeFields but uses a cache to avoid repeated work.
 | 
			
		||||
func cachedTypeFields(t reflect.Type) []field {
 | 
			
		||||
	fieldCache.RLock()
 | 
			
		||||
	f := fieldCache.m[t]
 | 
			
		||||
	fieldCache.RUnlock()
 | 
			
		||||
	if f != nil {
 | 
			
		||||
		return f
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Compute fields without lock.
 | 
			
		||||
	// Might duplicate effort but won't hold other computations back.
 | 
			
		||||
	f = typeFields(t)
 | 
			
		||||
	if f == nil {
 | 
			
		||||
		f = []field{}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fieldCache.Lock()
 | 
			
		||||
	if fieldCache.m == nil {
 | 
			
		||||
		fieldCache.m = map[reflect.Type][]field{}
 | 
			
		||||
	}
 | 
			
		||||
	fieldCache.m[t] = f
 | 
			
		||||
	fieldCache.Unlock()
 | 
			
		||||
	return f
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
The MIT License (MIT)
 | 
			
		||||
 | 
			
		||||
Copyright (c) 2014 Simon Eskildsen
 | 
			
		||||
 | 
			
		||||
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.
 | 
			
		||||
							
								
								
									
										26
									
								
								vendor/github.com/Sirupsen/logrus/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								vendor/github.com/Sirupsen/logrus/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,26 @@
 | 
			
		||||
/*
 | 
			
		||||
Package logrus is a structured logger for Go, completely API compatible with the standard library logger.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
The simplest way to use Logrus is simply the package-level exported logger:
 | 
			
		||||
 | 
			
		||||
  package main
 | 
			
		||||
 | 
			
		||||
  import (
 | 
			
		||||
    log "github.com/Sirupsen/logrus"
 | 
			
		||||
  )
 | 
			
		||||
 | 
			
		||||
  func main() {
 | 
			
		||||
    log.WithFields(log.Fields{
 | 
			
		||||
      "animal": "walrus",
 | 
			
		||||
      "number": 1,
 | 
			
		||||
      "size":   10,
 | 
			
		||||
    }).Info("A walrus appears")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
Output:
 | 
			
		||||
  time="2015-09-07T08:48:33Z" level=info msg="A walrus appears" animal=walrus number=1 size=10
 | 
			
		||||
 | 
			
		||||
For a full guide visit https://github.com/Sirupsen/logrus
 | 
			
		||||
*/
 | 
			
		||||
package logrus
 | 
			
		||||
							
								
								
									
										264
									
								
								vendor/github.com/Sirupsen/logrus/entry.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										264
									
								
								vendor/github.com/Sirupsen/logrus/entry.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,264 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Defines the key when adding errors using WithError.
 | 
			
		||||
var ErrorKey = "error"
 | 
			
		||||
 | 
			
		||||
// An entry is the final or intermediate Logrus logging entry. It contains all
 | 
			
		||||
// the fields passed with WithField{,s}. It's finally logged when Debug, Info,
 | 
			
		||||
// Warn, Error, Fatal or Panic is called on it. These objects can be reused and
 | 
			
		||||
// passed around as much as you wish to avoid field duplication.
 | 
			
		||||
type Entry struct {
 | 
			
		||||
	Logger *Logger
 | 
			
		||||
 | 
			
		||||
	// Contains all the fields set by the user.
 | 
			
		||||
	Data Fields
 | 
			
		||||
 | 
			
		||||
	// Time at which the log entry was created
 | 
			
		||||
	Time time.Time
 | 
			
		||||
 | 
			
		||||
	// Level the log entry was logged at: Debug, Info, Warn, Error, Fatal or Panic
 | 
			
		||||
	Level Level
 | 
			
		||||
 | 
			
		||||
	// Message passed to Debug, Info, Warn, Error, Fatal or Panic
 | 
			
		||||
	Message string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewEntry(logger *Logger) *Entry {
 | 
			
		||||
	return &Entry{
 | 
			
		||||
		Logger: logger,
 | 
			
		||||
		// Default is three fields, give a little extra room
 | 
			
		||||
		Data: make(Fields, 5),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Returns a reader for the entry, which is a proxy to the formatter.
 | 
			
		||||
func (entry *Entry) Reader() (*bytes.Buffer, error) {
 | 
			
		||||
	serialized, err := entry.Logger.Formatter.Format(entry)
 | 
			
		||||
	return bytes.NewBuffer(serialized), err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Returns the string representation from the reader and ultimately the
 | 
			
		||||
// formatter.
 | 
			
		||||
func (entry *Entry) String() (string, error) {
 | 
			
		||||
	reader, err := entry.Reader()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return reader.String(), err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Add an error as single field (using the key defined in ErrorKey) to the Entry.
 | 
			
		||||
func (entry *Entry) WithError(err error) *Entry {
 | 
			
		||||
	return entry.WithField(ErrorKey, err)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Add a single field to the Entry.
 | 
			
		||||
func (entry *Entry) WithField(key string, value interface{}) *Entry {
 | 
			
		||||
	return entry.WithFields(Fields{key: value})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Add a map of fields to the Entry.
 | 
			
		||||
func (entry *Entry) WithFields(fields Fields) *Entry {
 | 
			
		||||
	data := make(Fields, len(entry.Data)+len(fields))
 | 
			
		||||
	for k, v := range entry.Data {
 | 
			
		||||
		data[k] = v
 | 
			
		||||
	}
 | 
			
		||||
	for k, v := range fields {
 | 
			
		||||
		data[k] = v
 | 
			
		||||
	}
 | 
			
		||||
	return &Entry{Logger: entry.Logger, Data: data}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This function is not declared with a pointer value because otherwise
 | 
			
		||||
// race conditions will occur when using multiple goroutines
 | 
			
		||||
func (entry Entry) log(level Level, msg string) {
 | 
			
		||||
	entry.Time = time.Now()
 | 
			
		||||
	entry.Level = level
 | 
			
		||||
	entry.Message = msg
 | 
			
		||||
 | 
			
		||||
	if err := entry.Logger.Hooks.Fire(level, &entry); err != nil {
 | 
			
		||||
		entry.Logger.mu.Lock()
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "Failed to fire hook: %v\n", err)
 | 
			
		||||
		entry.Logger.mu.Unlock()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	reader, err := entry.Reader()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		entry.Logger.mu.Lock()
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "Failed to obtain reader, %v\n", err)
 | 
			
		||||
		entry.Logger.mu.Unlock()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	entry.Logger.mu.Lock()
 | 
			
		||||
	defer entry.Logger.mu.Unlock()
 | 
			
		||||
 | 
			
		||||
	_, err = io.Copy(entry.Logger.Out, reader)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "Failed to write to log, %v\n", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// To avoid Entry#log() returning a value that only would make sense for
 | 
			
		||||
	// panic() to use in Entry#Panic(), we avoid the allocation by checking
 | 
			
		||||
	// directly here.
 | 
			
		||||
	if level <= PanicLevel {
 | 
			
		||||
		panic(&entry)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Debug(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= DebugLevel {
 | 
			
		||||
		entry.log(DebugLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Print(args ...interface{}) {
 | 
			
		||||
	entry.Info(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Info(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= InfoLevel {
 | 
			
		||||
		entry.log(InfoLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warn(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= WarnLevel {
 | 
			
		||||
		entry.log(WarnLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warning(args ...interface{}) {
 | 
			
		||||
	entry.Warn(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Error(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= ErrorLevel {
 | 
			
		||||
		entry.log(ErrorLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Fatal(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= FatalLevel {
 | 
			
		||||
		entry.log(FatalLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Panic(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= PanicLevel {
 | 
			
		||||
		entry.log(PanicLevel, fmt.Sprint(args...))
 | 
			
		||||
	}
 | 
			
		||||
	panic(fmt.Sprint(args...))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Entry Printf family functions
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Debugf(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= DebugLevel {
 | 
			
		||||
		entry.Debug(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Infof(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= InfoLevel {
 | 
			
		||||
		entry.Info(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Printf(format string, args ...interface{}) {
 | 
			
		||||
	entry.Infof(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warnf(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= WarnLevel {
 | 
			
		||||
		entry.Warn(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warningf(format string, args ...interface{}) {
 | 
			
		||||
	entry.Warnf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Errorf(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= ErrorLevel {
 | 
			
		||||
		entry.Error(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Fatalf(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= FatalLevel {
 | 
			
		||||
		entry.Fatal(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Panicf(format string, args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= PanicLevel {
 | 
			
		||||
		entry.Panic(fmt.Sprintf(format, args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Entry Println family functions
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Debugln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= DebugLevel {
 | 
			
		||||
		entry.Debug(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Infoln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= InfoLevel {
 | 
			
		||||
		entry.Info(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Println(args ...interface{}) {
 | 
			
		||||
	entry.Infoln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warnln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= WarnLevel {
 | 
			
		||||
		entry.Warn(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Warningln(args ...interface{}) {
 | 
			
		||||
	entry.Warnln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Errorln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= ErrorLevel {
 | 
			
		||||
		entry.Error(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Fatalln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= FatalLevel {
 | 
			
		||||
		entry.Fatal(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (entry *Entry) Panicln(args ...interface{}) {
 | 
			
		||||
	if entry.Logger.Level >= PanicLevel {
 | 
			
		||||
		entry.Panic(entry.sprintlnn(args...))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Sprintlnn => Sprint no newline. This is to get the behavior of how
 | 
			
		||||
// fmt.Sprintln where spaces are always added between operands, regardless of
 | 
			
		||||
// their type. Instead of vendoring the Sprintln implementation to spare a
 | 
			
		||||
// string allocation, we do the simplest thing.
 | 
			
		||||
func (entry *Entry) sprintlnn(args ...interface{}) string {
 | 
			
		||||
	msg := fmt.Sprintln(args...)
 | 
			
		||||
	return msg[:len(msg)-1]
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										50
									
								
								vendor/github.com/Sirupsen/logrus/examples/basic/basic.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								vendor/github.com/Sirupsen/logrus/examples/basic/basic.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,50 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var log = logrus.New()
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.Formatter = new(logrus.JSONFormatter)
 | 
			
		||||
	log.Formatter = new(logrus.TextFormatter) // default
 | 
			
		||||
	log.Level = logrus.DebugLevel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	defer func() {
 | 
			
		||||
		err := recover()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.WithFields(logrus.Fields{
 | 
			
		||||
				"omg":    true,
 | 
			
		||||
				"err":    err,
 | 
			
		||||
				"number": 100,
 | 
			
		||||
			}).Fatal("The ice breaks!")
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"animal": "walrus",
 | 
			
		||||
		"number": 8,
 | 
			
		||||
	}).Debug("Started observing beach")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"animal": "walrus",
 | 
			
		||||
		"size":   10,
 | 
			
		||||
	}).Info("A group of walrus emerges from the ocean")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"omg":    true,
 | 
			
		||||
		"number": 122,
 | 
			
		||||
	}).Warn("The group's number increased tremendously!")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"temperature": -4,
 | 
			
		||||
	}).Debug("Temperature changes")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"animal": "orca",
 | 
			
		||||
		"size":   9009,
 | 
			
		||||
	}).Panic("It's over 9000!")
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										30
									
								
								vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,30 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/Sirupsen/logrus"
 | 
			
		||||
	"gopkg.in/gemnasium/logrus-airbrake-hook.v2"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var log = logrus.New()
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.Formatter = new(logrus.TextFormatter) // default
 | 
			
		||||
	log.Hooks.Add(airbrake.NewHook(123, "xyz", "development"))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"animal": "walrus",
 | 
			
		||||
		"size":   10,
 | 
			
		||||
	}).Info("A group of walrus emerges from the ocean")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"omg":    true,
 | 
			
		||||
		"number": 122,
 | 
			
		||||
	}).Warn("The group's number increased tremendously!")
 | 
			
		||||
 | 
			
		||||
	log.WithFields(logrus.Fields{
 | 
			
		||||
		"omg":    true,
 | 
			
		||||
		"number": 100,
 | 
			
		||||
	}).Fatal("The ice breaks!")
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										193
									
								
								vendor/github.com/Sirupsen/logrus/exported.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										193
									
								
								vendor/github.com/Sirupsen/logrus/exported.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,193 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"io"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	// std is the name of the standard logger in stdlib `log`
 | 
			
		||||
	std = New()
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func StandardLogger() *Logger {
 | 
			
		||||
	return std
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetOutput sets the standard logger output.
 | 
			
		||||
func SetOutput(out io.Writer) {
 | 
			
		||||
	std.mu.Lock()
 | 
			
		||||
	defer std.mu.Unlock()
 | 
			
		||||
	std.Out = out
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetFormatter sets the standard logger formatter.
 | 
			
		||||
func SetFormatter(formatter Formatter) {
 | 
			
		||||
	std.mu.Lock()
 | 
			
		||||
	defer std.mu.Unlock()
 | 
			
		||||
	std.Formatter = formatter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetLevel sets the standard logger level.
 | 
			
		||||
func SetLevel(level Level) {
 | 
			
		||||
	std.mu.Lock()
 | 
			
		||||
	defer std.mu.Unlock()
 | 
			
		||||
	std.Level = level
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetLevel returns the standard logger level.
 | 
			
		||||
func GetLevel() Level {
 | 
			
		||||
	std.mu.Lock()
 | 
			
		||||
	defer std.mu.Unlock()
 | 
			
		||||
	return std.Level
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AddHook adds a hook to the standard logger hooks.
 | 
			
		||||
func AddHook(hook Hook) {
 | 
			
		||||
	std.mu.Lock()
 | 
			
		||||
	defer std.mu.Unlock()
 | 
			
		||||
	std.Hooks.Add(hook)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WithError creates an entry from the standard logger and adds an error to it, using the value defined in ErrorKey as key.
 | 
			
		||||
func WithError(err error) *Entry {
 | 
			
		||||
	return std.WithField(ErrorKey, err)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WithField creates an entry from the standard logger and adds a field to
 | 
			
		||||
// it. If you want multiple fields, use `WithFields`.
 | 
			
		||||
//
 | 
			
		||||
// Note that it doesn't log until you call Debug, Print, Info, Warn, Fatal
 | 
			
		||||
// or Panic on the Entry it returns.
 | 
			
		||||
func WithField(key string, value interface{}) *Entry {
 | 
			
		||||
	return std.WithField(key, value)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WithFields creates an entry from the standard logger and adds multiple
 | 
			
		||||
// fields to it. This is simply a helper for `WithField`, invoking it
 | 
			
		||||
// once for each field.
 | 
			
		||||
//
 | 
			
		||||
// Note that it doesn't log until you call Debug, Print, Info, Warn, Fatal
 | 
			
		||||
// or Panic on the Entry it returns.
 | 
			
		||||
func WithFields(fields Fields) *Entry {
 | 
			
		||||
	return std.WithFields(fields)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Debug logs a message at level Debug on the standard logger.
 | 
			
		||||
func Debug(args ...interface{}) {
 | 
			
		||||
	std.Debug(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Print logs a message at level Info on the standard logger.
 | 
			
		||||
func Print(args ...interface{}) {
 | 
			
		||||
	std.Print(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Info logs a message at level Info on the standard logger.
 | 
			
		||||
func Info(args ...interface{}) {
 | 
			
		||||
	std.Info(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warn logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warn(args ...interface{}) {
 | 
			
		||||
	std.Warn(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warning logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warning(args ...interface{}) {
 | 
			
		||||
	std.Warning(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Error logs a message at level Error on the standard logger.
 | 
			
		||||
func Error(args ...interface{}) {
 | 
			
		||||
	std.Error(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Panic logs a message at level Panic on the standard logger.
 | 
			
		||||
func Panic(args ...interface{}) {
 | 
			
		||||
	std.Panic(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fatal logs a message at level Fatal on the standard logger.
 | 
			
		||||
func Fatal(args ...interface{}) {
 | 
			
		||||
	std.Fatal(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Debugf logs a message at level Debug on the standard logger.
 | 
			
		||||
func Debugf(format string, args ...interface{}) {
 | 
			
		||||
	std.Debugf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Printf logs a message at level Info on the standard logger.
 | 
			
		||||
func Printf(format string, args ...interface{}) {
 | 
			
		||||
	std.Printf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Infof logs a message at level Info on the standard logger.
 | 
			
		||||
func Infof(format string, args ...interface{}) {
 | 
			
		||||
	std.Infof(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warnf logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warnf(format string, args ...interface{}) {
 | 
			
		||||
	std.Warnf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warningf logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warningf(format string, args ...interface{}) {
 | 
			
		||||
	std.Warningf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Errorf logs a message at level Error on the standard logger.
 | 
			
		||||
func Errorf(format string, args ...interface{}) {
 | 
			
		||||
	std.Errorf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Panicf logs a message at level Panic on the standard logger.
 | 
			
		||||
func Panicf(format string, args ...interface{}) {
 | 
			
		||||
	std.Panicf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fatalf logs a message at level Fatal on the standard logger.
 | 
			
		||||
func Fatalf(format string, args ...interface{}) {
 | 
			
		||||
	std.Fatalf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Debugln logs a message at level Debug on the standard logger.
 | 
			
		||||
func Debugln(args ...interface{}) {
 | 
			
		||||
	std.Debugln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Println logs a message at level Info on the standard logger.
 | 
			
		||||
func Println(args ...interface{}) {
 | 
			
		||||
	std.Println(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Infoln logs a message at level Info on the standard logger.
 | 
			
		||||
func Infoln(args ...interface{}) {
 | 
			
		||||
	std.Infoln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warnln logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warnln(args ...interface{}) {
 | 
			
		||||
	std.Warnln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warningln logs a message at level Warn on the standard logger.
 | 
			
		||||
func Warningln(args ...interface{}) {
 | 
			
		||||
	std.Warningln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Errorln logs a message at level Error on the standard logger.
 | 
			
		||||
func Errorln(args ...interface{}) {
 | 
			
		||||
	std.Errorln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Panicln logs a message at level Panic on the standard logger.
 | 
			
		||||
func Panicln(args ...interface{}) {
 | 
			
		||||
	std.Panicln(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fatalln logs a message at level Fatal on the standard logger.
 | 
			
		||||
func Fatalln(args ...interface{}) {
 | 
			
		||||
	std.Fatalln(args...)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										48
									
								
								vendor/github.com/Sirupsen/logrus/formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										48
									
								
								vendor/github.com/Sirupsen/logrus/formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,48 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import "time"
 | 
			
		||||
 | 
			
		||||
const DefaultTimestampFormat = time.RFC3339
 | 
			
		||||
 | 
			
		||||
// The Formatter interface is used to implement a custom Formatter. It takes an
 | 
			
		||||
// `Entry`. It exposes all the fields, including the default ones:
 | 
			
		||||
//
 | 
			
		||||
// * `entry.Data["msg"]`. The message passed from Info, Warn, Error ..
 | 
			
		||||
// * `entry.Data["time"]`. The timestamp.
 | 
			
		||||
// * `entry.Data["level"]. The level the entry was logged at.
 | 
			
		||||
//
 | 
			
		||||
// Any additional fields added with `WithField` or `WithFields` are also in
 | 
			
		||||
// `entry.Data`. Format is expected to return an array of bytes which are then
 | 
			
		||||
// logged to `logger.Out`.
 | 
			
		||||
type Formatter interface {
 | 
			
		||||
	Format(*Entry) ([]byte, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This is to not silently overwrite `time`, `msg` and `level` fields when
 | 
			
		||||
// dumping it. If this code wasn't there doing:
 | 
			
		||||
//
 | 
			
		||||
//  logrus.WithField("level", 1).Info("hello")
 | 
			
		||||
//
 | 
			
		||||
// Would just silently drop the user provided level. Instead with this code
 | 
			
		||||
// it'll logged as:
 | 
			
		||||
//
 | 
			
		||||
//  {"level": "info", "fields.level": 1, "msg": "hello", "time": "..."}
 | 
			
		||||
//
 | 
			
		||||
// It's not exported because it's still using Data in an opinionated way. It's to
 | 
			
		||||
// avoid code duplication between the two default formatters.
 | 
			
		||||
func prefixFieldClashes(data Fields) {
 | 
			
		||||
	_, ok := data["time"]
 | 
			
		||||
	if ok {
 | 
			
		||||
		data["fields.time"] = data["time"]
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, ok = data["msg"]
 | 
			
		||||
	if ok {
 | 
			
		||||
		data["fields.msg"] = data["msg"]
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, ok = data["level"]
 | 
			
		||||
	if ok {
 | 
			
		||||
		data["fields.level"] = data["level"]
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										61
									
								
								vendor/github.com/Sirupsen/logrus/formatters/logstash/logstash.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										61
									
								
								vendor/github.com/Sirupsen/logrus/formatters/logstash/logstash.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,61 @@
 | 
			
		||||
package logstash
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Formatter generates json in logstash format.
 | 
			
		||||
// Logstash site: http://logstash.net/
 | 
			
		||||
type LogstashFormatter struct {
 | 
			
		||||
	Type string // if not empty use for logstash type field.
 | 
			
		||||
 | 
			
		||||
	// TimestampFormat sets the format used for timestamps.
 | 
			
		||||
	TimestampFormat string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *LogstashFormatter) Format(entry *logrus.Entry) ([]byte, error) {
 | 
			
		||||
	fields := make(logrus.Fields)
 | 
			
		||||
	for k, v := range entry.Data {
 | 
			
		||||
		fields[k] = v
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fields["@version"] = 1
 | 
			
		||||
 | 
			
		||||
	if f.TimestampFormat == "" {
 | 
			
		||||
		f.TimestampFormat = logrus.DefaultTimestampFormat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fields["@timestamp"] = entry.Time.Format(f.TimestampFormat)
 | 
			
		||||
 | 
			
		||||
	// set message field
 | 
			
		||||
	v, ok := entry.Data["message"]
 | 
			
		||||
	if ok {
 | 
			
		||||
		fields["fields.message"] = v
 | 
			
		||||
	}
 | 
			
		||||
	fields["message"] = entry.Message
 | 
			
		||||
 | 
			
		||||
	// set level field
 | 
			
		||||
	v, ok = entry.Data["level"]
 | 
			
		||||
	if ok {
 | 
			
		||||
		fields["fields.level"] = v
 | 
			
		||||
	}
 | 
			
		||||
	fields["level"] = entry.Level.String()
 | 
			
		||||
 | 
			
		||||
	// set type field
 | 
			
		||||
	if f.Type != "" {
 | 
			
		||||
		v, ok = entry.Data["type"]
 | 
			
		||||
		if ok {
 | 
			
		||||
			fields["fields.type"] = v
 | 
			
		||||
		}
 | 
			
		||||
		fields["type"] = f.Type
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	serialized, err := json.Marshal(fields)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, fmt.Errorf("Failed to marshal fields to JSON, %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	return append(serialized, '\n'), nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										34
									
								
								vendor/github.com/Sirupsen/logrus/hooks.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								vendor/github.com/Sirupsen/logrus/hooks.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,34 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
// A hook to be fired when logging on the logging levels returned from
 | 
			
		||||
// `Levels()` on your implementation of the interface. Note that this is not
 | 
			
		||||
// fired in a goroutine or a channel with workers, you should handle such
 | 
			
		||||
// functionality yourself if your call is non-blocking and you don't wish for
 | 
			
		||||
// the logging calls for levels returned from `Levels()` to block.
 | 
			
		||||
type Hook interface {
 | 
			
		||||
	Levels() []Level
 | 
			
		||||
	Fire(*Entry) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Internal type for storing the hooks on a logger instance.
 | 
			
		||||
type LevelHooks map[Level][]Hook
 | 
			
		||||
 | 
			
		||||
// Add a hook to an instance of logger. This is called with
 | 
			
		||||
// `log.Hooks.Add(new(MyHook))` where `MyHook` implements the `Hook` interface.
 | 
			
		||||
func (hooks LevelHooks) Add(hook Hook) {
 | 
			
		||||
	for _, level := range hook.Levels() {
 | 
			
		||||
		hooks[level] = append(hooks[level], hook)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fire all the hooks for the passed level. Used by `entry.log` to fire
 | 
			
		||||
// appropriate hooks for a log entry.
 | 
			
		||||
func (hooks LevelHooks) Fire(level Level, entry *Entry) error {
 | 
			
		||||
	for _, hook := range hooks[level] {
 | 
			
		||||
		if err := hook.Fire(entry); err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										54
									
								
								vendor/github.com/Sirupsen/logrus/hooks/syslog/syslog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								vendor/github.com/Sirupsen/logrus/hooks/syslog/syslog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,54 @@
 | 
			
		||||
// +build !windows,!nacl,!plan9
 | 
			
		||||
 | 
			
		||||
package logrus_syslog
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/Sirupsen/logrus"
 | 
			
		||||
	"log/syslog"
 | 
			
		||||
	"os"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// SyslogHook to send logs via syslog.
 | 
			
		||||
type SyslogHook struct {
 | 
			
		||||
	Writer        *syslog.Writer
 | 
			
		||||
	SyslogNetwork string
 | 
			
		||||
	SyslogRaddr   string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Creates a hook to be added to an instance of logger. This is called with
 | 
			
		||||
// `hook, err := NewSyslogHook("udp", "localhost:514", syslog.LOG_DEBUG, "")`
 | 
			
		||||
// `if err == nil { log.Hooks.Add(hook) }`
 | 
			
		||||
func NewSyslogHook(network, raddr string, priority syslog.Priority, tag string) (*SyslogHook, error) {
 | 
			
		||||
	w, err := syslog.Dial(network, raddr, priority, tag)
 | 
			
		||||
	return &SyslogHook{w, network, raddr}, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (hook *SyslogHook) Fire(entry *logrus.Entry) error {
 | 
			
		||||
	line, err := entry.String()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "Unable to read entry, %v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	switch entry.Level {
 | 
			
		||||
	case logrus.PanicLevel:
 | 
			
		||||
		return hook.Writer.Crit(line)
 | 
			
		||||
	case logrus.FatalLevel:
 | 
			
		||||
		return hook.Writer.Crit(line)
 | 
			
		||||
	case logrus.ErrorLevel:
 | 
			
		||||
		return hook.Writer.Err(line)
 | 
			
		||||
	case logrus.WarnLevel:
 | 
			
		||||
		return hook.Writer.Warning(line)
 | 
			
		||||
	case logrus.InfoLevel:
 | 
			
		||||
		return hook.Writer.Info(line)
 | 
			
		||||
	case logrus.DebugLevel:
 | 
			
		||||
		return hook.Writer.Debug(line)
 | 
			
		||||
	default:
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (hook *SyslogHook) Levels() []logrus.Level {
 | 
			
		||||
	return logrus.AllLevels
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										67
									
								
								vendor/github.com/Sirupsen/logrus/hooks/test/test.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										67
									
								
								vendor/github.com/Sirupsen/logrus/hooks/test/test.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,67 @@
 | 
			
		||||
package test
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
 | 
			
		||||
	"github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// test.Hook is a hook designed for dealing with logs in test scenarios.
 | 
			
		||||
type Hook struct {
 | 
			
		||||
	Entries []*logrus.Entry
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Installs a test hook for the global logger.
 | 
			
		||||
func NewGlobal() *Hook {
 | 
			
		||||
 | 
			
		||||
	hook := new(Hook)
 | 
			
		||||
	logrus.AddHook(hook)
 | 
			
		||||
 | 
			
		||||
	return hook
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Installs a test hook for a given local logger.
 | 
			
		||||
func NewLocal(logger *logrus.Logger) *Hook {
 | 
			
		||||
 | 
			
		||||
	hook := new(Hook)
 | 
			
		||||
	logger.Hooks.Add(hook)
 | 
			
		||||
 | 
			
		||||
	return hook
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Creates a discarding logger and installs the test hook.
 | 
			
		||||
func NewNullLogger() (*logrus.Logger, *Hook) {
 | 
			
		||||
 | 
			
		||||
	logger := logrus.New()
 | 
			
		||||
	logger.Out = ioutil.Discard
 | 
			
		||||
 | 
			
		||||
	return logger, NewLocal(logger)
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *Hook) Fire(e *logrus.Entry) error {
 | 
			
		||||
	t.Entries = append(t.Entries, e)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *Hook) Levels() []logrus.Level {
 | 
			
		||||
	return logrus.AllLevels
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// LastEntry returns the last entry that was logged or nil.
 | 
			
		||||
func (t *Hook) LastEntry() (l *logrus.Entry) {
 | 
			
		||||
 | 
			
		||||
	if i := len(t.Entries) - 1; i < 0 {
 | 
			
		||||
		return nil
 | 
			
		||||
	} else {
 | 
			
		||||
		return t.Entries[i]
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Reset removes all Entries from this test hook.
 | 
			
		||||
func (t *Hook) Reset() {
 | 
			
		||||
	t.Entries = make([]*logrus.Entry, 0)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										41
									
								
								vendor/github.com/Sirupsen/logrus/json_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								vendor/github.com/Sirupsen/logrus/json_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,41 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type JSONFormatter struct {
 | 
			
		||||
	// TimestampFormat sets the format used for marshaling timestamps.
 | 
			
		||||
	TimestampFormat string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *JSONFormatter) Format(entry *Entry) ([]byte, error) {
 | 
			
		||||
	data := make(Fields, len(entry.Data)+3)
 | 
			
		||||
	for k, v := range entry.Data {
 | 
			
		||||
		switch v := v.(type) {
 | 
			
		||||
		case error:
 | 
			
		||||
			// Otherwise errors are ignored by `encoding/json`
 | 
			
		||||
			// https://github.com/Sirupsen/logrus/issues/137
 | 
			
		||||
			data[k] = v.Error()
 | 
			
		||||
		default:
 | 
			
		||||
			data[k] = v
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	prefixFieldClashes(data)
 | 
			
		||||
 | 
			
		||||
	timestampFormat := f.TimestampFormat
 | 
			
		||||
	if timestampFormat == "" {
 | 
			
		||||
		timestampFormat = DefaultTimestampFormat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	data["time"] = entry.Time.Format(timestampFormat)
 | 
			
		||||
	data["msg"] = entry.Message
 | 
			
		||||
	data["level"] = entry.Level.String()
 | 
			
		||||
 | 
			
		||||
	serialized, err := json.Marshal(data)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, fmt.Errorf("Failed to marshal fields to JSON, %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	return append(serialized, '\n'), nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										212
									
								
								vendor/github.com/Sirupsen/logrus/logger.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										212
									
								
								vendor/github.com/Sirupsen/logrus/logger.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,212 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Logger struct {
 | 
			
		||||
	// The logs are `io.Copy`'d to this in a mutex. It's common to set this to a
 | 
			
		||||
	// file, or leave it default which is `os.Stderr`. You can also set this to
 | 
			
		||||
	// something more adventorous, such as logging to Kafka.
 | 
			
		||||
	Out io.Writer
 | 
			
		||||
	// Hooks for the logger instance. These allow firing events based on logging
 | 
			
		||||
	// levels and log entries. For example, to send errors to an error tracking
 | 
			
		||||
	// service, log to StatsD or dump the core on fatal errors.
 | 
			
		||||
	Hooks LevelHooks
 | 
			
		||||
	// All log entries pass through the formatter before logged to Out. The
 | 
			
		||||
	// included formatters are `TextFormatter` and `JSONFormatter` for which
 | 
			
		||||
	// TextFormatter is the default. In development (when a TTY is attached) it
 | 
			
		||||
	// logs with colors, but to a file it wouldn't. You can easily implement your
 | 
			
		||||
	// own that implements the `Formatter` interface, see the `README` or included
 | 
			
		||||
	// formatters for examples.
 | 
			
		||||
	Formatter Formatter
 | 
			
		||||
	// The logging level the logger should log at. This is typically (and defaults
 | 
			
		||||
	// to) `logrus.Info`, which allows Info(), Warn(), Error() and Fatal() to be
 | 
			
		||||
	// logged. `logrus.Debug` is useful in
 | 
			
		||||
	Level Level
 | 
			
		||||
	// Used to sync writing to the log.
 | 
			
		||||
	mu sync.Mutex
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Creates a new logger. Configuration should be set by changing `Formatter`,
 | 
			
		||||
// `Out` and `Hooks` directly on the default logger instance. You can also just
 | 
			
		||||
// instantiate your own:
 | 
			
		||||
//
 | 
			
		||||
//    var log = &Logger{
 | 
			
		||||
//      Out: os.Stderr,
 | 
			
		||||
//      Formatter: new(JSONFormatter),
 | 
			
		||||
//      Hooks: make(LevelHooks),
 | 
			
		||||
//      Level: logrus.DebugLevel,
 | 
			
		||||
//    }
 | 
			
		||||
//
 | 
			
		||||
// It's recommended to make this a global instance called `log`.
 | 
			
		||||
func New() *Logger {
 | 
			
		||||
	return &Logger{
 | 
			
		||||
		Out:       os.Stderr,
 | 
			
		||||
		Formatter: new(TextFormatter),
 | 
			
		||||
		Hooks:     make(LevelHooks),
 | 
			
		||||
		Level:     InfoLevel,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Adds a field to the log entry, note that you it doesn't log until you call
 | 
			
		||||
// Debug, Print, Info, Warn, Fatal or Panic. It only creates a log entry.
 | 
			
		||||
// If you want multiple fields, use `WithFields`.
 | 
			
		||||
func (logger *Logger) WithField(key string, value interface{}) *Entry {
 | 
			
		||||
	return NewEntry(logger).WithField(key, value)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Adds a struct of fields to the log entry. All it does is call `WithField` for
 | 
			
		||||
// each `Field`.
 | 
			
		||||
func (logger *Logger) WithFields(fields Fields) *Entry {
 | 
			
		||||
	return NewEntry(logger).WithFields(fields)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Add an error as single field to the log entry.  All it does is call
 | 
			
		||||
// `WithError` for the given `error`.
 | 
			
		||||
func (logger *Logger) WithError(err error) *Entry {
 | 
			
		||||
	return NewEntry(logger).WithError(err)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Debugf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= DebugLevel {
 | 
			
		||||
		NewEntry(logger).Debugf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Infof(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= InfoLevel {
 | 
			
		||||
		NewEntry(logger).Infof(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Printf(format string, args ...interface{}) {
 | 
			
		||||
	NewEntry(logger).Printf(format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warnf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warnf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warningf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warnf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Errorf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= ErrorLevel {
 | 
			
		||||
		NewEntry(logger).Errorf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Fatalf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= FatalLevel {
 | 
			
		||||
		NewEntry(logger).Fatalf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Panicf(format string, args ...interface{}) {
 | 
			
		||||
	if logger.Level >= PanicLevel {
 | 
			
		||||
		NewEntry(logger).Panicf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Debug(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= DebugLevel {
 | 
			
		||||
		NewEntry(logger).Debug(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Info(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= InfoLevel {
 | 
			
		||||
		NewEntry(logger).Info(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Print(args ...interface{}) {
 | 
			
		||||
	NewEntry(logger).Info(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warn(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warn(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warning(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warn(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Error(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= ErrorLevel {
 | 
			
		||||
		NewEntry(logger).Error(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Fatal(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= FatalLevel {
 | 
			
		||||
		NewEntry(logger).Fatal(args...)
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Panic(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= PanicLevel {
 | 
			
		||||
		NewEntry(logger).Panic(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Debugln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= DebugLevel {
 | 
			
		||||
		NewEntry(logger).Debugln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Infoln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= InfoLevel {
 | 
			
		||||
		NewEntry(logger).Infoln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Println(args ...interface{}) {
 | 
			
		||||
	NewEntry(logger).Println(args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warnln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warnln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Warningln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= WarnLevel {
 | 
			
		||||
		NewEntry(logger).Warnln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Errorln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= ErrorLevel {
 | 
			
		||||
		NewEntry(logger).Errorln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Fatalln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= FatalLevel {
 | 
			
		||||
		NewEntry(logger).Fatalln(args...)
 | 
			
		||||
	}
 | 
			
		||||
	os.Exit(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Panicln(args ...interface{}) {
 | 
			
		||||
	if logger.Level >= PanicLevel {
 | 
			
		||||
		NewEntry(logger).Panicln(args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										143
									
								
								vendor/github.com/Sirupsen/logrus/logrus.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										143
									
								
								vendor/github.com/Sirupsen/logrus/logrus.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,143 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"log"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Fields type, used to pass to `WithFields`.
 | 
			
		||||
type Fields map[string]interface{}
 | 
			
		||||
 | 
			
		||||
// Level type
 | 
			
		||||
type Level uint8
 | 
			
		||||
 | 
			
		||||
// Convert the Level to a string. E.g. PanicLevel becomes "panic".
 | 
			
		||||
func (level Level) String() string {
 | 
			
		||||
	switch level {
 | 
			
		||||
	case DebugLevel:
 | 
			
		||||
		return "debug"
 | 
			
		||||
	case InfoLevel:
 | 
			
		||||
		return "info"
 | 
			
		||||
	case WarnLevel:
 | 
			
		||||
		return "warning"
 | 
			
		||||
	case ErrorLevel:
 | 
			
		||||
		return "error"
 | 
			
		||||
	case FatalLevel:
 | 
			
		||||
		return "fatal"
 | 
			
		||||
	case PanicLevel:
 | 
			
		||||
		return "panic"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return "unknown"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ParseLevel takes a string level and returns the Logrus log level constant.
 | 
			
		||||
func ParseLevel(lvl string) (Level, error) {
 | 
			
		||||
	switch strings.ToLower(lvl) {
 | 
			
		||||
	case "panic":
 | 
			
		||||
		return PanicLevel, nil
 | 
			
		||||
	case "fatal":
 | 
			
		||||
		return FatalLevel, nil
 | 
			
		||||
	case "error":
 | 
			
		||||
		return ErrorLevel, nil
 | 
			
		||||
	case "warn", "warning":
 | 
			
		||||
		return WarnLevel, nil
 | 
			
		||||
	case "info":
 | 
			
		||||
		return InfoLevel, nil
 | 
			
		||||
	case "debug":
 | 
			
		||||
		return DebugLevel, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var l Level
 | 
			
		||||
	return l, fmt.Errorf("not a valid logrus Level: %q", lvl)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// A constant exposing all logging levels
 | 
			
		||||
var AllLevels = []Level{
 | 
			
		||||
	PanicLevel,
 | 
			
		||||
	FatalLevel,
 | 
			
		||||
	ErrorLevel,
 | 
			
		||||
	WarnLevel,
 | 
			
		||||
	InfoLevel,
 | 
			
		||||
	DebugLevel,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// These are the different logging levels. You can set the logging level to log
 | 
			
		||||
// on your instance of logger, obtained with `logrus.New()`.
 | 
			
		||||
const (
 | 
			
		||||
	// PanicLevel level, highest level of severity. Logs and then calls panic with the
 | 
			
		||||
	// message passed to Debug, Info, ...
 | 
			
		||||
	PanicLevel Level = iota
 | 
			
		||||
	// FatalLevel level. Logs and then calls `os.Exit(1)`. It will exit even if the
 | 
			
		||||
	// logging level is set to Panic.
 | 
			
		||||
	FatalLevel
 | 
			
		||||
	// ErrorLevel level. Logs. Used for errors that should definitely be noted.
 | 
			
		||||
	// Commonly used for hooks to send errors to an error tracking service.
 | 
			
		||||
	ErrorLevel
 | 
			
		||||
	// WarnLevel level. Non-critical entries that deserve eyes.
 | 
			
		||||
	WarnLevel
 | 
			
		||||
	// InfoLevel level. General operational entries about what's going on inside the
 | 
			
		||||
	// application.
 | 
			
		||||
	InfoLevel
 | 
			
		||||
	// DebugLevel level. Usually only enabled when debugging. Very verbose logging.
 | 
			
		||||
	DebugLevel
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Won't compile if StdLogger can't be realized by a log.Logger
 | 
			
		||||
var (
 | 
			
		||||
	_ StdLogger = &log.Logger{}
 | 
			
		||||
	_ StdLogger = &Entry{}
 | 
			
		||||
	_ StdLogger = &Logger{}
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// StdLogger is what your logrus-enabled library should take, that way
 | 
			
		||||
// it'll accept a stdlib logger and a logrus logger. There's no standard
 | 
			
		||||
// interface, this is the closest we get, unfortunately.
 | 
			
		||||
type StdLogger interface {
 | 
			
		||||
	Print(...interface{})
 | 
			
		||||
	Printf(string, ...interface{})
 | 
			
		||||
	Println(...interface{})
 | 
			
		||||
 | 
			
		||||
	Fatal(...interface{})
 | 
			
		||||
	Fatalf(string, ...interface{})
 | 
			
		||||
	Fatalln(...interface{})
 | 
			
		||||
 | 
			
		||||
	Panic(...interface{})
 | 
			
		||||
	Panicf(string, ...interface{})
 | 
			
		||||
	Panicln(...interface{})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// The FieldLogger interface generalizes the Entry and Logger types
 | 
			
		||||
type FieldLogger interface {
 | 
			
		||||
	WithField(key string, value interface{}) *Entry
 | 
			
		||||
	WithFields(fields Fields) *Entry
 | 
			
		||||
	WithError(err error) *Entry
 | 
			
		||||
 | 
			
		||||
	Debugf(format string, args ...interface{})
 | 
			
		||||
	Infof(format string, args ...interface{})
 | 
			
		||||
	Printf(format string, args ...interface{})
 | 
			
		||||
	Warnf(format string, args ...interface{})
 | 
			
		||||
	Warningf(format string, args ...interface{})
 | 
			
		||||
	Errorf(format string, args ...interface{})
 | 
			
		||||
	Fatalf(format string, args ...interface{})
 | 
			
		||||
	Panicf(format string, args ...interface{})
 | 
			
		||||
 | 
			
		||||
	Debug(args ...interface{})
 | 
			
		||||
	Info(args ...interface{})
 | 
			
		||||
	Print(args ...interface{})
 | 
			
		||||
	Warn(args ...interface{})
 | 
			
		||||
	Warning(args ...interface{})
 | 
			
		||||
	Error(args ...interface{})
 | 
			
		||||
	Fatal(args ...interface{})
 | 
			
		||||
	Panic(args ...interface{})
 | 
			
		||||
 | 
			
		||||
	Debugln(args ...interface{})
 | 
			
		||||
	Infoln(args ...interface{})
 | 
			
		||||
	Println(args ...interface{})
 | 
			
		||||
	Warnln(args ...interface{})
 | 
			
		||||
	Warningln(args ...interface{})
 | 
			
		||||
	Errorln(args ...interface{})
 | 
			
		||||
	Fatalln(args ...interface{})
 | 
			
		||||
	Panicln(args ...interface{})
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										9
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
			
		||||
// +build darwin freebsd openbsd netbsd dragonfly
 | 
			
		||||
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import "syscall"
 | 
			
		||||
 | 
			
		||||
const ioctlReadTermios = syscall.TIOCGETA
 | 
			
		||||
 | 
			
		||||
type Termios syscall.Termios
 | 
			
		||||
							
								
								
									
										12
									
								
								vendor/github.com/Sirupsen/logrus/terminal_linux.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								vendor/github.com/Sirupsen/logrus/terminal_linux.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
// Based on ssh/terminal:
 | 
			
		||||
// Copyright 2013 The Go Authors. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import "syscall"
 | 
			
		||||
 | 
			
		||||
const ioctlReadTermios = syscall.TCGETS
 | 
			
		||||
 | 
			
		||||
type Termios syscall.Termios
 | 
			
		||||
							
								
								
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
// Based on ssh/terminal:
 | 
			
		||||
// Copyright 2011 The Go Authors. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// +build linux darwin freebsd openbsd netbsd dragonfly
 | 
			
		||||
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"syscall"
 | 
			
		||||
	"unsafe"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// IsTerminal returns true if stderr's file descriptor is a terminal.
 | 
			
		||||
func IsTerminal() bool {
 | 
			
		||||
	fd := syscall.Stderr
 | 
			
		||||
	var termios Termios
 | 
			
		||||
	_, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(fd), ioctlReadTermios, uintptr(unsafe.Pointer(&termios)), 0, 0, 0)
 | 
			
		||||
	return err == 0
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										15
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
// +build solaris
 | 
			
		||||
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"os"
 | 
			
		||||
 | 
			
		||||
	"golang.org/x/sys/unix"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// IsTerminal returns true if the given file descriptor is a terminal.
 | 
			
		||||
func IsTerminal() bool {
 | 
			
		||||
	_, err := unix.IoctlGetTermios(int(os.Stdout.Fd()), unix.TCGETA)
 | 
			
		||||
	return err == nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										27
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,27 @@
 | 
			
		||||
// Based on ssh/terminal:
 | 
			
		||||
// Copyright 2011 The Go Authors. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// +build windows
 | 
			
		||||
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"syscall"
 | 
			
		||||
	"unsafe"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var kernel32 = syscall.NewLazyDLL("kernel32.dll")
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	procGetConsoleMode = kernel32.NewProc("GetConsoleMode")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// IsTerminal returns true if stderr's file descriptor is a terminal.
 | 
			
		||||
func IsTerminal() bool {
 | 
			
		||||
	fd := syscall.Stderr
 | 
			
		||||
	var st uint32
 | 
			
		||||
	r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(fd), uintptr(unsafe.Pointer(&st)), 0)
 | 
			
		||||
	return r != 0 && e == 0
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										161
									
								
								vendor/github.com/Sirupsen/logrus/text_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										161
									
								
								vendor/github.com/Sirupsen/logrus/text_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,161 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"runtime"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	nocolor = 0
 | 
			
		||||
	red     = 31
 | 
			
		||||
	green   = 32
 | 
			
		||||
	yellow  = 33
 | 
			
		||||
	blue    = 34
 | 
			
		||||
	gray    = 37
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	baseTimestamp time.Time
 | 
			
		||||
	isTerminal    bool
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	baseTimestamp = time.Now()
 | 
			
		||||
	isTerminal = IsTerminal()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func miniTS() int {
 | 
			
		||||
	return int(time.Since(baseTimestamp) / time.Second)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type TextFormatter struct {
 | 
			
		||||
	// Set to true to bypass checking for a TTY before outputting colors.
 | 
			
		||||
	ForceColors bool
 | 
			
		||||
 | 
			
		||||
	// Force disabling colors.
 | 
			
		||||
	DisableColors bool
 | 
			
		||||
 | 
			
		||||
	// Disable timestamp logging. useful when output is redirected to logging
 | 
			
		||||
	// system that already adds timestamps.
 | 
			
		||||
	DisableTimestamp bool
 | 
			
		||||
 | 
			
		||||
	// Enable logging the full timestamp when a TTY is attached instead of just
 | 
			
		||||
	// the time passed since beginning of execution.
 | 
			
		||||
	FullTimestamp bool
 | 
			
		||||
 | 
			
		||||
	// TimestampFormat to use for display when a full timestamp is printed
 | 
			
		||||
	TimestampFormat string
 | 
			
		||||
 | 
			
		||||
	// The fields are sorted by default for a consistent output. For applications
 | 
			
		||||
	// that log extremely frequently and don't use the JSON formatter this may not
 | 
			
		||||
	// be desired.
 | 
			
		||||
	DisableSorting bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *TextFormatter) Format(entry *Entry) ([]byte, error) {
 | 
			
		||||
	var keys []string = make([]string, 0, len(entry.Data))
 | 
			
		||||
	for k := range entry.Data {
 | 
			
		||||
		keys = append(keys, k)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if !f.DisableSorting {
 | 
			
		||||
		sort.Strings(keys)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b := &bytes.Buffer{}
 | 
			
		||||
 | 
			
		||||
	prefixFieldClashes(entry.Data)
 | 
			
		||||
 | 
			
		||||
	isColorTerminal := isTerminal && (runtime.GOOS != "windows")
 | 
			
		||||
	isColored := (f.ForceColors || isColorTerminal) && !f.DisableColors
 | 
			
		||||
 | 
			
		||||
	timestampFormat := f.TimestampFormat
 | 
			
		||||
	if timestampFormat == "" {
 | 
			
		||||
		timestampFormat = DefaultTimestampFormat
 | 
			
		||||
	}
 | 
			
		||||
	if isColored {
 | 
			
		||||
		f.printColored(b, entry, keys, timestampFormat)
 | 
			
		||||
	} else {
 | 
			
		||||
		if !f.DisableTimestamp {
 | 
			
		||||
			f.appendKeyValue(b, "time", entry.Time.Format(timestampFormat))
 | 
			
		||||
		}
 | 
			
		||||
		f.appendKeyValue(b, "level", entry.Level.String())
 | 
			
		||||
		if entry.Message != "" {
 | 
			
		||||
			f.appendKeyValue(b, "msg", entry.Message)
 | 
			
		||||
		}
 | 
			
		||||
		for _, key := range keys {
 | 
			
		||||
			f.appendKeyValue(b, key, entry.Data[key])
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.WriteByte('\n')
 | 
			
		||||
	return b.Bytes(), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *TextFormatter) printColored(b *bytes.Buffer, entry *Entry, keys []string, timestampFormat string) {
 | 
			
		||||
	var levelColor int
 | 
			
		||||
	switch entry.Level {
 | 
			
		||||
	case DebugLevel:
 | 
			
		||||
		levelColor = gray
 | 
			
		||||
	case WarnLevel:
 | 
			
		||||
		levelColor = yellow
 | 
			
		||||
	case ErrorLevel, FatalLevel, PanicLevel:
 | 
			
		||||
		levelColor = red
 | 
			
		||||
	default:
 | 
			
		||||
		levelColor = blue
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	levelText := strings.ToUpper(entry.Level.String())[0:4]
 | 
			
		||||
 | 
			
		||||
	if !f.FullTimestamp {
 | 
			
		||||
		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%04d] %-44s ", levelColor, levelText, miniTS(), entry.Message)
 | 
			
		||||
	} else {
 | 
			
		||||
		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%s] %-44s ", levelColor, levelText, entry.Time.Format(timestampFormat), entry.Message)
 | 
			
		||||
	}
 | 
			
		||||
	for _, k := range keys {
 | 
			
		||||
		v := entry.Data[k]
 | 
			
		||||
		fmt.Fprintf(b, " \x1b[%dm%s\x1b[0m=%+v", levelColor, k, v)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func needsQuoting(text string) bool {
 | 
			
		||||
	for _, ch := range text {
 | 
			
		||||
		if !((ch >= 'a' && ch <= 'z') ||
 | 
			
		||||
			(ch >= 'A' && ch <= 'Z') ||
 | 
			
		||||
			(ch >= '0' && ch <= '9') ||
 | 
			
		||||
			ch == '-' || ch == '.') {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *TextFormatter) appendKeyValue(b *bytes.Buffer, key string, value interface{}) {
 | 
			
		||||
 | 
			
		||||
	b.WriteString(key)
 | 
			
		||||
	b.WriteByte('=')
 | 
			
		||||
 | 
			
		||||
	switch value := value.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		if needsQuoting(value) {
 | 
			
		||||
			b.WriteString(value)
 | 
			
		||||
		} else {
 | 
			
		||||
			fmt.Fprintf(b, "%q", value)
 | 
			
		||||
		}
 | 
			
		||||
	case error:
 | 
			
		||||
		errmsg := value.Error()
 | 
			
		||||
		if needsQuoting(errmsg) {
 | 
			
		||||
			b.WriteString(errmsg)
 | 
			
		||||
		} else {
 | 
			
		||||
			fmt.Fprintf(b, "%q", value)
 | 
			
		||||
		}
 | 
			
		||||
	default:
 | 
			
		||||
		fmt.Fprint(b, value)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.WriteByte(' ')
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										31
									
								
								vendor/github.com/Sirupsen/logrus/writer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								vendor/github.com/Sirupsen/logrus/writer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,31 @@
 | 
			
		||||
package logrus
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"io"
 | 
			
		||||
	"runtime"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) Writer() *io.PipeWriter {
 | 
			
		||||
	reader, writer := io.Pipe()
 | 
			
		||||
 | 
			
		||||
	go logger.writerScanner(reader)
 | 
			
		||||
	runtime.SetFinalizer(writer, writerFinalizer)
 | 
			
		||||
 | 
			
		||||
	return writer
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (logger *Logger) writerScanner(reader *io.PipeReader) {
 | 
			
		||||
	scanner := bufio.NewScanner(reader)
 | 
			
		||||
	for scanner.Scan() {
 | 
			
		||||
		logger.Print(scanner.Text())
 | 
			
		||||
	}
 | 
			
		||||
	if err := scanner.Err(); err != nil {
 | 
			
		||||
		logger.Errorf("Error while reading from Writer: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
	reader.Close()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func writerFinalizer(writer *io.PipeWriter) {
 | 
			
		||||
	writer.Close()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										13
									
								
								vendor/github.com/alecthomas/log4go/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/alecthomas/log4go/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
			
		||||
Copyright (c) 2010, Kyle Lemons <kyle@kylelemons.net>. All rights reserved.
 | 
			
		||||
 | 
			
		||||
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
 | 
			
		||||
 * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
 | 
			
		||||
 * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the
 | 
			
		||||
   documentation and/or other materials provided with the distribution.
 | 
			
		||||
 | 
			
		||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 | 
			
		||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 | 
			
		||||
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 | 
			
		||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 | 
			
		||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 | 
			
		||||
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | 
			
		||||
							
								
								
									
										288
									
								
								vendor/github.com/alecthomas/log4go/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										288
									
								
								vendor/github.com/alecthomas/log4go/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,288 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/xml"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"os"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type xmlProperty struct {
 | 
			
		||||
	Name  string `xml:"name,attr"`
 | 
			
		||||
	Value string `xml:",chardata"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type xmlFilter struct {
 | 
			
		||||
	Enabled  string        `xml:"enabled,attr"`
 | 
			
		||||
	Tag      string        `xml:"tag"`
 | 
			
		||||
	Level    string        `xml:"level"`
 | 
			
		||||
	Type     string        `xml:"type"`
 | 
			
		||||
	Property []xmlProperty `xml:"property"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type xmlLoggerConfig struct {
 | 
			
		||||
	Filter []xmlFilter `xml:"filter"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Load XML configuration; see examples/example.xml for documentation
 | 
			
		||||
func (log Logger) LoadConfiguration(filename string) {
 | 
			
		||||
	log.Close()
 | 
			
		||||
 | 
			
		||||
	// Open the configuration file
 | 
			
		||||
	fd, err := os.Open(filename)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not open %q for reading: %s\n", filename, err)
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	contents, err := ioutil.ReadAll(fd)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not read %q: %s\n", filename, err)
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	xc := new(xmlLoggerConfig)
 | 
			
		||||
	if err := xml.Unmarshal(contents, xc); err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not parse XML configuration in %q: %s\n", filename, err)
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, xmlfilt := range xc.Filter {
 | 
			
		||||
		var filt LogWriter
 | 
			
		||||
		var lvl Level
 | 
			
		||||
		bad, good, enabled := false, true, false
 | 
			
		||||
 | 
			
		||||
		// Check required children
 | 
			
		||||
		if len(xmlfilt.Enabled) == 0 {
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required attribute %s for filter missing in %s\n", "enabled", filename)
 | 
			
		||||
			bad = true
 | 
			
		||||
		} else {
 | 
			
		||||
			enabled = xmlfilt.Enabled != "false"
 | 
			
		||||
		}
 | 
			
		||||
		if len(xmlfilt.Tag) == 0 {
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "tag", filename)
 | 
			
		||||
			bad = true
 | 
			
		||||
		}
 | 
			
		||||
		if len(xmlfilt.Type) == 0 {
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "type", filename)
 | 
			
		||||
			bad = true
 | 
			
		||||
		}
 | 
			
		||||
		if len(xmlfilt.Level) == 0 {
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "level", filename)
 | 
			
		||||
			bad = true
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		switch xmlfilt.Level {
 | 
			
		||||
		case "FINEST":
 | 
			
		||||
			lvl = FINEST
 | 
			
		||||
		case "FINE":
 | 
			
		||||
			lvl = FINE
 | 
			
		||||
		case "DEBUG":
 | 
			
		||||
			lvl = DEBUG
 | 
			
		||||
		case "TRACE":
 | 
			
		||||
			lvl = TRACE
 | 
			
		||||
		case "INFO":
 | 
			
		||||
			lvl = INFO
 | 
			
		||||
		case "WARNING":
 | 
			
		||||
			lvl = WARNING
 | 
			
		||||
		case "ERROR":
 | 
			
		||||
			lvl = ERROR
 | 
			
		||||
		case "CRITICAL":
 | 
			
		||||
			lvl = CRITICAL
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter has unknown value in %s: %s\n", "level", filename, xmlfilt.Level)
 | 
			
		||||
			bad = true
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Just so all of the required attributes are errored at the same time if missing
 | 
			
		||||
		if bad {
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		switch xmlfilt.Type {
 | 
			
		||||
		case "console":
 | 
			
		||||
			filt, good = xmlToConsoleLogWriter(filename, xmlfilt.Property, enabled)
 | 
			
		||||
		case "file":
 | 
			
		||||
			filt, good = xmlToFileLogWriter(filename, xmlfilt.Property, enabled)
 | 
			
		||||
		case "xml":
 | 
			
		||||
			filt, good = xmlToXMLLogWriter(filename, xmlfilt.Property, enabled)
 | 
			
		||||
		case "socket":
 | 
			
		||||
			filt, good = xmlToSocketLogWriter(filename, xmlfilt.Property, enabled)
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not load XML configuration in %s: unknown filter type \"%s\"\n", filename, xmlfilt.Type)
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Just so all of the required params are errored at the same time if wrong
 | 
			
		||||
		if !good {
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// If we're disabled (syntax and correctness checks only), don't add to logger
 | 
			
		||||
		if !enabled {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		log[xmlfilt.Tag] = &Filter{lvl, filt}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func xmlToConsoleLogWriter(filename string, props []xmlProperty, enabled bool) (*ConsoleLogWriter, bool) {
 | 
			
		||||
	// Parse properties
 | 
			
		||||
	for _, prop := range props {
 | 
			
		||||
		switch prop.Name {
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for console filter in %s\n", prop.Name, filename)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If it's disabled, we're just checking syntax
 | 
			
		||||
	if !enabled {
 | 
			
		||||
		return nil, true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return NewConsoleLogWriter(), true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Parse a number with K/M/G suffixes based on thousands (1000) or 2^10 (1024)
 | 
			
		||||
func strToNumSuffix(str string, mult int) int {
 | 
			
		||||
	num := 1
 | 
			
		||||
	if len(str) > 1 {
 | 
			
		||||
		switch str[len(str)-1] {
 | 
			
		||||
		case 'G', 'g':
 | 
			
		||||
			num *= mult
 | 
			
		||||
			fallthrough
 | 
			
		||||
		case 'M', 'm':
 | 
			
		||||
			num *= mult
 | 
			
		||||
			fallthrough
 | 
			
		||||
		case 'K', 'k':
 | 
			
		||||
			num *= mult
 | 
			
		||||
			str = str[0 : len(str)-1]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	parsed, _ := strconv.Atoi(str)
 | 
			
		||||
	return parsed * num
 | 
			
		||||
}
 | 
			
		||||
func xmlToFileLogWriter(filename string, props []xmlProperty, enabled bool) (*FileLogWriter, bool) {
 | 
			
		||||
	file := ""
 | 
			
		||||
	format := "[%D %T] [%L] (%S) %M"
 | 
			
		||||
	maxlines := 0
 | 
			
		||||
	maxsize := 0
 | 
			
		||||
	daily := false
 | 
			
		||||
	rotate := false
 | 
			
		||||
 | 
			
		||||
	// Parse properties
 | 
			
		||||
	for _, prop := range props {
 | 
			
		||||
		switch prop.Name {
 | 
			
		||||
		case "filename":
 | 
			
		||||
			file = strings.Trim(prop.Value, " \r\n")
 | 
			
		||||
		case "format":
 | 
			
		||||
			format = strings.Trim(prop.Value, " \r\n")
 | 
			
		||||
		case "maxlines":
 | 
			
		||||
			maxlines = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1000)
 | 
			
		||||
		case "maxsize":
 | 
			
		||||
			maxsize = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1024)
 | 
			
		||||
		case "daily":
 | 
			
		||||
			daily = strings.Trim(prop.Value, " \r\n") != "false"
 | 
			
		||||
		case "rotate":
 | 
			
		||||
			rotate = strings.Trim(prop.Value, " \r\n") != "false"
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for file filter in %s\n", prop.Name, filename)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Check properties
 | 
			
		||||
	if len(file) == 0 {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for file filter missing in %s\n", "filename", filename)
 | 
			
		||||
		return nil, false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If it's disabled, we're just checking syntax
 | 
			
		||||
	if !enabled {
 | 
			
		||||
		return nil, true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	flw := NewFileLogWriter(file, rotate)
 | 
			
		||||
	flw.SetFormat(format)
 | 
			
		||||
	flw.SetRotateLines(maxlines)
 | 
			
		||||
	flw.SetRotateSize(maxsize)
 | 
			
		||||
	flw.SetRotateDaily(daily)
 | 
			
		||||
	return flw, true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func xmlToXMLLogWriter(filename string, props []xmlProperty, enabled bool) (*FileLogWriter, bool) {
 | 
			
		||||
	file := ""
 | 
			
		||||
	maxrecords := 0
 | 
			
		||||
	maxsize := 0
 | 
			
		||||
	daily := false
 | 
			
		||||
	rotate := false
 | 
			
		||||
 | 
			
		||||
	// Parse properties
 | 
			
		||||
	for _, prop := range props {
 | 
			
		||||
		switch prop.Name {
 | 
			
		||||
		case "filename":
 | 
			
		||||
			file = strings.Trim(prop.Value, " \r\n")
 | 
			
		||||
		case "maxrecords":
 | 
			
		||||
			maxrecords = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1000)
 | 
			
		||||
		case "maxsize":
 | 
			
		||||
			maxsize = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1024)
 | 
			
		||||
		case "daily":
 | 
			
		||||
			daily = strings.Trim(prop.Value, " \r\n") != "false"
 | 
			
		||||
		case "rotate":
 | 
			
		||||
			rotate = strings.Trim(prop.Value, " \r\n") != "false"
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for xml filter in %s\n", prop.Name, filename)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Check properties
 | 
			
		||||
	if len(file) == 0 {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for xml filter missing in %s\n", "filename", filename)
 | 
			
		||||
		return nil, false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If it's disabled, we're just checking syntax
 | 
			
		||||
	if !enabled {
 | 
			
		||||
		return nil, true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	xlw := NewXMLLogWriter(file, rotate)
 | 
			
		||||
	xlw.SetRotateLines(maxrecords)
 | 
			
		||||
	xlw.SetRotateSize(maxsize)
 | 
			
		||||
	xlw.SetRotateDaily(daily)
 | 
			
		||||
	return xlw, true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func xmlToSocketLogWriter(filename string, props []xmlProperty, enabled bool) (SocketLogWriter, bool) {
 | 
			
		||||
	endpoint := ""
 | 
			
		||||
	protocol := "udp"
 | 
			
		||||
 | 
			
		||||
	// Parse properties
 | 
			
		||||
	for _, prop := range props {
 | 
			
		||||
		switch prop.Name {
 | 
			
		||||
		case "endpoint":
 | 
			
		||||
			endpoint = strings.Trim(prop.Value, " \r\n")
 | 
			
		||||
		case "protocol":
 | 
			
		||||
			protocol = strings.Trim(prop.Value, " \r\n")
 | 
			
		||||
		default:
 | 
			
		||||
			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for file filter in %s\n", prop.Name, filename)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Check properties
 | 
			
		||||
	if len(endpoint) == 0 {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for file filter missing in %s\n", "endpoint", filename)
 | 
			
		||||
		return nil, false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If it's disabled, we're just checking syntax
 | 
			
		||||
	if !enabled {
 | 
			
		||||
		return nil, true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return NewSocketLogWriter(protocol, endpoint), true
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										14
									
								
								vendor/github.com/alecthomas/log4go/examples/ConsoleLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/alecthomas/log4go/examples/ConsoleLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
import l4g "code.google.com/p/log4go"
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	log := l4g.NewLogger()
 | 
			
		||||
	defer log.Close()
 | 
			
		||||
	log.AddFilter("stdout", l4g.DEBUG, l4g.NewConsoleLogWriter())
 | 
			
		||||
	log.Info("The time is now: %s", time.Now().Format("15:04:05 MST 2006/01/02"))
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										57
									
								
								vendor/github.com/alecthomas/log4go/examples/FileLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										57
									
								
								vendor/github.com/alecthomas/log4go/examples/FileLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,57 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
import l4g "code.google.com/p/log4go"
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	filename = "flw.log"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	// Get a new logger instance
 | 
			
		||||
	log := l4g.NewLogger()
 | 
			
		||||
 | 
			
		||||
	// Create a default logger that is logging messages of FINE or higher
 | 
			
		||||
	log.AddFilter("file", l4g.FINE, l4g.NewFileLogWriter(filename, false))
 | 
			
		||||
	log.Close()
 | 
			
		||||
 | 
			
		||||
	/* Can also specify manually via the following: (these are the defaults) */
 | 
			
		||||
	flw := l4g.NewFileLogWriter(filename, false)
 | 
			
		||||
	flw.SetFormat("[%D %T] [%L] (%S) %M")
 | 
			
		||||
	flw.SetRotate(false)
 | 
			
		||||
	flw.SetRotateSize(0)
 | 
			
		||||
	flw.SetRotateLines(0)
 | 
			
		||||
	flw.SetRotateDaily(false)
 | 
			
		||||
	log.AddFilter("file", l4g.FINE, flw)
 | 
			
		||||
 | 
			
		||||
	// Log some experimental messages
 | 
			
		||||
	log.Finest("Everything is created now (notice that I will not be printing to the file)")
 | 
			
		||||
	log.Info("The time is now: %s", time.Now().Format("15:04:05 MST 2006/01/02"))
 | 
			
		||||
	log.Critical("Time to close out!")
 | 
			
		||||
 | 
			
		||||
	// Close the log
 | 
			
		||||
	log.Close()
 | 
			
		||||
 | 
			
		||||
	// Print what was logged to the file (yes, I know I'm skipping error checking)
 | 
			
		||||
	fd, _ := os.Open(filename)
 | 
			
		||||
	in := bufio.NewReader(fd)
 | 
			
		||||
	fmt.Print("Messages logged to file were: (line numbers not included)\n")
 | 
			
		||||
	for lineno := 1; ; lineno++ {
 | 
			
		||||
		line, err := in.ReadString('\n')
 | 
			
		||||
		if err == io.EOF {
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		fmt.Printf("%3d:\t%s", lineno, line)
 | 
			
		||||
	}
 | 
			
		||||
	fd.Close()
 | 
			
		||||
 | 
			
		||||
	// Remove the file so it's not lying around
 | 
			
		||||
	os.Remove(filename)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										42
									
								
								vendor/github.com/alecthomas/log4go/examples/SimpleNetLogServer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								vendor/github.com/alecthomas/log4go/examples/SimpleNetLogServer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,42 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net"
 | 
			
		||||
	"os"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	port = flag.String("p", "12124", "Port number to listen on")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func e(err error) {
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Printf("Erroring out: %s\n", err)
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	// Bind to the port
 | 
			
		||||
	bind, err := net.ResolveUDPAddr("0.0.0.0:" + *port)
 | 
			
		||||
	e(err)
 | 
			
		||||
 | 
			
		||||
	// Create listener
 | 
			
		||||
	listener, err := net.ListenUDP("udp", bind)
 | 
			
		||||
	e(err)
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Listening to port %s...\n", *port)
 | 
			
		||||
	for {
 | 
			
		||||
		// read into a new buffer
 | 
			
		||||
		buffer := make([]byte, 1024)
 | 
			
		||||
		_, _, err := listener.ReadFrom(buffer)
 | 
			
		||||
		e(err)
 | 
			
		||||
 | 
			
		||||
		// log to standard output
 | 
			
		||||
		fmt.Println(string(buffer))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										18
									
								
								vendor/github.com/alecthomas/log4go/examples/SocketLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/alecthomas/log4go/examples/SocketLogWriter_Manual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
import l4g "code.google.com/p/log4go"
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	log := l4g.NewLogger()
 | 
			
		||||
	log.AddFilter("network", l4g.FINEST, l4g.NewSocketLogWriter("udp", "192.168.1.255:12124"))
 | 
			
		||||
 | 
			
		||||
	// Run `nc -u -l -p 12124` or similar before you run this to see the following message
 | 
			
		||||
	log.Info("The time is now: %s", time.Now().Format("15:04:05 MST 2006/01/02"))
 | 
			
		||||
 | 
			
		||||
	// This makes sure the output stream buffer is written
 | 
			
		||||
	log.Close()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										13
									
								
								vendor/github.com/alecthomas/log4go/examples/XMLConfigurationExample.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/alecthomas/log4go/examples/XMLConfigurationExample.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import l4g "code.google.com/p/log4go"
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	// Load the configuration (isn't this easy?)
 | 
			
		||||
	l4g.LoadConfiguration("example.xml")
 | 
			
		||||
 | 
			
		||||
	// And now we're ready!
 | 
			
		||||
	l4g.Finest("This will only go to those of you really cool UDP kids!  If you change enabled=true.")
 | 
			
		||||
	l4g.Debug("Oh no!  %d + %d = %d!", 2, 2, 2+2)
 | 
			
		||||
	l4g.Info("About that time, eh chaps?")
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										264
									
								
								vendor/github.com/alecthomas/log4go/filelog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										264
									
								
								vendor/github.com/alecthomas/log4go/filelog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,264 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// This log writer sends output to a file
 | 
			
		||||
type FileLogWriter struct {
 | 
			
		||||
	rec chan *LogRecord
 | 
			
		||||
	rot chan bool
 | 
			
		||||
 | 
			
		||||
	// The opened file
 | 
			
		||||
	filename string
 | 
			
		||||
	file     *os.File
 | 
			
		||||
 | 
			
		||||
	// The logging format
 | 
			
		||||
	format string
 | 
			
		||||
 | 
			
		||||
	// File header/trailer
 | 
			
		||||
	header, trailer string
 | 
			
		||||
 | 
			
		||||
	// Rotate at linecount
 | 
			
		||||
	maxlines          int
 | 
			
		||||
	maxlines_curlines int
 | 
			
		||||
 | 
			
		||||
	// Rotate at size
 | 
			
		||||
	maxsize         int
 | 
			
		||||
	maxsize_cursize int
 | 
			
		||||
 | 
			
		||||
	// Rotate daily
 | 
			
		||||
	daily          bool
 | 
			
		||||
	daily_opendate int
 | 
			
		||||
 | 
			
		||||
	// Keep old logfiles (.001, .002, etc)
 | 
			
		||||
	rotate    bool
 | 
			
		||||
	maxbackup int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This is the FileLogWriter's output method
 | 
			
		||||
func (w *FileLogWriter) LogWrite(rec *LogRecord) {
 | 
			
		||||
	w.rec <- rec
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (w *FileLogWriter) Close() {
 | 
			
		||||
	close(w.rec)
 | 
			
		||||
	w.file.Sync()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewFileLogWriter creates a new LogWriter which writes to the given file and
 | 
			
		||||
// has rotation enabled if rotate is true.
 | 
			
		||||
//
 | 
			
		||||
// If rotate is true, any time a new log file is opened, the old one is renamed
 | 
			
		||||
// with a .### extension to preserve it.  The various Set* methods can be used
 | 
			
		||||
// to configure log rotation based on lines, size, and daily.
 | 
			
		||||
//
 | 
			
		||||
// The standard log-line format is:
 | 
			
		||||
//   [%D %T] [%L] (%S) %M
 | 
			
		||||
func NewFileLogWriter(fname string, rotate bool) *FileLogWriter {
 | 
			
		||||
	w := &FileLogWriter{
 | 
			
		||||
		rec:       make(chan *LogRecord, LogBufferLength),
 | 
			
		||||
		rot:       make(chan bool),
 | 
			
		||||
		filename:  fname,
 | 
			
		||||
		format:    "[%D %T] [%L] (%S) %M",
 | 
			
		||||
		rotate:    rotate,
 | 
			
		||||
		maxbackup: 999,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// open the file for the first time
 | 
			
		||||
	if err := w.intRotate(); err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	go func() {
 | 
			
		||||
		defer func() {
 | 
			
		||||
			if w.file != nil {
 | 
			
		||||
				fmt.Fprint(w.file, FormatLogRecord(w.trailer, &LogRecord{Created: time.Now()}))
 | 
			
		||||
				w.file.Close()
 | 
			
		||||
			}
 | 
			
		||||
		}()
 | 
			
		||||
 | 
			
		||||
		for {
 | 
			
		||||
			select {
 | 
			
		||||
			case <-w.rot:
 | 
			
		||||
				if err := w.intRotate(); err != nil {
 | 
			
		||||
					fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
			case rec, ok := <-w.rec:
 | 
			
		||||
				if !ok {
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				now := time.Now()
 | 
			
		||||
				if (w.maxlines > 0 && w.maxlines_curlines >= w.maxlines) ||
 | 
			
		||||
					(w.maxsize > 0 && w.maxsize_cursize >= w.maxsize) ||
 | 
			
		||||
					(w.daily && now.Day() != w.daily_opendate) {
 | 
			
		||||
					if err := w.intRotate(); err != nil {
 | 
			
		||||
						fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err)
 | 
			
		||||
						return
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				// Perform the write
 | 
			
		||||
				n, err := fmt.Fprint(w.file, FormatLogRecord(w.format, rec))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				// Update the counts
 | 
			
		||||
				w.maxlines_curlines++
 | 
			
		||||
				w.maxsize_cursize += n
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Request that the logs rotate
 | 
			
		||||
func (w *FileLogWriter) Rotate() {
 | 
			
		||||
	w.rot <- true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// If this is called in a threaded context, it MUST be synchronized
 | 
			
		||||
func (w *FileLogWriter) intRotate() error {
 | 
			
		||||
	// Close any log file that may be open
 | 
			
		||||
	if w.file != nil {
 | 
			
		||||
		fmt.Fprint(w.file, FormatLogRecord(w.trailer, &LogRecord{Created: time.Now()}))
 | 
			
		||||
		w.file.Close()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If we are keeping log files, move it to the next available number
 | 
			
		||||
	if w.rotate {
 | 
			
		||||
		_, err := os.Lstat(w.filename)
 | 
			
		||||
		if err == nil { // file exists
 | 
			
		||||
			// Find the next available number
 | 
			
		||||
			num := 1
 | 
			
		||||
			fname := ""
 | 
			
		||||
			if w.daily && time.Now().Day() != w.daily_opendate {
 | 
			
		||||
				yesterday := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
 | 
			
		||||
 | 
			
		||||
				for ; err == nil && num <= 999; num++ {
 | 
			
		||||
					fname = w.filename + fmt.Sprintf(".%s.%03d", yesterday, num)
 | 
			
		||||
					_, err = os.Lstat(fname)
 | 
			
		||||
				}
 | 
			
		||||
				// return error if the last file checked still existed
 | 
			
		||||
				if err == nil {
 | 
			
		||||
					return fmt.Errorf("Rotate: Cannot find free log number to rename %s\n", w.filename)
 | 
			
		||||
				}
 | 
			
		||||
			} else {
 | 
			
		||||
				num = w.maxbackup - 1
 | 
			
		||||
				for ; num >= 1; num-- {
 | 
			
		||||
					fname = w.filename + fmt.Sprintf(".%d", num)
 | 
			
		||||
					nfname := w.filename + fmt.Sprintf(".%d", num+1)
 | 
			
		||||
					_, err = os.Lstat(fname)
 | 
			
		||||
					if err == nil {
 | 
			
		||||
						os.Rename(fname, nfname)
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			w.file.Close()
 | 
			
		||||
			// Rename the file to its newfound home
 | 
			
		||||
			err = os.Rename(w.filename, fname)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return fmt.Errorf("Rotate: %s\n", err)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Open the log file
 | 
			
		||||
	fd, err := os.OpenFile(w.filename, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0660)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	w.file = fd
 | 
			
		||||
 | 
			
		||||
	now := time.Now()
 | 
			
		||||
	fmt.Fprint(w.file, FormatLogRecord(w.header, &LogRecord{Created: now}))
 | 
			
		||||
 | 
			
		||||
	// Set the daily open date to the current date
 | 
			
		||||
	w.daily_opendate = now.Day()
 | 
			
		||||
 | 
			
		||||
	// initialize rotation values
 | 
			
		||||
	w.maxlines_curlines = 0
 | 
			
		||||
	w.maxsize_cursize = 0
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set the logging format (chainable).  Must be called before the first log
 | 
			
		||||
// message is written.
 | 
			
		||||
func (w *FileLogWriter) SetFormat(format string) *FileLogWriter {
 | 
			
		||||
	w.format = format
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set the logfile header and footer (chainable).  Must be called before the first log
 | 
			
		||||
// message is written.  These are formatted similar to the FormatLogRecord (e.g.
 | 
			
		||||
// you can use %D and %T in your header/footer for date and time).
 | 
			
		||||
func (w *FileLogWriter) SetHeadFoot(head, foot string) *FileLogWriter {
 | 
			
		||||
	w.header, w.trailer = head, foot
 | 
			
		||||
	if w.maxlines_curlines == 0 {
 | 
			
		||||
		fmt.Fprint(w.file, FormatLogRecord(w.header, &LogRecord{Created: time.Now()}))
 | 
			
		||||
	}
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set rotate at linecount (chainable). Must be called before the first log
 | 
			
		||||
// message is written.
 | 
			
		||||
func (w *FileLogWriter) SetRotateLines(maxlines int) *FileLogWriter {
 | 
			
		||||
	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateLines: %v\n", maxlines)
 | 
			
		||||
	w.maxlines = maxlines
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set rotate at size (chainable). Must be called before the first log message
 | 
			
		||||
// is written.
 | 
			
		||||
func (w *FileLogWriter) SetRotateSize(maxsize int) *FileLogWriter {
 | 
			
		||||
	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateSize: %v\n", maxsize)
 | 
			
		||||
	w.maxsize = maxsize
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set rotate daily (chainable). Must be called before the first log message is
 | 
			
		||||
// written.
 | 
			
		||||
func (w *FileLogWriter) SetRotateDaily(daily bool) *FileLogWriter {
 | 
			
		||||
	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateDaily: %v\n", daily)
 | 
			
		||||
	w.daily = daily
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Set max backup files. Must be called before the first log message
 | 
			
		||||
// is written.
 | 
			
		||||
func (w *FileLogWriter) SetRotateMaxBackup(maxbackup int) *FileLogWriter {
 | 
			
		||||
	w.maxbackup = maxbackup
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// SetRotate changes whether or not the old logs are kept. (chainable) Must be
 | 
			
		||||
// called before the first log message is written.  If rotate is false, the
 | 
			
		||||
// files are overwritten; otherwise, they are rotated to another file before the
 | 
			
		||||
// new log is opened.
 | 
			
		||||
func (w *FileLogWriter) SetRotate(rotate bool) *FileLogWriter {
 | 
			
		||||
	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotate: %v\n", rotate)
 | 
			
		||||
	w.rotate = rotate
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewXMLLogWriter is a utility method for creating a FileLogWriter set up to
 | 
			
		||||
// output XML record log messages instead of line-based ones.
 | 
			
		||||
func NewXMLLogWriter(fname string, rotate bool) *FileLogWriter {
 | 
			
		||||
	return NewFileLogWriter(fname, rotate).SetFormat(
 | 
			
		||||
		`	<record level="%L">
 | 
			
		||||
		<timestamp>%D %T</timestamp>
 | 
			
		||||
		<source>%S</source>
 | 
			
		||||
		<message>%M</message>
 | 
			
		||||
	</record>`).SetHeadFoot("<log created=\"%D %T\">", "</log>")
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										484
									
								
								vendor/github.com/alecthomas/log4go/log4go.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										484
									
								
								vendor/github.com/alecthomas/log4go/log4go.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,484 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
// Package log4go provides level-based and highly configurable logging.
 | 
			
		||||
//
 | 
			
		||||
// Enhanced Logging
 | 
			
		||||
//
 | 
			
		||||
// This is inspired by the logging functionality in Java.  Essentially, you create a Logger
 | 
			
		||||
// object and create output filters for it.  You can send whatever you want to the Logger,
 | 
			
		||||
// and it will filter that based on your settings and send it to the outputs.  This way, you
 | 
			
		||||
// can put as much debug code in your program as you want, and when you're done you can filter
 | 
			
		||||
// out the mundane messages so only the important ones show up.
 | 
			
		||||
//
 | 
			
		||||
// Utility functions are provided to make life easier. Here is some example code to get started:
 | 
			
		||||
//
 | 
			
		||||
// log := log4go.NewLogger()
 | 
			
		||||
// log.AddFilter("stdout", log4go.DEBUG, log4go.NewConsoleLogWriter())
 | 
			
		||||
// log.AddFilter("log",    log4go.FINE,  log4go.NewFileLogWriter("example.log", true))
 | 
			
		||||
// log.Info("The time is now: %s", time.LocalTime().Format("15:04:05 MST 2006/01/02"))
 | 
			
		||||
//
 | 
			
		||||
// The first two lines can be combined with the utility NewDefaultLogger:
 | 
			
		||||
//
 | 
			
		||||
// log := log4go.NewDefaultLogger(log4go.DEBUG)
 | 
			
		||||
// log.AddFilter("log",    log4go.FINE,  log4go.NewFileLogWriter("example.log", true))
 | 
			
		||||
// log.Info("The time is now: %s", time.LocalTime().Format("15:04:05 MST 2006/01/02"))
 | 
			
		||||
//
 | 
			
		||||
// Usage notes:
 | 
			
		||||
// - The ConsoleLogWriter does not display the source of the message to standard
 | 
			
		||||
//   output, but the FileLogWriter does.
 | 
			
		||||
// - The utility functions (Info, Debug, Warn, etc) derive their source from the
 | 
			
		||||
//   calling function, and this incurs extra overhead.
 | 
			
		||||
//
 | 
			
		||||
// Changes from 2.0:
 | 
			
		||||
// - The external interface has remained mostly stable, but a lot of the
 | 
			
		||||
//   internals have been changed, so if you depended on any of this or created
 | 
			
		||||
//   your own LogWriter, then you will probably have to update your code.  In
 | 
			
		||||
//   particular, Logger is now a map and ConsoleLogWriter is now a channel
 | 
			
		||||
//   behind-the-scenes, and the LogWrite method no longer has return values.
 | 
			
		||||
//
 | 
			
		||||
// Future work: (please let me know if you think I should work on any of these particularly)
 | 
			
		||||
// - Log file rotation
 | 
			
		||||
// - Logging configuration files ala log4j
 | 
			
		||||
// - Have the ability to remove filters?
 | 
			
		||||
// - Have GetInfoChannel, GetDebugChannel, etc return a chan string that allows
 | 
			
		||||
//   for another method of logging
 | 
			
		||||
// - Add an XML filter type
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"runtime"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Version information
 | 
			
		||||
const (
 | 
			
		||||
	L4G_VERSION = "log4go-v3.0.1"
 | 
			
		||||
	L4G_MAJOR   = 3
 | 
			
		||||
	L4G_MINOR   = 0
 | 
			
		||||
	L4G_BUILD   = 1
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
/****** Constants ******/
 | 
			
		||||
 | 
			
		||||
// These are the integer logging levels used by the logger
 | 
			
		||||
type Level int
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	FINEST Level = iota
 | 
			
		||||
	FINE
 | 
			
		||||
	DEBUG
 | 
			
		||||
	TRACE
 | 
			
		||||
	INFO
 | 
			
		||||
	WARNING
 | 
			
		||||
	ERROR
 | 
			
		||||
	CRITICAL
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Logging level strings
 | 
			
		||||
var (
 | 
			
		||||
	levelStrings = [...]string{"FNST", "FINE", "DEBG", "TRAC", "INFO", "WARN", "EROR", "CRIT"}
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func (l Level) String() string {
 | 
			
		||||
	if l < 0 || int(l) > len(levelStrings) {
 | 
			
		||||
		return "UNKNOWN"
 | 
			
		||||
	}
 | 
			
		||||
	return levelStrings[int(l)]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/****** Variables ******/
 | 
			
		||||
var (
 | 
			
		||||
	// LogBufferLength specifies how many log messages a particular log4go
 | 
			
		||||
	// logger can buffer at a time before writing them.
 | 
			
		||||
	LogBufferLength = 32
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
/****** LogRecord ******/
 | 
			
		||||
 | 
			
		||||
// A LogRecord contains all of the pertinent information for each message
 | 
			
		||||
type LogRecord struct {
 | 
			
		||||
	Level   Level     // The log level
 | 
			
		||||
	Created time.Time // The time at which the log message was created (nanoseconds)
 | 
			
		||||
	Source  string    // The message source
 | 
			
		||||
	Message string    // The log message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/****** LogWriter ******/
 | 
			
		||||
 | 
			
		||||
// This is an interface for anything that should be able to write logs
 | 
			
		||||
type LogWriter interface {
 | 
			
		||||
	// This will be called to log a LogRecord message.
 | 
			
		||||
	LogWrite(rec *LogRecord)
 | 
			
		||||
 | 
			
		||||
	// This should clean up anything lingering about the LogWriter, as it is called before
 | 
			
		||||
	// the LogWriter is removed.  LogWrite should not be called after Close.
 | 
			
		||||
	Close()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/****** Logger ******/
 | 
			
		||||
 | 
			
		||||
// A Filter represents the log level below which no log records are written to
 | 
			
		||||
// the associated LogWriter.
 | 
			
		||||
type Filter struct {
 | 
			
		||||
	Level Level
 | 
			
		||||
	LogWriter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// A Logger represents a collection of Filters through which log messages are
 | 
			
		||||
// written.
 | 
			
		||||
type Logger map[string]*Filter
 | 
			
		||||
 | 
			
		||||
// Create a new logger.
 | 
			
		||||
//
 | 
			
		||||
// DEPRECATED: Use make(Logger) instead.
 | 
			
		||||
func NewLogger() Logger {
 | 
			
		||||
	os.Stderr.WriteString("warning: use of deprecated NewLogger\n")
 | 
			
		||||
	return make(Logger)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Create a new logger with a "stdout" filter configured to send log messages at
 | 
			
		||||
// or above lvl to standard output.
 | 
			
		||||
//
 | 
			
		||||
// DEPRECATED: use NewDefaultLogger instead.
 | 
			
		||||
func NewConsoleLogger(lvl Level) Logger {
 | 
			
		||||
	os.Stderr.WriteString("warning: use of deprecated NewConsoleLogger\n")
 | 
			
		||||
	return Logger{
 | 
			
		||||
		"stdout": &Filter{lvl, NewConsoleLogWriter()},
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Create a new logger with a "stdout" filter configured to send log messages at
 | 
			
		||||
// or above lvl to standard output.
 | 
			
		||||
func NewDefaultLogger(lvl Level) Logger {
 | 
			
		||||
	return Logger{
 | 
			
		||||
		"stdout": &Filter{lvl, NewConsoleLogWriter()},
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Closes all log writers in preparation for exiting the program or a
 | 
			
		||||
// reconfiguration of logging.  Calling this is not really imperative, unless
 | 
			
		||||
// you want to guarantee that all log messages are written.  Close removes
 | 
			
		||||
// all filters (and thus all LogWriters) from the logger.
 | 
			
		||||
func (log Logger) Close() {
 | 
			
		||||
	// Close all open loggers
 | 
			
		||||
	for name, filt := range log {
 | 
			
		||||
		filt.Close()
 | 
			
		||||
		delete(log, name)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Add a new LogWriter to the Logger which will only log messages at lvl or
 | 
			
		||||
// higher.  This function should not be called from multiple goroutines.
 | 
			
		||||
// Returns the logger for chaining.
 | 
			
		||||
func (log Logger) AddFilter(name string, lvl Level, writer LogWriter) Logger {
 | 
			
		||||
	log[name] = &Filter{lvl, writer}
 | 
			
		||||
	return log
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/******* Logging *******/
 | 
			
		||||
// Send a formatted log message internally
 | 
			
		||||
func (log Logger) intLogf(lvl Level, format string, args ...interface{}) {
 | 
			
		||||
	skip := true
 | 
			
		||||
 | 
			
		||||
	// Determine if any logging will be done
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl >= filt.Level {
 | 
			
		||||
			skip = false
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if skip {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Determine caller func
 | 
			
		||||
	pc, _, lineno, ok := runtime.Caller(2)
 | 
			
		||||
	src := ""
 | 
			
		||||
	if ok {
 | 
			
		||||
		src = fmt.Sprintf("%s:%d", runtime.FuncForPC(pc).Name(), lineno)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	msg := format
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		msg = fmt.Sprintf(format, args...)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Make the log record
 | 
			
		||||
	rec := &LogRecord{
 | 
			
		||||
		Level:   lvl,
 | 
			
		||||
		Created: time.Now(),
 | 
			
		||||
		Source:  src,
 | 
			
		||||
		Message: msg,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Dispatch the logs
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl < filt.Level {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		filt.LogWrite(rec)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Send a closure log message internally
 | 
			
		||||
func (log Logger) intLogc(lvl Level, closure func() string) {
 | 
			
		||||
	skip := true
 | 
			
		||||
 | 
			
		||||
	// Determine if any logging will be done
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl >= filt.Level {
 | 
			
		||||
			skip = false
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if skip {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Determine caller func
 | 
			
		||||
	pc, _, lineno, ok := runtime.Caller(2)
 | 
			
		||||
	src := ""
 | 
			
		||||
	if ok {
 | 
			
		||||
		src = fmt.Sprintf("%s:%d", runtime.FuncForPC(pc).Name(), lineno)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Make the log record
 | 
			
		||||
	rec := &LogRecord{
 | 
			
		||||
		Level:   lvl,
 | 
			
		||||
		Created: time.Now(),
 | 
			
		||||
		Source:  src,
 | 
			
		||||
		Message: closure(),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Dispatch the logs
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl < filt.Level {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		filt.LogWrite(rec)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Send a log message with manual level, source, and message.
 | 
			
		||||
func (log Logger) Log(lvl Level, source, message string) {
 | 
			
		||||
	skip := true
 | 
			
		||||
 | 
			
		||||
	// Determine if any logging will be done
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl >= filt.Level {
 | 
			
		||||
			skip = false
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if skip {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Make the log record
 | 
			
		||||
	rec := &LogRecord{
 | 
			
		||||
		Level:   lvl,
 | 
			
		||||
		Created: time.Now(),
 | 
			
		||||
		Source:  source,
 | 
			
		||||
		Message: message,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Dispatch the logs
 | 
			
		||||
	for _, filt := range log {
 | 
			
		||||
		if lvl < filt.Level {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		filt.LogWrite(rec)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Logf logs a formatted log message at the given log level, using the caller as
 | 
			
		||||
// its source.
 | 
			
		||||
func (log Logger) Logf(lvl Level, format string, args ...interface{}) {
 | 
			
		||||
	log.intLogf(lvl, format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Logc logs a string returned by the closure at the given log level, using the caller as
 | 
			
		||||
// its source.  If no log message would be written, the closure is never called.
 | 
			
		||||
func (log Logger) Logc(lvl Level, closure func() string) {
 | 
			
		||||
	log.intLogc(lvl, closure)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Finest logs a message at the finest log level.
 | 
			
		||||
// See Debug for an explanation of the arguments.
 | 
			
		||||
func (log Logger) Finest(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = FINEST
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		log.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		log.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fine logs a message at the fine log level.
 | 
			
		||||
// See Debug for an explanation of the arguments.
 | 
			
		||||
func (log Logger) Fine(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = FINE
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		log.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		log.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Debug is a utility method for debug log messages.
 | 
			
		||||
// The behavior of Debug depends on the first argument:
 | 
			
		||||
// - arg0 is a string
 | 
			
		||||
//   When given a string as the first argument, this behaves like Logf but with
 | 
			
		||||
//   the DEBUG log level: the first argument is interpreted as a format for the
 | 
			
		||||
//   latter arguments.
 | 
			
		||||
// - arg0 is a func()string
 | 
			
		||||
//   When given a closure of type func()string, this logs the string returned by
 | 
			
		||||
//   the closure iff it will be logged.  The closure runs at most one time.
 | 
			
		||||
// - arg0 is interface{}
 | 
			
		||||
//   When given anything else, the log message will be each of the arguments
 | 
			
		||||
//   formatted with %v and separated by spaces (ala Sprint).
 | 
			
		||||
func (log Logger) Debug(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = DEBUG
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		log.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		log.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Trace logs a message at the trace log level.
 | 
			
		||||
// See Debug for an explanation of the arguments.
 | 
			
		||||
func (log Logger) Trace(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = TRACE
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		log.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		log.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Info logs a message at the info log level.
 | 
			
		||||
// See Debug for an explanation of the arguments.
 | 
			
		||||
func (log Logger) Info(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = INFO
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		log.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		log.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Warn logs a message at the warning log level and returns the formatted error.
 | 
			
		||||
// At the warning level and higher, there is no performance benefit if the
 | 
			
		||||
// message is not actually logged, because all formats are processed and all
 | 
			
		||||
// closures are executed to format the error message.
 | 
			
		||||
// See Debug for further explanation of the arguments.
 | 
			
		||||
func (log Logger) Warn(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = WARNING
 | 
			
		||||
	)
 | 
			
		||||
	var msg string
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		msg = fmt.Sprintf(first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		msg = first()
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
	log.intLogf(lvl, msg)
 | 
			
		||||
	return errors.New(msg)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Error logs a message at the error log level and returns the formatted error,
 | 
			
		||||
// See Warn for an explanation of the performance and Debug for an explanation
 | 
			
		||||
// of the parameters.
 | 
			
		||||
func (log Logger) Error(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = ERROR
 | 
			
		||||
	)
 | 
			
		||||
	var msg string
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		msg = fmt.Sprintf(first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		msg = first()
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
	log.intLogf(lvl, msg)
 | 
			
		||||
	return errors.New(msg)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Critical logs a message at the critical log level and returns the formatted error,
 | 
			
		||||
// See Warn for an explanation of the performance and Debug for an explanation
 | 
			
		||||
// of the parameters.
 | 
			
		||||
func (log Logger) Critical(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = CRITICAL
 | 
			
		||||
	)
 | 
			
		||||
	var msg string
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		msg = fmt.Sprintf(first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		msg = first()
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
	log.intLogf(lvl, msg)
 | 
			
		||||
	return errors.New(msg)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										126
									
								
								vendor/github.com/alecthomas/log4go/pattlog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										126
									
								
								vendor/github.com/alecthomas/log4go/pattlog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,126 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	FORMAT_DEFAULT = "[%D %T] [%L] (%S) %M"
 | 
			
		||||
	FORMAT_SHORT   = "[%t %d] [%L] %M"
 | 
			
		||||
	FORMAT_ABBREV  = "[%L] %M"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type formatCacheType struct {
 | 
			
		||||
	LastUpdateSeconds    int64
 | 
			
		||||
	shortTime, shortDate string
 | 
			
		||||
	longTime, longDate   string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var formatCache = &formatCacheType{}
 | 
			
		||||
 | 
			
		||||
// Known format codes:
 | 
			
		||||
// %T - Time (15:04:05 MST)
 | 
			
		||||
// %t - Time (15:04)
 | 
			
		||||
// %D - Date (2006/01/02)
 | 
			
		||||
// %d - Date (01/02/06)
 | 
			
		||||
// %L - Level (FNST, FINE, DEBG, TRAC, WARN, EROR, CRIT)
 | 
			
		||||
// %S - Source
 | 
			
		||||
// %M - Message
 | 
			
		||||
// Ignores unknown formats
 | 
			
		||||
// Recommended: "[%D %T] [%L] (%S) %M"
 | 
			
		||||
func FormatLogRecord(format string, rec *LogRecord) string {
 | 
			
		||||
	if rec == nil {
 | 
			
		||||
		return "<nil>"
 | 
			
		||||
	}
 | 
			
		||||
	if len(format) == 0 {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	out := bytes.NewBuffer(make([]byte, 0, 64))
 | 
			
		||||
	secs := rec.Created.UnixNano() / 1e9
 | 
			
		||||
 | 
			
		||||
	cache := *formatCache
 | 
			
		||||
	if cache.LastUpdateSeconds != secs {
 | 
			
		||||
		month, day, year := rec.Created.Month(), rec.Created.Day(), rec.Created.Year()
 | 
			
		||||
		hour, minute, second := rec.Created.Hour(), rec.Created.Minute(), rec.Created.Second()
 | 
			
		||||
		zone, _ := rec.Created.Zone()
 | 
			
		||||
		updated := &formatCacheType{
 | 
			
		||||
			LastUpdateSeconds: secs,
 | 
			
		||||
			shortTime:         fmt.Sprintf("%02d:%02d", hour, minute),
 | 
			
		||||
			shortDate:         fmt.Sprintf("%02d/%02d/%02d", day, month, year%100),
 | 
			
		||||
			longTime:          fmt.Sprintf("%02d:%02d:%02d %s", hour, minute, second, zone),
 | 
			
		||||
			longDate:          fmt.Sprintf("%04d/%02d/%02d", year, month, day),
 | 
			
		||||
		}
 | 
			
		||||
		cache = *updated
 | 
			
		||||
		formatCache = updated
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Split the string into pieces by % signs
 | 
			
		||||
	pieces := bytes.Split([]byte(format), []byte{'%'})
 | 
			
		||||
 | 
			
		||||
	// Iterate over the pieces, replacing known formats
 | 
			
		||||
	for i, piece := range pieces {
 | 
			
		||||
		if i > 0 && len(piece) > 0 {
 | 
			
		||||
			switch piece[0] {
 | 
			
		||||
			case 'T':
 | 
			
		||||
				out.WriteString(cache.longTime)
 | 
			
		||||
			case 't':
 | 
			
		||||
				out.WriteString(cache.shortTime)
 | 
			
		||||
			case 'D':
 | 
			
		||||
				out.WriteString(cache.longDate)
 | 
			
		||||
			case 'd':
 | 
			
		||||
				out.WriteString(cache.shortDate)
 | 
			
		||||
			case 'L':
 | 
			
		||||
				out.WriteString(levelStrings[rec.Level])
 | 
			
		||||
			case 'S':
 | 
			
		||||
				out.WriteString(rec.Source)
 | 
			
		||||
			case 's':
 | 
			
		||||
				slice := strings.Split(rec.Source, "/")
 | 
			
		||||
				out.WriteString(slice[len(slice)-1])
 | 
			
		||||
			case 'M':
 | 
			
		||||
				out.WriteString(rec.Message)
 | 
			
		||||
			}
 | 
			
		||||
			if len(piece) > 1 {
 | 
			
		||||
				out.Write(piece[1:])
 | 
			
		||||
			}
 | 
			
		||||
		} else if len(piece) > 0 {
 | 
			
		||||
			out.Write(piece)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
 | 
			
		||||
	return out.String()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This is the standard writer that prints to standard output.
 | 
			
		||||
type FormatLogWriter chan *LogRecord
 | 
			
		||||
 | 
			
		||||
// This creates a new FormatLogWriter
 | 
			
		||||
func NewFormatLogWriter(out io.Writer, format string) FormatLogWriter {
 | 
			
		||||
	records := make(FormatLogWriter, LogBufferLength)
 | 
			
		||||
	go records.run(out, format)
 | 
			
		||||
	return records
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (w FormatLogWriter) run(out io.Writer, format string) {
 | 
			
		||||
	for rec := range w {
 | 
			
		||||
		fmt.Fprint(out, FormatLogRecord(format, rec))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This is the FormatLogWriter's output method.  This will block if the output
 | 
			
		||||
// buffer is full.
 | 
			
		||||
func (w FormatLogWriter) LogWrite(rec *LogRecord) {
 | 
			
		||||
	w <- rec
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Close stops the logger from sending messages to standard output.  Attempts to
 | 
			
		||||
// send log messages to this logger after a Close have undefined behavior.
 | 
			
		||||
func (w FormatLogWriter) Close() {
 | 
			
		||||
	close(w)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										57
									
								
								vendor/github.com/alecthomas/log4go/socklog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										57
									
								
								vendor/github.com/alecthomas/log4go/socklog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,57 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net"
 | 
			
		||||
	"os"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// This log writer sends output to a socket
 | 
			
		||||
type SocketLogWriter chan *LogRecord
 | 
			
		||||
 | 
			
		||||
// This is the SocketLogWriter's output method
 | 
			
		||||
func (w SocketLogWriter) LogWrite(rec *LogRecord) {
 | 
			
		||||
	w <- rec
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (w SocketLogWriter) Close() {
 | 
			
		||||
	close(w)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewSocketLogWriter(proto, hostport string) SocketLogWriter {
 | 
			
		||||
	sock, err := net.Dial(proto, hostport)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "NewSocketLogWriter(%q): %s\n", hostport, err)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	w := SocketLogWriter(make(chan *LogRecord, LogBufferLength))
 | 
			
		||||
 | 
			
		||||
	go func() {
 | 
			
		||||
		defer func() {
 | 
			
		||||
			if sock != nil && proto == "tcp" {
 | 
			
		||||
				sock.Close()
 | 
			
		||||
			}
 | 
			
		||||
		}()
 | 
			
		||||
 | 
			
		||||
		for rec := range w {
 | 
			
		||||
			// Marshall into JSON
 | 
			
		||||
			js, err := json.Marshal(rec)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				fmt.Fprint(os.Stderr, "SocketLogWriter(%q): %s", hostport, err)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			_, err = sock.Write(js)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				fmt.Fprint(os.Stderr, "SocketLogWriter(%q): %s", hostport, err)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	return w
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										49
									
								
								vendor/github.com/alecthomas/log4go/termlog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								vendor/github.com/alecthomas/log4go/termlog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,49 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var stdout io.Writer = os.Stdout
 | 
			
		||||
 | 
			
		||||
// This is the standard writer that prints to standard output.
 | 
			
		||||
type ConsoleLogWriter struct {
 | 
			
		||||
	format string
 | 
			
		||||
	w      chan *LogRecord
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This creates a new ConsoleLogWriter
 | 
			
		||||
func NewConsoleLogWriter() *ConsoleLogWriter {
 | 
			
		||||
	consoleWriter := &ConsoleLogWriter{
 | 
			
		||||
		format: "[%T %D] [%L] (%S) %M",
 | 
			
		||||
		w:      make(chan *LogRecord, LogBufferLength),
 | 
			
		||||
	}
 | 
			
		||||
	go consoleWriter.run(stdout)
 | 
			
		||||
	return consoleWriter
 | 
			
		||||
}
 | 
			
		||||
func (c *ConsoleLogWriter) SetFormat(format string) {
 | 
			
		||||
	c.format = format
 | 
			
		||||
}
 | 
			
		||||
func (c *ConsoleLogWriter) run(out io.Writer) {
 | 
			
		||||
	for rec := range c.w {
 | 
			
		||||
		fmt.Fprint(out, FormatLogRecord(c.format, rec))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This is the ConsoleLogWriter's output method.  This will block if the output
 | 
			
		||||
// buffer is full.
 | 
			
		||||
func (c *ConsoleLogWriter) LogWrite(rec *LogRecord) {
 | 
			
		||||
	c.w <- rec
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Close stops the logger from sending messages to standard output.  Attempts to
 | 
			
		||||
// send log messages to this logger after a Close have undefined behavior.
 | 
			
		||||
func (c *ConsoleLogWriter) Close() {
 | 
			
		||||
	close(c.w)
 | 
			
		||||
	time.Sleep(50 * time.Millisecond) // Try to give console I/O time to complete
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										278
									
								
								vendor/github.com/alecthomas/log4go/wrapper.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										278
									
								
								vendor/github.com/alecthomas/log4go/wrapper.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,278 @@
 | 
			
		||||
// Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved.
 | 
			
		||||
 | 
			
		||||
package log4go
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	Global Logger
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	Global = NewDefaultLogger(DEBUG)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Wrapper for (*Logger).LoadConfiguration
 | 
			
		||||
func LoadConfiguration(filename string) {
 | 
			
		||||
	Global.LoadConfiguration(filename)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Wrapper for (*Logger).AddFilter
 | 
			
		||||
func AddFilter(name string, lvl Level, writer LogWriter) {
 | 
			
		||||
	Global.AddFilter(name, lvl, writer)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Wrapper for (*Logger).Close (closes and removes all logwriters)
 | 
			
		||||
func Close() {
 | 
			
		||||
	Global.Close()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Crash(args ...interface{}) {
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		Global.intLogf(CRITICAL, strings.Repeat(" %v", len(args))[1:], args...)
 | 
			
		||||
	}
 | 
			
		||||
	panic(args)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Logs the given message and crashes the program
 | 
			
		||||
func Crashf(format string, args ...interface{}) {
 | 
			
		||||
	Global.intLogf(CRITICAL, format, args...)
 | 
			
		||||
	Global.Close() // so that hopefully the messages get logged
 | 
			
		||||
	panic(fmt.Sprintf(format, args...))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Exit(args ...interface{}) {
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		Global.intLogf(ERROR, strings.Repeat(" %v", len(args))[1:], args...)
 | 
			
		||||
	}
 | 
			
		||||
	Global.Close() // so that hopefully the messages get logged
 | 
			
		||||
	os.Exit(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Exitf(format string, args ...interface{}) {
 | 
			
		||||
	Global.intLogf(ERROR, format, args...)
 | 
			
		||||
	Global.Close() // so that hopefully the messages get logged
 | 
			
		||||
	os.Exit(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Stderr(args ...interface{}) {
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		Global.intLogf(ERROR, strings.Repeat(" %v", len(args))[1:], args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Stderrf(format string, args ...interface{}) {
 | 
			
		||||
	Global.intLogf(ERROR, format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Stdout(args ...interface{}) {
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		Global.intLogf(INFO, strings.Repeat(" %v", len(args))[1:], args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Compatibility with `log`
 | 
			
		||||
func Stdoutf(format string, args ...interface{}) {
 | 
			
		||||
	Global.intLogf(INFO, format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Send a log message manually
 | 
			
		||||
// Wrapper for (*Logger).Log
 | 
			
		||||
func Log(lvl Level, source, message string) {
 | 
			
		||||
	Global.Log(lvl, source, message)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Send a formatted log message easily
 | 
			
		||||
// Wrapper for (*Logger).Logf
 | 
			
		||||
func Logf(lvl Level, format string, args ...interface{}) {
 | 
			
		||||
	Global.intLogf(lvl, format, args...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Send a closure log message
 | 
			
		||||
// Wrapper for (*Logger).Logc
 | 
			
		||||
func Logc(lvl Level, closure func() string) {
 | 
			
		||||
	Global.intLogc(lvl, closure)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for finest log messages (see Debug() for parameter explanation)
 | 
			
		||||
// Wrapper for (*Logger).Finest
 | 
			
		||||
func Finest(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = FINEST
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		Global.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for fine log messages (see Debug() for parameter explanation)
 | 
			
		||||
// Wrapper for (*Logger).Fine
 | 
			
		||||
func Fine(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = FINE
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		Global.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for debug log messages
 | 
			
		||||
// When given a string as the first argument, this behaves like Logf but with the DEBUG log level (e.g. the first argument is interpreted as a format for the latter arguments)
 | 
			
		||||
// When given a closure of type func()string, this logs the string returned by the closure iff it will be logged.  The closure runs at most one time.
 | 
			
		||||
// When given anything else, the log message will be each of the arguments formatted with %v and separated by spaces (ala Sprint).
 | 
			
		||||
// Wrapper for (*Logger).Debug
 | 
			
		||||
func Debug(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = DEBUG
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		Global.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for trace log messages (see Debug() for parameter explanation)
 | 
			
		||||
// Wrapper for (*Logger).Trace
 | 
			
		||||
func Trace(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = TRACE
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		Global.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for info log messages (see Debug() for parameter explanation)
 | 
			
		||||
// Wrapper for (*Logger).Info
 | 
			
		||||
func Info(arg0 interface{}, args ...interface{}) {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = INFO
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		Global.intLogc(lvl, first)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for warn log messages (returns an error for easy function returns) (see Debug() for parameter explanation)
 | 
			
		||||
// These functions will execute a closure exactly once, to build the error message for the return
 | 
			
		||||
// Wrapper for (*Logger).Warn
 | 
			
		||||
func Warn(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = WARNING
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
		return errors.New(fmt.Sprintf(first, args...))
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		str := first()
 | 
			
		||||
		Global.intLogf(lvl, "%s", str)
 | 
			
		||||
		return errors.New(str)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...))
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for error log messages (returns an error for easy function returns) (see Debug() for parameter explanation)
 | 
			
		||||
// These functions will execute a closure exactly once, to build the error message for the return
 | 
			
		||||
// Wrapper for (*Logger).Error
 | 
			
		||||
func Error(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = ERROR
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
		return errors.New(fmt.Sprintf(first, args...))
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		str := first()
 | 
			
		||||
		Global.intLogf(lvl, "%s", str)
 | 
			
		||||
		return errors.New(str)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...))
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Utility for critical log messages (returns an error for easy function returns) (see Debug() for parameter explanation)
 | 
			
		||||
// These functions will execute a closure exactly once, to build the error message for the return
 | 
			
		||||
// Wrapper for (*Logger).Critical
 | 
			
		||||
func Critical(arg0 interface{}, args ...interface{}) error {
 | 
			
		||||
	const (
 | 
			
		||||
		lvl = CRITICAL
 | 
			
		||||
	)
 | 
			
		||||
	switch first := arg0.(type) {
 | 
			
		||||
	case string:
 | 
			
		||||
		// Use the string as a format string
 | 
			
		||||
		Global.intLogf(lvl, first, args...)
 | 
			
		||||
		return errors.New(fmt.Sprintf(first, args...))
 | 
			
		||||
	case func() string:
 | 
			
		||||
		// Log the closure (no other arguments used)
 | 
			
		||||
		str := first()
 | 
			
		||||
		Global.intLogf(lvl, "%s", str)
 | 
			
		||||
		return errors.New(str)
 | 
			
		||||
	default:
 | 
			
		||||
		// Build a format string so that it will be similar to Sprint
 | 
			
		||||
		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...)
 | 
			
		||||
		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...))
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										28
									
								
								vendor/github.com/bwmarrin/discordgo/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								vendor/github.com/bwmarrin/discordgo/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,28 @@
 | 
			
		||||
Copyright (c) 2015, Bruce Marriner
 | 
			
		||||
All rights reserved.
 | 
			
		||||
 | 
			
		||||
Redistribution and use in source and binary forms, with or without
 | 
			
		||||
modification, are permitted provided that the following conditions are met:
 | 
			
		||||
 | 
			
		||||
* Redistributions of source code must retain the above copyright notice, this
 | 
			
		||||
  list of conditions and the following disclaimer.
 | 
			
		||||
 | 
			
		||||
* Redistributions in binary form must reproduce the above copyright notice,
 | 
			
		||||
  this list of conditions and the following disclaimer in the documentation
 | 
			
		||||
  and/or other materials provided with the distribution.
 | 
			
		||||
 | 
			
		||||
* Neither the name of discordgo nor the names of its
 | 
			
		||||
  contributors may be used to endorse or promote products derived from
 | 
			
		||||
  this software without specific prior written permission.
 | 
			
		||||
 | 
			
		||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 | 
			
		||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | 
			
		||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 | 
			
		||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 | 
			
		||||
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 | 
			
		||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 | 
			
		||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 | 
			
		||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 | 
			
		||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 | 
			
		||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										257
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										257
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,257 @@
 | 
			
		||||
// Discordgo - Discord bindings for Go
 | 
			
		||||
// Available at https://github.com/bwmarrin/discordgo
 | 
			
		||||
 | 
			
		||||
// Copyright 2015-2016 Bruce Marriner <bruce@sqls.net>.  All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// This file contains high level helper functions and easy entry points for the
 | 
			
		||||
// entire discordgo package.  These functions are beling developed and are very
 | 
			
		||||
// experimental at this point.  They will most likley change so please use the
 | 
			
		||||
// low level functions if that's a problem.
 | 
			
		||||
 | 
			
		||||
// Package discordgo provides Discord binding for Go
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"reflect"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// VERSION of Discordgo, follows Symantic Versioning. (http://semver.org/)
 | 
			
		||||
const VERSION = "0.13.0"
 | 
			
		||||
 | 
			
		||||
// New creates a new Discord session and will automate some startup
 | 
			
		||||
// tasks if given enough information to do so.  Currently you can pass zero
 | 
			
		||||
// arguments and it will return an empty Discord session.
 | 
			
		||||
// There are 3 ways to call New:
 | 
			
		||||
//     With a single auth token - All requests will use the token blindly,
 | 
			
		||||
//         no verification of the token will be done and requests may fail.
 | 
			
		||||
//     With an email and password - Discord will sign in with the provided
 | 
			
		||||
//         credentials.
 | 
			
		||||
//     With an email, password and auth token - Discord will verify the auth
 | 
			
		||||
//         token, if it is invalid it will sign in with the provided
 | 
			
		||||
//         credentials. This is the Discord recommended way to sign in.
 | 
			
		||||
func New(args ...interface{}) (s *Session, err error) {
 | 
			
		||||
 | 
			
		||||
	// Create an empty Session interface.
 | 
			
		||||
	s = &Session{
 | 
			
		||||
		State:                  NewState(),
 | 
			
		||||
		StateEnabled:           true,
 | 
			
		||||
		Compress:               true,
 | 
			
		||||
		ShouldReconnectOnError: true,
 | 
			
		||||
		ShardID:                0,
 | 
			
		||||
		ShardCount:             1,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If no arguments are passed return the empty Session interface.
 | 
			
		||||
	if args == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Variables used below when parsing func arguments
 | 
			
		||||
	var auth, pass string
 | 
			
		||||
 | 
			
		||||
	// Parse passed arguments
 | 
			
		||||
	for _, arg := range args {
 | 
			
		||||
 | 
			
		||||
		switch v := arg.(type) {
 | 
			
		||||
 | 
			
		||||
		case []string:
 | 
			
		||||
			if len(v) > 3 {
 | 
			
		||||
				err = fmt.Errorf("Too many string parameters provided.")
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			// First string is either token or username
 | 
			
		||||
			if len(v) > 0 {
 | 
			
		||||
				auth = v[0]
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			// If second string exists, it must be a password.
 | 
			
		||||
			if len(v) > 1 {
 | 
			
		||||
				pass = v[1]
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			// If third string exists, it must be an auth token.
 | 
			
		||||
			if len(v) > 2 {
 | 
			
		||||
				s.Token = v[2]
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
		case string:
 | 
			
		||||
			// First string must be either auth token or username.
 | 
			
		||||
			// Second string must be a password.
 | 
			
		||||
			// Only 2 input strings are supported.
 | 
			
		||||
 | 
			
		||||
			if auth == "" {
 | 
			
		||||
				auth = v
 | 
			
		||||
			} else if pass == "" {
 | 
			
		||||
				pass = v
 | 
			
		||||
			} else if s.Token == "" {
 | 
			
		||||
				s.Token = v
 | 
			
		||||
			} else {
 | 
			
		||||
				err = fmt.Errorf("Too many string parameters provided.")
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			//		case Config:
 | 
			
		||||
			// TODO: Parse configuration struct
 | 
			
		||||
 | 
			
		||||
		default:
 | 
			
		||||
			err = fmt.Errorf("Unsupported parameter type provided.")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If only one string was provided, assume it is an auth token.
 | 
			
		||||
	// Otherwise get auth token from Discord, if a token was specified
 | 
			
		||||
	// Discord will verify it for free, or log the user in if it is
 | 
			
		||||
	// invalid.
 | 
			
		||||
	if pass == "" {
 | 
			
		||||
		s.Token = auth
 | 
			
		||||
	} else {
 | 
			
		||||
		err = s.Login(auth, pass)
 | 
			
		||||
		if err != nil || s.Token == "" {
 | 
			
		||||
			err = fmt.Errorf("Unable to fetch discord authentication token. %v", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// The Session is now able to have RestAPI methods called on it.
 | 
			
		||||
	// It is recommended that you now call Open() so that events will trigger.
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// validateHandler takes an event handler func, and returns the type of event.
 | 
			
		||||
// eg.
 | 
			
		||||
//     Session.validateHandler(func (s *discordgo.Session, m *discordgo.MessageCreate))
 | 
			
		||||
//     will return the reflect.Type of *discordgo.MessageCreate
 | 
			
		||||
func (s *Session) validateHandler(handler interface{}) reflect.Type {
 | 
			
		||||
 | 
			
		||||
	handlerType := reflect.TypeOf(handler)
 | 
			
		||||
 | 
			
		||||
	if handlerType.NumIn() != 2 {
 | 
			
		||||
		panic("Unable to add event handler, handler must be of the type func(*discordgo.Session, *discordgo.EventType).")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if handlerType.In(0) != reflect.TypeOf(s) {
 | 
			
		||||
		panic("Unable to add event handler, first argument must be of type *discordgo.Session.")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	eventType := handlerType.In(1)
 | 
			
		||||
 | 
			
		||||
	// Support handlers of type interface{}, this is a special handler, which is triggered on every event.
 | 
			
		||||
	if eventType.Kind() == reflect.Interface {
 | 
			
		||||
		eventType = nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return eventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AddHandler allows you to add an event handler that will be fired anytime
 | 
			
		||||
// the Discord WSAPI event that matches the interface fires.
 | 
			
		||||
// eventToInterface in events.go has a list of all the Discord WSAPI events
 | 
			
		||||
// and their respective interface.
 | 
			
		||||
// eg:
 | 
			
		||||
//     Session.AddHandler(func(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
//     })
 | 
			
		||||
//
 | 
			
		||||
// or:
 | 
			
		||||
//     Session.AddHandler(func(s *discordgo.Session, m *discordgo.PresenceUpdate) {
 | 
			
		||||
//     })
 | 
			
		||||
// The return value of this method is a function, that when called will remove the
 | 
			
		||||
// event handler.
 | 
			
		||||
func (s *Session) AddHandler(handler interface{}) func() {
 | 
			
		||||
 | 
			
		||||
	s.initialize()
 | 
			
		||||
 | 
			
		||||
	eventType := s.validateHandler(handler)
 | 
			
		||||
 | 
			
		||||
	s.handlersMu.Lock()
 | 
			
		||||
	defer s.handlersMu.Unlock()
 | 
			
		||||
 | 
			
		||||
	h := reflect.ValueOf(handler)
 | 
			
		||||
 | 
			
		||||
	s.handlers[eventType] = append(s.handlers[eventType], h)
 | 
			
		||||
 | 
			
		||||
	// This must be done as we need a consistent reference to the
 | 
			
		||||
	// reflected value, otherwise a RemoveHandler method would have
 | 
			
		||||
	// been nice.
 | 
			
		||||
	return func() {
 | 
			
		||||
		s.handlersMu.Lock()
 | 
			
		||||
		defer s.handlersMu.Unlock()
 | 
			
		||||
 | 
			
		||||
		handlers := s.handlers[eventType]
 | 
			
		||||
		for i, v := range handlers {
 | 
			
		||||
			if h == v {
 | 
			
		||||
				s.handlers[eventType] = append(handlers[:i], handlers[i+1:]...)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handle calls any handlers that match the event type and any handlers of
 | 
			
		||||
// interface{}.
 | 
			
		||||
func (s *Session) handle(event interface{}) {
 | 
			
		||||
 | 
			
		||||
	s.handlersMu.RLock()
 | 
			
		||||
	defer s.handlersMu.RUnlock()
 | 
			
		||||
 | 
			
		||||
	if s.handlers == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	handlerParameters := []reflect.Value{reflect.ValueOf(s), reflect.ValueOf(event)}
 | 
			
		||||
 | 
			
		||||
	if handlers, ok := s.handlers[nil]; ok {
 | 
			
		||||
		for _, handler := range handlers {
 | 
			
		||||
			go handler.Call(handlerParameters)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if handlers, ok := s.handlers[reflect.TypeOf(event)]; ok {
 | 
			
		||||
		for _, handler := range handlers {
 | 
			
		||||
			go handler.Call(handlerParameters)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// initialize adds all internal handlers and state tracking handlers.
 | 
			
		||||
func (s *Session) initialize() {
 | 
			
		||||
 | 
			
		||||
	s.log(LogInformational, "called")
 | 
			
		||||
 | 
			
		||||
	s.handlersMu.Lock()
 | 
			
		||||
	if s.handlers != nil {
 | 
			
		||||
		s.handlersMu.Unlock()
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	s.handlers = map[interface{}][]reflect.Value{}
 | 
			
		||||
	s.handlersMu.Unlock()
 | 
			
		||||
 | 
			
		||||
	s.AddHandler(s.onReady)
 | 
			
		||||
	s.AddHandler(s.onResumed)
 | 
			
		||||
	s.AddHandler(s.onVoiceServerUpdate)
 | 
			
		||||
	s.AddHandler(s.onVoiceStateUpdate)
 | 
			
		||||
	s.AddHandler(s.State.onInterface)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onReady handles the ready event.
 | 
			
		||||
func (s *Session) onReady(se *Session, r *Ready) {
 | 
			
		||||
 | 
			
		||||
	// Store the SessionID within the Session struct.
 | 
			
		||||
	s.sessionID = r.SessionID
 | 
			
		||||
 | 
			
		||||
	// Start the heartbeat to keep the connection alive.
 | 
			
		||||
	go s.heartbeat(s.wsConn, s.listening, r.HeartbeatInterval)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onResumed handles the resumed event.
 | 
			
		||||
func (s *Session) onResumed(se *Session, r *Resumed) {
 | 
			
		||||
 | 
			
		||||
	// Start the heartbeat to keep the connection alive.
 | 
			
		||||
	go s.heartbeat(s.wsConn, s.listening, r.HeartbeatInterval)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										99
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										99
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,99 @@
 | 
			
		||||
// Discordgo - Discord bindings for Go
 | 
			
		||||
// Available at https://github.com/bwmarrin/discordgo
 | 
			
		||||
 | 
			
		||||
// Copyright 2015-2016 Bruce Marriner <bruce@sqls.net>.  All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// This file contains variables for all known Discord end points.  All functions
 | 
			
		||||
// throughout the Discordgo package use these variables for all connections
 | 
			
		||||
// to Discord.  These are all exported and you may modify them if needed.
 | 
			
		||||
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
// Known Discord API Endpoints.
 | 
			
		||||
var (
 | 
			
		||||
	EndpointStatus     = "https://status.discordapp.com/api/v2/"
 | 
			
		||||
	EndpointSm         = EndpointStatus + "scheduled-maintenances/"
 | 
			
		||||
	EndpointSmActive   = EndpointSm + "active.json"
 | 
			
		||||
	EndpointSmUpcoming = EndpointSm + "upcoming.json"
 | 
			
		||||
 | 
			
		||||
	EndpointDiscord  = "https://discordapp.com/"
 | 
			
		||||
	EndpointAPI      = EndpointDiscord + "api/"
 | 
			
		||||
	EndpointGuilds   = EndpointAPI + "guilds/"
 | 
			
		||||
	EndpointChannels = EndpointAPI + "channels/"
 | 
			
		||||
	EndpointUsers    = EndpointAPI + "users/"
 | 
			
		||||
	EndpointGateway  = EndpointAPI + "gateway"
 | 
			
		||||
 | 
			
		||||
	EndpointAuth           = EndpointAPI + "auth/"
 | 
			
		||||
	EndpointLogin          = EndpointAuth + "login"
 | 
			
		||||
	EndpointLogout         = EndpointAuth + "logout"
 | 
			
		||||
	EndpointVerify         = EndpointAuth + "verify"
 | 
			
		||||
	EndpointVerifyResend   = EndpointAuth + "verify/resend"
 | 
			
		||||
	EndpointForgotPassword = EndpointAuth + "forgot"
 | 
			
		||||
	EndpointResetPassword  = EndpointAuth + "reset"
 | 
			
		||||
	EndpointRegister       = EndpointAuth + "register"
 | 
			
		||||
 | 
			
		||||
	EndpointVoice        = EndpointAPI + "/voice/"
 | 
			
		||||
	EndpointVoiceRegions = EndpointVoice + "regions"
 | 
			
		||||
	EndpointVoiceIce     = EndpointVoice + "ice"
 | 
			
		||||
 | 
			
		||||
	EndpointTutorial           = EndpointAPI + "tutorial/"
 | 
			
		||||
	EndpointTutorialIndicators = EndpointTutorial + "indicators"
 | 
			
		||||
 | 
			
		||||
	EndpointTrack        = EndpointAPI + "track"
 | 
			
		||||
	EndpointSso          = EndpointAPI + "sso"
 | 
			
		||||
	EndpointReport       = EndpointAPI + "report"
 | 
			
		||||
	EndpointIntegrations = EndpointAPI + "integrations"
 | 
			
		||||
 | 
			
		||||
	EndpointUser              = func(uID string) string { return EndpointUsers + uID }
 | 
			
		||||
	EndpointUserAvatar        = func(uID, aID string) string { return EndpointUsers + uID + "/avatars/" + aID + ".jpg" }
 | 
			
		||||
	EndpointUserSettings      = func(uID string) string { return EndpointUsers + uID + "/settings" }
 | 
			
		||||
	EndpointUserGuilds        = func(uID string) string { return EndpointUsers + uID + "/guilds" }
 | 
			
		||||
	EndpointUserGuild         = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID }
 | 
			
		||||
	EndpointUserGuildSettings = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID + "/settings" }
 | 
			
		||||
	EndpointUserChannels      = func(uID string) string { return EndpointUsers + uID + "/channels" }
 | 
			
		||||
	EndpointUserDevices       = func(uID string) string { return EndpointUsers + uID + "/devices" }
 | 
			
		||||
	EndpointUserConnections   = func(uID string) string { return EndpointUsers + uID + "/connections" }
 | 
			
		||||
 | 
			
		||||
	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID }
 | 
			
		||||
	EndpointGuildInivtes         = func(gID string) string { return EndpointGuilds + gID + "/invites" }
 | 
			
		||||
	EndpointGuildChannels        = func(gID string) string { return EndpointGuilds + gID + "/channels" }
 | 
			
		||||
	EndpointGuildMembers         = func(gID string) string { return EndpointGuilds + gID + "/members" }
 | 
			
		||||
	EndpointGuildMember          = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID }
 | 
			
		||||
	EndpointGuildBans            = func(gID string) string { return EndpointGuilds + gID + "/bans" }
 | 
			
		||||
	EndpointGuildBan             = func(gID, uID string) string { return EndpointGuilds + gID + "/bans/" + uID }
 | 
			
		||||
	EndpointGuildIntegrations    = func(gID string) string { return EndpointGuilds + gID + "/integrations" }
 | 
			
		||||
	EndpointGuildIntegration     = func(gID, iID string) string { return EndpointGuilds + gID + "/integrations/" + iID }
 | 
			
		||||
	EndpointGuildIntegrationSync = func(gID, iID string) string { return EndpointGuilds + gID + "/integrations/" + iID + "/sync" }
 | 
			
		||||
	EndpointGuildRoles           = func(gID string) string { return EndpointGuilds + gID + "/roles" }
 | 
			
		||||
	EndpointGuildRole            = func(gID, rID string) string { return EndpointGuilds + gID + "/roles/" + rID }
 | 
			
		||||
	EndpointGuildInvites         = func(gID string) string { return EndpointGuilds + gID + "/invites" }
 | 
			
		||||
	EndpointGuildEmbed           = func(gID string) string { return EndpointGuilds + gID + "/embed" }
 | 
			
		||||
	EndpointGuildPrune           = func(gID string) string { return EndpointGuilds + gID + "/prune" }
 | 
			
		||||
	EndpointGuildIcon            = func(gID, hash string) string { return EndpointGuilds + gID + "/icons/" + hash + ".jpg" }
 | 
			
		||||
	EndpointGuildSplash          = func(gID, hash string) string { return EndpointGuilds + gID + "/splashes/" + hash + ".jpg" }
 | 
			
		||||
 | 
			
		||||
	EndpointChannel                   = func(cID string) string { return EndpointChannels + cID }
 | 
			
		||||
	EndpointChannelPermissions        = func(cID string) string { return EndpointChannels + cID + "/permissions" }
 | 
			
		||||
	EndpointChannelPermission         = func(cID, tID string) string { return EndpointChannels + cID + "/permissions/" + tID }
 | 
			
		||||
	EndpointChannelInvites            = func(cID string) string { return EndpointChannels + cID + "/invites" }
 | 
			
		||||
	EndpointChannelTyping             = func(cID string) string { return EndpointChannels + cID + "/typing" }
 | 
			
		||||
	EndpointChannelMessages           = func(cID string) string { return EndpointChannels + cID + "/messages" }
 | 
			
		||||
	EndpointChannelMessage            = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID }
 | 
			
		||||
	EndpointChannelMessageAck         = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" }
 | 
			
		||||
	EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk_delete" }
 | 
			
		||||
	EndpointChannelMessagesPins       = func(cID string) string { return EndpointChannel(cID) + "/pins" }
 | 
			
		||||
	EndpointChannelMessagePin         = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID }
 | 
			
		||||
 | 
			
		||||
	EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID }
 | 
			
		||||
 | 
			
		||||
	EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" }
 | 
			
		||||
 | 
			
		||||
	EndpointEmoji = func(eID string) string { return EndpointAPI + "emojis/" + eID + ".png" }
 | 
			
		||||
 | 
			
		||||
	EndpointOauth2          = EndpointAPI + "oauth2/"
 | 
			
		||||
	EndpointApplications    = EndpointOauth2 + "applications"
 | 
			
		||||
	EndpointApplication     = func(aID string) string { return EndpointApplications + "/" + aID }
 | 
			
		||||
	EndpointApplicationsBot = func(aID string) string { return EndpointApplications + "/" + aID + "/bot" }
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										159
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										159
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,159 @@
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
// eventToInterface is a mapping of Discord WSAPI events to their
 | 
			
		||||
// DiscordGo event container.
 | 
			
		||||
// Each Discord WSAPI event maps to a unique interface.
 | 
			
		||||
// Use Session.AddHandler with one of these types to handle that
 | 
			
		||||
// type of event.
 | 
			
		||||
// eg:
 | 
			
		||||
//     Session.AddHandler(func(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
//     })
 | 
			
		||||
//
 | 
			
		||||
// or:
 | 
			
		||||
//     Session.AddHandler(func(s *discordgo.Session, m *discordgo.PresenceUpdate) {
 | 
			
		||||
//     })
 | 
			
		||||
var eventToInterface = map[string]interface{}{
 | 
			
		||||
	"CHANNEL_CREATE":             ChannelCreate{},
 | 
			
		||||
	"CHANNEL_UPDATE":             ChannelUpdate{},
 | 
			
		||||
	"CHANNEL_DELETE":             ChannelDelete{},
 | 
			
		||||
	"GUILD_CREATE":               GuildCreate{},
 | 
			
		||||
	"GUILD_UPDATE":               GuildUpdate{},
 | 
			
		||||
	"GUILD_DELETE":               GuildDelete{},
 | 
			
		||||
	"GUILD_BAN_ADD":              GuildBanAdd{},
 | 
			
		||||
	"GUILD_BAN_REMOVE":           GuildBanRemove{},
 | 
			
		||||
	"GUILD_MEMBER_ADD":           GuildMemberAdd{},
 | 
			
		||||
	"GUILD_MEMBER_UPDATE":        GuildMemberUpdate{},
 | 
			
		||||
	"GUILD_MEMBER_REMOVE":        GuildMemberRemove{},
 | 
			
		||||
	"GUILD_ROLE_CREATE":          GuildRoleCreate{},
 | 
			
		||||
	"GUILD_ROLE_UPDATE":          GuildRoleUpdate{},
 | 
			
		||||
	"GUILD_ROLE_DELETE":          GuildRoleDelete{},
 | 
			
		||||
	"GUILD_INTEGRATIONS_UPDATE":  GuildIntegrationsUpdate{},
 | 
			
		||||
	"GUILD_EMOJIS_UPDATE":        GuildEmojisUpdate{},
 | 
			
		||||
	"MESSAGE_ACK":                MessageAck{},
 | 
			
		||||
	"MESSAGE_CREATE":             MessageCreate{},
 | 
			
		||||
	"MESSAGE_UPDATE":             MessageUpdate{},
 | 
			
		||||
	"MESSAGE_DELETE":             MessageDelete{},
 | 
			
		||||
	"PRESENCE_UPDATE":            PresenceUpdate{},
 | 
			
		||||
	"PRESENCES_REPLACE":          PresencesReplace{},
 | 
			
		||||
	"READY":                      Ready{},
 | 
			
		||||
	"USER_UPDATE":                UserUpdate{},
 | 
			
		||||
	"USER_SETTINGS_UPDATE":       UserSettingsUpdate{},
 | 
			
		||||
	"USER_GUILD_SETTINGS_UPDATE": UserGuildSettingsUpdate{},
 | 
			
		||||
	"TYPING_START":               TypingStart{},
 | 
			
		||||
	"VOICE_SERVER_UPDATE":        VoiceServerUpdate{},
 | 
			
		||||
	"VOICE_STATE_UPDATE":         VoiceStateUpdate{},
 | 
			
		||||
	"RESUMED":                    Resumed{},
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Connect is an empty struct for an event.
 | 
			
		||||
type Connect struct{}
 | 
			
		||||
 | 
			
		||||
// Disconnect is an empty struct for an event.
 | 
			
		||||
type Disconnect struct{}
 | 
			
		||||
 | 
			
		||||
// RateLimit is a struct for the RateLimited event
 | 
			
		||||
type RateLimit struct {
 | 
			
		||||
	*TooManyRequests
 | 
			
		||||
	URL string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// MessageCreate is a wrapper struct for an event.
 | 
			
		||||
type MessageCreate struct {
 | 
			
		||||
	*Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// MessageUpdate is a wrapper struct for an event.
 | 
			
		||||
type MessageUpdate struct {
 | 
			
		||||
	*Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// MessageDelete is a wrapper struct for an event.
 | 
			
		||||
type MessageDelete struct {
 | 
			
		||||
	*Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelCreate is a wrapper struct for an event.
 | 
			
		||||
type ChannelCreate struct {
 | 
			
		||||
	*Channel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelUpdate is a wrapper struct for an event.
 | 
			
		||||
type ChannelUpdate struct {
 | 
			
		||||
	*Channel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelDelete is a wrapper struct for an event.
 | 
			
		||||
type ChannelDelete struct {
 | 
			
		||||
	*Channel
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildCreate is a wrapper struct for an event.
 | 
			
		||||
type GuildCreate struct {
 | 
			
		||||
	*Guild
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildUpdate is a wrapper struct for an event.
 | 
			
		||||
type GuildUpdate struct {
 | 
			
		||||
	*Guild
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildDelete is a wrapper struct for an event.
 | 
			
		||||
type GuildDelete struct {
 | 
			
		||||
	*Guild
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildBanAdd is a wrapper struct for an event.
 | 
			
		||||
type GuildBanAdd struct {
 | 
			
		||||
	*GuildBan
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildBanRemove is a wrapper struct for an event.
 | 
			
		||||
type GuildBanRemove struct {
 | 
			
		||||
	*GuildBan
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildMemberAdd is a wrapper struct for an event.
 | 
			
		||||
type GuildMemberAdd struct {
 | 
			
		||||
	*Member
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildMemberUpdate is a wrapper struct for an event.
 | 
			
		||||
type GuildMemberUpdate struct {
 | 
			
		||||
	*Member
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildMemberRemove is a wrapper struct for an event.
 | 
			
		||||
type GuildMemberRemove struct {
 | 
			
		||||
	*Member
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildRoleCreate is a wrapper struct for an event.
 | 
			
		||||
type GuildRoleCreate struct {
 | 
			
		||||
	*GuildRole
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GuildRoleUpdate is a wrapper struct for an event.
 | 
			
		||||
type GuildRoleUpdate struct {
 | 
			
		||||
	*GuildRole
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// PresencesReplace is an array of Presences for an event.
 | 
			
		||||
type PresencesReplace []*Presence
 | 
			
		||||
 | 
			
		||||
// VoiceStateUpdate is a wrapper struct for an event.
 | 
			
		||||
type VoiceStateUpdate struct {
 | 
			
		||||
	*VoiceState
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// UserUpdate is a wrapper struct for an event.
 | 
			
		||||
type UserUpdate struct {
 | 
			
		||||
	*User
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// UserSettingsUpdate is a map for an event.
 | 
			
		||||
type UserSettingsUpdate map[string]interface{}
 | 
			
		||||
 | 
			
		||||
// UserGuildSettingsUpdate is a map for an event.
 | 
			
		||||
type UserGuildSettingsUpdate struct {
 | 
			
		||||
	*UserGuildSettings
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										186
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										186
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,186 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/binary"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flag.StringVar(&token, "t", "", "Account Token")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var token string
 | 
			
		||||
var buffer = make([][]byte, 0)
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	if token == "" {
 | 
			
		||||
		fmt.Println("No token provided. Please run: airhorn -t <bot token>")
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Load the sound file.
 | 
			
		||||
	err := loadSound()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error loading sound: ", err)
 | 
			
		||||
		fmt.Println("Please copy $GOPATH/src/github.com/bwmarrin/examples/airhorn/airhorn.dca to this directory.")
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided token.
 | 
			
		||||
	dg, err := discordgo.New(token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error creating Discord session: ", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Register ready as a callback for the ready events.
 | 
			
		||||
	dg.AddHandler(ready)
 | 
			
		||||
 | 
			
		||||
	// Register messageCreate as a callback for the messageCreate events.
 | 
			
		||||
	dg.AddHandler(messageCreate)
 | 
			
		||||
 | 
			
		||||
	// Register guildCreate as a callback for the guildCreate events.
 | 
			
		||||
	dg.AddHandler(guildCreate)
 | 
			
		||||
 | 
			
		||||
	// Open the websocket and begin listening.
 | 
			
		||||
	err = dg.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error opening Discord session: ", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Airhorn is now running.  Press CTRL-C to exit.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ready(s *discordgo.Session, event *discordgo.Ready) {
 | 
			
		||||
	// Set the playing status.
 | 
			
		||||
	_ = s.UpdateStatus(0, "!airhorn")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This function will be called (due to AddHandler above) every time a new
 | 
			
		||||
// message is created on any channel that the autenticated bot has access to.
 | 
			
		||||
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
	if strings.HasPrefix(m.Content, "!airhorn") {
 | 
			
		||||
		// Find the channel that the message came from.
 | 
			
		||||
		c, err := s.State.Channel(m.ChannelID)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			// Could not find channel.
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Find the guild for that channel.
 | 
			
		||||
		g, err := s.State.Guild(c.GuildID)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			// Could not find guild.
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Look for the message sender in that guilds current voice states.
 | 
			
		||||
		for _, vs := range g.VoiceStates {
 | 
			
		||||
			if vs.UserID == m.Author.ID {
 | 
			
		||||
				err = playSound(s, g.ID, vs.ChannelID)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					fmt.Println("Error playing sound:", err)
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This function will be called (due to AddHandler above) every time a new
 | 
			
		||||
// guild is joined.
 | 
			
		||||
func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) {
 | 
			
		||||
	if event.Guild.Unavailable != nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, channel := range event.Guild.Channels {
 | 
			
		||||
		if channel.ID == event.Guild.ID {
 | 
			
		||||
			_, _ = s.ChannelMessageSend(channel.ID, "Airhorn is ready! Type !airhorn while in a voice channel to play a sound.")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// loadSound attempts to load an encoded sound file from disk.
 | 
			
		||||
func loadSound() error {
 | 
			
		||||
	file, err := os.Open("airhorn.dca")
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error opening dca file :", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var opuslen int16
 | 
			
		||||
 | 
			
		||||
	for {
 | 
			
		||||
		// Read opus frame length from dca file.
 | 
			
		||||
		err = binary.Read(file, binary.LittleEndian, &opuslen)
 | 
			
		||||
 | 
			
		||||
		// If this is the end of the file, just return.
 | 
			
		||||
		if err == io.EOF || err == io.ErrUnexpectedEOF {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println("Error reading from dca file :", err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Read encoded pcm from dca file.
 | 
			
		||||
		InBuf := make([]byte, opuslen)
 | 
			
		||||
		err = binary.Read(file, binary.LittleEndian, &InBuf)
 | 
			
		||||
 | 
			
		||||
		// Should not be any end of file errors
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println("Error reading from dca file :", err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Append encoded pcm data to the buffer.
 | 
			
		||||
		buffer = append(buffer, InBuf)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// playSound plays the current buffer to the provided channel.
 | 
			
		||||
func playSound(s *discordgo.Session, guildID, channelID string) (err error) {
 | 
			
		||||
	// Join the provided voice channel.
 | 
			
		||||
	vc, err := s.ChannelVoiceJoin(guildID, channelID, false, true)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Sleep for a specified amount of time before playing the sound
 | 
			
		||||
	time.Sleep(250 * time.Millisecond)
 | 
			
		||||
 | 
			
		||||
	// Start speaking.
 | 
			
		||||
	_ = vc.Speaking(true)
 | 
			
		||||
 | 
			
		||||
	// Send the buffer data.
 | 
			
		||||
	for _, buff := range buffer {
 | 
			
		||||
		vc.OpusSend <- buff
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Stop speaking
 | 
			
		||||
	_ = vc.Speaking(false)
 | 
			
		||||
 | 
			
		||||
	// Sleep for a specificed amount of time before ending.
 | 
			
		||||
	time.Sleep(250 * time.Millisecond)
 | 
			
		||||
 | 
			
		||||
	// Disconnect from the provided voice channel.
 | 
			
		||||
	_ = vc.Disconnect()
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										98
									
								
								vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,98 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line options
 | 
			
		||||
var (
 | 
			
		||||
	Email    string
 | 
			
		||||
	Password string
 | 
			
		||||
	Token    string
 | 
			
		||||
	AppName  string
 | 
			
		||||
	DeleteID string
 | 
			
		||||
	ListOnly bool
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.StringVar(&DeleteID, "d", "", "Application ID to delete")
 | 
			
		||||
	flag.BoolVar(&ListOnly, "l", false, "List Applications Only")
 | 
			
		||||
	flag.StringVar(&AppName, "a", "", "App/Bot Name")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	var err error
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password, Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If -l set, only display a list of existing applications
 | 
			
		||||
	// for the given account.
 | 
			
		||||
	if ListOnly {
 | 
			
		||||
		aps, err2 := dg.Applications()
 | 
			
		||||
		if err2 != nil {
 | 
			
		||||
			fmt.Println("error fetching applications,", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		for k, v := range aps {
 | 
			
		||||
			fmt.Printf("%d : --------------------------------------\n", k)
 | 
			
		||||
			fmt.Printf("ID: %s\n", v.ID)
 | 
			
		||||
			fmt.Printf("Name: %s\n", v.Name)
 | 
			
		||||
			fmt.Printf("Secret: %s\n", v.Secret)
 | 
			
		||||
			fmt.Printf("Description: %s\n", v.Description)
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if -d set, delete the given Application
 | 
			
		||||
	if DeleteID != "" {
 | 
			
		||||
		err = dg.ApplicationDelete(DeleteID)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println("error deleting application,", err)
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Create a new application.
 | 
			
		||||
	ap := &discordgo.Application{}
 | 
			
		||||
	ap.Name = AppName
 | 
			
		||||
	ap, err = dg.ApplicationCreate(ap)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating new applicaiton,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Application created successfully:\n")
 | 
			
		||||
	fmt.Printf("ID: %s\n", ap.ID)
 | 
			
		||||
	fmt.Printf("Name: %s\n", ap.Name)
 | 
			
		||||
	fmt.Printf("Secret: %s\n\n", ap.Secret)
 | 
			
		||||
 | 
			
		||||
	// Create the bot account under the application we just created
 | 
			
		||||
	bot, err := dg.ApplicationBotCreate(ap.ID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating bot account,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Bot account created successfully.\n")
 | 
			
		||||
	fmt.Printf("ID: %s\n", bot.ID)
 | 
			
		||||
	fmt.Printf("Username: %s\n", bot.Username)
 | 
			
		||||
	fmt.Printf("Token: %s\n\n", bot.Token)
 | 
			
		||||
	fmt.Println("Please save the above posted info in a secure place.")
 | 
			
		||||
	fmt.Println("You will need that information to login with your bot account.")
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										73
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/localfile/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										73
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/localfile/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,73 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/base64"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Email       string
 | 
			
		||||
	Password    string
 | 
			
		||||
	Token       string
 | 
			
		||||
	Avatar      string
 | 
			
		||||
	BotID       string
 | 
			
		||||
	BotUsername string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.StringVar(&Avatar, "f", "./avatar.jpg", "Avatar File Name")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	// Use discordgo.New(Token) to just use a token for login.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password, Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	bot, err := dg.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error fetching the bot details,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	BotID = bot.ID
 | 
			
		||||
	BotUsername = bot.Username
 | 
			
		||||
	changeAvatar(dg)
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Bot is now running.  Press CTRL-C to exit.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Helper function to change the avatar
 | 
			
		||||
func changeAvatar(s *discordgo.Session) {
 | 
			
		||||
	img, err := ioutil.ReadFile(Avatar)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println(err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	base64 := base64.StdEncoding.EncodeToString(img)
 | 
			
		||||
 | 
			
		||||
	avatar := fmt.Sprintf("data:%s;base64,%s", http.DetectContentType(img), base64)
 | 
			
		||||
 | 
			
		||||
	_, err = s.UserUpdate("", "", BotUsername, avatar, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println(err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										86
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/url/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										86
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/url/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,86 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/base64"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Email       string
 | 
			
		||||
	Password    string
 | 
			
		||||
	Token       string
 | 
			
		||||
	URL         string
 | 
			
		||||
	BotID       string
 | 
			
		||||
	BotUsername string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.StringVar(&URL, "l", "http://bwmarrin.github.io/discordgo/img/discordgo.png", "Link to the avatar image")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	// Use discordgo.New(Token) to just use a token for login.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password, Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	bot, err := dg.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error fetching the bot details,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	BotID = bot.ID
 | 
			
		||||
	BotUsername = bot.Username
 | 
			
		||||
	changeAvatar(dg)
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Bot is now running.  Press CTRL-C to exit.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Helper function to change the avatar
 | 
			
		||||
func changeAvatar(s *discordgo.Session) {
 | 
			
		||||
 | 
			
		||||
	resp, err := http.Get(URL)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error retrieving the file, ", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	defer func() {
 | 
			
		||||
		_ = resp.Body.Close()
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	img, err := ioutil.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error reading the response, ", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	base64 := base64.StdEncoding.EncodeToString(img)
 | 
			
		||||
 | 
			
		||||
	avatar := fmt.Sprintf("data:%s;base64,%s", http.DetectContentType(img), base64)
 | 
			
		||||
 | 
			
		||||
	_, err = s.UserUpdate("", "", BotUsername, avatar, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error setting the avatar, ", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										33
									
								
								vendor/github.com/bwmarrin/discordgo/examples/mytoken/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								vendor/github.com/bwmarrin/discordgo/examples/mytoken/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,33 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Email    string
 | 
			
		||||
	Password string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Your Authentication Token is:\n\n%s\n", dg.Token)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										58
									
								
								vendor/github.com/bwmarrin/discordgo/examples/new_basic/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										58
									
								
								vendor/github.com/bwmarrin/discordgo/examples/new_basic/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,58 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Email    string
 | 
			
		||||
	Password string
 | 
			
		||||
	Token    string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	// Use discordgo.New(Token) to just use a token for login.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password, Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Register messageCreate as a callback for the messageCreate events.
 | 
			
		||||
	dg.AddHandler(messageCreate)
 | 
			
		||||
 | 
			
		||||
	// Open the websocket and begin listening.
 | 
			
		||||
	err = dg.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error opening connection,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Bot is now running.  Press CTRL-C to exit.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This function will be called (due to AddHandler above) every time a new
 | 
			
		||||
// message is created on any channel that the autenticated bot has access to.
 | 
			
		||||
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
 | 
			
		||||
	// Print message to stdout.
 | 
			
		||||
	fmt.Printf("%20s %20s %20s > %s\n", m.ChannelID, time.Now().Format(time.Stamp), m.Author.Username, m.Content)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										78
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										78
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,78 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Email    string
 | 
			
		||||
	Password string
 | 
			
		||||
	Token    string
 | 
			
		||||
	BotID    string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	// Create a new Discord session using the provided login information.
 | 
			
		||||
	dg, err := discordgo.New(Email, Password, Token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Get the account information.
 | 
			
		||||
	u, err := dg.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error obtaining account details,", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Store the account ID for later use.
 | 
			
		||||
	BotID = u.ID
 | 
			
		||||
 | 
			
		||||
	// Register messageCreate as a callback for the messageCreate events.
 | 
			
		||||
	dg.AddHandler(messageCreate)
 | 
			
		||||
 | 
			
		||||
	// Open the websocket and begin listening.
 | 
			
		||||
	err = dg.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error opening connection,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Bot is now running.  Press CTRL-C to exit.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This function will be called (due to AddHandler above) every time a new
 | 
			
		||||
// message is created on any channel that the autenticated bot has access to.
 | 
			
		||||
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
 | 
			
		||||
	// Ignore all messages created by the bot itself
 | 
			
		||||
	if m.Author.ID == BotID {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If the message is "ping" reply with "Pong!"
 | 
			
		||||
	if m.Content == "ping" {
 | 
			
		||||
		_, _ = s.ChannelMessageSend(m.ChannelID, "Pong!")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If the message is "pong" reply with "Ping!"
 | 
			
		||||
	if m.Content == "pong" {
 | 
			
		||||
		_, _ = s.ChannelMessageSend(m.ChannelID, "Ping!")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										95
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										95
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,95 @@
 | 
			
		||||
// Discordgo - Discord bindings for Go
 | 
			
		||||
// Available at https://github.com/bwmarrin/discordgo
 | 
			
		||||
 | 
			
		||||
// Copyright 2015-2016 Bruce Marriner <bruce@sqls.net>.  All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// This file contains code related to discordgo package logging
 | 
			
		||||
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"log"
 | 
			
		||||
	"runtime"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
 | 
			
		||||
	// LogError level is used for critical errors that could lead to data loss
 | 
			
		||||
	// or panic that would not be returned to a calling function.
 | 
			
		||||
	LogError int = iota
 | 
			
		||||
 | 
			
		||||
	// LogWarning level is used for very abnormal events and errors that are
 | 
			
		||||
	// also returend to a calling function.
 | 
			
		||||
	LogWarning
 | 
			
		||||
 | 
			
		||||
	// LogInformational level is used for normal non-error activity
 | 
			
		||||
	LogInformational
 | 
			
		||||
 | 
			
		||||
	// LogDebug level is for very detailed non-error activity.  This is
 | 
			
		||||
	// very spammy and will impact performance.
 | 
			
		||||
	LogDebug
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// msglog provides package wide logging consistancy for discordgo
 | 
			
		||||
// the format, a...  portion this command follows that of fmt.Printf
 | 
			
		||||
//   msgL   : LogLevel of the message
 | 
			
		||||
//   caller : 1 + the number of callers away from the message source
 | 
			
		||||
//   format : Printf style message format
 | 
			
		||||
//   a ...  : comma seperated list of values to pass
 | 
			
		||||
func msglog(msgL, caller int, format string, a ...interface{}) {
 | 
			
		||||
 | 
			
		||||
	pc, file, line, _ := runtime.Caller(caller)
 | 
			
		||||
 | 
			
		||||
	files := strings.Split(file, "/")
 | 
			
		||||
	file = files[len(files)-1]
 | 
			
		||||
 | 
			
		||||
	name := runtime.FuncForPC(pc).Name()
 | 
			
		||||
	fns := strings.Split(name, ".")
 | 
			
		||||
	name = fns[len(fns)-1]
 | 
			
		||||
 | 
			
		||||
	msg := fmt.Sprintf(format, a...)
 | 
			
		||||
 | 
			
		||||
	log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// helper function that wraps msglog for the Session struct
 | 
			
		||||
// This adds a check to insure the message is only logged
 | 
			
		||||
// if the session log level is equal or higher than the
 | 
			
		||||
// message log level
 | 
			
		||||
func (s *Session) log(msgL int, format string, a ...interface{}) {
 | 
			
		||||
 | 
			
		||||
	if msgL > s.LogLevel {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	msglog(msgL, 2, format, a...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// helper function that wraps msglog for the VoiceConnection struct
 | 
			
		||||
// This adds a check to insure the message is only logged
 | 
			
		||||
// if the voice connection log level is equal or higher than the
 | 
			
		||||
// message log level
 | 
			
		||||
func (v *VoiceConnection) log(msgL int, format string, a ...interface{}) {
 | 
			
		||||
 | 
			
		||||
	if msgL > v.LogLevel {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	msglog(msgL, 2, format, a...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// printJSON is a helper function to display JSON data in a easy to read format.
 | 
			
		||||
/* NOT USED ATM
 | 
			
		||||
func printJSON(body []byte) {
 | 
			
		||||
	var prettyJSON bytes.Buffer
 | 
			
		||||
	error := json.Indent(&prettyJSON, body, "", "\t")
 | 
			
		||||
	if error != nil {
 | 
			
		||||
		log.Print("JSON parse error: ", error)
 | 
			
		||||
	}
 | 
			
		||||
	log.Println(string(prettyJSON.Bytes()))
 | 
			
		||||
}
 | 
			
		||||
*/
 | 
			
		||||
							
								
								
									
										82
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,82 @@
 | 
			
		||||
// Discordgo - Discord bindings for Go
 | 
			
		||||
// Available at https://github.com/bwmarrin/discordgo
 | 
			
		||||
 | 
			
		||||
// Copyright 2015-2016 Bruce Marriner <bruce@sqls.net>.  All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
// This file contains code related to the Message struct
 | 
			
		||||
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"regexp"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// A Message stores all data related to a specific Discord message.
 | 
			
		||||
type Message struct {
 | 
			
		||||
	ID              string               `json:"id"`
 | 
			
		||||
	ChannelID       string               `json:"channel_id"`
 | 
			
		||||
	Content         string               `json:"content"`
 | 
			
		||||
	Timestamp       string               `json:"timestamp"`
 | 
			
		||||
	EditedTimestamp string               `json:"edited_timestamp"`
 | 
			
		||||
	MentionRoles    []string             `json:"mention_roles"`
 | 
			
		||||
	Tts             bool                 `json:"tts"`
 | 
			
		||||
	MentionEveryone bool                 `json:"mention_everyone"`
 | 
			
		||||
	Author          *User                `json:"author"`
 | 
			
		||||
	Attachments     []*MessageAttachment `json:"attachments"`
 | 
			
		||||
	Embeds          []*MessageEmbed      `json:"embeds"`
 | 
			
		||||
	Mentions        []*User              `json:"mentions"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// A MessageAttachment stores data for message attachments.
 | 
			
		||||
type MessageAttachment struct {
 | 
			
		||||
	ID       string `json:"id"`
 | 
			
		||||
	URL      string `json:"url"`
 | 
			
		||||
	ProxyURL string `json:"proxy_url"`
 | 
			
		||||
	Filename string `json:"filename"`
 | 
			
		||||
	Width    int    `json:"width"`
 | 
			
		||||
	Height   int    `json:"height"`
 | 
			
		||||
	Size     int    `json:"size"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// An MessageEmbed stores data for message embeds.
 | 
			
		||||
type MessageEmbed struct {
 | 
			
		||||
	URL         string `json:"url"`
 | 
			
		||||
	Type        string `json:"type"`
 | 
			
		||||
	Title       string `json:"title"`
 | 
			
		||||
	Description string `json:"description"`
 | 
			
		||||
	Thumbnail   *struct {
 | 
			
		||||
		URL      string `json:"url"`
 | 
			
		||||
		ProxyURL string `json:"proxy_url"`
 | 
			
		||||
		Width    int    `json:"width"`
 | 
			
		||||
		Height   int    `json:"height"`
 | 
			
		||||
	} `json:"thumbnail"`
 | 
			
		||||
	Provider *struct {
 | 
			
		||||
		URL  string `json:"url"`
 | 
			
		||||
		Name string `json:"name"`
 | 
			
		||||
	} `json:"provider"`
 | 
			
		||||
	Author *struct {
 | 
			
		||||
		URL  string `json:"url"`
 | 
			
		||||
		Name string `json:"name"`
 | 
			
		||||
	} `json:"author"`
 | 
			
		||||
	Video *struct {
 | 
			
		||||
		URL    string `json:"url"`
 | 
			
		||||
		Width  int    `json:"width"`
 | 
			
		||||
		Height int    `json:"height"`
 | 
			
		||||
	} `json:"video"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ContentWithMentionsReplaced will replace all @<id> mentions with the
 | 
			
		||||
// username of the mention.
 | 
			
		||||
func (m *Message) ContentWithMentionsReplaced() string {
 | 
			
		||||
	if m.Mentions == nil {
 | 
			
		||||
		return m.Content
 | 
			
		||||
	}
 | 
			
		||||
	content := m.Content
 | 
			
		||||
	for _, user := range m.Mentions {
 | 
			
		||||
		content = regexp.MustCompile(fmt.Sprintf("<@!?(%s)>", user.ID)).ReplaceAllString(content, "@"+user.Username)
 | 
			
		||||
	}
 | 
			
		||||
	return content
 | 
			
		||||
}
 | 
			
		||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user