Compare commits
	
		
			259 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | 7a16146304 | ||
|   | 3d3809a21b | ||
|   | 29465397dd | ||
|   | d300bb1735 | ||
|   | 2e703472f1 | ||
|   | 8fede90b9e | ||
|   | d128f157c4 | ||
|   | 4fcedabfd0 | ||
|   | 246c8e4f74 | ||
|   | 4d2207aba7 | ||
|   | 17b8b86d68 | ||
|   | fdb57230a3 | ||
|   | 7469732bbc | ||
|   | d1dd6c3440 | ||
|   | 02612c0061 | ||
|   | a4db63a773 | ||
|   | 035c2b906a | ||
|   | 6ea8be5749 | ||
|   | 36024d5439 | ||
|   | 8d52c98373 | ||
|   | b4a4eb0057 | ||
|   | b469c8ddbd | ||
|   | eee0036c7f | ||
|   | 89c66b9430 | ||
|   | bd38319d83 | ||
|   | 33dffd5ea8 | ||
|   | 57176dadd4 | ||
|   | dd449a8705 | ||
|   | 587ad9f41d | ||
|   | a16ad8bf3b | ||
|   | 1e0490bd36 | ||
|   | 8afc641f0c | ||
|   | 2e4d58cb92 | ||
|   | 02d7e2db65 | ||
|   | f935c573e9 | ||
|   | 4a25e66c00 | ||
|   | 95f4e3448e | ||
|   | eacb1c1771 | ||
|   | 07fd825349 | ||
|   | be15cc8a36 | ||
|   | 2f68519b3c | ||
|   | efe641f202 | ||
|   | 9bd663046a | ||
|   | 11b07f01ba | ||
|   | 6c2f370e6b | ||
|   | 936bccccd2 | ||
|   | c30ffeb81e | ||
|   | e05a323afd | ||
|   | 80895deae2 | ||
|   | eddc691fc9 | ||
|   | deb2d7194d | ||
|   | fd8cfb11fb | ||
|   | 9407aa4600 | ||
|   | 263b8da37d | ||
|   | b95988b4e2 | ||
|   | 35025e164a | ||
|   | 32bbab8518 | ||
|   | 84c0b745af | ||
|   | 8b286fb009 | ||
|   | 386fa58b67 | ||
|   | c5cfbc2297 | ||
|   | cd0a2beb11 | ||
|   | 73f01ad8d8 | ||
|   | 930b639cc9 | ||
|   | 58483ea70c | ||
|   | 072cac0347 | ||
|   | 956d7cf3f3 | ||
|   | 7558a2162e | ||
|   | 62b165c0b4 | ||
|   | fe258e1b67 | ||
|   | dc37232100 | ||
|   | 163f55f9c2 | ||
|   | 2d16fd085e | ||
|   | e1a5f5bca5 | ||
|   | 6e772ee189 | ||
|   | 2b0f178ba3 | ||
|   | 79e6c9fa6c | ||
|   | 1426ddec5f | ||
|   | 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 | 
							
								
								
									
										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)) | ||||||
| @@ -2,10 +2,10 @@ FROM alpine:edge | |||||||
| ENTRYPOINT ["/bin/matterbridge"] | ENTRYPOINT ["/bin/matterbridge"] | ||||||
|  |  | ||||||
| COPY . /go/src/github.com/42wim/matterbridge | COPY . /go/src/github.com/42wim/matterbridge | ||||||
| RUN apk update && apk add go git \ | RUN apk update && apk add go git gcc musl-dev ca-certificates \ | ||||||
|         && cd /go/src/github.com/42wim/matterbridge \ |         && cd /go/src/github.com/42wim/matterbridge \ | ||||||
|         && export GOPATH=/go \ |         && export GOPATH=/go \ | ||||||
|         && go get \ |         && go get \ | ||||||
|         && go build -o /bin/matterbridge \ |         && go build -x -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge \ | ||||||
|         && rm -rf /go \ |         && rm -rf /go \ | ||||||
|         && apk del --purge git 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. | ||||||
							
								
								
									
										231
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										231
									
								
								README.md
									
									
									
									
									
								
							| @@ -1,16 +1,50 @@ | |||||||
