forked from lug/matterbridge
		
	Compare commits
	
		
			295 Commits
		
	
	
		
			v0.16.0-rc
			...
			v1.8.0
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | 222cccf388 | ||
|   | bab308508e | ||
|   | dedb83c867 | ||
|   | 723a90cdd6 | ||
|   | 67d2398fa8 | ||
|   | 5f3b6ec007 | ||
|   | 55ab0c12f1 | ||
|   | d1227b5fc9 | ||
|   | 6ea368c383 | ||
|   | e92b6de09f | ||
|   | e622587db4 | ||
|   | f2efc06d1f | ||
|   | a2b94452db | ||
|   | 4c506f7cc3 | ||
|   | 7886f05e88 | ||
|   | f58be0d1c1 | ||
|   | 1152394bc1 | ||
|   | a082b5a590 | ||
|   | bae9484df2 | ||
|   | 6f78485878 | ||
|   | fd0fe3390b | ||
|   | 2522158127 | ||
|   | 8be107cecc | ||
|   | 5aab158c0b | ||
|   | 1d33e60e36 | ||
|   | 83c28cb857 | ||
|   | df5bce27b0 | ||
|   | 2b15739b48 | ||
|   | 3480c88e90 | ||
|   | 432cd0f99d | ||
|   | e8b3e9b22d | ||
|   | d4a47671ea | ||
|   | 0bcd1e62f3 | ||
|   | 80822b7fff | ||
|   | 78f1011f52 | ||
|   | 67f6257617 | ||
|   | 169c614489 | ||
|   | da908c438a | ||
|   | 9c9c4bf1f9 | ||
|   | 7764493298 | ||
|   | 64a20ee61b | ||
|   | 62d1af8c37 | ||
|   | 0f5274fdf6 | ||
|   | 2e2187ebf4 | ||
|   | 762c3350f4 | ||
|   | e1a4d7f77e | ||
|   | a7a4554a85 | ||
|   | 6bd808ce91 | ||
|   | a5c143bc46 | ||
|   | 87c9cac756 | ||
|   | 6a047f8722 | ||
|   | 6523494e83 | ||
|   | 7c6ce8bb90 | ||
|   | dafbfe4021 | ||
|   | a4d5c94d9b | ||
|   | 7119e378a7 | ||
|   | e1dc3032c1 | ||
|   | 5de03b8921 | ||
|   | 7631d43c48 | ||
|   | d0b2ee5c85 | ||
|   | 8830a5a1df | ||
|   | ee87626a93 | ||
|   | 9f15d38c1c | ||
|   | 4a96a977c0 | ||
|   | 9a95293bdf | ||
|   | 0b3a06d263 | ||
|   | 9a6249c4f5 | ||
|   | 50bd51e461 | ||
|   | 04f8013314 | ||
|   | a0aaf0057a | ||
|   | 8e78b3e6be | ||
|   | 57a503818d | ||
|   | 25d2ff3e9b | ||
|   | 31902d3e57 | ||
|   | 16f3fa6bae | ||
|   | 1f706673cf | ||
|   | fac5f69ad2 | ||
|   | 97c944bb63 | ||
|   | d0c4fe78ee | ||
|   | 265457b451 | ||
|   | 4a4a29c9f6 | ||
|   | 0a91b9e1c9 | ||
|   | f56163295c | ||
|   | d1c87c068b | ||
|   | fa20761110 | ||
|   | e4a0e0a0e9 | ||
|   | d30ae19e2a | ||
|   | 5c919e6bff | ||
|   | 434393d1c3 | ||
|   | af9aa5d7cb | ||
|   | 05eb75442a | ||
|   | 3496ed0c7e | ||
|   | 1b89604c7a | ||
|   | 67a9d133e9 | ||
|   | ed9118b346 | ||
|   | 59e55cfbd5 | ||
|   | 788d3b32ac | ||
|   | 1d414cf2fd | ||
|   | cc3c168162 | ||
|   | 1ee6837f0e | ||
|   | 27dcea7c5b | ||
|   | dcda7f7b8c | ||
|   | e0cbb69a4f | ||
|   | 7ec95f786d | ||
|   | 1efe40add5 | ||
|   | cbd73ee313 | ||
|   | 34227a7a39 | ||
|   | 71cb9b2d1d | ||
|   | cd4c9b194f | ||
|   | 98762a0235 | ||
|   | 2fd1fd9573 | ||
|   | aff3964078 | ||
|   | 2778580397 | ||
|   | 962062fe44 | ||
|   | 0578b21270 | ||
|   | 36a800c3f5 | ||
|   | 6d21f84187 | ||
|   | f1e9833310 | ||
|   | 46f5acc4f9 | ||
|   | 95d4dcaeb3 | ||
|   | 64c542e614 | ||
|   | 13d081ea80 | ||
|   | c0f9d86287 | ||
|   | bcdecdaa73 | ||
|   | daac3ebca2 | ||
|   | 639f9cf966 | ||
|   | 4fc48b5aa4 | ||
|   | 307ff77b42 | ||
|   | 9b500bc5f7 | ||
|   | e313154134 | ||
|   | 27e94c438d | ||
|   | 58392876df | ||
|   | 115c4b1aa7 | ||
|   | ba5649d259 | ||
|   | 1b30575510 | ||
|   | 7dbebd3ea7 | ||
|   | 6f18790352 | ||
|   | d1e04a2ece | ||
|   | bea0bbd0c2 | ||
|   | 0530503ef2 | ||
|   | d1e8ff814b | ||
|   | 4f8ae761a2 | ||
|   | b530e92834 | ||
|   | b2a6777995 | ||
|   | b461fc5e40 | ||
|   | b7a8c6b60f | ||
|   | 41aa8ad799 | ||
|   | 7973baedd0 | ||
|   | 299b71d982 | ||
|   | 76aafe1fa8 | ||
|   | 95a0229aaf | ||
|   | 915a8fbad7 | ||
|   | d4d7fef313 | ||
|   | 4e1dc9f885 | ||
|   | 155ae80d22 | ||
|   | c7e336efd9 | ||
|   | ac3c65a0cc | ||
|   | df74df475b | ||
|   | a61e2db7cb | ||
|   | 7aabe12acf | ||
|   | c4b75e5754 | ||
|   | 6a7adb20a8 | ||
|   | b49fb2b69c | ||
|   | 4bda29cb38 | ||
|   | 5f14141ec9 | ||
|   | c088e45d85 | ||
|   | d59c51a94b | ||
|   | 47b7fae61b | ||
|   | 1a40b0c1e9 | ||
|   | 27d886826c | ||
|   | 18981cb636 | ||
|   | ffa8f65aa8 | ||
|   | 82588b00c5 | ||
|   | 603449e850 | ||
|   | 248d88c849 | ||
|   | d19535fa21 | ||
|   | 49204cafcc | ||
|   | 812db2d267 | ||
|   | 14490bea9f | ||
|   | 0352970051 | ||
|   | ed01820722 | ||
|   | 90a61f15cc | ||
|   | 86cd7f1ba6 | ||
|   | d6ee55e35f | ||
|   | aef64eec32 | ||
|   | c4193d5ccd | ||
|   | 0c94186818 | ||
|   | 9039720013 | ||
|   | a3470f8aec | ||
|   | 01badde21d | ||
|   | a37b232dd9 | ||
|   | 579ee48385 | ||
|   | dd985d1dad | ||
|   | d2caea70a2 | ||
|   | 21143cf5ee | ||
|   | dc2aed698d | ||
|   | 37c350f19f | ||
|   | 9e03fcf162 | ||
|   | 8d4521c1df | ||
|   | 9226252336 | ||
|   | f4fb83e787 | ||
|   | e7fcb25107 | ||
|   | 5a85258f74 | ||
|   | 2f7df2df43 | ||
|   | ad3a753718 | ||
|   | e45c551880 | ||
|   | e59d338d4e | ||
|   | 7a86044f7a | ||
|   | 8b98f605bc | ||
|   | 7c773ebae0 | ||
|   | e84417430d | ||
|   | 5a8d7b5f6d | ||
|   | cfb8107138 | ||
|   | 43bd779fb7 | ||
|   | 7f9a400776 | ||
|   | ce1c5873ac | ||
|   | 85ff1995fd | ||
|   | b963f83c6a | ||
|   | f6297ebbb0 | ||
|   | a5259f56c5 | ||
|   | 3f75ed9c18 | ||
|   | 49ece51167 | ||
|   | e77c3eb20a | ||
|   | 59b2a5f8d0 | ||
|   | 28710d0bc7 | ||
|   | ad4d461606 | ||
|   | 67905089ba | ||
|   | f2483af561 | ||
|   | c28b87641e | ||
|   | f8e6a69d6e | ||
|   | 54216cec4b | ||
|   | 12989bbd99 | ||
|   | 38d09dba2e | ||
|   | fafd0c68e9 | ||
|   | 41195c8e48 | ||
|   | a97804548e | ||
|   | ba653c0841 | ||
|   | 5b191f78a0 | ||
|   | 83ef61287e | ||
|   | 3527e09bc5 | ||
|   | ddc5b3268f | ||
|   | 22307b1934 | ||
|   | bd97357f8d | ||
|   | 10dab1366e | ||
|   | 52fc94c1fe | ||
|   | c1c7961dd6 | ||
|   | d3eef051b1 | ||
|   | 57654df81e | ||
|   | 0f791d7a9a | ||
|   | 58779e0d65 | ||
|   | 4ac361b5fd | ||
|   | 1e2f27c061 | ||
|   | 0302e4da82 | ||
|   | dc8743e0c0 | ||
|   | cc5ce3d5ae | ||
|   | caaf6f3012 | ||
|   | c5de8fd1cc | ||
|   | c9f23869e3 | ||
|   | 61208c0e35 | ||
|   | dcffc74255 | ||
|   | 23e23be1a6 | ||
|   | 710427248a | ||
|   | a868042de2 | ||
|   | 15296cd8b4 | ||
|   | 717023245f | ||
|   | 320be5bffa | ||
|   | 778abea2d9 | ||
|   | 835a1ac3a6 | ||
|   | 20a7ef33f1 | ||
|   | e72612c7ff | ||
|   | 04e0f001b0 | ||
|   | 5db24aa901 | ||
|   | aec5e3d77b | ||
|   | 335ddf8db5 | ||
|   | 4abaf2b236 | ||
|   | 183d212431 | ||
|   | e99532fb89 | ||
|   | 4aa646f6b0 | ||
|   | 9dcd51fb80 | ||
|   | 6dee988b76 | ||
|   | 5af40db396 | ||
|   | b3553bee7a | ||
|   | ac19c94b9f | ||
|   | 845f7dc331 | ||
|   | 2adeae37e1 | ||
|   | 16eb12b2a0 | ||
|   | 8411f2aa32 | ||
|   | e8acc49cbd | ||
|   | 4bed073c65 | ||
|   | 272735fb26 | ||
|   | b75cf2c189 | ||
|   | 1aaa992250 | ||
|   | 6256c066f1 | ||
|   | 870b89a8f0 | ||
|   | 65ac96913c | 
							
								
								
									
										28
									
								
								.github/ISSUE_TEMPLATE.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								.github/ISSUE_TEMPLATE.md
									
									
									
									
										vendored
									
									
								
							| @@ -1,20 +1,36 @@ | |||||||
| Please answer the following questions.  | <!-- This is a bug report template. By following the instructions below and | ||||||
|  | filling out the sections with your information, you will help the us to get all | ||||||
|  | the necessary data to fix your issue. | ||||||
|  |  | ||||||
| ### Which version of matterbridge are you using? | You can also preview your report before submitting it. | ||||||
| run ```matterbridge -version``` |  | ||||||
|  |  | ||||||
| ### If you're having problems with mattermost please specify mattermost version.  | Text between <!-- and --> marks will be invisible in the report. | ||||||
|  | --> | ||||||
|  |  | ||||||
|  | <!-- If you have a configuration problem, please first try to create a basic configuration following the instructions on [the wiki](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) before filing an issue. --> | ||||||
|  |  | ||||||
|  |  | ||||||
|  | ### Environment | ||||||
|  | <!-- run `matterbridge -version` --> | ||||||
|  | <!-- If you're having problems with mattermost also specify the mattermost version. --> | ||||||
|  | Version: | ||||||
|  |  | ||||||
|  | <!-- What operating system are you using ? (be as specific as possible) --> | ||||||
|  | Operating system: | ||||||
|  |  | ||||||
|  | <!-- If you compiled matterbridge yourself: | ||||||
|  |        * Specify the output of `go version`  | ||||||
|  |        * Specify the output of `git rev-parse HEAD` --> | ||||||
|  |  | ||||||
| ### Please describe the expected behavior. | ### Please describe the expected behavior. | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Please describe the actual behavior.  | ### Please describe the actual behavior.  | ||||||
| #### Use logs from running ```matterbridge -debug``` if possible. | <!-- Use logs from running `matterbridge -debug` if possible. --> | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Any steps to reproduce the behavior? | ### Any steps to reproduce the behavior? | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Please add your configuration file  | ### Please add your configuration file  | ||||||
| #### (be sure to exclude or anonymize private data (tokens/passwords)) | <!-- (be sure to exclude or anonymize private data (tokens/passwords)) --> | ||||||
|   | |||||||
							
								
								
									
										49
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,49 @@ | |||||||
|  | language: go | ||||||
|  | go: | ||||||
|  |     #- 1.7.x | ||||||
|  |     - 1.9.x | ||||||
|  |       # - tip | ||||||
|  |  | ||||||
|  | # we have everything vendored | ||||||
|  | install: true | ||||||
|  |  | ||||||
|  | env: | ||||||
|  |     - GOOS=linux GOARCH=amd64 | ||||||
|  |       #    - GOOS=windows GOARCH=amd64 | ||||||
|  |       #- GOOS=linux GOARCH=arm | ||||||
|  |  | ||||||
|  | matrix: | ||||||
|  |   # It's ok if our code fails on unstable development versions of Go. | ||||||
|  |   allow_failures: | ||||||
|  |     - go: tip | ||||||
|  |   # Don't wait for tip tests to finish. Mark the test run green if the | ||||||
|  |   # tests pass on the stable versions of Go. | ||||||
|  |   fast_finish: true | ||||||
|  |  | ||||||
|  | notifications: | ||||||
|  |       email: false | ||||||
|  |  | ||||||
|  | before_script: | ||||||
|  |   - MY_VERSION=$(git describe --tags) | ||||||
|  |   - GO_FILES=$(find . -iname '*.go' | grep -v /vendor/)  # All the .go files, excluding vendor/ | ||||||
|  |   - PKGS=$(go list ./... | grep -v /vendor/)             # All the import paths, excluding vendor/ | ||||||
|  | #  - go get github.com/golang/lint/golint                 # Linter | ||||||
|  |   - go get honnef.co/go/tools/cmd/megacheck              # Badass static analyzer/linter | ||||||
|  |  | ||||||
|  | # Anything in before_script: that returns a nonzero exit code will | ||||||
|  | # flunk the build and immediately stop. It's sorta like having | ||||||
|  | # set -e enabled in bash.  | ||||||
|  | script: | ||||||
|  |  #- test -z $(gofmt -s -l $GO_FILES)  # Fail if a .go file hasn't been formatted with gofmt | ||||||
|  |   - go test -v -race $PKGS            # Run all the tests with the race detector enabled | ||||||
|  |   - go vet $PKGS                      # go vet is the official Go static analyzer | ||||||
|  |   - megacheck $PKGS                   # "go vet on steroids" + linter | ||||||
|  |   - /bin/bash ci/bintray.sh | ||||||
|  |   #- golint -set_exit_status $PKGS     # one last linter | ||||||
|  |  | ||||||
|  | deploy: | ||||||
|  |   provider: bintray | ||||||
|  |   file: ci/deploy.json | ||||||
|  |   user: 42wim | ||||||
|  |   key: | ||||||
|  |      secure: "CeXXe6JOmt7HYR81MdWLua0ltQHhDdkIeRGBFbgd7hkb1wi8eF9DgpAcQrTso8NIlHNZmSAP46uhFgsRvkuezzX0ygalZ7DCJyAyn3sAMEh+UQSHV1WGThRehTtidqRGjetzsIGSwdrJOWil+XTfbO1Z8DGzfakhSuAZka8CM4BAoe3YeP9rYK8h+84x0GHfczvsLtXZ3mWLvQuwe4pK6+ItBCUg0ae7O7ZUpWHy0xQQkkWztY/6RAzXfaG7DuGjIw+20fhx3WOXRNpHCtZ6Bc3qERCpk0s1HhlQWlrN9wDaFTBWYwlvSnNgvxxMbNXJ6RrRJ0l0bA7FUswYwyroxhzrGLdzWDg8dHaQkypocngdalfhpsnoO9j3ApJhomUFJ3UoEq5nOGRUrKn8MPi+dP0zE4kNQ3e4VNa1ufNrvfpWolMg3xh8OXuhQdD5wIM5zFAbRJLqWSCVAjPq4DDPecmvXBOlIial7oa312lN5qnBnUjvAcxszZ+FUyDHT1Grxzna4tMwxY9obPzZUzm7359AOCCwIQFVB8GLqD2nwIstcXS0zGRz+fhviPipHuBa02q5bGUZwmkvrSNab0s8Jo7pCrel2Rz3nWPKaiCfq2WjbW1CLheSMkOQrjsdUd1hhbqNWFPUjJPInTc77NAKCfm5runv5uyowRLh4NNd0sI=" | ||||||
							
								
								
									
										115
									
								
								README-0.6.md
									
									
									
									
									
								
							
							
						
						
									
										115
									
								
								README-0.6.md
									
									
									
									
									
								
							| @@ -1,115 +0,0 @@ | |||||||
| # 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. |  | ||||||
							
								
								
									
										83
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										83
									
								
								README.md
									
									
									
									
									
								
							| @@ -1,18 +1,25 @@ | |||||||
| # 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) | Click on one of the badges below to join the chat    | ||||||
|  |  | ||||||
|  | [](https://gitter.im/42wim/matterbridge) [](https://webchat.freenode.net/?channels=matterbridgechat) [](https://discord.gg/AkKPtrQ) [](https://riot.im/app/#/room/#matterbridge:matrix.org) [](https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA) [](https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e) [](https://inverse.chat) [](https://www.twitch.tv/matterbridge) | ||||||
|  |  | ||||||
|  | [](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion) | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
| Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix and Steam. | Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix, Steam and ssh-chat | ||||||
| Has a REST API.   | Has a REST API.   | ||||||
|  | Minecraft server chat support via [MatterLink](https://github.com/elytra/MatterLink) | ||||||
|  |  | ||||||
| # Table of Contents | # Table of Contents | ||||||
|  * [Features](#features) |  * [Features](https://github.com/42wim/matterbridge/wiki/Features) | ||||||
|  * [Requirements](#requirements) |  * [Requirements](#requirements) | ||||||
|  |  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) | ||||||
|  * [Installing](#installing) |  * [Installing](#installing) | ||||||
|    * [Binaries](#binaries) |    * [Binaries](#binaries) | ||||||
|    * [Building](#building) |    * [Building](#building) | ||||||
|  * [Configuration](#configuration) |  * [Configuration](#configuration) | ||||||
|  |    * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) | ||||||
|    * [Examples](#examples)  |    * [Examples](#examples)  | ||||||
|  * [Running](#running) |  * [Running](#running) | ||||||
|    * [Docker](#docker) |    * [Docker](#docker) | ||||||
| @@ -21,16 +28,25 @@ Has a REST API. | |||||||
|  * [Thanks](#thanks) |  * [Thanks](#thanks) | ||||||
|  |  | ||||||
| # Features | # Features | ||||||
| * Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam.  | * [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols) | ||||||
|   Pick and mix. | * [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols) | ||||||
| * Matterbridge can also work with private groups on your mattermost/slack. | * [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes) | ||||||
| * Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts. | * [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling) | ||||||
| * The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways). | * [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing) | ||||||
| * REST API to read/post messages to bridges (WIP). | * [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups) | ||||||
|  | * [API](https://github.com/42wim/matterbridge/wiki/Features#api) | ||||||
|  |  | ||||||
|  | ## API | ||||||
|  | The API is very basic at the moment and rather undocumented. | ||||||
|  |  | ||||||
|  | Used by at least 2 projects. Feel free to make a PR to add your project to this list. | ||||||
|  |  | ||||||
|  | * [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Server chat) | ||||||
|  | * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||||
|  |  | ||||||
| # Requirements | # Requirements | ||||||
| Accounts to one of the supported bridges | Accounts to one of the supported bridges | ||||||
| * [Mattermost](https://github.com/mattermost/platform/) 3.5.x - 3.10.x | * [Mattermost](https://github.com/mattermost/platform/) 3.8.x - 3.10.x, 4.x | ||||||
| * [IRC](http://www.mirc.com/servers.html) | * [IRC](http://www.mirc.com/servers.html) | ||||||
| * [XMPP](https://jabber.org) | * [XMPP](https://jabber.org) | ||||||
| * [Gitter](https://gitter.im) | * [Gitter](https://gitter.im) | ||||||
| @@ -41,15 +57,19 @@ Accounts to one of the supported bridges | |||||||
| * [Rocket.chat](https://rocket.chat) | * [Rocket.chat](https://rocket.chat) | ||||||
| * [Matrix](https://matrix.org) | * [Matrix](https://matrix.org) | ||||||
| * [Steam](https://store.steampowered.com/) | * [Steam](https://store.steampowered.com/) | ||||||
|  | * [Twitch](https://twitch.tv) | ||||||
|  | * [Ssh-chat](https://github.com/shazow/ssh-chat) | ||||||
|  |  | ||||||
|  | # Screenshots | ||||||
|  | See https://github.com/42wim/matterbridge/wiki | ||||||
|  |  | ||||||
| # Installing | # Installing | ||||||
| ## Binaries | ## Binaries | ||||||
| Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/) | * Latest stable release [v1.8.0](https://github.com/42wim/matterbridge/releases/latest) | ||||||
| * Latest rc release (with steam support) [v0.16.0-rc2](https://github.com/42wim/matterbridge/releases/latest) | * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)   | ||||||
| * Latest stable release [v0.15.0](https://github.com/42wim/matterbridge/releases/tag/v0.15.0) |  | ||||||
|  |  | ||||||
| ## Building | ## 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.8+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH] (https://golang.org/doc/code.html#GOPATH) | ||||||
|  |  | ||||||
| ``` | ``` | ||||||
| cd $GOPATH | cd $GOPATH | ||||||
| @@ -64,12 +84,12 @@ matterbridge | |||||||
| ``` | ``` | ||||||
|  |  | ||||||
| # Configuration | # Configuration | ||||||
| * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | ## Basic configuration | ||||||
| * [matterbridge.toml.simple](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.simple) for a simple example. |  | ||||||
|  |  | ||||||
| ## Create a configuration. |  | ||||||
| See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||||
|  |  | ||||||
|  | ## Advanced configuration | ||||||
|  | * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | ||||||
|  |  | ||||||
| ## Examples  | ## Examples  | ||||||
| ### Bridge mattermost (off-topic) - irc (#testing) | ### Bridge mattermost (off-topic) - irc (#testing) | ||||||
| ``` | ``` | ||||||
| @@ -80,12 +100,12 @@ See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config | |||||||
|  |  | ||||||
| [mattermost] | [mattermost] | ||||||
|     [mattermost.work] |     [mattermost.work] | ||||||
|     useAPI=true |  | ||||||
|     Server="yourmattermostserver.tld" |     Server="yourmattermostserver.tld" | ||||||
|     Team="yourteam" |     Team="yourteam" | ||||||
|     Login="yourlogin" |     Login="yourlogin" | ||||||
|     Password="yourpass" |     Password="yourpass" | ||||||
|     PrefixMessagesWithNick=true |     PrefixMessagesWithNick=true | ||||||
|  |     RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| [[gateway]] | [[gateway]] | ||||||
| name="mygateway" | name="mygateway" | ||||||
| @@ -103,7 +123,6 @@ enable=true | |||||||
| ``` | ``` | ||||||
| [slack] | [slack] | ||||||
| [slack.test] | [slack.test] | ||||||
| useAPI=true |  | ||||||
| Token="yourslacktoken" | Token="yourslacktoken" | ||||||
| PrefixMessagesWithNick=true | PrefixMessagesWithNick=true | ||||||
|  |  | ||||||
| @@ -129,11 +148,8 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | |||||||
| ``` | ``` | ||||||
|  |  | ||||||
| # Running | # 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") | See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||||
|  |  | ||||||
| ``` | ``` | ||||||
| Usage of ./matterbridge: | Usage of ./matterbridge: | ||||||
| @@ -158,18 +174,11 @@ See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.m | |||||||
|  |  | ||||||
| # FAQ | # FAQ | ||||||
|  |  | ||||||
| Please look at [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for more information first. | See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | ||||||
|  |  | ||||||
| ## 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. |  | ||||||
|  |  | ||||||
|  | Want to tip ?  | ||||||
|  | * eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f | ||||||
|  | * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | ||||||
|  |  | ||||||
| # Thanks | # Thanks | ||||||
| Matterbridge wouldn't exist without these libraries: | Matterbridge wouldn't exist without these libraries: | ||||||
| @@ -177,10 +186,10 @@ Matterbridge wouldn't exist without these libraries: | |||||||
| * echo - https://github.com/labstack/echo | * echo - https://github.com/labstack/echo | ||||||
| * gitter - https://github.com/sromku/go-gitter | * gitter - https://github.com/sromku/go-gitter | ||||||
| * gops - https://github.com/google/gops | * gops - https://github.com/google/gops | ||||||
| * irc - https://github.com/thoj/go-ircevent | * irc - https://github.com/lrstanley/girc | ||||||
| * mattermost - https://github.com/mattermost/platform | * mattermost - https://github.com/mattermost/platform | ||||||
| * matrix - https://github.com/matrix-org/gomatrix | * matrix - https://github.com/matrix-org/gomatrix | ||||||
| * slack - https://github.com/nlopes/slack | * slack - https://github.com/nlopes/slack | ||||||
|  | * steam - https://github.com/Philipp15b/go-steam | ||||||
| * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | ||||||
| * xmpp - https://github.com/mattn/go-xmpp | * xmpp - https://github.com/mattn/go-xmpp | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,21 +1,21 @@ | |||||||
| package api | package api | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"encoding/json" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/labstack/echo" | 	"github.com/labstack/echo" | ||||||
| 	"github.com/labstack/echo/middleware" | 	"github.com/labstack/echo/middleware" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| 	"github.com/zfjagann/golang-ring" | 	"github.com/zfjagann/golang-ring" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"sync" | 	"sync" | ||||||
|  | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Api struct { | type Api struct { | ||||||
| 	Config   *config.Protocol |  | ||||||
| 	Remote   chan config.Message |  | ||||||
| 	Account  string |  | ||||||
| 	Messages ring.Ring | 	Messages ring.Ring | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| type ApiMessage struct { | type ApiMessage struct { | ||||||
| @@ -30,26 +30,30 @@ var flog *log.Entry | |||||||
| var protocol = "api" | var protocol = "api" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Api { | func New(cfg *config.BridgeConfig) *Api { | ||||||
| 	b := &Api{} | 	b := &Api{BridgeConfig: cfg} | ||||||
| 	e := echo.New() | 	e := echo.New() | ||||||
|  | 	e.HideBanner = true | ||||||
|  | 	e.HidePort = true | ||||||
| 	b.Messages = ring.Ring{} | 	b.Messages = ring.Ring{} | ||||||
| 	b.Messages.SetCapacity(cfg.Buffer) | 	b.Messages.SetCapacity(b.Config.Buffer) | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.Remote = c |  | ||||||
| 	if b.Config.Token != "" { | 	if b.Config.Token != "" { | ||||||
| 		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) { | 		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) { | ||||||
| 			return key == b.Config.Token, nil | 			return key == b.Config.Token, nil | ||||||
| 		})) | 		})) | ||||||
| 	} | 	} | ||||||
| 	e.GET("/api/messages", b.handleMessages) | 	e.GET("/api/messages", b.handleMessages) | ||||||
|  | 	e.GET("/api/stream", b.handleStream) | ||||||
| 	e.POST("/api/message", b.handlePostMessage) | 	e.POST("/api/message", b.handlePostMessage) | ||||||
| 	go func() { | 	go func() { | ||||||
| 		flog.Fatal(e.Start(cfg.BindAddress)) | 		if b.Config.BindAddress == "" { | ||||||
|  | 			flog.Fatalf("No BindAddress configured.") | ||||||
|  | 		} | ||||||
|  | 		flog.Infof("Listening on %s", b.Config.BindAddress) | ||||||
|  | 		flog.Fatal(e.Start(b.Config.BindAddress)) | ||||||
| 	}() | 	}() | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -61,34 +65,35 @@ func (b *Api) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
| func (b *Api) JoinChannel(channel string) error { | func (b *Api) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) Send(msg config.Message) error { | func (b *Api) Send(msg config.Message) (string, error) { | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	b.Messages.Enqueue(&msg) | 	b.Messages.Enqueue(&msg) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) handlePostMessage(c echo.Context) error { | func (b *Api) handlePostMessage(c echo.Context) error { | ||||||
| 	message := &ApiMessage{} | 	message := config.Message{} | ||||||
| 	if err := c.Bind(message); err != nil { | 	if err := c.Bind(&message); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	// these values are fixed | ||||||
|  | 	message.Channel = "api" | ||||||
|  | 	message.Protocol = "api" | ||||||
|  | 	message.Account = b.Account | ||||||
|  | 	message.ID = "" | ||||||
|  | 	message.Timestamp = time.Now() | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api") | 	flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api") | ||||||
| 	b.Remote <- config.Message{ | 	b.Remote <- message | ||||||
| 		Text:     message.Text, |  | ||||||
| 		Username: message.Username, |  | ||||||
| 		UserID:   message.UserID, |  | ||||||
| 		Channel:  "api", |  | ||||||
| 		Avatar:   message.Avatar, |  | ||||||
| 		Account:  b.Account, |  | ||||||
| 		Gateway:  message.Gateway, |  | ||||||
| 		Protocol: "api", |  | ||||||
| 	} |  | ||||||
| 	return c.JSON(http.StatusOK, message) | 	return c.JSON(http.StatusOK, message) | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -99,3 +104,24 @@ func (b *Api) handleMessages(c echo.Context) error { | |||||||
| 	b.Messages = ring.Ring{} | 	b.Messages = ring.Ring{} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Api) handleStream(c echo.Context) error { | ||||||
|  | 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON) | ||||||
|  | 	c.Response().WriteHeader(http.StatusOK) | ||||||
|  | 	closeNotifier := c.Response().CloseNotify() | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-closeNotifier: | ||||||
|  | 			return nil | ||||||
|  | 		default: | ||||||
|  | 			msg := b.Messages.Dequeue() | ||||||
|  | 			if msg != nil { | ||||||
|  | 				if err := json.NewEncoder(c.Response()).Encode(msg); err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				c.Response().Flush() | ||||||
|  | 			} | ||||||
|  | 			time.Sleep(200 * time.Millisecond) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -10,18 +10,19 @@ import ( | |||||||
| 	"github.com/42wim/matterbridge/bridge/mattermost" | 	"github.com/42wim/matterbridge/bridge/mattermost" | ||||||
| 	"github.com/42wim/matterbridge/bridge/rocketchat" | 	"github.com/42wim/matterbridge/bridge/rocketchat" | ||||||
| 	"github.com/42wim/matterbridge/bridge/slack" | 	"github.com/42wim/matterbridge/bridge/slack" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/sshchat" | ||||||
| 	"github.com/42wim/matterbridge/bridge/steam" | 	"github.com/42wim/matterbridge/bridge/steam" | ||||||
| 	"github.com/42wim/matterbridge/bridge/telegram" | 	"github.com/42wim/matterbridge/bridge/telegram" | ||||||
| 	"github.com/42wim/matterbridge/bridge/xmpp" | 	"github.com/42wim/matterbridge/bridge/xmpp" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
|  |  | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bridger interface { | type Bridger interface { | ||||||
| 	Send(msg config.Message) error | 	Send(msg config.Message) (string, error) | ||||||
| 	Connect() error | 	Connect() error | ||||||
| 	JoinChannel(channel string) error | 	JoinChannel(channel config.ChannelInfo) error | ||||||
| 	Disconnect() error | 	Disconnect() error | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -35,6 +36,12 @@ type Bridge struct { | |||||||
| 	Joined   map[string]bool | 	Joined   map[string]bool | ||||||
| } | } | ||||||
|  |  | ||||||
|  | var flog *log.Entry | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	flog = log.WithFields(log.Fields{"prefix": "bridge"}) | ||||||
|  | } | ||||||
|  |  | ||||||
| func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge { | func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge { | ||||||
| 	b := new(Bridge) | 	b := new(Bridge) | ||||||
| 	b.Channels = make(map[string]config.ChannelInfo) | 	b.Channels = make(map[string]config.ChannelInfo) | ||||||
| @@ -45,66 +52,62 @@ func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Brid | |||||||
| 	b.Protocol = protocol | 	b.Protocol = protocol | ||||||
| 	b.Account = bridge.Account | 	b.Account = bridge.Account | ||||||
| 	b.Joined = make(map[string]bool) | 	b.Joined = make(map[string]bool) | ||||||
|  | 	bridgeConfig := &config.BridgeConfig{General: &cfg.General, Account: bridge.Account, Remote: c} | ||||||
|  |  | ||||||
| 	// override config from environment | 	// override config from environment | ||||||
| 	config.OverrideCfgFromEnv(cfg, protocol, name) | 	config.OverrideCfgFromEnv(cfg, protocol, name) | ||||||
| 	switch protocol { | 	switch protocol { | ||||||
| 	case "mattermost": | 	case "mattermost": | ||||||
| 		b.Config = cfg.Mattermost[name] | 		bridgeConfig.Config = cfg.Mattermost[name] | ||||||
| 		b.Bridger = bmattermost.New(cfg.Mattermost[name], bridge.Account, c) | 		b.Bridger = bmattermost.New(bridgeConfig) | ||||||
| 	case "irc": | 	case "irc": | ||||||
| 		b.Config = cfg.IRC[name] | 		bridgeConfig.Config = cfg.IRC[name] | ||||||
| 		b.Bridger = birc.New(cfg.IRC[name], bridge.Account, c) | 		b.Bridger = birc.New(bridgeConfig) | ||||||
| 	case "gitter": | 	case "gitter": | ||||||
| 		b.Config = cfg.Gitter[name] | 		bridgeConfig.Config = cfg.Gitter[name] | ||||||
| 		b.Bridger = bgitter.New(cfg.Gitter[name], bridge.Account, c) | 		b.Bridger = bgitter.New(bridgeConfig) | ||||||
| 	case "slack": | 	case "slack": | ||||||
| 		b.Config = cfg.Slack[name] | 		bridgeConfig.Config = cfg.Slack[name] | ||||||
| 		b.Bridger = bslack.New(cfg.Slack[name], bridge.Account, c) | 		b.Bridger = bslack.New(bridgeConfig) | ||||||
| 	case "xmpp": | 	case "xmpp": | ||||||
| 		b.Config = cfg.Xmpp[name] | 		bridgeConfig.Config = cfg.Xmpp[name] | ||||||
| 		b.Bridger = bxmpp.New(cfg.Xmpp[name], bridge.Account, c) | 		b.Bridger = bxmpp.New(bridgeConfig) | ||||||
| 	case "discord": | 	case "discord": | ||||||
| 		b.Config = cfg.Discord[name] | 		bridgeConfig.Config = cfg.Discord[name] | ||||||
| 		b.Bridger = bdiscord.New(cfg.Discord[name], bridge.Account, c) | 		b.Bridger = bdiscord.New(bridgeConfig) | ||||||
| 	case "telegram": | 	case "telegram": | ||||||
| 		b.Config = cfg.Telegram[name] | 		bridgeConfig.Config = cfg.Telegram[name] | ||||||
| 		b.Bridger = btelegram.New(cfg.Telegram[name], bridge.Account, c) | 		b.Bridger = btelegram.New(bridgeConfig) | ||||||
| 	case "rocketchat": | 	case "rocketchat": | ||||||
| 		b.Config = cfg.Rocketchat[name] | 		bridgeConfig.Config = cfg.Rocketchat[name] | ||||||
| 		b.Bridger = brocketchat.New(cfg.Rocketchat[name], bridge.Account, c) | 		b.Bridger = brocketchat.New(bridgeConfig) | ||||||
| 	case "matrix": | 	case "matrix": | ||||||
| 		b.Config = cfg.Matrix[name] | 		bridgeConfig.Config = cfg.Matrix[name] | ||||||
| 		b.Bridger = bmatrix.New(cfg.Matrix[name], bridge.Account, c) | 		b.Bridger = bmatrix.New(bridgeConfig) | ||||||
| 	case "steam": | 	case "steam": | ||||||
| 		b.Config = cfg.Steam[name] | 		bridgeConfig.Config = cfg.Steam[name] | ||||||
| 		b.Bridger = bsteam.New(cfg.Steam[name], bridge.Account, c) | 		b.Bridger = bsteam.New(bridgeConfig) | ||||||
|  | 	case "sshchat": | ||||||
|  | 		bridgeConfig.Config = cfg.Sshchat[name] | ||||||
|  | 		b.Bridger = bsshchat.New(bridgeConfig) | ||||||
| 	case "api": | 	case "api": | ||||||
| 		b.Config = cfg.Api[name] | 		bridgeConfig.Config = cfg.Api[name] | ||||||
| 		b.Bridger = api.New(cfg.Api[name], bridge.Account, c) | 		b.Bridger = api.New(bridgeConfig) | ||||||
| 	} | 	} | ||||||
|  | 	b.Config = bridgeConfig.Config | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bridge) JoinChannels() error { | func (b *Bridge) JoinChannels() error { | ||||||
| 	err := b.joinChannels(b.Channels, b.Joined) | 	err := b.joinChannels(b.Channels, b.Joined) | ||||||
| 	if err != nil { |  | ||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | ||||||
| 	mychannel := "" |  | ||||||
| 	for ID, channel := range channels { | 	for ID, channel := range channels { | ||||||
| 		if !exists[ID] { | 		if !exists[ID] { | ||||||
| 			mychannel = channel.Name | 			flog.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID) | ||||||
| 			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID) | 			err := b.JoinChannel(channel) | ||||||
| 			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(mychannel) |  | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
|   | |||||||
| @@ -11,8 +11,13 @@ import ( | |||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	EVENT_JOIN_LEAVE        = "join_leave" | 	EVENT_JOIN_LEAVE        = "join_leave" | ||||||
|  | 	EVENT_TOPIC_CHANGE      = "topic_change" | ||||||
| 	EVENT_FAILURE           = "failure" | 	EVENT_FAILURE           = "failure" | ||||||
|  | 	EVENT_FILE_FAILURE_SIZE = "file_failure_size" | ||||||
|  | 	EVENT_AVATAR_DOWNLOAD   = "avatar_download" | ||||||
| 	EVENT_REJOIN_CHANNELS   = "rejoin_channels" | 	EVENT_REJOIN_CHANNELS   = "rejoin_channels" | ||||||
|  | 	EVENT_USER_ACTION       = "user_action" | ||||||
|  | 	EVENT_MSG_DELETE        = "msg_delete" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Message struct { | type Message struct { | ||||||
| @@ -26,6 +31,18 @@ type Message struct { | |||||||
| 	Protocol  string    `json:"protocol"` | 	Protocol  string    `json:"protocol"` | ||||||
| 	Gateway   string    `json:"gateway"` | 	Gateway   string    `json:"gateway"` | ||||||
| 	Timestamp time.Time `json:"timestamp"` | 	Timestamp time.Time `json:"timestamp"` | ||||||
|  | 	ID        string    `json:"id"` | ||||||
|  | 	Extra     map[string][]interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type FileInfo struct { | ||||||
|  | 	Name    string | ||||||
|  | 	Data    *[]byte | ||||||
|  | 	Comment string | ||||||
|  | 	URL     string | ||||||
|  | 	Size    int64 | ||||||
|  | 	Avatar  bool | ||||||
|  | 	SHA     string | ||||||
| } | } | ||||||
|  |  | ||||||
| type ChannelInfo struct { | type ChannelInfo struct { | ||||||
| @@ -33,7 +50,6 @@ type ChannelInfo struct { | |||||||
| 	Account     string | 	Account     string | ||||||
| 	Direction   string | 	Direction   string | ||||||
| 	ID          string | 	ID          string | ||||||
| 	GID         map[string]bool |  | ||||||
| 	SameChannel map[string]bool | 	SameChannel map[string]bool | ||||||
| 	Options     ChannelOptions | 	Options     ChannelOptions | ||||||
| } | } | ||||||
| @@ -42,18 +58,30 @@ type Protocol struct { | |||||||
| 	AuthCode               string // steam | 	AuthCode               string // steam | ||||||
| 	BindAddress            string // mattermost, slack // DEPRECATED | 	BindAddress            string // mattermost, slack // DEPRECATED | ||||||
| 	Buffer                 int    // api | 	Buffer                 int    // api | ||||||
|  | 	Charset                string // irc | ||||||
|  | 	Debug                  bool   // general | ||||||
| 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||||
| 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||||
| 	IconURL                string // mattermost, slack | 	IconURL                string // mattermost, slack | ||||||
| 	IgnoreNicks            string // all protocols | 	IgnoreNicks            string // all protocols | ||||||
| 	IgnoreMessages         string // all protocols | 	IgnoreMessages         string // all protocols | ||||||
| 	Jid                    string // xmpp | 	Jid                    string // xmpp | ||||||
|  | 	Label                  string // all protocols | ||||||
| 	Login                  string // mattermost, matrix | 	Login                  string // mattermost, matrix | ||||||
|  | 	MediaDownloadSize      int    // all protocols | ||||||
|  | 	MediaServerDownload    string | ||||||
|  | 	MediaServerUpload      string | ||||||
|  | 	MessageDelay           int        // IRC, time in millisecond to wait between messages | ||||||
|  | 	MessageFormat          string     // telegram | ||||||
|  | 	MessageLength          int        // IRC, max length of a message allowed | ||||||
|  | 	MessageQueue           int        // IRC, size of message queue for flood control | ||||||
|  | 	MessageSplit           bool       // IRC, split long messages with newlines on MessageLength instead of clipping | ||||||
| 	Muc                    string     // xmpp | 	Muc                    string     // xmpp | ||||||
| 	Name                   string     // all protocols | 	Name                   string     // all protocols | ||||||
| 	Nick                   string     // all protocols | 	Nick                   string     // all protocols | ||||||
| 	NickFormatter          string     // mattermost, slack | 	NickFormatter          string     // mattermost, slack | ||||||
| 	NickServNick           string     // IRC | 	NickServNick           string     // IRC | ||||||
|  | 	NickServUsername       string     // IRC | ||||||
| 	NickServPassword       string     // IRC | 	NickServPassword       string     // IRC | ||||||
| 	NicksPerRow            int        // mattermost, slack | 	NicksPerRow            int        // mattermost, slack | ||||||
| 	NoHomeServerSuffix     bool       // matrix | 	NoHomeServerSuffix     bool       // matrix | ||||||
| @@ -61,15 +89,16 @@ type Protocol struct { | |||||||
| 	Password               string     // IRC,mattermost,XMPP,matrix | 	Password               string     // IRC,mattermost,XMPP,matrix | ||||||
| 	PrefixMessagesWithNick bool       // mattemost, slack | 	PrefixMessagesWithNick bool       // mattemost, slack | ||||||
| 	Protocol               string     // all protocols | 	Protocol               string     // all protocols | ||||||
| 	MessageQueue           int    // IRC, size of message queue for flood control | 	RejoinDelay            int        // IRC | ||||||
| 	MessageDelay           int    // IRC, time in millisecond to wait between messages | 	ReplaceMessages        [][]string // all protocols | ||||||
| 	MessageLength          int    // IRC, max length of a message allowed | 	ReplaceNicks           [][]string // all protocols | ||||||
| 	MessageFormat          string // telegram |  | ||||||
| 	RemoteNickFormat       string     // all protocols | 	RemoteNickFormat       string     // all protocols | ||||||
| 	Server                 string     // IRC,mattermost,XMPP,discord | 	Server                 string     // IRC,mattermost,XMPP,discord | ||||||
| 	ShowJoinPart           bool       // all protocols | 	ShowJoinPart           bool       // all protocols | ||||||
|  | 	ShowTopicChange        bool       // slack | ||||||
| 	ShowEmbeds             bool       // discord | 	ShowEmbeds             bool       // discord | ||||||
| 	SkipTLSVerify          bool       // IRC, mattermost | 	SkipTLSVerify          bool       // IRC, mattermost | ||||||
|  | 	StripNick              bool       // all protocols | ||||||
| 	Team                   string     // mattermost | 	Team                   string     // mattermost | ||||||
| 	Token                  string     // gitter, slack, discord, api | 	Token                  string     // gitter, slack, discord, api | ||||||
| 	URL                    string     // mattermost, slack // DEPRECATED | 	URL                    string     // mattermost, slack // DEPRECATED | ||||||
| @@ -77,6 +106,7 @@ type Protocol struct { | |||||||
| 	UseSASL                bool       // IRC | 	UseSASL                bool       // IRC | ||||||
| 	UseTLS                 bool       // IRC | 	UseTLS                 bool       // IRC | ||||||
| 	UseFirstName           bool       // telegram | 	UseFirstName           bool       // telegram | ||||||
|  | 	UseUserName            bool       // discord | ||||||
| 	UseInsecureURL         bool       // telegram | 	UseInsecureURL         bool       // telegram | ||||||
| 	WebhookBindAddress     string     // mattermost, slack | 	WebhookBindAddress     string     // mattermost, slack | ||||||
| 	WebhookURL             string     // mattermost, slack | 	WebhookURL             string     // mattermost, slack | ||||||
| @@ -85,6 +115,7 @@ type Protocol struct { | |||||||
|  |  | ||||||
| type ChannelOptions struct { | type ChannelOptions struct { | ||||||
| 	Key        string // irc | 	Key        string // irc | ||||||
|  | 	WebhookURL string // discord | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bridge struct { | type Bridge struct { | ||||||
| @@ -121,11 +152,19 @@ type Config struct { | |||||||
| 	Discord            map[string]Protocol | 	Discord            map[string]Protocol | ||||||
| 	Telegram           map[string]Protocol | 	Telegram           map[string]Protocol | ||||||
| 	Rocketchat         map[string]Protocol | 	Rocketchat         map[string]Protocol | ||||||
|  | 	Sshchat            map[string]Protocol | ||||||
| 	General            Protocol | 	General            Protocol | ||||||
| 	Gateway            []Gateway | 	Gateway            []Gateway | ||||||
| 	SameChannelGateway []SameChannelGateway | 	SameChannelGateway []SameChannelGateway | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type BridgeConfig struct { | ||||||
|  | 	Config  Protocol | ||||||
|  | 	General *Protocol | ||||||
|  | 	Account string | ||||||
|  | 	Remote  chan Message | ||||||
|  | } | ||||||
|  |  | ||||||
| func NewConfig(cfgfile string) *Config { | func NewConfig(cfgfile string) *Config { | ||||||
| 	var cfg Config | 	var cfg Config | ||||||
| 	if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil { | 	if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil { | ||||||
| @@ -153,6 +192,9 @@ func NewConfig(cfgfile string) *Config { | |||||||
| 	if fail { | 	if fail { | ||||||
| 		log.Fatalf("Fix your config. Please see changelog for more information") | 		log.Fatalf("Fix your config. Please see changelog for more information") | ||||||
| 	} | 	} | ||||||
|  | 	if cfg.General.MediaDownloadSize == 0 { | ||||||
|  | 		cfg.General.MediaDownloadSize = 1000000 | ||||||
|  | 	} | ||||||
| 	return &cfg | 	return &cfg | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -209,7 +251,7 @@ func Deprecated(cfg Protocol, account string) bool { | |||||||
| 		log.Printf("ERROR: %s BindAddress is deprecated, you need to change it to WebhookBindAddress.", account) | 		log.Printf("ERROR: %s BindAddress is deprecated, you need to change it to WebhookBindAddress.", account) | ||||||
| 	} else if cfg.URL != "" { | 	} else if cfg.URL != "" { | ||||||
| 		log.Printf("ERROR: %s URL is deprecated, you need to change it to WebhookURL.", account) | 		log.Printf("ERROR: %s URL is deprecated, you need to change it to WebhookURL.", account) | ||||||
| 	} else if cfg.UseAPI == true { | 	} else if cfg.UseAPI { | ||||||
| 		log.Printf("ERROR: %s UseAPI is deprecated, it's enabled by default, please remove it from your config file.", account) | 		log.Printf("ERROR: %s UseAPI is deprecated, it's enabled by default, please remove it from your config file.", account) | ||||||
| 	} else { | 	} else { | ||||||
| 		return false | 		return false | ||||||
|   | |||||||
| @@ -1,9 +1,11 @@ | |||||||
| package bdiscord | package bdiscord | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| @@ -11,9 +13,6 @@ import ( | |||||||
|  |  | ||||||
| type bdiscord struct { | type bdiscord struct { | ||||||
| 	c              *discordgo.Session | 	c              *discordgo.Session | ||||||
| 	Config        *config.Protocol |  | ||||||
| 	Remote        chan config.Message |  | ||||||
| 	Account       string |  | ||||||
| 	Channels       []*discordgo.Channel | 	Channels       []*discordgo.Channel | ||||||
| 	Nick           string | 	Nick           string | ||||||
| 	UseChannelID   bool | 	UseChannelID   bool | ||||||
| @@ -21,27 +20,25 @@ type bdiscord struct { | |||||||
| 	guildID        string | 	guildID        string | ||||||
| 	webhookID      string | 	webhookID      string | ||||||
| 	webhookToken   string | 	webhookToken   string | ||||||
|  | 	channelInfoMap map[string]*config.ChannelInfo | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "discord" | var protocol = "discord" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord { | func New(cfg *config.BridgeConfig) *bdiscord { | ||||||
| 	b := &bdiscord{} | 	b := &bdiscord{BridgeConfig: cfg} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.userMemberMap = make(map[string]*discordgo.Member) | 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||||
|  | 	b.channelInfoMap = make(map[string]*config.ChannelInfo) | ||||||
| 	if b.Config.WebhookURL != "" { | 	if b.Config.WebhookURL != "" { | ||||||
| 		flog.Debug("Configuring Discord Incoming Webhook") | 		flog.Debug("Configuring Discord Incoming Webhook") | ||||||
| 		webhookURLSplit := strings.Split(b.Config.WebhookURL, "/") | 		b.webhookID, b.webhookToken = b.splitURL(b.Config.WebhookURL) | ||||||
| 		b.webhookToken = webhookURLSplit[len(webhookURLSplit)-1] |  | ||||||
| 		b.webhookID = webhookURLSplit[len(webhookURLSplit)-2] |  | ||||||
| 	} | 	} | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -66,12 +63,13 @@ func (b *bdiscord) Connect() error { | |||||||
| 	b.c.AddHandler(b.messageCreate) | 	b.c.AddHandler(b.messageCreate) | ||||||
| 	b.c.AddHandler(b.memberUpdate) | 	b.c.AddHandler(b.memberUpdate) | ||||||
| 	b.c.AddHandler(b.messageUpdate) | 	b.c.AddHandler(b.messageUpdate) | ||||||
|  | 	b.c.AddHandler(b.messageDelete) | ||||||
| 	err = b.c.Open() | 	err = b.c.Open() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	guilds, err := b.c.UserGuilds() | 	guilds, err := b.c.UserGuilds(100, "", "") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| @@ -99,37 +97,96 @@ func (b *bdiscord) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) JoinChannel(channel string) error { | func (b *bdiscord) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	idcheck := strings.Split(channel, "ID:") | 	b.channelInfoMap[channel.ID] = &channel | ||||||
|  | 	idcheck := strings.Split(channel.Name, "ID:") | ||||||
| 	if len(idcheck) > 1 { | 	if len(idcheck) > 1 { | ||||||
| 		b.UseChannelID = true | 		b.UseChannelID = true | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) Send(msg config.Message) error { | func (b *bdiscord) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	channelID := b.getChannelID(msg.Channel) | 	channelID := b.getChannelID(msg.Channel) | ||||||
| 	if channelID == "" { | 	if channelID == "" { | ||||||
| 		flog.Errorf("Could not find channelID for %v", msg.Channel) | 		flog.Errorf("Could not find channelID for %v", msg.Channel) | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL == "" { | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "_" + msg.Text + "_" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wID := b.webhookID | ||||||
|  | 	wToken := b.webhookToken | ||||||
|  | 	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok { | ||||||
|  | 		if ci.Options.WebhookURL != "" { | ||||||
|  | 			wID, wToken = b.splitURL(ci.Options.WebhookURL) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if wID == "" { | ||||||
| 		flog.Debugf("Broadcasting using token (API)") | 		flog.Debugf("Broadcasting using token (API)") | ||||||
| 		b.c.ChannelMessageSend(channelID, msg.Username+msg.Text) | 		if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	} else { | 			if msg.ID == "" { | ||||||
|  | 				return "", nil | ||||||
|  | 			} | ||||||
|  | 			err := b.c.ChannelMessageDelete(channelID, msg.ID) | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		if msg.ID != "" { | ||||||
|  | 			_, err := b.c.ChannelMessageEdit(channelID, msg.ID, msg.Username+msg.Text) | ||||||
|  | 			return msg.ID, err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if msg.Extra != nil { | ||||||
|  | 			for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 				b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text) | ||||||
|  | 			} | ||||||
|  | 			// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 			if len(msg.Extra["file"]) > 0 { | ||||||
|  | 				var err error | ||||||
|  | 				for _, f := range msg.Extra["file"] { | ||||||
|  | 					fi := f.(config.FileInfo) | ||||||
|  | 					files := []*discordgo.File{} | ||||||
|  | 					files = append(files, &discordgo.File{fi.Name, "", bytes.NewReader(*fi.Data)}) | ||||||
|  | 					_, err = b.c.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{Content: msg.Username + fi.Comment, Files: files}) | ||||||
|  | 					if err != nil { | ||||||
|  | 						flog.Errorf("file upload failed: %#v", err) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				return "", nil | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		res, err := b.c.ChannelMessageSend(channelID, msg.Username+msg.Text) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return res.ID, err | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Broadcasting using Webhook") | 	flog.Debugf("Broadcasting using Webhook") | ||||||
| 		b.c.WebhookExecute( | 	err := b.c.WebhookExecute( | ||||||
| 			b.webhookID, | 		wID, | ||||||
| 			b.webhookToken, | 		wToken, | ||||||
| 		true, | 		true, | ||||||
| 		&discordgo.WebhookParams{ | 		&discordgo.WebhookParams{ | ||||||
| 			Content:   msg.Text, | 			Content:   msg.Text, | ||||||
| 			Username:  msg.Username, | 			Username:  msg.Username, | ||||||
| 			AvatarURL: msg.Avatar, | 			AvatarURL: msg.Avatar, | ||||||
| 		}) | 		}) | ||||||
|  | 	return "", err | ||||||
| } | } | ||||||
| 	return nil |  | ||||||
|  | func (b *bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { | ||||||
|  | 	rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EVENT_MSG_DELETE, Text: config.EVENT_MSG_DELETE} | ||||||
|  | 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||||
|  | 	if b.UseChannelID { | ||||||
|  | 		rmsg.Channel = "ID:" + m.ChannelID | ||||||
|  | 	} | ||||||
|  | 	flog.Debugf("Sending message from %s to gateway", b.Account) | ||||||
|  | 	flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 	b.Remote <- rmsg | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | ||||||
| @@ -145,45 +202,68 @@ func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdat | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | ||||||
|  | 	var err error | ||||||
| 	// not relay our own messages | 	// not relay our own messages | ||||||
| 	if m.Author.Username == b.Nick { | 	if m.Author.Username == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// if using webhooks, do not relay if it's ours | 	// if using webhooks, do not relay if it's ours | ||||||
| 	if b.Config.WebhookURL != "" && m.Author.Bot && m.Author.ID == b.webhookID { | 	if b.useWebhook() && m.Author.Bot && b.isWebhookID(m.Author.ID) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if len(m.Attachments) > 0 { | 	if len(m.Attachments) > 0 { | ||||||
| 		for _, attach := range m.Attachments { | 		for _, attach := range m.Attachments { | ||||||
| 			m.Content = m.Content + "\n" + attach.URL | 			m.Content = m.Content + "\n" + attach.URL | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	if m.Content == "" { |  | ||||||
| 		return | 	var text string | ||||||
| 	} | 	if m.Content != "" { | ||||||
| 		flog.Debugf("Receiving message %#v", m.Message) | 		flog.Debugf("Receiving message %#v", m.Message) | ||||||
| 	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) | 		m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||||
| 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | ||||||
|  | 		text, err = m.ContentWithMoreMentionsReplaced(b.c) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("ContentWithMoreMentionsReplaced failed: %s", err) | ||||||
|  | 			text = m.ContentWithMentionsReplaced() | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", | ||||||
|  | 		UserID: m.Author.ID, ID: m.ID} | ||||||
|  |  | ||||||
|  | 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||||
|  | 	if b.UseChannelID { | ||||||
|  | 		rmsg.Channel = "ID:" + m.ChannelID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !b.Config.UseUserName { | ||||||
|  | 		rmsg.Username = b.getNick(m.Author) | ||||||
|  | 	} else { | ||||||
|  | 		rmsg.Username = m.Author.Username | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	text := m.ContentWithMentionsReplaced() |  | ||||||
| 	if b.Config.ShowEmbeds && m.Message.Embeds != nil { | 	if b.Config.ShowEmbeds && m.Message.Embeds != nil { | ||||||
| 		for _, embed := range m.Message.Embeds { | 		for _, embed := range m.Message.Embeds { | ||||||
| 			text = text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | 			text = text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// no empty messages | ||||||
|  | 	if text == "" { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	text, ok := b.replaceAction(text) | ||||||
|  | 	if ok { | ||||||
|  | 		rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rmsg.Text = text | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account) | 	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||||
| 	b.Remote <- config.Message{Username: username, Text: text, Channel: channelName, | 	flog.Debugf("Message is %#v", rmsg) | ||||||
| 		Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", | 	b.Remote <- rmsg | ||||||
| 		UserID: m.Author.ID} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | func (b *bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||||
| @@ -244,18 +324,6 @@ func (b *bdiscord) getChannelName(id string) string { | |||||||
| 	return "" | 	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) replaceChannelMentions(text string) string { | func (b *bdiscord) replaceChannelMentions(text string) string { | ||||||
| 	var err error | 	var err error | ||||||
| 	re := regexp.MustCompile("<#[0-9]+>") | 	re := regexp.MustCompile("<#[0-9]+>") | ||||||
| @@ -275,8 +343,56 @@ func (b *bdiscord) replaceChannelMentions(text string) string { | |||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *bdiscord) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||||
|  | 		return strings.Replace(text, "_", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|  |  | ||||||
| func (b *bdiscord) stripCustomoji(text string) string { | func (b *bdiscord) stripCustomoji(text string) string { | ||||||
| 	// <:doge:302803592035958784> | 	// <:doge:302803592035958784> | ||||||
| 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||||
| 	return re.ReplaceAllString(text, `$1`) | 	return re.ReplaceAllString(text, `$1`) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // splitURL splits a webhookURL and returns the id and token | ||||||
|  | func (b *bdiscord) splitURL(url string) (string, string) { | ||||||
|  | 	webhookURLSplit := strings.Split(url, "/") | ||||||
|  | 	if len(webhookURLSplit) != 7 { | ||||||
|  | 		log.Fatalf("%s is no correct discord WebhookURL", url) | ||||||
|  | 	} | ||||||
|  | 	return webhookURLSplit[len(webhookURLSplit)-2], webhookURLSplit[len(webhookURLSplit)-1] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // useWebhook returns true if we have a webhook defined somewhere | ||||||
|  | func (b *bdiscord) useWebhook() bool { | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	for _, channel := range b.channelInfoMap { | ||||||
|  | 		if channel.Options.WebhookURL != "" { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // isWebhookID returns true if the specified id is used in a defined webhook | ||||||
|  | func (b *bdiscord) isWebhookID(id string) bool { | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		wID, _ := b.splitURL(b.Config.WebhookURL) | ||||||
|  | 		if wID == id { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, channel := range b.channelInfoMap { | ||||||
|  | 		if channel.Options.WebhookURL != "" { | ||||||
|  | 			wID, _ := b.splitURL(channel.Options.WebhookURL) | ||||||
|  | 			if wID == id { | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -2,41 +2,37 @@ package bgitter | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"github.com/42wim/go-gitter" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/sromku/go-gitter" | 	log "github.com/sirupsen/logrus" | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bgitter struct { | type Bgitter struct { | ||||||
| 	c     *gitter.Gitter | 	c     *gitter.Gitter | ||||||
| 	Config  *config.Protocol | 	User  *gitter.User | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| 	Users []gitter.User | 	Users []gitter.User | ||||||
| 	Rooms []gitter.Room | 	Rooms []gitter.Room | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "gitter" | var protocol = "gitter" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bgitter { | func New(cfg *config.BridgeConfig) *Bgitter { | ||||||
| 	b := &Bgitter{} | 	return &Bgitter{BridgeConfig: cfg} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) Connect() error { | func (b *Bgitter) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Info("Connecting") | 	flog.Info("Connecting") | ||||||
| 	b.c = gitter.New(b.Config.Token) | 	b.c = gitter.New(b.Config.Token) | ||||||
| 	_, err = b.c.GetUser() | 	b.User, err = b.c.GetUser() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| @@ -51,10 +47,10 @@ func (b *Bgitter) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) JoinChannel(channel string) error { | func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	roomID, err := b.c.GetRoomId(channel) | 	roomID, err := b.c.GetRoomId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel) | 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel.Name) | ||||||
| 	} | 	} | ||||||
| 	room, err := b.c.GetRoom(roomID) | 	room, err := b.c.GetRoom(roomID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -78,29 +74,80 @@ func (b *Bgitter) JoinChannel(channel string) error { | |||||||
| 		for event := range stream.Event { | 		for event := range stream.Event { | ||||||
| 			switch ev := event.Data.(type) { | 			switch ev := event.Data.(type) { | ||||||
| 			case *gitter.MessageReceived: | 			case *gitter.MessageReceived: | ||||||
| 				// check for ZWSP to see if it's not an echo | 				if ev.Message.From.ID != b.User.ID { | ||||||
| 				if !strings.HasSuffix(ev.Message.Text, "") { |  | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | 					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | ||||||
| 					b.Remote <- config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | 					rmsg := config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | ||||||
| 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID} | 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID, | ||||||
|  | 						ID: ev.Message.ID} | ||||||
|  | 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | ||||||
|  | 						rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 					b.Remote <- rmsg | ||||||
| 				} | 				} | ||||||
| 			case *gitter.GitterConnectionClosed: | 			case *gitter.GitterConnectionClosed: | ||||||
| 				flog.Errorf("connection with gitter closed for room %s", room) | 				flog.Errorf("connection with gitter closed for room %s", room) | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	}(stream, room.Name) | 	}(stream, room.URI) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) Send(msg config.Message) error { | func (b *Bgitter) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	roomID := b.getRoomID(msg.Channel) | 	roomID := b.getRoomID(msg.Channel) | ||||||
| 	if roomID == "" { | 	if roomID == "" { | ||||||
| 		flog.Errorf("Could not find roomID for %v", msg.Channel) | 		flog.Errorf("Could not find roomID for %v", msg.Channel) | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	// add ZWSP because gitter echoes our own messages | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	return b.c.SendMessage(roomID, msg.Username+msg.Text+" ") | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		// gitter has no delete message api | ||||||
|  | 		_, err := b.c.UpdateMessage(roomID, msg.ID, "") | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		flog.Debugf("updating message with id %s", msg.ID) | ||||||
|  | 		_, err := b.c.UpdateMessage(roomID, msg.ID, msg.Username+msg.Text) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.c.SendMessage(roomID, rmsg.Username+rmsg.Text) | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				if fi.Comment != "" { | ||||||
|  | 					msg.Text += fi.Comment + ": " | ||||||
|  | 				} | ||||||
|  | 				if fi.URL != "" { | ||||||
|  | 					msg.Text = fi.URL | ||||||
|  | 				} | ||||||
|  | 				_, err := b.c.SendMessage(roomID, msg.Username+msg.Text) | ||||||
|  | 				if err != nil { | ||||||
|  | 					return "", err | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	resp, err := b.c.SendMessage(roomID, msg.Username+msg.Text) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return resp.ID, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) getRoomID(channel string) string { | func (b *Bgitter) getRoomID(channel string) string { | ||||||
|   | |||||||
							
								
								
									
										63
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | |||||||
|  | package helper | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func DownloadFile(url string) (*[]byte, error) { | ||||||
|  | 	var buf bytes.Buffer | ||||||
|  | 	client := &http.Client{ | ||||||
|  | 		Timeout: time.Second * 5, | ||||||
|  | 	} | ||||||
|  | 	req, err := http.NewRequest("GET", url, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	resp, err := client.Do(req) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer resp.Body.Close() | ||||||
|  | 	io.Copy(&buf, resp.Body) | ||||||
|  | 	data := buf.Bytes() | ||||||
|  | 	return &data, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func SplitStringLength(input string, length int) string { | ||||||
|  | 	a := []rune(input) | ||||||
|  | 	str := "" | ||||||
|  | 	for i, r := range a { | ||||||
|  | 		str = str + string(r) | ||||||
|  | 		if i > 0 && (i+1)%length == 0 { | ||||||
|  | 			str += "\n" | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return str | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handle all the stuff we put into extra | ||||||
|  | func HandleExtra(msg *config.Message, general *config.Protocol) []config.Message { | ||||||
|  | 	extra := msg.Extra | ||||||
|  | 	rmsg := []config.Message{} | ||||||
|  | 	if len(extra[config.EVENT_FILE_FAILURE_SIZE]) > 0 { | ||||||
|  | 		for _, f := range extra[config.EVENT_FILE_FAILURE_SIZE] { | ||||||
|  | 			fi := f.(config.FileInfo) | ||||||
|  | 			text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize) | ||||||
|  | 			rmsg = append(rmsg, config.Message{Text: text, Username: "<system> ", Channel: msg.Channel}) | ||||||
|  | 		} | ||||||
|  | 		return rmsg | ||||||
|  | 	} | ||||||
|  | 	return rmsg | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func GetAvatar(av map[string]string, userid string, general *config.Protocol) string { | ||||||
|  | 	if sha, ok := av[userid]; ok { | ||||||
|  | 		return general.MediaServerDownload + "/" + sha + "/" + userid + ".png" | ||||||
|  | 	} | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
| @@ -4,6 +4,7 @@ import ( | |||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | /* | ||||||
| func tableformatter(nicks []string, nicksPerRow int, continued bool) string { | func tableformatter(nicks []string, nicksPerRow int, continued bool) string { | ||||||
| 	result := "|IRC users" | 	result := "|IRC users" | ||||||
| 	if continued { | 	if continued { | ||||||
| @@ -29,6 +30,7 @@ func tableformatter(nicks []string, nicksPerRow int, continued bool) string { | |||||||
| 	} | 	} | ||||||
| 	return result | 	return result | ||||||
| } | } | ||||||
|  | */ | ||||||
|  |  | ||||||
| func plainformatter(nicks []string, nicksPerRow int) string { | func plainformatter(nicks []string, nicksPerRow int) string { | ||||||
| 	return strings.Join(nicks, ", ") + " currently on IRC" | 	return strings.Join(nicks, ", ") + " currently on IRC" | ||||||
|   | |||||||
| @@ -1,50 +1,50 @@ | |||||||
| package birc | package birc | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	"github.com/lrstanley/girc" | ||||||
| 	"github.com/paulrosania/go-charset/charset" | 	"github.com/paulrosania/go-charset/charset" | ||||||
| 	_ "github.com/paulrosania/go-charset/data" | 	_ "github.com/paulrosania/go-charset/data" | ||||||
| 	"github.com/saintfish/chardet" | 	"github.com/saintfish/chardet" | ||||||
| 	ircm "github.com/sorcix/irc" | 	log "github.com/sirupsen/logrus" | ||||||
| 	"github.com/thoj/go-ircevent" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
|  | 	"net" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"sort" | 	"sort" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  | 	"unicode/utf8" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Birc struct { | type Birc struct { | ||||||
| 	i               *irc.Connection | 	i               *girc.Client | ||||||
| 	Nick            string | 	Nick            string | ||||||
| 	names           map[string][]string | 	names           map[string][]string | ||||||
| 	Config          *config.Protocol |  | ||||||
| 	Remote          chan config.Message |  | ||||||
| 	connected       chan struct{} | 	connected       chan struct{} | ||||||
| 	Local           chan config.Message // local queue for flood control | 	Local           chan config.Message // local queue for flood control | ||||||
| 	Account         string |  | ||||||
| 	FirstConnection bool | 	FirstConnection bool | ||||||
|  |  | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "irc" | var protocol = "irc" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Birc { | func New(cfg *config.BridgeConfig) *Birc { | ||||||
| 	b := &Birc{} | 	b := &Birc{} | ||||||
| 	b.Config = &cfg | 	b.BridgeConfig = cfg | ||||||
| 	b.Nick = b.Config.Nick | 	b.Nick = b.Config.Nick | ||||||
| 	b.Remote = c |  | ||||||
| 	b.names = make(map[string][]string) | 	b.names = make(map[string][]string) | ||||||
| 	b.Account = account |  | ||||||
| 	b.connected = make(chan struct{}) | 	b.connected = make(chan struct{}) | ||||||
| 	if b.Config.MessageDelay == 0 { | 	if b.Config.MessageDelay == 0 { | ||||||
| 		b.Config.MessageDelay = 1300 | 		b.Config.MessageDelay = 1300 | ||||||
| @@ -62,9 +62,9 @@ func New(cfg config.Protocol, account string, c chan config.Message) *Birc { | |||||||
| func (b *Birc) Command(msg *config.Message) string { | func (b *Birc) Command(msg *config.Message) string { | ||||||
| 	switch msg.Text { | 	switch msg.Text { | ||||||
| 	case "!users": | 	case "!users": | ||||||
| 		b.i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames) | 		b.i.Handlers.Add(girc.RPL_NAMREPLY, b.storeNames) | ||||||
| 		b.i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames) | 		b.i.Handlers.Add(girc.RPL_ENDOFNAMES, b.endNames) | ||||||
| 		b.i.SendRaw("NAMES " + msg.Channel) | 		b.i.Cmd.SendRaw("NAMES " + msg.Channel) | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
| @@ -72,25 +72,60 @@ func (b *Birc) Command(msg *config.Message) string { | |||||||
| func (b *Birc) Connect() error { | func (b *Birc) Connect() error { | ||||||
| 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | ||||||
| 	flog.Infof("Connecting %s", b.Config.Server) | 	flog.Infof("Connecting %s", b.Config.Server) | ||||||
| 	i := irc.IRC(b.Config.Nick, b.Config.Nick) | 	server, portstr, err := net.SplitHostPort(b.Config.Server) | ||||||
| 	if log.GetLevel() == log.DebugLevel { |  | ||||||
| 		i.Debug = true |  | ||||||
| 	} |  | ||||||
| 	i.UseTLS = b.Config.UseTLS |  | ||||||
| 	i.UseSASL = b.Config.UseSASL |  | ||||||
| 	i.SASLLogin = b.Config.NickServNick |  | ||||||
| 	i.SASLPassword = b.Config.NickServPassword |  | ||||||
| 	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify} |  | ||||||
| 	i.KeepAlive = time.Minute |  | ||||||
| 	i.PingFreq = time.Minute |  | ||||||
| 	if b.Config.Password != "" { |  | ||||||
| 		i.Password = b.Config.Password |  | ||||||
| 	} |  | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, b.handleNewConnection) |  | ||||||
| 	err := i.Connect(b.Config.Server) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	port, err := strconv.Atoi(portstr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	// fix strict user handling of girc | ||||||
|  | 	user := b.Config.Nick | ||||||
|  | 	for !girc.IsValidUser(user) { | ||||||
|  | 		if len(user) == 1 { | ||||||
|  | 			user = "matterbridge" | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		user = user[1:] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	i := girc.New(girc.Config{ | ||||||
|  | 		Server:     server, | ||||||
|  | 		ServerPass: b.Config.Password, | ||||||
|  | 		Port:       port, | ||||||
|  | 		Nick:       b.Config.Nick, | ||||||
|  | 		User:       user, | ||||||
|  | 		Name:       b.Config.Nick, | ||||||
|  | 		SSL:        b.Config.UseTLS, | ||||||
|  | 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, ServerName: server}, | ||||||
|  | 		PingDelay:  time.Minute, | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	if b.Config.UseSASL { | ||||||
|  | 		i.Config.SASL = &girc.SASLPlain{b.Config.NickServNick, b.Config.NickServPassword} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	i.Handlers.Add(girc.RPL_WELCOME, b.handleNewConnection) | ||||||
|  | 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||||
|  | 	i.Handlers.Add("*", b.handleOther) | ||||||
|  | 	go func() { | ||||||
|  | 		for { | ||||||
|  | 			if err := i.Connect(); err != nil { | ||||||
|  | 				flog.Errorf("error: %s", err) | ||||||
|  | 				flog.Info("reconnecting in 30 seconds...") | ||||||
|  | 				time.Sleep(30 * time.Second) | ||||||
|  | 				i.Handlers.Clear(girc.RPL_WELCOME) | ||||||
|  | 				i.Handlers.Add(girc.RPL_WELCOME, func(client *girc.Client, event girc.Event) { | ||||||
|  | 					b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
|  | 					// set our correct nick on reconnect if necessary | ||||||
|  | 					b.Nick = event.Source.Name | ||||||
|  | 				}) | ||||||
|  | 			} else { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
| 	b.i = i | 	b.i = i | ||||||
| 	select { | 	select { | ||||||
| 	case <-b.connected: | 	case <-b.connected: | ||||||
| @@ -98,15 +133,8 @@ func (b *Birc) Connect() error { | |||||||
| 	case <-time.After(time.Second * 30): | 	case <-time.After(time.Second * 30): | ||||||
| 		return fmt.Errorf("connection timed out") | 		return fmt.Errorf("connection timed out") | ||||||
| 	} | 	} | ||||||
| 	i.Debug = false | 	//i.Debug = false | ||||||
| 	// clear on reconnects | 	i.Handlers.Clear("*") | ||||||
| 	i.ClearCallback(ircm.RPL_WELCOME) |  | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, func(event *irc.Event) { |  | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} |  | ||||||
| 		// set our correct nick on reconnect if necessary |  | ||||||
| 		b.Nick = event.Nick |  | ||||||
| 	}) |  | ||||||
| 	go i.Loop() |  | ||||||
| 	go b.doSend() | 	go b.doSend() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -117,46 +145,96 @@ func (b *Birc) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) JoinChannel(channel string) error { | func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.i.Join(channel) | 	if channel.Options.Key != "" { | ||||||
|  | 		flog.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||||
|  | 		b.i.Cmd.JoinKey(channel.Name, channel.Options.Key) | ||||||
|  | 	} else { | ||||||
|  | 		b.i.Cmd.Join(channel.Name) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Send(msg config.Message) error { | func (b *Birc) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	// ignore delete messages | ||||||
| 	if msg.Account == b.Account { | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
|  | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	if strings.HasPrefix(msg.Text, "!") { | 	if strings.HasPrefix(msg.Text, "!") { | ||||||
| 		b.Command(&msg) | 		b.Command(&msg) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if b.Config.Charset != "" { | ||||||
|  | 		buf := new(bytes.Buffer) | ||||||
|  | 		w, err := charset.NewWriter(b.Config.Charset, buf) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("charset from utf-8 conversion failed: %s", err) | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		fmt.Fprintf(w, msg.Text) | ||||||
|  | 		w.Close() | ||||||
|  | 		msg.Text = buf.String() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.Local <- rmsg | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				if fi.Comment != "" { | ||||||
|  | 					msg.Text += fi.Comment + ": " | ||||||
|  | 				} | ||||||
|  | 				if fi.URL != "" { | ||||||
|  | 					msg.Text = fi.URL | ||||||
|  | 				} | ||||||
|  | 				b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||||
|  | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// split long messages on messageLength, to avoid clipped messages #281 | ||||||
|  | 	if b.Config.MessageSplit { | ||||||
|  | 		msg.Text = helper.SplitStringLength(msg.Text, b.Config.MessageLength) | ||||||
|  | 	} | ||||||
| 	for _, text := range strings.Split(msg.Text, "\n") { | 	for _, text := range strings.Split(msg.Text, "\n") { | ||||||
| 		if len(text) > b.Config.MessageLength { | 		if len(text) > b.Config.MessageLength { | ||||||
| 			text = text[:b.Config.MessageLength] + " <message clipped>" | 			text = text[:b.Config.MessageLength-len(" <message clipped>")] | ||||||
|  | 			if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||||
|  | 				text = text[:len(text)-size] | ||||||
|  | 			} | ||||||
|  | 			text += " <message clipped>" | ||||||
| 		} | 		} | ||||||
| 		if len(b.Local) < b.Config.MessageQueue { | 		if len(b.Local) < b.Config.MessageQueue { | ||||||
| 			if len(b.Local) == b.Config.MessageQueue-1 { | 			if len(b.Local) == b.Config.MessageQueue-1 { | ||||||
| 				text = text + " <message clipped>" | 				text = text + " <message clipped>" | ||||||
| 			} | 			} | ||||||
| 			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel} | 			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||||
| 		} else { | 		} else { | ||||||
| 			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | 			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) doSend() { | func (b *Birc) doSend() { | ||||||
| 	rate := time.Millisecond * time.Duration(b.Config.MessageDelay) | 	rate := time.Millisecond * time.Duration(b.Config.MessageDelay) | ||||||
| 	throttle := time.Tick(rate) | 	throttle := time.NewTicker(rate) | ||||||
| 	for msg := range b.Local { | 	for msg := range b.Local { | ||||||
| 		<-throttle | 		<-throttle.C | ||||||
| 		b.i.Privmsg(msg.Channel, msg.Username+msg.Text) | 		if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 			b.i.Cmd.Action(msg.Channel, msg.Username+msg.Text) | ||||||
|  | 		} else { | ||||||
|  | 			b.i.Cmd.Message(msg.Channel, msg.Username+msg.Text) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) endNames(event *irc.Event) { | func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[1] | 	channel := event.Params[1] | ||||||
| 	sort.Strings(b.names[channel]) | 	sort.Strings(b.names[channel]) | ||||||
| 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | ||||||
| 	continued := false | 	continued := false | ||||||
| @@ -169,94 +247,107 @@ func (b *Birc) endNames(event *irc.Event) { | |||||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | ||||||
| 		Channel: channel, Account: b.Account} | 		Channel: channel, Account: b.Account} | ||||||
| 	b.names[channel] = nil | 	b.names[channel] = nil | ||||||
| 	b.i.ClearCallback(ircm.RPL_NAMREPLY) | 	b.i.Handlers.Clear(girc.RPL_NAMREPLY) | ||||||
| 	b.i.ClearCallback(ircm.RPL_ENDOFNAMES) | 	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNewConnection(event *irc.Event) { | func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||||
| 	flog.Debug("Registering callbacks") | 	flog.Debug("Registering callbacks") | ||||||
| 	i := b.i | 	i := b.i | ||||||
| 	b.Nick = event.Arguments[0] | 	b.Nick = event.Params[0] | ||||||
| 	i.AddCallback("PRIVMSG", b.handlePrivMsg) |  | ||||||
| 	i.AddCallback("CTCP_ACTION", b.handlePrivMsg) | 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||||
| 	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||||
| 	i.AddCallback(ircm.NOTICE, b.handleNotice) | 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||||
| 	//i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) }) | 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||||
| 	i.AddCallback("PING", func(e *irc.Event) { | 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||||
| 		i.SendRaw("PONG :" + e.Message()) | 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||||
| 		flog.Debugf("PING/PONG") | 	i.Handlers.Add("PART", b.handleJoinPart) | ||||||
| 	}) | 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||||
| 	i.AddCallback("JOIN", b.handleJoinPart) | 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||||
| 	i.AddCallback("PART", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("QUIT", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("KICK", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("*", b.handleOther) |  | ||||||
| 	// we are now fully connected | 	// we are now fully connected | ||||||
| 	b.connected <- struct{}{} | 	b.connected <- struct{}{} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleJoinPart(event *irc.Event) { | func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[0] | 	if len(event.Params) == 0 { | ||||||
| 	if event.Code == "KICK" { | 		flog.Debugf("handleJoinPart: empty Params? %#v", event) | ||||||
| 		flog.Infof("Got kicked from %s by %s", channel, event.Nick) | 		return | ||||||
|  | 	} | ||||||
|  | 	channel := event.Params[0] | ||||||
|  | 	if event.Command == "KICK" { | ||||||
|  | 		flog.Infof("Got kicked from %s by %s", channel, event.Source.Name) | ||||||
|  | 		time.Sleep(time.Duration(b.Config.RejoinDelay) * time.Second) | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if event.Code == "QUIT" { | 	if event.Command == "QUIT" { | ||||||
| 		if event.Nick == b.Nick && strings.Contains(event.Raw, "Ping timeout") { | 		if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") { | ||||||
| 			flog.Infof("%s reconnecting ..", b.Account) | 			flog.Infof("%s reconnecting ..", b.Account) | ||||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	if event.Nick != b.Nick { | 	if event.Source.Name != b.Nick { | ||||||
| 		flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | 		flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: event.Nick + " " + strings.ToLower(event.Code) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | 		b.Remote <- config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("handle %#v", event) | 	flog.Debugf("handle %#v", event) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNotice(event *irc.Event) { | func (b *Birc) handleNotice(client *girc.Client, event girc.Event) { | ||||||
| 	if strings.Contains(event.Message(), "This nickname is registered") && event.Nick == b.Config.NickServNick { | 	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.Config.NickServNick { | ||||||
| 		b.i.Privmsg(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword) | 		b.i.Cmd.Message(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword) | ||||||
| 	} else { | 	} else { | ||||||
| 		b.handlePrivMsg(event) | 		b.handlePrivMsg(client, event) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleOther(event *irc.Event) { | func (b *Birc) handleOther(client *girc.Client, event girc.Event) { | ||||||
| 	switch event.Code { | 	switch event.Command { | ||||||
| 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("%#v", event.Raw) | 	flog.Debugf("%#v", event.String()) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handlePrivMsg(event *irc.Event) { | func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) { | ||||||
|  | 	if strings.EqualFold(b.Config.NickServNick, "Q@CServe.quakenet.org") { | ||||||
|  | 		flog.Debugf("Authenticating %s against %s", b.Config.NickServUsername, b.Config.NickServNick) | ||||||
|  | 		b.i.Cmd.Message(b.Config.NickServNick, "AUTH "+b.Config.NickServUsername+" "+b.Config.NickServPassword) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||||
| 	b.Nick = b.i.GetNick() | 	b.Nick = b.i.GetNick() | ||||||
| 	// freenode doesn't send 001 as first reply | 	// freenode doesn't send 001 as first reply | ||||||
| 	if event.Code == "NOTICE" { | 	if event.Command == "NOTICE" { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// don't forward queries to the bot | 	// don't forward queries to the bot | ||||||
| 	if event.Arguments[0] == b.Nick { | 	if event.Params[0] == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// don't forward message from ourself | 	// don't forward message from ourself | ||||||
| 	if event.Nick == b.Nick { | 	if event.Source.Name == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("handlePrivMsg() %s %s %#v", event.Nick, event.Message(), event) | 	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||||
|  | 	flog.Debugf("handlePrivMsg() %s %s %#v", event.Source.Name, event.Trailing, event) | ||||||
| 	msg := "" | 	msg := "" | ||||||
| 	if event.Code == "CTCP_ACTION" { | 	if event.IsAction() { | ||||||
| 		msg = event.Nick + " " | 		rmsg.Event = config.EVENT_USER_ACTION | ||||||
| 	} | 	} | ||||||
| 	msg += event.Message() | 	msg += event.StripAction() | ||||||
| 	// strip IRC colors | 	// strip IRC colors | ||||||
| 	re := regexp.MustCompile(`[[:cntrl:]](\d+,|)\d+`) | 	re := regexp.MustCompile(`[[:cntrl:]](?:\d{1,2}(?:,\d{1,2})?)?`) | ||||||
| 	msg = re.ReplaceAllString(msg, "") | 	msg = re.ReplaceAllString(msg, "") | ||||||
|  |  | ||||||
|  | 	var r io.Reader | ||||||
|  | 	var err error | ||||||
|  | 	mycharset := b.Config.Charset | ||||||
|  | 	if mycharset == "" { | ||||||
| 		// detect what were sending so that we convert it to utf-8 | 		// detect what were sending so that we convert it to utf-8 | ||||||
| 		detector := chardet.NewTextDetector() | 		detector := chardet.NewTextDetector() | ||||||
| 		result, err := detector.DetectBest([]byte(msg)) | 		result, err := detector.DetectBest([]byte(msg)) | ||||||
| @@ -265,12 +356,13 @@ func (b *Birc) handlePrivMsg(event *irc.Event) { | |||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) | 		flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) | ||||||
| 	var r io.Reader | 		mycharset = result.Charset | ||||||
| 	r, err = charset.NewReader(result.Charset, strings.NewReader(msg)) |  | ||||||
| 		// if we're not sure, just pick ISO-8859-1 | 		// if we're not sure, just pick ISO-8859-1 | ||||||
| 		if result.Confidence < 80 { | 		if result.Confidence < 80 { | ||||||
| 		r, err = charset.NewReader("ISO-8859-1", strings.NewReader(msg)) | 			mycharset = "ISO-8859-1" | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  | 	r, err = charset.NewReader(mycharset, strings.NewReader(msg)) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Errorf("charset to utf-8 conversion failed: %s", err) | 		flog.Errorf("charset to utf-8 conversion failed: %s", err) | ||||||
| 		return | 		return | ||||||
| @@ -278,48 +370,35 @@ func (b *Birc) handlePrivMsg(event *irc.Event) { | |||||||
| 	output, _ := ioutil.ReadAll(r) | 	output, _ := ioutil.ReadAll(r) | ||||||
| 	msg = string(output) | 	msg = string(output) | ||||||
|  |  | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", event.Arguments[0], b.Account) | 	flog.Debugf("Sending message from %s on %s to gateway", event.Params[0], b.Account) | ||||||
| 	b.Remote <- config.Message{Username: event.Nick, Text: msg, Channel: event.Arguments[0], Account: b.Account, UserID: event.User + "@" + event.Host} | 	rmsg.Text = msg | ||||||
|  | 	b.Remote <- rmsg | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleTopicWhoTime(event *irc.Event) { | func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) { | ||||||
| 	parts := strings.Split(event.Arguments[2], "!") | 	parts := strings.Split(event.Params[2], "!") | ||||||
| 	t, err := strconv.ParseInt(event.Arguments[3], 10, 64) | 	t, err := strconv.ParseInt(event.Params[3], 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Errorf("Invalid time stamp: %s", event.Arguments[3]) | 		flog.Errorf("Invalid time stamp: %s", event.Params[3]) | ||||||
| 	} | 	} | ||||||
| 	user := parts[0] | 	user := parts[0] | ||||||
| 	if len(parts) > 1 { | 	if len(parts) > 1 { | ||||||
| 		user += " [" + parts[1] + "]" | 		user += " [" + parts[1] + "]" | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0)) | 	flog.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0)) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) nicksPerRow() int { | func (b *Birc) nicksPerRow() int { | ||||||
| 	return 4 | 	return 4 | ||||||
| 	/* |  | ||||||
| 		if b.Config.Mattermost.NicksPerRow < 1 { |  | ||||||
| 			return 4 |  | ||||||
| 		} |  | ||||||
| 		return b.Config.Mattermost.NicksPerRow |  | ||||||
| 	*/ |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) storeNames(event *irc.Event) { | func (b *Birc) storeNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[2] | 	channel := event.Params[2] | ||||||
| 	b.names[channel] = append( | 	b.names[channel] = append( | ||||||
| 		b.names[channel], | 		b.names[channel], | ||||||
| 		strings.Split(strings.TrimSpace(event.Message()), " ")...) | 		strings.Split(strings.TrimSpace(event.Trailing), " ")...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) formatnicks(nicks []string, continued bool) string { | func (b *Birc) formatnicks(nicks []string, continued bool) string { | ||||||
| 	return plainformatter(nicks, b.nicksPerRow()) | 	return plainformatter(nicks, b.nicksPerRow()) | ||||||
| 	/* |  | ||||||
| 		switch b.Config.Mattermost.NickFormatter { |  | ||||||
| 		case "table": |  | ||||||
| 			return tableformatter(nicks, b.nicksPerRow(), continued) |  | ||||||
| 		default: |  | ||||||
| 			return plainformatter(nicks, b.nicksPerRow()) |  | ||||||
| 		} |  | ||||||
| 	*/ |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,37 +1,36 @@ | |||||||
| package bmatrix | package bmatrix | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"mime" | ||||||
| 	"regexp" | 	"regexp" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	matrix "github.com/matrix-org/gomatrix" | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	matrix "github.com/matterbridge/gomatrix" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bmatrix struct { | type Bmatrix struct { | ||||||
| 	mc      *matrix.Client | 	mc      *matrix.Client | ||||||
| 	Config  *config.Protocol |  | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| 	UserID  string | 	UserID  string | ||||||
| 	RoomMap map[string]string | 	RoomMap map[string]string | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "matrix" | var protocol = "matrix" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bmatrix { | func New(cfg *config.BridgeConfig) *Bmatrix { | ||||||
| 	b := &Bmatrix{} | 	b := &Bmatrix{BridgeConfig: cfg} | ||||||
| 	b.RoomMap = make(map[string]string) | 	b.RoomMap = make(map[string]string) | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.Remote = c |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -63,23 +62,92 @@ func (b *Bmatrix) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) JoinChannel(channel string) error { | func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	resp, err := b.mc.JoinRoom(channel, "", nil) | 	resp, err := b.mc.JoinRoom(channel.Name, "", nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	b.RoomMap[resp.RoomID] = channel | 	b.RoomMap[resp.RoomID] = channel.Name | ||||||
| 	b.Unlock() | 	b.Unlock() | ||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) Send(msg config.Message) error { | func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	channel := b.getRoomID(msg.Channel) | 	channel := b.getRoomID(msg.Channel) | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{}) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return resp.EventID, err | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Sending to channel %s", channel) | 	flog.Debugf("Sending to channel %s", channel) | ||||||
| 	b.mc.SendText(channel, msg.Username+msg.Text) | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
| 	return nil | 		resp, err := b.mc.SendMessageEvent(channel, "m.room.message", | ||||||
|  | 			matrix.TextMessage{"m.emote", msg.Username + msg.Text}) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return resp.EventID, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.mc.SendText(channel, rmsg.Username+rmsg.Text) | ||||||
|  | 		} | ||||||
|  | 		// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				content := bytes.NewReader(*fi.Data) | ||||||
|  | 				sp := strings.Split(fi.Name, ".") | ||||||
|  | 				mtype := mime.TypeByExtension("." + sp[len(sp)-1]) | ||||||
|  | 				if strings.Contains(mtype, "image") || | ||||||
|  | 					strings.Contains(mtype, "video") { | ||||||
|  | 					if fi.Comment != "" { | ||||||
|  | 						_, err := b.mc.SendText(channel, msg.Username+fi.Comment) | ||||||
|  | 						if err != nil { | ||||||
|  | 							flog.Errorf("file comment failed: %#v", err) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("uploading file: %s %s", fi.Name, mtype) | ||||||
|  | 					res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) | ||||||
|  | 					if err != nil { | ||||||
|  | 						flog.Errorf("file upload failed: %#v", err) | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					if strings.Contains(mtype, "video") { | ||||||
|  | 						flog.Debugf("sendVideo %s", res.ContentURI) | ||||||
|  | 						_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) | ||||||
|  | 						if err != nil { | ||||||
|  | 							flog.Errorf("sendVideo failed: %#v", err) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 					if strings.Contains(mtype, "image") { | ||||||
|  | 						flog.Debugf("sendImage %s", res.ContentURI) | ||||||
|  | 						_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) | ||||||
|  | 						if err != nil { | ||||||
|  | 							flog.Errorf("sendImage failed: %#v", err) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("result: %#v", res) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	resp, err := b.mc.SendText(channel, msg.Username+msg.Text) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return resp.EventID, err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) getRoomID(channel string) string { | func (b *Bmatrix) getRoomID(channel string) string { | ||||||
| @@ -92,10 +160,24 @@ func (b *Bmatrix) getRoomID(channel string) string { | |||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) handlematrix() error { | func (b *Bmatrix) handlematrix() error { | ||||||
| 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||||
| 	syncer.OnEventType("m.room.message", func(ev *matrix.Event) { | 	syncer.OnEventType("m.room.redaction", b.handleEvent) | ||||||
| 		if ev.Content["msgtype"].(string) == "m.text" && ev.Sender != b.UserID { | 	syncer.OnEventType("m.room.message", b.handleEvent) | ||||||
|  | 	go func() { | ||||||
|  | 		for { | ||||||
|  | 			if err := b.mc.Sync(); err != nil { | ||||||
|  | 				flog.Println("Sync() returned ", err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||||
|  | 	flog.Debugf("Received: %#v", ev) | ||||||
|  | 	if ev.Sender != b.UserID { | ||||||
| 		b.RLock() | 		b.RLock() | ||||||
| 		channel, ok := b.RoomMap[ev.RoomID] | 		channel, ok := b.RoomMap[ev.RoomID] | ||||||
| 		b.RUnlock() | 		b.RUnlock() | ||||||
| @@ -108,17 +190,61 @@ func (b *Bmatrix) handlematrix() error { | |||||||
| 			re := regexp.MustCompile("(.*?):.*") | 			re := regexp.MustCompile("(.*?):.*") | ||||||
| 			username = re.ReplaceAllString(username, `$1`) | 			username = re.ReplaceAllString(username, `$1`) | ||||||
| 		} | 		} | ||||||
|  | 		var text string | ||||||
|  | 		text, _ = ev.Content["body"].(string) | ||||||
|  | 		rmsg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: ev.Sender} | ||||||
|  | 		rmsg.ID = ev.ID | ||||||
|  | 		if ev.Type == "m.room.redaction" { | ||||||
|  | 			rmsg.Event = config.EVENT_MSG_DELETE | ||||||
|  | 			rmsg.ID = ev.Redacts | ||||||
|  | 			rmsg.Text = config.EVENT_MSG_DELETE | ||||||
|  | 			b.Remote <- rmsg | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		if ev.Content["msgtype"].(string) == "m.emote" { | ||||||
|  | 			rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 		} | ||||||
|  | 		if ev.Content["msgtype"] != nil && ev.Content["msgtype"].(string) == "m.image" || | ||||||
|  | 			ev.Content["msgtype"].(string) == "m.video" || | ||||||
|  | 			ev.Content["msgtype"].(string) == "m.file" { | ||||||
|  | 			flog.Debugf("ev: %#v", ev) | ||||||
|  | 			rmsg.Extra = make(map[string][]interface{}) | ||||||
|  | 			url := ev.Content["url"].(string) | ||||||
|  | 			url = strings.Replace(url, "mxc://", b.Config.Server+"/_matrix/media/v1/download/", -1) | ||||||
|  | 			info := ev.Content["info"].(map[string]interface{}) | ||||||
|  | 			size := info["size"].(float64) | ||||||
|  | 			name := ev.Content["body"].(string) | ||||||
|  | 			// check if we have an image uploaded without extension | ||||||
|  | 			if !strings.Contains(name, ".") { | ||||||
|  | 				if ev.Content["msgtype"].(string) == "m.image" { | ||||||
|  | 					if mtype, ok := ev.Content["mimetype"].(string); ok { | ||||||
|  | 						mext, _ := mime.ExtensionsByType(mtype) | ||||||
|  | 						if len(mext) > 0 { | ||||||
|  | 							name = name + mext[0] | ||||||
|  | 						} | ||||||
|  | 					} else { | ||||||
|  | 						// just a default .png extension if we don't have mime info | ||||||
|  | 						name = name + ".png" | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			flog.Debugf("trying to download %#v with size %#v", name, size) | ||||||
|  | 			if size <= float64(b.General.MediaDownloadSize) { | ||||||
|  | 				data, err := helper.DownloadFile(url) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("download %s failed %#v", url, err) | ||||||
|  | 				} else { | ||||||
|  | 					flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url)) | ||||||
|  | 					rmsg.Extra["file"] = append(rmsg.Extra["file"], config.FileInfo{Name: name, Data: data}) | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, b.General.MediaDownloadSize) | ||||||
|  | 				rmsg.Event = config.EVENT_FILE_FAILURE_SIZE | ||||||
|  | 				rmsg.Extra[rmsg.Event] = append(rmsg.Extra[rmsg.Event], config.FileInfo{Name: name, Size: int64(size)}) | ||||||
|  | 			} | ||||||
|  | 			rmsg.Text = "" | ||||||
|  | 		} | ||||||
| 		flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) | 		flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) | ||||||
| 			b.Remote <- config.Message{Username: username, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account, UserID: ev.Sender} | 		b.Remote <- rmsg | ||||||
| 		} |  | ||||||
| 		flog.Debugf("Received: %#v", ev) |  | ||||||
| 	}) |  | ||||||
| 	go func() { |  | ||||||
| 		for { |  | ||||||
| 			if err := b.mc.Sync(); err != nil { |  | ||||||
| 				flog.Println("Sync() returned ", err) |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 	}() |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,10 +1,14 @@ | |||||||
| package bmattermost | package bmattermost | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/42wim/matterbridge/matterclient" | 	"github.com/42wim/matterbridge/matterclient" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type MMhook struct { | type MMhook struct { | ||||||
| @@ -14,7 +18,6 @@ type MMhook struct { | |||||||
| type MMapi struct { | type MMapi struct { | ||||||
| 	mc    *matterclient.MMClient | 	mc    *matterclient.MMClient | ||||||
| 	mmMap map[string]string | 	mmMap map[string]string | ||||||
| 	mmIgnoreNicks []string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| type MMMessage struct { | type MMMessage struct { | ||||||
| @@ -22,30 +25,28 @@ type MMMessage struct { | |||||||
| 	Channel  string | 	Channel  string | ||||||
| 	Username string | 	Username string | ||||||
| 	UserID   string | 	UserID   string | ||||||
|  | 	ID       string | ||||||
|  | 	Event    string | ||||||
|  | 	Extra    map[string][]interface{} | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bmattermost struct { | type Bmattermost struct { | ||||||
| 	MMhook | 	MMhook | ||||||
| 	MMapi | 	MMapi | ||||||
| 	Config  *config.Protocol |  | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	name    string |  | ||||||
| 	TeamId string | 	TeamId string | ||||||
| 	Account string | 	*config.BridgeConfig | ||||||
|  | 	avatarMap map[string]string | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "mattermost" | var protocol = "mattermost" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bmattermost { | func New(cfg *config.BridgeConfig) *Bmattermost { | ||||||
| 	b := &Bmattermost{} | 	b := &Bmattermost{BridgeConfig: cfg, avatarMap: make(map[string]string)} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.mmMap = make(map[string]string) | 	b.mmMap = make(map[string]string) | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -55,87 +56,208 @@ func (b *Bmattermost) Command(cmd string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Connect() error { | func (b *Bmattermost) Connect() error { | ||||||
| 	if b.Config.WebhookURL != "" && b.Config.WebhookBindAddress != "" { | 	if b.Config.WebhookBindAddress != "" { | ||||||
| 		flog.Info("Connecting using webhookurl and webhookbindaddress") | 		if b.Config.WebhookURL != "" { | ||||||
|  | 			flog.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
| 	} else if b.Config.WebhookURL != "" { | 		} else if b.Config.Token != "" { | ||||||
| 		flog.Info("Connecting using webhookurl (for posting) and token") | 			flog.Info("Connecting using token (sending)") | ||||||
| 		b.mh = matterhook.New(b.Config.WebhookURL, | 			err := b.apiLogin() | ||||||
| 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, |  | ||||||
| 				DisableServer: true}) |  | ||||||
| 	} else { |  | ||||||
| 		flog.Info("Connecting using token") |  | ||||||
| 		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 { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 		flog.Info("Connection succeeded") | 		} else if b.Config.Login != "" { | ||||||
| 		b.TeamId = b.mc.GetTeamId() | 			flog.Info("Connecting using login/password (sending)") | ||||||
| 		go b.mc.WsReceiver() | 			err := b.apiLogin() | ||||||
| 		go b.mc.StatusLoop() | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			flog.Info("Connecting using webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
|  | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
|  | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
| 		} | 		} | ||||||
| 		go b.handleMatter() | 		go b.handleMatter() | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		flog.Info("Connecting using webhookurl (sending)") | ||||||
|  | 		b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
|  | 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
|  | 				DisableServer: true}) | ||||||
|  | 		if b.Config.Token != "" { | ||||||
|  | 			flog.Info("Connecting using token (receiving)") | ||||||
|  | 			err := b.apiLogin() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			go b.handleMatter() | ||||||
|  | 		} else if b.Config.Login != "" { | ||||||
|  | 			flog.Info("Connecting using login/password (receiving)") | ||||||
|  | 			err := b.apiLogin() | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			go b.handleMatter() | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	} else if b.Config.Token != "" { | ||||||
|  | 		flog.Info("Connecting using token (sending and receiving)") | ||||||
|  | 		err := b.apiLogin() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		go b.handleMatter() | ||||||
|  | 	} else if b.Config.Login != "" { | ||||||
|  | 		flog.Info("Connecting using login/password (sending and receiving)") | ||||||
|  | 		err := b.apiLogin() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		go b.handleMatter() | ||||||
|  | 	} | ||||||
|  | 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Login == "" && b.Config.Token == "" { | ||||||
|  | 		return errors.New("No connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured.") | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Disconnect() error { | func (b *Bmattermost) Disconnect() error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) JoinChannel(channel string) error { | func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	// we can only join channels using the API | 	// we can only join channels using the API | ||||||
| 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | ||||||
| 		return b.mc.JoinChannel(b.mc.GetChannelId(channel, "")) | 		id := b.mc.GetChannelId(channel.Name, "") | ||||||
|  | 		if id == "" { | ||||||
|  | 			return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) | ||||||
|  | 		} | ||||||
|  | 		return b.mc.JoinChannel(id) | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Send(msg config.Message) error { | func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "*" + msg.Text + "*" | ||||||
|  | 	} | ||||||
| 	nick := msg.Username | 	nick := msg.Username | ||||||
| 	message := msg.Text | 	message := msg.Text | ||||||
| 	channel := msg.Channel | 	channel := msg.Channel | ||||||
|  |  | ||||||
|  | 	// map the file SHA to our user (caches the avatar) | ||||||
|  | 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||||
|  | 		fi := msg.Extra["file"][0].(config.FileInfo) | ||||||
|  | 		/* if we have a sha we have successfully uploaded the file to the media server, | ||||||
|  | 		so we can now cache the sha */ | ||||||
|  | 		if fi.SHA != "" { | ||||||
|  | 			flog.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID) | ||||||
|  | 			b.avatarMap[msg.UserID] = fi.SHA | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if b.Config.PrefixMessagesWithNick { | 	if b.Config.PrefixMessagesWithNick { | ||||||
| 		message = nick + message | 		message = nick + message | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL != "" { | 	if b.Config.WebhookURL != "" { | ||||||
|  |  | ||||||
|  | 		if msg.Extra != nil { | ||||||
|  | 			for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 				matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: channel, UserName: rmsg.Username, | ||||||
|  | 					Text: rmsg.Text, Props: make(map[string]interface{})} | ||||||
|  | 				matterMessage.Props["matterbridge"] = true | ||||||
|  | 				b.mh.Send(matterMessage) | ||||||
|  | 			} | ||||||
|  | 			if len(msg.Extra["file"]) > 0 { | ||||||
|  | 				for _, f := range msg.Extra["file"] { | ||||||
|  | 					fi := f.(config.FileInfo) | ||||||
|  | 					if fi.URL != "" { | ||||||
|  | 						message += fi.URL | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | ||||||
| 		matterMessage.IconURL = msg.Avatar | 		matterMessage.IconURL = msg.Avatar | ||||||
| 		matterMessage.Channel = channel | 		matterMessage.Channel = channel | ||||||
| 		matterMessage.UserName = nick | 		matterMessage.UserName = nick | ||||||
| 		matterMessage.Type = "" | 		matterMessage.Type = "" | ||||||
| 		matterMessage.Text = message | 		matterMessage.Text = message | ||||||
|  | 		matterMessage.Props = make(map[string]interface{}) | ||||||
|  | 		matterMessage.Props["matterbridge"] = true | ||||||
| 		err := b.mh.Send(matterMessage) | 		err := b.mh.Send(matterMessage) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			flog.Info(err) | 			flog.Info(err) | ||||||
| 			return err | 			return "", err | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message) | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
| 	return nil | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		return msg.ID, b.mc.DeleteMessage(msg.ID) | ||||||
|  | 	} | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.mc.PostMessage(b.mc.GetChannelId(channel, ""), rmsg.Username+rmsg.Text) | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var err error | ||||||
|  | 			var res, id string | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				id, err = b.mc.UploadFile(*fi.Data, b.mc.GetChannelId(channel, ""), fi.Name) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Debugf("ERROR %#v", err) | ||||||
|  | 					return "", err | ||||||
|  | 				} | ||||||
|  | 				message = fi.Comment | ||||||
|  | 				if b.Config.PrefixMessagesWithNick { | ||||||
|  | 					message = nick + fi.Comment | ||||||
|  | 				} | ||||||
|  | 				res, err = b.mc.PostMessageWithFiles(b.mc.GetChannelId(channel, ""), message, []string{id}) | ||||||
|  | 			} | ||||||
|  | 			return res, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		return b.mc.EditMessage(msg.ID, message) | ||||||
|  | 	} | ||||||
|  | 	return b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) handleMatter() { | func (b *Bmattermost) handleMatter() { | ||||||
| 	mchan := make(chan *MMMessage) | 	mchan := make(chan *MMMessage) | ||||||
| 	if b.Config.WebhookBindAddress != "" && b.Config.WebhookURL != "" { | 	if b.Config.WebhookBindAddress != "" { | ||||||
| 		flog.Debugf("Choosing webhooks based receiving") | 		flog.Debugf("Choosing webhooks based receiving") | ||||||
| 		go b.handleMatterHook(mchan) | 		go b.handleMatterHook(mchan) | ||||||
| 	} else { | 	} else { | ||||||
| 		flog.Debugf("Choosing login (api) based receiving") | 		if b.Config.Token != "" { | ||||||
|  | 			flog.Debugf("Choosing token based receiving") | ||||||
|  | 		} else { | ||||||
|  | 			flog.Debugf("Choosing login/password based receiving") | ||||||
|  | 		} | ||||||
| 		go b.handleMatterClient(mchan) | 		go b.handleMatterClient(mchan) | ||||||
| 	} | 	} | ||||||
| 	for message := range mchan { | 	for message := range mchan { | ||||||
|  | 		avatar := helper.GetAvatar(b.avatarMap, message.UserID, b.General) | ||||||
|  | 		rmsg := config.Message{Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID, ID: message.ID, Event: message.Event, Extra: message.Extra, Avatar: avatar} | ||||||
|  | 		text, ok := b.replaceAction(message.Text) | ||||||
|  | 		if ok { | ||||||
|  | 			rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 		} | ||||||
|  | 		rmsg.Text = text | ||||||
| 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | ||||||
| 		b.Remote <- config.Message{Text: message.Text, Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID} | 		flog.Debugf("Message is %#v", rmsg) | ||||||
|  | 		b.Remote <- rmsg | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -152,22 +274,68 @@ func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) { | |||||||
| 		if (message.Raw.Event == "post_edited") && b.Config.EditDisable { | 		if (message.Raw.Event == "post_edited") && b.Config.EditDisable { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||||
|  | 		if b.General.MediaServerUpload != "" { | ||||||
|  | 			b.handleDownloadAvatar(message.UserID, message.Channel) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		m := &MMMessage{Extra: make(map[string][]interface{})} | ||||||
|  |  | ||||||
|  | 		props := message.Post.Props | ||||||
|  | 		if props != nil { | ||||||
|  | 			if _, ok := props["matterbridge"].(bool); ok { | ||||||
|  | 				flog.Debugf("sent by matterbridge, ignoring") | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if _, ok := props["override_username"].(string); ok { | ||||||
|  | 				message.Username = props["override_username"].(string) | ||||||
|  | 			} | ||||||
|  | 			if _, ok := props["attachments"].([]interface{}); ok { | ||||||
|  | 				m.Extra["attachments"] = props["attachments"].([]interface{}) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 		// do not post our own messages back to irc | 		// do not post our own messages back to irc | ||||||
| 		// only listen to message from our team | 		// only listen to message from our team | ||||||
| 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited") && | 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited" || message.Raw.Event == "post_deleted") && | ||||||
| 			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId { | 			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId { | ||||||
|  | 			// if the message has reactions don't repost it (for now, until we can correlate reaction with message) | ||||||
|  | 			if message.Post.HasReactions { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
| 			flog.Debugf("Receiving from matterclient %#v", message) | 			flog.Debugf("Receiving from matterclient %#v", message) | ||||||
| 			m := &MMMessage{} |  | ||||||
| 			m.UserID = message.UserID | 			m.UserID = message.UserID | ||||||
| 			m.Username = message.Username | 			m.Username = message.Username | ||||||
| 			m.Channel = message.Channel | 			m.Channel = message.Channel | ||||||
| 			m.Text = message.Text | 			m.Text = message.Text | ||||||
|  | 			m.ID = message.Post.Id | ||||||
| 			if message.Raw.Event == "post_edited" && !b.Config.EditDisable { | 			if message.Raw.Event == "post_edited" && !b.Config.EditDisable { | ||||||
| 				m.Text = message.Text + b.Config.EditSuffix | 				m.Text = message.Text + b.Config.EditSuffix | ||||||
| 			} | 			} | ||||||
|  | 			if message.Raw.Event == "post_deleted" { | ||||||
|  | 				m.Event = config.EVENT_MSG_DELETE | ||||||
|  | 			} | ||||||
| 			if len(message.Post.FileIds) > 0 { | 			if len(message.Post.FileIds) > 0 { | ||||||
| 				for _, link := range b.mc.GetPublicLinks(message.Post.FileIds) { | 				for _, id := range message.Post.FileIds { | ||||||
| 					m.Text = m.Text + "\n" + link | 					url, _ := b.mc.Client.GetFileLink(id) | ||||||
|  | 					finfo, resp := b.mc.Client.GetFileInfo(id) | ||||||
|  | 					if resp.Error != nil { | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("trying to download %#v fileid %#v with size %#v", finfo.Name, finfo.Id, finfo.Size) | ||||||
|  | 					if int(finfo.Size) > b.General.MediaDownloadSize { | ||||||
|  | 						flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", finfo.Name, finfo.Size, b.General.MediaDownloadSize) | ||||||
|  | 						m.Event = config.EVENT_FILE_FAILURE_SIZE | ||||||
|  | 						m.Extra[m.Event] = append(m.Extra[m.Event], config.FileInfo{Name: finfo.Name, Comment: message.Text, Size: int64(finfo.Size)}) | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					data, resp := b.mc.Client.DownloadFile(id, true) | ||||||
|  | 					if resp.Error != nil { | ||||||
|  | 						flog.Errorf("download %s failed %#v", finfo.Name, resp.Error) | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					flog.Debugf("download OK %#v %#v", finfo.Name, len(data)) | ||||||
|  | 					m.Extra["file"] = append(m.Extra["file"], config.FileInfo{Name: finfo.Name, Data: &data, URL: url, Comment: message.Text}) | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 			mchan <- m | 			mchan <- m | ||||||
| @@ -187,3 +355,59 @@ func (b *Bmattermost) handleMatterHook(mchan chan *MMMessage) { | |||||||
| 		mchan <- m | 		mchan <- m | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bmattermost) apiLogin() error { | ||||||
|  | 	password := b.Config.Password | ||||||
|  | 	if b.Config.Token != "" { | ||||||
|  | 		password = "MMAUTHTOKEN=" + b.Config.Token | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.mc = matterclient.New(b.Config.Login, password, | ||||||
|  | 		b.Config.Team, b.Config.Server) | ||||||
|  | 	if b.General.Debug { | ||||||
|  | 		b.mc.SetLogLevel("debug") | ||||||
|  | 	} | ||||||
|  | 	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() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bmattermost) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | ||||||
|  | 		return strings.Replace(text, "*", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleDownloadAvatar downloads the avatar of userid from channel | ||||||
|  | // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||||
|  | // logs an error message if it fails | ||||||
|  | func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||||
|  | 	var name string | ||||||
|  | 	msg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: userid, Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})} | ||||||
|  | 	if _, ok := b.avatarMap[userid]; !ok { | ||||||
|  | 		data, resp := b.mc.Client.GetProfileImage(userid, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			flog.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error) | ||||||
|  | 		} | ||||||
|  | 		if len(data) <= b.General.MediaDownloadSize { | ||||||
|  | 			name = userid + ".png" | ||||||
|  | 			flog.Debugf("download OK %#v %#v", name, len(data)) | ||||||
|  | 			msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: &data, Avatar: true}) | ||||||
|  | 			flog.Debugf("Sending avatar download message from %#v on %s to gateway", userid, b.Account) | ||||||
|  | 			flog.Debugf("Message is %#v", msg) | ||||||
|  | 			b.Remote <- msg | ||||||
|  | 		} else { | ||||||
|  | 			flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, len(data), b.General.MediaDownloadSize) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -2,9 +2,10 @@ package brocketchat | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/42wim/matterbridge/hook/rockethook" | 	"github.com/42wim/matterbridge/hook/rockethook" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type MMhook struct { | type MMhook struct { | ||||||
| @@ -14,25 +15,18 @@ type MMhook struct { | |||||||
|  |  | ||||||
| type Brocketchat struct { | type Brocketchat struct { | ||||||
| 	MMhook | 	MMhook | ||||||
| 	Config  *config.Protocol | 	*config.BridgeConfig | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	name    string |  | ||||||
| 	Account string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "rocketchat" | var protocol = "rocketchat" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Brocketchat { | func New(cfg *config.BridgeConfig) *Brocketchat { | ||||||
| 	b := &Brocketchat{} | 	return &Brocketchat{BridgeConfig: cfg} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Command(cmd string) string { | func (b *Brocketchat) Command(cmd string) string { | ||||||
| @@ -54,12 +48,32 @@ func (b *Brocketchat) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) JoinChannel(channel string) error { | func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Send(msg config.Message) error { | func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: rmsg.Channel, UserName: rmsg.Username, | ||||||
|  | 				Text: rmsg.Text} | ||||||
|  | 			b.mh.Send(matterMessage) | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				if fi.URL != "" { | ||||||
|  | 					msg.Text += fi.URL | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | ||||||
| 	matterMessage.Channel = msg.Channel | 	matterMessage.Channel = msg.Channel | ||||||
| 	matterMessage.UserName = msg.Username | 	matterMessage.UserName = msg.Username | ||||||
| @@ -68,9 +82,9 @@ func (b *Brocketchat) Send(msg config.Message) error { | |||||||
| 	err := b.mh.Send(matterMessage) | 	err := b.mh.Send(matterMessage) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Info(err) | 		flog.Info(err) | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) handleRocketHook() { | func (b *Brocketchat) handleRocketHook() { | ||||||
|   | |||||||
| @@ -1,11 +1,17 @@ | |||||||
| package bslack | package bslack | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
| 	"github.com/nlopes/slack" | 	"github.com/nlopes/slack" | ||||||
|  | 	"html" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -22,29 +28,23 @@ type MMMessage struct { | |||||||
| type Bslack struct { | type Bslack struct { | ||||||
| 	mh       *matterhook.Client | 	mh       *matterhook.Client | ||||||
| 	sc       *slack.Client | 	sc       *slack.Client | ||||||
| 	Config   *config.Protocol |  | ||||||
| 	rtm      *slack.RTM | 	rtm      *slack.RTM | ||||||
| 	Plus     bool | 	Plus     bool | ||||||
| 	Remote   chan config.Message |  | ||||||
| 	Users    []slack.User | 	Users    []slack.User | ||||||
| 	Account  string |  | ||||||
| 	si       *slack.Info | 	si       *slack.Info | ||||||
| 	channels []slack.Channel | 	channels []slack.Channel | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "slack" | var protocol = "slack" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bslack { | func New(cfg *config.BridgeConfig) *Bslack { | ||||||
| 	b := &Bslack{} | 	return &Bslack{BridgeConfig: cfg} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Command(cmd string) string { | func (b *Bslack) Command(cmd string) string { | ||||||
| @@ -52,38 +52,68 @@ func (b *Bslack) Command(cmd string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Connect() error { | func (b *Bslack) Connect() error { | ||||||
| 	if b.Config.WebhookURL != "" && b.Config.WebhookBindAddress != "" { | 	if b.Config.WebhookBindAddress != "" { | ||||||
| 		flog.Info("Connecting using webhookurl and webhookbindaddress") | 		if b.Config.WebhookURL != "" { | ||||||
|  | 			flog.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
| 			matterhook.Config{BindAddress: b.Config.WebhookBindAddress}) | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
| 	} else if b.Config.WebhookURL != "" { | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
| 		flog.Info("Connecting using webhookurl (for posting) and token") | 		} else if b.Config.Token != "" { | ||||||
| 		b.mh = matterhook.New(b.Config.WebhookURL, | 			flog.Info("Connecting using token (sending)") | ||||||
| 			matterhook.Config{DisableServer: true}) |  | ||||||
| 	} else { |  | ||||||
| 		flog.Info("Connecting using token") |  | ||||||
| 			b.sc = slack.New(b.Config.Token) | 			b.sc = slack.New(b.Config.Token) | ||||||
| 			b.rtm = b.sc.NewRTM() | 			b.rtm = b.sc.NewRTM() | ||||||
| 			go b.rtm.ManageConnection() | 			go b.rtm.ManageConnection() | ||||||
|  | 			flog.Info("Connecting using webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
|  | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
|  | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
|  | 		} else { | ||||||
|  | 			flog.Info("Connecting using webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
|  | 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
|  | 					BindAddress: b.Config.WebhookBindAddress}) | ||||||
| 		} | 		} | ||||||
| 	flog.Info("Connection succeeded") |  | ||||||
| 		go b.handleSlack() | 		go b.handleSlack() | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		flog.Info("Connecting using webhookurl (sending)") | ||||||
|  | 		b.mh = matterhook.New(b.Config.WebhookURL, | ||||||
|  | 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | ||||||
|  | 				DisableServer: true}) | ||||||
|  | 		if b.Config.Token != "" { | ||||||
|  | 			flog.Info("Connecting using token (receiving)") | ||||||
|  | 			b.sc = slack.New(b.Config.Token) | ||||||
|  | 			b.rtm = b.sc.NewRTM() | ||||||
|  | 			go b.rtm.ManageConnection() | ||||||
|  | 			go b.handleSlack() | ||||||
|  | 		} | ||||||
|  | 	} else if b.Config.Token != "" { | ||||||
|  | 		flog.Info("Connecting using token (sending and receiving)") | ||||||
|  | 		b.sc = slack.New(b.Config.Token) | ||||||
|  | 		b.rtm = b.sc.NewRTM() | ||||||
|  | 		go b.rtm.ManageConnection() | ||||||
|  | 		go b.handleSlack() | ||||||
|  | 	} | ||||||
|  | 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Token == "" { | ||||||
|  | 		return errors.New("No connection method found. See that you have WebhookBindAddress, WebhookURL or Token configured.") | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (b *Bslack) Disconnect() error { | func (b *Bslack) Disconnect() error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) JoinChannel(channel string) error { | func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	// we can only join channels using the API | 	// we can only join channels using the API | ||||||
| 	if b.Config.WebhookURL == "" || b.Config.WebhookBindAddress == "" { | 	if b.sc != nil { | ||||||
| 		if strings.HasPrefix(b.Config.Token, "xoxb") { | 		if strings.HasPrefix(b.Config.Token, "xoxb") { | ||||||
| 			// TODO check if bot has already joined channel | 			// TODO check if bot has already joined channel | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 		_, err := b.sc.JoinChannel(channel) | 		_, err := b.sc.JoinChannel(channel.Name) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			if err.Error() != "name_taken" { | 			if err.Error() != "name_taken" { | ||||||
| 				return err | 				return err | ||||||
| @@ -93,8 +123,11 @@ func (b *Bslack) JoinChannel(channel string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Send(msg config.Message) error { | func (b *Bslack) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Event == config.EVENT_USER_ACTION { | ||||||
|  | 		msg.Text = "_" + msg.Text + "_" | ||||||
|  | 	} | ||||||
| 	nick := msg.Username | 	nick := msg.Username | ||||||
| 	message := msg.Text | 	message := msg.Text | ||||||
| 	channel := msg.Channel | 	channel := msg.Channel | ||||||
| @@ -102,6 +135,22 @@ func (b *Bslack) Send(msg config.Message) error { | |||||||
| 		message = nick + " " + message | 		message = nick + " " + message | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL != "" { | 	if b.Config.WebhookURL != "" { | ||||||
|  | 		if msg.Extra != nil { | ||||||
|  | 			for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 				matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: channel, UserName: rmsg.Username, | ||||||
|  | 					Text: rmsg.Text} | ||||||
|  | 				b.mh.Send(matterMessage) | ||||||
|  | 			} | ||||||
|  | 			if len(msg.Extra["file"]) > 0 { | ||||||
|  | 				for _, f := range msg.Extra["file"] { | ||||||
|  | 					fi := f.(config.FileInfo) | ||||||
|  | 					if fi.URL != "" { | ||||||
|  | 						message += fi.URL | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | ||||||
| 		matterMessage.Channel = channel | 		matterMessage.Channel = channel | ||||||
| 		matterMessage.UserName = nick | 		matterMessage.UserName = nick | ||||||
| @@ -110,31 +159,73 @@ func (b *Bslack) Send(msg config.Message) error { | |||||||
| 		err := b.mh.Send(matterMessage) | 		err := b.mh.Send(matterMessage) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			flog.Info(err) | 			flog.Info(err) | ||||||
| 			return err | 			return "", err | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	schannel, err := b.getChannelByName(channel) | 	schannel, err := b.getChannelByName(channel) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	np := slack.NewPostMessageParameters() | 	np := slack.NewPostMessageParameters() | ||||||
| 	if b.Config.PrefixMessagesWithNick == true { | 	if b.Config.PrefixMessagesWithNick { | ||||||
| 		np.AsUser = true | 		np.AsUser = true | ||||||
| 	} | 	} | ||||||
| 	np.Username = nick | 	np.Username = nick | ||||||
| 	np.IconURL = config.GetIconURL(&msg, b.Config) | 	np.IconURL = config.GetIconURL(&msg, &b.Config) | ||||||
| 	if msg.Avatar != "" { | 	if msg.Avatar != "" { | ||||||
| 		np.IconURL = msg.Avatar | 		np.IconURL = msg.Avatar | ||||||
| 	} | 	} | ||||||
| 	b.sc.PostMessage(schannel.ID, message, np) | 	np.Attachments = append(np.Attachments, slack.Attachment{CallbackID: "matterbridge"}) | ||||||
|  | 	np.Attachments = append(np.Attachments, b.createAttach(msg.Extra)...) | ||||||
|  |  | ||||||
| 	/* | 	// replace mentions | ||||||
| 	   newmsg := b.rtm.NewOutgoingMessage(message, schannel.ID) | 	np.LinkNames = 1 | ||||||
| 	   b.rtm.SendMessage(newmsg) |  | ||||||
| 	*/ |  | ||||||
|  |  | ||||||
| 	return nil | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		// some protocols echo deletes, but with empty ID | ||||||
|  | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		// we get a "slack <ID>", split it | ||||||
|  | 		ts := strings.Fields(msg.ID) | ||||||
|  | 		b.sc.DeleteMessage(schannel.ID, ts[1]) | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	// if we have no ID it means we're creating a new message, not updating an existing one | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		ts := strings.Fields(msg.ID) | ||||||
|  | 		b.sc.UpdateMessage(schannel.ID, ts[1], message) | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.sc.PostMessage(schannel.ID, rmsg.Username+rmsg.Text, np) | ||||||
|  | 		} | ||||||
|  | 		// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var err error | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				_, err = b.sc.UploadFile(slack.FileUploadParameters{ | ||||||
|  | 					Reader:         bytes.NewReader(*fi.Data), | ||||||
|  | 					Filename:       fi.Name, | ||||||
|  | 					Channels:       []string{schannel.ID}, | ||||||
|  | 					InitialComment: fi.Comment, | ||||||
|  | 				}) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("uploadfile %#v", err) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, id, err := b.sc.PostMessage(schannel.ID, message, np) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return "slack " + id, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) getAvatar(user string) string { | func (b *Bslack) getAvatar(user string) string { | ||||||
| @@ -175,7 +266,7 @@ func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) { | |||||||
|  |  | ||||||
| func (b *Bslack) handleSlack() { | func (b *Bslack) handleSlack() { | ||||||
| 	mchan := make(chan *MMMessage) | 	mchan := make(chan *MMMessage) | ||||||
| 	if b.Config.WebhookBindAddress != "" && b.Config.WebhookURL != "" { | 	if b.Config.WebhookBindAddress != "" { | ||||||
| 		flog.Debugf("Choosing webhooks based receiving") | 		flog.Debugf("Choosing webhooks based receiving") | ||||||
| 		go b.handleMatterHook(mchan) | 		go b.handleMatterHook(mchan) | ||||||
| 	} else { | 	} else { | ||||||
| @@ -189,47 +280,140 @@ func (b *Bslack) handleSlack() { | |||||||
| 		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name { | 		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		texts := strings.Split(message.Text, "\n") | 		if (message.Text == "" || message.Username == "") && message.Raw.SubType != "message_deleted" { | ||||||
| 		for _, text := range texts { | 			continue | ||||||
| 			text = b.replaceURL(text) |  | ||||||
| 			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), UserID: message.UserID} |  | ||||||
| 		} | 		} | ||||||
|  | 		text := message.Text | ||||||
|  | 		text = b.replaceURL(text) | ||||||
|  | 		text = html.UnescapeString(text) | ||||||
|  | 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | ||||||
|  | 		msg := config.Message{Text: text, Username: message.Username, Channel: message.Channel, Account: b.Account, Avatar: b.getAvatar(message.Username), UserID: message.UserID, ID: "slack " + message.Raw.Timestamp, Extra: make(map[string][]interface{})} | ||||||
|  | 		if message.Raw.SubType == "me_message" { | ||||||
|  | 			msg.Event = config.EVENT_USER_ACTION | ||||||
|  | 		} | ||||||
|  | 		if message.Raw.SubType == "channel_leave" || message.Raw.SubType == "channel_join" { | ||||||
|  | 			msg.Username = "system" | ||||||
|  | 			msg.Event = config.EVENT_JOIN_LEAVE | ||||||
|  | 		} | ||||||
|  | 		// edited messages have a submessage, use this timestamp | ||||||
|  | 		if message.Raw.SubMessage != nil { | ||||||
|  | 			msg.ID = "slack " + message.Raw.SubMessage.Timestamp | ||||||
|  | 		} | ||||||
|  | 		if message.Raw.SubType == "message_deleted" { | ||||||
|  | 			msg.Text = config.EVENT_MSG_DELETE | ||||||
|  | 			msg.Event = config.EVENT_MSG_DELETE | ||||||
|  | 			msg.ID = "slack " + message.Raw.DeletedTimestamp | ||||||
|  | 		} | ||||||
|  | 		if message.Raw.SubType == "channel_topic" || message.Raw.SubType == "channel_purpose" { | ||||||
|  | 			msg.Event = config.EVENT_TOPIC_CHANGE | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||||
|  | 		if message.Raw.File != nil { | ||||||
|  | 			// limit to 1MB for now | ||||||
|  | 			comment := "" | ||||||
|  | 			results := regexp.MustCompile(`.*?commented: (.*)`).FindAllStringSubmatch(msg.Text, -1) | ||||||
|  | 			if len(results) > 0 { | ||||||
|  | 				comment = results[0][1] | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if message.Raw.File.Size > b.General.MediaDownloadSize { | ||||||
|  | 				flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", message.Raw.File.Name, message.Raw.File.Size, b.General.MediaDownloadSize) | ||||||
|  | 				msg.Event = config.EVENT_FILE_FAILURE_SIZE | ||||||
|  | 				msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: message.Raw.File.Name, Comment: comment, Size: int64(message.Raw.File.Size)}) | ||||||
|  | 			} else { | ||||||
|  | 				data, err := b.downloadFile(message.Raw.File.URLPrivateDownload) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("download %s failed %#v", message.Raw.File.URLPrivateDownload, err) | ||||||
|  | 				} else { | ||||||
|  | 					msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: message.Raw.File.Name, Data: data, Comment: comment}) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("Message is %#v", msg) | ||||||
|  | 		b.Remote <- msg | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | ||||||
| 	count := 0 |  | ||||||
| 	for msg := range b.rtm.IncomingEvents { | 	for msg := range b.rtm.IncomingEvents { | ||||||
|  | 		if msg.Type != "user_typing" && msg.Type != "latency_report" { | ||||||
|  | 			flog.Debugf("Receiving from slackclient %#v", msg.Data) | ||||||
|  | 		} | ||||||
| 		switch ev := msg.Data.(type) { | 		switch ev := msg.Data.(type) { | ||||||
| 		case *slack.MessageEvent: | 		case *slack.MessageEvent: | ||||||
| 			// ignore first message | 			if ev.SubType == "pinned_item" || ev.SubType == "unpinned_item" { | ||||||
| 			if count > 0 { | 				continue | ||||||
| 				flog.Debugf("Receiving from slackclient %#v", ev) | 			} | ||||||
| 				if !b.Config.EditDisable && ev.SubMessage != nil { | 			if len(ev.Attachments) > 0 { | ||||||
|  | 				// skip messages we made ourselves | ||||||
|  | 				if ev.Attachments[0].CallbackID == "matterbridge" { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if !b.Config.EditDisable && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { | ||||||
| 				flog.Debugf("SubMessage %#v", ev.SubMessage) | 				flog.Debugf("SubMessage %#v", ev.SubMessage) | ||||||
| 				ev.User = ev.SubMessage.User | 				ev.User = ev.SubMessage.User | ||||||
| 				ev.Text = ev.SubMessage.Text + b.Config.EditSuffix | 				ev.Text = ev.SubMessage.Text + b.Config.EditSuffix | ||||||
|  |  | ||||||
|  | 				// it seems ev.SubMessage.Edited == nil when slack unfurls | ||||||
|  | 				// do not forward these messages #266 | ||||||
|  | 				if ev.SubMessage.Edited == nil { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
| 			} | 			} | ||||||
| 			// use our own func because rtm.GetChannelInfo doesn't work for private channels | 			// use our own func because rtm.GetChannelInfo doesn't work for private channels | ||||||
| 			channel, err := b.getChannelByID(ev.Channel) | 			channel, err := b.getChannelByID(ev.Channel) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
|  | 			m := &MMMessage{} | ||||||
|  | 			if ev.BotID == "" && ev.SubType != "message_deleted" && ev.SubType != "file_comment" { | ||||||
| 				user, err := b.rtm.GetUserInfo(ev.User) | 				user, err := b.rtm.GetUserInfo(ev.User) | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| 					continue | 					continue | ||||||
| 				} | 				} | ||||||
| 				m := &MMMessage{} |  | ||||||
| 				m.UserID = user.ID | 				m.UserID = user.ID | ||||||
| 				m.Username = user.Name | 				m.Username = user.Name | ||||||
|  | 				if user.Profile.DisplayName != "" { | ||||||
|  | 					m.Username = user.Profile.DisplayName | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
| 			m.Channel = channel.Name | 			m.Channel = channel.Name | ||||||
| 			m.Text = ev.Text | 			m.Text = ev.Text | ||||||
|  | 			if m.Text == "" { | ||||||
|  | 				for _, attach := range ev.Attachments { | ||||||
|  | 					if attach.Text != "" { | ||||||
|  | 						m.Text = attach.Text | ||||||
|  | 					} else { | ||||||
|  | 						m.Text = attach.Fallback | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
| 			m.Raw = ev | 			m.Raw = ev | ||||||
| 			m.Text = b.replaceMention(m.Text) | 			m.Text = b.replaceMention(m.Text) | ||||||
| 				mchan <- m | 			m.Text = b.replaceVariable(m.Text) | ||||||
|  | 			m.Text = b.replaceChannel(m.Text) | ||||||
|  | 			// when using webhookURL we can't check if it's our webhook or not for now | ||||||
|  | 			if ev.BotID != "" && b.Config.WebhookURL == "" { | ||||||
|  | 				bot, err := b.rtm.GetBotInfo(ev.BotID) | ||||||
|  | 				if err != nil { | ||||||
|  | 					continue | ||||||
| 				} | 				} | ||||||
| 			count++ | 				if bot.Name != "" { | ||||||
|  | 					m.Username = bot.Name | ||||||
|  | 					if ev.Username != "" { | ||||||
|  | 						m.Username = ev.Username | ||||||
|  | 					} | ||||||
|  | 					m.UserID = bot.ID | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if ev.SubType == "file_comment" { | ||||||
|  | 				m.Username = "system" | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			mchan <- m | ||||||
| 		case *slack.OutgoingErrorEvent: | 		case *slack.OutgoingErrorEvent: | ||||||
| 			flog.Debugf("%#v", ev.Error()) | 			flog.Debugf("%#v", ev.Error()) | ||||||
| 		case *slack.ChannelJoinedEvent: | 		case *slack.ChannelJoinedEvent: | ||||||
| @@ -248,6 +432,8 @@ func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | |||||||
| 			} | 			} | ||||||
| 		case *slack.InvalidAuthEvent: | 		case *slack.InvalidAuthEvent: | ||||||
| 			flog.Fatalf("Invalid Token %#v", ev) | 			flog.Fatalf("Invalid Token %#v", ev) | ||||||
|  | 		case *slack.ConnectionErrorEvent: | ||||||
|  | 			flog.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj) | ||||||
| 		default: | 		default: | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -261,6 +447,8 @@ func (b *Bslack) handleMatterHook(mchan chan *MMMessage) { | |||||||
| 		m.Username = message.UserName | 		m.Username = message.UserName | ||||||
| 		m.Text = message.Text | 		m.Text = message.Text | ||||||
| 		m.Text = b.replaceMention(m.Text) | 		m.Text = b.replaceMention(m.Text) | ||||||
|  | 		m.Text = b.replaceVariable(m.Text) | ||||||
|  | 		m.Text = b.replaceChannel(m.Text) | ||||||
| 		m.Channel = message.ChannelName | 		m.Channel = message.ChannelName | ||||||
| 		if m.Username == "slackbot" { | 		if m.Username == "slackbot" { | ||||||
| 			continue | 			continue | ||||||
| @@ -272,25 +460,91 @@ func (b *Bslack) handleMatterHook(mchan chan *MMMessage) { | |||||||
| func (b *Bslack) userName(id string) string { | func (b *Bslack) userName(id string) string { | ||||||
| 	for _, u := range b.Users { | 	for _, u := range b.Users { | ||||||
| 		if u.ID == id { | 		if u.ID == id { | ||||||
|  | 			if u.Profile.DisplayName != "" { | ||||||
|  | 				return u.Profile.DisplayName | ||||||
|  | 			} | ||||||
| 			return u.Name | 			return u.Name | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
| func (b *Bslack) replaceMention(text string) string { | func (b *Bslack) replaceMention(text string) string { | ||||||
| 	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1) | 	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1) | ||||||
| 	for _, r := range results { | 	for _, r := range results { | ||||||
| 		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1) | 		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1) | ||||||
|  |  | ||||||
| 	} | 	} | ||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
|  | func (b *Bslack) replaceChannel(text string) string { | ||||||
|  | 	results := regexp.MustCompile(`<#[a-zA-Z0-9]+\|(.+?)>`).FindAllStringSubmatch(text, -1) | ||||||
|  | 	for _, r := range results { | ||||||
|  | 		text = strings.Replace(text, r[0], "#"+r[1], -1) | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#variables | ||||||
|  | func (b *Bslack) replaceVariable(text string) string { | ||||||
|  | 	results := regexp.MustCompile(`<!([a-zA-Z0-9]+)(\|.+?)?>`).FindAllStringSubmatch(text, -1) | ||||||
|  | 	for _, r := range results { | ||||||
|  | 		text = strings.Replace(text, r[0], "@"+r[1], -1) | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_urls | ||||||
| func (b *Bslack) replaceURL(text string) string { | func (b *Bslack) replaceURL(text string) string { | ||||||
| 	results := regexp.MustCompile(`<(.*?)\|.*?>`).FindAllStringSubmatch(text, -1) | 	results := regexp.MustCompile(`<(.*?)(\|.*?)?>`).FindAllStringSubmatch(text, -1) | ||||||
| 	for _, r := range results { | 	for _, r := range results { | ||||||
| 		text = strings.Replace(text, r[0], r[1], -1) | 		text = strings.Replace(text, r[0], r[1], -1) | ||||||
| 	} | 	} | ||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) createAttach(extra map[string][]interface{}) []slack.Attachment { | ||||||
|  | 	var attachs []slack.Attachment | ||||||
|  | 	for _, v := range extra["attachments"] { | ||||||
|  | 		entry := v.(map[string]interface{}) | ||||||
|  | 		s := slack.Attachment{} | ||||||
|  | 		s.Fallback = entry["fallback"].(string) | ||||||
|  | 		s.Color = entry["color"].(string) | ||||||
|  | 		s.Pretext = entry["pretext"].(string) | ||||||
|  | 		s.AuthorName = entry["author_name"].(string) | ||||||
|  | 		s.AuthorLink = entry["author_link"].(string) | ||||||
|  | 		s.AuthorIcon = entry["author_icon"].(string) | ||||||
|  | 		s.Title = entry["title"].(string) | ||||||
|  | 		s.TitleLink = entry["title_link"].(string) | ||||||
|  | 		s.Text = entry["text"].(string) | ||||||
|  | 		s.ImageURL = entry["image_url"].(string) | ||||||
|  | 		s.ThumbURL = entry["thumb_url"].(string) | ||||||
|  | 		s.Footer = entry["footer"].(string) | ||||||
|  | 		s.FooterIcon = entry["footer_icon"].(string) | ||||||
|  | 		attachs = append(attachs, s) | ||||||
|  | 	} | ||||||
|  | 	return attachs | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) downloadFile(url string) (*[]byte, error) { | ||||||
|  | 	var buf bytes.Buffer | ||||||
|  | 	client := &http.Client{ | ||||||
|  | 		Timeout: time.Second * 5, | ||||||
|  | 	} | ||||||
|  | 	req, err := http.NewRequest("GET", url, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	req.Header.Add("Authorization", "Bearer "+b.Config.Token) | ||||||
|  | 	resp, err := client.Do(req) | ||||||
|  | 	if err != nil { | ||||||
|  | 		resp.Body.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	io.Copy(&buf, resp.Body) | ||||||
|  | 	data := buf.Bytes() | ||||||
|  | 	resp.Body.Close() | ||||||
|  | 	return &data, nil | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										139
									
								
								bridge/sshchat/sshchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										139
									
								
								bridge/sshchat/sshchat.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,139 @@ | |||||||
|  | package bsshchat | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bufio" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	"github.com/shazow/ssh-chat/sshd" | ||||||
|  | 	"io" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Bsshchat struct { | ||||||
|  | 	r *bufio.Scanner | ||||||
|  | 	w io.WriteCloser | ||||||
|  | 	*config.BridgeConfig | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var flog *log.Entry | ||||||
|  | var protocol = "sshchat" | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg *config.BridgeConfig) *Bsshchat { | ||||||
|  | 	return &Bsshchat{BridgeConfig: cfg} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bsshchat) Connect() error { | ||||||
|  | 	var err error | ||||||
|  | 	flog.Infof("Connecting %s", b.Config.Server) | ||||||
|  | 	go func() { | ||||||
|  | 		err = sshd.ConnectShell(b.Config.Server, b.Config.Nick, func(r io.Reader, w io.WriteCloser) error { | ||||||
|  | 			b.r = bufio.NewScanner(r) | ||||||
|  | 			b.w = w | ||||||
|  | 			b.r.Scan() | ||||||
|  | 			w.Write([]byte("/theme mono\r\n")) | ||||||
|  | 			b.handleSshChat() | ||||||
|  | 			return nil | ||||||
|  | 		}) | ||||||
|  | 	}() | ||||||
|  | 	if err != nil { | ||||||
|  | 		flog.Debugf("%#v", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	flog.Info("Connection succeeded") | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bsshchat) Disconnect() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bsshchat) JoinChannel(channel config.ChannelInfo) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bsshchat) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.w.Write([]byte(rmsg.Username + rmsg.Text + "\r\n")) | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				if fi.Comment != "" { | ||||||
|  | 					msg.Text += fi.Comment + ": " | ||||||
|  | 				} | ||||||
|  | 				if fi.URL != "" { | ||||||
|  | 					msg.Text = fi.URL | ||||||
|  | 				} | ||||||
|  | 				b.w.Write([]byte(msg.Username + msg.Text)) | ||||||
|  | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	b.w.Write([]byte(msg.Username + msg.Text + "\r\n")) | ||||||
|  | 	return "", nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | /* | ||||||
|  | func (b *Bsshchat) sshchatKeepAlive() chan bool { | ||||||
|  | 	done := make(chan bool) | ||||||
|  | 	go func() { | ||||||
|  | 		ticker := time.NewTicker(90 * time.Second) | ||||||
|  | 		defer ticker.Stop() | ||||||
|  | 		for { | ||||||
|  | 			select { | ||||||
|  | 			case <-ticker.C: | ||||||
|  | 				flog.Debugf("PING") | ||||||
|  | 				err := b.xc.PingC2S("", "") | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Debugf("PING failed %#v", err) | ||||||
|  | 				} | ||||||
|  | 			case <-done: | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	return done | ||||||
|  | } | ||||||
|  | */ | ||||||
|  |  | ||||||
|  | func stripPrompt(s string) string { | ||||||
|  | 	pos := strings.LastIndex(s, "\033[K") | ||||||
|  | 	if pos < 0 { | ||||||
|  | 		return s | ||||||
|  | 	} | ||||||
|  | 	return s[pos+3:] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bsshchat) handleSshChat() error { | ||||||
|  | 	/* | ||||||
|  | 		done := b.sshchatKeepAlive() | ||||||
|  | 		defer close(done) | ||||||
|  | 	*/ | ||||||
|  | 	wait := true | ||||||
|  | 	for { | ||||||
|  | 		if b.r.Scan() { | ||||||
|  | 			res := strings.Split(stripPrompt(b.r.Text()), ":") | ||||||
|  | 			if res[0] == "-> Set theme" { | ||||||
|  | 				wait = false | ||||||
|  | 				log.Debugf("mono found, allowing") | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if !wait { | ||||||
|  | 				flog.Debugf("message %#v", res) | ||||||
|  | 				rmsg := config.Message{Username: res[0], Text: strings.Join(res[1:], ":"), Channel: "sshchat", Account: b.Account, UserID: "nick"} | ||||||
|  | 				b.Remote <- rmsg | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -6,7 +6,7 @@ import ( | |||||||
| 	"github.com/Philipp15b/go-steam" | 	"github.com/Philipp15b/go-steam" | ||||||
| 	"github.com/Philipp15b/go-steam/protocol/steamlang" | 	"github.com/Philipp15b/go-steam/protocol/steamlang" | ||||||
| 	"github.com/Philipp15b/go-steam/steamid" | 	"github.com/Philipp15b/go-steam/steamid" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
| 	//"io/ioutil" | 	//"io/ioutil" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"sync" | 	"sync" | ||||||
| @@ -16,25 +16,20 @@ import ( | |||||||
| type Bsteam struct { | type Bsteam struct { | ||||||
| 	c         *steam.Client | 	c         *steam.Client | ||||||
| 	connected chan struct{} | 	connected chan struct{} | ||||||
| 	Config    *config.Protocol |  | ||||||
| 	Remote    chan config.Message |  | ||||||
| 	Account   string |  | ||||||
| 	userMap   map[steamid.SteamId]string | 	userMap   map[steamid.SteamId]string | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*config.BridgeConfig | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "steam" | var protocol = "steam" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bsteam { | func New(cfg *config.BridgeConfig) *Bsteam { | ||||||
| 	b := &Bsteam{} | 	b := &Bsteam{BridgeConfig: cfg} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.userMap = make(map[steamid.SteamId]string) | 	b.userMap = make(map[steamid.SteamId]string) | ||||||
| 	b.connected = make(chan struct{}) | 	b.connected = make(chan struct{}) | ||||||
| 	return b | 	return b | ||||||
| @@ -60,8 +55,8 @@ func (b *Bsteam) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) JoinChannel(channel string) error { | func (b *Bsteam) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	id, err := steamid.NewId(channel) | 	id, err := steamid.NewId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -69,13 +64,17 @@ func (b *Bsteam) JoinChannel(channel string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) Send(msg config.Message) error { | func (b *Bsteam) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	id, err := steamid.NewId(msg.Channel) | 	id, err := steamid.NewId(msg.Channel) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) getNick(id steamid.SteamId) string { | func (b *Bsteam) getNick(id steamid.SteamId) string { | ||||||
| @@ -101,8 +100,13 @@ func (b *Bsteam) handleEvents() { | |||||||
| 		case *steam.ChatMsgEvent: | 		case *steam.ChatMsgEvent: | ||||||
| 			flog.Debugf("Receiving ChatMsgEvent: %#v", e) | 			flog.Debugf("Receiving ChatMsgEvent: %#v", e) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | 			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | ||||||
|  | 			var channel int64 | ||||||
|  | 			if e.ChatRoomId == 0 { | ||||||
|  | 				channel = int64(e.ChatterId) | ||||||
|  | 			} else { | ||||||
| 				// for some reason we have to remove 0x18000000000000 | 				// for some reason we have to remove 0x18000000000000 | ||||||
| 			channel := int64(e.ChatRoomId) - 0x18000000000000 | 				channel = int64(e.ChatRoomId) - 0x18000000000000 | ||||||
|  | 			} | ||||||
| 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | ||||||
| 			b.Remote <- msg | 			b.Remote <- msg | ||||||
| 		case *steam.PersonaStateEvent: | 		case *steam.PersonaStateEvent: | ||||||
| @@ -136,7 +140,7 @@ func (b *Bsteam) handleEvents() { | |||||||
| 					myLoginInfo.AuthCode = code | 					myLoginInfo.AuthCode = code | ||||||
| 				} | 				} | ||||||
| 			default: | 			default: | ||||||
| 				log.Errorf("LogOnFailedEvent: ", e.Result) | 				log.Errorf("LogOnFailedEvent: %#v ", e.Result) | ||||||
| 				// TODO: Handle EResult_InvalidLoginAuthCode | 				// TODO: Handle EResult_InvalidLoginAuthCode | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
|   | |||||||
| @@ -1,33 +1,31 @@ | |||||||
| package btelegram | package btelegram | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"regexp" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Btelegram struct { | type Btelegram struct { | ||||||
| 	c *tgbotapi.BotAPI | 	c *tgbotapi.BotAPI | ||||||
| 	Config  *config.Protocol | 	*config.BridgeConfig | ||||||
| 	Remote  chan config.Message | 	avatarMap map[string]string // keep cache of userid and avatar sha | ||||||
| 	Account string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "telegram" | var protocol = "telegram" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Btelegram { | func New(cfg *config.BridgeConfig) *Btelegram { | ||||||
| 	b := &Btelegram{} | 	return &Btelegram{BridgeConfig: cfg, avatarMap: make(map[string]string)} | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Connect() error { | func (b *Btelegram) Connect() error { | ||||||
| @@ -38,7 +36,9 @@ func (b *Btelegram) Connect() error { | |||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	updates, err := b.c.GetUpdatesChan(tgbotapi.NewUpdate(0)) | 	u := tgbotapi.NewUpdate(0) | ||||||
|  | 	u.Timeout = 60 | ||||||
|  | 	updates, err := b.c.GetUpdatesChan(u) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		flog.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| @@ -53,34 +53,111 @@ func (b *Btelegram) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) JoinChannel(channel string) error { | func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Send(msg config.Message) error { | func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
| 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// map the file SHA to our user (caches the avatar) | ||||||
|  | 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||||
|  | 		fi := msg.Extra["file"][0].(config.FileInfo) | ||||||
|  | 		/* if we have a sha we have successfully uploaded the file to the media server, | ||||||
|  | 		so we can now cache the sha */ | ||||||
|  | 		if fi.SHA != "" { | ||||||
|  | 			flog.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID) | ||||||
|  | 			b.avatarMap[msg.UserID] = fi.SHA | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if b.Config.MessageFormat == "HTML" { | 	if b.Config.MessageFormat == "HTML" { | ||||||
| 		msg.Text = makeHTML(msg.Text) | 		msg.Text = makeHTML(msg.Text) | ||||||
| 	} | 	} | ||||||
| 	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text) |  | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		if msg.ID == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 		msgid, err := strconv.Atoi(msg.ID) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		_, err = b.c.DeleteMessage(tgbotapi.DeleteMessageConfig{ChatID: chatid, MessageID: msgid}) | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// edit the message if we have a msg ID | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		msgid, err := strconv.Atoi(msg.ID) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text) | ||||||
| 		if b.Config.MessageFormat == "HTML" { | 		if b.Config.MessageFormat == "HTML" { | ||||||
|  | 			flog.Debug("Using mode HTML") | ||||||
| 			m.ParseMode = tgbotapi.ModeHTML | 			m.ParseMode = tgbotapi.ModeHTML | ||||||
| 		} | 		} | ||||||
|  | 		if b.Config.MessageFormat == "Markdown" { | ||||||
|  | 			flog.Debug("Using mode markdown") | ||||||
|  | 			m.ParseMode = tgbotapi.ModeMarkdown | ||||||
|  | 		} | ||||||
| 		_, err = b.c.Send(m) | 		_, err = b.c.Send(m) | ||||||
| 	return err | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.sendMessage(chatid, rmsg.Username+rmsg.Text) | ||||||
|  | 		} | ||||||
|  | 		// check if we have files to upload (from slack, telegram or mattermost) | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			var c tgbotapi.Chattable | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				file := tgbotapi.FileBytes{fi.Name, *fi.Data} | ||||||
|  | 				re := regexp.MustCompile(".(jpg|png)$") | ||||||
|  | 				if re.MatchString(fi.Name) { | ||||||
|  | 					c = tgbotapi.NewPhotoUpload(chatid, file) | ||||||
|  | 				} else { | ||||||
|  | 					c = tgbotapi.NewDocumentUpload(chatid, file) | ||||||
|  | 				} | ||||||
|  | 				_, err := b.c.Send(c) | ||||||
|  | 				if err != nil { | ||||||
|  | 					log.Errorf("file upload failed: %#v", err) | ||||||
|  | 				} | ||||||
|  | 				if fi.Comment != "" { | ||||||
|  | 					b.sendMessage(chatid, msg.Username+fi.Comment) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return b.sendMessage(chatid, msg.Username+msg.Text) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||||
| 	for update := range updates { | 	for update := range updates { | ||||||
|  | 		flog.Debugf("Receiving from telegram: %#v", update.Message) | ||||||
|  | 		if update.Message == nil { | ||||||
|  | 			flog.Error("Getting nil messages, this shouldn't happen.") | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
| 		var message *tgbotapi.Message | 		var message *tgbotapi.Message | ||||||
| 		username := "" | 		username := "" | ||||||
| 		channel := "" | 		channel := "" | ||||||
| 		text := "" | 		text := "" | ||||||
|  |  | ||||||
|  | 		fmsg := config.Message{Extra: make(map[string][]interface{})} | ||||||
|  |  | ||||||
| 		// handle channels | 		// handle channels | ||||||
| 		if update.ChannelPost != nil { | 		if update.ChannelPost != nil { | ||||||
| 			message = update.ChannelPost | 			message = update.ChannelPost | ||||||
| @@ -109,28 +186,82 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | |||||||
| 			} | 			} | ||||||
| 			text = message.Text | 			text = message.Text | ||||||
| 			channel = strconv.FormatInt(message.Chat.ID, 10) | 			channel = strconv.FormatInt(message.Chat.ID, 10) | ||||||
|  | 			// only download avatars if we have a place to upload them (configured mediaserver) | ||||||
|  | 			if b.General.MediaServerUpload != "" { | ||||||
|  | 				b.handleDownloadAvatar(message.From.ID, channel) | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if username == "" { | 		if username == "" { | ||||||
| 			username = "unknown" | 			username = "unknown" | ||||||
| 		} | 		} | ||||||
| 		if message.Sticker != nil && b.Config.UseInsecureURL { | 		if message.Sticker != nil { | ||||||
| 			text = text + " " + b.getFileDirectURL(message.Sticker.FileID) | 			b.handleDownload(message.Sticker, message.Caption, &fmsg) | ||||||
| 		} | 		} | ||||||
| 		if message.Video != nil && b.Config.UseInsecureURL { | 		if message.Video != nil { | ||||||
| 			text = text + " " + b.getFileDirectURL(message.Video.FileID) | 			b.handleDownload(message.Video, message.Caption, &fmsg) | ||||||
| 		} | 		} | ||||||
| 		if message.Photo != nil && b.Config.UseInsecureURL { | 		if message.Photo != nil { | ||||||
| 			photos := *message.Photo | 			b.handleDownload(message.Photo, message.Caption, &fmsg) | ||||||
| 			// last photo is the biggest |  | ||||||
| 			text = text + " " + b.getFileDirectURL(photos[len(photos)-1].FileID) |  | ||||||
| 		} | 		} | ||||||
| 		if message.Document != nil && b.Config.UseInsecureURL { | 		if message.Document != nil { | ||||||
| 			text = text + " " + message.Document.FileName + " : " + b.getFileDirectURL(message.Document.FileID) | 			b.handleDownload(message.Document, message.Caption, &fmsg) | ||||||
| 		} | 		} | ||||||
| 		if text != "" { | 		if message.Voice != nil { | ||||||
|  | 			b.handleDownload(message.Voice, message.Caption, &fmsg) | ||||||
|  | 		} | ||||||
|  | 		if message.Audio != nil { | ||||||
|  | 			b.handleDownload(message.Audio, message.Caption, &fmsg) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// If UseInsecureURL is used we'll have a text in fmsg.Text | ||||||
|  | 		if fmsg.Text != "" { | ||||||
|  | 			text = text + fmsg.Text | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if message.ForwardFrom != nil { | ||||||
|  | 			usernameForward := "" | ||||||
|  | 			if b.Config.UseFirstName { | ||||||
|  | 				usernameForward = message.ForwardFrom.FirstName | ||||||
|  | 			} | ||||||
|  | 			if usernameForward == "" { | ||||||
|  | 				usernameForward = message.ForwardFrom.UserName | ||||||
|  | 				if usernameForward == "" { | ||||||
|  | 					usernameForward = message.ForwardFrom.FirstName | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if usernameForward == "" { | ||||||
|  | 				usernameForward = "unknown" | ||||||
|  | 			} | ||||||
|  | 			text = "Forwarded from " + usernameForward + ": " + text | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// quote the previous message | ||||||
|  | 		if message.ReplyToMessage != nil { | ||||||
|  | 			usernameReply := "" | ||||||
|  | 			if message.ReplyToMessage.From != nil { | ||||||
|  | 				if b.Config.UseFirstName { | ||||||
|  | 					usernameReply = message.ReplyToMessage.From.FirstName | ||||||
|  | 				} | ||||||
|  | 				if usernameReply == "" { | ||||||
|  | 					usernameReply = message.ReplyToMessage.From.UserName | ||||||
|  | 					if usernameReply == "" { | ||||||
|  | 						usernameReply = message.ReplyToMessage.From.FirstName | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if usernameReply == "" { | ||||||
|  | 				usernameReply = "unknown" | ||||||
|  | 			} | ||||||
|  | 			text = text + " (re @" + usernameReply + ":" + message.ReplyToMessage.Text + ")" | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if text != "" || len(fmsg.Extra) > 0 { | ||||||
|  | 			avatar := helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account) | 			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account) | ||||||
| 			b.Remote <- config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID)} | 			msg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID), ID: strconv.Itoa(message.MessageID), Extra: fmsg.Extra, Avatar: avatar} | ||||||
|  | 			flog.Debugf("Message is %#v", msg) | ||||||
|  | 			b.Remote <- msg | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -142,3 +273,130 @@ func (b *Btelegram) getFileDirectURL(id string) string { | |||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // handleDownloadAvatar downloads the avatar of userid from channel | ||||||
|  | // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||||
|  | // logs an error message if it fails | ||||||
|  | func (b *Btelegram) handleDownloadAvatar(userid int, channel string) { | ||||||
|  | 	msg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: strconv.Itoa(userid), Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})} | ||||||
|  | 	if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok { | ||||||
|  | 		photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1}) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("Userprofile download failed for %#v %s", userid, err) | ||||||
|  | 		} | ||||||
|  | 		if len(photos.Photos) > 0 { | ||||||
|  | 			photo := photos.Photos[0][0] | ||||||
|  | 			url := b.getFileDirectURL(photo.FileID) | ||||||
|  | 			name := strconv.Itoa(userid) + ".png" | ||||||
|  | 			flog.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize) | ||||||
|  | 			if photo.FileSize <= b.General.MediaDownloadSize { | ||||||
|  | 				data, err := helper.DownloadFile(url) | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Errorf("download %s failed %#v", url, err) | ||||||
|  | 				} else { | ||||||
|  | 					flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url)) | ||||||
|  | 					msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, Avatar: true}) | ||||||
|  | 					flog.Debugf("Sending avatar download message from %#v on %s to gateway", userid, b.Account) | ||||||
|  | 					flog.Debugf("Message is %#v", msg) | ||||||
|  | 					b.Remote <- msg | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, photo.FileSize, b.General.MediaDownloadSize) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Btelegram) handleDownload(file interface{}, comment string, msg *config.Message) { | ||||||
|  | 	size := 0 | ||||||
|  | 	url := "" | ||||||
|  | 	name := "" | ||||||
|  | 	text := "" | ||||||
|  | 	fileid := "" | ||||||
|  | 	switch v := file.(type) { | ||||||
|  | 	case *tgbotapi.Audio: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		urlPart := strings.Split(url, "/") | ||||||
|  | 		name = urlPart[len(urlPart)-1] | ||||||
|  | 		text = " " + url | ||||||
|  | 		fileid = v.FileID | ||||||
|  | 	case *tgbotapi.Voice: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		urlPart := strings.Split(url, "/") | ||||||
|  | 		name = urlPart[len(urlPart)-1] | ||||||
|  | 		text = " " + url | ||||||
|  | 		if !strings.HasSuffix(name, ".ogg") { | ||||||
|  | 			name = name + ".ogg" | ||||||
|  | 		} | ||||||
|  | 		fileid = v.FileID | ||||||
|  | 	case *tgbotapi.Sticker: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		urlPart := strings.Split(url, "/") | ||||||
|  | 		name = urlPart[len(urlPart)-1] | ||||||
|  | 		if !strings.HasSuffix(name, ".webp") { | ||||||
|  | 			name = name + ".webp" | ||||||
|  | 		} | ||||||
|  | 		text = " " + url | ||||||
|  | 		fileid = v.FileID | ||||||
|  | 	case *tgbotapi.Video: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		urlPart := strings.Split(url, "/") | ||||||
|  | 		name = urlPart[len(urlPart)-1] | ||||||
|  | 		text = " " + url | ||||||
|  | 		fileid = v.FileID | ||||||
|  | 	case *[]tgbotapi.PhotoSize: | ||||||
|  | 		photos := *v | ||||||
|  | 		size = photos[len(photos)-1].FileSize | ||||||
|  | 		url = b.getFileDirectURL(photos[len(photos)-1].FileID) | ||||||
|  | 		urlPart := strings.Split(url, "/") | ||||||
|  | 		name = urlPart[len(urlPart)-1] | ||||||
|  | 		text = " " + url | ||||||
|  | 	case *tgbotapi.Document: | ||||||
|  | 		size = v.FileSize | ||||||
|  | 		url = b.getFileDirectURL(v.FileID) | ||||||
|  | 		name = v.FileName | ||||||
|  | 		text = " " + v.FileName + " : " + url | ||||||
|  | 		fileid = v.FileID | ||||||
|  | 	} | ||||||
|  | 	if b.Config.UseInsecureURL { | ||||||
|  | 		flog.Debugf("Setting message text to :%s", text) | ||||||
|  | 		msg.Text = text | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||||
|  | 	flog.Debugf("trying to download %#v fileid %#v with size %#v", name, fileid, size) | ||||||
|  | 	if size <= b.General.MediaDownloadSize { | ||||||
|  | 		data, err := helper.DownloadFile(url) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("download %s failed %#v", url, err) | ||||||
|  | 		} else { | ||||||
|  | 			flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url)) | ||||||
|  | 			msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, Comment: comment}) | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, b.General.MediaDownloadSize) | ||||||
|  | 		msg.Event = config.EVENT_FILE_FAILURE_SIZE | ||||||
|  | 		msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: name, Comment: comment, Size: int64(size)}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Btelegram) sendMessage(chatid int64, text string) (string, error) { | ||||||
|  | 	m := tgbotapi.NewMessage(chatid, text) | ||||||
|  | 	if b.Config.MessageFormat == "HTML" { | ||||||
|  | 		flog.Debug("Using mode HTML") | ||||||
|  | 		m.ParseMode = tgbotapi.ModeHTML | ||||||
|  | 	} | ||||||
|  | 	if b.Config.MessageFormat == "Markdown" { | ||||||
|  | 		flog.Debug("Using mode markdown") | ||||||
|  | 		m.ParseMode = tgbotapi.ModeMarkdown | ||||||
|  | 	} | ||||||
|  | 	res, err := b.c.Send(m) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return strconv.Itoa(res.MessageID), nil | ||||||
|  | } | ||||||
|   | |||||||
| @@ -3,7 +3,9 @@ package bxmpp | |||||||
| import ( | import ( | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	"github.com/jpillora/backoff" | ||||||
| 	"github.com/mattn/go-xmpp" | 	"github.com/mattn/go-xmpp" | ||||||
|  |  | ||||||
| 	"strings" | 	"strings" | ||||||
| @@ -13,24 +15,19 @@ import ( | |||||||
| type Bxmpp struct { | type Bxmpp struct { | ||||||
| 	xc      *xmpp.Client | 	xc      *xmpp.Client | ||||||
| 	xmppMap map[string]string | 	xmppMap map[string]string | ||||||
| 	Config  *config.Protocol | 	*config.BridgeConfig | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | var flog *log.Entry | ||||||
| var protocol = "xmpp" | var protocol = "xmpp" | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	flog = log.WithFields(log.Fields{"prefix": protocol}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bxmpp { | func New(cfg *config.BridgeConfig) *Bxmpp { | ||||||
| 	b := &Bxmpp{} | 	b := &Bxmpp{BridgeConfig: cfg} | ||||||
| 	b.xmppMap = make(map[string]string) | 	b.xmppMap = make(map[string]string) | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.Remote = c |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -43,7 +40,29 @@ func (b *Bxmpp) Connect() error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	flog.Info("Connection succeeded") | ||||||
| 	go b.handleXmpp() | 	go func() { | ||||||
|  | 		initial := true | ||||||
|  | 		bf := &backoff.Backoff{ | ||||||
|  | 			Min:    time.Second, | ||||||
|  | 			Max:    5 * time.Minute, | ||||||
|  | 			Jitter: true, | ||||||
|  | 		} | ||||||
|  | 		for { | ||||||
|  | 			if initial { | ||||||
|  | 				b.handleXmpp() | ||||||
|  | 				initial = false | ||||||
|  | 			} | ||||||
|  | 			d := bf.Duration() | ||||||
|  | 			flog.Infof("Disconnected. Reconnecting in %s", d) | ||||||
|  | 			time.Sleep(d) | ||||||
|  | 			b.xc, err = b.createXMPP() | ||||||
|  | 			if err == nil { | ||||||
|  | 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||||
|  | 				b.handleXmpp() | ||||||
|  | 				bf.Reset() | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -51,15 +70,38 @@ func (b *Bxmpp) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) JoinChannel(channel string) error { | func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.xc.JoinMUCNoHistory(channel+"@"+b.Config.Muc, b.Config.Nick) | 	b.xc.JoinMUCNoHistory(channel.Name+"@"+b.Config.Muc, b.Config.Nick) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) Send(msg config.Message) error { | func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EVENT_MSG_DELETE { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	flog.Debugf("Receiving %#v", msg) | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: rmsg.Channel + "@" + b.Config.Muc, Text: rmsg.Username + rmsg.Text}) | ||||||
|  | 		} | ||||||
|  | 		if len(msg.Extra["file"]) > 0 { | ||||||
|  | 			for _, f := range msg.Extra["file"] { | ||||||
|  | 				fi := f.(config.FileInfo) | ||||||
|  | 				if fi.Comment != "" { | ||||||
|  | 					msg.Text += fi.Comment + ": " | ||||||
|  | 				} | ||||||
|  | 				if fi.URL != "" { | ||||||
|  | 					msg.Text += fi.URL | ||||||
|  | 				} | ||||||
| 				b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text}) | 				b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text}) | ||||||
| 	return nil | 			} | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	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) { | func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | ||||||
| @@ -75,7 +117,7 @@ func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | |||||||
| 		TLSConfig: tc, | 		TLSConfig: tc, | ||||||
|  |  | ||||||
| 		//StartTLS:      false, | 		//StartTLS:      false, | ||||||
| 		Debug:                        true, | 		Debug:                        b.General.Debug, | ||||||
| 		Session:                      true, | 		Session:                      true, | ||||||
| 		Status:                       "", | 		Status:                       "", | ||||||
| 		StatusMessage:                "", | 		StatusMessage:                "", | ||||||
| @@ -96,7 +138,11 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | |||||||
| 		for { | 		for { | ||||||
| 			select { | 			select { | ||||||
| 			case <-ticker.C: | 			case <-ticker.C: | ||||||
| 				b.xc.PingC2S("", "") | 				flog.Debugf("PING") | ||||||
|  | 				err := b.xc.PingC2S("", "") | ||||||
|  | 				if err != nil { | ||||||
|  | 					flog.Debugf("PING failed %#v", err) | ||||||
|  | 				} | ||||||
| 			case <-done: | 			case <-done: | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
| @@ -106,6 +152,7 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) handleXmpp() error { | func (b *Bxmpp) handleXmpp() error { | ||||||
|  | 	var ok bool | ||||||
| 	done := b.xmppKeepAlive() | 	done := b.xmppKeepAlive() | ||||||
| 	defer close(done) | 	defer close(done) | ||||||
| 	nodelay := time.Time{} | 	nodelay := time.Time{} | ||||||
| @@ -126,9 +173,14 @@ func (b *Bxmpp) handleXmpp() error { | |||||||
| 				if len(s) == 2 { | 				if len(s) == 2 { | ||||||
| 					nick = s[1] | 					nick = s[1] | ||||||
| 				} | 				} | ||||||
| 				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" { | 				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" && !strings.Contains(v.Text, "</subject>") { | ||||||
|  | 					rmsg := config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote} | ||||||
|  | 					rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||||
|  | 					if ok { | ||||||
|  | 						rmsg.Event = config.EVENT_USER_ACTION | ||||||
|  | 					} | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account) | 					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account) | ||||||
| 					b.Remote <- config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote} | 					b.Remote <- rmsg | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		case xmpp.Presence: | 		case xmpp.Presence: | ||||||
| @@ -136,3 +188,10 @@ func (b *Bxmpp) handleXmpp() error { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bxmpp) replaceAction(text string) (string, bool) { | ||||||
|  | 	if strings.HasPrefix(text, "/me ") { | ||||||
|  | 		return strings.Replace(text, "/me ", "", -1), true | ||||||
|  | 	} | ||||||
|  | 	return text, false | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										259
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						
									
										259
									
								
								changelog.md
									
									
									
									
									
								
							| @@ -1,3 +1,262 @@ | |||||||
|  | # v1.8.0 | ||||||
|  | ## New features | ||||||
|  | * general: Send chat notification if media is too big to be re-uploaded to MediaServer. See #359 | ||||||
|  | * general: Download (and upload) avatar images from mattermost and telegram when mediaserver is configured. Closes #362 | ||||||
|  | * general: Add label support in RemoteNickFormat | ||||||
|  | * general: Prettier info/debug log output | ||||||
|  | * mattermost: Download files and reupload to supported bridges (mattermost). Closes #357 | ||||||
|  | * slack: Add ShowTopicChange option. Allow/disable topic change messages (currently only from slack). Closes #353 | ||||||
|  | * slack: Add support for file comments (slack). Closes #346 | ||||||
|  | * telegram: Add comment to file upload from telegram. Show comments on all bridges. Closes #358 | ||||||
|  | * telegram: Add markdown support (telegram). #355 | ||||||
|  | * api: Give api access to whole config.Message (and events). Closes #374 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * discord: Check for a valid WebhookURL (discord). Closes #367 | ||||||
|  | * discord: Fix role mention replace issues | ||||||
|  | * irc: Truncate messages sent to IRC based on byte count (#368) | ||||||
|  | * mattermost: Add file download urls also to mattermost webhooks #356 | ||||||
|  | * telegram: Fix panic on nil messages (telegram). Closes #366 | ||||||
|  | * telegram: Fix the UseInsecureURL text (telegram). Closes #184 | ||||||
|  |  | ||||||
|  | # v1.7.1 | ||||||
|  | ## Bugfix | ||||||
|  | * telegram: Enable Long Polling for Telegram. Reduces bandwidth consumption. (#350) | ||||||
|  |  | ||||||
|  | # v1.7.0 | ||||||
|  | ## New features | ||||||
|  | * matrix: Add support for deleting messages from/to matrix (matrix). Closes #320 | ||||||
|  | * xmpp: Ignore <subject> messages (xmpp). #272 | ||||||
|  | * irc: Add twitch support (irc) to README / wiki | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * general: Change RemoteNickFormat replacement order. Closes #336 | ||||||
|  | * general: Make edits/delete work for bridges that gets reused. Closes #342 | ||||||
|  | * general: Lowercase irc channels in config. Closes #348 | ||||||
|  | * matrix: Fix possible panics (matrix). Closes #333 | ||||||
|  | * matrix: Add an extension to images without one (matrix). #331 | ||||||
|  | * api: Obey the Gateway value from the json (api). Closes #344 | ||||||
|  | * xmpp: Print only debug messages when specified (xmpp). Closes #345 | ||||||
|  | * xmpp: Allow xmpp to receive the extra messages (file uploads) when text is empty. #295 | ||||||
|  |  | ||||||
|  | # v1.6.3 | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Fix connection issues | ||||||
|  | * slack: Add more debug messages | ||||||
|  | * irc: Convert received IRC channel names to lowercase. Fixes #329 (#330) | ||||||
|  |  | ||||||
|  | # v1.6.2 | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: Crashes while connecting to Mattermost (regression). Closes #327 | ||||||
|  |  | ||||||
|  | # v1.6.1 | ||||||
|  | ## Bugfix | ||||||
|  | * general: Display of nicks not longer working (regression). Closes #323 | ||||||
|  |  | ||||||
|  | # v1.6.0 | ||||||
|  | ## New features | ||||||
|  | * sshchat: New protocol support added (https://github.com/shazow/ssh-chat) | ||||||
|  | * general: Allow specifying maximum download size of media using MediaDownloadSize (slack,telegram,matrix) | ||||||
|  | * api: Add (simple, one listener) long-polling support (api). Closes #307 | ||||||
|  | * telegram: Add support for forwarded messages. Closes #313 | ||||||
|  | * telegram: Add support for Audio/Voice files (telegram). Closes #314 | ||||||
|  | * irc: Add RejoinDelay option. Delay to rejoin after channel kick (irc). Closes #322 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * telegram: Also use HTML in edited messages (telegram). Closes #315 | ||||||
|  | * matrix: Fix panic (matrix). Closes #316 | ||||||
|  |  | ||||||
|  | # v1.5.1 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * irc: Fix irc ACTION regression (irc). Closes #306 | ||||||
|  | * irc: Split on UTF-8 for MessageSplit (irc). Closes #308 | ||||||
|  |  | ||||||
|  | # v1.5.0 | ||||||
|  | ## New features | ||||||
|  | * general: remote mediaserver support. See MediaServerDownload and MediaServerUpload in matterbridge.toml.sample | ||||||
|  |   more information on https://github.com/42wim/matterbridge/wiki/Mediaserver-setup-%5Badvanced%5D | ||||||
|  | * general: Add support for ReplaceNicks using regexp to replace nicks. Closes #269 (see matterbridge.toml.sample) | ||||||
|  | * general: Add support for ReplaceMessages using regexp to replace messages. #269 (see matterbridge.toml.sample) | ||||||
|  | * irc: Add MessageSplit option to split messages on MessageLength (irc). Closes #281 | ||||||
|  | * matrix: Add support for uploading images/video (matrix). Closes #302 | ||||||
|  | * matrix: Add support for uploaded images/video (matrix)  | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * telegram: Add webp extension to stickers if necessary (telegram) | ||||||
|  | * mattermost: Break when re-login fails (mattermost) | ||||||
|  |  | ||||||
|  | # v1.4.1 | ||||||
|  | ## Bugfix | ||||||
|  | * telegram: fix issue with uploading for images/documents/stickers | ||||||
|  | * slack: remove double messages sent to other bridges when uploading files | ||||||
|  | * irc: Fix strict user handling of girc (irc). Closes #298  | ||||||
|  |  | ||||||
|  | # v1.4.0 | ||||||
|  | ## Breaking changes | ||||||
|  | * general: `[general]` settings don't override the specific bridge settings | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * irc: Replace sorcix/irc and go-ircevent with girc, this should be give better reconnects | ||||||
|  | * steam: Add support for bridging to individual steam chats. (steam) (#294) | ||||||
|  | * telegram: Download files from telegram and reupload to supported bridges (telegram). #278 | ||||||
|  | * slack: Add support to upload files to slack, from bridges with private urls like slack/mattermost/telegram. (slack) | ||||||
|  | * discord: Add support to upload files to discord, from bridges with private urls like slack/mattermost/telegram. (discord) | ||||||
|  | * general: Add systemd service file (#291) | ||||||
|  | * general: Add support for DEBUG=1 envvar to enable debug. Closes #283 | ||||||
|  | * general: Add StripNick option, only allow alphanumerical nicks. Closes #285 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * gitter: Use room.URI instead of room.Name. (gitter) (#293) | ||||||
|  | * slack: Allow slack messages with variables (eg. @here) to be formatted correctly. (slack) (#288) | ||||||
|  | * slack: Resolve slack channel to human-readable name. (slack) (#282) | ||||||
|  | * slack: Use DisplayName instead of deprecated username (slack). Closes #276 | ||||||
|  | * slack: Allowed Slack bridge to extract simpler link format. (#287) | ||||||
|  | * irc: Strip irc colors correct, strip also ctrl chars (irc) | ||||||
|  |  | ||||||
|  | # v1.3.1 | ||||||
|  | ## New features | ||||||
|  | * Support mattermost 4.3.0 and every other 4.x as api4 should be stable (mattermost) | ||||||
|  | ## Bugfix | ||||||
|  | * Use bot username if specified (slack). Closes #273 | ||||||
|  |  | ||||||
|  | # v1.3.0 | ||||||
|  | ## New features | ||||||
|  | * Relay slack_attachments from mattermost to slack (slack). Closes #260 | ||||||
|  | * Add support for quoting previous message when replying (telegram). #237 | ||||||
|  | * Add support for Quakenet auth (irc). Closes #263 | ||||||
|  | * Download files (max size 1MB) from slack and reupload to mattermost (slack/mattermost). Closes #255 | ||||||
|  |  | ||||||
|  | ## Enhancements | ||||||
|  | * Backoff for 60 seconds when reconnecting too fast (irc) #267 | ||||||
|  | * Use override username if specified (mattermost). #260 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * Try to not forward slack unfurls. Closes #266 | ||||||
|  |  | ||||||
|  | # v1.2.0 | ||||||
|  | ## Breaking changes | ||||||
|  | * If you're running a discord bridge, update to this release before 16 october otherwise | ||||||
|  | it will stop working. (see https://discordapp.com/developers/docs/reference) | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * general: Add delete support. (actually delete the messages on bridges that support it) | ||||||
|  |     (mattermost,discord,gitter,slack,telegram) | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * Do not break messages on newline (slack). Closes #258  | ||||||
|  | * Update telegram library | ||||||
|  | * Update discord library (supports v6 API now). Old API is deprecated on 16 October | ||||||
|  |  | ||||||
|  | # v1.1.2 | ||||||
|  | ## New features | ||||||
|  | * general: also build darwin binaries | ||||||
|  | * mattermost: add support for mattermost 4.2.x | ||||||
|  |  | ||||||
|  | ## Bugfix  | ||||||
|  | * mattermost: Send images when text is empty regression. (mattermost). Closes #254 | ||||||
|  | * slack: also send the first messsage after connect. #252 | ||||||
|  |  | ||||||
|  | # v1.1.1 | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: fix public links | ||||||
|  |  | ||||||
|  | # v1.1.0 | ||||||
|  | ## New features | ||||||
|  | * general: Add better editing support. (actually edit the messages on bridges that support it) | ||||||
|  | 	(mattermost,discord,gitter,slack,telegram) | ||||||
|  | * mattermost: use API v4 (removes support for mattermost < 3.8) | ||||||
|  | * mattermost: add support for personal access tokens (since mattermost 4.1) | ||||||
|  | 	Use ```Token="yourtoken"``` in mattermost config | ||||||
|  | 	See https://docs.mattermost.com/developer/personal-access-tokens.html for more info | ||||||
|  | * matrix: Relay notices (matrix). Closes #243 | ||||||
|  | * irc: Add a charset option. Closes #247 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Handle leave/join events (slack). Closes #246 | ||||||
|  | * slack: Replace mentions from other bridges. (slack). Closes #233 | ||||||
|  | * gitter: remove ZWSP after messages | ||||||
|  |  | ||||||
|  | # v1.0.1 | ||||||
|  | ## New features | ||||||
|  | * mattermost: add support for mattermost 4.1.x | ||||||
|  | * discord: allow a webhookURL per channel #239 | ||||||
|  |  | ||||||
|  | # v1.0.0 | ||||||
|  | ## New features | ||||||
|  | * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | ||||||
|  | * discord: Shows the username instead of the server nickname #234 | ||||||
|  |  | ||||||
|  | # v1.0.0-rc1 | ||||||
|  | ## New features | ||||||
|  | * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * general: Handle same account in multiple gateways better | ||||||
|  | * mattermost: ignore edited messages with reactions | ||||||
|  | * mattermost: Fix double posting of edited messages by using lru cache | ||||||
|  | * irc: update vendor | ||||||
|  |  | ||||||
|  | # v0.16.3 | ||||||
|  | ## Bugfix | ||||||
|  | * general: Fix in/out logic. Closes #224  | ||||||
|  | * general: Fix message modification | ||||||
|  | * slack: Disable message from other bots when using webhooks (slack) | ||||||
|  | * mattermost: Return better error messages on mattermost connect | ||||||
|  |  | ||||||
|  | # v0.16.2 | ||||||
|  | ## New features | ||||||
|  | * general: binary builds against latest commit are now available on https://bintray.com/42wim/nightly/Matterbridge/_latestVersion | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: fix loop introduced by relaying message of other bots #219 | ||||||
|  | * slack: Suppress parent message when child message is received #218 | ||||||
|  | * mattermost: fix regression when using webhookurl and webhookbindaddress #221 | ||||||
|  |  | ||||||
|  | # v0.16.1 | ||||||
|  | ## New features | ||||||
|  | * slack: also relay messages of other bots #213 | ||||||
|  | * mattermost: show also links if public links have not been enabled. | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost, slack: fix connecting logic #216 | ||||||
|  |  | ||||||
|  | # v0.16.0 | ||||||
|  | ## Breaking Changes | ||||||
|  | * URL,UseAPI,BindAddress is deprecated. Your config has to be updated. | ||||||
|  |   * URL => WebhookURL | ||||||
|  |   * BindAddress => WebhookBindAddress | ||||||
|  |   * UseAPI => removed  | ||||||
|  |   This change allows you to specify a WebhookURL and a token (slack,discord), so that | ||||||
|  |   messages will be sent with the webhook, but received via the token (API) | ||||||
|  |   If you have not specified WebhookURL and WebhookBindAddress the API (login or token)  | ||||||
|  |   will be used automatically. (no need for UseAPI) | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * mattermost: add support for mattermost 4.0 | ||||||
|  | * steam: New protocol support added (http://store.steampowered.com/) | ||||||
|  | * discord: Support for embedded messages (sent by other bots) | ||||||
|  |   Shows title, description and URL of embedded messages (sent by other bots) | ||||||
|  |   To enable add ```ShowEmbeds=true``` to your discord config  | ||||||
|  | * discord: ```WebhookURL``` posting support added (thanks @saury07) #204 | ||||||
|  |   Discord API does not allow to change the name of the user posting, but webhooks does. | ||||||
|  |  | ||||||
|  | ## Changes | ||||||
|  | * general: all :emoji: will be converted to unicode, providing consistent emojis across all bridges | ||||||
|  | * telegram: Add ```UseInsecureURL``` option for telegram (default false) | ||||||
|  |   WARNING! If enabled this will relay GIF/stickers/documents and other attachments as URLs | ||||||
|  |   Those URLs will contain your bot-token. This may not be what you want. | ||||||
|  |   For now there is no secure way to relay GIF/stickers/documents without seeing your token. | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * irc: detect charset and try to convert it to utf-8 before sending it to other bridges. #209 #210 | ||||||
|  | * slack: Remove label from URLs (slack). #205 | ||||||
|  | * slack: Relay <>& correctly to other bridges #215 | ||||||
|  | * steam: Fix channel id bug in steam (channels are off by 0x18000000000000) | ||||||
|  | * general: various improvements | ||||||
|  | * general: samechannelgateway now relays messages correct again #207 | ||||||
|  |  | ||||||
|  |  | ||||||
| # v0.16.0-rc2 | # v0.16.0-rc2 | ||||||
| ## Breaking Changes | ## Breaking Changes | ||||||
| * URL,UseAPI,BindAddress is deprecated. Your config has to be updated. | * URL,UseAPI,BindAddress is deprecated. Your config has to be updated. | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								ci/bintray.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										27
									
								
								ci/bintray.sh
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,27 @@ | |||||||
|  | #!/bin/bash | ||||||
|  | go version |grep go1.9 || exit | ||||||
|  | VERSION=$(git describe --tags) | ||||||
|  | mkdir ci/binaries | ||||||
|  | GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-windows-amd64.exe | ||||||
|  | GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-amd64 | ||||||
|  | GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | ||||||
|  | GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-darwin-amd64 | ||||||
|  | cd ci | ||||||
|  | cat > deploy.json <<EOF | ||||||
|  | { | ||||||
|  |     "package": { | ||||||
|  |         "name": "Matterbridge", | ||||||
|  |         "repo": "nightly", | ||||||
|  |         "subject": "42wim" | ||||||
|  |     }, | ||||||
|  |     "version": { | ||||||
|  |         "name": "$VERSION" | ||||||
|  |     }, | ||||||
|  |     "files": | ||||||
|  |         [ | ||||||
|  |         {"includePattern": "ci/binaries/(.*)", "uploadPattern":"\$1"} | ||||||
|  |         ], | ||||||
|  |     "publish": true | ||||||
|  | } | ||||||
|  | EOF | ||||||
|  |  | ||||||
							
								
								
									
										11
									
								
								contrib/matterbridge.service
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								contrib/matterbridge.service
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | |||||||
|  | [Unit] | ||||||
|  | Description=matterbridge | ||||||
|  | After=network.target | ||||||
|  |  | ||||||
|  | [Service] | ||||||
|  | ExecStart=/usr/bin/matterbridge -conf /etc/matterbridge/bridge.toml | ||||||
|  | User=matterbridge | ||||||
|  | Group=matterbridge | ||||||
|  |  | ||||||
|  | [Install] | ||||||
|  | WantedBy=multi-user.target | ||||||
							
								
								
									
										11
									
								
								docker/arm/Dockerfile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								docker/arm/Dockerfile
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | |||||||
|  | FROM cmosh/alpine-arm:edge | ||||||
|  | ENTRYPOINT ["/bin/matterbridge"] | ||||||
|  |  | ||||||
|  | COPY . /go/src/github.com/42wim/matterbridge | ||||||
|  | RUN apk update && apk add go git gcc musl-dev ca-certificates \ | ||||||
|  |         && cd /go/src/github.com/42wim/matterbridge \ | ||||||
|  |         && export GOPATH=/go \ | ||||||
|  |         && go get \ | ||||||
|  |         && go build -x -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge \ | ||||||
|  |         && rm -rf /go \ | ||||||
|  |         && apk del --purge git go gcc musl-dev | ||||||
| @@ -1,11 +1,16 @@ | |||||||
| package gateway | package gateway | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge" | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
| 	//	"github.com/davecgh/go-spew/spew" | 	//	"github.com/davecgh/go-spew/spew" | ||||||
|  | 	"crypto/sha1" | ||||||
|  | 	"github.com/hashicorp/golang-lru" | ||||||
|  | 	"github.com/peterhellberg/emojilib" | ||||||
|  | 	"net/http" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -13,62 +18,48 @@ import ( | |||||||
|  |  | ||||||
| type Gateway struct { | type Gateway struct { | ||||||
| 	*config.Config | 	*config.Config | ||||||
|  | 	Router         *Router | ||||||
| 	MyConfig       *config.Gateway | 	MyConfig       *config.Gateway | ||||||
| 	Bridges        map[string]*bridge.Bridge | 	Bridges        map[string]*bridge.Bridge | ||||||
| 	Channels       map[string]*config.ChannelInfo | 	Channels       map[string]*config.ChannelInfo | ||||||
| 	ChannelOptions map[string]config.ChannelOptions | 	ChannelOptions map[string]config.ChannelOptions | ||||||
| 	Names           map[string]bool |  | ||||||
| 	Name            string |  | ||||||
| 	Message        chan config.Message | 	Message        chan config.Message | ||||||
| 	DestChannelFunc func(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo | 	Name           string | ||||||
|  | 	Messages       *lru.Cache | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg *config.Config) *Gateway { | type BrMsgID struct { | ||||||
| 	gw := &Gateway{} | 	br        *bridge.Bridge | ||||||
| 	gw.Config = cfg | 	ID        string | ||||||
| 	gw.Channels = make(map[string]*config.ChannelInfo) | 	ChannelID string | ||||||
| 	gw.Message = make(chan config.Message) | } | ||||||
| 	gw.Bridges = make(map[string]*bridge.Bridge) |  | ||||||
| 	gw.Names = make(map[string]bool) | var flog *log.Entry | ||||||
| 	gw.DestChannelFunc = gw.getDestChannel |  | ||||||
|  | func init() { | ||||||
|  | 	flog = log.WithFields(log.Fields{"prefix": "gateway"}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func New(cfg config.Gateway, r *Router) *Gateway { | ||||||
|  | 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | ||||||
|  | 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | ||||||
|  | 	cache, _ := lru.New(5000) | ||||||
|  | 	gw.Messages = cache | ||||||
|  | 	gw.AddConfig(&cfg) | ||||||
| 	return gw | 	return gw | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||||
| 	for _, br := range gw.Bridges { | 	br := gw.Router.getBridge(cfg.Account) | ||||||
| 		if br.Account == cfg.Account { | 	if br == nil { | ||||||
| 			gw.mapChannelsToBridge(br) | 		br = bridge.New(gw.Config, cfg, gw.Message) | ||||||
| 			err := br.JoinChannels() |  | ||||||
| 			if err != nil { |  | ||||||
| 				return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) |  | ||||||
| 	} | 	} | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	log.Infof("Starting bridge: %s ", cfg.Account) |  | ||||||
| 	br := bridge.New(gw.Config, cfg, gw.Message) |  | ||||||
| 	gw.mapChannelsToBridge(br) | 	gw.mapChannelsToBridge(br) | ||||||
| 	gw.Bridges[cfg.Account] = br | 	gw.Bridges[cfg.Account] = br | ||||||
| 	err := br.Connect() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) |  | ||||||
| 	} |  | ||||||
| 	err = br.JoinChannels() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) |  | ||||||
| 	} |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | ||||||
| 	if gw.Names[cfg.Name] { |  | ||||||
| 		return fmt.Errorf("Gateway with name %s already exists", cfg.Name) |  | ||||||
| 	} |  | ||||||
| 	if cfg.Name == "" { |  | ||||||
| 		return fmt.Errorf("%s", "Gateway without name found") |  | ||||||
| 	} |  | ||||||
| 	log.Infof("Starting gateway: %s", cfg.Name) |  | ||||||
| 	gw.Names[cfg.Name] = true |  | ||||||
| 	gw.Name = cfg.Name | 	gw.Name = cfg.Name | ||||||
| 	gw.MyConfig = cfg | 	gw.MyConfig = cfg | ||||||
| 	gw.mapChannels() | 	gw.mapChannels() | ||||||
| @@ -89,49 +80,14 @@ func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| 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) { | func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | ||||||
| 	br.Disconnect() | 	br.Disconnect() | ||||||
| 	time.Sleep(time.Second * 5) | 	time.Sleep(time.Second * 5) | ||||||
| RECONNECT: | RECONNECT: | ||||||
| 	log.Infof("Reconnecting %s", br.Account) | 	flog.Infof("Reconnecting %s", br.Account) | ||||||
| 	err := br.Connect() | 	err := br.Connect() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err) | 		flog.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err) | ||||||
| 		time.Sleep(time.Second * 60) | 		time.Sleep(time.Second * 60) | ||||||
| 		goto RECONNECT | 		goto RECONNECT | ||||||
| 	} | 	} | ||||||
| @@ -139,107 +95,181 @@ RECONNECT: | |||||||
| 	br.JoinChannels() | 	br.JoinChannels() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) mapChannels() error { | func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | ||||||
| 	for _, br := range append(gw.MyConfig.Out, gw.MyConfig.InOut...) { | 	for _, br := range cfg { | ||||||
| 		if isApi(br.Account) { | 		if isApi(br.Account) { | ||||||
| 			br.Channel = "api" | 			br.Channel = "api" | ||||||
| 		} | 		} | ||||||
|  | 		// make sure to lowercase irc channels in config #348 | ||||||
|  | 		if strings.HasPrefix(br.Account, "irc.") { | ||||||
|  | 			br.Channel = strings.ToLower(br.Channel) | ||||||
|  | 		} | ||||||
| 		ID := br.Channel + br.Account | 		ID := br.Channel + br.Account | ||||||
| 		_, ok := gw.Channels[ID] | 		if _, ok := gw.Channels[ID]; !ok { | ||||||
| 		if !ok { | 			channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account, | ||||||
| 			channel := &config.ChannelInfo{Name: br.Channel, Direction: "out", ID: ID, Options: br.Options, Account: br.Account, | 				SameChannel: make(map[string]bool)} | ||||||
| 				GID: make(map[string]bool), SameChannel: make(map[string]bool)} |  | ||||||
| 			channel.GID[gw.Name] = true |  | ||||||
| 			channel.SameChannel[gw.Name] = br.SameChannel | 			channel.SameChannel[gw.Name] = br.SameChannel | ||||||
| 			gw.Channels[channel.ID] = channel | 			gw.Channels[channel.ID] = channel | ||||||
|  | 		} else { | ||||||
|  | 			// if we already have a key and it's not our current direction it means we have a bidirectional inout | ||||||
|  | 			if gw.Channels[ID].Direction != direction { | ||||||
|  | 				gw.Channels[ID].Direction = "inout" | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 		gw.Channels[ID].GID[gw.Name] = true |  | ||||||
| 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | 		gw.Channels[ID].SameChannel[gw.Name] = br.SameChannel | ||||||
| 	} | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| 	for _, br := range append(gw.MyConfig.In, gw.MyConfig.InOut...) { | func (gw *Gateway) mapChannels() error { | ||||||
| 		if isApi(br.Account) { | 	gw.mapChannelConfig(gw.MyConfig.In, "in") | ||||||
| 			br.Channel = "api" | 	gw.mapChannelConfig(gw.MyConfig.Out, "out") | ||||||
| 		} | 	gw.mapChannelConfig(gw.MyConfig.InOut, "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 | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo { | func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo { | ||||||
| 	var channels []config.ChannelInfo | 	var channels []config.ChannelInfo | ||||||
|  |  | ||||||
|  | 	// for messages received from the api check that the gateway is the specified one | ||||||
|  | 	if msg.Protocol == "api" && gw.Name != msg.Gateway { | ||||||
|  | 		return channels | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// if source channel is in only, do nothing | ||||||
|  | 	for _, channel := range gw.Channels { | ||||||
|  | 		// lookup the channel from the message | ||||||
|  | 		if channel.ID == getChannelID(*msg) { | ||||||
|  | 			// we only have destinations if the original message is from an "in" (sending) channel | ||||||
|  | 			if !strings.Contains(channel.Direction, "in") { | ||||||
|  | 				return channels | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
| 	for _, channel := range gw.Channels { | 	for _, channel := range gw.Channels { | ||||||
| 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		// add gateway to message |  | ||||||
| 		gw.validGatewayDest(msg, channel) |  | ||||||
|  |  | ||||||
| 		// do samechannelgateway logic | 		// do samechannelgateway flogic | ||||||
| 		if channel.SameChannel[msg.Gateway] { | 		if channel.SameChannel[msg.Gateway] { | ||||||
| 			if msg.Channel == channel.Name && msg.Account != dest.Account { | 			if msg.Channel == channel.Name && msg.Account != dest.Account { | ||||||
| 				channels = append(channels, *channel) | 				channels = append(channels, *channel) | ||||||
| 			} | 			} | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  | 		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) { | ||||||
| 		if channel.Direction == "out" && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) { |  | ||||||
| 			channels = append(channels, *channel) | 			channels = append(channels, *channel) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) { | func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||||
| 	// only relay join/part when configged | 	var brMsgIDs []*BrMsgID | ||||||
| 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart { |  | ||||||
| 		return | 	// TODO refactor | ||||||
| 	} | 	// only slack now, check will have to be done in the different bridges. | ||||||
| 	// broadcast to every out channel (irc QUIT) | 	// we need to check if we can't use fallback or text in other bridges | ||||||
| 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | 	if msg.Extra != nil { | ||||||
| 		log.Debug("empty channel") | 		if dest.Protocol != "discord" && | ||||||
| 		return | 			dest.Protocol != "slack" && | ||||||
| 	} | 			dest.Protocol != "mattermost" && | ||||||
| 	originchannel := msg.Channel | 			dest.Protocol != "telegram" && | ||||||
| 	origmsg := msg | 			dest.Protocol != "matrix" && | ||||||
| 	for _, channel := range gw.DestChannelFunc(&msg, *dest) { | 			dest.Protocol != "xmpp" && | ||||||
| 		// do not send to ourself | 			len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) == 0 { | ||||||
| 		if channel.ID == getChannelID(origmsg) { | 			if msg.Text == "" { | ||||||
| 			continue | 				return brMsgIDs | ||||||
| 		} |  | ||||||
| 		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) |  | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Avatar downloads are only relevant for telegram and mattermost for now | ||||||
|  | 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||||
|  | 		if dest.Protocol != "mattermost" && | ||||||
|  | 			dest.Protocol != "telegram" { | ||||||
|  | 			return brMsgIDs | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	// only relay join/part when configged | ||||||
|  | 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart { | ||||||
|  | 		return brMsgIDs | ||||||
|  | 	} | ||||||
|  | 	if msg.Event == config.EVENT_TOPIC_CHANGE && !gw.Bridges[dest.Account].Config.ShowTopicChange { | ||||||
|  | 		return brMsgIDs | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// broadcast to every out channel (irc QUIT) | ||||||
|  | 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | ||||||
|  | 		flog.Debug("empty channel") | ||||||
|  | 		return brMsgIDs | ||||||
|  | 	} | ||||||
|  | 	originchannel := msg.Channel | ||||||
|  | 	origmsg := msg | ||||||
|  | 	channels := gw.getDestChannel(&msg, *dest) | ||||||
|  | 	for _, channel := range channels { | ||||||
|  | 		// Only send the avatar download event to ourselves. | ||||||
|  | 		if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||||
|  | 			if channel.ID != getChannelID(origmsg) { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			// do not send to ourself for any other event | ||||||
|  | 			if channel.ID == getChannelID(origmsg) { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name) | ||||||
|  | 		msg.Channel = channel.Name | ||||||
|  | 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||||
|  | 		msg.Username = gw.modifyUsername(origmsg, dest) | ||||||
|  | 		msg.ID = "" | ||||||
|  | 		if res, ok := gw.Messages.Get(origmsg.ID); ok { | ||||||
|  | 			IDs := res.([]*BrMsgID) | ||||||
|  | 			for _, id := range IDs { | ||||||
|  | 				// check protocol, bridge name and channelname | ||||||
|  | 				// for people that reuse the same bridge multiple times. see #342 | ||||||
|  | 				if dest.Protocol == id.br.Protocol && dest.Name == id.br.Name && channel.ID == id.ChannelID { | ||||||
|  | 					msg.ID = id.ID | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		// for api we need originchannel as channel | ||||||
|  | 		if dest.Protocol == "api" { | ||||||
|  | 			msg.Channel = originchannel | ||||||
|  | 		} | ||||||
|  | 		mID, err := dest.Send(msg) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Println(err) | ||||||
|  | 		} | ||||||
|  | 		// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||||
|  | 		if mID != "" { | ||||||
|  | 			brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID, channel.ID}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return brMsgIDs | ||||||
|  | } | ||||||
|  |  | ||||||
| func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||||
|  | 	// if we don't have the bridge, ignore it | ||||||
|  | 	if _, ok := gw.Bridges[msg.Account]; !ok { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
| 	if msg.Text == "" { | 	if msg.Text == "" { | ||||||
| 		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | 		// we have an attachment or actual bytes | ||||||
|  | 		if msg.Extra != nil && | ||||||
|  | 			(msg.Extra["attachments"] != nil || | ||||||
|  | 				len(msg.Extra["file"]) > 0 || | ||||||
|  | 				len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) > 0) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) { | 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) { | ||||||
| 		if msg.Username == entry { | 		if msg.Username == entry { | ||||||
| 			log.Debugf("ignoring %s from %s", msg.Username, msg.Account) | 			flog.Debugf("ignoring %s from %s", msg.Username, msg.Account) | ||||||
| 			return true | 			return true | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -248,11 +278,11 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | |||||||
| 		if entry != "" { | 		if entry != "" { | ||||||
| 			re, err := regexp.Compile(entry) | 			re, err := regexp.Compile(entry) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				log.Errorf("incorrect regexp %s for %s", entry, msg.Account) | 				flog.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
| 			if re.MatchString(msg.Text) { | 			if re.MatchString(msg.Text) { | ||||||
| 				log.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account) | 				flog.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account) | ||||||
| 				return true | 				return true | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -260,13 +290,31 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | |||||||
| 	return false | 	return false | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) modifyUsername(msg *config.Message, dest *bridge.Bridge) { | func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string { | ||||||
| 	br := gw.Bridges[msg.Account] | 	br := gw.Bridges[msg.Account] | ||||||
| 	msg.Protocol = br.Protocol | 	msg.Protocol = br.Protocol | ||||||
| 	nick := gw.Config.General.RemoteNickFormat | 	if gw.Config.General.StripNick || dest.Config.StripNick { | ||||||
| 	if nick == "" { | 		re := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||||
| 		nick = dest.Config.RemoteNickFormat | 		msg.Username = re.ReplaceAllString(msg.Username, "") | ||||||
| 	} | 	} | ||||||
|  | 	nick := dest.Config.RemoteNickFormat | ||||||
|  | 	if nick == "" { | ||||||
|  | 		nick = gw.Config.General.RemoteNickFormat | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// loop to replace nicks | ||||||
|  | 	for _, outer := range br.Config.ReplaceNicks { | ||||||
|  | 		search := outer[0] | ||||||
|  | 		replace := outer[1] | ||||||
|  | 		// TODO move compile to bridge init somewhere | ||||||
|  | 		re, err := regexp.Compile(search) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		msg.Username = re.ReplaceAllString(msg.Username, replace) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if len(msg.Username) > 0 { | 	if len(msg.Username) > 0 { | ||||||
| 		// fix utf-8 issue #193 | 		// fix utf-8 issue #193 | ||||||
| 		i := 0 | 		i := 0 | ||||||
| @@ -279,13 +327,14 @@ func (gw *Gateway) modifyUsername(msg *config.Message, dest *bridge.Bridge) { | |||||||
| 		} | 		} | ||||||
| 		nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1) | 		nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1) | ||||||
| 	} | 	} | ||||||
| 	nick = strings.Replace(nick, "{NICK}", msg.Username, -1) |  | ||||||
| 	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1) | 	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1) | ||||||
| 	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1) | 	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1) | ||||||
| 	msg.Username = nick | 	nick = strings.Replace(nick, "{LABEL}", br.Config.Label, -1) | ||||||
|  | 	nick = strings.Replace(nick, "{NICK}", msg.Username, -1) | ||||||
|  | 	return nick | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) modifyAvatar(msg *config.Message, dest *bridge.Bridge) { | func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string { | ||||||
| 	iconurl := gw.Config.General.IconURL | 	iconurl := gw.Config.General.IconURL | ||||||
| 	if iconurl == "" { | 	if iconurl == "" { | ||||||
| 		iconurl = dest.Config.IconURL | 		iconurl = dest.Config.IconURL | ||||||
| @@ -294,6 +343,61 @@ func (gw *Gateway) modifyAvatar(msg *config.Message, dest *bridge.Bridge) { | |||||||
| 	if msg.Avatar == "" { | 	if msg.Avatar == "" { | ||||||
| 		msg.Avatar = iconurl | 		msg.Avatar = iconurl | ||||||
| 	} | 	} | ||||||
|  | 	return msg.Avatar | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||||
|  | 	// replace :emoji: to unicode | ||||||
|  | 	msg.Text = emojilib.Replace(msg.Text) | ||||||
|  | 	br := gw.Bridges[msg.Account] | ||||||
|  | 	// loop to replace messages | ||||||
|  | 	for _, outer := range br.Config.ReplaceMessages { | ||||||
|  | 		search := outer[0] | ||||||
|  | 		replace := outer[1] | ||||||
|  | 		// TODO move compile to bridge init somewhere | ||||||
|  | 		re, err := regexp.Compile(search) | ||||||
|  | 		if err != nil { | ||||||
|  | 			flog.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		msg.Text = re.ReplaceAllString(msg.Text, replace) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// messages from api have Gateway specified, don't overwrite | ||||||
|  | 	if msg.Protocol != "api" { | ||||||
|  | 		msg.Gateway = gw.Name | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) handleFiles(msg *config.Message) { | ||||||
|  | 	if msg.Extra == nil || gw.Config.General.MediaServerUpload == "" { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if len(msg.Extra["file"]) > 0 { | ||||||
|  | 		client := &http.Client{ | ||||||
|  | 			Timeout: time.Second * 5, | ||||||
|  | 		} | ||||||
|  | 		for i, f := range msg.Extra["file"] { | ||||||
|  | 			fi := f.(config.FileInfo) | ||||||
|  | 			sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data)) | ||||||
|  | 			reader := bytes.NewReader(*fi.Data) | ||||||
|  | 			url := gw.Config.General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name | ||||||
|  | 			durl := gw.Config.General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||||
|  | 			extra := msg.Extra["file"][i].(config.FileInfo) | ||||||
|  | 			extra.URL = durl | ||||||
|  | 			req, _ := http.NewRequest("PUT", url, reader) | ||||||
|  | 			req.Header.Set("Content-Type", "binary/octet-stream") | ||||||
|  | 			_, err := client.Do(req) | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Errorf("mediaserver upload failed: %#v", err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			flog.Debugf("mediaserver download URL = %s", durl) | ||||||
|  | 			// we uploaded the file successfully. Add the SHA | ||||||
|  | 			extra.SHA = sha1sum | ||||||
|  | 			msg.Extra["file"][i] = extra | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func getChannelID(msg config.Message) string { | func getChannelID(msg config.Message) string { | ||||||
| @@ -301,40 +405,9 @@ func getChannelID(msg config.Message) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | ||||||
| 	GIDmap := gw.Channels[getChannelID(*msg)].GID | 	return msg.Gateway == gw.Name | ||||||
|  |  | ||||||
| 	// gateway is specified in message (probably from api) |  | ||||||
| 	if msg.Gateway != "" { |  | ||||||
| 		return channel.GID[msg.Gateway] |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// check if we are running a samechannelgateway. |  | ||||||
| 	// if it is and the channel name matches it's ok, otherwise we shouldn't use this channel. |  | ||||||
| 	for k, _ := range GIDmap { |  | ||||||
| 		if channel.SameChannel[k] == true { |  | ||||||
| 			if msg.Channel == channel.Name { |  | ||||||
| 				// add the gateway to our message |  | ||||||
| 				msg.Gateway = k |  | ||||||
| 				return true |  | ||||||
| 			} else { |  | ||||||
| 				return false |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	// check if we are in the correct gateway |  | ||||||
| 	for k, _ := range GIDmap { |  | ||||||
| 		if channel.GID[k] == true { |  | ||||||
| 			// add the gateway to our message |  | ||||||
| 			msg.Gateway = k |  | ||||||
| 			return true |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func isApi(account string) bool { | func isApi(account string) bool { | ||||||
| 	if strings.HasPrefix(account, "api.") { | 	return strings.HasPrefix(account, "api.") | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										288
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										288
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,288 @@ | |||||||
|  | package gateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/BurntSushi/toml" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  | 	"strconv" | ||||||
|  |  | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var testconfig = ` | ||||||
|  | [irc.freenode] | ||||||
|  | [mattermost.test] | ||||||
|  | [gitter.42wim] | ||||||
|  | [discord.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge1" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting" | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |     #channel="matterbridge/Lobby" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general" | ||||||
|  |      | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="slack.test" | ||||||
|  |     channel="testing" | ||||||
|  | 	` | ||||||
|  |  | ||||||
|  | var testconfig2 = ` | ||||||
|  | [irc.freenode] | ||||||
|  | [mattermost.test] | ||||||
|  | [gitter.42wim] | ||||||
|  | [discord.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge1" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting" | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general" | ||||||
|  |      | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="slack.test" | ||||||
|  |     channel="testing" | ||||||
|  | [[gateway]] | ||||||
|  |     name = "bridge2" | ||||||
|  |     enable=true | ||||||
|  |      | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account = "irc.freenode" | ||||||
|  |     channel = "#wimtesting2" | ||||||
|  |      | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="gitter.42wim" | ||||||
|  |     channel="42wim/testroom" | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account = "discord.test" | ||||||
|  |     channel = "general2" | ||||||
|  | 	` | ||||||
|  | var testconfig3 = ` | ||||||
|  | [irc.zzz] | ||||||
|  | [telegram.zzz] | ||||||
|  | [slack.zzz] | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main"		 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="-1111111111111" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="slack.zzz" | ||||||
|  |     channel="irc"	 | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="announcements" | ||||||
|  | enable=true | ||||||
|  | 	 | ||||||
|  |     [[gateway.in]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="-2222222222222"	 | ||||||
|  | 	 | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main"		 | ||||||
|  | 	 | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-help"	 | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--333333333333"	 | ||||||
|  |  | ||||||
|  |     [[gateway.out]] | ||||||
|  |     account="slack.zzz" | ||||||
|  |     channel="general"		 | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge2" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-help"	 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--444444444444"	 | ||||||
|  |  | ||||||
|  | 	 | ||||||
|  | [[gateway]] | ||||||
|  | name="bridge3" | ||||||
|  | enable=true | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="irc.zzz" | ||||||
|  |     channel="#main-telegram"	 | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="telegram.zzz" | ||||||
|  |     channel="--333333333333" | ||||||
|  | ` | ||||||
|  |  | ||||||
|  | func maketestRouter(input string) *Router { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(input, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	r, err := NewRouter(cfg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	return r | ||||||
|  | } | ||||||
|  | func TestNewRouter(t *testing.T) { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	r, err := NewRouter(cfg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	assert.Equal(t, 1, len(r.Gateways)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
|  |  | ||||||
|  | 	r = maketestRouter(testconfig2) | ||||||
|  | 	assert.Equal(t, 2, len(r.Gateways)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | ||||||
|  | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) | ||||||
|  | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
|  | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "out", | ||||||
|  | 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||||
|  | 		SameChannel: map[string]bool{"bridge2": false}}, | ||||||
|  | 		r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "in", | ||||||
|  | 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||||
|  | 		SameChannel: map[string]bool{"bridge1": false}}, | ||||||
|  | 		r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) | ||||||
|  | 	assert.Equal(t, &config.ChannelInfo{Name: "general", Direction: "inout", | ||||||
|  | 		ID: "generaldiscord.test", Account: "discord.test", | ||||||
|  | 		SameChannel: map[string]bool{"bridge1": false}}, | ||||||
|  | 		r.Gateways["bridge1"].Channels["generaldiscord.test"]) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestGetDestChannel(t *testing.T) { | ||||||
|  | 	r := maketestRouter(testconfig2) | ||||||
|  | 	msg := &config.Message{Text: "test", Channel: "general", Account: "discord.test", Gateway: "bridge1", Protocol: "discord", Username: "test"} | ||||||
|  | 	for _, br := range r.Gateways["bridge1"].Bridges { | ||||||
|  | 		switch br.Account { | ||||||
|  | 		case "discord.test": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "discord.test", Direction: "inout", ID: "generaldiscord.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||||
|  | 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "slack.test": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo{{Name: "testing", Account: "slack.test", Direction: "out", ID: "testingslack.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||||
|  | 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "gitter.42wim": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		case "irc.freenode": | ||||||
|  | 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestGetDestChannelAdvanced(t *testing.T) { | ||||||
|  | 	r := maketestRouter(testconfig3) | ||||||
|  | 	var msgs []*config.Message | ||||||
|  | 	i := 0 | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		for _, channel := range gw.Channels { | ||||||
|  | 			msgs = append(msgs, &config.Message{Text: "text" + strconv.Itoa(i), Channel: channel.Name, Account: channel.Account, Gateway: gw.Name, Username: "user" + strconv.Itoa(i)}) | ||||||
|  | 			i++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	hits := make(map[string]int) | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		for _, br := range gw.Bridges { | ||||||
|  | 			for _, msg := range msgs { | ||||||
|  | 				channels := gw.getDestChannel(msg, *br) | ||||||
|  | 				if gw.Name != msg.Gateway { | ||||||
|  | 					assert.Equal(t, []config.ChannelInfo(nil), channels) | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 				switch gw.Name { | ||||||
|  | 				case "bridge": | ||||||
|  | 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz" || msg.Account == "slack.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "inout", ID: "#mainirc.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "-1111111111111", Account: "telegram.zzz", Direction: "inout", ID: "-1111111111111telegram.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "slack.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "irc", Account: "slack.zzz", Direction: "inout", ID: "ircslack.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "bridge2": | ||||||
|  | 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-help", Account: "irc.zzz", Direction: "inout", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "--444444444444", Account: "telegram.zzz", Direction: "inout", ID: "--444444444444telegram.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "bridge3": | ||||||
|  | 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||||
|  | 						hits[gw.Name]++ | ||||||
|  | 						switch br.Account { | ||||||
|  | 						case "irc.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-telegram", Account: "irc.zzz", Direction: "inout", ID: "#main-telegramirc.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						case "telegram.zzz": | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "inout", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				case "announcements": | ||||||
|  | 					if msg.Channel != "-2222222222222" && msg.Account != "telegram" { | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo(nil), channels) | ||||||
|  | 						continue | ||||||
|  | 					} | ||||||
|  | 					hits[gw.Name]++ | ||||||
|  | 					switch br.Account { | ||||||
|  | 					case "irc.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "out", ID: "#mainirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}, {Name: "#main-help", Account: "irc.zzz", Direction: "out", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					case "slack.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "slack.zzz", Direction: "out", ID: "generalslack.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					case "telegram.zzz": | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "out", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits) | ||||||
|  | } | ||||||
							
								
								
									
										114
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,114 @@ | |||||||
|  | package gateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/gateway/samechannel" | ||||||
|  | 	//log "github.com/sirupsen/logrus" | ||||||
|  | 	//	"github.com/davecgh/go-spew/spew" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Router struct { | ||||||
|  | 	Gateways map[string]*Gateway | ||||||
|  | 	Message  chan config.Message | ||||||
|  | 	*config.Config | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewRouter(cfg *config.Config) (*Router, error) { | ||||||
|  | 	r := &Router{} | ||||||
|  | 	r.Config = cfg | ||||||
|  | 	r.Message = make(chan config.Message) | ||||||
|  | 	r.Gateways = make(map[string]*Gateway) | ||||||
|  | 	sgw := samechannelgateway.New(cfg) | ||||||
|  | 	gwconfigs := sgw.GetConfig() | ||||||
|  |  | ||||||
|  | 	for _, entry := range append(gwconfigs, cfg.Gateway...) { | ||||||
|  | 		if !entry.Enable { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if entry.Name == "" { | ||||||
|  | 			return nil, fmt.Errorf("%s", "Gateway without name found") | ||||||
|  | 		} | ||||||
|  | 		if _, ok := r.Gateways[entry.Name]; ok { | ||||||
|  | 			return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name) | ||||||
|  | 		} | ||||||
|  | 		r.Gateways[entry.Name] = New(entry, r) | ||||||
|  | 	} | ||||||
|  | 	return r, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) Start() error { | ||||||
|  | 	m := make(map[string]*bridge.Bridge) | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		flog.Infof("Parsing gateway %s", gw.Name) | ||||||
|  | 		for _, br := range gw.Bridges { | ||||||
|  | 			m[br.Account] = br | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, br := range m { | ||||||
|  | 		flog.Infof("Starting bridge: %s ", br.Account) | ||||||
|  | 		err := br.Connect() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||||
|  | 		} | ||||||
|  | 		err = br.JoinChannels() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	go r.handleReceive() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) getBridge(account string) *bridge.Bridge { | ||||||
|  | 	for _, gw := range r.Gateways { | ||||||
|  | 		if br, ok := gw.Bridges[account]; ok { | ||||||
|  | 			return br | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *Router) handleReceive() { | ||||||
|  | 	for msg := range r.Message { | ||||||
|  | 		if msg.Event == config.EVENT_FAILURE { | ||||||
|  | 		Loop: | ||||||
|  | 			for _, gw := range r.Gateways { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						go gw.reconnectBridge(br) | ||||||
|  | 						break Loop | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if msg.Event == config.EVENT_REJOIN_CHANNELS { | ||||||
|  | 			for _, gw := range r.Gateways { | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					if msg.Account == br.Account { | ||||||
|  | 						br.Joined = make(map[string]bool) | ||||||
|  | 						br.JoinChannels() | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for _, gw := range r.Gateways { | ||||||
|  | 			// record all the message ID's of the different bridges | ||||||
|  | 			var msgIDs []*BrMsgID | ||||||
|  | 			if !gw.ignoreMessage(&msg) { | ||||||
|  | 				msg.Timestamp = time.Now() | ||||||
|  | 				gw.modifyMessage(&msg) | ||||||
|  | 				gw.handleFiles(&msg) | ||||||
|  | 				for _, br := range gw.Bridges { | ||||||
|  | 					msgIDs = append(msgIDs, gw.handleMessage(msg, br)...) | ||||||
|  | 				} | ||||||
|  | 				// only add the message ID if it doesn't already exists | ||||||
|  | 				if _, ok := gw.Messages.Get(msg.ID); !ok && msg.ID != "" { | ||||||
|  | 					gw.Messages.Add(msg.ID, msgIDs) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										31
									
								
								gateway/samechannel/samechannel_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								gateway/samechannel/samechannel_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | package samechannelgateway | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/BurntSushi/toml" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  |  | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var testconfig = ` | ||||||
|  | [mattermost.test] | ||||||
|  | [slack.test] | ||||||
|  |  | ||||||
|  | [[samechannelgateway]] | ||||||
|  |    enable = true | ||||||
|  |    name = "blah" | ||||||
|  |       accounts = [ "mattermost.test","slack.test" ] | ||||||
|  |       channels = [ "testing","testing2","testing10"] | ||||||
|  | ` | ||||||
|  |  | ||||||
|  | func TestGetConfig(t *testing.T) { | ||||||
|  | 	var cfg *config.Config | ||||||
|  | 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | 	sgw := New(cfg) | ||||||
|  | 	configs := sgw.GetConfig() | ||||||
|  | 	assert.Equal(t, []config.Gateway{{Name: "blah", Enable: true, In: []config.Bridge(nil), Out: []config.Bridge(nil), InOut: []config.Bridge{{Account: "mattermost.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}}}}, configs) | ||||||
|  | } | ||||||
| @@ -99,10 +99,9 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) { | |||||||
|  |  | ||||||
| // Receive returns an incoming message from mattermost outgoing webhooks URL. | // Receive returns an incoming message from mattermost outgoing webhooks URL. | ||||||
| func (c *Client) Receive() Message { | func (c *Client) Receive() Message { | ||||||
| 	for { | 	var msg Message | ||||||
| 		select { | 	for msg = range c.In { | ||||||
| 		case msg := <-c.In: |  | ||||||
| 		return msg | 		return msg | ||||||
| 	} | 	} | ||||||
| 	} | 	return msg | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,22 +5,21 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/gateway" | 	"github.com/42wim/matterbridge/gateway" | ||||||
| 	"github.com/42wim/matterbridge/gateway/samechannel" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/google/gops/agent" | 	"github.com/google/gops/agent" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	prefixed "github.com/x-cray/logrus-prefixed-formatter" | ||||||
|  | 	"os" | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	version = "0.16.0-rc2" | 	version = "1.8.0" | ||||||
| 	githash string | 	githash string | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
|  | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true}) | ||||||
|  | 	flog := log.WithFields(log.Fields{"prefix": "main"}) | ||||||
| 	flagConfig := flag.String("conf", "matterbridge.toml", "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") | ||||||
| @@ -34,32 +33,25 @@ func main() { | |||||||
| 		fmt.Printf("version: %s %s\n", version, githash) | 		fmt.Printf("version: %s %s\n", version, githash) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if *flagDebug { | 	if *flagDebug || os.Getenv("DEBUG") == "1" { | ||||||
| 		log.Info("Enabling debug") | 		log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||||
|  | 		flog.Info("Enabling debug") | ||||||
| 		log.SetLevel(log.DebugLevel) | 		log.SetLevel(log.DebugLevel) | ||||||
| 	} | 	} | ||||||
| 	log.Printf("Running version %s %s", version, githash) | 	flog.Printf("Running version %s %s", version, githash) | ||||||
| 	if strings.Contains(version, "-dev") { | 	if strings.Contains(version, "-dev") { | ||||||
| 		log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | 		flog.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||||
| 	} | 	} | ||||||
| 	cfg := config.NewConfig(*flagConfig) | 	cfg := config.NewConfig(*flagConfig) | ||||||
|  | 	cfg.General.Debug = *flagDebug | ||||||
| 	g := gateway.New(cfg) | 	r, err := gateway.NewRouter(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 { | 	if err != nil { | ||||||
| 			log.Fatalf("Starting gateway failed: %s", err) | 		flog.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	} | 	} | ||||||
| 	} | 	err = r.Start() | ||||||
| 	err := g.Start() |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Fatalf("Starting gateway failed: %s", err) | 		flog.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	} | 	} | ||||||
| 	log.Printf("Gateway(s) started succesfully. Now relaying messages") | 	flog.Printf("Gateway(s) started succesfully. Now relaying messages") | ||||||
| 	select {} | 	select {} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -32,16 +32,38 @@ UseSASL=false | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| SkipTLSVerify=true | SkipTLSVerify=true | ||||||
|  |  | ||||||
|  | #If you know your charset, you can specify it manually.  | ||||||
|  | #Otherwise it tries to detect this automatically. Select one below | ||||||
|  | # "iso-8859-2:1987", "iso-8859-9:1989", "866", "latin9", "iso-8859-10:1992", "iso-ir-109", "hebrew",  | ||||||
|  | # "cp932", "iso-8859-15", "cp437", "utf-16be", "iso-8859-3:1988", "windows-1251", "utf16", "latin6",  | ||||||
|  | # "latin3", "iso-8859-1:1987", "iso-8859-9", "utf-16le", "big5", "cp819", "asmo-708", "utf-8",  | ||||||
|  | # "ibm437", "iso-ir-157", "iso-ir-144", "latin4", "850", "iso-8859-5", "iso-8859-5:1988", "l3",  | ||||||
|  | # "windows-31j", "utf8", "iso-8859-3", "437", "greek", "iso-8859-8", "l6", "l9-iso-8859-15",  | ||||||
|  | # "iso-8859-2", "latin2", "iso-ir-100", "iso-8859-6", "arabic", "iso-ir-148", "us-ascii", "x-sjis",  | ||||||
|  | # "utf16be", "iso-8859-8:1988", "utf16le", "l4", "utf-16", "iso-ir-138", "iso-8859-7", "iso-8859-7:1987",  | ||||||
|  | # "windows-1252", "l2", "koi8-r", "iso8859-1", "latin1", "ecma-114", "iso-ir-110", "elot-928",  | ||||||
|  | # "iso-ir-126", "iso-8859-1", "iso-ir-127", "cp850", "cyrillic", "greek8", "windows-1250", "iso-latin-1",  | ||||||
|  | # "l5", "ibm866", "cp866", "ms-kanji", "ibm850", "ecma-118", "iso-ir-101", "ibm819", "l1", "iso-8859-6:1987",  | ||||||
|  | # "latin5", "ascii", "sjis", "iso-8859-10", "iso-8859-4", "iso-8859-4:1988", "shift-jis | ||||||
|  | # The select charset will be converted to utf-8 when sent to other bridges. | ||||||
|  | #OPTIONAL (default "") | ||||||
|  | Charset="" | ||||||
|  |  | ||||||
| #Your nick on irc.  | #Your nick on irc.  | ||||||
| #REQUIRED | #REQUIRED | ||||||
| Nick="matterbot" | Nick="matterbot" | ||||||
|  |  | ||||||
| #If you registered your bot with a service like Nickserv on freenode.  | #If you registered your bot with a service like Nickserv on freenode.  | ||||||
| #Also being used when UseSASL=true | #Also being used when UseSASL=true | ||||||
|  | # | ||||||
|  | #Note: if you want do to quakenet auth, set NickServNick="Q@CServe.quakenet.org" | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| NickServNick="nickserv" | NickServNick="nickserv" | ||||||
| NickServPassword="secret" | NickServPassword="secret" | ||||||
|  |  | ||||||
|  | #OPTIONAL only used for quakenet auth | ||||||
|  | NickServUsername="username" | ||||||
|  |  | ||||||
| #Flood control | #Flood control | ||||||
| #Delay in milliseconds between each message send to the IRC server | #Delay in milliseconds between each message send to the IRC server | ||||||
| #OPTIONAL (default 1300) | #OPTIONAL (default 1300) | ||||||
| @@ -58,6 +80,15 @@ MessageQueue=30 | |||||||
| #OPTIONAL (default 400) | #OPTIONAL (default 400) | ||||||
| MessageLength=400 | MessageLength=400 | ||||||
|  |  | ||||||
|  | #Split messages on MessageLength instead of showing the <message clipped> | ||||||
|  | #WARNING: this could lead to flooding | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | MessageSplit=false | ||||||
|  |  | ||||||
|  | #Delay in seconds to rejoin a channel when kicked | ||||||
|  | #OPTIONAL (default 0) | ||||||
|  | RejoinDelay=0 | ||||||
|  |  | ||||||
| #Nicks you want to ignore.  | #Nicks you want to ignore.  | ||||||
| #Messages from those users will not be sent to other bridges. | #Messages from those users will not be sent to other bridges. | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| @@ -69,19 +100,51 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #The string "{NOPINGNICK}" (case sensitive) will be replaced by the actual nick / username, but with a ZWSP inside the nick, so the irc user with the same nick won't get pinged. See https://github.com/42wim/matterbridge/issues/175 for more information | #The string "{NOPINGNICK}" (case sensitive) will be replaced by the actual nick / username, but with a ZWSP inside the nick, so the irc user with the same nick won't get pinged. See https://github.com/42wim/matterbridge/issues/175 for more information | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #XMPP section | #XMPP section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -127,18 +190,49 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #Messages you want to replace. | ||||||
|  | #It replaces outgoing messages from the bridge. | ||||||
|  | #So you need to place it by the sending bridge definition. | ||||||
|  | #Regular expressions supported | ||||||
|  | #Some examples: | ||||||
|  | #This replaces cat => dog and sleep => awake | ||||||
|  | #ReplaceMessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #This Replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #ReplaceMessages=[ ["[0-9]","number"] ] | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #Nicks you want to replace. | ||||||
|  | #See ReplaceMessages for syntaxA | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #hipchat section | #hipchat section | ||||||
| @@ -177,18 +271,49 @@ IgnoreNicks="spammer1 spammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #mattermost section | #mattermost section | ||||||
| @@ -213,6 +338,11 @@ Team="yourteam" | |||||||
| Login="yourlogin" | Login="yourlogin" | ||||||
| Password="yourpass" | Password="yourpass" | ||||||
|  |  | ||||||
|  | #personal access token of the bot. | ||||||
|  | #new feature since mattermost 4.1. See https://docs.mattermost.com/developer/personal-access-tokens.html | ||||||
|  | #OPTIONAL (you can use token instead of login/password) | ||||||
|  | #Token="abcdefghijklm" | ||||||
|  |  | ||||||
| #Enable this to make a http connection (instead of https) to your mattermost.  | #Enable this to make a http connection (instead of https) to your mattermost.  | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| NoTLS=false | NoTLS=false | ||||||
| @@ -282,18 +412,50 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #Gitter section | #Gitter section | ||||||
| #Best to make a dedicated gitter account for the bot. | #Best to make a dedicated gitter account for the bot. | ||||||
| @@ -321,18 +483,50 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #slack section | #slack section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -361,7 +555,6 @@ WebhookURL="https://hooks.slack.com/services/yourhook" | |||||||
| #AND DEDICATED BOT USER WHEN POSSIBLE! | #AND DEDICATED BOT USER WHEN POSSIBLE! | ||||||
| #Address to listen on for outgoing webhook requests from slack | #Address to listen on for outgoing webhook requests from slack | ||||||
| #See account settings - integrations - outgoing webhooks on slack | #See account settings - integrations - outgoing webhooks on slack | ||||||
| #This setting will not be used when useAPI is eanbled |  | ||||||
| #webhooks | #webhooks | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| WebhookBindAddress="0.0.0.0:9999" | WebhookBindAddress="0.0.0.0:9999" | ||||||
| @@ -369,6 +562,7 @@ WebhookBindAddress="0.0.0.0:9999" | |||||||
| #Icon that will be showed in slack | #Icon that will be showed in slack | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL | #OPTIONAL | ||||||
| IconURL="https://robohash.org/{NICK}.png?size=48x48" | IconURL="https://robohash.org/{NICK}.png?size=48x48" | ||||||
| @@ -408,18 +602,50 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #discord section | #discord section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -443,7 +669,12 @@ Server="yourservername" | |||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowEmbeds=false | ShowEmbeds=false | ||||||
|  |  | ||||||
|  | #Shows the username (minus the discriminator) instead of the server nickname | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | UseUserName=false | ||||||
|  |  | ||||||
| #Specify WebhookURL. If given, will relay messages using the Webhook, which gives a better look to messages. | #Specify WebhookURL. If given, will relay messages using the Webhook, which gives a better look to messages. | ||||||
|  | #This only works if you have one discord channel, if you have multiple discord channels you'll have to specify it in the gateway config | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| WebhookURL="Yourwebhooktokenhere" | WebhookURL="Yourwebhooktokenhere" | ||||||
|  |  | ||||||
| @@ -466,18 +697,50 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #telegram section | #telegram section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -528,18 +791,49 @@ IgnoreNicks="spammer1 spammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #rocketchat section | #rocketchat section | ||||||
| @@ -592,18 +886,50 @@ IgnoreNicks="ircspammer1 ircspammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #matrix section | #matrix section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -647,18 +973,50 @@ IgnoreNicks="spammer1 spammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #steam section | #steam section | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -696,18 +1054,49 @@ IgnoreNicks="spammer1 spammer2" | |||||||
| #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | #OPTIONAL (example below ignores messages starting with ~~ or messages containing badword | ||||||
| IgnoreMessages="^~~ badword" | IgnoreMessages="^~~ badword" | ||||||
|  |  | ||||||
|  | #messages you want to replace. | ||||||
|  | #it replaces outgoing messages from the bridge. | ||||||
|  | #so you need to place it by the sending bridge definition. | ||||||
|  | #regular expressions supported | ||||||
|  | #some examples: | ||||||
|  | #this replaces cat => dog and sleep => awake | ||||||
|  | #replacemessages=[ ["cat","dog"], ["sleep","awake"] ] | ||||||
|  | #this replaces every number with number.  123 => numbernumbernumber | ||||||
|  | #replacemessages=[ ["[0-9]","number"] ] | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceMessages=[ ["cat","dog"] ] | ||||||
|  |  | ||||||
|  | #nicks you want to replace. | ||||||
|  | #see replacemessages for syntaxa | ||||||
|  | #optional (default empty) | ||||||
|  | ReplaceNicks=[ ["user--","user"] ] | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
| #Enable to show users joins/parts from other bridges  | #Enable to show users joins/parts from other bridges  | ||||||
| #Only works hiding/show messages from irc and mattermost bridge for now | #Currently works for messages from the following bridges: irc, mattermost, slack | ||||||
| #OPTIONAL (default false) | #OPTIONAL (default false) | ||||||
| ShowJoinPart=false | ShowJoinPart=false | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  | #Enable to show topic changes from other bridges  | ||||||
|  | #Only works hiding/show topic changes from slack bridge for now | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | ShowTopicChange=false | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #API | #API | ||||||
| @@ -730,9 +1119,14 @@ Buffer=1000 | |||||||
| #OPTIONAL (no authorization if token is empty) | #OPTIONAL (no authorization if token is empty) | ||||||
| Token="mytoken" | Token="mytoken" | ||||||
|  |  | ||||||
|  | #extra label that can be used in the RemoteNickFormat | ||||||
|  | #optional (default empty) | ||||||
|  | Label="" | ||||||
|  |  | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="{NICK}" | RemoteNickFormat="{NICK}" | ||||||
| @@ -742,15 +1136,45 @@ RemoteNickFormat="{NICK}" | |||||||
| ################################################################### | ################################################################### | ||||||
| #General configuration | #General configuration | ||||||
| ################################################################### | ################################################################### | ||||||
| #Settings here override specific settings for each protocol | # Settings here are defaults that each protocol can override | ||||||
| [general] | [general] | ||||||
| #RemoteNickFormat defines how remote users appear on this bridge  | #RemoteNickFormat defines how remote users appear on this bridge  | ||||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||||
|  | #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||||
| #OPTIONAL (default empty) | #OPTIONAL (default empty) | ||||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||||
|  |  | ||||||
|  | #StripNick only allows alphanumerical nicks. See https://github.com/42wim/matterbridge/issues/285 | ||||||
|  | #It will strip other characters from the nick | ||||||
|  | #OPTIONAL (default false) | ||||||
|  | StripNick=false | ||||||
|  |  | ||||||
|  |  | ||||||
|  | #MediaServerUpload and MediaServerDownload are used for uploading images/files/video to | ||||||
|  | #a remote "mediaserver" (a webserver like caddy for example). | ||||||
|  | #When configured images/files uploaded on bridges like mattermost,slack, telegram will be downloaded | ||||||
|  | #and uploaded again to MediaServerUpload URL | ||||||
|  | #The MediaServerDownload will be used so that bridges without native uploading support: | ||||||
|  | #gitter, irc and xmpp will be shown links to the files on MediaServerDownload | ||||||
|  | # | ||||||
|  | #More information https://github.com/42wim/matterbridge/wiki/Mediaserver-setup-%5Badvanced%5D | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | MediaServerUpload="https://user:pass@yourserver.com/upload" | ||||||
|  | #OPTIONAL (default empty) | ||||||
|  | MediaServerDownload="https://youserver.com/download" | ||||||
|  |  | ||||||
|  | #MediaDownloadSize is the maximum size of attachments, videos, images | ||||||
|  | #matterbridge will download and upload this file to bridges that also support uploading files. | ||||||
|  | #eg downloading from slack to upload it to mattermost | ||||||
|  | # | ||||||
|  | #It will only download from bridges that don't have public links available, which are for the moment | ||||||
|  | #slack, telegram, matrix and mattermost | ||||||
|  | # | ||||||
|  | #Optional (default 1000000 (1 megabyte)) | ||||||
|  | MediaDownloadSize=1000000 | ||||||
|  |  | ||||||
| ################################################################### | ################################################################### | ||||||
| #Gateway configuration | #Gateway configuration | ||||||
| ################################################################### | ################################################################### | ||||||
| @@ -786,7 +1210,7 @@ enable=true | |||||||
|     #mattermost - channel (the channel name as seen in the URL, not the displayname) |     #mattermost - channel (the channel name as seen in the URL, not the displayname) | ||||||
|     #gitter     - username/room  |     #gitter     - username/room  | ||||||
|     #xmpp       - channel |     #xmpp       - channel | ||||||
|     #slack      - channel (the channel name as seen in the URL, not the displayname) |     #slack      - channel (without the #) | ||||||
|     #discord    - channel (without the #) |     #discord    - channel (without the #) | ||||||
|     #           - ID:123456789 (where 123456789 is the channel ID)  |     #           - ID:123456789 (where 123456789 is the channel ID)  | ||||||
|     #               (https://github.com/42wim/matterbridge/issues/57) |     #               (https://github.com/42wim/matterbridge/issues/57) | ||||||
| @@ -829,6 +1253,14 @@ enable=true | |||||||
|         #OPTIONAL - your irc channel key |         #OPTIONAL - your irc channel key | ||||||
|         key="yourkey" |         key="yourkey" | ||||||
|  |  | ||||||
|  |     [[gateway.inout]] | ||||||
|  |     account="discord.game" | ||||||
|  |     channel="mygreatgame" | ||||||
|  |  | ||||||
|  |         #OPTIONAL - webhookurl only works for discord (it needs a different URL for each cahnnel) | ||||||
|  |         [gateway.inout.options] | ||||||
|  |         webhookurl=""https://discordapp.com/api/webhooks/123456789123456789/C9WPqExYWONPDZabcdef-def1434FGFjstasJX9pYht73y" | ||||||
|  |  | ||||||
|     #API example |     #API example | ||||||
|     #[[gateway.inout]] |     #[[gateway.inout]] | ||||||
|     #account="api.local" |     #account="api.local" | ||||||
|   | |||||||
| @@ -6,7 +6,6 @@ | |||||||
|  |  | ||||||
| [mattermost] | [mattermost] | ||||||
|     [mattermost.work] |     [mattermost.work] | ||||||
|     useAPI=true |  | ||||||
|     #do not prefix it wit http:// or https:// |     #do not prefix it wit http:// or https:// | ||||||
|     Server="yourmattermostserver.domain"  |     Server="yourmattermostserver.domain"  | ||||||
|     Team="yourteam" |     Team="yourteam" | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| package matterclient | package matterclient | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"crypto/md5" | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"errors" | 	"errors" | ||||||
| @@ -8,14 +9,15 @@ import ( | |||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"net/http/cookiejar" | 	"net/http/cookiejar" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strconv" |  | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	log "github.com/Sirupsen/logrus" | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	prefixed "github.com/x-cray/logrus-prefixed-formatter" | ||||||
|  |  | ||||||
| 	"github.com/gorilla/websocket" | 	"github.com/gorilla/websocket" | ||||||
|  | 	"github.com/hashicorp/golang-lru" | ||||||
| 	"github.com/jpillora/backoff" | 	"github.com/jpillora/backoff" | ||||||
| 	"github.com/mattermost/platform/model" | 	"github.com/mattermost/platform/model" | ||||||
| ) | ) | ||||||
| @@ -43,8 +45,8 @@ type Message struct { | |||||||
| type Team struct { | type Team struct { | ||||||
| 	Team         *model.Team | 	Team         *model.Team | ||||||
| 	Id           string | 	Id           string | ||||||
| 	Channels     *model.ChannelList | 	Channels     []*model.Channel | ||||||
| 	MoreChannels *model.ChannelList | 	MoreChannels []*model.Channel | ||||||
| 	Users        map[string]*model.User | 	Users        map[string]*model.User | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -53,7 +55,7 @@ type MMClient struct { | |||||||
| 	*Credentials | 	*Credentials | ||||||
| 	Team          *Team | 	Team          *Team | ||||||
| 	OtherTeams    []*Team | 	OtherTeams    []*Team | ||||||
| 	Client        *model.Client | 	Client        *model.Client4 | ||||||
| 	User          *model.User | 	User          *model.User | ||||||
| 	Users         map[string]*model.User | 	Users         map[string]*model.User | ||||||
| 	MessageChan   chan *Message | 	MessageChan   chan *Message | ||||||
| @@ -66,16 +68,22 @@ type MMClient struct { | |||||||
| 	WsPingChan    chan *model.WebSocketResponse | 	WsPingChan    chan *model.WebSocketResponse | ||||||
| 	ServerVersion string | 	ServerVersion string | ||||||
| 	OnWsConnect   func() | 	OnWsConnect   func() | ||||||
|  | 	lruCache      *lru.Cache | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(login, pass, team, server string) *MMClient { | func New(login, pass, team, server string) *MMClient { | ||||||
| 	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server} | 	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 := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | ||||||
| 	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"}) | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true}) | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | 	mmclient.log = log.WithFields(log.Fields{"prefix": "matterclient"}) | ||||||
|  | 	mmclient.lruCache, _ = lru.New(500) | ||||||
| 	return mmclient | 	return mmclient | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) SetDebugLog() { | ||||||
|  | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) SetLogLevel(level string) { | func (m *MMClient) SetLogLevel(level string) { | ||||||
| 	l, err := log.ParseLevel(level) | 	l, err := log.ParseLevel(level) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -88,7 +96,7 @@ func (m *MMClient) SetLogLevel(level string) { | |||||||
| func (m *MMClient) Login() error { | func (m *MMClient) Login() error { | ||||||
| 	// check if this is a first connect or a reconnection | 	// check if this is a first connect or a reconnection | ||||||
| 	firstConnection := true | 	firstConnection := true | ||||||
| 	if m.WsConnected == true { | 	if m.WsConnected { | ||||||
| 		firstConnection = false | 		firstConnection = false | ||||||
| 	} | 	} | ||||||
| 	m.WsConnected = false | 	m.WsConnected = false | ||||||
| @@ -105,18 +113,21 @@ func (m *MMClient) Login() error { | |||||||
| 		uriScheme = "http://" | 		uriScheme = "http://" | ||||||
| 	} | 	} | ||||||
| 	// login to mattermost | 	// login to mattermost | ||||||
| 	m.Client = model.NewClient(uriScheme + m.Credentials.Server) | 	m.Client = model.NewAPIv4Client(uriScheme + m.Credentials.Server) | ||||||
| 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | ||||||
| 	m.Client.HttpClient.Timeout = time.Second * 10 | 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
| 		d := b.Duration() | 		d := b.Duration() | ||||||
| 		// bogus call to get the serverversion | 		// bogus call to get the serverversion | ||||||
| 		m.Client.GetClientProperties() | 		_, resp := m.Client.Logout() | ||||||
| 		if firstConnection && !supportedVersion(m.Client.ServerVersion) { | 		if resp.Error != nil { | ||||||
| 			return fmt.Errorf("unsupported mattermost version: %s", m.Client.ServerVersion) | 			return fmt.Errorf("%#v", resp.Error.Error()) | ||||||
| 		} | 		} | ||||||
| 		m.ServerVersion = m.Client.ServerVersion | 		if firstConnection && !supportedVersion(resp.ServerVersion) { | ||||||
|  | 			return fmt.Errorf("unsupported mattermost version: %s", resp.ServerVersion) | ||||||
|  | 		} | ||||||
|  | 		m.ServerVersion = resp.ServerVersion | ||||||
| 		if m.ServerVersion == "" { | 		if m.ServerVersion == "" { | ||||||
| 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||||
| 			time.Sleep(d) | 			time.Sleep(d) | ||||||
| @@ -127,30 +138,33 @@ func (m *MMClient) Login() error { | |||||||
| 	} | 	} | ||||||
| 	b.Reset() | 	b.Reset() | ||||||
|  |  | ||||||
| 	var myinfo *model.Result | 	var resp *model.Response | ||||||
|  | 	//var myinfo *model.Result | ||||||
| 	var appErr *model.AppError | 	var appErr *model.AppError | ||||||
| 	var logmsg = "trying login" | 	var logmsg = "trying login" | ||||||
| 	for { | 	for { | ||||||
| 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | ||||||
| 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||||
| 			m.log.Debugf(logmsg+" with %s", model.SESSION_COOKIE_TOKEN) | 			m.log.Debugf(logmsg + " with token") | ||||||
| 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||||
| 			if len(token) != 2 { | 			if len(token) != 2 { | ||||||
| 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||||
| 			} | 			} | ||||||
| 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | ||||||
| 			m.Client.MockSession(token[1]) | 			m.Client.AuthToken = token[1] | ||||||
| 			myinfo, appErr = m.Client.GetMe("") | 			m.Client.AuthType = model.HEADER_BEARER | ||||||
| 			if appErr != nil { | 			m.User, resp = m.Client.GetMe("") | ||||||
| 				return errors.New(appErr.DetailedError) | 			if resp.Error != nil { | ||||||
|  | 				return resp.Error | ||||||
| 			} | 			} | ||||||
| 			if myinfo.Data.(*model.User) == nil { | 			if m.User == nil { | ||||||
| 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | ||||||
| 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | ||||||
| 			} | 			} | ||||||
| 		} else { | 		} else { | ||||||
| 			myinfo, appErr = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | 			m.User, resp = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | ||||||
| 		} | 		} | ||||||
|  | 		appErr = resp.Error | ||||||
| 		if appErr != nil { | 		if appErr != nil { | ||||||
| 			d := b.Duration() | 			d := b.Duration() | ||||||
| 			m.log.Debug(appErr.DetailedError) | 			m.log.Debug(appErr.DetailedError) | ||||||
| @@ -178,8 +192,6 @@ func (m *MMClient) Login() error { | |||||||
| 	if m.Team == nil { | 	if m.Team == nil { | ||||||
| 		return errors.New("team not found") | 		return errors.New("team not found") | ||||||
| 	} | 	} | ||||||
| 	// set our team id as default route |  | ||||||
| 	m.Client.SetTeamId(m.Team.Id) |  | ||||||
|  |  | ||||||
| 	m.wsConnect() | 	m.wsConnect() | ||||||
|  |  | ||||||
| @@ -200,7 +212,7 @@ func (m *MMClient) wsConnect() { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// setup websocket connection | 	// setup websocket connection | ||||||
| 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V3 + "/users/websocket" | 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V4 + "/websocket" | ||||||
| 	header := http.Header{} | 	header := http.Header{} | ||||||
| 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||||
|  |  | ||||||
| @@ -234,9 +246,9 @@ func (m *MMClient) Logout() error { | |||||||
| 		m.log.Debug("Not invalidating session in logout, credential is a token") | 		m.log.Debug("Not invalidating session in logout, credential is a token") | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	_, err := m.Client.Logout() | 	_, resp := m.Client.Logout() | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return err | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -267,9 +279,19 @@ func (m *MMClient) WsReceiver() { | |||||||
| 			m.log.Debugf("WsReceiver event: %#v", event) | 			m.log.Debugf("WsReceiver event: %#v", event) | ||||||
| 			msg := &Message{Raw: &event, Team: m.Credentials.Team} | 			msg := &Message{Raw: &event, Team: m.Credentials.Team} | ||||||
| 			m.parseMessage(msg) | 			m.parseMessage(msg) | ||||||
|  | 			// check if we didn't empty the message | ||||||
|  | 			if msg.Text != "" { | ||||||
| 				m.MessageChan <- msg | 				m.MessageChan <- msg | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
|  | 			// if we have file attached but the message is empty, also send it | ||||||
|  | 			if msg.Post != nil { | ||||||
|  | 				if msg.Text != "" || len(msg.Post.FileIds) > 0 || msg.Post.Type == "slack_attachment" { | ||||||
|  | 					m.MessageChan <- msg | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		var response model.WebSocketResponse | 		var response model.WebSocketResponse | ||||||
| 		if err := json.Unmarshal(rawMsg, &response); err == nil && response.IsValid() { | 		if err := json.Unmarshal(rawMsg, &response); err == nil && response.IsValid() { | ||||||
| @@ -282,7 +304,7 @@ func (m *MMClient) WsReceiver() { | |||||||
|  |  | ||||||
| func (m *MMClient) parseMessage(rmsg *Message) { | func (m *MMClient) parseMessage(rmsg *Message) { | ||||||
| 	switch rmsg.Raw.Event { | 	switch rmsg.Raw.Event { | ||||||
| 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED: | 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED, model.WEBSOCKET_EVENT_POST_DELETED: | ||||||
| 		m.parseActionPost(rmsg) | 		m.parseActionPost(rmsg) | ||||||
| 		/* | 		/* | ||||||
| 			case model.ACTION_USER_REMOVED: | 			case model.ACTION_USER_REMOVED: | ||||||
| @@ -303,6 +325,13 @@ func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) parseActionPost(rmsg *Message) { | func (m *MMClient) parseActionPost(rmsg *Message) { | ||||||
|  | 	// add post to cache, if it already exists don't relay this again. | ||||||
|  | 	// this should fix reposts | ||||||
|  | 	if ok, _ := m.lruCache.ContainsOrAdd(digestString(rmsg.Raw.Data["post"].(string)), true); ok { | ||||||
|  | 		m.log.Debugf("message %#v in cache, not processing again", rmsg.Raw.Data["post"].(string)) | ||||||
|  | 		rmsg.Text = "" | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
| 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | ||||||
| 	// we don't have the user, refresh the userlist | 	// we don't have the user, refresh the userlist | ||||||
| 	if m.GetUser(data.UserId) == nil { | 	if m.GetUser(data.UserId) == nil { | ||||||
| @@ -329,37 +358,37 @@ func (m *MMClient) parseActionPost(rmsg *Message) { | |||||||
| 	} | 	} | ||||||
| 	rmsg.Text = data.Message | 	rmsg.Text = data.Message | ||||||
| 	rmsg.Post = data | 	rmsg.Post = data | ||||||
| 	return |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateUsers() error { | func (m *MMClient) UpdateUsers() error { | ||||||
| 	mmusers, err := m.Client.GetProfiles(0, 50000, "") | 	mmusers, resp := m.Client.GetUsers(0, 50000, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Users = mmusers.Data.(map[string]*model.User) | 	for _, user := range mmusers { | ||||||
|  | 		m.Users[user.Id] = user | ||||||
|  | 	} | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannels() error { | func (m *MMClient) UpdateChannels() error { | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} |  | ||||||
| 	var mmchannels2 *model.Result |  | ||||||
| 	if m.mmVersion() >= 3.08 { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannelsPage(0, 5000) |  | ||||||
| 	} else { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannels("") |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return errors.New(err.DetailedError) |  | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) | 	m.Team.Channels = mmchannels | ||||||
| 	m.Team.MoreChannels = mmchannels2.Data.(*model.ChannelList) | 	m.Unlock() | ||||||
|  |  | ||||||
|  | 	mmchannels, resp = m.Client.GetPublicChannelsForTeam(m.Team.Id, 0, 5000, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return errors.New(resp.Error.DetailedError) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	m.Team.MoreChannels = mmchannels | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -372,14 +401,14 @@ func (m *MMClient) GetChannelName(channelId string) string { | |||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		if t.Channels != nil { | 		if t.Channels != nil { | ||||||
| 			for _, channel := range *t.Channels { | 			for _, channel := range t.Channels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			for _, channel := range *t.MoreChannels { | 			for _, channel := range t.MoreChannels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| @@ -397,7 +426,7 @@ func (m *MMClient) GetChannelId(name string, teamId string) string { | |||||||
| 	} | 	} | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id == teamId { | 		if t.Id == teamId { | ||||||
| 			for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 				if channel.Name == name { | 				if channel.Name == name { | ||||||
| 					return channel.Id | 					return channel.Id | ||||||
| 				} | 				} | ||||||
| @@ -411,7 +440,7 @@ func (m *MMClient) GetChannelTeamId(id string) string { | |||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range append(m.OtherTeams, m.Team) { | 	for _, t := range append(m.OtherTeams, m.Team) { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == id { | 			if channel.Id == id { | ||||||
| 				return channel.TeamId | 				return channel.TeamId | ||||||
| 			} | 			} | ||||||
| @@ -424,7 +453,7 @@ func (m *MMClient) GetChannelHeader(channelId string) string { | |||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == channelId { | 			if channel.Id == channelId { | ||||||
| 				return channel.Header | 				return channel.Header | ||||||
| 			} | 			} | ||||||
| @@ -434,55 +463,85 @@ func (m *MMClient) GetChannelHeader(channelId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) PostMessage(channelId string, text string) { | func (m *MMClient) PostMessage(channelId string, text string) (string, error) { | ||||||
| 	post := &model.Post{ChannelId: channelId, Message: text} | 	post := &model.Post{ChannelId: channelId, Message: text} | ||||||
| 	m.Client.CreatePost(post) | 	res, resp := m.Client.CreatePost(post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) PostMessageWithFiles(channelId string, text string, fileIds []string) (string, error) { | ||||||
|  | 	post := &model.Post{ChannelId: channelId, Message: text, FileIds: fileIds} | ||||||
|  | 	res, resp := m.Client.CreatePost(post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) EditMessage(postId string, text string) (string, error) { | ||||||
|  | 	post := &model.Post{Message: text} | ||||||
|  | 	res, resp := m.Client.UpdatePost(postId, post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) DeleteMessage(postId string) error { | ||||||
|  | 	_, resp := m.Client.DeletePost(postId) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) JoinChannel(channelId string) error { | func (m *MMClient) JoinChannel(channelId string) error { | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, c := range *m.Team.Channels { | 	for _, c := range m.Team.Channels { | ||||||
| 		if c.Id == channelId { | 		if c.Id == channelId { | ||||||
| 			m.log.Debug("Not joining ", channelId, " already joined.") | 			m.log.Debug("Not joining ", channelId, " already joined.") | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	m.log.Debug("Joining ", channelId) | 	m.log.Debug("Joining ", channelId) | ||||||
| 	_, err := m.Client.JoinChannel(channelId) | 	_, resp := m.Client.AddChannelMember(channelId, m.User.Id) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New("failed to join") | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | ||||||
| 	res, err := m.Client.GetPostsSince(channelId, time) | 	res, resp := m.Client.GetPostsSince(channelId, time) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) SearchPosts(query string) *model.PostList { | func (m *MMClient) SearchPosts(query string) *model.PostList { | ||||||
| 	res, err := m.Client.SearchPosts(query, false) | 	res, resp := m.Client.SearchPosts(m.Team.Id, query, false) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | ||||||
| 	res, err := m.Client.GetPosts(channelId, 0, limit, "") | 	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPublicLink(filename string) string { | func (m *MMClient) GetPublicLink(filename string) string { | ||||||
| 	res, err := m.Client.GetPublicLink(filename) | 	res, resp := m.Client.GetFileLink(filename) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| @@ -491,8 +550,27 @@ func (m *MMClient) GetPublicLink(filename string) string { | |||||||
| func (m *MMClient) GetPublicLinks(filenames []string) []string { | func (m *MMClient) GetPublicLinks(filenames []string) []string { | ||||||
| 	var output []string | 	var output []string | ||||||
| 	for _, f := range filenames { | 	for _, f := range filenames { | ||||||
| 		res, err := m.Client.GetPublicLink(f) | 		res, resp := m.Client.GetFileLink(f) | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		output = append(output, res) | ||||||
|  | 	} | ||||||
|  | 	return output | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) GetFileLinks(filenames []string) []string { | ||||||
|  | 	uriScheme := "https://" | ||||||
|  | 	if m.NoTLS { | ||||||
|  | 		uriScheme = "http://" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var output []string | ||||||
|  | 	for _, f := range filenames { | ||||||
|  | 		res, resp := m.Client.GetFileLink(f) | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			// public links is probably disabled, create the link ourselves | ||||||
|  | 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V3+"/files/"+f+"/get") | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		output = append(output, res) | 		output = append(output, res) | ||||||
| @@ -501,42 +579,43 @@ func (m *MMClient) GetPublicLinks(filenames []string) []string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | ||||||
| 	data := make(map[string]string) | 	channel := &model.Channel{Id: channelId, Header: header} | ||||||
| 	data["channel_id"] = channelId |  | ||||||
| 	data["channel_header"] = header |  | ||||||
| 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||||
| 	_, err := m.Client.UpdateChannelHeader(data) | 	_, resp := m.Client.UpdateChannel(channel) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		log.Error(err) | 		log.Error(resp.Error) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateLastViewed(channelId string) { | func (m *MMClient) UpdateLastViewed(channelId string) { | ||||||
| 	m.log.Debugf("posting lastview %#v", channelId) | 	m.log.Debugf("posting lastview %#v", channelId) | ||||||
| 	if m.mmVersion() >= 3.08 { | 	view := &model.ChannelView{ChannelId: channelId} | ||||||
| 		view := model.ChannelView{ChannelId: channelId} | 	_, resp := m.Client.ViewChannel(m.User.Id, view) | ||||||
| 		res, _ := m.Client.ViewChannel(view) | 	if resp.Error != nil { | ||||||
| 		if res == false { | 		m.log.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error) | ||||||
| 			m.log.Errorf("ChannelView update for %s failed", channelId) |  | ||||||
| 		} |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	_, err := m.Client.UpdateLastViewedAt(channelId, true) |  | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Error(err) |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UpdateUserNick(nick string) error { | ||||||
|  | 	user := m.User | ||||||
|  | 	user.Nickname = nick | ||||||
|  | 	_, resp := m.Client.UpdateUser(user) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) UsernamesInChannel(channelId string) []string { | func (m *MMClient) UsernamesInChannel(channelId string) []string { | ||||||
| 	res, err := m.Client.GetProfilesInChannel(channelId, 0, 50000, "") | 	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, err) | 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error) | ||||||
| 		return []string{} | 		return []string{} | ||||||
| 	} | 	} | ||||||
| 	members := res.Data.(map[string]*model.User) | 	allusers := m.GetUsers() | ||||||
| 	result := []string{} | 	result := []string{} | ||||||
| 	for _, member := range members { | 	for _, member := range *res { | ||||||
| 		result = append(result, member.Nickname) | 		result = append(result, allusers[member.UserId].Nickname) | ||||||
| 	} | 	} | ||||||
| 	return result | 	return result | ||||||
| } | } | ||||||
| @@ -560,22 +639,15 @@ func (m *MMClient) createCookieJar(token string) *cookiejar.Jar { | |||||||
| func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | ||||||
| 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||||
| 	// create DM channel (only happens on first message) | 	// create DM channel (only happens on first message) | ||||||
| 	_, err := m.Client.CreateDirectChannel(toUserId) | 	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err) | 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||||
|  |  | ||||||
| 	// update our channels | 	// update our channels | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	m.UpdateChannels() | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Debug("SendDirectMessage: Couldn't update channels") |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	// build & send the message | 	// build & send the message | ||||||
| 	msg = strings.Replace(msg, "\r", "", -1) | 	msg = strings.Replace(msg, "\r", "", -1) | ||||||
| @@ -601,10 +673,10 @@ func (m *MMClient) GetChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	// our primary team channels first | 	// our primary team channels first | ||||||
| 	channels = append(channels, *m.Team.Channels...) | 	channels = append(channels, m.Team.Channels...) | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id != m.Team.Id { | 		if t.Id != m.Team.Id { | ||||||
| 			channels = append(channels, *t.Channels...) | 			channels = append(channels, t.Channels...) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| @@ -616,7 +688,7 @@ func (m *MMClient) GetMoreChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.MoreChannels...) | 		channels = append(channels, t.MoreChannels...) | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
| @@ -627,9 +699,9 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.Channels...) | 		channels = append(channels, t.Channels...) | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			channels = append(channels, *t.MoreChannels...) | 			channels = append(channels, t.MoreChannels...) | ||||||
| 		} | 		} | ||||||
| 		for _, c := range channels { | 		for _, c := range channels { | ||||||
| 			if c.Id == channelId { | 			if c.Id == channelId { | ||||||
| @@ -643,12 +715,11 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string { | |||||||
| func (m *MMClient) GetLastViewedAt(channelId string) int64 { | func (m *MMClient) GetLastViewedAt(channelId string) int64 { | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	res, err := m.Client.GetChannel(channelId, "") | 	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return model.GetMillis() | 		return model.GetMillis() | ||||||
| 	} | 	} | ||||||
| 	data := res.Data.(*model.ChannelData) | 	return res.LastViewedAt | ||||||
| 	return data.Member.LastViewedAt |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetUsers() map[string]*model.User { | func (m *MMClient) GetUsers() map[string]*model.User { | ||||||
| @@ -664,14 +735,13 @@ func (m *MMClient) GetUsers() map[string]*model.User { | |||||||
| func (m *MMClient) GetUser(userId string) *model.User { | func (m *MMClient) GetUser(userId string) *model.User { | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	u, ok := m.Users[userId] | 	_, ok := m.Users[userId] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		res, err := m.Client.GetProfilesByIds([]string{userId}) | 		res, resp := m.Client.GetUser(userId, "") | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 		u = res.Data.(map[string]*model.User)[userId] | 		m.Users[userId] = res | ||||||
| 		m.Users[userId] = u |  | ||||||
| 	} | 	} | ||||||
| 	return m.Users[userId] | 	return m.Users[userId] | ||||||
| } | } | ||||||
| @@ -685,36 +755,36 @@ func (m *MMClient) GetUserName(userId string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetStatus(userId string) string { | func (m *MMClient) GetStatus(userId string) string { | ||||||
| 	res, err := m.Client.GetStatusesByIds([]string{userId}) | 	res, resp := m.Client.GetUserStatus(userId, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	status := res.Data.(map[string]string) | 	if res.Status == model.STATUS_AWAY { | ||||||
| 	if status[userId] == model.STATUS_AWAY { |  | ||||||
| 		return "away" | 		return "away" | ||||||
| 	} | 	} | ||||||
| 	if status[userId] == model.STATUS_ONLINE { | 	if res.Status == model.STATUS_ONLINE { | ||||||
| 		return "online" | 		return "online" | ||||||
| 	} | 	} | ||||||
| 	return "offline" | 	return "offline" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetStatuses() map[string]string { | func (m *MMClient) GetStatuses() map[string]string { | ||||||
| 	var ok bool | 	var ids []string | ||||||
| 	statuses := make(map[string]string) | 	statuses := make(map[string]string) | ||||||
| 	res, err := m.Client.GetStatuses() | 	for id := range m.Users { | ||||||
| 	if err != nil { | 		ids = append(ids, id) | ||||||
|  | 	} | ||||||
|  | 	res, resp := m.Client.GetUsersStatusesByIds(ids) | ||||||
|  | 	if resp.Error != nil { | ||||||
| 		return statuses | 		return statuses | ||||||
| 	} | 	} | ||||||
| 	if statuses, ok = res.Data.(map[string]string); ok { | 	for _, status := range res { | ||||||
| 		for userId, status := range statuses { | 		statuses[status.UserId] = "offline" | ||||||
| 			statuses[userId] = "offline" | 		if status.Status == model.STATUS_AWAY { | ||||||
| 			if status == model.STATUS_AWAY { | 			statuses[status.UserId] = "away" | ||||||
| 				statuses[userId] = "away" |  | ||||||
| 			} |  | ||||||
| 			if status == model.STATUS_ONLINE { |  | ||||||
| 				statuses[userId] = "online" |  | ||||||
| 		} | 		} | ||||||
|  | 		if status.Status == model.STATUS_ONLINE { | ||||||
|  | 			statuses[status.UserId] = "online" | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return statuses | 	return statuses | ||||||
| @@ -724,6 +794,14 @@ func (m *MMClient) GetTeamId() string { | |||||||
| 	return m.Team.Id | 	return m.Team.Id | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { | ||||||
|  | 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return f.FileInfos[0].Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) StatusLoop() { | func (m *MMClient) StatusLoop() { | ||||||
| 	retries := 0 | 	retries := 0 | ||||||
| 	backoff := time.Second * 60 | 	backoff := time.Second * 60 | ||||||
| @@ -744,9 +822,14 @@ func (m *MMClient) StatusLoop() { | |||||||
| 				backoff = time.Second * 60 | 				backoff = time.Second * 60 | ||||||
| 			case <-time.After(time.Second * 5): | 			case <-time.After(time.Second * 5): | ||||||
| 				if retries > 3 { | 				if retries > 3 { | ||||||
|  | 					m.log.Debug("StatusLoop() timeout") | ||||||
| 					m.Logout() | 					m.Logout() | ||||||
| 					m.WsQuit = false | 					m.WsQuit = false | ||||||
| 					m.Login() | 					err := m.Login() | ||||||
|  | 					if err != nil { | ||||||
|  | 						log.Errorf("Login failed: %#v", err) | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
| 					if m.OnWsConnect != nil { | 					if m.OnWsConnect != nil { | ||||||
| 						m.OnWsConnect() | 						m.OnWsConnect() | ||||||
| 					} | 					} | ||||||
| @@ -765,40 +848,39 @@ func (m *MMClient) StatusLoop() { | |||||||
| func (m *MMClient) initUser() error { | func (m *MMClient) initUser() error { | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	initLoad, err := m.Client.GetInitialLoad() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	initData := initLoad.Data.(*model.InitialLoad) |  | ||||||
| 	m.User = initData.User |  | ||||||
| 	// we only load all team data on initial login. | 	// we only load all team data on initial login. | ||||||
| 	// all other updates are for channels from our (primary) team only. | 	// all other updates are for channels from our (primary) team only. | ||||||
| 	//m.log.Debug("initUser(): loading all team data") | 	//m.log.Debug("initUser(): loading all team data") | ||||||
| 	for _, v := range initData.Teams { | 	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "") | ||||||
| 		m.Client.SetTeamId(v.Id) | 	if resp.Error != nil { | ||||||
| 		mmusers, err := m.Client.GetProfiles(0, 50000, "") | 		return resp.Error | ||||||
| 		if err != nil { |  | ||||||
| 			return errors.New(err.DetailedError) |  | ||||||
| 	} | 	} | ||||||
| 		t := &Team{Team: v, Users: mmusers.Data.(map[string]*model.User), Id: v.Id} | 	for _, team := range teams { | ||||||
| 		mmchannels, err := m.Client.GetChannels("") | 		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "") | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return errors.New(err.DetailedError) | 			return errors.New(resp.Error.DetailedError) | ||||||
| 		} | 		} | ||||||
| 		t.Channels = mmchannels.Data.(*model.ChannelList) | 		usermap := make(map[string]*model.User) | ||||||
| 		if m.mmVersion() >= 3.08 { | 		for _, user := range mmusers { | ||||||
| 			mmchannels, err = m.Client.GetMoreChannelsPage(0, 5000) | 			usermap[user.Id] = user | ||||||
| 		} else { |  | ||||||
| 			mmchannels, err = m.Client.GetMoreChannels("") |  | ||||||
| 		} | 		} | ||||||
| 		if err != nil { |  | ||||||
| 			return errors.New(err.DetailedError) | 		t := &Team{Team: team, Users: usermap, Id: team.Id} | ||||||
|  |  | ||||||
|  | 		mmchannels, resp := m.Client.GetChannelsForTeamForUser(team.Id, m.User.Id, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
| 		} | 		} | ||||||
| 		t.MoreChannels = mmchannels.Data.(*model.ChannelList) | 		t.Channels = mmchannels | ||||||
|  | 		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
|  | 		} | ||||||
|  | 		t.MoreChannels = mmchannels | ||||||
| 		m.OtherTeams = append(m.OtherTeams, t) | 		m.OtherTeams = append(m.OtherTeams, t) | ||||||
| 		if v.Name == m.Credentials.Team { | 		if team.Name == m.Credentials.Team { | ||||||
| 			m.Team = t | 			m.Team = t | ||||||
| 			m.log.Debugf("initUser(): found our team %s (id: %s)", v.Name, v.Id) | 			m.log.Debugf("initUser(): found our team %s (id: %s)", team.Name, team.Id) | ||||||
| 		} | 		} | ||||||
| 		// add all users | 		// add all users | ||||||
| 		for k, v := range t.Users { | 		for k, v := range t.Users { | ||||||
| @@ -819,22 +901,16 @@ func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) err | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) mmVersion() float64 { |  | ||||||
| 	v, _ := strconv.ParseFloat(string(m.ServerVersion[0:2])+"0"+string(m.ServerVersion[2]), 64) |  | ||||||
| 	if string(m.ServerVersion[4]) == "." { |  | ||||||
| 		v, _ = strconv.ParseFloat(m.ServerVersion[0:4], 64) |  | ||||||
| 	} |  | ||||||
| 	return v |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func supportedVersion(version string) bool { | func supportedVersion(version string) bool { | ||||||
| 	if strings.HasPrefix(version, "3.5.0") || | 	if strings.HasPrefix(version, "3.8.0") || | ||||||
| 		strings.HasPrefix(version, "3.6.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.7.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.8.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.9.0") || | 		strings.HasPrefix(version, "3.9.0") || | ||||||
| 		strings.HasPrefix(version, "3.10.0") { | 		strings.HasPrefix(version, "3.10.0") || | ||||||
|  | 		strings.HasPrefix(version, "4.") { | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| 	return false | 	return false | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func digestString(s string) string { | ||||||
|  | 	return fmt.Sprintf("%x", md5.Sum([]byte(s))) | ||||||
|  | } | ||||||
|   | |||||||
| @@ -24,6 +24,7 @@ type OMessage struct { | |||||||
| 	Text        string                 `json:"text"` | 	Text        string                 `json:"text"` | ||||||
| 	Attachments interface{}            `json:"attachments,omitempty"` | 	Attachments interface{}            `json:"attachments,omitempty"` | ||||||
| 	Type        string                 `json:"type,omitempty"` | 	Type        string                 `json:"type,omitempty"` | ||||||
|  | 	Props       map[string]interface{} `json:"props"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // IMessage for mattermost outgoing webhook. (received from mattermost) | // IMessage for mattermost outgoing webhook. (received from mattermost) | ||||||
| @@ -43,6 +44,7 @@ type IMessage struct { | |||||||
| 	ServiceId   string `schema:"service_id"` | 	ServiceId   string `schema:"service_id"` | ||||||
| 	Text        string `schema:"text"` | 	Text        string `schema:"text"` | ||||||
| 	TriggerWord string `schema:"trigger_word"` | 	TriggerWord string `schema:"trigger_word"` | ||||||
|  | 	FileIDs     string `schema:"file_ids"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Client for Mattermost. | // Client for Mattermost. | ||||||
| @@ -134,12 +136,11 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) { | |||||||
|  |  | ||||||
| // Receive returns an incoming message from mattermost outgoing webhooks URL. | // Receive returns an incoming message from mattermost outgoing webhooks URL. | ||||||
| func (c *Client) Receive() IMessage { | func (c *Client) Receive() IMessage { | ||||||
| 	for { | 	var msg IMessage | ||||||
| 		select { | 	for msg := range c.In { | ||||||
| 		case msg := <-c.In: |  | ||||||
| 		return msg | 		return msg | ||||||
| 	} | 	} | ||||||
| 	} | 	return msg | ||||||
| } | } | ||||||
|  |  | ||||||
| // Send sends a msg to mattermost incoming webhooks URL. | // Send sends a msg to mattermost incoming webhooks URL. | ||||||
|   | |||||||
							
								
								
									
										50
									
								
								migration.md
									
									
									
									
									
								
							
							
						
						
									
										50
									
								
								migration.md
									
									
									
									
									
								
							| @@ -1,50 +0,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" |  | ||||||
| ``` |  | ||||||
| @@ -205,17 +205,43 @@ func (gitter *Gitter) GetMessage(roomID, messageID string) (*Message, error) { | |||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // SendMessage sends a message to a room | // SendMessage sends a message to a room | ||||||
| func (gitter *Gitter) SendMessage(roomID, text string) error { | func (gitter *Gitter) SendMessage(roomID, text string) (*Message, error) { | ||||||
| 
 | 
 | ||||||
| 	message := Message{Text: text} | 	message := Message{Text: text} | ||||||
| 	body, _ := json.Marshal(message) | 	body, _ := json.Marshal(message) | ||||||
| 	_, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | 	response, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		gitter.log(err) | 		gitter.log(err) | ||||||
| 		return err | 		return nil, err | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	return nil | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // UpdateMessage updates a message in a room | ||||||
|  | func (gitter *Gitter) UpdateMessage(roomID, msgID, text string) (*Message, error) { | ||||||
|  | 
 | ||||||
|  | 	message := Message{Text: text} | ||||||
|  | 	body, _ := json.Marshal(message) | ||||||
|  | 	response, err := gitter.put(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages/"+msgID, body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // JoinRoom joins a room | // JoinRoom joins a room | ||||||
| @@ -414,6 +440,39 @@ func (gitter *Gitter) post(url string, body []byte) ([]byte, error) { | |||||||
| 	return result, nil | 	return result, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | func (gitter *Gitter) put(url string, body []byte) ([]byte, error) { | ||||||
|  | 	r, err := http.NewRequest("PUT", url, bytes.NewBuffer(body)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	r.Header.Set("Content-Type", "application/json") | ||||||
|  | 	r.Header.Set("Accept", "application/json") | ||||||
|  | 	r.Header.Set("Authorization", "Bearer "+gitter.config.token) | ||||||
|  | 
 | ||||||
|  | 	resp, err := gitter.config.client.Do(r) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer resp.Body.Close() | ||||||
|  | 
 | ||||||
|  | 	if resp.StatusCode != http.StatusOK { | ||||||
|  | 		err = APIError{What: fmt.Sprintf("Status code: %v", resp.StatusCode)} | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	result, err := ioutil.ReadAll(resp.Body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return result, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| func (gitter *Gitter) delete(url string) ([]byte, error) { | func (gitter *Gitter) delete(url string) ([]byte, error) { | ||||||
| 	r, err := http.NewRequest("delete", url, nil) | 	r, err := http.NewRequest("delete", url, nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
							
								
								
									
										117
									
								
								vendor/github.com/thoj/go-ircevent/irc.go → vendor/github.com/42wim/go-ircevent/irc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										117
									
								
								vendor/github.com/thoj/go-ircevent/irc.go → vendor/github.com/42wim/go-ircevent/irc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -87,6 +87,17 @@ func (irc *Connection) readLoop() { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // Unescape tag values as defined in the IRCv3.2 message tags spec | ||||||
|  | // http://ircv3.net/specs/core/message-tags-3.2.html | ||||||
|  | func unescapeTagValue(value string) string { | ||||||
|  | 	value = strings.Replace(value, "\\:", ";", -1) | ||||||
|  | 	value = strings.Replace(value, "\\s", " ", -1) | ||||||
|  | 	value = strings.Replace(value, "\\\\", "\\", -1) | ||||||
|  | 	value = strings.Replace(value, "\\r", "\r", -1) | ||||||
|  | 	value = strings.Replace(value, "\\n", "\n", -1) | ||||||
|  | 	return value | ||||||
|  | } | ||||||
|  | 
 | ||||||
| //Parse raw irc messages | //Parse raw irc messages | ||||||
| func parseToEvent(msg string) (*Event, error) { | func parseToEvent(msg string) (*Event, error) { | ||||||
| 	msg = strings.TrimSuffix(msg, "\n") //Remove \r\n | 	msg = strings.TrimSuffix(msg, "\n") //Remove \r\n | ||||||
| @@ -95,6 +106,26 @@ func parseToEvent(msg string) (*Event, error) { | |||||||
| 	if len(msg) < 5 { | 	if len(msg) < 5 { | ||||||
| 		return nil, errors.New("Malformed msg from server") | 		return nil, errors.New("Malformed msg from server") | ||||||
| 	} | 	} | ||||||
|  | 
 | ||||||
|  | 	if msg[0] == '@' { | ||||||
|  | 		// IRCv3 Message Tags | ||||||
|  | 		if i := strings.Index(msg, " "); i > -1 { | ||||||
|  | 			event.Tags = make(map[string]string) | ||||||
|  | 			tags := strings.Split(msg[1:i], ";") | ||||||
|  | 			for _, data := range tags { | ||||||
|  | 				parts := strings.SplitN(data, "=", 2) | ||||||
|  | 				if len(parts) == 1 { | ||||||
|  | 					event.Tags[parts[0]] = "" | ||||||
|  | 				} else { | ||||||
|  | 					event.Tags[parts[0]] = unescapeTagValue(parts[1]) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			msg = msg[i+1 : len(msg)] | ||||||
|  | 		} else { | ||||||
|  | 			return nil, errors.New("Malformed msg from server") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	if msg[0] == ':' { | 	if msg[0] == ':' { | ||||||
| 		if i := strings.Index(msg, " "); i > -1 { | 		if i := strings.Index(msg, " "); i > -1 { | ||||||
| 			event.Source = msg[1:i] | 			event.Source = msg[1:i] | ||||||
| @@ -196,12 +227,17 @@ func (irc *Connection) isQuitting() bool { | |||||||
| // Main loop to control the connection. | // Main loop to control the connection. | ||||||
| func (irc *Connection) Loop() { | func (irc *Connection) Loop() { | ||||||
| 	errChan := irc.ErrorChan() | 	errChan := irc.ErrorChan() | ||||||
|  | 	connTime := time.Now() | ||||||
| 	for !irc.isQuitting() { | 	for !irc.isQuitting() { | ||||||
| 		err := <-errChan | 		err := <-errChan | ||||||
| 		close(irc.end) | 		close(irc.end) | ||||||
| 		irc.Wait() | 		irc.Wait() | ||||||
| 		for !irc.isQuitting() { | 		for !irc.isQuitting() { | ||||||
| 			irc.Log.Printf("Error, disconnected: %s\n", err) | 			irc.Log.Printf("Error, disconnected: %s\n", err) | ||||||
|  | 			if time.Now().Sub(connTime) < time.Second*5 { | ||||||
|  | 				irc.Log.Println("Rreconnecting too fast, sleeping 60 seconds") | ||||||
|  | 				time.Sleep(60 * time.Second) | ||||||
|  | 			} | ||||||
| 			if err = irc.Reconnect(); err != nil { | 			if err = irc.Reconnect(); err != nil { | ||||||
| 				irc.Log.Printf("Error while reconnecting: %s\n", err) | 				irc.Log.Printf("Error while reconnecting: %s\n", err) | ||||||
| 				time.Sleep(60 * time.Second) | 				time.Sleep(60 * time.Second) | ||||||
| @@ -210,6 +246,7 @@ func (irc *Connection) Loop() { | |||||||
| 				break | 				break | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | 		connTime = time.Now() | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| @@ -430,26 +467,84 @@ func (irc *Connection) Connect(server string) error { | |||||||
| 		irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password) | 		irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	resChan := make(chan *SASLResult) | 	err = irc.negotiateCaps() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("NICK %s\r\n", irc.nick) | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Negotiate IRCv3 capabilities | ||||||
|  | func (irc *Connection) negotiateCaps() error { | ||||||
|  | 	saslResChan := make(chan *SASLResult) | ||||||
|  | 	if irc.UseSASL { | ||||||
|  | 		irc.RequestCaps = append(irc.RequestCaps, "sasl") | ||||||
|  | 		irc.setupSASLCallbacks(saslResChan) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if len(irc.RequestCaps) == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	cap_chan := make(chan bool, len(irc.RequestCaps)) | ||||||
|  | 	irc.AddCallback("CAP", func(e *Event) { | ||||||
|  | 		if len(e.Arguments) != 3 { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		command := e.Arguments[1] | ||||||
|  | 
 | ||||||
|  | 		if command == "LS" { | ||||||
|  | 			missing_caps := len(irc.RequestCaps) | ||||||
|  | 			for _, cap_name := range strings.Split(e.Arguments[2], " ") { | ||||||
|  | 				for _, req_cap := range irc.RequestCaps { | ||||||
|  | 					if cap_name == req_cap { | ||||||
|  | 						irc.pwrite <- fmt.Sprintf("CAP REQ :%s\r\n", cap_name) | ||||||
|  | 						missing_caps-- | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			for i := 0; i < missing_caps; i++ { | ||||||
|  | 				cap_chan <- true | ||||||
|  | 			} | ||||||
|  | 		} else if command == "ACK" || command == "NAK" { | ||||||
|  | 			for _, cap_name := range strings.Split(strings.TrimSpace(e.Arguments[2]), " ") { | ||||||
|  | 				if cap_name == "" { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 
 | ||||||
|  | 				if command == "ACK" { | ||||||
|  | 					irc.AcknowledgedCaps = append(irc.AcknowledgedCaps, cap_name) | ||||||
|  | 				} | ||||||
|  | 				cap_chan <- true | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	irc.pwrite <- "CAP LS\r\n" | ||||||
|  | 
 | ||||||
| 	if irc.UseSASL { | 	if irc.UseSASL { | ||||||
| 		irc.setupSASLCallbacks(resChan) |  | ||||||
| 		irc.pwrite <- fmt.Sprintf("CAP LS\r\n") |  | ||||||
| 		// request SASL |  | ||||||
| 		irc.pwrite <- fmt.Sprintf("CAP REQ :sasl\r\n") |  | ||||||
| 		// if sasl request doesn't complete in 15 seconds, close chan and timeout |  | ||||||
| 		select { | 		select { | ||||||
| 		case res := <-resChan: | 		case res := <-saslResChan: | ||||||
| 			if res.Failed { | 			if res.Failed { | ||||||
| 				close(resChan) | 				close(saslResChan) | ||||||
| 				return res.Err | 				return res.Err | ||||||
| 			} | 			} | ||||||
| 		case <-time.After(time.Second * 15): | 		case <-time.After(time.Second * 15): | ||||||
| 			close(resChan) | 			close(saslResChan) | ||||||
| 			return errors.New("SASL setup timed out. This shouldn't happen.") | 			return errors.New("SASL setup timed out. This shouldn't happen.") | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	irc.pwrite <- fmt.Sprintf("NICK %s\r\n", irc.nick) | 
 | ||||||
| 	irc.pwrite <- fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user) | 	// Wait for all capabilities to be ACKed or NAKed before ending negotiation | ||||||
|  | 	for i := 0; i < len(irc.RequestCaps); i++ { | ||||||
|  | 		<-cap_chan | ||||||
|  | 	} | ||||||
|  | 	irc.pwrite <- fmt.Sprintf("CAP END\r\n") | ||||||
|  | 
 | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| @@ -43,7 +43,6 @@ func (irc *Connection) setupSASLCallbacks(result chan<- *SASLResult) { | |||||||
| 		result <- &SASLResult{true, errors.New(e.Arguments[1])} | 		result <- &SASLResult{true, errors.New(e.Arguments[1])} | ||||||
| 	}) | 	}) | ||||||
| 	irc.AddCallback("903", func(e *Event) { | 	irc.AddCallback("903", func(e *Event) { | ||||||
| 		irc.SendRaw("CAP END") |  | ||||||
| 		result <- &SASLResult{false, nil} | 		result <- &SASLResult{false, nil} | ||||||
| 	}) | 	}) | ||||||
| 	irc.AddCallback("904", func(e *Event) { | 	irc.AddCallback("904", func(e *Event) { | ||||||
| @@ -20,6 +20,8 @@ type Connection struct { | |||||||
| 	Password         string | 	Password         string | ||||||
| 	UseTLS           bool | 	UseTLS           bool | ||||||
| 	UseSASL          bool | 	UseSASL          bool | ||||||
|  | 	RequestCaps      []string | ||||||
|  | 	AcknowledgedCaps []string | ||||||
| 	SASLLogin        string | 	SASLLogin        string | ||||||
| 	SASLPassword     string | 	SASLPassword     string | ||||||
| 	SASLMech         string | 	SASLMech         string | ||||||
| @@ -59,6 +61,7 @@ type Event struct { | |||||||
| 	Source     string //<host> | 	Source     string //<host> | ||||||
| 	User       string //<usr> | 	User       string //<usr> | ||||||
| 	Arguments  []string | 	Arguments  []string | ||||||
|  | 	Tags       map[string]string | ||||||
| 	Connection *Connection | 	Connection *Connection | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
							
								
								
									
										30
									
								
								vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										30
									
								
								vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,30 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"github.com/Sirupsen/logrus" |  | ||||||
| 	"gopkg.in/gemnasium/logrus-airbrake-hook.v2" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var log = logrus.New() |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.Formatter = new(logrus.TextFormatter) // default |  | ||||||
| 	log.Hooks.Add(airbrake.NewHook(123, "xyz", "development")) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	log.WithFields(logrus.Fields{ |  | ||||||
| 		"animal": "walrus", |  | ||||||
| 		"size":   10, |  | ||||||
| 	}).Info("A group of walrus emerges from the ocean") |  | ||||||
|  |  | ||||||
| 	log.WithFields(logrus.Fields{ |  | ||||||
| 		"omg":    true, |  | ||||||
| 		"number": 122, |  | ||||||
| 	}).Warn("The group's number increased tremendously!") |  | ||||||
|  |  | ||||||
| 	log.WithFields(logrus.Fields{ |  | ||||||
| 		"omg":    true, |  | ||||||
| 		"number": 100, |  | ||||||
| 	}).Fatal("The ice breaks!") |  | ||||||
| } |  | ||||||
							
								
								
									
										67
									
								
								vendor/github.com/Sirupsen/logrus/hooks/test/test.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										67
									
								
								vendor/github.com/Sirupsen/logrus/hooks/test/test.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,67 +0,0 @@ | |||||||
| package test |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"io/ioutil" |  | ||||||
|  |  | ||||||
| 	"github.com/Sirupsen/logrus" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // test.Hook is a hook designed for dealing with logs in test scenarios. |  | ||||||
| type Hook struct { |  | ||||||
| 	Entries []*logrus.Entry |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Installs a test hook for the global logger. |  | ||||||
| func NewGlobal() *Hook { |  | ||||||
|  |  | ||||||
| 	hook := new(Hook) |  | ||||||
| 	logrus.AddHook(hook) |  | ||||||
|  |  | ||||||
| 	return hook |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Installs a test hook for a given local logger. |  | ||||||
| func NewLocal(logger *logrus.Logger) *Hook { |  | ||||||
|  |  | ||||||
| 	hook := new(Hook) |  | ||||||
| 	logger.Hooks.Add(hook) |  | ||||||
|  |  | ||||||
| 	return hook |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Creates a discarding logger and installs the test hook. |  | ||||||
| func NewNullLogger() (*logrus.Logger, *Hook) { |  | ||||||
|  |  | ||||||
| 	logger := logrus.New() |  | ||||||
| 	logger.Out = ioutil.Discard |  | ||||||
|  |  | ||||||
| 	return logger, NewLocal(logger) |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (t *Hook) Fire(e *logrus.Entry) error { |  | ||||||
| 	t.Entries = append(t.Entries, e) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (t *Hook) Levels() []logrus.Level { |  | ||||||
| 	return logrus.AllLevels |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // LastEntry returns the last entry that was logged or nil. |  | ||||||
| func (t *Hook) LastEntry() (l *logrus.Entry) { |  | ||||||
|  |  | ||||||
| 	if i := len(t.Entries) - 1; i < 0 { |  | ||||||
| 		return nil |  | ||||||
| 	} else { |  | ||||||
| 		return t.Entries[i] |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Reset removes all Entries from this test hook. |  | ||||||
| func (t *Hook) Reset() { |  | ||||||
| 	t.Entries = make([]*logrus.Entry, 0) |  | ||||||
| } |  | ||||||
							
								
								
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_appengine.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_appengine.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,10 +0,0 @@ | |||||||
| // +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 |  | ||||||
| } |  | ||||||
							
								
								
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										10
									
								
								vendor/github.com/Sirupsen/logrus/terminal_bsd.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,10 +0,0 @@ | |||||||
| // +build darwin freebsd openbsd netbsd dragonfly |  | ||||||
| // +build !appengine |  | ||||||
|  |  | ||||||
| package logrus |  | ||||||
|  |  | ||||||
| import "syscall" |  | ||||||
|  |  | ||||||
| const ioctlReadTermios = syscall.TIOCGETA |  | ||||||
|  |  | ||||||
| type Termios syscall.Termios |  | ||||||
							
								
								
									
										28
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,28 +0,0 @@ | |||||||
| // Based on ssh/terminal: |  | ||||||
| // Copyright 2011 The Go Authors. All rights reserved. |  | ||||||
| // Use of this source code is governed by a BSD-style |  | ||||||
| // license that can be found in the LICENSE file. |  | ||||||
|  |  | ||||||
| // +build linux darwin freebsd openbsd netbsd dragonfly |  | ||||||
| // +build !appengine |  | ||||||
|  |  | ||||||
| package logrus |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"io" |  | ||||||
| 	"os" |  | ||||||
| 	"syscall" |  | ||||||
| 	"unsafe" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // IsTerminal returns true if stderr's file descriptor is a terminal. |  | ||||||
| func IsTerminal(f io.Writer) bool { |  | ||||||
| 	var termios Termios |  | ||||||
| 	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 |  | ||||||
| 	default: |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/Sirupsen/logrus/terminal_solaris.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,21 +0,0 @@ | |||||||
| // +build solaris,!appengine |  | ||||||
|  |  | ||||||
| package logrus |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"io" |  | ||||||
| 	"os" |  | ||||||
|  |  | ||||||
| 	"golang.org/x/sys/unix" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // IsTerminal returns true if the given file descriptor is a terminal. |  | ||||||
| func IsTerminal(f io.Writer) bool { |  | ||||||
| 	switch v := f.(type) { |  | ||||||
| 	case *os.File: |  | ||||||
| 		_, err := unix.IoctlGetTermios(int(v.Fd()), unix.TCGETA) |  | ||||||
| 		return err == nil |  | ||||||
| 	default: |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										33
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										33
									
								
								vendor/github.com/Sirupsen/logrus/terminal_windows.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,33 +0,0 @@ | |||||||
| // Based on ssh/terminal: |  | ||||||
| // Copyright 2011 The Go Authors. All rights reserved. |  | ||||||
| // Use of this source code is governed by a BSD-style |  | ||||||
| // license that can be found in the LICENSE file. |  | ||||||
|  |  | ||||||
| // +build windows,!appengine |  | ||||||
|  |  | ||||||
| package logrus |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"io" |  | ||||||
| 	"os" |  | ||||||
| 	"syscall" |  | ||||||
| 	"unsafe" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var kernel32 = syscall.NewLazyDLL("kernel32.dll") |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	procGetConsoleMode = kernel32.NewProc("GetConsoleMode") |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // IsTerminal returns true if stderr's file descriptor is a terminal. |  | ||||||
| func IsTerminal(f io.Writer) bool { |  | ||||||
| 	switch v := f.(type) { |  | ||||||
| 	case *os.File: |  | ||||||
| 		var st uint32 |  | ||||||
| 		r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(v.Fd()), uintptr(unsafe.Pointer(&st)), 0) |  | ||||||
| 		return r != 0 && e == 0 |  | ||||||
| 	default: |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										35
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										35
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -13,10 +13,18 @@ | |||||||
| // Package discordgo provides Discord binding for Go | // Package discordgo provides Discord binding for Go | ||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
| import "fmt" | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net/http" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  |  | ||||||
| // VERSION of Discordgo, follows Symantic Versioning. (http://semver.org/) | // VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/) | ||||||
| const VERSION = "0.15.0" | const VERSION = "0.18.0" | ||||||
|  |  | ||||||
|  | // ErrMFA will be risen by New when the user has 2FA. | ||||||
|  | var ErrMFA = errors.New("account has 2FA enabled") | ||||||
|  |  | ||||||
| // New creates a new Discord session and will automate some startup | // New creates a new Discord session and will automate some startup | ||||||
| // tasks if given enough information to do so.  Currently you can pass zero | // tasks if given enough information to do so.  Currently you can pass zero | ||||||
| @@ -31,18 +39,27 @@ const VERSION = "0.15.0" | |||||||
| //     With an email, password and auth token - Discord will verify the auth | //     With an email, password and auth token - Discord will verify the auth | ||||||
| //         token, if it is invalid it will sign in with the provided | //         token, if it is invalid it will sign in with the provided | ||||||
| //         credentials. This is the Discord recommended way to sign in. | //         credentials. This is the Discord recommended way to sign in. | ||||||
|  | // | ||||||
|  | // NOTE: While email/pass authentication is supported by DiscordGo it is | ||||||
|  | // HIGHLY DISCOURAGED by Discord. Please only use email/pass to obtain a token | ||||||
|  | // and then use that authentication token for all future connections. | ||||||
|  | // Also, doing any form of automation with a user (non Bot) account may result | ||||||
|  | // in that account being permanently banned from Discord. | ||||||
| func New(args ...interface{}) (s *Session, err error) { | func New(args ...interface{}) (s *Session, err error) { | ||||||
|  |  | ||||||
| 	// Create an empty Session interface. | 	// Create an empty Session interface. | ||||||
| 	s = &Session{ | 	s = &Session{ | ||||||
| 		State:                  NewState(), | 		State:                  NewState(), | ||||||
| 		ratelimiter:            NewRatelimiter(), | 		Ratelimiter:            NewRatelimiter(), | ||||||
| 		StateEnabled:           true, | 		StateEnabled:           true, | ||||||
| 		Compress:               true, | 		Compress:               true, | ||||||
| 		ShouldReconnectOnError: true, | 		ShouldReconnectOnError: true, | ||||||
| 		ShardID:                0, | 		ShardID:                0, | ||||||
| 		ShardCount:             1, | 		ShardCount:             1, | ||||||
| 		MaxRestRetries:         3, | 		MaxRestRetries:         3, | ||||||
|  | 		Client:                 &http.Client{Timeout: (20 * time.Second)}, | ||||||
|  | 		sequence:               new(int64), | ||||||
|  | 		LastHeartbeatAck:       time.Now().UTC(), | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// If no arguments are passed return the empty Session interface. | 	// If no arguments are passed return the empty Session interface. | ||||||
| @@ -60,7 +77,7 @@ func New(args ...interface{}) (s *Session, err error) { | |||||||
|  |  | ||||||
| 		case []string: | 		case []string: | ||||||
| 			if len(v) > 3 { | 			if len(v) > 3 { | ||||||
| 				err = fmt.Errorf("Too many string parameters provided.") | 				err = fmt.Errorf("too many string parameters provided") | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| @@ -91,7 +108,7 @@ func New(args ...interface{}) (s *Session, err error) { | |||||||
| 			} else if s.Token == "" { | 			} else if s.Token == "" { | ||||||
| 				s.Token = v | 				s.Token = v | ||||||
| 			} else { | 			} else { | ||||||
| 				err = fmt.Errorf("Too many string parameters provided.") | 				err = fmt.Errorf("too many string parameters provided") | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| @@ -99,7 +116,7 @@ func New(args ...interface{}) (s *Session, err error) { | |||||||
| 			// TODO: Parse configuration struct | 			// TODO: Parse configuration struct | ||||||
|  |  | ||||||
| 		default: | 		default: | ||||||
| 			err = fmt.Errorf("Unsupported parameter type provided.") | 			err = fmt.Errorf("unsupported parameter type provided") | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -113,7 +130,11 @@ func New(args ...interface{}) (s *Session, err error) { | |||||||
| 	} else { | 	} else { | ||||||
| 		err = s.Login(auth, pass) | 		err = s.Login(auth, pass) | ||||||
| 		if err != nil || s.Token == "" { | 		if err != nil || s.Token == "" { | ||||||
|  | 			if s.MFA { | ||||||
|  | 				err = ErrMFA | ||||||
|  | 			} else { | ||||||
| 				err = fmt.Errorf("Unable to fetch discord authentication token. %v", err) | 				err = fmt.Errorf("Unable to fetch discord authentication token. %v", err) | ||||||
|  | 			} | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										31
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										31
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -11,6 +11,9 @@ | |||||||
|  |  | ||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
|  | // APIVersion is the Discord API version used for the REST and Websocket API. | ||||||
|  | var APIVersion = "6" | ||||||
|  |  | ||||||
| // Known Discord API Endpoints. | // Known Discord API Endpoints. | ||||||
| var ( | var ( | ||||||
| 	EndpointStatus     = "https://status.discordapp.com/api/v2/" | 	EndpointStatus     = "https://status.discordapp.com/api/v2/" | ||||||
| @@ -19,13 +22,21 @@ var ( | |||||||
| 	EndpointSmUpcoming = EndpointSm + "upcoming.json" | 	EndpointSmUpcoming = EndpointSm + "upcoming.json" | ||||||
|  |  | ||||||
| 	EndpointDiscord    = "https://discordapp.com/" | 	EndpointDiscord    = "https://discordapp.com/" | ||||||
| 	EndpointAPI      = EndpointDiscord + "api/" | 	EndpointAPI        = EndpointDiscord + "api/v" + APIVersion + "/" | ||||||
| 	EndpointGuilds     = EndpointAPI + "guilds/" | 	EndpointGuilds     = EndpointAPI + "guilds/" | ||||||
| 	EndpointChannels   = EndpointAPI + "channels/" | 	EndpointChannels   = EndpointAPI + "channels/" | ||||||
| 	EndpointUsers      = EndpointAPI + "users/" | 	EndpointUsers      = EndpointAPI + "users/" | ||||||
| 	EndpointGateway    = EndpointAPI + "gateway" | 	EndpointGateway    = EndpointAPI + "gateway" | ||||||
|  | 	EndpointGatewayBot = EndpointGateway + "/bot" | ||||||
| 	EndpointWebhooks   = EndpointAPI + "webhooks/" | 	EndpointWebhooks   = EndpointAPI + "webhooks/" | ||||||
|  |  | ||||||
|  | 	EndpointCDN             = "https://cdn.discordapp.com/" | ||||||
|  | 	EndpointCDNAttachments  = EndpointCDN + "attachments/" | ||||||
|  | 	EndpointCDNAvatars      = EndpointCDN + "avatars/" | ||||||
|  | 	EndpointCDNIcons        = EndpointCDN + "icons/" | ||||||
|  | 	EndpointCDNSplashes     = EndpointCDN + "splashes/" | ||||||
|  | 	EndpointCDNChannelIcons = EndpointCDN + "channel-icons/" | ||||||
|  |  | ||||||
| 	EndpointAuth           = EndpointAPI + "auth/" | 	EndpointAuth           = EndpointAPI + "auth/" | ||||||
| 	EndpointLogin          = EndpointAuth + "login" | 	EndpointLogin          = EndpointAuth + "login" | ||||||
| 	EndpointLogout         = EndpointAuth + "logout" | 	EndpointLogout         = EndpointAuth + "logout" | ||||||
| @@ -48,7 +59,8 @@ var ( | |||||||
| 	EndpointIntegrations = EndpointAPI + "integrations" | 	EndpointIntegrations = EndpointAPI + "integrations" | ||||||
|  |  | ||||||
| 	EndpointUser               = func(uID string) string { return EndpointUsers + uID } | 	EndpointUser               = func(uID string) string { return EndpointUsers + uID } | ||||||
| 	EndpointUserAvatar        = func(uID, aID string) string { return EndpointUsers + uID + "/avatars/" + aID + ".jpg" } | 	EndpointUserAvatar         = func(uID, aID string) string { return EndpointCDNAvatars + uID + "/" + aID + ".png" } | ||||||
|  | 	EndpointUserAvatarAnimated = func(uID, aID string) string { return EndpointCDNAvatars + uID + "/" + aID + ".gif" } | ||||||
| 	EndpointUserSettings       = func(uID string) string { return EndpointUsers + uID + "/settings" } | 	EndpointUserSettings       = func(uID string) string { return EndpointUsers + uID + "/settings" } | ||||||
| 	EndpointUserGuilds         = func(uID string) string { return EndpointUsers + uID + "/guilds" } | 	EndpointUserGuilds         = func(uID string) string { return EndpointUsers + uID + "/guilds" } | ||||||
| 	EndpointUserGuild          = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID } | 	EndpointUserGuild          = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID } | ||||||
| @@ -56,9 +68,9 @@ var ( | |||||||
| 	EndpointUserChannels       = func(uID string) string { return EndpointUsers + uID + "/channels" } | 	EndpointUserChannels       = func(uID string) string { return EndpointUsers + uID + "/channels" } | ||||||
| 	EndpointUserDevices        = func(uID string) string { return EndpointUsers + uID + "/devices" } | 	EndpointUserDevices        = func(uID string) string { return EndpointUsers + uID + "/devices" } | ||||||
| 	EndpointUserConnections    = func(uID string) string { return EndpointUsers + uID + "/connections" } | 	EndpointUserConnections    = func(uID string) string { return EndpointUsers + uID + "/connections" } | ||||||
|  | 	EndpointUserNotes          = func(uID string) string { return EndpointUsers + "@me/notes/" + uID } | ||||||
|  |  | ||||||
| 	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID } | 	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID } | ||||||
| 	EndpointGuildInivtes         = func(gID string) string { return EndpointGuilds + gID + "/invites" } |  | ||||||
| 	EndpointGuildChannels        = func(gID string) string { return EndpointGuilds + gID + "/channels" } | 	EndpointGuildChannels        = func(gID string) string { return EndpointGuilds + gID + "/channels" } | ||||||
| 	EndpointGuildMembers         = func(gID string) string { return EndpointGuilds + gID + "/members" } | 	EndpointGuildMembers         = func(gID string) string { return EndpointGuilds + gID + "/members" } | ||||||
| 	EndpointGuildMember          = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID } | 	EndpointGuildMember          = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID } | ||||||
| @@ -73,8 +85,8 @@ var ( | |||||||
| 	EndpointGuildInvites         = func(gID string) string { return EndpointGuilds + gID + "/invites" } | 	EndpointGuildInvites         = func(gID string) string { return EndpointGuilds + gID + "/invites" } | ||||||
| 	EndpointGuildEmbed           = func(gID string) string { return EndpointGuilds + gID + "/embed" } | 	EndpointGuildEmbed           = func(gID string) string { return EndpointGuilds + gID + "/embed" } | ||||||
| 	EndpointGuildPrune           = func(gID string) string { return EndpointGuilds + gID + "/prune" } | 	EndpointGuildPrune           = func(gID string) string { return EndpointGuilds + gID + "/prune" } | ||||||
| 	EndpointGuildIcon            = func(gID, hash string) string { return EndpointGuilds + gID + "/icons/" + hash + ".jpg" } | 	EndpointGuildIcon            = func(gID, hash string) string { return EndpointCDNIcons + gID + "/" + hash + ".png" } | ||||||
| 	EndpointGuildSplash          = func(gID, hash string) string { return EndpointGuilds + gID + "/splashes/" + hash + ".jpg" } | 	EndpointGuildSplash          = func(gID, hash string) string { return EndpointCDNSplashes + gID + "/" + hash + ".png" } | ||||||
| 	EndpointGuildWebhooks        = func(gID string) string { return EndpointGuilds + gID + "/webhooks" } | 	EndpointGuildWebhooks        = func(gID string) string { return EndpointGuilds + gID + "/webhooks" } | ||||||
|  |  | ||||||
| 	EndpointChannel                   = func(cID string) string { return EndpointChannels + cID } | 	EndpointChannel                   = func(cID string) string { return EndpointChannels + cID } | ||||||
| @@ -85,14 +97,19 @@ var ( | |||||||
| 	EndpointChannelMessages           = func(cID string) string { return EndpointChannels + cID + "/messages" } | 	EndpointChannelMessages           = func(cID string) string { return EndpointChannels + cID + "/messages" } | ||||||
| 	EndpointChannelMessage            = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID } | 	EndpointChannelMessage            = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID } | ||||||
| 	EndpointChannelMessageAck         = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" } | 	EndpointChannelMessageAck         = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" } | ||||||
| 	EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk_delete" } | 	EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk-delete" } | ||||||
| 	EndpointChannelMessagesPins       = func(cID string) string { return EndpointChannel(cID) + "/pins" } | 	EndpointChannelMessagesPins       = func(cID string) string { return EndpointChannel(cID) + "/pins" } | ||||||
| 	EndpointChannelMessagePin         = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID } | 	EndpointChannelMessagePin         = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID } | ||||||
|  |  | ||||||
|  | 	EndpointGroupIcon = func(cID, hash string) string { return EndpointCDNChannelIcons + cID + "/" + hash + ".png" } | ||||||
|  |  | ||||||
| 	EndpointChannelWebhooks = func(cID string) string { return EndpointChannel(cID) + "/webhooks" } | 	EndpointChannelWebhooks = func(cID string) string { return EndpointChannel(cID) + "/webhooks" } | ||||||
| 	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID } | 	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID } | ||||||
| 	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token } | 	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token } | ||||||
|  |  | ||||||
|  | 	EndpointMessageReactionsAll = func(cID, mID string) string { | ||||||
|  | 		return EndpointChannelMessage(cID, mID) + "/reactions" | ||||||
|  | 	} | ||||||
| 	EndpointMessageReactions = func(cID, mID, eID string) string { | 	EndpointMessageReactions = func(cID, mID, eID string) string { | ||||||
| 		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID | 		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID | ||||||
| 	} | 	} | ||||||
| @@ -104,6 +121,8 @@ var ( | |||||||
| 	EndpointRelationship        = func(uID string) string { return EndpointRelationships() + "/" + uID } | 	EndpointRelationship        = func(uID string) string { return EndpointRelationships() + "/" + uID } | ||||||
| 	EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" } | 	EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" } | ||||||
|  |  | ||||||
|  | 	EndpointGuildCreate = EndpointAPI + "guilds" | ||||||
|  |  | ||||||
| 	EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID } | 	EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID } | ||||||
|  |  | ||||||
| 	EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" } | 	EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" } | ||||||
|   | |||||||
							
								
								
									
										38
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										38
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,14 +1,12 @@ | |||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
| import "fmt" |  | ||||||
|  |  | ||||||
| // EventHandler is an interface for Discord events. | // EventHandler is an interface for Discord events. | ||||||
| type EventHandler interface { | type EventHandler interface { | ||||||
| 	// Type returns the type of event this handler belongs to. | 	// Type returns the type of event this handler belongs to. | ||||||
| 	Type() string | 	Type() string | ||||||
|  |  | ||||||
| 	// Handle is called whenever an event of Type() happens. | 	// Handle is called whenever an event of Type() happens. | ||||||
| 	// It is the recievers responsibility to type assert that the interface | 	// It is the receivers responsibility to type assert that the interface | ||||||
| 	// is the expected struct. | 	// is the expected struct. | ||||||
| 	Handle(*Session, interface{}) | 	Handle(*Session, interface{}) | ||||||
| } | } | ||||||
| @@ -45,12 +43,15 @@ var registeredInterfaceProviders = map[string]EventInterfaceProvider{} | |||||||
|  |  | ||||||
| // registerInterfaceProvider registers a provider so that DiscordGo can | // registerInterfaceProvider registers a provider so that DiscordGo can | ||||||
| // access it's New() method. | // access it's New() method. | ||||||
| func registerInterfaceProvider(eh EventInterfaceProvider) error { | func registerInterfaceProvider(eh EventInterfaceProvider) { | ||||||
| 	if _, ok := registeredInterfaceProviders[eh.Type()]; ok { | 	if _, ok := registeredInterfaceProviders[eh.Type()]; ok { | ||||||
| 		return fmt.Errorf("event %s already registered", eh.Type()) | 		return | ||||||
|  | 		// XXX: | ||||||
|  | 		// if we should error here, we need to do something with it. | ||||||
|  | 		// fmt.Errorf("event %s already registered", eh.Type()) | ||||||
| 	} | 	} | ||||||
| 	registeredInterfaceProviders[eh.Type()] = eh | 	registeredInterfaceProviders[eh.Type()] = eh | ||||||
| 	return nil | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // eventHandlerInstance is a wrapper around an event handler, as functions | // eventHandlerInstance is a wrapper around an event handler, as functions | ||||||
| @@ -155,13 +156,21 @@ func (s *Session) removeEventHandlerInstance(t string, ehi *eventHandlerInstance | |||||||
| // Handles calling permanent and once handlers for an event type. | // Handles calling permanent and once handlers for an event type. | ||||||
| func (s *Session) handle(t string, i interface{}) { | func (s *Session) handle(t string, i interface{}) { | ||||||
| 	for _, eh := range s.handlers[t] { | 	for _, eh := range s.handlers[t] { | ||||||
|  | 		if s.SyncEvents { | ||||||
|  | 			eh.eventHandler.Handle(s, i) | ||||||
|  | 		} else { | ||||||
| 			go eh.eventHandler.Handle(s, i) | 			go eh.eventHandler.Handle(s, i) | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if len(s.onceHandlers[t]) > 0 { | 	if len(s.onceHandlers[t]) > 0 { | ||||||
| 		for _, eh := range s.onceHandlers[t] { | 		for _, eh := range s.onceHandlers[t] { | ||||||
|  | 			if s.SyncEvents { | ||||||
|  | 				eh.eventHandler.Handle(s, i) | ||||||
|  | 			} else { | ||||||
| 				go eh.eventHandler.Handle(s, i) | 				go eh.eventHandler.Handle(s, i) | ||||||
| 			} | 			} | ||||||
|  | 		} | ||||||
| 		s.onceHandlers[t] = nil | 		s.onceHandlers[t] = nil | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -210,14 +219,15 @@ func (s *Session) onInterface(i interface{}) { | |||||||
| 		setGuildIds(t.Guild) | 		setGuildIds(t.Guild) | ||||||
| 	case *GuildUpdate: | 	case *GuildUpdate: | ||||||
| 		setGuildIds(t.Guild) | 		setGuildIds(t.Guild) | ||||||
| 	case *Resumed: |  | ||||||
| 		s.onResumed(t) |  | ||||||
| 	case *VoiceServerUpdate: | 	case *VoiceServerUpdate: | ||||||
| 		go s.onVoiceServerUpdate(t) | 		go s.onVoiceServerUpdate(t) | ||||||
| 	case *VoiceStateUpdate: | 	case *VoiceStateUpdate: | ||||||
| 		go s.onVoiceStateUpdate(t) | 		go s.onVoiceStateUpdate(t) | ||||||
| 	} | 	} | ||||||
| 	s.State.onInterface(s, i) | 	err := s.State.OnInterface(s, i) | ||||||
|  | 	if err != nil { | ||||||
|  | 		s.log(LogDebug, "error dispatching internal event, %s", err) | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // onReady handles the ready event. | // onReady handles the ready event. | ||||||
| @@ -225,14 +235,4 @@ func (s *Session) onReady(r *Ready) { | |||||||
|  |  | ||||||
| 	// Store the SessionID within the Session struct. | 	// Store the SessionID within the Session struct. | ||||||
| 	s.sessionID = r.SessionID | 	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) |  | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										93
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										93
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -31,8 +31,10 @@ const ( | |||||||
| 	messageAckEventType               = "MESSAGE_ACK" | 	messageAckEventType               = "MESSAGE_ACK" | ||||||
| 	messageCreateEventType            = "MESSAGE_CREATE" | 	messageCreateEventType            = "MESSAGE_CREATE" | ||||||
| 	messageDeleteEventType            = "MESSAGE_DELETE" | 	messageDeleteEventType            = "MESSAGE_DELETE" | ||||||
|  | 	messageDeleteBulkEventType        = "MESSAGE_DELETE_BULK" | ||||||
| 	messageReactionAddEventType       = "MESSAGE_REACTION_ADD" | 	messageReactionAddEventType       = "MESSAGE_REACTION_ADD" | ||||||
| 	messageReactionRemoveEventType    = "MESSAGE_REACTION_REMOVE" | 	messageReactionRemoveEventType    = "MESSAGE_REACTION_REMOVE" | ||||||
|  | 	messageReactionRemoveAllEventType = "MESSAGE_REACTION_REMOVE_ALL" | ||||||
| 	messageUpdateEventType            = "MESSAGE_UPDATE" | 	messageUpdateEventType            = "MESSAGE_UPDATE" | ||||||
| 	presenceUpdateEventType           = "PRESENCE_UPDATE" | 	presenceUpdateEventType           = "PRESENCE_UPDATE" | ||||||
| 	presencesReplaceEventType         = "PRESENCES_REPLACE" | 	presencesReplaceEventType         = "PRESENCES_REPLACE" | ||||||
| @@ -43,6 +45,7 @@ const ( | |||||||
| 	resumedEventType                  = "RESUMED" | 	resumedEventType                  = "RESUMED" | ||||||
| 	typingStartEventType              = "TYPING_START" | 	typingStartEventType              = "TYPING_START" | ||||||
| 	userGuildSettingsUpdateEventType  = "USER_GUILD_SETTINGS_UPDATE" | 	userGuildSettingsUpdateEventType  = "USER_GUILD_SETTINGS_UPDATE" | ||||||
|  | 	userNoteUpdateEventType           = "USER_NOTE_UPDATE" | ||||||
| 	userSettingsUpdateEventType       = "USER_SETTINGS_UPDATE" | 	userSettingsUpdateEventType       = "USER_SETTINGS_UPDATE" | ||||||
| 	userUpdateEventType               = "USER_UPDATE" | 	userUpdateEventType               = "USER_UPDATE" | ||||||
| 	voiceServerUpdateEventType        = "VOICE_SERVER_UPDATE" | 	voiceServerUpdateEventType        = "VOICE_SERVER_UPDATE" | ||||||
| @@ -137,11 +140,6 @@ func (eh connectEventHandler) Type() string { | |||||||
| 	return connectEventType | 	return connectEventType | ||||||
| } | } | ||||||
|  |  | ||||||
| // New returns a new instance of Connect. |  | ||||||
| func (eh connectEventHandler) New() interface{} { |  | ||||||
| 	return &Connect{} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Handle is the handler for Connect events. | // Handle is the handler for Connect events. | ||||||
| func (eh connectEventHandler) Handle(s *Session, i interface{}) { | func (eh connectEventHandler) Handle(s *Session, i interface{}) { | ||||||
| 	if t, ok := i.(*Connect); ok { | 	if t, ok := i.(*Connect); ok { | ||||||
| @@ -157,11 +155,6 @@ func (eh disconnectEventHandler) Type() string { | |||||||
| 	return disconnectEventType | 	return disconnectEventType | ||||||
| } | } | ||||||
|  |  | ||||||
| // New returns a new instance of Disconnect. |  | ||||||
| func (eh disconnectEventHandler) New() interface{} { |  | ||||||
| 	return &Disconnect{} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Handle is the handler for Disconnect events. | // Handle is the handler for Disconnect events. | ||||||
| func (eh disconnectEventHandler) Handle(s *Session, i interface{}) { | func (eh disconnectEventHandler) Handle(s *Session, i interface{}) { | ||||||
| 	if t, ok := i.(*Disconnect); ok { | 	if t, ok := i.(*Disconnect); ok { | ||||||
| @@ -177,11 +170,6 @@ func (eh eventEventHandler) Type() string { | |||||||
| 	return eventEventType | 	return eventEventType | ||||||
| } | } | ||||||
|  |  | ||||||
| // New returns a new instance of Event. |  | ||||||
| func (eh eventEventHandler) New() interface{} { |  | ||||||
| 	return &Event{} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Handle is the handler for Event events. | // Handle is the handler for Event events. | ||||||
| func (eh eventEventHandler) Handle(s *Session, i interface{}) { | func (eh eventEventHandler) Handle(s *Session, i interface{}) { | ||||||
| 	if t, ok := i.(*Event); ok { | 	if t, ok := i.(*Event); ok { | ||||||
| @@ -529,6 +517,26 @@ func (eh messageDeleteEventHandler) Handle(s *Session, i interface{}) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // messageDeleteBulkEventHandler is an event handler for MessageDeleteBulk events. | ||||||
|  | type messageDeleteBulkEventHandler func(*Session, *MessageDeleteBulk) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageDeleteBulk events. | ||||||
|  | func (eh messageDeleteBulkEventHandler) Type() string { | ||||||
|  | 	return messageDeleteBulkEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageDeleteBulk. | ||||||
|  | func (eh messageDeleteBulkEventHandler) New() interface{} { | ||||||
|  | 	return &MessageDeleteBulk{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageDeleteBulk events. | ||||||
|  | func (eh messageDeleteBulkEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageDeleteBulk); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // messageReactionAddEventHandler is an event handler for MessageReactionAdd events. | // messageReactionAddEventHandler is an event handler for MessageReactionAdd events. | ||||||
| type messageReactionAddEventHandler func(*Session, *MessageReactionAdd) | type messageReactionAddEventHandler func(*Session, *MessageReactionAdd) | ||||||
|  |  | ||||||
| @@ -569,6 +577,26 @@ func (eh messageReactionRemoveEventHandler) Handle(s *Session, i interface{}) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // messageReactionRemoveAllEventHandler is an event handler for MessageReactionRemoveAll events. | ||||||
|  | type messageReactionRemoveAllEventHandler func(*Session, *MessageReactionRemoveAll) | ||||||
|  |  | ||||||
|  | // Type returns the event type for MessageReactionRemoveAll events. | ||||||
|  | func (eh messageReactionRemoveAllEventHandler) Type() string { | ||||||
|  | 	return messageReactionRemoveAllEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of MessageReactionRemoveAll. | ||||||
|  | func (eh messageReactionRemoveAllEventHandler) New() interface{} { | ||||||
|  | 	return &MessageReactionRemoveAll{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for MessageReactionRemoveAll events. | ||||||
|  | func (eh messageReactionRemoveAllEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*MessageReactionRemoveAll); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // messageUpdateEventHandler is an event handler for MessageUpdate events. | // messageUpdateEventHandler is an event handler for MessageUpdate events. | ||||||
| type messageUpdateEventHandler func(*Session, *MessageUpdate) | type messageUpdateEventHandler func(*Session, *MessageUpdate) | ||||||
|  |  | ||||||
| @@ -637,11 +665,6 @@ func (eh rateLimitEventHandler) Type() string { | |||||||
| 	return rateLimitEventType | 	return rateLimitEventType | ||||||
| } | } | ||||||
|  |  | ||||||
| // New returns a new instance of RateLimit. |  | ||||||
| func (eh rateLimitEventHandler) New() interface{} { |  | ||||||
| 	return &RateLimit{} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Handle is the handler for RateLimit events. | // Handle is the handler for RateLimit events. | ||||||
| func (eh rateLimitEventHandler) Handle(s *Session, i interface{}) { | func (eh rateLimitEventHandler) Handle(s *Session, i interface{}) { | ||||||
| 	if t, ok := i.(*RateLimit); ok { | 	if t, ok := i.(*RateLimit); ok { | ||||||
| @@ -769,6 +792,26 @@ func (eh userGuildSettingsUpdateEventHandler) Handle(s *Session, i interface{}) | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // userNoteUpdateEventHandler is an event handler for UserNoteUpdate events. | ||||||
|  | type userNoteUpdateEventHandler func(*Session, *UserNoteUpdate) | ||||||
|  |  | ||||||
|  | // Type returns the event type for UserNoteUpdate events. | ||||||
|  | func (eh userNoteUpdateEventHandler) Type() string { | ||||||
|  | 	return userNoteUpdateEventType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New returns a new instance of UserNoteUpdate. | ||||||
|  | func (eh userNoteUpdateEventHandler) New() interface{} { | ||||||
|  | 	return &UserNoteUpdate{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Handle is the handler for UserNoteUpdate events. | ||||||
|  | func (eh userNoteUpdateEventHandler) Handle(s *Session, i interface{}) { | ||||||
|  | 	if t, ok := i.(*UserNoteUpdate); ok { | ||||||
|  | 		eh(s, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // userSettingsUpdateEventHandler is an event handler for UserSettingsUpdate events. | // userSettingsUpdateEventHandler is an event handler for UserSettingsUpdate events. | ||||||
| type userSettingsUpdateEventHandler func(*Session, *UserSettingsUpdate) | type userSettingsUpdateEventHandler func(*Session, *UserSettingsUpdate) | ||||||
|  |  | ||||||
| @@ -901,10 +944,14 @@ func handlerForInterface(handler interface{}) EventHandler { | |||||||
| 		return messageCreateEventHandler(v) | 		return messageCreateEventHandler(v) | ||||||
| 	case func(*Session, *MessageDelete): | 	case func(*Session, *MessageDelete): | ||||||
| 		return messageDeleteEventHandler(v) | 		return messageDeleteEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageDeleteBulk): | ||||||
|  | 		return messageDeleteBulkEventHandler(v) | ||||||
| 	case func(*Session, *MessageReactionAdd): | 	case func(*Session, *MessageReactionAdd): | ||||||
| 		return messageReactionAddEventHandler(v) | 		return messageReactionAddEventHandler(v) | ||||||
| 	case func(*Session, *MessageReactionRemove): | 	case func(*Session, *MessageReactionRemove): | ||||||
| 		return messageReactionRemoveEventHandler(v) | 		return messageReactionRemoveEventHandler(v) | ||||||
|  | 	case func(*Session, *MessageReactionRemoveAll): | ||||||
|  | 		return messageReactionRemoveAllEventHandler(v) | ||||||
| 	case func(*Session, *MessageUpdate): | 	case func(*Session, *MessageUpdate): | ||||||
| 		return messageUpdateEventHandler(v) | 		return messageUpdateEventHandler(v) | ||||||
| 	case func(*Session, *PresenceUpdate): | 	case func(*Session, *PresenceUpdate): | ||||||
| @@ -925,6 +972,8 @@ func handlerForInterface(handler interface{}) EventHandler { | |||||||
| 		return typingStartEventHandler(v) | 		return typingStartEventHandler(v) | ||||||
| 	case func(*Session, *UserGuildSettingsUpdate): | 	case func(*Session, *UserGuildSettingsUpdate): | ||||||
| 		return userGuildSettingsUpdateEventHandler(v) | 		return userGuildSettingsUpdateEventHandler(v) | ||||||
|  | 	case func(*Session, *UserNoteUpdate): | ||||||
|  | 		return userNoteUpdateEventHandler(v) | ||||||
| 	case func(*Session, *UserSettingsUpdate): | 	case func(*Session, *UserSettingsUpdate): | ||||||
| 		return userSettingsUpdateEventHandler(v) | 		return userSettingsUpdateEventHandler(v) | ||||||
| 	case func(*Session, *UserUpdate): | 	case func(*Session, *UserUpdate): | ||||||
| @@ -937,6 +986,7 @@ func handlerForInterface(handler interface{}) EventHandler { | |||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	registerInterfaceProvider(channelCreateEventHandler(nil)) | 	registerInterfaceProvider(channelCreateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(channelDeleteEventHandler(nil)) | 	registerInterfaceProvider(channelDeleteEventHandler(nil)) | ||||||
| @@ -959,8 +1009,10 @@ func init() { | |||||||
| 	registerInterfaceProvider(messageAckEventHandler(nil)) | 	registerInterfaceProvider(messageAckEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(messageCreateEventHandler(nil)) | 	registerInterfaceProvider(messageCreateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(messageDeleteEventHandler(nil)) | 	registerInterfaceProvider(messageDeleteEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageDeleteBulkEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(messageReactionAddEventHandler(nil)) | 	registerInterfaceProvider(messageReactionAddEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(messageReactionRemoveEventHandler(nil)) | 	registerInterfaceProvider(messageReactionRemoveEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(messageReactionRemoveAllEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(messageUpdateEventHandler(nil)) | 	registerInterfaceProvider(messageUpdateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(presenceUpdateEventHandler(nil)) | 	registerInterfaceProvider(presenceUpdateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(presencesReplaceEventHandler(nil)) | 	registerInterfaceProvider(presencesReplaceEventHandler(nil)) | ||||||
| @@ -970,6 +1022,7 @@ func init() { | |||||||
| 	registerInterfaceProvider(resumedEventHandler(nil)) | 	registerInterfaceProvider(resumedEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(typingStartEventHandler(nil)) | 	registerInterfaceProvider(typingStartEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(userGuildSettingsUpdateEventHandler(nil)) | 	registerInterfaceProvider(userGuildSettingsUpdateEventHandler(nil)) | ||||||
|  | 	registerInterfaceProvider(userNoteUpdateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(userSettingsUpdateEventHandler(nil)) | 	registerInterfaceProvider(userSettingsUpdateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(userUpdateEventHandler(nil)) | 	registerInterfaceProvider(userUpdateEventHandler(nil)) | ||||||
| 	registerInterfaceProvider(voiceServerUpdateEventHandler(nil)) | 	registerInterfaceProvider(voiceServerUpdateEventHandler(nil)) | ||||||
|   | |||||||
							
								
								
									
										23
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										23
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -2,7 +2,6 @@ package discordgo | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"time" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // This file contains all the possible structs that can be | // This file contains all the possible structs that can be | ||||||
| @@ -28,7 +27,7 @@ type RateLimit struct { | |||||||
| // Event provides a basic initial struct for all websocket events. | // Event provides a basic initial struct for all websocket events. | ||||||
| type Event struct { | type Event struct { | ||||||
| 	Operation int             `json:"op"` | 	Operation int             `json:"op"` | ||||||
| 	Sequence  int             `json:"s"` | 	Sequence  int64           `json:"s"` | ||||||
| 	Type      string          `json:"t"` | 	Type      string          `json:"t"` | ||||||
| 	RawData   json.RawMessage `json:"d"` | 	RawData   json.RawMessage `json:"d"` | ||||||
| 	// Struct contains one of the other types in this file. | 	// Struct contains one of the other types in this file. | ||||||
| @@ -39,7 +38,6 @@ type Event struct { | |||||||
| type Ready struct { | type Ready struct { | ||||||
| 	Version         int          `json:"v"` | 	Version         int          `json:"v"` | ||||||
| 	SessionID       string       `json:"session_id"` | 	SessionID       string       `json:"session_id"` | ||||||
| 	HeartbeatInterval time.Duration `json:"heartbeat_interval"` |  | ||||||
| 	User            *User        `json:"user"` | 	User            *User        `json:"user"` | ||||||
| 	ReadState       []*ReadState `json:"read_state"` | 	ReadState       []*ReadState `json:"read_state"` | ||||||
| 	PrivateChannels []*Channel   `json:"private_channels"` | 	PrivateChannels []*Channel   `json:"private_channels"` | ||||||
| @@ -50,6 +48,7 @@ type Ready struct { | |||||||
| 	UserGuildSettings []*UserGuildSettings `json:"user_guild_settings"` | 	UserGuildSettings []*UserGuildSettings `json:"user_guild_settings"` | ||||||
| 	Relationships     []*Relationship      `json:"relationships"` | 	Relationships     []*Relationship      `json:"relationships"` | ||||||
| 	Presences         []*Presence          `json:"presences"` | 	Presences         []*Presence          `json:"presences"` | ||||||
|  | 	Notes             map[string]string    `json:"notes"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelCreate is the data for a ChannelCreate event. | // ChannelCreate is the data for a ChannelCreate event. | ||||||
| @@ -179,6 +178,11 @@ type MessageReactionRemove struct { | |||||||
| 	*MessageReaction | 	*MessageReaction | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MessageReactionRemoveAll is the data for a MessageReactionRemoveAll event. | ||||||
|  | type MessageReactionRemoveAll struct { | ||||||
|  | 	*MessageReaction | ||||||
|  | } | ||||||
|  |  | ||||||
| // PresencesReplace is the data for a PresencesReplace event. | // PresencesReplace is the data for a PresencesReplace event. | ||||||
| type PresencesReplace []*Presence | type PresencesReplace []*Presence | ||||||
|  |  | ||||||
| @@ -191,7 +195,6 @@ type PresenceUpdate struct { | |||||||
|  |  | ||||||
| // Resumed is the data for a Resumed event. | // Resumed is the data for a Resumed event. | ||||||
| type Resumed struct { | type Resumed struct { | ||||||
| 	HeartbeatInterval time.Duration `json:"heartbeat_interval"` |  | ||||||
| 	Trace []string `json:"_trace"` | 	Trace []string `json:"_trace"` | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -225,6 +228,12 @@ type UserGuildSettingsUpdate struct { | |||||||
| 	*UserGuildSettings | 	*UserGuildSettings | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // UserNoteUpdate is the data for a UserNoteUpdate event. | ||||||
|  | type UserNoteUpdate struct { | ||||||
|  | 	ID   string `json:"id"` | ||||||
|  | 	Note string `json:"note"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // VoiceServerUpdate is the data for a VoiceServerUpdate event. | // VoiceServerUpdate is the data for a VoiceServerUpdate event. | ||||||
| type VoiceServerUpdate struct { | type VoiceServerUpdate struct { | ||||||
| 	Token    string `json:"token"` | 	Token    string `json:"token"` | ||||||
| @@ -236,3 +245,9 @@ type VoiceServerUpdate struct { | |||||||
| type VoiceStateUpdate struct { | type VoiceStateUpdate struct { | ||||||
| 	*VoiceState | 	*VoiceState | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MessageDeleteBulk is the data for a MessageDeleteBulk event | ||||||
|  | type MessageDeleteBulk struct { | ||||||
|  | 	Messages  []string `json:"ids"` | ||||||
|  | 	ChannelID string   `json:"channel_id"` | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										41
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										41
									
								
								vendor/github.com/bwmarrin/discordgo/examples/airhorn/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -6,7 +6,9 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"io" | 	"io" | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"os/signal" | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"syscall" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
| @@ -21,6 +23,7 @@ var token string | |||||||
| var buffer = make([][]byte, 0) | var buffer = make([][]byte, 0) | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
|  |  | ||||||
| 	if token == "" { | 	if token == "" { | ||||||
| 		fmt.Println("No token provided. Please run: airhorn -t <bot token>") | 		fmt.Println("No token provided. Please run: airhorn -t <bot token>") | ||||||
| 		return | 		return | ||||||
| @@ -56,21 +59,37 @@ func main() { | |||||||
| 		fmt.Println("Error opening Discord session: ", err) | 		fmt.Println("Error opening Discord session: ", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Wait here until CTRL-C or other term signal is received. | ||||||
| 	fmt.Println("Airhorn is now running.  Press CTRL-C to exit.") | 	fmt.Println("Airhorn is now running.  Press CTRL-C to exit.") | ||||||
| 	// Simple way to keep program running until CTRL-C is pressed. | 	sc := make(chan os.Signal, 1) | ||||||
| 	<-make(chan struct{}) | 	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) | ||||||
| 	return | 	<-sc | ||||||
|  |  | ||||||
|  | 	// Cleanly close down the Discord session. | ||||||
|  | 	dg.Close() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // This function will be called (due to AddHandler above) when the bot receives | ||||||
|  | // the "ready" event from Discord. | ||||||
| func ready(s *discordgo.Session, event *discordgo.Ready) { | func ready(s *discordgo.Session, event *discordgo.Ready) { | ||||||
|  |  | ||||||
| 	// Set the playing status. | 	// Set the playing status. | ||||||
| 	_ = s.UpdateStatus(0, "!airhorn") | 	s.UpdateStatus(0, "!airhorn") | ||||||
| } | } | ||||||
|  |  | ||||||
| // This function will be called (due to AddHandler above) every time a new | // 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. | // message is created on any channel that the autenticated bot has access to. | ||||||
| func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | ||||||
|  |  | ||||||
|  | 	// Ignore all messages created by the bot itself | ||||||
|  | 	// This isn't required in this specific example but it's a good practice. | ||||||
|  | 	if m.Author.ID == s.State.User.ID { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// check if the message is "!airhorn" | ||||||
| 	if strings.HasPrefix(m.Content, "!airhorn") { | 	if strings.HasPrefix(m.Content, "!airhorn") { | ||||||
|  |  | ||||||
| 		// Find the channel that the message came from. | 		// Find the channel that the message came from. | ||||||
| 		c, err := s.State.Channel(m.ChannelID) | 		c, err := s.State.Channel(m.ChannelID) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| @@ -85,7 +104,7 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | |||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// Look for the message sender in that guilds current voice states. | 		// Look for the message sender in that guild's current voice states. | ||||||
| 		for _, vs := range g.VoiceStates { | 		for _, vs := range g.VoiceStates { | ||||||
| 			if vs.UserID == m.Author.ID { | 			if vs.UserID == m.Author.ID { | ||||||
| 				err = playSound(s, g.ID, vs.ChannelID) | 				err = playSound(s, g.ID, vs.ChannelID) | ||||||
| @@ -102,6 +121,7 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | |||||||
| // This function will be called (due to AddHandler above) every time a new | // This function will be called (due to AddHandler above) every time a new | ||||||
| // guild is joined. | // guild is joined. | ||||||
| func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) { | func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) { | ||||||
|  |  | ||||||
| 	if event.Guild.Unavailable { | 	if event.Guild.Unavailable { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -116,8 +136,8 @@ func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) { | |||||||
|  |  | ||||||
| // loadSound attempts to load an encoded sound file from disk. | // loadSound attempts to load an encoded sound file from disk. | ||||||
| func loadSound() error { | func loadSound() error { | ||||||
| 	file, err := os.Open("airhorn.dca") |  | ||||||
|  |  | ||||||
|  | 	file, err := os.Open("airhorn.dca") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		fmt.Println("Error opening dca file :", err) | 		fmt.Println("Error opening dca file :", err) | ||||||
| 		return err | 		return err | ||||||
| @@ -131,7 +151,7 @@ func loadSound() error { | |||||||
|  |  | ||||||
| 		// If this is the end of the file, just return. | 		// If this is the end of the file, just return. | ||||||
| 		if err == io.EOF || err == io.ErrUnexpectedEOF { | 		if err == io.EOF || err == io.ErrUnexpectedEOF { | ||||||
| 			file.Close() | 			err := file.Close() | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| @@ -160,6 +180,7 @@ func loadSound() error { | |||||||
|  |  | ||||||
| // playSound plays the current buffer to the provided channel. | // playSound plays the current buffer to the provided channel. | ||||||
| func playSound(s *discordgo.Session, guildID, channelID string) (err error) { | func playSound(s *discordgo.Session, guildID, channelID string) (err error) { | ||||||
|  |  | ||||||
| 	// Join the provided voice channel. | 	// Join the provided voice channel. | ||||||
| 	vc, err := s.ChannelVoiceJoin(guildID, channelID, false, true) | 	vc, err := s.ChannelVoiceJoin(guildID, channelID, false, true) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -170,7 +191,7 @@ func playSound(s *discordgo.Session, guildID, channelID string) (err error) { | |||||||
| 	time.Sleep(250 * time.Millisecond) | 	time.Sleep(250 * time.Millisecond) | ||||||
|  |  | ||||||
| 	// Start speaking. | 	// Start speaking. | ||||||
| 	_ = vc.Speaking(true) | 	vc.Speaking(true) | ||||||
|  |  | ||||||
| 	// Send the buffer data. | 	// Send the buffer data. | ||||||
| 	for _, buff := range buffer { | 	for _, buff := range buffer { | ||||||
| @@ -178,13 +199,13 @@ func playSound(s *discordgo.Session, guildID, channelID string) (err error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Stop speaking | 	// Stop speaking | ||||||
| 	_ = vc.Speaking(false) | 	vc.Speaking(false) | ||||||
|  |  | ||||||
| 	// Sleep for a specificed amount of time before ending. | 	// Sleep for a specificed amount of time before ending. | ||||||
| 	time.Sleep(250 * time.Millisecond) | 	time.Sleep(250 * time.Millisecond) | ||||||
|  |  | ||||||
| 	// Disconnect from the provided voice channel. | 	// Disconnect from the provided voice channel. | ||||||
| 	_ = vc.Disconnect() | 	vc.Disconnect() | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										57
									
								
								vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										57
									
								
								vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,38 +1,42 @@ | |||||||
| package main | package main | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"encoding/json" | ||||||
| 	"flag" | 	"flag" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  |  | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Variables used for command line options | // Variables used for command line options | ||||||
| var ( | var ( | ||||||
| 	Email    string |  | ||||||
| 	Password string |  | ||||||
| 	Token    string | 	Token    string | ||||||
| 	AppName  string | 	Name     string | ||||||
| 	DeleteID string | 	DeleteID string | ||||||
| 	ListOnly bool | 	ListOnly bool | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
|  |  | ||||||
| 	flag.StringVar(&Email, "e", "", "Account Email") | 	flag.StringVar(&Token, "t", "", "Owner Account Token") | ||||||
| 	flag.StringVar(&Password, "p", "", "Account Password") | 	flag.StringVar(&Name, "n", "", "Name to give App/Bot") | ||||||
| 	flag.StringVar(&Token, "t", "", "Account Token") |  | ||||||
| 	flag.StringVar(&DeleteID, "d", "", "Application ID to delete") | 	flag.StringVar(&DeleteID, "d", "", "Application ID to delete") | ||||||
| 	flag.BoolVar(&ListOnly, "l", false, "List Applications Only") | 	flag.BoolVar(&ListOnly, "l", false, "List Applications Only") | ||||||
| 	flag.StringVar(&AppName, "a", "", "App/Bot Name") |  | ||||||
| 	flag.Parse() | 	flag.Parse() | ||||||
|  |  | ||||||
|  | 	if Token == "" { | ||||||
|  | 		flag.Usage() | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
|  |  | ||||||
| 	var err error | 	var err error | ||||||
|  |  | ||||||
| 	// Create a new Discord session using the provided login information. | 	// Create a new Discord session using the provided login information. | ||||||
| 	dg, err := discordgo.New(Email, Password, Token) | 	dg, err := discordgo.New(Token) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		fmt.Println("error creating Discord session,", err) | 		fmt.Println("error creating Discord session,", err) | ||||||
| 		return | 		return | ||||||
| @@ -41,18 +45,17 @@ func main() { | |||||||
| 	// If -l set, only display a list of existing applications | 	// If -l set, only display a list of existing applications | ||||||
| 	// for the given account. | 	// for the given account. | ||||||
| 	if ListOnly { | 	if ListOnly { | ||||||
| 		aps, err2 := dg.Applications() |  | ||||||
| 		if err2 != nil { | 		aps, err := dg.Applications() | ||||||
|  | 		if err != nil { | ||||||
| 			fmt.Println("error fetching applications,", err) | 			fmt.Println("error fetching applications,", err) | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		for k, v := range aps { | 		for _, v := range aps { | ||||||
| 			fmt.Printf("%d : --------------------------------------\n", k) | 			fmt.Println("-----------------------------------------------------") | ||||||
| 			fmt.Printf("ID: %s\n", v.ID) | 			b, _ := json.MarshalIndent(v, "", " ") | ||||||
| 			fmt.Printf("Name: %s\n", v.Name) | 			fmt.Println(string(b)) | ||||||
| 			fmt.Printf("Secret: %s\n", v.Secret) |  | ||||||
| 			fmt.Printf("Description: %s\n", v.Description) |  | ||||||
| 		} | 		} | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -66,19 +69,23 @@ func main() { | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if Name == "" { | ||||||
|  | 		flag.Usage() | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// Create a new application. | 	// Create a new application. | ||||||
| 	ap := &discordgo.Application{} | 	ap := &discordgo.Application{} | ||||||
| 	ap.Name = AppName | 	ap.Name = Name | ||||||
| 	ap, err = dg.ApplicationCreate(ap) | 	ap, err = dg.ApplicationCreate(ap) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		fmt.Println("error creating new applicaiton,", err) | 		fmt.Println("error creating new application,", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	fmt.Printf("Application created successfully:\n") | 	fmt.Printf("Application created successfully:\n") | ||||||
| 	fmt.Printf("ID: %s\n", ap.ID) | 	b, _ := json.MarshalIndent(ap, "", " ") | ||||||
| 	fmt.Printf("Name: %s\n", ap.Name) | 	fmt.Println(string(b)) | ||||||
| 	fmt.Printf("Secret: %s\n\n", ap.Secret) |  | ||||||
|  |  | ||||||
| 	// Create the bot account under the application we just created | 	// Create the bot account under the application we just created | ||||||
| 	bot, err := dg.ApplicationBotCreate(ap.ID) | 	bot, err := dg.ApplicationBotCreate(ap.ID) | ||||||
| @@ -88,11 +95,9 @@ func main() { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	fmt.Printf("Bot account created successfully.\n") | 	fmt.Printf("Bot account created successfully.\n") | ||||||
| 	fmt.Printf("ID: %s\n", bot.ID) | 	b, _ = json.MarshalIndent(bot, "", " ") | ||||||
| 	fmt.Printf("Username: %s\n", bot.Username) | 	fmt.Println(string(b)) | ||||||
| 	fmt.Printf("Token: %s\n\n", bot.Token) |  | ||||||
| 	fmt.Println("Please save the above posted info in a secure place.") | 	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.") | 	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
									
									
								
							
							
						
						
									
										73
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/localfile/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,73 +0,0 @@ | |||||||
| 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) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										89
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										89
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/main.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,89 @@ | |||||||
|  | package main | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"encoding/base64" | ||||||
|  | 	"flag" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"net/http" | ||||||
|  | 	"os" | ||||||
|  |  | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Variables used for command line parameters | ||||||
|  | var ( | ||||||
|  | 	Token      string | ||||||
|  | 	AvatarFile string | ||||||
|  | 	AvatarURL  string | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  |  | ||||||
|  | 	flag.StringVar(&Token, "t", "", "Bot Token") | ||||||
|  | 	flag.StringVar(&AvatarFile, "f", "", "Avatar File Name") | ||||||
|  | 	flag.StringVar(&AvatarURL, "u", "", "URL to the avatar image") | ||||||
|  | 	flag.Parse() | ||||||
|  |  | ||||||
|  | 	if Token == "" || (AvatarFile == "" && AvatarURL == "") { | ||||||
|  | 		flag.Usage() | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func main() { | ||||||
|  |  | ||||||
|  | 	// Create a new Discord session using the provided login information. | ||||||
|  | 	dg, err := discordgo.New("Bot " + Token) | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println("error creating Discord session,", err) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Declare these here so they can be used in the below two if blocks and | ||||||
|  | 	// still carry over to the end of this function. | ||||||
|  | 	var base64img string | ||||||
|  | 	var contentType string | ||||||
|  |  | ||||||
|  | 	// If we're using a URL link for the Avatar | ||||||
|  | 	if AvatarURL != "" { | ||||||
|  |  | ||||||
|  | 		resp, err := http.Get(AvatarURL) | ||||||
|  | 		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 | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		contentType = http.DetectContentType(img) | ||||||
|  | 		base64img = base64.StdEncoding.EncodeToString(img) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If we're using a local file for the Avatar | ||||||
|  | 	if AvatarFile != "" { | ||||||
|  | 		img, err := ioutil.ReadFile(AvatarFile) | ||||||
|  | 		if err != nil { | ||||||
|  | 			fmt.Println(err) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		contentType = http.DetectContentType(img) | ||||||
|  | 		base64img = base64.StdEncoding.EncodeToString(img) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Now lets format our base64 image into the proper format Discord wants | ||||||
|  | 	// and then call UserUpdate to set it as our user's Avatar. | ||||||
|  | 	avatar := fmt.Sprintf("data:%s;base64,%s", contentType, base64img) | ||||||
|  | 	_, err = dg.UserUpdate("", "", "", avatar, "") | ||||||
|  | 	if err != nil { | ||||||
|  | 		fmt.Println(err) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										86
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/url/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										86
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/url/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,86 +0,0 @@ | |||||||
| 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) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| } |  | ||||||
							
								
								
									
										7
									
								
								vendor/github.com/bwmarrin/discordgo/examples/mytoken/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										7
									
								
								vendor/github.com/bwmarrin/discordgo/examples/mytoken/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,7 @@ package main | |||||||
| import ( | import ( | ||||||
| 	"flag" | 	"flag" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  |  | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
| ) | ) | ||||||
| @@ -18,6 +19,11 @@ func init() { | |||||||
| 	flag.StringVar(&Email, "e", "", "Account Email") | 	flag.StringVar(&Email, "e", "", "Account Email") | ||||||
| 	flag.StringVar(&Password, "p", "", "Account Password") | 	flag.StringVar(&Password, "p", "", "Account Password") | ||||||
| 	flag.Parse() | 	flag.Parse() | ||||||
|  |  | ||||||
|  | 	if Email == "" || Password == "" { | ||||||
|  | 		flag.Usage() | ||||||
|  | 		os.Exit(1) | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
| @@ -29,5 +35,6 @@ func main() { | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Print out your token. | ||||||
| 	fmt.Printf("Your Authentication Token is:\n\n%s\n", dg.Token) | 	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
									
									
								
							
							
						
						
									
										53
									
								
								vendor/github.com/bwmarrin/discordgo/examples/new_basic/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,53 +0,0 @@ | |||||||
| 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) |  | ||||||
| } |  | ||||||
							
								
								
									
										35
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										35
									
								
								vendor/github.com/bwmarrin/discordgo/examples/pingpong/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,9 @@ package main | |||||||
| import ( | import ( | ||||||
| 	"flag" | 	"flag" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"os/signal" | ||||||
|  | 	"syscall" | ||||||
|  |  | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"github.com/bwmarrin/discordgo" | ||||||
| ) | ) | ||||||
| @@ -10,7 +13,6 @@ import ( | |||||||
| // Variables used for command line parameters | // Variables used for command line parameters | ||||||
| var ( | var ( | ||||||
| 	Token string | 	Token string | ||||||
| 	BotID string |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| @@ -28,29 +30,24 @@ func main() { | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Get the account information. | 	// Register the messageCreate func as a callback for MessageCreate events. | ||||||
| 	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) | 	dg.AddHandler(messageCreate) | ||||||
|  |  | ||||||
| 	// Open the websocket and begin listening. | 	// Open a websocket connection to Discord and begin listening. | ||||||
| 	err = dg.Open() | 	err = dg.Open() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		fmt.Println("error opening connection,", err) | 		fmt.Println("error opening connection,", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Wait here until CTRL-C or other term signal is received. | ||||||
| 	fmt.Println("Bot is now running.  Press CTRL-C to exit.") | 	fmt.Println("Bot is now running.  Press CTRL-C to exit.") | ||||||
| 	// Simple way to keep program running until CTRL-C is pressed. | 	sc := make(chan os.Signal, 1) | ||||||
| 	<-make(chan struct{}) | 	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) | ||||||
| 	return | 	<-sc | ||||||
|  |  | ||||||
|  | 	// Cleanly close down the Discord session. | ||||||
|  | 	dg.Close() | ||||||
| } | } | ||||||
|  |  | ||||||
| // This function will be called (due to AddHandler above) every time a new | // This function will be called (due to AddHandler above) every time a new | ||||||
| @@ -58,17 +55,17 @@ func main() { | |||||||
| func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | ||||||
|  |  | ||||||
| 	// Ignore all messages created by the bot itself | 	// Ignore all messages created by the bot itself | ||||||
| 	if m.Author.ID == BotID { | 	// This isn't required in this specific example but it's a good practice. | ||||||
|  | 	if m.Author.ID == s.State.User.ID { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// If the message is "ping" reply with "Pong!" | 	// If the message is "ping" reply with "Pong!" | ||||||
| 	if m.Content == "ping" { | 	if m.Content == "ping" { | ||||||
| 		_, _ = s.ChannelMessageSend(m.ChannelID, "Pong!") | 		s.ChannelMessageSend(m.ChannelID, "Pong!") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// If the message is "pong" reply with "Ping!" | 	// If the message is "pong" reply with "Ping!" | ||||||
| 	if m.Content == "pong" { | 	if m.Content == "pong" { | ||||||
| 		_, _ = s.ChannelMessageSend(m.ChannelID, "Ping!") | 		s.ChannelMessageSend(m.ChannelID, "Ping!") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										12
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										12
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -23,7 +23,7 @@ const ( | |||||||
| 	LogError int = iota | 	LogError int = iota | ||||||
|  |  | ||||||
| 	// LogWarning level is used for very abnormal events and errors that are | 	// LogWarning level is used for very abnormal events and errors that are | ||||||
| 	// also returend to a calling function. | 	// also returned to a calling function. | ||||||
| 	LogWarning | 	LogWarning | ||||||
|  |  | ||||||
| 	// LogInformational level is used for normal non-error activity | 	// LogInformational level is used for normal non-error activity | ||||||
| @@ -34,14 +34,21 @@ const ( | |||||||
| 	LogDebug | 	LogDebug | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Logger can be used to replace the standard logging for discordgo | ||||||
|  | var Logger func(msgL, caller int, format string, a ...interface{}) | ||||||
|  |  | ||||||
| // msglog provides package wide logging consistancy for discordgo | // msglog provides package wide logging consistancy for discordgo | ||||||
| // the format, a...  portion this command follows that of fmt.Printf | // the format, a...  portion this command follows that of fmt.Printf | ||||||
| //   msgL   : LogLevel of the message | //   msgL   : LogLevel of the message | ||||||
| //   caller : 1 + the number of callers away from the message source | //   caller : 1 + the number of callers away from the message source | ||||||
| //   format : Printf style message format | //   format : Printf style message format | ||||||
| //   a ...  : comma seperated list of values to pass | //   a ...  : comma separated list of values to pass | ||||||
| func msglog(msgL, caller int, format string, a ...interface{}) { | func msglog(msgL, caller int, format string, a ...interface{}) { | ||||||
|  |  | ||||||
|  | 	if Logger != nil { | ||||||
|  | 		Logger(msgL, caller, format, a...) | ||||||
|  | 	} else { | ||||||
|  |  | ||||||
| 		pc, file, line, _ := runtime.Caller(caller) | 		pc, file, line, _ := runtime.Caller(caller) | ||||||
|  |  | ||||||
| 		files := strings.Split(file, "/") | 		files := strings.Split(file, "/") | ||||||
| @@ -55,6 +62,7 @@ func msglog(msgL, caller int, format string, a ...interface{}) { | |||||||
|  |  | ||||||
| 		log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg) | 		log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg) | ||||||
| 	} | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // helper function that wraps msglog for the Session struct | // helper function that wraps msglog for the Session struct | ||||||
| // This adds a check to insure the message is only logged | // This adds a check to insure the message is only logged | ||||||
|   | |||||||
							
								
								
									
										136
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										136
									
								
								vendor/github.com/bwmarrin/discordgo/message.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -10,8 +10,24 @@ | |||||||
| package discordgo | package discordgo | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"io" | ||||||
| 	"regexp" | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // MessageType is the type of Message | ||||||
|  | type MessageType int | ||||||
|  |  | ||||||
|  | // Block contains the valid known MessageType values | ||||||
|  | const ( | ||||||
|  | 	MessageTypeDefault MessageType = iota | ||||||
|  | 	MessageTypeRecipientAdd | ||||||
|  | 	MessageTypeRecipientRemove | ||||||
|  | 	MessageTypeCall | ||||||
|  | 	MessageTypeChannelNameChange | ||||||
|  | 	MessageTypeChannelIconChange | ||||||
|  | 	MessageTypeChannelPinnedMessage | ||||||
|  | 	MessageTypeGuildMemberJoin | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // A Message stores all data related to a specific Discord message. | // A Message stores all data related to a specific Discord message. | ||||||
| @@ -29,6 +45,58 @@ type Message struct { | |||||||
| 	Embeds          []*MessageEmbed      `json:"embeds"` | 	Embeds          []*MessageEmbed      `json:"embeds"` | ||||||
| 	Mentions        []*User              `json:"mentions"` | 	Mentions        []*User              `json:"mentions"` | ||||||
| 	Reactions       []*MessageReactions  `json:"reactions"` | 	Reactions       []*MessageReactions  `json:"reactions"` | ||||||
|  | 	Type            MessageType          `json:"type"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // File stores info about files you e.g. send in messages. | ||||||
|  | type File struct { | ||||||
|  | 	Name        string | ||||||
|  | 	ContentType string | ||||||
|  | 	Reader      io.Reader | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageSend stores all parameters you can send with ChannelMessageSendComplex. | ||||||
|  | type MessageSend struct { | ||||||
|  | 	Content string        `json:"content,omitempty"` | ||||||
|  | 	Embed   *MessageEmbed `json:"embed,omitempty"` | ||||||
|  | 	Tts     bool          `json:"tts"` | ||||||
|  | 	Files   []*File       `json:"-"` | ||||||
|  |  | ||||||
|  | 	// TODO: Remove this when compatibility is not required. | ||||||
|  | 	File *File `json:"-"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageEdit is used to chain parameters via ChannelMessageEditComplex, which | ||||||
|  | // is also where you should get the instance from. | ||||||
|  | type MessageEdit struct { | ||||||
|  | 	Content *string       `json:"content,omitempty"` | ||||||
|  | 	Embed   *MessageEmbed `json:"embed,omitempty"` | ||||||
|  |  | ||||||
|  | 	ID      string | ||||||
|  | 	Channel string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewMessageEdit returns a MessageEdit struct, initialized | ||||||
|  | // with the Channel and ID. | ||||||
|  | func NewMessageEdit(channelID string, messageID string) *MessageEdit { | ||||||
|  | 	return &MessageEdit{ | ||||||
|  | 		Channel: channelID, | ||||||
|  | 		ID:      messageID, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetContent is the same as setting the variable Content, | ||||||
|  | // except it doesn't take a pointer. | ||||||
|  | func (m *MessageEdit) SetContent(str string) *MessageEdit { | ||||||
|  | 	m.Content = &str | ||||||
|  | 	return m | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetEmbed is a convenience function for setting the embed, | ||||||
|  | // so you can chain commands. | ||||||
|  | func (m *MessageEdit) SetEmbed(embed *MessageEmbed) *MessageEdit { | ||||||
|  | 	m.Embed = embed | ||||||
|  | 	return m | ||||||
| } | } | ||||||
|  |  | ||||||
| // A MessageAttachment stores data for message attachments. | // A MessageAttachment stores data for message attachments. | ||||||
| @@ -120,13 +188,65 @@ type MessageReactions struct { | |||||||
|  |  | ||||||
| // ContentWithMentionsReplaced will replace all @<id> mentions with the | // ContentWithMentionsReplaced will replace all @<id> mentions with the | ||||||
| // username of the mention. | // username of the mention. | ||||||
| func (m *Message) ContentWithMentionsReplaced() string { | func (m *Message) ContentWithMentionsReplaced() (content string) { | ||||||
| 	if m.Mentions == nil { | 	content = m.Content | ||||||
| 		return m.Content |  | ||||||
| 	} |  | ||||||
| 	content := m.Content |  | ||||||
| 	for _, user := range m.Mentions { | 	for _, user := range m.Mentions { | ||||||
| 		content = regexp.MustCompile(fmt.Sprintf("<@!?(%s)>", user.ID)).ReplaceAllString(content, "@"+user.Username) | 		content = strings.NewReplacer( | ||||||
|  | 			"<@"+user.ID+">", "@"+user.Username, | ||||||
|  | 			"<@!"+user.ID+">", "@"+user.Username, | ||||||
|  | 		).Replace(content) | ||||||
| 	} | 	} | ||||||
| 	return content | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var patternChannels = regexp.MustCompile("<#[^>]*>") | ||||||
|  |  | ||||||
|  | // ContentWithMoreMentionsReplaced will replace all @<id> mentions with the | ||||||
|  | // username of the mention, but also role IDs and more. | ||||||
|  | func (m *Message) ContentWithMoreMentionsReplaced(s *Session) (content string, err error) { | ||||||
|  | 	content = m.Content | ||||||
|  |  | ||||||
|  | 	if !s.StateEnabled { | ||||||
|  | 		content = m.ContentWithMentionsReplaced() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel, err := s.State.Channel(m.ChannelID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		content = m.ContentWithMentionsReplaced() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, user := range m.Mentions { | ||||||
|  | 		nick := user.Username | ||||||
|  |  | ||||||
|  | 		member, err := s.State.Member(channel.GuildID, user.ID) | ||||||
|  | 		if err == nil && member.Nick != "" { | ||||||
|  | 			nick = member.Nick | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		content = strings.NewReplacer( | ||||||
|  | 			"<@"+user.ID+">", "@"+user.Username, | ||||||
|  | 			"<@!"+user.ID+">", "@"+nick, | ||||||
|  | 		).Replace(content) | ||||||
|  | 	} | ||||||
|  | 	for _, roleID := range m.MentionRoles { | ||||||
|  | 		role, err := s.State.Role(channel.GuildID, roleID) | ||||||
|  | 		if err != nil || !role.Mentionable { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		content = strings.Replace(content, "<@&"+role.ID+">", "@"+role.Name, -1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	content = patternChannels.ReplaceAllStringFunc(content, func(mention string) string { | ||||||
|  | 		channel, err := s.State.Channel(mention[2 : len(mention)-1]) | ||||||
|  | 		if err != nil || channel.Type == ChannelTypeGuildVoice { | ||||||
|  | 			return mention | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return "#" + channel.Name | ||||||
|  | 	}) | ||||||
|  | 	return | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										5
									
								
								vendor/github.com/bwmarrin/discordgo/oauth2.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										5
									
								
								vendor/github.com/bwmarrin/discordgo/oauth2.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -21,7 +21,12 @@ type Application struct { | |||||||
| 	Icon                string    `json:"icon,omitempty"` | 	Icon                string    `json:"icon,omitempty"` | ||||||
| 	Secret              string    `json:"secret,omitempty"` | 	Secret              string    `json:"secret,omitempty"` | ||||||
| 	RedirectURIs        *[]string `json:"redirect_uris,omitempty"` | 	RedirectURIs        *[]string `json:"redirect_uris,omitempty"` | ||||||
|  | 	BotRequireCodeGrant bool      `json:"bot_require_code_grant,omitempty"` | ||||||
|  | 	BotPublic           bool      `json:"bot_public,omitempty"` | ||||||
|  | 	RPCApplicationState int       `json:"rpc_application_state,omitempty"` | ||||||
|  | 	Flags               int       `json:"flags,omitempty"` | ||||||
| 	Owner               *User     `json:"owner"` | 	Owner               *User     `json:"owner"` | ||||||
|  | 	Bot                 *User     `json:"bot"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Application returns an Application structure of a specific Application | // Application returns an Application structure of a specific Application | ||||||
|   | |||||||
							
								
								
									
										139
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										139
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,16 +3,26 @@ package discordgo | |||||||
| import ( | import ( | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  | 	"sync/atomic" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // customRateLimit holds information for defining a custom rate limit | ||||||
|  | type customRateLimit struct { | ||||||
|  | 	suffix   string | ||||||
|  | 	requests int | ||||||
|  | 	reset    time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
| // RateLimiter holds all ratelimit buckets | // RateLimiter holds all ratelimit buckets | ||||||
| type RateLimiter struct { | type RateLimiter struct { | ||||||
| 	sync.Mutex | 	sync.Mutex | ||||||
| 	global          *Bucket | 	global           *int64 | ||||||
| 	buckets          map[string]*Bucket | 	buckets          map[string]*Bucket | ||||||
| 	globalRateLimit  time.Duration | 	globalRateLimit  time.Duration | ||||||
|  | 	customRateLimits []*customRateLimit | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewRatelimiter returns a new RateLimiter | // NewRatelimiter returns a new RateLimiter | ||||||
| @@ -20,12 +30,19 @@ func NewRatelimiter() *RateLimiter { | |||||||
|  |  | ||||||
| 	return &RateLimiter{ | 	return &RateLimiter{ | ||||||
| 		buckets: make(map[string]*Bucket), | 		buckets: make(map[string]*Bucket), | ||||||
| 		global:  &Bucket{Key: "global"}, | 		global:  new(int64), | ||||||
|  | 		customRateLimits: []*customRateLimit{ | ||||||
|  | 			&customRateLimit{ | ||||||
|  | 				suffix:   "//reactions//", | ||||||
|  | 				requests: 1, | ||||||
|  | 				reset:    200 * time.Millisecond, | ||||||
|  | 			}, | ||||||
|  | 		}, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // getBucket retrieves or creates a bucket | // GetBucket retrieves or creates a bucket | ||||||
| func (r *RateLimiter) getBucket(key string) *Bucket { | func (r *RateLimiter) GetBucket(key string) *Bucket { | ||||||
| 	r.Lock() | 	r.Lock() | ||||||
| 	defer r.Unlock() | 	defer r.Unlock() | ||||||
|  |  | ||||||
| @@ -34,34 +51,54 @@ func (r *RateLimiter) getBucket(key string) *Bucket { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b := &Bucket{ | 	b := &Bucket{ | ||||||
| 		remaining: 1, | 		Remaining: 1, | ||||||
| 		Key:       key, | 		Key:       key, | ||||||
| 		global:    r.global, | 		global:    r.global, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Check if there is a custom ratelimit set for this bucket ID. | ||||||
|  | 	for _, rl := range r.customRateLimits { | ||||||
|  | 		if strings.HasSuffix(b.Key, rl.suffix) { | ||||||
|  | 			b.customRateLimit = rl | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	r.buckets[key] = b | 	r.buckets[key] = b | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| // LockBucket Locks until a request can be made | // GetWaitTime returns the duration you should wait for a Bucket | ||||||
| func (r *RateLimiter) LockBucket(bucketID string) *Bucket { | func (r *RateLimiter) GetWaitTime(b *Bucket, minRemaining int) time.Duration { | ||||||
|  |  | ||||||
| 	b := r.getBucket(bucketID) |  | ||||||
|  |  | ||||||
| 	b.Lock() |  | ||||||
|  |  | ||||||
| 	// If we ran out of calls and the reset time is still ahead of us | 	// If we ran out of calls and the reset time is still ahead of us | ||||||
| 	// then we need to take it easy and relax a little | 	// then we need to take it easy and relax a little | ||||||
| 	if b.remaining < 1 && b.reset.After(time.Now()) { | 	if b.Remaining < minRemaining && b.reset.After(time.Now()) { | ||||||
| 		time.Sleep(b.reset.Sub(time.Now())) | 		return b.reset.Sub(time.Now()) | ||||||
|  |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Check for global ratelimits | 	// Check for global ratelimits | ||||||
| 	r.global.Lock() | 	sleepTo := time.Unix(0, atomic.LoadInt64(r.global)) | ||||||
| 	r.global.Unlock() | 	if now := time.Now(); now.Before(sleepTo) { | ||||||
|  | 		return sleepTo.Sub(now) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	b.remaining-- | 	return 0 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LockBucket Locks until a request can be made | ||||||
|  | func (r *RateLimiter) LockBucket(bucketID string) *Bucket { | ||||||
|  | 	return r.LockBucketObject(r.GetBucket(bucketID)) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LockBucketObject Locks an already resolved bucket until a request can be made | ||||||
|  | func (r *RateLimiter) LockBucketObject(b *Bucket) *Bucket { | ||||||
|  | 	b.Lock() | ||||||
|  |  | ||||||
|  | 	if wait := r.GetWaitTime(b, 1); wait > 0 { | ||||||
|  | 		time.Sleep(wait) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.Remaining-- | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -69,17 +106,33 @@ func (r *RateLimiter) LockBucket(bucketID string) *Bucket { | |||||||
| type Bucket struct { | type Bucket struct { | ||||||
| 	sync.Mutex | 	sync.Mutex | ||||||
| 	Key       string | 	Key       string | ||||||
| 	remaining int | 	Remaining int | ||||||
| 	limit     int | 	limit     int | ||||||
| 	reset     time.Time | 	reset     time.Time | ||||||
| 	global    *Bucket | 	global    *int64 | ||||||
|  |  | ||||||
|  | 	lastReset       time.Time | ||||||
|  | 	customRateLimit *customRateLimit | ||||||
|  | 	Userdata        interface{} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Release unlocks the bucket and reads the headers to update the buckets ratelimit info | // Release unlocks the bucket and reads the headers to update the buckets ratelimit info | ||||||
| // and locks up the whole thing in case if there's a global ratelimit. | // and locks up the whole thing in case if there's a global ratelimit. | ||||||
| func (b *Bucket) Release(headers http.Header) error { | func (b *Bucket) Release(headers http.Header) error { | ||||||
|  |  | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the bucket uses a custom ratelimiter | ||||||
|  | 	if rl := b.customRateLimit; rl != nil { | ||||||
|  | 		if time.Now().Sub(b.lastReset) >= rl.reset { | ||||||
|  | 			b.Remaining = rl.requests - 1 | ||||||
|  | 			b.lastReset = time.Now() | ||||||
|  | 		} | ||||||
|  | 		if b.Remaining < 1 { | ||||||
|  | 			b.reset = time.Now().Add(rl.reset) | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if headers == nil { | 	if headers == nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| @@ -89,41 +142,25 @@ func (b *Bucket) Release(headers http.Header) error { | |||||||
| 	global := headers.Get("X-RateLimit-Global") | 	global := headers.Get("X-RateLimit-Global") | ||||||
| 	retryAfter := headers.Get("Retry-After") | 	retryAfter := headers.Get("Retry-After") | ||||||
|  |  | ||||||
| 	// If it's global just keep the main ratelimit mutex locked | 	// Update global and per bucket reset time if the proper headers are available | ||||||
| 	if global != "" { | 	// If global is set, then it will block all buckets until after Retry-After | ||||||
| 		parsedAfter, err := strconv.Atoi(retryAfter) | 	// If Retry-After without global is provided it will use that for the new reset | ||||||
| 		if err != nil { | 	// time since it's more accurate than X-RateLimit-Reset. | ||||||
| 			return err | 	// If Retry-After after is not proided, it will update the reset time from X-RateLimit-Reset | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Lock it in a new goroutine so that this isn't a blocking call |  | ||||||
| 		go func() { |  | ||||||
| 			// Make sure if several requests were waiting we don't sleep for n * retry-after |  | ||||||
| 			// where n is the amount of requests that were going on |  | ||||||
| 			sleepTo := time.Now().Add(time.Duration(parsedAfter) * time.Millisecond) |  | ||||||
|  |  | ||||||
| 			b.global.Lock() |  | ||||||
|  |  | ||||||
| 			sleepDuration := sleepTo.Sub(time.Now()) |  | ||||||
| 			if sleepDuration > 0 { |  | ||||||
| 				time.Sleep(sleepDuration) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			b.global.Unlock() |  | ||||||
| 		}() |  | ||||||
|  |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Update reset time if either retry after or reset headers are present |  | ||||||
| 	// Prefer retryafter because it's more accurate with time sync and whatnot |  | ||||||
| 	if retryAfter != "" { | 	if retryAfter != "" { | ||||||
| 		parsedAfter, err := strconv.ParseInt(retryAfter, 10, 64) | 		parsedAfter, err := strconv.ParseInt(retryAfter, 10, 64) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		b.reset = time.Now().Add(time.Duration(parsedAfter) * time.Millisecond) |  | ||||||
|  |  | ||||||
|  | 		resetAt := time.Now().Add(time.Duration(parsedAfter) * time.Millisecond) | ||||||
|  |  | ||||||
|  | 		// Lock either this single bucket or all buckets | ||||||
|  | 		if global != "" { | ||||||
|  | 			atomic.StoreInt64(b.global, resetAt.UnixNano()) | ||||||
|  | 		} else { | ||||||
|  | 			b.reset = resetAt | ||||||
|  | 		} | ||||||
| 	} else if reset != "" { | 	} else if reset != "" { | ||||||
| 		// Calculate the reset time by using the date header returned from discord | 		// Calculate the reset time by using the date header returned from discord | ||||||
| 		discordTime, err := http.ParseTime(headers.Get("Date")) | 		discordTime, err := http.ParseTime(headers.Get("Date")) | ||||||
| @@ -150,7 +187,7 @@ func (b *Bucket) Release(headers http.Header) error { | |||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		b.remaining = int(parsedRemaining) | 		b.Remaining = int(parsedRemaining) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
|   | |||||||
							
								
								
									
										498
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										498
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -23,14 +23,22 @@ import ( | |||||||
| 	"log" | 	"log" | ||||||
| 	"mime/multipart" | 	"mime/multipart" | ||||||
| 	"net/http" | 	"net/http" | ||||||
|  | 	"net/textproto" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // ErrJSONUnmarshal is returned for JSON Unmarshall errors. | // All error constants | ||||||
| var ErrJSONUnmarshal = errors.New("json unmarshal") | var ( | ||||||
|  | 	ErrJSONUnmarshal           = errors.New("json unmarshal") | ||||||
|  | 	ErrStatusOffline           = errors.New("You can't set your Status to offline") | ||||||
|  | 	ErrVerificationLevelBounds = errors.New("VerificationLevel out of bounds, should be between 0 and 3") | ||||||
|  | 	ErrPruneDaysBounds         = errors.New("the number of days should be more than or equal to 1") | ||||||
|  | 	ErrGuildNoIcon             = errors.New("guild does not have an icon set") | ||||||
|  | 	ErrGuildNoSplash           = errors.New("guild does not have a splash set") | ||||||
|  | ) | ||||||
|  |  | ||||||
| // Request is the same as RequestWithBucketID but the bucket id is the same as the urlStr | // Request is the same as RequestWithBucketID but the bucket id is the same as the urlStr | ||||||
| func (s *Session) Request(method, urlStr string, data interface{}) (response []byte, err error) { | func (s *Session) Request(method, urlStr string, data interface{}) (response []byte, err error) { | ||||||
| @@ -57,9 +65,11 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID | |||||||
| 	if bucketID == "" { | 	if bucketID == "" { | ||||||
| 		bucketID = strings.SplitN(urlStr, "?", 2)[0] | 		bucketID = strings.SplitN(urlStr, "?", 2)[0] | ||||||
| 	} | 	} | ||||||
|  | 	return s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucket(bucketID), sequence) | ||||||
|  | } | ||||||
|  |  | ||||||
| 	bucket := s.ratelimiter.LockBucket(bucketID) | // RequestWithLockedBucket makes a request using a bucket that's already been locked | ||||||
|  | func (s *Session) RequestWithLockedBucket(method, urlStr, contentType string, b []byte, bucket *Bucket, sequence int) (response []byte, err error) { | ||||||
| 	if s.Debug { | 	if s.Debug { | ||||||
| 		log.Printf("API REQUEST %8s :: %s\n", method, urlStr) | 		log.Printf("API REQUEST %8s :: %s\n", method, urlStr) | ||||||
| 		log.Printf("API REQUEST  PAYLOAD :: [%s]\n", string(b)) | 		log.Printf("API REQUEST  PAYLOAD :: [%s]\n", string(b)) | ||||||
| @@ -87,9 +97,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	client := &http.Client{Timeout: (20 * time.Second)} | 	resp, err := s.Client.Do(req) | ||||||
|  |  | ||||||
| 	resp, err := client.Do(req) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		bucket.Release(nil) | 		bucket.Release(nil) | ||||||
| 		return | 		return | ||||||
| @@ -133,7 +141,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID | |||||||
| 		if sequence < s.MaxRestRetries { | 		if sequence < s.MaxRestRetries { | ||||||
|  |  | ||||||
| 			s.log(LogInformational, "%s Failed (%s), Retrying...", urlStr, resp.Status) | 			s.log(LogInformational, "%s Failed (%s), Retrying...", urlStr, resp.Status) | ||||||
| 			response, err = s.request(method, urlStr, contentType, b, bucketID, sequence+1) | 			response, err = s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucketObject(bucket), sequence+1) | ||||||
| 		} else { | 		} else { | ||||||
| 			err = fmt.Errorf("Exceeded Max retries HTTP %s, %s", resp.Status, response) | 			err = fmt.Errorf("Exceeded Max retries HTTP %s, %s", resp.Status, response) | ||||||
| 		} | 		} | ||||||
| @@ -152,7 +160,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID | |||||||
| 		// we can make the above smarter | 		// we can make the above smarter | ||||||
| 		// this method can cause longer delays than required | 		// this method can cause longer delays than required | ||||||
|  |  | ||||||
| 		response, err = s.request(method, urlStr, contentType, b, bucketID, sequence) | 		response, err = s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucketObject(bucket), sequence) | ||||||
|  |  | ||||||
| 	default: // Error condition | 	default: // Error condition | ||||||
| 		err = newRestError(req, resp, response) | 		err = newRestError(req, resp, response) | ||||||
| @@ -175,6 +183,12 @@ func unmarshal(data []byte, v interface{}) error { | |||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
|  |  | ||||||
| // Login asks the Discord server for an authentication token. | // Login asks the Discord server for an authentication token. | ||||||
|  | // | ||||||
|  | // NOTE: While email/pass authentication is supported by DiscordGo it is | ||||||
|  | // HIGHLY DISCOURAGED by Discord. Please only use email/pass to obtain a token | ||||||
|  | // and then use that authentication token for all future connections. | ||||||
|  | // Also, doing any form of automation with a user (non Bot) account may result | ||||||
|  | // in that account being permanently banned from Discord. | ||||||
| func (s *Session) Login(email, password string) (err error) { | func (s *Session) Login(email, password string) (err error) { | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| @@ -189,6 +203,7 @@ func (s *Session) Login(email, password string) (err error) { | |||||||
|  |  | ||||||
| 	temp := struct { | 	temp := struct { | ||||||
| 		Token string `json:"token"` | 		Token string `json:"token"` | ||||||
|  | 		MFA   bool   `json:"mfa"` | ||||||
| 	}{} | 	}{} | ||||||
|  |  | ||||||
| 	err = unmarshal(response, &temp) | 	err = unmarshal(response, &temp) | ||||||
| @@ -197,6 +212,7 @@ func (s *Session) Login(email, password string) (err error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	s.Token = temp.Token | 	s.Token = temp.Token | ||||||
|  | 	s.MFA = temp.MFA | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -264,15 +280,21 @@ func (s *Session) User(userID string) (st *User, err error) { | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // UserAvatar returns an image.Image of a users Avatar. | // UserAvatar is deprecated. Please use UserAvatarDecode | ||||||
| // userID    : A user ID or "@me" which is a shortcut of current user ID | // userID    : A user ID or "@me" which is a shortcut of current user ID | ||||||
| func (s *Session) UserAvatar(userID string) (img image.Image, err error) { | func (s *Session) UserAvatar(userID string) (img image.Image, err error) { | ||||||
| 	u, err := s.User(userID) | 	u, err := s.User(userID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  | 	img, err = s.UserAvatarDecode(u) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("GET", EndpointUserAvatar(userID, u.Avatar), nil, EndpointUserAvatar("", "")) | // UserAvatarDecode returns an image.Image of a user's Avatar | ||||||
|  | // user : The user which avatar should be retrieved | ||||||
|  | func (s *Session) UserAvatarDecode(u *User) (img image.Image, err error) { | ||||||
|  | 	body, err := s.RequestWithBucketID("GET", EndpointUserAvatar(u.ID, u.Avatar), nil, EndpointUserAvatar("", "")) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -290,9 +312,9 @@ func (s *Session) UserUpdate(email, password, username, avatar, newPassword stri | |||||||
| 	// If left blank, avatar will be set to null/blank | 	// If left blank, avatar will be set to null/blank | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| 		Email       string `json:"email"` | 		Email       string `json:"email,omitempty"` | ||||||
| 		Password    string `json:"password"` | 		Password    string `json:"password,omitempty"` | ||||||
| 		Username    string `json:"username"` | 		Username    string `json:"username,omitempty"` | ||||||
| 		Avatar      string `json:"avatar,omitempty"` | 		Avatar      string `json:"avatar,omitempty"` | ||||||
| 		NewPassword string `json:"new_password,omitempty"` | 		NewPassword string `json:"new_password,omitempty"` | ||||||
| 	}{email, password, username, avatar, newPassword} | 	}{email, password, username, avatar, newPassword} | ||||||
| @@ -322,7 +344,7 @@ func (s *Session) UserSettings() (st *Settings, err error) { | |||||||
| // status   : The new status (Actual valid status are 'online','idle','dnd','invisible') | // status   : The new status (Actual valid status are 'online','idle','dnd','invisible') | ||||||
| func (s *Session) UserUpdateStatus(status Status) (st *Settings, err error) { | func (s *Session) UserUpdateStatus(status Status) (st *Settings, err error) { | ||||||
| 	if status == StatusOffline { | 	if status == StatusOffline { | ||||||
| 		err = errors.New("You can't set your Status to offline") | 		err = ErrStatusOffline | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -370,9 +392,30 @@ func (s *Session) UserChannelCreate(recipientID string) (st *Channel, err error) | |||||||
| } | } | ||||||
|  |  | ||||||
| // UserGuilds returns an array of UserGuild structures for all guilds. | // UserGuilds returns an array of UserGuild structures for all guilds. | ||||||
| func (s *Session) UserGuilds() (st []*UserGuild, err error) { | // limit     : The number guilds that can be returned. (max 100) | ||||||
|  | // beforeID  : If provided all guilds returned will be before given ID. | ||||||
|  | // afterID   : If provided all guilds returned will be after given ID. | ||||||
|  | func (s *Session) UserGuilds(limit int, beforeID, afterID string) (st []*UserGuild, err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("GET", EndpointUserGuilds("@me"), nil, EndpointUserGuilds("")) | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if limit > 0 { | ||||||
|  | 		v.Set("limit", strconv.Itoa(limit)) | ||||||
|  | 	} | ||||||
|  | 	if afterID != "" { | ||||||
|  | 		v.Set("after", afterID) | ||||||
|  | 	} | ||||||
|  | 	if beforeID != "" { | ||||||
|  | 		v.Set("before", beforeID) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	uri := EndpointUserGuilds("@me") | ||||||
|  |  | ||||||
|  | 	if len(v) > 0 { | ||||||
|  | 		uri = fmt.Sprintf("%s?%s", uri, v.Encode()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	body, err := s.RequestWithBucketID("GET", uri, nil, EndpointUserGuilds("")) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -402,6 +445,13 @@ func (s *Session) UserGuildSettingsEdit(guildID string, settings *UserGuildSetti | |||||||
| // NOTE: This function is now deprecated and will be removed in the future. | // NOTE: This function is now deprecated and will be removed in the future. | ||||||
| // Please see the same function inside state.go | // Please see the same function inside state.go | ||||||
| func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions int, err error) { | func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions int, err error) { | ||||||
|  | 	// Try to just get permissions from state. | ||||||
|  | 	apermissions, err = s.State.UserChannelPermissions(userID, channelID) | ||||||
|  | 	if err == nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Otherwise try get as much data from state as possible, falling back to the network. | ||||||
| 	channel, err := s.State.Channel(channelID) | 	channel, err := s.State.Channel(channelID) | ||||||
| 	if err != nil || channel == nil { | 	if err != nil || channel == nil { | ||||||
| 		channel, err = s.Channel(channelID) | 		channel, err = s.Channel(channelID) | ||||||
| @@ -431,6 +481,19 @@ func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	return memberPermissions(guild, channel, member), nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Calculates the permissions for a member. | ||||||
|  | // https://support.discordapp.com/hc/en-us/articles/206141927-How-is-the-permission-hierarchy-structured- | ||||||
|  | func memberPermissions(guild *Guild, channel *Channel, member *Member) (apermissions int) { | ||||||
|  | 	userID := member.User.ID | ||||||
|  |  | ||||||
|  | 	if userID == guild.OwnerID { | ||||||
|  | 		apermissions = PermissionAll | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, role := range guild.Roles { | 	for _, role := range guild.Roles { | ||||||
| 		if role.ID == guild.ID { | 		if role.ID == guild.ID { | ||||||
| 			apermissions |= role.Permissions | 			apermissions |= role.Permissions | ||||||
| @@ -447,20 +510,35 @@ func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if apermissions&PermissionAdministrator > 0 { | 	if apermissions&PermissionAdministrator == PermissionAdministrator { | ||||||
| 		apermissions |= PermissionAll | 		apermissions |= PermissionAll | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Member overwrites can override role overrides, so do two passes | 	// Apply @everyone overrides from the channel. | ||||||
| 	for _, overwrite := range channel.PermissionOverwrites { | 	for _, overwrite := range channel.PermissionOverwrites { | ||||||
| 		for _, roleID := range member.Roles { | 		if guild.ID == overwrite.ID { | ||||||
| 			if overwrite.Type == "role" && roleID == overwrite.ID { |  | ||||||
| 			apermissions &= ^overwrite.Deny | 			apermissions &= ^overwrite.Deny | ||||||
| 			apermissions |= overwrite.Allow | 			apermissions |= overwrite.Allow | ||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	denies := 0 | ||||||
|  | 	allows := 0 | ||||||
|  |  | ||||||
|  | 	// Member overwrites can override role overrides, so do two passes | ||||||
|  | 	for _, overwrite := range channel.PermissionOverwrites { | ||||||
|  | 		for _, roleID := range member.Roles { | ||||||
|  | 			if overwrite.Type == "role" && roleID == overwrite.ID { | ||||||
|  | 				denies |= overwrite.Deny | ||||||
|  | 				allows |= overwrite.Allow | ||||||
|  | 				break | ||||||
| 			} | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	apermissions &= ^denies | ||||||
|  | 	apermissions |= allows | ||||||
|  |  | ||||||
| 	for _, overwrite := range channel.PermissionOverwrites { | 	for _, overwrite := range channel.PermissionOverwrites { | ||||||
| 		if overwrite.Type == "member" && overwrite.ID == userID { | 		if overwrite.Type == "member" && overwrite.ID == userID { | ||||||
| @@ -470,11 +548,11 @@ func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if apermissions&PermissionAdministrator > 0 { | 	if apermissions&PermissionAdministrator == PermissionAdministrator { | ||||||
| 		apermissions |= PermissionAllChannel | 		apermissions |= PermissionAllChannel | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return | 	return apermissions | ||||||
| } | } | ||||||
|  |  | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
| @@ -509,7 +587,7 @@ func (s *Session) GuildCreate(name string) (st *Guild, err error) { | |||||||
| 		Name string `json:"name"` | 		Name string `json:"name"` | ||||||
| 	}{name} | 	}{name} | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("POST", EndpointGuilds, data, EndpointGuilds) | 	body, err := s.RequestWithBucketID("POST", EndpointGuildCreate, data, EndpointGuildCreate) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -527,7 +605,7 @@ func (s *Session) GuildEdit(guildID string, g GuildParams) (st *Guild, err error | |||||||
| 	if g.VerificationLevel != nil { | 	if g.VerificationLevel != nil { | ||||||
| 		val := *g.VerificationLevel | 		val := *g.VerificationLevel | ||||||
| 		if val < 0 || val > 3 { | 		if val < 0 || val > 3 { | ||||||
| 			err = errors.New("VerificationLevel out of bounds, should be between 0 and 3") | 			err = ErrVerificationLevelBounds | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -551,13 +629,7 @@ func (s *Session) GuildEdit(guildID string, g GuildParams) (st *Guild, err error | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	data := struct { | 	body, err := s.RequestWithBucketID("PATCH", EndpointGuild(guildID), g, EndpointGuild(guildID)) | ||||||
| 		Name              string             `json:"name,omitempty"` |  | ||||||
| 		Region            string             `json:"region,omitempty"` |  | ||||||
| 		VerificationLevel *VerificationLevel `json:"verification_level,omitempty"` |  | ||||||
| 	}{g.Name, g.Region, g.VerificationLevel} |  | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("PATCH", EndpointGuild(guildID), data, EndpointGuild(guildID)) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -607,11 +679,28 @@ func (s *Session) GuildBans(guildID string) (st []*GuildBan, err error) { | |||||||
| // userID    : The ID of a User | // userID    : The ID of a User | ||||||
| // days      : The number of days of previous comments to delete. | // days      : The number of days of previous comments to delete. | ||||||
| func (s *Session) GuildBanCreate(guildID, userID string, days int) (err error) { | func (s *Session) GuildBanCreate(guildID, userID string, days int) (err error) { | ||||||
|  | 	return s.GuildBanCreateWithReason(guildID, userID, "", days) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildBanCreateWithReason bans the given user from the given guild also providing a reaso. | ||||||
|  | // guildID   : The ID of a Guild. | ||||||
|  | // userID    : The ID of a User | ||||||
|  | // reason    : The reason for this ban | ||||||
|  | // days      : The number of days of previous comments to delete. | ||||||
|  | func (s *Session) GuildBanCreateWithReason(guildID, userID, reason string, days int) (err error) { | ||||||
|  |  | ||||||
| 	uri := EndpointGuildBan(guildID, userID) | 	uri := EndpointGuildBan(guildID, userID) | ||||||
|  |  | ||||||
|  | 	queryParams := url.Values{} | ||||||
| 	if days > 0 { | 	if days > 0 { | ||||||
| 		uri = fmt.Sprintf("%s?delete-message-days=%d", uri, days) | 		queryParams.Set("delete-message-days", strconv.Itoa(days)) | ||||||
|  | 	} | ||||||
|  | 	if reason != "" { | ||||||
|  | 		queryParams.Set("reason", reason) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(queryParams) > 0 { | ||||||
|  | 		uri += "?" + queryParams.Encode() | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("PUT", uri, nil, EndpointGuildBan(guildID, "")) | 	_, err = s.RequestWithBucketID("PUT", uri, nil, EndpointGuildBan(guildID, "")) | ||||||
| @@ -677,7 +766,21 @@ func (s *Session) GuildMember(guildID, userID string) (st *Member, err error) { | |||||||
| // userID    : The ID of a User | // userID    : The ID of a User | ||||||
| func (s *Session) GuildMemberDelete(guildID, userID string) (err error) { | func (s *Session) GuildMemberDelete(guildID, userID string) (err error) { | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("DELETE", EndpointGuildMember(guildID, userID), nil, EndpointGuildMember(guildID, "")) | 	return s.GuildMemberDeleteWithReason(guildID, userID, "") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GuildMemberDeleteWithReason removes the given user from the given guild. | ||||||
|  | // guildID   : The ID of a Guild. | ||||||
|  | // userID    : The ID of a User | ||||||
|  | // reason    : The reason for the kick | ||||||
|  | func (s *Session) GuildMemberDeleteWithReason(guildID, userID, reason string) (err error) { | ||||||
|  |  | ||||||
|  | 	uri := EndpointGuildMember(guildID, userID) | ||||||
|  | 	if reason != "" { | ||||||
|  | 		uri += "?reason=" + url.QueryEscape(reason) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, err = s.RequestWithBucketID("DELETE", uri, nil, EndpointGuildMember(guildID, "")) | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -722,12 +825,17 @@ func (s *Session) GuildMemberMove(guildID, userID, channelID string) (err error) | |||||||
| // GuildMemberNickname updates the nickname of a guild member | // GuildMemberNickname updates the nickname of a guild member | ||||||
| // guildID   : The ID of a guild | // guildID   : The ID of a guild | ||||||
| // userID    : The ID of a user | // userID    : The ID of a user | ||||||
|  | // userID    : The ID of a user or "@me" which is a shortcut of the current user ID | ||||||
| func (s *Session) GuildMemberNickname(guildID, userID, nickname string) (err error) { | func (s *Session) GuildMemberNickname(guildID, userID, nickname string) (err error) { | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| 		Nick string `json:"nick"` | 		Nick string `json:"nick"` | ||||||
| 	}{nickname} | 	}{nickname} | ||||||
|  |  | ||||||
|  | 	if userID == "@me" { | ||||||
|  | 		userID += "/nick" | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("PATCH", EndpointGuildMember(guildID, userID), data, EndpointGuildMember(guildID, "")) | 	_, err = s.RequestWithBucketID("PATCH", EndpointGuildMember(guildID, userID), data, EndpointGuildMember(guildID, "")) | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -738,7 +846,7 @@ func (s *Session) GuildMemberNickname(guildID, userID, nickname string) (err err | |||||||
| //  roleID 	  : The ID of a Role to be assigned to the user. | //  roleID 	  : The ID of a Role to be assigned to the user. | ||||||
| func (s *Session) GuildMemberRoleAdd(guildID, userID, roleID string) (err error) { | func (s *Session) GuildMemberRoleAdd(guildID, userID, roleID string) (err error) { | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("PUT", EndpointGuildMemberRole(guildID, userID, roleID), nil, EndpointGuildMemberRole(guildID, userID, roleID)) | 	_, err = s.RequestWithBucketID("PUT", EndpointGuildMemberRole(guildID, userID, roleID), nil, EndpointGuildMemberRole(guildID, "", "")) | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -749,7 +857,7 @@ func (s *Session) GuildMemberRoleAdd(guildID, userID, roleID string) (err error) | |||||||
| //  roleID 	  : The ID of a Role to be removed from the user. | //  roleID 	  : The ID of a Role to be removed from the user. | ||||||
| func (s *Session) GuildMemberRoleRemove(guildID, userID, roleID string) (err error) { | func (s *Session) GuildMemberRoleRemove(guildID, userID, roleID string) (err error) { | ||||||
|  |  | ||||||
| 	_, err = s.RequestWithBucketID("DELETE", EndpointGuildMemberRole(guildID, userID, roleID), nil, EndpointGuildMemberRole(guildID, userID, roleID)) | 	_, err = s.RequestWithBucketID("DELETE", EndpointGuildMemberRole(guildID, userID, roleID), nil, EndpointGuildMemberRole(guildID, "", "")) | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -801,7 +909,7 @@ func (s *Session) GuildChannelsReorder(guildID string, channels []*Channel) (err | |||||||
| // GuildInvites returns an array of Invite structures for the given guild | // GuildInvites returns an array of Invite structures for the given guild | ||||||
| // guildID   : The ID of a Guild. | // guildID   : The ID of a Guild. | ||||||
| func (s *Session) GuildInvites(guildID string) (st []*Invite, err error) { | func (s *Session) GuildInvites(guildID string) (st []*Invite, err error) { | ||||||
| 	body, err := s.RequestWithBucketID("GET", EndpointGuildInvites(guildID), nil, EndpointGuildInivtes(guildID)) | 	body, err := s.RequestWithBucketID("GET", EndpointGuildInvites(guildID), nil, EndpointGuildInvites(guildID)) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -851,6 +959,7 @@ func (s *Session) GuildRoleEdit(guildID, roleID, name string, color int, hoist b | |||||||
| 	// Prevent sending a color int that is too big. | 	// Prevent sending a color int that is too big. | ||||||
| 	if color > 0xFFFFFF { | 	if color > 0xFFFFFF { | ||||||
| 		err = fmt.Errorf("color value cannot be larger than 0xFFFFFF") | 		err = fmt.Errorf("color value cannot be larger than 0xFFFFFF") | ||||||
|  | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| @@ -904,7 +1013,7 @@ func (s *Session) GuildPruneCount(guildID string, days uint32) (count uint32, er | |||||||
| 	count = 0 | 	count = 0 | ||||||
|  |  | ||||||
| 	if days <= 0 { | 	if days <= 0 { | ||||||
| 		err = errors.New("The number of days should be more than or equal to 1.") | 		err = ErrPruneDaysBounds | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -914,6 +1023,9 @@ func (s *Session) GuildPruneCount(guildID string, days uint32) (count uint32, er | |||||||
|  |  | ||||||
| 	uri := EndpointGuildPrune(guildID) + fmt.Sprintf("?days=%d", days) | 	uri := EndpointGuildPrune(guildID) + fmt.Sprintf("?days=%d", days) | ||||||
| 	body, err := s.RequestWithBucketID("GET", uri, nil, EndpointGuildPrune(guildID)) | 	body, err := s.RequestWithBucketID("GET", uri, nil, EndpointGuildPrune(guildID)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	err = unmarshal(body, &p) | 	err = unmarshal(body, &p) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -934,7 +1046,7 @@ func (s *Session) GuildPrune(guildID string, days uint32) (count uint32, err err | |||||||
| 	count = 0 | 	count = 0 | ||||||
|  |  | ||||||
| 	if days <= 0 { | 	if days <= 0 { | ||||||
| 		err = errors.New("The number of days should be more than or equal to 1.") | 		err = ErrPruneDaysBounds | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -1036,7 +1148,7 @@ func (s *Session) GuildIcon(guildID string) (img image.Image, err error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if g.Icon == "" { | 	if g.Icon == "" { | ||||||
| 		err = errors.New("Guild does not have an icon set.") | 		err = ErrGuildNoIcon | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -1058,7 +1170,7 @@ func (s *Session) GuildSplash(guildID string) (img image.Image, err error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if g.Splash == "" { | 	if g.Splash == "" { | ||||||
| 		err = errors.New("Guild does not have a splash set.") | 		err = ErrGuildNoSplash | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -1098,7 +1210,7 @@ func (s *Session) GuildEmbedEdit(guildID string, enabled bool, channelID string) | |||||||
| // Functions specific to Discord Channels | // Functions specific to Discord Channels | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
|  |  | ||||||
| // Channel returns a Channel strucutre of a specific Channel. | // Channel returns a Channel structure of a specific Channel. | ||||||
| // channelID  : The ID of the Channel you want returned. | // channelID  : The ID of the Channel you want returned. | ||||||
| func (s *Session) Channel(channelID string) (st *Channel, err error) { | func (s *Session) Channel(channelID string) (st *Channel, err error) { | ||||||
| 	body, err := s.RequestWithBucketID("GET", EndpointChannel(channelID), nil, EndpointChannel(channelID)) | 	body, err := s.RequestWithBucketID("GET", EndpointChannel(channelID), nil, EndpointChannel(channelID)) | ||||||
| @@ -1113,12 +1225,16 @@ func (s *Session) Channel(channelID string) (st *Channel, err error) { | |||||||
| // ChannelEdit edits the given channel | // ChannelEdit edits the given channel | ||||||
| // channelID  : The ID of a Channel | // channelID  : The ID of a Channel | ||||||
| // name       : The new name to assign the channel. | // name       : The new name to assign the channel. | ||||||
| func (s *Session) ChannelEdit(channelID, name string) (st *Channel, err error) { | func (s *Session) ChannelEdit(channelID, name string) (*Channel, error) { | ||||||
|  | 	return s.ChannelEditComplex(channelID, &ChannelEdit{ | ||||||
| 	data := struct { | 		Name: name, | ||||||
| 		Name string `json:"name"` | 	}) | ||||||
| 	}{name} | } | ||||||
|  |  | ||||||
|  | // ChannelEditComplex edits an existing channel, replacing the parameters entirely with ChannelEdit struct | ||||||
|  | // channelID  : The ID of a Channel | ||||||
|  | // data          : The channel struct to send | ||||||
|  | func (s *Session) ChannelEditComplex(channelID string, data *ChannelEdit) (st *Channel, err error) { | ||||||
| 	body, err := s.RequestWithBucketID("PATCH", EndpointChannel(channelID), data, EndpointChannel(channelID)) | 	body, err := s.RequestWithBucketID("PATCH", EndpointChannel(channelID), data, EndpointChannel(channelID)) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| @@ -1156,7 +1272,8 @@ func (s *Session) ChannelTyping(channelID string) (err error) { | |||||||
| // limit     : The number messages that can be returned. (max 100) | // limit     : The number messages that can be returned. (max 100) | ||||||
| // beforeID  : If provided all messages returned will be before given ID. | // beforeID  : If provided all messages returned will be before given ID. | ||||||
| // afterID   : If provided all messages returned will be after given ID. | // afterID   : If provided all messages returned will be after given ID. | ||||||
| func (s *Session) ChannelMessages(channelID string, limit int, beforeID, afterID string) (st []*Message, err error) { | // aroundID  : If provided all messages returned will be around given ID. | ||||||
|  | func (s *Session) ChannelMessages(channelID string, limit int, beforeID, afterID, aroundID string) (st []*Message, err error) { | ||||||
|  |  | ||||||
| 	uri := EndpointChannelMessages(channelID) | 	uri := EndpointChannelMessages(channelID) | ||||||
|  |  | ||||||
| @@ -1170,6 +1287,9 @@ func (s *Session) ChannelMessages(channelID string, limit int, beforeID, afterID | |||||||
| 	if beforeID != "" { | 	if beforeID != "" { | ||||||
| 		v.Set("before", beforeID) | 		v.Set("before", beforeID) | ||||||
| 	} | 	} | ||||||
|  | 	if aroundID != "" { | ||||||
|  | 		v.Set("around", aroundID) | ||||||
|  | 	} | ||||||
| 	if len(v) > 0 { | 	if len(v) > 0 { | ||||||
| 		uri = fmt.Sprintf("%s?%s", uri, v.Encode()) | 		uri = fmt.Sprintf("%s?%s", uri, v.Encode()) | ||||||
| 	} | 	} | ||||||
| @@ -1212,20 +1332,92 @@ func (s *Session) ChannelMessageAck(channelID, messageID, lastToken string) (st | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // channelMessageSend sends a message to the given channel. | // ChannelMessageSend sends a message to the given channel. | ||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // content   : The message to send. | // content   : The message to send. | ||||||
| // tts       : Whether to send the message with TTS. | func (s *Session) ChannelMessageSend(channelID string, content string) (*Message, error) { | ||||||
| func (s *Session) channelMessageSend(channelID, content string, tts bool) (st *Message, err error) { | 	return s.ChannelMessageSendComplex(channelID, &MessageSend{ | ||||||
|  | 		Content: content, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  |  | ||||||
| 	// TODO: nonce string ? | var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"") | ||||||
| 	data := struct { |  | ||||||
| 		Content string `json:"content"` |  | ||||||
| 		TTS     bool   `json:"tts"` |  | ||||||
| 	}{content, tts} |  | ||||||
|  |  | ||||||
| 	// Send the message to the given channel | // ChannelMessageSendComplex sends a message to the given channel. | ||||||
| 	response, err := s.RequestWithBucketID("POST", EndpointChannelMessages(channelID), data, EndpointChannelMessages(channelID)) | // channelID : The ID of a Channel. | ||||||
|  | // data      : The message struct to send. | ||||||
|  | func (s *Session) ChannelMessageSendComplex(channelID string, data *MessageSend) (st *Message, err error) { | ||||||
|  | 	if data.Embed != nil && data.Embed.Type == "" { | ||||||
|  | 		data.Embed.Type = "rich" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	endpoint := EndpointChannelMessages(channelID) | ||||||
|  |  | ||||||
|  | 	// TODO: Remove this when compatibility is not required. | ||||||
|  | 	files := data.Files | ||||||
|  | 	if data.File != nil { | ||||||
|  | 		if files == nil { | ||||||
|  | 			files = []*File{data.File} | ||||||
|  | 		} else { | ||||||
|  | 			err = fmt.Errorf("cannot specify both File and Files") | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var response []byte | ||||||
|  | 	if len(files) > 0 { | ||||||
|  | 		body := &bytes.Buffer{} | ||||||
|  | 		bodywriter := multipart.NewWriter(body) | ||||||
|  |  | ||||||
|  | 		var payload []byte | ||||||
|  | 		payload, err = json.Marshal(data) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		var p io.Writer | ||||||
|  |  | ||||||
|  | 		h := make(textproto.MIMEHeader) | ||||||
|  | 		h.Set("Content-Disposition", `form-data; name="payload_json"`) | ||||||
|  | 		h.Set("Content-Type", "application/json") | ||||||
|  |  | ||||||
|  | 		p, err = bodywriter.CreatePart(h) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if _, err = p.Write(payload); err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i, file := range files { | ||||||
|  | 			h := make(textproto.MIMEHeader) | ||||||
|  | 			h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file%d"; filename="%s"`, i, quoteEscaper.Replace(file.Name))) | ||||||
|  | 			contentType := file.ContentType | ||||||
|  | 			if contentType == "" { | ||||||
|  | 				contentType = "application/octet-stream" | ||||||
|  | 			} | ||||||
|  | 			h.Set("Content-Type", contentType) | ||||||
|  |  | ||||||
|  | 			p, err = bodywriter.CreatePart(h) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if _, err = io.Copy(p, file.Reader); err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		err = bodywriter.Close() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		response, err = s.request("POST", endpoint, bodywriter.FormDataContentType(), body.Bytes(), endpoint, 0) | ||||||
|  | 	} else { | ||||||
|  | 		response, err = s.RequestWithBucketID("POST", endpoint, data, endpoint) | ||||||
|  | 	} | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -1234,55 +1426,42 @@ func (s *Session) channelMessageSend(channelID, content string, tts bool) (st *M | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessageSend sends a message to the given channel. |  | ||||||
| // channelID : The ID of a Channel. |  | ||||||
| // content   : The message to send. |  | ||||||
| func (s *Session) ChannelMessageSend(channelID string, content string) (st *Message, err error) { |  | ||||||
|  |  | ||||||
| 	return s.channelMessageSend(channelID, content, false) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ChannelMessageSendTTS sends a message to the given channel with Text to Speech. | // ChannelMessageSendTTS sends a message to the given channel with Text to Speech. | ||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // content   : The message to send. | // content   : The message to send. | ||||||
| func (s *Session) ChannelMessageSendTTS(channelID string, content string) (st *Message, err error) { | func (s *Session) ChannelMessageSendTTS(channelID string, content string) (*Message, error) { | ||||||
|  | 	return s.ChannelMessageSendComplex(channelID, &MessageSend{ | ||||||
| 	return s.channelMessageSend(channelID, content, true) | 		Content: content, | ||||||
|  | 		Tts:     true, | ||||||
|  | 	}) | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessageSendEmbed sends a message to the given channel with embedded data (bot only). | // ChannelMessageSendEmbed sends a message to the given channel with embedded data. | ||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // embed     : The embed data to send. | // embed     : The embed data to send. | ||||||
| func (s *Session) ChannelMessageSendEmbed(channelID string, embed *MessageEmbed) (st *Message, err error) { | func (s *Session) ChannelMessageSendEmbed(channelID string, embed *MessageEmbed) (*Message, error) { | ||||||
| 	if embed != nil && embed.Type == "" { | 	return s.ChannelMessageSendComplex(channelID, &MessageSend{ | ||||||
| 		embed.Type = "rich" | 		Embed: embed, | ||||||
| 	} | 	}) | ||||||
|  |  | ||||||
| 	data := struct { |  | ||||||
| 		Embed *MessageEmbed `json:"embed"` |  | ||||||
| 	}{embed} |  | ||||||
|  |  | ||||||
| 	// Send the message to the given channel |  | ||||||
| 	response, err := s.RequestWithBucketID("POST", EndpointChannelMessages(channelID), data, EndpointChannelMessages(channelID)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = unmarshal(response, &st) |  | ||||||
| 	return |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessageEdit edits an existing message, replacing it entirely with | // ChannelMessageEdit edits an existing message, replacing it entirely with | ||||||
| // the given content. | // the given content. | ||||||
| // channeld  : The ID of a Channel | // channelID  : The ID of a Channel | ||||||
| // messageID : the ID of a Message | // messageID  : The ID of a Message | ||||||
| func (s *Session) ChannelMessageEdit(channelID, messageID, content string) (st *Message, err error) { | // content    : The contents of the message | ||||||
|  | func (s *Session) ChannelMessageEdit(channelID, messageID, content string) (*Message, error) { | ||||||
|  | 	return s.ChannelMessageEditComplex(NewMessageEdit(channelID, messageID).SetContent(content)) | ||||||
|  | } | ||||||
|  |  | ||||||
| 	data := struct { | // ChannelMessageEditComplex edits an existing message, replacing it entirely with | ||||||
| 		Content string `json:"content"` | // the given MessageEdit struct | ||||||
| 	}{content} | func (s *Session) ChannelMessageEditComplex(m *MessageEdit) (st *Message, err error) { | ||||||
|  | 	if m.Embed != nil && m.Embed.Type == "" { | ||||||
|  | 		m.Embed.Type = "rich" | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	response, err := s.RequestWithBucketID("PATCH", EndpointChannelMessage(channelID, messageID), data, EndpointChannelMessage(channelID, "")) | 	response, err := s.RequestWithBucketID("PATCH", EndpointChannelMessage(m.Channel, m.ID), m, EndpointChannelMessage(m.Channel, "")) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -1291,26 +1470,12 @@ func (s *Session) ChannelMessageEdit(channelID, messageID, content string) (st * | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessageEditEmbed edits an existing message with embedded data (bot only). | // ChannelMessageEditEmbed edits an existing message with embedded data. | ||||||
| // channelID : The ID of a Channel | // channelID : The ID of a Channel | ||||||
| // messageID : The ID of a Message | // messageID : The ID of a Message | ||||||
| // embed     : The embed data to send | // embed     : The embed data to send | ||||||
| func (s *Session) ChannelMessageEditEmbed(channelID, messageID string, embed *MessageEmbed) (st *Message, err error) { | func (s *Session) ChannelMessageEditEmbed(channelID, messageID string, embed *MessageEmbed) (*Message, error) { | ||||||
| 	if embed != nil && embed.Type == "" { | 	return s.ChannelMessageEditComplex(NewMessageEdit(channelID, messageID).SetEmbed(embed)) | ||||||
| 		embed.Type = "rich" |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	data := struct { |  | ||||||
| 		Embed *MessageEmbed `json:"embed"` |  | ||||||
| 	}{embed} |  | ||||||
|  |  | ||||||
| 	response, err := s.RequestWithBucketID("PATCH", EndpointChannelMessage(channelID, messageID), data, EndpointChannelMessage(channelID, "")) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = unmarshal(response, &st) |  | ||||||
| 	return |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessageDelete deletes a message from the Channel. | // ChannelMessageDelete deletes a message from the Channel. | ||||||
| @@ -1321,7 +1486,7 @@ func (s *Session) ChannelMessageDelete(channelID, messageID string) (err error) | |||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelMessagesBulkDelete bulk deletes the messages from the channel for the provided messageIDs. | // ChannelMessagesBulkDelete bulk deletes the messages from the channel for the provided messageIDs. | ||||||
| // If only one messageID is in the slice call channelMessageDelete funciton. | // If only one messageID is in the slice call channelMessageDelete function. | ||||||
| // If the slice is empty do nothing. | // If the slice is empty do nothing. | ||||||
| // channelID : The ID of the channel for the messages to delete. | // channelID : The ID of the channel for the messages to delete. | ||||||
| // messages  : The IDs of the messages to be deleted. A slice of string IDs. A maximum of 100 messages. | // messages  : The IDs of the messages to be deleted. A slice of string IDs. A maximum of 100 messages. | ||||||
| @@ -1385,48 +1550,18 @@ func (s *Session) ChannelMessagesPinned(channelID string) (st []*Message, err er | |||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // name: The name of the file. | // name: The name of the file. | ||||||
| // io.Reader : A reader for the file contents. | // io.Reader : A reader for the file contents. | ||||||
| func (s *Session) ChannelFileSend(channelID, name string, r io.Reader) (st *Message, err error) { | func (s *Session) ChannelFileSend(channelID, name string, r io.Reader) (*Message, error) { | ||||||
| 	return s.ChannelFileSendWithMessage(channelID, "", name, r) | 	return s.ChannelMessageSendComplex(channelID, &MessageSend{File: &File{Name: name, Reader: r}}) | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelFileSendWithMessage sends a file to the given channel with an message. | // ChannelFileSendWithMessage sends a file to the given channel with an message. | ||||||
|  | // DEPRECATED. Use ChannelMessageSendComplex instead. | ||||||
| // channelID : The ID of a Channel. | // channelID : The ID of a Channel. | ||||||
| // content: Optional Message content. | // content: Optional Message content. | ||||||
| // name: The name of the file. | // name: The name of the file. | ||||||
| // io.Reader : A reader for the file contents. | // io.Reader : A reader for the file contents. | ||||||
| func (s *Session) ChannelFileSendWithMessage(channelID, content string, name string, r io.Reader) (st *Message, err error) { | func (s *Session) ChannelFileSendWithMessage(channelID, content string, name string, r io.Reader) (*Message, error) { | ||||||
|  | 	return s.ChannelMessageSendComplex(channelID, &MessageSend{File: &File{Name: name, Reader: r}, Content: content}) | ||||||
| 	body := &bytes.Buffer{} |  | ||||||
| 	bodywriter := multipart.NewWriter(body) |  | ||||||
|  |  | ||||||
| 	if len(content) != 0 { |  | ||||||
| 		if err := bodywriter.WriteField("content", content); err != nil { |  | ||||||
| 			return nil, err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	writer, err := bodywriter.CreateFormFile("file", name) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	_, err = io.Copy(writer, r) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = bodywriter.Close() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	response, err := s.request("POST", EndpointChannelMessages(channelID), bodywriter.FormDataContentType(), body.Bytes(), EndpointChannelMessages(channelID), 0) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = unmarshal(response, &st) |  | ||||||
| 	return |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelInvites returns an array of Invite structures for the given channel | // ChannelInvites returns an array of Invite structures for the given channel | ||||||
| @@ -1444,16 +1579,14 @@ func (s *Session) ChannelInvites(channelID string) (st []*Invite, err error) { | |||||||
|  |  | ||||||
| // ChannelInviteCreate creates a new invite for the given channel. | // ChannelInviteCreate creates a new invite for the given channel. | ||||||
| // channelID   : The ID of a Channel | // channelID   : The ID of a Channel | ||||||
| // i           : An Invite struct with the values MaxAge, MaxUses, Temporary, | // i           : An Invite struct with the values MaxAge, MaxUses and Temporary defined. | ||||||
| //               and XkcdPass defined. |  | ||||||
| func (s *Session) ChannelInviteCreate(channelID string, i Invite) (st *Invite, err error) { | func (s *Session) ChannelInviteCreate(channelID string, i Invite) (st *Invite, err error) { | ||||||
|  |  | ||||||
| 	data := struct { | 	data := struct { | ||||||
| 		MaxAge    int  `json:"max_age"` | 		MaxAge    int  `json:"max_age"` | ||||||
| 		MaxUses   int  `json:"max_uses"` | 		MaxUses   int  `json:"max_uses"` | ||||||
| 		Temporary bool `json:"temporary"` | 		Temporary bool `json:"temporary"` | ||||||
| 		XKCDPass  string `json:"xkcdpass"` | 	}{i.MaxAge, i.MaxUses, i.Temporary} | ||||||
| 	}{i.MaxAge, i.MaxUses, i.Temporary, i.XkcdPass} |  | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("POST", EndpointChannelInvites(channelID), data, EndpointChannelInvites(channelID)) | 	body, err := s.RequestWithBucketID("POST", EndpointChannelInvites(channelID), data, EndpointChannelInvites(channelID)) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -1493,7 +1626,7 @@ func (s *Session) ChannelPermissionDelete(channelID, targetID string) (err error | |||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
|  |  | ||||||
| // Invite returns an Invite structure of the given invite | // Invite returns an Invite structure of the given invite | ||||||
| // inviteID : The invite code (or maybe xkcdpass?) | // inviteID : The invite code | ||||||
| func (s *Session) Invite(inviteID string) (st *Invite, err error) { | func (s *Session) Invite(inviteID string) (st *Invite, err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("GET", EndpointInvite(inviteID), nil, EndpointInvite("")) | 	body, err := s.RequestWithBucketID("GET", EndpointInvite(inviteID), nil, EndpointInvite("")) | ||||||
| @@ -1506,7 +1639,7 @@ func (s *Session) Invite(inviteID string) (st *Invite, err error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| // InviteDelete deletes an existing invite | // InviteDelete deletes an existing invite | ||||||
| // inviteID   : the code (or maybe xkcdpass?) of an invite | // inviteID   : the code of an invite | ||||||
| func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) { | func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("DELETE", EndpointInvite(inviteID), nil, EndpointInvite("")) | 	body, err := s.RequestWithBucketID("DELETE", EndpointInvite(inviteID), nil, EndpointInvite("")) | ||||||
| @@ -1519,7 +1652,7 @@ func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| // InviteAccept accepts an Invite to a Guild or Channel | // InviteAccept accepts an Invite to a Guild or Channel | ||||||
| // inviteID : The invite code (or maybe xkcdpass?) | // inviteID : The invite code | ||||||
| func (s *Session) InviteAccept(inviteID string) (st *Invite, err error) { | func (s *Session) InviteAccept(inviteID string) (st *Invite, err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("POST", EndpointInvite(inviteID), nil, EndpointInvite("")) | 	body, err := s.RequestWithBucketID("POST", EndpointInvite(inviteID), nil, EndpointInvite("")) | ||||||
| @@ -1563,7 +1696,7 @@ func (s *Session) VoiceICE() (st *VoiceICE, err error) { | |||||||
| // Functions specific to Discord Websockets | // Functions specific to Discord Websockets | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
|  |  | ||||||
| // Gateway returns the a websocket Gateway address | // Gateway returns the websocket Gateway address | ||||||
| func (s *Session) Gateway() (gateway string, err error) { | func (s *Session) Gateway() (gateway string, err error) { | ||||||
|  |  | ||||||
| 	response, err := s.RequestWithBucketID("GET", EndpointGateway, nil, EndpointGateway) | 	response, err := s.RequestWithBucketID("GET", EndpointGateway, nil, EndpointGateway) | ||||||
| @@ -1591,6 +1724,28 @@ func (s *Session) Gateway() (gateway string, err error) { | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // GatewayBot returns the websocket Gateway address and the recommended number of shards | ||||||
|  | func (s *Session) GatewayBot() (st *GatewayBotResponse, err error) { | ||||||
|  |  | ||||||
|  | 	response, err := s.RequestWithBucketID("GET", EndpointGatewayBot, nil, EndpointGatewayBot) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	err = unmarshal(response, &st) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Ensure the gateway always has a trailing slash. | ||||||
|  | 	// MacOS will fail to connect if we add query params without a trailing slash on the base domain. | ||||||
|  | 	if !strings.HasSuffix(st.URL, "/") { | ||||||
|  | 		st.URL += "/" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
| // Functions specific to Webhooks | // Functions specific to Webhooks | ||||||
|  |  | ||||||
| // WebhookCreate returns a new Webhook. | // WebhookCreate returns a new Webhook. | ||||||
| @@ -1716,14 +1871,9 @@ func (s *Session) WebhookEditWithToken(webhookID, token, name, avatar string) (s | |||||||
|  |  | ||||||
| // WebhookDelete deletes a webhook for a given ID | // WebhookDelete deletes a webhook for a given ID | ||||||
| // webhookID: The ID of a webhook. | // webhookID: The ID of a webhook. | ||||||
| func (s *Session) WebhookDelete(webhookID string) (st *Webhook, err error) { | func (s *Session) WebhookDelete(webhookID string) (err error) { | ||||||
|  |  | ||||||
| 	body, err := s.RequestWithBucketID("DELETE", EndpointWebhook(webhookID), nil, EndpointWebhooks) | 	_, err = s.RequestWithBucketID("DELETE", EndpointWebhook(webhookID), nil, EndpointWebhooks) | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = unmarshal(body, &st) |  | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -1781,6 +1931,16 @@ func (s *Session) MessageReactionRemove(channelID, messageID, emojiID, userID st | |||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MessageReactionsRemoveAll deletes all reactions from a message | ||||||
|  | // channelID : The channel ID | ||||||
|  | // messageID : The message ID. | ||||||
|  | func (s *Session) MessageReactionsRemoveAll(channelID, messageID string) error { | ||||||
|  |  | ||||||
|  | 	_, err := s.RequestWithBucketID("DELETE", EndpointMessageReactionsAll(channelID, messageID), nil, EndpointMessageReactionsAll(channelID, messageID)) | ||||||
|  |  | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
| // MessageReactions gets all the users reactions for a specific emoji. | // MessageReactions gets all the users reactions for a specific emoji. | ||||||
| // channelID : The channel ID. | // channelID : The channel ID. | ||||||
| // messageID : The message ID. | // messageID : The message ID. | ||||||
| @@ -1808,6 +1968,20 @@ func (s *Session) MessageReactions(channelID, messageID, emojiID string, limit i | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ------------------------------------------------------------------------------------------------ | ||||||
|  | // Functions specific to user notes | ||||||
|  | // ------------------------------------------------------------------------------------------------ | ||||||
|  |  | ||||||
|  | // UserNoteSet sets the note for a specific user. | ||||||
|  | func (s *Session) UserNoteSet(userID string, message string) (err error) { | ||||||
|  | 	data := struct { | ||||||
|  | 		Note string `json:"note"` | ||||||
|  | 	}{message} | ||||||
|  |  | ||||||
|  | 	_, err = s.RequestWithBucketID("PUT", EndpointUserNotes(userID), data, EndpointUserNotes("")) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
| // Functions specific to Discord Relationships (Friends list) | // Functions specific to Discord Relationships (Friends list) | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
|   | |||||||
							
								
								
									
										336
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										336
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -14,11 +14,16 @@ package discordgo | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"sort" | ||||||
| 	"sync" | 	"sync" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // ErrNilState is returned when the state is nil. | // ErrNilState is returned when the state is nil. | ||||||
| var ErrNilState = errors.New("State not instantiated, please use discordgo.New() or assign Session.State.") | var ErrNilState = errors.New("state not instantiated, please use discordgo.New() or assign Session.State") | ||||||
|  |  | ||||||
|  | // ErrStateNotFound is returned when the state cache | ||||||
|  | // requested is not found | ||||||
|  | var ErrStateNotFound = errors.New("state cache not found") | ||||||
|  |  | ||||||
| // A State contains the current known state. | // A State contains the current known state. | ||||||
| // As discord sends this in a READY blob, it seems reasonable to simply | // As discord sends this in a READY blob, it seems reasonable to simply | ||||||
| @@ -33,9 +38,11 @@ type State struct { | |||||||
| 	TrackMembers    bool | 	TrackMembers    bool | ||||||
| 	TrackRoles      bool | 	TrackRoles      bool | ||||||
| 	TrackVoice      bool | 	TrackVoice      bool | ||||||
|  | 	TrackPresences  bool | ||||||
|  |  | ||||||
| 	guildMap   map[string]*Guild | 	guildMap   map[string]*Guild | ||||||
| 	channelMap map[string]*Channel | 	channelMap map[string]*Channel | ||||||
|  | 	memberMap  map[string]map[string]*Member | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewState creates an empty state. | // NewState creates an empty state. | ||||||
| @@ -50,11 +57,21 @@ func NewState() *State { | |||||||
| 		TrackMembers:   true, | 		TrackMembers:   true, | ||||||
| 		TrackRoles:     true, | 		TrackRoles:     true, | ||||||
| 		TrackVoice:     true, | 		TrackVoice:     true, | ||||||
|  | 		TrackPresences: true, | ||||||
| 		guildMap:       make(map[string]*Guild), | 		guildMap:       make(map[string]*Guild), | ||||||
| 		channelMap:     make(map[string]*Channel), | 		channelMap:     make(map[string]*Channel), | ||||||
|  | 		memberMap:      make(map[string]map[string]*Member), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (s *State) createMemberMap(guild *Guild) { | ||||||
|  | 	members := make(map[string]*Member) | ||||||
|  | 	for _, m := range guild.Members { | ||||||
|  | 		members[m.User.ID] = m | ||||||
|  | 	} | ||||||
|  | 	s.memberMap[guild.ID] = members | ||||||
|  | } | ||||||
|  |  | ||||||
| // GuildAdd adds a guild to the current world state, or | // GuildAdd adds a guild to the current world state, or | ||||||
| // updates it if it already exists. | // updates it if it already exists. | ||||||
| func (s *State) GuildAdd(guild *Guild) error { | func (s *State) GuildAdd(guild *Guild) error { | ||||||
| @@ -70,6 +87,14 @@ func (s *State) GuildAdd(guild *Guild) error { | |||||||
| 		s.channelMap[c.ID] = c | 		s.channelMap[c.ID] = c | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// If this guild contains a new member slice, we must regenerate the member map so the pointers stay valid | ||||||
|  | 	if guild.Members != nil { | ||||||
|  | 		s.createMemberMap(guild) | ||||||
|  | 	} else if _, ok := s.memberMap[guild.ID]; !ok { | ||||||
|  | 		// Even if we have no new member slice, we still initialize the member map for this guild if it doesn't exist | ||||||
|  | 		s.memberMap[guild.ID] = make(map[string]*Member) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if g, ok := s.guildMap[guild.ID]; ok { | 	if g, ok := s.guildMap[guild.ID]; ok { | ||||||
| 		// We are about to replace `g` in the state with `guild`, but first we need to | 		// We are about to replace `g` in the state with `guild`, but first we need to | ||||||
| 		// make sure we preserve any fields that the `guild` doesn't contain from `g`. | 		// make sure we preserve any fields that the `guild` doesn't contain from `g`. | ||||||
| @@ -143,7 +168,108 @@ func (s *State) Guild(guildID string) (*Guild, error) { | |||||||
| 		return g, nil | 		return g, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Guild not found.") | 	return nil, ErrStateNotFound | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PresenceAdd adds a presence to the current world state, or | ||||||
|  | // updates it if it already exists. | ||||||
|  | func (s *State) PresenceAdd(guildID string, presence *Presence) error { | ||||||
|  | 	if s == nil { | ||||||
|  | 		return ErrNilState | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	guild, err := s.Guild(guildID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	s.Lock() | ||||||
|  | 	defer s.Unlock() | ||||||
|  |  | ||||||
|  | 	for i, p := range guild.Presences { | ||||||
|  | 		if p.User.ID == presence.User.ID { | ||||||
|  | 			//guild.Presences[i] = presence | ||||||
|  |  | ||||||
|  | 			//Update status | ||||||
|  | 			guild.Presences[i].Game = presence.Game | ||||||
|  | 			guild.Presences[i].Roles = presence.Roles | ||||||
|  | 			if presence.Status != "" { | ||||||
|  | 				guild.Presences[i].Status = presence.Status | ||||||
|  | 			} | ||||||
|  | 			if presence.Nick != "" { | ||||||
|  | 				guild.Presences[i].Nick = presence.Nick | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			//Update the optionally sent user information | ||||||
|  | 			//ID Is a mandatory field so you should not need to check if it is empty | ||||||
|  | 			guild.Presences[i].User.ID = presence.User.ID | ||||||
|  |  | ||||||
|  | 			if presence.User.Avatar != "" { | ||||||
|  | 				guild.Presences[i].User.Avatar = presence.User.Avatar | ||||||
|  | 			} | ||||||
|  | 			if presence.User.Discriminator != "" { | ||||||
|  | 				guild.Presences[i].User.Discriminator = presence.User.Discriminator | ||||||
|  | 			} | ||||||
|  | 			if presence.User.Email != "" { | ||||||
|  | 				guild.Presences[i].User.Email = presence.User.Email | ||||||
|  | 			} | ||||||
|  | 			if presence.User.Token != "" { | ||||||
|  | 				guild.Presences[i].User.Token = presence.User.Token | ||||||
|  | 			} | ||||||
|  | 			if presence.User.Username != "" { | ||||||
|  | 				guild.Presences[i].User.Username = presence.User.Username | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	guild.Presences = append(guild.Presences, presence) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PresenceRemove removes a presence from the current world state. | ||||||
|  | func (s *State) PresenceRemove(guildID string, presence *Presence) error { | ||||||
|  | 	if s == nil { | ||||||
|  | 		return ErrNilState | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	guild, err := s.Guild(guildID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	s.Lock() | ||||||
|  | 	defer s.Unlock() | ||||||
|  |  | ||||||
|  | 	for i, p := range guild.Presences { | ||||||
|  | 		if p.User.ID == presence.User.ID { | ||||||
|  | 			guild.Presences = append(guild.Presences[:i], guild.Presences[i+1:]...) | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ErrStateNotFound | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Presence gets a presence by ID from a guild. | ||||||
|  | func (s *State) Presence(guildID, userID string) (*Presence, error) { | ||||||
|  | 	if s == nil { | ||||||
|  | 		return nil, ErrNilState | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	guild, err := s.Guild(guildID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, p := range guild.Presences { | ||||||
|  | 		if p.User.ID == userID { | ||||||
|  | 			return p, nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // TODO: Consider moving Guild state update methods onto *Guild. | // TODO: Consider moving Guild state update methods onto *Guild. | ||||||
| @@ -163,14 +289,19 @@ func (s *State) MemberAdd(member *Member) error { | |||||||
| 	s.Lock() | 	s.Lock() | ||||||
| 	defer s.Unlock() | 	defer s.Unlock() | ||||||
|  |  | ||||||
| 	for i, m := range guild.Members { | 	members, ok := s.memberMap[member.GuildID] | ||||||
| 		if m.User.ID == member.User.ID { | 	if !ok { | ||||||
| 			guild.Members[i] = member | 		return ErrStateNotFound | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	m, ok := members[member.User.ID] | ||||||
|  | 	if !ok { | ||||||
|  | 		members[member.User.ID] = member | ||||||
| 		guild.Members = append(guild.Members, member) | 		guild.Members = append(guild.Members, member) | ||||||
|  | 	} else { | ||||||
|  | 		*m = *member // Update the actual data, which will also update the member pointer in the slice | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -188,6 +319,17 @@ func (s *State) MemberRemove(member *Member) error { | |||||||
| 	s.Lock() | 	s.Lock() | ||||||
| 	defer s.Unlock() | 	defer s.Unlock() | ||||||
|  |  | ||||||
|  | 	members, ok := s.memberMap[member.GuildID] | ||||||
|  | 	if !ok { | ||||||
|  | 		return ErrStateNotFound | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, ok = members[member.User.ID] | ||||||
|  | 	if !ok { | ||||||
|  | 		return ErrStateNotFound | ||||||
|  | 	} | ||||||
|  | 	delete(members, member.User.ID) | ||||||
|  |  | ||||||
| 	for i, m := range guild.Members { | 	for i, m := range guild.Members { | ||||||
| 		if m.User.ID == member.User.ID { | 		if m.User.ID == member.User.ID { | ||||||
| 			guild.Members = append(guild.Members[:i], guild.Members[i+1:]...) | 			guild.Members = append(guild.Members[:i], guild.Members[i+1:]...) | ||||||
| @@ -195,7 +337,7 @@ func (s *State) MemberRemove(member *Member) error { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return errors.New("Member not found.") | 	return ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // Member gets a member by ID from a guild. | // Member gets a member by ID from a guild. | ||||||
| @@ -204,21 +346,20 @@ func (s *State) Member(guildID, userID string) (*Member, error) { | |||||||
| 		return nil, ErrNilState | 		return nil, ErrNilState | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	guild, err := s.Guild(guildID) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	s.RLock() | 	s.RLock() | ||||||
| 	defer s.RUnlock() | 	defer s.RUnlock() | ||||||
|  |  | ||||||
| 	for _, m := range guild.Members { | 	members, ok := s.memberMap[guildID] | ||||||
| 		if m.User.ID == userID { | 	if !ok { | ||||||
| 			return m, nil | 		return nil, ErrStateNotFound | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Member not found.") | 	m, ok := members[userID] | ||||||
|  | 	if ok { | ||||||
|  | 		return m, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // RoleAdd adds a role to the current world state, or | // RoleAdd adds a role to the current world state, or | ||||||
| @@ -268,7 +409,7 @@ func (s *State) RoleRemove(guildID, roleID string) error { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return errors.New("Role not found.") | 	return ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // Role gets a role by ID from a guild. | // Role gets a role by ID from a guild. | ||||||
| @@ -291,10 +432,10 @@ func (s *State) Role(guildID, roleID string) (*Role, error) { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Role not found.") | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // ChannelAdd adds a guild to the current world state, or | // ChannelAdd adds a channel to the current world state, or | ||||||
| // updates it if it already exists. | // updates it if it already exists. | ||||||
| // Channels may exist either as PrivateChannels or inside | // Channels may exist either as PrivateChannels or inside | ||||||
| // a guild. | // a guild. | ||||||
| @@ -319,12 +460,12 @@ func (s *State) ChannelAdd(channel *Channel) error { | |||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if channel.IsPrivate { | 	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM { | ||||||
| 		s.PrivateChannels = append(s.PrivateChannels, channel) | 		s.PrivateChannels = append(s.PrivateChannels, channel) | ||||||
| 	} else { | 	} else { | ||||||
| 		guild, ok := s.guildMap[channel.GuildID] | 		guild, ok := s.guildMap[channel.GuildID] | ||||||
| 		if !ok { | 		if !ok { | ||||||
| 			return errors.New("Guild for channel not found.") | 			return ErrStateNotFound | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		guild.Channels = append(guild.Channels, channel) | 		guild.Channels = append(guild.Channels, channel) | ||||||
| @@ -346,7 +487,7 @@ func (s *State) ChannelRemove(channel *Channel) error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if channel.IsPrivate { | 	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM { | ||||||
| 		s.Lock() | 		s.Lock() | ||||||
| 		defer s.Unlock() | 		defer s.Unlock() | ||||||
|  |  | ||||||
| @@ -390,7 +531,7 @@ func (s *State) PrivateChannel(channelID string) (*Channel, error) { | |||||||
| 	return s.Channel(channelID) | 	return s.Channel(channelID) | ||||||
| } | } | ||||||
|  |  | ||||||
| // Channel gets a channel by ID, it will look in all guilds an private channels. | // Channel gets a channel by ID, it will look in all guilds and private channels. | ||||||
| func (s *State) Channel(channelID string) (*Channel, error) { | func (s *State) Channel(channelID string) (*Channel, error) { | ||||||
| 	if s == nil { | 	if s == nil { | ||||||
| 		return nil, ErrNilState | 		return nil, ErrNilState | ||||||
| @@ -403,7 +544,7 @@ func (s *State) Channel(channelID string) (*Channel, error) { | |||||||
| 		return c, nil | 		return c, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Channel not found.") | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // Emoji returns an emoji for a guild and emoji id. | // Emoji returns an emoji for a guild and emoji id. | ||||||
| @@ -426,7 +567,7 @@ func (s *State) Emoji(guildID, emojiID string) (*Emoji, error) { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Emoji not found.") | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // EmojiAdd adds an emoji to the current world state. | // EmojiAdd adds an emoji to the current world state. | ||||||
| @@ -523,7 +664,12 @@ func (s *State) MessageRemove(message *Message) error { | |||||||
| 		return ErrNilState | 		return ErrNilState | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	c, err := s.Channel(message.ChannelID) | 	return s.messageRemoveByID(message.ChannelID, message.ID) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // messageRemoveByID removes a message by channelID and messageID from the world state. | ||||||
|  | func (s *State) messageRemoveByID(channelID, messageID string) error { | ||||||
|  | 	c, err := s.Channel(channelID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -532,13 +678,13 @@ func (s *State) MessageRemove(message *Message) error { | |||||||
| 	defer s.Unlock() | 	defer s.Unlock() | ||||||
|  |  | ||||||
| 	for i, m := range c.Messages { | 	for i, m := range c.Messages { | ||||||
| 		if m.ID == message.ID { | 		if m.ID == messageID { | ||||||
| 			c.Messages = append(c.Messages[:i], c.Messages[i+1:]...) | 			c.Messages = append(c.Messages[:i], c.Messages[i+1:]...) | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return errors.New("Message not found.") | 	return ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *State) voiceStateUpdate(update *VoiceStateUpdate) error { | func (s *State) voiceStateUpdate(update *VoiceStateUpdate) error { | ||||||
| @@ -592,7 +738,7 @@ func (s *State) Message(channelID, messageID string) (*Message, error) { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil, errors.New("Message not found.") | 	return nil, ErrStateNotFound | ||||||
| } | } | ||||||
|  |  | ||||||
| // OnReady takes a Ready event and updates all internal state. | // OnReady takes a Ready event and updates all internal state. | ||||||
| @@ -610,7 +756,6 @@ func (s *State) onReady(se *Session, r *Ready) (err error) { | |||||||
| 		ready := Ready{ | 		ready := Ready{ | ||||||
| 			Version:   r.Version, | 			Version:   r.Version, | ||||||
| 			SessionID: r.SessionID, | 			SessionID: r.SessionID, | ||||||
| 			HeartbeatInterval: r.HeartbeatInterval, |  | ||||||
| 			User:      r.User, | 			User:      r.User, | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| @@ -623,6 +768,7 @@ func (s *State) onReady(se *Session, r *Ready) (err error) { | |||||||
|  |  | ||||||
| 	for _, g := range s.Guilds { | 	for _, g := range s.Guilds { | ||||||
| 		s.guildMap[g.ID] = g | 		s.guildMap[g.ID] = g | ||||||
|  | 		s.createMemberMap(g) | ||||||
|  |  | ||||||
| 		for _, c := range g.Channels { | 		for _, c := range g.Channels { | ||||||
| 			s.channelMap[c.ID] = c | 			s.channelMap[c.ID] = c | ||||||
| @@ -636,8 +782,8 @@ func (s *State) onReady(se *Session, r *Ready) (err error) { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // onInterface handles all events related to states. | // OnInterface handles all events related to states. | ||||||
| func (s *State) onInterface(se *Session, i interface{}) (err error) { | func (s *State) OnInterface(se *Session, i interface{}) (err error) { | ||||||
| 	if s == nil { | 	if s == nil { | ||||||
| 		return ErrNilState | 		return ErrNilState | ||||||
| 	} | 	} | ||||||
| @@ -670,6 +816,13 @@ func (s *State) onInterface(se *Session, i interface{}) (err error) { | |||||||
| 		if s.TrackMembers { | 		if s.TrackMembers { | ||||||
| 			err = s.MemberRemove(t.Member) | 			err = s.MemberRemove(t.Member) | ||||||
| 		} | 		} | ||||||
|  | 	case *GuildMembersChunk: | ||||||
|  | 		if s.TrackMembers { | ||||||
|  | 			for i := range t.Members { | ||||||
|  | 				t.Members[i].GuildID = t.GuildID | ||||||
|  | 				err = s.MemberAdd(t.Members[i]) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 	case *GuildRoleCreate: | 	case *GuildRoleCreate: | ||||||
| 		if s.TrackRoles { | 		if s.TrackRoles { | ||||||
| 			err = s.RoleAdd(t.GuildID, t.Role) | 			err = s.RoleAdd(t.GuildID, t.Role) | ||||||
| @@ -710,10 +863,55 @@ func (s *State) onInterface(se *Session, i interface{}) (err error) { | |||||||
| 		if s.MaxMessageCount != 0 { | 		if s.MaxMessageCount != 0 { | ||||||
| 			err = s.MessageRemove(t.Message) | 			err = s.MessageRemove(t.Message) | ||||||
| 		} | 		} | ||||||
|  | 	case *MessageDeleteBulk: | ||||||
|  | 		if s.MaxMessageCount != 0 { | ||||||
|  | 			for _, mID := range t.Messages { | ||||||
|  | 				s.messageRemoveByID(t.ChannelID, mID) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 	case *VoiceStateUpdate: | 	case *VoiceStateUpdate: | ||||||
| 		if s.TrackVoice { | 		if s.TrackVoice { | ||||||
| 			err = s.voiceStateUpdate(t) | 			err = s.voiceStateUpdate(t) | ||||||
| 		} | 		} | ||||||
|  | 	case *PresenceUpdate: | ||||||
|  | 		if s.TrackPresences { | ||||||
|  | 			s.PresenceAdd(t.GuildID, &t.Presence) | ||||||
|  | 		} | ||||||
|  | 		if s.TrackMembers { | ||||||
|  | 			if t.Status == StatusOffline { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			var m *Member | ||||||
|  | 			m, err = s.Member(t.GuildID, t.User.ID) | ||||||
|  |  | ||||||
|  | 			if err != nil { | ||||||
|  | 				// Member not found; this is a user coming online | ||||||
|  | 				m = &Member{ | ||||||
|  | 					GuildID: t.GuildID, | ||||||
|  | 					Nick:    t.Nick, | ||||||
|  | 					User:    t.User, | ||||||
|  | 					Roles:   t.Roles, | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 			} else { | ||||||
|  |  | ||||||
|  | 				if t.Nick != "" { | ||||||
|  | 					m.Nick = t.Nick | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				if t.User.Username != "" { | ||||||
|  | 					m.User.Username = t.User.Username | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				// PresenceUpdates always contain a list of roles, so there's no need to check for an empty list here | ||||||
|  | 				m.Roles = t.Roles | ||||||
|  |  | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			err = s.MemberAdd(m) | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| @@ -747,48 +945,46 @@ func (s *State) UserChannelPermissions(userID, channelID string) (apermissions i | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, role := range guild.Roles { | 	return memberPermissions(guild, channel, member), nil | ||||||
| 		if role.ID == guild.ID { |  | ||||||
| 			apermissions |= role.Permissions |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| 	for _, role := range guild.Roles { | // UserColor returns the color of a user in a channel. | ||||||
|  | // While colors are defined at a Guild level, determining for a channel is more useful in message handlers. | ||||||
|  | // 0 is returned in cases of error, which is the color of @everyone. | ||||||
|  | // userID    : The ID of the user to calculate the color for. | ||||||
|  | // channelID   : The ID of the channel to calculate the color for. | ||||||
|  | func (s *State) UserColor(userID, channelID string) int { | ||||||
|  | 	if s == nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	channel, err := s.Channel(channelID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	guild, err := s.Guild(channel.GuildID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	member, err := s.Member(guild.ID, userID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return 0 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	roles := Roles(guild.Roles) | ||||||
|  | 	sort.Sort(roles) | ||||||
|  |  | ||||||
|  | 	for _, role := range roles { | ||||||
| 		for _, roleID := range member.Roles { | 		for _, roleID := range member.Roles { | ||||||
| 			if role.ID == roleID { | 			if role.ID == roleID { | ||||||
| 				apermissions |= role.Permissions | 				if role.Color != 0 { | ||||||
| 				break | 					return role.Color | ||||||
|  | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if apermissions&PermissionAdministrator > 0 { | 	return 0 | ||||||
| 		apermissions |= PermissionAll |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Member overwrites can override role overrides, so do two passes |  | ||||||
| 	for _, overwrite := range channel.PermissionOverwrites { |  | ||||||
| 		for _, roleID := range member.Roles { |  | ||||||
| 			if overwrite.Type == "role" && roleID == overwrite.ID { |  | ||||||
| 				apermissions &= ^overwrite.Deny |  | ||||||
| 				apermissions |= overwrite.Allow |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for _, overwrite := range channel.PermissionOverwrites { |  | ||||||
| 		if overwrite.Type == "member" && overwrite.ID == userID { |  | ||||||
| 			apermissions &= ^overwrite.Deny |  | ||||||
| 			apermissions |= overwrite.Allow |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if apermissions&PermissionAdministrator > 0 { |  | ||||||
| 		apermissions |= PermissionAllChannel |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return |  | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										237
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										237
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -13,7 +13,7 @@ package discordgo | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"strconv" | 	"net/http" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| @@ -28,6 +28,7 @@ type Session struct { | |||||||
|  |  | ||||||
| 	// Authentication token for this session | 	// Authentication token for this session | ||||||
| 	Token string | 	Token string | ||||||
|  | 	MFA   bool | ||||||
|  |  | ||||||
| 	// Debug for printing JSON request/responses | 	// Debug for printing JSON request/responses | ||||||
| 	Debug    bool // Deprecated, will be removed. | 	Debug    bool // Deprecated, will be removed. | ||||||
| @@ -48,6 +49,10 @@ type Session struct { | |||||||
| 	// active guilds and the members of the guilds. | 	// active guilds and the members of the guilds. | ||||||
| 	StateEnabled bool | 	StateEnabled bool | ||||||
|  |  | ||||||
|  | 	// Whether or not to call event handlers synchronously. | ||||||
|  | 	// e.g false = launch event handlers in their own goroutines. | ||||||
|  | 	SyncEvents bool | ||||||
|  |  | ||||||
| 	// Exposed but should not be modified by User. | 	// Exposed but should not be modified by User. | ||||||
|  |  | ||||||
| 	// Whether the Data Websocket is ready | 	// Whether the Data Websocket is ready | ||||||
| @@ -73,6 +78,15 @@ type Session struct { | |||||||
| 	// StateEnabled is true. | 	// StateEnabled is true. | ||||||
| 	State *State | 	State *State | ||||||
|  |  | ||||||
|  | 	// The http client used for REST requests | ||||||
|  | 	Client *http.Client | ||||||
|  |  | ||||||
|  | 	// Stores the last HeartbeatAck that was recieved (in UTC) | ||||||
|  | 	LastHeartbeatAck time.Time | ||||||
|  |  | ||||||
|  | 	// used to deal with rate limits | ||||||
|  | 	Ratelimiter *RateLimiter | ||||||
|  |  | ||||||
| 	// Event handlers | 	// Event handlers | ||||||
| 	handlersMu   sync.RWMutex | 	handlersMu   sync.RWMutex | ||||||
| 	handlers     map[string][]*eventHandlerInstance | 	handlers     map[string][]*eventHandlerInstance | ||||||
| @@ -84,11 +98,8 @@ type Session struct { | |||||||
| 	// When nil, the session is not listening. | 	// When nil, the session is not listening. | ||||||
| 	listening chan interface{} | 	listening chan interface{} | ||||||
|  |  | ||||||
| 	// used to deal with rate limits |  | ||||||
| 	ratelimiter *RateLimiter |  | ||||||
|  |  | ||||||
| 	// sequence tracks the current gateway api websocket sequence number | 	// sequence tracks the current gateway api websocket sequence number | ||||||
| 	sequence int | 	sequence *int64 | ||||||
|  |  | ||||||
| 	// stores sessions current Discord Gateway | 	// stores sessions current Discord Gateway | ||||||
| 	gateway string | 	gateway string | ||||||
| @@ -100,12 +111,6 @@ type Session struct { | |||||||
| 	wsMutex sync.Mutex | 	wsMutex sync.Mutex | ||||||
| } | } | ||||||
|  |  | ||||||
| type rateLimitMutex struct { |  | ||||||
| 	sync.Mutex |  | ||||||
| 	url map[string]*sync.Mutex |  | ||||||
| 	// bucket map[string]*sync.Mutex // TODO :) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // A VoiceRegion stores data for a specific voice region server. | // A VoiceRegion stores data for a specific voice region server. | ||||||
| type VoiceRegion struct { | type VoiceRegion struct { | ||||||
| 	ID       string `json:"id"` | 	ID       string `json:"id"` | ||||||
| @@ -137,25 +142,50 @@ type Invite struct { | |||||||
| 	MaxAge    int       `json:"max_age"` | 	MaxAge    int       `json:"max_age"` | ||||||
| 	Uses      int       `json:"uses"` | 	Uses      int       `json:"uses"` | ||||||
| 	MaxUses   int       `json:"max_uses"` | 	MaxUses   int       `json:"max_uses"` | ||||||
| 	XkcdPass  string    `json:"xkcdpass"` |  | ||||||
| 	Revoked   bool      `json:"revoked"` | 	Revoked   bool      `json:"revoked"` | ||||||
| 	Temporary bool      `json:"temporary"` | 	Temporary bool      `json:"temporary"` | ||||||
|  | 	Unique    bool      `json:"unique"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ChannelType is the type of a Channel | ||||||
|  | type ChannelType int | ||||||
|  |  | ||||||
|  | // Block contains known ChannelType values | ||||||
|  | const ( | ||||||
|  | 	ChannelTypeGuildText ChannelType = iota | ||||||
|  | 	ChannelTypeDM | ||||||
|  | 	ChannelTypeGuildVoice | ||||||
|  | 	ChannelTypeGroupDM | ||||||
|  | 	ChannelTypeGuildCategory | ||||||
|  | ) | ||||||
|  |  | ||||||
| // A Channel holds all data related to an individual Discord channel. | // A Channel holds all data related to an individual Discord channel. | ||||||
| type Channel struct { | type Channel struct { | ||||||
| 	ID                   string                 `json:"id"` | 	ID                   string                 `json:"id"` | ||||||
| 	GuildID              string                 `json:"guild_id"` | 	GuildID              string                 `json:"guild_id"` | ||||||
| 	Name                 string                 `json:"name"` | 	Name                 string                 `json:"name"` | ||||||
| 	Topic                string                 `json:"topic"` | 	Topic                string                 `json:"topic"` | ||||||
| 	Type                 string                 `json:"type"` | 	Type                 ChannelType            `json:"type"` | ||||||
| 	LastMessageID        string                 `json:"last_message_id"` | 	LastMessageID        string                 `json:"last_message_id"` | ||||||
|  | 	NSFW                 bool                   `json:"nsfw"` | ||||||
| 	Position             int                    `json:"position"` | 	Position             int                    `json:"position"` | ||||||
| 	Bitrate              int                    `json:"bitrate"` | 	Bitrate              int                    `json:"bitrate"` | ||||||
| 	IsPrivate            bool                   `json:"is_private"` | 	Recipients           []*User                `json:"recipients"` | ||||||
| 	Recipient            *User                  `json:"recipient"` |  | ||||||
| 	Messages             []*Message             `json:"-"` | 	Messages             []*Message             `json:"-"` | ||||||
| 	PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"` | 	PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"` | ||||||
|  | 	ParentID             string                 `json:"parent_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A ChannelEdit holds Channel Feild data for a channel edit. | ||||||
|  | type ChannelEdit struct { | ||||||
|  | 	Name                 string                 `json:"name,omitempty"` | ||||||
|  | 	Topic                string                 `json:"topic,omitempty"` | ||||||
|  | 	NSFW                 bool                   `json:"nsfw,omitempty"` | ||||||
|  | 	Position             int                    `json:"position"` | ||||||
|  | 	Bitrate              int                    `json:"bitrate,omitempty"` | ||||||
|  | 	UserLimit            int                    `json:"user_limit,omitempty"` | ||||||
|  | 	PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites,omitempty"` | ||||||
|  | 	ParentID             string                 `json:"parent_id,omitempty"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // A PermissionOverwrite holds permission overwrite data for a Channel | // A PermissionOverwrite holds permission overwrite data for a Channel | ||||||
| @@ -173,6 +203,7 @@ type Emoji struct { | |||||||
| 	Roles         []string `json:"roles"` | 	Roles         []string `json:"roles"` | ||||||
| 	Managed       bool     `json:"managed"` | 	Managed       bool     `json:"managed"` | ||||||
| 	RequireColons bool     `json:"require_colons"` | 	RequireColons bool     `json:"require_colons"` | ||||||
|  | 	Animated      bool     `json:"animated"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // APIName returns an correctly formatted API name for use in the MessageReactions endpoints. | // APIName returns an correctly formatted API name for use in the MessageReactions endpoints. | ||||||
| @@ -186,7 +217,7 @@ func (e *Emoji) APIName() string { | |||||||
| 	return e.ID | 	return e.ID | ||||||
| } | } | ||||||
|  |  | ||||||
| // VerificationLevel type defination | // VerificationLevel type definition | ||||||
| type VerificationLevel int | type VerificationLevel int | ||||||
|  |  | ||||||
| // Constants for VerificationLevel levels from 0 to 3 inclusive | // Constants for VerificationLevel levels from 0 to 3 inclusive | ||||||
| @@ -235,9 +266,15 @@ type UserGuild struct { | |||||||
|  |  | ||||||
| // A GuildParams stores all the data needed to update discord guild settings | // A GuildParams stores all the data needed to update discord guild settings | ||||||
| type GuildParams struct { | type GuildParams struct { | ||||||
| 	Name              string             `json:"name"` | 	Name                        string             `json:"name,omitempty"` | ||||||
| 	Region            string             `json:"region"` | 	Region                      string             `json:"region,omitempty"` | ||||||
| 	VerificationLevel *VerificationLevel `json:"verification_level"` | 	VerificationLevel           *VerificationLevel `json:"verification_level,omitempty"` | ||||||
|  | 	DefaultMessageNotifications int                `json:"default_message_notifications,omitempty"` // TODO: Separate type? | ||||||
|  | 	AfkChannelID                string             `json:"afk_channel_id,omitempty"` | ||||||
|  | 	AfkTimeout                  int                `json:"afk_timeout,omitempty"` | ||||||
|  | 	Icon                        string             `json:"icon,omitempty"` | ||||||
|  | 	OwnerID                     string             `json:"owner_id,omitempty"` | ||||||
|  | 	Splash                      string             `json:"splash,omitempty"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // A Role stores information about Discord guild member roles. | // A Role stores information about Discord guild member roles. | ||||||
| @@ -252,6 +289,21 @@ type Role struct { | |||||||
| 	Permissions int    `json:"permissions"` | 	Permissions int    `json:"permissions"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Roles are a collection of Role | ||||||
|  | type Roles []*Role | ||||||
|  |  | ||||||
|  | func (r Roles) Len() int { | ||||||
|  | 	return len(r) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r Roles) Less(i, j int) bool { | ||||||
|  | 	return r[i].Position > r[j].Position | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r Roles) Swap(i, j int) { | ||||||
|  | 	r[i], r[j] = r[j], r[i] | ||||||
|  | } | ||||||
|  |  | ||||||
| // A VoiceState stores the voice states of Guilds | // A VoiceState stores the voice states of Guilds | ||||||
| type VoiceState struct { | type VoiceState struct { | ||||||
| 	UserID    string `json:"user_id"` | 	UserID    string `json:"user_id"` | ||||||
| @@ -272,45 +324,59 @@ type Presence struct { | |||||||
| 	Game   *Game    `json:"game"` | 	Game   *Game    `json:"game"` | ||||||
| 	Nick   string   `json:"nick"` | 	Nick   string   `json:"nick"` | ||||||
| 	Roles  []string `json:"roles"` | 	Roles  []string `json:"roles"` | ||||||
|  | 	Since  *int     `json:"since"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // GameType is the type of "game" (see GameType* consts) in the Game struct | ||||||
|  | type GameType int | ||||||
|  |  | ||||||
|  | // Valid GameType values | ||||||
|  | const ( | ||||||
|  | 	GameTypeGame GameType = iota | ||||||
|  | 	GameTypeStreaming | ||||||
|  | ) | ||||||
|  |  | ||||||
| // A Game struct holds the name of the "playing .." game for a user | // A Game struct holds the name of the "playing .." game for a user | ||||||
| type Game struct { | type Game struct { | ||||||
| 	Name          string     `json:"name"` | 	Name          string     `json:"name"` | ||||||
| 	Type int    `json:"type"` | 	Type          GameType   `json:"type"` | ||||||
| 	URL  string `json:"url"` | 	URL           string     `json:"url,omitempty"` | ||||||
|  | 	Details       string     `json:"details,omitempty"` | ||||||
|  | 	State         string     `json:"state,omitempty"` | ||||||
|  | 	TimeStamps    TimeStamps `json:"timestamps,omitempty"` | ||||||
|  | 	Assets        Assets     `json:"assets,omitempty"` | ||||||
|  | 	ApplicationID string     `json:"application_id,omitempty"` | ||||||
|  | 	Instance      int8       `json:"instance,omitempty"` | ||||||
|  | 	// TODO: Party and Secrets (unknown structure) | ||||||
| } | } | ||||||
|  |  | ||||||
| // UnmarshalJSON unmarshals json to Game struct | // A TimeStamps struct contains start and end times used in the rich presence "playing .." Game | ||||||
| func (g *Game) UnmarshalJSON(bytes []byte) error { | type TimeStamps struct { | ||||||
| 	temp := &struct { | 	EndTimestamp   int64 `json:"end,omitempty"` | ||||||
| 		Name string          `json:"name"` | 	StartTimestamp int64 `json:"start,omitempty"` | ||||||
| 		Type json.RawMessage `json:"type"` | } | ||||||
| 		URL  string          `json:"url"` |  | ||||||
|  | // UnmarshalJSON unmarshals JSON into TimeStamps struct | ||||||
|  | func (t *TimeStamps) UnmarshalJSON(b []byte) error { | ||||||
|  | 	temp := struct { | ||||||
|  | 		End   float64 `json:"end,omitempty"` | ||||||
|  | 		Start float64 `json:"start,omitempty"` | ||||||
| 	}{} | 	}{} | ||||||
| 	err := json.Unmarshal(bytes, temp) | 	err := json.Unmarshal(b, &temp) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	g.Name = temp.Name | 	t.EndTimestamp = int64(temp.End) | ||||||
| 	g.URL = temp.URL | 	t.StartTimestamp = int64(temp.Start) | ||||||
|  |  | ||||||
| 	if temp.Type != nil { |  | ||||||
| 		err = json.Unmarshal(temp.Type, &g.Type) |  | ||||||
| 		if err == nil { |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| 		s := "" | // An Assets struct contains assets and labels used in the rich presence "playing .." Game | ||||||
| 		err = json.Unmarshal(temp.Type, &s) | type Assets struct { | ||||||
| 		if err == nil { | 	LargeImageID string `json:"large_image,omitempty"` | ||||||
| 			g.Type, err = strconv.Atoi(s) | 	SmallImageID string `json:"small_image,omitempty"` | ||||||
| 		} | 	LargeText    string `json:"large_text,omitempty"` | ||||||
|  | 	SmallText    string `json:"small_text,omitempty"` | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // A Member stores user information for Guild members. | // A Member stores user information for Guild members. | ||||||
| @@ -324,19 +390,6 @@ type Member struct { | |||||||
| 	Roles    []string `json:"roles"` | 	Roles    []string `json:"roles"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // A User stores all data for an individual Discord user. |  | ||||||
| type User struct { |  | ||||||
| 	ID            string `json:"id"` |  | ||||||
| 	Email         string `json:"email"` |  | ||||||
| 	Username      string `json:"username"` |  | ||||||
| 	Avatar        string `json:"Avatar"` |  | ||||||
| 	Discriminator string `json:"discriminator"` |  | ||||||
| 	Token         string `json:"token"` |  | ||||||
| 	Verified      bool   `json:"verified"` |  | ||||||
| 	MFAEnabled    bool   `json:"mfa_enabled"` |  | ||||||
| 	Bot           bool   `json:"bot"` |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // A Settings stores data for a specific users Discord client settings. | // A Settings stores data for a specific users Discord client settings. | ||||||
| type Settings struct { | type Settings struct { | ||||||
| 	RenderEmbeds           bool               `json:"render_embeds"` | 	RenderEmbeds           bool               `json:"render_embeds"` | ||||||
| @@ -356,7 +409,7 @@ type Settings struct { | |||||||
| 	DeveloperMode          bool               `json:"developer_mode"` | 	DeveloperMode          bool               `json:"developer_mode"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Status type defination | // Status type definition | ||||||
| type Status string | type Status string | ||||||
|  |  | ||||||
| // Constants for Status with the different current available status | // Constants for Status with the different current available status | ||||||
| @@ -502,6 +555,12 @@ type MessageReaction struct { | |||||||
| 	ChannelID string `json:"channel_id"` | 	ChannelID string `json:"channel_id"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // GatewayBotResponse stores the data for the gateway/bot response | ||||||
|  | type GatewayBotResponse struct { | ||||||
|  | 	URL    string `json:"url"` | ||||||
|  | 	Shards int    `json:"shards"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Constants for the different bit offsets of text channel permissions | // Constants for the different bit offsets of text channel permissions | ||||||
| const ( | const ( | ||||||
| 	PermissionReadMessages = 1 << (iota + 10) | 	PermissionReadMessages = 1 << (iota + 10) | ||||||
| @@ -542,6 +601,8 @@ const ( | |||||||
| 	PermissionAdministrator | 	PermissionAdministrator | ||||||
| 	PermissionManageChannels | 	PermissionManageChannels | ||||||
| 	PermissionManageServer | 	PermissionManageServer | ||||||
|  | 	PermissionAddReactions | ||||||
|  | 	PermissionViewAuditLogs | ||||||
|  |  | ||||||
| 	PermissionAllText = PermissionReadMessages | | 	PermissionAllText = PermissionReadMessages | | ||||||
| 		PermissionSendMessages | | 		PermissionSendMessages | | ||||||
| @@ -561,9 +622,65 @@ const ( | |||||||
| 		PermissionAllVoice | | 		PermissionAllVoice | | ||||||
| 		PermissionCreateInstantInvite | | 		PermissionCreateInstantInvite | | ||||||
| 		PermissionManageRoles | | 		PermissionManageRoles | | ||||||
| 		PermissionManageChannels | 		PermissionManageChannels | | ||||||
|  | 		PermissionAddReactions | | ||||||
|  | 		PermissionViewAuditLogs | ||||||
| 	PermissionAll = PermissionAllChannel | | 	PermissionAll = PermissionAllChannel | | ||||||
| 		PermissionKickMembers | | 		PermissionKickMembers | | ||||||
| 		PermissionBanMembers | | 		PermissionBanMembers | | ||||||
| 		PermissionManageServer | 		PermissionManageServer | | ||||||
|  | 		PermissionAdministrator | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Block contains Discord JSON Error Response codes | ||||||
|  | const ( | ||||||
|  | 	ErrCodeUnknownAccount     = 10001 | ||||||
|  | 	ErrCodeUnknownApplication = 10002 | ||||||
|  | 	ErrCodeUnknownChannel     = 10003 | ||||||
|  | 	ErrCodeUnknownGuild       = 10004 | ||||||
|  | 	ErrCodeUnknownIntegration = 10005 | ||||||
|  | 	ErrCodeUnknownInvite      = 10006 | ||||||
|  | 	ErrCodeUnknownMember      = 10007 | ||||||
|  | 	ErrCodeUnknownMessage     = 10008 | ||||||
|  | 	ErrCodeUnknownOverwrite   = 10009 | ||||||
|  | 	ErrCodeUnknownProvider    = 10010 | ||||||
|  | 	ErrCodeUnknownRole        = 10011 | ||||||
|  | 	ErrCodeUnknownToken       = 10012 | ||||||
|  | 	ErrCodeUnknownUser        = 10013 | ||||||
|  | 	ErrCodeUnknownEmoji       = 10014 | ||||||
|  |  | ||||||
|  | 	ErrCodeBotsCannotUseEndpoint  = 20001 | ||||||
|  | 	ErrCodeOnlyBotsCanUseEndpoint = 20002 | ||||||
|  |  | ||||||
|  | 	ErrCodeMaximumGuildsReached     = 30001 | ||||||
|  | 	ErrCodeMaximumFriendsReached    = 30002 | ||||||
|  | 	ErrCodeMaximumPinsReached       = 30003 | ||||||
|  | 	ErrCodeMaximumGuildRolesReached = 30005 | ||||||
|  | 	ErrCodeTooManyReactions         = 30010 | ||||||
|  |  | ||||||
|  | 	ErrCodeUnauthorized = 40001 | ||||||
|  |  | ||||||
|  | 	ErrCodeMissingAccess                             = 50001 | ||||||
|  | 	ErrCodeInvalidAccountType                        = 50002 | ||||||
|  | 	ErrCodeCannotExecuteActionOnDMChannel            = 50003 | ||||||
|  | 	ErrCodeEmbedCisabled                             = 50004 | ||||||
|  | 	ErrCodeCannotEditFromAnotherUser                 = 50005 | ||||||
|  | 	ErrCodeCannotSendEmptyMessage                    = 50006 | ||||||
|  | 	ErrCodeCannotSendMessagesToThisUser              = 50007 | ||||||
|  | 	ErrCodeCannotSendMessagesInVoiceChannel          = 50008 | ||||||
|  | 	ErrCodeChannelVerificationLevelTooHigh           = 50009 | ||||||
|  | 	ErrCodeOAuth2ApplicationDoesNotHaveBot           = 50010 | ||||||
|  | 	ErrCodeOAuth2ApplicationLimitReached             = 50011 | ||||||
|  | 	ErrCodeInvalidOAuthState                         = 50012 | ||||||
|  | 	ErrCodeMissingPermissions                        = 50013 | ||||||
|  | 	ErrCodeInvalidAuthenticationToken                = 50014 | ||||||
|  | 	ErrCodeNoteTooLong                               = 50015 | ||||||
|  | 	ErrCodeTooFewOrTooManyMessagesToDelete           = 50016 | ||||||
|  | 	ErrCodeCanOnlyPinMessageToOriginatingChannel     = 50019 | ||||||
|  | 	ErrCodeCannotExecuteActionOnSystemMessage        = 50021 | ||||||
|  | 	ErrCodeMessageProvidedTooOldForBulkDelete        = 50034 | ||||||
|  | 	ErrCodeInvalidFormBody                           = 50035 | ||||||
|  | 	ErrCodeInviteAcceptedToGuildApplicationsBotNotIn = 50036 | ||||||
|  |  | ||||||
|  | 	ErrCodeReactionBlocked = 90001 | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										7
									
								
								vendor/github.com/bwmarrin/discordgo/tools/cmd/eventhandlers/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										7
									
								
								vendor/github.com/bwmarrin/discordgo/tools/cmd/eventhandlers/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -37,18 +37,18 @@ type {{privateName .}}EventHandler func(*Session, *{{.}}) | |||||||
| func (eh {{privateName .}}EventHandler) Type() string { | func (eh {{privateName .}}EventHandler) Type() string { | ||||||
|   return {{privateName .}}EventType |   return {{privateName .}}EventType | ||||||
| } | } | ||||||
|  | {{if isDiscordEvent .}} | ||||||
| // New returns a new instance of {{.}}. | // New returns a new instance of {{.}}. | ||||||
| func (eh {{privateName .}}EventHandler) New() interface{} { | func (eh {{privateName .}}EventHandler) New() interface{} { | ||||||
|   return &{{.}}{} |   return &{{.}}{} | ||||||
| } | }{{end}} | ||||||
|  |  | ||||||
| // Handle is the handler for {{.}} events. | // Handle is the handler for {{.}} events. | ||||||
| func (eh {{privateName .}}EventHandler) Handle(s *Session, i interface{}) { | func (eh {{privateName .}}EventHandler) Handle(s *Session, i interface{}) { | ||||||
|   if t, ok := i.(*{{.}}); ok { |   if t, ok := i.(*{{.}}); ok { | ||||||
|     eh(s, t) |     eh(s, t) | ||||||
|   } |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| {{end}} | {{end}} | ||||||
| func handlerForInterface(handler interface{}) EventHandler { | func handlerForInterface(handler interface{}) EventHandler { | ||||||
|   switch v := handler.(type) { |   switch v := handler.(type) { | ||||||
| @@ -60,6 +60,7 @@ func handlerForInterface(handler interface{}) EventHandler { | |||||||
|  |  | ||||||
|   return nil |   return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func init() { {{range .}}{{if isDiscordEvent .}} | func init() { {{range .}}{{if isDiscordEvent .}} | ||||||
|   registerInterfaceProvider({{privateName .}}EventHandler(nil)){{end}}{{end}} |   registerInterfaceProvider({{privateName .}}EventHandler(nil)){{end}}{{end}} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										47
									
								
								vendor/github.com/bwmarrin/discordgo/user.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								vendor/github.com/bwmarrin/discordgo/user.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,47 @@ | |||||||
|  | package discordgo | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // A User stores all data for an individual Discord user. | ||||||
|  | type User struct { | ||||||
|  | 	ID            string `json:"id"` | ||||||
|  | 	Email         string `json:"email"` | ||||||
|  | 	Username      string `json:"username"` | ||||||
|  | 	Avatar        string `json:"avatar"` | ||||||
|  | 	Discriminator string `json:"discriminator"` | ||||||
|  | 	Token         string `json:"token"` | ||||||
|  | 	Verified      bool   `json:"verified"` | ||||||
|  | 	MFAEnabled    bool   `json:"mfa_enabled"` | ||||||
|  | 	Bot           bool   `json:"bot"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String returns a unique identifier of the form username#discriminator | ||||||
|  | func (u *User) String() string { | ||||||
|  | 	return fmt.Sprintf("%s#%s", u.Username, u.Discriminator) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Mention return a string which mentions the user | ||||||
|  | func (u *User) Mention() string { | ||||||
|  | 	return fmt.Sprintf("<@%s>", u.ID) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AvatarURL returns a URL to the user's avatar. | ||||||
|  | //    size:    The size of the user's avatar as a power of two | ||||||
|  | //             if size is an empty string, no size parameter will | ||||||
|  | //             be added to the URL. | ||||||
|  | func (u *User) AvatarURL(size string) string { | ||||||
|  | 	var URL string | ||||||
|  | 	if strings.HasPrefix(u.Avatar, "a_") { | ||||||
|  | 		URL = EndpointUserAvatarAnimated(u.ID, u.Avatar) | ||||||
|  | 	} else { | ||||||
|  | 		URL = EndpointUserAvatar(u.ID, u.Avatar) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if size != "" { | ||||||
|  | 		return URL + "?size=" + size | ||||||
|  | 	} | ||||||
|  | 	return URL | ||||||
|  | } | ||||||
							
								
								
									
										79
									
								
								vendor/github.com/bwmarrin/discordgo/voice.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										79
									
								
								vendor/github.com/bwmarrin/discordgo/voice.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -13,9 +13,7 @@ import ( | |||||||
| 	"encoding/binary" | 	"encoding/binary" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"log" |  | ||||||
| 	"net" | 	"net" | ||||||
| 	"runtime" |  | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -70,7 +68,7 @@ type VoiceConnection struct { | |||||||
| 	voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler | 	voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler | ||||||
| } | } | ||||||
|  |  | ||||||
| // VoiceSpeakingUpdateHandler type provides a function defination for the | // VoiceSpeakingUpdateHandler type provides a function definition for the | ||||||
| // VoiceSpeakingUpdate event | // VoiceSpeakingUpdate event | ||||||
| type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate) | type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate) | ||||||
|  |  | ||||||
| @@ -93,18 +91,22 @@ func (v *VoiceConnection) Speaking(b bool) (err error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if v.wsConn == nil { | 	if v.wsConn == nil { | ||||||
| 		return fmt.Errorf("No VoiceConnection websocket.") | 		return fmt.Errorf("no VoiceConnection websocket") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	data := voiceSpeakingOp{5, voiceSpeakingData{b, 0}} | 	data := voiceSpeakingOp{5, voiceSpeakingData{b, 0}} | ||||||
| 	v.wsMutex.Lock() | 	v.wsMutex.Lock() | ||||||
| 	err = v.wsConn.WriteJSON(data) | 	err = v.wsConn.WriteJSON(data) | ||||||
| 	v.wsMutex.Unlock() | 	v.wsMutex.Unlock() | ||||||
|  |  | ||||||
|  | 	v.Lock() | ||||||
|  | 	defer v.Unlock() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		v.speaking = false | 		v.speaking = false | ||||||
| 		log.Println("Speaking() write json error:", err) | 		v.log(LogError, "Speaking() write json error:", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	v.speaking = b | 	v.speaking = b | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| @@ -139,9 +141,9 @@ func (v *VoiceConnection) Disconnect() (err error) { | |||||||
| 	// Send a OP4 with a nil channel to disconnect | 	// Send a OP4 with a nil channel to disconnect | ||||||
| 	if v.sessionID != "" { | 	if v.sessionID != "" { | ||||||
| 		data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, nil, true, true}} | 		data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, nil, true, true}} | ||||||
| 		v.wsMutex.Lock() | 		v.session.wsMutex.Lock() | ||||||
| 		err = v.session.wsConn.WriteJSON(data) | 		err = v.session.wsConn.WriteJSON(data) | ||||||
| 		v.wsMutex.Unlock() | 		v.session.wsMutex.Unlock() | ||||||
| 		v.sessionID = "" | 		v.sessionID = "" | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -149,7 +151,10 @@ func (v *VoiceConnection) Disconnect() (err error) { | |||||||
| 	v.Close() | 	v.Close() | ||||||
|  |  | ||||||
| 	v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID) | 	v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID) | ||||||
|  |  | ||||||
|  | 	v.session.Lock() | ||||||
| 	delete(v.session.VoiceConnections, v.GuildID) | 	delete(v.session.VoiceConnections, v.GuildID) | ||||||
|  | 	v.session.Unlock() | ||||||
|  |  | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
| @@ -175,7 +180,7 @@ func (v *VoiceConnection) Close() { | |||||||
| 		v.log(LogInformational, "closing udp") | 		v.log(LogInformational, "closing udp") | ||||||
| 		err := v.udpConn.Close() | 		err := v.udpConn.Close() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			log.Println("error closing udp connection: ", err) | 			v.log(LogError, "error closing udp connection: ", err) | ||||||
| 		} | 		} | ||||||
| 		v.udpConn = nil | 		v.udpConn = nil | ||||||
| 	} | 	} | ||||||
| @@ -185,7 +190,9 @@ func (v *VoiceConnection) Close() { | |||||||
|  |  | ||||||
| 		// To cleanly close a connection, a client should send a close | 		// To cleanly close a connection, a client should send a close | ||||||
| 		// frame and wait for the server to close the connection. | 		// frame and wait for the server to close the connection. | ||||||
|  | 		v.wsMutex.Lock() | ||||||
| 		err := v.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) | 		err := v.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) | ||||||
|  | 		v.wsMutex.Unlock() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			v.log(LogError, "error closing websocket, %s", err) | 			v.log(LogError, "error closing websocket, %s", err) | ||||||
| 		} | 		} | ||||||
| @@ -239,19 +246,22 @@ type voiceOP2 struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // WaitUntilConnected waits for the Voice Connection to | // WaitUntilConnected waits for the Voice Connection to | ||||||
| // become ready, if it does not become ready it retuns an err | // become ready, if it does not become ready it returns an err | ||||||
| func (v *VoiceConnection) waitUntilConnected() error { | func (v *VoiceConnection) waitUntilConnected() error { | ||||||
|  |  | ||||||
| 	v.log(LogInformational, "called") | 	v.log(LogInformational, "called") | ||||||
|  |  | ||||||
| 	i := 0 | 	i := 0 | ||||||
| 	for { | 	for { | ||||||
| 		if v.Ready { | 		v.RLock() | ||||||
|  | 		ready := v.Ready | ||||||
|  | 		v.RUnlock() | ||||||
|  | 		if ready { | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if i > 10 { | 		if i > 10 { | ||||||
| 			return fmt.Errorf("Timeout waiting for voice.") | 			return fmt.Errorf("timeout waiting for voice") | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		time.Sleep(1 * time.Second) | 		time.Sleep(1 * time.Second) | ||||||
| @@ -282,7 +292,7 @@ func (v *VoiceConnection) open() (err error) { | |||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
| 		if i > 20 { // only loop for up to 1 second total | 		if i > 20 { // only loop for up to 1 second total | ||||||
| 			return fmt.Errorf("Did not receive voice Session ID in time.") | 			return fmt.Errorf("did not receive voice Session ID in time") | ||||||
| 		} | 		} | ||||||
| 		time.Sleep(50 * time.Millisecond) | 		time.Sleep(50 * time.Millisecond) | ||||||
| 		i++ | 		i++ | ||||||
| @@ -409,8 +419,6 @@ func (v *VoiceConnection) onEvent(message []byte) { | |||||||
| 			go v.opusReceiver(v.udpConn, v.close, v.OpusRecv) | 			go v.opusReceiver(v.udpConn, v.close, v.OpusRecv) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// Send the ready event |  | ||||||
| 		v.connected <- true |  | ||||||
| 		return | 		return | ||||||
|  |  | ||||||
| 	case 3: // HEARTBEAT response | 	case 3: // HEARTBEAT response | ||||||
| @@ -418,6 +426,9 @@ func (v *VoiceConnection) onEvent(message []byte) { | |||||||
| 		return | 		return | ||||||
|  |  | ||||||
| 	case 4: // udp encryption secret key | 	case 4: // udp encryption secret key | ||||||
|  | 		v.Lock() | ||||||
|  | 		defer v.Unlock() | ||||||
|  |  | ||||||
| 		v.op4 = voiceOP4{} | 		v.op4 = voiceOP4{} | ||||||
| 		if err := json.Unmarshal(e.RawData, &v.op4); err != nil { | 		if err := json.Unmarshal(e.RawData, &v.op4); err != nil { | ||||||
| 			v.log(LogError, "OP4 unmarshall error, %s, %s", err, string(e.RawData)) | 			v.log(LogError, "OP4 unmarshall error, %s, %s", err, string(e.RawData)) | ||||||
| @@ -466,6 +477,7 @@ func (v *VoiceConnection) wsHeartbeat(wsConn *websocket.Conn, close <-chan struc | |||||||
|  |  | ||||||
| 	var err error | 	var err error | ||||||
| 	ticker := time.NewTicker(i * time.Millisecond) | 	ticker := time.NewTicker(i * time.Millisecond) | ||||||
|  | 	defer ticker.Stop() | ||||||
| 	for { | 	for { | ||||||
| 		v.log(LogDebug, "sending heartbeat packet") | 		v.log(LogDebug, "sending heartbeat packet") | ||||||
| 		v.wsMutex.Lock() | 		v.wsMutex.Lock() | ||||||
| @@ -616,6 +628,7 @@ func (v *VoiceConnection) udpKeepAlive(udpConn *net.UDPConn, close <-chan struct | |||||||
| 	packet := make([]byte, 8) | 	packet := make([]byte, 8) | ||||||
|  |  | ||||||
| 	ticker := time.NewTicker(i) | 	ticker := time.NewTicker(i) | ||||||
|  | 	defer ticker.Stop() | ||||||
| 	for { | 	for { | ||||||
|  |  | ||||||
| 		binary.LittleEndian.PutUint64(packet, sequence) | 		binary.LittleEndian.PutUint64(packet, sequence) | ||||||
| @@ -644,12 +657,16 @@ func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{} | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	runtime.LockOSThread() |  | ||||||
|  |  | ||||||
| 	// VoiceConnection is now ready to receive audio packets | 	// VoiceConnection is now ready to receive audio packets | ||||||
| 	// TODO: this needs reviewed as I think there must be a better way. | 	// TODO: this needs reviewed as I think there must be a better way. | ||||||
|  | 	v.Lock() | ||||||
| 	v.Ready = true | 	v.Ready = true | ||||||
| 	defer func() { v.Ready = false }() | 	v.Unlock() | ||||||
|  | 	defer func() { | ||||||
|  | 		v.Lock() | ||||||
|  | 		v.Ready = false | ||||||
|  | 		v.Unlock() | ||||||
|  | 	}() | ||||||
|  |  | ||||||
| 	var sequence uint16 | 	var sequence uint16 | ||||||
| 	var timestamp uint32 | 	var timestamp uint32 | ||||||
| @@ -665,6 +682,7 @@ func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{} | |||||||
|  |  | ||||||
| 	// start a send loop that loops until buf chan is closed | 	// start a send loop that loops until buf chan is closed | ||||||
| 	ticker := time.NewTicker(time.Millisecond * time.Duration(size/(rate/1000))) | 	ticker := time.NewTicker(time.Millisecond * time.Duration(size/(rate/1000))) | ||||||
|  | 	defer ticker.Stop() | ||||||
| 	for { | 	for { | ||||||
|  |  | ||||||
| 		// Get data from chan.  If chan is closed, return. | 		// Get data from chan.  If chan is closed, return. | ||||||
| @@ -678,7 +696,10 @@ func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{} | |||||||
| 			// else, continue loop | 			// else, continue loop | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if !v.speaking { | 		v.RLock() | ||||||
|  | 		speaking := v.speaking | ||||||
|  | 		v.RUnlock() | ||||||
|  | 		if !speaking { | ||||||
| 			err := v.Speaking(true) | 			err := v.Speaking(true) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				v.log(LogError, "error sending speaking packet, %s", err) | 				v.log(LogError, "error sending speaking packet, %s", err) | ||||||
| @@ -691,7 +712,9 @@ func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{} | |||||||
|  |  | ||||||
| 		// encrypt the opus data | 		// encrypt the opus data | ||||||
| 		copy(nonce[:], udpHeader) | 		copy(nonce[:], udpHeader) | ||||||
|  | 		v.RLock() | ||||||
| 		sendbuf := secretbox.Seal(udpHeader, recvbuf, &nonce, &v.op4.SecretKey) | 		sendbuf := secretbox.Seal(udpHeader, recvbuf, &nonce, &v.op4.SecretKey) | ||||||
|  | 		v.RUnlock() | ||||||
|  |  | ||||||
| 		// block here until we're exactly at the right time :) | 		// block here until we're exactly at the right time :) | ||||||
| 		// Then send rtp audio packet to Discord over UDP | 		// Then send rtp audio packet to Discord over UDP | ||||||
| @@ -742,7 +765,6 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	p := Packet{} |  | ||||||
| 	recvbuf := make([]byte, 1024) | 	recvbuf := make([]byte, 1024) | ||||||
| 	var nonce [24]byte | 	var nonce [24]byte | ||||||
|  |  | ||||||
| @@ -773,11 +795,12 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// For now, skip anything except audio. | 		// For now, skip anything except audio. | ||||||
| 		if rlen < 12 || recvbuf[0] != 0x80 { | 		if rlen < 12 || (recvbuf[0] != 0x80 && recvbuf[0] != 0x90) { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// build a audio packet struct | 		// build a audio packet struct | ||||||
|  | 		p := Packet{} | ||||||
| 		p.Type = recvbuf[0:2] | 		p.Type = recvbuf[0:2] | ||||||
| 		p.Sequence = binary.BigEndian.Uint16(recvbuf[2:4]) | 		p.Sequence = binary.BigEndian.Uint16(recvbuf[2:4]) | ||||||
| 		p.Timestamp = binary.BigEndian.Uint32(recvbuf[4:8]) | 		p.Timestamp = binary.BigEndian.Uint32(recvbuf[4:8]) | ||||||
| @@ -786,8 +809,17 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct | |||||||
| 		copy(nonce[:], recvbuf[0:12]) | 		copy(nonce[:], recvbuf[0:12]) | ||||||
| 		p.Opus, _ = secretbox.Open(nil, recvbuf[12:rlen], &nonce, &v.op4.SecretKey) | 		p.Opus, _ = secretbox.Open(nil, recvbuf[12:rlen], &nonce, &v.op4.SecretKey) | ||||||
|  |  | ||||||
|  | 		if len(p.Opus) > 8 && recvbuf[0] == 0x90 { | ||||||
|  | 			// Extension bit is set, first 8 bytes is the extended header | ||||||
|  | 			p.Opus = p.Opus[8:] | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		if c != nil { | 		if c != nil { | ||||||
| 			c <- &p | 			select { | ||||||
|  | 			case c <- &p: | ||||||
|  | 			case <-close: | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -825,7 +857,7 @@ func (v *VoiceConnection) reconnect() { | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if v.session.DataReady == false || v.session.wsConn == nil { | 		if v.session.DataReady == false || v.session.wsConn == nil { | ||||||
| 			v.log(LogInformational, "cannot reconenct to channel %s with unready session", v.ChannelID) | 			v.log(LogInformational, "cannot reconnect to channel %s with unready session", v.ChannelID) | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| @@ -837,6 +869,8 @@ func (v *VoiceConnection) reconnect() { | |||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		v.log(LogInformational, "error reconnecting to channel %s, %s", v.ChannelID, err) | ||||||
|  |  | ||||||
| 		// if the reconnect above didn't work lets just send a disconnect | 		// if the reconnect above didn't work lets just send a disconnect | ||||||
| 		// packet to reset things. | 		// packet to reset things. | ||||||
| 		// Send a OP4 with a nil channel to disconnect | 		// Send a OP4 with a nil channel to disconnect | ||||||
| @@ -848,6 +882,5 @@ func (v *VoiceConnection) reconnect() { | |||||||
| 			v.log(LogError, "error sending disconnect packet, %s", err) | 			v.log(LogError, "error sending disconnect packet, %s", err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		v.log(LogInformational, "error reconnecting to channel %s, %s", v.ChannelID, err) |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										342
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										342
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -19,33 +19,141 @@ import ( | |||||||
| 	"io" | 	"io" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"runtime" | 	"runtime" | ||||||
|  | 	"sync/atomic" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/gorilla/websocket" | 	"github.com/gorilla/websocket" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // ErrWSAlreadyOpen is thrown when you attempt to open | ||||||
|  | // a websocket that already is open. | ||||||
|  | var ErrWSAlreadyOpen = errors.New("web socket already opened") | ||||||
|  |  | ||||||
|  | // ErrWSNotFound is thrown when you attempt to use a websocket | ||||||
|  | // that doesn't exist | ||||||
|  | var ErrWSNotFound = errors.New("no websocket connection exists") | ||||||
|  |  | ||||||
|  | // ErrWSShardBounds is thrown when you try to use a shard ID that is | ||||||
|  | // less than the total shard count | ||||||
|  | var ErrWSShardBounds = errors.New("ShardID must be less than ShardCount") | ||||||
|  |  | ||||||
| type resumePacket struct { | type resumePacket struct { | ||||||
| 	Op   int `json:"op"` | 	Op   int `json:"op"` | ||||||
| 	Data struct { | 	Data struct { | ||||||
| 		Token     string `json:"token"` | 		Token     string `json:"token"` | ||||||
| 		SessionID string `json:"session_id"` | 		SessionID string `json:"session_id"` | ||||||
| 		Sequence  int    `json:"seq"` | 		Sequence  int64  `json:"seq"` | ||||||
| 	} `json:"d"` | 	} `json:"d"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Open opens a websocket connection to Discord. | // Open creates a websocket connection to Discord. | ||||||
| func (s *Session) Open() (err error) { | // See: https://discordapp.com/developers/docs/topics/gateway#connecting | ||||||
|  | func (s *Session) Open() error { | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	// Prevent Open or other major Session functions from | ||||||
|  | 	// being called while Open is still running. | ||||||
| 	s.Lock() | 	s.Lock() | ||||||
| 	defer func() { | 	defer s.Unlock() | ||||||
|  |  | ||||||
|  | 	// If the websock is already open, bail out here. | ||||||
|  | 	if s.wsConn != nil { | ||||||
|  | 		return ErrWSAlreadyOpen | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Get the gateway to use for the Websocket connection | ||||||
|  | 	if s.gateway == "" { | ||||||
|  | 		s.gateway, err = s.Gateway() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			s.Unlock() | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Add the version and encoding to the URL | ||||||
|  | 		s.gateway = s.gateway + "?v=" + APIVersion + "&encoding=json" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Connect to the Gateway | ||||||
|  | 	s.log(LogInformational, "connecting to gateway %s", s.gateway) | ||||||
|  | 	header := http.Header{} | ||||||
|  | 	header.Add("accept-encoding", "zlib") | ||||||
|  | 	s.wsConn, _, err = websocket.DefaultDialer.Dial(s.gateway, header) | ||||||
|  | 	if err != nil { | ||||||
|  | 		s.log(LogWarning, "error connecting to gateway %s, %s", s.gateway, err) | ||||||
|  | 		s.gateway = "" // clear cached gateway | ||||||
|  | 		s.wsConn = nil // Just to be safe. | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	defer func() { | ||||||
|  | 		// because of this, all code below must set err to the error | ||||||
|  | 		// when exiting with an error :)  Maybe someone has a better | ||||||
|  | 		// way :) | ||||||
|  | 		if err != nil { | ||||||
|  | 			s.wsConn.Close() | ||||||
|  | 			s.wsConn = nil | ||||||
| 		} | 		} | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
|  | 	// The first response from Discord should be an Op 10 (Hello) Packet. | ||||||
|  | 	// When processed by onEvent the heartbeat goroutine will be started. | ||||||
|  | 	mt, m, err := s.wsConn.ReadMessage() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	e, err := s.onEvent(mt, m) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if e.Operation != 10 { | ||||||
|  | 		err = fmt.Errorf("expecting Op 10, got Op %d instead", e.Operation) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	s.log(LogInformational, "Op 10 Hello Packet received from Discord") | ||||||
|  | 	s.LastHeartbeatAck = time.Now().UTC() | ||||||
|  | 	var h helloOp | ||||||
|  | 	if err = json.Unmarshal(e.RawData, &h); err != nil { | ||||||
|  | 		err = fmt.Errorf("error unmarshalling helloOp, %s", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Now we send either an Op 2 Identity if this is a brand new | ||||||
|  | 	// connection or Op 6 Resume if we are resuming an existing connection. | ||||||
|  | 	sequence := atomic.LoadInt64(s.sequence) | ||||||
|  | 	if s.sessionID == "" && sequence == 0 { | ||||||
|  |  | ||||||
|  | 		// Send Op 2 Identity Packet | ||||||
|  | 		err = s.identify() | ||||||
|  | 		if err != nil { | ||||||
|  | 			err = fmt.Errorf("error sending identify packet to gateway, %s, %s", s.gateway, err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 	} else { | ||||||
|  |  | ||||||
|  | 		// Send Op 6 Resume Packet | ||||||
|  | 		p := resumePacket{} | ||||||
|  | 		p.Op = 6 | ||||||
|  | 		p.Data.Token = s.Token | ||||||
|  | 		p.Data.SessionID = s.sessionID | ||||||
|  | 		p.Data.Sequence = sequence | ||||||
|  |  | ||||||
|  | 		s.log(LogInformational, "sending resume packet to gateway") | ||||||
|  | 		s.wsMutex.Lock() | ||||||
|  | 		err = s.wsConn.WriteJSON(p) | ||||||
|  | 		s.wsMutex.Unlock() | ||||||
|  | 		if err != nil { | ||||||
|  | 			err = fmt.Errorf("error sending gateway resume packet, %s, %s", s.gateway, err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// A basic state is a hard requirement for Voice. | 	// A basic state is a hard requirement for Voice. | ||||||
|  | 	// We create it here so the below READY/RESUMED packet can populate | ||||||
|  | 	// the state :) | ||||||
|  | 	// XXX: Move to New() func? | ||||||
| 	if s.State == nil { | 	if s.State == nil { | ||||||
| 		state := NewState() | 		state := NewState() | ||||||
| 		state.TrackChannels = false | 		state.TrackChannels = false | ||||||
| @@ -56,75 +164,42 @@ func (s *Session) Open() (err error) { | |||||||
| 		s.State = state | 		s.State = state | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if s.wsConn != nil { | 	// Now Discord should send us a READY or RESUMED packet. | ||||||
| 		err = errors.New("Web socket already opened.") | 	mt, m, err = s.wsConn.ReadMessage() | ||||||
| 		return | 	if err != nil { | ||||||
|  | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	e, err = s.onEvent(mt, m) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if e.Type != `READY` && e.Type != `RESUMED` { | ||||||
|  | 		// This is not fatal, but it does not follow their API documentation. | ||||||
|  | 		s.log(LogWarning, "Expected READY/RESUMED, instead got:\n%#v\n", e) | ||||||
|  | 	} | ||||||
|  | 	s.log(LogInformational, "First Packet:\n%#v\n", e) | ||||||
|  |  | ||||||
|  | 	s.log(LogInformational, "We are now connected to Discord, emitting connect event") | ||||||
|  | 	s.handleEvent(connectEventType, &Connect{}) | ||||||
|  |  | ||||||
|  | 	// A VoiceConnections map is a hard requirement for Voice. | ||||||
|  | 	// XXX: can this be moved to when opening a voice connection? | ||||||
| 	if s.VoiceConnections == nil { | 	if s.VoiceConnections == nil { | ||||||
| 		s.log(LogInformational, "creating new VoiceConnections map") | 		s.log(LogInformational, "creating new VoiceConnections map") | ||||||
| 		s.VoiceConnections = make(map[string]*VoiceConnection) | 		s.VoiceConnections = make(map[string]*VoiceConnection) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Get the gateway to use for the Websocket connection | 	// Create listening chan outside of listen, as it needs to happen inside the | ||||||
| 	if s.gateway == "" { | 	// mutex lock and needs to exist before calling heartbeat and listen | ||||||
| 		s.gateway, err = s.Gateway() | 	// go rountines. | ||||||
| 		if err != nil { |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Add the version and encoding to the URL |  | ||||||
| 		s.gateway = fmt.Sprintf("%s?v=4&encoding=json", s.gateway) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	header := http.Header{} |  | ||||||
| 	header.Add("accept-encoding", "zlib") |  | ||||||
|  |  | ||||||
| 	s.log(LogInformational, "connecting to gateway %s", s.gateway) |  | ||||||
| 	s.wsConn, _, err = websocket.DefaultDialer.Dial(s.gateway, header) |  | ||||||
| 	if err != nil { |  | ||||||
| 		s.log(LogWarning, "error connecting to gateway %s, %s", s.gateway, err) |  | ||||||
| 		s.gateway = "" // clear cached gateway |  | ||||||
| 		// TODO: should we add a retry block here? |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if s.sessionID != "" && s.sequence > 0 { |  | ||||||
|  |  | ||||||
| 		p := resumePacket{} |  | ||||||
| 		p.Op = 6 |  | ||||||
| 		p.Data.Token = s.Token |  | ||||||
| 		p.Data.SessionID = s.sessionID |  | ||||||
| 		p.Data.Sequence = s.sequence |  | ||||||
|  |  | ||||||
| 		s.log(LogInformational, "sending resume packet to gateway") |  | ||||||
| 		err = s.wsConn.WriteJSON(p) |  | ||||||
| 		if err != nil { |  | ||||||
| 			s.log(LogWarning, "error sending gateway resume packet, %s, %s", s.gateway, err) |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 	} else { |  | ||||||
|  |  | ||||||
| 		err = s.identify() |  | ||||||
| 		if err != nil { |  | ||||||
| 			s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err) |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Create listening outside of listen, as it needs to happen inside the mutex |  | ||||||
| 	// lock. |  | ||||||
| 	s.listening = make(chan interface{}) | 	s.listening = make(chan interface{}) | ||||||
|  |  | ||||||
|  | 	// Start sending heartbeats and reading messages from Discord. | ||||||
|  | 	go s.heartbeat(s.wsConn, s.listening, h.HeartbeatInterval) | ||||||
| 	go s.listen(s.wsConn, s.listening) | 	go s.listen(s.wsConn, s.listening) | ||||||
|  |  | ||||||
| 	s.Unlock() |  | ||||||
|  |  | ||||||
| 	s.log(LogInformational, "emit connect event") |  | ||||||
| 	s.handleEvent(connectEventType, &Connect{}) |  | ||||||
|  |  | ||||||
| 	s.log(LogInformational, "exiting") | 	s.log(LogInformational, "exiting") | ||||||
| 	return | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // listen polls the websocket connection for events, it will stop when the | // listen polls the websocket connection for events, it will stop when the | ||||||
| @@ -177,13 +252,21 @@ func (s *Session) listen(wsConn *websocket.Conn, listening <-chan interface{}) { | |||||||
|  |  | ||||||
| type heartbeatOp struct { | type heartbeatOp struct { | ||||||
| 	Op   int   `json:"op"` | 	Op   int   `json:"op"` | ||||||
| 	Data int `json:"d"` | 	Data int64 `json:"d"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type helloOp struct { | ||||||
|  | 	HeartbeatInterval time.Duration `json:"heartbeat_interval"` | ||||||
|  | 	Trace             []string      `json:"_trace"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FailedHeartbeatAcks is the Number of heartbeat intervals to wait until forcing a connection restart. | ||||||
|  | const FailedHeartbeatAcks time.Duration = 5 * time.Millisecond | ||||||
|  |  | ||||||
| // heartbeat sends regular heartbeats to Discord so it knows the client | // heartbeat sends regular heartbeats to Discord so it knows the client | ||||||
| // is still connected.  If you do not send these heartbeats Discord will | // is still connected.  If you do not send these heartbeats Discord will | ||||||
| // disconnect the websocket connection after a few seconds. | // disconnect the websocket connection after a few seconds. | ||||||
| func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}, i time.Duration) { | func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}, heartbeatIntervalMsec time.Duration) { | ||||||
|  |  | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
| @@ -192,19 +275,26 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{} | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var err error | 	var err error | ||||||
| 	ticker := time.NewTicker(i * time.Millisecond) | 	ticker := time.NewTicker(heartbeatIntervalMsec * time.Millisecond) | ||||||
|  | 	defer ticker.Stop() | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
|  | 		s.RLock() | ||||||
| 		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", s.sequence) | 		last := s.LastHeartbeatAck | ||||||
|  | 		s.RUnlock() | ||||||
|  | 		sequence := atomic.LoadInt64(s.sequence) | ||||||
|  | 		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", sequence) | ||||||
| 		s.wsMutex.Lock() | 		s.wsMutex.Lock() | ||||||
| 		err = wsConn.WriteJSON(heartbeatOp{1, s.sequence}) | 		err = wsConn.WriteJSON(heartbeatOp{1, sequence}) | ||||||
| 		s.wsMutex.Unlock() | 		s.wsMutex.Unlock() | ||||||
|  | 		if err != nil || time.Now().UTC().Sub(last) > (heartbeatIntervalMsec*FailedHeartbeatAcks) { | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err) | 				s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err) | ||||||
| 			s.Lock() | 			} else { | ||||||
| 			s.DataReady = false | 				s.log(LogError, "haven't gotten a heartbeat ACK in %v, triggering a reconnection", time.Now().UTC().Sub(last)) | ||||||
| 			s.Unlock() | 			} | ||||||
|  | 			s.Close() | ||||||
|  | 			s.reconnect() | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		s.Lock() | 		s.Lock() | ||||||
| @@ -220,14 +310,17 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{} | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| type updateStatusData struct { | // UpdateStatusData ia provided to UpdateStatusComplex() | ||||||
| 	IdleSince *int  `json:"idle_since"` | type UpdateStatusData struct { | ||||||
|  | 	IdleSince *int   `json:"since"` | ||||||
| 	Game      *Game  `json:"game"` | 	Game      *Game  `json:"game"` | ||||||
|  | 	AFK       bool   `json:"afk"` | ||||||
|  | 	Status    string `json:"status"` | ||||||
| } | } | ||||||
|  |  | ||||||
| type updateStatusOp struct { | type updateStatusOp struct { | ||||||
| 	Op   int              `json:"op"` | 	Op   int              `json:"op"` | ||||||
| 	Data updateStatusData `json:"d"` | 	Data UpdateStatusData `json:"d"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // UpdateStreamingStatus is used to update the user's streaming status. | // UpdateStreamingStatus is used to update the user's streaming status. | ||||||
| @@ -239,21 +332,18 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err | |||||||
|  |  | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
| 	s.RLock() | 	usd := UpdateStatusData{ | ||||||
| 	defer s.RUnlock() | 		Status: "online", | ||||||
| 	if s.wsConn == nil { |  | ||||||
| 		return errors.New("no websocket connection exists") |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var usd updateStatusData |  | ||||||
| 	if idle > 0 { | 	if idle > 0 { | ||||||
| 		usd.IdleSince = &idle | 		usd.IdleSince = &idle | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if game != "" { | 	if game != "" { | ||||||
| 		gameType := 0 | 		gameType := GameTypeGame | ||||||
| 		if url != "" { | 		if url != "" { | ||||||
| 			gameType = 1 | 			gameType = GameTypeStreaming | ||||||
| 		} | 		} | ||||||
| 		usd.Game = &Game{ | 		usd.Game = &Game{ | ||||||
| 			Name: game, | 			Name: game, | ||||||
| @@ -262,6 +352,18 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	return s.UpdateStatusComplex(usd) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UpdateStatusComplex allows for sending the raw status update data untouched by discordgo. | ||||||
|  | func (s *Session) UpdateStatusComplex(usd UpdateStatusData) (err error) { | ||||||
|  |  | ||||||
|  | 	s.RLock() | ||||||
|  | 	defer s.RUnlock() | ||||||
|  | 	if s.wsConn == nil { | ||||||
|  | 		return ErrWSNotFound | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	s.wsMutex.Lock() | 	s.wsMutex.Lock() | ||||||
| 	err = s.wsConn.WriteJSON(updateStatusOp{3, usd}) | 	err = s.wsConn.WriteJSON(updateStatusOp{3, usd}) | ||||||
| 	s.wsMutex.Unlock() | 	s.wsMutex.Unlock() | ||||||
| @@ -299,7 +401,7 @@ func (s *Session) RequestGuildMembers(guildID, query string, limit int) (err err | |||||||
| 	s.RLock() | 	s.RLock() | ||||||
| 	defer s.RUnlock() | 	defer s.RUnlock() | ||||||
| 	if s.wsConn == nil { | 	if s.wsConn == nil { | ||||||
| 		return errors.New("no websocket connection exists") | 		return ErrWSNotFound | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	data := requestGuildMembersData{ | 	data := requestGuildMembersData{ | ||||||
| @@ -323,9 +425,7 @@ func (s *Session) RequestGuildMembers(guildID, query string, limit int) (err err | |||||||
| // | // | ||||||
| // If you use the AddHandler() function to register a handler for the | // If you use the AddHandler() function to register a handler for the | ||||||
| // "OnEvent" event then all events will be passed to that handler. | // "OnEvent" event then all events will be passed to that handler. | ||||||
| // | func (s *Session) onEvent(messageType int, message []byte) (*Event, error) { | ||||||
| // TODO: You may also register a custom event handler entirely using... |  | ||||||
| func (s *Session) onEvent(messageType int, message []byte) { |  | ||||||
|  |  | ||||||
| 	var err error | 	var err error | ||||||
| 	var reader io.Reader | 	var reader io.Reader | ||||||
| @@ -337,7 +437,7 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 		z, err2 := zlib.NewReader(reader) | 		z, err2 := zlib.NewReader(reader) | ||||||
| 		if err2 != nil { | 		if err2 != nil { | ||||||
| 			s.log(LogError, "error uncompressing websocket message, %s", err) | 			s.log(LogError, "error uncompressing websocket message, %s", err) | ||||||
| 			return | 			return nil, err2 | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		defer func() { | 		defer func() { | ||||||
| @@ -355,7 +455,7 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 	decoder := json.NewDecoder(reader) | 	decoder := json.NewDecoder(reader) | ||||||
| 	if err = decoder.Decode(&e); err != nil { | 	if err = decoder.Decode(&e); err != nil { | ||||||
| 		s.log(LogError, "error decoding websocket message, %s", err) | 		s.log(LogError, "error decoding websocket message, %s", err) | ||||||
| 		return | 		return e, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	s.log(LogDebug, "Op: %d, Seq: %d, Type: %s, Data: %s\n\n", e.Operation, e.Sequence, e.Type, string(e.RawData)) | 	s.log(LogDebug, "Op: %d, Seq: %d, Type: %s, Data: %s\n\n", e.Operation, e.Sequence, e.Type, string(e.RawData)) | ||||||
| @@ -365,20 +465,23 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 	if e.Operation == 1 { | 	if e.Operation == 1 { | ||||||
| 		s.log(LogInformational, "sending heartbeat in response to Op1") | 		s.log(LogInformational, "sending heartbeat in response to Op1") | ||||||
| 		s.wsMutex.Lock() | 		s.wsMutex.Lock() | ||||||
| 		err = s.wsConn.WriteJSON(heartbeatOp{1, s.sequence}) | 		err = s.wsConn.WriteJSON(heartbeatOp{1, atomic.LoadInt64(s.sequence)}) | ||||||
| 		s.wsMutex.Unlock() | 		s.wsMutex.Unlock() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			s.log(LogError, "error sending heartbeat in response to Op1") | 			s.log(LogError, "error sending heartbeat in response to Op1") | ||||||
| 			return | 			return e, err | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		return | 		return e, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Reconnect | 	// Reconnect | ||||||
| 	// Must immediately disconnect from gateway and reconnect to new gateway. | 	// Must immediately disconnect from gateway and reconnect to new gateway. | ||||||
| 	if e.Operation == 7 { | 	if e.Operation == 7 { | ||||||
| 		// TODO | 		s.log(LogInformational, "Closing and reconnecting in response to Op7") | ||||||
|  | 		s.Close() | ||||||
|  | 		s.reconnect() | ||||||
|  | 		return e, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Invalid Session | 	// Invalid Session | ||||||
| @@ -390,10 +493,23 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 		err = s.identify() | 		err = s.identify() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err) | 			s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err) | ||||||
| 			return | 			return e, err | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		return | 		return e, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Operation == 10 { | ||||||
|  | 		// Op10 is handled by Open() | ||||||
|  | 		return e, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if e.Operation == 11 { | ||||||
|  | 		s.Lock() | ||||||
|  | 		s.LastHeartbeatAck = time.Now().UTC() | ||||||
|  | 		s.Unlock() | ||||||
|  | 		s.log(LogInformational, "got heartbeat ACK") | ||||||
|  | 		return e, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Do not try to Dispatch a non-Dispatch Message | 	// Do not try to Dispatch a non-Dispatch Message | ||||||
| @@ -401,11 +517,11 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
| 		// But we probably should be doing something with them. | 		// But we probably should be doing something with them. | ||||||
| 		// TEMP | 		// TEMP | ||||||
| 		s.log(LogWarning, "unknown Op: %d, Seq: %d, Type: %s, Data: %s, message: %s", e.Operation, e.Sequence, e.Type, string(e.RawData), string(message)) | 		s.log(LogWarning, "unknown Op: %d, Seq: %d, Type: %s, Data: %s, message: %s", e.Operation, e.Sequence, e.Type, string(e.RawData), string(message)) | ||||||
| 		return | 		return e, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Store the message sequence | 	// Store the message sequence | ||||||
| 	s.sequence = e.Sequence | 	atomic.StoreInt64(s.sequence, e.Sequence) | ||||||
|  |  | ||||||
| 	// Map event to registered event handlers and pass it along to any registered handlers. | 	// Map event to registered event handlers and pass it along to any registered handlers. | ||||||
| 	if eh, ok := registeredInterfaceProviders[e.Type]; ok { | 	if eh, ok := registeredInterfaceProviders[e.Type]; ok { | ||||||
| @@ -430,6 +546,8 @@ func (s *Session) onEvent(messageType int, message []byte) { | |||||||
|  |  | ||||||
| 	// For legacy reasons, we send the raw event also, this could be useful for handling unknown events. | 	// For legacy reasons, we send the raw event also, this could be useful for handling unknown events. | ||||||
| 	s.handleEvent(eventEventType, e) | 	s.handleEvent(eventEventType, e) | ||||||
|  |  | ||||||
|  | 	return e, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // ------------------------------------------------------------------------------------------------ | // ------------------------------------------------------------------------------------------------ | ||||||
| @@ -458,18 +576,24 @@ func (s *Session) ChannelVoiceJoin(gID, cID string, mute, deaf bool) (voice *Voi | |||||||
|  |  | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
|  | 	s.RLock() | ||||||
| 	voice, _ = s.VoiceConnections[gID] | 	voice, _ = s.VoiceConnections[gID] | ||||||
|  | 	s.RUnlock() | ||||||
|  |  | ||||||
| 	if voice == nil { | 	if voice == nil { | ||||||
| 		voice = &VoiceConnection{} | 		voice = &VoiceConnection{} | ||||||
|  | 		s.Lock() | ||||||
| 		s.VoiceConnections[gID] = voice | 		s.VoiceConnections[gID] = voice | ||||||
|  | 		s.Unlock() | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	voice.Lock() | ||||||
| 	voice.GuildID = gID | 	voice.GuildID = gID | ||||||
| 	voice.ChannelID = cID | 	voice.ChannelID = cID | ||||||
| 	voice.deaf = deaf | 	voice.deaf = deaf | ||||||
| 	voice.mute = mute | 	voice.mute = mute | ||||||
| 	voice.session = s | 	voice.session = s | ||||||
|  | 	voice.Unlock() | ||||||
|  |  | ||||||
| 	// Send the request to Discord that we want to join the voice channel | 	// Send the request to Discord that we want to join the voice channel | ||||||
| 	data := voiceChannelJoinOp{4, voiceChannelJoinData{&gID, &cID, mute, deaf}} | 	data := voiceChannelJoinOp{4, voiceChannelJoinData{&gID, &cID, mute, deaf}} | ||||||
| @@ -500,7 +624,9 @@ func (s *Session) onVoiceStateUpdate(st *VoiceStateUpdate) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Check if we have a voice connection to update | 	// Check if we have a voice connection to update | ||||||
|  | 	s.RLock() | ||||||
| 	voice, exists := s.VoiceConnections[st.GuildID] | 	voice, exists := s.VoiceConnections[st.GuildID] | ||||||
|  | 	s.RUnlock() | ||||||
| 	if !exists { | 	if !exists { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| @@ -511,8 +637,11 @@ func (s *Session) onVoiceStateUpdate(st *VoiceStateUpdate) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Store the SessionID for later use. | 	// Store the SessionID for later use. | ||||||
|  | 	voice.Lock() | ||||||
| 	voice.UserID = st.UserID | 	voice.UserID = st.UserID | ||||||
| 	voice.sessionID = st.SessionID | 	voice.sessionID = st.SessionID | ||||||
|  | 	voice.ChannelID = st.ChannelID | ||||||
|  | 	voice.Unlock() | ||||||
| } | } | ||||||
|  |  | ||||||
| // onVoiceServerUpdate handles the Voice Server Update data websocket event. | // onVoiceServerUpdate handles the Voice Server Update data websocket event. | ||||||
| @@ -524,7 +653,9 @@ func (s *Session) onVoiceServerUpdate(st *VoiceServerUpdate) { | |||||||
|  |  | ||||||
| 	s.log(LogInformational, "called") | 	s.log(LogInformational, "called") | ||||||
|  |  | ||||||
|  | 	s.RLock() | ||||||
| 	voice, exists := s.VoiceConnections[st.GuildID] | 	voice, exists := s.VoiceConnections[st.GuildID] | ||||||
|  | 	s.RUnlock() | ||||||
|  |  | ||||||
| 	// If no VoiceConnection exists, just skip this | 	// If no VoiceConnection exists, just skip this | ||||||
| 	if !exists { | 	if !exists { | ||||||
| @@ -536,11 +667,13 @@ func (s *Session) onVoiceServerUpdate(st *VoiceServerUpdate) { | |||||||
| 	voice.Close() | 	voice.Close() | ||||||
|  |  | ||||||
| 	// Store values for later use | 	// Store values for later use | ||||||
|  | 	voice.Lock() | ||||||
| 	voice.token = st.Token | 	voice.token = st.Token | ||||||
| 	voice.endpoint = st.Endpoint | 	voice.endpoint = st.Endpoint | ||||||
| 	voice.GuildID = st.GuildID | 	voice.GuildID = st.GuildID | ||||||
|  | 	voice.Unlock() | ||||||
|  |  | ||||||
| 	// Open a conenction to the voice server | 	// Open a connection to the voice server | ||||||
| 	err := voice.open() | 	err := voice.open() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		s.log(LogError, "onVoiceServerUpdate voice.open, %s", err) | 		s.log(LogError, "onVoiceServerUpdate voice.open, %s", err) | ||||||
| @@ -588,7 +721,7 @@ func (s *Session) identify() error { | |||||||
| 	if s.ShardCount > 1 { | 	if s.ShardCount > 1 { | ||||||
|  |  | ||||||
| 		if s.ShardID >= s.ShardCount { | 		if s.ShardID >= s.ShardCount { | ||||||
| 			return errors.New("ShardID must be less than ShardCount") | 			return ErrWSShardBounds | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		data.Shard = &[2]int{s.ShardID, s.ShardCount} | 		data.Shard = &[2]int{s.ShardID, s.ShardCount} | ||||||
| @@ -628,6 +761,8 @@ func (s *Session) reconnect() { | |||||||
| 				// However, there seems to be cases where something "weird" | 				// However, there seems to be cases where something "weird" | ||||||
| 				// happens.  So we're doing this for now just to improve | 				// happens.  So we're doing this for now just to improve | ||||||
| 				// stability in those edge cases. | 				// stability in those edge cases. | ||||||
|  | 				s.RLock() | ||||||
|  | 				defer s.RUnlock() | ||||||
| 				for _, v := range s.VoiceConnections { | 				for _, v := range s.VoiceConnections { | ||||||
|  |  | ||||||
| 					s.log(LogInformational, "reconnecting voice connection to guild %s", v.GuildID) | 					s.log(LogInformational, "reconnecting voice connection to guild %s", v.GuildID) | ||||||
| @@ -641,6 +776,13 @@ func (s *Session) reconnect() { | |||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
|  | 			// Certain race conditions can call reconnect() twice. If this happens, we | ||||||
|  | 			// just break out of the reconnect loop | ||||||
|  | 			if err == ErrWSAlreadyOpen { | ||||||
|  | 				s.log(LogInformational, "Websocket already exists, no need to reconnect") | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  |  | ||||||
| 			s.log(LogError, "error reconnecting to gateway, %s", err) | 			s.log(LogError, "error reconnecting to gateway, %s", err) | ||||||
|  |  | ||||||
| 			<-time.After(wait * time.Second) | 			<-time.After(wait * time.Second) | ||||||
| @@ -675,7 +817,9 @@ func (s *Session) Close() (err error) { | |||||||
| 		s.log(LogInformational, "sending close frame") | 		s.log(LogInformational, "sending close frame") | ||||||
| 		// To cleanly close a connection, a client should send a close | 		// To cleanly close a connection, a client should send a close | ||||||
| 		// frame and wait for the server to close the connection. | 		// frame and wait for the server to close the connection. | ||||||
|  | 		s.wsMutex.Lock() | ||||||
| 		err := s.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) | 		err := s.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) | ||||||
|  | 		s.wsMutex.Unlock() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			s.log(LogInformational, "error closing websocket, %s", err) | 			s.log(LogInformational, "error closing websocket, %s", err) | ||||||
| 		} | 		} | ||||||
|   | |||||||
							
								
								
									
										197
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										197
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -191,7 +191,11 @@ func (bot *BotAPI) UploadFile(endpoint string, params map[string]string, fieldna | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var apiResp APIResponse | 	var apiResp APIResponse | ||||||
| 	json.Unmarshal(bytes, &apiResp) |  | ||||||
|  | 	err = json.Unmarshal(bytes, &apiResp) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if !apiResp.Ok { | 	if !apiResp.Ok { | ||||||
| 		return APIResponse{}, errors.New(apiResp.Description) | 		return APIResponse{}, errors.New(apiResp.Description) | ||||||
| @@ -438,14 +442,7 @@ func (bot *BotAPI) SetWebhook(config WebhookConfig) (APIResponse, error) { | |||||||
| 		return APIResponse{}, err | 		return APIResponse{}, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var apiResp APIResponse | 	return resp, nil | ||||||
| 	json.Unmarshal(resp.Result, &apiResp) |  | ||||||
|  |  | ||||||
| 	if bot.Debug { |  | ||||||
| 		log.Printf("setWebhook resp: %+v\n", apiResp) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return apiResp, nil |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // GetWebhookInfo allows you to fetch information about a webhook and if | // GetWebhookInfo allows you to fetch information about a webhook and if | ||||||
| @@ -550,7 +547,7 @@ func (bot *BotAPI) AnswerCallbackQuery(config CallbackConfig) (APIResponse, erro | |||||||
| // KickChatMember kicks a user from a chat. Note that this only will work | // KickChatMember kicks a user from a chat. Note that this only will work | ||||||
| // in supergroups, and requires the bot to be an admin. Also note they | // in supergroups, and requires the bot to be an admin. Also note they | ||||||
| // will be unable to rejoin until they are unbanned. | // will be unable to rejoin until they are unbanned. | ||||||
| func (bot *BotAPI) KickChatMember(config ChatMemberConfig) (APIResponse, error) { | func (bot *BotAPI) KickChatMember(config KickChatMemberConfig) (APIResponse, error) { | ||||||
| 	v := url.Values{} | 	v := url.Values{} | ||||||
|  |  | ||||||
| 	if config.SuperGroupUsername == "" { | 	if config.SuperGroupUsername == "" { | ||||||
| @@ -560,6 +557,10 @@ func (bot *BotAPI) KickChatMember(config ChatMemberConfig) (APIResponse, error) | |||||||
| 	} | 	} | ||||||
| 	v.Add("user_id", strconv.Itoa(config.UserID)) | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if config.UntilDate != 0 { | ||||||
|  | 		v.Add("until_date", strconv.FormatInt(config.UntilDate, 10)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	bot.debugLog("kickChatMember", v, nil) | 	bot.debugLog("kickChatMember", v, nil) | ||||||
|  |  | ||||||
| 	return bot.MakeRequest("kickChatMember", v) | 	return bot.MakeRequest("kickChatMember", v) | ||||||
| @@ -677,14 +678,16 @@ func (bot *BotAPI) GetChatMember(config ChatConfigWithUser) (ChatMember, error) | |||||||
| } | } | ||||||
|  |  | ||||||
| // UnbanChatMember unbans a user from a chat. Note that this only will work | // UnbanChatMember unbans a user from a chat. Note that this only will work | ||||||
| // in supergroups, and requires the bot to be an admin. | // in supergroups and channels, and requires the bot to be an admin. | ||||||
| func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) { | func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) { | ||||||
| 	v := url.Values{} | 	v := url.Values{} | ||||||
|  |  | ||||||
| 	if config.SuperGroupUsername == "" { | 	if config.SuperGroupUsername != "" { | ||||||
| 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) |  | ||||||
| 	} else { |  | ||||||
| 		v.Add("chat_id", config.SuperGroupUsername) | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
| 	} | 	} | ||||||
| 	v.Add("user_id", strconv.Itoa(config.UserID)) | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
| @@ -693,6 +696,82 @@ func (bot *BotAPI) UnbanChatMember(config ChatMemberConfig) (APIResponse, error) | |||||||
| 	return bot.MakeRequest("unbanChatMember", v) | 	return bot.MakeRequest("unbanChatMember", v) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // RestrictChatMember to restrict a user in a supergroup. The bot must be an | ||||||
|  | //administrator in the supergroup for this to work and must have the | ||||||
|  | //appropriate admin rights. Pass True for all boolean parameters to lift | ||||||
|  | //restrictions from a user. Returns True on success. | ||||||
|  | func (bot *BotAPI) RestrictChatMember(config RestrictChatMemberConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} | ||||||
|  | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if &config.CanSendMessages != nil { | ||||||
|  | 		v.Add("can_send_messages", strconv.FormatBool(*config.CanSendMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanSendMediaMessages != nil { | ||||||
|  | 		v.Add("can_send_media_messages", strconv.FormatBool(*config.CanSendMediaMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanSendOtherMessages != nil { | ||||||
|  | 		v.Add("can_send_other_messages", strconv.FormatBool(*config.CanSendOtherMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanAddWebPagePreviews != nil { | ||||||
|  | 		v.Add("can_add_web_page_previews", strconv.FormatBool(*config.CanAddWebPagePreviews)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("restrictChatMember", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("restrictChatMember", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (bot *BotAPI) PromoteChatMember(config PromoteChatMemberConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} else if config.ChannelUsername != "" { | ||||||
|  | 		v.Add("chat_id", config.ChannelUsername) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} | ||||||
|  | 	v.Add("user_id", strconv.Itoa(config.UserID)) | ||||||
|  |  | ||||||
|  | 	if &config.CanChangeInfo != nil { | ||||||
|  | 		v.Add("can_change_info", strconv.FormatBool(*config.CanChangeInfo)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPostMessages != nil { | ||||||
|  | 		v.Add("can_post_messages", strconv.FormatBool(*config.CanPostMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanEditMessages != nil { | ||||||
|  | 		v.Add("can_edit_messages", strconv.FormatBool(*config.CanEditMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanDeleteMessages != nil { | ||||||
|  | 		v.Add("can_delete_messages", strconv.FormatBool(*config.CanDeleteMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanInviteUsers != nil { | ||||||
|  | 		v.Add("can_invite_users", strconv.FormatBool(*config.CanInviteUsers)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanRestrictMembers != nil { | ||||||
|  | 		v.Add("can_restrict_members", strconv.FormatBool(*config.CanRestrictMembers)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPinMessages != nil { | ||||||
|  | 		v.Add("can_pin_messages", strconv.FormatBool(*config.CanPinMessages)) | ||||||
|  | 	} | ||||||
|  | 	if &config.CanPromoteMembers != nil { | ||||||
|  | 		v.Add("can_promote_members", strconv.FormatBool(*config.CanPromoteMembers)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("promoteChatMember", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("promoteChatMember", v) | ||||||
|  | } | ||||||
|  |  | ||||||
| // GetGameHighScores allows you to get the high scores for a game. | // GetGameHighScores allows you to get the high scores for a game. | ||||||
| func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHighScore, error) { | func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHighScore, error) { | ||||||
| 	v, _ := config.values() | 	v, _ := config.values() | ||||||
| @@ -707,3 +786,93 @@ func (bot *BotAPI) GetGameHighScores(config GetGameHighScoresConfig) ([]GameHigh | |||||||
|  |  | ||||||
| 	return highScores, err | 	return highScores, err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // AnswerShippingQuery allows you to reply to Update with shipping_query parameter. | ||||||
|  | func (bot *BotAPI) AnswerShippingQuery(config ShippingConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("shipping_query_id", config.ShippingQueryID) | ||||||
|  | 	v.Add("ok", strconv.FormatBool(config.OK)) | ||||||
|  | 	if config.OK == true { | ||||||
|  | 		data, err := json.Marshal(config.ShippingOptions) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return APIResponse{}, err | ||||||
|  | 		} | ||||||
|  | 		v.Add("shipping_options", string(data)) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("error_message", config.ErrorMessage) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("answerShippingQuery", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("answerShippingQuery", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AnswerPreCheckoutQuery allows you to reply to Update with pre_checkout_query. | ||||||
|  | func (bot *BotAPI) AnswerPreCheckoutQuery(config PreCheckoutConfig) (APIResponse, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("pre_checkout_query_id", config.PreCheckoutQueryID) | ||||||
|  | 	v.Add("ok", strconv.FormatBool(config.OK)) | ||||||
|  | 	if config.OK != true { | ||||||
|  | 		v.Add("error", config.ErrorMessage) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog("answerPreCheckoutQuery", v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest("answerPreCheckoutQuery", v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DeleteMessage deletes a message in a chat | ||||||
|  | func (bot *BotAPI) DeleteMessage(config DeleteMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetInviteLink get InviteLink for a chat | ||||||
|  | func (bot *BotAPI) GetInviteLink(config ChatConfig) (string, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	if config.SuperGroupUsername == "" { | ||||||
|  | 		v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	} else { | ||||||
|  | 		v.Add("chat_id", config.SuperGroupUsername) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	resp, err := bot.MakeRequest("exportChatInviteLink", v) | ||||||
|  |  | ||||||
|  | 	var inviteLink string | ||||||
|  | 	err = json.Unmarshal(resp.Result, &inviteLink) | ||||||
|  |  | ||||||
|  | 	return inviteLink, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pin message in supergroup | ||||||
|  | func (bot *BotAPI) PinChatMessage(config PinChatMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Unpin message in supergroup | ||||||
|  | func (bot *BotAPI) UnpinChatMessage(config UnpinChatMessageConfig) (APIResponse, error) { | ||||||
|  | 	v, err := config.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return APIResponse{}, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bot.debugLog(config.method(), v, nil) | ||||||
|  |  | ||||||
|  | 	return bot.MakeRequest(config.method(), v) | ||||||
|  | } | ||||||
							
								
								
									
										243
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										243
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -349,6 +349,7 @@ func (config AudioConfig) method() string { | |||||||
| // DocumentConfig contains information about a SendDocument request. | // DocumentConfig contains information about a SendDocument request. | ||||||
| type DocumentConfig struct { | type DocumentConfig struct { | ||||||
| 	BaseFile | 	BaseFile | ||||||
|  | 	Caption string | ||||||
| } | } | ||||||
|  |  | ||||||
| // values returns a url.Values representation of DocumentConfig. | // values returns a url.Values representation of DocumentConfig. | ||||||
| @@ -359,6 +360,9 @@ func (config DocumentConfig) values() (url.Values, error) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	v.Add(config.name(), config.FileID) | 	v.Add(config.name(), config.FileID) | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		v.Add("caption", config.Caption) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return v, nil | 	return v, nil | ||||||
| } | } | ||||||
| @@ -367,6 +371,10 @@ func (config DocumentConfig) values() (url.Values, error) { | |||||||
| func (config DocumentConfig) params() (map[string]string, error) { | func (config DocumentConfig) params() (map[string]string, error) { | ||||||
| 	params, _ := config.BaseFile.params() | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -443,6 +451,10 @@ func (config VideoConfig) values() (url.Values, error) { | |||||||
| func (config VideoConfig) params() (map[string]string, error) { | func (config VideoConfig) params() (map[string]string, error) { | ||||||
| 	params, _ := config.BaseFile.params() | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -456,6 +468,57 @@ func (config VideoConfig) method() string { | |||||||
| 	return "sendVideo" | 	return "sendVideo" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // VideoNoteConfig contains information about a SendVideoNote request. | ||||||
|  | type VideoNoteConfig struct { | ||||||
|  | 	BaseFile | ||||||
|  | 	Duration int | ||||||
|  | 	Length   int | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // values returns a url.Values representation of VideoNoteConfig. | ||||||
|  | func (config VideoNoteConfig) values() (url.Values, error) { | ||||||
|  | 	v, err := config.BaseChat.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	v.Add(config.name(), config.FileID) | ||||||
|  | 	if config.Duration != 0 { | ||||||
|  | 		v.Add("duration", strconv.Itoa(config.Duration)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Telegram API seems to have a bug, if no length is provided or it is 0, it will send an error response | ||||||
|  | 	if config.Length != 0 { | ||||||
|  | 		v.Add("length", strconv.Itoa(config.Length)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // params returns a map[string]string representation of VideoNoteConfig. | ||||||
|  | func (config VideoNoteConfig) params() (map[string]string, error) { | ||||||
|  | 	params, _ := config.BaseFile.params() | ||||||
|  |  | ||||||
|  | 	if config.Length != 0 { | ||||||
|  | 		params["length"] = strconv.Itoa(config.Length) | ||||||
|  | 	} | ||||||
|  | 	if config.Duration != 0 { | ||||||
|  | 		params["duration"] = strconv.Itoa(config.Duration) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return params, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // name returns the field name for the VideoNote. | ||||||
|  | func (config VideoNoteConfig) name() string { | ||||||
|  | 	return "video_note" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // method returns Telegram API method name for sending VideoNote. | ||||||
|  | func (config VideoNoteConfig) method() string { | ||||||
|  | 	return "sendVideoNote" | ||||||
|  | } | ||||||
|  |  | ||||||
| // VoiceConfig contains information about a SendVoice request. | // VoiceConfig contains information about a SendVoice request. | ||||||
| type VoiceConfig struct { | type VoiceConfig struct { | ||||||
| 	BaseFile | 	BaseFile | ||||||
| @@ -474,6 +537,9 @@ func (config VoiceConfig) values() (url.Values, error) { | |||||||
| 	if config.Duration != 0 { | 	if config.Duration != 0 { | ||||||
| 		v.Add("duration", strconv.Itoa(config.Duration)) | 		v.Add("duration", strconv.Itoa(config.Duration)) | ||||||
| 	} | 	} | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		v.Add("caption", config.Caption) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return v, nil | 	return v, nil | ||||||
| } | } | ||||||
| @@ -485,6 +551,9 @@ func (config VoiceConfig) params() (map[string]string, error) { | |||||||
| 	if config.Duration != 0 { | 	if config.Duration != 0 { | ||||||
| 		params["duration"] = strconv.Itoa(config.Duration) | 		params["duration"] = strconv.Itoa(config.Duration) | ||||||
| 	} | 	} | ||||||
|  | 	if config.Caption != "" { | ||||||
|  | 		params["caption"] = config.Caption | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return params, nil | 	return params, nil | ||||||
| } | } | ||||||
| @@ -814,9 +883,39 @@ type CallbackConfig struct { | |||||||
| type ChatMemberConfig struct { | type ChatMemberConfig struct { | ||||||
| 	ChatID             int64 | 	ChatID             int64 | ||||||
| 	SuperGroupUsername string | 	SuperGroupUsername string | ||||||
|  | 	ChannelUsername    string | ||||||
| 	UserID             int | 	UserID             int | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // KickChatMemberConfig contains extra fields to kick user | ||||||
|  | type KickChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	UntilDate int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RestrictChatMemberConfig contains fields to restrict members of chat | ||||||
|  | type RestrictChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	UntilDate             int64 | ||||||
|  | 	CanSendMessages       *bool | ||||||
|  | 	CanSendMediaMessages  *bool | ||||||
|  | 	CanSendOtherMessages  *bool | ||||||
|  | 	CanAddWebPagePreviews *bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PromoteChatMemberConfig contains fields to promote members of chat | ||||||
|  | type PromoteChatMemberConfig struct { | ||||||
|  | 	ChatMemberConfig | ||||||
|  | 	CanChangeInfo      *bool | ||||||
|  | 	CanPostMessages    *bool | ||||||
|  | 	CanEditMessages    *bool | ||||||
|  | 	CanDeleteMessages  *bool | ||||||
|  | 	CanInviteUsers     *bool | ||||||
|  | 	CanRestrictMembers *bool | ||||||
|  | 	CanPinMessages     *bool | ||||||
|  | 	CanPromoteMembers  *bool | ||||||
|  | } | ||||||
|  |  | ||||||
| // ChatConfig contains information about getting information on a chat. | // ChatConfig contains information about getting information on a chat. | ||||||
| type ChatConfig struct { | type ChatConfig struct { | ||||||
| 	ChatID             int64 | 	ChatID             int64 | ||||||
| @@ -830,3 +929,147 @@ type ChatConfigWithUser struct { | |||||||
| 	SuperGroupUsername string | 	SuperGroupUsername string | ||||||
| 	UserID             int | 	UserID             int | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // InvoiceConfig contains information for sendInvoice request. | ||||||
|  | type InvoiceConfig struct { | ||||||
|  | 	BaseChat | ||||||
|  | 	Title               string          // required | ||||||
|  | 	Description         string          // required | ||||||
|  | 	Payload             string          // required | ||||||
|  | 	ProviderToken       string          // required | ||||||
|  | 	StartParameter      string          // required | ||||||
|  | 	Currency            string          // required | ||||||
|  | 	Prices              *[]LabeledPrice // required | ||||||
|  | 	PhotoURL            string | ||||||
|  | 	PhotoSize           int | ||||||
|  | 	PhotoWidth          int | ||||||
|  | 	PhotoHeight         int | ||||||
|  | 	NeedName            bool | ||||||
|  | 	NeedPhoneNumber     bool | ||||||
|  | 	NeedEmail           bool | ||||||
|  | 	NeedShippingAddress bool | ||||||
|  | 	IsFlexible          bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config InvoiceConfig) values() (url.Values, error) { | ||||||
|  | 	v, err := config.BaseChat.values() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  | 	v.Add("title", config.Title) | ||||||
|  | 	v.Add("description", config.Description) | ||||||
|  | 	v.Add("payload", config.Payload) | ||||||
|  | 	v.Add("provider_token", config.ProviderToken) | ||||||
|  | 	v.Add("start_parameter", config.StartParameter) | ||||||
|  | 	v.Add("currency", config.Currency) | ||||||
|  | 	data, err := json.Marshal(config.Prices) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return v, err | ||||||
|  | 	} | ||||||
|  | 	v.Add("prices", string(data)) | ||||||
|  | 	if config.PhotoURL != "" { | ||||||
|  | 		v.Add("photo_url", config.PhotoURL) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoSize != 0 { | ||||||
|  | 		v.Add("photo_size", strconv.Itoa(config.PhotoSize)) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoWidth != 0 { | ||||||
|  | 		v.Add("photo_width", strconv.Itoa(config.PhotoWidth)) | ||||||
|  | 	} | ||||||
|  | 	if config.PhotoHeight != 0 { | ||||||
|  | 		v.Add("photo_height", strconv.Itoa(config.PhotoHeight)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedName != false { | ||||||
|  | 		v.Add("need_name", strconv.FormatBool(config.NeedName)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedPhoneNumber != false { | ||||||
|  | 		v.Add("need_phone_number", strconv.FormatBool(config.NeedPhoneNumber)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedEmail != false { | ||||||
|  | 		v.Add("need_email", strconv.FormatBool(config.NeedEmail)) | ||||||
|  | 	} | ||||||
|  | 	if config.NeedShippingAddress != false { | ||||||
|  | 		v.Add("need_shipping_address", strconv.FormatBool(config.NeedShippingAddress)) | ||||||
|  | 	} | ||||||
|  | 	if config.IsFlexible != false { | ||||||
|  | 		v.Add("is_flexible", strconv.FormatBool(config.IsFlexible)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config InvoiceConfig) method() string { | ||||||
|  | 	return "sendInvoice" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingConfig contains information for answerShippingQuery request. | ||||||
|  | type ShippingConfig struct { | ||||||
|  | 	ShippingQueryID string // required | ||||||
|  | 	OK              bool   // required | ||||||
|  | 	ShippingOptions *[]ShippingOption | ||||||
|  | 	ErrorMessage    string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PreCheckoutConfig conatins information for answerPreCheckoutQuery request. | ||||||
|  | type PreCheckoutConfig struct { | ||||||
|  | 	PreCheckoutQueryID string // required | ||||||
|  | 	OK                 bool   // required | ||||||
|  | 	ErrorMessage       string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DeleteMessageConfig contains information of a message in a chat to delete. | ||||||
|  | type DeleteMessageConfig struct { | ||||||
|  | 	ChatID    int64 | ||||||
|  | 	MessageID int | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config DeleteMessageConfig) method() string { | ||||||
|  | 	return "deleteMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config DeleteMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	v.Add("message_id", strconv.Itoa(config.MessageID)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PinChatMessageConfig contains information of a message in a chat to pin. | ||||||
|  | type PinChatMessageConfig struct { | ||||||
|  | 	ChatID int64 | ||||||
|  | 	MessageID int | ||||||
|  | 	DisableNotification bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config PinChatMessageConfig) method() string { | ||||||
|  | 	return "pinChatMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config PinChatMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  | 	v.Add("message_id", strconv.Itoa(config.MessageID)) | ||||||
|  | 	v.Add("disable_notification", strconv.FormatBool(config.DisableNotification)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnpinChatMessageConfig contains information of chat to unpin. | ||||||
|  | type UnpinChatMessageConfig struct { | ||||||
|  | 	ChatID int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config UnpinChatMessageConfig) method() string { | ||||||
|  | 	return "unpinChatMessage" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (config UnpinChatMessageConfig) values() (url.Values, error) { | ||||||
|  | 	v := url.Values{} | ||||||
|  |  | ||||||
|  | 	v.Add("chat_id", strconv.FormatInt(config.ChatID, 10)) | ||||||
|  |  | ||||||
|  | 	return v, nil | ||||||
|  | } | ||||||
							
								
								
									
										44
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										44
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -194,6 +194,37 @@ func NewVideoShare(chatID int64, fileID string) VideoConfig { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // NewVideoNoteUpload creates a new video note uploader. | ||||||
|  | // | ||||||
|  | // chatID is where to send it, file is a string path to the file, | ||||||
|  | // FileReader, or FileBytes. | ||||||
|  | func NewVideoNoteUpload(chatID int64, length int, file interface{}) VideoNoteConfig { | ||||||
|  | 	return VideoNoteConfig{ | ||||||
|  | 		BaseFile: BaseFile{ | ||||||
|  | 			BaseChat:    BaseChat{ChatID: chatID}, | ||||||
|  | 			File:        file, | ||||||
|  | 			UseExisting: false, | ||||||
|  | 		}, | ||||||
|  | 		Length: length, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewVideoNoteShare shares an existing video. | ||||||
|  | // You may use this to reshare an existing video without reuploading it. | ||||||
|  | // | ||||||
|  | // chatID is where to send it, fileID is the ID of the video | ||||||
|  | // already uploaded. | ||||||
|  | func NewVideoNoteShare(chatID int64, length int, fileID string) VideoNoteConfig { | ||||||
|  | 	return VideoNoteConfig{ | ||||||
|  | 		BaseFile: BaseFile{ | ||||||
|  | 			BaseChat:    BaseChat{ChatID: chatID}, | ||||||
|  | 			FileID:      fileID, | ||||||
|  | 			UseExisting: true, | ||||||
|  | 		}, | ||||||
|  | 		Length: length, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewVoiceUpload creates a new voice uploader. | // NewVoiceUpload creates a new voice uploader. | ||||||
| // | // | ||||||
| // chatID is where to send it, file is a string path to the file, | // chatID is where to send it, file is a string path to the file, | ||||||
| @@ -609,3 +640,16 @@ func NewCallbackWithAlert(id, text string) CallbackConfig { | |||||||
| 		ShowAlert:       true, | 		ShowAlert:       true, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // NewInvoice created a new Invoice request to the user. | ||||||
|  | func NewInvoice(chatID int64, title, description, payload, providerToken, startParameter, currency string, prices *[]LabeledPrice) InvoiceConfig { | ||||||
|  | 	return InvoiceConfig{ | ||||||
|  | 		BaseChat:       BaseChat{ChatID: chatID}, | ||||||
|  | 		Title:          title, | ||||||
|  | 		Description:    description, | ||||||
|  | 		Payload:        payload, | ||||||
|  | 		ProviderToken:  providerToken, | ||||||
|  | 		StartParameter: startParameter, | ||||||
|  | 		Currency:       currency, | ||||||
|  | 		Prices:         prices} | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										113
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										113
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -35,6 +35,8 @@ type Update struct { | |||||||
| 	InlineQuery        *InlineQuery        `json:"inline_query"` | 	InlineQuery        *InlineQuery        `json:"inline_query"` | ||||||
| 	ChosenInlineResult *ChosenInlineResult `json:"chosen_inline_result"` | 	ChosenInlineResult *ChosenInlineResult `json:"chosen_inline_result"` | ||||||
| 	CallbackQuery      *CallbackQuery      `json:"callback_query"` | 	CallbackQuery      *CallbackQuery      `json:"callback_query"` | ||||||
|  | 	ShippingQuery      *ShippingQuery      `json:"shipping_query"` | ||||||
|  | 	PreCheckoutQuery   *PreCheckoutQuery   `json:"pre_checkout_query"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // UpdatesChannel is the channel for getting updates. | // UpdatesChannel is the channel for getting updates. | ||||||
| @@ -53,6 +55,7 @@ type User struct { | |||||||
| 	FirstName    string `json:"first_name"` | 	FirstName    string `json:"first_name"` | ||||||
| 	LastName     string `json:"last_name"`     // optional | 	LastName     string `json:"last_name"`     // optional | ||||||
| 	UserName     string `json:"username"`      // optional | 	UserName     string `json:"username"`      // optional | ||||||
|  | 	LanguageCode string `json:"language_code"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // String displays a simple text version of a user. | // String displays a simple text version of a user. | ||||||
| @@ -78,6 +81,12 @@ type GroupChat struct { | |||||||
| 	Title string `json:"title"` | 	Title string `json:"title"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ChatPhoto represents a chat photo. | ||||||
|  | type ChatPhoto struct { | ||||||
|  | 	SmallFileID string `json:"small_file_id"` | ||||||
|  | 	BigFileID   string `json:"big_file_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Chat contains information about the place a message was sent. | // Chat contains information about the place a message was sent. | ||||||
| type Chat struct { | type Chat struct { | ||||||
| 	ID                  int64      `json:"id"` | 	ID                  int64      `json:"id"` | ||||||
| @@ -87,6 +96,9 @@ type Chat struct { | |||||||
| 	FirstName           string     `json:"first_name"`                     // optional | 	FirstName           string     `json:"first_name"`                     // optional | ||||||
| 	LastName            string     `json:"last_name"`                      // optional | 	LastName            string     `json:"last_name"`                      // optional | ||||||
| 	AllMembersAreAdmins bool       `json:"all_members_are_administrators"` // optional | 	AllMembersAreAdmins bool       `json:"all_members_are_administrators"` // optional | ||||||
|  | 	Photo               *ChatPhoto `json:"photo"` | ||||||
|  | 	Description         string     `json:"description,omitempty"` // optional | ||||||
|  | 	InviteLink          string     `json:"invite_link,omitempty"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // IsPrivate returns if the Chat is a private conversation. | // IsPrivate returns if the Chat is a private conversation. | ||||||
| @@ -135,12 +147,13 @@ type Message struct { | |||||||
| 	Photo                 *[]PhotoSize       `json:"photo"`                   // optional | 	Photo                 *[]PhotoSize       `json:"photo"`                   // optional | ||||||
| 	Sticker               *Sticker           `json:"sticker"`                 // optional | 	Sticker               *Sticker           `json:"sticker"`                 // optional | ||||||
| 	Video                 *Video             `json:"video"`                   // optional | 	Video                 *Video             `json:"video"`                   // optional | ||||||
|  | 	VideoNote             *VideoNote         `json:"video_note"`              // optional | ||||||
| 	Voice                 *Voice             `json:"voice"`                   // optional | 	Voice                 *Voice             `json:"voice"`                   // optional | ||||||
| 	Caption               string             `json:"caption"`                 // optional | 	Caption               string             `json:"caption"`                 // optional | ||||||
| 	Contact               *Contact           `json:"contact"`                 // optional | 	Contact               *Contact           `json:"contact"`                 // optional | ||||||
| 	Location              *Location          `json:"location"`                // optional | 	Location              *Location          `json:"location"`                // optional | ||||||
| 	Venue                 *Venue             `json:"venue"`                   // optional | 	Venue                 *Venue             `json:"venue"`                   // optional | ||||||
| 	NewChatMember         *User            `json:"new_chat_member"`         // optional | 	NewChatMembers        *[]User            `json:"new_chat_members"`        // optional | ||||||
| 	LeftChatMember        *User              `json:"left_chat_member"`        // optional | 	LeftChatMember        *User              `json:"left_chat_member"`        // optional | ||||||
| 	NewChatTitle          string             `json:"new_chat_title"`          // optional | 	NewChatTitle          string             `json:"new_chat_title"`          // optional | ||||||
| 	NewChatPhoto          *[]PhotoSize       `json:"new_chat_photo"`          // optional | 	NewChatPhoto          *[]PhotoSize       `json:"new_chat_photo"`          // optional | ||||||
| @@ -151,6 +164,8 @@ type Message struct { | |||||||
| 	MigrateToChatID       int64              `json:"migrate_to_chat_id"`      // optional | 	MigrateToChatID       int64              `json:"migrate_to_chat_id"`      // optional | ||||||
| 	MigrateFromChatID     int64              `json:"migrate_from_chat_id"`    // optional | 	MigrateFromChatID     int64              `json:"migrate_from_chat_id"`    // optional | ||||||
| 	PinnedMessage         *Message           `json:"pinned_message"`          // optional | 	PinnedMessage         *Message           `json:"pinned_message"`          // optional | ||||||
|  | 	Invoice               *Invoice           `json:"invoice"`                 // optional | ||||||
|  | 	SuccessfulPayment     *SuccessfulPayment `json:"successful_payment"`      // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // Time converts the message timestamp into a Time. | // Time converts the message timestamp into a Time. | ||||||
| @@ -263,6 +278,15 @@ type Video struct { | |||||||
| 	FileSize  int        `json:"file_size"` // optional | 	FileSize  int        `json:"file_size"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // VideoNote contains information about a video. | ||||||
|  | type VideoNote struct { | ||||||
|  | 	FileID    string     `json:"file_id"` | ||||||
|  | 	Length    int        `json:"length"` | ||||||
|  | 	Duration  int        `json:"duration"` | ||||||
|  | 	Thumbnail *PhotoSize `json:"thumb"`     // optional | ||||||
|  | 	FileSize  int        `json:"file_size"` // optional | ||||||
|  | } | ||||||
|  |  | ||||||
| // Voice contains information about a voice. | // Voice contains information about a voice. | ||||||
| type Voice struct { | type Voice struct { | ||||||
| 	FileID   string `json:"file_id"` | 	FileID   string `json:"file_id"` | ||||||
| @@ -361,6 +385,7 @@ type InlineKeyboardButton struct { | |||||||
| 	SwitchInlineQuery            *string       `json:"switch_inline_query,omitempty"`              // optional | 	SwitchInlineQuery            *string       `json:"switch_inline_query,omitempty"`              // optional | ||||||
| 	SwitchInlineQueryCurrentChat *string       `json:"switch_inline_query_current_chat,omitempty"` // optional | 	SwitchInlineQueryCurrentChat *string       `json:"switch_inline_query_current_chat,omitempty"` // optional | ||||||
| 	CallbackGame                 *CallbackGame `json:"callback_game,omitempty"`                    // optional | 	CallbackGame                 *CallbackGame `json:"callback_game,omitempty"`                    // optional | ||||||
|  | 	Pay                          bool          `json:"pay,omitempty"`			       // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // CallbackQuery is data sent when a keyboard button with callback data | // CallbackQuery is data sent when a keyboard button with callback data | ||||||
| @@ -386,6 +411,20 @@ type ForceReply struct { | |||||||
| type ChatMember struct { | type ChatMember struct { | ||||||
| 	User                  *User  `json:"user"` | 	User                  *User  `json:"user"` | ||||||
| 	Status                string `json:"status"` | 	Status                string `json:"status"` | ||||||
|  | 	UntilDate             int64  `json:"until_date,omitempty"`                // optional | ||||||
|  | 	CanBeEdited           bool   `json:"can_be_edited,omitempty"`             // optional | ||||||
|  | 	CanChangeInfo         bool   `json:"can_change_info,omitempty"`           // optional | ||||||
|  | 	CanPostMessages       bool   `json:"can_post_messages,omitempty"`         // optional | ||||||
|  | 	CanEditMessages       bool   `json:"can_edit_messages,omitempty"`         // optional | ||||||
|  | 	CanDeleteMessages     bool   `json:"can_delete_messages,omitempty"`       // optional | ||||||
|  | 	CanInviteUsers        bool   `json:"can_invite_users,omitempty"`          // optional | ||||||
|  | 	CanRestrictMembers    bool   `json:"can_restrict_members,omitempty"`      // optional | ||||||
|  | 	CanPinMessages        bool   `json:"can_pin_messages,omitempty"`          // optional | ||||||
|  | 	CanPromoteMembers     bool   `json:"can_promote_members,omitempty"`       // optional | ||||||
|  | 	CanSendMessages       bool   `json:"can_send_messages,omitempty"`         // optional | ||||||
|  | 	CanSendMediaMessages  bool   `json:"can_send_media_messages,omitempty"`   // optional | ||||||
|  | 	CanSendOtherMessages  bool   `json:"can_send_other_messages,omitempty"`   // optional | ||||||
|  | 	CanAddWebPagePreviews bool   `json:"can_add_web_page_previews,omitempty"` // optional | ||||||
| } | } | ||||||
|  |  | ||||||
| // IsCreator returns if the ChatMember was the creator of the chat. | // IsCreator returns if the ChatMember was the creator of the chat. | ||||||
| @@ -493,6 +532,7 @@ type InlineQueryResultGIF struct { | |||||||
| 	URL                 string                `json:"gif_url"` // required | 	URL                 string                `json:"gif_url"` // required | ||||||
| 	Width               int                   `json:"gif_width"` | 	Width               int                   `json:"gif_width"` | ||||||
| 	Height              int                   `json:"gif_height"` | 	Height              int                   `json:"gif_height"` | ||||||
|  | 	Duration            int                   `json:"gif_duration"` | ||||||
| 	ThumbURL            string                `json:"thumb_url"` | 	ThumbURL            string                `json:"thumb_url"` | ||||||
| 	Title               string                `json:"title"` | 	Title               string                `json:"title"` | ||||||
| 	Caption             string                `json:"caption"` | 	Caption             string                `json:"caption"` | ||||||
| @@ -507,6 +547,7 @@ type InlineQueryResultMPEG4GIF struct { | |||||||
| 	URL                 string                `json:"mpeg4_url"` // required | 	URL                 string                `json:"mpeg4_url"` // required | ||||||
| 	Width               int                   `json:"mpeg4_width"` | 	Width               int                   `json:"mpeg4_width"` | ||||||
| 	Height              int                   `json:"mpeg4_height"` | 	Height              int                   `json:"mpeg4_height"` | ||||||
|  | 	Duration            int                   `json:"mpeg4_duration"` | ||||||
| 	ThumbURL            string                `json:"thumb_url"` | 	ThumbURL            string                `json:"thumb_url"` | ||||||
| 	Title               string                `json:"title"` | 	Title               string                `json:"title"` | ||||||
| 	Caption             string                `json:"caption"` | 	Caption             string                `json:"caption"` | ||||||
| @@ -635,3 +676,73 @@ type InputContactMessageContent struct { | |||||||
| 	FirstName   string `json:"first_name"` | 	FirstName   string `json:"first_name"` | ||||||
| 	LastName    string `json:"last_name"` | 	LastName    string `json:"last_name"` | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Invoice contains basic information about an invoice. | ||||||
|  | type Invoice struct { | ||||||
|  | 	Title          string `json:"title"` | ||||||
|  | 	Description    string `json:"description"` | ||||||
|  | 	StartParameter string `json:"start_parameter"` | ||||||
|  | 	Currency       string `json:"currency"` | ||||||
|  | 	TotalAmount    int    `json:"total_amount"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // LabeledPrice represents a portion of the price for goods or services. | ||||||
|  | type LabeledPrice struct { | ||||||
|  | 	Label  string `json:"label"` | ||||||
|  | 	Amount int    `json:"amount"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingAddress represents a shipping address. | ||||||
|  | type ShippingAddress struct { | ||||||
|  | 	CountryCode string `json:"country_code"` | ||||||
|  | 	State       string `json:"state"` | ||||||
|  | 	City        string `json:"city"` | ||||||
|  | 	StreetLine1 string `json:"street_line1"` | ||||||
|  | 	StreetLine2 string `json:"street_line2"` | ||||||
|  | 	PostCode    string `json:"post_code"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // OrderInfo represents information about an order. | ||||||
|  | type OrderInfo struct { | ||||||
|  | 	Name            string           `json:"name,omitempty"` | ||||||
|  | 	PhoneNumber     string           `json:"phone_number,omitempty"` | ||||||
|  | 	Email           string           `json:"email,omitempty"` | ||||||
|  | 	ShippingAddress *ShippingAddress `json:"shipping_address,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingOption represents one shipping option. | ||||||
|  | type ShippingOption struct { | ||||||
|  | 	ID     string          `json:"id"` | ||||||
|  | 	Title  string          `json:"title"` | ||||||
|  | 	Prices *[]LabeledPrice `json:"prices"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SuccessfulPayment contains basic information about a successful payment. | ||||||
|  | type SuccessfulPayment struct { | ||||||
|  | 	Currency                string     `json:"currency"` | ||||||
|  | 	TotalAmount             int        `json:"total_amount"` | ||||||
|  | 	InvoicePayload          string     `json:"invoice_payload"` | ||||||
|  | 	ShippingOptionID        string     `json:"shipping_option_id,omitempty"` | ||||||
|  | 	OrderInfo               *OrderInfo `json:"order_info,omitempty"` | ||||||
|  | 	TelegramPaymentChargeID string     `json:"telegram_payment_charge_id"` | ||||||
|  | 	ProviderPaymentChargeID string     `json:"provider_payment_charge_id"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ShippingQuery contains information about an incoming shipping query. | ||||||
|  | type ShippingQuery struct { | ||||||
|  | 	ID              string           `json:"id"` | ||||||
|  | 	From            *User            `json:"from"` | ||||||
|  | 	InvoicePayload  string           `json:"invoice_payload"` | ||||||
|  | 	ShippingAddress *ShippingAddress `json:"shipping_address"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PreCheckoutQuery contains information about an incoming pre-checkout query. | ||||||
|  | type PreCheckoutQuery struct { | ||||||
|  | 	ID               string     `json:"id"` | ||||||
|  | 	From             *User      `json:"from"` | ||||||
|  | 	Currency         string     `json:"currency"` | ||||||
|  | 	TotalAmount      int        `json:"total_amount"` | ||||||
|  | 	InvoicePayload   string     `json:"invoice_payload"` | ||||||
|  | 	ShippingOptionID string     `json:"shipping_option_id,omitempty"` | ||||||
|  | 	OrderInfo        *OrderInfo `json:"order_info,omitempty"` | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										77
									
								
								vendor/github.com/gorilla/websocket/proxy.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										77
									
								
								vendor/github.com/gorilla/websocket/proxy.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,77 @@ | |||||||
|  | // Copyright 2017 The Gorilla WebSocket Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  |  | ||||||
|  | package websocket | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bufio" | ||||||
|  | 	"encoding/base64" | ||||||
|  | 	"errors" | ||||||
|  | 	"net" | ||||||
|  | 	"net/http" | ||||||
|  | 	"net/url" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type netDialerFunc func(netowrk, addr string) (net.Conn, error) | ||||||
|  |  | ||||||
|  | func (fn netDialerFunc) Dial(network, addr string) (net.Conn, error) { | ||||||
|  | 	return fn(network, addr) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func init() { | ||||||
|  | 	proxy_RegisterDialerType("http", func(proxyURL *url.URL, forwardDialer proxy_Dialer) (proxy_Dialer, error) { | ||||||
|  | 		return &httpProxyDialer{proxyURL: proxyURL, fowardDial: forwardDialer.Dial}, nil | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type httpProxyDialer struct { | ||||||
|  | 	proxyURL   *url.URL | ||||||
|  | 	fowardDial func(network, addr string) (net.Conn, error) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (hpd *httpProxyDialer) Dial(network string, addr string) (net.Conn, error) { | ||||||
|  | 	hostPort, _ := hostPortNoPort(hpd.proxyURL) | ||||||
|  | 	conn, err := hpd.fowardDial(network, hostPort) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	connectHeader := make(http.Header) | ||||||
|  | 	if user := hpd.proxyURL.User; user != nil { | ||||||
|  | 		proxyUser := user.Username() | ||||||
|  | 		if proxyPassword, passwordSet := user.Password(); passwordSet { | ||||||
|  | 			credential := base64.StdEncoding.EncodeToString([]byte(proxyUser + ":" + proxyPassword)) | ||||||
|  | 			connectHeader.Set("Proxy-Authorization", "Basic "+credential) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	connectReq := &http.Request{ | ||||||
|  | 		Method: "CONNECT", | ||||||
|  | 		URL:    &url.URL{Opaque: addr}, | ||||||
|  | 		Host:   addr, | ||||||
|  | 		Header: connectHeader, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := connectReq.Write(conn); err != nil { | ||||||
|  | 		conn.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Read response. It's OK to use and discard buffered reader here becaue | ||||||
|  | 	// the remote server does not speak until spoken to. | ||||||
|  | 	br := bufio.NewReader(conn) | ||||||
|  | 	resp, err := http.ReadResponse(br, connectReq) | ||||||
|  | 	if err != nil { | ||||||
|  | 		conn.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if resp.StatusCode != 200 { | ||||||
|  | 		conn.Close() | ||||||
|  | 		f := strings.SplitN(resp.Status, " ", 2) | ||||||
|  | 		return nil, errors.New(f[1]) | ||||||
|  | 	} | ||||||
|  | 	return conn, nil | ||||||
|  | } | ||||||
							
								
								
									
										473
									
								
								vendor/github.com/gorilla/websocket/x_net_proxy.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										473
									
								
								vendor/github.com/gorilla/websocket/x_net_proxy.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,473 @@ | |||||||
|  | // Code generated by golang.org/x/tools/cmd/bundle. DO NOT EDIT. | ||||||
|  | //go:generate bundle -o x_net_proxy.go golang.org/x/net/proxy | ||||||
|  |  | ||||||
|  | // Package proxy provides support for a variety of protocols to proxy network | ||||||
|  | // data. | ||||||
|  | // | ||||||
|  |  | ||||||
|  | package websocket | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"io" | ||||||
|  | 	"net" | ||||||
|  | 	"net/url" | ||||||
|  | 	"os" | ||||||
|  | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type proxy_direct struct{} | ||||||
|  |  | ||||||
|  | // Direct is a direct proxy: one that makes network connections directly. | ||||||
|  | var proxy_Direct = proxy_direct{} | ||||||
|  |  | ||||||
|  | func (proxy_direct) Dial(network, addr string) (net.Conn, error) { | ||||||
|  | 	return net.Dial(network, addr) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A PerHost directs connections to a default Dialer unless the host name | ||||||
|  | // requested matches one of a number of exceptions. | ||||||
|  | type proxy_PerHost struct { | ||||||
|  | 	def, bypass proxy_Dialer | ||||||
|  |  | ||||||
|  | 	bypassNetworks []*net.IPNet | ||||||
|  | 	bypassIPs      []net.IP | ||||||
|  | 	bypassZones    []string | ||||||
|  | 	bypassHosts    []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewPerHost returns a PerHost Dialer that directs connections to either | ||||||
|  | // defaultDialer or bypass, depending on whether the connection matches one of | ||||||
|  | // the configured rules. | ||||||
|  | func proxy_NewPerHost(defaultDialer, bypass proxy_Dialer) *proxy_PerHost { | ||||||
|  | 	return &proxy_PerHost{ | ||||||
|  | 		def:    defaultDialer, | ||||||
|  | 		bypass: bypass, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Dial connects to the address addr on the given network through either | ||||||
|  | // defaultDialer or bypass. | ||||||
|  | func (p *proxy_PerHost) Dial(network, addr string) (c net.Conn, err error) { | ||||||
|  | 	host, _, err := net.SplitHostPort(addr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return p.dialerForRequest(host).Dial(network, addr) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (p *proxy_PerHost) dialerForRequest(host string) proxy_Dialer { | ||||||
|  | 	if ip := net.ParseIP(host); ip != nil { | ||||||
|  | 		for _, net := range p.bypassNetworks { | ||||||
|  | 			if net.Contains(ip) { | ||||||
|  | 				return p.bypass | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for _, bypassIP := range p.bypassIPs { | ||||||
|  | 			if bypassIP.Equal(ip) { | ||||||
|  | 				return p.bypass | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return p.def | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, zone := range p.bypassZones { | ||||||
|  | 		if strings.HasSuffix(host, zone) { | ||||||
|  | 			return p.bypass | ||||||
|  | 		} | ||||||
|  | 		if host == zone[1:] { | ||||||
|  | 			// For a zone ".example.com", we match "example.com" | ||||||
|  | 			// too. | ||||||
|  | 			return p.bypass | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, bypassHost := range p.bypassHosts { | ||||||
|  | 		if bypassHost == host { | ||||||
|  | 			return p.bypass | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return p.def | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddFromString parses a string that contains comma-separated values | ||||||
|  | // specifying hosts that should use the bypass proxy. Each value is either an | ||||||
|  | // IP address, a CIDR range, a zone (*.example.com) or a host name | ||||||
|  | // (localhost). A best effort is made to parse the string and errors are | ||||||
|  | // ignored. | ||||||
|  | func (p *proxy_PerHost) AddFromString(s string) { | ||||||
|  | 	hosts := strings.Split(s, ",") | ||||||
|  | 	for _, host := range hosts { | ||||||
|  | 		host = strings.TrimSpace(host) | ||||||
|  | 		if len(host) == 0 { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if strings.Contains(host, "/") { | ||||||
|  | 			// We assume that it's a CIDR address like 127.0.0.0/8 | ||||||
|  | 			if _, net, err := net.ParseCIDR(host); err == nil { | ||||||
|  | 				p.AddNetwork(net) | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if ip := net.ParseIP(host); ip != nil { | ||||||
|  | 			p.AddIP(ip) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if strings.HasPrefix(host, "*.") { | ||||||
|  | 			p.AddZone(host[1:]) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		p.AddHost(host) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddIP specifies an IP address that will use the bypass proxy. Note that | ||||||
|  | // this will only take effect if a literal IP address is dialed. A connection | ||||||
|  | // to a named host will never match an IP. | ||||||
|  | func (p *proxy_PerHost) AddIP(ip net.IP) { | ||||||
|  | 	p.bypassIPs = append(p.bypassIPs, ip) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddNetwork specifies an IP range that will use the bypass proxy. Note that | ||||||
|  | // this will only take effect if a literal IP address is dialed. A connection | ||||||
|  | // to a named host will never match. | ||||||
|  | func (p *proxy_PerHost) AddNetwork(net *net.IPNet) { | ||||||
|  | 	p.bypassNetworks = append(p.bypassNetworks, net) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddZone specifies a DNS suffix that will use the bypass proxy. A zone of | ||||||
|  | // "example.com" matches "example.com" and all of its subdomains. | ||||||
|  | func (p *proxy_PerHost) AddZone(zone string) { | ||||||
|  | 	if strings.HasSuffix(zone, ".") { | ||||||
|  | 		zone = zone[:len(zone)-1] | ||||||
|  | 	} | ||||||
|  | 	if !strings.HasPrefix(zone, ".") { | ||||||
|  | 		zone = "." + zone | ||||||
|  | 	} | ||||||
|  | 	p.bypassZones = append(p.bypassZones, zone) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AddHost specifies a host name that will use the bypass proxy. | ||||||
|  | func (p *proxy_PerHost) AddHost(host string) { | ||||||
|  | 	if strings.HasSuffix(host, ".") { | ||||||
|  | 		host = host[:len(host)-1] | ||||||
|  | 	} | ||||||
|  | 	p.bypassHosts = append(p.bypassHosts, host) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // A Dialer is a means to establish a connection. | ||||||
|  | type proxy_Dialer interface { | ||||||
|  | 	// Dial connects to the given address via the proxy. | ||||||
|  | 	Dial(network, addr string) (c net.Conn, err error) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Auth contains authentication parameters that specific Dialers may require. | ||||||
|  | type proxy_Auth struct { | ||||||
|  | 	User, Password string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromEnvironment returns the dialer specified by the proxy related variables in | ||||||
|  | // the environment. | ||||||
|  | func proxy_FromEnvironment() proxy_Dialer { | ||||||
|  | 	allProxy := proxy_allProxyEnv.Get() | ||||||
|  | 	if len(allProxy) == 0 { | ||||||
|  | 		return proxy_Direct | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	proxyURL, err := url.Parse(allProxy) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return proxy_Direct | ||||||
|  | 	} | ||||||
|  | 	proxy, err := proxy_FromURL(proxyURL, proxy_Direct) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return proxy_Direct | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	noProxy := proxy_noProxyEnv.Get() | ||||||
|  | 	if len(noProxy) == 0 { | ||||||
|  | 		return proxy | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	perHost := proxy_NewPerHost(proxy, proxy_Direct) | ||||||
|  | 	perHost.AddFromString(noProxy) | ||||||
|  | 	return perHost | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // proxySchemes is a map from URL schemes to a function that creates a Dialer | ||||||
|  | // from a URL with such a scheme. | ||||||
|  | var proxy_proxySchemes map[string]func(*url.URL, proxy_Dialer) (proxy_Dialer, error) | ||||||
|  |  | ||||||
|  | // RegisterDialerType takes a URL scheme and a function to generate Dialers from | ||||||
|  | // a URL with that scheme and a forwarding Dialer. Registered schemes are used | ||||||
|  | // by FromURL. | ||||||
|  | func proxy_RegisterDialerType(scheme string, f func(*url.URL, proxy_Dialer) (proxy_Dialer, error)) { | ||||||
|  | 	if proxy_proxySchemes == nil { | ||||||
|  | 		proxy_proxySchemes = make(map[string]func(*url.URL, proxy_Dialer) (proxy_Dialer, error)) | ||||||
|  | 	} | ||||||
|  | 	proxy_proxySchemes[scheme] = f | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromURL returns a Dialer given a URL specification and an underlying | ||||||
|  | // Dialer for it to make network requests. | ||||||
|  | func proxy_FromURL(u *url.URL, forward proxy_Dialer) (proxy_Dialer, error) { | ||||||
|  | 	var auth *proxy_Auth | ||||||
|  | 	if u.User != nil { | ||||||
|  | 		auth = new(proxy_Auth) | ||||||
|  | 		auth.User = u.User.Username() | ||||||
|  | 		if p, ok := u.User.Password(); ok { | ||||||
|  | 			auth.Password = p | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch u.Scheme { | ||||||
|  | 	case "socks5": | ||||||
|  | 		return proxy_SOCKS5("tcp", u.Host, auth, forward) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the scheme doesn't match any of the built-in schemes, see if it | ||||||
|  | 	// was registered by another package. | ||||||
|  | 	if proxy_proxySchemes != nil { | ||||||
|  | 		if f, ok := proxy_proxySchemes[u.Scheme]; ok { | ||||||
|  | 			return f(u, forward) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil, errors.New("proxy: unknown scheme: " + u.Scheme) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	proxy_allProxyEnv = &proxy_envOnce{ | ||||||
|  | 		names: []string{"ALL_PROXY", "all_proxy"}, | ||||||
|  | 	} | ||||||
|  | 	proxy_noProxyEnv = &proxy_envOnce{ | ||||||
|  | 		names: []string{"NO_PROXY", "no_proxy"}, | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // envOnce looks up an environment variable (optionally by multiple | ||||||
|  | // names) once. It mitigates expensive lookups on some platforms | ||||||
|  | // (e.g. Windows). | ||||||
|  | // (Borrowed from net/http/transport.go) | ||||||
|  | type proxy_envOnce struct { | ||||||
|  | 	names []string | ||||||
|  | 	once  sync.Once | ||||||
|  | 	val   string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *proxy_envOnce) Get() string { | ||||||
|  | 	e.once.Do(e.init) | ||||||
|  | 	return e.val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *proxy_envOnce) init() { | ||||||
|  | 	for _, n := range e.names { | ||||||
|  | 		e.val = os.Getenv(n) | ||||||
|  | 		if e.val != "" { | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SOCKS5 returns a Dialer that makes SOCKSv5 connections to the given address | ||||||
|  | // with an optional username and password. See RFC 1928 and RFC 1929. | ||||||
|  | func proxy_SOCKS5(network, addr string, auth *proxy_Auth, forward proxy_Dialer) (proxy_Dialer, error) { | ||||||
|  | 	s := &proxy_socks5{ | ||||||
|  | 		network: network, | ||||||
|  | 		addr:    addr, | ||||||
|  | 		forward: forward, | ||||||
|  | 	} | ||||||
|  | 	if auth != nil { | ||||||
|  | 		s.user = auth.User | ||||||
|  | 		s.password = auth.Password | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return s, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type proxy_socks5 struct { | ||||||
|  | 	user, password string | ||||||
|  | 	network, addr  string | ||||||
|  | 	forward        proxy_Dialer | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const proxy_socks5Version = 5 | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	proxy_socks5AuthNone     = 0 | ||||||
|  | 	proxy_socks5AuthPassword = 2 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const proxy_socks5Connect = 1 | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	proxy_socks5IP4    = 1 | ||||||
|  | 	proxy_socks5Domain = 3 | ||||||
|  | 	proxy_socks5IP6    = 4 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var proxy_socks5Errors = []string{ | ||||||
|  | 	"", | ||||||
|  | 	"general failure", | ||||||
|  | 	"connection forbidden", | ||||||
|  | 	"network unreachable", | ||||||
|  | 	"host unreachable", | ||||||
|  | 	"connection refused", | ||||||
|  | 	"TTL expired", | ||||||
|  | 	"command not supported", | ||||||
|  | 	"address type not supported", | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Dial connects to the address addr on the given network via the SOCKS5 proxy. | ||||||
|  | func (s *proxy_socks5) Dial(network, addr string) (net.Conn, error) { | ||||||
|  | 	switch network { | ||||||
|  | 	case "tcp", "tcp6", "tcp4": | ||||||
|  | 	default: | ||||||
|  | 		return nil, errors.New("proxy: no support for SOCKS5 proxy connections of type " + network) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	conn, err := s.forward.Dial(s.network, s.addr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	if err := s.connect(conn, addr); err != nil { | ||||||
|  | 		conn.Close() | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return conn, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // connect takes an existing connection to a socks5 proxy server, | ||||||
|  | // and commands the server to extend that connection to target, | ||||||
|  | // which must be a canonical address with a host and port. | ||||||
|  | func (s *proxy_socks5) connect(conn net.Conn, target string) error { | ||||||
|  | 	host, portStr, err := net.SplitHostPort(target) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	port, err := strconv.Atoi(portStr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return errors.New("proxy: failed to parse port number: " + portStr) | ||||||
|  | 	} | ||||||
|  | 	if port < 1 || port > 0xffff { | ||||||
|  | 		return errors.New("proxy: port number out of range: " + portStr) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// the size here is just an estimate | ||||||
|  | 	buf := make([]byte, 0, 6+len(host)) | ||||||
|  |  | ||||||
|  | 	buf = append(buf, proxy_socks5Version) | ||||||
|  | 	if len(s.user) > 0 && len(s.user) < 256 && len(s.password) < 256 { | ||||||
|  | 		buf = append(buf, 2 /* num auth methods */, proxy_socks5AuthNone, proxy_socks5AuthPassword) | ||||||
|  | 	} else { | ||||||
|  | 		buf = append(buf, 1 /* num auth methods */, proxy_socks5AuthNone) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, err := conn.Write(buf); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to write greeting to SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, err := io.ReadFull(conn, buf[:2]); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to read greeting from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  | 	if buf[0] != 5 { | ||||||
|  | 		return errors.New("proxy: SOCKS5 proxy at " + s.addr + " has unexpected version " + strconv.Itoa(int(buf[0]))) | ||||||
|  | 	} | ||||||
|  | 	if buf[1] == 0xff { | ||||||
|  | 		return errors.New("proxy: SOCKS5 proxy at " + s.addr + " requires authentication") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// See RFC 1929 | ||||||
|  | 	if buf[1] == proxy_socks5AuthPassword { | ||||||
|  | 		buf = buf[:0] | ||||||
|  | 		buf = append(buf, 1 /* password protocol version */) | ||||||
|  | 		buf = append(buf, uint8(len(s.user))) | ||||||
|  | 		buf = append(buf, s.user...) | ||||||
|  | 		buf = append(buf, uint8(len(s.password))) | ||||||
|  | 		buf = append(buf, s.password...) | ||||||
|  |  | ||||||
|  | 		if _, err := conn.Write(buf); err != nil { | ||||||
|  | 			return errors.New("proxy: failed to write authentication request to SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if _, err := io.ReadFull(conn, buf[:2]); err != nil { | ||||||
|  | 			return errors.New("proxy: failed to read authentication reply from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if buf[1] != 0 { | ||||||
|  | 			return errors.New("proxy: SOCKS5 proxy at " + s.addr + " rejected username/password") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	buf = buf[:0] | ||||||
|  | 	buf = append(buf, proxy_socks5Version, proxy_socks5Connect, 0 /* reserved */) | ||||||
|  |  | ||||||
|  | 	if ip := net.ParseIP(host); ip != nil { | ||||||
|  | 		if ip4 := ip.To4(); ip4 != nil { | ||||||
|  | 			buf = append(buf, proxy_socks5IP4) | ||||||
|  | 			ip = ip4 | ||||||
|  | 		} else { | ||||||
|  | 			buf = append(buf, proxy_socks5IP6) | ||||||
|  | 		} | ||||||
|  | 		buf = append(buf, ip...) | ||||||
|  | 	} else { | ||||||
|  | 		if len(host) > 255 { | ||||||
|  | 			return errors.New("proxy: destination host name too long: " + host) | ||||||
|  | 		} | ||||||
|  | 		buf = append(buf, proxy_socks5Domain) | ||||||
|  | 		buf = append(buf, byte(len(host))) | ||||||
|  | 		buf = append(buf, host...) | ||||||
|  | 	} | ||||||
|  | 	buf = append(buf, byte(port>>8), byte(port)) | ||||||
|  |  | ||||||
|  | 	if _, err := conn.Write(buf); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to write connect request to SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, err := io.ReadFull(conn, buf[:4]); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to read connect reply from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	failure := "unknown error" | ||||||
|  | 	if int(buf[1]) < len(proxy_socks5Errors) { | ||||||
|  | 		failure = proxy_socks5Errors[buf[1]] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(failure) > 0 { | ||||||
|  | 		return errors.New("proxy: SOCKS5 proxy at " + s.addr + " failed to connect: " + failure) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	bytesToDiscard := 0 | ||||||
|  | 	switch buf[3] { | ||||||
|  | 	case proxy_socks5IP4: | ||||||
|  | 		bytesToDiscard = net.IPv4len | ||||||
|  | 	case proxy_socks5IP6: | ||||||
|  | 		bytesToDiscard = net.IPv6len | ||||||
|  | 	case proxy_socks5Domain: | ||||||
|  | 		_, err := io.ReadFull(conn, buf[:1]) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return errors.New("proxy: failed to read domain length from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 		} | ||||||
|  | 		bytesToDiscard = int(buf[0]) | ||||||
|  | 	default: | ||||||
|  | 		return errors.New("proxy: got unknown address type " + strconv.Itoa(int(buf[3])) + " from SOCKS5 proxy at " + s.addr) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if cap(buf) < bytesToDiscard { | ||||||
|  | 		buf = make([]byte, bytesToDiscard) | ||||||
|  | 	} else { | ||||||
|  | 		buf = buf[:bytesToDiscard] | ||||||
|  | 	} | ||||||
|  | 	if _, err := io.ReadFull(conn, buf); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to read address from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Also need to discard the port number | ||||||
|  | 	if _, err := io.ReadFull(conn, buf[:2]); err != nil { | ||||||
|  | 		return errors.New("proxy: failed to read port from SOCKS5 proxy at " + s.addr + ": " + err.Error()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										212
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										212
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,212 @@ | |||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	// Default2QRecentRatio is the ratio of the 2Q cache dedicated | ||||||
|  | 	// to recently added entries that have only been accessed once. | ||||||
|  | 	Default2QRecentRatio = 0.25 | ||||||
|  |  | ||||||
|  | 	// Default2QGhostEntries is the default ratio of ghost | ||||||
|  | 	// entries kept to track entries recently evicted | ||||||
|  | 	Default2QGhostEntries = 0.50 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // TwoQueueCache is a thread-safe fixed size 2Q cache. | ||||||
|  | // 2Q is an enhancement over the standard LRU cache | ||||||
|  | // in that it tracks both frequently and recently used | ||||||
|  | // entries separately. This avoids a burst in access to new | ||||||
|  | // entries from evicting frequently used entries. It adds some | ||||||
|  | // additional tracking overhead to the standard LRU cache, and is | ||||||
|  | // computationally about 2x the cost, and adds some metadata over | ||||||
|  | // head. The ARCCache is similar, but does not require setting any | ||||||
|  | // parameters. | ||||||
|  | type TwoQueueCache struct { | ||||||
|  | 	size       int | ||||||
|  | 	recentSize int | ||||||
|  |  | ||||||
|  | 	recent      *simplelru.LRU | ||||||
|  | 	frequent    *simplelru.LRU | ||||||
|  | 	recentEvict *simplelru.LRU | ||||||
|  | 	lock        sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New2Q creates a new TwoQueueCache using the default | ||||||
|  | // values for the parameters. | ||||||
|  | func New2Q(size int) (*TwoQueueCache, error) { | ||||||
|  | 	return New2QParams(size, Default2QRecentRatio, Default2QGhostEntries) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New2QParams creates a new TwoQueueCache using the provided | ||||||
|  | // parameter values. | ||||||
|  | func New2QParams(size int, recentRatio float64, ghostRatio float64) (*TwoQueueCache, error) { | ||||||
|  | 	if size <= 0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid size") | ||||||
|  | 	} | ||||||
|  | 	if recentRatio < 0.0 || recentRatio > 1.0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid recent ratio") | ||||||
|  | 	} | ||||||
|  | 	if ghostRatio < 0.0 || ghostRatio > 1.0 { | ||||||
|  | 		return nil, fmt.Errorf("invalid ghost ratio") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Determine the sub-sizes | ||||||
|  | 	recentSize := int(float64(size) * recentRatio) | ||||||
|  | 	evictSize := int(float64(size) * ghostRatio) | ||||||
|  |  | ||||||
|  | 	// Allocate the LRUs | ||||||
|  | 	recent, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	frequent, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	recentEvict, err := simplelru.NewLRU(evictSize, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Initialize the cache | ||||||
|  | 	c := &TwoQueueCache{ | ||||||
|  | 		size:        size, | ||||||
|  | 		recentSize:  recentSize, | ||||||
|  | 		recent:      recent, | ||||||
|  | 		frequent:    frequent, | ||||||
|  | 		recentEvict: recentEvict, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if this is a frequent value | ||||||
|  | 	if val, ok := c.frequent.Get(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the value is contained in recent, then we | ||||||
|  | 	// promote it to frequent | ||||||
|  | 	if val, ok := c.recent.Peek(key); ok { | ||||||
|  | 		c.recent.Remove(key) | ||||||
|  | 		c.frequent.Add(key, val) | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// No hit | ||||||
|  | 	return nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Add(key, value interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the value is frequently used already, | ||||||
|  | 	// and just update the value | ||||||
|  | 	if c.frequent.Contains(key) { | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is recently used, and promote | ||||||
|  | 	// the value into the frequent list | ||||||
|  | 	if c.recent.Contains(key) { | ||||||
|  | 		c.recent.Remove(key) | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the value was recently evicted, add it to the | ||||||
|  | 	// frequently used list | ||||||
|  | 	if c.recentEvict.Contains(key) { | ||||||
|  | 		c.ensureSpace(true) | ||||||
|  | 		c.recentEvict.Remove(key) | ||||||
|  | 		c.frequent.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add to the recently seen list | ||||||
|  | 	c.ensureSpace(false) | ||||||
|  | 	c.recent.Add(key, value) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ensureSpace is used to ensure we have space in the cache | ||||||
|  | func (c *TwoQueueCache) ensureSpace(recentEvict bool) { | ||||||
|  | 	// If we have space, nothing to do | ||||||
|  | 	recentLen := c.recent.Len() | ||||||
|  | 	freqLen := c.frequent.Len() | ||||||
|  | 	if recentLen+freqLen < c.size { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If the recent buffer is larger than | ||||||
|  | 	// the target, evict from there | ||||||
|  | 	if recentLen > 0 && (recentLen > c.recentSize || (recentLen == c.recentSize && !recentEvict)) { | ||||||
|  | 		k, _, _ := c.recent.RemoveOldest() | ||||||
|  | 		c.recentEvict.Add(k, nil) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Remove from the frequent list otherwise | ||||||
|  | 	c.frequent.RemoveOldest() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.recent.Len() + c.frequent.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	k1 := c.frequent.Keys() | ||||||
|  | 	k2 := c.recent.Keys() | ||||||
|  | 	return append(k1, k2...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	if c.frequent.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.recent.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.recentEvict.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	c.recent.Purge() | ||||||
|  | 	c.frequent.Purge() | ||||||
|  | 	c.recentEvict.Purge() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.frequent.Contains(key) || c.recent.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TwoQueueCache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	if val, ok := c.frequent.Peek(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  | 	return c.recent.Peek(key) | ||||||
|  | } | ||||||
							
								
								
									
										362
									
								
								vendor/github.com/hashicorp/golang-lru/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										362
									
								
								vendor/github.com/hashicorp/golang-lru/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,362 @@ | |||||||
|  | Mozilla Public License, version 2.0 | ||||||
|  |  | ||||||
|  | 1. Definitions | ||||||
|  |  | ||||||
|  | 1.1. "Contributor" | ||||||
|  |  | ||||||
|  |      means each individual or legal entity that creates, contributes to the | ||||||
|  |      creation of, or owns Covered Software. | ||||||
|  |  | ||||||
|  | 1.2. "Contributor Version" | ||||||
|  |  | ||||||
|  |      means the combination of the Contributions of others (if any) used by a | ||||||
|  |      Contributor and that particular Contributor's Contribution. | ||||||
|  |  | ||||||
|  | 1.3. "Contribution" | ||||||
|  |  | ||||||
|  |      means Covered Software of a particular Contributor. | ||||||
|  |  | ||||||
|  | 1.4. "Covered Software" | ||||||
|  |  | ||||||
|  |      means Source Code Form to which the initial Contributor has attached the | ||||||
|  |      notice in Exhibit A, the Executable Form of such Source Code Form, and | ||||||
|  |      Modifications of such Source Code Form, in each case including portions | ||||||
|  |      thereof. | ||||||
|  |  | ||||||
|  | 1.5. "Incompatible With Secondary Licenses" | ||||||
|  |      means | ||||||
|  |  | ||||||
|  |      a. that the initial Contributor has attached the notice described in | ||||||
|  |         Exhibit B to the Covered Software; or | ||||||
|  |  | ||||||
|  |      b. that the Covered Software was made available under the terms of | ||||||
|  |         version 1.1 or earlier of the License, but not also under the terms of | ||||||
|  |         a Secondary License. | ||||||
|  |  | ||||||
|  | 1.6. "Executable Form" | ||||||
|  |  | ||||||
|  |      means any form of the work other than Source Code Form. | ||||||
|  |  | ||||||
|  | 1.7. "Larger Work" | ||||||
|  |  | ||||||
|  |      means a work that combines Covered Software with other material, in a | ||||||
|  |      separate file or files, that is not Covered Software. | ||||||
|  |  | ||||||
|  | 1.8. "License" | ||||||
|  |  | ||||||
|  |      means this document. | ||||||
|  |  | ||||||
|  | 1.9. "Licensable" | ||||||
|  |  | ||||||
|  |      means having the right to grant, to the maximum extent possible, whether | ||||||
|  |      at the time of the initial grant or subsequently, any and all of the | ||||||
|  |      rights conveyed by this License. | ||||||
|  |  | ||||||
|  | 1.10. "Modifications" | ||||||
|  |  | ||||||
|  |      means any of the following: | ||||||
|  |  | ||||||
|  |      a. any file in Source Code Form that results from an addition to, | ||||||
|  |         deletion from, or modification of the contents of Covered Software; or | ||||||
|  |  | ||||||
|  |      b. any new file in Source Code Form that contains any Covered Software. | ||||||
|  |  | ||||||
|  | 1.11. "Patent Claims" of a Contributor | ||||||
|  |  | ||||||
|  |       means any patent claim(s), including without limitation, method, | ||||||
|  |       process, and apparatus claims, in any patent Licensable by such | ||||||
|  |       Contributor that would be infringed, but for the grant of the License, | ||||||
|  |       by the making, using, selling, offering for sale, having made, import, | ||||||
|  |       or transfer of either its Contributions or its Contributor Version. | ||||||
|  |  | ||||||
|  | 1.12. "Secondary License" | ||||||
|  |  | ||||||
|  |       means either the GNU General Public License, Version 2.0, the GNU Lesser | ||||||
|  |       General Public License, Version 2.1, the GNU Affero General Public | ||||||
|  |       License, Version 3.0, or any later versions of those licenses. | ||||||
|  |  | ||||||
|  | 1.13. "Source Code Form" | ||||||
|  |  | ||||||
|  |       means the form of the work preferred for making modifications. | ||||||
|  |  | ||||||
|  | 1.14. "You" (or "Your") | ||||||
|  |  | ||||||
|  |       means an individual or a legal entity exercising rights under this | ||||||
|  |       License. For legal entities, "You" includes any entity that controls, is | ||||||
|  |       controlled by, or is under common control with You. For purposes of this | ||||||
|  |       definition, "control" means (a) the power, direct or indirect, to cause | ||||||
|  |       the direction or management of such entity, whether by contract or | ||||||
|  |       otherwise, or (b) ownership of more than fifty percent (50%) of the | ||||||
|  |       outstanding shares or beneficial ownership of such entity. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 2. License Grants and Conditions | ||||||
|  |  | ||||||
|  | 2.1. Grants | ||||||
|  |  | ||||||
|  |      Each Contributor hereby grants You a world-wide, royalty-free, | ||||||
|  |      non-exclusive license: | ||||||
|  |  | ||||||
|  |      a. under intellectual property rights (other than patent or trademark) | ||||||
|  |         Licensable by such Contributor to use, reproduce, make available, | ||||||
|  |         modify, display, perform, distribute, and otherwise exploit its | ||||||
|  |         Contributions, either on an unmodified basis, with Modifications, or | ||||||
|  |         as part of a Larger Work; and | ||||||
|  |  | ||||||
|  |      b. under Patent Claims of such Contributor to make, use, sell, offer for | ||||||
|  |         sale, have made, import, and otherwise transfer either its | ||||||
|  |         Contributions or its Contributor Version. | ||||||
|  |  | ||||||
|  | 2.2. Effective Date | ||||||
|  |  | ||||||
|  |      The licenses granted in Section 2.1 with respect to any Contribution | ||||||
|  |      become effective for each Contribution on the date the Contributor first | ||||||
|  |      distributes such Contribution. | ||||||
|  |  | ||||||
|  | 2.3. Limitations on Grant Scope | ||||||
|  |  | ||||||
|  |      The licenses granted in this Section 2 are the only rights granted under | ||||||
|  |      this License. No additional rights or licenses will be implied from the | ||||||
|  |      distribution or licensing of Covered Software under this License. | ||||||
|  |      Notwithstanding Section 2.1(b) above, no patent license is granted by a | ||||||
|  |      Contributor: | ||||||
|  |  | ||||||
|  |      a. for any code that a Contributor has removed from Covered Software; or | ||||||
|  |  | ||||||
|  |      b. for infringements caused by: (i) Your and any other third party's | ||||||
|  |         modifications of Covered Software, or (ii) the combination of its | ||||||
|  |         Contributions with other software (except as part of its Contributor | ||||||
|  |         Version); or | ||||||
|  |  | ||||||
|  |      c. under Patent Claims infringed by Covered Software in the absence of | ||||||
|  |         its Contributions. | ||||||
|  |  | ||||||
|  |      This License does not grant any rights in the trademarks, service marks, | ||||||
|  |      or logos of any Contributor (except as may be necessary to comply with | ||||||
|  |      the notice requirements in Section 3.4). | ||||||
|  |  | ||||||
|  | 2.4. Subsequent Licenses | ||||||
|  |  | ||||||
|  |      No Contributor makes additional grants as a result of Your choice to | ||||||
|  |      distribute the Covered Software under a subsequent version of this | ||||||
|  |      License (see Section 10.2) or under the terms of a Secondary License (if | ||||||
|  |      permitted under the terms of Section 3.3). | ||||||
|  |  | ||||||
|  | 2.5. Representation | ||||||
|  |  | ||||||
|  |      Each Contributor represents that the Contributor believes its | ||||||
|  |      Contributions are its original creation(s) or it has sufficient rights to | ||||||
|  |      grant the rights to its Contributions conveyed by this License. | ||||||
|  |  | ||||||
|  | 2.6. Fair Use | ||||||
|  |  | ||||||
|  |      This License is not intended to limit any rights You have under | ||||||
|  |      applicable copyright doctrines of fair use, fair dealing, or other | ||||||
|  |      equivalents. | ||||||
|  |  | ||||||
|  | 2.7. Conditions | ||||||
|  |  | ||||||
|  |      Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted in | ||||||
|  |      Section 2.1. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 3. Responsibilities | ||||||
|  |  | ||||||
|  | 3.1. Distribution of Source Form | ||||||
|  |  | ||||||
|  |      All distribution of Covered Software in Source Code Form, including any | ||||||
|  |      Modifications that You create or to which You contribute, must be under | ||||||
|  |      the terms of this License. You must inform recipients that the Source | ||||||
|  |      Code Form of the Covered Software is governed by the terms of this | ||||||
|  |      License, and how they can obtain a copy of this License. You may not | ||||||
|  |      attempt to alter or restrict the recipients' rights in the Source Code | ||||||
|  |      Form. | ||||||
|  |  | ||||||
|  | 3.2. Distribution of Executable Form | ||||||
|  |  | ||||||
|  |      If You distribute Covered Software in Executable Form then: | ||||||
|  |  | ||||||
|  |      a. such Covered Software must also be made available in Source Code Form, | ||||||
|  |         as described in Section 3.1, and You must inform recipients of the | ||||||
|  |         Executable Form how they can obtain a copy of such Source Code Form by | ||||||
|  |         reasonable means in a timely manner, at a charge no more than the cost | ||||||
|  |         of distribution to the recipient; and | ||||||
|  |  | ||||||
|  |      b. You may distribute such Executable Form under the terms of this | ||||||
|  |         License, or sublicense it under different terms, provided that the | ||||||
|  |         license for the Executable Form does not attempt to limit or alter the | ||||||
|  |         recipients' rights in the Source Code Form under this License. | ||||||
|  |  | ||||||
|  | 3.3. Distribution of a Larger Work | ||||||
|  |  | ||||||
|  |      You may create and distribute a Larger Work under terms of Your choice, | ||||||
|  |      provided that You also comply with the requirements of this License for | ||||||
|  |      the Covered Software. If the Larger Work is a combination of Covered | ||||||
|  |      Software with a work governed by one or more Secondary Licenses, and the | ||||||
|  |      Covered Software is not Incompatible With Secondary Licenses, this | ||||||
|  |      License permits You to additionally distribute such Covered Software | ||||||
|  |      under the terms of such Secondary License(s), so that the recipient of | ||||||
|  |      the Larger Work may, at their option, further distribute the Covered | ||||||
|  |      Software under the terms of either this License or such Secondary | ||||||
|  |      License(s). | ||||||
|  |  | ||||||
|  | 3.4. Notices | ||||||
|  |  | ||||||
|  |      You may not remove or alter the substance of any license notices | ||||||
|  |      (including copyright notices, patent notices, disclaimers of warranty, or | ||||||
|  |      limitations of liability) contained within the Source Code Form of the | ||||||
|  |      Covered Software, except that You may alter any license notices to the | ||||||
|  |      extent required to remedy known factual inaccuracies. | ||||||
|  |  | ||||||
|  | 3.5. Application of Additional Terms | ||||||
|  |  | ||||||
|  |      You may choose to offer, and to charge a fee for, warranty, support, | ||||||
|  |      indemnity or liability obligations to one or more recipients of Covered | ||||||
|  |      Software. However, You may do so only on Your own behalf, and not on | ||||||
|  |      behalf of any Contributor. You must make it absolutely clear that any | ||||||
|  |      such warranty, support, indemnity, or liability obligation is offered by | ||||||
|  |      You alone, and You hereby agree to indemnify every Contributor for any | ||||||
|  |      liability incurred by such Contributor as a result of warranty, support, | ||||||
|  |      indemnity or liability terms You offer. You may include additional | ||||||
|  |      disclaimers of warranty and limitations of liability specific to any | ||||||
|  |      jurisdiction. | ||||||
|  |  | ||||||
|  | 4. Inability to Comply Due to Statute or Regulation | ||||||
|  |  | ||||||
|  |    If it is impossible for You to comply with any of the terms of this License | ||||||
|  |    with respect to some or all of the Covered Software due to statute, | ||||||
|  |    judicial order, or regulation then You must: (a) comply with the terms of | ||||||
|  |    this License to the maximum extent possible; and (b) describe the | ||||||
|  |    limitations and the code they affect. Such description must be placed in a | ||||||
|  |    text file included with all distributions of the Covered Software under | ||||||
|  |    this License. Except to the extent prohibited by statute or regulation, | ||||||
|  |    such description must be sufficiently detailed for a recipient of ordinary | ||||||
|  |    skill to be able to understand it. | ||||||
|  |  | ||||||
|  | 5. Termination | ||||||
|  |  | ||||||
|  | 5.1. The rights granted under this License will terminate automatically if You | ||||||
|  |      fail to comply with any of its terms. However, if You become compliant, | ||||||
|  |      then the rights granted under this License from a particular Contributor | ||||||
|  |      are reinstated (a) provisionally, unless and until such Contributor | ||||||
|  |      explicitly and finally terminates Your grants, and (b) on an ongoing | ||||||
|  |      basis, if such Contributor fails to notify You of the non-compliance by | ||||||
|  |      some reasonable means prior to 60 days after You have come back into | ||||||
|  |      compliance. Moreover, Your grants from a particular Contributor are | ||||||
|  |      reinstated on an ongoing basis if such Contributor notifies You of the | ||||||
|  |      non-compliance by some reasonable means, this is the first time You have | ||||||
|  |      received notice of non-compliance with this License from such | ||||||
|  |      Contributor, and You become compliant prior to 30 days after Your receipt | ||||||
|  |      of the notice. | ||||||
|  |  | ||||||
|  | 5.2. If You initiate litigation against any entity by asserting a patent | ||||||
|  |      infringement claim (excluding declaratory judgment actions, | ||||||
|  |      counter-claims, and cross-claims) alleging that a Contributor Version | ||||||
|  |      directly or indirectly infringes any patent, then the rights granted to | ||||||
|  |      You by any and all Contributors for the Covered Software under Section | ||||||
|  |      2.1 of this License shall terminate. | ||||||
|  |  | ||||||
|  | 5.3. In the event of termination under Sections 5.1 or 5.2 above, all end user | ||||||
|  |      license agreements (excluding distributors and resellers) which have been | ||||||
|  |      validly granted by You or Your distributors under this License prior to | ||||||
|  |      termination shall survive termination. | ||||||
|  |  | ||||||
|  | 6. Disclaimer of Warranty | ||||||
|  |  | ||||||
|  |    Covered Software is provided under this License on an "as is" basis, | ||||||
|  |    without warranty of any kind, either expressed, implied, or statutory, | ||||||
|  |    including, without limitation, warranties that the Covered Software is free | ||||||
|  |    of defects, merchantable, fit for a particular purpose or non-infringing. | ||||||
|  |    The entire risk as to the quality and performance of the Covered Software | ||||||
|  |    is with You. Should any Covered Software prove defective in any respect, | ||||||
|  |    You (not any Contributor) assume the cost of any necessary servicing, | ||||||
|  |    repair, or correction. This disclaimer of warranty constitutes an essential | ||||||
|  |    part of this License. No use of  any Covered Software is authorized under | ||||||
|  |    this License except under this disclaimer. | ||||||
|  |  | ||||||
|  | 7. Limitation of Liability | ||||||
|  |  | ||||||
|  |    Under no circumstances and under no legal theory, whether tort (including | ||||||
|  |    negligence), contract, or otherwise, shall any Contributor, or anyone who | ||||||
|  |    distributes Covered Software as permitted above, be liable to You for any | ||||||
|  |    direct, indirect, special, incidental, or consequential damages of any | ||||||
|  |    character including, without limitation, damages for lost profits, loss of | ||||||
|  |    goodwill, work stoppage, computer failure or malfunction, or any and all | ||||||
|  |    other commercial damages or losses, even if such party shall have been | ||||||
|  |    informed of the possibility of such damages. This limitation of liability | ||||||
|  |    shall not apply to liability for death or personal injury resulting from | ||||||
|  |    such party's negligence to the extent applicable law prohibits such | ||||||
|  |    limitation. Some jurisdictions do not allow the exclusion or limitation of | ||||||
|  |    incidental or consequential damages, so this exclusion and limitation may | ||||||
|  |    not apply to You. | ||||||
|  |  | ||||||
|  | 8. Litigation | ||||||
|  |  | ||||||
|  |    Any litigation relating to this License may be brought only in the courts | ||||||
|  |    of a jurisdiction where the defendant maintains its principal place of | ||||||
|  |    business and such litigation shall be governed by laws of that | ||||||
|  |    jurisdiction, without reference to its conflict-of-law provisions. Nothing | ||||||
|  |    in this Section shall prevent a party's ability to bring cross-claims or | ||||||
|  |    counter-claims. | ||||||
|  |  | ||||||
|  | 9. Miscellaneous | ||||||
|  |  | ||||||
|  |    This License represents the complete agreement concerning the subject | ||||||
|  |    matter hereof. If any provision of this License is held to be | ||||||
|  |    unenforceable, such provision shall be reformed only to the extent | ||||||
|  |    necessary to make it enforceable. Any law or regulation which provides that | ||||||
|  |    the language of a contract shall be construed against the drafter shall not | ||||||
|  |    be used to construe this License against a Contributor. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 10. Versions of the License | ||||||
|  |  | ||||||
|  | 10.1. New Versions | ||||||
|  |  | ||||||
|  |       Mozilla Foundation is the license steward. Except as provided in Section | ||||||
|  |       10.3, no one other than the license steward has the right to modify or | ||||||
|  |       publish new versions of this License. Each version will be given a | ||||||
|  |       distinguishing version number. | ||||||
|  |  | ||||||
|  | 10.2. Effect of New Versions | ||||||
|  |  | ||||||
|  |       You may distribute the Covered Software under the terms of the version | ||||||
|  |       of the License under which You originally received the Covered Software, | ||||||
|  |       or under the terms of any subsequent version published by the license | ||||||
|  |       steward. | ||||||
|  |  | ||||||
|  | 10.3. Modified Versions | ||||||
|  |  | ||||||
|  |       If you create software not governed by this License, and you want to | ||||||
|  |       create a new license for such software, you may create and use a | ||||||
|  |       modified version of this License if you rename the license and remove | ||||||
|  |       any references to the name of the license steward (except to note that | ||||||
|  |       such modified license differs from this License). | ||||||
|  |  | ||||||
|  | 10.4. Distributing Source Code Form that is Incompatible With Secondary | ||||||
|  |       Licenses If You choose to distribute Source Code Form that is | ||||||
|  |       Incompatible With Secondary Licenses under the terms of this version of | ||||||
|  |       the License, the notice described in Exhibit B of this License must be | ||||||
|  |       attached. | ||||||
|  |  | ||||||
|  | Exhibit A - Source Code Form License Notice | ||||||
|  |  | ||||||
|  |       This Source Code Form is subject to the | ||||||
|  |       terms of the Mozilla Public License, v. | ||||||
|  |       2.0. If a copy of the MPL was not | ||||||
|  |       distributed with this file, You can | ||||||
|  |       obtain one at | ||||||
|  |       http://mozilla.org/MPL/2.0/. | ||||||
|  |  | ||||||
|  | If it is not possible or desirable to put the notice in a particular file, | ||||||
|  | then You may include the notice in a location (such as a LICENSE file in a | ||||||
|  | relevant directory) where a recipient would be likely to look for such a | ||||||
|  | notice. | ||||||
|  |  | ||||||
|  | You may add additional accurate notices of copyright ownership. | ||||||
|  |  | ||||||
|  | Exhibit B - "Incompatible With Secondary Licenses" Notice | ||||||
|  |  | ||||||
|  |       This Source Code Form is "Incompatible | ||||||
|  |       With Secondary Licenses", as defined by | ||||||
|  |       the Mozilla Public License, v. 2.0. | ||||||
							
								
								
									
										257
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										257
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,257 @@ | |||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // ARCCache is a thread-safe fixed size Adaptive Replacement Cache (ARC). | ||||||
|  | // ARC is an enhancement over the standard LRU cache in that tracks both | ||||||
|  | // frequency and recency of use. This avoids a burst in access to new | ||||||
|  | // entries from evicting the frequently used older entries. It adds some | ||||||
|  | // additional tracking overhead to a standard LRU cache, computationally | ||||||
|  | // it is roughly 2x the cost, and the extra memory overhead is linear | ||||||
|  | // with the size of the cache. ARC has been patented by IBM, but is | ||||||
|  | // similar to the TwoQueueCache (2Q) which requires setting parameters. | ||||||
|  | type ARCCache struct { | ||||||
|  | 	size int // Size is the total capacity of the cache | ||||||
|  | 	p    int // P is the dynamic preference towards T1 or T2 | ||||||
|  |  | ||||||
|  | 	t1 *simplelru.LRU // T1 is the LRU for recently accessed items | ||||||
|  | 	b1 *simplelru.LRU // B1 is the LRU for evictions from t1 | ||||||
|  |  | ||||||
|  | 	t2 *simplelru.LRU // T2 is the LRU for frequently accessed items | ||||||
|  | 	b2 *simplelru.LRU // B2 is the LRU for evictions from t2 | ||||||
|  |  | ||||||
|  | 	lock sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewARC creates an ARC of the given size | ||||||
|  | func NewARC(size int) (*ARCCache, error) { | ||||||
|  | 	// Create the sub LRUs | ||||||
|  | 	b1, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	b2, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	t1, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	t2, err := simplelru.NewLRU(size, nil) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Initialize the ARC | ||||||
|  | 	c := &ARCCache{ | ||||||
|  | 		size: size, | ||||||
|  | 		p:    0, | ||||||
|  | 		t1:   t1, | ||||||
|  | 		b1:   b1, | ||||||
|  | 		t2:   t2, | ||||||
|  | 		b2:   b2, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *ARCCache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Ff the value is contained in T1 (recent), then | ||||||
|  | 	// promote it to T2 (frequent) | ||||||
|  | 	if val, ok := c.t1.Peek(key); ok { | ||||||
|  | 		c.t1.Remove(key) | ||||||
|  | 		c.t2.Add(key, val) | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is contained in T2 (frequent) | ||||||
|  | 	if val, ok := c.t2.Get(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// No hit | ||||||
|  | 	return nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache. | ||||||
|  | func (c *ARCCache) Add(key, value interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	// Check if the value is contained in T1 (recent), and potentially | ||||||
|  | 	// promote it to frequent T2 | ||||||
|  | 	if c.t1.Contains(key) { | ||||||
|  | 		c.t1.Remove(key) | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if the value is already in T2 (frequent) and update it | ||||||
|  | 	if c.t2.Contains(key) { | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if this value was recently evicted as part of the | ||||||
|  | 	// recently used list | ||||||
|  | 	if c.b1.Contains(key) { | ||||||
|  | 		// T1 set is too small, increase P appropriately | ||||||
|  | 		delta := 1 | ||||||
|  | 		b1Len := c.b1.Len() | ||||||
|  | 		b2Len := c.b2.Len() | ||||||
|  | 		if b2Len > b1Len { | ||||||
|  | 			delta = b2Len / b1Len | ||||||
|  | 		} | ||||||
|  | 		if c.p+delta >= c.size { | ||||||
|  | 			c.p = c.size | ||||||
|  | 		} else { | ||||||
|  | 			c.p += delta | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Potentially need to make room in the cache | ||||||
|  | 		if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 			c.replace(false) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remove from B1 | ||||||
|  | 		c.b1.Remove(key) | ||||||
|  |  | ||||||
|  | 		// Add the key to the frequently used list | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Check if this value was recently evicted as part of the | ||||||
|  | 	// frequently used list | ||||||
|  | 	if c.b2.Contains(key) { | ||||||
|  | 		// T2 set is too small, decrease P appropriately | ||||||
|  | 		delta := 1 | ||||||
|  | 		b1Len := c.b1.Len() | ||||||
|  | 		b2Len := c.b2.Len() | ||||||
|  | 		if b1Len > b2Len { | ||||||
|  | 			delta = b1Len / b2Len | ||||||
|  | 		} | ||||||
|  | 		if delta >= c.p { | ||||||
|  | 			c.p = 0 | ||||||
|  | 		} else { | ||||||
|  | 			c.p -= delta | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Potentially need to make room in the cache | ||||||
|  | 		if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 			c.replace(true) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remove from B2 | ||||||
|  | 		c.b2.Remove(key) | ||||||
|  |  | ||||||
|  | 		// Add the key to the frequntly used list | ||||||
|  | 		c.t2.Add(key, value) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Potentially need to make room in the cache | ||||||
|  | 	if c.t1.Len()+c.t2.Len() >= c.size { | ||||||
|  | 		c.replace(false) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Keep the size of the ghost buffers trim | ||||||
|  | 	if c.b1.Len() > c.size-c.p { | ||||||
|  | 		c.b1.RemoveOldest() | ||||||
|  | 	} | ||||||
|  | 	if c.b2.Len() > c.p { | ||||||
|  | 		c.b2.RemoveOldest() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add to the recently seen list | ||||||
|  | 	c.t1.Add(key, value) | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // replace is used to adaptively evict from either T1 or T2 | ||||||
|  | // based on the current learned value of P | ||||||
|  | func (c *ARCCache) replace(b2ContainsKey bool) { | ||||||
|  | 	t1Len := c.t1.Len() | ||||||
|  | 	if t1Len > 0 && (t1Len > c.p || (t1Len == c.p && b2ContainsKey)) { | ||||||
|  | 		k, _, ok := c.t1.RemoveOldest() | ||||||
|  | 		if ok { | ||||||
|  | 			c.b1.Add(k, nil) | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		k, _, ok := c.t2.RemoveOldest() | ||||||
|  | 		if ok { | ||||||
|  | 			c.b2.Add(k, nil) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of cached entries | ||||||
|  | func (c *ARCCache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.t1.Len() + c.t2.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns all the cached keys | ||||||
|  | func (c *ARCCache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	k1 := c.t1.Keys() | ||||||
|  | 	k2 := c.t2.Keys() | ||||||
|  | 	return append(k1, k2...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove is used to purge a key from the cache | ||||||
|  | func (c *ARCCache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	if c.t1.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.t2.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.b1.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if c.b2.Remove(key) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to clear the cache | ||||||
|  | func (c *ARCCache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	c.t1.Purge() | ||||||
|  | 	c.t2.Purge() | ||||||
|  | 	c.b1.Purge() | ||||||
|  | 	c.b2.Purge() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Contains is used to check if the cache contains a key | ||||||
|  | // without updating recency or frequency. | ||||||
|  | func (c *ARCCache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.t1.Contains(key) || c.t2.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Peek is used to inspect the cache value of a key | ||||||
|  | // without updating recency or frequency. | ||||||
|  | func (c *ARCCache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	if val, ok := c.t1.Peek(key); ok { | ||||||
|  | 		return val, ok | ||||||
|  | 	} | ||||||
|  | 	return c.t2.Peek(key) | ||||||
|  | } | ||||||
							
								
								
									
										114
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,114 @@ | |||||||
|  | // This package provides a simple LRU cache. It is based on the | ||||||
|  | // LRU implementation in groupcache: | ||||||
|  | // https://github.com/golang/groupcache/tree/master/lru | ||||||
|  | package lru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/hashicorp/golang-lru/simplelru" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Cache is a thread-safe fixed size LRU cache. | ||||||
|  | type Cache struct { | ||||||
|  | 	lru  *simplelru.LRU | ||||||
|  | 	lock sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New creates an LRU of the given size | ||||||
|  | func New(size int) (*Cache, error) { | ||||||
|  | 	return NewWithEvict(size, nil) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewWithEvict constructs a fixed size cache with the given eviction | ||||||
|  | // callback. | ||||||
|  | func NewWithEvict(size int, onEvicted func(key interface{}, value interface{})) (*Cache, error) { | ||||||
|  | 	lru, err := simplelru.NewLRU(size, simplelru.EvictCallback(onEvicted)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	c := &Cache{ | ||||||
|  | 		lru: lru, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to completely clear the cache | ||||||
|  | func (c *Cache) Purge() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.Purge() | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||||
|  | func (c *Cache) Add(key, value interface{}) bool { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	return c.lru.Add(key, value) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *Cache) Get(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  | 	return c.lru.Get(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Check if a key is in the cache, without updating the recent-ness | ||||||
|  | // or deleting it for being stale. | ||||||
|  | func (c *Cache) Contains(key interface{}) bool { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Contains(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Returns the key value (or undefined if not found) without updating | ||||||
|  | // the "recently used"-ness of the key. | ||||||
|  | func (c *Cache) Peek(key interface{}) (interface{}, bool) { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Peek(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ContainsOrAdd checks if a key is in the cache  without updating the | ||||||
|  | // recent-ness or deleting it for being stale,  and if not, adds the value. | ||||||
|  | // Returns whether found and whether an eviction occurred. | ||||||
|  | func (c *Cache) ContainsOrAdd(key, value interface{}) (ok, evict bool) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	defer c.lock.Unlock() | ||||||
|  |  | ||||||
|  | 	if c.lru.Contains(key) { | ||||||
|  | 		return true, false | ||||||
|  | 	} else { | ||||||
|  | 		evict := c.lru.Add(key, value) | ||||||
|  | 		return false, evict | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove removes the provided key from the cache. | ||||||
|  | func (c *Cache) Remove(key interface{}) { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.Remove(key) | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RemoveOldest removes the oldest item from the cache. | ||||||
|  | func (c *Cache) RemoveOldest() { | ||||||
|  | 	c.lock.Lock() | ||||||
|  | 	c.lru.RemoveOldest() | ||||||
|  | 	c.lock.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns a slice of the keys in the cache, from oldest to newest. | ||||||
|  | func (c *Cache) Keys() []interface{} { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Keys() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of items in the cache. | ||||||
|  | func (c *Cache) Len() int { | ||||||
|  | 	c.lock.RLock() | ||||||
|  | 	defer c.lock.RUnlock() | ||||||
|  | 	return c.lru.Len() | ||||||
|  | } | ||||||
							
								
								
									
										160
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										160
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,160 @@ | |||||||
|  | package simplelru | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"container/list" | ||||||
|  | 	"errors" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // EvictCallback is used to get a callback when a cache entry is evicted | ||||||
|  | type EvictCallback func(key interface{}, value interface{}) | ||||||
|  |  | ||||||
|  | // LRU implements a non-thread safe fixed size LRU cache | ||||||
|  | type LRU struct { | ||||||
|  | 	size      int | ||||||
|  | 	evictList *list.List | ||||||
|  | 	items     map[interface{}]*list.Element | ||||||
|  | 	onEvict   EvictCallback | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // entry is used to hold a value in the evictList | ||||||
|  | type entry struct { | ||||||
|  | 	key   interface{} | ||||||
|  | 	value interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewLRU constructs an LRU of the given size | ||||||
|  | func NewLRU(size int, onEvict EvictCallback) (*LRU, error) { | ||||||
|  | 	if size <= 0 { | ||||||
|  | 		return nil, errors.New("Must provide a positive size") | ||||||
|  | 	} | ||||||
|  | 	c := &LRU{ | ||||||
|  | 		size:      size, | ||||||
|  | 		evictList: list.New(), | ||||||
|  | 		items:     make(map[interface{}]*list.Element), | ||||||
|  | 		onEvict:   onEvict, | ||||||
|  | 	} | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Purge is used to completely clear the cache | ||||||
|  | func (c *LRU) Purge() { | ||||||
|  | 	for k, v := range c.items { | ||||||
|  | 		if c.onEvict != nil { | ||||||
|  | 			c.onEvict(k, v.Value.(*entry).value) | ||||||
|  | 		} | ||||||
|  | 		delete(c.items, k) | ||||||
|  | 	} | ||||||
|  | 	c.evictList.Init() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||||
|  | func (c *LRU) Add(key, value interface{}) bool { | ||||||
|  | 	// Check for existing item | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.evictList.MoveToFront(ent) | ||||||
|  | 		ent.Value.(*entry).value = value | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Add new item | ||||||
|  | 	ent := &entry{key, value} | ||||||
|  | 	entry := c.evictList.PushFront(ent) | ||||||
|  | 	c.items[key] = entry | ||||||
|  |  | ||||||
|  | 	evict := c.evictList.Len() > c.size | ||||||
|  | 	// Verify size not exceeded | ||||||
|  | 	if evict { | ||||||
|  | 		c.removeOldest() | ||||||
|  | 	} | ||||||
|  | 	return evict | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get looks up a key's value from the cache. | ||||||
|  | func (c *LRU) Get(key interface{}) (value interface{}, ok bool) { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.evictList.MoveToFront(ent) | ||||||
|  | 		return ent.Value.(*entry).value, true | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Check if a key is in the cache, without updating the recent-ness | ||||||
|  | // or deleting it for being stale. | ||||||
|  | func (c *LRU) Contains(key interface{}) (ok bool) { | ||||||
|  | 	_, ok = c.items[key] | ||||||
|  | 	return ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Returns the key value (or undefined if not found) without updating | ||||||
|  | // the "recently used"-ness of the key. | ||||||
|  | func (c *LRU) Peek(key interface{}) (value interface{}, ok bool) { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		return ent.Value.(*entry).value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Remove removes the provided key from the cache, returning if the | ||||||
|  | // key was contained. | ||||||
|  | func (c *LRU) Remove(key interface{}) bool { | ||||||
|  | 	if ent, ok := c.items[key]; ok { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RemoveOldest removes the oldest item from the cache. | ||||||
|  | func (c *LRU) RemoveOldest() (interface{}, interface{}, bool) { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 		kv := ent.Value.(*entry) | ||||||
|  | 		return kv.key, kv.value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetOldest returns the oldest entry | ||||||
|  | func (c *LRU) GetOldest() (interface{}, interface{}, bool) { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		kv := ent.Value.(*entry) | ||||||
|  | 		return kv.key, kv.value, true | ||||||
|  | 	} | ||||||
|  | 	return nil, nil, false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Keys returns a slice of the keys in the cache, from oldest to newest. | ||||||
|  | func (c *LRU) Keys() []interface{} { | ||||||
|  | 	keys := make([]interface{}, len(c.items)) | ||||||
|  | 	i := 0 | ||||||
|  | 	for ent := c.evictList.Back(); ent != nil; ent = ent.Prev() { | ||||||
|  | 		keys[i] = ent.Value.(*entry).key | ||||||
|  | 		i++ | ||||||
|  | 	} | ||||||
|  | 	return keys | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of items in the cache. | ||||||
|  | func (c *LRU) Len() int { | ||||||
|  | 	return c.evictList.Len() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // removeOldest removes the oldest item from the cache. | ||||||
|  | func (c *LRU) removeOldest() { | ||||||
|  | 	ent := c.evictList.Back() | ||||||
|  | 	if ent != nil { | ||||||
|  | 		c.removeElement(ent) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // removeElement is used to remove a given list element from the cache | ||||||
|  | func (c *LRU) removeElement(e *list.Element) { | ||||||
|  | 	c.evictList.Remove(e) | ||||||
|  | 	kv := e.Value.(*entry) | ||||||
|  | 	delete(c.items, kv.key) | ||||||
|  | 	if c.onEvict != nil { | ||||||
|  | 		c.onEvict(kv.key, kv.value) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										18
									
								
								vendor/github.com/labstack/echo/context.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/labstack/echo/context.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -274,13 +274,6 @@ func (c *context) Param(name string) string { | |||||||
| 			if n == name { | 			if n == name { | ||||||
| 				return c.pvalues[i] | 				return c.pvalues[i] | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// Param name with aliases |  | ||||||
| 			for _, p := range strings.Split(n, ",") { |  | ||||||
| 				if p == name { |  | ||||||
| 					return c.pvalues[i] |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| @@ -494,14 +487,9 @@ func (c *context) Stream(code int, contentType string, r io.Reader) (err error) | |||||||
| } | } | ||||||
|  |  | ||||||
| func (c *context) File(file string) (err error) { | func (c *context) File(file string) (err error) { | ||||||
| 	file, err = url.QueryUnescape(file) // Issue #839 |  | ||||||
| 	if err != nil { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	f, err := os.Open(file) | 	f, err := os.Open(file) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return ErrNotFound | 		return NotFoundHandler(c) | ||||||
| 	} | 	} | ||||||
| 	defer f.Close() | 	defer f.Close() | ||||||
|  |  | ||||||
| @@ -510,7 +498,7 @@ func (c *context) File(file string) (err error) { | |||||||
| 		file = filepath.Join(file, indexPage) | 		file = filepath.Join(file, indexPage) | ||||||
| 		f, err = os.Open(file) | 		f, err = os.Open(file) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return ErrNotFound | 			return NotFoundHandler(c) | ||||||
| 		} | 		} | ||||||
| 		defer f.Close() | 		defer f.Close() | ||||||
| 		if fi, err = f.Stat(); err != nil { | 		if fi, err = f.Stat(); err != nil { | ||||||
| @@ -530,7 +518,7 @@ func (c *context) Inline(file, name string) (err error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (c *context) contentDisposition(file, name, dispositionType string) (err error) { | func (c *context) contentDisposition(file, name, dispositionType string) (err error) { | ||||||
| 	c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%s", dispositionType, name)) | 	c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%q", dispositionType, name)) | ||||||
| 	c.File(file) | 	c.File(file) | ||||||
| 	return | 	return | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										137
									
								
								vendor/github.com/labstack/echo/echo.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										137
									
								
								vendor/github.com/labstack/echo/echo.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -72,14 +72,15 @@ type ( | |||||||
| 		TLSServer        *http.Server | 		TLSServer        *http.Server | ||||||
| 		Listener         net.Listener | 		Listener         net.Listener | ||||||
| 		TLSListener      net.Listener | 		TLSListener      net.Listener | ||||||
|  | 		AutoTLSManager   autocert.Manager | ||||||
| 		DisableHTTP2     bool | 		DisableHTTP2     bool | ||||||
| 		Debug            bool | 		Debug            bool | ||||||
| 		HideBanner       bool | 		HideBanner       bool | ||||||
|  | 		HidePort         bool | ||||||
| 		HTTPErrorHandler HTTPErrorHandler | 		HTTPErrorHandler HTTPErrorHandler | ||||||
| 		Binder           Binder | 		Binder           Binder | ||||||
| 		Validator        Validator | 		Validator        Validator | ||||||
| 		Renderer         Renderer | 		Renderer         Renderer | ||||||
| 		AutoTLSManager   autocert.Manager |  | ||||||
| 		// Mutex            sync.RWMutex | 		// Mutex            sync.RWMutex | ||||||
| 		Logger Logger | 		Logger Logger | ||||||
| 	} | 	} | ||||||
| @@ -88,13 +89,14 @@ type ( | |||||||
| 	Route struct { | 	Route struct { | ||||||
| 		Method string `json:"method"` | 		Method string `json:"method"` | ||||||
| 		Path   string `json:"path"` | 		Path   string `json:"path"` | ||||||
| 		Handler string `json:"handler"` | 		Name   string `json:"name"` | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// HTTPError represents an error that occurred while handling a request. | 	// HTTPError represents an error that occurred while handling a request. | ||||||
| 	HTTPError struct { | 	HTTPError struct { | ||||||
| 		Code    int | 		Code    int | ||||||
| 		Message interface{} | 		Message interface{} | ||||||
|  | 		Inner   error // Stores the error returned by an external dependency | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// MiddlewareFunc defines a function to process middleware. | 	// MiddlewareFunc defines a function to process middleware. | ||||||
| @@ -121,7 +123,7 @@ type ( | |||||||
|  |  | ||||||
| 	// i is the interface for Echo and Group. | 	// i is the interface for Echo and Group. | ||||||
| 	i interface { | 	i interface { | ||||||
| 		GET(string, HandlerFunc, ...MiddlewareFunc) | 		GET(string, HandlerFunc, ...MiddlewareFunc) *Route | ||||||
| 	} | 	} | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -212,7 +214,7 @@ const ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	version = "3.1.0" | 	version = "3.2.6" | ||||||
| 	website = "https://echo.labstack.com" | 	website = "https://echo.labstack.com" | ||||||
| 	// http://patorjk.com/software/taag/#p=display&f=Small%20Slant&t=Echo | 	// http://patorjk.com/software/taag/#p=display&f=Small%20Slant&t=Echo | ||||||
| 	banner = ` | 	banner = ` | ||||||
| @@ -282,7 +284,7 @@ func New() (e *Echo) { | |||||||
| 	e.TLSServer.Handler = e | 	e.TLSServer.Handler = e | ||||||
| 	e.HTTPErrorHandler = e.DefaultHTTPErrorHandler | 	e.HTTPErrorHandler = e.DefaultHTTPErrorHandler | ||||||
| 	e.Binder = &DefaultBinder{} | 	e.Binder = &DefaultBinder{} | ||||||
| 	e.Logger.SetLevel(log.OFF) | 	e.Logger.SetLevel(log.ERROR) | ||||||
| 	e.stdLogger = stdLog.New(e.Logger.Output(), e.Logger.Prefix()+": ", 0) | 	e.stdLogger = stdLog.New(e.Logger.Output(), e.Logger.Prefix()+": ", 0) | ||||||
| 	e.pool.New = func() interface{} { | 	e.pool.New = func() interface{} { | ||||||
| 		return e.NewContext(nil, nil) | 		return e.NewContext(nil, nil) | ||||||
| @@ -319,6 +321,9 @@ func (e *Echo) DefaultHTTPErrorHandler(err error, c Context) { | |||||||
| 	if he, ok := err.(*HTTPError); ok { | 	if he, ok := err.(*HTTPError); ok { | ||||||
| 		code = he.Code | 		code = he.Code | ||||||
| 		msg = he.Message | 		msg = he.Message | ||||||
|  | 		if he.Inner != nil { | ||||||
|  | 			msg = fmt.Sprintf("%v, %v", err, he.Inner) | ||||||
|  | 		} | ||||||
| 	} else if e.Debug { | 	} else if e.Debug { | ||||||
| 		msg = err.Error() | 		msg = err.Error() | ||||||
| 	} else { | 	} else { | ||||||
| @@ -328,20 +333,20 @@ func (e *Echo) DefaultHTTPErrorHandler(err error, c Context) { | |||||||
| 		msg = Map{"message": msg} | 		msg = Map{"message": msg} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	e.Logger.Error(err) | ||||||
|  |  | ||||||
|  | 	// Send response | ||||||
| 	if !c.Response().Committed { | 	if !c.Response().Committed { | ||||||
| 		if c.Request().Method == HEAD { // Issue #608 | 		if c.Request().Method == HEAD { // Issue #608 | ||||||
| 			if err := c.NoContent(code); err != nil { | 			err = c.NoContent(code) | ||||||
| 				goto ERROR |  | ||||||
| 			} |  | ||||||
| 		} else { | 		} else { | ||||||
| 			if err := c.JSON(code, msg); err != nil { | 			err = c.JSON(code, msg) | ||||||
| 				goto ERROR |  | ||||||
| 		} | 		} | ||||||
| 		} | 		if err != nil { | ||||||
| 	} |  | ||||||
| ERROR: |  | ||||||
| 			e.Logger.Error(err) | 			e.Logger.Error(err) | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Pre adds middleware to the chain which is run before router. | // Pre adds middleware to the chain which is run before router. | ||||||
| func (e *Echo) Pre(middleware ...MiddlewareFunc) { | func (e *Echo) Pre(middleware ...MiddlewareFunc) { | ||||||
| @@ -355,104 +360,114 @@ func (e *Echo) Use(middleware ...MiddlewareFunc) { | |||||||
|  |  | ||||||
| // CONNECT registers a new CONNECT route for a path with matching handler in the | // CONNECT registers a new CONNECT route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(CONNECT, path, h, m...) | 	return e.Add(CONNECT, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // DELETE registers a new DELETE route for a path with matching handler in the router | // DELETE registers a new DELETE route for a path with matching handler in the router | ||||||
| // with optional route-level middleware. | // with optional route-level middleware. | ||||||
| func (e *Echo) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(DELETE, path, h, m...) | 	return e.Add(DELETE, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // GET registers a new GET route for a path with matching handler in the router | // GET registers a new GET route for a path with matching handler in the router | ||||||
| // with optional route-level middleware. | // with optional route-level middleware. | ||||||
| func (e *Echo) GET(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) GET(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(GET, path, h, m...) | 	return e.Add(GET, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // HEAD registers a new HEAD route for a path with matching handler in the | // HEAD registers a new HEAD route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(HEAD, path, h, m...) | 	return e.Add(HEAD, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // OPTIONS registers a new OPTIONS route for a path with matching handler in the | // OPTIONS registers a new OPTIONS route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(OPTIONS, path, h, m...) | 	return e.Add(OPTIONS, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // PATCH registers a new PATCH route for a path with matching handler in the | // PATCH registers a new PATCH route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(PATCH, path, h, m...) | 	return e.Add(PATCH, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // POST registers a new POST route for a path with matching handler in the | // POST registers a new POST route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) POST(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) POST(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(POST, path, h, m...) | 	return e.Add(POST, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // PUT registers a new PUT route for a path with matching handler in the | // PUT registers a new PUT route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(PUT, path, h, m...) | 	return e.Add(PUT, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // TRACE registers a new TRACE route for a path with matching handler in the | // TRACE registers a new TRACE route for a path with matching handler in the | ||||||
| // router with optional route-level middleware. | // router with optional route-level middleware. | ||||||
| func (e *Echo) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (e *Echo) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	e.add(TRACE, path, h, m...) | 	return e.Add(TRACE, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // Any registers a new route for all HTTP methods and path with matching handler | // Any registers a new route for all HTTP methods and path with matching handler | ||||||
| // in the router with optional route-level middleware. | // in the router with optional route-level middleware. | ||||||
| func (e *Echo) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | func (e *Echo) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route { | ||||||
| 	for _, m := range methods { | 	routes := make([]*Route, len(methods)) | ||||||
| 		e.add(m, path, handler, middleware...) | 	for i, m := range methods { | ||||||
|  | 		routes[i] = e.Add(m, path, handler, middleware...) | ||||||
| 	} | 	} | ||||||
|  | 	return routes | ||||||
| } | } | ||||||
|  |  | ||||||
| // Match registers a new route for multiple HTTP methods and path with matching | // Match registers a new route for multiple HTTP methods and path with matching | ||||||
| // handler in the router with optional route-level middleware. | // handler in the router with optional route-level middleware. | ||||||
| func (e *Echo) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | func (e *Echo) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route { | ||||||
| 	for _, m := range methods { | 	routes := make([]*Route, len(methods)) | ||||||
| 		e.add(m, path, handler, middleware...) | 	for i, m := range methods { | ||||||
|  | 		routes[i] = e.Add(m, path, handler, middleware...) | ||||||
| 	} | 	} | ||||||
|  | 	return routes | ||||||
| } | } | ||||||
|  |  | ||||||
| // Static registers a new route with path prefix to serve static files from the | // Static registers a new route with path prefix to serve static files from the | ||||||
| // provided root directory. | // provided root directory. | ||||||
| func (e *Echo) Static(prefix, root string) { | func (e *Echo) Static(prefix, root string) *Route { | ||||||
| 	if root == "" { | 	if root == "" { | ||||||
| 		root = "." // For security we want to restrict to CWD. | 		root = "." // For security we want to restrict to CWD. | ||||||
| 	} | 	} | ||||||
| 	static(e, prefix, root) | 	return static(e, prefix, root) | ||||||
| } | } | ||||||
|  |  | ||||||
| func static(i i, prefix, root string) { | func static(i i, prefix, root string) *Route { | ||||||
| 	h := func(c Context) error { | 	h := func(c Context) error { | ||||||
| 		name := filepath.Join(root, path.Clean("/"+c.Param("*"))) // "/"+ for security | 		p, err := PathUnescape(c.Param("*")) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		name := filepath.Join(root, path.Clean("/"+p)) // "/"+ for security | ||||||
| 		return c.File(name) | 		return c.File(name) | ||||||
| 	} | 	} | ||||||
| 	i.GET(prefix, h) | 	i.GET(prefix, h) | ||||||
| 	if prefix == "/" { | 	if prefix == "/" { | ||||||
| 		i.GET(prefix+"*", h) | 		return i.GET(prefix+"*", h) | ||||||
| 	} else { |  | ||||||
| 		i.GET(prefix+"/*", h) |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	return i.GET(prefix+"/*", h) | ||||||
| } | } | ||||||
|  |  | ||||||
| // File registers a new route with path to serve a static file. | // File registers a new route with path to serve a static file. | ||||||
| func (e *Echo) File(path, file string) { | func (e *Echo) File(path, file string) *Route { | ||||||
| 	e.GET(path, func(c Context) error { | 	return e.GET(path, func(c Context) error { | ||||||
| 		return c.File(file) | 		return c.File(file) | ||||||
| 	}) | 	}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (e *Echo) add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | // Add registers a new route for an HTTP method and path with matching handler | ||||||
|  | // in the router with optional route-level middleware. | ||||||
|  | func (e *Echo) Add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) *Route { | ||||||
| 	name := handlerName(handler) | 	name := handlerName(handler) | ||||||
| 	e.router.Add(method, path, func(c Context) error { | 	e.router.Add(method, path, func(c Context) error { | ||||||
| 		h := handler | 		h := handler | ||||||
| @@ -465,9 +480,10 @@ func (e *Echo) add(method, path string, handler HandlerFunc, middleware ...Middl | |||||||
| 	r := &Route{ | 	r := &Route{ | ||||||
| 		Method: method, | 		Method: method, | ||||||
| 		Path:   path, | 		Path:   path, | ||||||
| 		Handler: name, | 		Name:   name, | ||||||
| 	} | 	} | ||||||
| 	e.router.routes[method+path] = r | 	e.router.routes[method+path] = r | ||||||
|  | 	return r | ||||||
| } | } | ||||||
|  |  | ||||||
| // Group creates a new router group with prefix and optional group-level middleware. | // Group creates a new router group with prefix and optional group-level middleware. | ||||||
| @@ -479,12 +495,22 @@ func (e *Echo) Group(prefix string, m ...MiddlewareFunc) (g *Group) { | |||||||
|  |  | ||||||
| // URI generates a URI from handler. | // URI generates a URI from handler. | ||||||
| func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string { | func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string { | ||||||
|  | 	name := handlerName(handler) | ||||||
|  | 	return e.Reverse(name, params...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // URL is an alias for `URI` function. | ||||||
|  | func (e *Echo) URL(h HandlerFunc, params ...interface{}) string { | ||||||
|  | 	return e.URI(h, params...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Reverse generates an URL from route name and provided parameters. | ||||||
|  | func (e *Echo) Reverse(name string, params ...interface{}) string { | ||||||
| 	uri := new(bytes.Buffer) | 	uri := new(bytes.Buffer) | ||||||
| 	ln := len(params) | 	ln := len(params) | ||||||
| 	n := 0 | 	n := 0 | ||||||
| 	name := handlerName(handler) |  | ||||||
| 	for _, r := range e.router.routes { | 	for _, r := range e.router.routes { | ||||||
| 		if r.Handler == name { | 		if r.Name == name { | ||||||
| 			for i, l := 0, len(r.Path); i < l; i++ { | 			for i, l := 0, len(r.Path); i < l; i++ { | ||||||
| 				if r.Path[i] == ':' && n < ln { | 				if r.Path[i] == ':' && n < ln { | ||||||
| 					for ; i < l && r.Path[i] != '/'; i++ { | 					for ; i < l && r.Path[i] != '/'; i++ { | ||||||
| @@ -502,11 +528,6 @@ func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string { | |||||||
| 	return uri.String() | 	return uri.String() | ||||||
| } | } | ||||||
|  |  | ||||||
| // URL is an alias for `URI` function. |  | ||||||
| func (e *Echo) URL(h HandlerFunc, params ...interface{}) string { |  | ||||||
| 	return e.URI(h, params...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Routes returns the registered routes. | // Routes returns the registered routes. | ||||||
| func (e *Echo) Routes() []*Route { | func (e *Echo) Routes() []*Route { | ||||||
| 	routes := []*Route{} | 	routes := []*Route{} | ||||||
| @@ -624,7 +645,7 @@ func (e *Echo) StartServer(s *http.Server) (err error) { | |||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		if !e.HideBanner { | 		if !e.HidePort { | ||||||
| 			e.colorer.Printf("⇨ http server started on %s\n", e.colorer.Green(e.Listener.Addr())) | 			e.colorer.Printf("⇨ http server started on %s\n", e.colorer.Green(e.Listener.Addr())) | ||||||
| 		} | 		} | ||||||
| 		return s.Serve(e.Listener) | 		return s.Serve(e.Listener) | ||||||
| @@ -636,7 +657,7 @@ func (e *Echo) StartServer(s *http.Server) (err error) { | |||||||
| 		} | 		} | ||||||
| 		e.TLSListener = tls.NewListener(l, s.TLSConfig) | 		e.TLSListener = tls.NewListener(l, s.TLSConfig) | ||||||
| 	} | 	} | ||||||
| 	if !e.HideBanner { | 	if !e.HidePort { | ||||||
| 		e.colorer.Printf("⇨ https server started on %s\n", e.colorer.Green(e.TLSListener.Addr())) | 		e.colorer.Printf("⇨ https server started on %s\n", e.colorer.Green(e.TLSListener.Addr())) | ||||||
| 	} | 	} | ||||||
| 	return s.Serve(e.TLSListener) | 	return s.Serve(e.TLSListener) | ||||||
| @@ -653,7 +674,7 @@ func NewHTTPError(code int, message ...interface{}) *HTTPError { | |||||||
|  |  | ||||||
| // Error makes it compatible with `error` interface. | // Error makes it compatible with `error` interface. | ||||||
| func (he *HTTPError) Error() string { | func (he *HTTPError) Error() string { | ||||||
| 	return fmt.Sprintf("code=%d, message=%s", he.Code, he.Message) | 	return fmt.Sprintf("code=%d, message=%v", he.Code, he.Message) | ||||||
| } | } | ||||||
|  |  | ||||||
| // WrapHandler wraps `http.Handler` into `echo.HandlerFunc`. | // WrapHandler wraps `http.Handler` into `echo.HandlerFunc`. | ||||||
|   | |||||||
							
								
								
									
										63
									
								
								vendor/github.com/labstack/echo/group.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										63
									
								
								vendor/github.com/labstack/echo/group.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -20,68 +20,74 @@ func (g *Group) Use(middleware ...MiddlewareFunc) { | |||||||
| 	g.middleware = append(g.middleware, middleware...) | 	g.middleware = append(g.middleware, middleware...) | ||||||
| 	// Allow all requests to reach the group as they might get dropped if router | 	// Allow all requests to reach the group as they might get dropped if router | ||||||
| 	// doesn't find a match, making none of the group middleware process. | 	// doesn't find a match, making none of the group middleware process. | ||||||
| 	g.echo.Any(path.Clean(g.prefix+"/*"), func(c Context) error { | 	for _, p := range []string{"", "/*"} { | ||||||
| 		return ErrNotFound | 		g.echo.Any(path.Clean(g.prefix+p), func(c Context) error { | ||||||
|  | 			return NotFoundHandler(c) | ||||||
| 		}, g.middleware...) | 		}, g.middleware...) | ||||||
| 	} | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // CONNECT implements `Echo#CONNECT()` for sub-routes within the Group. | // CONNECT implements `Echo#CONNECT()` for sub-routes within the Group. | ||||||
| func (g *Group) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(CONNECT, path, h, m...) | 	return g.Add(CONNECT, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // DELETE implements `Echo#DELETE()` for sub-routes within the Group. | // DELETE implements `Echo#DELETE()` for sub-routes within the Group. | ||||||
| func (g *Group) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(DELETE, path, h, m...) | 	return g.Add(DELETE, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // GET implements `Echo#GET()` for sub-routes within the Group. | // GET implements `Echo#GET()` for sub-routes within the Group. | ||||||
| func (g *Group) GET(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) GET(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(GET, path, h, m...) | 	return g.Add(GET, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // HEAD implements `Echo#HEAD()` for sub-routes within the Group. | // HEAD implements `Echo#HEAD()` for sub-routes within the Group. | ||||||
| func (g *Group) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(HEAD, path, h, m...) | 	return g.Add(HEAD, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // OPTIONS implements `Echo#OPTIONS()` for sub-routes within the Group. | // OPTIONS implements `Echo#OPTIONS()` for sub-routes within the Group. | ||||||
| func (g *Group) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(OPTIONS, path, h, m...) | 	return g.Add(OPTIONS, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // PATCH implements `Echo#PATCH()` for sub-routes within the Group. | // PATCH implements `Echo#PATCH()` for sub-routes within the Group. | ||||||
| func (g *Group) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(PATCH, path, h, m...) | 	return g.Add(PATCH, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // POST implements `Echo#POST()` for sub-routes within the Group. | // POST implements `Echo#POST()` for sub-routes within the Group. | ||||||
| func (g *Group) POST(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) POST(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(POST, path, h, m...) | 	return g.Add(POST, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // PUT implements `Echo#PUT()` for sub-routes within the Group. | // PUT implements `Echo#PUT()` for sub-routes within the Group. | ||||||
| func (g *Group) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(PUT, path, h, m...) | 	return g.Add(PUT, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // TRACE implements `Echo#TRACE()` for sub-routes within the Group. | // TRACE implements `Echo#TRACE()` for sub-routes within the Group. | ||||||
| func (g *Group) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) { | func (g *Group) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route { | ||||||
| 	g.add(TRACE, path, h, m...) | 	return g.Add(TRACE, path, h, m...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // Any implements `Echo#Any()` for sub-routes within the Group. | // Any implements `Echo#Any()` for sub-routes within the Group. | ||||||
| func (g *Group) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | func (g *Group) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route { | ||||||
| 	for _, m := range methods { | 	routes := make([]*Route, len(methods)) | ||||||
| 		g.add(m, path, handler, middleware...) | 	for i, m := range methods { | ||||||
|  | 		routes[i] = g.Add(m, path, handler, middleware...) | ||||||
| 	} | 	} | ||||||
|  | 	return routes | ||||||
| } | } | ||||||
|  |  | ||||||
| // Match implements `Echo#Match()` for sub-routes within the Group. | // Match implements `Echo#Match()` for sub-routes within the Group. | ||||||
| func (g *Group) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | func (g *Group) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route { | ||||||
| 	for _, m := range methods { | 	routes := make([]*Route, len(methods)) | ||||||
| 		g.add(m, path, handler, middleware...) | 	for i, m := range methods { | ||||||
|  | 		routes[i] = g.Add(m, path, handler, middleware...) | ||||||
| 	} | 	} | ||||||
|  | 	return routes | ||||||
| } | } | ||||||
|  |  | ||||||
| // Group creates a new sub-group with prefix and optional sub-group-level middleware. | // Group creates a new sub-group with prefix and optional sub-group-level middleware. | ||||||
| @@ -102,12 +108,13 @@ func (g *Group) File(path, file string) { | |||||||
| 	g.echo.File(g.prefix+path, file) | 	g.echo.File(g.prefix+path, file) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (g *Group) add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) { | // Add implements `Echo#Add()` for sub-routes within the Group. | ||||||
|  | func (g *Group) Add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) *Route { | ||||||
| 	// Combine into a new slice to avoid accidentally passing the same slice for | 	// Combine into a new slice to avoid accidentally passing the same slice for | ||||||
| 	// multiple routes, which would lead to later add() calls overwriting the | 	// multiple routes, which would lead to later add() calls overwriting the | ||||||
| 	// middleware from earlier calls. | 	// middleware from earlier calls. | ||||||
| 	m := []MiddlewareFunc{} | 	m := []MiddlewareFunc{} | ||||||
| 	m = append(m, g.middleware...) | 	m = append(m, g.middleware...) | ||||||
| 	m = append(m, middleware...) | 	m = append(m, middleware...) | ||||||
| 	g.echo.add(method, g.prefix+path, handler, m...) | 	return g.echo.Add(method, g.prefix+path, handler, m...) | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										8
									
								
								vendor/github.com/labstack/echo/middleware/basic_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								vendor/github.com/labstack/echo/middleware/basic_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,7 @@ package middleware | |||||||
| import ( | import ( | ||||||
| 	"encoding/base64" | 	"encoding/base64" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
| 	"github.com/labstack/echo" | 	"github.com/labstack/echo" | ||||||
| ) | ) | ||||||
| @@ -27,7 +28,7 @@ type ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	basic        = "Basic" | 	basic        = "basic" | ||||||
| 	defaultRealm = "Restricted" | 	defaultRealm = "Restricted" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -54,7 +55,7 @@ func BasicAuth(fn BasicAuthValidator) echo.MiddlewareFunc { | |||||||
| func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc { | func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc { | ||||||
| 	// Defaults | 	// Defaults | ||||||
| 	if config.Validator == nil { | 	if config.Validator == nil { | ||||||
| 		panic("basic-auth middleware requires a validator function") | 		panic("echo: basic-auth middleware requires a validator function") | ||||||
| 	} | 	} | ||||||
| 	if config.Skipper == nil { | 	if config.Skipper == nil { | ||||||
| 		config.Skipper = DefaultBasicAuthConfig.Skipper | 		config.Skipper = DefaultBasicAuthConfig.Skipper | ||||||
| @@ -72,7 +73,7 @@ func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc { | |||||||
| 			auth := c.Request().Header.Get(echo.HeaderAuthorization) | 			auth := c.Request().Header.Get(echo.HeaderAuthorization) | ||||||
| 			l := len(basic) | 			l := len(basic) | ||||||
|  |  | ||||||
| 			if len(auth) > l+1 && auth[:l] == basic { | 			if len(auth) > l+1 && strings.ToLower(auth[:l]) == basic { | ||||||
| 				b, err := base64.StdEncoding.DecodeString(auth[l+1:]) | 				b, err := base64.StdEncoding.DecodeString(auth[l+1:]) | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| 					return err | 					return err | ||||||
| @@ -87,6 +88,7 @@ func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc { | |||||||
| 						} else if valid { | 						} else if valid { | ||||||
| 							return next(c) | 							return next(c) | ||||||
| 						} | 						} | ||||||
|  | 						break | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
|   | |||||||
							
								
								
									
										112
									
								
								vendor/github.com/labstack/echo/middleware/body_dump.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										112
									
								
								vendor/github.com/labstack/echo/middleware/body_dump.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,112 @@ | |||||||
|  | package middleware | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bufio" | ||||||
|  | 	"bytes" | ||||||
|  | 	"io/ioutil" | ||||||
|  | 	"net" | ||||||
|  | 	"net/http" | ||||||
|  |  | ||||||
|  | 	"io" | ||||||
|  |  | ||||||
|  | 	"github.com/labstack/echo" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	// BodyDumpConfig defines the config for BodyDump middleware. | ||||||
|  | 	BodyDumpConfig struct { | ||||||
|  | 		// Skipper defines a function to skip middleware. | ||||||
|  | 		Skipper Skipper | ||||||
|  |  | ||||||
|  | 		// Handler receives request and response payload. | ||||||
|  | 		// Required. | ||||||
|  | 		Handler BodyDumpHandler | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// BodyDumpHandler receives the request and response payload. | ||||||
|  | 	BodyDumpHandler func(echo.Context, []byte, []byte) | ||||||
|  |  | ||||||
|  | 	bodyDumpResponseWriter struct { | ||||||
|  | 		io.Writer | ||||||
|  | 		http.ResponseWriter | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// DefaultBodyDumpConfig is the default BodyDump middleware config. | ||||||
|  | 	DefaultBodyDumpConfig = BodyDumpConfig{ | ||||||
|  | 		Skipper: DefaultSkipper, | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // BodyDump returns a BodyDump middleware. | ||||||
|  | // | ||||||
|  | // BodyLimit middleware captures the request and response payload and calls the | ||||||
|  | // registered handler. | ||||||
|  | func BodyDump(handler BodyDumpHandler) echo.MiddlewareFunc { | ||||||
|  | 	c := DefaultBodyDumpConfig | ||||||
|  | 	c.Handler = handler | ||||||
|  | 	return BodyDumpWithConfig(c) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // BodyDumpWithConfig returns a BodyDump middleware with config. | ||||||
|  | // See: `BodyDump()`. | ||||||
|  | func BodyDumpWithConfig(config BodyDumpConfig) echo.MiddlewareFunc { | ||||||
|  | 	// Defaults | ||||||
|  | 	if config.Handler == nil { | ||||||
|  | 		panic("echo: body-dump middleware requires a handler function") | ||||||
|  | 	} | ||||||
|  | 	if config.Skipper == nil { | ||||||
|  | 		config.Skipper = DefaultBodyDumpConfig.Skipper | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return func(next echo.HandlerFunc) echo.HandlerFunc { | ||||||
|  | 		return func(c echo.Context) (err error) { | ||||||
|  | 			if config.Skipper(c) { | ||||||
|  | 				return next(c) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Request | ||||||
|  | 			reqBody := []byte{} | ||||||
|  | 			if c.Request().Body != nil { // Read | ||||||
|  | 				reqBody, _ = ioutil.ReadAll(c.Request().Body) | ||||||
|  | 			} | ||||||
|  | 			c.Request().Body = ioutil.NopCloser(bytes.NewBuffer(reqBody)) // Reset | ||||||
|  |  | ||||||
|  | 			// Response | ||||||
|  | 			resBody := new(bytes.Buffer) | ||||||
|  | 			mw := io.MultiWriter(c.Response().Writer, resBody) | ||||||
|  | 			writer := &bodyDumpResponseWriter{Writer: mw, ResponseWriter: c.Response().Writer} | ||||||
|  | 			c.Response().Writer = writer | ||||||
|  |  | ||||||
|  | 			if err = next(c); err != nil { | ||||||
|  | 				c.Error(err) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Callback | ||||||
|  | 			config.Handler(c, reqBody, resBody.Bytes()) | ||||||
|  |  | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *bodyDumpResponseWriter) WriteHeader(code int) { | ||||||
|  | 	w.ResponseWriter.WriteHeader(code) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *bodyDumpResponseWriter) Write(b []byte) (int, error) { | ||||||
|  | 	return w.Writer.Write(b) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *bodyDumpResponseWriter) Flush() { | ||||||
|  | 	w.ResponseWriter.(http.Flusher).Flush() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *bodyDumpResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { | ||||||
|  | 	return w.ResponseWriter.(http.Hijacker).Hijack() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *bodyDumpResponseWriter) CloseNotify() <-chan bool { | ||||||
|  | 	return w.ResponseWriter.(http.CloseNotifier).CloseNotify() | ||||||
|  | } | ||||||
							
								
								
									
										6
									
								
								vendor/github.com/labstack/echo/middleware/body_limit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								vendor/github.com/labstack/echo/middleware/body_limit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -17,7 +17,7 @@ type ( | |||||||
|  |  | ||||||
| 		// Maximum allowed size for a request body, it can be specified | 		// Maximum allowed size for a request body, it can be specified | ||||||
| 		// as `4x` or `4xB`, where x is one of the multiple from K, M, G, T or P. | 		// as `4x` or `4xB`, where x is one of the multiple from K, M, G, T or P. | ||||||
| 		Limit string `json:"limit"` | 		Limit string `yaml:"limit"` | ||||||
| 		limit int64 | 		limit int64 | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -30,7 +30,7 @@ type ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultBodyLimitConfig is the default Gzip middleware config. | 	// DefaultBodyLimitConfig is the default BodyLimit middleware config. | ||||||
| 	DefaultBodyLimitConfig = BodyLimitConfig{ | 	DefaultBodyLimitConfig = BodyLimitConfig{ | ||||||
| 		Skipper: DefaultSkipper, | 		Skipper: DefaultSkipper, | ||||||
| 	} | 	} | ||||||
| @@ -60,7 +60,7 @@ func BodyLimitWithConfig(config BodyLimitConfig) echo.MiddlewareFunc { | |||||||
|  |  | ||||||
| 	limit, err := bytes.Parse(config.Limit) | 	limit, err := bytes.Parse(config.Limit) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		panic(fmt.Errorf("invalid body-limit=%s", config.Limit)) | 		panic(fmt.Errorf("echo: invalid body-limit=%s", config.Limit)) | ||||||
| 	} | 	} | ||||||
| 	config.limit = limit | 	config.limit = limit | ||||||
| 	pool := limitedReaderPool(config) | 	pool := limitedReaderPool(config) | ||||||
|   | |||||||
							
								
								
									
										5
									
								
								vendor/github.com/labstack/echo/middleware/compress.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										5
									
								
								vendor/github.com/labstack/echo/middleware/compress.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -20,7 +20,7 @@ type ( | |||||||
|  |  | ||||||
| 		// Gzip compression level. | 		// Gzip compression level. | ||||||
| 		// Optional. Default value -1. | 		// Optional. Default value -1. | ||||||
| 		Level int `json:"level"` | 		Level int `yaml:"level"` | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	gzipResponseWriter struct { | 	gzipResponseWriter struct { | ||||||
| @@ -67,7 +67,7 @@ func GzipWithConfig(config GzipConfig) echo.MiddlewareFunc { | |||||||
| 			res := c.Response() | 			res := c.Response() | ||||||
| 			res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding) | 			res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding) | ||||||
| 			if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) { | 			if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) { | ||||||
| 				res.Header().Add(echo.HeaderContentEncoding, gzipScheme) // Issue #806 | 				res.Header().Set(echo.HeaderContentEncoding, gzipScheme) // Issue #806 | ||||||
| 				rw := res.Writer | 				rw := res.Writer | ||||||
| 				w, err := gzip.NewWriterLevel(rw, config.Level) | 				w, err := gzip.NewWriterLevel(rw, config.Level) | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| @@ -98,6 +98,7 @@ func (w *gzipResponseWriter) WriteHeader(code int) { | |||||||
| 	if code == http.StatusNoContent { // Issue #489 | 	if code == http.StatusNoContent { // Issue #489 | ||||||
| 		w.ResponseWriter.Header().Del(echo.HeaderContentEncoding) | 		w.ResponseWriter.Header().Del(echo.HeaderContentEncoding) | ||||||
| 	} | 	} | ||||||
|  | 	w.Header().Del(echo.HeaderContentLength) // Issue #444 | ||||||
| 	w.ResponseWriter.WriteHeader(code) | 	w.ResponseWriter.WriteHeader(code) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										12
									
								
								vendor/github.com/labstack/echo/middleware/cors.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										12
									
								
								vendor/github.com/labstack/echo/middleware/cors.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -16,34 +16,34 @@ type ( | |||||||
|  |  | ||||||
| 		// AllowOrigin defines a list of origins that may access the resource. | 		// AllowOrigin defines a list of origins that may access the resource. | ||||||
| 		// Optional. Default value []string{"*"}. | 		// Optional. Default value []string{"*"}. | ||||||
| 		AllowOrigins []string `json:"allow_origins"` | 		AllowOrigins []string `yaml:"allow_origins"` | ||||||
|  |  | ||||||
| 		// AllowMethods defines a list methods allowed when accessing the resource. | 		// AllowMethods defines a list methods allowed when accessing the resource. | ||||||
| 		// This is used in response to a preflight request. | 		// This is used in response to a preflight request. | ||||||
| 		// Optional. Default value DefaultCORSConfig.AllowMethods. | 		// Optional. Default value DefaultCORSConfig.AllowMethods. | ||||||
| 		AllowMethods []string `json:"allow_methods"` | 		AllowMethods []string `yaml:"allow_methods"` | ||||||
|  |  | ||||||
| 		// AllowHeaders defines a list of request headers that can be used when | 		// AllowHeaders defines a list of request headers that can be used when | ||||||
| 		// making the actual request. This in response to a preflight request. | 		// making the actual request. This in response to a preflight request. | ||||||
| 		// Optional. Default value []string{}. | 		// Optional. Default value []string{}. | ||||||
| 		AllowHeaders []string `json:"allow_headers"` | 		AllowHeaders []string `yaml:"allow_headers"` | ||||||
|  |  | ||||||
| 		// AllowCredentials indicates whether or not the response to the request | 		// AllowCredentials indicates whether or not the response to the request | ||||||
| 		// can be exposed when the credentials flag is true. When used as part of | 		// can be exposed when the credentials flag is true. When used as part of | ||||||
| 		// a response to a preflight request, this indicates whether or not the | 		// a response to a preflight request, this indicates whether or not the | ||||||
| 		// actual request can be made using credentials. | 		// actual request can be made using credentials. | ||||||
| 		// Optional. Default value false. | 		// Optional. Default value false. | ||||||
| 		AllowCredentials bool `json:"allow_credentials"` | 		AllowCredentials bool `yaml:"allow_credentials"` | ||||||
|  |  | ||||||
| 		// ExposeHeaders defines a whitelist headers that clients are allowed to | 		// ExposeHeaders defines a whitelist headers that clients are allowed to | ||||||
| 		// access. | 		// access. | ||||||
| 		// Optional. Default value []string{}. | 		// Optional. Default value []string{}. | ||||||
| 		ExposeHeaders []string `json:"expose_headers"` | 		ExposeHeaders []string `yaml:"expose_headers"` | ||||||
|  |  | ||||||
| 		// MaxAge indicates how long (in seconds) the results of a preflight request | 		// MaxAge indicates how long (in seconds) the results of a preflight request | ||||||
| 		// can be cached. | 		// can be cached. | ||||||
| 		// Optional. Default value 0. | 		// Optional. Default value 0. | ||||||
| 		MaxAge int `json:"max_age"` | 		MaxAge int `yaml:"max_age"` | ||||||
| 	} | 	} | ||||||
| ) | ) | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										18
									
								
								vendor/github.com/labstack/echo/middleware/csrf.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/labstack/echo/middleware/csrf.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -18,7 +18,7 @@ type ( | |||||||
| 		Skipper Skipper | 		Skipper Skipper | ||||||
|  |  | ||||||
| 		// TokenLength is the length of the generated token. | 		// TokenLength is the length of the generated token. | ||||||
| 		TokenLength uint8 `json:"token_length"` | 		TokenLength uint8 `yaml:"token_length"` | ||||||
| 		// Optional. Default value 32. | 		// Optional. Default value 32. | ||||||
|  |  | ||||||
| 		// TokenLookup is a string in the form of "<source>:<key>" that is used | 		// TokenLookup is a string in the form of "<source>:<key>" that is used | ||||||
| @@ -28,35 +28,35 @@ type ( | |||||||
| 		// - "header:<name>" | 		// - "header:<name>" | ||||||
| 		// - "form:<name>" | 		// - "form:<name>" | ||||||
| 		// - "query:<name>" | 		// - "query:<name>" | ||||||
| 		TokenLookup string `json:"token_lookup"` | 		TokenLookup string `yaml:"token_lookup"` | ||||||
|  |  | ||||||
| 		// Context key to store generated CSRF token into context. | 		// Context key to store generated CSRF token into context. | ||||||
| 		// Optional. Default value "csrf". | 		// Optional. Default value "csrf". | ||||||
| 		ContextKey string `json:"context_key"` | 		ContextKey string `yaml:"context_key"` | ||||||
|  |  | ||||||
| 		// Name of the CSRF cookie. This cookie will store CSRF token. | 		// Name of the CSRF cookie. This cookie will store CSRF token. | ||||||
| 		// Optional. Default value "csrf". | 		// Optional. Default value "csrf". | ||||||
| 		CookieName string `json:"cookie_name"` | 		CookieName string `yaml:"cookie_name"` | ||||||
|  |  | ||||||
| 		// Domain of the CSRF cookie. | 		// Domain of the CSRF cookie. | ||||||
| 		// Optional. Default value none. | 		// Optional. Default value none. | ||||||
| 		CookieDomain string `json:"cookie_domain"` | 		CookieDomain string `yaml:"cookie_domain"` | ||||||
|  |  | ||||||
| 		// Path of the CSRF cookie. | 		// Path of the CSRF cookie. | ||||||
| 		// Optional. Default value none. | 		// Optional. Default value none. | ||||||
| 		CookiePath string `json:"cookie_path"` | 		CookiePath string `yaml:"cookie_path"` | ||||||
|  |  | ||||||
| 		// Max age (in seconds) of the CSRF cookie. | 		// Max age (in seconds) of the CSRF cookie. | ||||||
| 		// Optional. Default value 86400 (24hr). | 		// Optional. Default value 86400 (24hr). | ||||||
| 		CookieMaxAge int `json:"cookie_max_age"` | 		CookieMaxAge int `yaml:"cookie_max_age"` | ||||||
|  |  | ||||||
| 		// Indicates if CSRF cookie is secure. | 		// Indicates if CSRF cookie is secure. | ||||||
| 		// Optional. Default value false. | 		// Optional. Default value false. | ||||||
| 		CookieSecure bool `json:"cookie_secure"` | 		CookieSecure bool `yaml:"cookie_secure"` | ||||||
|  |  | ||||||
| 		// Indicates if CSRF cookie is HTTP only. | 		// Indicates if CSRF cookie is HTTP only. | ||||||
| 		// Optional. Default value false. | 		// Optional. Default value false. | ||||||
| 		CookieHTTPOnly bool `json:"cookie_http_only"` | 		CookieHTTPOnly bool `yaml:"cookie_http_only"` | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// csrfTokenExtractor defines a function that takes `echo.Context` and returns | 	// csrfTokenExtractor defines a function that takes `echo.Context` and returns | ||||||
|   | |||||||
							
								
								
									
										26
									
								
								vendor/github.com/labstack/echo/middleware/jwt.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										26
									
								
								vendor/github.com/labstack/echo/middleware/jwt.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,7 +1,6 @@ | |||||||
| package middleware | package middleware | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" |  | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"reflect" | 	"reflect" | ||||||
| @@ -57,6 +56,12 @@ const ( | |||||||
| 	AlgorithmHS256 = "HS256" | 	AlgorithmHS256 = "HS256" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Errors | ||||||
|  | var ( | ||||||
|  | 	ErrJWTMissing = echo.NewHTTPError(http.StatusBadRequest, "Missing or malformed jwt") | ||||||
|  | 	ErrJWTInvalid = echo.NewHTTPError(http.StatusUnauthorized, "Invalid or expired jwt") | ||||||
|  | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultJWTConfig is the default JWT auth middleware config. | 	// DefaultJWTConfig is the default JWT auth middleware config. | ||||||
| 	DefaultJWTConfig = JWTConfig{ | 	DefaultJWTConfig = JWTConfig{ | ||||||
| @@ -77,7 +82,7 @@ var ( | |||||||
| // | // | ||||||
| // See: https://jwt.io/introduction | // See: https://jwt.io/introduction | ||||||
| // See `JWTConfig.TokenLookup` | // See `JWTConfig.TokenLookup` | ||||||
| func JWT(key []byte) echo.MiddlewareFunc { | func JWT(key interface{}) echo.MiddlewareFunc { | ||||||
| 	c := DefaultJWTConfig | 	c := DefaultJWTConfig | ||||||
| 	c.SigningKey = key | 	c.SigningKey = key | ||||||
| 	return JWTWithConfig(c) | 	return JWTWithConfig(c) | ||||||
| @@ -134,14 +139,15 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc { | |||||||
|  |  | ||||||
| 			auth, err := extractor(c) | 			auth, err := extractor(c) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return echo.NewHTTPError(http.StatusBadRequest, err.Error()) | 				return err | ||||||
| 			} | 			} | ||||||
| 			token := new(jwt.Token) | 			token := new(jwt.Token) | ||||||
| 			// Issue #647, #656 | 			// Issue #647, #656 | ||||||
| 			if _, ok := config.Claims.(jwt.MapClaims); ok { | 			if _, ok := config.Claims.(jwt.MapClaims); ok { | ||||||
| 				token, err = jwt.Parse(auth, config.keyFunc) | 				token, err = jwt.Parse(auth, config.keyFunc) | ||||||
| 			} else { | 			} else { | ||||||
| 				claims := reflect.ValueOf(config.Claims).Interface().(jwt.Claims) | 				t := reflect.ValueOf(config.Claims).Type().Elem() | ||||||
|  | 				claims := reflect.New(t).Interface().(jwt.Claims) | ||||||
| 				token, err = jwt.ParseWithClaims(auth, claims, config.keyFunc) | 				token, err = jwt.ParseWithClaims(auth, claims, config.keyFunc) | ||||||
| 			} | 			} | ||||||
| 			if err == nil && token.Valid { | 			if err == nil && token.Valid { | ||||||
| @@ -149,7 +155,11 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc { | |||||||
| 				c.Set(config.ContextKey, token) | 				c.Set(config.ContextKey, token) | ||||||
| 				return next(c) | 				return next(c) | ||||||
| 			} | 			} | ||||||
| 			return echo.ErrUnauthorized | 			return &echo.HTTPError{ | ||||||
|  | 				Code:    ErrJWTInvalid.Code, | ||||||
|  | 				Message: ErrJWTInvalid.Message, | ||||||
|  | 				Inner:   err, | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -162,7 +172,7 @@ func jwtFromHeader(header string, authScheme string) jwtExtractor { | |||||||
| 		if len(auth) > l+1 && auth[:l] == authScheme { | 		if len(auth) > l+1 && auth[:l] == authScheme { | ||||||
| 			return auth[l+1:], nil | 			return auth[l+1:], nil | ||||||
| 		} | 		} | ||||||
| 		return "", errors.New("Missing or invalid jwt in the request header") | 		return "", ErrJWTMissing | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -171,7 +181,7 @@ func jwtFromQuery(param string) jwtExtractor { | |||||||
| 	return func(c echo.Context) (string, error) { | 	return func(c echo.Context) (string, error) { | ||||||
| 		token := c.QueryParam(param) | 		token := c.QueryParam(param) | ||||||
| 		if token == "" { | 		if token == "" { | ||||||
| 			return "", errors.New("Missing jwt in the query string") | 			return "", ErrJWTMissing | ||||||
| 		} | 		} | ||||||
| 		return token, nil | 		return token, nil | ||||||
| 	} | 	} | ||||||
| @@ -182,7 +192,7 @@ func jwtFromCookie(name string) jwtExtractor { | |||||||
| 	return func(c echo.Context) (string, error) { | 	return func(c echo.Context) (string, error) { | ||||||
| 		cookie, err := c.Cookie(name) | 		cookie, err := c.Cookie(name) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return "", errors.New("Missing jwt in the cookie") | 			return "", ErrJWTMissing | ||||||
| 		} | 		} | ||||||
| 		return cookie.Value, nil | 		return cookie.Value, nil | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										18
									
								
								vendor/github.com/labstack/echo/middleware/key_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/labstack/echo/middleware/key_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -20,7 +20,8 @@ type ( | |||||||
| 		// Possible values: | 		// Possible values: | ||||||
| 		// - "header:<name>" | 		// - "header:<name>" | ||||||
| 		// - "query:<name>" | 		// - "query:<name>" | ||||||
| 		KeyLookup string `json:"key_lookup"` | 		// - "form:<name>" | ||||||
|  | 		KeyLookup string `yaml:"key_lookup"` | ||||||
|  |  | ||||||
| 		// AuthScheme to be used in the Authorization header. | 		// AuthScheme to be used in the Authorization header. | ||||||
| 		// Optional. Default value "Bearer". | 		// Optional. Default value "Bearer". | ||||||
| @@ -72,7 +73,7 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc { | |||||||
| 		config.KeyLookup = DefaultKeyAuthConfig.KeyLookup | 		config.KeyLookup = DefaultKeyAuthConfig.KeyLookup | ||||||
| 	} | 	} | ||||||
| 	if config.Validator == nil { | 	if config.Validator == nil { | ||||||
| 		panic("key-auth middleware requires a validator function") | 		panic("echo: key-auth middleware requires a validator function") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Initialize | 	// Initialize | ||||||
| @@ -81,6 +82,8 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc { | |||||||
| 	switch parts[0] { | 	switch parts[0] { | ||||||
| 	case "query": | 	case "query": | ||||||
| 		extractor = keyFromQuery(parts[1]) | 		extractor = keyFromQuery(parts[1]) | ||||||
|  | 	case "form": | ||||||
|  | 		extractor = keyFromForm(parts[1]) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return func(next echo.HandlerFunc) echo.HandlerFunc { | 	return func(next echo.HandlerFunc) echo.HandlerFunc { | ||||||
| @@ -134,3 +137,14 @@ func keyFromQuery(param string) keyExtractor { | |||||||
| 		return key, nil | 		return key, nil | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // keyFromForm returns a `keyExtractor` that extracts key from the form. | ||||||
|  | func keyFromForm(param string) keyExtractor { | ||||||
|  | 	return func(c echo.Context) (string, error) { | ||||||
|  | 		key := c.FormValue(param) | ||||||
|  | 		if key == "" { | ||||||
|  | 			return "", errors.New("Missing key in the form") | ||||||
|  | 		} | ||||||
|  | 		return key, nil | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user