| # matterbridge | # matterbridge | ||||||
|  | [](https://gitter.im/42wim/matterbridge) [](https://webchat.freenode.net/?channels=matterbridgechat) [](https://discord.gg/AkKPtrQ) [](https://riot.im/app/#/room/#matterbridge:matrix.org) | ||||||
|  |  | ||||||
| Simple bridge between mattermost and IRC. Uses the in/outgoing webhooks.   |  | ||||||
| Relays public channel messages between mattermost and IRC.   |  | ||||||
|  |  | ||||||
| Requires mattermost 1.2.0+ | Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp) and Matrix with REST API. | ||||||
|  |  | ||||||
| There is also [matterbridge-plus] (https://github.com/42wim/matterbridge-plus) which uses the mattermost API and needs a dedicated user (bot). But requires no incoming/outgoing webhook setup.  | # Table of Contents | ||||||
|  |  * [Features](#features) | ||||||
|  |  * [Requirements](#requirements) | ||||||
|  |  * [Installing](#installing) | ||||||
|  |    * [Binaries](#binaries) | ||||||
|  |    * [Building](#building) | ||||||
|  |  * [Configuration](#configuration) | ||||||
|  |    * [Examples](#examples)  | ||||||
|  |  * [Running](#running) | ||||||
|  |    * [Docker](#docker) | ||||||
|  |  * [Changelog](#changelog) | ||||||
|  |  * [FAQ](#faq) | ||||||
|  |  * [Thanks](#thanks) | ||||||
|  |  | ||||||
| ## binaries | # Features | ||||||
| Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/tag/v0.4.2) | * Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp) and Matrix. Pick and mix. | ||||||
|  | * Matterbridge can also work with private groups on your mattermost/slack. | ||||||
|  | * 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). | ||||||
|  | * REST API to read/post messages to bridges (WIP). | ||||||
|  |  | ||||||
| ## building | # Requirements | ||||||
|  | Accounts to one of the supported bridges | ||||||
|  | * [Mattermost](https://github.com/mattermost/platform/) 3.5.x - 3.9.x | ||||||
|  | * [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) | ||||||
|  | * [Matrix](https://matrix.org) | ||||||
|  |  | ||||||
|  | # Installing | ||||||
|  | ## Binaries | ||||||
|  | Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/) | ||||||
|  | * Latest stable release [v0.12.0](https://github.com/42wim/matterbridge/releases/latest) | ||||||
|  |  | ||||||
|  | ## 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) | 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) | ||||||
|  |  | ||||||
| ``` | ``` | ||||||
| @@ -25,92 +59,121 @@ $ ls bin/ | |||||||
| matterbridge | matterbridge | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| ## running | # Configuration | ||||||
| 1) Copy the matterbridge.conf.sample to matterbridge.conf in the same directory as the matterbridge binary.   | * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | ||||||
| 2) Edit matterbridge.conf with the settings for your environment. See below for more config information.   | * [matterbridge.toml.simple](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.simple) for a simple example. | ||||||
| 3) Now you can run matterbridge.  |  | ||||||
|  |  | ||||||
|  | ## Examples | ||||||
|  | ### Bridge mattermost (off-topic) - irc (#testing) | ||||||
| ``` | ``` | ||||||
| Usage of matterbridge: | [irc] | ||||||
|   -conf="matterbridge.conf": config file |     [irc.freenode] | ||||||
| ``` |     Server="irc.freenode.net:6667" | ||||||
|  |     Nick="yourbotname" | ||||||
| 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 |  | ||||||
|  |  | ||||||
| ## config |  | ||||||
| ### matterbridge |  | ||||||
| matterbridge looks for matterbridge.conf 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" |  | ||||||
| UseSlackCircumfix=false |  | ||||||
| #Freenode nickserv |  | ||||||
| NickServNick="nickserv" |  | ||||||
| #Password for nickserv |  | ||||||
| NickServPassword="secret" |  | ||||||
| #Ignore the messages from these nicks. They will not be sent to mattermost |  | ||||||
| IgnoreNicks="ircspammer1 ircspammer2" |  | ||||||
|  |  | ||||||
| [mattermost] | [mattermost] | ||||||
| #url is your incoming webhook url (account settings - integrations - incoming webhooks) |     [mattermost.work] | ||||||
| url="http://mattermost.yourdomain.com/hooks/incomingwebhookkey"   |     useAPI=true | ||||||
| #port the bridge webserver will listen on |     Server="yourmattermostserver.tld" | ||||||
| port=9999 |     Team="yourteam" | ||||||
| #address the webserver will bind to |     Login="yourlogin" | ||||||
| BindAddress="0.0.0.0" |     Password="yourpass" | ||||||
| showjoinpart=true #show irc users joining and parting |     PrefixMessagesWithNick=true | ||||||
| #the token you get from the outgoing webhook in mattermost. If empty no token check will be done. |  | ||||||
| #if you use multiple IRC channel (see below, this must be empty!) |  | ||||||
| token=yourtokenfrommattermost |  | ||||||
| #disable certificate checking (selfsigned certificates) |  | ||||||
| #SkipTLSVerify=true |  | ||||||
| #whether to prefix messages from IRC to mattermost with the sender's nick. Useful if username overrides for incoming webhooks isn't enabled on the mattermost server |  | ||||||
| PrefixMessagesWithNick=false |  | ||||||
| #how to format the list of IRC nicks when displayed in mattermost. Possible options are "table" and "plain" |  | ||||||
| NickFormatter=plain |  | ||||||
| #how many nicks to list per row for formatters that support this |  | ||||||
| NicksPerRow=4 |  | ||||||
| #Ignore the messages from these nicks. They will not be sent to irc |  | ||||||
| IgnoreNicks="mmbot spammer2" |  | ||||||
|  |  | ||||||
| #multiple channel config | [[gateway]] | ||||||
| #token you can find in your outgoing webhook | name="mygateway" | ||||||
| [Token "outgoingwebhooktoken1"]  | enable=true | ||||||
| IRCChannel="#off-topic" |     [[gateway.inout]] | ||||||
| MMChannel="off-topic" |     account="irc.freenode" | ||||||
|  |     channel="#testing" | ||||||
|  |  | ||||||
| [Token "outgoingwebhooktoken2"] |     [[gateway.inout]] | ||||||
| IRCChannel="#testing" |     account="mattermost.work" | ||||||
| MMChannel="testing" |     channel="off-topic" | ||||||
|  |  | ||||||
| [general] |  | ||||||
| #request your API key on https://github.com/giphy/GiphyAPI. This is a public beta key |  | ||||||
| GiphyApiKey="dc6zaTOxFJmzC" |  | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| ### mattermost | ### Bridge slack (#general) - discord (general) | ||||||
| You'll have to configure the incoming en outgoing webhooks.  | ``` | ||||||
|  | [slack] | ||||||
|  | [slack.test] | ||||||
|  | useAPI=true | ||||||
|  | Token="yourslacktoken" | ||||||
|  | PrefixMessagesWithNick=true | ||||||
|  |  | ||||||
| * incoming webhooks | [discord] | ||||||
| Go to "account settings" - integrations - "incoming webhooks".   | [discord.test] | ||||||
| Choose a channel at "Add a new incoming webhook", this will create a webhook URL right below.   | Token="yourdiscordtoken" | ||||||
| This URL should be set in the matterbridge.conf in the [mattermost] section (see above)   | Server="yourdiscordservername" | ||||||
|  |  | ||||||
| * outgoing webhooks | [general] | ||||||
| Go to "account settings" - integrations - "outgoing webhooks".   | RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | ||||||
| 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) | [[gateway]] | ||||||
|  |     name = "mygateway" | ||||||
|  |     enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel="general" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account ="slack.test" | ||||||
|  |     channel = "general" | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | # Running | ||||||
|  | 1) Copy the matterbridge.toml.sample to matterbridge.toml  | ||||||
|  | 2) Edit matterbridge.toml with the settings for your environment.  | ||||||
|  | 3) Now you can run matterbridge.  (```./matterbridge```)    | ||||||
|  |  | ||||||
|  | (Matterbridge will only look for the config file in your current directory, if it isn't there specify -conf "/path/toyour/matterbridge.toml") | ||||||
|  |  | ||||||
|  | ``` | ||||||
|  | Usage of ./matterbridge: | ||||||
|  |   -conf string | ||||||
|  |         config file (default "matterbridge.toml") | ||||||
|  |   -debug | ||||||
|  |         enable debug | ||||||
|  |   -gops | ||||||
|  |         enable gops agent | ||||||
|  |   -version | ||||||
|  |         show version | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | ## Docker | ||||||
|  | Create your matterbridge.toml file locally eg in ```/tmp/matterbridge.toml``` | ||||||
|  | ``` | ||||||
|  | docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | # Changelog | ||||||
|  | See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md) | ||||||
|  |  | ||||||
|  | # 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 API version you'll need to: | ||||||
|  | * setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.toml. | ||||||
|  |  | ||||||
|  | Also look at the ```RemoteNickFormat``` setting. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | # Thanks | ||||||
|  | Matterbridge wouldn't exist without these libraries: | ||||||
|  | * discord - https://github.com/bwmarrin/discordgo | ||||||
|  | * echo - https://github.com/labstack/echo | ||||||
|  | * gitter - https://github.com/sromku/go-gitter | ||||||
|  | * gops - https://github.com/google/gops | ||||||
|  | * irc - https://github.com/thoj/go-ircevent | ||||||
|  | * mattermost - https://github.com/mattermost/platform | ||||||
|  | * matrix - https://github.com/matrix-org/gomatrix | ||||||
|  | * slack - https://github.com/nlopes/slack | ||||||
|  | * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | ||||||
|  | * xmpp - https://github.com/mattn/go-xmpp | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										91
									
								
								bridge/api/api.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										91
									
								
								bridge/api/api.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,91 @@ | |||||||
|  | package api | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/labstack/echo" | ||||||
|  | 	"github.com/zfjagann/golang-ring" | ||||||
|  | 	"net/http" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Api struct { | ||||||
|  | 	Config   *config.Protocol | ||||||
|  | 	Remote   chan config.Message | ||||||
|  | 	Account  string | ||||||
|  | 	Messages ring.Ring | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type ApiMessage struct { | ||||||
|  | 	Text     string `json:"text"` | ||||||
|  | 	Username string `json:"username"` | ||||||
|  | 	Avatar   string `json:"avatar"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var flog *log.Entry | ||||||
|  | var protocol = "api" | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	flog = log.WithFields(log.Fields{"module": protocol}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg config.Protocol, account string, c chan config.Message) *Api { | ||||||
|  | 	b := &Api{} | ||||||
|  | 	e := echo.New() | ||||||
|  | 	b.Messages = ring.Ring{} | ||||||
|  | 	b.Messages.SetCapacity(cfg.Buffer) | ||||||
|  | 	b.Config = &cfg | ||||||
|  | 	b.Account = account | ||||||
|  | 	b.Remote = c | ||||||
|  | 	e.GET("/api/messages", b.handleMessages) | ||||||
|  | 	e.POST("/api/message", b.handlePostMessage) | ||||||
|  | 	go func() { | ||||||
|  | 		flog.Fatal(e.Start(cfg.BindAddress)) | ||||||
|  | 	}() | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Api) Connect() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | func (b *Api) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  | func (b *Api) JoinChannel(channel string) error { | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Api) Send(msg config.Message) error { | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  | 	b.Messages.Enqueue(&msg) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Api) handlePostMessage(c echo.Context) error { | ||||||
|  | 	message := &ApiMessage{} | ||||||
|  | 	if err := c.Bind(message); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.Remote <- config.Message{ | ||||||
|  | 		Text:     message.Text, | ||||||
|  | 		Username: message.Username, | ||||||
|  | 		Channel:  "api", | ||||||
|  | 		Avatar:   message.Avatar, | ||||||
|  | 		Account:  b.Account, | ||||||
|  | 	} | ||||||
|  | 	return c.JSON(http.StatusOK, message) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Api) handleMessages(c echo.Context) error { | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  | 	for _, msg := range b.Messages.Values() { | ||||||
|  | 		c.JSONPretty(http.StatusOK, msg, " ") | ||||||
|  | 	} | ||||||
|  | 	b.Messages = ring.Ring{} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										111
									
								
								bridge/bridge.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										111
									
								
								bridge/bridge.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,111 @@ | |||||||
|  | package bridge | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/api" | ||||||
|  | 	"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/matrix" | ||||||
|  | 	"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" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  |  | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Bridger interface { | ||||||
|  | 	Send(msg config.Message) error | ||||||
|  | 	Connect() error | ||||||
|  | 	JoinChannel(channel string) error | ||||||
|  | 	Disconnect() error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Bridge struct { | ||||||
|  | 	Config config.Protocol | ||||||
|  | 	Bridger | ||||||
|  | 	Name     string | ||||||
|  | 	Account  string | ||||||
|  | 	Protocol string | ||||||
|  | 	Channels map[string]config.ChannelInfo | ||||||
|  | 	Joined   map[string]bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge { | ||||||
|  | 	b := new(Bridge) | ||||||
|  | 	b.Channels = make(map[string]config.ChannelInfo) | ||||||
|  | 	accInfo := strings.Split(bridge.Account, ".") | ||||||
|  | 	protocol := accInfo[0] | ||||||
|  | 	name := accInfo[1] | ||||||
|  | 	b.Name = name | ||||||
|  | 	b.Protocol = protocol | ||||||
|  | 	b.Account = bridge.Account | ||||||
|  | 	b.Joined = make(map[string]bool) | ||||||
|  |  | ||||||
|  | 	// 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) | ||||||
|  | 	case "matrix": | ||||||
|  | 		b.Config = cfg.Matrix[name] | ||||||
|  | 		b.Bridger = bmatrix.New(cfg.Matrix[name], bridge.Account, c) | ||||||
|  | 	case "api": | ||||||
|  | 		b.Config = cfg.Api[name] | ||||||
|  | 		b.Bridger = api.New(cfg.Api[name], bridge.Account, c) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) JoinChannels() error { | ||||||
|  | 	err := b.joinChannels(b.Channels, b.Joined) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | ||||||
|  | 	mychannel := "" | ||||||
|  | 	for ID, channel := range channels { | ||||||
|  | 		if !exists[ID] { | ||||||
|  | 			mychannel = channel.Name | ||||||
|  | 			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID) | ||||||
|  | 			if b.Protocol == "irc" && channel.Options.Key != "" { | ||||||
|  | 				log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||||
|  | 				mychannel = mychannel + " " + channel.Options.Key | ||||||
|  | 			} | ||||||
|  | 			err := b.JoinChannel(channel.Name) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			exists[ID] = true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										170
									
								
								bridge/config/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										170
									
								
								bridge/config/config.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,170 @@ | |||||||
|  | package config | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/BurntSushi/toml" | ||||||
|  | 	"log" | ||||||
|  | 	"os" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	EVENT_JOIN_LEAVE      = "join_leave" | ||||||
|  | 	EVENT_FAILURE         = "failure" | ||||||
|  | 	EVENT_REJOIN_CHANNELS = "rejoin_channels" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Message struct { | ||||||
|  | 	Text      string | ||||||
|  | 	Channel   string | ||||||
|  | 	Username  string | ||||||
|  | 	Avatar    string | ||||||
|  | 	Account   string | ||||||
|  | 	Event     string | ||||||
|  | 	Protocol  string | ||||||
|  | 	Timestamp time.Time | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type ChannelInfo struct { | ||||||
|  | 	Name        string | ||||||
|  | 	Account     string | ||||||
|  | 	Direction   string | ||||||
|  | 	ID          string | ||||||
|  | 	GID         map[string]bool | ||||||
|  | 	SameChannel map[string]bool | ||||||
|  | 	Options     ChannelOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Protocol struct { | ||||||
|  | 	BindAddress            string // mattermost, slack | ||||||
|  | 	Buffer                 int    // api | ||||||
|  | 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||||
|  | 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||||
|  | 	IconURL                string // mattermost, slack | ||||||
|  | 	IgnoreNicks            string // all protocols | ||||||
|  | 	Jid                    string // xmpp | ||||||
|  | 	Login                  string // mattermost, matrix | ||||||
|  | 	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,matrix | ||||||
|  | 	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 | ||||||
|  | 	MessageFormat          string // telegram | ||||||
|  | 	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, matrix | ||||||
|  | 	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 | ||||||
|  | 	SameChannel bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 { | ||||||
|  | 	Api                map[string]Protocol | ||||||
|  | 	IRC                map[string]Protocol | ||||||
|  | 	Mattermost         map[string]Protocol | ||||||
|  | 	Matrix             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 | ||||||
|  | } | ||||||
							
								
								
									
										218
									
								
								bridge/discord/discord.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										218
									
								
								bridge/discord/discord.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,218 @@ | |||||||
|  | package bdiscord | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type bdiscord struct { | ||||||
|  | 	c             *discordgo.Session | ||||||
|  | 	Config        *config.Protocol | ||||||
|  | 	Remote        chan config.Message | ||||||
|  | 	Account       string | ||||||
|  | 	Channels      []*discordgo.Channel | ||||||
|  | 	Nick          string | ||||||
|  | 	UseChannelID  bool | ||||||
|  | 	userMemberMap map[string]*discordgo.Member | ||||||
|  | 	guildID       string | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||||
|  | 	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) | ||||||
|  | 	b.c.AddHandler(b.memberUpdate) | ||||||
|  | 	b.c.AddHandler(b.messageUpdate) | ||||||
|  | 	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) | ||||||
|  | 			b.guildID = guild.ID | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Debugf("%#v", err) | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) Disconnect() error { | ||||||
|  | 	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) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | ||||||
|  | 	if b.Config.EditDisable { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	// only when message is actually edited | ||||||
|  | 	if m.Message.EditedTimestamp != "" { | ||||||
|  | 		flog.Debugf("Sending edit message") | ||||||
|  | 		m.Content = m.Content + b.Config.EditSuffix | ||||||
|  | 		b.messageCreate(s, (*discordgo.MessageCreate)(m)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 	} | ||||||
|  | 	username := b.getNick(m.Author) | ||||||
|  | 	if len(m.MentionRoles) > 0 { | ||||||
|  | 		m.Message.Content = b.replaceRoleMentions(m.Message.Content) | ||||||
|  | 	} | ||||||
|  | 	m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||||
|  | 	b.Remote <- config.Message{Username: 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) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||||
|  | 	b.Lock() | ||||||
|  | 	if _, ok := b.userMemberMap[m.Member.User.ID]; ok { | ||||||
|  | 		flog.Debugf("%s: memberupdate: user %s (nick %s) changes nick to %s", b.Account, m.Member.User.Username, b.userMemberMap[m.Member.User.ID].Nick, m.Member.Nick) | ||||||
|  | 	} | ||||||
|  | 	b.userMemberMap[m.Member.User.ID] = m.Member | ||||||
|  | 	b.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) getNick(user *discordgo.User) string { | ||||||
|  | 	var err error | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  | 	if _, ok := b.userMemberMap[user.ID]; ok { | ||||||
|  | 		if b.userMemberMap[user.ID].Nick != "" { | ||||||
|  | 			// only return if nick is set | ||||||
|  | 			return b.userMemberMap[user.ID].Nick | ||||||
|  | 		} | ||||||
|  | 		// otherwise return username | ||||||
|  | 		return user.Username | ||||||
|  | 	} | ||||||
|  | 	// if we didn't find nick, search for it | ||||||
|  | 	b.userMemberMap[user.ID], err = b.c.GuildMember(b.guildID, user.ID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return user.Username | ||||||
|  | 	} | ||||||
|  | 	// only return if nick is set | ||||||
|  | 	if b.userMemberMap[user.ID].Nick != "" { | ||||||
|  | 		return b.userMemberMap[user.ID].Nick | ||||||
|  | 	} | ||||||
|  | 	return user.Username | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) replaceRoleMentions(text string) string { | ||||||
|  | 	roles, err := b.c.GuildRoles(b.guildID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		flog.Debugf("%#v", string(err.(*discordgo.RESTError).ResponseBody)) | ||||||
|  | 		return text | ||||||
|  | 	} | ||||||
|  | 	for _, role := range roles { | ||||||
|  | 		text = strings.Replace(text, "<@&"+role.ID+">", "@"+role.Name, -1) | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) stripCustomoji(text string) string { | ||||||
|  | 	// <:doge:302803592035958784> | ||||||
|  | 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||||
|  | 	return re.ReplaceAllString(text, `$1`) | ||||||
|  | } | ||||||
							
								
								
									
										125
									
								
								bridge/gitter/gitter.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										125
									
								
								bridge/gitter/gitter.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,125 @@ | |||||||
|  | package bgitter | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/sromku/go-gitter" | ||||||
|  | 	"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) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bgitter) JoinChannel(channel string) error { | ||||||
|  | 	roomID, err := b.c.GetRoomId(channel) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel) | ||||||
|  | 	} | ||||||
|  | 	room, err := b.c.GetRoom(roomID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.Rooms = append(b.Rooms, *room) | ||||||
|  | 	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.Name) | ||||||
|  | 	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 | ||||||
|  | } | ||||||
| @@ -1,4 +1,4 @@ | |||||||
| package bridge | package birc | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
| 	"strings" | 	"strings" | ||||||
							
								
								
									
										272
									
								
								bridge/irc/irc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										272
									
								
								bridge/irc/irc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,272 @@ | |||||||
|  | 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 | ||||||
|  | 	} | ||||||
|  | 	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 { | ||||||
|  | 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | ||||||
|  | 	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) Disconnect() error { | ||||||
|  | 	//b.i.Disconnect() | ||||||
|  | 	close(b.Local) | ||||||
|  | 	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) | ||||||
|  | 	} | ||||||
|  | 	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("KICK", b.handleJoinPart) | ||||||
|  | 	i.AddCallback("*", b.handleOther) | ||||||
|  | 	// we are now fully connected | ||||||
|  | 	b.connected <- struct{}{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Birc) handleJoinPart(event *irc.Event) { | ||||||
|  | 	channel := event.Arguments[0] | ||||||
|  | 	if event.Code == "KICK" { | ||||||
|  | 		flog.Infof("Got kicked from %s by %s", channel, event.Nick) | ||||||
|  | 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if event.Code == "QUIT" { | ||||||
|  | 		if event.Nick == b.Nick && strings.Contains(event.Raw, "Ping timeout") { | ||||||
|  | 			flog.Infof("%s reconnecting ..", b.Account) | ||||||
|  | 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||||
|  | 	b.Remote <- config.Message{Username: "system", Text: event.Nick + " " + strings.ToLower(event.Code) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||||
|  | 	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()) | ||||||
|  | 		} | ||||||
|  | 	*/ | ||||||
|  | } | ||||||
							
								
								
									
										117
									
								
								bridge/matrix/matrix.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										117
									
								
								bridge/matrix/matrix.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,117 @@ | |||||||
|  | package bmatrix | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	matrix "github.com/matrix-org/gomatrix" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Bmatrix struct { | ||||||
|  | 	mc      *matrix.Client | ||||||
|  | 	Config  *config.Protocol | ||||||
|  | 	Remote  chan config.Message | ||||||
|  | 	Account string | ||||||
|  | 	UserID  string | ||||||
|  | 	RoomMap map[string]string | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var flog *log.Entry | ||||||
|  | var protocol = "matrix" | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	flog = log.WithFields(log.Fields{"module": protocol}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg config.Protocol, account string, c chan config.Message) *Bmatrix { | ||||||
|  | 	b := &Bmatrix{} | ||||||
|  | 	b.RoomMap = make(map[string]string) | ||||||
|  | 	b.Config = &cfg | ||||||
|  | 	b.Account = account | ||||||
|  | 	b.Remote = c | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) Connect() error { | ||||||
|  | 	var err error | ||||||
|  | 	flog.Infof("Connecting %s", b.Config.Server) | ||||||
|  | 	b.mc, err = matrix.NewClient(b.Config.Server, "", "") | ||||||
|  | 	if err != nil { | ||||||
|  | 		flog.Debugf("%#v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	resp, err := b.mc.Login(&matrix.ReqLogin{ | ||||||
|  | 		Type:     "m.login.password", | ||||||
|  | 		User:     b.Config.Login, | ||||||
|  | 		Password: b.Config.Password, | ||||||
|  | 	}) | ||||||
|  | 	if err != nil { | ||||||
|  | 		flog.Debugf("%#v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.mc.SetCredentials(resp.UserID, resp.AccessToken) | ||||||
|  | 	b.UserID = resp.UserID | ||||||
|  | 	flog.Info("Connection succeeded") | ||||||
|  | 	go b.handlematrix() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) JoinChannel(channel string) error { | ||||||
|  | 	resp, err := b.mc.JoinRoom(channel, "", nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.Lock() | ||||||
|  | 	b.RoomMap[resp.RoomID] = channel | ||||||
|  | 	b.Unlock() | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) Send(msg config.Message) error { | ||||||
|  | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	channel := b.getRoomID(msg.Channel) | ||||||
|  | 	flog.Debugf("Sending to channel %s", channel) | ||||||
|  | 	b.mc.SendText(channel, msg.Username+msg.Text) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) getRoomID(channel string) string { | ||||||
|  | 	b.RLock() | ||||||
|  | 	defer b.RUnlock() | ||||||
|  | 	for ID, name := range b.RoomMap { | ||||||
|  | 		if name == channel { | ||||||
|  | 			return ID | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  | func (b *Bmatrix) handlematrix() error { | ||||||
|  | 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||||
|  | 	syncer.OnEventType("m.room.message", func(ev *matrix.Event) { | ||||||
|  | 		if ev.Content["msgtype"].(string) == "m.text" && ev.Sender != b.UserID { | ||||||
|  | 			b.RLock() | ||||||
|  | 			channel, ok := b.RoomMap[ev.RoomID] | ||||||
|  | 			b.RUnlock() | ||||||
|  | 			if !ok { | ||||||
|  | 				flog.Debugf("Unknown room %s", ev.RoomID) | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) | ||||||
|  | 			b.Remote <- config.Message{Username: ev.Sender, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account} | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("Received: %#v", ev) | ||||||
|  | 	}) | ||||||
|  | 	go func() { | ||||||
|  | 		for { | ||||||
|  | 			if err := b.mc.Sync(); err != nil { | ||||||
|  | 				flog.Println("Sync() returned ", err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										176
									
								
								bridge/mattermost/mattermost.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										176
									
								
								bridge/mattermost/mattermost.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,176 @@ | |||||||
|  | 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.mc.StatusLoop() | ||||||
|  | 	} | ||||||
|  | 	go b.handleMatter() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmattermost) Disconnect() error { | ||||||
|  | 	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 { | ||||||
|  | 		message = nick + message | ||||||
|  | 	} | ||||||
|  | 	if !b.Config.UseAPI { | ||||||
|  | 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | ||||||
|  | 		matterMessage.IconURL = msg.Avatar | ||||||
|  | 		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 { | ||||||
|  | 		flog.Debugf("%#v", message.Raw.Data) | ||||||
|  | 		if message.Type == "system_join_leave" || | ||||||
|  | 			message.Type == "system_join_channel" || | ||||||
|  | 			message.Type == "system_leave_channel" { | ||||||
|  | 			flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||||
|  | 			b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		// do not post our own messages back to irc | ||||||
|  | 		// only listen to message from our team | ||||||
|  | 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited") && | ||||||
|  | 			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 message.Raw.Event == "post_edited" && !b.Config.EditDisable { | ||||||
|  | 				m.Text = message.Text + b.Config.EditSuffix | ||||||
|  | 			} | ||||||
|  | 			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 | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										87
									
								
								bridge/rocketchat/rocketchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										87
									
								
								bridge/rocketchat/rocketchat.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,87 @@ | |||||||
|  | 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) Disconnect() error { | ||||||
|  | 	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} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										282
									
								
								bridge/slack/slack.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										282
									
								
								bridge/slack/slack.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,282 @@ | |||||||
|  | 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) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) JoinChannel(channel string) error { | ||||||
|  | 	// we can only join channels using the API | ||||||
|  | 	if b.Config.UseAPI { | ||||||
|  | 		if strings.HasPrefix(b.Config.Token, "xoxb") { | ||||||
|  | 			// TODO check if bot has already joined channel | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		_, err := b.sc.JoinChannel(channel) | ||||||
|  | 		if err != nil { | ||||||
|  | 			if err.Error() != "name_taken" { | ||||||
|  | 				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) getChannelByID(ID string) (*slack.Channel, error) { | ||||||
|  | 	if b.channels == nil { | ||||||
|  | 		return nil, fmt.Errorf("%s: channel %s not found (no channels found)", b.Account, ID) | ||||||
|  | 	} | ||||||
|  | 	for _, channel := range b.channels { | ||||||
|  | 		if channel.ID == ID { | ||||||
|  | 			return &channel, nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, ID) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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) | ||||||
|  | 				if !b.Config.EditDisable && ev.SubMessage != nil { | ||||||
|  | 					flog.Debugf("SubMessage %#v", ev.SubMessage) | ||||||
|  | 					ev.User = ev.SubMessage.User | ||||||
|  | 					ev.Text = ev.SubMessage.Text + b.Config.EditSuffix | ||||||
|  | 				} | ||||||
|  | 				// use our own func because rtm.GetChannelInfo doesn't work for private channels | ||||||
|  | 				channel, err := b.getChannelByID(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 | ||||||
|  | 		if m.Username == "slackbot" { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		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 | ||||||
|  | } | ||||||
							
								
								
									
										64
									
								
								bridge/telegram/html.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								bridge/telegram/html.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,64 @@ | |||||||
|  | package btelegram | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"github.com/russross/blackfriday" | ||||||
|  | 	"html" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | 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 makeHTML(input string) string { | ||||||
|  | 	return string(blackfriday.Markdown([]byte(input), | ||||||
|  | 		&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)) | ||||||
|  | } | ||||||
							
								
								
									
										117
									
								
								bridge/telegram/telegram.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										117
									
								
								bridge/telegram/telegram.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,117 @@ | |||||||
|  | package btelegram | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"strconv" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | 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) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Btelegram) JoinChannel(channel string) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if b.Config.MessageFormat == "HTML" { | ||||||
|  | 		msg.Text = makeHTML(msg.Text) | ||||||
|  | 	} | ||||||
|  | 	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text) | ||||||
|  | 	if b.Config.MessageFormat == "HTML" { | ||||||
|  | 		m.ParseMode = tgbotapi.ModeHTML | ||||||
|  | 	} | ||||||
|  | 	_, err = b.c.Send(m) | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||||
|  | 	username := "" | ||||||
|  | 	text := "" | ||||||
|  | 	channel := "" | ||||||
|  | 	for update := range updates { | ||||||
|  | 		var message *tgbotapi.Message | ||||||
|  | 		// handle channels | ||||||
|  | 		if update.ChannelPost != nil { | ||||||
|  | 			message = update.ChannelPost | ||||||
|  | 		} | ||||||
|  | 		if update.EditedChannelPost != nil && !b.Config.EditDisable { | ||||||
|  | 			message = update.EditedChannelPost | ||||||
|  | 			message.Text = message.Text + b.Config.EditSuffix | ||||||
|  | 		} | ||||||
|  | 		// handle groups | ||||||
|  | 		if update.Message != nil { | ||||||
|  | 			message = update.Message | ||||||
|  | 		} | ||||||
|  | 		if update.EditedMessage != nil && !b.Config.EditDisable { | ||||||
|  | 			message = update.EditedMessage | ||||||
|  | 			message.Text = message.Text + b.Config.EditSuffix | ||||||
|  | 		} | ||||||
|  | 		if message.From != nil { | ||||||
|  | 			username = message.From.FirstName | ||||||
|  | 			if username == "" { | ||||||
|  | 				username = message.From.UserName | ||||||
|  | 			} | ||||||
|  | 			text = message.Text | ||||||
|  | 			channel = strconv.FormatInt(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} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										138
									
								
								bridge/xmpp/xmpp.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										138
									
								
								bridge/xmpp/xmpp.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,138 @@ | |||||||
|  | package bxmpp | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/mattn/go-xmpp" | ||||||
|  |  | ||||||
|  | 	"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) Disconnect() error { | ||||||
|  | 	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 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										245
									
								
								changelog.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										245
									
								
								changelog.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,245 @@ | |||||||
|  | # v0.12.0 | ||||||
|  | ## Changes | ||||||
|  | * general: edited messages are now being sent by default on discord/mattermost/telegram/slack. See "New Features" | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * general: add support for edited messages.  | ||||||
|  |   Add new keyword EditDisable (false/true), default false. Which means by default edited messages will be sent to other bridges. | ||||||
|  |   Add new keyword EditSuffix , default "". You can change this eg to "(edited)", this will be appended to every edit message. | ||||||
|  |  | ||||||
|  | ## Enhancements | ||||||
|  | * discord: Strip custom emoji metadata (discord). Closes #148 | ||||||
|  |  | ||||||
|  | # v0.11.0 | ||||||
|  | ## New features | ||||||
|  | * general: reusing the same account on multiple gateways now also reuses the connection. | ||||||
|  |   This is particuarly useful for irc. See #87 | ||||||
|  | * general: the Name is now REQUIRED and needs to be UNIQUE for each gateway configuration | ||||||
|  | * telegram:  Support edited messages (telegram). See #141 | ||||||
|  | * mattermost: Add support for showing/hiding join/leave messages from mattermost. Closes #147 | ||||||
|  | * mattermost: Reconnect on session removal/timeout (mattermost) | ||||||
|  | * irc:  Rejoin channel when kicked (irc). | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: Remove space after nick (mattermost). Closes #142 | ||||||
|  | * mattermost: Modify iconurl correctly (mattermost). | ||||||
|  | * irc: Fix join/leave regression (irc) | ||||||
|  |  | ||||||
|  | # v0.10.3 | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Allow bot tokens for now without warning (slack). Closes #140 (fixes user_is_bot message on channel join) | ||||||
|  |  | ||||||
|  | # v0.10.2 | ||||||
|  | ## New features | ||||||
|  | * general: gops agent added. Allows for more debugging. See #134 | ||||||
|  | * general: toml inline table support added for config file | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * all: vendored libs updated | ||||||
|  |  | ||||||
|  | ## Changes | ||||||
|  | * general: add more informative messages on startup | ||||||
|  |  | ||||||
|  | # v0.10.1 | ||||||
|  | ## Bugfix | ||||||
|  | * gitter: Fix sending messages on new channel join. | ||||||
|  |  | ||||||
|  | # v0.10.0 | ||||||
|  | ## New features | ||||||
|  | * matrix: New protocol support added (https://matrix.org) | ||||||
|  | * mattermost: works with mattermost release v3.7.0 | ||||||
|  | * discord: Replace role ids in mentions to role names (discord). Closes #133 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: Add ReadTimeout to close lingering connections (mattermost). See #125 | ||||||
|  | * gitter: Join rooms not already joined by the bot (gitter). See #135 | ||||||
|  | * general: Fail when bridge is unable to join a channel (general) | ||||||
|  |  | ||||||
|  | ## Changes | ||||||
|  | * telegram: Do not use HTML parsemode by default. Set ```MessageFormat="HTML"``` to use it. Closes #126 | ||||||
|  |  | ||||||
|  | # v0.9.3 | ||||||
|  | ## New features | ||||||
|  | * API: rest interface to read / post messages (see API section in matterbridge.toml.sample) | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: fix receiving messages from private channels #118 | ||||||
|  | * slack: fix echo when using webhooks #119 | ||||||
|  | * mattermost: reconnecting should work better now | ||||||
|  | * irc: keeps reconnecting (every 60 seconds) now after ping timeout/disconnects. | ||||||
|  |  | ||||||
|  | # v0.9.2 | ||||||
|  | ## New features | ||||||
|  | * slack: support private channels #118 | ||||||
|  |  | ||||||
|  | ## 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 | ||||||
|  |  | ||||||
|  | # 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" | ||||||
|  | ``` | ||||||
							
								
								
									
										277
									
								
								gateway/gateway.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										277
									
								
								gateway/gateway.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,277 @@ | |||||||
|  | package gateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Gateway struct { | ||||||
|  | 	*config.Config | ||||||
|  | 	MyConfig        *config.Gateway | ||||||
|  | 	Bridges         map[string]*bridge.Bridge | ||||||
|  | 	Channels        map[string]*config.ChannelInfo | ||||||
|  | 	ChannelOptions  map[string]config.ChannelOptions | ||||||
|  | 	Names           map[string]bool | ||||||
|  | 	Name            string | ||||||
|  | 	Message         chan config.Message | ||||||
|  | 	DestChannelFunc func(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg *config.Config) *Gateway { | ||||||
|  | 	gw := &Gateway{} | ||||||
|  | 	gw.Config = cfg | ||||||
|  | 	gw.Channels = make(map[string]*config.ChannelInfo) | ||||||
|  | 	gw.Message = make(chan config.Message) | ||||||
|  | 	gw.Bridges = make(map[string]*bridge.Bridge) | ||||||
|  | 	gw.Names = make(map[string]bool) | ||||||
|  | 	gw.DestChannelFunc = gw.getDestChannel | ||||||
|  | 	return gw | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||||
|  | 	for _, br := range gw.Bridges { | ||||||
|  | 		if br.Account == cfg.Account { | ||||||
|  | 			gw.mapChannelsToBridge(br) | ||||||
|  | 			err := br.JoinChannels() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||||
|  | 			} | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	log.Infof("Starting bridge: %s ", cfg.Account) | ||||||
|  | 	br := bridge.New(gw.Config, cfg, gw.Message) | ||||||
|  | 	gw.mapChannelsToBridge(br) | ||||||
|  | 	gw.Bridges[cfg.Account] = br | ||||||
|  | 	err := br.Connect() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||||
|  | 	} | ||||||
|  | 	err = br.JoinChannels() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | ||||||
|  | 	if gw.Names[cfg.Name] { | ||||||
|  | 		return fmt.Errorf("Gateway with name %s already exists", cfg.Name) | ||||||
|  | 	} | ||||||
|  | 	if cfg.Name == "" { | ||||||
|  | 		return fmt.Errorf("%s", "Gateway without name found") | ||||||
|  | 	} | ||||||
|  | 	log.Infof("Starting gateway: %s", cfg.Name) | ||||||
|  | 	gw.Names[cfg.Name] = true | ||||||
|  | 	gw.Name = cfg.Name | ||||||
|  | 	gw.MyConfig = cfg | ||||||
|  | 	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 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) { | ||||||
|  | 	for ID, channel := range gw.Channels { | ||||||
|  | 		if br.Account == channel.Account { | ||||||
|  | 			br.Channels[ID] = *channel | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) Start() error { | ||||||
|  | 	go gw.handleReceive() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) handleReceive() { | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case msg := <-gw.Message: | ||||||
|  | 			if msg.Event == config.EVENT_FAILURE { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						go gw.reconnectBridge(br) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if msg.Event == config.EVENT_REJOIN_CHANNELS { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						br.Joined = make(map[string]bool) | ||||||
|  | 						br.JoinChannels() | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if !gw.ignoreMessage(&msg) { | ||||||
|  | 				msg.Timestamp = time.Now() | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					gw.handleMessage(msg, br) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | ||||||
|  | 	br.Disconnect() | ||||||
|  | 	time.Sleep(time.Second * 5) | ||||||
|  | RECONNECT: | ||||||
|  | 	log.Infof("Reconnecting %s", br.Account) | ||||||
|  | 	err := br.Connect() | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err) | ||||||
|  | 		time.Sleep(time.Second * 60) | ||||||
|  | 		goto RECONNECT | ||||||
|  | 	} | ||||||
|  | 	br.Joined = make(map[string]bool) | ||||||
|  | 	br.JoinChannels() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) mapChannels() error { | ||||||
|  | 	for _, br := range append(gw.MyConfig.Out, gw.MyConfig.InOut...) { | ||||||
|  | 		ID := br.Channel + br.Account | ||||||
|  | 		_, ok := gw.Channels[ID] | ||||||
|  | 		if !ok { | ||||||
|  | 			channel := &config.ChannelInfo{Name: br.Channel, Direction: "out", ID: ID, Options: br.Options, Account: br.Account, | ||||||
|  | 				GID: make(map[string]bool), SameChannel: make(map[string]bool)} | ||||||
|  | 			channel.GID[gw.Name] = true | ||||||
|  | 			channel.SameChannel[gw.Name] = br.SameChannel | ||||||
|  | 			gw.Channels[channel.ID] = channel | ||||||
|  | 		} | ||||||
|  | 		gw.Channels[ID].GID[gw.Name] = true | ||||||
|  | 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, br := range append(gw.MyConfig.In, gw.MyConfig.InOut...) { | ||||||
|  | 		ID := br.Channel + br.Account | ||||||
|  | 		_, ok := gw.Channels[ID] | ||||||
|  | 		if !ok { | ||||||
|  | 			channel := &config.ChannelInfo{Name: br.Channel, Direction: "in", ID: ID, Options: br.Options, Account: br.Account, | ||||||
|  | 				GID: make(map[string]bool), SameChannel: make(map[string]bool)} | ||||||
|  | 			channel.GID[gw.Name] = true | ||||||
|  | 			channel.SameChannel[gw.Name] = br.SameChannel | ||||||
|  | 			gw.Channels[channel.ID] = channel | ||||||
|  | 		} | ||||||
|  | 		gw.Channels[ID].GID[gw.Name] = true | ||||||
|  | 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo { | ||||||
|  | 	var channels []config.ChannelInfo | ||||||
|  | 	for _, channel := range gw.Channels { | ||||||
|  | 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if channel.Direction == "out" && channel.Account == dest.Account && gw.validGatewayDest(*msg, channel) { | ||||||
|  | 			channels = append(channels, *channel) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return channels | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 	} | ||||||
|  | 	// broadcast to every out channel (irc QUIT) | ||||||
|  | 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | ||||||
|  | 		log.Debug("empty channel") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	originchannel := msg.Channel | ||||||
|  | 	for _, channel := range gw.DestChannelFunc(&msg, *dest) { | ||||||
|  | 		// do not send to ourself | ||||||
|  | 		if channel.ID == getChannelID(msg) { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		log.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name) | ||||||
|  | 		msg.Channel = channel.Name | ||||||
|  | 		gw.modifyAvatar(&msg, dest) | ||||||
|  | 		gw.modifyUsername(&msg, dest) | ||||||
|  | 		// for api we need originchannel as channel | ||||||
|  | 		if dest.Protocol == "api" { | ||||||
|  | 			msg.Channel = originchannel | ||||||
|  | 		} | ||||||
|  | 		err := dest.Send(msg) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Println(err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||||
|  | 	if msg.Text == "" { | ||||||
|  | 		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	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) modifyUsername(msg *config.Message, dest *bridge.Bridge) { | ||||||
|  | 	br := gw.Bridges[msg.Account] | ||||||
|  | 	msg.Protocol = br.Protocol | ||||||
|  | 	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 *Gateway) modifyAvatar(msg *config.Message, dest *bridge.Bridge) { | ||||||
|  | 	iconurl := gw.Config.General.IconURL | ||||||
|  | 	if iconurl == "" { | ||||||
|  | 		iconurl = dest.Config.IconURL | ||||||
|  | 	} | ||||||
|  | 	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1) | ||||||
|  | 	if msg.Avatar == "" { | ||||||
|  | 		msg.Avatar = iconurl | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func getChannelID(msg config.Message) string { | ||||||
|  | 	return msg.Channel + msg.Account | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) validGatewayDest(msg config.Message, channel *config.ChannelInfo) bool { | ||||||
|  | 	GIDmap := gw.Channels[getChannelID(msg)].GID | ||||||
|  | 	// check if we are running a samechannelgateway. | ||||||
|  | 	// if it is and the channel name matches it's ok, otherwise we shouldn't use this channel. | ||||||
|  | 	for k, _ := range GIDmap { | ||||||
|  | 		if channel.SameChannel[k] == true { | ||||||
|  | 			if msg.Channel == channel.Name { | ||||||
|  | 				return true | ||||||
|  | 			} else { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	// check if we are in the correct gateway | ||||||
|  | 	for k, _ := range GIDmap { | ||||||
|  | 		if channel.GID[k] == true { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
							
								
								
									
										28
									
								
								gateway/samechannel/samechannel.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								gateway/samechannel/samechannel.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | |||||||
|  | package samechannelgateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type SameChannelGateway struct { | ||||||
|  | 	*config.Config | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg *config.Config) *SameChannelGateway { | ||||||
|  | 	return &SameChannelGateway{Config: cfg} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (sgw *SameChannelGateway) GetConfig() []config.Gateway { | ||||||
|  | 	var gwconfigs []config.Gateway | ||||||
|  | 	cfg := sgw.Config | ||||||
|  | 	for _, gw := range cfg.SameChannelGateway { | ||||||
|  | 		gwconfig := config.Gateway{Name: gw.Name, Enable: gw.Enable} | ||||||
|  | 		for _, account := range gw.Accounts { | ||||||
|  | 			for _, channel := range gw.Channels { | ||||||
|  | 				gwconfig.InOut = append(gwconfig.InOut, config.Bridge{Account: account, Channel: channel, SameChannel: true}) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		gwconfigs = append(gwconfigs, gwconfig) | ||||||
|  | 	} | ||||||
|  | 	return gwconfigs | ||||||
|  | } | ||||||
							
								
								
									
										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,39 +0,0 @@ | |||||||
| [IRC] |  | ||||||
| server="irc.freenode.net" |  | ||||||
| port=6667 |  | ||||||
| UseTLS=false |  | ||||||
| SkipTLSVerify=true |  | ||||||
| nick="matterbot" |  | ||||||
| channel="#matterbridge" |  | ||||||
| UseSlackCircumfix=false |  | ||||||
| #NickServNick="nickserv" |  | ||||||
| #NickServPassword="secret" |  | ||||||
| IgnoreNicks="ircspammer1 ircspammer2" |  | ||||||
|  |  | ||||||
| [mattermost] |  | ||||||
| url="http://yourdomain/hooks/yourhookkey" |  | ||||||
| port=9999 |  | ||||||
| showjoinpart=true |  | ||||||
| #remove token when using multiple channels! |  | ||||||
| token=yourtokenfrommattermost |  | ||||||
| IconURL="http://youricon.png" |  | ||||||
| #SkipTLSVerify=true |  | ||||||
| #BindAddress="0.0.0.0" |  | ||||||
| PrefixMessagesWithNick=false |  | ||||||
| NickFormatter=plain |  | ||||||
| NicksPerRow=4 |  | ||||||
| IgnoreNicks="mmbot spammer2" |  | ||||||
|  |  | ||||||
| [general] |  | ||||||
| GiphyAPIKey=dc6zaTOxFJmzC |  | ||||||
|  |  | ||||||
| #multiple channel config |  | ||||||
| #token you can find in your outgoing webhook |  | ||||||
| [Token "outgoingwebhooktoken1"]  |  | ||||||
| IRCChannel="#off-topic" |  | ||||||
| MMChannel="off-topic" |  | ||||||
|  |  | ||||||
| [Token "outgoingwebhooktoken2"] |  | ||||||
| IRCChannel="#testing" |  | ||||||
| MMChannel="testing" |  | ||||||
|  |  | ||||||
| @@ -3,31 +3,64 @@ package main | |||||||
| import ( | import ( | ||||||
| 	"flag" | 	"flag" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge-plus/bridge" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/gateway" | ||||||
|  | 	"github.com/42wim/matterbridge/gateway/samechannel" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/Sirupsen/logrus" | ||||||
|  | 	"github.com/google/gops/agent" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var Version = "0.4.2" | var ( | ||||||
|  | 	version = "0.12.0" | ||||||
|  | 	githash string | ||||||
|  | ) | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
| 	flagConfig := flag.String("conf", "matterbridge.conf", "config file") | 	flagConfig := flag.String("conf", "matterbridge.toml", "config file") | ||||||
| 	flagDebug := flag.Bool("debug", false, "enable debug") | 	flagDebug := flag.Bool("debug", false, "enable debug") | ||||||
| 	flagVersion := flag.Bool("version", false, "show version") | 	flagVersion := flag.Bool("version", false, "show version") | ||||||
|  | 	flagGops := flag.Bool("gops", false, "enable gops agent") | ||||||
| 	flag.Parse() | 	flag.Parse() | ||||||
|  | 	if *flagGops { | ||||||
|  | 		agent.Listen(&agent.Options{}) | ||||||
|  | 		defer agent.Close() | ||||||
|  | 	} | ||||||
| 	if *flagVersion { | 	if *flagVersion { | ||||||
| 		fmt.Println("Version:", Version) | 		fmt.Printf("version: %s %s\n", version, githash) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flag.Parse() | 	flag.Parse() | ||||||
| 	if *flagDebug { | 	if *flagDebug { | ||||||
| 		log.Info("enabling debug") | 		log.Info("Enabling debug") | ||||||
| 		log.SetLevel(log.DebugLevel) | 		log.SetLevel(log.DebugLevel) | ||||||
| 	} | 	} | ||||||
| 	fmt.Println("running version", Version) | 	log.Printf("Running version %s %s", version, githash) | ||||||
| 	bridge.NewBridge("matterbot", bridge.NewConfig(*flagConfig), "legacy") | 	if strings.Contains(version, "-dev") { | ||||||
|  | 		log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||||
|  | 	} | ||||||
|  | 	cfg := config.NewConfig(*flagConfig) | ||||||
|  |  | ||||||
|  | 	g := gateway.New(cfg) | ||||||
|  | 	sgw := samechannelgateway.New(cfg) | ||||||
|  | 	gwconfigs := sgw.GetConfig() | ||||||
|  | 	for _, gw := range append(gwconfigs, cfg.Gateway...) { | ||||||
|  | 		if !gw.Enable { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		err := g.AddConfig(&gw) | ||||||
|  | 		if err != nil { | ||||||
|  | 			log.Fatalf("Starting gateway failed: %s", err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	err := g.Start() | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatalf("Starting gateway failed: %s", err) | ||||||
|  | 	} | ||||||
|  | 	log.Printf("Gateway(s) started succesfully. Now relaying messages") | ||||||
| 	select {} | 	select {} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										707
									
								
								matterbridge.toml.sample
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										707
									
								
								matterbridge.toml.sample
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,707 @@ | |||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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. (do not prefix it with http or https) | ||||||
|  | #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 | ||||||
|  |  | ||||||
|  | #Disable sending of edits to other bridges | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | EditDisable=false | ||||||
|  |  | ||||||
|  | #Message to be appended to every edited message | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | EditSuffix=" (edited)" | ||||||
|  |  | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 | ||||||
|  |  | ||||||
|  | #NOT RECOMMENDED TO USE INCOMING/OUTGOING WEBHOOK. USE SLACK API | ||||||
|  | #AND DEDICATED BOT USER WHEN POSSIBLE! | ||||||
|  | #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" | ||||||
|  |  | ||||||
|  | #NOT RECOMMENDED TO USE INCOMING/OUTGOING WEBHOOK. USE SLACK API | ||||||
|  | #AND DEDICATED BOT USER WHEN POSSIBLE! | ||||||
|  | #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 (RECOMMENDED) | ||||||
|  | #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. | ||||||
|  | #Use https://api.slack.com/custom-integrations/legacy-tokens | ||||||
|  | #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 | ||||||
|  |  | ||||||
|  | #Disable sending of edits to other bridges | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | EditDisable=true | ||||||
|  |  | ||||||
|  | #Message to be appended to every edited message | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | EditSuffix=" (edited)" | ||||||
|  |  | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 | ||||||
|  | #If you want roles/groups mentions to be shown with names instead of ID, you'll need to give your bot the "Manage Roles" permission. | ||||||
|  | #REQUIRED | ||||||
|  | Token="Yourtokenhere" | ||||||
|  |  | ||||||
|  | #REQUIRED | ||||||
|  | Server="yourservername" | ||||||
|  |  | ||||||
|  | #Disable sending of edits to other bridges | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | EditDisable=false | ||||||
|  |  | ||||||
|  | #Message to be appended to every edited message | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | EditSuffix=" (edited)" | ||||||
|  |  | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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" | ||||||
|  |  | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | #Only supported format is "HTML", messages will be sent in html parsemode. | ||||||
|  | #See https://core.telegram.org/bots/api#html-style | ||||||
|  | MessageFormat="" | ||||||
|  |  | ||||||
|  | #Disable sending of edits to other bridges | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | EditDisable=false | ||||||
|  |  | ||||||
|  | #Message to be appended to every edited message | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | EditSuffix=" (edited)" | ||||||
|  |  | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | ################################################################### | ||||||
|  | #matrix section | ||||||
|  | ################################################################### | ||||||
|  | [matrix] | ||||||
|  | #You can configure multiple servers "[matrix.name]" or "[matrix.name2]" | ||||||
|  | #In this example we use [matrix.neo] | ||||||
|  | #REQUIRED | ||||||
|  |  | ||||||
|  | [matrix.neo] | ||||||
|  | #Server is your homeserver (eg https://matrix.org) | ||||||
|  | #REQUIRED | ||||||
|  | Server="https://matrix.org" | ||||||
|  |  | ||||||
|  | #login/pass of your bot.  | ||||||
|  | #Use a dedicated user for this and not your own!  | ||||||
|  | #Messages sent from this user will not be relayed to avoid loops. | ||||||
|  | #REQUIRED  | ||||||
|  | Login="yourlogin" | ||||||
|  | Password="yourpass" | ||||||
|  |  | ||||||
|  | #Whether to prefix messages from other bridges to matrix with the sender's nick.  | ||||||
|  | #Useful if username overrides for incoming webhooks isn't enabled on the  | ||||||
|  | #matrix server. If you set PrefixMessagesWithNick to true, each message  | ||||||
|  | #from bridge to matrix 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="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 works hiding/show messages from irc and mattermost bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowJoinPart=false | ||||||
|  |  | ||||||
|  |  | ||||||
|  | ################################################################### | ||||||
|  | #API | ||||||
|  | ################################################################### | ||||||
|  | [api] | ||||||
|  | #You can configure multiple API hooks | ||||||
|  | #In this example we use [api.local] | ||||||
|  | #REQUIRED | ||||||
|  |  | ||||||
|  | [api.local] | ||||||
|  | #Address to listen on for API | ||||||
|  | #REQUIRED  | ||||||
|  | BindAddress="127.0.0.1:4242" | ||||||
|  |  | ||||||
|  | #Amount of messages to keep in memory | ||||||
|  | Buffer=1000 | ||||||
|  |  | ||||||
|  | #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="{NICK}" | ||||||
|  |  | ||||||
|  | ################################################################### | ||||||
|  | #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]] | ||||||
|  | #REQUIRED and UNIQUE | ||||||
|  | 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) | ||||||
|  |     #matrix     - #channel:server (eg #yourchannel:matrix.org) | ||||||
|  |     #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" | ||||||
|  |  | ||||||
|  |     #API example | ||||||
|  |     #[[gateway.inout]] | ||||||
|  |     #account="api.local" | ||||||
|  |     #channel="api" | ||||||
|  |     #To send data to the api: | ||||||
|  |     #curl -XPOST -H 'Content-Type: application/json'  -d '{"text":"test","username":"randomuser"}' http://localhost:4242/api/message | ||||||
|  |  | ||||||
|  | #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]] | ||||||
|  |    name="samechannel1" | ||||||
|  |    enable = false | ||||||
|  |    accounts = [ "mattermost.work","slack.hobby" ] | ||||||
|  |    channels = [ "testing","testing2","testing3"] | ||||||
							
								
								
									
										42
									
								
								matterbridge.toml.simple
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								matterbridge.toml.simple
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,42 @@ | |||||||
|  | [irc] | ||||||
|  |     [irc.freenode] | ||||||
|  |     Server="irc.freenode.net:6667" | ||||||
|  |     Nick="matterbot" | ||||||
|  |  | ||||||
|  | [mattermost] | ||||||
|  |     [mattermost.work] | ||||||
|  |     useAPI=true | ||||||
|  |     #do not prefix it wit http:// or https:// | ||||||
|  |     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" | ||||||
|  |  | ||||||
|  | #simpler config possible since v0.10.2 | ||||||
|  | #[[gateway]] | ||||||
|  | #name="gateway2" | ||||||
|  | #enable=true | ||||||
|  | #inout = [ | ||||||
|  | #    { account="irc.freenode", channel="#testing", options={key="channelkey"}}, | ||||||
|  | #    { account="mattermost.work", channel="off-topic" }, | ||||||
|  | #] | ||||||
							
								
								
									
										758
									
								
								matterclient/matterclient.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										758
									
								
								matterclient/matterclient.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,758 @@ | |||||||
|  | package matterclient | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net/http" | ||||||
|  | 	"net/http/cookiejar" | ||||||
|  | 	"net/url" | ||||||
|  | 	"strconv" | ||||||
|  | 	"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     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 | ||||||
|  | 	ServerVersion string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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}, Proxy: http.ProxyFromEnvironment} | ||||||
|  | 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		d := b.Duration() | ||||||
|  | 		// bogus call to get the serverversion | ||||||
|  | 		m.Client.GetClientProperties() | ||||||
|  | 		if firstConnection && !supportedVersion(m.Client.ServerVersion) { | ||||||
|  | 			return fmt.Errorf("unsupported mattermost version: %s", m.Client.ServerVersion) | ||||||
|  | 		} | ||||||
|  | 		m.ServerVersion = m.Client.ServerVersion | ||||||
|  | 		if m.ServerVersion == "" { | ||||||
|  | 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||||
|  | 			time.Sleep(d) | ||||||
|  | 		} else { | ||||||
|  | 			m.log.Infof("Found version %s", m.ServerVersion) | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	b.Reset() | ||||||
|  |  | ||||||
|  | 	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_V3 + "/users/websocket" | ||||||
|  | 	header := http.Header{} | ||||||
|  | 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||||
|  |  | ||||||
|  | 	m.log.Debugf("WsClient: making connection: %s", wsurl) | ||||||
|  | 	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.log.Debug("WsClient: connected") | ||||||
|  | 	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, model.WEBSOCKET_EVENT_POST_EDITED: | ||||||
|  | 		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.Type = data.Type | ||||||
|  | 	teamid, _ := rmsg.Raw.Data["team_id"].(string) | ||||||
|  | 	// edit messsages have no team_id for some reason | ||||||
|  | 	if teamid == "" { | ||||||
|  | 		// we can find the team_id from the channelid | ||||||
|  | 		result, _ := m.Client.GetChannel(data.ChannelId, "") | ||||||
|  | 		teamid = result.Data.(*model.ChannelData).Channel.TeamId | ||||||
|  | 		rmsg.Raw.Data["team_id"] = teamid | ||||||
|  | 	} | ||||||
|  | 	if teamid != "" { | ||||||
|  | 		rmsg.Team = m.GetTeamName(teamid) | ||||||
|  | 	} | ||||||
|  | 	// 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) | ||||||
|  | 	} | ||||||
|  | 	var mmchannels2 *model.Result | ||||||
|  | 	if m.mmVersion() >= 3.8 { | ||||||
|  | 		mmchannels2, err = m.Client.GetMoreChannelsPage(0, 5000) | ||||||
|  | 	} else { | ||||||
|  | 		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) | ||||||
|  | 	if m.mmVersion() >= 3.8 { | ||||||
|  | 		view := model.ChannelView{ChannelId: channelId} | ||||||
|  | 		res, _ := m.Client.ViewChannel(view) | ||||||
|  | 		if res == false { | ||||||
|  | 			m.log.Errorf("ChannelView update for %s failed", channelId) | ||||||
|  | 		} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	_, err := m.Client.UpdateLastViewedAt(channelId, true) | ||||||
|  | 	if err != nil { | ||||||
|  | 		m.log.Error(err) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) 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) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||||
|  |  | ||||||
|  | 	// update our channels | ||||||
|  | 	mmchannels, err := m.Client.GetChannels("") | ||||||
|  | 	if err != nil { | ||||||
|  | 		m.log.Debug("SendDirectMessage: Couldn't update channels") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	m.Lock() | ||||||
|  | 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) | ||||||
|  | 	m.Unlock() | ||||||
|  |  | ||||||
|  | 	// build & send the message | ||||||
|  | 	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() | ||||||
|  | 				go m.WsReceiver() | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		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) | ||||||
|  | 		if m.mmVersion() >= 3.8 { | ||||||
|  | 			mmchannels, err = m.Client.GetMoreChannelsPage(0, 5000) | ||||||
|  | 		} else { | ||||||
|  | 			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 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) mmVersion() float64 { | ||||||
|  | 	v, _ := strconv.ParseFloat(m.ServerVersion[0:3], 64) | ||||||
|  | 	return v | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func supportedVersion(version string) bool { | ||||||
|  | 	if strings.HasPrefix(version, "3.5.0") || | ||||||
|  | 		strings.HasPrefix(version, "3.6.0") || | ||||||
|  | 		strings.HasPrefix(version, "3.7.0") || | ||||||
|  | 		strings.HasPrefix(version, "3.8.0") || | ||||||
|  | 		strings.HasPrefix(version, "3.9.0") { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
| @@ -10,8 +10,9 @@ import ( | |||||||
| 	"io" | 	"io" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
| 	"log" | 	"log" | ||||||
|  | 	"net" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // OMessage for mattermost incoming webhook. (send to mattermost) | // OMessage for mattermost incoming webhook. (send to mattermost) | ||||||
| @@ -27,6 +28,8 @@ type OMessage struct { | |||||||
|  |  | ||||||
| // IMessage for mattermost outgoing webhook. (received from mattermost) | // IMessage for mattermost outgoing webhook. (received from mattermost) | ||||||
| type IMessage struct { | type IMessage struct { | ||||||
|  | 	BotID       string `schema:"bot_id"` | ||||||
|  | 	BotName     string `schema:"bot_name"` | ||||||
| 	Token       string `schema:"token"` | 	Token       string `schema:"token"` | ||||||
| 	TeamID      string `schema:"team_id"` | 	TeamID      string `schema:"team_id"` | ||||||
| 	TeamDomain  string `schema:"team_domain"` | 	TeamDomain  string `schema:"team_domain"` | ||||||
| @@ -36,6 +39,8 @@ type IMessage struct { | |||||||
| 	UserID      string `schema:"user_id"` | 	UserID      string `schema:"user_id"` | ||||||
| 	UserName    string `schema:"user_name"` | 	UserName    string `schema:"user_name"` | ||||||
| 	PostId      string `schema:"post_id"` | 	PostId      string `schema:"post_id"` | ||||||
|  | 	RawText     string `schema:"raw_text"` | ||||||
|  | 	ServiceId   string `schema:"service_id"` | ||||||
| 	Text        string `schema:"text"` | 	Text        string `schema:"text"` | ||||||
| 	TriggerWord string `schema:"trigger_word"` | 	TriggerWord string `schema:"trigger_word"` | ||||||
| } | } | ||||||
| @@ -51,7 +56,6 @@ type Client struct { | |||||||
|  |  | ||||||
| // Config for client. | // Config for client. | ||||||
| type Config struct { | type Config struct { | ||||||
| 	Port               int    // Port to listen on. |  | ||||||
| 	BindAddress        string // Address to listen on | 	BindAddress        string // Address to listen on | ||||||
| 	Token              string // Only allow this token from Mattermost. (Allow everything when empty) | 	Token              string // Only allow this token from Mattermost. (Allow everything when empty) | ||||||
| 	InsecureSkipVerify bool   // disable certificate checking | 	InsecureSkipVerify bool   // disable certificate checking | ||||||
| @@ -61,15 +65,15 @@ type Config struct { | |||||||
| // New Mattermost client. | // New Mattermost client. | ||||||
| func New(url string, config Config) *Client { | func New(url string, config Config) *Client { | ||||||
| 	c := &Client{Url: url, In: make(chan IMessage), Out: make(chan OMessage), Config: config} | 	c := &Client{Url: url, In: make(chan IMessage), Out: make(chan OMessage), Config: config} | ||||||
| 	if c.Port == 0 { |  | ||||||
| 		c.Port = 9999 |  | ||||||
| 	} |  | ||||||
| 	c.BindAddress += ":" |  | ||||||
| 	tr := &http.Transport{ | 	tr := &http.Transport{ | ||||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, | 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, | ||||||
| 	} | 	} | ||||||
| 	c.httpclient = &http.Client{Transport: tr} | 	c.httpclient = &http.Client{Transport: tr} | ||||||
| 	if !c.DisableServer { | 	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 | 	return c | ||||||
| @@ -79,8 +83,14 @@ func New(url string, config Config) *Client { | |||||||
| func (c *Client) StartServer() { | func (c *Client) StartServer() { | ||||||
| 	mux := http.NewServeMux() | 	mux := http.NewServeMux() | ||||||
| 	mux.Handle("/", c) | 	mux.Handle("/", c) | ||||||
| 	log.Printf("Listening on http://%v:%v...\n", c.BindAddress, c.Port) | 	srv := &http.Server{ | ||||||
| 	if err := http.ListenAndServe((c.BindAddress + strconv.Itoa(c.Port)), mux); err != nil { | 		ReadTimeout:  5 * time.Second, | ||||||
|  | 		WriteTimeout: 10 * time.Second, | ||||||
|  | 		Handler:      mux, | ||||||
|  | 		Addr:         c.BindAddress, | ||||||
|  | 	} | ||||||
|  | 	log.Printf("Listening on http://%v...\n", c.BindAddress) | ||||||
|  | 	if err := srv.ListenAndServe(); err != nil { | ||||||
| 		log.Fatal(err) | 		log.Fatal(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" | ||||||
|  | ``` | ||||||
							
								
								
									
										434
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										434
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,434 +0,0 @@ | |||||||
| 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
									
									
								
							
							
						
						
									
										68
									
								
								vendor/github.com/42wim/matterbridge-plus/bridge/config.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,68 +0,0 @@ | |||||||
| 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 |  | ||||||
| } |  | ||||||
							
								
								
									
										441
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/matterclient.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										441
									
								
								vendor/github.com/42wim/matterbridge-plus/matterclient/matterclient.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,441 +0,0 @@ | |||||||
| 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/toml-lang/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 newline 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 | ||||||
|  | } | ||||||
							
								
								
									
										953
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										953
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,953 @@ | |||||||
|  | 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 | ||||||
|  | 	itemInlineTableStart | ||||||
|  | 	itemInlineTableEnd | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	eof              = 0 | ||||||
|  | 	comma            = ',' | ||||||
|  | 	tableStart       = '[' | ||||||
|  | 	tableEnd         = ']' | ||||||
|  | 	arrayTableStart  = '[' | ||||||
|  | 	arrayTableEnd    = ']' | ||||||
|  | 	tableSep         = '.' | ||||||
|  | 	keySep           = '=' | ||||||
|  | 	arrayStart       = '[' | ||||||
|  | 	arrayEnd         = ']' | ||||||
|  | 	commentStart     = '#' | ||||||
|  | 	stringStart      = '"' | ||||||
|  | 	stringEnd        = '"' | ||||||
|  | 	rawStringStart   = '\'' | ||||||
|  | 	rawStringEnd     = '\'' | ||||||
|  | 	inlineTableStart = '{' | ||||||
|  | 	inlineTableEnd   = '}' | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type stateFn func(lx *lexer) stateFn | ||||||
|  |  | ||||||
|  | type lexer struct { | ||||||
|  | 	input string | ||||||
|  | 	start int | ||||||
|  | 	pos   int | ||||||
|  | 	line  int | ||||||
|  | 	state stateFn | ||||||
|  | 	items chan item | ||||||
|  |  | ||||||
|  | 	// Allow for backing up up to three runes. | ||||||
|  | 	// This is necessary because TOML contains 3-rune tokens (""" and '''). | ||||||
|  | 	prevWidths [3]int | ||||||
|  | 	nprev      int // how many of prevWidths are in use | ||||||
|  | 	// If we emit an eof, we can still back up, but it is not OK to call | ||||||
|  | 	// next again. | ||||||
|  | 	atEOF bool | ||||||
|  |  | ||||||
|  | 	// 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, | ||||||
|  | 		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.atEOF { | ||||||
|  | 		panic("next called after EOF") | ||||||
|  | 	} | ||||||
|  | 	if lx.pos >= len(lx.input) { | ||||||
|  | 		lx.atEOF = true | ||||||
|  | 		return eof | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if lx.input[lx.pos] == '\n' { | ||||||
|  | 		lx.line++ | ||||||
|  | 	} | ||||||
|  | 	lx.prevWidths[2] = lx.prevWidths[1] | ||||||
|  | 	lx.prevWidths[1] = lx.prevWidths[0] | ||||||
|  | 	if lx.nprev < 3 { | ||||||
|  | 		lx.nprev++ | ||||||
|  | 	} | ||||||
|  | 	r, w := utf8.DecodeRuneInString(lx.input[lx.pos:]) | ||||||
|  | 	lx.prevWidths[0] = w | ||||||
|  | 	lx.pos += w | ||||||
|  | 	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 twice between calls to next. | ||||||
|  | func (lx *lexer) backup() { | ||||||
|  | 	if lx.atEOF { | ||||||
|  | 		lx.atEOF = false | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if lx.nprev < 1 { | ||||||
|  | 		panic("backed up too far") | ||||||
|  | 	} | ||||||
|  | 	w := lx.prevWidths[0] | ||||||
|  | 	lx.prevWidths[0] = lx.prevWidths[1] | ||||||
|  | 	lx.prevWidths[1] = lx.prevWidths[2] | ||||||
|  | 	lx.nprev-- | ||||||
|  | 	lx.pos -= w | ||||||
|  | 	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 (newlines, 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 newline. 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 newline for us. | ||||||
|  | 		lx.push(lexTop) | ||||||
|  | 		return lexCommentStart | ||||||
|  | 	case isWhitespace(r): | ||||||
|  | 		return lexTopEnd | ||||||
|  | 	case isNL(r): | ||||||
|  | 		lx.ignore() | ||||||
|  | 		return lexTop | ||||||
|  | 	case r == eof: | ||||||
|  | 		lx.emit(itemEOF) | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	return lx.errorf("expected a top-level item to end with a newline, "+ | ||||||
|  | 		"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 newlines. | ||||||
|  | 	// In array syntax, the array states are responsible for ignoring newlines. | ||||||
|  | 	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 inlineTableStart: | ||||||
|  | 		lx.ignore() | ||||||
|  | 		lx.emit(itemInlineTableStart) | ||||||
|  | 		return lexInlineTableValue | ||||||
|  | 	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 newlines 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 == comma: | ||||||
|  | 		return lx.errorf("unexpected comma") | ||||||
|  | 	case r == arrayEnd: | ||||||
|  | 		// NOTE(caleb): The spec isn't clear about whether you can have | ||||||
|  | 		// a trailing comma or not, so we'll allow it. | ||||||
|  | 		return lexArrayEnd | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	lx.backup() | ||||||
|  | 	lx.push(lexArrayValueEnd) | ||||||
|  | 	return lexValue | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lexArrayValueEnd consumes everything between the end of an array value and | ||||||
|  | // the next value (or the end of the array): it ignores whitespace and newlines | ||||||
|  | // 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 == comma: | ||||||
|  | 		lx.ignore() | ||||||
|  | 		return lexArrayValue // move on to the next value | ||||||
|  | 	case r == arrayEnd: | ||||||
|  | 		return lexArrayEnd | ||||||
|  | 	} | ||||||
|  | 	return lx.errorf( | ||||||
|  | 		"expected a comma or array terminator %q, but got %q instead", | ||||||
|  | 		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() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lexInlineTableValue consumes one key/value pair in an inline table. | ||||||
|  | // It assumes that '{' or ',' have already been consumed. Whitespace is ignored. | ||||||
|  | func lexInlineTableValue(lx *lexer) stateFn { | ||||||
|  | 	r := lx.next() | ||||||
|  | 	switch { | ||||||
|  | 	case isWhitespace(r): | ||||||
|  | 		return lexSkip(lx, lexInlineTableValue) | ||||||
|  | 	case isNL(r): | ||||||
|  | 		return lx.errorf("newlines not allowed within inline tables") | ||||||
|  | 	case r == commentStart: | ||||||
|  | 		lx.push(lexInlineTableValue) | ||||||
|  | 		return lexCommentStart | ||||||
|  | 	case r == comma: | ||||||
|  | 		return lx.errorf("unexpected comma") | ||||||
|  | 	case r == inlineTableEnd: | ||||||
|  | 		return lexInlineTableEnd | ||||||
|  | 	} | ||||||
|  | 	lx.backup() | ||||||
|  | 	lx.push(lexInlineTableValueEnd) | ||||||
|  | 	return lexKeyStart | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lexInlineTableValueEnd consumes everything between the end of an inline table | ||||||
|  | // key/value pair and the next pair (or the end of the table): | ||||||
|  | // it ignores whitespace and expects either a ',' or a '}'. | ||||||
|  | func lexInlineTableValueEnd(lx *lexer) stateFn { | ||||||
|  | 	r := lx.next() | ||||||
|  | 	switch { | ||||||
|  | 	case isWhitespace(r): | ||||||
|  | 		return lexSkip(lx, lexInlineTableValueEnd) | ||||||
|  | 	case isNL(r): | ||||||
|  | 		return lx.errorf("newlines not allowed within inline tables") | ||||||
|  | 	case r == commentStart: | ||||||
|  | 		lx.push(lexInlineTableValueEnd) | ||||||
|  | 		return lexCommentStart | ||||||
|  | 	case r == comma: | ||||||
|  | 		lx.ignore() | ||||||
|  | 		return lexInlineTableValue | ||||||
|  | 	case r == inlineTableEnd: | ||||||
|  | 		return lexInlineTableEnd | ||||||
|  | 	} | ||||||
|  | 	return lx.errorf("expected a comma or an inline table terminator %q, "+ | ||||||
|  | 		"but got %q instead", inlineTableEnd, r) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // lexInlineTableEnd finishes the lexing of an inline table. | ||||||
|  | // It assumes that a '}' has just been consumed. | ||||||
|  | func lexInlineTableEnd(lx *lexer) stateFn { | ||||||
|  | 	lx.ignore() | ||||||
|  | 	lx.emit(itemInlineTableEnd) | ||||||
|  | 	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 r == eof: | ||||||
|  | 		return lx.errorf("unexpected EOF") | ||||||
|  | 	case isNL(r): | ||||||
|  | 		return lx.errorf("strings cannot contain newlines") | ||||||
|  | 	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 { | ||||||
|  | 	switch lx.next() { | ||||||
|  | 	case eof: | ||||||
|  | 		return lx.errorf("unexpected EOF") | ||||||
|  | 	case '\\': | ||||||
|  | 		return lexMultilineStringEscape | ||||||
|  | 	case 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 r == eof: | ||||||
|  | 		return lx.errorf("unexpected EOF") | ||||||
|  | 	case isNL(r): | ||||||
|  | 		return lx.errorf("strings cannot contain newlines") | ||||||
|  | 	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 { | ||||||
|  | 	switch lx.next() { | ||||||
|  | 	case eof: | ||||||
|  | 		return lx.errorf("unexpected EOF") | ||||||
|  | 	case 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 %q 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 %q 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 newline 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) | ||||||
|  | } | ||||||
							
								
								
									
										592
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										592
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,592 @@ | |||||||
|  | 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) | ||||||
|  | 	case itemInlineTableStart: | ||||||
|  | 		var ( | ||||||
|  | 			hash         = make(map[string]interface{}) | ||||||
|  | 			outerContext = p.context | ||||||
|  | 			outerKey     = p.currentKey | ||||||
|  | 		) | ||||||
|  |  | ||||||
|  | 		p.context = append(p.context, p.currentKey) | ||||||
|  | 		p.currentKey = "" | ||||||
|  | 		for it := p.next(); it.typ != itemInlineTableEnd; it = p.next() { | ||||||
|  | 			if it.typ != itemKeyStart { | ||||||
|  | 				p.bug("Expected key start but instead found %q, around line %d", | ||||||
|  | 					it.val, p.approxLine) | ||||||
|  | 			} | ||||||
|  | 			if it.typ == itemCommentStart { | ||||||
|  | 				p.expect(itemText) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// retrieve key | ||||||
|  | 			k := p.next() | ||||||
|  | 			p.approxLine = k.line | ||||||
|  | 			kname := p.keyString(k) | ||||||
|  |  | ||||||
|  | 			// retrieve value | ||||||
|  | 			p.currentKey = kname | ||||||
|  | 			val, typ := p.value(p.next()) | ||||||
|  | 			// make sure we keep metadata up to date | ||||||
|  | 			p.setType(kname, typ) | ||||||
|  | 			p.ordered = append(p.ordered, p.context.add(p.currentKey)) | ||||||
|  | 			hash[kname] = val | ||||||
|  | 		} | ||||||
|  | 		p.context = outerContext | ||||||
|  | 		p.currentKey = outerKey | ||||||
|  | 		return hash, tomlHash | ||||||
|  | 	} | ||||||
|  | 	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 | ||||||
|  | } | ||||||
							
								
								
									
										22
									
								
								vendor/github.com/GeertJohan/go.rice/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								vendor/github.com/GeertJohan/go.rice/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | |||||||
|  | Copyright (c) 2013, Geert-Johan Riemer | ||||||
|  | All rights reserved. | ||||||
|  |  | ||||||
|  | Redistribution and use in source and binary forms, with or without | ||||||
|  | modification, are permitted provided that the following conditions are met: | ||||||
|  |  | ||||||
|  | 1. Redistributions of source code must retain the above copyright notice, this | ||||||
|  |    list of conditions and the following disclaimer. | ||||||
|  | 2. 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 OWNER 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. | ||||||
							
								
								
									
										138
									
								
								vendor/github.com/GeertJohan/go.rice/appended.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										138
									
								
								vendor/github.com/GeertJohan/go.rice/appended.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,138 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"archive/zip" | ||||||
|  | 	"log" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/daaku/go.zipexe" | ||||||
|  | 	"github.com/kardianos/osext" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // appendedBox defines an appended box | ||||||
|  | type appendedBox struct { | ||||||
|  | 	Name  string                   // box name | ||||||
|  | 	Files map[string]*appendedFile // appended files (*zip.File) by full path | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type appendedFile struct { | ||||||
|  | 	zipFile  *zip.File | ||||||
|  | 	dir      bool | ||||||
|  | 	dirInfo  *appendedDirInfo | ||||||
|  | 	children []*appendedFile | ||||||
|  | 	content  []byte | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // appendedBoxes is a public register of appendes boxes | ||||||
|  | var appendedBoxes = make(map[string]*appendedBox) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	// find if exec is appended | ||||||
|  | 	thisFile, err := osext.Executable() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return // not appended or cant find self executable | ||||||
|  | 	} | ||||||
|  | 	closer, rd, err := zipexe.OpenCloser(thisFile) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return // not appended | ||||||
|  | 	} | ||||||
|  | 	defer closer.Close() | ||||||
|  |  | ||||||
|  | 	for _, f := range rd.File { | ||||||
|  | 		// get box and file name from f.Name | ||||||
|  | 		fileParts := strings.SplitN(strings.TrimLeft(filepath.ToSlash(f.Name), "/"), "/", 2) | ||||||
|  | 		boxName := fileParts[0] | ||||||
|  | 		var fileName string | ||||||
|  | 		if len(fileParts) > 1 { | ||||||
|  | 			fileName = fileParts[1] | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// find box or create new one if doesn't exist | ||||||
|  | 		box := appendedBoxes[boxName] | ||||||
|  | 		if box == nil { | ||||||
|  | 			box = &appendedBox{ | ||||||
|  | 				Name:  boxName, | ||||||
|  | 				Files: make(map[string]*appendedFile), | ||||||
|  | 			} | ||||||
|  | 			appendedBoxes[boxName] = box | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// create and add file to box | ||||||
|  | 		af := &appendedFile{ | ||||||
|  | 			zipFile: f, | ||||||
|  | 		} | ||||||
|  | 		if f.Comment == "dir" { | ||||||
|  | 			af.dir = true | ||||||
|  | 			af.dirInfo = &appendedDirInfo{ | ||||||
|  | 				name: filepath.Base(af.zipFile.Name), | ||||||
|  | 				//++ TODO: use zip modtime when that is set correctly: af.zipFile.ModTime() | ||||||
|  | 				time: time.Now(), | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			// this is a file, we need it's contents so we can create a bytes.Reader when the file is opened | ||||||
|  | 			// make a new byteslice | ||||||
|  | 			af.content = make([]byte, af.zipFile.FileInfo().Size()) | ||||||
|  | 			// ignore reading empty files from zip (empty file still is a valid file to be read though!) | ||||||
|  | 			if len(af.content) > 0 { | ||||||
|  | 				// open io.ReadCloser | ||||||
|  | 				rc, err := af.zipFile.Open() | ||||||
|  | 				if err != nil { | ||||||
|  | 					af.content = nil // this will cause an error when the file is being opened or seeked (which is good) | ||||||
|  | 					// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet.. | ||||||
|  | 					log.Printf("error opening appended file %s: %v", af.zipFile.Name, err) | ||||||
|  | 				} else { | ||||||
|  | 					_, err = rc.Read(af.content) | ||||||
|  | 					rc.Close() | ||||||
|  | 					if err != nil { | ||||||
|  | 						af.content = nil // this will cause an error when the file is being opened or seeked (which is good) | ||||||
|  | 						// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet.. | ||||||
|  | 						log.Printf("error reading data for appended file %s: %v", af.zipFile.Name, err) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// add appendedFile to box file list | ||||||
|  | 		box.Files[fileName] = af | ||||||
|  |  | ||||||
|  | 		// add to parent dir (if any) | ||||||
|  | 		dirName := filepath.Dir(fileName) | ||||||
|  | 		if dirName == "." { | ||||||
|  | 			dirName = "" | ||||||
|  | 		} | ||||||
|  | 		if fileName != "" { // don't make box root dir a child of itself | ||||||
|  | 			if dir := box.Files[dirName]; dir != nil { | ||||||
|  | 				dir.children = append(dir.children, af) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // implements os.FileInfo. | ||||||
|  | // used for Readdir() | ||||||
|  | type appendedDirInfo struct { | ||||||
|  | 	name string | ||||||
|  | 	time time.Time | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (adi *appendedDirInfo) Name() string { | ||||||
|  | 	return adi.name | ||||||
|  | } | ||||||
|  | func (adi *appendedDirInfo) Size() int64 { | ||||||
|  | 	return 0 | ||||||
|  | } | ||||||
|  | func (adi *appendedDirInfo) Mode() os.FileMode { | ||||||
|  | 	return os.ModeDir | ||||||
|  | } | ||||||
|  | func (adi *appendedDirInfo) ModTime() time.Time { | ||||||
|  | 	return adi.time | ||||||
|  | } | ||||||
|  | func (adi *appendedDirInfo) IsDir() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  | func (adi *appendedDirInfo) Sys() interface{} { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										337
									
								
								vendor/github.com/GeertJohan/go.rice/box.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										337
									
								
								vendor/github.com/GeertJohan/go.rice/box.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,337 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Box abstracts a directory for resources/files. | ||||||
|  | // It can either load files from disk, or from embedded code (when `rice --embed` was ran). | ||||||
|  | type Box struct { | ||||||
|  | 	name         string | ||||||
|  | 	absolutePath string | ||||||
|  | 	embed        *embedded.EmbeddedBox | ||||||
|  | 	appendd      *appendedBox | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var defaultLocateOrder = []LocateMethod{LocateEmbedded, LocateAppended, LocateFS} | ||||||
|  |  | ||||||
|  | func findBox(name string, order []LocateMethod) (*Box, error) { | ||||||
|  | 	b := &Box{name: name} | ||||||
|  |  | ||||||
|  | 	// no support for absolute paths since gopath can be different on different machines. | ||||||
|  | 	// therefore, required box must be located relative to package requiring it. | ||||||
|  | 	if filepath.IsAbs(name) { | ||||||
|  | 		return nil, errors.New("given name/path is absolute") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  | 	for _, method := range order { | ||||||
|  | 		switch method { | ||||||
|  | 		case LocateEmbedded: | ||||||
|  | 			if embed := embedded.EmbeddedBoxes[name]; embed != nil { | ||||||
|  | 				b.embed = embed | ||||||
|  | 				return b, nil | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 		case LocateAppended: | ||||||
|  | 			appendedBoxName := strings.Replace(name, `/`, `-`, -1) | ||||||
|  | 			if appendd := appendedBoxes[appendedBoxName]; appendd != nil { | ||||||
|  | 				b.appendd = appendd | ||||||
|  | 				return b, nil | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 		case LocateFS: | ||||||
|  | 			// resolve absolute directory path | ||||||
|  | 			err := b.resolveAbsolutePathFromCaller() | ||||||
|  | 			if err != nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// check if absolutePath exists on filesystem | ||||||
|  | 			info, err := os.Stat(b.absolutePath) | ||||||
|  | 			if err != nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// check if absolutePath is actually a directory | ||||||
|  | 			if !info.IsDir() { | ||||||
|  | 				err = errors.New("given name/path is not a directory") | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			return b, nil | ||||||
|  | 		case LocateWorkingDirectory: | ||||||
|  | 			// resolve absolute directory path | ||||||
|  | 			err := b.resolveAbsolutePathFromWorkingDirectory() | ||||||
|  | 			if err != nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// check if absolutePath exists on filesystem | ||||||
|  | 			info, err := os.Stat(b.absolutePath) | ||||||
|  | 			if err != nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// check if absolutePath is actually a directory | ||||||
|  | 			if !info.IsDir() { | ||||||
|  | 				err = errors.New("given name/path is not a directory") | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			return b, nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err == nil { | ||||||
|  | 		err = fmt.Errorf("could not locate box %q", name) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FindBox returns a Box instance for given name. | ||||||
|  | // When the given name is a relative path, it's base path will be the calling pkg/cmd's source root. | ||||||
|  | // When the given name is absolute, it's absolute. derp. | ||||||
|  | // Make sure the path doesn't contain any sensitive information as it might be placed into generated go source (embedded). | ||||||
|  | func FindBox(name string) (*Box, error) { | ||||||
|  | 	return findBox(name, defaultLocateOrder) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustFindBox returns a Box instance for given name, like FindBox does. | ||||||
|  | // It does not return an error, instead it panics when an error occurs. | ||||||
|  | func MustFindBox(name string) *Box { | ||||||
|  | 	box, err := findBox(name, defaultLocateOrder) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return box | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // This is injected as a mutable function literal so that we can mock it out in | ||||||
|  | // tests and return a fixed test file. | ||||||
|  | var resolveAbsolutePathFromCaller = func(name string, nStackFrames int) (string, error) { | ||||||
|  | 	_, callingGoFile, _, ok := runtime.Caller(nStackFrames) | ||||||
|  | 	if !ok { | ||||||
|  | 		return "", errors.New("couldn't find caller on stack") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// resolve to proper path | ||||||
|  | 	pkgDir := filepath.Dir(callingGoFile) | ||||||
|  | 	// fix for go cover | ||||||
|  | 	const coverPath = "_test/_obj_test" | ||||||
|  | 	if !filepath.IsAbs(pkgDir) { | ||||||
|  | 		if i := strings.Index(pkgDir, coverPath); i >= 0 { | ||||||
|  | 			pkgDir = pkgDir[:i] + pkgDir[i+len(coverPath):]            // remove coverPath | ||||||
|  | 			pkgDir = filepath.Join(os.Getenv("GOPATH"), "src", pkgDir) // make absolute | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return filepath.Join(pkgDir, name), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Box) resolveAbsolutePathFromCaller() error { | ||||||
|  | 	path, err := resolveAbsolutePathFromCaller(b.name, 4) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.absolutePath = path | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Box) resolveAbsolutePathFromWorkingDirectory() error { | ||||||
|  | 	path, err := os.Getwd() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.absolutePath = filepath.Join(path, b.name) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsEmbedded indicates wether this box was embedded into the application | ||||||
|  | func (b *Box) IsEmbedded() bool { | ||||||
|  | 	return b.embed != nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsAppended indicates wether this box was appended to the application | ||||||
|  | func (b *Box) IsAppended() bool { | ||||||
|  | 	return b.appendd != nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Time returns how actual the box is. | ||||||
|  | // When the box is embedded, it's value is saved in the embedding code. | ||||||
|  | // When the box is live, this methods returns time.Now() | ||||||
|  | func (b *Box) Time() time.Time { | ||||||
|  | 	if b.IsEmbedded() { | ||||||
|  | 		return b.embed.Time | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	//++ TODO: return time for appended box | ||||||
|  |  | ||||||
|  | 	return time.Now() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Open opens a File from the box | ||||||
|  | // If there is an error, it will be of type *os.PathError. | ||||||
|  | func (b *Box) Open(name string) (*File, error) { | ||||||
|  | 	if Debug { | ||||||
|  | 		fmt.Printf("Open(%s)\n", name) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if b.IsEmbedded() { | ||||||
|  | 		if Debug { | ||||||
|  | 			fmt.Println("Box is embedded") | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// trim prefix (paths are relative to box) | ||||||
|  | 		name = strings.TrimLeft(name, "/") | ||||||
|  | 		if Debug { | ||||||
|  | 			fmt.Printf("Trying %s\n", name) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// search for file | ||||||
|  | 		ef := b.embed.Files[name] | ||||||
|  | 		if ef == nil { | ||||||
|  | 			if Debug { | ||||||
|  | 				fmt.Println("Didn't find file in embed") | ||||||
|  | 			} | ||||||
|  | 			// file not found, try dir | ||||||
|  | 			ed := b.embed.Dirs[name] | ||||||
|  | 			if ed == nil { | ||||||
|  | 				if Debug { | ||||||
|  | 					fmt.Println("Didn't find dir in embed") | ||||||
|  | 				} | ||||||
|  | 				// dir not found, error out | ||||||
|  | 				return nil, &os.PathError{ | ||||||
|  | 					Op:   "open", | ||||||
|  | 					Path: name, | ||||||
|  | 					Err:  os.ErrNotExist, | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if Debug { | ||||||
|  | 				fmt.Println("Found dir. Returning virtual dir") | ||||||
|  | 			} | ||||||
|  | 			vd := newVirtualDir(ed) | ||||||
|  | 			return &File{virtualD: vd}, nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// box is embedded | ||||||
|  | 		if Debug { | ||||||
|  | 			fmt.Println("Found file. Returning virtual file") | ||||||
|  | 		} | ||||||
|  | 		vf := newVirtualFile(ef) | ||||||
|  | 		return &File{virtualF: vf}, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if b.IsAppended() { | ||||||
|  | 		// trim prefix (paths are relative to box) | ||||||
|  | 		name = strings.TrimLeft(name, "/") | ||||||
|  |  | ||||||
|  | 		// search for file | ||||||
|  | 		appendedFile := b.appendd.Files[name] | ||||||
|  | 		if appendedFile == nil { | ||||||
|  | 			return nil, &os.PathError{ | ||||||
|  | 				Op:   "open", | ||||||
|  | 				Path: name, | ||||||
|  | 				Err:  os.ErrNotExist, | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// create new file | ||||||
|  | 		f := &File{ | ||||||
|  | 			appendedF: appendedFile, | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// if this file is a directory, we want to be able to read and seek | ||||||
|  | 		if !appendedFile.dir { | ||||||
|  | 			// looks like malformed data in zip, error now | ||||||
|  | 			if appendedFile.content == nil { | ||||||
|  | 				return nil, &os.PathError{ | ||||||
|  | 					Op:   "open", | ||||||
|  | 					Path: "name", | ||||||
|  | 					Err:  errors.New("error reading data from zip file"), | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			// create new bytes.Reader | ||||||
|  | 			f.appendedFileReader = bytes.NewReader(appendedFile.content) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// all done | ||||||
|  | 		return f, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// perform os open | ||||||
|  | 	if Debug { | ||||||
|  | 		fmt.Printf("Using os.Open(%s)", filepath.Join(b.absolutePath, name)) | ||||||
|  | 	} | ||||||
|  | 	file, err := os.Open(filepath.Join(b.absolutePath, name)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return &File{realF: file}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Bytes returns the content of the file with given name as []byte. | ||||||
|  | func (b *Box) Bytes(name string) ([]byte, error) { | ||||||
|  | 	file, err := b.Open(name) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer file.Close() | ||||||
|  |  | ||||||
|  | 	content, err := ioutil.ReadAll(file) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return content, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustBytes returns the content of the file with given name as []byte. | ||||||
|  | // panic's on error. | ||||||
|  | func (b *Box) MustBytes(name string) []byte { | ||||||
|  | 	bts, err := b.Bytes(name) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return bts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns the content of the file with given name as string. | ||||||
|  | func (b *Box) String(name string) (string, error) { | ||||||
|  | 	// check if box is embedded, optimized fast path | ||||||
|  | 	if b.IsEmbedded() { | ||||||
|  | 		// find file in embed | ||||||
|  | 		ef := b.embed.Files[name] | ||||||
|  | 		if ef == nil { | ||||||
|  | 			return "", os.ErrNotExist | ||||||
|  | 		} | ||||||
|  | 		// return as string | ||||||
|  | 		return ef.Content, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bts, err := b.Bytes(name) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return string(bts), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustString returns the content of the file with given name as string. | ||||||
|  | // panic's on error. | ||||||
|  | func (b *Box) MustString(name string) string { | ||||||
|  | 	str, err := b.String(name) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return str | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name returns the name of the box | ||||||
|  | func (b *Box) Name() string { | ||||||
|  | 	return b.name | ||||||
|  | } | ||||||
							
								
								
									
										39
									
								
								vendor/github.com/GeertJohan/go.rice/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								vendor/github.com/GeertJohan/go.rice/config.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,39 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | // LocateMethod defines how a box is located. | ||||||
|  | type LocateMethod int | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	LocateFS               = LocateMethod(iota) // Locate on the filesystem according to package path. | ||||||
|  | 	LocateAppended                              // Locate boxes appended to the executable. | ||||||
|  | 	LocateEmbedded                              // Locate embedded boxes. | ||||||
|  | 	LocateWorkingDirectory                      // Locate on the binary working directory | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Config allows customizing the box lookup behavior. | ||||||
|  | type Config struct { | ||||||
|  | 	// LocateOrder defines the priority order that boxes are searched for. By | ||||||
|  | 	// default, the package global FindBox searches for embedded boxes first, | ||||||
|  | 	// then appended boxes, and then finally boxes on the filesystem.  That | ||||||
|  | 	// search order may be customized by provided the ordered list here. Leaving | ||||||
|  | 	// out a particular method will omit that from the search space. For | ||||||
|  | 	// example, []LocateMethod{LocateEmbedded, LocateAppended} will never search | ||||||
|  | 	// the filesystem for boxes. | ||||||
|  | 	LocateOrder []LocateMethod | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FindBox searches for boxes using the LocateOrder of the config. | ||||||
|  | func (c *Config) FindBox(boxName string) (*Box, error) { | ||||||
|  | 	return findBox(boxName, c.LocateOrder) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustFindBox searches for boxes using the LocateOrder of the config, like | ||||||
|  | // FindBox does.  It does not return an error, instead it panics when an error | ||||||
|  | // occurs. | ||||||
|  | func (c *Config) MustFindBox(boxName string) *Box { | ||||||
|  | 	box, err := findBox(boxName, c.LocateOrder) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return box | ||||||
|  | } | ||||||
							
								
								
									
										4
									
								
								vendor/github.com/GeertJohan/go.rice/debug.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								vendor/github.com/GeertJohan/go.rice/debug.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | // Debug can be set to true to enable debugging. | ||||||
|  | var Debug = false | ||||||
							
								
								
									
										90
									
								
								vendor/github.com/GeertJohan/go.rice/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										90
									
								
								vendor/github.com/GeertJohan/go.rice/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,90 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"os" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // re-type to make exported methods invisible to user (godoc) | ||||||
|  | // they're not required for the user | ||||||
|  | // embeddedDirInfo implements os.FileInfo | ||||||
|  | type embeddedDirInfo embedded.EmbeddedDir | ||||||
|  |  | ||||||
|  | // Name returns the base name of the directory | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) Name() string { | ||||||
|  | 	return ed.Filename | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Size always returns 0 | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) Size() int64 { | ||||||
|  | 	return 0 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Mode returns the file mode bits | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) Mode() os.FileMode { | ||||||
|  | 	return os.FileMode(0555 | os.ModeDir) // dr-xr-xr-x | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ModTime returns the modification time | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) ModTime() time.Time { | ||||||
|  | 	return ed.DirModTime | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsDir returns the abbreviation for Mode().IsDir() (always true) | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) IsDir() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Sys returns the underlying data source (always nil) | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ed *embeddedDirInfo) Sys() interface{} { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // re-type to make exported methods invisible to user (godoc) | ||||||
|  | // they're not required for the user | ||||||
|  | // embeddedFileInfo implements os.FileInfo | ||||||
|  | type embeddedFileInfo embedded.EmbeddedFile | ||||||
|  |  | ||||||
|  | // Name returns the base name of the file | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) Name() string { | ||||||
|  | 	return ef.Filename | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Size returns the length in bytes for regular files; system-dependent for others | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) Size() int64 { | ||||||
|  | 	return int64(len(ef.Content)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Mode returns the file mode bits | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) Mode() os.FileMode { | ||||||
|  | 	return os.FileMode(0555) // r-xr-xr-x | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ModTime returns the modification time | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) ModTime() time.Time { | ||||||
|  | 	return ef.FileModTime | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsDir returns the abbreviation for Mode().IsDir() (always false) | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) IsDir() bool { | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Sys returns the underlying data source (always nil) | ||||||
|  | // (implementing os.FileInfo) | ||||||
|  | func (ef *embeddedFileInfo) Sys() interface{} { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/embedded/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/embedded/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,80 @@ | |||||||
|  | // Package embedded defines embedded data types that are shared between the go.rice package and generated code. | ||||||
|  | package embedded | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	EmbedTypeGo   = 0 | ||||||
|  | 	EmbedTypeSyso = 1 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // EmbeddedBox defines an embedded box | ||||||
|  | type EmbeddedBox struct { | ||||||
|  | 	Name      string                   // box name | ||||||
|  | 	Time      time.Time                // embed time | ||||||
|  | 	EmbedType int                      // kind of embedding | ||||||
|  | 	Files     map[string]*EmbeddedFile // ALL embedded files by full path | ||||||
|  | 	Dirs      map[string]*EmbeddedDir  // ALL embedded dirs by full path | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Link creates the ChildDirs and ChildFiles links in all EmbeddedDir's | ||||||
|  | func (e *EmbeddedBox) Link() { | ||||||
|  | 	for path, ed := range e.Dirs { | ||||||
|  | 		fmt.Println(path) | ||||||
|  | 		ed.ChildDirs = make([]*EmbeddedDir, 0) | ||||||
|  | 		ed.ChildFiles = make([]*EmbeddedFile, 0) | ||||||
|  | 	} | ||||||
|  | 	for path, ed := range e.Dirs { | ||||||
|  | 		parentDirpath, _ := filepath.Split(path) | ||||||
|  | 		if strings.HasSuffix(parentDirpath, "/") { | ||||||
|  | 			parentDirpath = parentDirpath[:len(parentDirpath)-1] | ||||||
|  | 		} | ||||||
|  | 		parentDir := e.Dirs[parentDirpath] | ||||||
|  | 		if parentDir == nil { | ||||||
|  | 			panic("parentDir `" + parentDirpath + "` is missing in embedded box") | ||||||
|  | 		} | ||||||
|  | 		parentDir.ChildDirs = append(parentDir.ChildDirs, ed) | ||||||
|  | 	} | ||||||
|  | 	for path, ef := range e.Files { | ||||||
|  | 		dirpath, _ := filepath.Split(path) | ||||||
|  | 		if strings.HasSuffix(dirpath, "/") { | ||||||
|  | 			dirpath = dirpath[:len(dirpath)-1] | ||||||
|  | 		} | ||||||
|  | 		dir := e.Dirs[dirpath] | ||||||
|  | 		if dir == nil { | ||||||
|  | 			panic("dir `" + dirpath + "` is missing in embedded box") | ||||||
|  | 		} | ||||||
|  | 		dir.ChildFiles = append(dir.ChildFiles, ef) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // EmbeddedDir is instanced in the code generated by the rice tool and contains all necicary information about an embedded file | ||||||
|  | type EmbeddedDir struct { | ||||||
|  | 	Filename   string | ||||||
|  | 	DirModTime time.Time | ||||||
|  | 	ChildDirs  []*EmbeddedDir  // direct childs, as returned by virtualDir.Readdir() | ||||||
|  | 	ChildFiles []*EmbeddedFile // direct childs, as returned by virtualDir.Readdir() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // EmbeddedFile is instanced in the code generated by the rice tool and contains all necicary information about an embedded file | ||||||
|  | type EmbeddedFile struct { | ||||||
|  | 	Filename    string // filename | ||||||
|  | 	FileModTime time.Time | ||||||
|  | 	Content     string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // EmbeddedBoxes is a public register of embedded boxes | ||||||
|  | var EmbeddedBoxes = make(map[string]*EmbeddedBox) | ||||||
|  |  | ||||||
|  | // RegisterEmbeddedBox registers an EmbeddedBox | ||||||
|  | func RegisterEmbeddedBox(name string, box *EmbeddedBox) { | ||||||
|  | 	if _, exists := EmbeddedBoxes[name]; exists { | ||||||
|  | 		panic(fmt.Sprintf("EmbeddedBox with name `%s` exists already", name)) | ||||||
|  | 	} | ||||||
|  | 	EmbeddedBoxes[name] = box | ||||||
|  | } | ||||||
							
								
								
									
										69
									
								
								vendor/github.com/GeertJohan/go.rice/example/example.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										69
									
								
								vendor/github.com/GeertJohan/go.rice/example/example.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,69 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"encoding/hex" | ||||||
|  | 	"fmt" | ||||||
|  | 	"log" | ||||||
|  | 	"net/http" | ||||||
|  | 	"os" | ||||||
|  | 	"text/template" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.rice" | ||||||
|  | 	"github.com/davecgh/go-spew/spew" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func main() { | ||||||
|  | 	conf := rice.Config{ | ||||||
|  | 		LocateOrder: []rice.LocateMethod{rice.LocateEmbedded, rice.LocateAppended, rice.LocateFS}, | ||||||
|  | 	} | ||||||
|  | 	box, err := conf.FindBox("example-files") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatalf("error opening rice.Box: %s\n", err) | ||||||
|  | 	} | ||||||
|  | 	// spew.Dump(box) | ||||||
|  |  | ||||||
|  | 	contentString, err := box.String("file.txt") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatalf("could not read file contents as string: %s\n", err) | ||||||
|  | 	} | ||||||
|  | 	log.Printf("Read some file contents as string:\n%s\n", contentString) | ||||||
|  |  | ||||||
|  | 	contentBytes, err := box.Bytes("file.txt") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatalf("could not read file contents as byteSlice: %s\n", err) | ||||||
|  | 	} | ||||||
|  | 	log.Printf("Read some file contents as byteSlice:\n%s\n", hex.Dump(contentBytes)) | ||||||
|  |  | ||||||
|  | 	file, err := box.Open("file.txt") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatalf("could not open file: %s\n", err) | ||||||
|  | 	} | ||||||
|  | 	spew.Dump(file) | ||||||
|  |  | ||||||
|  | 	// find/create a rice.Box | ||||||
|  | 	templateBox, err := rice.FindBox("example-templates") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	// get file contents as string | ||||||
|  | 	templateString, err := templateBox.String("message.tmpl") | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	// parse and execute the template | ||||||
|  | 	tmplMessage, err := template.New("message").Parse(templateString) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	tmplMessage.Execute(os.Stdout, map[string]string{"Message": "Hello, world!"}) | ||||||
|  |  | ||||||
|  | 	http.Handle("/", http.FileServer(box.HTTPBox())) | ||||||
|  | 	go func() { | ||||||
|  | 		fmt.Println("Serving files on :8080, press ctrl-C to exit") | ||||||
|  | 		err := http.ListenAndServe(":8080", nil) | ||||||
|  | 		if err != nil { | ||||||
|  | 			log.Fatalf("error serving files: %v", err) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	select {} | ||||||
|  | } | ||||||
							
								
								
									
										144
									
								
								vendor/github.com/GeertJohan/go.rice/file.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										144
									
								
								vendor/github.com/GeertJohan/go.rice/file.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,144 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"errors" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // File implements the io.Reader, io.Seeker, io.Closer and http.File interfaces | ||||||
|  | type File struct { | ||||||
|  | 	// File abstracts file methods so the user doesn't see the difference between rice.virtualFile, rice.virtualDir and os.File | ||||||
|  | 	// TODO: maybe use internal File interface and four implementations: *os.File, appendedFile, virtualFile, virtualDir | ||||||
|  |  | ||||||
|  | 	// real file on disk | ||||||
|  | 	realF *os.File | ||||||
|  |  | ||||||
|  | 	// when embedded (go) | ||||||
|  | 	virtualF *virtualFile | ||||||
|  | 	virtualD *virtualDir | ||||||
|  |  | ||||||
|  | 	// when appended (zip) | ||||||
|  | 	appendedF          *appendedFile | ||||||
|  | 	appendedFileReader *bytes.Reader | ||||||
|  | 	// TODO: is appendedFileReader subject of races? Might need a lock here.. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Close is like (*os.File).Close() | ||||||
|  | // Visit http://golang.org/pkg/os/#File.Close for more information | ||||||
|  | func (f *File) Close() error { | ||||||
|  | 	if f.appendedF != nil { | ||||||
|  | 		if f.appendedFileReader == nil { | ||||||
|  | 			return errors.New("already closed") | ||||||
|  | 		} | ||||||
|  | 		f.appendedFileReader = nil | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	if f.virtualF != nil { | ||||||
|  | 		return f.virtualF.close() | ||||||
|  | 	} | ||||||
|  | 	if f.virtualD != nil { | ||||||
|  | 		return f.virtualD.close() | ||||||
|  | 	} | ||||||
|  | 	return f.realF.Close() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Stat is like (*os.File).Stat() | ||||||
|  | // Visit http://golang.org/pkg/os/#File.Stat for more information | ||||||
|  | func (f *File) Stat() (os.FileInfo, error) { | ||||||
|  | 	if f.appendedF != nil { | ||||||
|  | 		if f.appendedF.dir { | ||||||
|  | 			return f.appendedF.dirInfo, nil | ||||||
|  | 		} | ||||||
|  | 		if f.appendedFileReader == nil { | ||||||
|  | 			return nil, errors.New("file is closed") | ||||||
|  | 		} | ||||||
|  | 		return f.appendedF.zipFile.FileInfo(), nil | ||||||
|  | 	} | ||||||
|  | 	if f.virtualF != nil { | ||||||
|  | 		return f.virtualF.stat() | ||||||
|  | 	} | ||||||
|  | 	if f.virtualD != nil { | ||||||
|  | 		return f.virtualD.stat() | ||||||
|  | 	} | ||||||
|  | 	return f.realF.Stat() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Readdir is like (*os.File).Readdir() | ||||||
|  | // Visit http://golang.org/pkg/os/#File.Readdir for more information | ||||||
|  | func (f *File) Readdir(count int) ([]os.FileInfo, error) { | ||||||
|  | 	if f.appendedF != nil { | ||||||
|  | 		if f.appendedF.dir { | ||||||
|  | 			fi := make([]os.FileInfo, 0, len(f.appendedF.children)) | ||||||
|  | 			for _, childAppendedFile := range f.appendedF.children { | ||||||
|  | 				if childAppendedFile.dir { | ||||||
|  | 					fi = append(fi, childAppendedFile.dirInfo) | ||||||
|  | 				} else { | ||||||
|  | 					fi = append(fi, childAppendedFile.zipFile.FileInfo()) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return fi, nil | ||||||
|  | 		} | ||||||
|  | 		//++ TODO: is os.ErrInvalid the correct error for Readdir on file? | ||||||
|  | 		return nil, os.ErrInvalid | ||||||
|  | 	} | ||||||
|  | 	if f.virtualF != nil { | ||||||
|  | 		return f.virtualF.readdir(count) | ||||||
|  | 	} | ||||||
|  | 	if f.virtualD != nil { | ||||||
|  | 		return f.virtualD.readdir(count) | ||||||
|  | 	} | ||||||
|  | 	return f.realF.Readdir(count) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Read is like (*os.File).Read() | ||||||
|  | // Visit http://golang.org/pkg/os/#File.Read for more information | ||||||
|  | func (f *File) Read(bts []byte) (int, error) { | ||||||
|  | 	if f.appendedF != nil { | ||||||
|  | 		if f.appendedFileReader == nil { | ||||||
|  | 			return 0, &os.PathError{ | ||||||
|  | 				Op:   "read", | ||||||
|  | 				Path: filepath.Base(f.appendedF.zipFile.Name), | ||||||
|  | 				Err:  errors.New("file is closed"), | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if f.appendedF.dir { | ||||||
|  | 			return 0, &os.PathError{ | ||||||
|  | 				Op:   "read", | ||||||
|  | 				Path: filepath.Base(f.appendedF.zipFile.Name), | ||||||
|  | 				Err:  errors.New("is a directory"), | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return f.appendedFileReader.Read(bts) | ||||||
|  | 	} | ||||||
|  | 	if f.virtualF != nil { | ||||||
|  | 		return f.virtualF.read(bts) | ||||||
|  | 	} | ||||||
|  | 	if f.virtualD != nil { | ||||||
|  | 		return f.virtualD.read(bts) | ||||||
|  | 	} | ||||||
|  | 	return f.realF.Read(bts) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Seek is like (*os.File).Seek() | ||||||
|  | // Visit http://golang.org/pkg/os/#File.Seek for more information | ||||||
|  | func (f *File) Seek(offset int64, whence int) (int64, error) { | ||||||
|  | 	if f.appendedF != nil { | ||||||
|  | 		if f.appendedFileReader == nil { | ||||||
|  | 			return 0, &os.PathError{ | ||||||
|  | 				Op:   "seek", | ||||||
|  | 				Path: filepath.Base(f.appendedF.zipFile.Name), | ||||||
|  | 				Err:  errors.New("file is closed"), | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return f.appendedFileReader.Seek(offset, whence) | ||||||
|  | 	} | ||||||
|  | 	if f.virtualF != nil { | ||||||
|  | 		return f.virtualF.seek(offset, whence) | ||||||
|  | 	} | ||||||
|  | 	if f.virtualD != nil { | ||||||
|  | 		return f.virtualD.seek(offset, whence) | ||||||
|  | 	} | ||||||
|  | 	return f.realF.Seek(offset, whence) | ||||||
|  | } | ||||||
							
								
								
									
										21
									
								
								vendor/github.com/GeertJohan/go.rice/http.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/GeertJohan/go.rice/http.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,21 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"net/http" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // HTTPBox implements http.FileSystem which allows the use of Box with a http.FileServer. | ||||||
|  | //   e.g.: http.Handle("/", http.FileServer(rice.MustFindBox("http-files").HTTPBox())) | ||||||
|  | type HTTPBox struct { | ||||||
|  | 	*Box | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // HTTPBox creates a new HTTPBox from an existing Box | ||||||
|  | func (b *Box) HTTPBox() *HTTPBox { | ||||||
|  | 	return &HTTPBox{b} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Open returns a File using the http.File interface | ||||||
|  | func (hb *HTTPBox) Open(name string) (http.File, error) { | ||||||
|  | 	return hb.Box.Open(name) | ||||||
|  | } | ||||||
							
								
								
									
										172
									
								
								vendor/github.com/GeertJohan/go.rice/rice/append.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										172
									
								
								vendor/github.com/GeertJohan/go.rice/rice/append.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,172 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"archive/zip" | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"io" | ||||||
|  | 	"os" | ||||||
|  | 	"os/exec" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/daaku/go.zipexe" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func operationAppend(pkgs []*build.Package) { | ||||||
|  | 	if runtime.GOOS == "windows" { | ||||||
|  | 		_, err := exec.LookPath("zip") | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Println("#### WARNING ! ####") | ||||||
|  | 			fmt.Println("`rice append` is known not to work under windows because the `zip` command is not available. Please let me know if you got this to work (and how).") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// MARKED FOR DELETION | ||||||
|  | 	// This is actually not required, the append command now has the option --exec required. | ||||||
|  | 	// // check if package is a command | ||||||
|  | 	// if !pkg.IsCommand() { | ||||||
|  | 	// 	fmt.Println("Error: can not append to non-main package. Please follow instructions at github.com/GeertJohan/go.rice") | ||||||
|  | 	// 	os.Exit(1) | ||||||
|  | 	// } | ||||||
|  |  | ||||||
|  | 	// create tmp zipfile | ||||||
|  | 	tmpZipfileName := filepath.Join(os.TempDir(), fmt.Sprintf("ricebox-%d-%s.zip", time.Now().Unix(), randomString(10))) | ||||||
|  | 	verbosef("Will create tmp zipfile: %s\n", tmpZipfileName) | ||||||
|  | 	tmpZipfile, err := os.Create(tmpZipfileName) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error creating tmp zipfile: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | 	defer func() { | ||||||
|  | 		tmpZipfile.Close() | ||||||
|  | 		os.Remove(tmpZipfileName) | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	// find abs path for binary file | ||||||
|  | 	binfileName, err := filepath.Abs(flags.Append.Executable) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error finding absolute path for executable to append: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | 	verbosef("Will append to file: %s\n", binfileName) | ||||||
|  |  | ||||||
|  | 	// check that command doesn't already have zip appended | ||||||
|  | 	if rd, _ := zipexe.Open(binfileName); rd != nil { | ||||||
|  | 		fmt.Printf("Cannot append to already appended executable. Please remove %s and build a fresh one.\n", binfileName) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// open binfile | ||||||
|  | 	binfile, err := os.OpenFile(binfileName, os.O_WRONLY, os.ModeAppend) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error: unable to open executable file: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// create zip.Writer | ||||||
|  | 	zipWriter := zip.NewWriter(tmpZipfile) | ||||||
|  |  | ||||||
|  | 	for _, pkg := range pkgs { | ||||||
|  | 		// find boxes for this command | ||||||
|  | 		boxMap := findBoxes(pkg) | ||||||
|  |  | ||||||
|  | 		// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? | ||||||
|  | 		if len(boxMap) == 0 { | ||||||
|  | 			fmt.Printf("no calls to rice.FindBox() or rice.MustFindBox() found in import path `%s`\n", pkg.ImportPath) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		verbosef("\n") | ||||||
|  |  | ||||||
|  | 		for boxname := range boxMap { | ||||||
|  | 			appendedBoxName := strings.Replace(boxname, `/`, `-`, -1) | ||||||
|  |  | ||||||
|  | 			// walk box path's and insert files | ||||||
|  | 			boxPath := filepath.Clean(filepath.Join(pkg.Dir, boxname)) | ||||||
|  | 			filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { | ||||||
|  | 				if info == nil { | ||||||
|  | 					fmt.Printf("Error: box \"%s\" not found on disk\n", path) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				// create zipFilename | ||||||
|  | 				zipFileName := filepath.Join(appendedBoxName, strings.TrimPrefix(path, boxPath)) | ||||||
|  | 				// write directories as empty file with comment "dir" | ||||||
|  | 				if info.IsDir() { | ||||||
|  | 					_, err := zipWriter.CreateHeader(&zip.FileHeader{ | ||||||
|  | 						Name:    zipFileName, | ||||||
|  | 						Comment: "dir", | ||||||
|  | 					}) | ||||||
|  | 					if err != nil { | ||||||
|  | 						fmt.Printf("Error creating dir in tmp zip: %s\n", err) | ||||||
|  | 						os.Exit(1) | ||||||
|  | 					} | ||||||
|  | 					return nil | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				// create zipFileWriter | ||||||
|  | 				zipFileHeader, err := zip.FileInfoHeader(info) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("Error creating zip FileHeader: %v\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				zipFileHeader.Name = zipFileName | ||||||
|  | 				zipFileWriter, err := zipWriter.CreateHeader(zipFileHeader) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("Error creating file in tmp zip: %s\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				srcFile, err := os.Open(path) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("Error opening file to append: %s\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				_, err = io.Copy(zipFileWriter, srcFile) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("Error copying file contents to zip: %s\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				srcFile.Close() | ||||||
|  |  | ||||||
|  | 				return nil | ||||||
|  | 			}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	err = zipWriter.Close() | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error closing tmp zipfile: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	err = tmpZipfile.Sync() | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error syncing tmp zipfile: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | 	_, err = tmpZipfile.Seek(0, 0) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error seeking tmp zipfile: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | 	_, err = binfile.Seek(0, 2) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error seeking bin file: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, err = io.Copy(binfile, tmpZipfile) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error appending zipfile to executable: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	zipA := exec.Command("zip", "-A", binfileName) | ||||||
|  | 	err = zipA.Run() | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("Error setting zip offset: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										33
									
								
								vendor/github.com/GeertJohan/go.rice/rice/clean.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								vendor/github.com/GeertJohan/go.rice/rice/clean.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,33 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func operationClean(pkg *build.Package) { | ||||||
|  | 	filepath.Walk(pkg.Dir, func(filename string, info os.FileInfo, err error) error { | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error walking pkg dir to clean files: %v\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		if info.IsDir() { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		verbosef("checking file '%s'\n", filename) | ||||||
|  | 		if filepath.Base(filename) == "rice-box.go" || | ||||||
|  | 			strings.HasSuffix(filename, ".rice-box.go") || | ||||||
|  | 			strings.HasSuffix(filename, ".rice-box.syso") { | ||||||
|  | 			err := os.Remove(filename) | ||||||
|  | 			if err != nil { | ||||||
|  | 				fmt.Printf("error removing file (%s): %s\n", filename, err) | ||||||
|  | 				os.Exit(-1) | ||||||
|  | 			} | ||||||
|  | 			verbosef("removed file '%s'\n", filename) | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	}) | ||||||
|  | } | ||||||
							
								
								
									
										158
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-go.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										158
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-go.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,158 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"go/format" | ||||||
|  | 	"io" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"log" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const boxFilename = "rice-box.go" | ||||||
|  |  | ||||||
|  | func operationEmbedGo(pkg *build.Package) { | ||||||
|  |  | ||||||
|  | 	boxMap := findBoxes(pkg) | ||||||
|  |  | ||||||
|  | 	// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? | ||||||
|  | 	if len(boxMap) == 0 { | ||||||
|  | 		fmt.Println("no calls to rice.FindBox() found") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	verbosef("\n") | ||||||
|  | 	var boxes []*boxDataType | ||||||
|  |  | ||||||
|  | 	for boxname := range boxMap { | ||||||
|  | 		// find path and filename for this box | ||||||
|  | 		boxPath := filepath.Join(pkg.Dir, boxname) | ||||||
|  |  | ||||||
|  | 		// Check to see if the path for the box is a symbolic link.  If so, simply | ||||||
|  | 		// box what the symbolic link points to.  Note: the filepath.Walk function | ||||||
|  | 		// will NOT follow any nested symbolic links.  This only handles the case | ||||||
|  | 		// where the root of the box is a symbolic link. | ||||||
|  | 		symPath, serr := os.Readlink(boxPath) | ||||||
|  | 		if serr == nil { | ||||||
|  | 			boxPath = symPath | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// verbose info | ||||||
|  | 		verbosef("embedding box '%s' to '%s'\n", boxname, boxFilename) | ||||||
|  |  | ||||||
|  | 		// read box metadata | ||||||
|  | 		boxInfo, ierr := os.Stat(boxPath) | ||||||
|  | 		if ierr != nil { | ||||||
|  | 			fmt.Printf("Error: unable to access box at %s\n", boxPath) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// create box datastructure (used by template) | ||||||
|  | 		box := &boxDataType{ | ||||||
|  | 			BoxName: boxname, | ||||||
|  | 			UnixNow: boxInfo.ModTime().Unix(), | ||||||
|  | 			Files:   make([]*fileDataType, 0), | ||||||
|  | 			Dirs:    make(map[string]*dirDataType), | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if !boxInfo.IsDir() { | ||||||
|  | 			fmt.Printf("Error: Box %s must point to a directory but points to %s instead\n", | ||||||
|  | 				boxname, boxPath) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// fill box datastructure with file data | ||||||
|  | 		filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { | ||||||
|  | 			if err != nil { | ||||||
|  | 				fmt.Printf("error walking box: %s\n", err) | ||||||
|  | 				os.Exit(1) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			filename := strings.TrimPrefix(path, boxPath) | ||||||
|  | 			filename = strings.Replace(filename, "\\", "/", -1) | ||||||
|  | 			filename = strings.TrimPrefix(filename, "/") | ||||||
|  | 			if info.IsDir() { | ||||||
|  | 				dirData := &dirDataType{ | ||||||
|  | 					Identifier: "dir" + nextIdentifier(), | ||||||
|  | 					FileName:   filename, | ||||||
|  | 					ModTime:    info.ModTime().Unix(), | ||||||
|  | 					ChildFiles: make([]*fileDataType, 0), | ||||||
|  | 					ChildDirs:  make([]*dirDataType, 0), | ||||||
|  | 				} | ||||||
|  | 				verbosef("\tincludes dir: '%s'\n", dirData.FileName) | ||||||
|  | 				box.Dirs[dirData.FileName] = dirData | ||||||
|  |  | ||||||
|  | 				// add tree entry (skip for root, it'll create a recursion) | ||||||
|  | 				if dirData.FileName != "" { | ||||||
|  | 					pathParts := strings.Split(dirData.FileName, "/") | ||||||
|  | 					parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] | ||||||
|  | 					parentDir.ChildDirs = append(parentDir.ChildDirs, dirData) | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				fileData := &fileDataType{ | ||||||
|  | 					Identifier: "file" + nextIdentifier(), | ||||||
|  | 					FileName:   filename, | ||||||
|  | 					ModTime:    info.ModTime().Unix(), | ||||||
|  | 				} | ||||||
|  | 				verbosef("\tincludes file: '%s'\n", fileData.FileName) | ||||||
|  | 				fileData.Content, err = ioutil.ReadFile(path) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("error reading file content while walking box: %s\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				box.Files = append(box.Files, fileData) | ||||||
|  |  | ||||||
|  | 				// add tree entry | ||||||
|  | 				pathParts := strings.Split(fileData.FileName, "/") | ||||||
|  | 				parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] | ||||||
|  | 				if parentDir == nil { | ||||||
|  | 					fmt.Printf("Error: parent of %s is not within the box\n", path) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				parentDir.ChildFiles = append(parentDir.ChildFiles, fileData) | ||||||
|  | 			} | ||||||
|  | 			return nil | ||||||
|  | 		}) | ||||||
|  | 		boxes = append(boxes, box) | ||||||
|  |  | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	embedSourceUnformated := bytes.NewBuffer(make([]byte, 0)) | ||||||
|  |  | ||||||
|  | 	// execute template to buffer | ||||||
|  | 	err := tmplEmbeddedBox.Execute( | ||||||
|  | 		embedSourceUnformated, | ||||||
|  | 		embedFileDataType{pkg.Name, boxes}, | ||||||
|  | 	) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Printf("error writing embedded box to file (template execute): %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// format the source code | ||||||
|  | 	embedSource, err := format.Source(embedSourceUnformated.Bytes()) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Printf("error formatting embedSource: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// create go file for box | ||||||
|  | 	boxFile, err := os.Create(filepath.Join(pkg.Dir, boxFilename)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Printf("error creating embedded box file: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | 	defer boxFile.Close() | ||||||
|  |  | ||||||
|  | 	// write source to file | ||||||
|  | 	_, err = io.Copy(boxFile, bytes.NewBuffer(embedSource)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		log.Printf("error writing embedSource to file: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | } | ||||||
							
								
								
									
										204
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-syso.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										204
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-syso.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,204 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"encoding/gob" | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"io" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 	"text/template" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | 	"github.com/akavel/rsrc/coff" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type sizedReader struct { | ||||||
|  | 	*bytes.Reader | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s sizedReader) Size() int64 { | ||||||
|  | 	return int64(s.Len()) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var tmplEmbeddedSysoHelper *template.Template | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	var err error | ||||||
|  | 	tmplEmbeddedSysoHelper, err = template.New("embeddedSysoHelper").Parse(`package {{.Package}} | ||||||
|  | // ############# GENERATED CODE ##################### | ||||||
|  | // ## This file was generated by the rice tool. | ||||||
|  | // ## Do not edit unless you know what you're doing. | ||||||
|  | // ################################################## | ||||||
|  |  | ||||||
|  | // extern char _bricebox_{{.Symname}}[], _ericebox_{{.Symname}}; | ||||||
|  | // int get_{{.Symname}}_length() { | ||||||
|  | // 	return &_ericebox_{{.Symname}} - _bricebox_{{.Symname}}; | ||||||
|  | // } | ||||||
|  | import "C" | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"encoding/gob" | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | 	"unsafe" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	ptr := unsafe.Pointer(&C._bricebox_{{.Symname}}) | ||||||
|  | 	bts := C.GoBytes(ptr, C.get_{{.Symname}}_length()) | ||||||
|  | 	embeddedBox := &embedded.EmbeddedBox{} | ||||||
|  | 	err := gob.NewDecoder(bytes.NewReader(bts)).Decode(embeddedBox) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic("error decoding embedded box: "+err.Error()) | ||||||
|  | 	} | ||||||
|  | 	embeddedBox.Link() | ||||||
|  | 	embedded.RegisterEmbeddedBox(embeddedBox.Name, embeddedBox) | ||||||
|  | }`) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic("could not parse template embeddedSysoHelper: " + err.Error()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type embeddedSysoHelperData struct { | ||||||
|  | 	Package string | ||||||
|  | 	Symname string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func operationEmbedSyso(pkg *build.Package) { | ||||||
|  |  | ||||||
|  | 	regexpSynameReplacer := regexp.MustCompile(`[^a-z0-9_]`) | ||||||
|  |  | ||||||
|  | 	boxMap := findBoxes(pkg) | ||||||
|  |  | ||||||
|  | 	// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? | ||||||
|  | 	if len(boxMap) == 0 { | ||||||
|  | 		fmt.Println("no calls to rice.FindBox() found") | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	verbosef("\n") | ||||||
|  |  | ||||||
|  | 	for boxname := range boxMap { | ||||||
|  | 		// find path and filename for this box | ||||||
|  | 		boxPath := filepath.Join(pkg.Dir, boxname) | ||||||
|  | 		boxFilename := strings.Replace(boxname, "/", "-", -1) | ||||||
|  | 		boxFilename = strings.Replace(boxFilename, "..", "back", -1) | ||||||
|  | 		boxFilename = strings.Replace(boxFilename, ".", "-", -1) | ||||||
|  |  | ||||||
|  | 		// verbose info | ||||||
|  | 		verbosef("embedding box '%s'\n", boxname) | ||||||
|  | 		verbosef("\tto file %s\n", boxFilename) | ||||||
|  |  | ||||||
|  | 		// read box metadata | ||||||
|  | 		boxInfo, ierr := os.Stat(boxPath) | ||||||
|  | 		if ierr != nil { | ||||||
|  | 			fmt.Printf("Error: unable to access box at %s\n", boxPath) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// create box datastructure (used by template) | ||||||
|  | 		box := &embedded.EmbeddedBox{ | ||||||
|  | 			Name:      boxname, | ||||||
|  | 			Time:      boxInfo.ModTime(), | ||||||
|  | 			EmbedType: embedded.EmbedTypeSyso, | ||||||
|  | 			Files:     make(map[string]*embedded.EmbeddedFile), | ||||||
|  | 			Dirs:      make(map[string]*embedded.EmbeddedDir), | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// fill box datastructure with file data | ||||||
|  | 		filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { | ||||||
|  | 			if err != nil { | ||||||
|  | 				fmt.Printf("error walking box: %s\n", err) | ||||||
|  | 				os.Exit(1) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			filename := strings.TrimPrefix(path, boxPath) | ||||||
|  | 			filename = strings.Replace(filename, "\\", "/", -1) | ||||||
|  | 			filename = strings.TrimPrefix(filename, "/") | ||||||
|  | 			if info.IsDir() { | ||||||
|  | 				embeddedDir := &embedded.EmbeddedDir{ | ||||||
|  | 					Filename:   filename, | ||||||
|  | 					DirModTime: info.ModTime(), | ||||||
|  | 				} | ||||||
|  | 				verbosef("\tincludes dir: '%s'\n", embeddedDir.Filename) | ||||||
|  | 				box.Dirs[embeddedDir.Filename] = embeddedDir | ||||||
|  |  | ||||||
|  | 				// add tree entry (skip for root, it'll create a recursion) | ||||||
|  | 				if embeddedDir.Filename != "" { | ||||||
|  | 					pathParts := strings.Split(embeddedDir.Filename, "/") | ||||||
|  | 					parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] | ||||||
|  | 					parentDir.ChildDirs = append(parentDir.ChildDirs, embeddedDir) | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				embeddedFile := &embedded.EmbeddedFile{ | ||||||
|  | 					Filename:    filename, | ||||||
|  | 					FileModTime: info.ModTime(), | ||||||
|  | 					Content:     "", | ||||||
|  | 				} | ||||||
|  | 				verbosef("\tincludes file: '%s'\n", embeddedFile.Filename) | ||||||
|  | 				contentBytes, err := ioutil.ReadFile(path) | ||||||
|  | 				if err != nil { | ||||||
|  | 					fmt.Printf("error reading file content while walking box: %s\n", err) | ||||||
|  | 					os.Exit(1) | ||||||
|  | 				} | ||||||
|  | 				embeddedFile.Content = string(contentBytes) | ||||||
|  | 				box.Files[embeddedFile.Filename] = embeddedFile | ||||||
|  | 			} | ||||||
|  | 			return nil | ||||||
|  | 		}) | ||||||
|  |  | ||||||
|  | 		// encode embedded box to gob file | ||||||
|  | 		boxGobBuf := &bytes.Buffer{} | ||||||
|  | 		err := gob.NewEncoder(boxGobBuf).Encode(box) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error encoding box to gob: %v\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		verbosef("gob-encoded embeddedBox is %d bytes large\n", boxGobBuf.Len()) | ||||||
|  |  | ||||||
|  | 		// write coff | ||||||
|  | 		symname := regexpSynameReplacer.ReplaceAllString(boxname, "_") | ||||||
|  | 		createCoffSyso(boxname, symname, "386", boxGobBuf.Bytes()) | ||||||
|  | 		createCoffSyso(boxname, symname, "amd64", boxGobBuf.Bytes()) | ||||||
|  |  | ||||||
|  | 		// write go | ||||||
|  | 		sysoHelperData := embeddedSysoHelperData{ | ||||||
|  | 			Package: pkg.Name, | ||||||
|  | 			Symname: symname, | ||||||
|  | 		} | ||||||
|  | 		fileSysoHelper, err := os.Create(boxFilename + ".rice-box.go") | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error creating syso helper: %v\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		err = tmplEmbeddedSysoHelper.Execute(fileSysoHelper, sysoHelperData) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error executing tmplEmbeddedSysoHelper: %v\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func createCoffSyso(boxFilename string, symname string, arch string, data []byte) { | ||||||
|  | 	boxCoff := coff.NewRDATA() | ||||||
|  | 	switch arch { | ||||||
|  | 	case "386": | ||||||
|  | 	case "amd64": | ||||||
|  | 		boxCoff.FileHeader.Machine = 0x8664 | ||||||
|  | 	default: | ||||||
|  | 		panic("invalid arch") | ||||||
|  | 	} | ||||||
|  | 	boxCoff.AddData("_bricebox_"+symname, sizedReader{bytes.NewReader(data)}) | ||||||
|  | 	boxCoff.AddData("_ericebox_"+symname, io.NewSectionReader(strings.NewReader("\000\000"), 0, 2)) // TODO: why? copied from rsrc, which copied it from as-generated | ||||||
|  | 	boxCoff.Freeze() | ||||||
|  | 	err := writeCoff(boxCoff, boxFilename+"_"+arch+".rice-box.syso") | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("error writing %s coff/.syso: %v\n", arch, err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										150
									
								
								vendor/github.com/GeertJohan/go.rice/rice/find.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										150
									
								
								vendor/github.com/GeertJohan/go.rice/rice/find.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,150 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/ast" | ||||||
|  | 	"go/build" | ||||||
|  | 	"go/parser" | ||||||
|  | 	"go/token" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func badArgument(fileset *token.FileSet, p token.Pos) { | ||||||
|  | 	pos := fileset.Position(p) | ||||||
|  | 	filename := pos.Filename | ||||||
|  | 	base, err := os.Getwd() | ||||||
|  | 	if err == nil { | ||||||
|  | 		rpath, perr := filepath.Rel(base, pos.Filename) | ||||||
|  | 		if perr == nil { | ||||||
|  | 			filename = rpath | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	msg := fmt.Sprintf("%s:%d: Error: found call to rice.FindBox, "+ | ||||||
|  | 		"but argument must be a string literal.\n", filename, pos.Line) | ||||||
|  | 	fmt.Println(msg) | ||||||
|  | 	os.Exit(1) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func findBoxes(pkg *build.Package) map[string]bool { | ||||||
|  | 	// create map of boxes to embed | ||||||
|  | 	var boxMap = make(map[string]bool) | ||||||
|  |  | ||||||
|  | 	// create one list of files for this package | ||||||
|  | 	filenames := make([]string, 0, len(pkg.GoFiles)+len(pkg.CgoFiles)) | ||||||
|  | 	filenames = append(filenames, pkg.GoFiles...) | ||||||
|  | 	filenames = append(filenames, pkg.CgoFiles...) | ||||||
|  |  | ||||||
|  | 	// loop over files, search for rice.FindBox(..) calls | ||||||
|  | 	for _, filename := range filenames { | ||||||
|  | 		// find full filepath | ||||||
|  | 		fullpath := filepath.Join(pkg.Dir, filename) | ||||||
|  | 		if strings.HasSuffix(filename, "rice-box.go") { | ||||||
|  | 			// Ignore *.rice-box.go files | ||||||
|  | 			verbosef("skipping file %q\n", fullpath) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		verbosef("scanning file %q\n", fullpath) | ||||||
|  |  | ||||||
|  | 		fset := token.NewFileSet() | ||||||
|  | 		f, err := parser.ParseFile(fset, fullpath, nil, 0) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Println(err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		var riceIsImported bool | ||||||
|  | 		ricePkgName := "rice" | ||||||
|  | 		for _, imp := range f.Imports { | ||||||
|  | 			if strings.HasSuffix(imp.Path.Value, "go.rice\"") { | ||||||
|  | 				if imp.Name != nil { | ||||||
|  | 					ricePkgName = imp.Name.Name | ||||||
|  | 				} | ||||||
|  | 				riceIsImported = true | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if !riceIsImported { | ||||||
|  | 			// Rice wasn't imported, so we won't find a box. | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if ricePkgName == "_" { | ||||||
|  | 			// Rice pkg is unnamed, so we won't find a box. | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Inspect AST, looking for calls to (Must)?FindBox. | ||||||
|  | 		// First parameter of the func must be a basic literal. | ||||||
|  | 		// Identifiers won't be resolved. | ||||||
|  | 		var nextIdentIsBoxFunc bool | ||||||
|  | 		var nextBasicLitParamIsBoxName bool | ||||||
|  | 		var boxCall token.Pos | ||||||
|  | 		var variableToRemember string | ||||||
|  | 		var validVariablesForBoxes map[string]bool = make(map[string]bool) | ||||||
|  |  | ||||||
|  | 		ast.Inspect(f, func(node ast.Node) bool { | ||||||
|  | 			if node == nil { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 			switch x := node.(type) { | ||||||
|  | 			// this case fixes the var := func() style assignments, not assignments to vars declared separately from the assignment. | ||||||
|  | 			case *ast.AssignStmt: | ||||||
|  | 				var assign = node.(*ast.AssignStmt) | ||||||
|  | 				name, found := assign.Lhs[0].(*ast.Ident) | ||||||
|  | 				if found { | ||||||
|  | 					variableToRemember = name.Name | ||||||
|  | 					composite, first := assign.Rhs[0].(*ast.CompositeLit) | ||||||
|  | 					if first { | ||||||
|  | 						riceSelector, second := composite.Type.(*ast.SelectorExpr) | ||||||
|  |  | ||||||
|  | 						if second { | ||||||
|  | 							callCorrect := riceSelector.Sel.Name == "Config" | ||||||
|  | 							packageName, third := riceSelector.X.(*ast.Ident) | ||||||
|  |  | ||||||
|  | 							if third && callCorrect && packageName.Name == ricePkgName { | ||||||
|  | 								validVariablesForBoxes[name.Name] = true | ||||||
|  | 								verbosef("\tfound variable, saving to scan for boxes: %q\n", name.Name) | ||||||
|  | 							} | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			case *ast.Ident: | ||||||
|  | 				if nextIdentIsBoxFunc || ricePkgName == "." { | ||||||
|  | 					nextIdentIsBoxFunc = false | ||||||
|  | 					if x.Name == "FindBox" || x.Name == "MustFindBox" { | ||||||
|  | 						nextBasicLitParamIsBoxName = true | ||||||
|  | 						boxCall = x.Pos() | ||||||
|  | 					} | ||||||
|  | 				} else { | ||||||
|  | 					if x.Name == ricePkgName || validVariablesForBoxes[x.Name] { | ||||||
|  | 						nextIdentIsBoxFunc = true | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			case *ast.BasicLit: | ||||||
|  | 				if nextBasicLitParamIsBoxName { | ||||||
|  | 					if x.Kind == token.STRING { | ||||||
|  | 						nextBasicLitParamIsBoxName = false | ||||||
|  | 						// trim "" or `` | ||||||
|  | 						name := x.Value[1 : len(x.Value)-1] | ||||||
|  | 						boxMap[name] = true | ||||||
|  | 						verbosef("\tfound box %q\n", name) | ||||||
|  | 					} else { | ||||||
|  | 						badArgument(fset, boxCall) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 			default: | ||||||
|  | 				if nextIdentIsBoxFunc { | ||||||
|  | 					nextIdentIsBoxFunc = false | ||||||
|  | 				} | ||||||
|  | 				if nextBasicLitParamIsBoxName { | ||||||
|  | 					badArgument(fset, boxCall) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return true | ||||||
|  | 		}) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return boxMap | ||||||
|  | } | ||||||
							
								
								
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/rice/flags.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/rice/flags.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,80 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"os" | ||||||
|  |  | ||||||
|  | 	goflags "github.com/jessevdk/go-flags" // rename import to `goflags` (file scope) so we can use `var flags` (package scope) | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // flags | ||||||
|  | var flags struct { | ||||||
|  | 	Verbose     bool     `long:"verbose" short:"v" description:"Show verbose debug information"` | ||||||
|  | 	ImportPaths []string `long:"import-path" short:"i" description:"Import path(s) to use. Using PWD when left empty. Specify multiple times for more import paths to append"` | ||||||
|  |  | ||||||
|  | 	Append struct { | ||||||
|  | 		Executable string `long:"exec" description:"Executable to append" required:"true"` | ||||||
|  | 	} `command:"append"` | ||||||
|  |  | ||||||
|  | 	EmbedGo   struct{} `command:"embed-go" alias:"embed"` | ||||||
|  | 	EmbedSyso struct{} `command:"embed-syso"` | ||||||
|  | 	Clean     struct{} `command:"clean"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // flags parser | ||||||
|  | var flagsParser *goflags.Parser | ||||||
|  |  | ||||||
|  | // initFlags parses the given flags. | ||||||
|  | // when the user asks for help (-h or --help): the application exists with status 0 | ||||||
|  | // when unexpected flags is given: the application exits with status 1 | ||||||
|  | func parseArguments() { | ||||||
|  | 	// create flags parser in global var, for flagsParser.Active.Name (operation) | ||||||
|  | 	flagsParser = goflags.NewParser(&flags, goflags.Default) | ||||||
|  |  | ||||||
|  | 	// parse flags | ||||||
|  | 	args, err := flagsParser.Parse() | ||||||
|  | 	if err != nil { | ||||||
|  | 		// assert the err to be a flags.Error | ||||||
|  | 		flagError := err.(*goflags.Error) | ||||||
|  | 		if flagError.Type == goflags.ErrHelp { | ||||||
|  | 			// user asked for help on flags. | ||||||
|  | 			// program can exit successfully | ||||||
|  | 			os.Exit(0) | ||||||
|  | 		} | ||||||
|  | 		if flagError.Type == goflags.ErrUnknownFlag { | ||||||
|  | 			fmt.Println("Use --help to view available options.") | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		if flagError.Type == goflags.ErrRequired { | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		fmt.Printf("Error parsing flags: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// error on left-over arguments | ||||||
|  | 	if len(args) > 0 { | ||||||
|  | 		fmt.Printf("Unexpected arguments: %s\nUse --help to view available options.", args) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// default ImportPath to pwd when not set | ||||||
|  | 	if len(flags.ImportPaths) == 0 { | ||||||
|  | 		pwd, err := os.Getwd() | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error getting pwd: %s\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		verbosef("using pwd as import path\n") | ||||||
|  | 		// find non-absolute path for this pwd | ||||||
|  | 		pkg, err := build.ImportDir(pwd, build.FindOnly) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Printf("error using current directory as import path: %s\n", err) | ||||||
|  | 			os.Exit(1) | ||||||
|  | 		} | ||||||
|  | 		flags.ImportPaths = append(flags.ImportPaths, pkg.ImportPath) | ||||||
|  | 		verbosef("using import paths: %s\n", flags.ImportPaths) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										14
									
								
								vendor/github.com/GeertJohan/go.rice/rice/identifier.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/GeertJohan/go.rice/rice/identifier.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"strconv" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.incremental" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var identifierCount incremental.Uint64 | ||||||
|  |  | ||||||
|  | func nextIdentifier() string { | ||||||
|  | 	num := identifierCount.Next() | ||||||
|  | 	return strconv.FormatUint(num, 36) // 0123456789abcdefghijklmnopqrstuvwxyz | ||||||
|  | } | ||||||
							
								
								
									
										68
									
								
								vendor/github.com/GeertJohan/go.rice/rice/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										68
									
								
								vendor/github.com/GeertJohan/go.rice/rice/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,68 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"go/build" | ||||||
|  | 	"log" | ||||||
|  | 	"os" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func main() { | ||||||
|  | 	// parser arguments | ||||||
|  | 	parseArguments() | ||||||
|  |  | ||||||
|  | 	// find package for path | ||||||
|  | 	var pkgs []*build.Package | ||||||
|  | 	for _, importPath := range flags.ImportPaths { | ||||||
|  | 		pkg := pkgForPath(importPath) | ||||||
|  | 		pkgs = append(pkgs, pkg) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// switch on the operation to perform | ||||||
|  | 	switch flagsParser.Active.Name { | ||||||
|  | 	case "embed", "embed-go": | ||||||
|  | 		for _, pkg := range pkgs { | ||||||
|  | 			operationEmbedGo(pkg) | ||||||
|  | 		} | ||||||
|  | 	case "embed-syso": | ||||||
|  | 		log.Println("WARNING: embedding .syso is experimental..") | ||||||
|  | 		for _, pkg := range pkgs { | ||||||
|  | 			operationEmbedSyso(pkg) | ||||||
|  | 		} | ||||||
|  | 	case "append": | ||||||
|  | 		operationAppend(pkgs) | ||||||
|  | 	case "clean": | ||||||
|  | 		for _, pkg := range pkgs { | ||||||
|  | 			operationClean(pkg) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// all done | ||||||
|  | 	verbosef("\n") | ||||||
|  | 	verbosef("rice finished successfully\n") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // helper function to get *build.Package for given path | ||||||
|  | func pkgForPath(path string) *build.Package { | ||||||
|  | 	// get pwd for relative imports | ||||||
|  | 	pwd, err := os.Getwd() | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("error getting pwd (required for relative imports): %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// read full package information | ||||||
|  | 	pkg, err := build.Import(path, pwd, 0) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("error reading package: %s\n", err) | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return pkg | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func verbosef(format string, stuff ...interface{}) { | ||||||
|  | 	if flags.Verbose { | ||||||
|  | 		log.Printf(format, stuff...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										98
									
								
								vendor/github.com/GeertJohan/go.rice/rice/templates.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								vendor/github.com/GeertJohan/go.rice/rice/templates.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,98 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"text/template" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var tmplEmbeddedBox *template.Template | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	// parse embedded box template | ||||||
|  | 	tmplEmbeddedBox, err = template.New("embeddedBox").Parse(`package {{.Package}} | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | {{range .Boxes}} | ||||||
|  | func init() { | ||||||
|  |  | ||||||
|  | 	// define files | ||||||
|  | 	{{range .Files}}{{.Identifier}} := &embedded.EmbeddedFile{ | ||||||
|  | 		Filename:    ` + "`" + `{{.FileName}}` + "`" + `, | ||||||
|  | 		FileModTime: time.Unix({{.ModTime}}, 0), | ||||||
|  | 		Content:     string({{.Content | printf "%q"}}),  | ||||||
|  | 	} | ||||||
|  | 	{{end}} | ||||||
|  |  | ||||||
|  | 	// define dirs | ||||||
|  | 	{{range .Dirs}}{{.Identifier}} := &embedded.EmbeddedDir{ | ||||||
|  | 		Filename:    ` + "`" + `{{.FileName}}` + "`" + `, | ||||||
|  | 		DirModTime: time.Unix({{.ModTime}}, 0), | ||||||
|  | 		ChildFiles:  []*embedded.EmbeddedFile{ | ||||||
|  | 			{{range .ChildFiles}}{{.Identifier}}, // {{.FileName}} | ||||||
|  | 			{{end}} | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	{{end}} | ||||||
|  |  | ||||||
|  | 	// link ChildDirs | ||||||
|  | 	{{range .Dirs}}{{.Identifier}}.ChildDirs = []*embedded.EmbeddedDir{ | ||||||
|  | 		{{range .ChildDirs}}{{.Identifier}}, // {{.FileName}} | ||||||
|  | 		{{end}} | ||||||
|  | 	} | ||||||
|  | 	{{end}} | ||||||
|  |  | ||||||
|  | 	// register embeddedBox | ||||||
|  | 	embedded.RegisterEmbeddedBox(` + "`" + `{{.BoxName}}` + "`" + `, &embedded.EmbeddedBox{ | ||||||
|  | 		Name: ` + "`" + `{{.BoxName}}` + "`" + `, | ||||||
|  | 		Time: time.Unix({{.UnixNow}}, 0), | ||||||
|  | 		Dirs: map[string]*embedded.EmbeddedDir{ | ||||||
|  | 			{{range .Dirs}}"{{.FileName}}": {{.Identifier}}, | ||||||
|  | 			{{end}} | ||||||
|  | 		}, | ||||||
|  | 		Files: map[string]*embedded.EmbeddedFile{ | ||||||
|  | 			{{range .Files}}"{{.FileName}}": {{.Identifier}}, | ||||||
|  | 			{{end}} | ||||||
|  | 		}, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | {{end}}`) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Printf("error parsing embedded box template: %s\n", err) | ||||||
|  | 		os.Exit(-1) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type embedFileDataType struct { | ||||||
|  | 	Package string | ||||||
|  | 	Boxes   []*boxDataType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type boxDataType struct { | ||||||
|  | 	BoxName string | ||||||
|  | 	UnixNow int64 | ||||||
|  | 	Files   []*fileDataType | ||||||
|  | 	Dirs    map[string]*dirDataType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type fileDataType struct { | ||||||
|  | 	Identifier string | ||||||
|  | 	FileName   string | ||||||
|  | 	Content    []byte | ||||||
|  | 	ModTime    int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type dirDataType struct { | ||||||
|  | 	Identifier string | ||||||
|  | 	FileName   string | ||||||
|  | 	Content    []byte | ||||||
|  | 	ModTime    int64 | ||||||
|  | 	ChildDirs  []*dirDataType | ||||||
|  | 	ChildFiles []*fileDataType | ||||||
|  | } | ||||||
							
								
								
									
										22
									
								
								vendor/github.com/GeertJohan/go.rice/rice/util.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								vendor/github.com/GeertJohan/go.rice/rice/util.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"math/rand" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // randomString generates a pseudo-random alpha-numeric string with given length. | ||||||
|  | func randomString(length int) string { | ||||||
|  | 	rand.Seed(time.Now().UnixNano()) | ||||||
|  | 	k := make([]rune, length) | ||||||
|  | 	for i := 0; i < length; i++ { | ||||||
|  | 		c := rand.Intn(35) | ||||||
|  | 		if c < 10 { | ||||||
|  | 			c += 48 // numbers (0-9) (0+48 == 48 == '0', 9+48 == 57 == '9') | ||||||
|  | 		} else { | ||||||
|  | 			c += 87 // lower case alphabets (a-z) (10+87 == 97 == 'a', 35+87 == 122 = 'z') | ||||||
|  | 		} | ||||||
|  | 		k[i] = rune(c) | ||||||
|  | 	} | ||||||
|  | 	return string(k) | ||||||
|  | } | ||||||
							
								
								
									
										42
									
								
								vendor/github.com/GeertJohan/go.rice/rice/writecoff.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								vendor/github.com/GeertJohan/go.rice/rice/writecoff.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,42 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"reflect" | ||||||
|  |  | ||||||
|  | 	"github.com/akavel/rsrc/binutil" | ||||||
|  | 	"github.com/akavel/rsrc/coff" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // copied from github.com/akavel/rsrc | ||||||
|  | // LICENSE: MIT | ||||||
|  | // Copyright 2013-2014 The rsrc Authors. (https://github.com/akavel/rsrc/blob/master/AUTHORS) | ||||||
|  | func writeCoff(coff *coff.Coff, fnameout string) error { | ||||||
|  | 	out, err := os.Create(fnameout) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer out.Close() | ||||||
|  | 	w := binutil.Writer{W: out} | ||||||
|  |  | ||||||
|  | 	// write the resulting file to disk | ||||||
|  | 	binutil.Walk(coff, func(v reflect.Value, path string) error { | ||||||
|  | 		if binutil.Plain(v.Kind()) { | ||||||
|  | 			w.WriteLE(v.Interface()) | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		vv, ok := v.Interface().(binutil.SizedReader) | ||||||
|  | 		if ok { | ||||||
|  | 			w.WriteFromSized(vv) | ||||||
|  | 			return binutil.WALK_SKIP | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	if w.Err != nil { | ||||||
|  | 		return fmt.Errorf("Error writing output file: %s", w.Err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										19
									
								
								vendor/github.com/GeertJohan/go.rice/sort.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								vendor/github.com/GeertJohan/go.rice/sort.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,19 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import "os" | ||||||
|  |  | ||||||
|  | // SortByName allows an array of os.FileInfo objects | ||||||
|  | // to be easily sorted by filename using sort.Sort(SortByName(array)) | ||||||
|  | type SortByName []os.FileInfo | ||||||
|  |  | ||||||
|  | func (f SortByName) Len() int           { return len(f) } | ||||||
|  | func (f SortByName) Less(i, j int) bool { return f[i].Name() < f[j].Name() } | ||||||
|  | func (f SortByName) Swap(i, j int)      { f[i], f[j] = f[j], f[i] } | ||||||
|  |  | ||||||
|  | // SortByModified allows an array of os.FileInfo objects | ||||||
|  | // to be easily sorted by modified date using sort.Sort(SortByModified(array)) | ||||||
|  | type SortByModified []os.FileInfo | ||||||
|  |  | ||||||
|  | func (f SortByModified) Len() int           { return len(f) } | ||||||
|  | func (f SortByModified) Less(i, j int) bool { return f[i].ModTime().Unix() > f[j].ModTime().Unix() } | ||||||
|  | func (f SortByModified) Swap(i, j int)      { f[i], f[j] = f[j], f[i] } | ||||||
							
								
								
									
										252
									
								
								vendor/github.com/GeertJohan/go.rice/virtual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										252
									
								
								vendor/github.com/GeertJohan/go.rice/virtual.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,252 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"io" | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"sort" | ||||||
|  |  | ||||||
|  | 	"github.com/GeertJohan/go.rice/embedded" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | //++ TODO: IDEA: merge virtualFile and virtualDir, this decreases work done by rice.File | ||||||
|  |  | ||||||
|  | // Error indicating some function is not implemented yet (but available to satisfy an interface) | ||||||
|  | var ErrNotImplemented = errors.New("not implemented yet") | ||||||
|  |  | ||||||
|  | // virtualFile is a 'stateful' virtual file. | ||||||
|  | // virtualFile wraps an *EmbeddedFile for a call to Box.Open() and virtualizes 'read cursor' (offset) and 'closing'. | ||||||
|  | // virtualFile is only internally visible and should be exposed through rice.File | ||||||
|  | type virtualFile struct { | ||||||
|  | 	*embedded.EmbeddedFile       // the actual embedded file, embedded to obtain methods | ||||||
|  | 	offset                 int64 // read position on the virtual file | ||||||
|  | 	closed                 bool  // closed when true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // create a new virtualFile for given EmbeddedFile | ||||||
|  | func newVirtualFile(ef *embedded.EmbeddedFile) *virtualFile { | ||||||
|  | 	vf := &virtualFile{ | ||||||
|  | 		EmbeddedFile: ef, | ||||||
|  | 		offset:       0, | ||||||
|  | 		closed:       false, | ||||||
|  | 	} | ||||||
|  | 	return vf | ||||||
|  | } | ||||||
|  |  | ||||||
|  | //++ TODO check for nil pointers in all these methods. When so: return os.PathError with Err: os.ErrInvalid | ||||||
|  |  | ||||||
|  | func (vf *virtualFile) close() error { | ||||||
|  | 	if vf.closed { | ||||||
|  | 		return &os.PathError{ | ||||||
|  | 			Op:   "close", | ||||||
|  | 			Path: vf.EmbeddedFile.Filename, | ||||||
|  | 			Err:  errors.New("already closed"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	vf.EmbeddedFile = nil | ||||||
|  | 	vf.closed = true | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vf *virtualFile) stat() (os.FileInfo, error) { | ||||||
|  | 	if vf.closed { | ||||||
|  | 		return nil, &os.PathError{ | ||||||
|  | 			Op:   "stat", | ||||||
|  | 			Path: vf.EmbeddedFile.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return (*embeddedFileInfo)(vf.EmbeddedFile), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vf *virtualFile) readdir(count int) ([]os.FileInfo, error) { | ||||||
|  | 	if vf.closed { | ||||||
|  | 		return nil, &os.PathError{ | ||||||
|  | 			Op:   "readdir", | ||||||
|  | 			Path: vf.EmbeddedFile.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	//TODO: return proper error for a readdir() call on a file | ||||||
|  | 	return nil, ErrNotImplemented | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vf *virtualFile) read(bts []byte) (int, error) { | ||||||
|  | 	if vf.closed { | ||||||
|  | 		return 0, &os.PathError{ | ||||||
|  | 			Op:   "read", | ||||||
|  | 			Path: vf.EmbeddedFile.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	end := vf.offset + int64(len(bts)) | ||||||
|  |  | ||||||
|  | 	if end >= int64(len(vf.Content)) { | ||||||
|  | 		// end of file, so return what we have + EOF | ||||||
|  | 		n := copy(bts, vf.Content[vf.offset:]) | ||||||
|  | 		vf.offset = 0 | ||||||
|  | 		return n, io.EOF | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	n := copy(bts, vf.Content[vf.offset:end]) | ||||||
|  | 	vf.offset += int64(n) | ||||||
|  | 	return n, nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vf *virtualFile) seek(offset int64, whence int) (int64, error) { | ||||||
|  | 	if vf.closed { | ||||||
|  | 		return 0, &os.PathError{ | ||||||
|  | 			Op:   "seek", | ||||||
|  | 			Path: vf.EmbeddedFile.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	var e error | ||||||
|  |  | ||||||
|  | 	//++ TODO: check if this is correct implementation for seek | ||||||
|  | 	switch whence { | ||||||
|  | 	case os.SEEK_SET: | ||||||
|  | 		//++ check if new offset isn't out of bounds, set e when it is, then break out of switch | ||||||
|  | 		vf.offset = offset | ||||||
|  | 	case os.SEEK_CUR: | ||||||
|  | 		//++ check if new offset isn't out of bounds, set e when it is, then break out of switch | ||||||
|  | 		vf.offset += offset | ||||||
|  | 	case os.SEEK_END: | ||||||
|  | 		//++ check if new offset isn't out of bounds, set e when it is, then break out of switch | ||||||
|  | 		vf.offset = int64(len(vf.EmbeddedFile.Content)) - offset | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e != nil { | ||||||
|  | 		return 0, &os.PathError{ | ||||||
|  | 			Op:   "seek", | ||||||
|  | 			Path: vf.Filename, | ||||||
|  | 			Err:  e, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return vf.offset, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // virtualDir is a 'stateful' virtual directory. | ||||||
|  | // virtualDir wraps an *EmbeddedDir for a call to Box.Open() and virtualizes 'closing'. | ||||||
|  | // virtualDir is only internally visible and should be exposed through rice.File | ||||||
|  | type virtualDir struct { | ||||||
|  | 	*embedded.EmbeddedDir | ||||||
|  | 	offset int // readdir position on the directory | ||||||
|  | 	closed bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // create a new virtualDir for given EmbeddedDir | ||||||
|  | func newVirtualDir(ed *embedded.EmbeddedDir) *virtualDir { | ||||||
|  | 	vd := &virtualDir{ | ||||||
|  | 		EmbeddedDir: ed, | ||||||
|  | 		offset:      0, | ||||||
|  | 		closed:      false, | ||||||
|  | 	} | ||||||
|  | 	return vd | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vd *virtualDir) close() error { | ||||||
|  | 	//++ TODO: needs sync mutex? | ||||||
|  | 	if vd.closed { | ||||||
|  | 		return &os.PathError{ | ||||||
|  | 			Op:   "close", | ||||||
|  | 			Path: vd.EmbeddedDir.Filename, | ||||||
|  | 			Err:  errors.New("already closed"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	vd.closed = true | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vd *virtualDir) stat() (os.FileInfo, error) { | ||||||
|  | 	if vd.closed { | ||||||
|  | 		return nil, &os.PathError{ | ||||||
|  | 			Op:   "stat", | ||||||
|  | 			Path: vd.EmbeddedDir.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return (*embeddedDirInfo)(vd.EmbeddedDir), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vd *virtualDir) readdir(n int) (fi []os.FileInfo, err error) { | ||||||
|  |  | ||||||
|  | 	if vd.closed { | ||||||
|  | 		return nil, &os.PathError{ | ||||||
|  | 			Op:   "readdir", | ||||||
|  | 			Path: vd.EmbeddedDir.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Build up the array of our contents | ||||||
|  | 	var files []os.FileInfo | ||||||
|  |  | ||||||
|  | 	// Add the child directories | ||||||
|  | 	for _, child := range vd.ChildDirs { | ||||||
|  | 		child.Filename = filepath.Base(child.Filename) | ||||||
|  | 		files = append(files, (*embeddedDirInfo)(child)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add the child files | ||||||
|  | 	for _, child := range vd.ChildFiles { | ||||||
|  | 		child.Filename = filepath.Base(child.Filename) | ||||||
|  | 		files = append(files, (*embeddedFileInfo)(child)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Sort it by filename (lexical order) | ||||||
|  | 	sort.Sort(SortByName(files)) | ||||||
|  |  | ||||||
|  | 	// Return all contents if that's what is requested | ||||||
|  | 	if n <= 0 { | ||||||
|  | 		vd.offset = 0 | ||||||
|  | 		return files, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If user has requested past the end of our list | ||||||
|  | 	// return what we can and send an EOF | ||||||
|  | 	if vd.offset+n >= len(files) { | ||||||
|  | 		offset := vd.offset | ||||||
|  | 		vd.offset = 0 | ||||||
|  | 		return files[offset:], io.EOF | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	offset := vd.offset | ||||||
|  | 	vd.offset += n | ||||||
|  | 	return files[offset : offset+n], nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vd *virtualDir) read(bts []byte) (int, error) { | ||||||
|  | 	if vd.closed { | ||||||
|  | 		return 0, &os.PathError{ | ||||||
|  | 			Op:   "read", | ||||||
|  | 			Path: vd.EmbeddedDir.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return 0, &os.PathError{ | ||||||
|  | 		Op:   "read", | ||||||
|  | 		Path: vd.EmbeddedDir.Filename, | ||||||
|  | 		Err:  errors.New("is a directory"), | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (vd *virtualDir) seek(offset int64, whence int) (int64, error) { | ||||||
|  | 	if vd.closed { | ||||||
|  | 		return 0, &os.PathError{ | ||||||
|  | 			Op:   "seek", | ||||||
|  | 			Path: vd.EmbeddedDir.Filename, | ||||||
|  | 			Err:  errors.New("bad file descriptor"), | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return 0, &os.PathError{ | ||||||
|  | 		Op:   "seek", | ||||||
|  | 		Path: vd.Filename, | ||||||
|  | 		Err:  errors.New("is a directory"), | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										122
									
								
								vendor/github.com/GeertJohan/go.rice/walk.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								vendor/github.com/GeertJohan/go.rice/walk.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,122 @@ | |||||||
|  | package rice | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"os" | ||||||
|  | 	"path/filepath" | ||||||
|  | 	"sort" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Walk is like filepath.Walk() | ||||||
|  | // Visit http://golang.org/pkg/path/filepath/#Walk for more information | ||||||
|  | func (b *Box) Walk(path string, walkFn filepath.WalkFunc) error { | ||||||
|  |  | ||||||
|  | 	pathFile, err := b.Open(path) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer pathFile.Close() | ||||||
|  |  | ||||||
|  | 	pathInfo, err := pathFile.Stat() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if b.IsAppended() || b.IsEmbedded() { | ||||||
|  | 		return b.walk(path, pathInfo, walkFn) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// We don't have any embedded or appended box so use live filesystem mode | ||||||
|  | 	return filepath.Walk(b.absolutePath+string(os.PathSeparator)+path, func(path string, info os.FileInfo, err error) error { | ||||||
|  |  | ||||||
|  | 		// Strip out the box name from the returned paths | ||||||
|  | 		path = strings.TrimPrefix(path, b.absolutePath+string(os.PathSeparator)) | ||||||
|  | 		return walkFn(path, info, err) | ||||||
|  |  | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // walk recursively descends path. | ||||||
|  | // See walk() in $GOROOT/src/pkg/path/filepath/path.go | ||||||
|  | func (b *Box) walk(path string, info os.FileInfo, walkFn filepath.WalkFunc) error { | ||||||
|  |  | ||||||
|  | 	err := walkFn(path, info, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		if info.IsDir() && err == filepath.SkipDir { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !info.IsDir() { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	names, err := b.readDirNames(path) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return walkFn(path, info, err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, name := range names { | ||||||
|  |  | ||||||
|  | 		filename := filepath.Join(path, name) | ||||||
|  | 		fileObject, err := b.Open(filename) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		defer fileObject.Close() | ||||||
|  |  | ||||||
|  | 		fileInfo, err := fileObject.Stat() | ||||||
|  | 		if err != nil { | ||||||
|  | 			if err := walkFn(filename, fileInfo, err); err != nil && err != filepath.SkipDir { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			err = b.walk(filename, fileInfo, walkFn) | ||||||
|  | 			if err != nil { | ||||||
|  | 				if !fileInfo.IsDir() || err != filepath.SkipDir { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // readDirNames reads the directory named by path and returns a sorted list of directory entries. | ||||||
|  | // See readDirNames() in $GOROOT/pkg/path/filepath/path.go | ||||||
|  | func (b *Box) readDirNames(path string) ([]string, error) { | ||||||
|  |  | ||||||
|  | 	f, err := b.Open(path) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer f.Close() | ||||||
|  |  | ||||||
|  | 	stat, err := f.Stat() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !stat.IsDir() { | ||||||
|  | 		return nil, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	infos, err := f.Readdir(0) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var names []string | ||||||
|  |  | ||||||
|  | 	for _, info := range infos { | ||||||
|  | 		names = append(names, info.Name()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	sort.Strings(names) | ||||||
|  | 	return names, nil | ||||||
|  |  | ||||||
|  | } | ||||||
							
								
								
									
										64
									
								
								vendor/github.com/Sirupsen/logrus/alt_exit.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								vendor/github.com/Sirupsen/logrus/alt_exit.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,64 @@ | |||||||
|  | package logrus | ||||||
|  |  | ||||||
|  | // The following code was sourced and modified from the | ||||||
|  | // https://github.com/tebeka/atexit package governed by the following license: | ||||||
|  | // | ||||||
|  | // Copyright (c) 2012 Miki Tebeka <miki.tebeka@gmail.com>. | ||||||
|  | // | ||||||
|  | // 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. | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var handlers = []func(){} | ||||||
|  |  | ||||||
|  | func runHandler(handler func()) { | ||||||
|  | 	defer func() { | ||||||
|  | 		if err := recover(); err != nil { | ||||||
|  | 			fmt.Fprintln(os.Stderr, "Error: Logrus exit handler error:", err) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	handler() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func runHandlers() { | ||||||
|  | 	for _, handler := range handlers { | ||||||
|  | 		runHandler(handler) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Exit runs all the Logrus atexit handlers and then terminates the program using os.Exit(code) | ||||||
|  | func Exit(code int) { | ||||||
|  | 	runHandlers() | ||||||
|  | 	os.Exit(code) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RegisterExitHandler adds a Logrus Exit handler, call logrus.Exit to invoke | ||||||
|  | // all handlers. The handlers will also be invoked when any Fatal log entry is | ||||||
|  | // made. | ||||||
|  | // | ||||||
|  | // This method is useful when a caller wishes to use logrus to log a fatal | ||||||
|  | // message but also needs to gracefully shutdown. An example usecase could be | ||||||
|  | // closing database connections, or sending a alert that the application is | ||||||
|  | // closing. | ||||||
|  | func RegisterExitHandler(handler func()) { | ||||||
|  | 	handlers = append(handlers, handler) | ||||||
|  | } | ||||||
							
								
								
									
										51
									
								
								vendor/github.com/Sirupsen/logrus/entry.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										51
									
								
								vendor/github.com/Sirupsen/logrus/entry.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,11 +3,21 @@ package logrus | |||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"io" |  | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | var bufferPool *sync.Pool | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	bufferPool = &sync.Pool{ | ||||||
|  | 		New: func() interface{} { | ||||||
|  | 			return new(bytes.Buffer) | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Defines the key when adding errors using WithError. | // Defines the key when adding errors using WithError. | ||||||
| var ErrorKey = "error" | var ErrorKey = "error" | ||||||
|  |  | ||||||
| @@ -29,6 +39,9 @@ type Entry struct { | |||||||
|  |  | ||||||
| 	// Message passed to Debug, Info, Warn, Error, Fatal or Panic | 	// Message passed to Debug, Info, Warn, Error, Fatal or Panic | ||||||
| 	Message string | 	Message string | ||||||
|  |  | ||||||
|  | 	// When formatter is called in entry.log(), an Buffer may be set to entry | ||||||
|  | 	Buffer *bytes.Buffer | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewEntry(logger *Logger) *Entry { | func NewEntry(logger *Logger) *Entry { | ||||||
| @@ -39,21 +52,15 @@ func NewEntry(logger *Logger) *Entry { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // 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 | // Returns the string representation from the reader and ultimately the | ||||||
| // formatter. | // formatter. | ||||||
| func (entry *Entry) String() (string, error) { | func (entry *Entry) String() (string, error) { | ||||||
| 	reader, err := entry.Reader() | 	serialized, err := entry.Logger.Formatter.Format(entry) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return "", err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  | 	str := string(serialized) | ||||||
| 	return reader.String(), err | 	return str, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // Add an error as single field (using the key defined in ErrorKey) to the Entry. | // Add an error as single field (using the key defined in ErrorKey) to the Entry. | ||||||
| @@ -81,6 +88,7 @@ func (entry *Entry) WithFields(fields Fields) *Entry { | |||||||
| // This function is not declared with a pointer value because otherwise | // This function is not declared with a pointer value because otherwise | ||||||
| // race conditions will occur when using multiple goroutines | // race conditions will occur when using multiple goroutines | ||||||
| func (entry Entry) log(level Level, msg string) { | func (entry Entry) log(level Level, msg string) { | ||||||
|  | 	var buffer *bytes.Buffer | ||||||
| 	entry.Time = time.Now() | 	entry.Time = time.Now() | ||||||
| 	entry.Level = level | 	entry.Level = level | ||||||
| 	entry.Message = msg | 	entry.Message = msg | ||||||
| @@ -90,21 +98,24 @@ func (entry Entry) log(level Level, msg string) { | |||||||
| 		fmt.Fprintf(os.Stderr, "Failed to fire hook: %v\n", err) | 		fmt.Fprintf(os.Stderr, "Failed to fire hook: %v\n", err) | ||||||
| 		entry.Logger.mu.Unlock() | 		entry.Logger.mu.Unlock() | ||||||
| 	} | 	} | ||||||
|  | 	buffer = bufferPool.Get().(*bytes.Buffer) | ||||||
| 	reader, err := entry.Reader() | 	buffer.Reset() | ||||||
|  | 	defer bufferPool.Put(buffer) | ||||||
|  | 	entry.Buffer = buffer | ||||||
|  | 	serialized, err := entry.Logger.Formatter.Format(&entry) | ||||||
|  | 	entry.Buffer = nil | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		entry.Logger.mu.Lock() | 		entry.Logger.mu.Lock() | ||||||
| 		fmt.Fprintf(os.Stderr, "Failed to obtain reader, %v\n", err) | 		fmt.Fprintf(os.Stderr, "Failed to obtain reader, %v\n", err) | ||||||
| 		entry.Logger.mu.Unlock() | 		entry.Logger.mu.Unlock() | ||||||
| 	} | 	} else { | ||||||
|  |  | ||||||
| 		entry.Logger.mu.Lock() | 		entry.Logger.mu.Lock() | ||||||
| 	defer entry.Logger.mu.Unlock() | 		_, err = entry.Logger.Out.Write(serialized) | ||||||
|  |  | ||||||
| 	_, err = io.Copy(entry.Logger.Out, reader) |  | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			fmt.Fprintf(os.Stderr, "Failed to write to log, %v\n", err) | 			fmt.Fprintf(os.Stderr, "Failed to write to log, %v\n", err) | ||||||
| 		} | 		} | ||||||
|  | 		entry.Logger.mu.Unlock() | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// To avoid Entry#log() returning a value that only would make sense for | 	// 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 | 	// panic() to use in Entry#Panic(), we avoid the allocation by checking | ||||||
| @@ -150,7 +161,7 @@ func (entry *Entry) Fatal(args ...interface{}) { | |||||||
| 	if entry.Logger.Level >= FatalLevel { | 	if entry.Logger.Level >= FatalLevel { | ||||||
| 		entry.log(FatalLevel, fmt.Sprint(args...)) | 		entry.log(FatalLevel, fmt.Sprint(args...)) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (entry *Entry) Panic(args ...interface{}) { | func (entry *Entry) Panic(args ...interface{}) { | ||||||
| @@ -198,7 +209,7 @@ func (entry *Entry) Fatalf(format string, args ...interface{}) { | |||||||
| 	if entry.Logger.Level >= FatalLevel { | 	if entry.Logger.Level >= FatalLevel { | ||||||
| 		entry.Fatal(fmt.Sprintf(format, args...)) | 		entry.Fatal(fmt.Sprintf(format, args...)) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (entry *Entry) Panicf(format string, args ...interface{}) { | func (entry *Entry) Panicf(format string, args ...interface{}) { | ||||||
| @@ -245,7 +256,7 @@ func (entry *Entry) Fatalln(args ...interface{}) { | |||||||
| 	if entry.Logger.Level >= FatalLevel { | 	if entry.Logger.Level >= FatalLevel { | ||||||
| 		entry.Fatal(entry.sprintlnn(args...)) | 		entry.Fatal(entry.sprintlnn(args...)) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (entry *Entry) Panicln(args ...interface{}) { | func (entry *Entry) Panicln(args ...interface{}) { | ||||||
|   | |||||||
							
								
								
									
										9
									
								
								vendor/github.com/Sirupsen/logrus/examples/basic/basic.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										9
									
								
								vendor/github.com/Sirupsen/logrus/examples/basic/basic.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -2,6 +2,7 @@ package main | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/Sirupsen/logrus" | 	"github.com/Sirupsen/logrus" | ||||||
|  | 	// "os" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var log = logrus.New() | var log = logrus.New() | ||||||
| @@ -9,6 +10,14 @@ var log = logrus.New() | |||||||
| func init() { | func init() { | ||||||
| 	log.Formatter = new(logrus.JSONFormatter) | 	log.Formatter = new(logrus.JSONFormatter) | ||||||
| 	log.Formatter = new(logrus.TextFormatter) // default | 	log.Formatter = new(logrus.TextFormatter) // default | ||||||
|  |  | ||||||
|  | 	// file, err := os.OpenFile("logrus.log", os.O_CREATE|os.O_WRONLY, 0666) | ||||||
|  | 	// if err == nil { | ||||||
|  | 	// 	log.Out = file | ||||||
|  | 	// } else { | ||||||
|  | 	// 	log.Info("Failed to log to file, using default stderr") | ||||||
|  | 	// } | ||||||
|  |  | ||||||
| 	log.Level = logrus.DebugLevel | 	log.Level = logrus.DebugLevel | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										15
									
								
								vendor/github.com/Sirupsen/logrus/formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										15
									
								
								vendor/github.com/Sirupsen/logrus/formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -31,18 +31,15 @@ type Formatter interface { | |||||||
| // It's not exported because it's still using Data in an opinionated way. It's to | // 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. | // avoid code duplication between the two default formatters. | ||||||
| func prefixFieldClashes(data Fields) { | func prefixFieldClashes(data Fields) { | ||||||
| 	_, ok := data["time"] | 	if t, ok := data["time"]; ok { | ||||||
| 	if ok { | 		data["fields.time"] = t | ||||||
| 		data["fields.time"] = data["time"] |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	_, ok = data["msg"] | 	if m, ok := data["msg"]; ok { | ||||||
| 	if ok { | 		data["fields.msg"] = m | ||||||
| 		data["fields.msg"] = data["msg"] |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	_, ok = data["level"] | 	if l, ok := data["level"]; ok { | ||||||
| 	if ok { | 		data["fields.level"] = l | ||||||
| 		data["fields.level"] = data["level"] |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										61
									
								
								vendor/github.com/Sirupsen/logrus/formatters/logstash/logstash.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										61
									
								
								vendor/github.com/Sirupsen/logrus/formatters/logstash/logstash.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,61 +0,0 @@ | |||||||
| 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 |  | ||||||
| } |  | ||||||
							
								
								
									
										39
									
								
								vendor/github.com/Sirupsen/logrus/json_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										39
									
								
								vendor/github.com/Sirupsen/logrus/json_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -5,9 +5,40 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | type fieldKey string | ||||||
|  | type FieldMap map[fieldKey]string | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	FieldKeyMsg   = "msg" | ||||||
|  | 	FieldKeyLevel = "level" | ||||||
|  | 	FieldKeyTime  = "time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func (f FieldMap) resolve(key fieldKey) string { | ||||||
|  | 	if k, ok := f[key]; ok { | ||||||
|  | 		return k | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return string(key) | ||||||
|  | } | ||||||
|  |  | ||||||
| type JSONFormatter struct { | type JSONFormatter struct { | ||||||
| 	// TimestampFormat sets the format used for marshaling timestamps. | 	// TimestampFormat sets the format used for marshaling timestamps. | ||||||
| 	TimestampFormat string | 	TimestampFormat string | ||||||
|  |  | ||||||
|  | 	// DisableTimestamp allows disabling automatic timestamps in output | ||||||
|  | 	DisableTimestamp bool | ||||||
|  |  | ||||||
|  | 	// FieldMap allows users to customize the names of keys for various fields. | ||||||
|  | 	// As an example: | ||||||
|  | 	// formatter := &JSONFormatter{ | ||||||
|  | 	//   	FieldMap: FieldMap{ | ||||||
|  | 	// 		 FieldKeyTime: "@timestamp", | ||||||
|  | 	// 		 FieldKeyLevel: "@level", | ||||||
|  | 	// 		 FieldKeyLevel: "@message", | ||||||
|  | 	//    }, | ||||||
|  | 	// } | ||||||
|  | 	FieldMap FieldMap | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *JSONFormatter) Format(entry *Entry) ([]byte, error) { | func (f *JSONFormatter) Format(entry *Entry) ([]byte, error) { | ||||||
| @@ -29,9 +60,11 @@ func (f *JSONFormatter) Format(entry *Entry) ([]byte, error) { | |||||||
| 		timestampFormat = DefaultTimestampFormat | 		timestampFormat = DefaultTimestampFormat | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	data["time"] = entry.Time.Format(timestampFormat) | 	if !f.DisableTimestamp { | ||||||
| 	data["msg"] = entry.Message | 		data[f.FieldMap.resolve(FieldKeyTime)] = entry.Time.Format(timestampFormat) | ||||||
| 	data["level"] = entry.Level.String() | 	} | ||||||
|  | 	data[f.FieldMap.resolve(FieldKeyMsg)] = entry.Message | ||||||
|  | 	data[f.FieldMap.resolve(FieldKeyLevel)] = entry.Level.String() | ||||||
|  |  | ||||||
| 	serialized, err := json.Marshal(data) | 	serialized, err := json.Marshal(data) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
|   | |||||||
							
								
								
									
										162
									
								
								vendor/github.com/Sirupsen/logrus/logger.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										162
									
								
								vendor/github.com/Sirupsen/logrus/logger.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -26,8 +26,31 @@ type Logger struct { | |||||||
| 	// to) `logrus.Info`, which allows Info(), Warn(), Error() and Fatal() to be | 	// to) `logrus.Info`, which allows Info(), Warn(), Error() and Fatal() to be | ||||||
| 	// logged. `logrus.Debug` is useful in | 	// logged. `logrus.Debug` is useful in | ||||||
| 	Level Level | 	Level Level | ||||||
| 	// Used to sync writing to the log. | 	// Used to sync writing to the log. Locking is enabled by Default | ||||||
| 	mu sync.Mutex | 	mu MutexWrap | ||||||
|  | 	// Reusable empty entry | ||||||
|  | 	entryPool sync.Pool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type MutexWrap struct { | ||||||
|  | 	lock     sync.Mutex | ||||||
|  | 	disabled bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (mw *MutexWrap) Lock() { | ||||||
|  | 	if !mw.disabled { | ||||||
|  | 		mw.lock.Lock() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (mw *MutexWrap) Unlock() { | ||||||
|  | 	if !mw.disabled { | ||||||
|  | 		mw.lock.Unlock() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (mw *MutexWrap) Disable() { | ||||||
|  | 	mw.disabled = true | ||||||
| } | } | ||||||
|  |  | ||||||
| // Creates a new logger. Configuration should be set by changing `Formatter`, | // Creates a new logger. Configuration should be set by changing `Formatter`, | ||||||
| @@ -51,162 +74,235 @@ func New() *Logger { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Adds a field to the log entry, note that you it doesn't log until you call | func (logger *Logger) newEntry() *Entry { | ||||||
|  | 	entry, ok := logger.entryPool.Get().(*Entry) | ||||||
|  | 	if ok { | ||||||
|  | 		return entry | ||||||
|  | 	} | ||||||
|  | 	return NewEntry(logger) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (logger *Logger) releaseEntry(entry *Entry) { | ||||||
|  | 	logger.entryPool.Put(entry) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Adds a field to the log entry, note that it doesn't log until you call | ||||||
| // Debug, Print, Info, Warn, Fatal or Panic. It only creates a log entry. | // Debug, Print, Info, Warn, Fatal or Panic. It only creates a log entry. | ||||||
| // If you want multiple fields, use `WithFields`. | // If you want multiple fields, use `WithFields`. | ||||||
| func (logger *Logger) WithField(key string, value interface{}) *Entry { | func (logger *Logger) WithField(key string, value interface{}) *Entry { | ||||||
| 	return NewEntry(logger).WithField(key, value) | 	entry := logger.newEntry() | ||||||
|  | 	defer logger.releaseEntry(entry) | ||||||
|  | 	return entry.WithField(key, value) | ||||||
| } | } | ||||||
|  |  | ||||||
| // Adds a struct of fields to the log entry. All it does is call `WithField` for | // Adds a struct of fields to the log entry. All it does is call `WithField` for | ||||||
| // each `Field`. | // each `Field`. | ||||||
| func (logger *Logger) WithFields(fields Fields) *Entry { | func (logger *Logger) WithFields(fields Fields) *Entry { | ||||||
| 	return NewEntry(logger).WithFields(fields) | 	entry := logger.newEntry() | ||||||
|  | 	defer logger.releaseEntry(entry) | ||||||
|  | 	return entry.WithFields(fields) | ||||||
| } | } | ||||||
|  |  | ||||||
| // Add an error as single field to the log entry.  All it does is call | // Add an error as single field to the log entry.  All it does is call | ||||||
| // `WithError` for the given `error`. | // `WithError` for the given `error`. | ||||||
| func (logger *Logger) WithError(err error) *Entry { | func (logger *Logger) WithError(err error) *Entry { | ||||||
| 	return NewEntry(logger).WithError(err) | 	entry := logger.newEntry() | ||||||
|  | 	defer logger.releaseEntry(entry) | ||||||
|  | 	return entry.WithError(err) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Debugf(format string, args ...interface{}) { | func (logger *Logger) Debugf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= DebugLevel { | 	if logger.Level >= DebugLevel { | ||||||
| 		NewEntry(logger).Debugf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Debugf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Infof(format string, args ...interface{}) { | func (logger *Logger) Infof(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= InfoLevel { | 	if logger.Level >= InfoLevel { | ||||||
| 		NewEntry(logger).Infof(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Infof(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Printf(format string, args ...interface{}) { | func (logger *Logger) Printf(format string, args ...interface{}) { | ||||||
| 	NewEntry(logger).Printf(format, args...) | 	entry := logger.newEntry() | ||||||
|  | 	entry.Printf(format, args...) | ||||||
|  | 	logger.releaseEntry(entry) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warnf(format string, args ...interface{}) { | func (logger *Logger) Warnf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warnf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warnf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warningf(format string, args ...interface{}) { | func (logger *Logger) Warningf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warnf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warnf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Errorf(format string, args ...interface{}) { | func (logger *Logger) Errorf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= ErrorLevel { | 	if logger.Level >= ErrorLevel { | ||||||
| 		NewEntry(logger).Errorf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Errorf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Fatalf(format string, args ...interface{}) { | func (logger *Logger) Fatalf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= FatalLevel { | 	if logger.Level >= FatalLevel { | ||||||
| 		NewEntry(logger).Fatalf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Fatalf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Panicf(format string, args ...interface{}) { | func (logger *Logger) Panicf(format string, args ...interface{}) { | ||||||
| 	if logger.Level >= PanicLevel { | 	if logger.Level >= PanicLevel { | ||||||
| 		NewEntry(logger).Panicf(format, args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Panicf(format, args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Debug(args ...interface{}) { | func (logger *Logger) Debug(args ...interface{}) { | ||||||
| 	if logger.Level >= DebugLevel { | 	if logger.Level >= DebugLevel { | ||||||
| 		NewEntry(logger).Debug(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Debug(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Info(args ...interface{}) { | func (logger *Logger) Info(args ...interface{}) { | ||||||
| 	if logger.Level >= InfoLevel { | 	if logger.Level >= InfoLevel { | ||||||
| 		NewEntry(logger).Info(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Info(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Print(args ...interface{}) { | func (logger *Logger) Print(args ...interface{}) { | ||||||
| 	NewEntry(logger).Info(args...) | 	entry := logger.newEntry() | ||||||
|  | 	entry.Info(args...) | ||||||
|  | 	logger.releaseEntry(entry) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warn(args ...interface{}) { | func (logger *Logger) Warn(args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warn(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warn(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warning(args ...interface{}) { | func (logger *Logger) Warning(args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warn(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warn(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Error(args ...interface{}) { | func (logger *Logger) Error(args ...interface{}) { | ||||||
| 	if logger.Level >= ErrorLevel { | 	if logger.Level >= ErrorLevel { | ||||||
| 		NewEntry(logger).Error(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Error(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Fatal(args ...interface{}) { | func (logger *Logger) Fatal(args ...interface{}) { | ||||||
| 	if logger.Level >= FatalLevel { | 	if logger.Level >= FatalLevel { | ||||||
| 		NewEntry(logger).Fatal(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Fatal(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Panic(args ...interface{}) { | func (logger *Logger) Panic(args ...interface{}) { | ||||||
| 	if logger.Level >= PanicLevel { | 	if logger.Level >= PanicLevel { | ||||||
| 		NewEntry(logger).Panic(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Panic(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Debugln(args ...interface{}) { | func (logger *Logger) Debugln(args ...interface{}) { | ||||||
| 	if logger.Level >= DebugLevel { | 	if logger.Level >= DebugLevel { | ||||||
| 		NewEntry(logger).Debugln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Debugln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Infoln(args ...interface{}) { | func (logger *Logger) Infoln(args ...interface{}) { | ||||||
| 	if logger.Level >= InfoLevel { | 	if logger.Level >= InfoLevel { | ||||||
| 		NewEntry(logger).Infoln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Infoln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Println(args ...interface{}) { | func (logger *Logger) Println(args ...interface{}) { | ||||||
| 	NewEntry(logger).Println(args...) | 	entry := logger.newEntry() | ||||||
|  | 	entry.Println(args...) | ||||||
|  | 	logger.releaseEntry(entry) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warnln(args ...interface{}) { | func (logger *Logger) Warnln(args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warnln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warnln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Warningln(args ...interface{}) { | func (logger *Logger) Warningln(args ...interface{}) { | ||||||
| 	if logger.Level >= WarnLevel { | 	if logger.Level >= WarnLevel { | ||||||
| 		NewEntry(logger).Warnln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Warnln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Errorln(args ...interface{}) { | func (logger *Logger) Errorln(args ...interface{}) { | ||||||
| 	if logger.Level >= ErrorLevel { | 	if logger.Level >= ErrorLevel { | ||||||
| 		NewEntry(logger).Errorln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Errorln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Fatalln(args ...interface{}) { | func (logger *Logger) Fatalln(args ...interface{}) { | ||||||
| 	if logger.Level >= FatalLevel { | 	if logger.Level >= FatalLevel { | ||||||
| 		NewEntry(logger).Fatalln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Fatalln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| 	os.Exit(1) | 	Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) Panicln(args ...interface{}) { | func (logger *Logger) Panicln(args ...interface{}) { | ||||||
| 	if logger.Level >= PanicLevel { | 	if logger.Level >= PanicLevel { | ||||||
| 		NewEntry(logger).Panicln(args...) | 		entry := logger.newEntry() | ||||||
|  | 		entry.Panicln(args...) | ||||||
|  | 		logger.releaseEntry(entry) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | //When file is opened with appending mode, it's safe to | ||||||
|  | //write concurrently to a file (within 4k message on Linux). | ||||||
|  | //In these cases user can choose to disable the lock. | ||||||
|  | func (logger *Logger) SetNoLock() { | ||||||
|  | 	logger.mu.Disable() | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_appengine.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_appengine.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,10 @@ | |||||||
|  | // +build appengine | ||||||
|  |  | ||||||
|  | package logrus | ||||||
|  |  | ||||||
|  | import "io" | ||||||
|  |  | ||||||
|  | // IsTerminal returns true if stderr's file descriptor is a terminal. | ||||||
|  | func IsTerminal(f io.Writer) bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
							
								
								
									
										1
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,4 +1,5 @@ | |||||||
| // +build darwin freebsd openbsd netbsd dragonfly | // +build darwin freebsd openbsd netbsd dragonfly | ||||||
|  | // +build !appengine | ||||||
|  |  | ||||||
| package logrus | package logrus | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										2
									
								
								vendor/github.com/Sirupsen/logrus/terminal_linux.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/Sirupsen/logrus/terminal_linux.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,8 @@ | |||||||
| // Use of this source code is governed by a BSD-style | // Use of this source code is governed by a BSD-style | ||||||
| // license that can be found in the LICENSE file. | // license that can be found in the LICENSE file. | ||||||
|  |  | ||||||
|  | // +build !appengine | ||||||
|  |  | ||||||
| package logrus | package logrus | ||||||
|  |  | ||||||
| import "syscall" | import "syscall" | ||||||
|   | |||||||
							
								
								
									
										13
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -4,18 +4,25 @@ | |||||||
| // license that can be found in the LICENSE file. | // license that can be found in the LICENSE file. | ||||||
|  |  | ||||||
| // +build linux darwin freebsd openbsd netbsd dragonfly | // +build linux darwin freebsd openbsd netbsd dragonfly | ||||||
|  | // +build !appengine | ||||||
|  |  | ||||||
| package logrus | package logrus | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"io" | ||||||
|  | 	"os" | ||||||
| 	"syscall" | 	"syscall" | ||||||
| 	"unsafe" | 	"unsafe" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // IsTerminal returns true if stderr's file descriptor is a terminal. | // IsTerminal returns true if stderr's file descriptor is a terminal. | ||||||
| func IsTerminal() bool { | func IsTerminal(f io.Writer) bool { | ||||||
| 	fd := syscall.Stderr |  | ||||||
| 	var termios Termios | 	var termios Termios | ||||||
| 	_, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(fd), ioctlReadTermios, uintptr(unsafe.Pointer(&termios)), 0, 0, 0) | 	switch v := f.(type) { | ||||||
|  | 	case *os.File: | ||||||
|  | 		_, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(v.Fd()), ioctlReadTermios, uintptr(unsafe.Pointer(&termios)), 0, 0, 0) | ||||||
| 		return err == 0 | 		return err == 0 | ||||||
|  | 	default: | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										12
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										12
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,15 +1,21 @@ | |||||||
| // +build solaris | // +build solaris,!appengine | ||||||
|  |  | ||||||
| package logrus | package logrus | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"io" | ||||||
| 	"os" | 	"os" | ||||||
|  |  | ||||||
| 	"golang.org/x/sys/unix" | 	"golang.org/x/sys/unix" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // IsTerminal returns true if the given file descriptor is a terminal. | // IsTerminal returns true if the given file descriptor is a terminal. | ||||||
| func IsTerminal() bool { | func IsTerminal(f io.Writer) bool { | ||||||
| 	_, err := unix.IoctlGetTermios(int(os.Stdout.Fd()), unix.TCGETA) | 	switch v := f.(type) { | ||||||
|  | 	case *os.File: | ||||||
|  | 		_, err := unix.IoctlGetTermios(int(v.Fd()), unix.TCGETA) | ||||||
| 		return err == nil | 		return err == nil | ||||||
|  | 	default: | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										14
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,11 +3,13 @@ | |||||||
| // Use of this source code is governed by a BSD-style | // Use of this source code is governed by a BSD-style | ||||||
| // license that can be found in the LICENSE file. | // license that can be found in the LICENSE file. | ||||||
|  |  | ||||||
| // +build windows | // +build windows,!appengine | ||||||
|  |  | ||||||
| package logrus | package logrus | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"io" | ||||||
|  | 	"os" | ||||||
| 	"syscall" | 	"syscall" | ||||||
| 	"unsafe" | 	"unsafe" | ||||||
| ) | ) | ||||||
| @@ -19,9 +21,13 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| // IsTerminal returns true if stderr's file descriptor is a terminal. | // IsTerminal returns true if stderr's file descriptor is a terminal. | ||||||
| func IsTerminal() bool { | func IsTerminal(f io.Writer) bool { | ||||||
| 	fd := syscall.Stderr | 	switch v := f.(type) { | ||||||
|  | 	case *os.File: | ||||||
| 		var st uint32 | 		var st uint32 | ||||||
| 	r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(fd), uintptr(unsafe.Pointer(&st)), 0) | 		r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(v.Fd()), uintptr(unsafe.Pointer(&st)), 0) | ||||||
| 		return r != 0 && e == 0 | 		return r != 0 && e == 0 | ||||||
|  | 	default: | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										78
									
								
								vendor/github.com/Sirupsen/logrus/text_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										78
									
								
								vendor/github.com/Sirupsen/logrus/text_formatter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,9 +3,9 @@ package logrus | |||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"runtime" |  | ||||||
| 	"sort" | 	"sort" | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -20,16 +20,10 @@ const ( | |||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	baseTimestamp time.Time | 	baseTimestamp time.Time | ||||||
| 	isTerminal    bool |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	baseTimestamp = time.Now() | 	baseTimestamp = time.Now() | ||||||
| 	isTerminal = IsTerminal() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func miniTS() int { |  | ||||||
| 	return int(time.Since(baseTimestamp) / time.Second) |  | ||||||
| } | } | ||||||
|  |  | ||||||
| type TextFormatter struct { | type TextFormatter struct { | ||||||
| @@ -54,10 +48,32 @@ type TextFormatter struct { | |||||||
| 	// that log extremely frequently and don't use the JSON formatter this may not | 	// that log extremely frequently and don't use the JSON formatter this may not | ||||||
| 	// be desired. | 	// be desired. | ||||||
| 	DisableSorting bool | 	DisableSorting bool | ||||||
|  |  | ||||||
|  | 	// QuoteEmptyFields will wrap empty fields in quotes if true | ||||||
|  | 	QuoteEmptyFields bool | ||||||
|  |  | ||||||
|  | 	// QuoteCharacter can be set to the override the default quoting character " | ||||||
|  | 	// with something else. For example: ', or `. | ||||||
|  | 	QuoteCharacter string | ||||||
|  |  | ||||||
|  | 	// Whether the logger's out is to a terminal | ||||||
|  | 	isTerminal bool | ||||||
|  |  | ||||||
|  | 	sync.Once | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (f *TextFormatter) init(entry *Entry) { | ||||||
|  | 	if len(f.QuoteCharacter) == 0 { | ||||||
|  | 		f.QuoteCharacter = "\"" | ||||||
|  | 	} | ||||||
|  | 	if entry.Logger != nil { | ||||||
|  | 		f.isTerminal = IsTerminal(entry.Logger.Out) | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *TextFormatter) Format(entry *Entry) ([]byte, error) { | func (f *TextFormatter) Format(entry *Entry) ([]byte, error) { | ||||||
| 	var keys []string = make([]string, 0, len(entry.Data)) | 	var b *bytes.Buffer | ||||||
|  | 	keys := make([]string, 0, len(entry.Data)) | ||||||
| 	for k := range entry.Data { | 	for k := range entry.Data { | ||||||
| 		keys = append(keys, k) | 		keys = append(keys, k) | ||||||
| 	} | 	} | ||||||
| @@ -65,13 +81,17 @@ func (f *TextFormatter) Format(entry *Entry) ([]byte, error) { | |||||||
| 	if !f.DisableSorting { | 	if !f.DisableSorting { | ||||||
| 		sort.Strings(keys) | 		sort.Strings(keys) | ||||||
| 	} | 	} | ||||||
|  | 	if entry.Buffer != nil { | ||||||
| 	b := &bytes.Buffer{} | 		b = entry.Buffer | ||||||
|  | 	} else { | ||||||
|  | 		b = &bytes.Buffer{} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	prefixFieldClashes(entry.Data) | 	prefixFieldClashes(entry.Data) | ||||||
|  |  | ||||||
| 	isColorTerminal := isTerminal && (runtime.GOOS != "windows") | 	f.Do(func() { f.init(entry) }) | ||||||
| 	isColored := (f.ForceColors || isColorTerminal) && !f.DisableColors |  | ||||||
|  | 	isColored := (f.ForceColors || f.isTerminal) && !f.DisableColors | ||||||
|  |  | ||||||
| 	timestampFormat := f.TimestampFormat | 	timestampFormat := f.TimestampFormat | ||||||
| 	if timestampFormat == "" { | 	if timestampFormat == "" { | ||||||
| @@ -111,51 +131,59 @@ func (f *TextFormatter) printColored(b *bytes.Buffer, entry *Entry, keys []strin | |||||||
|  |  | ||||||
| 	levelText := strings.ToUpper(entry.Level.String())[0:4] | 	levelText := strings.ToUpper(entry.Level.String())[0:4] | ||||||
|  |  | ||||||
| 	if !f.FullTimestamp { | 	if f.DisableTimestamp { | ||||||
| 		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%04d] %-44s ", levelColor, levelText, miniTS(), entry.Message) | 		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m %-44s ", levelColor, levelText, entry.Message) | ||||||
|  | 	} else if !f.FullTimestamp { | ||||||
|  | 		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%04d] %-44s ", levelColor, levelText, int(entry.Time.Sub(baseTimestamp)/time.Second), entry.Message) | ||||||
| 	} else { | 	} else { | ||||||
| 		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%s] %-44s ", levelColor, levelText, entry.Time.Format(timestampFormat), entry.Message) | 		fmt.Fprintf(b, "\x1b[%dm%s\x1b[0m[%s] %-44s ", levelColor, levelText, entry.Time.Format(timestampFormat), entry.Message) | ||||||
| 	} | 	} | ||||||
| 	for _, k := range keys { | 	for _, k := range keys { | ||||||
| 		v := entry.Data[k] | 		v := entry.Data[k] | ||||||
| 		fmt.Fprintf(b, " \x1b[%dm%s\x1b[0m=%+v", levelColor, k, v) | 		fmt.Fprintf(b, " \x1b[%dm%s\x1b[0m=", levelColor, k) | ||||||
|  | 		f.appendValue(b, v) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func needsQuoting(text string) bool { | func (f *TextFormatter) needsQuoting(text string) bool { | ||||||
|  | 	if f.QuoteEmptyFields && len(text) == 0 { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
| 	for _, ch := range text { | 	for _, ch := range text { | ||||||
| 		if !((ch >= 'a' && ch <= 'z') || | 		if !((ch >= 'a' && ch <= 'z') || | ||||||
| 			(ch >= 'A' && ch <= 'Z') || | 			(ch >= 'A' && ch <= 'Z') || | ||||||
| 			(ch >= '0' && ch <= '9') || | 			(ch >= '0' && ch <= '9') || | ||||||
| 			ch == '-' || ch == '.') { | 			ch == '-' || ch == '.') { | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 			return true | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *TextFormatter) appendKeyValue(b *bytes.Buffer, key string, value interface{}) { | func (f *TextFormatter) appendKeyValue(b *bytes.Buffer, key string, value interface{}) { | ||||||
|  |  | ||||||
| 	b.WriteString(key) | 	b.WriteString(key) | ||||||
| 	b.WriteByte('=') | 	b.WriteByte('=') | ||||||
|  | 	f.appendValue(b, value) | ||||||
|  | 	b.WriteByte(' ') | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (f *TextFormatter) appendValue(b *bytes.Buffer, value interface{}) { | ||||||
| 	switch value := value.(type) { | 	switch value := value.(type) { | ||||||
| 	case string: | 	case string: | ||||||
| 		if needsQuoting(value) { | 		if !f.needsQuoting(value) { | ||||||
| 			b.WriteString(value) | 			b.WriteString(value) | ||||||
| 		} else { | 		} else { | ||||||
| 			fmt.Fprintf(b, "%q", value) | 			fmt.Fprintf(b, "%s%v%s", f.QuoteCharacter, value, f.QuoteCharacter) | ||||||
| 		} | 		} | ||||||
| 	case error: | 	case error: | ||||||
| 		errmsg := value.Error() | 		errmsg := value.Error() | ||||||
| 		if needsQuoting(errmsg) { | 		if !f.needsQuoting(errmsg) { | ||||||
| 			b.WriteString(errmsg) | 			b.WriteString(errmsg) | ||||||
| 		} else { | 		} else { | ||||||
| 			fmt.Fprintf(b, "%q", value) | 			fmt.Fprintf(b, "%s%v%s", f.QuoteCharacter, errmsg, f.QuoteCharacter) | ||||||
| 		} | 		} | ||||||
| 	default: | 	default: | ||||||
| 		fmt.Fprint(b, value) | 		fmt.Fprint(b, value) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.WriteByte(' ') |  | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										39
									
								
								vendor/github.com/Sirupsen/logrus/writer.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										39
									
								
								vendor/github.com/Sirupsen/logrus/writer.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -7,21 +7,52 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| func (logger *Logger) Writer() *io.PipeWriter { | func (logger *Logger) Writer() *io.PipeWriter { | ||||||
|  | 	return logger.WriterLevel(InfoLevel) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (logger *Logger) WriterLevel(level Level) *io.PipeWriter { | ||||||
|  | 	return NewEntry(logger).WriterLevel(level) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (entry *Entry) Writer() *io.PipeWriter { | ||||||
|  | 	return entry.WriterLevel(InfoLevel) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (entry *Entry) WriterLevel(level Level) *io.PipeWriter { | ||||||
| 	reader, writer := io.Pipe() | 	reader, writer := io.Pipe() | ||||||
|  |  | ||||||
| 	go logger.writerScanner(reader) | 	var printFunc func(args ...interface{}) | ||||||
|  |  | ||||||
|  | 	switch level { | ||||||
|  | 	case DebugLevel: | ||||||
|  | 		printFunc = entry.Debug | ||||||
|  | 	case InfoLevel: | ||||||
|  | 		printFunc = entry.Info | ||||||
|  | 	case WarnLevel: | ||||||
|  | 		printFunc = entry.Warn | ||||||
|  | 	case ErrorLevel: | ||||||
|  | 		printFunc = entry.Error | ||||||
|  | 	case FatalLevel: | ||||||
|  | 		printFunc = entry.Fatal | ||||||
|  | 	case PanicLevel: | ||||||
|  | 		printFunc = entry.Panic | ||||||
|  | 	default: | ||||||
|  | 		printFunc = entry.Print | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	go entry.writerScanner(reader, printFunc) | ||||||
| 	runtime.SetFinalizer(writer, writerFinalizer) | 	runtime.SetFinalizer(writer, writerFinalizer) | ||||||
|  |  | ||||||
| 	return writer | 	return writer | ||||||
| } | } | ||||||
|  |  | ||||||
| func (logger *Logger) writerScanner(reader *io.PipeReader) { | func (entry *Entry) writerScanner(reader *io.PipeReader, printFunc func(args ...interface{})) { | ||||||
| 	scanner := bufio.NewScanner(reader) | 	scanner := bufio.NewScanner(reader) | ||||||
| 	for scanner.Scan() { | 	for scanner.Scan() { | ||||||
| 		logger.Print(scanner.Text()) | 		printFunc(scanner.Text()) | ||||||
| 	} | 	} | ||||||
| 	if err := scanner.Err(); err != nil { | 	if err := scanner.Err(); err != nil { | ||||||
| 		logger.Errorf("Error while reading from Writer: %s", err) | 		entry.Errorf("Error while reading from Writer: %s", err) | ||||||
| 	} | 	} | ||||||
| 	reader.Close() | 	reader.Close() | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										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. | ||||||
|  |  | ||||||
							
								
								
									
										125
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										125
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,125 @@ | |||||||
|  | // 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" | ||||||
|  |  | ||||||
|  | // VERSION of Discordgo, follows Symantic Versioning. (http://semver.org/) | ||||||
|  | const VERSION = "0.15.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. | ||||||
|  | //         IF THE TOKEN IS FOR A BOT, IT MUST BE PREFIXED WITH `BOT ` | ||||||
|  | //         eg: `"Bot <token>"` | ||||||
|  | //     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(), | ||||||
|  | 		ratelimiter:            NewRatelimiter(), | ||||||
|  | 		StateEnabled:           true, | ||||||
|  | 		Compress:               true, | ||||||
|  | 		ShouldReconnectOnError: true, | ||||||
|  | 		ShardID:                0, | ||||||
|  | 		ShardCount:             1, | ||||||
|  | 		MaxRestRetries:         3, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// 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 | ||||||
|  | } | ||||||
							
								
								
									
										117
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										117
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,117 @@ | |||||||
|  | // 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" | ||||||
|  | 	EndpointWebhooks = EndpointAPI + "webhooks/" | ||||||
|  |  | ||||||
|  | 	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 } | ||||||
|  | 	EndpointGuildMemberRole      = func(gID, uID, rID string) string { return EndpointGuilds + gID + "/members/" + uID + "/roles/" + rID } | ||||||
|  | 	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" } | ||||||
|  | 	EndpointGuildWebhooks        = func(gID string) string { return EndpointGuilds + gID + "/webhooks" } | ||||||
|  |  | ||||||
|  | 	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 } | ||||||
|  |  | ||||||
|  | 	EndpointChannelWebhooks = func(cID string) string { return EndpointChannel(cID) + "/webhooks" } | ||||||
|  | 	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID } | ||||||
|  | 	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token } | ||||||
|  |  | ||||||
|  | 	EndpointMessageReactions = func(cID, mID, eID string) string { | ||||||
|  | 		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID | ||||||
|  | 	} | ||||||
|  | 	EndpointMessageReaction = func(cID, mID, eID, uID string) string { | ||||||
|  | 		return EndpointMessageReactions(cID, mID, eID) + "/" + uID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	EndpointRelationships       = func() string { return EndpointUsers + "@me" + "/relationships" } | ||||||
|  | 	EndpointRelationship        = func(uID string) string { return EndpointRelationships() + "/" + uID } | ||||||
|  | 	EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" } | ||||||
|  |  | ||||||
|  | 	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" } | ||||||
|  | ) | ||||||
							
								
								
									
										238
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										238
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,238 @@ | |||||||
|  | package discordgo | ||||||
|  |  | ||||||
|  | import "fmt" | ||||||
|  |  | ||||||
|  | // EventHandler is an interface for Discord events. | ||||||
|  | type EventHandler interface { | ||||||
|  | 	// Type returns the type of event this handler belongs to. | ||||||
|  | 	Type() string | ||||||
|  |  | ||||||
|  | 	// Handle is called whenever an event of Type() happens. | ||||||
|  | 	// It is the recievers responsibility to type assert that the interface | ||||||
|  | 	// is the expected struct. | ||||||
|  | 	Handle(*Session, interface{}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // EventInterfaceProvider is an interface for providing empty interfaces for | ||||||
|  | // Discord events. | ||||||
|  | type EventInterfaceProvider interface { | ||||||
|  | 	// Type is the type of event this handler belongs to. | ||||||
|  | 	Type() string | ||||||
|  |  | ||||||
|  | 	// New returns a new instance of the struct this event handler handles. | ||||||
|  | 	// This is called once per event. | ||||||
|  | 	// The struct is provided to all handlers of the same Type(). | ||||||
|  | 	New() interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // interfaceEventType is the event handler type for interface{} events. | ||||||
|  | const interfaceEventType = "__INTERFACE__" | ||||||
|  |  | ||||||
|  | // interfaceEventHandler is an event handler for interface{} events. | ||||||
|  | type interfaceEventHandler func(*Session, interface{}) | ||||||
|  |  | ||||||
|  | // Type returns the event type for interface{} events. | ||||||
|  | func (eh interfaceEventHandler) Type() string { | ||||||
|  | 	return interfaceEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for an interface{} event. | ||||||
|  | func (eh interfaceEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	eh(s, i) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var registeredInterfaceProviders = map[string]EventInterfaceProvider{} | ||||||
|  |  | ||||||
|  | // registerInterfaceProvider registers a provider so that DiscordGo can | ||||||
|  | // access it's New() method. | ||||||
|  | func registerInterfaceProvider(eh EventInterfaceProvider) error { | ||||||
|  | 	if _, ok := registeredInterfaceProviders[eh.Type()]; ok { | ||||||
|  | 		return fmt.Errorf("event %s already registered", eh.Type()) | ||||||
|  | 	} | ||||||
|  | 	registeredInterfaceProviders[eh.Type()] = eh | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // eventHandlerInstance is a wrapper around an event handler, as functions | ||||||
|  | // cannot be compared directly. | ||||||
|  | type eventHandlerInstance struct { | ||||||
|  | 	eventHandler EventHandler | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // addEventHandler adds an event handler that will be fired anytime | ||||||
|  | // the Discord WSAPI matching eventHandler.Type() fires. | ||||||
|  | func (s *Session) addEventHandler(eventHandler EventHandler) func() { | ||||||
|  | 	s.handlersMu.Lock() | ||||||
|  | 	defer s.handlersMu.Unlock() | ||||||
|  |  | ||||||
|  | 	if s.handlers == nil { | ||||||
|  | 		s.handlers = map[string][]*eventHandlerInstance{} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ehi := &eventHandlerInstance{eventHandler} | ||||||
|  | 	s.handlers[eventHandler.Type()] = append(s.handlers[eventHandler.Type()], ehi) | ||||||
|  |  | ||||||
|  | 	return func() { | ||||||
|  | 		s.removeEventHandlerInstance(eventHandler.Type(), ehi) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // addEventHandler adds an event handler that will be fired the next time | ||||||
|  | // the Discord WSAPI matching eventHandler.Type() fires. | ||||||
|  | func (s *Session) addEventHandlerOnce(eventHandler EventHandler) func() { | ||||||
|  | 	s.handlersMu.Lock() | ||||||
|  | 	defer s.handlersMu.Unlock() | ||||||
|  |  | ||||||
|  | 	if s.onceHandlers == nil { | ||||||
|  | 		s.onceHandlers = map[string][]*eventHandlerInstance{} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ehi := &eventHandlerInstance{eventHandler} | ||||||
|  | 	s.onceHandlers[eventHandler.Type()] = append(s.onceHandlers[eventHandler.Type()], ehi) | ||||||
|  |  | ||||||
|  | 	return func() { | ||||||
|  | 		s.removeEventHandlerInstance(eventHandler.Type(), ehi) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddHandler allows you to add an event handler that will be fired anytime | ||||||
|  | // the Discord WSAPI event that matches the function fires. | ||||||
|  | // events.go contains all the Discord WSAPI events that can be fired. | ||||||
|  | // 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() { | ||||||
|  | 	eh := handlerForInterface(handler) | ||||||
|  |  | ||||||
|  | 	if eh == nil { | ||||||
|  | 		s.log(LogError, "Invalid handler type, handler will never be called") | ||||||
|  | 		return func() {} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return s.addEventHandler(eh) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddHandlerOnce allows you to add an event handler that will be fired the next time | ||||||
|  | // the Discord WSAPI event that matches the function fires. | ||||||
|  | // See AddHandler for more details. | ||||||
|  | func (s *Session) AddHandlerOnce(handler interface{}) func() { | ||||||
|  | 	eh := handlerForInterface(handler) | ||||||
|  |  | ||||||
|  | 	if eh == nil { | ||||||
|  | 		s.log(LogError, "Invalid handler type, handler will never be called") | ||||||
|  | 		return func() {} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return s.addEventHandlerOnce(eh) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // removeEventHandler instance removes an event handler instance. | ||||||
|  | func (s *Session) removeEventHandlerInstance(t string, ehi *eventHandlerInstance) { | ||||||
|  | 	s.handlersMu.Lock() | ||||||
|  | 	defer s.handlersMu.Unlock() | ||||||
|  |  | ||||||
|  | 	handlers := s.handlers[t] | ||||||
|  | 	for i := range handlers { | ||||||
|  | 		if handlers[i] == ehi { | ||||||
|  | 			s.handlers[t] = append(handlers[:i], handlers[i+1:]...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	onceHandlers := s.onceHandlers[t] | ||||||
|  | 	for i := range onceHandlers { | ||||||
|  | 		if onceHandlers[i] == ehi { | ||||||
|  | 			s.onceHandlers[t] = append(onceHandlers[:i], handlers[i+1:]...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handles calling permanent and once handlers for an event type. | ||||||
|  | func (s *Session) handle(t string, i interface{}) { | ||||||
|  | 	for _, eh := range s.handlers[t] { | ||||||
|  | 		go eh.eventHandler.Handle(s, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(s.onceHandlers[t]) > 0 { | ||||||
|  | 		for _, eh := range s.onceHandlers[t] { | ||||||
|  | 			go eh.eventHandler.Handle(s, i) | ||||||
|  | 		} | ||||||
|  | 		s.onceHandlers[t] = nil | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handles an event type by calling internal methods, firing handlers and firing the | ||||||
|  | // interface{} event. | ||||||
|  | func (s *Session) handleEvent(t string, i interface{}) { | ||||||
|  | 	s.handlersMu.RLock() | ||||||
|  | 	defer s.handlersMu.RUnlock() | ||||||
|  |  | ||||||
|  | 	// All events are dispatched internally first. | ||||||
|  | 	s.onInterface(i) | ||||||
|  |  | ||||||
|  | 	// Then they are dispatched to anyone handling interface{} events. | ||||||
|  | 	s.handle(interfaceEventType, i) | ||||||
|  |  | ||||||
|  | 	// Finally they are dispatched to any typed handlers. | ||||||
|  | 	s.handle(t, i) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // setGuildIds will set the GuildID on all the members of a guild. | ||||||
|  | // This is done as event data does not have it set. | ||||||
|  | func setGuildIds(g *Guild) { | ||||||
|  | 	for _, c := range g.Channels { | ||||||
|  | 		c.GuildID = g.ID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, m := range g.Members { | ||||||
|  | 		m.GuildID = g.ID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, vs := range g.VoiceStates { | ||||||
|  | 		vs.GuildID = g.ID | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // onInterface handles all internal events and routes them to the appropriate internal handler. | ||||||
|  | func (s *Session) onInterface(i interface{}) { | ||||||
|  | 	switch t := i.(type) { | ||||||
|  | 	case *Ready: | ||||||
|  | 		for _, g := range t.Guilds { | ||||||
|  | 			setGuildIds(g) | ||||||
|  | 		} | ||||||
|  | 		s.onReady(t) | ||||||
|  | 	case *GuildCreate: | ||||||
|  | 		setGuildIds(t.Guild) | ||||||
|  | 	case *GuildUpdate: | ||||||
|  | 		setGuildIds(t.Guild) | ||||||
|  | 	case *Resumed: | ||||||
|  | 		s.onResumed(t) | ||||||
|  | 	case *VoiceServerUpdate: | ||||||
|  | 		go s.onVoiceServerUpdate(t) | ||||||
|  | 	case *VoiceStateUpdate: | ||||||
|  | 		go s.onVoiceStateUpdate(t) | ||||||
|  | 	} | ||||||
|  | 	s.State.onInterface(s, i) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // onReady handles the ready event. | ||||||
|  | func (s *Session) onReady(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(r *Resumed) { | ||||||
|  |  | ||||||
|  | 	// Start the heartbeat to keep the connection alive. | ||||||
|  | 	go s.heartbeat(s.wsConn, s.listening, r.HeartbeatInterval) | ||||||
|  | } | ||||||
							
								
								
									
										977
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										977
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,977 @@ | |||||||
|  | // Code generated by \"eventhandlers\"; DO NOT EDIT | ||||||
|  | // See events.go | ||||||
|  |  | ||||||
|  | package discordgo | ||||||
|  |  | ||||||
|  | // Following are all the event types. | ||||||
|  | // Event type values are used to match the events returned by Discord. | ||||||
|  | // EventTypes surrounded by __ are synthetic and are internal to DiscordGo. | ||||||
|  | const ( | ||||||
|  | 	channelCreateEventType           = "CHANNEL_CREATE" | ||||||
|  | 	channelDeleteEventType           = "CHANNEL_DELETE" | ||||||
|  | 	channelPinsUpdateEventType       = "CHANNEL_PINS_UPDATE" | ||||||
|  | 	channelUpdateEventType           = "CHANNEL_UPDATE" | ||||||
|  | 	connectEventType                 = "__CONNECT__" | ||||||
|  | 	disconnectEventType              = "__DISCONNECT__" | ||||||
|  | 	eventEventType                   = "__EVENT__" | ||||||
|  | 	guildBanAddEventType             = "GUILD_BAN_ADD" | ||||||
|  | 	guildBanRemoveEventType          = "GUILD_BAN_REMOVE" | ||||||
|  | 	guildCreateEventType             = "GUILD_CREATE" | ||||||
|  | 	guildDeleteEventType             = "GUILD_DELETE" | ||||||
|  | 	guildEmojisUpdateEventType       = "GUILD_EMOJIS_UPDATE" | ||||||
|  | 	guildIntegrationsUpdateEventType = "GUILD_INTEGRATIONS_UPDATE" | ||||||
|  | 	guildMemberAddEventType          = "GUILD_MEMBER_ADD" | ||||||
|  | 	guildMemberRemoveEventType       = "GUILD_MEMBER_REMOVE" | ||||||
|  | 	guildMemberUpdateEventType       = "GUILD_MEMBER_UPDATE" | ||||||
|  | 	guildMembersChunkEventType       = "GUILD_MEMBERS_CHUNK" | ||||||
|  | 	guildRoleCreateEventType         = "GUILD_ROLE_CREATE" | ||||||
|  | 	guildRoleDeleteEventType         = "GUILD_ROLE_DELETE" | ||||||
|  | 	guildRoleUpdateEventType         = "GUILD_ROLE_UPDATE" | ||||||
|  | 	guildUpdateEventType             = "GUILD_UPDATE" | ||||||
|  | 	messageAckEventType              = "MESSAGE_ACK" | ||||||
|  | 	messageCreateEventType           = "MESSAGE_CREATE" | ||||||
|  | 	messageDeleteEventType           = "MESSAGE_DELETE" | ||||||
|  | 	messageReactionAddEventType      = "MESSAGE_REACTION_ADD" | ||||||
|  | 	messageReactionRemoveEventType   = "MESSAGE_REACTION_REMOVE" | ||||||
|  | 	messageUpdateEventType           = "MESSAGE_UPDATE" | ||||||
|  | 	presenceUpdateEventType          = "PRESENCE_UPDATE" | ||||||
|  | 	presencesReplaceEventType        = "PRESENCES_REPLACE" | ||||||
|  | 	rateLimitEventType               = "__RATE_LIMIT__" | ||||||
|  | 	readyEventType                   = "READY" | ||||||
|  | 	relationshipAddEventType         = "RELATIONSHIP_ADD" | ||||||
|  | 	relationshipRemoveEventType      = "RELATIONSHIP_REMOVE" | ||||||
|  | 	resumedEventType                 = "RESUMED" | ||||||
|  | 	typingStartEventType             = "TYPING_START" | ||||||
|  | 	userGuildSettingsUpdateEventType = "USER_GUILD_SETTINGS_UPDATE" | ||||||
|  | 	userSettingsUpdateEventType      = "USER_SETTINGS_UPDATE" | ||||||
|  | 	userUpdateEventType              = "USER_UPDATE" | ||||||
|  | 	voiceServerUpdateEventType       = "VOICE_SERVER_UPDATE" | ||||||
|  | 	voiceStateUpdateEventType        = "VOICE_STATE_UPDATE" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // channelCreateEventHandler is an event handler for ChannelCreate events. | ||||||
|  | type channelCreateEventHandler func(*Session, *ChannelCreate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for ChannelCreate events. | ||||||
|  | func (eh channelCreateEventHandler) Type() string { | ||||||
|  | 	return channelCreateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of ChannelCreate. | ||||||
|  | func (eh channelCreateEventHandler) New() interface{} { | ||||||
|  | 	return &ChannelCreate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for ChannelCreate events. | ||||||
|  | func (eh channelCreateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*ChannelCreate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // channelDeleteEventHandler is an event handler for ChannelDelete events. | ||||||
|  | type channelDeleteEventHandler func(*Session, *ChannelDelete) | ||||||
|  |  | ||||||
|  | // Type returns the event type for ChannelDelete events. | ||||||
|  | func (eh channelDeleteEventHandler) Type() string { | ||||||
|  | 	return channelDeleteEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of ChannelDelete. | ||||||
|  | func (eh channelDeleteEventHandler) New() interface{} { | ||||||
|  | 	return &ChannelDelete{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for ChannelDelete events. | ||||||
|  | func (eh channelDeleteEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*ChannelDelete); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // channelPinsUpdateEventHandler is an event handler for ChannelPinsUpdate events. | ||||||
|  | type channelPinsUpdateEventHandler func(*Session, *ChannelPinsUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for ChannelPinsUpdate events. | ||||||
|  | func (eh channelPinsUpdateEventHandler) Type() string { | ||||||
|  | 	return channelPinsUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of ChannelPinsUpdate. | ||||||
|  | func (eh channelPinsUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &ChannelPinsUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for ChannelPinsUpdate events. | ||||||
|  | func (eh channelPinsUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*ChannelPinsUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // channelUpdateEventHandler is an event handler for ChannelUpdate events. | ||||||
|  | type channelUpdateEventHandler func(*Session, *ChannelUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for ChannelUpdate events. | ||||||
|  | func (eh channelUpdateEventHandler) Type() string { | ||||||
|  | 	return channelUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of ChannelUpdate. | ||||||
|  | func (eh channelUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &ChannelUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for ChannelUpdate events. | ||||||
|  | func (eh channelUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*ChannelUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // connectEventHandler is an event handler for Connect events. | ||||||
|  | type connectEventHandler func(*Session, *Connect) | ||||||
|  |  | ||||||
|  | // Type returns the event type for Connect events. | ||||||
|  | func (eh connectEventHandler) Type() string { | ||||||
|  | 	return connectEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of Connect. | ||||||
|  | func (eh connectEventHandler) New() interface{} { | ||||||
|  | 	return &Connect{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for Connect events. | ||||||
|  | func (eh connectEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*Connect); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // disconnectEventHandler is an event handler for Disconnect events. | ||||||
|  | type disconnectEventHandler func(*Session, *Disconnect) | ||||||
|  |  | ||||||
|  | // Type returns the event type for Disconnect events. | ||||||
|  | func (eh disconnectEventHandler) Type() string { | ||||||
|  | 	return disconnectEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of Disconnect. | ||||||
|  | func (eh disconnectEventHandler) New() interface{} { | ||||||
|  | 	return &Disconnect{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for Disconnect events. | ||||||
|  | func (eh disconnectEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*Disconnect); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // eventEventHandler is an event handler for Event events. | ||||||
|  | type eventEventHandler func(*Session, *Event) | ||||||
|  |  | ||||||
|  | // Type returns the event type for Event events. | ||||||
|  | func (eh eventEventHandler) Type() string { | ||||||
|  | 	return eventEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of Event. | ||||||
|  | func (eh eventEventHandler) New() interface{} { | ||||||
|  | 	return &Event{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for Event events. | ||||||
|  | func (eh eventEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*Event); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildBanAddEventHandler is an event handler for GuildBanAdd events. | ||||||
|  | type guildBanAddEventHandler func(*Session, *GuildBanAdd) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildBanAdd events. | ||||||
|  | func (eh guildBanAddEventHandler) Type() string { | ||||||
|  | 	return guildBanAddEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildBanAdd. | ||||||
|  | func (eh guildBanAddEventHandler) New() interface{} { | ||||||
|  | 	return &GuildBanAdd{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildBanAdd events. | ||||||
|  | func (eh guildBanAddEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildBanAdd); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildBanRemoveEventHandler is an event handler for GuildBanRemove events. | ||||||
|  | type guildBanRemoveEventHandler func(*Session, *GuildBanRemove) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildBanRemove events. | ||||||
|  | func (eh guildBanRemoveEventHandler) Type() string { | ||||||
|  | 	return guildBanRemoveEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildBanRemove. | ||||||
|  | func (eh guildBanRemoveEventHandler) New() interface{} { | ||||||
|  | 	return &GuildBanRemove{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildBanRemove events. | ||||||
|  | func (eh guildBanRemoveEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildBanRemove); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildCreateEventHandler is an event handler for GuildCreate events. | ||||||
|  | type guildCreateEventHandler func(*Session, *GuildCreate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildCreate events. | ||||||
|  | func (eh guildCreateEventHandler) Type() string { | ||||||
|  | 	return guildCreateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildCreate. | ||||||
|  | func (eh guildCreateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildCreate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildCreate events. | ||||||
|  | func (eh guildCreateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildCreate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildDeleteEventHandler is an event handler for GuildDelete events. | ||||||
|  | type guildDeleteEventHandler func(*Session, *GuildDelete) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildDelete events. | ||||||
|  | func (eh guildDeleteEventHandler) Type() string { | ||||||
|  | 	return guildDeleteEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildDelete. | ||||||
|  | func (eh guildDeleteEventHandler) New() interface{} { | ||||||
|  | 	return &GuildDelete{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildDelete events. | ||||||
|  | func (eh guildDeleteEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildDelete); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildEmojisUpdateEventHandler is an event handler for GuildEmojisUpdate events. | ||||||
|  | type guildEmojisUpdateEventHandler func(*Session, *GuildEmojisUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildEmojisUpdate events. | ||||||
|  | func (eh guildEmojisUpdateEventHandler) Type() string { | ||||||
|  | 	return guildEmojisUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildEmojisUpdate. | ||||||
|  | func (eh guildEmojisUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildEmojisUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildEmojisUpdate events. | ||||||
|  | func (eh guildEmojisUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildEmojisUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildIntegrationsUpdateEventHandler is an event handler for GuildIntegrationsUpdate events. | ||||||
|  | type guildIntegrationsUpdateEventHandler func(*Session, *GuildIntegrationsUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildIntegrationsUpdate events. | ||||||
|  | func (eh guildIntegrationsUpdateEventHandler) Type() string { | ||||||
|  | 	return guildIntegrationsUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildIntegrationsUpdate. | ||||||
|  | func (eh guildIntegrationsUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildIntegrationsUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildIntegrationsUpdate events. | ||||||
|  | func (eh guildIntegrationsUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildIntegrationsUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildMemberAddEventHandler is an event handler for GuildMemberAdd events. | ||||||
|  | type guildMemberAddEventHandler func(*Session, *GuildMemberAdd) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildMemberAdd events. | ||||||
|  | func (eh guildMemberAddEventHandler) Type() string { | ||||||
|  | 	return guildMemberAddEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildMemberAdd. | ||||||
|  | func (eh guildMemberAddEventHandler) New() interface{} { | ||||||
|  | 	return &GuildMemberAdd{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildMemberAdd events. | ||||||
|  | func (eh guildMemberAddEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildMemberAdd); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildMemberRemoveEventHandler is an event handler for GuildMemberRemove events. | ||||||
|  | type guildMemberRemoveEventHandler func(*Session, *GuildMemberRemove) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildMemberRemove events. | ||||||
|  | func (eh guildMemberRemoveEventHandler) Type() string { | ||||||
|  | 	return guildMemberRemoveEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildMemberRemove. | ||||||
|  | func (eh guildMemberRemoveEventHandler) New() interface{} { | ||||||
|  | 	return &GuildMemberRemove{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildMemberRemove events. | ||||||
|  | func (eh guildMemberRemoveEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildMemberRemove); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildMemberUpdateEventHandler is an event handler for GuildMemberUpdate events. | ||||||
|  | type guildMemberUpdateEventHandler func(*Session, *GuildMemberUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildMemberUpdate events. | ||||||
|  | func (eh guildMemberUpdateEventHandler) Type() string { | ||||||
|  | 	return guildMemberUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildMemberUpdate. | ||||||
|  | func (eh guildMemberUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildMemberUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildMemberUpdate events. | ||||||
|  | func (eh guildMemberUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildMemberUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildMembersChunkEventHandler is an event handler for GuildMembersChunk events. | ||||||
|  | type guildMembersChunkEventHandler func(*Session, *GuildMembersChunk) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildMembersChunk events. | ||||||
|  | func (eh guildMembersChunkEventHandler) Type() string { | ||||||
|  | 	return guildMembersChunkEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildMembersChunk. | ||||||
|  | func (eh guildMembersChunkEventHandler) New() interface{} { | ||||||
|  | 	return &GuildMembersChunk{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildMembersChunk events. | ||||||
|  | func (eh guildMembersChunkEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildMembersChunk); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildRoleCreateEventHandler is an event handler for GuildRoleCreate events. | ||||||
|  | type guildRoleCreateEventHandler func(*Session, *GuildRoleCreate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildRoleCreate events. | ||||||
|  | func (eh guildRoleCreateEventHandler) Type() string { | ||||||
|  | 	return guildRoleCreateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildRoleCreate. | ||||||
|  | func (eh guildRoleCreateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildRoleCreate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildRoleCreate events. | ||||||
|  | func (eh guildRoleCreateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildRoleCreate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildRoleDeleteEventHandler is an event handler for GuildRoleDelete events. | ||||||
|  | type guildRoleDeleteEventHandler func(*Session, *GuildRoleDelete) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildRoleDelete events. | ||||||
|  | func (eh guildRoleDeleteEventHandler) Type() string { | ||||||
|  | 	return guildRoleDeleteEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildRoleDelete. | ||||||
|  | func (eh guildRoleDeleteEventHandler) New() interface{} { | ||||||
|  | 	return &GuildRoleDelete{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildRoleDelete events. | ||||||
|  | func (eh guildRoleDeleteEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildRoleDelete); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildRoleUpdateEventHandler is an event handler for GuildRoleUpdate events. | ||||||
|  | type guildRoleUpdateEventHandler func(*Session, *GuildRoleUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildRoleUpdate events. | ||||||
|  | func (eh guildRoleUpdateEventHandler) Type() string { | ||||||
|  | 	return guildRoleUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildRoleUpdate. | ||||||
|  | func (eh guildRoleUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildRoleUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildRoleUpdate events. | ||||||
|  | func (eh guildRoleUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildRoleUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // guildUpdateEventHandler is an event handler for GuildUpdate events. | ||||||
|  | type guildUpdateEventHandler func(*Session, *GuildUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for GuildUpdate events. | ||||||
|  | func (eh guildUpdateEventHandler) Type() string { | ||||||
|  | 	return guildUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of GuildUpdate. | ||||||
|  | func (eh guildUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &GuildUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for GuildUpdate events. | ||||||
|  | func (eh guildUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*GuildUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageAckEventHandler is an event handler for MessageAck events. | ||||||
|  | type messageAckEventHandler func(*Session, *MessageAck) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageAck events. | ||||||
|  | func (eh messageAckEventHandler) Type() string { | ||||||
|  | 	return messageAckEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageAck. | ||||||
|  | func (eh messageAckEventHandler) New() interface{} { | ||||||
|  | 	return &MessageAck{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageAck events. | ||||||
|  | func (eh messageAckEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageAck); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageCreateEventHandler is an event handler for MessageCreate events. | ||||||
|  | type messageCreateEventHandler func(*Session, *MessageCreate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageCreate events. | ||||||
|  | func (eh messageCreateEventHandler) Type() string { | ||||||
|  | 	return messageCreateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageCreate. | ||||||
|  | func (eh messageCreateEventHandler) New() interface{} { | ||||||
|  | 	return &MessageCreate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageCreate events. | ||||||
|  | func (eh messageCreateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageCreate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageDeleteEventHandler is an event handler for MessageDelete events. | ||||||
|  | type messageDeleteEventHandler func(*Session, *MessageDelete) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageDelete events. | ||||||
|  | func (eh messageDeleteEventHandler) Type() string { | ||||||
|  | 	return messageDeleteEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageDelete. | ||||||
|  | func (eh messageDeleteEventHandler) New() interface{} { | ||||||
|  | 	return &MessageDelete{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageDelete events. | ||||||
|  | func (eh messageDeleteEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageDelete); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageReactionAddEventHandler is an event handler for MessageReactionAdd events. | ||||||
|  | type messageReactionAddEventHandler func(*Session, *MessageReactionAdd) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageReactionAdd events. | ||||||
|  | func (eh messageReactionAddEventHandler) Type() string { | ||||||
|  | 	return messageReactionAddEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageReactionAdd. | ||||||
|  | func (eh messageReactionAddEventHandler) New() interface{} { | ||||||
|  | 	return &MessageReactionAdd{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageReactionAdd events. | ||||||
|  | func (eh messageReactionAddEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageReactionAdd); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageReactionRemoveEventHandler is an event handler for MessageReactionRemove events. | ||||||
|  | type messageReactionRemoveEventHandler func(*Session, *MessageReactionRemove) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageReactionRemove events. | ||||||
|  | func (eh messageReactionRemoveEventHandler) Type() string { | ||||||
|  | 	return messageReactionRemoveEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageReactionRemove. | ||||||
|  | func (eh messageReactionRemoveEventHandler) New() interface{} { | ||||||
|  | 	return &MessageReactionRemove{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageReactionRemove events. | ||||||
|  | func (eh messageReactionRemoveEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageReactionRemove); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageUpdateEventHandler is an event handler for MessageUpdate events. | ||||||
|  | type messageUpdateEventHandler func(*Session, *MessageUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageUpdate events. | ||||||
|  | func (eh messageUpdateEventHandler) Type() string { | ||||||
|  | 	return messageUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageUpdate. | ||||||
|  | func (eh messageUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &MessageUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageUpdate events. | ||||||
|  | func (eh messageUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // presenceUpdateEventHandler is an event handler for PresenceUpdate events. | ||||||
|  | type presenceUpdateEventHandler func(*Session, *PresenceUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for PresenceUpdate events. | ||||||
|  | func (eh presenceUpdateEventHandler) Type() string { | ||||||
|  | 	return presenceUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of PresenceUpdate. | ||||||
|  | func (eh presenceUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &PresenceUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for PresenceUpdate events. | ||||||
|  | func (eh presenceUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*PresenceUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // presencesReplaceEventHandler is an event handler for PresencesReplace events. | ||||||
|  | type presencesReplaceEventHandler func(*Session, *PresencesReplace) | ||||||
|  |  | ||||||
|  | // Type returns the event type for PresencesReplace events. | ||||||
|  | func (eh presencesReplaceEventHandler) Type() string { | ||||||
|  | 	return presencesReplaceEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of PresencesReplace. | ||||||
|  | func (eh presencesReplaceEventHandler) New() interface{} { | ||||||
|  | 	return &PresencesReplace{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for PresencesReplace events. | ||||||
|  | func (eh presencesReplaceEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*PresencesReplace); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // rateLimitEventHandler is an event handler for RateLimit events. | ||||||
|  | type rateLimitEventHandler func(*Session, *RateLimit) | ||||||
|  |  | ||||||
|  | // Type returns the event type for RateLimit events. | ||||||
|  | func (eh rateLimitEventHandler) Type() string { | ||||||
|  | 	return rateLimitEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of RateLimit. | ||||||
|  | func (eh rateLimitEventHandler) New() interface{} { | ||||||
|  | 	return &RateLimit{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for RateLimit events. | ||||||
|  | func (eh rateLimitEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*RateLimit); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // readyEventHandler is an event handler for Ready events. | ||||||
|  | type readyEventHandler func(*Session, *Ready) | ||||||
|  |  | ||||||
|  | // Type returns the event type for Ready events. | ||||||
|  | func (eh readyEventHandler) Type() string { | ||||||
|  | 	return readyEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of Ready. | ||||||
|  | func (eh readyEventHandler) New() interface{} { | ||||||
|  | 	return &Ready{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for Ready events. | ||||||
|  | func (eh readyEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*Ready); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // relationshipAddEventHandler is an event handler for RelationshipAdd events. | ||||||
|  | type relationshipAddEventHandler func(*Session, *RelationshipAdd) | ||||||
|  |  | ||||||
|  | // Type returns the event type for RelationshipAdd events. | ||||||
|  | func (eh relationshipAddEventHandler) Type() string { | ||||||
|  | 	return relationshipAddEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of RelationshipAdd. | ||||||
|  | func (eh relationshipAddEventHandler) New() interface{} { | ||||||
|  | 	return &RelationshipAdd{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for RelationshipAdd events. | ||||||
|  | func (eh relationshipAddEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*RelationshipAdd); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // relationshipRemoveEventHandler is an event handler for RelationshipRemove events. | ||||||
|  | type relationshipRemoveEventHandler func(*Session, *RelationshipRemove) | ||||||
|  |  | ||||||
|  | // Type returns the event type for RelationshipRemove events. | ||||||
|  | func (eh relationshipRemoveEventHandler) Type() string { | ||||||
|  | 	return relationshipRemoveEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of RelationshipRemove. | ||||||
|  | func (eh relationshipRemoveEventHandler) New() interface{} { | ||||||
|  | 	return &RelationshipRemove{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for RelationshipRemove events. | ||||||
|  | func (eh relationshipRemoveEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*RelationshipRemove); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // resumedEventHandler is an event handler for Resumed events. | ||||||
|  | type resumedEventHandler func(*Session, *Resumed) | ||||||
|  |  | ||||||
|  | // Type returns the event type for Resumed events. | ||||||
|  | func (eh resumedEventHandler) Type() string { | ||||||
|  | 	return resumedEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of Resumed. | ||||||
|  | func (eh resumedEventHandler) New() interface{} { | ||||||
|  | 	return &Resumed{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for Resumed events. | ||||||
|  | func (eh resumedEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*Resumed); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // typingStartEventHandler is an event handler for TypingStart events. | ||||||
|  | type typingStartEventHandler func(*Session, *TypingStart) | ||||||
|  |  | ||||||
|  | // Type returns the event type for TypingStart events. | ||||||
|  | func (eh typingStartEventHandler) Type() string { | ||||||
|  | 	return typingStartEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of TypingStart. | ||||||
|  | func (eh typingStartEventHandler) New() interface{} { | ||||||
|  | 	return &TypingStart{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for TypingStart events. | ||||||
|  | func (eh typingStartEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*TypingStart); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // userGuildSettingsUpdateEventHandler is an event handler for UserGuildSettingsUpdate events. | ||||||
|  | type userGuildSettingsUpdateEventHandler func(*Session, *UserGuildSettingsUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for UserGuildSettingsUpdate events. | ||||||
|  | func (eh userGuildSettingsUpdateEventHandler) Type() string { | ||||||
|  | 	return userGuildSettingsUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of UserGuildSettingsUpdate. | ||||||
|  | func (eh userGuildSettingsUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &UserGuildSettingsUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for UserGuildSettingsUpdate events. | ||||||
|  | func (eh userGuildSettingsUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*UserGuildSettingsUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // userSettingsUpdateEventHandler is an event handler for UserSettingsUpdate events. | ||||||
|  | type userSettingsUpdateEventHandler func(*Session, *UserSettingsUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for UserSettingsUpdate events. | ||||||
|  | func (eh userSettingsUpdateEventHandler) Type() string { | ||||||
|  | 	return userSettingsUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of UserSettingsUpdate. | ||||||
|  | func (eh userSettingsUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &UserSettingsUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for UserSettingsUpdate events. | ||||||
|  | func (eh userSettingsUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*UserSettingsUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // userUpdateEventHandler is an event handler for UserUpdate events. | ||||||
|  | type userUpdateEventHandler func(*Session, *UserUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for UserUpdate events. | ||||||
|  | func (eh userUpdateEventHandler) Type() string { | ||||||
|  | 	return userUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of UserUpdate. | ||||||
|  | func (eh userUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &UserUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for UserUpdate events. | ||||||
|  | func (eh userUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*UserUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // voiceServerUpdateEventHandler is an event handler for VoiceServerUpdate events. | ||||||
|  | type voiceServerUpdateEventHandler func(*Session, *VoiceServerUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for VoiceServerUpdate events. | ||||||
|  | func (eh voiceServerUpdateEventHandler) Type() string { | ||||||
|  | 	return voiceServerUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of VoiceServerUpdate. | ||||||
|  | func (eh voiceServerUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &VoiceServerUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for VoiceServerUpdate events. | ||||||
|  | func (eh voiceServerUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*VoiceServerUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // voiceStateUpdateEventHandler is an event handler for VoiceStateUpdate events. | ||||||
|  | type voiceStateUpdateEventHandler func(*Session, *VoiceStateUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for VoiceStateUpdate events. | ||||||
|  | func (eh voiceStateUpdateEventHandler) Type() string { | ||||||
|  | 	return voiceStateUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of VoiceStateUpdate. | ||||||
|  | func (eh voiceStateUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &VoiceStateUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for VoiceStateUpdate events. | ||||||
|  | func (eh voiceStateUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*VoiceStateUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func handlerForInterface(handler interface{}) EventHandler { | ||||||
|  | 	switch v := handler.(type) { | ||||||
|  | 	case func(*Session, interface{}): | ||||||
|  | 		return interfaceEventHandler(v) | ||||||
|  | 	case func(*Session, *ChannelCreate): | ||||||
|  | 		return channelCreateEventHandler(v) | ||||||
|  | 	case func(*Session, *ChannelDelete): | ||||||
|  | 		return channelDeleteEventHandler(v) | ||||||
|  | 	case func(*Session, *ChannelPinsUpdate): | ||||||
|  | 		return channelPinsUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *ChannelUpdate): | ||||||
|  | 		return channelUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *Connect): | ||||||
|  | 		return connectEventHandler(v) | ||||||
|  | 	case func(*Session, *Disconnect): | ||||||
|  | 		return disconnectEventHandler(v) | ||||||
|  | 	case func(*Session, *Event): | ||||||
|  | 		return eventEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildBanAdd): | ||||||
|  | 		return guildBanAddEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildBanRemove): | ||||||
|  | 		return guildBanRemoveEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildCreate): | ||||||
|  | 		return guildCreateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildDelete): | ||||||
|  | 		return guildDeleteEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildEmojisUpdate): | ||||||
|  | 		return guildEmojisUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildIntegrationsUpdate): | ||||||
|  | 		return guildIntegrationsUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildMemberAdd): | ||||||
|  | 		return guildMemberAddEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildMemberRemove): | ||||||
|  | 		return guildMemberRemoveEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildMemberUpdate): | ||||||
|  | 		return guildMemberUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildMembersChunk): | ||||||
|  | 		return guildMembersChunkEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildRoleCreate): | ||||||
|  | 		return guildRoleCreateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildRoleDelete): | ||||||
|  | 		return guildRoleDeleteEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildRoleUpdate): | ||||||
|  | 		return guildRoleUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *GuildUpdate): | ||||||
|  | 		return guildUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageAck): | ||||||
|  | 		return messageAckEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageCreate): | ||||||
|  | 		return messageCreateEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageDelete): | ||||||
|  | 		return messageDeleteEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageReactionAdd): | ||||||
|  | 		return messageReactionAddEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageReactionRemove): | ||||||
|  | 		return messageReactionRemoveEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageUpdate): | ||||||
|  | 		return messageUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *PresenceUpdate): | ||||||
|  | 		return presenceUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *PresencesReplace): | ||||||
|  | 		return presencesReplaceEventHandler(v) | ||||||
|  | 	case func(*Session, *RateLimit): | ||||||
|  | 		return rateLimitEventHandler(v) | ||||||
|  | 	case func(*Session, *Ready): | ||||||
|  | 		return readyEventHandler(v) | ||||||
|  | 	case func(*Session, *RelationshipAdd): | ||||||
|  | 		return relationshipAddEventHandler(v) | ||||||
|  | 	case func(*Session, *RelationshipRemove): | ||||||
|  | 		return relationshipRemoveEventHandler(v) | ||||||
|  | 	case func(*Session, *Resumed): | ||||||
|  | 		return resumedEventHandler(v) | ||||||
|  | 	case func(*Session, *TypingStart): | ||||||
|  | 		return typingStartEventHandler(v) | ||||||
|  | 	case func(*Session, *UserGuildSettingsUpdate): | ||||||
|  | 		return userGuildSettingsUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *UserSettingsUpdate): | ||||||
|  | 		return userSettingsUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *UserUpdate): | ||||||
|  | 		return userUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *VoiceServerUpdate): | ||||||
|  | 		return voiceServerUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *VoiceStateUpdate): | ||||||
|  | 		return voiceStateUpdateEventHandler(v) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | func init() { | ||||||
|  | 	registerInterfaceProvider(channelCreateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(channelDeleteEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(channelPinsUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(channelUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildBanAddEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildBanRemoveEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildCreateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildDeleteEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildEmojisUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildIntegrationsUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildMemberAddEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildMemberRemoveEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildMemberUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildMembersChunkEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildRoleCreateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildRoleDeleteEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildRoleUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(guildUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageAckEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageCreateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageDeleteEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageReactionAddEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageReactionRemoveEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(presenceUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(presencesReplaceEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(readyEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(relationshipAddEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(relationshipRemoveEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(resumedEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(typingStartEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(userGuildSettingsUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(userSettingsUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(userUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(voiceServerUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(voiceStateUpdateEventHandler(nil)) | ||||||
|  | } | ||||||
							
								
								
									
										238
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										238
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,238 @@ | |||||||
|  | package discordgo | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // This file contains all the possible structs that can be | ||||||
|  | // handled by AddHandler/EventHandler. | ||||||
|  | // DO NOT ADD ANYTHING BUT EVENT HANDLER STRUCTS TO THIS FILE. | ||||||
|  | //go:generate go run tools/cmd/eventhandlers/main.go | ||||||
|  |  | ||||||
|  | // Connect is the data for a Connect event. | ||||||
|  | // This is a sythetic event and is not dispatched by Discord. | ||||||
|  | type Connect struct{} | ||||||
|  |  | ||||||
|  | // Disconnect is the data for a Disconnect event. | ||||||
|  | // This is a sythetic event and is not dispatched by Discord. | ||||||
|  | type Disconnect struct{} | ||||||
|  |  | ||||||
|  | // RateLimit is the data for a RateLimit event. | ||||||
|  | // This is a sythetic event and is not dispatched by Discord. | ||||||
|  | type RateLimit struct { | ||||||
|  | 	*TooManyRequests | ||||||
|  | 	URL string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Event provides a basic initial struct for all websocket events. | ||||||
|  | type Event struct { | ||||||
|  | 	Operation int             `json:"op"` | ||||||
|  | 	Sequence  int             `json:"s"` | ||||||
|  | 	Type      string          `json:"t"` | ||||||
|  | 	RawData   json.RawMessage `json:"d"` | ||||||
|  | 	// Struct contains one of the other types in this file. | ||||||
|  | 	Struct interface{} `json:"-"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A Ready stores all data for the websocket READY event. | ||||||
|  | type Ready struct { | ||||||
|  | 	Version           int           `json:"v"` | ||||||
|  | 	SessionID         string        `json:"session_id"` | ||||||
|  | 	HeartbeatInterval time.Duration `json:"heartbeat_interval"` | ||||||
|  | 	User              *User         `json:"user"` | ||||||
|  | 	ReadState         []*ReadState  `json:"read_state"` | ||||||
|  | 	PrivateChannels   []*Channel    `json:"private_channels"` | ||||||
|  | 	Guilds            []*Guild      `json:"guilds"` | ||||||
|  |  | ||||||
|  | 	// Undocumented fields | ||||||
|  | 	Settings          *Settings            `json:"user_settings"` | ||||||
|  | 	UserGuildSettings []*UserGuildSettings `json:"user_guild_settings"` | ||||||
|  | 	Relationships     []*Relationship      `json:"relationships"` | ||||||
|  | 	Presences         []*Presence          `json:"presences"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ChannelCreate is the data for a ChannelCreate event. | ||||||
|  | type ChannelCreate struct { | ||||||
|  | 	*Channel | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ChannelUpdate is the data for a ChannelUpdate event. | ||||||
|  | type ChannelUpdate struct { | ||||||
|  | 	*Channel | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ChannelDelete is the data for a ChannelDelete event. | ||||||
|  | type ChannelDelete struct { | ||||||
|  | 	*Channel | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ChannelPinsUpdate stores data for a ChannelPinsUpdate event. | ||||||
|  | type ChannelPinsUpdate struct { | ||||||
|  | 	LastPinTimestamp string `json:"last_pin_timestamp"` | ||||||
|  | 	ChannelID        string `json:"channel_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildCreate is the data for a GuildCreate event. | ||||||
|  | type GuildCreate struct { | ||||||
|  | 	*Guild | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildUpdate is the data for a GuildUpdate event. | ||||||
|  | type GuildUpdate struct { | ||||||
|  | 	*Guild | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildDelete is the data for a GuildDelete event. | ||||||
|  | type GuildDelete struct { | ||||||
|  | 	*Guild | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildBanAdd is the data for a GuildBanAdd event. | ||||||
|  | type GuildBanAdd struct { | ||||||
|  | 	User    *User  `json:"user"` | ||||||
|  | 	GuildID string `json:"guild_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildBanRemove is the data for a GuildBanRemove event. | ||||||
|  | type GuildBanRemove struct { | ||||||
|  | 	User    *User  `json:"user"` | ||||||
|  | 	GuildID string `json:"guild_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildMemberAdd is the data for a GuildMemberAdd event. | ||||||
|  | type GuildMemberAdd struct { | ||||||
|  | 	*Member | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildMemberUpdate is the data for a GuildMemberUpdate event. | ||||||
|  | type GuildMemberUpdate struct { | ||||||
|  | 	*Member | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildMemberRemove is the data for a GuildMemberRemove event. | ||||||
|  | type GuildMemberRemove struct { | ||||||
|  | 	*Member | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildRoleCreate is the data for a GuildRoleCreate event. | ||||||
|  | type GuildRoleCreate struct { | ||||||
|  | 	*GuildRole | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildRoleUpdate is the data for a GuildRoleUpdate event. | ||||||
|  | type GuildRoleUpdate struct { | ||||||
|  | 	*GuildRole | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A GuildRoleDelete is the data for a GuildRoleDelete event. | ||||||
|  | type GuildRoleDelete struct { | ||||||
|  | 	RoleID  string `json:"role_id"` | ||||||
|  | 	GuildID string `json:"guild_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A GuildEmojisUpdate is the data for a guild emoji update event. | ||||||
|  | type GuildEmojisUpdate struct { | ||||||
|  | 	GuildID string   `json:"guild_id"` | ||||||
|  | 	Emojis  []*Emoji `json:"emojis"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A GuildMembersChunk is the data for a GuildMembersChunk event. | ||||||
|  | type GuildMembersChunk struct { | ||||||
|  | 	GuildID string    `json:"guild_id"` | ||||||
|  | 	Members []*Member `json:"members"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildIntegrationsUpdate is the data for a GuildIntegrationsUpdate event. | ||||||
|  | type GuildIntegrationsUpdate struct { | ||||||
|  | 	GuildID string `json:"guild_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageAck is the data for a MessageAck event. | ||||||
|  | type MessageAck struct { | ||||||
|  | 	MessageID string `json:"message_id"` | ||||||
|  | 	ChannelID string `json:"channel_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageCreate is the data for a MessageCreate event. | ||||||
|  | type MessageCreate struct { | ||||||
|  | 	*Message | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageUpdate is the data for a MessageUpdate event. | ||||||
|  | type MessageUpdate struct { | ||||||
|  | 	*Message | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageDelete is the data for a MessageDelete event. | ||||||
|  | type MessageDelete struct { | ||||||
|  | 	*Message | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageReactionAdd is the data for a MessageReactionAdd event. | ||||||
|  | type MessageReactionAdd struct { | ||||||
|  | 	*MessageReaction | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageReactionRemove is the data for a MessageReactionRemove event. | ||||||
|  | type MessageReactionRemove struct { | ||||||
|  | 	*MessageReaction | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PresencesReplace is the data for a PresencesReplace event. | ||||||
|  | type PresencesReplace []*Presence | ||||||
|  |  | ||||||
|  | // PresenceUpdate is the data for a PresenceUpdate event. | ||||||
|  | type PresenceUpdate struct { | ||||||
|  | 	Presence | ||||||
|  | 	GuildID string   `json:"guild_id"` | ||||||
|  | 	Roles   []string `json:"roles"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Resumed is the data for a Resumed event. | ||||||
|  | type Resumed struct { | ||||||
|  | 	HeartbeatInterval time.Duration `json:"heartbeat_interval"` | ||||||
|  | 	Trace             []string      `json:"_trace"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RelationshipAdd is the data for a RelationshipAdd event. | ||||||
|  | type RelationshipAdd struct { | ||||||
|  | 	*Relationship | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RelationshipRemove is the data for a RelationshipRemove event. | ||||||
|  | type RelationshipRemove struct { | ||||||
|  | 	*Relationship | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // TypingStart is the data for a TypingStart event. | ||||||
|  | type TypingStart struct { | ||||||
|  | 	UserID    string `json:"user_id"` | ||||||
|  | 	ChannelID string `json:"channel_id"` | ||||||
|  | 	Timestamp int    `json:"timestamp"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UserUpdate is the data for a UserUpdate event. | ||||||
|  | type UserUpdate struct { | ||||||
|  | 	*User | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UserSettingsUpdate is the data for a UserSettingsUpdate event. | ||||||
|  | type UserSettingsUpdate map[string]interface{} | ||||||
|  |  | ||||||
|  | // UserGuildSettingsUpdate is the data for a UserGuildSettingsUpdate event. | ||||||
|  | type UserGuildSettingsUpdate struct { | ||||||
|  | 	*UserGuildSettings | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // VoiceServerUpdate is the data for a VoiceServerUpdate event. | ||||||
|  | type VoiceServerUpdate struct { | ||||||
|  | 	Token    string `json:"token"` | ||||||
|  | 	GuildID  string `json:"guild_id"` | ||||||
|  | 	Endpoint string `json:"endpoint"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // VoiceStateUpdate is the data for a VoiceStateUpdate event. | ||||||
|  | type VoiceStateUpdate struct { | ||||||
|  | 	*VoiceState | ||||||
|  | } | ||||||
							
								
								
									
										190
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										190
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,190 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"encoding/binary" | ||||||
|  | 	"flag" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io" | ||||||
|  | 	"os" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	flag.StringVar(&token, "t", "", "Bot 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 bot token. | ||||||
|  | 	dg, err := discordgo.New("Bot " + 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 { | ||||||
|  | 		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 { | ||||||
|  | 			file.Close() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			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) | ||||||
|  | } | ||||||
							
								
								
									
										53
									
								
								vendor/github.com/bwmarrin/discordgo/examples/new_basic/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								vendor/github.com/bwmarrin/discordgo/examples/new_basic/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,53 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"flag" | ||||||
|  | 	"fmt" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Variables used for command line parameters | ||||||
|  | var ( | ||||||
|  | 	Token string | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  |  | ||||||
|  | 	flag.StringVar(&Token, "t", "", "Bot Token") | ||||||
|  | 	flag.Parse() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func main() { | ||||||
|  |  | ||||||
|  | 	// Create a new Discord session using the provided bot token. | ||||||
|  | 	dg, err := discordgo.New("Bot " + 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) | ||||||
|  | } | ||||||
							
								
								
									
										74
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										74
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,74 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"flag" | ||||||
|  | 	"fmt" | ||||||
|  |  | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Variables used for command line parameters | ||||||
|  | var ( | ||||||
|  | 	Token string | ||||||
|  | 	BotID string | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  |  | ||||||
|  | 	flag.StringVar(&Token, "t", "", "Bot Token") | ||||||
|  | 	flag.Parse() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func main() { | ||||||
|  |  | ||||||
|  | 	// Create a new Discord session using the provided bot token. | ||||||
|  | 	dg, err := discordgo.New("Bot " + 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())) | ||||||
|  | } | ||||||
|  | */ | ||||||
							
								
								
									
										132
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										132
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,132 @@ | |||||||
|  | // 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       Timestamp            `json:"timestamp"` | ||||||
|  | 	EditedTimestamp Timestamp            `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"` | ||||||
|  | 	Reactions       []*MessageReactions  `json:"reactions"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedFooter is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedFooter struct { | ||||||
|  | 	Text         string `json:"text,omitempty"` | ||||||
|  | 	IconURL      string `json:"icon_url,omitempty"` | ||||||
|  | 	ProxyIconURL string `json:"proxy_icon_url,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedImage is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedImage struct { | ||||||
|  | 	URL      string `json:"url,omitempty"` | ||||||
|  | 	ProxyURL string `json:"proxy_url,omitempty"` | ||||||
|  | 	Width    int    `json:"width,omitempty"` | ||||||
|  | 	Height   int    `json:"height,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedThumbnail is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedThumbnail struct { | ||||||
|  | 	URL      string `json:"url,omitempty"` | ||||||
|  | 	ProxyURL string `json:"proxy_url,omitempty"` | ||||||
|  | 	Width    int    `json:"width,omitempty"` | ||||||
|  | 	Height   int    `json:"height,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedVideo is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedVideo struct { | ||||||
|  | 	URL      string `json:"url,omitempty"` | ||||||
|  | 	ProxyURL string `json:"proxy_url,omitempty"` | ||||||
|  | 	Width    int    `json:"width,omitempty"` | ||||||
|  | 	Height   int    `json:"height,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedProvider is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedProvider struct { | ||||||
|  | 	URL  string `json:"url,omitempty"` | ||||||
|  | 	Name string `json:"name,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedAuthor is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedAuthor struct { | ||||||
|  | 	URL          string `json:"url,omitempty"` | ||||||
|  | 	Name         string `json:"name,omitempty"` | ||||||
|  | 	IconURL      string `json:"icon_url,omitempty"` | ||||||
|  | 	ProxyIconURL string `json:"proxy_icon_url,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEmbedField is a part of a MessageEmbed struct. | ||||||
|  | type MessageEmbedField struct { | ||||||
|  | 	Name   string `json:"name,omitempty"` | ||||||
|  | 	Value  string `json:"value,omitempty"` | ||||||
|  | 	Inline bool   `json:"inline,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // An MessageEmbed stores data for message embeds. | ||||||
|  | type MessageEmbed struct { | ||||||
|  | 	URL         string                 `json:"url,omitempty"` | ||||||
|  | 	Type        string                 `json:"type,omitempty"` | ||||||
|  | 	Title       string                 `json:"title,omitempty"` | ||||||
|  | 	Description string                 `json:"description,omitempty"` | ||||||
|  | 	Timestamp   string                 `json:"timestamp,omitempty"` | ||||||
|  | 	Color       int                    `json:"color,omitempty"` | ||||||
|  | 	Footer      *MessageEmbedFooter    `json:"footer,omitempty"` | ||||||
|  | 	Image       *MessageEmbedImage     `json:"image,omitempty"` | ||||||
|  | 	Thumbnail   *MessageEmbedThumbnail `json:"thumbnail,omitempty"` | ||||||
|  | 	Video       *MessageEmbedVideo     `json:"video,omitempty"` | ||||||
|  | 	Provider    *MessageEmbedProvider  `json:"provider,omitempty"` | ||||||
|  | 	Author      *MessageEmbedAuthor    `json:"author,omitempty"` | ||||||
|  | 	Fields      []*MessageEmbedField   `json:"fields,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageReactions holds a reactions object for a message. | ||||||
|  | type MessageReactions struct { | ||||||
|  | 	Count int    `json:"count"` | ||||||
|  | 	Me    bool   `json:"me"` | ||||||
|  | 	Emoji *Emoji `json:"emoji"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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