forked from lug/matterbridge
		
	Compare commits
	
		
			398 Commits
		
	
	
		
			v0.16.0-rc
			...
			v1.10.1
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					dba259e9f1 | ||
| 
						 | 
					07885f5810 | ||
| 
						 | 
					696c518550 | ||
| 
						 | 
					411ef2691c | ||
| 
						 | 
					fc6074ea9f | ||
| 
						 | 
					ab1670e2ce | ||
| 
						 | 
					9142a33bbf | ||
| 
						 | 
					f6eefa4ecc | ||
| 
						 | 
					f1db166ac4 | ||
| 
						 | 
					887c2bc56d | ||
| 
						 | 
					f0738a93c3 | ||
| 
						 | 
					75381c2c6e | ||
| 
						 | 
					bf0b9959d1 | ||
| 
						 | 
					406a54b597 | ||
| 
						 | 
					be04d1a862 | ||
| 
						 | 
					85b2d5a124 | ||
| 
						 | 
					521a7ed7b0 | ||
| 
						 | 
					529b188164 | ||
| 
						 | 
					8d307d8134 | ||
| 
						 | 
					8c675b52bc | ||
| 
						 | 
					aa51aa2aa0 | ||
| 
						 | 
					86865c6da5 | ||
| 
						 | 
					45296100df | ||
| 
						 | 
					1605fbc012 | ||
| 
						 | 
					c6c92e273d | ||
| 
						 | 
					467b373c43 | ||
| 
						 | 
					72ce7f06e9 | ||
| 
						 | 
					346a7284f7 | ||
| 
						 | 
					ee4ac67081 | ||
| 
						 | 
					5a93d14d75 | ||
| 
						 | 
					96a47a60ad | ||
| 
						 | 
					b24a47ad7f | ||
| 
						 | 
					cd1fd1bb7c | ||
| 
						 | 
					d44df7b6e6 | ||
| 
						 | 
					9d1ac0c84b | ||
| 
						 | 
					76af9cba5a | ||
| 
						 | 
					b69fc30902 | ||
| 
						 | 
					c3174f4de9 | ||
| 
						 | 
					99ce68e9ba | ||
| 
						 | 
					0cf73673a9 | ||
| 
						 | 
					08f442dc7b | ||
| 
						 | 
					8a8b95228c | ||
| 
						 | 
					31a752fa21 | ||
| 
						 | 
					a83831e68d | ||
| 
						 | 
					a12a8d4fe2 | ||
| 
						 | 
					e57f3a7e6c | ||
| 
						 | 
					68fbed9281 | ||
| 
						 | 
					8bfaa007d5 | ||
| 
						 | 
					76360f89c1 | ||
| 
						 | 
					d525230abd | ||
| 
						 | 
					b4aa637d41 | ||
| 
						 | 
					7c4334d0de | ||
| 
						 | 
					062be8d7c9 | ||
| 
						 | 
					db25ee59c5 | ||
| 
						 | 
					4b0bc6d0bf | ||
| 
						 | 
					8c0b04b995 | ||
| 
						 | 
					e5989adf92 | ||
| 
						 | 
					9e5da2f9d7 | ||
| 
						 | 
					a284a228a3 | ||
| 
						 | 
					2133e0d1be | ||
| 
						 | 
					a6f37f1d61 | ||
| 
						 | 
					9de9151826 | ||
| 
						 | 
					fdd5ada98c | ||
| 
						 | 
					80fcf18e24 | ||
| 
						 | 
					ab94b5ca7a | ||
| 
						 | 
					8d2ce56c37 | ||
| 
						 | 
					1ec324354b | ||
| 
						 | 
					16be6601c8 | ||
| 
						 | 
					98027446c8 | ||
| 
						 | 
					f2f1d874e1 | ||
| 
						 | 
					25a72113b1 | ||
| 
						 | 
					79c4ad5015 | ||
| 
						 | 
					e24f1c7c87 | ||
| 
						 | 
					dbf8a326d5 | ||
| 
						 | 
					0bc9c70c66 | ||
| 
						 | 
					594d2155e3 | ||
| 
						 | 
					20dbd71306 | ||
| 
						 | 
					6a727b9723 | ||
| 
						 | 
					02a5bc096f | ||
| 
						 | 
					2110db6f0c | ||
| 
						 | 
					2bac867382 | ||
| 
						 | 
					5fbd8a3be0 | ||
| 
						 | 
					ad6440b603 | ||
| 
						 | 
					064b6a915f | ||
| 
						 | 
					1578ebb0e2 | ||
| 
						 | 
					73525a4bbc | ||
| 
						 | 
					d62f49d1fc | ||
| 
						 | 
					63b88e77f2 | ||
| 
						 | 
					3d8f15c20b | ||
| 
						 | 
					cac5d56d60 | ||
| 
						 | 
					bd2a672c14 | ||
| 
						 | 
					82396e73f5 | ||
| 
						 | 
					ba928b169d | ||
| 
						 | 
					4fed720f97 | ||
| 
						 | 
					78238c85d4 | ||
| 
						 | 
					4f2ae7b73f | ||
| 
						 | 
					f82a9cc7ac | ||
| 
						 | 
					cce7624ab8 | ||
| 
						 | 
					c5ecd09172 | ||
| 
						 | 
					7b21c1c2f4 | ||
| 
						 | 
					f8714d81f5 | ||
| 
						 | 
					8622656005 | ||
| 
						 | 
					52237fadb6 | ||
| 
						 | 
					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?
 | 
			
		||||
run ```matterbridge -version```
 | 
			
		||||
You can also preview your report before submitting it.
 | 
			
		||||
 | 
			
		||||
### 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 actual behavior. 
 | 
			
		||||
#### Use logs from running ```matterbridge -debug``` if possible.
 | 
			
		||||
<!-- Use logs from running `matterbridge -debug` if possible. -->
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Any steps to reproduce the behavior?
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
### Please add your configuration file 
 | 
			
		||||
#### (be sure to exclude or anonymize private data (tokens/passwords))
 | 
			
		||||
<!-- (be sure to exclude or anonymize private data (tokens/passwords)) -->
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										26
									
								
								.github/ISSUE_TEMPLATE/Bug_report.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								.github/ISSUE_TEMPLATE/Bug_report.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,26 @@
 | 
			
		||||
---
 | 
			
		||||
name: Bug report
 | 
			
		||||
about: Create a report to help us improve. (Check the FAQ on the wiki first)
 | 
			
		||||
 | 
			
		||||
---
 | 
			
		||||
 | 
			
		||||
**Describe the bug**
 | 
			
		||||
A clear and concise description of what the bug is.
 | 
			
		||||
 | 
			
		||||
**To Reproduce**
 | 
			
		||||
Steps to reproduce the behavior:
 | 
			
		||||
 | 
			
		||||
**Expected behavior**
 | 
			
		||||
A clear and concise description of what you expected to happen.
 | 
			
		||||
 | 
			
		||||
**Screenshots/debug logs**
 | 
			
		||||
If applicable, add screenshots to help explain your problem.
 | 
			
		||||
Use logs from running `matterbridge -debug` if possible.
 | 
			
		||||
 | 
			
		||||
**Environment (please complete the following information):**
 | 
			
		||||
 - OS: [e.g. linux]
 | 
			
		||||
 - Matterbridge version: output of  `matterbridge -version`
 | 
			
		||||
 - If self compiled: output of `git rev-parse HEAD`
 | 
			
		||||
 | 
			
		||||
**Additional context**
 | 
			
		||||
Please add your configuration file  (be sure to exclude or anonymize private data (tokens/passwords))
 | 
			
		||||
							
								
								
									
										17
									
								
								.github/ISSUE_TEMPLATE/Feature_request.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								.github/ISSUE_TEMPLATE/Feature_request.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
			
		||||
---
 | 
			
		||||
name: Feature request
 | 
			
		||||
about: Suggest an idea for this project
 | 
			
		||||
 | 
			
		||||
---
 | 
			
		||||
 | 
			
		||||
**Is your feature request related to a problem? Please describe.**
 | 
			
		||||
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
 | 
			
		||||
 | 
			
		||||
**Describe the solution you'd like**
 | 
			
		||||
A clear and concise description of what you want to happen.
 | 
			
		||||
 | 
			
		||||
**Describe alternatives you've considered**
 | 
			
		||||
A clear and concise description of any alternative solutions or features you've considered.
 | 
			
		||||
 | 
			
		||||
**Additional context**
 | 
			
		||||
Add any other context or screenshots about the feature request here.
 | 
			
		||||
							
								
								
									
										51
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,51 @@
 | 
			
		||||
language: go
 | 
			
		||||
go:
 | 
			
		||||
    #- 1.7.x
 | 
			
		||||
    - 1.10.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
 | 
			
		||||
  edge:
 | 
			
		||||
    branch: v1.8.47
 | 
			
		||||
  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.
 | 
			
		||||
							
								
								
									
										95
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										95
									
								
								README.md
									
									
									
									
									
								
							@@ -1,18 +1,28 @@
 | 
			
		||||
# 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://matterbridge.zulipchat.com/register/)
 | 
			
		||||
 | 
			
		||||
Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix and Steam.
 | 
			
		||||
Has a REST API.
 | 
			
		||||
[](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion)
 | 
			
		||||
 | 
			
		||||

 | 
			
		||||
 | 
			
		||||
Simple bridge between IRC, XMPP, Gitter, Mattermost, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix, Steam, ssh-chat and Zulip
 | 
			
		||||
Has a REST API.   
 | 
			
		||||
Minecraft server chat support via [MatterLink](https://github.com/elytra/MatterLink)
 | 
			
		||||
 | 
			
		||||
**Mattermost isn't required to run matterbridge. It bridges between any supported protocol.**   
 | 
			
		||||
(The name matterbridge is a remnant when it was only bridging mattermost)
 | 
			
		||||
 | 
			
		||||
# Table of Contents
 | 
			
		||||
 * [Features](#features)
 | 
			
		||||
 * [Features](https://github.com/42wim/matterbridge/wiki/Features)
 | 
			
		||||
 * [Requirements](#requirements)
 | 
			
		||||
 * [Screenshots](https://github.com/42wim/matterbridge/wiki/)
 | 
			
		||||
 * [Installing](#installing)
 | 
			
		||||
   * [Binaries](#binaries)
 | 
			
		||||
   * [Building](#building)
 | 
			
		||||
 * [Configuration](#configuration)
 | 
			
		||||
   * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config)
 | 
			
		||||
   * [Examples](#examples) 
 | 
			
		||||
 * [Running](#running)
 | 
			
		||||
   * [Docker](#docker)
 | 
			
		||||
@@ -21,16 +31,25 @@ Has a REST API.
 | 
			
		||||
 * [Thanks](#thanks)
 | 
			
		||||
 | 
			
		||||
# Features
 | 
			
		||||
* Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam. 
 | 
			
		||||
  Pick and mix.
 | 
			
		||||
* Matterbridge can also work with private groups on your mattermost/slack.
 | 
			
		||||
* Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts.
 | 
			
		||||
* The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways).
 | 
			
		||||
* REST API to read/post messages to bridges (WIP).
 | 
			
		||||
* [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols)
 | 
			
		||||
* [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols)
 | 
			
		||||
* [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes)
 | 
			
		||||
* [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling)
 | 
			
		||||
* [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing)
 | 
			
		||||
* [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
 | 
			
		||||
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)
 | 
			
		||||
* [XMPP](https://jabber.org)
 | 
			
		||||
* [Gitter](https://gitter.im)
 | 
			
		||||
@@ -41,15 +60,20 @@ Accounts to one of the supported bridges
 | 
			
		||||
* [Rocket.chat](https://rocket.chat)
 | 
			
		||||
* [Matrix](https://matrix.org)
 | 
			
		||||
* [Steam](https://store.steampowered.com/)
 | 
			
		||||
* [Twitch](https://twitch.tv)
 | 
			
		||||
* [Ssh-chat](https://github.com/shazow/ssh-chat)
 | 
			
		||||
* [Zulip](https://zulipchat.com)
 | 
			
		||||
 | 
			
		||||
# Screenshots
 | 
			
		||||
See https://github.com/42wim/matterbridge/wiki
 | 
			
		||||
 | 
			
		||||
# Installing
 | 
			
		||||
## Binaries
 | 
			
		||||
Binaries can be found [here] (https://github.com/42wim/matterbridge/releases/)
 | 
			
		||||
* Latest rc release (with steam support) [v0.16.0-rc2](https://github.com/42wim/matterbridge/releases/latest)
 | 
			
		||||
* Latest stable release [v0.15.0](https://github.com/42wim/matterbridge/releases/tag/v0.15.0)
 | 
			
		||||
* Latest stable release [v1.10.1](https://github.com/42wim/matterbridge/releases/latest)
 | 
			
		||||
* Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)  
 | 
			
		||||
 | 
			
		||||
## 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
 | 
			
		||||
@@ -64,12 +88,12 @@ matterbridge
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
# Configuration
 | 
			
		||||
* [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example.
 | 
			
		||||
* [matterbridge.toml.simple](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.simple) for a simple example.
 | 
			
		||||
 | 
			
		||||
## Create a configuration.
 | 
			
		||||
## Basic 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 
 | 
			
		||||
### Bridge mattermost (off-topic) - irc (#testing)
 | 
			
		||||
```
 | 
			
		||||
@@ -80,12 +104,12 @@ See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config
 | 
			
		||||
 | 
			
		||||
[mattermost]
 | 
			
		||||
    [mattermost.work]
 | 
			
		||||
    useAPI=true
 | 
			
		||||
    Server="yourmattermostserver.tld"
 | 
			
		||||
    Team="yourteam"
 | 
			
		||||
    Login="yourlogin"
 | 
			
		||||
    Password="yourpass"
 | 
			
		||||
    PrefixMessagesWithNick=true
 | 
			
		||||
    RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
[[gateway]]
 | 
			
		||||
name="mygateway"
 | 
			
		||||
@@ -103,7 +127,6 @@ enable=true
 | 
			
		||||
```
 | 
			
		||||
[slack]
 | 
			
		||||
[slack.test]
 | 
			
		||||
useAPI=true
 | 
			
		||||
Token="yourslacktoken"
 | 
			
		||||
PrefixMessagesWithNick=true
 | 
			
		||||
 | 
			
		||||
@@ -129,11 +152,8 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> "
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
# 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:
 | 
			
		||||
@@ -158,29 +178,26 @@ See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.m
 | 
			
		||||
 | 
			
		||||
# FAQ
 | 
			
		||||
 | 
			
		||||
Please look at [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for more information first.
 | 
			
		||||
 | 
			
		||||
## Mattermost doesn't show the IRC nicks
 | 
			
		||||
If you're running the webhooks version, this can be fixed by either:
 | 
			
		||||
* enabling "override usernames". See [mattermost documentation](http://docs.mattermost.com/developer/webhooks-incoming.html#enabling-incoming-webhooks)
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.toml.
 | 
			
		||||
 | 
			
		||||
If you're running the API version you'll need to:
 | 
			
		||||
* setting ```PrefixMessagesWithNick``` to ```true``` in ```mattermost``` section of your matterbridge.toml.
 | 
			
		||||
 | 
			
		||||
Also look at the ```RemoteNickFormat``` setting.
 | 
			
		||||
See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ)
 | 
			
		||||
 | 
			
		||||
Want to tip ? 
 | 
			
		||||
* eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f
 | 
			
		||||
* btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs
 | 
			
		||||
 | 
			
		||||
# Thanks
 | 
			
		||||
[](https://www.digitalocean.com/) for sponsoring demo/testing droplets.
 | 
			
		||||
 | 
			
		||||
Matterbridge wouldn't exist without these libraries:
 | 
			
		||||
* discord - https://github.com/bwmarrin/discordgo
 | 
			
		||||
* echo - https://github.com/labstack/echo
 | 
			
		||||
* gitter - https://github.com/sromku/go-gitter
 | 
			
		||||
* gops - https://github.com/google/gops
 | 
			
		||||
* irc - https://github.com/thoj/go-ircevent
 | 
			
		||||
* gozulipbot - https://github.com/ifo/gozulipbot
 | 
			
		||||
* irc - https://github.com/lrstanley/girc
 | 
			
		||||
* mattermost - https://github.com/mattermost/platform
 | 
			
		||||
* matrix - https://github.com/matrix-org/gomatrix
 | 
			
		||||
* slack - https://github.com/nlopes/slack
 | 
			
		||||
* steam - https://github.com/Philipp15b/go-steam
 | 
			
		||||
* telegram - https://github.com/go-telegram-bot-api/telegram-bot-api
 | 
			
		||||
* xmpp - https://github.com/mattn/go-xmpp
 | 
			
		||||
 | 
			
		||||
* zulip - https://github.com/ifo/gozulipbot
 | 
			
		||||
 
 | 
			
		||||
@@ -1,21 +1,21 @@
 | 
			
		||||
package api
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/labstack/echo"
 | 
			
		||||
	"github.com/labstack/echo/middleware"
 | 
			
		||||
	"github.com/zfjagann/golang-ring"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Api struct {
 | 
			
		||||
	Config   *config.Protocol
 | 
			
		||||
	Remote   chan config.Message
 | 
			
		||||
	Account  string
 | 
			
		||||
	Messages ring.Ring
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type ApiMessage struct {
 | 
			
		||||
@@ -26,30 +26,27 @@ type ApiMessage struct {
 | 
			
		||||
	Gateway  string `json:"gateway"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "api"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Api {
 | 
			
		||||
	b := &Api{}
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Api{Config: cfg}
 | 
			
		||||
	e := echo.New()
 | 
			
		||||
	e.HideBanner = true
 | 
			
		||||
	e.HidePort = true
 | 
			
		||||
	b.Messages = ring.Ring{}
 | 
			
		||||
	b.Messages.SetCapacity(cfg.Buffer)
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	if b.Config.Token != "" {
 | 
			
		||||
	b.Messages.SetCapacity(b.GetInt("Buffer"))
 | 
			
		||||
	if b.GetString("Token") != "" {
 | 
			
		||||
		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) {
 | 
			
		||||
			return key == b.Config.Token, nil
 | 
			
		||||
			return key == b.GetString("Token"), nil
 | 
			
		||||
		}))
 | 
			
		||||
	}
 | 
			
		||||
	e.GET("/api/messages", b.handleMessages)
 | 
			
		||||
	e.GET("/api/stream", b.handleStream)
 | 
			
		||||
	e.POST("/api/message", b.handlePostMessage)
 | 
			
		||||
	go func() {
 | 
			
		||||
		flog.Fatal(e.Start(cfg.BindAddress))
 | 
			
		||||
		if b.GetString("BindAddress") == "" {
 | 
			
		||||
			b.Log.Fatalf("No BindAddress configured.")
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Infof("Listening on %s", b.GetString("BindAddress"))
 | 
			
		||||
		b.Log.Fatal(e.Start(b.GetString("BindAddress")))
 | 
			
		||||
	}()
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
@@ -61,34 +58,35 @@ func (b *Api) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
func (b *Api) JoinChannel(channel string) error {
 | 
			
		||||
func (b *Api) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Api) Send(msg config.Message) error {
 | 
			
		||||
func (b *Api) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Lock()
 | 
			
		||||
	defer b.Unlock()
 | 
			
		||||
	// ignore delete messages
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
	b.Messages.Enqueue(&msg)
 | 
			
		||||
	return nil
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Api) handlePostMessage(c echo.Context) error {
 | 
			
		||||
	message := &ApiMessage{}
 | 
			
		||||
	if err := c.Bind(message); err != nil {
 | 
			
		||||
	message := config.Message{}
 | 
			
		||||
	if err := c.Bind(&message); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
 | 
			
		||||
	b.Remote <- config.Message{
 | 
			
		||||
		Text:     message.Text,
 | 
			
		||||
		Username: message.Username,
 | 
			
		||||
		UserID:   message.UserID,
 | 
			
		||||
		Channel:  "api",
 | 
			
		||||
		Avatar:   message.Avatar,
 | 
			
		||||
		Account:  b.Account,
 | 
			
		||||
		Gateway:  message.Gateway,
 | 
			
		||||
		Protocol: "api",
 | 
			
		||||
	}
 | 
			
		||||
	// these values are fixed
 | 
			
		||||
	message.Channel = "api"
 | 
			
		||||
	message.Protocol = "api"
 | 
			
		||||
	message.Account = b.Account
 | 
			
		||||
	message.ID = ""
 | 
			
		||||
	message.Timestamp = time.Now()
 | 
			
		||||
	b.Log.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
 | 
			
		||||
	b.Remote <- message
 | 
			
		||||
	return c.JSON(http.StatusOK, message)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -99,3 +97,24 @@ func (b *Api) handleMessages(c echo.Context) error {
 | 
			
		||||
	b.Messages = ring.Ring{}
 | 
			
		||||
	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)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										121
									
								
								bridge/bridge.go
									
									
									
									
									
								
							
							
						
						
									
										121
									
								
								bridge/bridge.go
									
									
									
									
									
								
							@@ -1,41 +1,42 @@
 | 
			
		||||
package bridge
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/api"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/discord"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/irc"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/matrix"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/mattermost"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/rocketchat"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/slack"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/steam"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/telegram"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/xmpp"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bridger interface {
 | 
			
		||||
	Send(msg config.Message) error
 | 
			
		||||
	Send(msg config.Message) (string, error)
 | 
			
		||||
	Connect() error
 | 
			
		||||
	JoinChannel(channel string) error
 | 
			
		||||
	JoinChannel(channel config.ChannelInfo) error
 | 
			
		||||
	Disconnect() error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
	Config config.Protocol
 | 
			
		||||
	Bridger
 | 
			
		||||
	Name     string
 | 
			
		||||
	Account  string
 | 
			
		||||
	Protocol string
 | 
			
		||||
	Channels map[string]config.ChannelInfo
 | 
			
		||||
	Joined   map[string]bool
 | 
			
		||||
	Log      *log.Entry
 | 
			
		||||
	Config   *config.Config
 | 
			
		||||
	General  *config.Protocol
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge {
 | 
			
		||||
type Config struct {
 | 
			
		||||
	//	General *config.Protocol
 | 
			
		||||
	Remote chan config.Message
 | 
			
		||||
	Log    *log.Entry
 | 
			
		||||
	*Bridge
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Factory is the factory function to create a bridge
 | 
			
		||||
type Factory func(*Config) Bridger
 | 
			
		||||
 | 
			
		||||
func New(bridge *config.Bridge) *Bridge {
 | 
			
		||||
	b := new(Bridge)
 | 
			
		||||
	b.Channels = make(map[string]config.ChannelInfo)
 | 
			
		||||
	accInfo := strings.Split(bridge.Account, ".")
 | 
			
		||||
@@ -45,66 +46,19 @@ func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Brid
 | 
			
		||||
	b.Protocol = protocol
 | 
			
		||||
	b.Account = bridge.Account
 | 
			
		||||
	b.Joined = make(map[string]bool)
 | 
			
		||||
 | 
			
		||||
	// override config from environment
 | 
			
		||||
	config.OverrideCfgFromEnv(cfg, protocol, name)
 | 
			
		||||
	switch protocol {
 | 
			
		||||
	case "mattermost":
 | 
			
		||||
		b.Config = cfg.Mattermost[name]
 | 
			
		||||
		b.Bridger = bmattermost.New(cfg.Mattermost[name], bridge.Account, c)
 | 
			
		||||
	case "irc":
 | 
			
		||||
		b.Config = cfg.IRC[name]
 | 
			
		||||
		b.Bridger = birc.New(cfg.IRC[name], bridge.Account, c)
 | 
			
		||||
	case "gitter":
 | 
			
		||||
		b.Config = cfg.Gitter[name]
 | 
			
		||||
		b.Bridger = bgitter.New(cfg.Gitter[name], bridge.Account, c)
 | 
			
		||||
	case "slack":
 | 
			
		||||
		b.Config = cfg.Slack[name]
 | 
			
		||||
		b.Bridger = bslack.New(cfg.Slack[name], bridge.Account, c)
 | 
			
		||||
	case "xmpp":
 | 
			
		||||
		b.Config = cfg.Xmpp[name]
 | 
			
		||||
		b.Bridger = bxmpp.New(cfg.Xmpp[name], bridge.Account, c)
 | 
			
		||||
	case "discord":
 | 
			
		||||
		b.Config = cfg.Discord[name]
 | 
			
		||||
		b.Bridger = bdiscord.New(cfg.Discord[name], bridge.Account, c)
 | 
			
		||||
	case "telegram":
 | 
			
		||||
		b.Config = cfg.Telegram[name]
 | 
			
		||||
		b.Bridger = btelegram.New(cfg.Telegram[name], bridge.Account, c)
 | 
			
		||||
	case "rocketchat":
 | 
			
		||||
		b.Config = cfg.Rocketchat[name]
 | 
			
		||||
		b.Bridger = brocketchat.New(cfg.Rocketchat[name], bridge.Account, c)
 | 
			
		||||
	case "matrix":
 | 
			
		||||
		b.Config = cfg.Matrix[name]
 | 
			
		||||
		b.Bridger = bmatrix.New(cfg.Matrix[name], bridge.Account, c)
 | 
			
		||||
	case "steam":
 | 
			
		||||
		b.Config = cfg.Steam[name]
 | 
			
		||||
		b.Bridger = bsteam.New(cfg.Steam[name], bridge.Account, c)
 | 
			
		||||
	case "api":
 | 
			
		||||
		b.Config = cfg.Api[name]
 | 
			
		||||
		b.Bridger = api.New(cfg.Api[name], bridge.Account, c)
 | 
			
		||||
	}
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) JoinChannels() error {
 | 
			
		||||
	err := b.joinChannels(b.Channels, b.Joined)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error {
 | 
			
		||||
	mychannel := ""
 | 
			
		||||
	for ID, channel := range channels {
 | 
			
		||||
		if !exists[ID] {
 | 
			
		||||
			mychannel = channel.Name
 | 
			
		||||
			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID)
 | 
			
		||||
			if b.Protocol == "irc" && channel.Options.Key != "" {
 | 
			
		||||
				log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name)
 | 
			
		||||
				mychannel = mychannel + " " + channel.Options.Key
 | 
			
		||||
			}
 | 
			
		||||
			err := b.JoinChannel(mychannel)
 | 
			
		||||
			b.Log.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID)
 | 
			
		||||
			err := b.JoinChannel(channel)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
@@ -113,3 +67,38 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) GetBool(key string) bool {
 | 
			
		||||
	if b.Config.GetBool(b.Account + "." + key) {
 | 
			
		||||
		return b.Config.GetBool(b.Account + "." + key)
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.GetBool("general." + key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) GetInt(key string) int {
 | 
			
		||||
	if b.Config.GetInt(b.Account+"."+key) != 0 {
 | 
			
		||||
		return b.Config.GetInt(b.Account + "." + key)
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.GetInt("general." + key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) GetString(key string) string {
 | 
			
		||||
	if b.Config.GetString(b.Account+"."+key) != "" {
 | 
			
		||||
		return b.Config.GetString(b.Account + "." + key)
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.GetString("general." + key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) GetStringSlice(key string) []string {
 | 
			
		||||
	if len(b.Config.GetStringSlice(b.Account+"."+key)) != 0 {
 | 
			
		||||
		return b.Config.GetStringSlice(b.Account + "." + key)
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.GetStringSlice("general." + key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bridge) GetStringSlice2D(key string) [][]string {
 | 
			
		||||
	if len(b.Config.GetStringSlice2D(b.Account+"."+key)) != 0 {
 | 
			
		||||
		return b.Config.GetStringSlice2D(b.Account + "." + key)
 | 
			
		||||
	}
 | 
			
		||||
	return b.Config.GetStringSlice2D("general." + key)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,18 +1,26 @@
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/BurntSushi/toml"
 | 
			
		||||
	"log"
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"github.com/fsnotify/fsnotify"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
	"github.com/spf13/viper"
 | 
			
		||||
	prefixed "github.com/x-cray/logrus-prefixed-formatter"
 | 
			
		||||
	"os"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	EVENT_JOIN_LEAVE      = "join_leave"
 | 
			
		||||
	EVENT_FAILURE         = "failure"
 | 
			
		||||
	EVENT_REJOIN_CHANNELS = "rejoin_channels"
 | 
			
		||||
	EVENT_JOIN_LEAVE        = "join_leave"
 | 
			
		||||
	EVENT_TOPIC_CHANGE      = "topic_change"
 | 
			
		||||
	EVENT_FAILURE           = "failure"
 | 
			
		||||
	EVENT_FILE_FAILURE_SIZE = "file_failure_size"
 | 
			
		||||
	EVENT_AVATAR_DOWNLOAD   = "avatar_download"
 | 
			
		||||
	EVENT_REJOIN_CHANNELS   = "rejoin_channels"
 | 
			
		||||
	EVENT_USER_ACTION       = "user_action"
 | 
			
		||||
	EVENT_MSG_DELETE        = "msg_delete"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
@@ -26,6 +34,18 @@ type Message struct {
 | 
			
		||||
	Protocol  string    `json:"protocol"`
 | 
			
		||||
	Gateway   string    `json:"gateway"`
 | 
			
		||||
	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 {
 | 
			
		||||
@@ -33,7 +53,6 @@ type ChannelInfo struct {
 | 
			
		||||
	Account     string
 | 
			
		||||
	Direction   string
 | 
			
		||||
	ID          string
 | 
			
		||||
	GID         map[string]bool
 | 
			
		||||
	SameChannel map[string]bool
 | 
			
		||||
	Options     ChannelOptions
 | 
			
		||||
}
 | 
			
		||||
@@ -42,49 +61,70 @@ type Protocol struct {
 | 
			
		||||
	AuthCode               string // steam
 | 
			
		||||
	BindAddress            string // mattermost, slack // DEPRECATED
 | 
			
		||||
	Buffer                 int    // api
 | 
			
		||||
	Charset                string // irc
 | 
			
		||||
	ColorNicks             bool   // only irc for now
 | 
			
		||||
	Debug                  bool   // general
 | 
			
		||||
	DebugLevel             int    // only for irc now
 | 
			
		||||
	EditSuffix             string // mattermost, slack, discord, telegram, gitter
 | 
			
		||||
	EditDisable            bool   // mattermost, slack, discord, telegram, gitter
 | 
			
		||||
	IconURL                string // mattermost, slack
 | 
			
		||||
	IgnoreNicks            string // all protocols
 | 
			
		||||
	IgnoreMessages         string // all protocols
 | 
			
		||||
	Jid                    string // xmpp
 | 
			
		||||
	Label                  string // all protocols
 | 
			
		||||
	Login                  string // mattermost, matrix
 | 
			
		||||
	Muc                    string // xmpp
 | 
			
		||||
	Name                   string // all protocols
 | 
			
		||||
	Nick                   string // all protocols
 | 
			
		||||
	NickFormatter          string // mattermost, slack
 | 
			
		||||
	NickServNick           string // IRC
 | 
			
		||||
	NickServPassword       string // IRC
 | 
			
		||||
	NicksPerRow            int    // mattermost, slack
 | 
			
		||||
	NoHomeServerSuffix     bool   // matrix
 | 
			
		||||
	NoTLS                  bool   // mattermost
 | 
			
		||||
	Password               string // IRC,mattermost,XMPP,matrix
 | 
			
		||||
	PrefixMessagesWithNick bool   // mattemost, slack
 | 
			
		||||
	Protocol               string //all protocols
 | 
			
		||||
	MessageQueue           int    // IRC, size of message queue for flood control
 | 
			
		||||
	MessageDelay           int    // IRC, time in millisecond to wait between messages
 | 
			
		||||
	MessageLength          int    // IRC, max length of a message allowed
 | 
			
		||||
	MessageFormat          string // telegram
 | 
			
		||||
	RemoteNickFormat       string // all protocols
 | 
			
		||||
	Server                 string // IRC,mattermost,XMPP,discord
 | 
			
		||||
	ShowJoinPart           bool   // all protocols
 | 
			
		||||
	ShowEmbeds             bool   // discord
 | 
			
		||||
	SkipTLSVerify          bool   // IRC, mattermost
 | 
			
		||||
	Team                   string // mattermost
 | 
			
		||||
	Token                  string // gitter, slack, discord, api
 | 
			
		||||
	URL                    string // mattermost, slack // DEPRECATED
 | 
			
		||||
	UseAPI                 bool   // mattermost, slack
 | 
			
		||||
	UseSASL                bool   // IRC
 | 
			
		||||
	UseTLS                 bool   // IRC
 | 
			
		||||
	UseFirstName           bool   // telegram
 | 
			
		||||
	UseInsecureURL         bool   // telegram
 | 
			
		||||
	WebhookBindAddress     string // mattermost, slack
 | 
			
		||||
	WebhookURL             string // mattermost, slack
 | 
			
		||||
	WebhookUse             string // mattermost, slack, discord
 | 
			
		||||
	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
 | 
			
		||||
	Name                   string     // all protocols
 | 
			
		||||
	Nick                   string     // all protocols
 | 
			
		||||
	NickFormatter          string     // mattermost, slack
 | 
			
		||||
	NickServNick           string     // IRC
 | 
			
		||||
	NickServUsername       string     // IRC
 | 
			
		||||
	NickServPassword       string     // IRC
 | 
			
		||||
	NicksPerRow            int        // mattermost, slack
 | 
			
		||||
	NoHomeServerSuffix     bool       // matrix
 | 
			
		||||
	NoSendJoinPart         bool       // all protocols
 | 
			
		||||
	NoTLS                  bool       // mattermost
 | 
			
		||||
	Password               string     // IRC,mattermost,XMPP,matrix
 | 
			
		||||
	PrefixMessagesWithNick bool       // mattemost, slack
 | 
			
		||||
	Protocol               string     // all protocols
 | 
			
		||||
	QuoteDisable           bool       // telegram
 | 
			
		||||
	QuoteFormat            string     // telegram
 | 
			
		||||
	RejoinDelay            int        // IRC
 | 
			
		||||
	ReplaceMessages        [][]string // all protocols
 | 
			
		||||
	ReplaceNicks           [][]string // all protocols
 | 
			
		||||
	RemoteNickFormat       string     // all protocols
 | 
			
		||||
	Server                 string     // IRC,mattermost,XMPP,discord
 | 
			
		||||
	ShowJoinPart           bool       // all protocols
 | 
			
		||||
	ShowTopicChange        bool       // slack
 | 
			
		||||
	ShowEmbeds             bool       // discord
 | 
			
		||||
	SkipTLSVerify          bool       // IRC, mattermost
 | 
			
		||||
	StripNick              bool       // all protocols
 | 
			
		||||
	Team                   string     // mattermost
 | 
			
		||||
	Token                  string     // gitter, slack, discord, api
 | 
			
		||||
	Topic                  string     // zulip
 | 
			
		||||
	URL                    string     // mattermost, slack // DEPRECATED
 | 
			
		||||
	UseAPI                 bool       // mattermost, slack
 | 
			
		||||
	UseSASL                bool       // IRC
 | 
			
		||||
	UseTLS                 bool       // IRC
 | 
			
		||||
	UseFirstName           bool       // telegram
 | 
			
		||||
	UseUserName            bool       // discord
 | 
			
		||||
	UseInsecureURL         bool       // telegram
 | 
			
		||||
	WebhookBindAddress     string     // mattermost, slack
 | 
			
		||||
	WebhookURL             string     // mattermost, slack
 | 
			
		||||
	WebhookUse             string     // mattermost, slack, discord
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type ChannelOptions struct {
 | 
			
		||||
	Key string // irc
 | 
			
		||||
	Key        string // irc
 | 
			
		||||
	WebhookURL string // discord
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bridge struct {
 | 
			
		||||
@@ -109,9 +149,9 @@ type SameChannelGateway struct {
 | 
			
		||||
	Accounts []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
type ConfigValues struct {
 | 
			
		||||
	Api                map[string]Protocol
 | 
			
		||||
	IRC                map[string]Protocol
 | 
			
		||||
	Irc                map[string]Protocol
 | 
			
		||||
	Mattermost         map[string]Protocol
 | 
			
		||||
	Matrix             map[string]Protocol
 | 
			
		||||
	Slack              map[string]Protocol
 | 
			
		||||
@@ -121,80 +161,118 @@ type Config struct {
 | 
			
		||||
	Discord            map[string]Protocol
 | 
			
		||||
	Telegram           map[string]Protocol
 | 
			
		||||
	Rocketchat         map[string]Protocol
 | 
			
		||||
	Sshchat            map[string]Protocol
 | 
			
		||||
	Zulip              map[string]Protocol
 | 
			
		||||
	General            Protocol
 | 
			
		||||
	Gateway            []Gateway
 | 
			
		||||
	SameChannelGateway []SameChannelGateway
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	v *viper.Viper
 | 
			
		||||
	*ConfigValues
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewConfig(cfgfile string) *Config {
 | 
			
		||||
	var cfg Config
 | 
			
		||||
	if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil {
 | 
			
		||||
	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false})
 | 
			
		||||
	flog := log.WithFields(log.Fields{"prefix": "config"})
 | 
			
		||||
	var cfg ConfigValues
 | 
			
		||||
	viper.SetConfigType("toml")
 | 
			
		||||
	viper.SetConfigFile(cfgfile)
 | 
			
		||||
	viper.SetEnvPrefix("matterbridge")
 | 
			
		||||
	viper.AddConfigPath(".")
 | 
			
		||||
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
 | 
			
		||||
	viper.AutomaticEnv()
 | 
			
		||||
	f, err := os.Open(cfgfile)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	fail := false
 | 
			
		||||
	for k, v := range cfg.Mattermost {
 | 
			
		||||
		res := Deprecated(v, "mattermost."+k)
 | 
			
		||||
		if res {
 | 
			
		||||
			fail = res
 | 
			
		||||
		}
 | 
			
		||||
	err = viper.ReadConfig(f)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	for k, v := range cfg.Slack {
 | 
			
		||||
		res := Deprecated(v, "slack."+k)
 | 
			
		||||
		if res {
 | 
			
		||||
			fail = res
 | 
			
		||||
		}
 | 
			
		||||
	err = viper.Unmarshal(&cfg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal("blah", err)
 | 
			
		||||
	}
 | 
			
		||||
	for k, v := range cfg.Rocketchat {
 | 
			
		||||
		res := Deprecated(v, "rocketchat."+k)
 | 
			
		||||
		if res {
 | 
			
		||||
			fail = res
 | 
			
		||||
		}
 | 
			
		||||
	mycfg := new(Config)
 | 
			
		||||
	mycfg.v = viper.GetViper()
 | 
			
		||||
	if cfg.General.MediaDownloadSize == 0 {
 | 
			
		||||
		cfg.General.MediaDownloadSize = 1000000
 | 
			
		||||
	}
 | 
			
		||||
	if fail {
 | 
			
		||||
		log.Fatalf("Fix your config. Please see changelog for more information")
 | 
			
		||||
	}
 | 
			
		||||
	return &cfg
 | 
			
		||||
	viper.WatchConfig()
 | 
			
		||||
	viper.OnConfigChange(func(e fsnotify.Event) {
 | 
			
		||||
		flog.Println("Config file changed:", e.Name)
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	mycfg.ConfigValues = &cfg
 | 
			
		||||
	return mycfg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func OverrideCfgFromEnv(cfg *Config, protocol string, account string) {
 | 
			
		||||
	var protoCfg Protocol
 | 
			
		||||
	val := reflect.ValueOf(cfg).Elem()
 | 
			
		||||
	// loop over the Config struct
 | 
			
		||||
	for i := 0; i < val.NumField(); i++ {
 | 
			
		||||
		typeField := val.Type().Field(i)
 | 
			
		||||
		// look for the protocol map (both lowercase)
 | 
			
		||||
		if strings.ToLower(typeField.Name) == protocol {
 | 
			
		||||
			// get the Protocol struct from the map
 | 
			
		||||
			data := val.Field(i).MapIndex(reflect.ValueOf(account))
 | 
			
		||||
			protoCfg = data.Interface().(Protocol)
 | 
			
		||||
			protoStruct := reflect.ValueOf(&protoCfg).Elem()
 | 
			
		||||
			// loop over the found protocol struct
 | 
			
		||||
			for i := 0; i < protoStruct.NumField(); i++ {
 | 
			
		||||
				typeField := protoStruct.Type().Field(i)
 | 
			
		||||
				// build our environment key (eg MATTERBRIDGE_MATTERMOST_WORK_LOGIN)
 | 
			
		||||
				key := "matterbridge_" + protocol + "_" + account + "_" + typeField.Name
 | 
			
		||||
				key = strings.ToUpper(key)
 | 
			
		||||
				// search the environment
 | 
			
		||||
				res := os.Getenv(key)
 | 
			
		||||
				// if it exists and the current field is a string
 | 
			
		||||
				// then update the current field
 | 
			
		||||
				if res != "" {
 | 
			
		||||
					fieldVal := protoStruct.Field(i)
 | 
			
		||||
					if fieldVal.Kind() == reflect.String {
 | 
			
		||||
						log.Printf("config: overriding %s from env with %s\n", key, res)
 | 
			
		||||
						fieldVal.Set(reflect.ValueOf(res))
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
func NewConfigFromString(input []byte) *Config {
 | 
			
		||||
	var cfg ConfigValues
 | 
			
		||||
	viper.SetConfigType("toml")
 | 
			
		||||
	err := viper.ReadConfig(bytes.NewBuffer(input))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	err = viper.Unmarshal(&cfg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	mycfg := new(Config)
 | 
			
		||||
	mycfg.v = viper.GetViper()
 | 
			
		||||
	mycfg.ConfigValues = &cfg
 | 
			
		||||
	return mycfg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetBool(key string) bool {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
	//	log.Debugf("getting bool %s = %#v", key, c.v.GetBool(key))
 | 
			
		||||
	return c.v.GetBool(key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetInt(key string) int {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
	//	log.Debugf("getting int %s = %d", key, c.v.GetInt(key))
 | 
			
		||||
	return c.v.GetInt(key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetString(key string) string {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
	//	log.Debugf("getting String %s = %s", key, c.v.GetString(key))
 | 
			
		||||
	return c.v.GetString(key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetStringSlice(key string) []string {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
	// log.Debugf("getting StringSlice %s = %#v", key, c.v.GetStringSlice(key))
 | 
			
		||||
	return c.v.GetStringSlice(key)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetStringSlice2D(key string) [][]string {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
	result := [][]string{}
 | 
			
		||||
	if res, ok := c.v.Get(key).([]interface{}); ok {
 | 
			
		||||
		for _, entry := range res {
 | 
			
		||||
			result2 := []string{}
 | 
			
		||||
			for _, entry2 := range entry.([]interface{}) {
 | 
			
		||||
				result2 = append(result2, entry2.(string))
 | 
			
		||||
			}
 | 
			
		||||
			// update the map with the modified Protocol (cfg.Protocol[account] = Protocol)
 | 
			
		||||
			val.Field(i).SetMapIndex(reflect.ValueOf(account), reflect.ValueOf(protoCfg))
 | 
			
		||||
			break
 | 
			
		||||
			result = append(result, result2)
 | 
			
		||||
		}
 | 
			
		||||
		return result
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetIconURL(msg *Message, cfg *Protocol) string {
 | 
			
		||||
	iconURL := cfg.IconURL
 | 
			
		||||
func GetIconURL(msg *Message, iconURL string) string {
 | 
			
		||||
	info := strings.Split(msg.Account, ".")
 | 
			
		||||
	protocol := info[0]
 | 
			
		||||
	name := info[1]
 | 
			
		||||
@@ -203,17 +281,3 @@ func GetIconURL(msg *Message, cfg *Protocol) string {
 | 
			
		||||
	iconURL = strings.Replace(iconURL, "{PROTOCOL}", protocol, -1)
 | 
			
		||||
	return iconURL
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Deprecated(cfg Protocol, account string) bool {
 | 
			
		||||
	if cfg.BindAddress != "" {
 | 
			
		||||
		log.Printf("ERROR: %s BindAddress is deprecated, you need to change it to WebhookBindAddress.", account)
 | 
			
		||||
	} else if cfg.URL != "" {
 | 
			
		||||
		log.Printf("ERROR: %s URL is deprecated, you need to change it to WebhookURL.", account)
 | 
			
		||||
	} else if cfg.UseAPI == true {
 | 
			
		||||
		log.Printf("ERROR: %s UseAPI is deprecated, it's enabled by default, please remove it from your config file.", account)
 | 
			
		||||
	} else {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
	//log.Fatalf("ERROR: Fix your config: %s", account)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,201 +1,291 @@
 | 
			
		||||
package bdiscord
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type bdiscord struct {
 | 
			
		||||
	c             *discordgo.Session
 | 
			
		||||
	Config        *config.Protocol
 | 
			
		||||
	Remote        chan config.Message
 | 
			
		||||
	Account       string
 | 
			
		||||
	Channels      []*discordgo.Channel
 | 
			
		||||
	Nick          string
 | 
			
		||||
	UseChannelID  bool
 | 
			
		||||
	userMemberMap map[string]*discordgo.Member
 | 
			
		||||
	guildID       string
 | 
			
		||||
	webhookID     string
 | 
			
		||||
	webhookToken  string
 | 
			
		||||
type Bdiscord struct {
 | 
			
		||||
	c              *discordgo.Session
 | 
			
		||||
	Channels       []*discordgo.Channel
 | 
			
		||||
	Nick           string
 | 
			
		||||
	UseChannelID   bool
 | 
			
		||||
	userMemberMap  map[string]*discordgo.Member
 | 
			
		||||
	guildID        string
 | 
			
		||||
	webhookID      string
 | 
			
		||||
	webhookToken   string
 | 
			
		||||
	channelInfoMap map[string]*config.ChannelInfo
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "discord"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord {
 | 
			
		||||
	b := &bdiscord{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Bdiscord{Config: cfg}
 | 
			
		||||
	b.userMemberMap = make(map[string]*discordgo.Member)
 | 
			
		||||
	if b.Config.WebhookURL != "" {
 | 
			
		||||
		flog.Debug("Configuring Discord Incoming Webhook")
 | 
			
		||||
		webhookURLSplit := strings.Split(b.Config.WebhookURL, "/")
 | 
			
		||||
		b.webhookToken = webhookURLSplit[len(webhookURLSplit)-1]
 | 
			
		||||
		b.webhookID = webhookURLSplit[len(webhookURLSplit)-2]
 | 
			
		||||
	b.channelInfoMap = make(map[string]*config.ChannelInfo)
 | 
			
		||||
	if b.GetString("WebhookURL") != "" {
 | 
			
		||||
		b.Log.Debug("Configuring Discord Incoming Webhook")
 | 
			
		||||
		b.webhookID, b.webhookToken = b.splitURL(b.GetString("WebhookURL"))
 | 
			
		||||
	}
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) Connect() error {
 | 
			
		||||
func (b *Bdiscord) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	if b.Config.WebhookURL == "" {
 | 
			
		||||
		flog.Info("Connecting using token")
 | 
			
		||||
	var token string
 | 
			
		||||
	b.Log.Info("Connecting")
 | 
			
		||||
	if b.GetString("WebhookURL") == "" {
 | 
			
		||||
		b.Log.Info("Connecting using token")
 | 
			
		||||
	} else {
 | 
			
		||||
		flog.Info("Connecting using webhookurl (for posting) and token")
 | 
			
		||||
		b.Log.Info("Connecting using webhookurl (for posting) and token")
 | 
			
		||||
	}
 | 
			
		||||
	if !strings.HasPrefix(b.Config.Token, "Bot ") {
 | 
			
		||||
		b.Config.Token = "Bot " + b.Config.Token
 | 
			
		||||
	if !strings.HasPrefix(b.GetString("Token"), "Bot ") {
 | 
			
		||||
		token = "Bot " + b.GetString("Token")
 | 
			
		||||
	}
 | 
			
		||||
	b.c, err = discordgo.New(b.Config.Token)
 | 
			
		||||
	b.c, err = discordgo.New(token)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	b.c.AddHandler(b.messageCreate)
 | 
			
		||||
	b.c.AddHandler(b.memberUpdate)
 | 
			
		||||
	b.c.AddHandler(b.messageUpdate)
 | 
			
		||||
	b.c.AddHandler(b.messageDelete)
 | 
			
		||||
	err = b.c.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	guilds, err := b.c.UserGuilds()
 | 
			
		||||
	guilds, err := b.c.UserGuilds(100, "", "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	userinfo, err := b.c.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Nick = userinfo.Username
 | 
			
		||||
	for _, guild := range guilds {
 | 
			
		||||
		if guild.Name == b.Config.Server {
 | 
			
		||||
		if guild.Name == b.GetString("Server") {
 | 
			
		||||
			b.Channels, err = b.c.GuildChannels(guild.ID)
 | 
			
		||||
			b.guildID = guild.ID
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				flog.Debugf("%#v", err)
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	for _, channel := range b.Channels {
 | 
			
		||||
		b.Log.Debugf("found channel %#v", channel)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
func (b *Bdiscord) Disconnect() error {
 | 
			
		||||
	return b.c.Close()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) JoinChannel(channel string) error {
 | 
			
		||||
	idcheck := strings.Split(channel, "ID:")
 | 
			
		||||
func (b *Bdiscord) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	b.channelInfoMap[channel.ID] = &channel
 | 
			
		||||
	idcheck := strings.Split(channel.Name, "ID:")
 | 
			
		||||
	if len(idcheck) > 1 {
 | 
			
		||||
		b.UseChannelID = true
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
func (b *Bdiscord) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	channelID := b.getChannelID(msg.Channel)
 | 
			
		||||
	if channelID == "" {
 | 
			
		||||
		flog.Errorf("Could not find channelID for %v", msg.Channel)
 | 
			
		||||
		return nil
 | 
			
		||||
		return "", fmt.Errorf("Could not find channelID for %v", msg.Channel)
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.WebhookURL == "" {
 | 
			
		||||
		flog.Debugf("Broadcasting using token (API)")
 | 
			
		||||
		b.c.ChannelMessageSend(channelID, msg.Username+msg.Text)
 | 
			
		||||
	} else {
 | 
			
		||||
		flog.Debugf("Broadcasting using Webhook")
 | 
			
		||||
		b.c.WebhookExecute(
 | 
			
		||||
			b.webhookID,
 | 
			
		||||
			b.webhookToken,
 | 
			
		||||
 | 
			
		||||
	// Make a action /me of the message
 | 
			
		||||
	if msg.Event == config.EVENT_USER_ACTION {
 | 
			
		||||
		msg.Text = "_" + msg.Text + "_"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// use initial webhook
 | 
			
		||||
	wID := b.webhookID
 | 
			
		||||
	wToken := b.webhookToken
 | 
			
		||||
 | 
			
		||||
	// check if have a channel specific webhook
 | 
			
		||||
	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok {
 | 
			
		||||
		if ci.Options.WebhookURL != "" {
 | 
			
		||||
			wID, wToken = b.splitURL(ci.Options.WebhookURL)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Use webhook to send the message
 | 
			
		||||
	if wID != "" {
 | 
			
		||||
		// skip events
 | 
			
		||||
		if msg.Event != "" {
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Debugf("Broadcasting using Webhook")
 | 
			
		||||
		for _, f := range msg.Extra["file"] {
 | 
			
		||||
			fi := f.(config.FileInfo)
 | 
			
		||||
			if fi.URL != "" {
 | 
			
		||||
				msg.Text += fi.URL + " "
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		err := b.c.WebhookExecute(
 | 
			
		||||
			wID,
 | 
			
		||||
			wToken,
 | 
			
		||||
			true,
 | 
			
		||||
			&discordgo.WebhookParams{
 | 
			
		||||
				Content:   msg.Text,
 | 
			
		||||
				Username:  msg.Username,
 | 
			
		||||
				AvatarURL: msg.Avatar,
 | 
			
		||||
			})
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
	b.Log.Debugf("Broadcasting using token (API)")
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		if msg.ID == "" {
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
		err := b.c.ChannelMessageDelete(channelID, msg.ID)
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	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 {
 | 
			
		||||
			return b.handleUploadFile(&msg, channelID)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message
 | 
			
		||||
	if msg.ID != "" {
 | 
			
		||||
		_, err := b.c.ChannelMessageEdit(channelID, msg.ID, msg.Username+msg.Text)
 | 
			
		||||
		return msg.ID, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	res, err := b.c.ChannelMessageSend(channelID, msg.Username+msg.Text)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return res.ID, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
 | 
			
		||||
	if b.Config.EditDisable {
 | 
			
		||||
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
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.Debugf("<= Sending message from %s to gateway", b.Account)
 | 
			
		||||
	b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
	b.Remote <- rmsg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) {
 | 
			
		||||
	if b.GetBool("EditDisable") {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// only when message is actually edited
 | 
			
		||||
	if m.Message.EditedTimestamp != "" {
 | 
			
		||||
		flog.Debugf("Sending edit message")
 | 
			
		||||
		m.Content = m.Content + b.Config.EditSuffix
 | 
			
		||||
		b.Log.Debugf("Sending edit message")
 | 
			
		||||
		m.Content = m.Content + b.GetString("EditSuffix")
 | 
			
		||||
		b.messageCreate(s, (*discordgo.MessageCreate)(m))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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
 | 
			
		||||
	if m.Author.Username == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// 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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// add the url of the attachments to content
 | 
			
		||||
	if len(m.Attachments) > 0 {
 | 
			
		||||
		for _, attach := range m.Attachments {
 | 
			
		||||
			m.Content = m.Content + "\n" + attach.URL
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if m.Content == "" {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	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.replaceChannelMentions(m.Message.Content)
 | 
			
		||||
 | 
			
		||||
	text := m.ContentWithMentionsReplaced()
 | 
			
		||||
	if b.Config.ShowEmbeds && m.Message.Embeds != nil {
 | 
			
		||||
		for _, embed := range m.Message.Embeds {
 | 
			
		||||
			text = text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n"
 | 
			
		||||
	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}
 | 
			
		||||
 | 
			
		||||
	if m.Content != "" {
 | 
			
		||||
		b.Log.Debugf("== Receiving event %#v", m.Message)
 | 
			
		||||
		m.Message.Content = b.stripCustomoji(m.Message.Content)
 | 
			
		||||
		m.Message.Content = b.replaceChannelMentions(m.Message.Content)
 | 
			
		||||
		rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Errorf("ContentWithMoreMentionsReplaced failed: %s", err)
 | 
			
		||||
			rmsg.Text = m.ContentWithMentionsReplaced()
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	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,
 | 
			
		||||
		Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg",
 | 
			
		||||
		UserID: m.Author.ID}
 | 
			
		||||
	// set channel name
 | 
			
		||||
	rmsg.Channel = b.getChannelName(m.ChannelID)
 | 
			
		||||
	if b.UseChannelID {
 | 
			
		||||
		rmsg.Channel = "ID:" + m.ChannelID
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// set username
 | 
			
		||||
	if !b.GetBool("UseUserName") {
 | 
			
		||||
		rmsg.Username = b.getNick(m.Author)
 | 
			
		||||
	} else {
 | 
			
		||||
		rmsg.Username = m.Author.Username
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if we have embedded content add it to text
 | 
			
		||||
	if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil {
 | 
			
		||||
		for _, embed := range m.Message.Embeds {
 | 
			
		||||
			rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// no empty messages
 | 
			
		||||
	if rmsg.Text == "" {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// do we have a /me action
 | 
			
		||||
	var ok bool
 | 
			
		||||
	rmsg.Text, ok = b.replaceAction(rmsg.Text)
 | 
			
		||||
	if ok {
 | 
			
		||||
		rmsg.Event = config.EVENT_USER_ACTION
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.Log.Debugf("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account)
 | 
			
		||||
	b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
	b.Remote <- rmsg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) {
 | 
			
		||||
func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) {
 | 
			
		||||
	b.Lock()
 | 
			
		||||
	if _, ok := b.userMemberMap[m.Member.User.ID]; ok {
 | 
			
		||||
		flog.Debugf("%s: memberupdate: user %s (nick %s) changes nick to %s", b.Account, m.Member.User.Username, b.userMemberMap[m.Member.User.ID].Nick, m.Member.Nick)
 | 
			
		||||
		b.Log.Debugf("%s: memberupdate: user %s (nick %s) changes nick to %s", b.Account, m.Member.User.Username, b.userMemberMap[m.Member.User.ID].Nick, m.Member.Nick)
 | 
			
		||||
	}
 | 
			
		||||
	b.userMemberMap[m.Member.User.ID] = m.Member
 | 
			
		||||
	b.Unlock()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) getNick(user *discordgo.User) string {
 | 
			
		||||
func (b *Bdiscord) getNick(user *discordgo.User) string {
 | 
			
		||||
	var err error
 | 
			
		||||
	b.Lock()
 | 
			
		||||
	defer b.Unlock()
 | 
			
		||||
@@ -222,7 +312,7 @@ func (b *bdiscord) getNick(user *discordgo.User) string {
 | 
			
		||||
	return user.Username
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) getChannelID(name string) string {
 | 
			
		||||
func (b *Bdiscord) getChannelID(name string) string {
 | 
			
		||||
	idcheck := strings.Split(name, "ID:")
 | 
			
		||||
	if len(idcheck) > 1 {
 | 
			
		||||
		return idcheck[1]
 | 
			
		||||
@@ -235,7 +325,7 @@ func (b *bdiscord) getChannelID(name string) string {
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) getChannelName(id string) string {
 | 
			
		||||
func (b *Bdiscord) getChannelName(id string) string {
 | 
			
		||||
	for _, channel := range b.Channels {
 | 
			
		||||
		if channel.ID == id {
 | 
			
		||||
			return channel.Name
 | 
			
		||||
@@ -244,19 +334,7 @@ func (b *bdiscord) getChannelName(id string) string {
 | 
			
		||||
	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
 | 
			
		||||
	re := regexp.MustCompile("<#[0-9]+>")
 | 
			
		||||
	text = re.ReplaceAllStringFunc(text, func(m string) string {
 | 
			
		||||
@@ -275,8 +353,71 @@ func (b *bdiscord) replaceChannelMentions(text string) string {
 | 
			
		||||
	return text
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *bdiscord) stripCustomoji(text string) string {
 | 
			
		||||
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 {
 | 
			
		||||
	// <:doge:302803592035958784>
 | 
			
		||||
	re := regexp.MustCompile("<(:.*?:)[0-9]+>")
 | 
			
		||||
	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 {
 | 
			
		||||
		b.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.GetString("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.GetString("WebhookURL") != "" {
 | 
			
		||||
		wID, _ := b.splitURL(b.GetString("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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Bdiscord) handleUploadFile(msg *config.Message, channelID string) (string, error) {
 | 
			
		||||
	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 {
 | 
			
		||||
			return "", fmt.Errorf("file upload failed: %#v", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -2,47 +2,38 @@ package bgitter
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/go-gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/sromku/go-gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bgitter struct {
 | 
			
		||||
	c       *gitter.Gitter
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
	Users   []gitter.User
 | 
			
		||||
	Rooms   []gitter.Room
 | 
			
		||||
	c     *gitter.Gitter
 | 
			
		||||
	User  *gitter.User
 | 
			
		||||
	Users []gitter.User
 | 
			
		||||
	Rooms []gitter.Room
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "gitter"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bgitter {
 | 
			
		||||
	b := &Bgitter{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Bgitter{Config: cfg}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	b.c = gitter.New(b.Config.Token)
 | 
			
		||||
	_, err = b.c.GetUser()
 | 
			
		||||
	b.Log.Info("Connecting")
 | 
			
		||||
	b.c = gitter.New(b.GetString("Token"))
 | 
			
		||||
	b.User, err = b.c.GetUser()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.Rooms, _ = b.c.GetRooms()
 | 
			
		||||
	b.Rooms, err = b.c.GetRooms()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -51,10 +42,10 @@ func (b *Bgitter) Disconnect() error {
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) JoinChannel(channel string) error {
 | 
			
		||||
	roomID, err := b.c.GetRoomId(channel)
 | 
			
		||||
func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	roomID, err := b.c.GetRoomId(channel.Name)
 | 
			
		||||
	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)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -78,29 +69,74 @@ func (b *Bgitter) JoinChannel(channel string) error {
 | 
			
		||||
		for event := range stream.Event {
 | 
			
		||||
			switch ev := event.Data.(type) {
 | 
			
		||||
			case *gitter.MessageReceived:
 | 
			
		||||
				// check for ZWSP to see if it's not an echo
 | 
			
		||||
				if !strings.HasSuffix(ev.Message.Text, "") {
 | 
			
		||||
					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account)
 | 
			
		||||
					b.Remote <- config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room,
 | 
			
		||||
						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID}
 | 
			
		||||
				// ignore message sent from ourselves
 | 
			
		||||
				if ev.Message.From.ID != b.User.ID {
 | 
			
		||||
					b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account)
 | 
			
		||||
					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,
 | 
			
		||||
						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)
 | 
			
		||||
					}
 | 
			
		||||
					b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
					b.Remote <- rmsg
 | 
			
		||||
				}
 | 
			
		||||
			case *gitter.GitterConnectionClosed:
 | 
			
		||||
				flog.Errorf("connection with gitter closed for room %s", room)
 | 
			
		||||
				b.Log.Errorf("connection with gitter closed for room %s", room)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}(stream, room.Name)
 | 
			
		||||
	}(stream, room.URI)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
func (b *Bgitter) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
	roomID := b.getRoomID(msg.Channel)
 | 
			
		||||
	if roomID == "" {
 | 
			
		||||
		flog.Errorf("Could not find roomID for %v", msg.Channel)
 | 
			
		||||
		return nil
 | 
			
		||||
		b.Log.Errorf("Could not find roomID for %v", msg.Channel)
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
	// add ZWSP because gitter echoes our own messages
 | 
			
		||||
	return b.c.SendMessage(roomID, msg.Username+msg.Text+" ")
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		if msg.ID == "" {
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
		// gitter has no delete message api so we edit message to ""
 | 
			
		||||
		_, err := b.c.UpdateMessage(roomID, msg.ID, "")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Upload a file (in gitter case send the upload URL because gitter has no native upload support)
 | 
			
		||||
	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 {
 | 
			
		||||
			return b.handleUploadFile(&msg, roomID)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message
 | 
			
		||||
	if msg.ID != "" {
 | 
			
		||||
		b.Log.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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	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 {
 | 
			
		||||
@@ -123,3 +159,23 @@ func (b *Bgitter) getAvatar(user string) string {
 | 
			
		||||
	}
 | 
			
		||||
	return avatar
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bgitter) handleUploadFile(msg *config.Message, roomID string) (string, error) {
 | 
			
		||||
	for _, f := range msg.Extra["file"] {
 | 
			
		||||
		fi := f.(config.FileInfo)
 | 
			
		||||
		if fi.Comment != "" {
 | 
			
		||||
			msg.Text += fi.Comment + ": "
 | 
			
		||||
		}
 | 
			
		||||
		if fi.URL != "" {
 | 
			
		||||
			msg.Text = fi.URL
 | 
			
		||||
			if fi.Comment != "" {
 | 
			
		||||
				msg.Text = fi.Comment + ": " + fi.URL
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		_, err := b.c.SendMessage(roomID, msg.Username+msg.Text)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										102
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										102
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,102 @@
 | 
			
		||||
package helper
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func DownloadFile(url string) (*[]byte, error) {
 | 
			
		||||
	return DownloadFileAuth(url, "")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func DownloadFileAuth(url string, auth string) (*[]byte, error) {
 | 
			
		||||
	var buf bytes.Buffer
 | 
			
		||||
	client := &http.Client{
 | 
			
		||||
		Timeout: time.Second * 5,
 | 
			
		||||
	}
 | 
			
		||||
	req, err := http.NewRequest("GET", url, nil)
 | 
			
		||||
	if auth != "" {
 | 
			
		||||
		req.Header.Add("Authorization", auth)
 | 
			
		||||
	}
 | 
			
		||||
	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 ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func HandleDownloadSize(flog *log.Entry, msg *config.Message, name string, size int64, general *config.Protocol) error {
 | 
			
		||||
	flog.Debugf("Trying to download %#v with size %#v", name, size)
 | 
			
		||||
	if int(size) > general.MediaDownloadSize {
 | 
			
		||||
		msg.Event = config.EVENT_FILE_FAILURE_SIZE
 | 
			
		||||
		msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: name, Comment: msg.Text, Size: size})
 | 
			
		||||
		return fmt.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, general.MediaDownloadSize)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func HandleDownloadData(flog *log.Entry, msg *config.Message, name, comment, url string, data *[]byte, general *config.Protocol) {
 | 
			
		||||
	var avatar bool
 | 
			
		||||
	flog.Debugf("Download OK %#v %#v", name, len(*data))
 | 
			
		||||
	if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
 | 
			
		||||
		avatar = true
 | 
			
		||||
	}
 | 
			
		||||
	msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, URL: url, Comment: comment, Avatar: avatar})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func RemoveEmptyNewLines(msg string) string {
 | 
			
		||||
	lines := ""
 | 
			
		||||
	for _, line := range strings.Split(msg, "\n") {
 | 
			
		||||
		if line != "" {
 | 
			
		||||
			lines += line + "\n"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	lines = strings.TrimRight(lines, "\n")
 | 
			
		||||
	return lines
 | 
			
		||||
}
 | 
			
		||||
@@ -4,6 +4,7 @@ import (
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
/*
 | 
			
		||||
func tableformatter(nicks []string, nicksPerRow int, continued bool) string {
 | 
			
		||||
	result := "|IRC users"
 | 
			
		||||
	if continued {
 | 
			
		||||
@@ -29,6 +30,7 @@ func tableformatter(nicks []string, nicksPerRow int, continued bool) string {
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
func plainformatter(nicks []string, nicksPerRow int) string {
 | 
			
		||||
	return strings.Join(nicks, ", ") + " currently on IRC"
 | 
			
		||||
 
 | 
			
		||||
@@ -1,59 +1,61 @@
 | 
			
		||||
package birc
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/dfordsoft/golib/ic"
 | 
			
		||||
	"github.com/lrstanley/girc"
 | 
			
		||||
	"github.com/paulrosania/go-charset/charset"
 | 
			
		||||
	_ "github.com/paulrosania/go-charset/data"
 | 
			
		||||
	"github.com/saintfish/chardet"
 | 
			
		||||
	ircm "github.com/sorcix/irc"
 | 
			
		||||
	"github.com/thoj/go-ircevent"
 | 
			
		||||
	"hash/crc32"
 | 
			
		||||
	"io"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
	"unicode/utf8"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Birc struct {
 | 
			
		||||
	i               *irc.Connection
 | 
			
		||||
	Nick            string
 | 
			
		||||
	names           map[string][]string
 | 
			
		||||
	Config          *config.Protocol
 | 
			
		||||
	Remote          chan config.Message
 | 
			
		||||
	connected       chan struct{}
 | 
			
		||||
	Local           chan config.Message // local queue for flood control
 | 
			
		||||
	Account         string
 | 
			
		||||
	FirstConnection bool
 | 
			
		||||
	i                                         *girc.Client
 | 
			
		||||
	Nick                                      string
 | 
			
		||||
	names                                     map[string][]string
 | 
			
		||||
	connected                                 chan struct{}
 | 
			
		||||
	Local                                     chan config.Message // local queue for flood control
 | 
			
		||||
	FirstConnection                           bool
 | 
			
		||||
	MessageDelay, MessageQueue, MessageLength int
 | 
			
		||||
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "irc"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Birc {
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Birc{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Nick = b.Config.Nick
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Config = cfg
 | 
			
		||||
	b.Nick = b.GetString("Nick")
 | 
			
		||||
	b.names = make(map[string][]string)
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.connected = make(chan struct{})
 | 
			
		||||
	if b.Config.MessageDelay == 0 {
 | 
			
		||||
		b.Config.MessageDelay = 1300
 | 
			
		||||
	if b.GetInt("MessageDelay") == 0 {
 | 
			
		||||
		b.MessageDelay = 1300
 | 
			
		||||
	} else {
 | 
			
		||||
		b.MessageDelay = b.GetInt("MessageDelay")
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.MessageQueue == 0 {
 | 
			
		||||
		b.Config.MessageQueue = 30
 | 
			
		||||
	if b.GetInt("MessageQueue") == 0 {
 | 
			
		||||
		b.MessageQueue = 30
 | 
			
		||||
	} else {
 | 
			
		||||
		b.MessageQueue = b.GetInt("MessageQueue")
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.MessageLength == 0 {
 | 
			
		||||
		b.Config.MessageLength = 400
 | 
			
		||||
	if b.GetInt("MessageLength") == 0 {
 | 
			
		||||
		b.MessageLength = 400
 | 
			
		||||
	} else {
 | 
			
		||||
		b.MessageLength = b.GetInt("MessageLength")
 | 
			
		||||
	}
 | 
			
		||||
	b.FirstConnection = true
 | 
			
		||||
	return b
 | 
			
		||||
@@ -62,101 +64,205 @@ func New(cfg config.Protocol, account string, c chan config.Message) *Birc {
 | 
			
		||||
func (b *Birc) Command(msg *config.Message) string {
 | 
			
		||||
	switch msg.Text {
 | 
			
		||||
	case "!users":
 | 
			
		||||
		b.i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames)
 | 
			
		||||
		b.i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames)
 | 
			
		||||
		b.i.SendRaw("NAMES " + msg.Channel)
 | 
			
		||||
		b.i.Handlers.Add(girc.RPL_NAMREPLY, b.storeNames)
 | 
			
		||||
		b.i.Handlers.Add(girc.RPL_ENDOFNAMES, b.endNames)
 | 
			
		||||
		b.i.Cmd.SendRaw("NAMES " + msg.Channel)
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Connect() error {
 | 
			
		||||
	b.Local = make(chan config.Message, b.Config.MessageQueue+10)
 | 
			
		||||
	flog.Infof("Connecting %s", b.Config.Server)
 | 
			
		||||
	i := irc.IRC(b.Config.Nick, b.Config.Nick)
 | 
			
		||||
	if log.GetLevel() == log.DebugLevel {
 | 
			
		||||
		i.Debug = true
 | 
			
		||||
	}
 | 
			
		||||
	i.UseTLS = b.Config.UseTLS
 | 
			
		||||
	i.UseSASL = b.Config.UseSASL
 | 
			
		||||
	i.SASLLogin = b.Config.NickServNick
 | 
			
		||||
	i.SASLPassword = b.Config.NickServPassword
 | 
			
		||||
	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify}
 | 
			
		||||
	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)
 | 
			
		||||
	b.Local = make(chan config.Message, b.MessageQueue+10)
 | 
			
		||||
	b.Log.Infof("Connecting %s", b.GetString("Server"))
 | 
			
		||||
	server, portstr, err := net.SplitHostPort(b.GetString("Server"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	port, err := strconv.Atoi(portstr)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	// fix strict user handling of girc
 | 
			
		||||
	user := b.GetString("Nick")
 | 
			
		||||
	for !girc.IsValidUser(user) {
 | 
			
		||||
		if len(user) == 1 {
 | 
			
		||||
			user = "matterbridge"
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		user = user[1:]
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	i := girc.New(girc.Config{
 | 
			
		||||
		Server:     server,
 | 
			
		||||
		ServerPass: b.GetString("Password"),
 | 
			
		||||
		Port:       port,
 | 
			
		||||
		Nick:       b.GetString("Nick"),
 | 
			
		||||
		User:       user,
 | 
			
		||||
		Name:       b.GetString("Nick"),
 | 
			
		||||
		SSL:        b.GetBool("UseTLS"),
 | 
			
		||||
		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server},
 | 
			
		||||
		PingDelay:  time.Minute,
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	if b.GetBool("UseSASL") {
 | 
			
		||||
		i.Config.SASL = &girc.SASLPlain{b.GetString("NickServNick"), b.GetString("NickServPassword")}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	i.Handlers.Add(girc.RPL_WELCOME, b.handleNewConnection)
 | 
			
		||||
	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth)
 | 
			
		||||
	i.Handlers.Add(girc.ALL_EVENTS, b.handleOther)
 | 
			
		||||
	go func() {
 | 
			
		||||
		for {
 | 
			
		||||
			if err := i.Connect(); err != nil {
 | 
			
		||||
				b.Log.Errorf("error: %s", err)
 | 
			
		||||
				b.Log.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
 | 
			
		||||
	select {
 | 
			
		||||
	case <-b.connected:
 | 
			
		||||
		flog.Info("Connection succeeded")
 | 
			
		||||
		b.Log.Info("Connection succeeded")
 | 
			
		||||
	case <-time.After(time.Second * 30):
 | 
			
		||||
		return fmt.Errorf("connection timed out")
 | 
			
		||||
	}
 | 
			
		||||
	i.Debug = false
 | 
			
		||||
	// clear on reconnects
 | 
			
		||||
	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()
 | 
			
		||||
	//i.Debug = false
 | 
			
		||||
	if b.GetInt("DebugLevel") == 0 {
 | 
			
		||||
		i.Handlers.Clear(girc.ALL_EVENTS)
 | 
			
		||||
	}
 | 
			
		||||
	go b.doSend()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Disconnect() error {
 | 
			
		||||
	//b.i.Disconnect()
 | 
			
		||||
	b.i.Close()
 | 
			
		||||
	close(b.Local)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) JoinChannel(channel string) error {
 | 
			
		||||
	b.i.Join(channel)
 | 
			
		||||
func (b *Birc) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	if channel.Options.Key != "" {
 | 
			
		||||
		b.Log.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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	if msg.Account == b.Account {
 | 
			
		||||
		return nil
 | 
			
		||||
func (b *Birc) Send(msg config.Message) (string, error) {
 | 
			
		||||
	// ignore delete messages
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	// we can be in between reconnects #385
 | 
			
		||||
	if !b.i.IsConnected() {
 | 
			
		||||
		b.Log.Error("Not connected to server, dropping message")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Execute a command
 | 
			
		||||
	if strings.HasPrefix(msg.Text, "!") {
 | 
			
		||||
		b.Command(&msg)
 | 
			
		||||
	}
 | 
			
		||||
	for _, text := range strings.Split(msg.Text, "\n") {
 | 
			
		||||
		if len(text) > b.Config.MessageLength {
 | 
			
		||||
			text = text[:b.Config.MessageLength] + " <message clipped>"
 | 
			
		||||
		}
 | 
			
		||||
		if len(b.Local) < b.Config.MessageQueue {
 | 
			
		||||
			if len(b.Local) == b.Config.MessageQueue-1 {
 | 
			
		||||
				text = text + " <message clipped>"
 | 
			
		||||
 | 
			
		||||
	// convert to specified charset
 | 
			
		||||
	if b.GetString("Charset") != "" {
 | 
			
		||||
		switch b.GetString("Charset") {
 | 
			
		||||
		case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp":
 | 
			
		||||
			msg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), msg.Text)
 | 
			
		||||
		default:
 | 
			
		||||
			buf := new(bytes.Buffer)
 | 
			
		||||
			w, err := charset.NewWriter(b.GetString("Charset"), buf)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Errorf("charset from utf-8 conversion failed: %s", err)
 | 
			
		||||
				return "", err
 | 
			
		||||
			}
 | 
			
		||||
			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel}
 | 
			
		||||
		} else {
 | 
			
		||||
			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local))
 | 
			
		||||
			fmt.Fprint(w, msg.Text)
 | 
			
		||||
			w.Close()
 | 
			
		||||
			msg.Text = buf.String()
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
	// Handle files
 | 
			
		||||
	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
 | 
			
		||||
					if fi.Comment != "" {
 | 
			
		||||
						msg.Text = fi.Comment + ": " + 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.GetBool("MessageSplit") {
 | 
			
		||||
		msg.Text = helper.SplitStringLength(msg.Text, b.MessageLength)
 | 
			
		||||
	}
 | 
			
		||||
	for _, text := range strings.Split(msg.Text, "\n") {
 | 
			
		||||
		if len(text) > b.MessageLength {
 | 
			
		||||
			text = text[:b.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.MessageQueue {
 | 
			
		||||
			if len(b.Local) == b.MessageQueue-1 {
 | 
			
		||||
				text = text + " <message clipped>"
 | 
			
		||||
			}
 | 
			
		||||
			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event}
 | 
			
		||||
		} else {
 | 
			
		||||
			b.Log.Debugf("flooding, dropping message (queue at %d)", len(b.Local))
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) doSend() {
 | 
			
		||||
	rate := time.Millisecond * time.Duration(b.Config.MessageDelay)
 | 
			
		||||
	throttle := time.Tick(rate)
 | 
			
		||||
	rate := time.Millisecond * time.Duration(b.MessageDelay)
 | 
			
		||||
	throttle := time.NewTicker(rate)
 | 
			
		||||
	for msg := range b.Local {
 | 
			
		||||
		<-throttle
 | 
			
		||||
		b.i.Privmsg(msg.Channel, msg.Username+msg.Text)
 | 
			
		||||
		<-throttle.C
 | 
			
		||||
		username := msg.Username
 | 
			
		||||
		if b.GetBool("Colornicks") {
 | 
			
		||||
			checksum := crc32.ChecksumIEEE([]byte(msg.Username))
 | 
			
		||||
			username = fmt.Sprintf("\x03%02d%s\x0F", checksum%0x10, msg.Username)
 | 
			
		||||
		}
 | 
			
		||||
		if msg.Event == config.EVENT_USER_ACTION {
 | 
			
		||||
			b.i.Cmd.Action(msg.Channel, username+msg.Text)
 | 
			
		||||
		} else {
 | 
			
		||||
			b.Log.Debugf("Sending to channel %s", msg.Channel)
 | 
			
		||||
			b.i.Cmd.Message(msg.Channel, username+msg.Text)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) endNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[1]
 | 
			
		||||
func (b *Birc) endNames(client *girc.Client, event girc.Event) {
 | 
			
		||||
	channel := event.Params[1]
 | 
			
		||||
	sort.Strings(b.names[channel])
 | 
			
		||||
	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
 | 
			
		||||
	continued := false
 | 
			
		||||
@@ -169,157 +275,188 @@ func (b *Birc) endNames(event *irc.Event) {
 | 
			
		||||
	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued),
 | 
			
		||||
		Channel: channel, Account: b.Account}
 | 
			
		||||
	b.names[channel] = nil
 | 
			
		||||
	b.i.ClearCallback(ircm.RPL_NAMREPLY)
 | 
			
		||||
	b.i.ClearCallback(ircm.RPL_ENDOFNAMES)
 | 
			
		||||
	b.i.Handlers.Clear(girc.RPL_NAMREPLY)
 | 
			
		||||
	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleNewConnection(event *irc.Event) {
 | 
			
		||||
	flog.Debug("Registering callbacks")
 | 
			
		||||
func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) {
 | 
			
		||||
	b.Log.Debug("Registering callbacks")
 | 
			
		||||
	i := b.i
 | 
			
		||||
	b.Nick = event.Arguments[0]
 | 
			
		||||
	i.AddCallback("PRIVMSG", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback("CTCP_ACTION", b.handlePrivMsg)
 | 
			
		||||
	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
 | 
			
		||||
	i.AddCallback(ircm.NOTICE, b.handleNotice)
 | 
			
		||||
	//i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) })
 | 
			
		||||
	i.AddCallback("PING", func(e *irc.Event) {
 | 
			
		||||
		i.SendRaw("PONG :" + e.Message())
 | 
			
		||||
		flog.Debugf("PING/PONG")
 | 
			
		||||
	})
 | 
			
		||||
	i.AddCallback("JOIN", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("PART", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("QUIT", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("KICK", b.handleJoinPart)
 | 
			
		||||
	i.AddCallback("*", b.handleOther)
 | 
			
		||||
	b.Nick = event.Params[0]
 | 
			
		||||
 | 
			
		||||
	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth)
 | 
			
		||||
	i.Handlers.Add("PRIVMSG", b.handlePrivMsg)
 | 
			
		||||
	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg)
 | 
			
		||||
	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
 | 
			
		||||
	i.Handlers.Add(girc.NOTICE, b.handleNotice)
 | 
			
		||||
	i.Handlers.Add("JOIN", b.handleJoinPart)
 | 
			
		||||
	i.Handlers.Add("PART", b.handleJoinPart)
 | 
			
		||||
	i.Handlers.Add("QUIT", b.handleJoinPart)
 | 
			
		||||
	i.Handlers.Add("KICK", b.handleJoinPart)
 | 
			
		||||
	// we are now fully connected
 | 
			
		||||
	b.connected <- struct{}{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleJoinPart(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[0]
 | 
			
		||||
	if event.Code == "KICK" {
 | 
			
		||||
		flog.Infof("Got kicked from %s by %s", channel, event.Nick)
 | 
			
		||||
func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) {
 | 
			
		||||
	if len(event.Params) == 0 {
 | 
			
		||||
		b.Log.Debugf("handleJoinPart: empty Params? %#v", event)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	channel := strings.ToLower(event.Params[0])
 | 
			
		||||
	if event.Command == "KICK" {
 | 
			
		||||
		b.Log.Infof("Got kicked from %s by %s", channel, event.Source.Name)
 | 
			
		||||
		time.Sleep(time.Duration(b.GetInt("RejoinDelay")) * time.Second)
 | 
			
		||||
		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if event.Code == "QUIT" {
 | 
			
		||||
		if event.Nick == b.Nick && strings.Contains(event.Raw, "Ping timeout") {
 | 
			
		||||
			flog.Infof("%s reconnecting ..", b.Account)
 | 
			
		||||
	if event.Command == "QUIT" {
 | 
			
		||||
		if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") {
 | 
			
		||||
			b.Log.Infof("%s reconnecting ..", b.Account)
 | 
			
		||||
			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE}
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if event.Nick != b.Nick {
 | 
			
		||||
		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}
 | 
			
		||||
	if event.Source.Name != b.Nick {
 | 
			
		||||
		if b.GetBool("nosendjoinpart") {
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account)
 | 
			
		||||
		msg := config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE}
 | 
			
		||||
		b.Log.Debugf("<= Message is %#v", msg)
 | 
			
		||||
		b.Remote <- msg
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("handle %#v", event)
 | 
			
		||||
	b.Log.Debugf("handle %#v", event)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleNotice(event *irc.Event) {
 | 
			
		||||
	if strings.Contains(event.Message(), "This nickname is registered") && event.Nick == b.Config.NickServNick {
 | 
			
		||||
		b.i.Privmsg(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword)
 | 
			
		||||
func (b *Birc) handleNotice(client *girc.Client, event girc.Event) {
 | 
			
		||||
	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.GetString("NickServNick") {
 | 
			
		||||
		b.i.Cmd.Message(b.GetString("NickServNick"), "IDENTIFY "+b.GetString("NickServPassword"))
 | 
			
		||||
	} else {
 | 
			
		||||
		b.handlePrivMsg(event)
 | 
			
		||||
		b.handlePrivMsg(client, event)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleOther(event *irc.Event) {
 | 
			
		||||
	switch event.Code {
 | 
			
		||||
func (b *Birc) handleOther(client *girc.Client, event girc.Event) {
 | 
			
		||||
	if b.GetInt("DebugLevel") == 1 {
 | 
			
		||||
		if event.Command != "CLIENT_STATE_UPDATED" &&
 | 
			
		||||
			event.Command != "CLIENT_GENERAL_UPDATED" {
 | 
			
		||||
			b.Log.Debugf("%#v", event.String())
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	switch event.Command {
 | 
			
		||||
	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005":
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("%#v", event.Raw)
 | 
			
		||||
	b.Log.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.GetString("NickServNick"), "Q@CServe.quakenet.org") {
 | 
			
		||||
		b.Log.Debugf("Authenticating %s against %s", b.GetString("NickServUsername"), b.GetString("NickServNick"))
 | 
			
		||||
		b.i.Cmd.Message(b.GetString("NickServNick"), "AUTH "+b.GetString("NickServUsername")+" "+b.GetString("NickServPassword"))
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) skipPrivMsg(event girc.Event) bool {
 | 
			
		||||
	// Our nick can be changed
 | 
			
		||||
	b.Nick = b.i.GetNick()
 | 
			
		||||
 | 
			
		||||
	// freenode doesn't send 001 as first reply
 | 
			
		||||
	if event.Code == "NOTICE" {
 | 
			
		||||
		return
 | 
			
		||||
	if event.Command == "NOTICE" {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	// don't forward queries to the bot
 | 
			
		||||
	if event.Arguments[0] == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	if event.Params[0] == b.Nick {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	// don't forward message from ourself
 | 
			
		||||
	if event.Nick == b.Nick {
 | 
			
		||||
		return
 | 
			
		||||
	if event.Source.Name == b.Nick {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("handlePrivMsg() %s %s %#v", event.Nick, event.Message(), event)
 | 
			
		||||
	msg := ""
 | 
			
		||||
	if event.Code == "CTCP_ACTION" {
 | 
			
		||||
		msg = event.Nick + " "
 | 
			
		||||
	}
 | 
			
		||||
	msg += event.Message()
 | 
			
		||||
	// strip IRC colors
 | 
			
		||||
	re := regexp.MustCompile(`[[:cntrl:]](\d+,|)\d+`)
 | 
			
		||||
	msg = re.ReplaceAllString(msg, "")
 | 
			
		||||
 | 
			
		||||
	// detect what were sending so that we convert it to utf-8
 | 
			
		||||
	detector := chardet.NewTextDetector()
 | 
			
		||||
	result, err := detector.DetectBest([]byte(msg))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Infof("detection failed for msg: %#v", msg)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence)
 | 
			
		||||
	var r io.Reader
 | 
			
		||||
	r, err = charset.NewReader(result.Charset, strings.NewReader(msg))
 | 
			
		||||
	// if we're not sure, just pick ISO-8859-1
 | 
			
		||||
	if result.Confidence < 80 {
 | 
			
		||||
		r, err = charset.NewReader("ISO-8859-1", strings.NewReader(msg))
 | 
			
		||||
	}
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Errorf("charset to utf-8 conversion failed: %s", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	output, _ := ioutil.ReadAll(r)
 | 
			
		||||
	msg = string(output)
 | 
			
		||||
 | 
			
		||||
	flog.Debugf("Sending message from %s on %s to gateway", event.Arguments[0], b.Account)
 | 
			
		||||
	b.Remote <- config.Message{Username: event.Nick, Text: msg, Channel: event.Arguments[0], Account: b.Account, UserID: event.User + "@" + event.Host}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleTopicWhoTime(event *irc.Event) {
 | 
			
		||||
	parts := strings.Split(event.Arguments[2], "!")
 | 
			
		||||
	t, err := strconv.ParseInt(event.Arguments[3], 10, 64)
 | 
			
		||||
func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) {
 | 
			
		||||
	if b.skipPrivMsg(event) {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host}
 | 
			
		||||
	b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Trailing, event)
 | 
			
		||||
 | 
			
		||||
	// set action event
 | 
			
		||||
	if event.IsAction() {
 | 
			
		||||
		rmsg.Event = config.EVENT_USER_ACTION
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// strip action, we made an event if it was an action
 | 
			
		||||
	rmsg.Text += event.StripAction()
 | 
			
		||||
 | 
			
		||||
	// strip IRC colors
 | 
			
		||||
	re := regexp.MustCompile(`[[:cntrl:]](?:\d{1,2}(?:,\d{1,2})?)?`)
 | 
			
		||||
	rmsg.Text = re.ReplaceAllString(rmsg.Text, "")
 | 
			
		||||
 | 
			
		||||
	// start detecting the charset
 | 
			
		||||
	var r io.Reader
 | 
			
		||||
	var err error
 | 
			
		||||
	mycharset := b.GetString("Charset")
 | 
			
		||||
	if mycharset == "" {
 | 
			
		||||
		// detect what were sending so that we convert it to utf-8
 | 
			
		||||
		detector := chardet.NewTextDetector()
 | 
			
		||||
		result, err := detector.DetectBest([]byte(rmsg.Text))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Infof("detection failed for rmsg.Text: %#v", rmsg.Text)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Debugf("detected %s confidence %#v", result.Charset, result.Confidence)
 | 
			
		||||
		mycharset = result.Charset
 | 
			
		||||
		// if we're not sure, just pick ISO-8859-1
 | 
			
		||||
		if result.Confidence < 80 {
 | 
			
		||||
			mycharset = "ISO-8859-1"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	switch mycharset {
 | 
			
		||||
	case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp":
 | 
			
		||||
		rmsg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), rmsg.Text)
 | 
			
		||||
	default:
 | 
			
		||||
		r, err = charset.NewReader(mycharset, strings.NewReader(rmsg.Text))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Errorf("charset to utf-8 conversion failed: %s", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		output, _ := ioutil.ReadAll(r)
 | 
			
		||||
		rmsg.Text = string(output)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.Log.Debugf("<= Sending message from %s on %s to gateway", event.Params[0], b.Account)
 | 
			
		||||
	b.Remote <- rmsg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) {
 | 
			
		||||
	parts := strings.Split(event.Params[2], "!")
 | 
			
		||||
	t, err := strconv.ParseInt(event.Params[3], 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Errorf("Invalid time stamp: %s", event.Arguments[3])
 | 
			
		||||
		b.Log.Errorf("Invalid time stamp: %s", event.Params[3])
 | 
			
		||||
	}
 | 
			
		||||
	user := parts[0]
 | 
			
		||||
	if len(parts) > 1 {
 | 
			
		||||
		user += " [" + parts[1] + "]"
 | 
			
		||||
	}
 | 
			
		||||
	flog.Debugf("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0))
 | 
			
		||||
	b.Log.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) nicksPerRow() int {
 | 
			
		||||
	return 4
 | 
			
		||||
	/*
 | 
			
		||||
		if b.Config.Mattermost.NicksPerRow < 1 {
 | 
			
		||||
			return 4
 | 
			
		||||
		}
 | 
			
		||||
		return b.Config.Mattermost.NicksPerRow
 | 
			
		||||
	*/
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) storeNames(event *irc.Event) {
 | 
			
		||||
	channel := event.Arguments[2]
 | 
			
		||||
func (b *Birc) storeNames(client *girc.Client, event girc.Event) {
 | 
			
		||||
	channel := event.Params[2]
 | 
			
		||||
	b.names[channel] = append(
 | 
			
		||||
		b.names[channel],
 | 
			
		||||
		strings.Split(strings.TrimSpace(event.Message()), " ")...)
 | 
			
		||||
		strings.Split(strings.TrimSpace(event.Trailing), " ")...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Birc) formatnicks(nicks []string, continued bool) string {
 | 
			
		||||
	return plainformatter(nicks, b.nicksPerRow())
 | 
			
		||||
	/*
 | 
			
		||||
		switch b.Config.Mattermost.NickFormatter {
 | 
			
		||||
		case "table":
 | 
			
		||||
			return tableformatter(nicks, b.nicksPerRow(), continued)
 | 
			
		||||
		default:
 | 
			
		||||
			return plainformatter(nicks, b.nicksPerRow())
 | 
			
		||||
		}
 | 
			
		||||
	*/
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,60 +1,50 @@
 | 
			
		||||
package bmatrix
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"sync"
 | 
			
		||||
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	matrix "github.com/matrix-org/gomatrix"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	matrix "github.com/matterbridge/gomatrix"
 | 
			
		||||
	"mime"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bmatrix struct {
 | 
			
		||||
	mc      *matrix.Client
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
	UserID  string
 | 
			
		||||
	RoomMap map[string]string
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "matrix"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bmatrix {
 | 
			
		||||
	b := &Bmatrix{}
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Bmatrix{Config: cfg}
 | 
			
		||||
	b.RoomMap = make(map[string]string)
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmatrix) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Infof("Connecting %s", b.Config.Server)
 | 
			
		||||
	b.mc, err = matrix.NewClient(b.Config.Server, "", "")
 | 
			
		||||
	b.Log.Infof("Connecting %s", b.GetString("Server"))
 | 
			
		||||
	b.mc, err = matrix.NewClient(b.GetString("Server"), "", "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp, err := b.mc.Login(&matrix.ReqLogin{
 | 
			
		||||
		Type:     "m.login.password",
 | 
			
		||||
		User:     b.Config.Login,
 | 
			
		||||
		Password: b.Config.Password,
 | 
			
		||||
		User:     b.GetString("Login"),
 | 
			
		||||
		Password: b.GetString("Password"),
 | 
			
		||||
	})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.mc.SetCredentials(resp.UserID, resp.AccessToken)
 | 
			
		||||
	b.UserID = resp.UserID
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	go b.handlematrix()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -63,23 +53,65 @@ func (b *Bmatrix) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmatrix) JoinChannel(channel string) error {
 | 
			
		||||
	resp, err := b.mc.JoinRoom(channel, "", nil)
 | 
			
		||||
func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	resp, err := b.mc.JoinRoom(channel.Name, "", nil)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Lock()
 | 
			
		||||
	b.RoomMap[resp.RoomID] = channel
 | 
			
		||||
	b.RoomMap[resp.RoomID] = channel.Name
 | 
			
		||||
	b.Unlock()
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmatrix) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
func (b *Bmatrix) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	channel := b.getRoomID(msg.Channel)
 | 
			
		||||
	flog.Debugf("Sending to channel %s", channel)
 | 
			
		||||
	b.mc.SendText(channel, msg.Username+msg.Text)
 | 
			
		||||
	return nil
 | 
			
		||||
	b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel)
 | 
			
		||||
 | 
			
		||||
	// Make a action /me of the message
 | 
			
		||||
	if msg.Event == config.EVENT_USER_ACTION {
 | 
			
		||||
		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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	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 {
 | 
			
		||||
			return b.handleUploadFile(&msg, channel)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message if we have an ID
 | 
			
		||||
	// matrix has no editing support
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	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 {
 | 
			
		||||
@@ -92,33 +124,188 @@ func (b *Bmatrix) getRoomID(channel string) string {
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmatrix) handlematrix() error {
 | 
			
		||||
	syncer := b.mc.Syncer.(*matrix.DefaultSyncer)
 | 
			
		||||
	syncer.OnEventType("m.room.message", func(ev *matrix.Event) {
 | 
			
		||||
		if ev.Content["msgtype"].(string) == "m.text" && ev.Sender != b.UserID {
 | 
			
		||||
			b.RLock()
 | 
			
		||||
			channel, ok := b.RoomMap[ev.RoomID]
 | 
			
		||||
			b.RUnlock()
 | 
			
		||||
			if !ok {
 | 
			
		||||
				flog.Debugf("Unknown room %s", ev.RoomID)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			username := ev.Sender[1:]
 | 
			
		||||
			if b.Config.NoHomeServerSuffix {
 | 
			
		||||
				re := regexp.MustCompile("(.*?):.*")
 | 
			
		||||
				username = re.ReplaceAllString(username, `$1`)
 | 
			
		||||
			}
 | 
			
		||||
			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}
 | 
			
		||||
		}
 | 
			
		||||
		flog.Debugf("Received: %#v", ev)
 | 
			
		||||
	})
 | 
			
		||||
	syncer.OnEventType("m.room.redaction", b.handleEvent)
 | 
			
		||||
	syncer.OnEventType("m.room.message", b.handleEvent)
 | 
			
		||||
	go func() {
 | 
			
		||||
		for {
 | 
			
		||||
			if err := b.mc.Sync(); err != nil {
 | 
			
		||||
				flog.Println("Sync() returned ", err)
 | 
			
		||||
				b.Log.Println("Sync() returned ", err)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmatrix) handleEvent(ev *matrix.Event) {
 | 
			
		||||
	b.Log.Debugf("== Receiving event: %#v", ev)
 | 
			
		||||
	if ev.Sender != b.UserID {
 | 
			
		||||
		b.RLock()
 | 
			
		||||
		channel, ok := b.RoomMap[ev.RoomID]
 | 
			
		||||
		b.RUnlock()
 | 
			
		||||
		if !ok {
 | 
			
		||||
			b.Log.Debugf("Unknown room %s", ev.RoomID)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// TODO download avatar
 | 
			
		||||
 | 
			
		||||
		// Create our message
 | 
			
		||||
		rmsg := config.Message{Username: ev.Sender[1:], Channel: channel, Account: b.Account, UserID: ev.Sender, ID: ev.ID}
 | 
			
		||||
 | 
			
		||||
		// Text must be a string
 | 
			
		||||
		if rmsg.Text, ok = ev.Content["body"].(string); !ok {
 | 
			
		||||
			b.Log.Errorf("Content[body] wasn't a %T ?", rmsg.Text)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Remove homeserver suffix if configured
 | 
			
		||||
		if b.GetBool("NoHomeServerSuffix") {
 | 
			
		||||
			re := regexp.MustCompile("(.*?):.*")
 | 
			
		||||
			rmsg.Username = re.ReplaceAllString(rmsg.Username, `$1`)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Delete event
 | 
			
		||||
		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
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Do we have a /me action
 | 
			
		||||
		if ev.Content["msgtype"].(string) == "m.emote" {
 | 
			
		||||
			rmsg.Event = config.EVENT_USER_ACTION
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Do we have attachments
 | 
			
		||||
		if b.containsAttachment(ev.Content) {
 | 
			
		||||
			err := b.handleDownloadFile(&rmsg, ev.Content)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Errorf("download failed: %#v", err)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Sender, b.Account)
 | 
			
		||||
		b.Remote <- rmsg
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleDownloadFile handles file download
 | 
			
		||||
func (b *Bmatrix) handleDownloadFile(rmsg *config.Message, content map[string]interface{}) error {
 | 
			
		||||
	var (
 | 
			
		||||
		ok                        bool
 | 
			
		||||
		url, name, msgtype, mtype string
 | 
			
		||||
		info                      map[string]interface{}
 | 
			
		||||
		size                      float64
 | 
			
		||||
	)
 | 
			
		||||
 | 
			
		||||
	rmsg.Extra = make(map[string][]interface{})
 | 
			
		||||
	if url, ok = content["url"].(string); !ok {
 | 
			
		||||
		return fmt.Errorf("url isn't a %T", url)
 | 
			
		||||
	}
 | 
			
		||||
	url = strings.Replace(url, "mxc://", b.GetString("Server")+"/_matrix/media/v1/download/", -1)
 | 
			
		||||
 | 
			
		||||
	if info, ok = content["info"].(map[string]interface{}); !ok {
 | 
			
		||||
		return fmt.Errorf("info isn't a %T", info)
 | 
			
		||||
	}
 | 
			
		||||
	if size, ok = info["size"].(float64); !ok {
 | 
			
		||||
		return fmt.Errorf("size isn't a %T", size)
 | 
			
		||||
	}
 | 
			
		||||
	if name, ok = content["body"].(string); !ok {
 | 
			
		||||
		return fmt.Errorf("name isn't a %T", name)
 | 
			
		||||
	}
 | 
			
		||||
	if msgtype, ok = content["msgtype"].(string); !ok {
 | 
			
		||||
		return fmt.Errorf("msgtype isn't a %T", msgtype)
 | 
			
		||||
	}
 | 
			
		||||
	if mtype, ok = info["mimetype"].(string); !ok {
 | 
			
		||||
		return fmt.Errorf("mtype isn't a %T", mtype)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// check if we have an image uploaded without extension
 | 
			
		||||
	if !strings.Contains(name, ".") {
 | 
			
		||||
		if msgtype == "m.image" {
 | 
			
		||||
			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"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// check if the size is ok
 | 
			
		||||
	err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	// actually download the file
 | 
			
		||||
	data, err := helper.DownloadFile(url)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("download %s failed %#v", url, err)
 | 
			
		||||
	}
 | 
			
		||||
	// add the downloaded data to the message
 | 
			
		||||
	helper.HandleDownloadData(b.Log, rmsg, name, "", url, data, b.General)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string) (string, error) {
 | 
			
		||||
	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 {
 | 
			
		||||
					b.Log.Errorf("file comment failed: %#v", err)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			b.Log.Debugf("uploading file: %s %s", fi.Name, mtype)
 | 
			
		||||
			res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data)))
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Errorf("file upload failed: %#v", err)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			if strings.Contains(mtype, "video") {
 | 
			
		||||
				b.Log.Debugf("sendVideo %s", res.ContentURI)
 | 
			
		||||
				_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					b.Log.Errorf("sendVideo failed: %#v", err)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			if strings.Contains(mtype, "image") {
 | 
			
		||||
				b.Log.Debugf("sendImage %s", res.ContentURI)
 | 
			
		||||
				_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					b.Log.Errorf("sendImage failed: %#v", err)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			b.Log.Debugf("result: %#v", res)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// skipMessages returns true if this message should not be handled
 | 
			
		||||
func (b *Bmatrix) containsAttachment(content map[string]interface{}) bool {
 | 
			
		||||
	// Skip empty messages
 | 
			
		||||
	if content["msgtype"] == nil {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Only allow image,video or file msgtypes
 | 
			
		||||
	if !(content["msgtype"].(string) == "m.image" ||
 | 
			
		||||
		content["msgtype"].(string) == "m.video" ||
 | 
			
		||||
		content["msgtype"].(string) == "m.file") {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,52 +1,29 @@
 | 
			
		||||
package bmattermost
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterclient"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/rs/xid"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMhook struct {
 | 
			
		||||
	mh *matterhook.Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMapi struct {
 | 
			
		||||
	mc            *matterclient.MMClient
 | 
			
		||||
	mmMap         map[string]string
 | 
			
		||||
	mmIgnoreNicks []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MMMessage struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	UserID   string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bmattermost struct {
 | 
			
		||||
	MMhook
 | 
			
		||||
	MMapi
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	name    string
 | 
			
		||||
	TeamId  string
 | 
			
		||||
	Account string
 | 
			
		||||
	mh     *matterhook.Client
 | 
			
		||||
	mc     *matterclient.MMClient
 | 
			
		||||
	uuid   string
 | 
			
		||||
	TeamID string
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
	avatarMap map[string]string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "mattermost"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bmattermost {
 | 
			
		||||
	b := &Bmattermost{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.mmMap = make(map[string]string)
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Bmattermost{Config: cfg, avatarMap: make(map[string]string)}
 | 
			
		||||
	b.uuid = xid.New().String()
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -55,33 +32,72 @@ func (b *Bmattermost) Command(cmd string) string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) Connect() error {
 | 
			
		||||
	if b.Config.WebhookURL != "" && b.Config.WebhookBindAddress != "" {
 | 
			
		||||
		flog.Info("Connecting using webhookurl and webhookbindaddress")
 | 
			
		||||
		b.mh = matterhook.New(b.Config.WebhookURL,
 | 
			
		||||
			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify,
 | 
			
		||||
				BindAddress: b.Config.WebhookBindAddress})
 | 
			
		||||
	} else if b.Config.WebhookURL != "" {
 | 
			
		||||
		flog.Info("Connecting using webhookurl (for posting) and token")
 | 
			
		||||
		b.mh = matterhook.New(b.Config.WebhookURL,
 | 
			
		||||
			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify,
 | 
			
		||||
	if b.GetString("WebhookBindAddress") != "" {
 | 
			
		||||
		if b.GetString("WebhookURL") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)")
 | 
			
		||||
			b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
					BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
		} else if b.GetString("Token") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using token (sending)")
 | 
			
		||||
			err := b.apiLogin()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
		} else if b.GetString("Login") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using login/password (sending)")
 | 
			
		||||
			err := b.apiLogin()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
		} else {
 | 
			
		||||
			b.Log.Info("Connecting using webhookbindaddress (receiving)")
 | 
			
		||||
			b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
					BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
		}
 | 
			
		||||
		go b.handleMatter()
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	if b.GetString("WebhookURL") != "" {
 | 
			
		||||
		b.Log.Info("Connecting using webhookurl (sending)")
 | 
			
		||||
		b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
			matterhook.Config{InsecureSkipVerify: b.GetBool("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 b.GetString("Token") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using token (receiving)")
 | 
			
		||||
			err := b.apiLogin()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
			go b.handleMatter()
 | 
			
		||||
		} else if b.GetString("Login") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using login/password (receiving)")
 | 
			
		||||
			err := b.apiLogin()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
			go b.handleMatter()
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	} else if b.GetString("Token") != "" {
 | 
			
		||||
		b.Log.Info("Connecting using token (sending and receiving)")
 | 
			
		||||
		err := b.apiLogin()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		flog.Info("Connection succeeded")
 | 
			
		||||
		b.TeamId = b.mc.GetTeamId()
 | 
			
		||||
		go b.mc.WsReceiver()
 | 
			
		||||
		go b.mc.StatusLoop()
 | 
			
		||||
		go b.handleMatter()
 | 
			
		||||
	} else if b.GetString("Login") != "" {
 | 
			
		||||
		b.Log.Info("Connecting using login/password (sending and receiving)")
 | 
			
		||||
		err := b.apiLogin()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		go b.handleMatter()
 | 
			
		||||
	}
 | 
			
		||||
	if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && b.GetString("Login") == "" && b.GetString("Token") == "" {
 | 
			
		||||
		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured")
 | 
			
		||||
	}
 | 
			
		||||
	go b.handleMatter()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -89,101 +105,345 @@ func (b *Bmattermost) Disconnect() error {
 | 
			
		||||
	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
 | 
			
		||||
	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" {
 | 
			
		||||
		return b.mc.JoinChannel(b.mc.GetChannelId(channel, ""))
 | 
			
		||||
	if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" {
 | 
			
		||||
		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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	nick := msg.Username
 | 
			
		||||
	message := msg.Text
 | 
			
		||||
	channel := msg.Channel
 | 
			
		||||
func (b *Bmattermost) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick {
 | 
			
		||||
		message = nick + message
 | 
			
		||||
	// Make a action /me of the message
 | 
			
		||||
	if msg.Event == config.EVENT_USER_ACTION {
 | 
			
		||||
		msg.Text = "*" + msg.Text + "*"
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.WebhookURL != "" {
 | 
			
		||||
		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
		matterMessage.IconURL = msg.Avatar
 | 
			
		||||
		matterMessage.Channel = channel
 | 
			
		||||
		matterMessage.UserName = nick
 | 
			
		||||
		matterMessage.Type = ""
 | 
			
		||||
		matterMessage.Text = message
 | 
			
		||||
		err := b.mh.Send(matterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.Info(err)
 | 
			
		||||
			return err
 | 
			
		||||
 | 
			
		||||
	// map the file SHA to our user (caches the avatar)
 | 
			
		||||
	if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
 | 
			
		||||
		return b.cacheAvatar(&msg)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Use webhook to send the message
 | 
			
		||||
	if b.GetString("WebhookURL") != "" {
 | 
			
		||||
		return b.sendWebhook(msg)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		if msg.ID == "" {
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
		return msg.ID, b.mc.DeleteMessage(msg.ID)
 | 
			
		||||
	}
 | 
			
		||||
	b.mc.PostMessage(b.mc.GetChannelId(channel, ""), message)
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, ""), rmsg.Username+rmsg.Text)
 | 
			
		||||
		}
 | 
			
		||||
		if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
			return b.handleUploadFile(&msg)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Prepend nick if configured
 | 
			
		||||
	if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
		msg.Text = msg.Username + msg.Text
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message if we have an ID
 | 
			
		||||
	if msg.ID != "" {
 | 
			
		||||
		return b.mc.EditMessage(msg.ID, msg.Text)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, ""), msg.Text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatter() {
 | 
			
		||||
	mchan := make(chan *MMMessage)
 | 
			
		||||
	if b.Config.WebhookBindAddress != "" && b.Config.WebhookURL != "" {
 | 
			
		||||
		flog.Debugf("Choosing webhooks based receiving")
 | 
			
		||||
		go b.handleMatterHook(mchan)
 | 
			
		||||
	messages := make(chan *config.Message)
 | 
			
		||||
	if b.GetString("WebhookBindAddress") != "" {
 | 
			
		||||
		b.Log.Debugf("Choosing webhooks based receiving")
 | 
			
		||||
		go b.handleMatterHook(messages)
 | 
			
		||||
	} else {
 | 
			
		||||
		flog.Debugf("Choosing login (api) based receiving")
 | 
			
		||||
		go b.handleMatterClient(mchan)
 | 
			
		||||
		if b.GetString("Token") != "" {
 | 
			
		||||
			b.Log.Debugf("Choosing token based receiving")
 | 
			
		||||
		} else {
 | 
			
		||||
			b.Log.Debugf("Choosing login/password based receiving")
 | 
			
		||||
		}
 | 
			
		||||
		go b.handleMatterClient(messages)
 | 
			
		||||
	}
 | 
			
		||||
	for message := range mchan {
 | 
			
		||||
		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account)
 | 
			
		||||
		b.Remote <- config.Message{Text: message.Text, Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID}
 | 
			
		||||
	var ok bool
 | 
			
		||||
	for message := range messages {
 | 
			
		||||
		message.Avatar = helper.GetAvatar(b.avatarMap, message.UserID, b.General)
 | 
			
		||||
		message.Account = b.Account
 | 
			
		||||
		message.Text, ok = b.replaceAction(message.Text)
 | 
			
		||||
		if ok {
 | 
			
		||||
			message.Event = config.EVENT_USER_ACTION
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
 | 
			
		||||
		b.Log.Debugf("<= Message is %#v", message)
 | 
			
		||||
		b.Remote <- *message
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) {
 | 
			
		||||
func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {
 | 
			
		||||
	for message := range b.mc.MessageChan {
 | 
			
		||||
		flog.Debugf("%#v", message.Raw.Data)
 | 
			
		||||
		if message.Type == "system_join_leave" ||
 | 
			
		||||
			message.Type == "system_join_channel" ||
 | 
			
		||||
			message.Type == "system_leave_channel" {
 | 
			
		||||
			flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account)
 | 
			
		||||
			b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE}
 | 
			
		||||
		b.Log.Debugf("%#v", message.Raw.Data)
 | 
			
		||||
 | 
			
		||||
		if b.skipMessage(message) {
 | 
			
		||||
			b.Log.Debugf("Skipped message: %#v", message)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		if (message.Raw.Event == "post_edited") && b.Config.EditDisable {
 | 
			
		||||
			continue
 | 
			
		||||
 | 
			
		||||
		// only download avatars if we have a place to upload them (configured mediaserver)
 | 
			
		||||
		if b.General.MediaServerUpload != "" {
 | 
			
		||||
			b.handleDownloadAvatar(message.UserID, message.Channel)
 | 
			
		||||
		}
 | 
			
		||||
		// do not post our own messages back to irc
 | 
			
		||||
		// only listen to message from our team
 | 
			
		||||
		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited") &&
 | 
			
		||||
			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId {
 | 
			
		||||
			flog.Debugf("Receiving from matterclient %#v", message)
 | 
			
		||||
			m := &MMMessage{}
 | 
			
		||||
			m.UserID = message.UserID
 | 
			
		||||
			m.Username = message.Username
 | 
			
		||||
			m.Channel = message.Channel
 | 
			
		||||
			m.Text = message.Text
 | 
			
		||||
			if message.Raw.Event == "post_edited" && !b.Config.EditDisable {
 | 
			
		||||
				m.Text = message.Text + b.Config.EditSuffix
 | 
			
		||||
 | 
			
		||||
		b.Log.Debugf("== Receiving event %#v", message)
 | 
			
		||||
 | 
			
		||||
		rmsg := &config.Message{Username: message.Username, UserID: message.UserID, Channel: message.Channel, Text: message.Text, ID: message.Post.Id, Extra: make(map[string][]interface{})}
 | 
			
		||||
 | 
			
		||||
		// handle mattermost post properties (override username and attachments)
 | 
			
		||||
		props := message.Post.Props
 | 
			
		||||
		if props != nil {
 | 
			
		||||
			if _, ok := props["override_username"].(string); ok {
 | 
			
		||||
				rmsg.Username = props["override_username"].(string)
 | 
			
		||||
			}
 | 
			
		||||
			if len(message.Post.FileIds) > 0 {
 | 
			
		||||
				for _, link := range b.mc.GetPublicLinks(message.Post.FileIds) {
 | 
			
		||||
					m.Text = m.Text + "\n" + link
 | 
			
		||||
			if _, ok := props["attachments"].([]interface{}); ok {
 | 
			
		||||
				rmsg.Extra["attachments"] = props["attachments"].([]interface{})
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// create a text for bridges that don't support native editing
 | 
			
		||||
		if message.Raw.Event == "post_edited" && !b.GetBool("EditDisable") {
 | 
			
		||||
			rmsg.Text = message.Text + b.GetString("EditSuffix")
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if message.Raw.Event == "post_deleted" {
 | 
			
		||||
			rmsg.Event = config.EVENT_MSG_DELETE
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if len(message.Post.FileIds) > 0 {
 | 
			
		||||
			for _, id := range message.Post.FileIds {
 | 
			
		||||
				err := b.handleDownloadFile(rmsg, id)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					b.Log.Errorf("download failed: %s", err)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			mchan <- m
 | 
			
		||||
		}
 | 
			
		||||
		messages <- rmsg
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) handleMatterHook(mchan chan *MMMessage) {
 | 
			
		||||
func (b *Bmattermost) handleMatterHook(messages chan *config.Message) {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.mh.Receive()
 | 
			
		||||
		flog.Debugf("Receiving from matterhook %#v", message)
 | 
			
		||||
		m := &MMMessage{}
 | 
			
		||||
		m.UserID = message.UserID
 | 
			
		||||
		m.Username = message.UserName
 | 
			
		||||
		m.Text = message.Text
 | 
			
		||||
		m.Channel = message.ChannelName
 | 
			
		||||
		mchan <- m
 | 
			
		||||
		b.Log.Debugf("Receiving from matterhook %#v", message)
 | 
			
		||||
		messages <- &config.Message{UserID: message.UserID, Username: message.UserName, Text: message.Text, Channel: message.ChannelName}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) apiLogin() error {
 | 
			
		||||
	password := b.GetString("Password")
 | 
			
		||||
	if b.GetString("Token") != "" {
 | 
			
		||||
		password = "MMAUTHTOKEN=" + b.GetString("Token")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server"))
 | 
			
		||||
	if b.GetBool("debug") {
 | 
			
		||||
		b.mc.SetLogLevel("debug")
 | 
			
		||||
	}
 | 
			
		||||
	b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify")
 | 
			
		||||
	b.mc.NoTLS = b.GetBool("NoTLS")
 | 
			
		||||
	b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))
 | 
			
		||||
	err := b.mc.Login()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	b.TeamID = b.mc.GetTeamId()
 | 
			
		||||
	go b.mc.WsReceiver()
 | 
			
		||||
	go b.mc.StatusLoop()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// replaceAction replace the message with the correct action (/me) code
 | 
			
		||||
func (b *Bmattermost) replaceAction(text string) (string, bool) {
 | 
			
		||||
	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") {
 | 
			
		||||
		return strings.Replace(text, "*", "", -1), true
 | 
			
		||||
	}
 | 
			
		||||
	return text, false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bmattermost) cacheAvatar(msg *config.Message) (string, error) {
 | 
			
		||||
	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 != "" {
 | 
			
		||||
		b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
 | 
			
		||||
		b.avatarMap[msg.UserID] = fi.SHA
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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) {
 | 
			
		||||
	rmsg := 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 {
 | 
			
		||||
			b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Error(err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General)
 | 
			
		||||
		b.Remote <- rmsg
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleDownloadFile handles file download
 | 
			
		||||
func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error {
 | 
			
		||||
	url, _ := b.mc.Client.GetFileLink(id)
 | 
			
		||||
	finfo, resp := b.mc.Client.GetFileInfo(id)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	data, resp := b.mc.Client.DownloadFile(id, true)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) {
 | 
			
		||||
	var err error
 | 
			
		||||
	var res, id string
 | 
			
		||||
	channelID := b.mc.GetChannelId(msg.Channel, "")
 | 
			
		||||
	for _, f := range msg.Extra["file"] {
 | 
			
		||||
		fi := f.(config.FileInfo)
 | 
			
		||||
		id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		msg.Text = fi.Comment
 | 
			
		||||
		if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
			msg.Text = msg.Username + msg.Text
 | 
			
		||||
		}
 | 
			
		||||
		res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, []string{id})
 | 
			
		||||
	}
 | 
			
		||||
	return res, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// sendWebhook uses the configured WebhookURL to send the message
 | 
			
		||||
func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) {
 | 
			
		||||
	// skip events
 | 
			
		||||
	if msg.Event != "" {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
		msg.Text = msg.Username + msg.Text
 | 
			
		||||
	}
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
 | 
			
		||||
			matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text, Props: make(map[string]interface{})}
 | 
			
		||||
			matterMessage.Props["matterbridge_"+b.uuid] = true
 | 
			
		||||
			b.mh.Send(matterMessage)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// webhook doesn't support file uploads, so we add the url manually
 | 
			
		||||
		if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
			for _, f := range msg.Extra["file"] {
 | 
			
		||||
				fi := f.(config.FileInfo)
 | 
			
		||||
				if fi.URL != "" {
 | 
			
		||||
					msg.Text += fi.URL
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
 | 
			
		||||
	matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: msg.Channel, UserName: msg.Username, Text: msg.Text, Props: make(map[string]interface{})}
 | 
			
		||||
	if msg.Avatar != "" {
 | 
			
		||||
		matterMessage.IconURL = msg.Avatar
 | 
			
		||||
	}
 | 
			
		||||
	matterMessage.Props["matterbridge_"+b.uuid] = true
 | 
			
		||||
	err := b.mh.Send(matterMessage)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		b.Log.Info(err)
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// skipMessages returns true if this message should not be handled
 | 
			
		||||
func (b *Bmattermost) skipMessage(message *matterclient.Message) bool {
 | 
			
		||||
	// Handle join/leave
 | 
			
		||||
	if message.Type == "system_join_leave" ||
 | 
			
		||||
		message.Type == "system_join_channel" ||
 | 
			
		||||
		message.Type == "system_leave_channel" {
 | 
			
		||||
		if b.GetBool("nosendjoinpart") {
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
		b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account)
 | 
			
		||||
		b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE}
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Handle edited messages
 | 
			
		||||
	if (message.Raw.Event == "post_edited") && b.GetBool("EditDisable") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Ignore messages sent from matterbridge
 | 
			
		||||
	if message.Post.Props != nil {
 | 
			
		||||
		if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok {
 | 
			
		||||
			b.Log.Debugf("sent by matterbridge, ignoring")
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Ignore messages sent from a user logged in as the bot
 | 
			
		||||
	if b.mc.User.Username == message.Username {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if the message has reactions don't repost it (for now, until we can correlate reaction with message)
 | 
			
		||||
	if message.Post.HasReactions {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// ignore messages from other teams than ours
 | 
			
		||||
	if message.Raw.Data["team_id"].(string) != b.TeamID {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// only handle posted, edited or deleted events
 | 
			
		||||
	if !(message.Raw.Event == "posted" || message.Raw.Event == "post_edited" || message.Raw.Event == "post_deleted") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,10 +1,11 @@
 | 
			
		||||
package brocketchat
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/42wim/matterbridge/hook/rockethook"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMhook struct {
 | 
			
		||||
@@ -14,25 +15,11 @@ type MMhook struct {
 | 
			
		||||
 | 
			
		||||
type Brocketchat struct {
 | 
			
		||||
	MMhook
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	name    string
 | 
			
		||||
	Account string
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "rocketchat"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Brocketchat {
 | 
			
		||||
	b := &Brocketchat{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Brocketchat{Config: cfg}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Command(cmd string) string {
 | 
			
		||||
@@ -40,11 +27,11 @@ func (b *Brocketchat) Command(cmd string) string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Connect() error {
 | 
			
		||||
	flog.Info("Connecting webhooks")
 | 
			
		||||
	b.mh = matterhook.New(b.Config.WebhookURL,
 | 
			
		||||
		matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify,
 | 
			
		||||
	b.Log.Info("Connecting webhooks")
 | 
			
		||||
	b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
		matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
			DisableServer: true})
 | 
			
		||||
	b.rh = rockethook.New(b.Config.WebhookURL, rockethook.Config{BindAddress: b.Config.WebhookBindAddress})
 | 
			
		||||
	b.rh = rockethook.New(b.GetString("WebhookURL"), rockethook.Config{BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
	go b.handleRocketHook()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -54,34 +41,55 @@ func (b *Brocketchat) Disconnect() error {
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) JoinChannel(channel string) error {
 | 
			
		||||
func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
func (b *Brocketchat) Send(msg config.Message) (string, error) {
 | 
			
		||||
	// ignore delete messages
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
 | 
			
		||||
			matterMessage := matterhook.OMessage{IconURL: 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
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
 | 
			
		||||
	matterMessage := matterhook.OMessage{IconURL: iconURL}
 | 
			
		||||
	matterMessage.Channel = msg.Channel
 | 
			
		||||
	matterMessage.UserName = msg.Username
 | 
			
		||||
	matterMessage.Type = ""
 | 
			
		||||
	matterMessage.Text = msg.Text
 | 
			
		||||
	err := b.mh.Send(matterMessage)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Info(err)
 | 
			
		||||
		return err
 | 
			
		||||
		b.Log.Info(err)
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Brocketchat) handleRocketHook() {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.rh.Receive()
 | 
			
		||||
		flog.Debugf("Receiving from rockethook %#v", message)
 | 
			
		||||
		b.Log.Debugf("Receiving from rockethook %#v", message)
 | 
			
		||||
		// do not loop
 | 
			
		||||
		if message.UserName == b.Config.Nick {
 | 
			
		||||
		if message.UserName == b.GetString("Nick") {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		flog.Debugf("Sending message from %s on %s to gateway", message.UserName, b.Account)
 | 
			
		||||
		b.Log.Debugf("<= Sending message from %s on %s to gateway", message.UserName, b.Account)
 | 
			
		||||
		b.Remote <- config.Message{Text: message.Text, Username: message.UserName, Channel: message.ChannelName, Account: b.Account, UserID: message.UserID}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,50 +1,40 @@
 | 
			
		||||
package bslack
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/nlopes/slack"
 | 
			
		||||
	"html"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/42wim/matterbridge/matterhook"
 | 
			
		||||
	"github.com/nlopes/slack"
 | 
			
		||||
	"github.com/rs/xid"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MMMessage struct {
 | 
			
		||||
	Text     string
 | 
			
		||||
	Channel  string
 | 
			
		||||
	Username string
 | 
			
		||||
	UserID   string
 | 
			
		||||
	Raw      *slack.MessageEvent
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Bslack struct {
 | 
			
		||||
	mh       *matterhook.Client
 | 
			
		||||
	sc       *slack.Client
 | 
			
		||||
	Config   *config.Protocol
 | 
			
		||||
	rtm      *slack.RTM
 | 
			
		||||
	Plus     bool
 | 
			
		||||
	Remote   chan config.Message
 | 
			
		||||
	Users    []slack.User
 | 
			
		||||
	Account  string
 | 
			
		||||
	si       *slack.Info
 | 
			
		||||
	channels []slack.Channel
 | 
			
		||||
	mh         *matterhook.Client
 | 
			
		||||
	sc         *slack.Client
 | 
			
		||||
	rtm        *slack.RTM
 | 
			
		||||
	Users      []slack.User
 | 
			
		||||
	Usergroups []slack.UserGroup
 | 
			
		||||
	si         *slack.Info
 | 
			
		||||
	channels   []slack.Channel
 | 
			
		||||
	uuid       string
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "slack"
 | 
			
		||||
const messageDeleted = "message_deleted"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bslack {
 | 
			
		||||
	b := &Bslack{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Bslack{Config: cfg, uuid: xid.New().String()}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Command(cmd string) string {
 | 
			
		||||
@@ -52,89 +42,174 @@ func (b *Bslack) Command(cmd string) string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Connect() error {
 | 
			
		||||
	if b.Config.WebhookURL != "" && b.Config.WebhookBindAddress != "" {
 | 
			
		||||
		flog.Info("Connecting using webhookurl and webhookbindaddress")
 | 
			
		||||
		b.mh = matterhook.New(b.Config.WebhookURL,
 | 
			
		||||
			matterhook.Config{BindAddress: b.Config.WebhookBindAddress})
 | 
			
		||||
	} else if b.Config.WebhookURL != "" {
 | 
			
		||||
		flog.Info("Connecting using webhookurl (for posting) and token")
 | 
			
		||||
		b.mh = matterhook.New(b.Config.WebhookURL,
 | 
			
		||||
			matterhook.Config{DisableServer: true})
 | 
			
		||||
	} else {
 | 
			
		||||
		flog.Info("Connecting using token")
 | 
			
		||||
		b.sc = slack.New(b.Config.Token)
 | 
			
		||||
	b.RLock()
 | 
			
		||||
	defer b.RUnlock()
 | 
			
		||||
	if b.GetString("WebhookBindAddress") != "" {
 | 
			
		||||
		if b.GetString("WebhookURL") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)")
 | 
			
		||||
			b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
					BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
		} else if b.GetString("Token") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using token (sending)")
 | 
			
		||||
			b.sc = slack.New(b.GetString("Token"))
 | 
			
		||||
			b.rtm = b.sc.NewRTM()
 | 
			
		||||
			go b.rtm.ManageConnection()
 | 
			
		||||
			b.Log.Info("Connecting using webhookbindaddress (receiving)")
 | 
			
		||||
			b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
					BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
		} else {
 | 
			
		||||
			b.Log.Info("Connecting using webhookbindaddress (receiving)")
 | 
			
		||||
			b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
					BindAddress: b.GetString("WebhookBindAddress")})
 | 
			
		||||
		}
 | 
			
		||||
		go b.handleSlack()
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	if b.GetString("WebhookURL") != "" {
 | 
			
		||||
		b.Log.Info("Connecting using webhookurl (sending)")
 | 
			
		||||
		b.mh = matterhook.New(b.GetString("WebhookURL"),
 | 
			
		||||
			matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
 | 
			
		||||
				DisableServer: true})
 | 
			
		||||
		if b.GetString("Token") != "" {
 | 
			
		||||
			b.Log.Info("Connecting using token (receiving)")
 | 
			
		||||
			b.sc = slack.New(b.GetString("Token"))
 | 
			
		||||
			b.rtm = b.sc.NewRTM()
 | 
			
		||||
			go b.rtm.ManageConnection()
 | 
			
		||||
			go b.handleSlack()
 | 
			
		||||
		}
 | 
			
		||||
	} else if b.GetString("Token") != "" {
 | 
			
		||||
		b.Log.Info("Connecting using token (sending and receiving)")
 | 
			
		||||
		b.sc = slack.New(b.GetString("Token"))
 | 
			
		||||
		b.rtm = b.sc.NewRTM()
 | 
			
		||||
		go b.rtm.ManageConnection()
 | 
			
		||||
		go b.handleSlack()
 | 
			
		||||
	}
 | 
			
		||||
	if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && b.GetString("Token") == "" {
 | 
			
		||||
		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token configured")
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	go b.handleSlack()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
	return b.rtm.Disconnect()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) JoinChannel(channel string) error {
 | 
			
		||||
func (b *Bslack) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	// we can only join channels using the API
 | 
			
		||||
	if b.Config.WebhookURL == "" || b.Config.WebhookBindAddress == "" {
 | 
			
		||||
		if strings.HasPrefix(b.Config.Token, "xoxb") {
 | 
			
		||||
	if b.sc != nil {
 | 
			
		||||
		if strings.HasPrefix(b.GetString("Token"), "xoxb") {
 | 
			
		||||
			// TODO check if bot has already joined channel
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		_, err := b.sc.JoinChannel(channel)
 | 
			
		||||
		_, err := b.sc.JoinChannel(channel.Name)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			if err.Error() != "name_taken" {
 | 
			
		||||
				return err
 | 
			
		||||
			switch err.Error() {
 | 
			
		||||
			case "name_taken", "restricted_action":
 | 
			
		||||
			case "default":
 | 
			
		||||
				{
 | 
			
		||||
					return err
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	nick := msg.Username
 | 
			
		||||
	message := msg.Text
 | 
			
		||||
	channel := msg.Channel
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick {
 | 
			
		||||
		message = nick + " " + message
 | 
			
		||||
func (b *Bslack) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	// Make a action /me of the message
 | 
			
		||||
	if msg.Event == config.EVENT_USER_ACTION {
 | 
			
		||||
		msg.Text = "_" + msg.Text + "_"
 | 
			
		||||
	}
 | 
			
		||||
	if b.Config.WebhookURL != "" {
 | 
			
		||||
		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
 | 
			
		||||
		matterMessage.Channel = channel
 | 
			
		||||
		matterMessage.UserName = nick
 | 
			
		||||
		matterMessage.Type = ""
 | 
			
		||||
		matterMessage.Text = message
 | 
			
		||||
		err := b.mh.Send(matterMessage)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			flog.Info(err)
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
 | 
			
		||||
	// Use webhook to send the message
 | 
			
		||||
	if b.GetString("WebhookURL") != "" {
 | 
			
		||||
		return b.sendWebhook(msg)
 | 
			
		||||
	}
 | 
			
		||||
	schannel, err := b.getChannelByName(channel)
 | 
			
		||||
 | 
			
		||||
	// get the slack channel
 | 
			
		||||
	schannel, err := b.getChannelByName(msg.Channel)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	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)
 | 
			
		||||
		_, _, err := b.sc.DeleteMessage(schannel.ID, ts[1])
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return msg.ID, err
 | 
			
		||||
		}
 | 
			
		||||
		return msg.ID, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Prepend nick if configured
 | 
			
		||||
	if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
		msg.Text = msg.Username + msg.Text
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message if we have an ID
 | 
			
		||||
	if msg.ID != "" {
 | 
			
		||||
		ts := strings.Fields(msg.ID)
 | 
			
		||||
		_, _, _, err := b.sc.UpdateMessage(schannel.ID, ts[1], msg.Text)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return msg.ID, err
 | 
			
		||||
		}
 | 
			
		||||
		return msg.ID, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// create slack new post parameters
 | 
			
		||||
	np := slack.NewPostMessageParameters()
 | 
			
		||||
	if b.Config.PrefixMessagesWithNick == true {
 | 
			
		||||
	if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
		np.AsUser = true
 | 
			
		||||
	}
 | 
			
		||||
	np.Username = nick
 | 
			
		||||
	np.IconURL = config.GetIconURL(&msg, b.Config)
 | 
			
		||||
	np.Username = msg.Username
 | 
			
		||||
	np.LinkNames = 1 // replace mentions
 | 
			
		||||
	np.IconURL = config.GetIconURL(&msg, b.GetString("iconurl"))
 | 
			
		||||
	if msg.Avatar != "" {
 | 
			
		||||
		np.IconURL = msg.Avatar
 | 
			
		||||
	}
 | 
			
		||||
	b.sc.PostMessage(schannel.ID, message, np)
 | 
			
		||||
	// add a callback ID so we can see we created it
 | 
			
		||||
	np.Attachments = append(np.Attachments, slack.Attachment{CallbackID: "matterbridge_" + b.uuid})
 | 
			
		||||
	// add file attachments
 | 
			
		||||
	np.Attachments = append(np.Attachments, b.createAttach(msg.Extra)...)
 | 
			
		||||
	// add slack attachments (from another slack bridge)
 | 
			
		||||
	if len(msg.Extra["slack_attachment"]) > 0 {
 | 
			
		||||
		for _, attach := range msg.Extra["slack_attachment"] {
 | 
			
		||||
			np.Attachments = append(np.Attachments, attach.([]slack.Attachment)...)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	/*
 | 
			
		||||
	   newmsg := b.rtm.NewOutgoingMessage(message, schannel.ID)
 | 
			
		||||
	   b.rtm.SendMessage(newmsg)
 | 
			
		||||
	*/
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	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 {
 | 
			
		||||
			b.handleUploadFile(&msg, schannel.ID)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	_, id, err := b.sc.PostMessage(schannel.ID, msg.Text, np)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return "slack " + id, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) Reload(cfg *bridge.Config) (string, error) {
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) getAvatar(user string) string {
 | 
			
		||||
@@ -174,70 +249,61 @@ func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleSlack() {
 | 
			
		||||
	mchan := make(chan *MMMessage)
 | 
			
		||||
	if b.Config.WebhookBindAddress != "" && b.Config.WebhookURL != "" {
 | 
			
		||||
		flog.Debugf("Choosing webhooks based receiving")
 | 
			
		||||
		go b.handleMatterHook(mchan)
 | 
			
		||||
	messages := make(chan *config.Message)
 | 
			
		||||
	if b.GetString("WebhookBindAddress") != "" {
 | 
			
		||||
		b.Log.Debugf("Choosing webhooks based receiving")
 | 
			
		||||
		go b.handleMatterHook(messages)
 | 
			
		||||
	} else {
 | 
			
		||||
		flog.Debugf("Choosing token based receiving")
 | 
			
		||||
		go b.handleSlackClient(mchan)
 | 
			
		||||
		b.Log.Debugf("Choosing token based receiving")
 | 
			
		||||
		go b.handleSlackClient(messages)
 | 
			
		||||
	}
 | 
			
		||||
	time.Sleep(time.Second)
 | 
			
		||||
	flog.Debug("Start listening for Slack messages")
 | 
			
		||||
	for message := range mchan {
 | 
			
		||||
		// do not send messages from ourself
 | 
			
		||||
		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		texts := strings.Split(message.Text, "\n")
 | 
			
		||||
		for _, text := range texts {
 | 
			
		||||
			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}
 | 
			
		||||
		}
 | 
			
		||||
	b.Log.Debug("Start listening for Slack messages")
 | 
			
		||||
	for message := range messages {
 | 
			
		||||
		b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
 | 
			
		||||
 | 
			
		||||
		// cleanup the message
 | 
			
		||||
		message.Text = b.replaceMention(message.Text)
 | 
			
		||||
		message.Text = b.replaceVariable(message.Text)
 | 
			
		||||
		message.Text = b.replaceChannel(message.Text)
 | 
			
		||||
		message.Text = b.replaceURL(message.Text)
 | 
			
		||||
		message.Text = html.UnescapeString(message.Text)
 | 
			
		||||
 | 
			
		||||
		// Add the avatar
 | 
			
		||||
		message.Avatar = b.getAvatar(strings.ToLower(message.Username))
 | 
			
		||||
 | 
			
		||||
		b.Log.Debugf("<= Message is %#v", message)
 | 
			
		||||
		b.Remote <- *message
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
 | 
			
		||||
	count := 0
 | 
			
		||||
func (b *Bslack) handleSlackClient(messages chan *config.Message) {
 | 
			
		||||
	for msg := range b.rtm.IncomingEvents {
 | 
			
		||||
		if msg.Type != "user_typing" && msg.Type != "latency_report" {
 | 
			
		||||
			b.Log.Debugf("== Receiving event %#v", msg.Data)
 | 
			
		||||
		}
 | 
			
		||||
		switch ev := msg.Data.(type) {
 | 
			
		||||
		case *slack.MessageEvent:
 | 
			
		||||
			// ignore first message
 | 
			
		||||
			if count > 0 {
 | 
			
		||||
				flog.Debugf("Receiving from slackclient %#v", ev)
 | 
			
		||||
				if !b.Config.EditDisable && ev.SubMessage != nil {
 | 
			
		||||
					flog.Debugf("SubMessage %#v", ev.SubMessage)
 | 
			
		||||
					ev.User = ev.SubMessage.User
 | 
			
		||||
					ev.Text = ev.SubMessage.Text + b.Config.EditSuffix
 | 
			
		||||
				}
 | 
			
		||||
				// use our own func because rtm.GetChannelInfo doesn't work for private channels
 | 
			
		||||
				channel, err := b.getChannelByID(ev.Channel)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				user, err := b.rtm.GetUserInfo(ev.User)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				m := &MMMessage{}
 | 
			
		||||
				m.UserID = user.ID
 | 
			
		||||
				m.Username = user.Name
 | 
			
		||||
				m.Channel = channel.Name
 | 
			
		||||
				m.Text = ev.Text
 | 
			
		||||
				m.Raw = ev
 | 
			
		||||
				m.Text = b.replaceMention(m.Text)
 | 
			
		||||
				mchan <- m
 | 
			
		||||
			if b.skipMessageEvent(ev) {
 | 
			
		||||
				b.Log.Debugf("Skipped message: %#v", ev)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			count++
 | 
			
		||||
			rmsg, err := b.handleMessageEvent(ev)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Errorf("%#v", err)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			messages <- rmsg
 | 
			
		||||
		case *slack.OutgoingErrorEvent:
 | 
			
		||||
			flog.Debugf("%#v", ev.Error())
 | 
			
		||||
			b.Log.Debugf("%#v", ev.Error())
 | 
			
		||||
		case *slack.ChannelJoinedEvent:
 | 
			
		||||
			b.Users, _ = b.sc.GetUsers()
 | 
			
		||||
			b.Usergroups, _ = b.sc.GetUserGroups()
 | 
			
		||||
		case *slack.ConnectedEvent:
 | 
			
		||||
			b.channels = ev.Info.Channels
 | 
			
		||||
			b.si = ev.Info
 | 
			
		||||
			b.Users, _ = b.sc.GetUsers()
 | 
			
		||||
			b.Usergroups, _ = b.sc.GetUserGroups()
 | 
			
		||||
			// add private channels
 | 
			
		||||
			groups, _ := b.sc.GetGroups(true)
 | 
			
		||||
			for _, g := range groups {
 | 
			
		||||
@@ -247,50 +313,369 @@ func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
 | 
			
		||||
				b.channels = append(b.channels, *channel)
 | 
			
		||||
			}
 | 
			
		||||
		case *slack.InvalidAuthEvent:
 | 
			
		||||
			flog.Fatalf("Invalid Token %#v", ev)
 | 
			
		||||
			b.Log.Fatalf("Invalid Token %#v", ev)
 | 
			
		||||
		case *slack.ConnectionErrorEvent:
 | 
			
		||||
			b.Log.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj)
 | 
			
		||||
		default:
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) handleMatterHook(mchan chan *MMMessage) {
 | 
			
		||||
func (b *Bslack) handleMatterHook(messages chan *config.Message) {
 | 
			
		||||
	for {
 | 
			
		||||
		message := b.mh.Receive()
 | 
			
		||||
		flog.Debugf("receiving from matterhook (slack) %#v", message)
 | 
			
		||||
		m := &MMMessage{}
 | 
			
		||||
		m.Username = message.UserName
 | 
			
		||||
		m.Text = message.Text
 | 
			
		||||
		m.Text = b.replaceMention(m.Text)
 | 
			
		||||
		m.Channel = message.ChannelName
 | 
			
		||||
		if m.Username == "slackbot" {
 | 
			
		||||
		b.Log.Debugf("receiving from matterhook (slack) %#v", message)
 | 
			
		||||
		if message.UserName == "slackbot" {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		mchan <- m
 | 
			
		||||
		messages <- &config.Message{Username: message.UserName, Text: message.Text, Channel: message.ChannelName}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) userName(id string) string {
 | 
			
		||||
	for _, u := range b.Users {
 | 
			
		||||
		if u.ID == id {
 | 
			
		||||
			if u.Profile.DisplayName != "" {
 | 
			
		||||
				return u.Profile.DisplayName
 | 
			
		||||
			}
 | 
			
		||||
			return u.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/*
 | 
			
		||||
func (b *Bslack) userGroupName(id string) string {
 | 
			
		||||
	for _, u := range b.Usergroups {
 | 
			
		||||
		if u.ID == id {
 | 
			
		||||
			return u.Name
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
// @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users
 | 
			
		||||
func (b *Bslack) replaceMention(text string) string {
 | 
			
		||||
	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1)
 | 
			
		||||
	for _, r := range results {
 | 
			
		||||
		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1)
 | 
			
		||||
 | 
			
		||||
	}
 | 
			
		||||
	return text
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bslack) replaceURL(text string) string {
 | 
			
		||||
	results := regexp.MustCompile(`<(.*?)\|.*?>`).FindAllStringSubmatch(text, -1)
 | 
			
		||||
// @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)
 | 
			
		||||
		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(`<!((?:subteam\^)?[a-zA-Z0-9]+)(?:\|@?(.+?))?>`).FindAllStringSubmatch(text, -1)
 | 
			
		||||
	for _, r := range results {
 | 
			
		||||
		if r[2] != "" {
 | 
			
		||||
			text = strings.Replace(text, r[0], "@"+r[2], -1)
 | 
			
		||||
		} else {
 | 
			
		||||
			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 {
 | 
			
		||||
	results := regexp.MustCompile(`<(.*?)(\|.*?)?>`).FindAllStringSubmatch(text, -1)
 | 
			
		||||
	for _, r := range results {
 | 
			
		||||
		if len(strings.TrimSpace(r[2])) == 1 { // A display text separator was found, but the text was blank
 | 
			
		||||
			text = strings.Replace(text, r[0], "", -1)
 | 
			
		||||
		} else {
 | 
			
		||||
			text = strings.Replace(text, r[0], r[1], -1)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleDownloadFile handles file download
 | 
			
		||||
func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) error {
 | 
			
		||||
	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
 | 
			
		||||
	// limit to 1MB for now
 | 
			
		||||
	comment := ""
 | 
			
		||||
	results := regexp.MustCompile(`.*?commented: (.*)`).FindAllStringSubmatch(rmsg.Text, -1)
 | 
			
		||||
	if len(results) > 0 {
 | 
			
		||||
		comment = results[0][1]
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err := helper.HandleDownloadSize(b.Log, rmsg, file.Name, int64(file.Size), b.General)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	// actually download the file
 | 
			
		||||
	data, err := helper.DownloadFileAuth(file.URLPrivateDownload, "Bearer "+b.GetString("Token"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("download %s failed %#v", file.URLPrivateDownload, err)
 | 
			
		||||
	}
 | 
			
		||||
	// add the downloaded data to the message
 | 
			
		||||
	helper.HandleDownloadData(b.Log, rmsg, file.Name, comment, file.URLPrivateDownload, data, b.General)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Bslack) handleUploadFile(msg *config.Message, channelID string) (string, error) {
 | 
			
		||||
	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{channelID},
 | 
			
		||||
			InitialComment: fi.Comment,
 | 
			
		||||
		})
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Errorf("uploadfile %#v", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleMessageEvent handles the message events
 | 
			
		||||
func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, error) {
 | 
			
		||||
	// update the userlist on a channel_join
 | 
			
		||||
	if ev.SubType == "channel_join" {
 | 
			
		||||
		b.Users, _ = b.sc.GetUsers()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Edit message
 | 
			
		||||
	if !b.GetBool("EditDisable") && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp {
 | 
			
		||||
		b.Log.Debugf("SubMessage %#v", ev.SubMessage)
 | 
			
		||||
		ev.User = ev.SubMessage.User
 | 
			
		||||
		ev.Text = ev.SubMessage.Text + b.GetString("EditSuffix")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// use our own func because rtm.GetChannelInfo doesn't work for private channels
 | 
			
		||||
	channel, err := b.getChannelByID(ev.Channel)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rmsg := config.Message{Text: ev.Text, Channel: channel.Name, Account: b.Account, ID: "slack " + ev.Timestamp, Extra: make(map[string][]interface{})}
 | 
			
		||||
 | 
			
		||||
	// find the user id and name
 | 
			
		||||
	if ev.User != "" && ev.SubType != messageDeleted && ev.SubType != "file_comment" {
 | 
			
		||||
		user, err := b.rtm.GetUserInfo(ev.User)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		rmsg.UserID = user.ID
 | 
			
		||||
		rmsg.Username = user.Name
 | 
			
		||||
		if user.Profile.DisplayName != "" {
 | 
			
		||||
			rmsg.Username = user.Profile.DisplayName
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// See if we have some text in the attachments
 | 
			
		||||
	if rmsg.Text == "" {
 | 
			
		||||
		for _, attach := range ev.Attachments {
 | 
			
		||||
			if attach.Text != "" {
 | 
			
		||||
				rmsg.Text = attach.Text
 | 
			
		||||
			} else {
 | 
			
		||||
				rmsg.Text = attach.Fallback
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// when using webhookURL we can't check if it's our webhook or not for now
 | 
			
		||||
	if rmsg.Username == "" && ev.BotID != "" && b.GetString("WebhookURL") == "" {
 | 
			
		||||
		bot, err := b.rtm.GetBotInfo(ev.BotID)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		if bot.Name != "" {
 | 
			
		||||
			rmsg.Username = bot.Name
 | 
			
		||||
			if ev.Username != "" {
 | 
			
		||||
				rmsg.Username = ev.Username
 | 
			
		||||
			}
 | 
			
		||||
			rmsg.UserID = bot.ID
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// fixes issues with matterircd users
 | 
			
		||||
		if bot.Name == "Slack API Tester" {
 | 
			
		||||
			user, err := b.rtm.GetUserInfo(ev.User)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return nil, err
 | 
			
		||||
			}
 | 
			
		||||
			rmsg.UserID = user.ID
 | 
			
		||||
			rmsg.Username = user.Name
 | 
			
		||||
			if user.Profile.DisplayName != "" {
 | 
			
		||||
				rmsg.Username = user.Profile.DisplayName
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// file comments are set by the system (because there is no username given)
 | 
			
		||||
	if ev.SubType == "file_comment" {
 | 
			
		||||
		rmsg.Username = "system"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// do we have a /me action
 | 
			
		||||
	if ev.SubType == "me_message" {
 | 
			
		||||
		rmsg.Event = config.EVENT_USER_ACTION
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Handle join/leave
 | 
			
		||||
	if ev.SubType == "channel_leave" || ev.SubType == "channel_join" {
 | 
			
		||||
		rmsg.Username = "system"
 | 
			
		||||
		rmsg.Event = config.EVENT_JOIN_LEAVE
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// edited messages have a submessage, use this timestamp
 | 
			
		||||
	if ev.SubMessage != nil {
 | 
			
		||||
		rmsg.ID = "slack " + ev.SubMessage.Timestamp
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// deleted message event
 | 
			
		||||
	if ev.SubType == messageDeleted {
 | 
			
		||||
		rmsg.Text = config.EVENT_MSG_DELETE
 | 
			
		||||
		rmsg.Event = config.EVENT_MSG_DELETE
 | 
			
		||||
		rmsg.ID = "slack " + ev.DeletedTimestamp
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// topic change event
 | 
			
		||||
	if ev.SubType == "channel_topic" || ev.SubType == "channel_purpose" {
 | 
			
		||||
		rmsg.Event = config.EVENT_TOPIC_CHANGE
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Only deleted messages can have a empty username and text
 | 
			
		||||
	if (rmsg.Text == "" || rmsg.Username == "") && ev.SubType != messageDeleted {
 | 
			
		||||
		// this is probably a webhook we couldn't resolve
 | 
			
		||||
		if ev.BotID != "" {
 | 
			
		||||
			return nil, fmt.Errorf("probably an incoming webhook we couldn't resolve (maybe ourselves)")
 | 
			
		||||
		}
 | 
			
		||||
		return nil, fmt.Errorf("empty message and not a deleted message")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// save the attachments, so that we can send them to other slack (compatible) bridges
 | 
			
		||||
	if len(ev.Attachments) > 0 {
 | 
			
		||||
		rmsg.Extra["slack_attachment"] = append(rmsg.Extra["slack_attachment"], ev.Attachments)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
 | 
			
		||||
	if ev.File != nil {
 | 
			
		||||
		err := b.handleDownloadFile(&rmsg, ev.File)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Errorf("download failed: %s", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return &rmsg, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// sendWebhook uses the configured WebhookURL to send the message
 | 
			
		||||
func (b *Bslack) sendWebhook(msg config.Message) (string, error) {
 | 
			
		||||
	// skip events
 | 
			
		||||
	if msg.Event != "" {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if b.GetBool("PrefixMessagesWithNick") {
 | 
			
		||||
		msg.Text = msg.Username + msg.Text
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
 | 
			
		||||
			matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: msg.Channel, UserName: rmsg.Username, Text: rmsg.Text}
 | 
			
		||||
			b.mh.Send(matterMessage)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// webhook doesn't support file uploads, so we add the url manually
 | 
			
		||||
		if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
			for _, f := range msg.Extra["file"] {
 | 
			
		||||
				fi := f.(config.FileInfo)
 | 
			
		||||
				if fi.URL != "" {
 | 
			
		||||
					msg.Text += " " + fi.URL
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if we have native slack_attachments add them
 | 
			
		||||
	var attachs []slack.Attachment
 | 
			
		||||
	if len(msg.Extra["slack_attachment"]) > 0 {
 | 
			
		||||
		for _, attach := range msg.Extra["slack_attachment"] {
 | 
			
		||||
			attachs = append(attachs, attach.([]slack.Attachment)...)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
 | 
			
		||||
	matterMessage := matterhook.OMessage{IconURL: iconURL, Attachments: attachs, Channel: msg.Channel, UserName: msg.Username, Text: msg.Text}
 | 
			
		||||
	if msg.Avatar != "" {
 | 
			
		||||
		matterMessage.IconURL = msg.Avatar
 | 
			
		||||
	}
 | 
			
		||||
	err := b.mh.Send(matterMessage)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		b.Log.Error(err)
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// skipMessageEvent skips event that need to be skipped :-)
 | 
			
		||||
func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool {
 | 
			
		||||
	if ev.SubType == "channel_leave" || ev.SubType == "channel_join" {
 | 
			
		||||
		return b.GetBool("nosendjoinpart")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// ignore pinned items
 | 
			
		||||
	if ev.SubType == "pinned_item" || ev.SubType == "unpinned_item" {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// do not send messages from ourself
 | 
			
		||||
	if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" && ev.Username == b.si.User.Name {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// skip messages we made ourselves
 | 
			
		||||
	if len(ev.Attachments) > 0 {
 | 
			
		||||
		if ev.Attachments[0].CallbackID == "matterbridge_"+b.uuid {
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if !b.GetBool("EditDisable") && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp {
 | 
			
		||||
		// it seems ev.SubMessage.Edited == nil when slack unfurls
 | 
			
		||||
		// do not forward these messages #266
 | 
			
		||||
		if ev.SubMessage.Edited == nil {
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										136
									
								
								bridge/sshchat/sshchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										136
									
								
								bridge/sshchat/sshchat.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,136 @@
 | 
			
		||||
package bsshchat
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/shazow/ssh-chat/sshd"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
	"io"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bsshchat struct {
 | 
			
		||||
	r *bufio.Scanner
 | 
			
		||||
	w io.WriteCloser
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Bsshchat{Config: cfg}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bsshchat) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	b.Log.Infof("Connecting %s", b.GetString("Server"))
 | 
			
		||||
	go func() {
 | 
			
		||||
		err = sshd.ConnectShell(b.GetString("Server"), b.GetString("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 {
 | 
			
		||||
		b.Log.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.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
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.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
 | 
			
		||||
					if fi.Comment != "" {
 | 
			
		||||
						msg.Text = fi.Comment + ": " + 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:
 | 
			
		||||
				b.Log.Debugf("PING")
 | 
			
		||||
				err := b.xc.PingC2S("", "")
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					b.Log.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 {
 | 
			
		||||
				b.Log.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
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@@ -2,11 +2,13 @@ package bsteam
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/Philipp15b/go-steam"
 | 
			
		||||
	"github.com/Philipp15b/go-steam/protocol/steamlang"
 | 
			
		||||
	"github.com/Philipp15b/go-steam/steamid"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	//"io/ioutil"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"sync"
 | 
			
		||||
@@ -16,38 +18,26 @@ import (
 | 
			
		||||
type Bsteam struct {
 | 
			
		||||
	c         *steam.Client
 | 
			
		||||
	connected chan struct{}
 | 
			
		||||
	Config    *config.Protocol
 | 
			
		||||
	Remote    chan config.Message
 | 
			
		||||
	Account   string
 | 
			
		||||
	userMap   map[steamid.SteamId]string
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "steam"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bsteam {
 | 
			
		||||
	b := &Bsteam{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Bsteam{Config: cfg}
 | 
			
		||||
	b.userMap = make(map[steamid.SteamId]string)
 | 
			
		||||
	b.connected = make(chan struct{})
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bsteam) Connect() error {
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	b.Log.Info("Connecting")
 | 
			
		||||
	b.c = steam.NewClient()
 | 
			
		||||
	go b.handleEvents()
 | 
			
		||||
	go b.c.Connect()
 | 
			
		||||
	select {
 | 
			
		||||
	case <-b.connected:
 | 
			
		||||
		flog.Info("Connection succeeded")
 | 
			
		||||
		b.Log.Info("Connection succeeded")
 | 
			
		||||
	case <-time.After(time.Second * 30):
 | 
			
		||||
		return fmt.Errorf("connection timed out")
 | 
			
		||||
	}
 | 
			
		||||
@@ -60,8 +50,8 @@ func (b *Bsteam) Disconnect() error {
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bsteam) JoinChannel(channel string) error {
 | 
			
		||||
	id, err := steamid.NewId(channel)
 | 
			
		||||
func (b *Bsteam) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	id, err := steamid.NewId(channel.Name)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
@@ -69,13 +59,41 @@ func (b *Bsteam) JoinChannel(channel string) error {
 | 
			
		||||
	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)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Handle files
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, 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
 | 
			
		||||
					if fi.Comment != "" {
 | 
			
		||||
						msg.Text = fi.Comment + ": " + fi.URL
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
				b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text)
 | 
			
		||||
			}
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text)
 | 
			
		||||
	return nil
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bsteam) getNick(id steamid.SteamId) string {
 | 
			
		||||
@@ -89,24 +107,29 @@ func (b *Bsteam) getNick(id steamid.SteamId) string {
 | 
			
		||||
 | 
			
		||||
func (b *Bsteam) handleEvents() {
 | 
			
		||||
	myLoginInfo := new(steam.LogOnDetails)
 | 
			
		||||
	myLoginInfo.Username = b.Config.Login
 | 
			
		||||
	myLoginInfo.Password = b.Config.Password
 | 
			
		||||
	myLoginInfo.AuthCode = b.Config.AuthCode
 | 
			
		||||
	myLoginInfo.Username = b.GetString("Login")
 | 
			
		||||
	myLoginInfo.Password = b.GetString("Password")
 | 
			
		||||
	myLoginInfo.AuthCode = b.GetString("AuthCode")
 | 
			
		||||
	// Attempt to read existing auth hash to avoid steam guard.
 | 
			
		||||
	// Maybe works
 | 
			
		||||
	//myLoginInfo.SentryFileHash, _ = ioutil.ReadFile("sentry")
 | 
			
		||||
	for event := range b.c.Events() {
 | 
			
		||||
		//flog.Info(event)
 | 
			
		||||
		//b.Log.Info(event)
 | 
			
		||||
		switch e := event.(type) {
 | 
			
		||||
		case *steam.ChatMsgEvent:
 | 
			
		||||
			flog.Debugf("Receiving ChatMsgEvent: %#v", e)
 | 
			
		||||
			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account)
 | 
			
		||||
			// for some reason we have to remove 0x18000000000000
 | 
			
		||||
			channel := int64(e.ChatRoomId) - 0x18000000000000
 | 
			
		||||
			b.Log.Debugf("Receiving ChatMsgEvent: %#v", e)
 | 
			
		||||
			b.Log.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
 | 
			
		||||
				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)}
 | 
			
		||||
			b.Remote <- msg
 | 
			
		||||
		case *steam.PersonaStateEvent:
 | 
			
		||||
			flog.Debugf("PersonaStateEvent: %#v\n", e)
 | 
			
		||||
			b.Log.Debugf("PersonaStateEvent: %#v\n", e)
 | 
			
		||||
			b.Lock()
 | 
			
		||||
			b.userMap[e.FriendId] = e.Name
 | 
			
		||||
			b.Unlock()
 | 
			
		||||
@@ -114,47 +137,47 @@ func (b *Bsteam) handleEvents() {
 | 
			
		||||
			b.c.Auth.LogOn(myLoginInfo)
 | 
			
		||||
		case *steam.MachineAuthUpdateEvent:
 | 
			
		||||
			/*
 | 
			
		||||
				flog.Info("authupdate", e)
 | 
			
		||||
				flog.Info("hash", e.Hash)
 | 
			
		||||
				b.Log.Info("authupdate", e)
 | 
			
		||||
				b.Log.Info("hash", e.Hash)
 | 
			
		||||
				ioutil.WriteFile("sentry", e.Hash, 0666)
 | 
			
		||||
			*/
 | 
			
		||||
		case *steam.LogOnFailedEvent:
 | 
			
		||||
			flog.Info("Logon failed", e)
 | 
			
		||||
			b.Log.Info("Logon failed", e)
 | 
			
		||||
			switch e.Result {
 | 
			
		||||
			case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode:
 | 
			
		||||
				{
 | 
			
		||||
					flog.Info("Steam guard isn't letting me in! Enter 2FA code:")
 | 
			
		||||
					b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:")
 | 
			
		||||
					var code string
 | 
			
		||||
					fmt.Scanf("%s", &code)
 | 
			
		||||
					myLoginInfo.TwoFactorCode = code
 | 
			
		||||
				}
 | 
			
		||||
			case steamlang.EResult_AccountLogonDenied:
 | 
			
		||||
				{
 | 
			
		||||
					flog.Info("Steam guard isn't letting me in! Enter auth code:")
 | 
			
		||||
					b.Log.Info("Steam guard isn't letting me in! Enter auth code:")
 | 
			
		||||
					var code string
 | 
			
		||||
					fmt.Scanf("%s", &code)
 | 
			
		||||
					myLoginInfo.AuthCode = code
 | 
			
		||||
				}
 | 
			
		||||
			default:
 | 
			
		||||
				log.Errorf("LogOnFailedEvent: ", e.Result)
 | 
			
		||||
				b.Log.Errorf("LogOnFailedEvent: %#v ", e.Result)
 | 
			
		||||
				// TODO: Handle EResult_InvalidLoginAuthCode
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
		case *steam.LoggedOnEvent:
 | 
			
		||||
			flog.Debugf("LoggedOnEvent: %#v", e)
 | 
			
		||||
			b.Log.Debugf("LoggedOnEvent: %#v", e)
 | 
			
		||||
			b.connected <- struct{}{}
 | 
			
		||||
			flog.Debugf("setting online")
 | 
			
		||||
			b.Log.Debugf("setting online")
 | 
			
		||||
			b.c.Social.SetPersonaState(steamlang.EPersonaState_Online)
 | 
			
		||||
		case *steam.DisconnectedEvent:
 | 
			
		||||
			flog.Info("Disconnected")
 | 
			
		||||
			flog.Info("Attempting to reconnect...")
 | 
			
		||||
			b.Log.Info("Disconnected")
 | 
			
		||||
			b.Log.Info("Attempting to reconnect...")
 | 
			
		||||
			b.c.Connect()
 | 
			
		||||
		case steam.FatalErrorEvent:
 | 
			
		||||
			flog.Error(e)
 | 
			
		||||
			b.Log.Error(e)
 | 
			
		||||
		case error:
 | 
			
		||||
			flog.Error(e)
 | 
			
		||||
			b.Log.Error(e)
 | 
			
		||||
		default:
 | 
			
		||||
			flog.Debugf("unknown event %#v", e)
 | 
			
		||||
			b.Log.Debugf("unknown event %#v", e)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -6,11 +6,11 @@ import (
 | 
			
		||||
	"html"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type customHtml struct {
 | 
			
		||||
type customHTML struct {
 | 
			
		||||
	blackfriday.Renderer
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) Paragraph(out *bytes.Buffer, text func() bool) {
 | 
			
		||||
func (options *customHTML) Paragraph(out *bytes.Buffer, text func() bool) {
 | 
			
		||||
	marker := out.Len()
 | 
			
		||||
 | 
			
		||||
	if !text() {
 | 
			
		||||
@@ -20,32 +20,32 @@ func (options *customHtml) Paragraph(out *bytes.Buffer, text func() bool) {
 | 
			
		||||
	out.WriteString("\n")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) BlockCode(out *bytes.Buffer, text []byte, lang string) {
 | 
			
		||||
func (options *customHTML) BlockCode(out *bytes.Buffer, text []byte, lang string) {
 | 
			
		||||
	out.WriteString("<pre>")
 | 
			
		||||
 | 
			
		||||
	out.WriteString(html.EscapeString(string(text)))
 | 
			
		||||
	out.WriteString("</pre>\n")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) Header(out *bytes.Buffer, text func() bool, level int, id string) {
 | 
			
		||||
func (options *customHTML) Header(out *bytes.Buffer, text func() bool, level int, id string) {
 | 
			
		||||
	options.Paragraph(out, text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) HRule(out *bytes.Buffer) {
 | 
			
		||||
func (options *customHTML) HRule(out *bytes.Buffer) {
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) BlockQuote(out *bytes.Buffer, text []byte) {
 | 
			
		||||
func (options *customHTML) BlockQuote(out *bytes.Buffer, text []byte) {
 | 
			
		||||
	out.WriteString("> ")
 | 
			
		||||
	out.Write(text)
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) List(out *bytes.Buffer, text func() bool, flags int) {
 | 
			
		||||
func (options *customHTML) List(out *bytes.Buffer, text func() bool, flags int) {
 | 
			
		||||
	options.Paragraph(out, text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (options *customHtml) ListItem(out *bytes.Buffer, text []byte, flags int) {
 | 
			
		||||
func (options *customHTML) ListItem(out *bytes.Buffer, text []byte, flags int) {
 | 
			
		||||
	out.WriteString("- ")
 | 
			
		||||
	out.Write(text)
 | 
			
		||||
	out.WriteByte('\n')
 | 
			
		||||
@@ -53,7 +53,7 @@ func (options *customHtml) ListItem(out *bytes.Buffer, text []byte, flags int) {
 | 
			
		||||
 | 
			
		||||
func makeHTML(input string) string {
 | 
			
		||||
	return string(blackfriday.Markdown([]byte(input),
 | 
			
		||||
		&customHtml{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")},
 | 
			
		||||
		&customHTML{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")},
 | 
			
		||||
		blackfriday.EXTENSION_NO_INTRA_EMPHASIS|
 | 
			
		||||
			blackfriday.EXTENSION_FENCED_CODE|
 | 
			
		||||
			blackfriday.EXTENSION_AUTOLINK|
 | 
			
		||||
 
 | 
			
		||||
@@ -1,136 +1,242 @@
 | 
			
		||||
package btelegram
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"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"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Btelegram struct {
 | 
			
		||||
	c       *tgbotapi.BotAPI
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
	c *tgbotapi.BotAPI
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
	avatarMap map[string]string // keep cache of userid and avatar sha
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "telegram"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Btelegram {
 | 
			
		||||
	b := &Btelegram{}
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	return b
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Btelegram{Config: cfg, avatarMap: make(map[string]string)}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Info("Connecting")
 | 
			
		||||
	b.c, err = tgbotapi.NewBotAPI(b.Config.Token)
 | 
			
		||||
	b.Log.Info("Connecting")
 | 
			
		||||
	b.c, err = tgbotapi.NewBotAPI(b.GetString("Token"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		b.Log.Debugf("%#v", 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 {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		b.Log.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	go b.handleRecv(updates)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) JoinChannel(channel string) error {
 | 
			
		||||
func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
func (b *Btelegram) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	// get the chatid
 | 
			
		||||
	chatid, err := strconv.ParseInt(msg.Channel, 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if b.Config.MessageFormat == "HTML" {
 | 
			
		||||
	// map the file SHA to our user (caches the avatar)
 | 
			
		||||
	if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
 | 
			
		||||
		return b.cacheAvatar(&msg)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if b.GetString("MessageFormat") == "HTML" {
 | 
			
		||||
		msg.Text = makeHTML(msg.Text)
 | 
			
		||||
	}
 | 
			
		||||
	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text)
 | 
			
		||||
	if b.Config.MessageFormat == "HTML" {
 | 
			
		||||
		m.ParseMode = tgbotapi.ModeHTML
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	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
 | 
			
		||||
	}
 | 
			
		||||
	_, err = b.c.Send(m)
 | 
			
		||||
	return err
 | 
			
		||||
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	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 {
 | 
			
		||||
			b.handleUploadFile(&msg, chatid)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 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.GetString("MessageFormat") == "HTML" {
 | 
			
		||||
			b.Log.Debug("Using mode HTML")
 | 
			
		||||
			m.ParseMode = tgbotapi.ModeHTML
 | 
			
		||||
		}
 | 
			
		||||
		if b.GetString("MessageFormat") == "Markdown" {
 | 
			
		||||
			b.Log.Debug("Using mode markdown")
 | 
			
		||||
			m.ParseMode = tgbotapi.ModeMarkdown
 | 
			
		||||
		}
 | 
			
		||||
		_, err = b.c.Send(m)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	return b.sendMessage(chatid, msg.Username, msg.Text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
 | 
			
		||||
	for update := range updates {
 | 
			
		||||
		b.Log.Debugf("== Receiving event: %#v", update.Message)
 | 
			
		||||
 | 
			
		||||
		if update.Message == nil && update.ChannelPost == nil && update.EditedMessage == nil && update.EditedChannelPost == nil {
 | 
			
		||||
			b.Log.Error("Getting nil messages, this shouldn't happen.")
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		var message *tgbotapi.Message
 | 
			
		||||
		username := ""
 | 
			
		||||
		channel := ""
 | 
			
		||||
		text := ""
 | 
			
		||||
 | 
			
		||||
		rmsg := config.Message{Account: b.Account, Extra: make(map[string][]interface{})}
 | 
			
		||||
 | 
			
		||||
		// handle channels
 | 
			
		||||
		if update.ChannelPost != nil {
 | 
			
		||||
			message = update.ChannelPost
 | 
			
		||||
			rmsg.Text = message.Text
 | 
			
		||||
		}
 | 
			
		||||
		if update.EditedChannelPost != nil && !b.Config.EditDisable {
 | 
			
		||||
 | 
			
		||||
		// edited channel message
 | 
			
		||||
		if update.EditedChannelPost != nil && !b.GetBool("EditDisable") {
 | 
			
		||||
			message = update.EditedChannelPost
 | 
			
		||||
			message.Text = message.Text + b.Config.EditSuffix
 | 
			
		||||
			rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix")
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// handle groups
 | 
			
		||||
		if update.Message != nil {
 | 
			
		||||
			message = update.Message
 | 
			
		||||
		}
 | 
			
		||||
		if update.EditedMessage != nil && !b.Config.EditDisable {
 | 
			
		||||
			message = update.EditedMessage
 | 
			
		||||
			message.Text = message.Text + b.Config.EditSuffix
 | 
			
		||||
		}
 | 
			
		||||
		if message.From != nil {
 | 
			
		||||
			if b.Config.UseFirstName {
 | 
			
		||||
				username = message.From.FirstName
 | 
			
		||||
			}
 | 
			
		||||
			if username == "" {
 | 
			
		||||
				username = message.From.UserName
 | 
			
		||||
				if username == "" {
 | 
			
		||||
					username = message.From.FirstName
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			text = message.Text
 | 
			
		||||
			channel = strconv.FormatInt(message.Chat.ID, 10)
 | 
			
		||||
			rmsg.Text = message.Text
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if username == "" {
 | 
			
		||||
			username = "unknown"
 | 
			
		||||
		// edited group message
 | 
			
		||||
		if update.EditedMessage != nil && !b.GetBool("EditDisable") {
 | 
			
		||||
			message = update.EditedMessage
 | 
			
		||||
			rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix")
 | 
			
		||||
		}
 | 
			
		||||
		if message.Sticker != nil && b.Config.UseInsecureURL {
 | 
			
		||||
			text = text + " " + b.getFileDirectURL(message.Sticker.FileID)
 | 
			
		||||
 | 
			
		||||
		// set the ID's from the channel or group message
 | 
			
		||||
		rmsg.ID = strconv.Itoa(message.MessageID)
 | 
			
		||||
		rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10)
 | 
			
		||||
 | 
			
		||||
		// handle username
 | 
			
		||||
		if message.From != nil {
 | 
			
		||||
			rmsg.UserID = strconv.Itoa(message.From.ID)
 | 
			
		||||
			if b.GetBool("UseFirstName") {
 | 
			
		||||
				rmsg.Username = message.From.FirstName
 | 
			
		||||
			}
 | 
			
		||||
			if rmsg.Username == "" {
 | 
			
		||||
				rmsg.Username = message.From.UserName
 | 
			
		||||
				if rmsg.Username == "" {
 | 
			
		||||
					rmsg.Username = message.From.FirstName
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			// only download avatars if we have a place to upload them (configured mediaserver)
 | 
			
		||||
			if b.General.MediaServerUpload != "" {
 | 
			
		||||
				b.handleDownloadAvatar(message.From.ID, rmsg.Channel)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if message.Video != nil && b.Config.UseInsecureURL {
 | 
			
		||||
			text = text + " " + b.getFileDirectURL(message.Video.FileID)
 | 
			
		||||
 | 
			
		||||
		// if we really didn't find a username, set it to unknown
 | 
			
		||||
		if rmsg.Username == "" {
 | 
			
		||||
			rmsg.Username = "unknown"
 | 
			
		||||
		}
 | 
			
		||||
		if message.Photo != nil && b.Config.UseInsecureURL {
 | 
			
		||||
			photos := *message.Photo
 | 
			
		||||
			// last photo is the biggest
 | 
			
		||||
			text = text + " " + b.getFileDirectURL(photos[len(photos)-1].FileID)
 | 
			
		||||
 | 
			
		||||
		// handle any downloads
 | 
			
		||||
		err := b.handleDownload(message, &rmsg)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			b.Log.Errorf("download failed: %s", err)
 | 
			
		||||
		}
 | 
			
		||||
		if message.Document != nil && b.Config.UseInsecureURL {
 | 
			
		||||
			text = text + " " + message.Document.FileName + " : " + b.getFileDirectURL(message.Document.FileID)
 | 
			
		||||
 | 
			
		||||
		// handle forwarded messages
 | 
			
		||||
		if message.ForwardFrom != nil {
 | 
			
		||||
			usernameForward := ""
 | 
			
		||||
			if b.GetBool("UseFirstName") {
 | 
			
		||||
				usernameForward = message.ForwardFrom.FirstName
 | 
			
		||||
			}
 | 
			
		||||
			if usernameForward == "" {
 | 
			
		||||
				usernameForward = message.ForwardFrom.UserName
 | 
			
		||||
				if usernameForward == "" {
 | 
			
		||||
					usernameForward = message.ForwardFrom.FirstName
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			if usernameForward == "" {
 | 
			
		||||
				usernameForward = "unknown"
 | 
			
		||||
			}
 | 
			
		||||
			rmsg.Text = "Forwarded from " + usernameForward + ": " + rmsg.Text
 | 
			
		||||
		}
 | 
			
		||||
		if text != "" {
 | 
			
		||||
			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account)
 | 
			
		||||
			b.Remote <- config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID)}
 | 
			
		||||
 | 
			
		||||
		// quote the previous message
 | 
			
		||||
		if message.ReplyToMessage != nil {
 | 
			
		||||
			usernameReply := ""
 | 
			
		||||
			if message.ReplyToMessage.From != nil {
 | 
			
		||||
				if b.GetBool("UseFirstName") {
 | 
			
		||||
					usernameReply = message.ReplyToMessage.From.FirstName
 | 
			
		||||
				}
 | 
			
		||||
				if usernameReply == "" {
 | 
			
		||||
					usernameReply = message.ReplyToMessage.From.UserName
 | 
			
		||||
					if usernameReply == "" {
 | 
			
		||||
						usernameReply = message.ReplyToMessage.From.FirstName
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			if usernameReply == "" {
 | 
			
		||||
				usernameReply = "unknown"
 | 
			
		||||
			}
 | 
			
		||||
			if !b.GetBool("QuoteDisable") {
 | 
			
		||||
				rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, message.ReplyToMessage.Text)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if rmsg.Text != "" || len(rmsg.Extra) > 0 {
 | 
			
		||||
			rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text)
 | 
			
		||||
			// channels don't have (always?) user information. see #410
 | 
			
		||||
			if message.From != nil {
 | 
			
		||||
				rmsg.Avatar = helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
 | 
			
		||||
			b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
			b.Remote <- rmsg
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@@ -142,3 +248,181 @@ func (b *Btelegram) getFileDirectURL(id string) string {
 | 
			
		||||
	}
 | 
			
		||||
	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) {
 | 
			
		||||
	rmsg := 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 {
 | 
			
		||||
			b.Log.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"
 | 
			
		||||
			b.Log.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize)
 | 
			
		||||
 | 
			
		||||
			err := helper.HandleDownloadSize(b.Log, &rmsg, name, int64(photo.FileSize), b.General)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Error(err)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			data, err := helper.DownloadFile(url)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				b.Log.Errorf("download %s failed %#v", url, err)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			helper.HandleDownloadData(b.Log, &rmsg, name, rmsg.Text, "", data, b.General)
 | 
			
		||||
			b.Remote <- rmsg
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleDownloadFile handles file download
 | 
			
		||||
func (b *Btelegram) handleDownload(message *tgbotapi.Message, rmsg *config.Message) error {
 | 
			
		||||
	size := 0
 | 
			
		||||
	var url, name, text string
 | 
			
		||||
 | 
			
		||||
	if message.Sticker != nil {
 | 
			
		||||
		v := message.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
 | 
			
		||||
	}
 | 
			
		||||
	if message.Video != nil {
 | 
			
		||||
		v := message.Video
 | 
			
		||||
		size = v.FileSize
 | 
			
		||||
		url = b.getFileDirectURL(v.FileID)
 | 
			
		||||
		urlPart := strings.Split(url, "/")
 | 
			
		||||
		name = urlPart[len(urlPart)-1]
 | 
			
		||||
		text = " " + url
 | 
			
		||||
	}
 | 
			
		||||
	if message.Photo != nil {
 | 
			
		||||
		photos := *message.Photo
 | 
			
		||||
		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
 | 
			
		||||
	}
 | 
			
		||||
	if message.Document != nil {
 | 
			
		||||
		v := message.Document
 | 
			
		||||
		size = v.FileSize
 | 
			
		||||
		url = b.getFileDirectURL(v.FileID)
 | 
			
		||||
		name = v.FileName
 | 
			
		||||
		text = " " + v.FileName + " : " + url
 | 
			
		||||
	}
 | 
			
		||||
	if message.Voice != nil {
 | 
			
		||||
		v := message.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"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if message.Audio != nil {
 | 
			
		||||
		v := message.Audio
 | 
			
		||||
		size = v.FileSize
 | 
			
		||||
		url = b.getFileDirectURL(v.FileID)
 | 
			
		||||
		urlPart := strings.Split(url, "/")
 | 
			
		||||
		name = urlPart[len(urlPart)-1]
 | 
			
		||||
		text = " " + url
 | 
			
		||||
	}
 | 
			
		||||
	// if name is empty we didn't match a thing to download
 | 
			
		||||
	if name == "" {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	// use the URL instead of native upload
 | 
			
		||||
	if b.GetBool("UseInsecureURL") {
 | 
			
		||||
		b.Log.Debugf("Setting message text to :%s", text)
 | 
			
		||||
		rmsg.Text = rmsg.Text + text
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
 | 
			
		||||
	err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	data, err := helper.DownloadFile(url)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) (string, error) {
 | 
			
		||||
	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 {
 | 
			
		||||
			b.Log.Errorf("file upload failed: %#v", err)
 | 
			
		||||
		}
 | 
			
		||||
		if fi.Comment != "" {
 | 
			
		||||
			b.sendMessage(chatid, msg.Username, fi.Comment)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) {
 | 
			
		||||
	m := tgbotapi.NewMessage(chatid, "")
 | 
			
		||||
	m.Text = username + text
 | 
			
		||||
	if b.GetString("MessageFormat") == "HTML" {
 | 
			
		||||
		b.Log.Debug("Using mode HTML")
 | 
			
		||||
		m.Text = username + text
 | 
			
		||||
		m.ParseMode = tgbotapi.ModeHTML
 | 
			
		||||
	}
 | 
			
		||||
	if b.GetString("MessageFormat") == "Markdown" {
 | 
			
		||||
		b.Log.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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) cacheAvatar(msg *config.Message) (string, error) {
 | 
			
		||||
	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 != "" {
 | 
			
		||||
		b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
 | 
			
		||||
		b.avatarMap[msg.UserID] = fi.SHA
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string {
 | 
			
		||||
	format := b.GetString("quoteformat")
 | 
			
		||||
	if format == "" {
 | 
			
		||||
		format = "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})"
 | 
			
		||||
	}
 | 
			
		||||
	format = strings.Replace(format, "{MESSAGE}", message, -1)
 | 
			
		||||
	format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1)
 | 
			
		||||
	format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1)
 | 
			
		||||
	return format
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -2,10 +2,11 @@ package bxmpp
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/mattn/go-xmpp"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	"github.com/jpillora/backoff"
 | 
			
		||||
	"github.com/matterbridge/go-xmpp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
@@ -13,37 +14,47 @@ import (
 | 
			
		||||
type Bxmpp struct {
 | 
			
		||||
	xc      *xmpp.Client
 | 
			
		||||
	xmppMap map[string]string
 | 
			
		||||
	Config  *config.Protocol
 | 
			
		||||
	Remote  chan config.Message
 | 
			
		||||
	Account string
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
var protocol = "xmpp"
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	flog = log.WithFields(log.Fields{"module": protocol})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bxmpp {
 | 
			
		||||
	b := &Bxmpp{}
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	b := &Bxmpp{Config: cfg}
 | 
			
		||||
	b.xmppMap = make(map[string]string)
 | 
			
		||||
	b.Config = &cfg
 | 
			
		||||
	b.Account = account
 | 
			
		||||
	b.Remote = c
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) Connect() error {
 | 
			
		||||
	var err error
 | 
			
		||||
	flog.Infof("Connecting %s", b.Config.Server)
 | 
			
		||||
	b.Log.Infof("Connecting %s", b.GetString("Server"))
 | 
			
		||||
	b.xc, err = b.createXMPP()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Debugf("%#v", err)
 | 
			
		||||
		b.Log.Debugf("%#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	flog.Info("Connection succeeded")
 | 
			
		||||
	go b.handleXmpp()
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	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()
 | 
			
		||||
			b.Log.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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -51,37 +62,54 @@ func (b *Bxmpp) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) JoinChannel(channel string) error {
 | 
			
		||||
	b.xc.JoinMUCNoHistory(channel+"@"+b.Config.Muc, b.Config.Nick)
 | 
			
		||||
func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	b.xc.JoinMUCNoHistory(channel.Name+"@"+b.GetString("Muc"), b.GetString("Nick"))
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) Send(msg config.Message) error {
 | 
			
		||||
	flog.Debugf("Receiving %#v", msg)
 | 
			
		||||
	b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text})
 | 
			
		||||
	return nil
 | 
			
		||||
func (b *Bxmpp) Send(msg config.Message) (string, error) {
 | 
			
		||||
	// ignore delete messages
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	// Upload a file (in xmpp case send the upload URL because xmpp has no native upload support)
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: rmsg.Channel + "@" + b.GetString("Muc"), Text: rmsg.Username + rmsg.Text})
 | 
			
		||||
		}
 | 
			
		||||
		if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
			return b.handleUploadFile(&msg)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	_, err := b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.GetString("Muc"), Text: msg.Username + msg.Text})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) createXMPP() (*xmpp.Client, error) {
 | 
			
		||||
	tc := new(tls.Config)
 | 
			
		||||
	tc.InsecureSkipVerify = b.Config.SkipTLSVerify
 | 
			
		||||
	tc.ServerName = strings.Split(b.Config.Server, ":")[0]
 | 
			
		||||
	tc.InsecureSkipVerify = b.GetBool("SkipTLSVerify")
 | 
			
		||||
	tc.ServerName = strings.Split(b.GetString("Server"), ":")[0]
 | 
			
		||||
	options := xmpp.Options{
 | 
			
		||||
		Host:      b.Config.Server,
 | 
			
		||||
		User:      b.Config.Jid,
 | 
			
		||||
		Password:  b.Config.Password,
 | 
			
		||||
		NoTLS:     true,
 | 
			
		||||
		StartTLS:  true,
 | 
			
		||||
		TLSConfig: tc,
 | 
			
		||||
 | 
			
		||||
		//StartTLS:      false,
 | 
			
		||||
		Debug:                        true,
 | 
			
		||||
		Host:                         b.GetString("Server"),
 | 
			
		||||
		User:                         b.GetString("Jid"),
 | 
			
		||||
		Password:                     b.GetString("Password"),
 | 
			
		||||
		NoTLS:                        true,
 | 
			
		||||
		StartTLS:                     true,
 | 
			
		||||
		TLSConfig:                    tc,
 | 
			
		||||
		Debug:                        b.GetBool("debug"),
 | 
			
		||||
		Logger:                       b.Log.Writer(),
 | 
			
		||||
		Session:                      true,
 | 
			
		||||
		Status:                       "",
 | 
			
		||||
		StatusMessage:                "",
 | 
			
		||||
		Resource:                     "",
 | 
			
		||||
		InsecureAllowUnencryptedAuth: false,
 | 
			
		||||
		//InsecureAllowUnencryptedAuth: true,
 | 
			
		||||
	}
 | 
			
		||||
	var err error
 | 
			
		||||
	b.xc, err = options.NewClient()
 | 
			
		||||
@@ -96,7 +124,11 @@ func (b *Bxmpp) xmppKeepAlive() chan bool {
 | 
			
		||||
		for {
 | 
			
		||||
			select {
 | 
			
		||||
			case <-ticker.C:
 | 
			
		||||
				b.xc.PingC2S("", "")
 | 
			
		||||
				b.Log.Debugf("PING")
 | 
			
		||||
				err := b.xc.PingC2S("", "")
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					b.Log.Debugf("PING failed %#v", err)
 | 
			
		||||
				}
 | 
			
		||||
			case <-done:
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
@@ -105,10 +137,10 @@ func (b *Bxmpp) xmppKeepAlive() chan bool {
 | 
			
		||||
	return done
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) handleXmpp() error {
 | 
			
		||||
func (b *Bxmpp) handleXMPP() error {
 | 
			
		||||
	var ok bool
 | 
			
		||||
	done := b.xmppKeepAlive()
 | 
			
		||||
	defer close(done)
 | 
			
		||||
	nodelay := time.Time{}
 | 
			
		||||
	for {
 | 
			
		||||
		m, err := b.xc.Recv()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
@@ -116,23 +148,99 @@ func (b *Bxmpp) handleXmpp() error {
 | 
			
		||||
		}
 | 
			
		||||
		switch v := m.(type) {
 | 
			
		||||
		case xmpp.Chat:
 | 
			
		||||
			var channel, nick string
 | 
			
		||||
			if v.Type == "groupchat" {
 | 
			
		||||
				s := strings.Split(v.Remote, "@")
 | 
			
		||||
				if len(s) >= 2 {
 | 
			
		||||
					channel = s[0]
 | 
			
		||||
				b.Log.Debugf("== Receiving %#v", v)
 | 
			
		||||
				// skip invalid messages
 | 
			
		||||
				if b.skipMessage(v) {
 | 
			
		||||
					continue
 | 
			
		||||
				}
 | 
			
		||||
				s = strings.Split(s[1], "/")
 | 
			
		||||
				if len(s) == 2 {
 | 
			
		||||
					nick = s[1]
 | 
			
		||||
				}
 | 
			
		||||
				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" {
 | 
			
		||||
					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account)
 | 
			
		||||
					b.Remote <- config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote}
 | 
			
		||||
				rmsg := config.Message{Username: b.parseNick(v.Remote), Text: v.Text, Channel: b.parseChannel(v.Remote), Account: b.Account, UserID: v.Remote}
 | 
			
		||||
 | 
			
		||||
				// check if we have an action event
 | 
			
		||||
				rmsg.Text, ok = b.replaceAction(rmsg.Text)
 | 
			
		||||
				if ok {
 | 
			
		||||
					rmsg.Event = config.EVENT_USER_ACTION
 | 
			
		||||
				}
 | 
			
		||||
				b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
 | 
			
		||||
				b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
				b.Remote <- rmsg
 | 
			
		||||
			}
 | 
			
		||||
		case xmpp.Presence:
 | 
			
		||||
			// do nothing
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) replaceAction(text string) (string, bool) {
 | 
			
		||||
	if strings.HasPrefix(text, "/me ") {
 | 
			
		||||
		return strings.Replace(text, "/me ", "", -1), true
 | 
			
		||||
	}
 | 
			
		||||
	return text, false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// handleUploadFile handles native upload of files
 | 
			
		||||
func (b *Bxmpp) handleUploadFile(msg *config.Message) (string, error) {
 | 
			
		||||
	var urldesc = ""
 | 
			
		||||
	for _, f := range msg.Extra["file"] {
 | 
			
		||||
		fi := f.(config.FileInfo)
 | 
			
		||||
		if fi.Comment != "" {
 | 
			
		||||
			msg.Text += fi.Comment + ": "
 | 
			
		||||
		}
 | 
			
		||||
		if fi.URL != "" {
 | 
			
		||||
			msg.Text = fi.URL
 | 
			
		||||
			if fi.Comment != "" {
 | 
			
		||||
				msg.Text = fi.Comment + ": " + fi.URL
 | 
			
		||||
				urldesc = fi.Comment
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		_, err := b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.GetString("Muc"), Text: msg.Username + msg.Text})
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		if fi.URL != "" {
 | 
			
		||||
			b.xc.SendOOB(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.GetString("Muc"), Ooburl: fi.URL, Oobdesc: urldesc})
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) parseNick(remote string) string {
 | 
			
		||||
	s := strings.Split(remote, "@")
 | 
			
		||||
	if len(s) > 0 {
 | 
			
		||||
		s = strings.Split(s[1], "/")
 | 
			
		||||
		if len(s) == 2 {
 | 
			
		||||
			return s[1] // nick
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bxmpp) parseChannel(remote string) string {
 | 
			
		||||
	s := strings.Split(remote, "@")
 | 
			
		||||
	if len(s) >= 2 {
 | 
			
		||||
		return s[0] // channel
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// skipMessage skips messages that need to be skipped
 | 
			
		||||
func (b *Bxmpp) skipMessage(message xmpp.Chat) bool {
 | 
			
		||||
	// skip messages from ourselves
 | 
			
		||||
	if b.parseNick(message.Remote) == b.GetString("Nick") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// skip empty messages
 | 
			
		||||
	if message.Text == "" {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// skip subject messages
 | 
			
		||||
	if strings.Contains(message.Text, "</subject>") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// skip delayed messages
 | 
			
		||||
	t := time.Time{}
 | 
			
		||||
	return message.Stamp != t
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										170
									
								
								bridge/zulip/zulip.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										170
									
								
								bridge/zulip/zulip.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,170 @@
 | 
			
		||||
package bzulip
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/helper"
 | 
			
		||||
	gzb "github.com/matterbridge/gozulipbot"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Bzulip struct {
 | 
			
		||||
	q       *gzb.Queue
 | 
			
		||||
	bot     *gzb.Bot
 | 
			
		||||
	streams map[int]string
 | 
			
		||||
	*bridge.Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *bridge.Config) bridge.Bridger {
 | 
			
		||||
	return &Bzulip{Config: cfg, streams: make(map[int]string)}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) Connect() error {
 | 
			
		||||
	bot := gzb.Bot{APIKey: b.GetString("token"), APIURL: b.GetString("server") + "/api/v1/", Email: b.GetString("login")}
 | 
			
		||||
	bot.Init()
 | 
			
		||||
	q, err := bot.RegisterAll()
 | 
			
		||||
	b.q = q
 | 
			
		||||
	b.bot = &bot
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		b.Log.Errorf("Connect() %#v", err)
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	// init stream
 | 
			
		||||
	b.getChannel(0)
 | 
			
		||||
	b.Log.Info("Connection succeeded")
 | 
			
		||||
	go b.handleQueue()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) Disconnect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) JoinChannel(channel config.ChannelInfo) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) Send(msg config.Message) (string, error) {
 | 
			
		||||
	b.Log.Debugf("=> Receiving %#v", msg)
 | 
			
		||||
 | 
			
		||||
	// Delete message
 | 
			
		||||
	if msg.Event == config.EVENT_MSG_DELETE {
 | 
			
		||||
		if msg.ID == "" {
 | 
			
		||||
			return "", nil
 | 
			
		||||
		}
 | 
			
		||||
		_, err := b.bot.UpdateMessage(msg.ID, "")
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Upload a file if it exists
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		for _, rmsg := range helper.HandleExtra(&msg, b.General) {
 | 
			
		||||
			b.sendMessage(rmsg)
 | 
			
		||||
		}
 | 
			
		||||
		if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
			return b.handleUploadFile(&msg)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// edit the message if we have a msg ID
 | 
			
		||||
	if msg.ID != "" {
 | 
			
		||||
		_, err := b.bot.UpdateMessage(msg.ID, msg.Username+msg.Text)
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Post normal message
 | 
			
		||||
	return b.sendMessage(msg)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) getChannel(id int) string {
 | 
			
		||||
	if name, ok := b.streams[id]; ok {
 | 
			
		||||
		return name
 | 
			
		||||
	}
 | 
			
		||||
	streams, err := b.bot.GetRawStreams()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		b.Log.Errorf("getChannel: %#v", err)
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	for _, stream := range streams.Streams {
 | 
			
		||||
		b.streams[stream.StreamID] = stream.Name
 | 
			
		||||
	}
 | 
			
		||||
	if name, ok := b.streams[id]; ok {
 | 
			
		||||
		return name
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) handleQueue() error {
 | 
			
		||||
	for {
 | 
			
		||||
		messages, _ := b.q.GetEvents()
 | 
			
		||||
		for _, m := range messages {
 | 
			
		||||
			b.Log.Debugf("== Receiving %#v", m)
 | 
			
		||||
			// ignore our own messages
 | 
			
		||||
			if m.SenderEmail == b.GetString("login") {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			rmsg := config.Message{Username: m.SenderFullName, Text: m.Content, Channel: b.getChannel(m.StreamID), Account: b.Account, UserID: strconv.Itoa(m.SenderID), Avatar: m.AvatarURL}
 | 
			
		||||
			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
 | 
			
		||||
			b.Log.Debugf("<= Message is %#v", rmsg)
 | 
			
		||||
			b.Remote <- rmsg
 | 
			
		||||
			b.q.LastEventID = m.ID
 | 
			
		||||
		}
 | 
			
		||||
		time.Sleep(time.Second * 3)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) sendMessage(msg config.Message) (string, error) {
 | 
			
		||||
	topic := "matterbridge"
 | 
			
		||||
	if b.GetString("topic") != "" {
 | 
			
		||||
		topic = b.GetString("topic")
 | 
			
		||||
	}
 | 
			
		||||
	m := gzb.Message{
 | 
			
		||||
		Stream:  msg.Channel,
 | 
			
		||||
		Topic:   topic,
 | 
			
		||||
		Content: msg.Username + msg.Text,
 | 
			
		||||
	}
 | 
			
		||||
	resp, err := b.bot.Message(m)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	if resp != nil {
 | 
			
		||||
		defer resp.Body.Close()
 | 
			
		||||
		res, err := ioutil.ReadAll(resp.Body)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		var jr struct {
 | 
			
		||||
			ID int `json:"id"`
 | 
			
		||||
		}
 | 
			
		||||
		err = json.Unmarshal(res, &jr)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
		return strconv.Itoa(jr.ID), nil
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (b *Bzulip) handleUploadFile(msg *config.Message) (string, error) {
 | 
			
		||||
	for _, f := range msg.Extra["file"] {
 | 
			
		||||
		fi := f.(config.FileInfo)
 | 
			
		||||
		if fi.Comment != "" {
 | 
			
		||||
			msg.Text += fi.Comment + ": "
 | 
			
		||||
		}
 | 
			
		||||
		if fi.URL != "" {
 | 
			
		||||
			msg.Text = fi.URL
 | 
			
		||||
			if fi.Comment != "" {
 | 
			
		||||
				msg.Text = fi.Comment + ": " + fi.URL
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		_, err := b.sendMessage(*msg)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										317
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						
									
										317
									
								
								changelog.md
									
									
									
									
									
								
							@@ -1,3 +1,320 @@
 | 
			
		||||
# v1.10.1
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: updated irc/xmpp/telegram libraries
 | 
			
		||||
* mattermost/slack/rocketchat: Fix iconurl regression. Closes #430
 | 
			
		||||
* mattermost/slack: Use uuid instead of userid. Fixes #429
 | 
			
		||||
* slack: Avatar spoofing from Slack to Discord with uppercase in nick doesn't work (#433)
 | 
			
		||||
* irc: Fix format string bug (irc) (#428)
 | 
			
		||||
* irc: Colorize username sent to IRC using its crc32 IEEE checksum (#423). See `ColorNicks` in matterbridge.toml.sample
 | 
			
		||||
* irc: Add support for CJK to/from utf-8 (irc). #400
 | 
			
		||||
* telegram: Add QuoteFormat option (telegram). Closes #41. See `QuoteFormat` in matterbridge.toml.sample
 | 
			
		||||
* xmpp: Send attached files to XMPP in different message with OOB data and without body (#421)
 | 
			
		||||
 | 
			
		||||
# v1.10.0
 | 
			
		||||
## New features
 | 
			
		||||
* general: Add support for reloading all settings automatically after changing config except connection and gateway configuration. Closes #373
 | 
			
		||||
* zulip: New protocol support added (https://zulipchat.com)
 | 
			
		||||
 | 
			
		||||
## Enhancements
 | 
			
		||||
* general: Handle file comment better
 | 
			
		||||
* steam: Handle file uploads to mediaserver (steam)
 | 
			
		||||
* slack: Properly set Slack user who initiated slash command (#394)
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* general: Use only alphanumeric for file uploads to mediaserver. Closes #416
 | 
			
		||||
* general: Fix crash on invalid filenames
 | 
			
		||||
* general: Fix regression in ReplaceMessages and ReplaceNicks. Closes #407
 | 
			
		||||
* telegram: Fix possible nil when using channels (telegram). #410
 | 
			
		||||
* telegram: Fix panic (telegram). Closes #410
 | 
			
		||||
* telegram: Handle channel posts correctly
 | 
			
		||||
* mattermost: Update GetFileLinks to API_V4
 | 
			
		||||
 | 
			
		||||
# v1.9.1
 | 
			
		||||
## New features
 | 
			
		||||
* telegram: Add QuoteDisable option (telegram). Closes #399. See QuoteDisable in matterbridge.toml.sample
 | 
			
		||||
## Enhancements
 | 
			
		||||
* discord: Send mediaserver link to Discord in Webhook mode (discord) (#405)
 | 
			
		||||
* mattermost: Print list of valid team names when team not found (#390)
 | 
			
		||||
* slack: Strip markdown URLs with blank text (slack) (#392)
 | 
			
		||||
## Bugfix
 | 
			
		||||
* slack/mattermost: Make our callbackid more unique. Fixes issue with running multiple matterbridge on the same channel (slack,mattermost)
 | 
			
		||||
* telegram: fix newlines in multiline messages #399
 | 
			
		||||
* telegram: Revert #378
 | 
			
		||||
 | 
			
		||||
# v1.9.0 (the refactor release)
 | 
			
		||||
## New features
 | 
			
		||||
* general: better debug messages
 | 
			
		||||
* general: better support for environment variables override
 | 
			
		||||
* general: Ability to disable sending join/leave messages to other gateways. #382
 | 
			
		||||
* slack: Allow Slack @usergroups to be parsed as human-friendly names #379
 | 
			
		||||
* slack: Provide better context for shared posts from Slack<=>Slack enhancement #369
 | 
			
		||||
* telegram: Convert nicks automatically into HTML when MessageFormat is set to HTML #378
 | 
			
		||||
* irc: Add DebugLevel option 
 | 
			
		||||
 | 
			
		||||
## Bugfix
 | 
			
		||||
* slack: Ignore restricted_action on channel join (slack). Closes #387
 | 
			
		||||
* slack: Add slack attachment support to matterhook
 | 
			
		||||
* slack: Update userlist on join (slack). Closes #372
 | 
			
		||||
 | 
			
		||||
# 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
 | 
			
		||||
## Breaking Changes
 | 
			
		||||
* 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.10 || 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,30 @@
 | 
			
		||||
package gateway
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/api"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/discord"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/gitter"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/irc"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/matrix"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/mattermost"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/rocketchat"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/slack"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/sshchat"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/steam"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/telegram"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/xmpp"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/zulip"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
	//	"github.com/davecgh/go-spew/spew"
 | 
			
		||||
	"crypto/sha1"
 | 
			
		||||
	"github.com/hashicorp/golang-lru"
 | 
			
		||||
	"github.com/peterhellberg/emojilib"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"path/filepath"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
@@ -13,62 +32,71 @@ import (
 | 
			
		||||
 | 
			
		||||
type Gateway struct {
 | 
			
		||||
	*config.Config
 | 
			
		||||
	MyConfig        *config.Gateway
 | 
			
		||||
	Bridges         map[string]*bridge.Bridge
 | 
			
		||||
	Channels        map[string]*config.ChannelInfo
 | 
			
		||||
	ChannelOptions  map[string]config.ChannelOptions
 | 
			
		||||
	Names           map[string]bool
 | 
			
		||||
	Name            string
 | 
			
		||||
	Message         chan config.Message
 | 
			
		||||
	DestChannelFunc func(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo
 | 
			
		||||
	Router         *Router
 | 
			
		||||
	MyConfig       *config.Gateway
 | 
			
		||||
	Bridges        map[string]*bridge.Bridge
 | 
			
		||||
	Channels       map[string]*config.ChannelInfo
 | 
			
		||||
	ChannelOptions map[string]config.ChannelOptions
 | 
			
		||||
	Message        chan config.Message
 | 
			
		||||
	Name           string
 | 
			
		||||
	Messages       *lru.Cache
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg *config.Config) *Gateway {
 | 
			
		||||
	gw := &Gateway{}
 | 
			
		||||
	gw.Config = cfg
 | 
			
		||||
	gw.Channels = make(map[string]*config.ChannelInfo)
 | 
			
		||||
	gw.Message = make(chan config.Message)
 | 
			
		||||
	gw.Bridges = make(map[string]*bridge.Bridge)
 | 
			
		||||
	gw.Names = make(map[string]bool)
 | 
			
		||||
	gw.DestChannelFunc = gw.getDestChannel
 | 
			
		||||
type BrMsgID struct {
 | 
			
		||||
	br        *bridge.Bridge
 | 
			
		||||
	ID        string
 | 
			
		||||
	ChannelID string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var flog *log.Entry
 | 
			
		||||
 | 
			
		||||
var bridgeMap = map[string]bridge.Factory{
 | 
			
		||||
	"api":        api.New,
 | 
			
		||||
	"discord":    bdiscord.New,
 | 
			
		||||
	"gitter":     bgitter.New,
 | 
			
		||||
	"irc":        birc.New,
 | 
			
		||||
	"mattermost": bmattermost.New,
 | 
			
		||||
	"matrix":     bmatrix.New,
 | 
			
		||||
	"rocketchat": brocketchat.New,
 | 
			
		||||
	"slack":      bslack.New,
 | 
			
		||||
	"sshchat":    bsshchat.New,
 | 
			
		||||
	"steam":      bsteam.New,
 | 
			
		||||
	"telegram":   btelegram.New,
 | 
			
		||||
	"xmpp":       bxmpp.New,
 | 
			
		||||
	"zulip":      bzulip.New,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) AddBridge(cfg *config.Bridge) error {
 | 
			
		||||
	for _, br := range gw.Bridges {
 | 
			
		||||
		if br.Account == cfg.Account {
 | 
			
		||||
			gw.mapChannelsToBridge(br)
 | 
			
		||||
			err := br.JoinChannels()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err)
 | 
			
		||||
			}
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	br := gw.Router.getBridge(cfg.Account)
 | 
			
		||||
	if br == nil {
 | 
			
		||||
		br = bridge.New(cfg)
 | 
			
		||||
		br.Config = gw.Router.Config
 | 
			
		||||
		br.General = &gw.General
 | 
			
		||||
		// set logging
 | 
			
		||||
		br.Log = log.WithFields(log.Fields{"prefix": "bridge"})
 | 
			
		||||
		brconfig := &bridge.Config{Remote: gw.Message, Log: log.WithFields(log.Fields{"prefix": br.Protocol}), Bridge: br}
 | 
			
		||||
		// add the actual bridger for this protocol to this bridge using the bridgeMap
 | 
			
		||||
		br.Bridger = bridgeMap[br.Protocol](brconfig)
 | 
			
		||||
	}
 | 
			
		||||
	log.Infof("Starting bridge: %s ", cfg.Account)
 | 
			
		||||
	br := bridge.New(gw.Config, cfg, gw.Message)
 | 
			
		||||
	gw.mapChannelsToBridge(br)
 | 
			
		||||
	gw.Bridges[cfg.Account] = br
 | 
			
		||||
	err := br.Connect()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
 | 
			
		||||
	}
 | 
			
		||||
	err = br.JoinChannels()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) AddConfig(cfg *config.Gateway) error {
 | 
			
		||||
	if gw.Names[cfg.Name] {
 | 
			
		||||
		return fmt.Errorf("Gateway with name %s already exists", cfg.Name)
 | 
			
		||||
	}
 | 
			
		||||
	if cfg.Name == "" {
 | 
			
		||||
		return fmt.Errorf("%s", "Gateway without name found")
 | 
			
		||||
	}
 | 
			
		||||
	log.Infof("Starting gateway: %s", cfg.Name)
 | 
			
		||||
	gw.Names[cfg.Name] = true
 | 
			
		||||
	gw.Name = cfg.Name
 | 
			
		||||
	gw.MyConfig = cfg
 | 
			
		||||
	gw.mapChannels()
 | 
			
		||||
@@ -89,49 +117,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) {
 | 
			
		||||
	br.Disconnect()
 | 
			
		||||
	time.Sleep(time.Second * 5)
 | 
			
		||||
RECONNECT:
 | 
			
		||||
	log.Infof("Reconnecting %s", br.Account)
 | 
			
		||||
	flog.Infof("Reconnecting %s", br.Account)
 | 
			
		||||
	err := br.Connect()
 | 
			
		||||
	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)
 | 
			
		||||
		goto RECONNECT
 | 
			
		||||
	}
 | 
			
		||||
@@ -139,120 +132,197 @@ RECONNECT:
 | 
			
		||||
	br.JoinChannels()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) mapChannels() error {
 | 
			
		||||
	for _, br := range append(gw.MyConfig.Out, gw.MyConfig.InOut...) {
 | 
			
		||||
func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) {
 | 
			
		||||
	for _, br := range cfg {
 | 
			
		||||
		if isApi(br.Account) {
 | 
			
		||||
			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
 | 
			
		||||
		_, ok := gw.Channels[ID]
 | 
			
		||||
		if !ok {
 | 
			
		||||
			channel := &config.ChannelInfo{Name: br.Channel, Direction: "out", ID: ID, Options: br.Options, Account: br.Account,
 | 
			
		||||
				GID: make(map[string]bool), SameChannel: make(map[string]bool)}
 | 
			
		||||
			channel.GID[gw.Name] = true
 | 
			
		||||
		if _, ok := gw.Channels[ID]; !ok {
 | 
			
		||||
			channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account,
 | 
			
		||||
				SameChannel: make(map[string]bool)}
 | 
			
		||||
			channel.SameChannel[gw.Name] = br.SameChannel
 | 
			
		||||
			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
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
	for _, br := range append(gw.MyConfig.In, gw.MyConfig.InOut...) {
 | 
			
		||||
		if isApi(br.Account) {
 | 
			
		||||
			br.Channel = "api"
 | 
			
		||||
		}
 | 
			
		||||
		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
 | 
			
		||||
	}
 | 
			
		||||
func (gw *Gateway) mapChannels() error {
 | 
			
		||||
	gw.mapChannelConfig(gw.MyConfig.In, "in")
 | 
			
		||||
	gw.mapChannelConfig(gw.MyConfig.Out, "out")
 | 
			
		||||
	gw.mapChannelConfig(gw.MyConfig.InOut, "inout")
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []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 {
 | 
			
		||||
		if _, ok := gw.Channels[getChannelID(*msg)]; !ok {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		// add gateway to message
 | 
			
		||||
		gw.validGatewayDest(msg, channel)
 | 
			
		||||
 | 
			
		||||
		// do samechannelgateway logic
 | 
			
		||||
		// do samechannelgateway flogic
 | 
			
		||||
		if channel.SameChannel[msg.Gateway] {
 | 
			
		||||
			if msg.Channel == channel.Name && msg.Account != dest.Account {
 | 
			
		||||
				channels = append(channels, *channel)
 | 
			
		||||
			}
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if channel.Direction == "out" && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) {
 | 
			
		||||
		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) {
 | 
			
		||||
			channels = append(channels, *channel)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return channels
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) {
 | 
			
		||||
	// only relay join/part when configged
 | 
			
		||||
	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart {
 | 
			
		||||
		return
 | 
			
		||||
func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID {
 | 
			
		||||
	var brMsgIDs []*BrMsgID
 | 
			
		||||
 | 
			
		||||
	// if we have an attached file, or other info
 | 
			
		||||
	if msg.Extra != nil {
 | 
			
		||||
		if len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) != 0 {
 | 
			
		||||
			if msg.Text == "" {
 | 
			
		||||
				return brMsgIDs
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 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 configured
 | 
			
		||||
	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].GetBool("ShowJoinPart") {
 | 
			
		||||
		return brMsgIDs
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// only relay topic change when configured
 | 
			
		||||
	if msg.Event == config.EVENT_TOPIC_CHANGE && !gw.Bridges[dest.Account].GetBool("ShowTopicChange") {
 | 
			
		||||
		return brMsgIDs
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// broadcast to every out channel (irc QUIT)
 | 
			
		||||
	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE {
 | 
			
		||||
		log.Debug("empty channel")
 | 
			
		||||
		return
 | 
			
		||||
		flog.Debug("empty channel")
 | 
			
		||||
		return brMsgIDs
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	originchannel := msg.Channel
 | 
			
		||||
	origmsg := msg
 | 
			
		||||
	for _, channel := range gw.DestChannelFunc(&msg, *dest) {
 | 
			
		||||
		// do not send to ourself
 | 
			
		||||
		if channel.ID == getChannelID(origmsg) {
 | 
			
		||||
			continue
 | 
			
		||||
	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
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		log.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name)
 | 
			
		||||
		flog.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)
 | 
			
		||||
		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
 | 
			
		||||
		}
 | 
			
		||||
		err := dest.Send(msg)
 | 
			
		||||
		mID, err := dest.Send(msg)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println(err)
 | 
			
		||||
			flog.Error(err)
 | 
			
		||||
		}
 | 
			
		||||
		// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice
 | 
			
		||||
		if mID != "" {
 | 
			
		||||
			flog.Debugf("mID %s: %s", dest.Account, mID)
 | 
			
		||||
			brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID, channel.ID})
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return brMsgIDs
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
 | 
			
		||||
	if msg.Text == "" {
 | 
			
		||||
		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account)
 | 
			
		||||
	// if we don't have the bridge, ignore it
 | 
			
		||||
	if _, ok := gw.Bridges[msg.Account]; !ok {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) {
 | 
			
		||||
 | 
			
		||||
	// check if we need to ignore a empty message
 | 
			
		||||
	if msg.Text == "" {
 | 
			
		||||
		// we have an attachment or actual bytes, do not ignore
 | 
			
		||||
		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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// is the username in IgnoreNicks field
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks")) {
 | 
			
		||||
		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
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// does the message match regex in IgnoreMessages field
 | 
			
		||||
	// TODO do not compile regexps everytime
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreMessages) {
 | 
			
		||||
	for _, entry := range strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages")) {
 | 
			
		||||
		if entry != "" {
 | 
			
		||||
			re, err := regexp.Compile(entry)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				log.Errorf("incorrect regexp %s for %s", entry, msg.Account)
 | 
			
		||||
				flog.Errorf("incorrect regexp %s for %s", entry, msg.Account)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			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
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
@@ -260,13 +330,31 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
 | 
			
		||||
	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]
 | 
			
		||||
	msg.Protocol = br.Protocol
 | 
			
		||||
	nick := gw.Config.General.RemoteNickFormat
 | 
			
		||||
	if nick == "" {
 | 
			
		||||
		nick = dest.Config.RemoteNickFormat
 | 
			
		||||
	if gw.Config.General.StripNick || dest.GetBool("StripNick") {
 | 
			
		||||
		re := regexp.MustCompile("[^a-zA-Z0-9]+")
 | 
			
		||||
		msg.Username = re.ReplaceAllString(msg.Username, "")
 | 
			
		||||
	}
 | 
			
		||||
	nick := dest.GetString("RemoteNickFormat")
 | 
			
		||||
	if nick == "" {
 | 
			
		||||
		nick = gw.Config.General.RemoteNickFormat
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// loop to replace nicks
 | 
			
		||||
	for _, outer := range br.GetStringSlice2D("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 {
 | 
			
		||||
		// fix utf-8 issue #193
 | 
			
		||||
		i := 0
 | 
			
		||||
@@ -279,62 +367,101 @@ 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, "{NICK}", msg.Username, -1)
 | 
			
		||||
 | 
			
		||||
	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
 | 
			
		||||
	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
 | 
			
		||||
	msg.Username = nick
 | 
			
		||||
	nick = strings.Replace(nick, "{LABEL}", br.GetString("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
 | 
			
		||||
	if iconurl == "" {
 | 
			
		||||
		iconurl = dest.Config.IconURL
 | 
			
		||||
		iconurl = dest.GetString("IconURL")
 | 
			
		||||
	}
 | 
			
		||||
	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1)
 | 
			
		||||
	if msg.Avatar == "" {
 | 
			
		||||
		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.GetStringSlice2D("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) {
 | 
			
		||||
	reg := regexp.MustCompile("[^a-zA-Z0-9]+")
 | 
			
		||||
	// if we don't have a attachfield or we don't have a mediaserver configured return
 | 
			
		||||
	if msg.Extra == nil || gw.Config.General.MediaServerUpload == "" {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// if we actually have files, start uploading them to the mediaserver
 | 
			
		||||
	if len(msg.Extra["file"]) > 0 {
 | 
			
		||||
		client := &http.Client{
 | 
			
		||||
			Timeout: time.Second * 5,
 | 
			
		||||
		}
 | 
			
		||||
		for i, f := range msg.Extra["file"] {
 | 
			
		||||
			fi := f.(config.FileInfo)
 | 
			
		||||
			ext := filepath.Ext(fi.Name)
 | 
			
		||||
			fi.Name = fi.Name[0 : len(fi.Name)-len(ext)]
 | 
			
		||||
			fi.Name = reg.ReplaceAllString(fi.Name, "_")
 | 
			
		||||
			fi.Name = fi.Name + ext
 | 
			
		||||
			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, err := http.NewRequest("PUT", url, reader)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				flog.Errorf("mediaserver upload failed: %#v", err)
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			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 (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool {
 | 
			
		||||
	return msg.Gateway == gw.Name
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getChannelID(msg config.Message) string {
 | 
			
		||||
	return msg.Channel + msg.Account
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool {
 | 
			
		||||
	GIDmap := gw.Channels[getChannelID(*msg)].GID
 | 
			
		||||
 | 
			
		||||
	// 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 {
 | 
			
		||||
	if strings.HasPrefix(account, "api.") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
	return strings.HasPrefix(account, "api.")
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										278
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										278
									
								
								gateway/gateway_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,278 @@
 | 
			
		||||
package gateway
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/stretchr/testify/assert"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"testing"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var testconfig = []byte(`
 | 
			
		||||
[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 = []byte(`
 | 
			
		||||
[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 = []byte(`
 | 
			
		||||
[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 []byte) *Router {
 | 
			
		||||
	cfg := config.NewConfigFromString(input)
 | 
			
		||||
	r, err := NewRouter(cfg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println(err)
 | 
			
		||||
	}
 | 
			
		||||
	return r
 | 
			
		||||
}
 | 
			
		||||
func TestNewRouter(t *testing.T) {
 | 
			
		||||
	r := maketestRouter(testconfig)
 | 
			
		||||
	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)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										110
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										110
									
								
								gateway/router.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,110 @@
 | 
			
		||||
package gateway
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/gateway/samechannel"
 | 
			
		||||
	//	"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{Message: make(chan config.Message), Gateways: make(map[string]*Gateway), Config: cfg}
 | 
			
		||||
	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.
 | 
			
		||||
func (c *Client) Receive() Message {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case msg := <-c.In:
 | 
			
		||||
			return msg
 | 
			
		||||
		}
 | 
			
		||||
	var msg Message
 | 
			
		||||
	for msg = range c.In {
 | 
			
		||||
		return msg
 | 
			
		||||
	}
 | 
			
		||||
	return msg
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										
											BIN
										
									
								
								img/matterbridge.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								img/matterbridge.gif
									
									
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							| 
		 After Width: | Height: | Size: 77 KiB  | 
@@ -5,22 +5,21 @@ import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/42wim/matterbridge/bridge/config"
 | 
			
		||||
	"github.com/42wim/matterbridge/gateway"
 | 
			
		||||
	"github.com/42wim/matterbridge/gateway/samechannel"
 | 
			
		||||
	log "github.com/Sirupsen/logrus"
 | 
			
		||||
	"github.com/google/gops/agent"
 | 
			
		||||
	log "github.com/sirupsen/logrus"
 | 
			
		||||
	prefixed "github.com/x-cray/logrus-prefixed-formatter"
 | 
			
		||||
	"os"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	version = "0.16.0-rc2"
 | 
			
		||||
	version = "1.10.1"
 | 
			
		||||
	githash string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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")
 | 
			
		||||
	flagDebug := flag.Bool("debug", false, "enable debug")
 | 
			
		||||
	flagVersion := flag.Bool("version", false, "show version")
 | 
			
		||||
@@ -34,32 +33,25 @@ func main() {
 | 
			
		||||
		fmt.Printf("version: %s %s\n", version, githash)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if *flagDebug {
 | 
			
		||||
		log.Info("Enabling debug")
 | 
			
		||||
	if *flagDebug || os.Getenv("DEBUG") == "1" {
 | 
			
		||||
		log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true})
 | 
			
		||||
		flog.Info("Enabling debug")
 | 
			
		||||
		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") {
 | 
			
		||||
		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)
 | 
			
		||||
 | 
			
		||||
	g := gateway.New(cfg)
 | 
			
		||||
	sgw := samechannelgateway.New(cfg)
 | 
			
		||||
	gwconfigs := sgw.GetConfig()
 | 
			
		||||
	for _, gw := range append(gwconfigs, cfg.Gateway...) {
 | 
			
		||||
		if !gw.Enable {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		err := g.AddConfig(&gw)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Fatalf("Starting gateway failed: %s", err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	err := g.Start()
 | 
			
		||||
	cfg.General.Debug = *flagDebug
 | 
			
		||||
	r, err := gateway.NewRouter(cfg)
 | 
			
		||||
	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")
 | 
			
		||||
	err = r.Start()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		flog.Fatalf("Starting gateway failed: %s", err)
 | 
			
		||||
	}
 | 
			
		||||
	flog.Printf("Gateway(s) started succesfully. Now relaying messages")
 | 
			
		||||
	select {}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -32,16 +32,41 @@ UseSASL=false
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
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. 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="matterbot"
 | 
			
		||||
 | 
			
		||||
#If you registered your bot with a service like Nickserv on freenode. 
 | 
			
		||||
#Also being used when UseSASL=true
 | 
			
		||||
#
 | 
			
		||||
#Note: if you want do to quakenet auth, set NickServNick="Q@CServe.quakenet.org"
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
NickServNick="nickserv"
 | 
			
		||||
NickServPassword="secret"
 | 
			
		||||
 | 
			
		||||
#OPTIONAL only used for quakenet auth
 | 
			
		||||
NickServUsername="username"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Flood control
 | 
			
		||||
#Delay in milliseconds between each message send to the IRC server
 | 
			
		||||
#OPTIONAL (default 1300)
 | 
			
		||||
@@ -58,6 +83,19 @@ MessageQueue=30
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
 | 
			
		||||
#ColorNicks will show each nickname in a different color.
 | 
			
		||||
#Only works in IRC right now.
 | 
			
		||||
ColorNicks=false
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
@@ -69,19 +107,56 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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 "{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)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#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)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
#Do not send joins/parts to other bridges
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoSendJoinPart=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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -116,6 +191,9 @@ Nick="xmppbot"
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
@@ -127,18 +205,49 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
@@ -166,6 +275,9 @@ Muc="conf.hipchat.com"
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Nick="yourlogin"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
@@ -177,18 +289,49 @@ IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
@@ -213,6 +356,11 @@ Team="yourteam"
 | 
			
		||||
Login="yourlogin"
 | 
			
		||||
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. 
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoTLS=false
 | 
			
		||||
@@ -247,6 +395,9 @@ IconURL="http://youricon.png"
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#how to format the list of IRC nicks when displayed in mattermost. 
 | 
			
		||||
#Possible options are "table" and "plain"
 | 
			
		||||
#OPTIONAL (default plain)
 | 
			
		||||
@@ -282,18 +433,55 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
#Do not send joins/parts to other bridges
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoSendJoinPart=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
 | 
			
		||||
#Best to make a dedicated gitter account for the bot.
 | 
			
		||||
@@ -310,6 +498,9 @@ ShowJoinPart=false
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
@@ -321,18 +512,50 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -361,7 +584,6 @@ WebhookURL="https://hooks.slack.com/services/yourhook"
 | 
			
		||||
#AND DEDICATED BOT USER WHEN POSSIBLE!
 | 
			
		||||
#Address to listen on for outgoing webhook requests from slack
 | 
			
		||||
#See account settings - integrations - outgoing webhooks on slack
 | 
			
		||||
#This setting will not be used when useAPI is eanbled
 | 
			
		||||
#webhooks
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
WebhookBindAddress="0.0.0.0:9999"
 | 
			
		||||
@@ -369,10 +591,14 @@ WebhookBindAddress="0.0.0.0:9999"
 | 
			
		||||
#Icon that will be showed in slack
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{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
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IconURL="https://robohash.org/{NICK}.png?size=48x48"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#how to format the list of IRC nicks when displayed in slack
 | 
			
		||||
#Possible options are "table" and "plain"
 | 
			
		||||
#OPTIONAL (default plain)
 | 
			
		||||
@@ -408,18 +634,55 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
ShowJoinPart=false
 | 
			
		||||
 | 
			
		||||
#Do not send joins/parts to other bridges
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoSendJoinPart=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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -439,11 +702,19 @@ Token="Yourtokenhere"
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Server="yourservername"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Shows title, description and URL of embedded messages (sent by other bots)
 | 
			
		||||
#OPTIONAL (default 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.
 | 
			
		||||
#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)
 | 
			
		||||
WebhookURL="Yourwebhooktokenhere"
 | 
			
		||||
 | 
			
		||||
@@ -466,18 +737,50 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -492,6 +795,9 @@ ShowJoinPart=false
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
#Only supported format is "HTML", messages will be sent in html parsemode.
 | 
			
		||||
#See https://core.telegram.org/bots/api#html-style
 | 
			
		||||
@@ -509,6 +815,14 @@ UseFirstName=false
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
UseInsecureURL=false
 | 
			
		||||
 | 
			
		||||
#Disable quoted/reply messages
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
QuoteDisable=false
 | 
			
		||||
 | 
			
		||||
#Format quoted/reply messages
 | 
			
		||||
#OPTIONAL (default "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})")
 | 
			
		||||
QuoteFormat="{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})"
 | 
			
		||||
 | 
			
		||||
#Disable sending of edits to other bridges
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
EditDisable=false
 | 
			
		||||
@@ -528,18 +842,54 @@ IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#
 | 
			
		||||
#WARNING: if you have set MessageFormat="HTML" be sure that this format matches the guidelines
 | 
			
		||||
#on https://core.telegram.org/bots/api#html-style otherwise the message will not go through to
 | 
			
		||||
#telegram! eg <{NICK}> should be <{NICK}>
 | 
			
		||||
#
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
@@ -574,6 +924,9 @@ NoTLS=false
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
SkipTLSVerify=true
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to rocketchat with the sender's nick. 
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#rocketchat server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
@@ -592,18 +945,50 @@ IgnoreNicks="ircspammer1 ircspammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -629,6 +1014,9 @@ Password="yourpass"
 | 
			
		||||
#OPTIONAL (default false)
 | 
			
		||||
NoHomeServerSuffix=false
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to matrix with the sender's nick. 
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#matrix server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
@@ -647,18 +1035,50 @@ IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -678,6 +1098,9 @@ Password="yourpass"
 | 
			
		||||
#OPTIONAL 
 | 
			
		||||
Authcode="ABCE12"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Whether to prefix messages from other bridges to matrix with the sender's nick. 
 | 
			
		||||
#Useful if username overrides for incoming webhooks isn't enabled on the 
 | 
			
		||||
#matrix server. If you set PrefixMessagesWithNick to true, each message 
 | 
			
		||||
@@ -696,18 +1119,133 @@ IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Only works hiding/show messages from irc and mattermost bridge for now
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
 | 
			
		||||
###################################################################
 | 
			
		||||
#zulip section
 | 
			
		||||
###################################################################
 | 
			
		||||
[zulip]
 | 
			
		||||
#You can configure multiple servers "[zulip.name]" or "[zulip.name2]"
 | 
			
		||||
#In this example we use [zulip.streamchat]
 | 
			
		||||
#REQUIRED
 | 
			
		||||
 | 
			
		||||
[zulip.streamchat]
 | 
			
		||||
#Token to connect with zulip API (called bot API key in Settings - Your bots)
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Token="Yourtokenhere"
 | 
			
		||||
 | 
			
		||||
#Username of the bot, normally called yourbot-bot@yourserver.zulipchat.com 
 | 
			
		||||
#See username in Settings - Your bots 
 | 
			
		||||
#REQUIRED
 | 
			
		||||
Login="yourbot-bot@yourserver.zulipchat.com"
 | 
			
		||||
 | 
			
		||||
#Servername of your zulip instance
 | 
			
		||||
#REQUIRED 
 | 
			
		||||
Server="https://yourserver.zulipchat.com"
 | 
			
		||||
 | 
			
		||||
#Topic of the messages matterbridge will use
 | 
			
		||||
#OPTIONAL (default "matterbridge")
 | 
			
		||||
Topic="matterbridge"
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#Nicks you want to ignore. 
 | 
			
		||||
#Messages from those users will not be sent to other bridges.
 | 
			
		||||
#OPTIONAL
 | 
			
		||||
IgnoreNicks="spammer1 spammer2"
 | 
			
		||||
 | 
			
		||||
#Messages you want to ignore. 
 | 
			
		||||
#Messages matching these regexp will be ignored and not sent to other bridges
 | 
			
		||||
#See https://regex-golang.appspot.com/assets/html/index.html for more regex info
 | 
			
		||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing 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 
 | 
			
		||||
#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 "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
 | 
			
		||||
 | 
			
		||||
#Enable to show users joins/parts from other bridges 
 | 
			
		||||
#Currently works for messages from the following bridges: irc, mattermost, slack
 | 
			
		||||
#OPTIONAL (default 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
 | 
			
		||||
@@ -730,9 +1268,14 @@ Buffer=1000
 | 
			
		||||
#OPTIONAL (no authorization if token is empty)
 | 
			
		||||
Token="mytoken"
 | 
			
		||||
 | 
			
		||||
#extra label that can be used in the RemoteNickFormat
 | 
			
		||||
#optional (default empty)
 | 
			
		||||
Label=""
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
RemoteNickFormat="{NICK}"
 | 
			
		||||
@@ -742,15 +1285,49 @@ RemoteNickFormat="{NICK}"
 | 
			
		||||
###################################################################
 | 
			
		||||
#General configuration
 | 
			
		||||
###################################################################
 | 
			
		||||
#Settings here override specific settings for each protocol
 | 
			
		||||
# Settings here are defaults that each protocol can override
 | 
			
		||||
[general]
 | 
			
		||||
 | 
			
		||||
## RELOADABLE SETTINGS
 | 
			
		||||
## Settings below can be reloaded by editing the file
 | 
			
		||||
 | 
			
		||||
#RemoteNickFormat defines how remote users appear on this bridge 
 | 
			
		||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
 | 
			
		||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
 | 
			
		||||
#The string "{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
 | 
			
		||||
#OPTIONAL (default empty)
 | 
			
		||||
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
 | 
			
		||||
###################################################################
 | 
			
		||||
@@ -786,7 +1363,7 @@ enable=true
 | 
			
		||||
    #mattermost - channel (the channel name as seen in the URL, not the displayname)
 | 
			
		||||
    #gitter     - username/room 
 | 
			
		||||
    #xmpp       - channel
 | 
			
		||||
    #slack      - channel (the channel name as seen in the URL, not the displayname)
 | 
			
		||||
    #slack      - channel (without the #)
 | 
			
		||||
    #discord    - channel (without the #)
 | 
			
		||||
    #           - ID:123456789 (where 123456789 is the channel ID) 
 | 
			
		||||
    #               (https://github.com/42wim/matterbridge/issues/57)
 | 
			
		||||
@@ -798,6 +1375,7 @@ enable=true
 | 
			
		||||
    #           - encrypted rooms are not supported in matrix
 | 
			
		||||
    #steam      - chatid (a large number). 
 | 
			
		||||
    #             The number in the URL when you click "enter chat room" in the browser
 | 
			
		||||
    #zulip      - stream (without the #)
 | 
			
		||||
    #                  
 | 
			
		||||
    #REQUIRED
 | 
			
		||||
    channel="#testing"
 | 
			
		||||
@@ -829,6 +1407,14 @@ enable=true
 | 
			
		||||
        #OPTIONAL - your irc channel key
 | 
			
		||||
        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
 | 
			
		||||
    #[[gateway.inout]]
 | 
			
		||||
    #account="api.local"
 | 
			
		||||
 
 | 
			
		||||
@@ -6,7 +6,6 @@
 | 
			
		||||
 | 
			
		||||
[mattermost]
 | 
			
		||||
    [mattermost.work]
 | 
			
		||||
    useAPI=true
 | 
			
		||||
    #do not prefix it wit http:// or https://
 | 
			
		||||
    Server="yourmattermostserver.domain" 
 | 
			
		||||
    Team="yourteam"
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,7 @@
 | 
			
		||||
package matterclient
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"crypto/md5"
 | 
			
		||||
	"crypto/tls"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
@@ -8,14 +9,15 @@ import (
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/cookiejar"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"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/hashicorp/golang-lru"
 | 
			
		||||
	"github.com/jpillora/backoff"
 | 
			
		||||
	"github.com/mattermost/platform/model"
 | 
			
		||||
)
 | 
			
		||||
@@ -43,8 +45,8 @@ type Message struct {
 | 
			
		||||
type Team struct {
 | 
			
		||||
	Team         *model.Team
 | 
			
		||||
	Id           string
 | 
			
		||||
	Channels     *model.ChannelList
 | 
			
		||||
	MoreChannels *model.ChannelList
 | 
			
		||||
	Channels     []*model.Channel
 | 
			
		||||
	MoreChannels []*model.Channel
 | 
			
		||||
	Users        map[string]*model.User
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -53,7 +55,7 @@ type MMClient struct {
 | 
			
		||||
	*Credentials
 | 
			
		||||
	Team          *Team
 | 
			
		||||
	OtherTeams    []*Team
 | 
			
		||||
	Client        *model.Client
 | 
			
		||||
	Client        *model.Client4
 | 
			
		||||
	User          *model.User
 | 
			
		||||
	Users         map[string]*model.User
 | 
			
		||||
	MessageChan   chan *Message
 | 
			
		||||
@@ -66,16 +68,22 @@ type MMClient struct {
 | 
			
		||||
	WsPingChan    chan *model.WebSocketResponse
 | 
			
		||||
	ServerVersion string
 | 
			
		||||
	OnWsConnect   func()
 | 
			
		||||
	lruCache      *lru.Cache
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(login, pass, team, server string) *MMClient {
 | 
			
		||||
	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server}
 | 
			
		||||
	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)}
 | 
			
		||||
	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"})
 | 
			
		||||
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
 | 
			
		||||
	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true})
 | 
			
		||||
	mmclient.log = log.WithFields(log.Fields{"prefix": "matterclient"})
 | 
			
		||||
	mmclient.lruCache, _ = lru.New(500)
 | 
			
		||||
	return mmclient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SetDebugLog() {
 | 
			
		||||
	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SetLogLevel(level string) {
 | 
			
		||||
	l, err := log.ParseLevel(level)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -88,7 +96,7 @@ func (m *MMClient) SetLogLevel(level string) {
 | 
			
		||||
func (m *MMClient) Login() error {
 | 
			
		||||
	// check if this is a first connect or a reconnection
 | 
			
		||||
	firstConnection := true
 | 
			
		||||
	if m.WsConnected == true {
 | 
			
		||||
	if m.WsConnected {
 | 
			
		||||
		firstConnection = false
 | 
			
		||||
	}
 | 
			
		||||
	m.WsConnected = false
 | 
			
		||||
@@ -105,18 +113,21 @@ func (m *MMClient) Login() error {
 | 
			
		||||
		uriScheme = "http://"
 | 
			
		||||
	}
 | 
			
		||||
	// 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.Timeout = time.Second * 10
 | 
			
		||||
 | 
			
		||||
	for {
 | 
			
		||||
		d := b.Duration()
 | 
			
		||||
		// bogus call to get the serverversion
 | 
			
		||||
		m.Client.GetClientProperties()
 | 
			
		||||
		if firstConnection && !supportedVersion(m.Client.ServerVersion) {
 | 
			
		||||
			return fmt.Errorf("unsupported mattermost version: %s", m.Client.ServerVersion)
 | 
			
		||||
		_, resp := m.Client.Logout()
 | 
			
		||||
		if resp.Error != nil {
 | 
			
		||||
			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 == "" {
 | 
			
		||||
			m.log.Debugf("Server not up yet, reconnecting in %s", d)
 | 
			
		||||
			time.Sleep(d)
 | 
			
		||||
@@ -127,30 +138,33 @@ func (m *MMClient) Login() error {
 | 
			
		||||
	}
 | 
			
		||||
	b.Reset()
 | 
			
		||||
 | 
			
		||||
	var myinfo *model.Result
 | 
			
		||||
	var resp *model.Response
 | 
			
		||||
	//var myinfo *model.Result
 | 
			
		||||
	var appErr *model.AppError
 | 
			
		||||
	var logmsg = "trying login"
 | 
			
		||||
	for {
 | 
			
		||||
		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server)
 | 
			
		||||
		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) {
 | 
			
		||||
			m.log.Debugf(logmsg+" with %s", model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			m.log.Debugf(logmsg + " with token")
 | 
			
		||||
			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=")
 | 
			
		||||
			if len(token) != 2 {
 | 
			
		||||
				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken")
 | 
			
		||||
			}
 | 
			
		||||
			m.Client.HttpClient.Jar = m.createCookieJar(token[1])
 | 
			
		||||
			m.Client.MockSession(token[1])
 | 
			
		||||
			myinfo, appErr = m.Client.GetMe("")
 | 
			
		||||
			if appErr != nil {
 | 
			
		||||
				return errors.New(appErr.DetailedError)
 | 
			
		||||
			m.Client.AuthToken = token[1]
 | 
			
		||||
			m.Client.AuthType = model.HEADER_BEARER
 | 
			
		||||
			m.User, resp = m.Client.GetMe("")
 | 
			
		||||
			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)
 | 
			
		||||
				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN)
 | 
			
		||||
			}
 | 
			
		||||
		} 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 {
 | 
			
		||||
			d := b.Duration()
 | 
			
		||||
			m.log.Debug(appErr.DetailedError)
 | 
			
		||||
@@ -176,10 +190,12 @@ func (m *MMClient) Login() error {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if m.Team == nil {
 | 
			
		||||
		return errors.New("team not found")
 | 
			
		||||
		validTeamNames := make([]string, len(m.OtherTeams))
 | 
			
		||||
		for i, t := range m.OtherTeams {
 | 
			
		||||
			validTeamNames[i] = t.Team.Name
 | 
			
		||||
		}
 | 
			
		||||
		return fmt.Errorf("Team '%s' not found in %v", m.Credentials.Team, validTeamNames)
 | 
			
		||||
	}
 | 
			
		||||
	// set our team id as default route
 | 
			
		||||
	m.Client.SetTeamId(m.Team.Id)
 | 
			
		||||
 | 
			
		||||
	m.wsConnect()
 | 
			
		||||
 | 
			
		||||
@@ -200,7 +216,7 @@ func (m *MMClient) wsConnect() {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 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.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken)
 | 
			
		||||
 | 
			
		||||
@@ -234,9 +250,9 @@ func (m *MMClient) Logout() error {
 | 
			
		||||
		m.log.Debug("Not invalidating session in logout, credential is a token")
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	_, err := m.Client.Logout()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	_, resp := m.Client.Logout()
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -267,7 +283,17 @@ func (m *MMClient) WsReceiver() {
 | 
			
		||||
			m.log.Debugf("WsReceiver event: %#v", event)
 | 
			
		||||
			msg := &Message{Raw: &event, Team: m.Credentials.Team}
 | 
			
		||||
			m.parseMessage(msg)
 | 
			
		||||
			m.MessageChan <- msg
 | 
			
		||||
			// check if we didn't empty the message
 | 
			
		||||
			if msg.Text != "" {
 | 
			
		||||
				m.MessageChan <- msg
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			// if we have file attached but the message is empty, also send it
 | 
			
		||||
			if msg.Post != nil {
 | 
			
		||||
				if msg.Text != "" || len(msg.Post.FileIds) > 0 || msg.Post.Type == "slack_attachment" {
 | 
			
		||||
					m.MessageChan <- msg
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
@@ -282,7 +308,7 @@ func (m *MMClient) WsReceiver() {
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) parseMessage(rmsg *Message) {
 | 
			
		||||
	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)
 | 
			
		||||
		/*
 | 
			
		||||
			case model.ACTION_USER_REMOVED:
 | 
			
		||||
@@ -303,6 +329,13 @@ func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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)))
 | 
			
		||||
	// we don't have the user, refresh the userlist
 | 
			
		||||
	if m.GetUser(data.UserId) == nil {
 | 
			
		||||
@@ -329,37 +362,37 @@ func (m *MMClient) parseActionPost(rmsg *Message) {
 | 
			
		||||
	}
 | 
			
		||||
	rmsg.Text = data.Message
 | 
			
		||||
	rmsg.Post = data
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateUsers() error {
 | 
			
		||||
	mmusers, err := m.Client.GetProfiles(0, 50000, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	mmusers, resp := m.Client.GetUsers(0, 50000, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return errors.New(resp.Error.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Users = mmusers.Data.(map[string]*model.User)
 | 
			
		||||
	for _, user := range mmusers {
 | 
			
		||||
		m.Users[user.Id] = user
 | 
			
		||||
	}
 | 
			
		||||
	m.Unlock()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannels() error {
 | 
			
		||||
	mmchannels, err := m.Client.GetChannels("")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	var mmchannels2 *model.Result
 | 
			
		||||
	if m.mmVersion() >= 3.08 {
 | 
			
		||||
		mmchannels2, err = m.Client.GetMoreChannelsPage(0, 5000)
 | 
			
		||||
	} else {
 | 
			
		||||
		mmchannels2, err = m.Client.GetMoreChannels("")
 | 
			
		||||
	}
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New(err.DetailedError)
 | 
			
		||||
	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return errors.New(resp.Error.DetailedError)
 | 
			
		||||
	}
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Team.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	m.Team.MoreChannels = mmchannels2.Data.(*model.ChannelList)
 | 
			
		||||
	m.Team.Channels = mmchannels
 | 
			
		||||
	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()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -372,14 +405,14 @@ func (m *MMClient) GetChannelName(channelId string) string {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		if t.Channels != nil {
 | 
			
		||||
			for _, channel := range *t.Channels {
 | 
			
		||||
			for _, channel := range t.Channels {
 | 
			
		||||
				if channel.Id == channelId {
 | 
			
		||||
					return channel.Name
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		if t.MoreChannels != nil {
 | 
			
		||||
			for _, channel := range *t.MoreChannels {
 | 
			
		||||
			for _, channel := range t.MoreChannels {
 | 
			
		||||
				if channel.Id == channelId {
 | 
			
		||||
					return channel.Name
 | 
			
		||||
				}
 | 
			
		||||
@@ -397,7 +430,7 @@ func (m *MMClient) GetChannelId(name string, teamId string) string {
 | 
			
		||||
	}
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		if t.Id == teamId {
 | 
			
		||||
			for _, channel := range append(*t.Channels, *t.MoreChannels...) {
 | 
			
		||||
			for _, channel := range append(t.Channels, t.MoreChannels...) {
 | 
			
		||||
				if channel.Name == name {
 | 
			
		||||
					return channel.Id
 | 
			
		||||
				}
 | 
			
		||||
@@ -411,7 +444,7 @@ func (m *MMClient) GetChannelTeamId(id string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	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 {
 | 
			
		||||
				return channel.TeamId
 | 
			
		||||
			}
 | 
			
		||||
@@ -424,7 +457,7 @@ func (m *MMClient) GetChannelHeader(channelId string) string {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		for _, channel := range append(*t.Channels, *t.MoreChannels...) {
 | 
			
		||||
		for _, channel := range append(t.Channels, t.MoreChannels...) {
 | 
			
		||||
			if channel.Id == channelId {
 | 
			
		||||
				return channel.Header
 | 
			
		||||
			}
 | 
			
		||||
@@ -434,55 +467,85 @@ func (m *MMClient) GetChannelHeader(channelId string) string {
 | 
			
		||||
	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}
 | 
			
		||||
	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 {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	for _, c := range *m.Team.Channels {
 | 
			
		||||
	for _, c := range m.Team.Channels {
 | 
			
		||||
		if c.Id == channelId {
 | 
			
		||||
			m.log.Debug("Not joining ", channelId, " already joined.")
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	m.log.Debug("Joining ", channelId)
 | 
			
		||||
	_, err := m.Client.JoinChannel(channelId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return errors.New("failed to join")
 | 
			
		||||
	_, resp := m.Client.AddChannelMember(channelId, m.User.Id)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPostsSince(channelId, time)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.GetPostsSince(channelId, time)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
	return res
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) SearchPosts(query string) *model.PostList {
 | 
			
		||||
	res, err := m.Client.SearchPosts(query, false)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.SearchPosts(m.Team.Id, query, false)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
	return res
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList {
 | 
			
		||||
	res, err := m.Client.GetPosts(channelId, 0, limit, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	return res.Data.(*model.PostList)
 | 
			
		||||
	return res
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetPublicLink(filename string) string {
 | 
			
		||||
	res, err := m.Client.GetPublicLink(filename)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.GetFileLink(filename)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	return res
 | 
			
		||||
@@ -491,8 +554,27 @@ func (m *MMClient) GetPublicLink(filename string) string {
 | 
			
		||||
func (m *MMClient) GetPublicLinks(filenames []string) []string {
 | 
			
		||||
	var output []string
 | 
			
		||||
	for _, f := range filenames {
 | 
			
		||||
		res, err := m.Client.GetPublicLink(f)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
		res, resp := m.Client.GetFileLink(f)
 | 
			
		||||
		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_V4+"/files/"+f)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		output = append(output, res)
 | 
			
		||||
@@ -501,42 +583,43 @@ func (m *MMClient) GetPublicLinks(filenames []string) []string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateChannelHeader(channelId string, header string) {
 | 
			
		||||
	data := make(map[string]string)
 | 
			
		||||
	data["channel_id"] = channelId
 | 
			
		||||
	data["channel_header"] = header
 | 
			
		||||
	channel := &model.Channel{Id: channelId, Header: header}
 | 
			
		||||
	m.log.Debugf("updating channelheader %#v, %#v", channelId, header)
 | 
			
		||||
	_, err := m.Client.UpdateChannelHeader(data)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Error(err)
 | 
			
		||||
	_, resp := m.Client.UpdateChannel(channel)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		log.Error(resp.Error)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateLastViewed(channelId string) {
 | 
			
		||||
	m.log.Debugf("posting lastview %#v", channelId)
 | 
			
		||||
	if m.mmVersion() >= 3.08 {
 | 
			
		||||
		view := model.ChannelView{ChannelId: channelId}
 | 
			
		||||
		res, _ := m.Client.ViewChannel(view)
 | 
			
		||||
		if res == false {
 | 
			
		||||
			m.log.Errorf("ChannelView update for %s failed", channelId)
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	_, err := m.Client.UpdateLastViewedAt(channelId, true)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Error(err)
 | 
			
		||||
	view := &model.ChannelView{ChannelId: channelId}
 | 
			
		||||
	_, resp := m.Client.ViewChannel(m.User.Id, view)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		m.log.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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 {
 | 
			
		||||
	res, err := m.Client.GetProfilesInChannel(channelId, 0, 50000, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, err)
 | 
			
		||||
	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error)
 | 
			
		||||
		return []string{}
 | 
			
		||||
	}
 | 
			
		||||
	members := res.Data.(map[string]*model.User)
 | 
			
		||||
	allusers := m.GetUsers()
 | 
			
		||||
	result := []string{}
 | 
			
		||||
	for _, member := range members {
 | 
			
		||||
		result = append(result, member.Nickname)
 | 
			
		||||
	for _, member := range *res {
 | 
			
		||||
		result = append(result, allusers[member.UserId].Nickname)
 | 
			
		||||
	}
 | 
			
		||||
	return result
 | 
			
		||||
}
 | 
			
		||||
@@ -560,22 +643,15 @@ func (m *MMClient) createCookieJar(token string) *cookiejar.Jar {
 | 
			
		||||
func (m *MMClient) SendDirectMessage(toUserId string, msg string) {
 | 
			
		||||
	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg)
 | 
			
		||||
	// create DM channel (only happens on first message)
 | 
			
		||||
	_, err := m.Client.CreateDirectChannel(toUserId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err)
 | 
			
		||||
	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	channelName := model.GetDMNameFromIds(toUserId, m.User.Id)
 | 
			
		||||
 | 
			
		||||
	// update our channels
 | 
			
		||||
	mmchannels, err := m.Client.GetChannels("")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		m.log.Debug("SendDirectMessage: Couldn't update channels")
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	m.Team.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
	m.Unlock()
 | 
			
		||||
	m.UpdateChannels()
 | 
			
		||||
 | 
			
		||||
	// build & send the message
 | 
			
		||||
	msg = strings.Replace(msg, "\r", "", -1)
 | 
			
		||||
@@ -601,10 +677,10 @@ func (m *MMClient) GetChannels() []*model.Channel {
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	// our primary team channels first
 | 
			
		||||
	channels = append(channels, *m.Team.Channels...)
 | 
			
		||||
	channels = append(channels, m.Team.Channels...)
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		if t.Id != m.Team.Id {
 | 
			
		||||
			channels = append(channels, *t.Channels...)
 | 
			
		||||
			channels = append(channels, t.Channels...)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return channels
 | 
			
		||||
@@ -616,7 +692,7 @@ func (m *MMClient) GetMoreChannels() []*model.Channel {
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		channels = append(channels, *t.MoreChannels...)
 | 
			
		||||
		channels = append(channels, t.MoreChannels...)
 | 
			
		||||
	}
 | 
			
		||||
	return channels
 | 
			
		||||
}
 | 
			
		||||
@@ -627,9 +703,9 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string {
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	var channels []*model.Channel
 | 
			
		||||
	for _, t := range m.OtherTeams {
 | 
			
		||||
		channels = append(channels, *t.Channels...)
 | 
			
		||||
		channels = append(channels, t.Channels...)
 | 
			
		||||
		if t.MoreChannels != nil {
 | 
			
		||||
			channels = append(channels, *t.MoreChannels...)
 | 
			
		||||
			channels = append(channels, t.MoreChannels...)
 | 
			
		||||
		}
 | 
			
		||||
		for _, c := range channels {
 | 
			
		||||
			if c.Id == channelId {
 | 
			
		||||
@@ -643,12 +719,11 @@ func (m *MMClient) GetTeamFromChannel(channelId string) string {
 | 
			
		||||
func (m *MMClient) GetLastViewedAt(channelId string) int64 {
 | 
			
		||||
	m.RLock()
 | 
			
		||||
	defer m.RUnlock()
 | 
			
		||||
	res, err := m.Client.GetChannel(channelId, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return model.GetMillis()
 | 
			
		||||
	}
 | 
			
		||||
	data := res.Data.(*model.ChannelData)
 | 
			
		||||
	return data.Member.LastViewedAt
 | 
			
		||||
	return res.LastViewedAt
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetUsers() map[string]*model.User {
 | 
			
		||||
@@ -664,14 +739,13 @@ func (m *MMClient) GetUsers() map[string]*model.User {
 | 
			
		||||
func (m *MMClient) GetUser(userId string) *model.User {
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	defer m.Unlock()
 | 
			
		||||
	u, ok := m.Users[userId]
 | 
			
		||||
	_, ok := m.Users[userId]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		res, err := m.Client.GetProfilesByIds([]string{userId})
 | 
			
		||||
		if err != nil {
 | 
			
		||||
		res, resp := m.Client.GetUser(userId, "")
 | 
			
		||||
		if resp.Error != nil {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		u = res.Data.(map[string]*model.User)[userId]
 | 
			
		||||
		m.Users[userId] = u
 | 
			
		||||
		m.Users[userId] = res
 | 
			
		||||
	}
 | 
			
		||||
	return m.Users[userId]
 | 
			
		||||
}
 | 
			
		||||
@@ -685,36 +759,44 @@ func (m *MMClient) GetUserName(userId string) string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetStatus(userId string) string {
 | 
			
		||||
	res, err := m.Client.GetStatusesByIds([]string{userId})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	res, resp := m.Client.GetUserStatus(userId, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return ""
 | 
			
		||||
	}
 | 
			
		||||
	status := res.Data.(map[string]string)
 | 
			
		||||
	if status[userId] == model.STATUS_AWAY {
 | 
			
		||||
	if res.Status == model.STATUS_AWAY {
 | 
			
		||||
		return "away"
 | 
			
		||||
	}
 | 
			
		||||
	if status[userId] == model.STATUS_ONLINE {
 | 
			
		||||
	if res.Status == model.STATUS_ONLINE {
 | 
			
		||||
		return "online"
 | 
			
		||||
	}
 | 
			
		||||
	return "offline"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) UpdateStatus(userId string, status string) error {
 | 
			
		||||
	_, resp := m.Client.UpdateUserStatus(userId, &model.Status{Status: status})
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MMClient) GetStatuses() map[string]string {
 | 
			
		||||
	var ok bool
 | 
			
		||||
	var ids []string
 | 
			
		||||
	statuses := make(map[string]string)
 | 
			
		||||
	res, err := m.Client.GetStatuses()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
	for id := range m.Users {
 | 
			
		||||
		ids = append(ids, id)
 | 
			
		||||
	}
 | 
			
		||||
	res, resp := m.Client.GetUsersStatusesByIds(ids)
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return statuses
 | 
			
		||||
	}
 | 
			
		||||
	if statuses, ok = res.Data.(map[string]string); ok {
 | 
			
		||||
		for userId, status := range statuses {
 | 
			
		||||
			statuses[userId] = "offline"
 | 
			
		||||
			if status == model.STATUS_AWAY {
 | 
			
		||||
				statuses[userId] = "away"
 | 
			
		||||
			}
 | 
			
		||||
			if status == model.STATUS_ONLINE {
 | 
			
		||||
				statuses[userId] = "online"
 | 
			
		||||
			}
 | 
			
		||||
	for _, status := range res {
 | 
			
		||||
		statuses[status.UserId] = "offline"
 | 
			
		||||
		if status.Status == model.STATUS_AWAY {
 | 
			
		||||
			statuses[status.UserId] = "away"
 | 
			
		||||
		}
 | 
			
		||||
		if status.Status == model.STATUS_ONLINE {
 | 
			
		||||
			statuses[status.UserId] = "online"
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return statuses
 | 
			
		||||
@@ -724,6 +806,14 @@ func (m *MMClient) GetTeamId() string {
 | 
			
		||||
	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() {
 | 
			
		||||
	retries := 0
 | 
			
		||||
	backoff := time.Second * 60
 | 
			
		||||
@@ -744,9 +834,14 @@ func (m *MMClient) StatusLoop() {
 | 
			
		||||
				backoff = time.Second * 60
 | 
			
		||||
			case <-time.After(time.Second * 5):
 | 
			
		||||
				if retries > 3 {
 | 
			
		||||
					m.log.Debug("StatusLoop() timeout")
 | 
			
		||||
					m.Logout()
 | 
			
		||||
					m.WsQuit = false
 | 
			
		||||
					m.Login()
 | 
			
		||||
					err := m.Login()
 | 
			
		||||
					if err != nil {
 | 
			
		||||
						log.Errorf("Login failed: %#v", err)
 | 
			
		||||
						break
 | 
			
		||||
					}
 | 
			
		||||
					if m.OnWsConnect != nil {
 | 
			
		||||
						m.OnWsConnect()
 | 
			
		||||
					}
 | 
			
		||||
@@ -765,40 +860,39 @@ func (m *MMClient) StatusLoop() {
 | 
			
		||||
func (m *MMClient) initUser() error {
 | 
			
		||||
	m.Lock()
 | 
			
		||||
	defer m.Unlock()
 | 
			
		||||
	initLoad, err := m.Client.GetInitialLoad()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	initData := initLoad.Data.(*model.InitialLoad)
 | 
			
		||||
	m.User = initData.User
 | 
			
		||||
	// we only load all team data on initial login.
 | 
			
		||||
	// all other updates are for channels from our (primary) team only.
 | 
			
		||||
	//m.log.Debug("initUser(): loading all team data")
 | 
			
		||||
	for _, v := range initData.Teams {
 | 
			
		||||
		m.Client.SetTeamId(v.Id)
 | 
			
		||||
		mmusers, err := m.Client.GetProfiles(0, 50000, "")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "")
 | 
			
		||||
	if resp.Error != nil {
 | 
			
		||||
		return resp.Error
 | 
			
		||||
	}
 | 
			
		||||
	for _, team := range teams {
 | 
			
		||||
		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "")
 | 
			
		||||
		if resp.Error != nil {
 | 
			
		||||
			return errors.New(resp.Error.DetailedError)
 | 
			
		||||
		}
 | 
			
		||||
		t := &Team{Team: v, Users: mmusers.Data.(map[string]*model.User), Id: v.Id}
 | 
			
		||||
		mmchannels, err := m.Client.GetChannels("")
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
		usermap := make(map[string]*model.User)
 | 
			
		||||
		for _, user := range mmusers {
 | 
			
		||||
			usermap[user.Id] = user
 | 
			
		||||
		}
 | 
			
		||||
		t.Channels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
		if m.mmVersion() >= 3.08 {
 | 
			
		||||
			mmchannels, err = m.Client.GetMoreChannelsPage(0, 5000)
 | 
			
		||||
		} else {
 | 
			
		||||
			mmchannels, err = m.Client.GetMoreChannels("")
 | 
			
		||||
 | 
			
		||||
		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
 | 
			
		||||
		}
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return errors.New(err.DetailedError)
 | 
			
		||||
		t.Channels = mmchannels
 | 
			
		||||
		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "")
 | 
			
		||||
		if resp.Error != nil {
 | 
			
		||||
			return resp.Error
 | 
			
		||||
		}
 | 
			
		||||
		t.MoreChannels = mmchannels.Data.(*model.ChannelList)
 | 
			
		||||
		t.MoreChannels = mmchannels
 | 
			
		||||
		m.OtherTeams = append(m.OtherTeams, t)
 | 
			
		||||
		if v.Name == m.Credentials.Team {
 | 
			
		||||
		if team.Name == m.Credentials.Team {
 | 
			
		||||
			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
 | 
			
		||||
		for k, v := range t.Users {
 | 
			
		||||
@@ -819,22 +913,16 @@ func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) err
 | 
			
		||||
	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 {
 | 
			
		||||
	if strings.HasPrefix(version, "3.5.0") ||
 | 
			
		||||
		strings.HasPrefix(version, "3.6.0") ||
 | 
			
		||||
		strings.HasPrefix(version, "3.7.0") ||
 | 
			
		||||
		strings.HasPrefix(version, "3.8.0") ||
 | 
			
		||||
	if strings.HasPrefix(version, "3.8.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 false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func digestString(s string) string {
 | 
			
		||||
	return fmt.Sprintf("%x", md5.Sum([]byte(s)))
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -7,6 +7,7 @@ import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gorilla/schema"
 | 
			
		||||
	"github.com/nlopes/slack"
 | 
			
		||||
	"io"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"log"
 | 
			
		||||
@@ -17,13 +18,14 @@ import (
 | 
			
		||||
 | 
			
		||||
// OMessage for mattermost incoming webhook. (send to mattermost)
 | 
			
		||||
type OMessage struct {
 | 
			
		||||
	Channel     string      `json:"channel,omitempty"`
 | 
			
		||||
	IconURL     string      `json:"icon_url,omitempty"`
 | 
			
		||||
	IconEmoji   string      `json:"icon_emoji,omitempty"`
 | 
			
		||||
	UserName    string      `json:"username,omitempty"`
 | 
			
		||||
	Text        string      `json:"text"`
 | 
			
		||||
	Attachments interface{} `json:"attachments,omitempty"`
 | 
			
		||||
	Type        string      `json:"type,omitempty"`
 | 
			
		||||
	Channel     string                 `json:"channel,omitempty"`
 | 
			
		||||
	IconURL     string                 `json:"icon_url,omitempty"`
 | 
			
		||||
	IconEmoji   string                 `json:"icon_emoji,omitempty"`
 | 
			
		||||
	UserName    string                 `json:"username,omitempty"`
 | 
			
		||||
	Text        string                 `json:"text"`
 | 
			
		||||
	Attachments []slack.Attachment     `json:"attachments,omitempty"`
 | 
			
		||||
	Type        string                 `json:"type,omitempty"`
 | 
			
		||||
	Props       map[string]interface{} `json:"props"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// IMessage for mattermost outgoing webhook. (received from mattermost)
 | 
			
		||||
@@ -43,6 +45,7 @@ type IMessage struct {
 | 
			
		||||
	ServiceId   string `schema:"service_id"`
 | 
			
		||||
	Text        string `schema:"text"`
 | 
			
		||||
	TriggerWord string `schema:"trigger_word"`
 | 
			
		||||
	FileIDs     string `schema:"file_ids"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Client for Mattermost.
 | 
			
		||||
@@ -134,12 +137,11 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 | 
			
		||||
 | 
			
		||||
// Receive returns an incoming message from mattermost outgoing webhooks URL.
 | 
			
		||||
func (c *Client) Receive() IMessage {
 | 
			
		||||
	for {
 | 
			
		||||
		select {
 | 
			
		||||
		case msg := <-c.In:
 | 
			
		||||
			return msg
 | 
			
		||||
		}
 | 
			
		||||
	var msg IMessage
 | 
			
		||||
	for msg := range c.In {
 | 
			
		||||
		return msg
 | 
			
		||||
	}
 | 
			
		||||
	return msg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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
 | 
			
		||||
func (gitter *Gitter) SendMessage(roomID, text string) error {
 | 
			
		||||
func (gitter *Gitter) SendMessage(roomID, text string) (*Message, error) {
 | 
			
		||||
 | 
			
		||||
	message := Message{Text: text}
 | 
			
		||||
	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 {
 | 
			
		||||
		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
 | 
			
		||||
@@ -265,7 +291,7 @@ func (gitter *Gitter) SearchRooms(room string) ([]Room, error) {
 | 
			
		||||
		Results []Room `json:"results"`
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room )
 | 
			
		||||
	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		gitter.log(err)
 | 
			
		||||
@@ -414,6 +440,39 @@ func (gitter *Gitter) post(url string, body []byte) ([]byte, error) {
 | 
			
		||||
	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) {
 | 
			
		||||
	r, err := http.NewRequest("delete", url, 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
 | 
			
		||||
func parseToEvent(msg string) (*Event, error) {
 | 
			
		||||
	msg = strings.TrimSuffix(msg, "\n") //Remove \r\n
 | 
			
		||||
@@ -95,6 +106,26 @@ func parseToEvent(msg string) (*Event, error) {
 | 
			
		||||
	if len(msg) < 5 {
 | 
			
		||||
		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 i := strings.Index(msg, " "); i > -1 {
 | 
			
		||||
			event.Source = msg[1:i]
 | 
			
		||||
@@ -196,12 +227,17 @@ func (irc *Connection) isQuitting() bool {
 | 
			
		||||
// Main loop to control the connection.
 | 
			
		||||
func (irc *Connection) Loop() {
 | 
			
		||||
	errChan := irc.ErrorChan()
 | 
			
		||||
	connTime := time.Now()
 | 
			
		||||
	for !irc.isQuitting() {
 | 
			
		||||
		err := <-errChan
 | 
			
		||||
		close(irc.end)
 | 
			
		||||
		irc.Wait()
 | 
			
		||||
		for !irc.isQuitting() {
 | 
			
		||||
			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 {
 | 
			
		||||
				irc.Log.Printf("Error while reconnecting: %s\n", err)
 | 
			
		||||
				time.Sleep(60 * time.Second)
 | 
			
		||||
@@ -210,6 +246,7 @@ func (irc *Connection) Loop() {
 | 
			
		||||
				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)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	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 {
 | 
			
		||||
		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 {
 | 
			
		||||
		case res := <-resChan:
 | 
			
		||||
		case res := <-saslResChan:
 | 
			
		||||
			if res.Failed {
 | 
			
		||||
				close(resChan)
 | 
			
		||||
				close(saslResChan)
 | 
			
		||||
				return res.Err
 | 
			
		||||
			}
 | 
			
		||||
		case <-time.After(time.Second * 15):
 | 
			
		||||
			close(resChan)
 | 
			
		||||
			close(saslResChan)
 | 
			
		||||
			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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -43,7 +43,6 @@ func (irc *Connection) setupSASLCallbacks(result chan<- *SASLResult) {
 | 
			
		||||
		result <- &SASLResult{true, errors.New(e.Arguments[1])}
 | 
			
		||||
	})
 | 
			
		||||
	irc.AddCallback("903", func(e *Event) {
 | 
			
		||||
		irc.SendRaw("CAP END")
 | 
			
		||||
		result <- &SASLResult{false, nil}
 | 
			
		||||
	})
 | 
			
		||||
	irc.AddCallback("904", func(e *Event) {
 | 
			
		||||
@@ -15,20 +15,22 @@ import (
 | 
			
		||||
type Connection struct {
 | 
			
		||||
	sync.Mutex
 | 
			
		||||
	sync.WaitGroup
 | 
			
		||||
	Debug        bool
 | 
			
		||||
	Error        chan error
 | 
			
		||||
	Password     string
 | 
			
		||||
	UseTLS       bool
 | 
			
		||||
	UseSASL      bool
 | 
			
		||||
	SASLLogin    string
 | 
			
		||||
	SASLPassword string
 | 
			
		||||
	SASLMech     string
 | 
			
		||||
	TLSConfig    *tls.Config
 | 
			
		||||
	Version      string
 | 
			
		||||
	Timeout      time.Duration
 | 
			
		||||
	PingFreq     time.Duration
 | 
			
		||||
	KeepAlive    time.Duration
 | 
			
		||||
	Server       string
 | 
			
		||||
	Debug            bool
 | 
			
		||||
	Error            chan error
 | 
			
		||||
	Password         string
 | 
			
		||||
	UseTLS           bool
 | 
			
		||||
	UseSASL          bool
 | 
			
		||||
	RequestCaps      []string
 | 
			
		||||
	AcknowledgedCaps []string
 | 
			
		||||
	SASLLogin        string
 | 
			
		||||
	SASLPassword     string
 | 
			
		||||
	SASLMech         string
 | 
			
		||||
	TLSConfig        *tls.Config
 | 
			
		||||
	Version          string
 | 
			
		||||
	Timeout          time.Duration
 | 
			
		||||
	PingFreq         time.Duration
 | 
			
		||||
	KeepAlive        time.Duration
 | 
			
		||||
	Server           string
 | 
			
		||||
 | 
			
		||||
	socket net.Conn
 | 
			
		||||
	pwrite chan string
 | 
			
		||||
@@ -59,6 +61,7 @@ type Event struct {
 | 
			
		||||
	Source     string //<host>
 | 
			
		||||
	User       string //<usr>
 | 
			
		||||
	Arguments  []string
 | 
			
		||||
	Tags       map[string]string
 | 
			
		||||
	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
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										362
									
								
								vendor/github.com/armon/consul-api/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										362
									
								
								vendor/github.com/armon/consul-api/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.
 | 
			
		||||
							
								
								
									
										140
									
								
								vendor/github.com/armon/consul-api/acl.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										140
									
								
								vendor/github.com/armon/consul-api/acl.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,140 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	// ACLCLientType is the client type token
 | 
			
		||||
	ACLClientType = "client"
 | 
			
		||||
 | 
			
		||||
	// ACLManagementType is the management type token
 | 
			
		||||
	ACLManagementType = "management"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// ACLEntry is used to represent an ACL entry
 | 
			
		||||
type ACLEntry struct {
 | 
			
		||||
	CreateIndex uint64
 | 
			
		||||
	ModifyIndex uint64
 | 
			
		||||
	ID          string
 | 
			
		||||
	Name        string
 | 
			
		||||
	Type        string
 | 
			
		||||
	Rules       string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ACL can be used to query the ACL endpoints
 | 
			
		||||
type ACL struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ACL returns a handle to the ACL endpoints
 | 
			
		||||
func (c *Client) ACL() *ACL {
 | 
			
		||||
	return &ACL{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Create is used to generate a new token with the given parameters
 | 
			
		||||
func (a *ACL) Create(acl *ACLEntry, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/acl/create")
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	r.obj = acl
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	var out struct{ ID string }
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out.ID, wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Update is used to update the rules of an existing token
 | 
			
		||||
func (a *ACL) Update(acl *ACLEntry, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/acl/update")
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	r.obj = acl
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	return wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Destroy is used to destroy a given ACL token ID
 | 
			
		||||
func (a *ACL) Destroy(id string, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/acl/destroy/"+id)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	return wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Clone is used to return a new token cloned from an existing one
 | 
			
		||||
func (a *ACL) Clone(id string, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/acl/clone/"+id)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	var out struct{ ID string }
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out.ID, wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Info is used to query for information about an ACL token
 | 
			
		||||
func (a *ACL) Info(id string, q *QueryOptions) (*ACLEntry, *QueryMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/acl/info/"+id)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*ACLEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if len(entries) > 0 {
 | 
			
		||||
		return entries[0], qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	return nil, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// List is used to get all the ACL tokens
 | 
			
		||||
func (a *ACL) List(q *QueryOptions) ([]*ACLEntry, *QueryMeta, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/acl/list")
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*ACLEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										272
									
								
								vendor/github.com/armon/consul-api/agent.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										272
									
								
								vendor/github.com/armon/consul-api/agent.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,272 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// AgentCheck represents a check known to the agent
 | 
			
		||||
type AgentCheck struct {
 | 
			
		||||
	Node        string
 | 
			
		||||
	CheckID     string
 | 
			
		||||
	Name        string
 | 
			
		||||
	Status      string
 | 
			
		||||
	Notes       string
 | 
			
		||||
	Output      string
 | 
			
		||||
	ServiceID   string
 | 
			
		||||
	ServiceName string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AgentService represents a service known to the agent
 | 
			
		||||
type AgentService struct {
 | 
			
		||||
	ID      string
 | 
			
		||||
	Service string
 | 
			
		||||
	Tags    []string
 | 
			
		||||
	Port    int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AgentMember represents a cluster member known to the agent
 | 
			
		||||
type AgentMember struct {
 | 
			
		||||
	Name        string
 | 
			
		||||
	Addr        string
 | 
			
		||||
	Port        uint16
 | 
			
		||||
	Tags        map[string]string
 | 
			
		||||
	Status      int
 | 
			
		||||
	ProtocolMin uint8
 | 
			
		||||
	ProtocolMax uint8
 | 
			
		||||
	ProtocolCur uint8
 | 
			
		||||
	DelegateMin uint8
 | 
			
		||||
	DelegateMax uint8
 | 
			
		||||
	DelegateCur uint8
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AgentServiceRegistration is used to register a new service
 | 
			
		||||
type AgentServiceRegistration struct {
 | 
			
		||||
	ID    string   `json:",omitempty"`
 | 
			
		||||
	Name  string   `json:",omitempty"`
 | 
			
		||||
	Tags  []string `json:",omitempty"`
 | 
			
		||||
	Port  int      `json:",omitempty"`
 | 
			
		||||
	Check *AgentServiceCheck
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AgentCheckRegistration is used to register a new check
 | 
			
		||||
type AgentCheckRegistration struct {
 | 
			
		||||
	ID    string `json:",omitempty"`
 | 
			
		||||
	Name  string `json:",omitempty"`
 | 
			
		||||
	Notes string `json:",omitempty"`
 | 
			
		||||
	AgentServiceCheck
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AgentServiceCheck is used to create an associated
 | 
			
		||||
// check for a service
 | 
			
		||||
type AgentServiceCheck struct {
 | 
			
		||||
	Script   string `json:",omitempty"`
 | 
			
		||||
	Interval string `json:",omitempty"`
 | 
			
		||||
	TTL      string `json:",omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Agent can be used to query the Agent endpoints
 | 
			
		||||
type Agent struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
 | 
			
		||||
	// cache the node name
 | 
			
		||||
	nodeName string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Agent returns a handle to the agent endpoints
 | 
			
		||||
func (c *Client) Agent() *Agent {
 | 
			
		||||
	return &Agent{c: c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Self is used to query the agent we are speaking to for
 | 
			
		||||
// information about itself
 | 
			
		||||
func (a *Agent) Self() (map[string]map[string]interface{}, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/agent/self")
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var out map[string]map[string]interface{}
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NodeName is used to get the node name of the agent
 | 
			
		||||
func (a *Agent) NodeName() (string, error) {
 | 
			
		||||
	if a.nodeName != "" {
 | 
			
		||||
		return a.nodeName, nil
 | 
			
		||||
	}
 | 
			
		||||
	info, err := a.Self()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	name := info["Config"]["NodeName"].(string)
 | 
			
		||||
	a.nodeName = name
 | 
			
		||||
	return name, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Checks returns the locally registered checks
 | 
			
		||||
func (a *Agent) Checks() (map[string]*AgentCheck, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/agent/checks")
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var out map[string]*AgentCheck
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Services returns the locally registered services
 | 
			
		||||
func (a *Agent) Services() (map[string]*AgentService, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/agent/services")
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var out map[string]*AgentService
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Members returns the known gossip members. The WAN
 | 
			
		||||
// flag can be used to query a server for WAN members.
 | 
			
		||||
func (a *Agent) Members(wan bool) ([]*AgentMember, error) {
 | 
			
		||||
	r := a.c.newRequest("GET", "/v1/agent/members")
 | 
			
		||||
	if wan {
 | 
			
		||||
		r.params.Set("wan", "1")
 | 
			
		||||
	}
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var out []*AgentMember
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ServiceRegister is used to register a new service with
 | 
			
		||||
// the local agent
 | 
			
		||||
func (a *Agent) ServiceRegister(service *AgentServiceRegistration) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/service/register")
 | 
			
		||||
	r.obj = service
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ServiceDeregister is used to deregister a service with
 | 
			
		||||
// the local agent
 | 
			
		||||
func (a *Agent) ServiceDeregister(serviceID string) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/service/deregister/"+serviceID)
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// PassTTL is used to set a TTL check to the passing state
 | 
			
		||||
func (a *Agent) PassTTL(checkID, note string) error {
 | 
			
		||||
	return a.UpdateTTL(checkID, note, "pass")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WarnTTL is used to set a TTL check to the warning state
 | 
			
		||||
func (a *Agent) WarnTTL(checkID, note string) error {
 | 
			
		||||
	return a.UpdateTTL(checkID, note, "warn")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// FailTTL is used to set a TTL check to the failing state
 | 
			
		||||
func (a *Agent) FailTTL(checkID, note string) error {
 | 
			
		||||
	return a.UpdateTTL(checkID, note, "fail")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// UpdateTTL is used to update the TTL of a check
 | 
			
		||||
func (a *Agent) UpdateTTL(checkID, note, status string) error {
 | 
			
		||||
	switch status {
 | 
			
		||||
	case "pass":
 | 
			
		||||
	case "warn":
 | 
			
		||||
	case "fail":
 | 
			
		||||
	default:
 | 
			
		||||
		return fmt.Errorf("Invalid status: %s", status)
 | 
			
		||||
	}
 | 
			
		||||
	endpoint := fmt.Sprintf("/v1/agent/check/%s/%s", status, checkID)
 | 
			
		||||
	r := a.c.newRequest("PUT", endpoint)
 | 
			
		||||
	r.params.Set("note", note)
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CheckRegister is used to register a new check with
 | 
			
		||||
// the local agent
 | 
			
		||||
func (a *Agent) CheckRegister(check *AgentCheckRegistration) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/check/register")
 | 
			
		||||
	r.obj = check
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CheckDeregister is used to deregister a check with
 | 
			
		||||
// the local agent
 | 
			
		||||
func (a *Agent) CheckDeregister(checkID string) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/check/deregister/"+checkID)
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Join is used to instruct the agent to attempt a join to
 | 
			
		||||
// another cluster member
 | 
			
		||||
func (a *Agent) Join(addr string, wan bool) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/join/"+addr)
 | 
			
		||||
	if wan {
 | 
			
		||||
		r.params.Set("wan", "1")
 | 
			
		||||
	}
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ForceLeave is used to have the agent eject a failed node
 | 
			
		||||
func (a *Agent) ForceLeave(node string) error {
 | 
			
		||||
	r := a.c.newRequest("PUT", "/v1/agent/force-leave/"+node)
 | 
			
		||||
	_, resp, err := requireOK(a.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										323
									
								
								vendor/github.com/armon/consul-api/api.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										323
									
								
								vendor/github.com/armon/consul-api/api.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,323 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// QueryOptions are used to parameterize a query
 | 
			
		||||
type QueryOptions struct {
 | 
			
		||||
	// Providing a datacenter overwrites the DC provided
 | 
			
		||||
	// by the Config
 | 
			
		||||
	Datacenter string
 | 
			
		||||
 | 
			
		||||
	// AllowStale allows any Consul server (non-leader) to service
 | 
			
		||||
	// a read. This allows for lower latency and higher throughput
 | 
			
		||||
	AllowStale bool
 | 
			
		||||
 | 
			
		||||
	// RequireConsistent forces the read to be fully consistent.
 | 
			
		||||
	// This is more expensive but prevents ever performing a stale
 | 
			
		||||
	// read.
 | 
			
		||||
	RequireConsistent bool
 | 
			
		||||
 | 
			
		||||
	// WaitIndex is used to enable a blocking query. Waits
 | 
			
		||||
	// until the timeout or the next index is reached
 | 
			
		||||
	WaitIndex uint64
 | 
			
		||||
 | 
			
		||||
	// WaitTime is used to bound the duration of a wait.
 | 
			
		||||
	// Defaults to that of the Config, but can be overriden.
 | 
			
		||||
	WaitTime time.Duration
 | 
			
		||||
 | 
			
		||||
	// Token is used to provide a per-request ACL token
 | 
			
		||||
	// which overrides the agent's default token.
 | 
			
		||||
	Token string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WriteOptions are used to parameterize a write
 | 
			
		||||
type WriteOptions struct {
 | 
			
		||||
	// Providing a datacenter overwrites the DC provided
 | 
			
		||||
	// by the Config
 | 
			
		||||
	Datacenter string
 | 
			
		||||
 | 
			
		||||
	// Token is used to provide a per-request ACL token
 | 
			
		||||
	// which overrides the agent's default token.
 | 
			
		||||
	Token string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// QueryMeta is used to return meta data about a query
 | 
			
		||||
type QueryMeta struct {
 | 
			
		||||
	// LastIndex. This can be used as a WaitIndex to perform
 | 
			
		||||
	// a blocking query
 | 
			
		||||
	LastIndex uint64
 | 
			
		||||
 | 
			
		||||
	// Time of last contact from the leader for the
 | 
			
		||||
	// server servicing the request
 | 
			
		||||
	LastContact time.Duration
 | 
			
		||||
 | 
			
		||||
	// Is there a known leader
 | 
			
		||||
	KnownLeader bool
 | 
			
		||||
 | 
			
		||||
	// How long did the request take
 | 
			
		||||
	RequestTime time.Duration
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WriteMeta is used to return meta data about a write
 | 
			
		||||
type WriteMeta struct {
 | 
			
		||||
	// How long did the request take
 | 
			
		||||
	RequestTime time.Duration
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// HttpBasicAuth is used to authenticate http client with HTTP Basic Authentication
 | 
			
		||||
type HttpBasicAuth struct {
 | 
			
		||||
	// Username to use for HTTP Basic Authentication
 | 
			
		||||
	Username string
 | 
			
		||||
 | 
			
		||||
	// Password to use for HTTP Basic Authentication
 | 
			
		||||
	Password string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Config is used to configure the creation of a client
 | 
			
		||||
type Config struct {
 | 
			
		||||
	// Address is the address of the Consul server
 | 
			
		||||
	Address string
 | 
			
		||||
 | 
			
		||||
	// Scheme is the URI scheme for the Consul server
 | 
			
		||||
	Scheme string
 | 
			
		||||
 | 
			
		||||
	// Datacenter to use. If not provided, the default agent datacenter is used.
 | 
			
		||||
	Datacenter string
 | 
			
		||||
 | 
			
		||||
	// HttpClient is the client to use. Default will be
 | 
			
		||||
	// used if not provided.
 | 
			
		||||
	HttpClient *http.Client
 | 
			
		||||
 | 
			
		||||
	// HttpAuth is the auth info to use for http access.
 | 
			
		||||
	HttpAuth *HttpBasicAuth
 | 
			
		||||
 | 
			
		||||
	// WaitTime limits how long a Watch will block. If not provided,
 | 
			
		||||
	// the agent default values will be used.
 | 
			
		||||
	WaitTime time.Duration
 | 
			
		||||
 | 
			
		||||
	// Token is used to provide a per-request ACL token
 | 
			
		||||
	// which overrides the agent's default token.
 | 
			
		||||
	Token string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// DefaultConfig returns a default configuration for the client
 | 
			
		||||
func DefaultConfig() *Config {
 | 
			
		||||
	return &Config{
 | 
			
		||||
		Address:    "127.0.0.1:8500",
 | 
			
		||||
		Scheme:     "http",
 | 
			
		||||
		HttpClient: http.DefaultClient,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Client provides a client to the Consul API
 | 
			
		||||
type Client struct {
 | 
			
		||||
	config Config
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewClient returns a new client
 | 
			
		||||
func NewClient(config *Config) (*Client, error) {
 | 
			
		||||
	// bootstrap the config
 | 
			
		||||
	defConfig := DefaultConfig()
 | 
			
		||||
 | 
			
		||||
	if len(config.Address) == 0 {
 | 
			
		||||
		config.Address = defConfig.Address
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(config.Scheme) == 0 {
 | 
			
		||||
		config.Scheme = defConfig.Scheme
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if config.HttpClient == nil {
 | 
			
		||||
		config.HttpClient = defConfig.HttpClient
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	client := &Client{
 | 
			
		||||
		config: *config,
 | 
			
		||||
	}
 | 
			
		||||
	return client, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// request is used to help build up a request
 | 
			
		||||
type request struct {
 | 
			
		||||
	config *Config
 | 
			
		||||
	method string
 | 
			
		||||
	url    *url.URL
 | 
			
		||||
	params url.Values
 | 
			
		||||
	body   io.Reader
 | 
			
		||||
	obj    interface{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// setQueryOptions is used to annotate the request with
 | 
			
		||||
// additional query options
 | 
			
		||||
func (r *request) setQueryOptions(q *QueryOptions) {
 | 
			
		||||
	if q == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if q.Datacenter != "" {
 | 
			
		||||
		r.params.Set("dc", q.Datacenter)
 | 
			
		||||
	}
 | 
			
		||||
	if q.AllowStale {
 | 
			
		||||
		r.params.Set("stale", "")
 | 
			
		||||
	}
 | 
			
		||||
	if q.RequireConsistent {
 | 
			
		||||
		r.params.Set("consistent", "")
 | 
			
		||||
	}
 | 
			
		||||
	if q.WaitIndex != 0 {
 | 
			
		||||
		r.params.Set("index", strconv.FormatUint(q.WaitIndex, 10))
 | 
			
		||||
	}
 | 
			
		||||
	if q.WaitTime != 0 {
 | 
			
		||||
		r.params.Set("wait", durToMsec(q.WaitTime))
 | 
			
		||||
	}
 | 
			
		||||
	if q.Token != "" {
 | 
			
		||||
		r.params.Set("token", q.Token)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// durToMsec converts a duration to a millisecond specified string
 | 
			
		||||
func durToMsec(dur time.Duration) string {
 | 
			
		||||
	return fmt.Sprintf("%dms", dur/time.Millisecond)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// setWriteOptions is used to annotate the request with
 | 
			
		||||
// additional write options
 | 
			
		||||
func (r *request) setWriteOptions(q *WriteOptions) {
 | 
			
		||||
	if q == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if q.Datacenter != "" {
 | 
			
		||||
		r.params.Set("dc", q.Datacenter)
 | 
			
		||||
	}
 | 
			
		||||
	if q.Token != "" {
 | 
			
		||||
		r.params.Set("token", q.Token)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// toHTTP converts the request to an HTTP request
 | 
			
		||||
func (r *request) toHTTP() (*http.Request, error) {
 | 
			
		||||
	// Encode the query parameters
 | 
			
		||||
	r.url.RawQuery = r.params.Encode()
 | 
			
		||||
 | 
			
		||||
	// Get the url sring
 | 
			
		||||
	urlRaw := r.url.String()
 | 
			
		||||
 | 
			
		||||
	// Check if we should encode the body
 | 
			
		||||
	if r.body == nil && r.obj != nil {
 | 
			
		||||
		if b, err := encodeBody(r.obj); err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		} else {
 | 
			
		||||
			r.body = b
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Create the HTTP request
 | 
			
		||||
	req, err := http.NewRequest(r.method, urlRaw, r.body)
 | 
			
		||||
 | 
			
		||||
	// Setup auth
 | 
			
		||||
	if err == nil && r.config.HttpAuth != nil {
 | 
			
		||||
		req.SetBasicAuth(r.config.HttpAuth.Username, r.config.HttpAuth.Password)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return req, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// newRequest is used to create a new request
 | 
			
		||||
func (c *Client) newRequest(method, path string) *request {
 | 
			
		||||
	r := &request{
 | 
			
		||||
		config: &c.config,
 | 
			
		||||
		method: method,
 | 
			
		||||
		url: &url.URL{
 | 
			
		||||
			Scheme: c.config.Scheme,
 | 
			
		||||
			Host:   c.config.Address,
 | 
			
		||||
			Path:   path,
 | 
			
		||||
		},
 | 
			
		||||
		params: make(map[string][]string),
 | 
			
		||||
	}
 | 
			
		||||
	if c.config.Datacenter != "" {
 | 
			
		||||
		r.params.Set("dc", c.config.Datacenter)
 | 
			
		||||
	}
 | 
			
		||||
	if c.config.WaitTime != 0 {
 | 
			
		||||
		r.params.Set("wait", durToMsec(r.config.WaitTime))
 | 
			
		||||
	}
 | 
			
		||||
	if c.config.Token != "" {
 | 
			
		||||
		r.params.Set("token", r.config.Token)
 | 
			
		||||
	}
 | 
			
		||||
	return r
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// doRequest runs a request with our client
 | 
			
		||||
func (c *Client) doRequest(r *request) (time.Duration, *http.Response, error) {
 | 
			
		||||
	req, err := r.toHTTP()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return 0, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	start := time.Now()
 | 
			
		||||
	resp, err := c.config.HttpClient.Do(req)
 | 
			
		||||
	diff := time.Now().Sub(start)
 | 
			
		||||
	return diff, resp, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// parseQueryMeta is used to help parse query meta-data
 | 
			
		||||
func parseQueryMeta(resp *http.Response, q *QueryMeta) error {
 | 
			
		||||
	header := resp.Header
 | 
			
		||||
 | 
			
		||||
	// Parse the X-Consul-Index
 | 
			
		||||
	index, err := strconv.ParseUint(header.Get("X-Consul-Index"), 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("Failed to parse X-Consul-Index: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	q.LastIndex = index
 | 
			
		||||
 | 
			
		||||
	// Parse the X-Consul-LastContact
 | 
			
		||||
	last, err := strconv.ParseUint(header.Get("X-Consul-LastContact"), 10, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("Failed to parse X-Consul-LastContact: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	q.LastContact = time.Duration(last) * time.Millisecond
 | 
			
		||||
 | 
			
		||||
	// Parse the X-Consul-KnownLeader
 | 
			
		||||
	switch header.Get("X-Consul-KnownLeader") {
 | 
			
		||||
	case "true":
 | 
			
		||||
		q.KnownLeader = true
 | 
			
		||||
	default:
 | 
			
		||||
		q.KnownLeader = false
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// decodeBody is used to JSON decode a body
 | 
			
		||||
func decodeBody(resp *http.Response, out interface{}) error {
 | 
			
		||||
	dec := json.NewDecoder(resp.Body)
 | 
			
		||||
	return dec.Decode(out)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// encodeBody is used to encode a request body
 | 
			
		||||
func encodeBody(obj interface{}) (io.Reader, error) {
 | 
			
		||||
	buf := bytes.NewBuffer(nil)
 | 
			
		||||
	enc := json.NewEncoder(buf)
 | 
			
		||||
	if err := enc.Encode(obj); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return buf, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// requireOK is used to wrap doRequest and check for a 200
 | 
			
		||||
func requireOK(d time.Duration, resp *http.Response, e error) (time.Duration, *http.Response, error) {
 | 
			
		||||
	if e != nil {
 | 
			
		||||
		return d, resp, e
 | 
			
		||||
	}
 | 
			
		||||
	if resp.StatusCode != 200 {
 | 
			
		||||
		var buf bytes.Buffer
 | 
			
		||||
		io.Copy(&buf, resp.Body)
 | 
			
		||||
		return d, resp, fmt.Errorf("Unexpected response code: %d (%s)", resp.StatusCode, buf.Bytes())
 | 
			
		||||
	}
 | 
			
		||||
	return d, resp, e
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										181
									
								
								vendor/github.com/armon/consul-api/catalog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										181
									
								
								vendor/github.com/armon/consul-api/catalog.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,181 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
type Node struct {
 | 
			
		||||
	Node    string
 | 
			
		||||
	Address string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CatalogService struct {
 | 
			
		||||
	Node        string
 | 
			
		||||
	Address     string
 | 
			
		||||
	ServiceID   string
 | 
			
		||||
	ServiceName string
 | 
			
		||||
	ServiceTags []string
 | 
			
		||||
	ServicePort int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CatalogNode struct {
 | 
			
		||||
	Node     *Node
 | 
			
		||||
	Services map[string]*AgentService
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CatalogRegistration struct {
 | 
			
		||||
	Node       string
 | 
			
		||||
	Address    string
 | 
			
		||||
	Datacenter string
 | 
			
		||||
	Service    *AgentService
 | 
			
		||||
	Check      *AgentCheck
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CatalogDeregistration struct {
 | 
			
		||||
	Node       string
 | 
			
		||||
	Address    string
 | 
			
		||||
	Datacenter string
 | 
			
		||||
	ServiceID  string
 | 
			
		||||
	CheckID    string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Catalog can be used to query the Catalog endpoints
 | 
			
		||||
type Catalog struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Catalog returns a handle to the catalog endpoints
 | 
			
		||||
func (c *Client) Catalog() *Catalog {
 | 
			
		||||
	return &Catalog{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Catalog) Register(reg *CatalogRegistration, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("PUT", "/v1/catalog/register")
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	r.obj = reg
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{}
 | 
			
		||||
	wm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	return wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Catalog) Deregister(dereg *CatalogDeregistration, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("PUT", "/v1/catalog/deregister")
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	r.obj = dereg
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{}
 | 
			
		||||
	wm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	return wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Datacenters is used to query for all the known datacenters
 | 
			
		||||
func (c *Catalog) Datacenters() ([]string, error) {
 | 
			
		||||
	r := c.c.newRequest("GET", "/v1/catalog/datacenters")
 | 
			
		||||
	_, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var out []string
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Nodes is used to query all the known nodes
 | 
			
		||||
func (c *Catalog) Nodes(q *QueryOptions) ([]*Node, *QueryMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("GET", "/v1/catalog/nodes")
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*Node
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Services is used to query for all known services
 | 
			
		||||
func (c *Catalog) Services(q *QueryOptions) (map[string][]string, *QueryMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("GET", "/v1/catalog/services")
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out map[string][]string
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Service is used to query catalog entries for a given service
 | 
			
		||||
func (c *Catalog) Service(service, tag string, q *QueryOptions) ([]*CatalogService, *QueryMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("GET", "/v1/catalog/service/"+service)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	if tag != "" {
 | 
			
		||||
		r.params.Set("tag", tag)
 | 
			
		||||
	}
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*CatalogService
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Node is used to query for service information about a single node
 | 
			
		||||
func (c *Catalog) Node(node string, q *QueryOptions) (*CatalogNode, *QueryMeta, error) {
 | 
			
		||||
	r := c.c.newRequest("GET", "/v1/catalog/node/"+node)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(c.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out *CatalogNode
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										104
									
								
								vendor/github.com/armon/consul-api/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										104
									
								
								vendor/github.com/armon/consul-api/event.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,104 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"strconv"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Event can be used to query the Event endpoints
 | 
			
		||||
type Event struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// UserEvent represents an event that was fired by the user
 | 
			
		||||
type UserEvent struct {
 | 
			
		||||
	ID            string
 | 
			
		||||
	Name          string
 | 
			
		||||
	Payload       []byte
 | 
			
		||||
	NodeFilter    string
 | 
			
		||||
	ServiceFilter string
 | 
			
		||||
	TagFilter     string
 | 
			
		||||
	Version       int
 | 
			
		||||
	LTime         uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Event returns a handle to the event endpoints
 | 
			
		||||
func (c *Client) Event() *Event {
 | 
			
		||||
	return &Event{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Fire is used to fire a new user event. Only the Name, Payload and Filters
 | 
			
		||||
// are respected. This returns the ID or an associated error. Cross DC requests
 | 
			
		||||
// are supported.
 | 
			
		||||
func (e *Event) Fire(params *UserEvent, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	r := e.c.newRequest("PUT", "/v1/event/fire/"+params.Name)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	if params.NodeFilter != "" {
 | 
			
		||||
		r.params.Set("node", params.NodeFilter)
 | 
			
		||||
	}
 | 
			
		||||
	if params.ServiceFilter != "" {
 | 
			
		||||
		r.params.Set("service", params.ServiceFilter)
 | 
			
		||||
	}
 | 
			
		||||
	if params.TagFilter != "" {
 | 
			
		||||
		r.params.Set("tag", params.TagFilter)
 | 
			
		||||
	}
 | 
			
		||||
	if params.Payload != nil {
 | 
			
		||||
		r.body = bytes.NewReader(params.Payload)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rtt, resp, err := requireOK(e.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	var out UserEvent
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out.ID, wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// List is used to get the most recent events an agent has received.
 | 
			
		||||
// This list can be optionally filtered by the name. This endpoint supports
 | 
			
		||||
// quasi-blocking queries. The index is not monotonic, nor does it provide provide
 | 
			
		||||
// LastContact or KnownLeader.
 | 
			
		||||
func (e *Event) List(name string, q *QueryOptions) ([]*UserEvent, *QueryMeta, error) {
 | 
			
		||||
	r := e.c.newRequest("GET", "/v1/event/list")
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	if name != "" {
 | 
			
		||||
		r.params.Set("name", name)
 | 
			
		||||
	}
 | 
			
		||||
	rtt, resp, err := requireOK(e.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*UserEvent
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// IDToIndex is a bit of a hack. This simulates the index generation to
 | 
			
		||||
// convert an event ID into a WaitIndex.
 | 
			
		||||
func (e *Event) IDToIndex(uuid string) uint64 {
 | 
			
		||||
	lower := uuid[0:8] + uuid[9:13] + uuid[14:18]
 | 
			
		||||
	upper := uuid[19:23] + uuid[24:36]
 | 
			
		||||
	lowVal, err := strconv.ParseUint(lower, 16, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		panic("Failed to convert " + lower)
 | 
			
		||||
	}
 | 
			
		||||
	highVal, err := strconv.ParseUint(upper, 16, 64)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		panic("Failed to convert " + upper)
 | 
			
		||||
	}
 | 
			
		||||
	return lowVal ^ highVal
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										136
									
								
								vendor/github.com/armon/consul-api/health.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										136
									
								
								vendor/github.com/armon/consul-api/health.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,136 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// HealthCheck is used to represent a single check
 | 
			
		||||
type HealthCheck struct {
 | 
			
		||||
	Node        string
 | 
			
		||||
	CheckID     string
 | 
			
		||||
	Name        string
 | 
			
		||||
	Status      string
 | 
			
		||||
	Notes       string
 | 
			
		||||
	Output      string
 | 
			
		||||
	ServiceID   string
 | 
			
		||||
	ServiceName string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ServiceEntry is used for the health service endpoint
 | 
			
		||||
type ServiceEntry struct {
 | 
			
		||||
	Node    *Node
 | 
			
		||||
	Service *AgentService
 | 
			
		||||
	Checks  []*HealthCheck
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Health can be used to query the Health endpoints
 | 
			
		||||
type Health struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Health returns a handle to the health endpoints
 | 
			
		||||
func (c *Client) Health() *Health {
 | 
			
		||||
	return &Health{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Node is used to query for checks belonging to a given node
 | 
			
		||||
func (h *Health) Node(node string, q *QueryOptions) ([]*HealthCheck, *QueryMeta, error) {
 | 
			
		||||
	r := h.c.newRequest("GET", "/v1/health/node/"+node)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(h.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*HealthCheck
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Checks is used to return the checks associated with a service
 | 
			
		||||
func (h *Health) Checks(service string, q *QueryOptions) ([]*HealthCheck, *QueryMeta, error) {
 | 
			
		||||
	r := h.c.newRequest("GET", "/v1/health/checks/"+service)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(h.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*HealthCheck
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Service is used to query health information along with service info
 | 
			
		||||
// for a given service. It can optionally do server-side filtering on a tag
 | 
			
		||||
// or nodes with passing health checks only.
 | 
			
		||||
func (h *Health) Service(service, tag string, passingOnly bool, q *QueryOptions) ([]*ServiceEntry, *QueryMeta, error) {
 | 
			
		||||
	r := h.c.newRequest("GET", "/v1/health/service/"+service)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	if tag != "" {
 | 
			
		||||
		r.params.Set("tag", tag)
 | 
			
		||||
	}
 | 
			
		||||
	if passingOnly {
 | 
			
		||||
		r.params.Set("passing", "1")
 | 
			
		||||
	}
 | 
			
		||||
	rtt, resp, err := requireOK(h.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*ServiceEntry
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// State is used to retrieve all the checks in a given state.
 | 
			
		||||
// The wildcard "any" state can also be used for all checks.
 | 
			
		||||
func (h *Health) State(state string, q *QueryOptions) ([]*HealthCheck, *QueryMeta, error) {
 | 
			
		||||
	switch state {
 | 
			
		||||
	case "any":
 | 
			
		||||
	case "warning":
 | 
			
		||||
	case "critical":
 | 
			
		||||
	case "passing":
 | 
			
		||||
	case "unknown":
 | 
			
		||||
	default:
 | 
			
		||||
		return nil, nil, fmt.Errorf("Unsupported state: %v", state)
 | 
			
		||||
	}
 | 
			
		||||
	r := h.c.newRequest("GET", "/v1/health/state/"+state)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(h.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var out []*HealthCheck
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out, qm, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										219
									
								
								vendor/github.com/armon/consul-api/kv.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										219
									
								
								vendor/github.com/armon/consul-api/kv.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,219 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// KVPair is used to represent a single K/V entry
 | 
			
		||||
type KVPair struct {
 | 
			
		||||
	Key         string
 | 
			
		||||
	CreateIndex uint64
 | 
			
		||||
	ModifyIndex uint64
 | 
			
		||||
	LockIndex   uint64
 | 
			
		||||
	Flags       uint64
 | 
			
		||||
	Value       []byte
 | 
			
		||||
	Session     string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// KVPairs is a list of KVPair objects
 | 
			
		||||
type KVPairs []*KVPair
 | 
			
		||||
 | 
			
		||||
// KV is used to manipulate the K/V API
 | 
			
		||||
type KV struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// KV is used to return a handle to the K/V apis
 | 
			
		||||
func (c *Client) KV() *KV {
 | 
			
		||||
	return &KV{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Get is used to lookup a single key
 | 
			
		||||
func (k *KV) Get(key string, q *QueryOptions) (*KVPair, *QueryMeta, error) {
 | 
			
		||||
	resp, qm, err := k.getInternal(key, nil, q)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if resp == nil {
 | 
			
		||||
		return nil, qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var entries []*KVPair
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if len(entries) > 0 {
 | 
			
		||||
		return entries[0], qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	return nil, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// List is used to lookup all keys under a prefix
 | 
			
		||||
func (k *KV) List(prefix string, q *QueryOptions) (KVPairs, *QueryMeta, error) {
 | 
			
		||||
	resp, qm, err := k.getInternal(prefix, map[string]string{"recurse": ""}, q)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if resp == nil {
 | 
			
		||||
		return nil, qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var entries []*KVPair
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Keys is used to list all the keys under a prefix. Optionally,
 | 
			
		||||
// a separator can be used to limit the responses.
 | 
			
		||||
func (k *KV) Keys(prefix, separator string, q *QueryOptions) ([]string, *QueryMeta, error) {
 | 
			
		||||
	params := map[string]string{"keys": ""}
 | 
			
		||||
	if separator != "" {
 | 
			
		||||
		params["separator"] = separator
 | 
			
		||||
	}
 | 
			
		||||
	resp, qm, err := k.getInternal(prefix, params, q)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if resp == nil {
 | 
			
		||||
		return nil, qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var entries []string
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k *KV) getInternal(key string, params map[string]string, q *QueryOptions) (*http.Response, *QueryMeta, error) {
 | 
			
		||||
	r := k.c.newRequest("GET", "/v1/kv/"+key)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	for param, val := range params {
 | 
			
		||||
		r.params.Set(param, val)
 | 
			
		||||
	}
 | 
			
		||||
	rtt, resp, err := k.c.doRequest(r)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	if resp.StatusCode == 404 {
 | 
			
		||||
		resp.Body.Close()
 | 
			
		||||
		return nil, qm, nil
 | 
			
		||||
	} else if resp.StatusCode != 200 {
 | 
			
		||||
		resp.Body.Close()
 | 
			
		||||
		return nil, nil, fmt.Errorf("Unexpected response code: %d", resp.StatusCode)
 | 
			
		||||
	}
 | 
			
		||||
	return resp, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Put is used to write a new value. Only the
 | 
			
		||||
// Key, Flags and Value is respected.
 | 
			
		||||
func (k *KV) Put(p *KVPair, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	params := make(map[string]string, 1)
 | 
			
		||||
	if p.Flags != 0 {
 | 
			
		||||
		params["flags"] = strconv.FormatUint(p.Flags, 10)
 | 
			
		||||
	}
 | 
			
		||||
	_, wm, err := k.put(p.Key, params, p.Value, q)
 | 
			
		||||
	return wm, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CAS is used for a Check-And-Set operation. The Key,
 | 
			
		||||
// ModifyIndex, Flags and Value are respected. Returns true
 | 
			
		||||
// on success or false on failures.
 | 
			
		||||
func (k *KV) CAS(p *KVPair, q *WriteOptions) (bool, *WriteMeta, error) {
 | 
			
		||||
	params := make(map[string]string, 2)
 | 
			
		||||
	if p.Flags != 0 {
 | 
			
		||||
		params["flags"] = strconv.FormatUint(p.Flags, 10)
 | 
			
		||||
	}
 | 
			
		||||
	params["cas"] = strconv.FormatUint(p.ModifyIndex, 10)
 | 
			
		||||
	return k.put(p.Key, params, p.Value, q)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Acquire is used for a lock acquisiiton operation. The Key,
 | 
			
		||||
// Flags, Value and Session are respected. Returns true
 | 
			
		||||
// on success or false on failures.
 | 
			
		||||
func (k *KV) Acquire(p *KVPair, q *WriteOptions) (bool, *WriteMeta, error) {
 | 
			
		||||
	params := make(map[string]string, 2)
 | 
			
		||||
	if p.Flags != 0 {
 | 
			
		||||
		params["flags"] = strconv.FormatUint(p.Flags, 10)
 | 
			
		||||
	}
 | 
			
		||||
	params["acquire"] = p.Session
 | 
			
		||||
	return k.put(p.Key, params, p.Value, q)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Release is used for a lock release operation. The Key,
 | 
			
		||||
// Flags, Value and Session are respected. Returns true
 | 
			
		||||
// on success or false on failures.
 | 
			
		||||
func (k *KV) Release(p *KVPair, q *WriteOptions) (bool, *WriteMeta, error) {
 | 
			
		||||
	params := make(map[string]string, 2)
 | 
			
		||||
	if p.Flags != 0 {
 | 
			
		||||
		params["flags"] = strconv.FormatUint(p.Flags, 10)
 | 
			
		||||
	}
 | 
			
		||||
	params["release"] = p.Session
 | 
			
		||||
	return k.put(p.Key, params, p.Value, q)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k *KV) put(key string, params map[string]string, body []byte, q *WriteOptions) (bool, *WriteMeta, error) {
 | 
			
		||||
	r := k.c.newRequest("PUT", "/v1/kv/"+key)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	for param, val := range params {
 | 
			
		||||
		r.params.Set(param, val)
 | 
			
		||||
	}
 | 
			
		||||
	r.body = bytes.NewReader(body)
 | 
			
		||||
	rtt, resp, err := requireOK(k.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return false, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &WriteMeta{}
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var buf bytes.Buffer
 | 
			
		||||
	if _, err := io.Copy(&buf, resp.Body); err != nil {
 | 
			
		||||
		return false, nil, fmt.Errorf("Failed to read response: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	res := strings.Contains(string(buf.Bytes()), "true")
 | 
			
		||||
	return res, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Delete is used to delete a single key
 | 
			
		||||
func (k *KV) Delete(key string, w *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	return k.deleteInternal(key, nil, w)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// DeleteTree is used to delete all keys under a prefix
 | 
			
		||||
func (k *KV) DeleteTree(prefix string, w *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	return k.deleteInternal(prefix, []string{"recurse"}, w)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (k *KV) deleteInternal(key string, params []string, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := k.c.newRequest("DELETE", "/v1/kv/"+key)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	for _, param := range params {
 | 
			
		||||
		r.params.Set(param, "")
 | 
			
		||||
	}
 | 
			
		||||
	rtt, resp, err := requireOK(k.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &WriteMeta{}
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
	return qm, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										204
									
								
								vendor/github.com/armon/consul-api/session.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										204
									
								
								vendor/github.com/armon/consul-api/session.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,204 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// SessionEntry represents a session in consul
 | 
			
		||||
type SessionEntry struct {
 | 
			
		||||
	CreateIndex uint64
 | 
			
		||||
	ID          string
 | 
			
		||||
	Name        string
 | 
			
		||||
	Node        string
 | 
			
		||||
	Checks      []string
 | 
			
		||||
	LockDelay   time.Duration
 | 
			
		||||
	Behavior    string
 | 
			
		||||
	TTL         string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Session can be used to query the Session endpoints
 | 
			
		||||
type Session struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Session returns a handle to the session endpoints
 | 
			
		||||
func (c *Client) Session() *Session {
 | 
			
		||||
	return &Session{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CreateNoChecks is like Create but is used specifically to create
 | 
			
		||||
// a session with no associated health checks.
 | 
			
		||||
func (s *Session) CreateNoChecks(se *SessionEntry, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	body := make(map[string]interface{})
 | 
			
		||||
	body["Checks"] = []string{}
 | 
			
		||||
	if se != nil {
 | 
			
		||||
		if se.Name != "" {
 | 
			
		||||
			body["Name"] = se.Name
 | 
			
		||||
		}
 | 
			
		||||
		if se.Node != "" {
 | 
			
		||||
			body["Node"] = se.Node
 | 
			
		||||
		}
 | 
			
		||||
		if se.LockDelay != 0 {
 | 
			
		||||
			body["LockDelay"] = durToMsec(se.LockDelay)
 | 
			
		||||
		}
 | 
			
		||||
		if se.Behavior != "" {
 | 
			
		||||
			body["Behavior"] = se.Behavior
 | 
			
		||||
		}
 | 
			
		||||
		if se.TTL != "" {
 | 
			
		||||
			body["TTL"] = se.TTL
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return s.create(body, q)
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Create makes a new session. Providing a session entry can
 | 
			
		||||
// customize the session. It can also be nil to use defaults.
 | 
			
		||||
func (s *Session) Create(se *SessionEntry, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	var obj interface{}
 | 
			
		||||
	if se != nil {
 | 
			
		||||
		body := make(map[string]interface{})
 | 
			
		||||
		obj = body
 | 
			
		||||
		if se.Name != "" {
 | 
			
		||||
			body["Name"] = se.Name
 | 
			
		||||
		}
 | 
			
		||||
		if se.Node != "" {
 | 
			
		||||
			body["Node"] = se.Node
 | 
			
		||||
		}
 | 
			
		||||
		if se.LockDelay != 0 {
 | 
			
		||||
			body["LockDelay"] = durToMsec(se.LockDelay)
 | 
			
		||||
		}
 | 
			
		||||
		if len(se.Checks) > 0 {
 | 
			
		||||
			body["Checks"] = se.Checks
 | 
			
		||||
		}
 | 
			
		||||
		if se.Behavior != "" {
 | 
			
		||||
			body["Behavior"] = se.Behavior
 | 
			
		||||
		}
 | 
			
		||||
		if se.TTL != "" {
 | 
			
		||||
			body["TTL"] = se.TTL
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return s.create(obj, q)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *Session) create(obj interface{}, q *WriteOptions) (string, *WriteMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("PUT", "/v1/session/create")
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	r.obj = obj
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	var out struct{ ID string }
 | 
			
		||||
	if err := decodeBody(resp, &out); err != nil {
 | 
			
		||||
		return "", nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return out.ID, wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Destroy invalides a given session
 | 
			
		||||
func (s *Session) Destroy(id string, q *WriteOptions) (*WriteMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("PUT", "/v1/session/destroy/"+id)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
	return wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Renew renews the TTL on a given session
 | 
			
		||||
func (s *Session) Renew(id string, q *WriteOptions) (*SessionEntry, *WriteMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("PUT", "/v1/session/renew/"+id)
 | 
			
		||||
	r.setWriteOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	wm := &WriteMeta{RequestTime: rtt}
 | 
			
		||||
 | 
			
		||||
	var entries []*SessionEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, wm, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(entries) > 0 {
 | 
			
		||||
		return entries[0], wm, nil
 | 
			
		||||
	}
 | 
			
		||||
	return nil, wm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Info looks up a single session
 | 
			
		||||
func (s *Session) Info(id string, q *QueryOptions) (*SessionEntry, *QueryMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("GET", "/v1/session/info/"+id)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*SessionEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(entries) > 0 {
 | 
			
		||||
		return entries[0], qm, nil
 | 
			
		||||
	}
 | 
			
		||||
	return nil, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// List gets sessions for a node
 | 
			
		||||
func (s *Session) Node(node string, q *QueryOptions) ([]*SessionEntry, *QueryMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("GET", "/v1/session/node/"+node)
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*SessionEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// List gets all active sessions
 | 
			
		||||
func (s *Session) List(q *QueryOptions) ([]*SessionEntry, *QueryMeta, error) {
 | 
			
		||||
	r := s.c.newRequest("GET", "/v1/session/list")
 | 
			
		||||
	r.setQueryOptions(q)
 | 
			
		||||
	rtt, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	qm := &QueryMeta{}
 | 
			
		||||
	parseQueryMeta(resp, qm)
 | 
			
		||||
	qm.RequestTime = rtt
 | 
			
		||||
 | 
			
		||||
	var entries []*SessionEntry
 | 
			
		||||
	if err := decodeBody(resp, &entries); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return entries, qm, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										43
									
								
								vendor/github.com/armon/consul-api/status.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								vendor/github.com/armon/consul-api/status.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,43 @@
 | 
			
		||||
package consulapi
 | 
			
		||||
 | 
			
		||||
// Status can be used to query the Status endpoints
 | 
			
		||||
type Status struct {
 | 
			
		||||
	c *Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Status returns a handle to the status endpoints
 | 
			
		||||
func (c *Client) Status() *Status {
 | 
			
		||||
	return &Status{c}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Leader is used to query for a known leader
 | 
			
		||||
func (s *Status) Leader() (string, error) {
 | 
			
		||||
	r := s.c.newRequest("GET", "/v1/status/leader")
 | 
			
		||||
	_, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var leader string
 | 
			
		||||
	if err := decodeBody(resp, &leader); err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return leader, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Peers is used to query for a known raft peers
 | 
			
		||||
func (s *Status) Peers() ([]string, error) {
 | 
			
		||||
	r := s.c.newRequest("GET", "/v1/status/peers")
 | 
			
		||||
	_, resp, err := requireOK(s.c.doRequest(r))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	var peers []string
 | 
			
		||||
	if err := decodeBody(resp, &peers); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return peers, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										37
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										37
									
								
								vendor/github.com/bwmarrin/discordgo/discord.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -13,10 +13,18 @@
 | 
			
		||||
// Package discordgo provides Discord binding for Go
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
import "fmt"
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// VERSION of Discordgo, follows Symantic Versioning. (http://semver.org/)
 | 
			
		||||
const VERSION = "0.15.0"
 | 
			
		||||
// VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/)
 | 
			
		||||
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
 | 
			
		||||
// 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
 | 
			
		||||
//         token, if it is invalid it will sign in with the provided
 | 
			
		||||
//         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) {
 | 
			
		||||
 | 
			
		||||
	// Create an empty Session interface.
 | 
			
		||||
	s = &Session{
 | 
			
		||||
		State:                  NewState(),
 | 
			
		||||
		ratelimiter:            NewRatelimiter(),
 | 
			
		||||
		Ratelimiter:            NewRatelimiter(),
 | 
			
		||||
		StateEnabled:           true,
 | 
			
		||||
		Compress:               true,
 | 
			
		||||
		ShouldReconnectOnError: true,
 | 
			
		||||
		ShardID:                0,
 | 
			
		||||
		ShardCount:             1,
 | 
			
		||||
		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.
 | 
			
		||||
@@ -60,7 +77,7 @@ func New(args ...interface{}) (s *Session, err error) {
 | 
			
		||||
 | 
			
		||||
		case []string:
 | 
			
		||||
			if len(v) > 3 {
 | 
			
		||||
				err = fmt.Errorf("Too many string parameters provided.")
 | 
			
		||||
				err = fmt.Errorf("too many string parameters provided")
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
@@ -91,7 +108,7 @@ func New(args ...interface{}) (s *Session, err error) {
 | 
			
		||||
			} else if s.Token == "" {
 | 
			
		||||
				s.Token = v
 | 
			
		||||
			} else {
 | 
			
		||||
				err = fmt.Errorf("Too many string parameters provided.")
 | 
			
		||||
				err = fmt.Errorf("too many string parameters provided")
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
@@ -99,7 +116,7 @@ func New(args ...interface{}) (s *Session, err error) {
 | 
			
		||||
			// TODO: Parse configuration struct
 | 
			
		||||
 | 
			
		||||
		default:
 | 
			
		||||
			err = fmt.Errorf("Unsupported parameter type provided.")
 | 
			
		||||
			err = fmt.Errorf("unsupported parameter type provided")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
@@ -113,7 +130,11 @@ func New(args ...interface{}) (s *Session, err error) {
 | 
			
		||||
	} else {
 | 
			
		||||
		err = s.Login(auth, pass)
 | 
			
		||||
		if err != nil || s.Token == "" {
 | 
			
		||||
			err = fmt.Errorf("Unable to fetch discord authentication token. %v", err)
 | 
			
		||||
			if s.MFA {
 | 
			
		||||
				err = ErrMFA
 | 
			
		||||
			} else {
 | 
			
		||||
				err = fmt.Errorf("Unable to fetch discord authentication token. %v", err)
 | 
			
		||||
			}
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										59
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										59
									
								
								vendor/github.com/bwmarrin/discordgo/endpoints.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -11,6 +11,9 @@
 | 
			
		||||
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
// APIVersion is the Discord API version used for the REST and Websocket API.
 | 
			
		||||
var APIVersion = "6"
 | 
			
		||||
 | 
			
		||||
// Known Discord API Endpoints.
 | 
			
		||||
var (
 | 
			
		||||
	EndpointStatus     = "https://status.discordapp.com/api/v2/"
 | 
			
		||||
@@ -18,13 +21,21 @@ var (
 | 
			
		||||
	EndpointSmActive   = EndpointSm + "active.json"
 | 
			
		||||
	EndpointSmUpcoming = EndpointSm + "upcoming.json"
 | 
			
		||||
 | 
			
		||||
	EndpointDiscord  = "https://discordapp.com/"
 | 
			
		||||
	EndpointAPI      = EndpointDiscord + "api/"
 | 
			
		||||
	EndpointGuilds   = EndpointAPI + "guilds/"
 | 
			
		||||
	EndpointChannels = EndpointAPI + "channels/"
 | 
			
		||||
	EndpointUsers    = EndpointAPI + "users/"
 | 
			
		||||
	EndpointGateway  = EndpointAPI + "gateway"
 | 
			
		||||
	EndpointWebhooks = EndpointAPI + "webhooks/"
 | 
			
		||||
	EndpointDiscord    = "https://discordapp.com/"
 | 
			
		||||
	EndpointAPI        = EndpointDiscord + "api/v" + APIVersion + "/"
 | 
			
		||||
	EndpointGuilds     = EndpointAPI + "guilds/"
 | 
			
		||||
	EndpointChannels   = EndpointAPI + "channels/"
 | 
			
		||||
	EndpointUsers      = EndpointAPI + "users/"
 | 
			
		||||
	EndpointGateway    = EndpointAPI + "gateway"
 | 
			
		||||
	EndpointGatewayBot = EndpointGateway + "/bot"
 | 
			
		||||
	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/"
 | 
			
		||||
	EndpointLogin          = EndpointAuth + "login"
 | 
			
		||||
@@ -47,18 +58,19 @@ var (
 | 
			
		||||
	EndpointReport       = EndpointAPI + "report"
 | 
			
		||||
	EndpointIntegrations = EndpointAPI + "integrations"
 | 
			
		||||
 | 
			
		||||
	EndpointUser              = func(uID string) string { return EndpointUsers + uID }
 | 
			
		||||
	EndpointUserAvatar        = func(uID, aID string) string { return EndpointUsers + uID + "/avatars/" + aID + ".jpg" }
 | 
			
		||||
	EndpointUserSettings      = func(uID string) string { return EndpointUsers + uID + "/settings" }
 | 
			
		||||
	EndpointUserGuilds        = func(uID string) string { return EndpointUsers + uID + "/guilds" }
 | 
			
		||||
	EndpointUserGuild         = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID }
 | 
			
		||||
	EndpointUserGuildSettings = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID + "/settings" }
 | 
			
		||||
	EndpointUserChannels      = func(uID string) string { return EndpointUsers + uID + "/channels" }
 | 
			
		||||
	EndpointUserDevices       = func(uID string) string { return EndpointUsers + uID + "/devices" }
 | 
			
		||||
	EndpointUserConnections   = func(uID string) string { return EndpointUsers + uID + "/connections" }
 | 
			
		||||
	EndpointUser               = func(uID string) string { return EndpointUsers + uID }
 | 
			
		||||
	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" }
 | 
			
		||||
	EndpointUserGuilds         = func(uID string) string { return EndpointUsers + uID + "/guilds" }
 | 
			
		||||
	EndpointUserGuild          = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID }
 | 
			
		||||
	EndpointUserGuildSettings  = func(uID, gID string) string { return EndpointUsers + uID + "/guilds/" + gID + "/settings" }
 | 
			
		||||
	EndpointUserChannels       = func(uID string) string { return EndpointUsers + uID + "/channels" }
 | 
			
		||||
	EndpointUserDevices        = func(uID string) string { return EndpointUsers + uID + "/devices" }
 | 
			
		||||
	EndpointUserConnections    = func(uID string) string { return EndpointUsers + uID + "/connections" }
 | 
			
		||||
	EndpointUserNotes          = func(uID string) string { return EndpointUsers + "@me/notes/" + uID }
 | 
			
		||||
 | 
			
		||||
	EndpointGuild                = func(gID string) string { return EndpointGuilds + gID }
 | 
			
		||||
	EndpointGuildInivtes         = func(gID string) string { return EndpointGuilds + gID + "/invites" }
 | 
			
		||||
	EndpointGuildChannels        = func(gID string) string { return EndpointGuilds + gID + "/channels" }
 | 
			
		||||
	EndpointGuildMembers         = func(gID string) string { return EndpointGuilds + gID + "/members" }
 | 
			
		||||
	EndpointGuildMember          = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID }
 | 
			
		||||
@@ -73,8 +85,8 @@ var (
 | 
			
		||||
	EndpointGuildInvites         = func(gID string) string { return EndpointGuilds + gID + "/invites" }
 | 
			
		||||
	EndpointGuildEmbed           = func(gID string) string { return EndpointGuilds + gID + "/embed" }
 | 
			
		||||
	EndpointGuildPrune           = func(gID string) string { return EndpointGuilds + gID + "/prune" }
 | 
			
		||||
	EndpointGuildIcon            = func(gID, hash string) string { return EndpointGuilds + gID + "/icons/" + hash + ".jpg" }
 | 
			
		||||
	EndpointGuildSplash          = func(gID, hash string) string { return EndpointGuilds + gID + "/splashes/" + hash + ".jpg" }
 | 
			
		||||
	EndpointGuildIcon            = func(gID, hash string) string { return EndpointCDNIcons + gID + "/" + hash + ".png" }
 | 
			
		||||
	EndpointGuildSplash          = func(gID, hash string) string { return EndpointCDNSplashes + gID + "/" + hash + ".png" }
 | 
			
		||||
	EndpointGuildWebhooks        = func(gID string) string { return EndpointGuilds + gID + "/webhooks" }
 | 
			
		||||
 | 
			
		||||
	EndpointChannel                   = func(cID string) string { return EndpointChannels + cID }
 | 
			
		||||
@@ -85,14 +97,19 @@ var (
 | 
			
		||||
	EndpointChannelMessages           = func(cID string) string { return EndpointChannels + cID + "/messages" }
 | 
			
		||||
	EndpointChannelMessage            = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID }
 | 
			
		||||
	EndpointChannelMessageAck         = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" }
 | 
			
		||||
	EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk_delete" }
 | 
			
		||||
	EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk-delete" }
 | 
			
		||||
	EndpointChannelMessagesPins       = func(cID string) string { return EndpointChannel(cID) + "/pins" }
 | 
			
		||||
	EndpointChannelMessagePin         = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID }
 | 
			
		||||
 | 
			
		||||
	EndpointGroupIcon = func(cID, hash string) string { return EndpointCDNChannelIcons + cID + "/" + hash + ".png" }
 | 
			
		||||
 | 
			
		||||
	EndpointChannelWebhooks = func(cID string) string { return EndpointChannel(cID) + "/webhooks" }
 | 
			
		||||
	EndpointWebhook         = func(wID string) string { return EndpointWebhooks + wID }
 | 
			
		||||
	EndpointWebhookToken    = func(wID, token string) string { return EndpointWebhooks + wID + "/" + token }
 | 
			
		||||
 | 
			
		||||
	EndpointMessageReactionsAll = func(cID, mID string) string {
 | 
			
		||||
		return EndpointChannelMessage(cID, mID) + "/reactions"
 | 
			
		||||
	}
 | 
			
		||||
	EndpointMessageReactions = func(cID, mID, eID string) string {
 | 
			
		||||
		return EndpointChannelMessage(cID, mID) + "/reactions/" + eID
 | 
			
		||||
	}
 | 
			
		||||
@@ -104,6 +121,8 @@ var (
 | 
			
		||||
	EndpointRelationship        = func(uID string) string { return EndpointRelationships() + "/" + uID }
 | 
			
		||||
	EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" }
 | 
			
		||||
 | 
			
		||||
	EndpointGuildCreate = EndpointAPI + "guilds"
 | 
			
		||||
 | 
			
		||||
	EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID }
 | 
			
		||||
 | 
			
		||||
	EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" }
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										42
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										42
									
								
								vendor/github.com/bwmarrin/discordgo/event.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -1,14 +1,12 @@
 | 
			
		||||
package discordgo
 | 
			
		||||
 | 
			
		||||
import "fmt"
 | 
			
		||||
 | 
			
		||||
// EventHandler is an interface for Discord events.
 | 
			
		||||
type EventHandler interface {
 | 
			
		||||
	// Type returns the type of event this handler belongs to.
 | 
			
		||||
	Type() string
 | 
			
		||||
 | 
			
		||||
	// Handle is called whenever an event of Type() happens.
 | 
			
		||||
	// It is the recievers responsibility to type assert that the interface
 | 
			
		||||
	// It is the receivers responsibility to type assert that the interface
 | 
			
		||||
	// is the expected struct.
 | 
			
		||||
	Handle(*Session, interface{})
 | 
			
		||||
}
 | 
			
		||||
@@ -45,12 +43,15 @@ var registeredInterfaceProviders = map[string]EventInterfaceProvider{}
 | 
			
		||||
 | 
			
		||||
// registerInterfaceProvider registers a provider so that DiscordGo can
 | 
			
		||||
// access it's New() method.
 | 
			
		||||
func registerInterfaceProvider(eh EventInterfaceProvider) error {
 | 
			
		||||
func registerInterfaceProvider(eh EventInterfaceProvider) {
 | 
			
		||||
	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
 | 
			
		||||
	return nil
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// eventHandlerInstance is a wrapper around an event handler, as functions
 | 
			
		||||
@@ -155,12 +156,20 @@ func (s *Session) removeEventHandlerInstance(t string, ehi *eventHandlerInstance
 | 
			
		||||
// Handles calling permanent and once handlers for an event type.
 | 
			
		||||
func (s *Session) handle(t string, i interface{}) {
 | 
			
		||||
	for _, eh := range s.handlers[t] {
 | 
			
		||||
		go eh.eventHandler.Handle(s, i)
 | 
			
		||||
		if s.SyncEvents {
 | 
			
		||||
			eh.eventHandler.Handle(s, i)
 | 
			
		||||
		} else {
 | 
			
		||||
			go eh.eventHandler.Handle(s, i)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(s.onceHandlers[t]) > 0 {
 | 
			
		||||
		for _, eh := range s.onceHandlers[t] {
 | 
			
		||||
			go eh.eventHandler.Handle(s, i)
 | 
			
		||||
			if s.SyncEvents {
 | 
			
		||||
				eh.eventHandler.Handle(s, i)
 | 
			
		||||
			} else {
 | 
			
		||||
				go eh.eventHandler.Handle(s, i)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		s.onceHandlers[t] = nil
 | 
			
		||||
	}
 | 
			
		||||
@@ -210,14 +219,15 @@ func (s *Session) onInterface(i interface{}) {
 | 
			
		||||
		setGuildIds(t.Guild)
 | 
			
		||||
	case *GuildUpdate:
 | 
			
		||||
		setGuildIds(t.Guild)
 | 
			
		||||
	case *Resumed:
 | 
			
		||||
		s.onResumed(t)
 | 
			
		||||
	case *VoiceServerUpdate:
 | 
			
		||||
		go s.onVoiceServerUpdate(t)
 | 
			
		||||
	case *VoiceStateUpdate:
 | 
			
		||||
		go s.onVoiceStateUpdate(t)
 | 
			
		||||
	}
 | 
			
		||||
	s.State.onInterface(s, i)
 | 
			
		||||
	err := s.State.OnInterface(s, i)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		s.log(LogDebug, "error dispatching internal event, %s", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onReady handles the ready event.
 | 
			
		||||
@@ -225,14 +235,4 @@ func (s *Session) onReady(r *Ready) {
 | 
			
		||||
 | 
			
		||||
	// Store the SessionID within the Session struct.
 | 
			
		||||
	s.sessionID = r.SessionID
 | 
			
		||||
 | 
			
		||||
	// Start the heartbeat to keep the connection alive.
 | 
			
		||||
	go s.heartbeat(s.wsConn, s.listening, r.HeartbeatInterval)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onResumed handles the resumed event.
 | 
			
		||||
func (s *Session) onResumed(r *Resumed) {
 | 
			
		||||
 | 
			
		||||
	// Start the heartbeat to keep the connection alive.
 | 
			
		||||
	go s.heartbeat(s.wsConn, s.listening, r.HeartbeatInterval)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										173
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										173
									
								
								vendor/github.com/bwmarrin/discordgo/eventhandlers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -7,46 +7,49 @@ package discordgo
 | 
			
		||||
// Event type values are used to match the events returned by Discord.
 | 
			
		||||
// EventTypes surrounded by __ are synthetic and are internal to DiscordGo.
 | 
			
		||||
const (
 | 
			
		||||
	channelCreateEventType           = "CHANNEL_CREATE"
 | 
			
		||||
	channelDeleteEventType           = "CHANNEL_DELETE"
 | 
			
		||||
	channelPinsUpdateEventType       = "CHANNEL_PINS_UPDATE"
 | 
			
		||||
	channelUpdateEventType           = "CHANNEL_UPDATE"
 | 
			
		||||
	connectEventType                 = "__CONNECT__"
 | 
			
		||||
	disconnectEventType              = "__DISCONNECT__"
 | 
			
		||||
	eventEventType                   = "__EVENT__"
 | 
			
		||||
	guildBanAddEventType             = "GUILD_BAN_ADD"
 | 
			
		||||
	guildBanRemoveEventType          = "GUILD_BAN_REMOVE"
 | 
			
		||||
	guildCreateEventType             = "GUILD_CREATE"
 | 
			
		||||
	guildDeleteEventType             = "GUILD_DELETE"
 | 
			
		||||
	guildEmojisUpdateEventType       = "GUILD_EMOJIS_UPDATE"
 | 
			
		||||
	guildIntegrationsUpdateEventType = "GUILD_INTEGRATIONS_UPDATE"
 | 
			
		||||
	guildMemberAddEventType          = "GUILD_MEMBER_ADD"
 | 
			
		||||
	guildMemberRemoveEventType       = "GUILD_MEMBER_REMOVE"
 | 
			
		||||
	guildMemberUpdateEventType       = "GUILD_MEMBER_UPDATE"
 | 
			
		||||
	guildMembersChunkEventType       = "GUILD_MEMBERS_CHUNK"
 | 
			
		||||
	guildRoleCreateEventType         = "GUILD_ROLE_CREATE"
 | 
			
		||||
	guildRoleDeleteEventType         = "GUILD_ROLE_DELETE"
 | 
			
		||||
	guildRoleUpdateEventType         = "GUILD_ROLE_UPDATE"
 | 
			
		||||
	guildUpdateEventType             = "GUILD_UPDATE"
 | 
			
		||||
	messageAckEventType              = "MESSAGE_ACK"
 | 
			
		||||
	messageCreateEventType           = "MESSAGE_CREATE"
 | 
			
		||||
	messageDeleteEventType           = "MESSAGE_DELETE"
 | 
			
		||||
	messageReactionAddEventType      = "MESSAGE_REACTION_ADD"
 | 
			
		||||
	messageReactionRemoveEventType   = "MESSAGE_REACTION_REMOVE"
 | 
			
		||||
	messageUpdateEventType           = "MESSAGE_UPDATE"
 | 
			
		||||
	presenceUpdateEventType          = "PRESENCE_UPDATE"
 | 
			
		||||
	presencesReplaceEventType        = "PRESENCES_REPLACE"
 | 
			
		||||
	rateLimitEventType               = "__RATE_LIMIT__"
 | 
			
		||||
	readyEventType                   = "READY"
 | 
			
		||||
	relationshipAddEventType         = "RELATIONSHIP_ADD"
 | 
			
		||||
	relationshipRemoveEventType      = "RELATIONSHIP_REMOVE"
 | 
			
		||||
	resumedEventType                 = "RESUMED"
 | 
			
		||||
	typingStartEventType             = "TYPING_START"
 | 
			
		||||
	userGuildSettingsUpdateEventType = "USER_GUILD_SETTINGS_UPDATE"
 | 
			
		||||
	userSettingsUpdateEventType      = "USER_SETTINGS_UPDATE"
 | 
			
		||||
	userUpdateEventType              = "USER_UPDATE"
 | 
			
		||||
	voiceServerUpdateEventType       = "VOICE_SERVER_UPDATE"
 | 
			
		||||
	voiceStateUpdateEventType        = "VOICE_STATE_UPDATE"
 | 
			
		||||
	channelCreateEventType            = "CHANNEL_CREATE"
 | 
			
		||||
	channelDeleteEventType            = "CHANNEL_DELETE"
 | 
			
		||||
	channelPinsUpdateEventType        = "CHANNEL_PINS_UPDATE"
 | 
			
		||||
	channelUpdateEventType            = "CHANNEL_UPDATE"
 | 
			
		||||
	connectEventType                  = "__CONNECT__"
 | 
			
		||||
	disconnectEventType               = "__DISCONNECT__"
 | 
			
		||||
	eventEventType                    = "__EVENT__"
 | 
			
		||||
	guildBanAddEventType              = "GUILD_BAN_ADD"
 | 
			
		||||
	guildBanRemoveEventType           = "GUILD_BAN_REMOVE"
 | 
			
		||||
	guildCreateEventType              = "GUILD_CREATE"
 | 
			
		||||
	guildDeleteEventType              = "GUILD_DELETE"
 | 
			
		||||
	guildEmojisUpdateEventType        = "GUILD_EMOJIS_UPDATE"
 | 
			
		||||
	guildIntegrationsUpdateEventType  = "GUILD_INTEGRATIONS_UPDATE"
 | 
			
		||||
	guildMemberAddEventType           = "GUILD_MEMBER_ADD"
 | 
			
		||||
	guildMemberRemoveEventType        = "GUILD_MEMBER_REMOVE"
 | 
			
		||||
	guildMemberUpdateEventType        = "GUILD_MEMBER_UPDATE"
 | 
			
		||||
	guildMembersChunkEventType        = "GUILD_MEMBERS_CHUNK"
 | 
			
		||||
	guildRoleCreateEventType          = "GUILD_ROLE_CREATE"
 | 
			
		||||
	guildRoleDeleteEventType          = "GUILD_ROLE_DELETE"
 | 
			
		||||
	guildRoleUpdateEventType          = "GUILD_ROLE_UPDATE"
 | 
			
		||||
	guildUpdateEventType              = "GUILD_UPDATE"
 | 
			
		||||
	messageAckEventType               = "MESSAGE_ACK"
 | 
			
		||||
	messageCreateEventType            = "MESSAGE_CREATE"
 | 
			
		||||
	messageDeleteEventType            = "MESSAGE_DELETE"
 | 
			
		||||
	messageDeleteBulkEventType        = "MESSAGE_DELETE_BULK"
 | 
			
		||||
	messageReactionAddEventType       = "MESSAGE_REACTION_ADD"
 | 
			
		||||
	messageReactionRemoveEventType    = "MESSAGE_REACTION_REMOVE"
 | 
			
		||||
	messageReactionRemoveAllEventType = "MESSAGE_REACTION_REMOVE_ALL"
 | 
			
		||||
	messageUpdateEventType            = "MESSAGE_UPDATE"
 | 
			
		||||
	presenceUpdateEventType           = "PRESENCE_UPDATE"
 | 
			
		||||
	presencesReplaceEventType         = "PRESENCES_REPLACE"
 | 
			
		||||
	rateLimitEventType                = "__RATE_LIMIT__"
 | 
			
		||||
	readyEventType                    = "READY"
 | 
			
		||||
	relationshipAddEventType          = "RELATIONSHIP_ADD"
 | 
			
		||||
	relationshipRemoveEventType       = "RELATIONSHIP_REMOVE"
 | 
			
		||||
	resumedEventType                  = "RESUMED"
 | 
			
		||||
	typingStartEventType              = "TYPING_START"
 | 
			
		||||
	userGuildSettingsUpdateEventType  = "USER_GUILD_SETTINGS_UPDATE"
 | 
			
		||||
	userNoteUpdateEventType           = "USER_NOTE_UPDATE"
 | 
			
		||||
	userSettingsUpdateEventType       = "USER_SETTINGS_UPDATE"
 | 
			
		||||
	userUpdateEventType               = "USER_UPDATE"
 | 
			
		||||
	voiceServerUpdateEventType        = "VOICE_SERVER_UPDATE"
 | 
			
		||||
	voiceStateUpdateEventType         = "VOICE_STATE_UPDATE"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// channelCreateEventHandler is an event handler for ChannelCreate events.
 | 
			
		||||
@@ -137,11 +140,6 @@ func (eh connectEventHandler) Type() string {
 | 
			
		||||
	return connectEventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New returns a new instance of Connect.
 | 
			
		||||
func (eh connectEventHandler) New() interface{} {
 | 
			
		||||
	return &Connect{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Handle is the handler for Connect events.
 | 
			
		||||
func (eh connectEventHandler) Handle(s *Session, i interface{}) {
 | 
			
		||||
	if t, ok := i.(*Connect); ok {
 | 
			
		||||
@@ -157,11 +155,6 @@ func (eh disconnectEventHandler) Type() string {
 | 
			
		||||
	return disconnectEventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New returns a new instance of Disconnect.
 | 
			
		||||
func (eh disconnectEventHandler) New() interface{} {
 | 
			
		||||
	return &Disconnect{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Handle is the handler for Disconnect events.
 | 
			
		||||
func (eh disconnectEventHandler) Handle(s *Session, i interface{}) {
 | 
			
		||||
	if t, ok := i.(*Disconnect); ok {
 | 
			
		||||
@@ -177,11 +170,6 @@ func (eh eventEventHandler) Type() string {
 | 
			
		||||
	return eventEventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New returns a new instance of Event.
 | 
			
		||||
func (eh eventEventHandler) New() interface{} {
 | 
			
		||||
	return &Event{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Handle is the handler for Event events.
 | 
			
		||||
func (eh eventEventHandler) Handle(s *Session, i interface{}) {
 | 
			
		||||
	if t, ok := i.(*Event); ok {
 | 
			
		||||
@@ -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.
 | 
			
		||||
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.
 | 
			
		||||
type messageUpdateEventHandler func(*Session, *MessageUpdate)
 | 
			
		||||
 | 
			
		||||
@@ -637,11 +665,6 @@ func (eh rateLimitEventHandler) Type() string {
 | 
			
		||||
	return rateLimitEventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// New returns a new instance of RateLimit.
 | 
			
		||||
func (eh rateLimitEventHandler) New() interface{} {
 | 
			
		||||
	return &RateLimit{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Handle is the handler for RateLimit events.
 | 
			
		||||
func (eh rateLimitEventHandler) Handle(s *Session, i interface{}) {
 | 
			
		||||
	if t, ok := i.(*RateLimit); ok {
 | 
			
		||||
@@ -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.
 | 
			
		||||
type userSettingsUpdateEventHandler func(*Session, *UserSettingsUpdate)
 | 
			
		||||
 | 
			
		||||
@@ -901,10 +944,14 @@ func handlerForInterface(handler interface{}) EventHandler {
 | 
			
		||||
		return messageCreateEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageDelete):
 | 
			
		||||
		return messageDeleteEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageDeleteBulk):
 | 
			
		||||
		return messageDeleteBulkEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageReactionAdd):
 | 
			
		||||
		return messageReactionAddEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageReactionRemove):
 | 
			
		||||
		return messageReactionRemoveEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageReactionRemoveAll):
 | 
			
		||||
		return messageReactionRemoveAllEventHandler(v)
 | 
			
		||||
	case func(*Session, *MessageUpdate):
 | 
			
		||||
		return messageUpdateEventHandler(v)
 | 
			
		||||
	case func(*Session, *PresenceUpdate):
 | 
			
		||||
@@ -925,6 +972,8 @@ func handlerForInterface(handler interface{}) EventHandler {
 | 
			
		||||
		return typingStartEventHandler(v)
 | 
			
		||||
	case func(*Session, *UserGuildSettingsUpdate):
 | 
			
		||||
		return userGuildSettingsUpdateEventHandler(v)
 | 
			
		||||
	case func(*Session, *UserNoteUpdate):
 | 
			
		||||
		return userNoteUpdateEventHandler(v)
 | 
			
		||||
	case func(*Session, *UserSettingsUpdate):
 | 
			
		||||
		return userSettingsUpdateEventHandler(v)
 | 
			
		||||
	case func(*Session, *UserUpdate):
 | 
			
		||||
@@ -937,6 +986,7 @@ func handlerForInterface(handler interface{}) EventHandler {
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	registerInterfaceProvider(channelCreateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(channelDeleteEventHandler(nil))
 | 
			
		||||
@@ -959,8 +1009,10 @@ func init() {
 | 
			
		||||
	registerInterfaceProvider(messageAckEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageCreateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageDeleteEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageDeleteBulkEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageReactionAddEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageReactionRemoveEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageReactionRemoveAllEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(messageUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(presenceUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(presencesReplaceEventHandler(nil))
 | 
			
		||||
@@ -970,6 +1022,7 @@ func init() {
 | 
			
		||||
	registerInterfaceProvider(resumedEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(typingStartEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(userGuildSettingsUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(userNoteUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(userSettingsUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(userUpdateEventHandler(nil))
 | 
			
		||||
	registerInterfaceProvider(voiceServerUpdateEventHandler(nil))
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										37
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										37
									
								
								vendor/github.com/bwmarrin/discordgo/events.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -2,7 +2,6 @@ package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// 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.
 | 
			
		||||
type Event struct {
 | 
			
		||||
	Operation int             `json:"op"`
 | 
			
		||||
	Sequence  int             `json:"s"`
 | 
			
		||||
	Sequence  int64           `json:"s"`
 | 
			
		||||
	Type      string          `json:"t"`
 | 
			
		||||
	RawData   json.RawMessage `json:"d"`
 | 
			
		||||
	// Struct contains one of the other types in this file.
 | 
			
		||||
@@ -37,19 +36,19 @@ type Event struct {
 | 
			
		||||
 | 
			
		||||
// A Ready stores all data for the websocket READY event.
 | 
			
		||||
type Ready struct {
 | 
			
		||||
	Version           int           `json:"v"`
 | 
			
		||||
	SessionID         string        `json:"session_id"`
 | 
			
		||||
	HeartbeatInterval time.Duration `json:"heartbeat_interval"`
 | 
			
		||||
	User              *User         `json:"user"`
 | 
			
		||||
	ReadState         []*ReadState  `json:"read_state"`
 | 
			
		||||
	PrivateChannels   []*Channel    `json:"private_channels"`
 | 
			
		||||
	Guilds            []*Guild      `json:"guilds"`
 | 
			
		||||
	Version         int          `json:"v"`
 | 
			
		||||
	SessionID       string       `json:"session_id"`
 | 
			
		||||
	User            *User        `json:"user"`
 | 
			
		||||
	ReadState       []*ReadState `json:"read_state"`
 | 
			
		||||
	PrivateChannels []*Channel   `json:"private_channels"`
 | 
			
		||||
	Guilds          []*Guild     `json:"guilds"`
 | 
			
		||||
 | 
			
		||||
	// Undocumented fields
 | 
			
		||||
	Settings          *Settings            `json:"user_settings"`
 | 
			
		||||
	UserGuildSettings []*UserGuildSettings `json:"user_guild_settings"`
 | 
			
		||||
	Relationships     []*Relationship      `json:"relationships"`
 | 
			
		||||
	Presences         []*Presence          `json:"presences"`
 | 
			
		||||
	Notes             map[string]string    `json:"notes"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelCreate is the data for a ChannelCreate event.
 | 
			
		||||
@@ -179,6 +178,11 @@ type MessageReactionRemove struct {
 | 
			
		||||
	*MessageReaction
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// MessageReactionRemoveAll is the data for a MessageReactionRemoveAll event.
 | 
			
		||||
type MessageReactionRemoveAll struct {
 | 
			
		||||
	*MessageReaction
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// PresencesReplace is the data for a PresencesReplace event.
 | 
			
		||||
type PresencesReplace []*Presence
 | 
			
		||||
 | 
			
		||||
@@ -191,8 +195,7 @@ type PresenceUpdate struct {
 | 
			
		||||
 | 
			
		||||
// Resumed is the data for a Resumed event.
 | 
			
		||||
type Resumed struct {
 | 
			
		||||
	HeartbeatInterval time.Duration `json:"heartbeat_interval"`
 | 
			
		||||
	Trace             []string      `json:"_trace"`
 | 
			
		||||
	Trace []string `json:"_trace"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// RelationshipAdd is the data for a RelationshipAdd event.
 | 
			
		||||
@@ -225,6 +228,12 @@ type UserGuildSettingsUpdate struct {
 | 
			
		||||
	*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.
 | 
			
		||||
type VoiceServerUpdate struct {
 | 
			
		||||
	Token    string `json:"token"`
 | 
			
		||||
@@ -236,3 +245,9 @@ type VoiceServerUpdate struct {
 | 
			
		||||
type VoiceStateUpdate struct {
 | 
			
		||||
	*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"
 | 
			
		||||
	"io"
 | 
			
		||||
	"os"
 | 
			
		||||
	"os/signal"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"syscall"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
@@ -21,6 +23,7 @@ var token string
 | 
			
		||||
var buffer = make([][]byte, 0)
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	if token == "" {
 | 
			
		||||
		fmt.Println("No token provided. Please run: airhorn -t <bot token>")
 | 
			
		||||
		return
 | 
			
		||||
@@ -56,21 +59,37 @@ func main() {
 | 
			
		||||
		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.")
 | 
			
		||||
	// Simple way to keep program running until CTRL-C is pressed.
 | 
			
		||||
	<-make(chan struct{})
 | 
			
		||||
	return
 | 
			
		||||
	sc := make(chan os.Signal, 1)
 | 
			
		||||
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
 | 
			
		||||
	<-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) {
 | 
			
		||||
 | 
			
		||||
	// 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
 | 
			
		||||
// message is created on any channel that the autenticated bot has access to.
 | 
			
		||||
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
 | 
			
		||||
	// Ignore all messages created by the bot itself
 | 
			
		||||
	// 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") {
 | 
			
		||||
 | 
			
		||||
		// Find the channel that the message came from.
 | 
			
		||||
		c, err := s.State.Channel(m.ChannelID)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
@@ -85,7 +104,7 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 | 
			
		||||
			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 {
 | 
			
		||||
			if vs.UserID == m.Author.ID {
 | 
			
		||||
				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
 | 
			
		||||
// guild is joined.
 | 
			
		||||
func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) {
 | 
			
		||||
 | 
			
		||||
	if event.Guild.Unavailable {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -116,8 +136,8 @@ func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) {
 | 
			
		||||
 | 
			
		||||
// loadSound attempts to load an encoded sound file from disk.
 | 
			
		||||
func loadSound() error {
 | 
			
		||||
	file, err := os.Open("airhorn.dca")
 | 
			
		||||
 | 
			
		||||
	file, err := os.Open("airhorn.dca")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("Error opening dca file :", err)
 | 
			
		||||
		return err
 | 
			
		||||
@@ -131,7 +151,7 @@ func loadSound() error {
 | 
			
		||||
 | 
			
		||||
		// If this is the end of the file, just return.
 | 
			
		||||
		if err == io.EOF || err == io.ErrUnexpectedEOF {
 | 
			
		||||
			file.Close()
 | 
			
		||||
			err := file.Close()
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
@@ -160,6 +180,7 @@ func loadSound() error {
 | 
			
		||||
 | 
			
		||||
// playSound plays the current buffer to the provided channel.
 | 
			
		||||
func playSound(s *discordgo.Session, guildID, channelID string) (err error) {
 | 
			
		||||
 | 
			
		||||
	// Join the provided voice channel.
 | 
			
		||||
	vc, err := s.ChannelVoiceJoin(guildID, channelID, false, true)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -170,7 +191,7 @@ func playSound(s *discordgo.Session, guildID, channelID string) (err error) {
 | 
			
		||||
	time.Sleep(250 * time.Millisecond)
 | 
			
		||||
 | 
			
		||||
	// Start speaking.
 | 
			
		||||
	_ = vc.Speaking(true)
 | 
			
		||||
	vc.Speaking(true)
 | 
			
		||||
 | 
			
		||||
	// Send the buffer data.
 | 
			
		||||
	for _, buff := range buffer {
 | 
			
		||||
@@ -178,13 +199,13 @@ func playSound(s *discordgo.Session, guildID, channelID string) (err error) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Stop speaking
 | 
			
		||||
	_ = vc.Speaking(false)
 | 
			
		||||
	vc.Speaking(false)
 | 
			
		||||
 | 
			
		||||
	// Sleep for a specificed amount of time before ending.
 | 
			
		||||
	time.Sleep(250 * time.Millisecond)
 | 
			
		||||
 | 
			
		||||
	// Disconnect from the provided voice channel.
 | 
			
		||||
	_ = vc.Disconnect()
 | 
			
		||||
	vc.Disconnect()
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Variables used for command line options
 | 
			
		||||
var (
 | 
			
		||||
	Email    string
 | 
			
		||||
	Password string
 | 
			
		||||
	Token    string
 | 
			
		||||
	AppName  string
 | 
			
		||||
	Name     string
 | 
			
		||||
	DeleteID string
 | 
			
		||||
	ListOnly bool
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Account Token")
 | 
			
		||||
	flag.StringVar(&Token, "t", "", "Owner Account Token")
 | 
			
		||||
	flag.StringVar(&Name, "n", "", "Name to give App/Bot")
 | 
			
		||||
	flag.StringVar(&DeleteID, "d", "", "Application ID to delete")
 | 
			
		||||
	flag.BoolVar(&ListOnly, "l", false, "List Applications Only")
 | 
			
		||||
	flag.StringVar(&AppName, "a", "", "App/Bot Name")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	if Token == "" {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	// 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 {
 | 
			
		||||
		fmt.Println("error creating Discord session,", err)
 | 
			
		||||
		return
 | 
			
		||||
@@ -41,18 +45,17 @@ func main() {
 | 
			
		||||
	// If -l set, only display a list of existing applications
 | 
			
		||||
	// for the given account.
 | 
			
		||||
	if ListOnly {
 | 
			
		||||
		aps, err2 := dg.Applications()
 | 
			
		||||
		if err2 != nil {
 | 
			
		||||
 | 
			
		||||
		aps, err := dg.Applications()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			fmt.Println("error fetching applications,", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		for k, v := range aps {
 | 
			
		||||
			fmt.Printf("%d : --------------------------------------\n", k)
 | 
			
		||||
			fmt.Printf("ID: %s\n", v.ID)
 | 
			
		||||
			fmt.Printf("Name: %s\n", v.Name)
 | 
			
		||||
			fmt.Printf("Secret: %s\n", v.Secret)
 | 
			
		||||
			fmt.Printf("Description: %s\n", v.Description)
 | 
			
		||||
		for _, v := range aps {
 | 
			
		||||
			fmt.Println("-----------------------------------------------------")
 | 
			
		||||
			b, _ := json.MarshalIndent(v, "", " ")
 | 
			
		||||
			fmt.Println(string(b))
 | 
			
		||||
		}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -66,19 +69,23 @@ func main() {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if Name == "" {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Create a new application.
 | 
			
		||||
	ap := &discordgo.Application{}
 | 
			
		||||
	ap.Name = AppName
 | 
			
		||||
	ap.Name = Name
 | 
			
		||||
	ap, err = dg.ApplicationCreate(ap)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error creating new applicaiton,", err)
 | 
			
		||||
		fmt.Println("error creating new application,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Application created successfully:\n")
 | 
			
		||||
	fmt.Printf("ID: %s\n", ap.ID)
 | 
			
		||||
	fmt.Printf("Name: %s\n", ap.Name)
 | 
			
		||||
	fmt.Printf("Secret: %s\n\n", ap.Secret)
 | 
			
		||||
	b, _ := json.MarshalIndent(ap, "", " ")
 | 
			
		||||
	fmt.Println(string(b))
 | 
			
		||||
 | 
			
		||||
	// Create the bot account under the application we just created
 | 
			
		||||
	bot, err := dg.ApplicationBotCreate(ap.ID)
 | 
			
		||||
@@ -88,11 +95,9 @@ func main() {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Printf("Bot account created successfully.\n")
 | 
			
		||||
	fmt.Printf("ID: %s\n", bot.ID)
 | 
			
		||||
	fmt.Printf("Username: %s\n", bot.Username)
 | 
			
		||||
	fmt.Printf("Token: %s\n\n", bot.Token)
 | 
			
		||||
	b, _ = json.MarshalIndent(bot, "", " ")
 | 
			
		||||
	fmt.Println(string(b))
 | 
			
		||||
 | 
			
		||||
	fmt.Println("Please save the above posted info in a secure place.")
 | 
			
		||||
	fmt.Println("You will need that information to login with your bot account.")
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										73
									
								
								vendor/github.com/bwmarrin/discordgo/examples/avatar/localfile/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										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 (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
@@ -18,6 +19,11 @@ func init() {
 | 
			
		||||
	flag.StringVar(&Email, "e", "", "Account Email")
 | 
			
		||||
	flag.StringVar(&Password, "p", "", "Account Password")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	if Email == "" || Password == "" {
 | 
			
		||||
		flag.Usage()
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
@@ -29,5 +35,6 @@ func main() {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Print out your 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 (
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"os/signal"
 | 
			
		||||
	"syscall"
 | 
			
		||||
 | 
			
		||||
	"github.com/bwmarrin/discordgo"
 | 
			
		||||
)
 | 
			
		||||
@@ -10,7 +13,6 @@ import (
 | 
			
		||||
// Variables used for command line parameters
 | 
			
		||||
var (
 | 
			
		||||
	Token string
 | 
			
		||||
	BotID string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
@@ -28,29 +30,24 @@ func main() {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Get the account information.
 | 
			
		||||
	u, err := dg.User("@me")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error obtaining account details,", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Store the account ID for later use.
 | 
			
		||||
	BotID = u.ID
 | 
			
		||||
 | 
			
		||||
	// Register messageCreate as a callback for the messageCreate events.
 | 
			
		||||
	// Register the messageCreate func as a callback for MessageCreate events.
 | 
			
		||||
	dg.AddHandler(messageCreate)
 | 
			
		||||
 | 
			
		||||
	// Open the websocket and begin listening.
 | 
			
		||||
	// Open a websocket connection to Discord and begin listening.
 | 
			
		||||
	err = dg.Open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Println("error opening connection,", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Wait here until CTRL-C or other term signal is received.
 | 
			
		||||
	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
 | 
			
		||||
	sc := make(chan os.Signal, 1)
 | 
			
		||||
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
 | 
			
		||||
	<-sc
 | 
			
		||||
 | 
			
		||||
	// Cleanly close down the Discord session.
 | 
			
		||||
	dg.Close()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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) {
 | 
			
		||||
 | 
			
		||||
	// 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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If the message is "ping" reply with "Pong!"
 | 
			
		||||
	if m.Content == "ping" {
 | 
			
		||||
		_, _ = s.ChannelMessageSend(m.ChannelID, "Pong!")
 | 
			
		||||
		s.ChannelMessageSend(m.ChannelID, "Pong!")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// If the message is "pong" reply with "Ping!"
 | 
			
		||||
	if m.Content == "pong" {
 | 
			
		||||
		_, _ = s.ChannelMessageSend(m.ChannelID, "Ping!")
 | 
			
		||||
		s.ChannelMessageSend(m.ChannelID, "Ping!")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										28
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								vendor/github.com/bwmarrin/discordgo/logging.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -23,7 +23,7 @@ const (
 | 
			
		||||
	LogError int = iota
 | 
			
		||||
 | 
			
		||||
	// 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
 | 
			
		||||
 | 
			
		||||
	// LogInformational level is used for normal non-error activity
 | 
			
		||||
@@ -34,26 +34,34 @@ const (
 | 
			
		||||
	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
 | 
			
		||||
// the format, a...  portion this command follows that of fmt.Printf
 | 
			
		||||
//   msgL   : LogLevel of the message
 | 
			
		||||
//   caller : 1 + the number of callers away from the message source
 | 
			
		||||
//   format : Printf style message format
 | 
			
		||||
//   a ...  : comma seperated list of values to pass
 | 
			
		||||
//   a ...  : comma separated list of values to pass
 | 
			
		||||
func msglog(msgL, caller int, format string, a ...interface{}) {
 | 
			
		||||
 | 
			
		||||
	pc, file, line, _ := runtime.Caller(caller)
 | 
			
		||||
	if Logger != nil {
 | 
			
		||||
		Logger(msgL, caller, format, a...)
 | 
			
		||||
	} else {
 | 
			
		||||
 | 
			
		||||
	files := strings.Split(file, "/")
 | 
			
		||||
	file = files[len(files)-1]
 | 
			
		||||
		pc, file, line, _ := runtime.Caller(caller)
 | 
			
		||||
 | 
			
		||||
	name := runtime.FuncForPC(pc).Name()
 | 
			
		||||
	fns := strings.Split(name, ".")
 | 
			
		||||
	name = fns[len(fns)-1]
 | 
			
		||||
		files := strings.Split(file, "/")
 | 
			
		||||
		file = files[len(files)-1]
 | 
			
		||||
 | 
			
		||||
	msg := fmt.Sprintf(format, a...)
 | 
			
		||||
		name := runtime.FuncForPC(pc).Name()
 | 
			
		||||
		fns := strings.Split(name, ".")
 | 
			
		||||
		name = fns[len(fns)-1]
 | 
			
		||||
 | 
			
		||||
	log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg)
 | 
			
		||||
		msg := fmt.Sprintf(format, a...)
 | 
			
		||||
 | 
			
		||||
		log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// helper function that wraps msglog for the Session struct
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										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
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"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.
 | 
			
		||||
@@ -29,6 +45,58 @@ type Message struct {
 | 
			
		||||
	Embeds          []*MessageEmbed      `json:"embeds"`
 | 
			
		||||
	Mentions        []*User              `json:"mentions"`
 | 
			
		||||
	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.
 | 
			
		||||
@@ -120,13 +188,65 @@ type MessageReactions struct {
 | 
			
		||||
 | 
			
		||||
// ContentWithMentionsReplaced will replace all @<id> mentions with the
 | 
			
		||||
// username of the mention.
 | 
			
		||||
func (m *Message) ContentWithMentionsReplaced() string {
 | 
			
		||||
	if m.Mentions == nil {
 | 
			
		||||
		return m.Content
 | 
			
		||||
	}
 | 
			
		||||
	content := m.Content
 | 
			
		||||
func (m *Message) ContentWithMentionsReplaced() (content string) {
 | 
			
		||||
	content = m.Content
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										19
									
								
								vendor/github.com/bwmarrin/discordgo/oauth2.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										19
									
								
								vendor/github.com/bwmarrin/discordgo/oauth2.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -15,13 +15,18 @@ package discordgo
 | 
			
		||||
 | 
			
		||||
// An Application struct stores values for a Discord OAuth2 Application
 | 
			
		||||
type Application struct {
 | 
			
		||||
	ID           string    `json:"id,omitempty"`
 | 
			
		||||
	Name         string    `json:"name"`
 | 
			
		||||
	Description  string    `json:"description,omitempty"`
 | 
			
		||||
	Icon         string    `json:"icon,omitempty"`
 | 
			
		||||
	Secret       string    `json:"secret,omitempty"`
 | 
			
		||||
	RedirectURIs *[]string `json:"redirect_uris,omitempty"`
 | 
			
		||||
	Owner        *User     `json:"owner"`
 | 
			
		||||
	ID                  string    `json:"id,omitempty"`
 | 
			
		||||
	Name                string    `json:"name"`
 | 
			
		||||
	Description         string    `json:"description,omitempty"`
 | 
			
		||||
	Icon                string    `json:"icon,omitempty"`
 | 
			
		||||
	Secret              string    `json:"secret,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"`
 | 
			
		||||
	Bot                 *User     `json:"bot"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Application returns an Application structure of a specific Application
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										143
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										143
									
								
								vendor/github.com/bwmarrin/discordgo/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -3,16 +3,26 @@ package discordgo
 | 
			
		||||
import (
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"sync/atomic"
 | 
			
		||||
	"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
 | 
			
		||||
type RateLimiter struct {
 | 
			
		||||
	sync.Mutex
 | 
			
		||||
	global          *Bucket
 | 
			
		||||
	buckets         map[string]*Bucket
 | 
			
		||||
	globalRateLimit time.Duration
 | 
			
		||||
	global           *int64
 | 
			
		||||
	buckets          map[string]*Bucket
 | 
			
		||||
	globalRateLimit  time.Duration
 | 
			
		||||
	customRateLimits []*customRateLimit
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewRatelimiter returns a new RateLimiter
 | 
			
		||||
@@ -20,12 +30,19 @@ func NewRatelimiter() *RateLimiter {
 | 
			
		||||
 | 
			
		||||
	return &RateLimiter{
 | 
			
		||||
		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
 | 
			
		||||
func (r *RateLimiter) getBucket(key string) *Bucket {
 | 
			
		||||
// GetBucket retrieves or creates a bucket
 | 
			
		||||
func (r *RateLimiter) GetBucket(key string) *Bucket {
 | 
			
		||||
	r.Lock()
 | 
			
		||||
	defer r.Unlock()
 | 
			
		||||
 | 
			
		||||
@@ -34,34 +51,54 @@ func (r *RateLimiter) getBucket(key string) *Bucket {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	b := &Bucket{
 | 
			
		||||
		remaining: 1,
 | 
			
		||||
		Remaining: 1,
 | 
			
		||||
		Key:       key,
 | 
			
		||||
		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
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// LockBucket Locks until a request can be made
 | 
			
		||||
func (r *RateLimiter) LockBucket(bucketID string) *Bucket {
 | 
			
		||||
 | 
			
		||||
	b := r.getBucket(bucketID)
 | 
			
		||||
 | 
			
		||||
	b.Lock()
 | 
			
		||||
 | 
			
		||||
// GetWaitTime returns the duration you should wait for a Bucket
 | 
			
		||||
func (r *RateLimiter) GetWaitTime(b *Bucket, minRemaining int) time.Duration {
 | 
			
		||||
	// 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
 | 
			
		||||
	if b.remaining < 1 && b.reset.After(time.Now()) {
 | 
			
		||||
		time.Sleep(b.reset.Sub(time.Now()))
 | 
			
		||||
 | 
			
		||||
	if b.Remaining < minRemaining && b.reset.After(time.Now()) {
 | 
			
		||||
		return b.reset.Sub(time.Now())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Check for global ratelimits
 | 
			
		||||
	r.global.Lock()
 | 
			
		||||
	r.global.Unlock()
 | 
			
		||||
	sleepTo := time.Unix(0, atomic.LoadInt64(r.global))
 | 
			
		||||
	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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -69,17 +106,33 @@ func (r *RateLimiter) LockBucket(bucketID string) *Bucket {
 | 
			
		||||
type Bucket struct {
 | 
			
		||||
	sync.Mutex
 | 
			
		||||
	Key       string
 | 
			
		||||
	remaining int
 | 
			
		||||
	Remaining int
 | 
			
		||||
	limit     int
 | 
			
		||||
	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
 | 
			
		||||
// and locks up the whole thing in case if there's a global ratelimit.
 | 
			
		||||
func (b *Bucket) Release(headers http.Header) error {
 | 
			
		||||
 | 
			
		||||
	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 {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
@@ -89,41 +142,25 @@ func (b *Bucket) Release(headers http.Header) error {
 | 
			
		||||
	global := headers.Get("X-RateLimit-Global")
 | 
			
		||||
	retryAfter := headers.Get("Retry-After")
 | 
			
		||||
 | 
			
		||||
	// If it's global just keep the main ratelimit mutex locked
 | 
			
		||||
	if global != "" {
 | 
			
		||||
		parsedAfter, err := strconv.Atoi(retryAfter)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// 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
 | 
			
		||||
	// Update global and per bucket reset time if the proper headers are available
 | 
			
		||||
	// If global is set, then it will block all buckets until after Retry-After
 | 
			
		||||
	// If Retry-After without global is provided it will use that for the new reset
 | 
			
		||||
	// time since it's more accurate than X-RateLimit-Reset.
 | 
			
		||||
	// If Retry-After after is not proided, it will update the reset time from X-RateLimit-Reset
 | 
			
		||||
	if retryAfter != "" {
 | 
			
		||||
		parsedAfter, err := strconv.ParseInt(retryAfter, 10, 64)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			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 != "" {
 | 
			
		||||
		// Calculate the reset time by using the date header returned from discord
 | 
			
		||||
		discordTime, err := http.ParseTime(headers.Get("Date"))
 | 
			
		||||
@@ -150,7 +187,7 @@ func (b *Bucket) Release(headers http.Header) error {
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		b.remaining = int(parsedRemaining)
 | 
			
		||||
		b.Remaining = int(parsedRemaining)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										502
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										502
									
								
								vendor/github.com/bwmarrin/discordgo/restapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -23,14 +23,22 @@ import (
 | 
			
		||||
	"log"
 | 
			
		||||
	"mime/multipart"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/textproto"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// ErrJSONUnmarshal is returned for JSON Unmarshall errors.
 | 
			
		||||
var ErrJSONUnmarshal = errors.New("json unmarshal")
 | 
			
		||||
// All error constants
 | 
			
		||||
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
 | 
			
		||||
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 == "" {
 | 
			
		||||
		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 {
 | 
			
		||||
		log.Printf("API REQUEST %8s :: %s\n", method, urlStr)
 | 
			
		||||
		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 := client.Do(req)
 | 
			
		||||
	resp, err := s.Client.Do(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		bucket.Release(nil)
 | 
			
		||||
		return
 | 
			
		||||
@@ -133,7 +141,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID
 | 
			
		||||
		if sequence < s.MaxRestRetries {
 | 
			
		||||
 | 
			
		||||
			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 {
 | 
			
		||||
			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
 | 
			
		||||
		// 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
 | 
			
		||||
		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.
 | 
			
		||||
//
 | 
			
		||||
// 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) {
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
@@ -189,6 +203,7 @@ func (s *Session) Login(email, password string) (err error) {
 | 
			
		||||
 | 
			
		||||
	temp := struct {
 | 
			
		||||
		Token string `json:"token"`
 | 
			
		||||
		MFA   bool   `json:"mfa"`
 | 
			
		||||
	}{}
 | 
			
		||||
 | 
			
		||||
	err = unmarshal(response, &temp)
 | 
			
		||||
@@ -197,6 +212,7 @@ func (s *Session) Login(email, password string) (err error) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	s.Token = temp.Token
 | 
			
		||||
	s.MFA = temp.MFA
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -264,15 +280,21 @@ func (s *Session) User(userID string) (st *User, err error) {
 | 
			
		||||
	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
 | 
			
		||||
func (s *Session) UserAvatar(userID string) (img image.Image, err error) {
 | 
			
		||||
	u, err := s.User(userID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		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 {
 | 
			
		||||
		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
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
		Email       string `json:"email"`
 | 
			
		||||
		Password    string `json:"password"`
 | 
			
		||||
		Username    string `json:"username"`
 | 
			
		||||
		Email       string `json:"email,omitempty"`
 | 
			
		||||
		Password    string `json:"password,omitempty"`
 | 
			
		||||
		Username    string `json:"username,omitempty"`
 | 
			
		||||
		Avatar      string `json:"avatar,omitempty"`
 | 
			
		||||
		NewPassword string `json:"new_password,omitempty"`
 | 
			
		||||
	}{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')
 | 
			
		||||
func (s *Session) UserUpdateStatus(status Status) (st *Settings, err error) {
 | 
			
		||||
	if status == StatusOffline {
 | 
			
		||||
		err = errors.New("You can't set your Status to offline")
 | 
			
		||||
		err = ErrStatusOffline
 | 
			
		||||
		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.
 | 
			
		||||
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 {
 | 
			
		||||
		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.
 | 
			
		||||
// Please see the same function inside state.go
 | 
			
		||||
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)
 | 
			
		||||
	if err != nil || channel == nil {
 | 
			
		||||
		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 {
 | 
			
		||||
		if role.ID == guild.ID {
 | 
			
		||||
			apermissions |= role.Permissions
 | 
			
		||||
@@ -447,21 +510,36 @@ func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if apermissions&PermissionAdministrator > 0 {
 | 
			
		||||
	if apermissions&PermissionAdministrator == PermissionAdministrator {
 | 
			
		||||
		apermissions |= PermissionAll
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Apply @everyone overrides from the channel.
 | 
			
		||||
	for _, overwrite := range channel.PermissionOverwrites {
 | 
			
		||||
		if guild.ID == overwrite.ID {
 | 
			
		||||
			apermissions &= ^overwrite.Deny
 | 
			
		||||
			apermissions |= overwrite.Allow
 | 
			
		||||
			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 {
 | 
			
		||||
				apermissions &= ^overwrite.Deny
 | 
			
		||||
				apermissions |= overwrite.Allow
 | 
			
		||||
				denies |= overwrite.Deny
 | 
			
		||||
				allows |= overwrite.Allow
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	apermissions &= ^denies
 | 
			
		||||
	apermissions |= allows
 | 
			
		||||
 | 
			
		||||
	for _, overwrite := range channel.PermissionOverwrites {
 | 
			
		||||
		if overwrite.Type == "member" && overwrite.ID == userID {
 | 
			
		||||
			apermissions &= ^overwrite.Deny
 | 
			
		||||
@@ -470,11 +548,11 @@ func (s *Session) UserChannelPermissions(userID, channelID string) (apermissions
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if apermissions&PermissionAdministrator > 0 {
 | 
			
		||||
	if apermissions&PermissionAdministrator == PermissionAdministrator {
 | 
			
		||||
		apermissions |= PermissionAllChannel
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
	return apermissions
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ------------------------------------------------------------------------------------------------
 | 
			
		||||
@@ -509,7 +587,7 @@ func (s *Session) GuildCreate(name string) (st *Guild, err error) {
 | 
			
		||||
		Name string `json:"name"`
 | 
			
		||||
	}{name}
 | 
			
		||||
 | 
			
		||||
	body, err := s.RequestWithBucketID("POST", EndpointGuilds, data, EndpointGuilds)
 | 
			
		||||
	body, err := s.RequestWithBucketID("POST", EndpointGuildCreate, data, EndpointGuildCreate)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -527,7 +605,7 @@ func (s *Session) GuildEdit(guildID string, g GuildParams) (st *Guild, err error
 | 
			
		||||
	if g.VerificationLevel != nil {
 | 
			
		||||
		val := *g.VerificationLevel
 | 
			
		||||
		if val < 0 || val > 3 {
 | 
			
		||||
			err = errors.New("VerificationLevel out of bounds, should be between 0 and 3")
 | 
			
		||||
			err = ErrVerificationLevelBounds
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
@@ -551,13 +629,7 @@ func (s *Session) GuildEdit(guildID string, g GuildParams) (st *Guild, err error
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
		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))
 | 
			
		||||
	body, err := s.RequestWithBucketID("PATCH", EndpointGuild(guildID), g, EndpointGuild(guildID))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -607,11 +679,28 @@ func (s *Session) GuildBans(guildID string) (st []*GuildBan, err error) {
 | 
			
		||||
// userID    : The ID of a User
 | 
			
		||||
// days      : The number of days of previous comments to delete.
 | 
			
		||||
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)
 | 
			
		||||
 | 
			
		||||
	queryParams := url.Values{}
 | 
			
		||||
	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, ""))
 | 
			
		||||
@@ -677,7 +766,21 @@ func (s *Session) GuildMember(guildID, userID string) (st *Member, err error) {
 | 
			
		||||
// userID    : The ID of a User
 | 
			
		||||
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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -722,12 +825,17 @@ func (s *Session) GuildMemberMove(guildID, userID, channelID string) (err error)
 | 
			
		||||
// GuildMemberNickname updates the nickname of a guild member
 | 
			
		||||
// guildID   : The ID of a guild
 | 
			
		||||
// 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) {
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
		Nick string `json:"nick"`
 | 
			
		||||
	}{nickname}
 | 
			
		||||
 | 
			
		||||
	if userID == "@me" {
 | 
			
		||||
		userID += "/nick"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, err = s.RequestWithBucketID("PATCH", EndpointGuildMember(guildID, userID), data, EndpointGuildMember(guildID, ""))
 | 
			
		||||
	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.
 | 
			
		||||
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
 | 
			
		||||
}
 | 
			
		||||
@@ -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.
 | 
			
		||||
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
 | 
			
		||||
}
 | 
			
		||||
@@ -801,7 +909,7 @@ func (s *Session) GuildChannelsReorder(guildID string, channels []*Channel) (err
 | 
			
		||||
// GuildInvites returns an array of Invite structures for the given guild
 | 
			
		||||
// guildID   : The ID of a Guild.
 | 
			
		||||
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 {
 | 
			
		||||
		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.
 | 
			
		||||
	if color > 0xFFFFFF {
 | 
			
		||||
		err = fmt.Errorf("color value cannot be larger than 0xFFFFFF")
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
@@ -904,7 +1013,7 @@ func (s *Session) GuildPruneCount(guildID string, days uint32) (count uint32, er
 | 
			
		||||
	count = 0
 | 
			
		||||
 | 
			
		||||
	if days <= 0 {
 | 
			
		||||
		err = errors.New("The number of days should be more than or equal to 1.")
 | 
			
		||||
		err = ErrPruneDaysBounds
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -914,6 +1023,9 @@ func (s *Session) GuildPruneCount(guildID string, days uint32) (count uint32, er
 | 
			
		||||
 | 
			
		||||
	uri := EndpointGuildPrune(guildID) + fmt.Sprintf("?days=%d", days)
 | 
			
		||||
	body, err := s.RequestWithBucketID("GET", uri, nil, EndpointGuildPrune(guildID))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = unmarshal(body, &p)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -934,7 +1046,7 @@ func (s *Session) GuildPrune(guildID string, days uint32) (count uint32, err err
 | 
			
		||||
	count = 0
 | 
			
		||||
 | 
			
		||||
	if days <= 0 {
 | 
			
		||||
		err = errors.New("The number of days should be more than or equal to 1.")
 | 
			
		||||
		err = ErrPruneDaysBounds
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -1036,7 +1148,7 @@ func (s *Session) GuildIcon(guildID string) (img image.Image, err error) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if g.Icon == "" {
 | 
			
		||||
		err = errors.New("Guild does not have an icon set.")
 | 
			
		||||
		err = ErrGuildNoIcon
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -1058,7 +1170,7 @@ func (s *Session) GuildSplash(guildID string) (img image.Image, err error) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if g.Splash == "" {
 | 
			
		||||
		err = errors.New("Guild does not have a splash set.")
 | 
			
		||||
		err = ErrGuildNoSplash
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -1098,7 +1210,7 @@ func (s *Session) GuildEmbedEdit(guildID string, enabled bool, channelID string)
 | 
			
		||||
// 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.
 | 
			
		||||
func (s *Session) Channel(channelID string) (st *Channel, err error) {
 | 
			
		||||
	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
 | 
			
		||||
// channelID  : The ID of a Channel
 | 
			
		||||
// name       : The new name to assign the channel.
 | 
			
		||||
func (s *Session) ChannelEdit(channelID, name string) (st *Channel, err error) {
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
		Name string `json:"name"`
 | 
			
		||||
	}{name}
 | 
			
		||||
func (s *Session) ChannelEdit(channelID, name string) (*Channel, error) {
 | 
			
		||||
	return s.ChannelEditComplex(channelID, &ChannelEdit{
 | 
			
		||||
		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))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return
 | 
			
		||||
@@ -1156,7 +1272,8 @@ func (s *Session) ChannelTyping(channelID string) (err error) {
 | 
			
		||||
// limit     : The number messages that can be returned. (max 100)
 | 
			
		||||
// beforeID  : If provided all messages returned will be before 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)
 | 
			
		||||
 | 
			
		||||
@@ -1170,6 +1287,9 @@ func (s *Session) ChannelMessages(channelID string, limit int, beforeID, afterID
 | 
			
		||||
	if beforeID != "" {
 | 
			
		||||
		v.Set("before", beforeID)
 | 
			
		||||
	}
 | 
			
		||||
	if aroundID != "" {
 | 
			
		||||
		v.Set("around", aroundID)
 | 
			
		||||
	}
 | 
			
		||||
	if len(v) > 0 {
 | 
			
		||||
		uri = fmt.Sprintf("%s?%s", uri, v.Encode())
 | 
			
		||||
	}
 | 
			
		||||
@@ -1212,20 +1332,92 @@ func (s *Session) ChannelMessageAck(channelID, messageID, lastToken string) (st
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// channelMessageSend sends a message to the given channel.
 | 
			
		||||
// ChannelMessageSend sends a message to the given channel.
 | 
			
		||||
// channelID : The ID of a Channel.
 | 
			
		||||
// content   : The message to send.
 | 
			
		||||
// tts       : Whether to send the message with TTS.
 | 
			
		||||
func (s *Session) channelMessageSend(channelID, content string, tts bool) (st *Message, err error) {
 | 
			
		||||
func (s *Session) ChannelMessageSend(channelID string, content string) (*Message, error) {
 | 
			
		||||
	return s.ChannelMessageSendComplex(channelID, &MessageSend{
 | 
			
		||||
		Content: content,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
	// TODO: nonce string ?
 | 
			
		||||
	data := struct {
 | 
			
		||||
		Content string `json:"content"`
 | 
			
		||||
		TTS     bool   `json:"tts"`
 | 
			
		||||
	}{content, tts}
 | 
			
		||||
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
 | 
			
		||||
 | 
			
		||||
	// Send the message to the given channel
 | 
			
		||||
	response, err := s.RequestWithBucketID("POST", EndpointChannelMessages(channelID), data, EndpointChannelMessages(channelID))
 | 
			
		||||
// ChannelMessageSendComplex sends a message to the given channel.
 | 
			
		||||
// 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 {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -1234,55 +1426,42 @@ func (s *Session) channelMessageSend(channelID, content string, tts bool) (st *M
 | 
			
		||||
	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.
 | 
			
		||||
// channelID : The ID of a Channel.
 | 
			
		||||
// content   : The message to send.
 | 
			
		||||
func (s *Session) ChannelMessageSendTTS(channelID string, content string) (st *Message, err error) {
 | 
			
		||||
 | 
			
		||||
	return s.channelMessageSend(channelID, content, true)
 | 
			
		||||
func (s *Session) ChannelMessageSendTTS(channelID string, content string) (*Message, error) {
 | 
			
		||||
	return s.ChannelMessageSendComplex(channelID, &MessageSend{
 | 
			
		||||
		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.
 | 
			
		||||
// embed     : The embed data to send.
 | 
			
		||||
func (s *Session) ChannelMessageSendEmbed(channelID string, embed *MessageEmbed) (st *Message, err error) {
 | 
			
		||||
	if embed != nil && embed.Type == "" {
 | 
			
		||||
		embed.Type = "rich"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
func (s *Session) ChannelMessageSendEmbed(channelID string, embed *MessageEmbed) (*Message, error) {
 | 
			
		||||
	return s.ChannelMessageSendComplex(channelID, &MessageSend{
 | 
			
		||||
		Embed: embed,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelMessageEdit edits an existing message, replacing it entirely with
 | 
			
		||||
// the given content.
 | 
			
		||||
// channeld  : The ID of a Channel
 | 
			
		||||
// messageID : the ID of a Message
 | 
			
		||||
func (s *Session) ChannelMessageEdit(channelID, messageID, content string) (st *Message, err error) {
 | 
			
		||||
// channelID  : The ID of a Channel
 | 
			
		||||
// messageID  : The ID of a Message
 | 
			
		||||
// 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 {
 | 
			
		||||
		Content string `json:"content"`
 | 
			
		||||
	}{content}
 | 
			
		||||
// ChannelMessageEditComplex edits an existing message, replacing it entirely with
 | 
			
		||||
// the given MessageEdit struct
 | 
			
		||||
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 {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -1291,26 +1470,12 @@ func (s *Session) ChannelMessageEdit(channelID, messageID, content string) (st *
 | 
			
		||||
	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
 | 
			
		||||
// messageID : The ID of a Message
 | 
			
		||||
// embed     : The embed data to send
 | 
			
		||||
func (s *Session) ChannelMessageEditEmbed(channelID, messageID string, embed *MessageEmbed) (st *Message, err error) {
 | 
			
		||||
	if embed != nil && embed.Type == "" {
 | 
			
		||||
		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
 | 
			
		||||
func (s *Session) ChannelMessageEditEmbed(channelID, messageID string, embed *MessageEmbed) (*Message, error) {
 | 
			
		||||
	return s.ChannelMessageEditComplex(NewMessageEdit(channelID, messageID).SetEmbed(embed))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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.
 | 
			
		||||
// 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.
 | 
			
		||||
// 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.
 | 
			
		||||
@@ -1385,48 +1550,18 @@ func (s *Session) ChannelMessagesPinned(channelID string) (st []*Message, err er
 | 
			
		||||
// channelID : The ID of a Channel.
 | 
			
		||||
// name: The name of the file.
 | 
			
		||||
// io.Reader : A reader for the file contents.
 | 
			
		||||
func (s *Session) ChannelFileSend(channelID, name string, r io.Reader) (st *Message, err error) {
 | 
			
		||||
	return s.ChannelFileSendWithMessage(channelID, "", name, r)
 | 
			
		||||
func (s *Session) ChannelFileSend(channelID, name string, r io.Reader) (*Message, error) {
 | 
			
		||||
	return s.ChannelMessageSendComplex(channelID, &MessageSend{File: &File{Name: name, Reader: r}})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ChannelFileSendWithMessage sends a file to the given channel with an message.
 | 
			
		||||
// DEPRECATED. Use ChannelMessageSendComplex instead.
 | 
			
		||||
// channelID : The ID of a Channel.
 | 
			
		||||
// content: Optional Message content.
 | 
			
		||||
// name: The name of the file.
 | 
			
		||||
// io.Reader : A reader for the file contents.
 | 
			
		||||
func (s *Session) ChannelFileSendWithMessage(channelID, content string, name string, r io.Reader) (st *Message, err error) {
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
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})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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.
 | 
			
		||||
// channelID   : The ID of a Channel
 | 
			
		||||
// i           : An Invite struct with the values MaxAge, MaxUses, Temporary,
 | 
			
		||||
//               and XkcdPass defined.
 | 
			
		||||
// i           : An Invite struct with the values MaxAge, MaxUses and Temporary defined.
 | 
			
		||||
func (s *Session) ChannelInviteCreate(channelID string, i Invite) (st *Invite, err error) {
 | 
			
		||||
 | 
			
		||||
	data := struct {
 | 
			
		||||
		MaxAge    int    `json:"max_age"`
 | 
			
		||||
		MaxUses   int    `json:"max_uses"`
 | 
			
		||||
		Temporary bool   `json:"temporary"`
 | 
			
		||||
		XKCDPass  string `json:"xkcdpass"`
 | 
			
		||||
	}{i.MaxAge, i.MaxUses, i.Temporary, i.XkcdPass}
 | 
			
		||||
		MaxAge    int  `json:"max_age"`
 | 
			
		||||
		MaxUses   int  `json:"max_uses"`
 | 
			
		||||
		Temporary bool `json:"temporary"`
 | 
			
		||||
	}{i.MaxAge, i.MaxUses, i.Temporary}
 | 
			
		||||
 | 
			
		||||
	body, err := s.RequestWithBucketID("POST", EndpointChannelInvites(channelID), data, EndpointChannelInvites(channelID))
 | 
			
		||||
	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
 | 
			
		||||
// inviteID : The invite code (or maybe xkcdpass?)
 | 
			
		||||
// inviteID : The invite code
 | 
			
		||||
func (s *Session) Invite(inviteID string) (st *Invite, err error) {
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
// 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) {
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
// inviteID : The invite code (or maybe xkcdpass?)
 | 
			
		||||
// inviteID : The invite code
 | 
			
		||||
func (s *Session) InviteAccept(inviteID string) (st *Invite, err error) {
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
// ------------------------------------------------------------------------------------------------
 | 
			
		||||
 | 
			
		||||
// Gateway returns the a websocket Gateway address
 | 
			
		||||
// Gateway returns the websocket Gateway address
 | 
			
		||||
func (s *Session) Gateway() (gateway string, err error) {
 | 
			
		||||
 | 
			
		||||
	response, err := s.RequestWithBucketID("GET", EndpointGateway, nil, EndpointGateway)
 | 
			
		||||
@@ -1591,6 +1724,28 @@ func (s *Session) Gateway() (gateway string, err error) {
 | 
			
		||||
	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
 | 
			
		||||
 | 
			
		||||
// 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
 | 
			
		||||
// 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)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = unmarshal(body, &st)
 | 
			
		||||
	_, err = s.RequestWithBucketID("DELETE", EndpointWebhook(webhookID), nil, EndpointWebhooks)
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
@@ -1781,6 +1931,16 @@ func (s *Session) MessageReactionRemove(channelID, messageID, emojiID, userID st
 | 
			
		||||
	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.
 | 
			
		||||
// channelID : The channel ID.
 | 
			
		||||
// messageID : The message ID.
 | 
			
		||||
@@ -1808,6 +1968,20 @@ func (s *Session) MessageReactions(channelID, messageID, emojiID string, limit i
 | 
			
		||||
	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)
 | 
			
		||||
// ------------------------------------------------------------------------------------------------
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										358
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										358
									
								
								vendor/github.com/bwmarrin/discordgo/state.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -14,11 +14,16 @@ package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// 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.
 | 
			
		||||
// As discord sends this in a READY blob, it seems reasonable to simply
 | 
			
		||||
@@ -33,9 +38,11 @@ type State struct {
 | 
			
		||||
	TrackMembers    bool
 | 
			
		||||
	TrackRoles      bool
 | 
			
		||||
	TrackVoice      bool
 | 
			
		||||
	TrackPresences  bool
 | 
			
		||||
 | 
			
		||||
	guildMap   map[string]*Guild
 | 
			
		||||
	channelMap map[string]*Channel
 | 
			
		||||
	memberMap  map[string]map[string]*Member
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewState creates an empty state.
 | 
			
		||||
@@ -45,16 +52,26 @@ func NewState() *State {
 | 
			
		||||
			PrivateChannels: []*Channel{},
 | 
			
		||||
			Guilds:          []*Guild{},
 | 
			
		||||
		},
 | 
			
		||||
		TrackChannels: true,
 | 
			
		||||
		TrackEmojis:   true,
 | 
			
		||||
		TrackMembers:  true,
 | 
			
		||||
		TrackRoles:    true,
 | 
			
		||||
		TrackVoice:    true,
 | 
			
		||||
		guildMap:      make(map[string]*Guild),
 | 
			
		||||
		channelMap:    make(map[string]*Channel),
 | 
			
		||||
		TrackChannels:  true,
 | 
			
		||||
		TrackEmojis:    true,
 | 
			
		||||
		TrackMembers:   true,
 | 
			
		||||
		TrackRoles:     true,
 | 
			
		||||
		TrackVoice:     true,
 | 
			
		||||
		TrackPresences: true,
 | 
			
		||||
		guildMap:       make(map[string]*Guild),
 | 
			
		||||
		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
 | 
			
		||||
// updates it if it already exists.
 | 
			
		||||
func (s *State) GuildAdd(guild *Guild) error {
 | 
			
		||||
@@ -70,6 +87,14 @@ func (s *State) GuildAdd(guild *Guild) error {
 | 
			
		||||
		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 {
 | 
			
		||||
		// 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`.
 | 
			
		||||
@@ -143,7 +168,108 @@ func (s *State) Guild(guildID string) (*Guild, error) {
 | 
			
		||||
		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.
 | 
			
		||||
@@ -163,14 +289,19 @@ func (s *State) MemberAdd(member *Member) error {
 | 
			
		||||
	s.Lock()
 | 
			
		||||
	defer s.Unlock()
 | 
			
		||||
 | 
			
		||||
	for i, m := range guild.Members {
 | 
			
		||||
		if m.User.ID == member.User.ID {
 | 
			
		||||
			guild.Members[i] = member
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	members, ok := s.memberMap[member.GuildID]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return ErrStateNotFound
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	m, ok := members[member.User.ID]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		members[member.User.ID] = member
 | 
			
		||||
		guild.Members = append(guild.Members, member)
 | 
			
		||||
	} else {
 | 
			
		||||
		*m = *member // Update the actual data, which will also update the member pointer in the slice
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	guild.Members = append(guild.Members, member)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -188,6 +319,17 @@ func (s *State) MemberRemove(member *Member) error {
 | 
			
		||||
	s.Lock()
 | 
			
		||||
	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 {
 | 
			
		||||
		if m.User.ID == member.User.ID {
 | 
			
		||||
			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.
 | 
			
		||||
@@ -204,21 +346,20 @@ func (s *State) Member(guildID, userID string) (*Member, error) {
 | 
			
		||||
		return nil, ErrNilState
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	guild, err := s.Guild(guildID)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	s.RLock()
 | 
			
		||||
	defer s.RUnlock()
 | 
			
		||||
 | 
			
		||||
	for _, m := range guild.Members {
 | 
			
		||||
		if m.User.ID == userID {
 | 
			
		||||
			return m, nil
 | 
			
		||||
		}
 | 
			
		||||
	members, ok := s.memberMap[guildID]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		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
 | 
			
		||||
@@ -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.
 | 
			
		||||
@@ -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.
 | 
			
		||||
// Channels may exist either as PrivateChannels or inside
 | 
			
		||||
// a guild.
 | 
			
		||||
@@ -319,12 +460,12 @@ func (s *State) ChannelAdd(channel *Channel) error {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if channel.IsPrivate {
 | 
			
		||||
	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM {
 | 
			
		||||
		s.PrivateChannels = append(s.PrivateChannels, channel)
 | 
			
		||||
	} else {
 | 
			
		||||
		guild, ok := s.guildMap[channel.GuildID]
 | 
			
		||||
		if !ok {
 | 
			
		||||
			return errors.New("Guild for channel not found.")
 | 
			
		||||
			return ErrStateNotFound
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		guild.Channels = append(guild.Channels, channel)
 | 
			
		||||
@@ -346,7 +487,7 @@ func (s *State) ChannelRemove(channel *Channel) error {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if channel.IsPrivate {
 | 
			
		||||
	if channel.Type == ChannelTypeDM || channel.Type == ChannelTypeGroupDM {
 | 
			
		||||
		s.Lock()
 | 
			
		||||
		defer s.Unlock()
 | 
			
		||||
 | 
			
		||||
@@ -390,7 +531,7 @@ func (s *State) PrivateChannel(channelID string) (*Channel, error) {
 | 
			
		||||
	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) {
 | 
			
		||||
	if s == nil {
 | 
			
		||||
		return nil, ErrNilState
 | 
			
		||||
@@ -403,7 +544,7 @@ func (s *State) Channel(channelID string) (*Channel, error) {
 | 
			
		||||
		return c, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil, errors.New("Channel not found.")
 | 
			
		||||
	return nil, ErrStateNotFound
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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.
 | 
			
		||||
@@ -523,7 +664,12 @@ func (s *State) MessageRemove(message *Message) error {
 | 
			
		||||
		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 {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
@@ -532,13 +678,13 @@ func (s *State) MessageRemove(message *Message) error {
 | 
			
		||||
	defer s.Unlock()
 | 
			
		||||
 | 
			
		||||
	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:]...)
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return errors.New("Message not found.")
 | 
			
		||||
	return ErrStateNotFound
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
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.
 | 
			
		||||
@@ -608,10 +754,9 @@ func (s *State) onReady(se *Session, r *Ready) (err error) {
 | 
			
		||||
	// if state is disabled, store the bare essentials.
 | 
			
		||||
	if !se.StateEnabled {
 | 
			
		||||
		ready := Ready{
 | 
			
		||||
			Version:           r.Version,
 | 
			
		||||
			SessionID:         r.SessionID,
 | 
			
		||||
			HeartbeatInterval: r.HeartbeatInterval,
 | 
			
		||||
			User:              r.User,
 | 
			
		||||
			Version:   r.Version,
 | 
			
		||||
			SessionID: r.SessionID,
 | 
			
		||||
			User:      r.User,
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		s.Ready = ready
 | 
			
		||||
@@ -623,6 +768,7 @@ func (s *State) onReady(se *Session, r *Ready) (err error) {
 | 
			
		||||
 | 
			
		||||
	for _, g := range s.Guilds {
 | 
			
		||||
		s.guildMap[g.ID] = g
 | 
			
		||||
		s.createMemberMap(g)
 | 
			
		||||
 | 
			
		||||
		for _, c := range g.Channels {
 | 
			
		||||
			s.channelMap[c.ID] = c
 | 
			
		||||
@@ -636,8 +782,8 @@ func (s *State) onReady(se *Session, r *Ready) (err error) {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onInterface handles all events related to states.
 | 
			
		||||
func (s *State) onInterface(se *Session, i interface{}) (err error) {
 | 
			
		||||
// OnInterface handles all events related to states.
 | 
			
		||||
func (s *State) OnInterface(se *Session, i interface{}) (err error) {
 | 
			
		||||
	if s == nil {
 | 
			
		||||
		return ErrNilState
 | 
			
		||||
	}
 | 
			
		||||
@@ -670,6 +816,13 @@ func (s *State) onInterface(se *Session, i interface{}) (err error) {
 | 
			
		||||
		if s.TrackMembers {
 | 
			
		||||
			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:
 | 
			
		||||
		if s.TrackRoles {
 | 
			
		||||
			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 {
 | 
			
		||||
			err = s.MessageRemove(t.Message)
 | 
			
		||||
		}
 | 
			
		||||
	case *MessageDeleteBulk:
 | 
			
		||||
		if s.MaxMessageCount != 0 {
 | 
			
		||||
			for _, mID := range t.Messages {
 | 
			
		||||
				s.messageRemoveByID(t.ChannelID, mID)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	case *VoiceStateUpdate:
 | 
			
		||||
		if s.TrackVoice {
 | 
			
		||||
			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
 | 
			
		||||
@@ -747,48 +945,46 @@ func (s *State) UserChannelPermissions(userID, channelID string) (apermissions i
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, role := range guild.Roles {
 | 
			
		||||
		if role.ID == guild.ID {
 | 
			
		||||
			apermissions |= role.Permissions
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	return memberPermissions(guild, channel, member), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, role := range guild.Roles {
 | 
			
		||||
	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 {
 | 
			
		||||
			if role.ID == roleID {
 | 
			
		||||
				apermissions |= role.Permissions
 | 
			
		||||
				break
 | 
			
		||||
				if role.Color != 0 {
 | 
			
		||||
					return role.Color
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if apermissions&PermissionAdministrator > 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
 | 
			
		||||
	return 0
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										243
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										243
									
								
								vendor/github.com/bwmarrin/discordgo/structs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -13,7 +13,7 @@ package discordgo
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
@@ -28,6 +28,7 @@ type Session struct {
 | 
			
		||||
 | 
			
		||||
	// Authentication token for this session
 | 
			
		||||
	Token string
 | 
			
		||||
	MFA   bool
 | 
			
		||||
 | 
			
		||||
	// Debug for printing JSON request/responses
 | 
			
		||||
	Debug    bool // Deprecated, will be removed.
 | 
			
		||||
@@ -48,6 +49,10 @@ type Session struct {
 | 
			
		||||
	// active guilds and the members of the guilds.
 | 
			
		||||
	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.
 | 
			
		||||
 | 
			
		||||
	// Whether the Data Websocket is ready
 | 
			
		||||
@@ -73,6 +78,15 @@ type Session struct {
 | 
			
		||||
	// StateEnabled is true.
 | 
			
		||||
	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
 | 
			
		||||
	handlersMu   sync.RWMutex
 | 
			
		||||
	handlers     map[string][]*eventHandlerInstance
 | 
			
		||||
@@ -84,11 +98,8 @@ type Session struct {
 | 
			
		||||
	// When nil, the session is not listening.
 | 
			
		||||
	listening chan interface{}
 | 
			
		||||
 | 
			
		||||
	// used to deal with rate limits
 | 
			
		||||
	ratelimiter *RateLimiter
 | 
			
		||||
 | 
			
		||||
	// sequence tracks the current gateway api websocket sequence number
 | 
			
		||||
	sequence int
 | 
			
		||||
	sequence *int64
 | 
			
		||||
 | 
			
		||||
	// stores sessions current Discord Gateway
 | 
			
		||||
	gateway string
 | 
			
		||||
@@ -100,12 +111,6 @@ type Session struct {
 | 
			
		||||
	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.
 | 
			
		||||
type VoiceRegion struct {
 | 
			
		||||
	ID       string `json:"id"`
 | 
			
		||||
@@ -137,25 +142,50 @@ type Invite struct {
 | 
			
		||||
	MaxAge    int       `json:"max_age"`
 | 
			
		||||
	Uses      int       `json:"uses"`
 | 
			
		||||
	MaxUses   int       `json:"max_uses"`
 | 
			
		||||
	XkcdPass  string    `json:"xkcdpass"`
 | 
			
		||||
	Revoked   bool      `json:"revoked"`
 | 
			
		||||
	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.
 | 
			
		||||
type Channel struct {
 | 
			
		||||
	ID                   string                 `json:"id"`
 | 
			
		||||
	GuildID              string                 `json:"guild_id"`
 | 
			
		||||
	Name                 string                 `json:"name"`
 | 
			
		||||
	Topic                string                 `json:"topic"`
 | 
			
		||||
	Type                 string                 `json:"type"`
 | 
			
		||||
	Type                 ChannelType            `json:"type"`
 | 
			
		||||
	LastMessageID        string                 `json:"last_message_id"`
 | 
			
		||||
	NSFW                 bool                   `json:"nsfw"`
 | 
			
		||||
	Position             int                    `json:"position"`
 | 
			
		||||
	Bitrate              int                    `json:"bitrate"`
 | 
			
		||||
	IsPrivate            bool                   `json:"is_private"`
 | 
			
		||||
	Recipient            *User                  `json:"recipient"`
 | 
			
		||||
	Recipients           []*User                `json:"recipients"`
 | 
			
		||||
	Messages             []*Message             `json:"-"`
 | 
			
		||||
	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
 | 
			
		||||
@@ -173,6 +203,7 @@ type Emoji struct {
 | 
			
		||||
	Roles         []string `json:"roles"`
 | 
			
		||||
	Managed       bool     `json:"managed"`
 | 
			
		||||
	RequireColons bool     `json:"require_colons"`
 | 
			
		||||
	Animated      bool     `json:"animated"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// VerificationLevel type defination
 | 
			
		||||
// VerificationLevel type definition
 | 
			
		||||
type VerificationLevel int
 | 
			
		||||
 | 
			
		||||
// 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
 | 
			
		||||
type GuildParams struct {
 | 
			
		||||
	Name              string             `json:"name"`
 | 
			
		||||
	Region            string             `json:"region"`
 | 
			
		||||
	VerificationLevel *VerificationLevel `json:"verification_level"`
 | 
			
		||||
	Name                        string             `json:"name,omitempty"`
 | 
			
		||||
	Region                      string             `json:"region,omitempty"`
 | 
			
		||||
	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.
 | 
			
		||||
@@ -252,6 +289,21 @@ type Role struct {
 | 
			
		||||
	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
 | 
			
		||||
type VoiceState struct {
 | 
			
		||||
	UserID    string `json:"user_id"`
 | 
			
		||||
@@ -272,47 +324,61 @@ type Presence struct {
 | 
			
		||||
	Game   *Game    `json:"game"`
 | 
			
		||||
	Nick   string   `json:"nick"`
 | 
			
		||||
	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
 | 
			
		||||
type Game struct {
 | 
			
		||||
	Name string `json:"name"`
 | 
			
		||||
	Type int    `json:"type"`
 | 
			
		||||
	URL  string `json:"url"`
 | 
			
		||||
	Name          string     `json:"name"`
 | 
			
		||||
	Type          GameType   `json:"type"`
 | 
			
		||||
	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
 | 
			
		||||
func (g *Game) UnmarshalJSON(bytes []byte) error {
 | 
			
		||||
	temp := &struct {
 | 
			
		||||
		Name string          `json:"name"`
 | 
			
		||||
		Type json.RawMessage `json:"type"`
 | 
			
		||||
		URL  string          `json:"url"`
 | 
			
		||||
// A TimeStamps struct contains start and end times used in the rich presence "playing .." Game
 | 
			
		||||
type TimeStamps struct {
 | 
			
		||||
	EndTimestamp   int64 `json:"end,omitempty"`
 | 
			
		||||
	StartTimestamp int64 `json:"start,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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 {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	g.Name = temp.Name
 | 
			
		||||
	g.URL = temp.URL
 | 
			
		||||
 | 
			
		||||
	if temp.Type != nil {
 | 
			
		||||
		err = json.Unmarshal(temp.Type, &g.Type)
 | 
			
		||||
		if err == nil {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		s := ""
 | 
			
		||||
		err = json.Unmarshal(temp.Type, &s)
 | 
			
		||||
		if err == nil {
 | 
			
		||||
			g.Type, err = strconv.Atoi(s)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	t.EndTimestamp = int64(temp.End)
 | 
			
		||||
	t.StartTimestamp = int64(temp.Start)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// An Assets struct contains assets and labels used in the rich presence "playing .." Game
 | 
			
		||||
type Assets struct {
 | 
			
		||||
	LargeImageID string `json:"large_image,omitempty"`
 | 
			
		||||
	SmallImageID string `json:"small_image,omitempty"`
 | 
			
		||||
	LargeText    string `json:"large_text,omitempty"`
 | 
			
		||||
	SmallText    string `json:"small_text,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// A Member stores user information for Guild members.
 | 
			
		||||
type Member struct {
 | 
			
		||||
	GuildID  string   `json:"guild_id"`
 | 
			
		||||
@@ -324,19 +390,6 @@ type Member struct {
 | 
			
		||||
	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.
 | 
			
		||||
type Settings struct {
 | 
			
		||||
	RenderEmbeds           bool               `json:"render_embeds"`
 | 
			
		||||
@@ -356,7 +409,7 @@ type Settings struct {
 | 
			
		||||
	DeveloperMode          bool               `json:"developer_mode"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Status type defination
 | 
			
		||||
// Status type definition
 | 
			
		||||
type Status string
 | 
			
		||||
 | 
			
		||||
// Constants for Status with the different current available status
 | 
			
		||||
@@ -502,6 +555,12 @@ type MessageReaction struct {
 | 
			
		||||
	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
 | 
			
		||||
const (
 | 
			
		||||
	PermissionReadMessages = 1 << (iota + 10)
 | 
			
		||||
@@ -542,6 +601,8 @@ const (
 | 
			
		||||
	PermissionAdministrator
 | 
			
		||||
	PermissionManageChannels
 | 
			
		||||
	PermissionManageServer
 | 
			
		||||
	PermissionAddReactions
 | 
			
		||||
	PermissionViewAuditLogs
 | 
			
		||||
 | 
			
		||||
	PermissionAllText = PermissionReadMessages |
 | 
			
		||||
		PermissionSendMessages |
 | 
			
		||||
@@ -561,9 +622,65 @@ const (
 | 
			
		||||
		PermissionAllVoice |
 | 
			
		||||
		PermissionCreateInstantInvite |
 | 
			
		||||
		PermissionManageRoles |
 | 
			
		||||
		PermissionManageChannels
 | 
			
		||||
		PermissionManageChannels |
 | 
			
		||||
		PermissionAddReactions |
 | 
			
		||||
		PermissionViewAuditLogs
 | 
			
		||||
	PermissionAll = PermissionAllChannel |
 | 
			
		||||
		PermissionKickMembers |
 | 
			
		||||
		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 {
 | 
			
		||||
  return {{privateName .}}EventType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
{{if isDiscordEvent .}}
 | 
			
		||||
// New returns a new instance of {{.}}.
 | 
			
		||||
func (eh {{privateName .}}EventHandler) New() interface{} {
 | 
			
		||||
  return &{{.}}{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}{{end}}
 | 
			
		||||
// Handle is the handler for {{.}} events.
 | 
			
		||||
func (eh {{privateName .}}EventHandler) Handle(s *Session, i interface{}) {
 | 
			
		||||
  if t, ok := i.(*{{.}}); ok {
 | 
			
		||||
    eh(s, t)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
{{end}}
 | 
			
		||||
func handlerForInterface(handler interface{}) EventHandler {
 | 
			
		||||
  switch v := handler.(type) {
 | 
			
		||||
@@ -60,6 +60,7 @@ func handlerForInterface(handler interface{}) EventHandler {
 | 
			
		||||
 | 
			
		||||
  return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func init() { {{range .}}{{if isDiscordEvent .}}
 | 
			
		||||
  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/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net"
 | 
			
		||||
	"runtime"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
@@ -70,7 +68,7 @@ type VoiceConnection struct {
 | 
			
		||||
	voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// VoiceSpeakingUpdateHandler type provides a function defination for the
 | 
			
		||||
// VoiceSpeakingUpdateHandler type provides a function definition for the
 | 
			
		||||
// VoiceSpeakingUpdate event
 | 
			
		||||
type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate)
 | 
			
		||||
 | 
			
		||||
@@ -93,18 +91,22 @@ func (v *VoiceConnection) Speaking(b bool) (err error) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if v.wsConn == nil {
 | 
			
		||||
		return fmt.Errorf("No VoiceConnection websocket.")
 | 
			
		||||
		return fmt.Errorf("no VoiceConnection websocket")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	data := voiceSpeakingOp{5, voiceSpeakingData{b, 0}}
 | 
			
		||||
	v.wsMutex.Lock()
 | 
			
		||||
	err = v.wsConn.WriteJSON(data)
 | 
			
		||||
	v.wsMutex.Unlock()
 | 
			
		||||
 | 
			
		||||
	v.Lock()
 | 
			
		||||
	defer v.Unlock()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		v.speaking = false
 | 
			
		||||
		log.Println("Speaking() write json error:", err)
 | 
			
		||||
		v.log(LogError, "Speaking() write json error:", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	v.speaking = b
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
@@ -139,9 +141,9 @@ func (v *VoiceConnection) Disconnect() (err error) {
 | 
			
		||||
	// Send a OP4 with a nil channel to disconnect
 | 
			
		||||
	if v.sessionID != "" {
 | 
			
		||||
		data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, nil, true, true}}
 | 
			
		||||
		v.wsMutex.Lock()
 | 
			
		||||
		v.session.wsMutex.Lock()
 | 
			
		||||
		err = v.session.wsConn.WriteJSON(data)
 | 
			
		||||
		v.wsMutex.Unlock()
 | 
			
		||||
		v.session.wsMutex.Unlock()
 | 
			
		||||
		v.sessionID = ""
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -149,7 +151,10 @@ func (v *VoiceConnection) Disconnect() (err error) {
 | 
			
		||||
	v.Close()
 | 
			
		||||
 | 
			
		||||
	v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID)
 | 
			
		||||
 | 
			
		||||
	v.session.Lock()
 | 
			
		||||
	delete(v.session.VoiceConnections, v.GuildID)
 | 
			
		||||
	v.session.Unlock()
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
@@ -175,7 +180,7 @@ func (v *VoiceConnection) Close() {
 | 
			
		||||
		v.log(LogInformational, "closing udp")
 | 
			
		||||
		err := v.udpConn.Close()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Println("error closing udp connection: ", err)
 | 
			
		||||
			v.log(LogError, "error closing udp connection: ", err)
 | 
			
		||||
		}
 | 
			
		||||
		v.udpConn = nil
 | 
			
		||||
	}
 | 
			
		||||
@@ -185,7 +190,9 @@ func (v *VoiceConnection) Close() {
 | 
			
		||||
 | 
			
		||||
		// To cleanly close a connection, a client should send a close
 | 
			
		||||
		// frame and wait for the server to close the connection.
 | 
			
		||||
		v.wsMutex.Lock()
 | 
			
		||||
		err := v.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
 | 
			
		||||
		v.wsMutex.Unlock()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			v.log(LogError, "error closing websocket, %s", err)
 | 
			
		||||
		}
 | 
			
		||||
@@ -239,19 +246,22 @@ type voiceOP2 struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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 {
 | 
			
		||||
 | 
			
		||||
	v.log(LogInformational, "called")
 | 
			
		||||
 | 
			
		||||
	i := 0
 | 
			
		||||
	for {
 | 
			
		||||
		if v.Ready {
 | 
			
		||||
		v.RLock()
 | 
			
		||||
		ready := v.Ready
 | 
			
		||||
		v.RUnlock()
 | 
			
		||||
		if ready {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if i > 10 {
 | 
			
		||||
			return fmt.Errorf("Timeout waiting for voice.")
 | 
			
		||||
			return fmt.Errorf("timeout waiting for voice")
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		time.Sleep(1 * time.Second)
 | 
			
		||||
@@ -282,7 +292,7 @@ func (v *VoiceConnection) open() (err error) {
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		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)
 | 
			
		||||
		i++
 | 
			
		||||
@@ -409,8 +419,6 @@ func (v *VoiceConnection) onEvent(message []byte) {
 | 
			
		||||
			go v.opusReceiver(v.udpConn, v.close, v.OpusRecv)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// Send the ready event
 | 
			
		||||
		v.connected <- true
 | 
			
		||||
		return
 | 
			
		||||
 | 
			
		||||
	case 3: // HEARTBEAT response
 | 
			
		||||
@@ -418,6 +426,9 @@ func (v *VoiceConnection) onEvent(message []byte) {
 | 
			
		||||
		return
 | 
			
		||||
 | 
			
		||||
	case 4: // udp encryption secret key
 | 
			
		||||
		v.Lock()
 | 
			
		||||
		defer v.Unlock()
 | 
			
		||||
 | 
			
		||||
		v.op4 = voiceOP4{}
 | 
			
		||||
		if err := json.Unmarshal(e.RawData, &v.op4); err != nil {
 | 
			
		||||
			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
 | 
			
		||||
	ticker := time.NewTicker(i * time.Millisecond)
 | 
			
		||||
	defer ticker.Stop()
 | 
			
		||||
	for {
 | 
			
		||||
		v.log(LogDebug, "sending heartbeat packet")
 | 
			
		||||
		v.wsMutex.Lock()
 | 
			
		||||
@@ -616,6 +628,7 @@ func (v *VoiceConnection) udpKeepAlive(udpConn *net.UDPConn, close <-chan struct
 | 
			
		||||
	packet := make([]byte, 8)
 | 
			
		||||
 | 
			
		||||
	ticker := time.NewTicker(i)
 | 
			
		||||
	defer ticker.Stop()
 | 
			
		||||
	for {
 | 
			
		||||
 | 
			
		||||
		binary.LittleEndian.PutUint64(packet, sequence)
 | 
			
		||||
@@ -644,12 +657,16 @@ func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{}
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	runtime.LockOSThread()
 | 
			
		||||
 | 
			
		||||
	// VoiceConnection is now ready to receive audio packets
 | 
			
		||||
	// TODO: this needs reviewed as I think there must be a better way.
 | 
			
		||||
	v.Lock()
 | 
			
		||||
	v.Ready = true
 | 
			
		||||
	defer func() { v.Ready = false }()
 | 
			
		||||
	v.Unlock()
 | 
			
		||||
	defer func() {
 | 
			
		||||
		v.Lock()
 | 
			
		||||
		v.Ready = false
 | 
			
		||||
		v.Unlock()
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	var sequence uint16
 | 
			
		||||
	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
 | 
			
		||||
	ticker := time.NewTicker(time.Millisecond * time.Duration(size/(rate/1000)))
 | 
			
		||||
	defer ticker.Stop()
 | 
			
		||||
	for {
 | 
			
		||||
 | 
			
		||||
		// 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
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if !v.speaking {
 | 
			
		||||
		v.RLock()
 | 
			
		||||
		speaking := v.speaking
 | 
			
		||||
		v.RUnlock()
 | 
			
		||||
		if !speaking {
 | 
			
		||||
			err := v.Speaking(true)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				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
 | 
			
		||||
		copy(nonce[:], udpHeader)
 | 
			
		||||
		v.RLock()
 | 
			
		||||
		sendbuf := secretbox.Seal(udpHeader, recvbuf, &nonce, &v.op4.SecretKey)
 | 
			
		||||
		v.RUnlock()
 | 
			
		||||
 | 
			
		||||
		// block here until we're exactly at the right time :)
 | 
			
		||||
		// Then send rtp audio packet to Discord over UDP
 | 
			
		||||
@@ -742,7 +765,6 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	p := Packet{}
 | 
			
		||||
	recvbuf := make([]byte, 1024)
 | 
			
		||||
	var nonce [24]byte
 | 
			
		||||
 | 
			
		||||
@@ -773,11 +795,12 @@ func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// For now, skip anything except audio.
 | 
			
		||||
		if rlen < 12 || recvbuf[0] != 0x80 {
 | 
			
		||||
		if rlen < 12 || (recvbuf[0] != 0x80 && recvbuf[0] != 0x90) {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// build a audio packet struct
 | 
			
		||||
		p := Packet{}
 | 
			
		||||
		p.Type = recvbuf[0:2]
 | 
			
		||||
		p.Sequence = binary.BigEndian.Uint16(recvbuf[2:4])
 | 
			
		||||
		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])
 | 
			
		||||
		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 {
 | 
			
		||||
			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 {
 | 
			
		||||
			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
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
@@ -837,6 +869,8 @@ func (v *VoiceConnection) reconnect() {
 | 
			
		||||
			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
 | 
			
		||||
		// packet to reset things.
 | 
			
		||||
		// 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(LogInformational, "error reconnecting to channel %s, %s", v.ChannelID, err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										350
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										350
									
								
								vendor/github.com/bwmarrin/discordgo/wsapi.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -19,33 +19,141 @@ import (
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"runtime"
 | 
			
		||||
	"sync/atomic"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"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 {
 | 
			
		||||
	Op   int `json:"op"`
 | 
			
		||||
	Data struct {
 | 
			
		||||
		Token     string `json:"token"`
 | 
			
		||||
		SessionID string `json:"session_id"`
 | 
			
		||||
		Sequence  int    `json:"seq"`
 | 
			
		||||
		Sequence  int64  `json:"seq"`
 | 
			
		||||
	} `json:"d"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Open opens a websocket connection to Discord.
 | 
			
		||||
func (s *Session) Open() (err error) {
 | 
			
		||||
 | 
			
		||||
// Open creates a websocket connection to Discord.
 | 
			
		||||
// See: https://discordapp.com/developers/docs/topics/gateway#connecting
 | 
			
		||||
func (s *Session) Open() error {
 | 
			
		||||
	s.log(LogInformational, "called")
 | 
			
		||||
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	// Prevent Open or other major Session functions from
 | 
			
		||||
	// being called while Open is still running.
 | 
			
		||||
	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 {
 | 
			
		||||
			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.
 | 
			
		||||
	// We create it here so the below READY/RESUMED packet can populate
 | 
			
		||||
	// the state :)
 | 
			
		||||
	// XXX: Move to New() func?
 | 
			
		||||
	if s.State == nil {
 | 
			
		||||
		state := NewState()
 | 
			
		||||
		state.TrackChannels = false
 | 
			
		||||
@@ -56,75 +164,42 @@ func (s *Session) Open() (err error) {
 | 
			
		||||
		s.State = state
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if s.wsConn != nil {
 | 
			
		||||
		err = errors.New("Web socket already opened.")
 | 
			
		||||
		return
 | 
			
		||||
	// Now Discord should send us a READY or RESUMED packet.
 | 
			
		||||
	mt, m, err = s.wsConn.ReadMessage()
 | 
			
		||||
	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 {
 | 
			
		||||
		s.log(LogInformational, "creating new VoiceConnections map")
 | 
			
		||||
		s.VoiceConnections = make(map[string]*VoiceConnection)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Get the gateway to use for the Websocket connection
 | 
			
		||||
	if s.gateway == "" {
 | 
			
		||||
		s.gateway, err = s.Gateway()
 | 
			
		||||
		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.
 | 
			
		||||
	// Create listening chan outside of listen, as it needs to happen inside the
 | 
			
		||||
	// mutex lock and needs to exist before calling heartbeat and listen
 | 
			
		||||
	// go rountines.
 | 
			
		||||
	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)
 | 
			
		||||
 | 
			
		||||
	s.Unlock()
 | 
			
		||||
 | 
			
		||||
	s.log(LogInformational, "emit connect event")
 | 
			
		||||
	s.handleEvent(connectEventType, &Connect{})
 | 
			
		||||
 | 
			
		||||
	s.log(LogInformational, "exiting")
 | 
			
		||||
	return
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// listen polls the websocket connection for events, it will stop when the
 | 
			
		||||
@@ -176,14 +251,22 @@ func (s *Session) listen(wsConn *websocket.Conn, listening <-chan interface{}) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type heartbeatOp struct {
 | 
			
		||||
	Op   int `json:"op"`
 | 
			
		||||
	Data int `json:"d"`
 | 
			
		||||
	Op   int   `json:"op"`
 | 
			
		||||
	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
 | 
			
		||||
// is still connected.  If you do not send these heartbeats Discord will
 | 
			
		||||
// 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")
 | 
			
		||||
 | 
			
		||||
@@ -192,19 +275,26 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var err error
 | 
			
		||||
	ticker := time.NewTicker(i * time.Millisecond)
 | 
			
		||||
	ticker := time.NewTicker(heartbeatIntervalMsec * time.Millisecond)
 | 
			
		||||
	defer ticker.Stop()
 | 
			
		||||
 | 
			
		||||
	for {
 | 
			
		||||
 | 
			
		||||
		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", s.sequence)
 | 
			
		||||
		s.RLock()
 | 
			
		||||
		last := s.LastHeartbeatAck
 | 
			
		||||
		s.RUnlock()
 | 
			
		||||
		sequence := atomic.LoadInt64(s.sequence)
 | 
			
		||||
		s.log(LogInformational, "sending gateway websocket heartbeat seq %d", sequence)
 | 
			
		||||
		s.wsMutex.Lock()
 | 
			
		||||
		err = wsConn.WriteJSON(heartbeatOp{1, s.sequence})
 | 
			
		||||
		err = wsConn.WriteJSON(heartbeatOp{1, sequence})
 | 
			
		||||
		s.wsMutex.Unlock()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err)
 | 
			
		||||
			s.Lock()
 | 
			
		||||
			s.DataReady = false
 | 
			
		||||
			s.Unlock()
 | 
			
		||||
		if err != nil || time.Now().UTC().Sub(last) > (heartbeatIntervalMsec*FailedHeartbeatAcks) {
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				s.log(LogError, "error sending heartbeat to gateway %s, %s", s.gateway, err)
 | 
			
		||||
			} else {
 | 
			
		||||
				s.log(LogError, "haven't gotten a heartbeat ACK in %v, triggering a reconnection", time.Now().UTC().Sub(last))
 | 
			
		||||
			}
 | 
			
		||||
			s.Close()
 | 
			
		||||
			s.reconnect()
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		s.Lock()
 | 
			
		||||
@@ -220,14 +310,17 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type updateStatusData struct {
 | 
			
		||||
	IdleSince *int  `json:"idle_since"`
 | 
			
		||||
	Game      *Game `json:"game"`
 | 
			
		||||
// UpdateStatusData ia provided to UpdateStatusComplex()
 | 
			
		||||
type UpdateStatusData struct {
 | 
			
		||||
	IdleSince *int   `json:"since"`
 | 
			
		||||
	Game      *Game  `json:"game"`
 | 
			
		||||
	AFK       bool   `json:"afk"`
 | 
			
		||||
	Status    string `json:"status"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type updateStatusOp struct {
 | 
			
		||||
	Op   int              `json:"op"`
 | 
			
		||||
	Data updateStatusData `json:"d"`
 | 
			
		||||
	Data UpdateStatusData `json:"d"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 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.RLock()
 | 
			
		||||
	defer s.RUnlock()
 | 
			
		||||
	if s.wsConn == nil {
 | 
			
		||||
		return errors.New("no websocket connection exists")
 | 
			
		||||
	usd := UpdateStatusData{
 | 
			
		||||
		Status: "online",
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var usd updateStatusData
 | 
			
		||||
	if idle > 0 {
 | 
			
		||||
		usd.IdleSince = &idle
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if game != "" {
 | 
			
		||||
		gameType := 0
 | 
			
		||||
		gameType := GameTypeGame
 | 
			
		||||
		if url != "" {
 | 
			
		||||
			gameType = 1
 | 
			
		||||
			gameType = GameTypeStreaming
 | 
			
		||||
		}
 | 
			
		||||
		usd.Game = &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()
 | 
			
		||||
	err = s.wsConn.WriteJSON(updateStatusOp{3, usd})
 | 
			
		||||
	s.wsMutex.Unlock()
 | 
			
		||||
@@ -299,7 +401,7 @@ func (s *Session) RequestGuildMembers(guildID, query string, limit int) (err err
 | 
			
		||||
	s.RLock()
 | 
			
		||||
	defer s.RUnlock()
 | 
			
		||||
	if s.wsConn == nil {
 | 
			
		||||
		return errors.New("no websocket connection exists")
 | 
			
		||||
		return ErrWSNotFound
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	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
 | 
			
		||||
// "OnEvent" event then all events will be passed to that handler.
 | 
			
		||||
//
 | 
			
		||||
// TODO: You may also register a custom event handler entirely using...
 | 
			
		||||
func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
func (s *Session) onEvent(messageType int, message []byte) (*Event, error) {
 | 
			
		||||
 | 
			
		||||
	var err error
 | 
			
		||||
	var reader io.Reader
 | 
			
		||||
@@ -337,7 +437,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
		z, err2 := zlib.NewReader(reader)
 | 
			
		||||
		if err2 != nil {
 | 
			
		||||
			s.log(LogError, "error uncompressing websocket message, %s", err)
 | 
			
		||||
			return
 | 
			
		||||
			return nil, err2
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		defer func() {
 | 
			
		||||
@@ -355,7 +455,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
	decoder := json.NewDecoder(reader)
 | 
			
		||||
	if err = decoder.Decode(&e); err != nil {
 | 
			
		||||
		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))
 | 
			
		||||
@@ -365,20 +465,23 @@ func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
	if e.Operation == 1 {
 | 
			
		||||
		s.log(LogInformational, "sending heartbeat in response to Op1")
 | 
			
		||||
		s.wsMutex.Lock()
 | 
			
		||||
		err = s.wsConn.WriteJSON(heartbeatOp{1, s.sequence})
 | 
			
		||||
		err = s.wsConn.WriteJSON(heartbeatOp{1, atomic.LoadInt64(s.sequence)})
 | 
			
		||||
		s.wsMutex.Unlock()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			s.log(LogError, "error sending heartbeat in response to Op1")
 | 
			
		||||
			return
 | 
			
		||||
			return e, err
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		return
 | 
			
		||||
		return e, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Reconnect
 | 
			
		||||
	// Must immediately disconnect from gateway and reconnect to new gateway.
 | 
			
		||||
	if e.Operation == 7 {
 | 
			
		||||
		// TODO
 | 
			
		||||
		s.log(LogInformational, "Closing and reconnecting in response to Op7")
 | 
			
		||||
		s.Close()
 | 
			
		||||
		s.reconnect()
 | 
			
		||||
		return e, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Invalid Session
 | 
			
		||||
@@ -390,10 +493,23 @@ func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
		err = s.identify()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			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
 | 
			
		||||
@@ -401,11 +517,11 @@ func (s *Session) onEvent(messageType int, message []byte) {
 | 
			
		||||
		// But we probably should be doing something with them.
 | 
			
		||||
		// 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))
 | 
			
		||||
		return
 | 
			
		||||
		return e, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 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.
 | 
			
		||||
	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.
 | 
			
		||||
	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.RLock()
 | 
			
		||||
	voice, _ = s.VoiceConnections[gID]
 | 
			
		||||
	s.RUnlock()
 | 
			
		||||
 | 
			
		||||
	if voice == nil {
 | 
			
		||||
		voice = &VoiceConnection{}
 | 
			
		||||
		s.Lock()
 | 
			
		||||
		s.VoiceConnections[gID] = voice
 | 
			
		||||
		s.Unlock()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	voice.Lock()
 | 
			
		||||
	voice.GuildID = gID
 | 
			
		||||
	voice.ChannelID = cID
 | 
			
		||||
	voice.deaf = deaf
 | 
			
		||||
	voice.mute = mute
 | 
			
		||||
	voice.session = s
 | 
			
		||||
	voice.Unlock()
 | 
			
		||||
 | 
			
		||||
	// Send the request to Discord that we want to join the voice channel
 | 
			
		||||
	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
 | 
			
		||||
	s.RLock()
 | 
			
		||||
	voice, exists := s.VoiceConnections[st.GuildID]
 | 
			
		||||
	s.RUnlock()
 | 
			
		||||
	if !exists {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -511,8 +637,11 @@ func (s *Session) onVoiceStateUpdate(st *VoiceStateUpdate) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Store the SessionID for later use.
 | 
			
		||||
	voice.Lock()
 | 
			
		||||
	voice.UserID = st.UserID
 | 
			
		||||
	voice.sessionID = st.SessionID
 | 
			
		||||
	voice.ChannelID = st.ChannelID
 | 
			
		||||
	voice.Unlock()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// onVoiceServerUpdate handles the Voice Server Update data websocket event.
 | 
			
		||||
@@ -524,7 +653,9 @@ func (s *Session) onVoiceServerUpdate(st *VoiceServerUpdate) {
 | 
			
		||||
 | 
			
		||||
	s.log(LogInformational, "called")
 | 
			
		||||
 | 
			
		||||
	s.RLock()
 | 
			
		||||
	voice, exists := s.VoiceConnections[st.GuildID]
 | 
			
		||||
	s.RUnlock()
 | 
			
		||||
 | 
			
		||||
	// If no VoiceConnection exists, just skip this
 | 
			
		||||
	if !exists {
 | 
			
		||||
@@ -536,11 +667,13 @@ func (s *Session) onVoiceServerUpdate(st *VoiceServerUpdate) {
 | 
			
		||||
	voice.Close()
 | 
			
		||||
 | 
			
		||||
	// Store values for later use
 | 
			
		||||
	voice.Lock()
 | 
			
		||||
	voice.token = st.Token
 | 
			
		||||
	voice.endpoint = st.Endpoint
 | 
			
		||||
	voice.GuildID = st.GuildID
 | 
			
		||||
	voice.Unlock()
 | 
			
		||||
 | 
			
		||||
	// Open a conenction to the voice server
 | 
			
		||||
	// Open a connection to the voice server
 | 
			
		||||
	err := voice.open()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		s.log(LogError, "onVoiceServerUpdate voice.open, %s", err)
 | 
			
		||||
@@ -588,7 +721,7 @@ func (s *Session) identify() error {
 | 
			
		||||
	if s.ShardCount > 1 {
 | 
			
		||||
 | 
			
		||||
		if s.ShardID >= s.ShardCount {
 | 
			
		||||
			return errors.New("ShardID must be less than ShardCount")
 | 
			
		||||
			return ErrWSShardBounds
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		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"
 | 
			
		||||
				// happens.  So we're doing this for now just to improve
 | 
			
		||||
				// stability in those edge cases.
 | 
			
		||||
				s.RLock()
 | 
			
		||||
				defer s.RUnlock()
 | 
			
		||||
				for _, v := range s.VoiceConnections {
 | 
			
		||||
 | 
			
		||||
					s.log(LogInformational, "reconnecting voice connection to guild %s", v.GuildID)
 | 
			
		||||
@@ -641,6 +776,13 @@ func (s *Session) reconnect() {
 | 
			
		||||
				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)
 | 
			
		||||
 | 
			
		||||
			<-time.After(wait * time.Second)
 | 
			
		||||
@@ -675,7 +817,9 @@ func (s *Session) Close() (err error) {
 | 
			
		||||
		s.log(LogInformational, "sending close frame")
 | 
			
		||||
		// To cleanly close a connection, a client should send a close
 | 
			
		||||
		// frame and wait for the server to close the connection.
 | 
			
		||||
		s.wsMutex.Lock()
 | 
			
		||||
		err := s.wsConn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
 | 
			
		||||
		s.wsMutex.Unlock()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			s.log(LogInformational, "error closing websocket, %s", err)
 | 
			
		||||
		}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										202
									
								
								vendor/github.com/coreos/etcd/client/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										202
									
								
								vendor/github.com/coreos/etcd/client/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,202 @@
 | 
			
		||||
 | 
			
		||||
                                 Apache License
 | 
			
		||||
                           Version 2.0, January 2004
 | 
			
		||||
                        http://www.apache.org/licenses/
 | 
			
		||||
 | 
			
		||||
   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
   1. Definitions.
 | 
			
		||||
 | 
			
		||||
      "License" shall mean the terms and conditions for use, reproduction,
 | 
			
		||||
      and distribution as defined by Sections 1 through 9 of this document.
 | 
			
		||||
 | 
			
		||||
      "Licensor" shall mean the copyright owner or entity authorized by
 | 
			
		||||
      the copyright owner that is granting the License.
 | 
			
		||||
 | 
			
		||||
      "Legal Entity" shall mean the union of the acting entity and all
 | 
			
		||||
      other entities that control, are controlled by, or are under common
 | 
			
		||||
      control with that entity. For the purposes of this definition,
 | 
			
		||||
      "control" means (i) the power, direct or indirect, to cause the
 | 
			
		||||
      direction or management of such entity, whether by contract or
 | 
			
		||||
      otherwise, or (ii) ownership of fifty percent (50%) or more of the
 | 
			
		||||
      outstanding shares, or (iii) beneficial ownership of such entity.
 | 
			
		||||
 | 
			
		||||
      "You" (or "Your") shall mean an individual or Legal Entity
 | 
			
		||||
      exercising permissions granted by this License.
 | 
			
		||||
 | 
			
		||||
      "Source" form shall mean the preferred form for making modifications,
 | 
			
		||||
      including but not limited to software source code, documentation
 | 
			
		||||
      source, and configuration files.
 | 
			
		||||
 | 
			
		||||
      "Object" form shall mean any form resulting from mechanical
 | 
			
		||||
      transformation or translation of a Source form, including but
 | 
			
		||||
      not limited to compiled object code, generated documentation,
 | 
			
		||||
      and conversions to other media types.
 | 
			
		||||
 | 
			
		||||
      "Work" shall mean the work of authorship, whether in Source or
 | 
			
		||||
      Object form, made available under the License, as indicated by a
 | 
			
		||||
      copyright notice that is included in or attached to the work
 | 
			
		||||
      (an example is provided in the Appendix below).
 | 
			
		||||
 | 
			
		||||
      "Derivative Works" shall mean any work, whether in Source or Object
 | 
			
		||||
      form, that is based on (or derived from) the Work and for which the
 | 
			
		||||
      editorial revisions, annotations, elaborations, or other modifications
 | 
			
		||||
      represent, as a whole, an original work of authorship. For the purposes
 | 
			
		||||
      of this License, Derivative Works shall not include works that remain
 | 
			
		||||
      separable from, or merely link (or bind by name) to the interfaces of,
 | 
			
		||||
      the Work and Derivative Works thereof.
 | 
			
		||||
 | 
			
		||||
      "Contribution" shall mean any work of authorship, including
 | 
			
		||||
      the original version of the Work and any modifications or additions
 | 
			
		||||
      to that Work or Derivative Works thereof, that is intentionally
 | 
			
		||||
      submitted to Licensor for inclusion in the Work by the copyright owner
 | 
			
		||||
      or by an individual or Legal Entity authorized to submit on behalf of
 | 
			
		||||
      the copyright owner. For the purposes of this definition, "submitted"
 | 
			
		||||
      means any form of electronic, verbal, or written communication sent
 | 
			
		||||
      to the Licensor or its representatives, including but not limited to
 | 
			
		||||
      communication on electronic mailing lists, source code control systems,
 | 
			
		||||
      and issue tracking systems that are managed by, or on behalf of, the
 | 
			
		||||
      Licensor for the purpose of discussing and improving the Work, but
 | 
			
		||||
      excluding communication that is conspicuously marked or otherwise
 | 
			
		||||
      designated in writing by the copyright owner as "Not a Contribution."
 | 
			
		||||
 | 
			
		||||
      "Contributor" shall mean Licensor and any individual or Legal Entity
 | 
			
		||||
      on behalf of whom a Contribution has been received by Licensor and
 | 
			
		||||
      subsequently incorporated within the Work.
 | 
			
		||||
 | 
			
		||||
   2. Grant of Copyright License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      copyright license to reproduce, prepare Derivative Works of,
 | 
			
		||||
      publicly display, publicly perform, sublicense, and distribute the
 | 
			
		||||
      Work and such Derivative Works in Source or Object form.
 | 
			
		||||
 | 
			
		||||
   3. Grant of Patent License. Subject to the terms and conditions of
 | 
			
		||||
      this License, each Contributor hereby grants to You a perpetual,
 | 
			
		||||
      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
 | 
			
		||||
      (except as stated in this section) patent license to make, have made,
 | 
			
		||||
      use, offer to sell, sell, import, and otherwise transfer the Work,
 | 
			
		||||
      where such license applies only to those patent claims licensable
 | 
			
		||||
      by such Contributor that are necessarily infringed by their
 | 
			
		||||
      Contribution(s) alone or by combination of their Contribution(s)
 | 
			
		||||
      with the Work to which such Contribution(s) was submitted. If You
 | 
			
		||||
      institute patent litigation against any entity (including a
 | 
			
		||||
      cross-claim or counterclaim in a lawsuit) alleging that the Work
 | 
			
		||||
      or a Contribution incorporated within the Work constitutes direct
 | 
			
		||||
      or contributory patent infringement, then any patent licenses
 | 
			
		||||
      granted to You under this License for that Work shall terminate
 | 
			
		||||
      as of the date such litigation is filed.
 | 
			
		||||
 | 
			
		||||
   4. Redistribution. You may reproduce and distribute copies of the
 | 
			
		||||
      Work or Derivative Works thereof in any medium, with or without
 | 
			
		||||
      modifications, and in Source or Object form, provided that You
 | 
			
		||||
      meet the following conditions:
 | 
			
		||||
 | 
			
		||||
      (a) You must give any other recipients of the Work or
 | 
			
		||||
          Derivative Works a copy of this License; and
 | 
			
		||||
 | 
			
		||||
      (b) You must cause any modified files to carry prominent notices
 | 
			
		||||
          stating that You changed the files; and
 | 
			
		||||
 | 
			
		||||
      (c) You must retain, in the Source form of any Derivative Works
 | 
			
		||||
          that You distribute, all copyright, patent, trademark, and
 | 
			
		||||
          attribution notices from the Source form of the Work,
 | 
			
		||||
          excluding those notices that do not pertain to any part of
 | 
			
		||||
          the Derivative Works; and
 | 
			
		||||
 | 
			
		||||
      (d) If the Work includes a "NOTICE" text file as part of its
 | 
			
		||||
          distribution, then any Derivative Works that You distribute must
 | 
			
		||||
          include a readable copy of the attribution notices contained
 | 
			
		||||
          within such NOTICE file, excluding those notices that do not
 | 
			
		||||
          pertain to any part of the Derivative Works, in at least one
 | 
			
		||||
          of the following places: within a NOTICE text file distributed
 | 
			
		||||
          as part of the Derivative Works; within the Source form or
 | 
			
		||||
          documentation, if provided along with the Derivative Works; or,
 | 
			
		||||
          within a display generated by the Derivative Works, if and
 | 
			
		||||
          wherever such third-party notices normally appear. The contents
 | 
			
		||||
          of the NOTICE file are for informational purposes only and
 | 
			
		||||
          do not modify the License. You may add Your own attribution
 | 
			
		||||
          notices within Derivative Works that You distribute, alongside
 | 
			
		||||
          or as an addendum to the NOTICE text from the Work, provided
 | 
			
		||||
          that such additional attribution notices cannot be construed
 | 
			
		||||
          as modifying the License.
 | 
			
		||||
 | 
			
		||||
      You may add Your own copyright statement to Your modifications and
 | 
			
		||||
      may provide additional or different license terms and conditions
 | 
			
		||||
      for use, reproduction, or distribution of Your modifications, or
 | 
			
		||||
      for any such Derivative Works as a whole, provided Your use,
 | 
			
		||||
      reproduction, and distribution of the Work otherwise complies with
 | 
			
		||||
      the conditions stated in this License.
 | 
			
		||||
 | 
			
		||||
   5. Submission of Contributions. Unless You explicitly state otherwise,
 | 
			
		||||
      any Contribution intentionally submitted for inclusion in the Work
 | 
			
		||||
      by You to the Licensor shall be under the terms and conditions of
 | 
			
		||||
      this License, without any additional terms or conditions.
 | 
			
		||||
      Notwithstanding the above, nothing herein shall supersede or modify
 | 
			
		||||
      the terms of any separate license agreement you may have executed
 | 
			
		||||
      with Licensor regarding such Contributions.
 | 
			
		||||
 | 
			
		||||
   6. Trademarks. This License does not grant permission to use the trade
 | 
			
		||||
      names, trademarks, service marks, or product names of the Licensor,
 | 
			
		||||
      except as required for reasonable and customary use in describing the
 | 
			
		||||
      origin of the Work and reproducing the content of the NOTICE file.
 | 
			
		||||
 | 
			
		||||
   7. Disclaimer of Warranty. Unless required by applicable law or
 | 
			
		||||
      agreed to in writing, Licensor provides the Work (and each
 | 
			
		||||
      Contributor provides its Contributions) on an "AS IS" BASIS,
 | 
			
		||||
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 | 
			
		||||
      implied, including, without limitation, any warranties or conditions
 | 
			
		||||
      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
 | 
			
		||||
      PARTICULAR PURPOSE. You are solely responsible for determining the
 | 
			
		||||
      appropriateness of using or redistributing the Work and assume any
 | 
			
		||||
      risks associated with Your exercise of permissions under this License.
 | 
			
		||||
 | 
			
		||||
   8. Limitation of Liability. In no event and under no legal theory,
 | 
			
		||||
      whether in tort (including negligence), contract, or otherwise,
 | 
			
		||||
      unless required by applicable law (such as deliberate and grossly
 | 
			
		||||
      negligent acts) or agreed to in writing, shall any Contributor be
 | 
			
		||||
      liable to You for damages, including any direct, indirect, special,
 | 
			
		||||
      incidental, or consequential damages of any character arising as a
 | 
			
		||||
      result of this License or out of the use or inability to use the
 | 
			
		||||
      Work (including but not limited to damages for loss of goodwill,
 | 
			
		||||
      work stoppage, computer failure or malfunction, or any and all
 | 
			
		||||
      other commercial damages or losses), even if such Contributor
 | 
			
		||||
      has been advised of the possibility of such damages.
 | 
			
		||||
 | 
			
		||||
   9. Accepting Warranty or Additional Liability. While redistributing
 | 
			
		||||
      the Work or Derivative Works thereof, You may choose to offer,
 | 
			
		||||
      and charge a fee for, acceptance of support, warranty, indemnity,
 | 
			
		||||
      or other liability obligations and/or rights consistent with this
 | 
			
		||||
      License. However, in accepting such obligations, You may act only
 | 
			
		||||
      on Your own behalf and on Your sole responsibility, not on behalf
 | 
			
		||||
      of any other Contributor, and only if You agree to indemnify,
 | 
			
		||||
      defend, and hold each Contributor harmless for any liability
 | 
			
		||||
      incurred by, or claims asserted against, such Contributor by reason
 | 
			
		||||
      of your accepting any such warranty or additional liability.
 | 
			
		||||
 | 
			
		||||
   END OF TERMS AND CONDITIONS
 | 
			
		||||
 | 
			
		||||
   APPENDIX: How to apply the Apache License to your work.
 | 
			
		||||
 | 
			
		||||
      To apply the Apache License to your work, attach the following
 | 
			
		||||
      boilerplate notice, with the fields enclosed by brackets "[]"
 | 
			
		||||
      replaced with your own identifying information. (Don't include
 | 
			
		||||
      the brackets!)  The text should be enclosed in the appropriate
 | 
			
		||||
      comment syntax for the file format. We also recommend that a
 | 
			
		||||
      file or class name and description of purpose be included on the
 | 
			
		||||
      same "printed page" as the copyright notice for easier
 | 
			
		||||
      identification within third-party archives.
 | 
			
		||||
 | 
			
		||||
   Copyright [yyyy] [name of copyright owner]
 | 
			
		||||
 | 
			
		||||
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
   you may not use this file except in compliance with the License.
 | 
			
		||||
   You may obtain a copy of the License at
 | 
			
		||||
 | 
			
		||||
       http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 | 
			
		||||
   Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
   See the License for the specific language governing permissions and
 | 
			
		||||
   limitations under the License.
 | 
			
		||||
							
								
								
									
										236
									
								
								vendor/github.com/coreos/etcd/client/auth_role.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										236
									
								
								vendor/github.com/coreos/etcd/client/auth_role.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,236 @@
 | 
			
		||||
// Copyright 2015 The etcd Authors
 | 
			
		||||
//
 | 
			
		||||
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
// you may not use this file except in compliance with the License.
 | 
			
		||||
// You may obtain a copy of the License at
 | 
			
		||||
//
 | 
			
		||||
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
//
 | 
			
		||||
// Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
// See the License for the specific language governing permissions and
 | 
			
		||||
// limitations under the License.
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/url"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Role struct {
 | 
			
		||||
	Role        string       `json:"role"`
 | 
			
		||||
	Permissions Permissions  `json:"permissions"`
 | 
			
		||||
	Grant       *Permissions `json:"grant,omitempty"`
 | 
			
		||||
	Revoke      *Permissions `json:"revoke,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Permissions struct {
 | 
			
		||||
	KV rwPermission `json:"kv"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type rwPermission struct {
 | 
			
		||||
	Read  []string `json:"read"`
 | 
			
		||||
	Write []string `json:"write"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type PermissionType int
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	ReadPermission PermissionType = iota
 | 
			
		||||
	WritePermission
 | 
			
		||||
	ReadWritePermission
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// NewAuthRoleAPI constructs a new AuthRoleAPI that uses HTTP to
 | 
			
		||||
// interact with etcd's role creation and modification features.
 | 
			
		||||
func NewAuthRoleAPI(c Client) AuthRoleAPI {
 | 
			
		||||
	return &httpAuthRoleAPI{
 | 
			
		||||
		client: c,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type AuthRoleAPI interface {
 | 
			
		||||
	// AddRole adds a role.
 | 
			
		||||
	AddRole(ctx context.Context, role string) error
 | 
			
		||||
 | 
			
		||||
	// RemoveRole removes a role.
 | 
			
		||||
	RemoveRole(ctx context.Context, role string) error
 | 
			
		||||
 | 
			
		||||
	// GetRole retrieves role details.
 | 
			
		||||
	GetRole(ctx context.Context, role string) (*Role, error)
 | 
			
		||||
 | 
			
		||||
	// GrantRoleKV grants a role some permission prefixes for the KV store.
 | 
			
		||||
	GrantRoleKV(ctx context.Context, role string, prefixes []string, permType PermissionType) (*Role, error)
 | 
			
		||||
 | 
			
		||||
	// RevokeRoleKV revokes some permission prefixes for a role on the KV store.
 | 
			
		||||
	RevokeRoleKV(ctx context.Context, role string, prefixes []string, permType PermissionType) (*Role, error)
 | 
			
		||||
 | 
			
		||||
	// ListRoles lists roles.
 | 
			
		||||
	ListRoles(ctx context.Context) ([]string, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpAuthRoleAPI struct {
 | 
			
		||||
	client httpClient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authRoleAPIAction struct {
 | 
			
		||||
	verb string
 | 
			
		||||
	name string
 | 
			
		||||
	role *Role
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authRoleAPIList struct{}
 | 
			
		||||
 | 
			
		||||
func (list *authRoleAPIList) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	u := v2AuthURL(ep, "roles", "")
 | 
			
		||||
	req, _ := http.NewRequest("GET", u.String(), nil)
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	return req
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (l *authRoleAPIAction) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	u := v2AuthURL(ep, "roles", l.name)
 | 
			
		||||
	if l.role == nil {
 | 
			
		||||
		req, _ := http.NewRequest(l.verb, u.String(), nil)
 | 
			
		||||
		return req
 | 
			
		||||
	}
 | 
			
		||||
	b, err := json.Marshal(l.role)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		panic(err)
 | 
			
		||||
	}
 | 
			
		||||
	body := bytes.NewReader(b)
 | 
			
		||||
	req, _ := http.NewRequest(l.verb, u.String(), body)
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	return req
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) ListRoles(ctx context.Context) ([]string, error) {
 | 
			
		||||
	resp, body, err := r.client.Do(ctx, &authRoleAPIList{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	var roleList struct {
 | 
			
		||||
		Roles []Role `json:"roles"`
 | 
			
		||||
	}
 | 
			
		||||
	if err = json.Unmarshal(body, &roleList); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	ret := make([]string, 0, len(roleList.Roles))
 | 
			
		||||
	for _, r := range roleList.Roles {
 | 
			
		||||
		ret = append(ret, r.Role)
 | 
			
		||||
	}
 | 
			
		||||
	return ret, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) AddRole(ctx context.Context, rolename string) error {
 | 
			
		||||
	role := &Role{
 | 
			
		||||
		Role: rolename,
 | 
			
		||||
	}
 | 
			
		||||
	return r.addRemoveRole(ctx, &authRoleAPIAction{
 | 
			
		||||
		verb: "PUT",
 | 
			
		||||
		name: rolename,
 | 
			
		||||
		role: role,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) RemoveRole(ctx context.Context, rolename string) error {
 | 
			
		||||
	return r.addRemoveRole(ctx, &authRoleAPIAction{
 | 
			
		||||
		verb: "DELETE",
 | 
			
		||||
		name: rolename,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) addRemoveRole(ctx context.Context, req *authRoleAPIAction) error {
 | 
			
		||||
	resp, body, err := r.client.Do(ctx, req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err := assertStatusCode(resp.StatusCode, http.StatusOK, http.StatusCreated); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err := json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return sec
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) GetRole(ctx context.Context, rolename string) (*Role, error) {
 | 
			
		||||
	return r.modRole(ctx, &authRoleAPIAction{
 | 
			
		||||
		verb: "GET",
 | 
			
		||||
		name: rolename,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func buildRWPermission(prefixes []string, permType PermissionType) rwPermission {
 | 
			
		||||
	var out rwPermission
 | 
			
		||||
	switch permType {
 | 
			
		||||
	case ReadPermission:
 | 
			
		||||
		out.Read = prefixes
 | 
			
		||||
	case WritePermission:
 | 
			
		||||
		out.Write = prefixes
 | 
			
		||||
	case ReadWritePermission:
 | 
			
		||||
		out.Read = prefixes
 | 
			
		||||
		out.Write = prefixes
 | 
			
		||||
	}
 | 
			
		||||
	return out
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) GrantRoleKV(ctx context.Context, rolename string, prefixes []string, permType PermissionType) (*Role, error) {
 | 
			
		||||
	rwp := buildRWPermission(prefixes, permType)
 | 
			
		||||
	role := &Role{
 | 
			
		||||
		Role: rolename,
 | 
			
		||||
		Grant: &Permissions{
 | 
			
		||||
			KV: rwp,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
	return r.modRole(ctx, &authRoleAPIAction{
 | 
			
		||||
		verb: "PUT",
 | 
			
		||||
		name: rolename,
 | 
			
		||||
		role: role,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) RevokeRoleKV(ctx context.Context, rolename string, prefixes []string, permType PermissionType) (*Role, error) {
 | 
			
		||||
	rwp := buildRWPermission(prefixes, permType)
 | 
			
		||||
	role := &Role{
 | 
			
		||||
		Role: rolename,
 | 
			
		||||
		Revoke: &Permissions{
 | 
			
		||||
			KV: rwp,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
	return r.modRole(ctx, &authRoleAPIAction{
 | 
			
		||||
		verb: "PUT",
 | 
			
		||||
		name: rolename,
 | 
			
		||||
		role: role,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *httpAuthRoleAPI) modRole(ctx context.Context, req *authRoleAPIAction) (*Role, error) {
 | 
			
		||||
	resp, body, err := r.client.Do(ctx, req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err = json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return nil, sec
 | 
			
		||||
	}
 | 
			
		||||
	var role Role
 | 
			
		||||
	if err = json.Unmarshal(body, &role); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return &role, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										319
									
								
								vendor/github.com/coreos/etcd/client/auth_user.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										319
									
								
								vendor/github.com/coreos/etcd/client/auth_user.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,319 @@
 | 
			
		||||
// Copyright 2015 The etcd Authors
 | 
			
		||||
//
 | 
			
		||||
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
// you may not use this file except in compliance with the License.
 | 
			
		||||
// You may obtain a copy of the License at
 | 
			
		||||
//
 | 
			
		||||
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
//
 | 
			
		||||
// Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
// See the License for the specific language governing permissions and
 | 
			
		||||
// limitations under the License.
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"path"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	defaultV2AuthPrefix = "/v2/auth"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type User struct {
 | 
			
		||||
	User     string   `json:"user"`
 | 
			
		||||
	Password string   `json:"password,omitempty"`
 | 
			
		||||
	Roles    []string `json:"roles"`
 | 
			
		||||
	Grant    []string `json:"grant,omitempty"`
 | 
			
		||||
	Revoke   []string `json:"revoke,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// userListEntry is the user representation given by the server for ListUsers
 | 
			
		||||
type userListEntry struct {
 | 
			
		||||
	User  string `json:"user"`
 | 
			
		||||
	Roles []Role `json:"roles"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type UserRoles struct {
 | 
			
		||||
	User  string `json:"user"`
 | 
			
		||||
	Roles []Role `json:"roles"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func v2AuthURL(ep url.URL, action string, name string) *url.URL {
 | 
			
		||||
	if name != "" {
 | 
			
		||||
		ep.Path = path.Join(ep.Path, defaultV2AuthPrefix, action, name)
 | 
			
		||||
		return &ep
 | 
			
		||||
	}
 | 
			
		||||
	ep.Path = path.Join(ep.Path, defaultV2AuthPrefix, action)
 | 
			
		||||
	return &ep
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewAuthAPI constructs a new AuthAPI that uses HTTP to
 | 
			
		||||
// interact with etcd's general auth features.
 | 
			
		||||
func NewAuthAPI(c Client) AuthAPI {
 | 
			
		||||
	return &httpAuthAPI{
 | 
			
		||||
		client: c,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type AuthAPI interface {
 | 
			
		||||
	// Enable auth.
 | 
			
		||||
	Enable(ctx context.Context) error
 | 
			
		||||
 | 
			
		||||
	// Disable auth.
 | 
			
		||||
	Disable(ctx context.Context) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpAuthAPI struct {
 | 
			
		||||
	client httpClient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *httpAuthAPI) Enable(ctx context.Context) error {
 | 
			
		||||
	return s.enableDisable(ctx, &authAPIAction{"PUT"})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *httpAuthAPI) Disable(ctx context.Context) error {
 | 
			
		||||
	return s.enableDisable(ctx, &authAPIAction{"DELETE"})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *httpAuthAPI) enableDisable(ctx context.Context, req httpAction) error {
 | 
			
		||||
	resp, body, err := s.client.Do(ctx, req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK, http.StatusCreated); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err = json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return sec
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authAPIAction struct {
 | 
			
		||||
	verb string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (l *authAPIAction) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	u := v2AuthURL(ep, "enable", "")
 | 
			
		||||
	req, _ := http.NewRequest(l.verb, u.String(), nil)
 | 
			
		||||
	return req
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authError struct {
 | 
			
		||||
	Message string `json:"message"`
 | 
			
		||||
	Code    int    `json:"-"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e authError) Error() string {
 | 
			
		||||
	return e.Message
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewAuthUserAPI constructs a new AuthUserAPI that uses HTTP to
 | 
			
		||||
// interact with etcd's user creation and modification features.
 | 
			
		||||
func NewAuthUserAPI(c Client) AuthUserAPI {
 | 
			
		||||
	return &httpAuthUserAPI{
 | 
			
		||||
		client: c,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type AuthUserAPI interface {
 | 
			
		||||
	// AddUser adds a user.
 | 
			
		||||
	AddUser(ctx context.Context, username string, password string) error
 | 
			
		||||
 | 
			
		||||
	// RemoveUser removes a user.
 | 
			
		||||
	RemoveUser(ctx context.Context, username string) error
 | 
			
		||||
 | 
			
		||||
	// GetUser retrieves user details.
 | 
			
		||||
	GetUser(ctx context.Context, username string) (*User, error)
 | 
			
		||||
 | 
			
		||||
	// GrantUser grants a user some permission roles.
 | 
			
		||||
	GrantUser(ctx context.Context, username string, roles []string) (*User, error)
 | 
			
		||||
 | 
			
		||||
	// RevokeUser revokes some permission roles from a user.
 | 
			
		||||
	RevokeUser(ctx context.Context, username string, roles []string) (*User, error)
 | 
			
		||||
 | 
			
		||||
	// ChangePassword changes the user's password.
 | 
			
		||||
	ChangePassword(ctx context.Context, username string, password string) (*User, error)
 | 
			
		||||
 | 
			
		||||
	// ListUsers lists the users.
 | 
			
		||||
	ListUsers(ctx context.Context) ([]string, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpAuthUserAPI struct {
 | 
			
		||||
	client httpClient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authUserAPIAction struct {
 | 
			
		||||
	verb     string
 | 
			
		||||
	username string
 | 
			
		||||
	user     *User
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authUserAPIList struct{}
 | 
			
		||||
 | 
			
		||||
func (list *authUserAPIList) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	u := v2AuthURL(ep, "users", "")
 | 
			
		||||
	req, _ := http.NewRequest("GET", u.String(), nil)
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	return req
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (l *authUserAPIAction) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	u := v2AuthURL(ep, "users", l.username)
 | 
			
		||||
	if l.user == nil {
 | 
			
		||||
		req, _ := http.NewRequest(l.verb, u.String(), nil)
 | 
			
		||||
		return req
 | 
			
		||||
	}
 | 
			
		||||
	b, err := json.Marshal(l.user)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		panic(err)
 | 
			
		||||
	}
 | 
			
		||||
	body := bytes.NewReader(b)
 | 
			
		||||
	req, _ := http.NewRequest(l.verb, u.String(), body)
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	return req
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) ListUsers(ctx context.Context) ([]string, error) {
 | 
			
		||||
	resp, body, err := u.client.Do(ctx, &authUserAPIList{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err = json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return nil, sec
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var userList struct {
 | 
			
		||||
		Users []userListEntry `json:"users"`
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err = json.Unmarshal(body, &userList); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	ret := make([]string, 0, len(userList.Users))
 | 
			
		||||
	for _, u := range userList.Users {
 | 
			
		||||
		ret = append(ret, u.User)
 | 
			
		||||
	}
 | 
			
		||||
	return ret, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) AddUser(ctx context.Context, username string, password string) error {
 | 
			
		||||
	user := &User{
 | 
			
		||||
		User:     username,
 | 
			
		||||
		Password: password,
 | 
			
		||||
	}
 | 
			
		||||
	return u.addRemoveUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "PUT",
 | 
			
		||||
		username: username,
 | 
			
		||||
		user:     user,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) RemoveUser(ctx context.Context, username string) error {
 | 
			
		||||
	return u.addRemoveUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "DELETE",
 | 
			
		||||
		username: username,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) addRemoveUser(ctx context.Context, req *authUserAPIAction) error {
 | 
			
		||||
	resp, body, err := u.client.Do(ctx, req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK, http.StatusCreated); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err = json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		return sec
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) GetUser(ctx context.Context, username string) (*User, error) {
 | 
			
		||||
	return u.modUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "GET",
 | 
			
		||||
		username: username,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) GrantUser(ctx context.Context, username string, roles []string) (*User, error) {
 | 
			
		||||
	user := &User{
 | 
			
		||||
		User:  username,
 | 
			
		||||
		Grant: roles,
 | 
			
		||||
	}
 | 
			
		||||
	return u.modUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "PUT",
 | 
			
		||||
		username: username,
 | 
			
		||||
		user:     user,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) RevokeUser(ctx context.Context, username string, roles []string) (*User, error) {
 | 
			
		||||
	user := &User{
 | 
			
		||||
		User:   username,
 | 
			
		||||
		Revoke: roles,
 | 
			
		||||
	}
 | 
			
		||||
	return u.modUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "PUT",
 | 
			
		||||
		username: username,
 | 
			
		||||
		user:     user,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) ChangePassword(ctx context.Context, username string, password string) (*User, error) {
 | 
			
		||||
	user := &User{
 | 
			
		||||
		User:     username,
 | 
			
		||||
		Password: password,
 | 
			
		||||
	}
 | 
			
		||||
	return u.modUser(ctx, &authUserAPIAction{
 | 
			
		||||
		verb:     "PUT",
 | 
			
		||||
		username: username,
 | 
			
		||||
		user:     user,
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (u *httpAuthUserAPI) modUser(ctx context.Context, req *authUserAPIAction) (*User, error) {
 | 
			
		||||
	resp, body, err := u.client.Do(ctx, req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if err = assertStatusCode(resp.StatusCode, http.StatusOK); err != nil {
 | 
			
		||||
		var sec authError
 | 
			
		||||
		err = json.Unmarshal(body, &sec)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return nil, sec
 | 
			
		||||
	}
 | 
			
		||||
	var user User
 | 
			
		||||
	if err = json.Unmarshal(body, &user); err != nil {
 | 
			
		||||
		var userR UserRoles
 | 
			
		||||
		if urerr := json.Unmarshal(body, &userR); urerr != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		user.User = userR.User
 | 
			
		||||
		for _, r := range userR.Roles {
 | 
			
		||||
			user.Roles = append(user.Roles, r.Role)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return &user, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										18
									
								
								vendor/github.com/coreos/etcd/client/cancelreq.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/coreos/etcd/client/cancelreq.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
			
		||||
// Copyright 2015 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.
 | 
			
		||||
 | 
			
		||||
// borrowed from golang/net/context/ctxhttp/cancelreq.go
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import "net/http"
 | 
			
		||||
 | 
			
		||||
func requestCanceler(tr CancelableTransport, req *http.Request) func() {
 | 
			
		||||
	ch := make(chan struct{})
 | 
			
		||||
	req.Cancel = ch
 | 
			
		||||
 | 
			
		||||
	return func() {
 | 
			
		||||
		close(ch)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										710
									
								
								vendor/github.com/coreos/etcd/client/client.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										710
									
								
								vendor/github.com/coreos/etcd/client/client.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,710 @@
 | 
			
		||||
// Copyright 2015 The etcd Authors
 | 
			
		||||
//
 | 
			
		||||
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
// you may not use this file except in compliance with the License.
 | 
			
		||||
// You may obtain a copy of the License at
 | 
			
		||||
//
 | 
			
		||||
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
//
 | 
			
		||||
// Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
// See the License for the specific language governing permissions and
 | 
			
		||||
// limitations under the License.
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"math/rand"
 | 
			
		||||
	"net"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"sort"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/coreos/etcd/version"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	ErrNoEndpoints           = errors.New("client: no endpoints available")
 | 
			
		||||
	ErrTooManyRedirects      = errors.New("client: too many redirects")
 | 
			
		||||
	ErrClusterUnavailable    = errors.New("client: etcd cluster is unavailable or misconfigured")
 | 
			
		||||
	ErrNoLeaderEndpoint      = errors.New("client: no leader endpoint available")
 | 
			
		||||
	errTooManyRedirectChecks = errors.New("client: too many redirect checks")
 | 
			
		||||
 | 
			
		||||
	// oneShotCtxValue is set on a context using WithValue(&oneShotValue) so
 | 
			
		||||
	// that Do() will not retry a request
 | 
			
		||||
	oneShotCtxValue interface{}
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var DefaultRequestTimeout = 5 * time.Second
 | 
			
		||||
 | 
			
		||||
var DefaultTransport CancelableTransport = &http.Transport{
 | 
			
		||||
	Proxy: http.ProxyFromEnvironment,
 | 
			
		||||
	Dial: (&net.Dialer{
 | 
			
		||||
		Timeout:   30 * time.Second,
 | 
			
		||||
		KeepAlive: 30 * time.Second,
 | 
			
		||||
	}).Dial,
 | 
			
		||||
	TLSHandshakeTimeout: 10 * time.Second,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type EndpointSelectionMode int
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	// EndpointSelectionRandom is the default value of the 'SelectionMode'.
 | 
			
		||||
	// As the name implies, the client object will pick a node from the members
 | 
			
		||||
	// of the cluster in a random fashion. If the cluster has three members, A, B,
 | 
			
		||||
	// and C, the client picks any node from its three members as its request
 | 
			
		||||
	// destination.
 | 
			
		||||
	EndpointSelectionRandom EndpointSelectionMode = iota
 | 
			
		||||
 | 
			
		||||
	// If 'SelectionMode' is set to 'EndpointSelectionPrioritizeLeader',
 | 
			
		||||
	// requests are sent directly to the cluster leader. This reduces
 | 
			
		||||
	// forwarding roundtrips compared to making requests to etcd followers
 | 
			
		||||
	// who then forward them to the cluster leader. In the event of a leader
 | 
			
		||||
	// failure, however, clients configured this way cannot prioritize among
 | 
			
		||||
	// the remaining etcd followers. Therefore, when a client sets 'SelectionMode'
 | 
			
		||||
	// to 'EndpointSelectionPrioritizeLeader', it must use 'client.AutoSync()' to
 | 
			
		||||
	// maintain its knowledge of current cluster state.
 | 
			
		||||
	//
 | 
			
		||||
	// This mode should be used with Client.AutoSync().
 | 
			
		||||
	EndpointSelectionPrioritizeLeader
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Config struct {
 | 
			
		||||
	// Endpoints defines a set of URLs (schemes, hosts and ports only)
 | 
			
		||||
	// that can be used to communicate with a logical etcd cluster. For
 | 
			
		||||
	// example, a three-node cluster could be provided like so:
 | 
			
		||||
	//
 | 
			
		||||
	// 	Endpoints: []string{
 | 
			
		||||
	//		"http://node1.example.com:2379",
 | 
			
		||||
	//		"http://node2.example.com:2379",
 | 
			
		||||
	//		"http://node3.example.com:2379",
 | 
			
		||||
	//	}
 | 
			
		||||
	//
 | 
			
		||||
	// If multiple endpoints are provided, the Client will attempt to
 | 
			
		||||
	// use them all in the event that one or more of them are unusable.
 | 
			
		||||
	//
 | 
			
		||||
	// If Client.Sync is ever called, the Client may cache an alternate
 | 
			
		||||
	// set of endpoints to continue operation.
 | 
			
		||||
	Endpoints []string
 | 
			
		||||
 | 
			
		||||
	// Transport is used by the Client to drive HTTP requests. If not
 | 
			
		||||
	// provided, DefaultTransport will be used.
 | 
			
		||||
	Transport CancelableTransport
 | 
			
		||||
 | 
			
		||||
	// CheckRedirect specifies the policy for handling HTTP redirects.
 | 
			
		||||
	// If CheckRedirect is not nil, the Client calls it before
 | 
			
		||||
	// following an HTTP redirect. The sole argument is the number of
 | 
			
		||||
	// requests that have already been made. If CheckRedirect returns
 | 
			
		||||
	// an error, Client.Do will not make any further requests and return
 | 
			
		||||
	// the error back it to the caller.
 | 
			
		||||
	//
 | 
			
		||||
	// If CheckRedirect is nil, the Client uses its default policy,
 | 
			
		||||
	// which is to stop after 10 consecutive requests.
 | 
			
		||||
	CheckRedirect CheckRedirectFunc
 | 
			
		||||
 | 
			
		||||
	// Username specifies the user credential to add as an authorization header
 | 
			
		||||
	Username string
 | 
			
		||||
 | 
			
		||||
	// Password is the password for the specified user to add as an authorization header
 | 
			
		||||
	// to the request.
 | 
			
		||||
	Password string
 | 
			
		||||
 | 
			
		||||
	// HeaderTimeoutPerRequest specifies the time limit to wait for response
 | 
			
		||||
	// header in a single request made by the Client. The timeout includes
 | 
			
		||||
	// connection time, any redirects, and header wait time.
 | 
			
		||||
	//
 | 
			
		||||
	// For non-watch GET request, server returns the response body immediately.
 | 
			
		||||
	// For PUT/POST/DELETE request, server will attempt to commit request
 | 
			
		||||
	// before responding, which is expected to take `100ms + 2 * RTT`.
 | 
			
		||||
	// For watch request, server returns the header immediately to notify Client
 | 
			
		||||
	// watch start. But if server is behind some kind of proxy, the response
 | 
			
		||||
	// header may be cached at proxy, and Client cannot rely on this behavior.
 | 
			
		||||
	//
 | 
			
		||||
	// Especially, wait request will ignore this timeout.
 | 
			
		||||
	//
 | 
			
		||||
	// One API call may send multiple requests to different etcd servers until it
 | 
			
		||||
	// succeeds. Use context of the API to specify the overall timeout.
 | 
			
		||||
	//
 | 
			
		||||
	// A HeaderTimeoutPerRequest of zero means no timeout.
 | 
			
		||||
	HeaderTimeoutPerRequest time.Duration
 | 
			
		||||
 | 
			
		||||
	// SelectionMode is an EndpointSelectionMode enum that specifies the
 | 
			
		||||
	// policy for choosing the etcd cluster node to which requests are sent.
 | 
			
		||||
	SelectionMode EndpointSelectionMode
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (cfg *Config) transport() CancelableTransport {
 | 
			
		||||
	if cfg.Transport == nil {
 | 
			
		||||
		return DefaultTransport
 | 
			
		||||
	}
 | 
			
		||||
	return cfg.Transport
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (cfg *Config) checkRedirect() CheckRedirectFunc {
 | 
			
		||||
	if cfg.CheckRedirect == nil {
 | 
			
		||||
		return DefaultCheckRedirect
 | 
			
		||||
	}
 | 
			
		||||
	return cfg.CheckRedirect
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CancelableTransport mimics net/http.Transport, but requires that
 | 
			
		||||
// the object also support request cancellation.
 | 
			
		||||
type CancelableTransport interface {
 | 
			
		||||
	http.RoundTripper
 | 
			
		||||
	CancelRequest(req *http.Request)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CheckRedirectFunc func(via int) error
 | 
			
		||||
 | 
			
		||||
// DefaultCheckRedirect follows up to 10 redirects, but no more.
 | 
			
		||||
var DefaultCheckRedirect CheckRedirectFunc = func(via int) error {
 | 
			
		||||
	if via > 10 {
 | 
			
		||||
		return ErrTooManyRedirects
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Client interface {
 | 
			
		||||
	// Sync updates the internal cache of the etcd cluster's membership.
 | 
			
		||||
	Sync(context.Context) error
 | 
			
		||||
 | 
			
		||||
	// AutoSync periodically calls Sync() every given interval.
 | 
			
		||||
	// The recommended sync interval is 10 seconds to 1 minute, which does
 | 
			
		||||
	// not bring too much overhead to server and makes client catch up the
 | 
			
		||||
	// cluster change in time.
 | 
			
		||||
	//
 | 
			
		||||
	// The example to use it:
 | 
			
		||||
	//
 | 
			
		||||
	//  for {
 | 
			
		||||
	//      err := client.AutoSync(ctx, 10*time.Second)
 | 
			
		||||
	//      if err == context.DeadlineExceeded || err == context.Canceled {
 | 
			
		||||
	//          break
 | 
			
		||||
	//      }
 | 
			
		||||
	//      log.Print(err)
 | 
			
		||||
	//  }
 | 
			
		||||
	AutoSync(context.Context, time.Duration) error
 | 
			
		||||
 | 
			
		||||
	// Endpoints returns a copy of the current set of API endpoints used
 | 
			
		||||
	// by Client to resolve HTTP requests. If Sync has ever been called,
 | 
			
		||||
	// this may differ from the initial Endpoints provided in the Config.
 | 
			
		||||
	Endpoints() []string
 | 
			
		||||
 | 
			
		||||
	// SetEndpoints sets the set of API endpoints used by Client to resolve
 | 
			
		||||
	// HTTP requests. If the given endpoints are not valid, an error will be
 | 
			
		||||
	// returned
 | 
			
		||||
	SetEndpoints(eps []string) error
 | 
			
		||||
 | 
			
		||||
	// GetVersion retrieves the current etcd server and cluster version
 | 
			
		||||
	GetVersion(ctx context.Context) (*version.Versions, error)
 | 
			
		||||
 | 
			
		||||
	httpClient
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func New(cfg Config) (Client, error) {
 | 
			
		||||
	c := &httpClusterClient{
 | 
			
		||||
		clientFactory: newHTTPClientFactory(cfg.transport(), cfg.checkRedirect(), cfg.HeaderTimeoutPerRequest),
 | 
			
		||||
		rand:          rand.New(rand.NewSource(int64(time.Now().Nanosecond()))),
 | 
			
		||||
		selectionMode: cfg.SelectionMode,
 | 
			
		||||
	}
 | 
			
		||||
	if cfg.Username != "" {
 | 
			
		||||
		c.credentials = &credentials{
 | 
			
		||||
			username: cfg.Username,
 | 
			
		||||
			password: cfg.Password,
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	if err := c.SetEndpoints(cfg.Endpoints); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return c, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpClient interface {
 | 
			
		||||
	Do(context.Context, httpAction) (*http.Response, []byte, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func newHTTPClientFactory(tr CancelableTransport, cr CheckRedirectFunc, headerTimeout time.Duration) httpClientFactory {
 | 
			
		||||
	return func(ep url.URL) httpClient {
 | 
			
		||||
		return &redirectFollowingHTTPClient{
 | 
			
		||||
			checkRedirect: cr,
 | 
			
		||||
			client: &simpleHTTPClient{
 | 
			
		||||
				transport:     tr,
 | 
			
		||||
				endpoint:      ep,
 | 
			
		||||
				headerTimeout: headerTimeout,
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type credentials struct {
 | 
			
		||||
	username string
 | 
			
		||||
	password string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpClientFactory func(url.URL) httpClient
 | 
			
		||||
 | 
			
		||||
type httpAction interface {
 | 
			
		||||
	HTTPRequest(url.URL) *http.Request
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type httpClusterClient struct {
 | 
			
		||||
	clientFactory httpClientFactory
 | 
			
		||||
	endpoints     []url.URL
 | 
			
		||||
	pinned        int
 | 
			
		||||
	credentials   *credentials
 | 
			
		||||
	sync.RWMutex
 | 
			
		||||
	rand          *rand.Rand
 | 
			
		||||
	selectionMode EndpointSelectionMode
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) getLeaderEndpoint(ctx context.Context, eps []url.URL) (string, error) {
 | 
			
		||||
	ceps := make([]url.URL, len(eps))
 | 
			
		||||
	copy(ceps, eps)
 | 
			
		||||
 | 
			
		||||
	// To perform a lookup on the new endpoint list without using the current
 | 
			
		||||
	// client, we'll copy it
 | 
			
		||||
	clientCopy := &httpClusterClient{
 | 
			
		||||
		clientFactory: c.clientFactory,
 | 
			
		||||
		credentials:   c.credentials,
 | 
			
		||||
		rand:          c.rand,
 | 
			
		||||
 | 
			
		||||
		pinned:    0,
 | 
			
		||||
		endpoints: ceps,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	mAPI := NewMembersAPI(clientCopy)
 | 
			
		||||
	leader, err := mAPI.Leader(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	if len(leader.ClientURLs) == 0 {
 | 
			
		||||
		return "", ErrNoLeaderEndpoint
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return leader.ClientURLs[0], nil // TODO: how to handle multiple client URLs?
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) parseEndpoints(eps []string) ([]url.URL, error) {
 | 
			
		||||
	if len(eps) == 0 {
 | 
			
		||||
		return []url.URL{}, ErrNoEndpoints
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	neps := make([]url.URL, len(eps))
 | 
			
		||||
	for i, ep := range eps {
 | 
			
		||||
		u, err := url.Parse(ep)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return []url.URL{}, err
 | 
			
		||||
		}
 | 
			
		||||
		neps[i] = *u
 | 
			
		||||
	}
 | 
			
		||||
	return neps, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) SetEndpoints(eps []string) error {
 | 
			
		||||
	neps, err := c.parseEndpoints(eps)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	c.Lock()
 | 
			
		||||
	defer c.Unlock()
 | 
			
		||||
 | 
			
		||||
	c.endpoints = shuffleEndpoints(c.rand, neps)
 | 
			
		||||
	// We're not doing anything for PrioritizeLeader here. This is
 | 
			
		||||
	// due to not having a context meaning we can't call getLeaderEndpoint
 | 
			
		||||
	// However, if you're using PrioritizeLeader, you've already been told
 | 
			
		||||
	// to regularly call sync, where we do have a ctx, and can figure the
 | 
			
		||||
	// leader. PrioritizeLeader is also quite a loose guarantee, so deal
 | 
			
		||||
	// with it
 | 
			
		||||
	c.pinned = 0
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) Do(ctx context.Context, act httpAction) (*http.Response, []byte, error) {
 | 
			
		||||
	action := act
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	leps := len(c.endpoints)
 | 
			
		||||
	eps := make([]url.URL, leps)
 | 
			
		||||
	n := copy(eps, c.endpoints)
 | 
			
		||||
	pinned := c.pinned
 | 
			
		||||
 | 
			
		||||
	if c.credentials != nil {
 | 
			
		||||
		action = &authedAction{
 | 
			
		||||
			act:         act,
 | 
			
		||||
			credentials: *c.credentials,
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	c.RUnlock()
 | 
			
		||||
 | 
			
		||||
	if leps == 0 {
 | 
			
		||||
		return nil, nil, ErrNoEndpoints
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if leps != n {
 | 
			
		||||
		return nil, nil, errors.New("unable to pick endpoint: copy failed")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var resp *http.Response
 | 
			
		||||
	var body []byte
 | 
			
		||||
	var err error
 | 
			
		||||
	cerr := &ClusterError{}
 | 
			
		||||
	isOneShot := ctx.Value(&oneShotCtxValue) != nil
 | 
			
		||||
 | 
			
		||||
	for i := pinned; i < leps+pinned; i++ {
 | 
			
		||||
		k := i % leps
 | 
			
		||||
		hc := c.clientFactory(eps[k])
 | 
			
		||||
		resp, body, err = hc.Do(ctx, action)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			cerr.Errors = append(cerr.Errors, err)
 | 
			
		||||
			if err == ctx.Err() {
 | 
			
		||||
				return nil, nil, ctx.Err()
 | 
			
		||||
			}
 | 
			
		||||
			if err == context.Canceled || err == context.DeadlineExceeded {
 | 
			
		||||
				return nil, nil, err
 | 
			
		||||
			}
 | 
			
		||||
		} else if resp.StatusCode/100 == 5 {
 | 
			
		||||
			switch resp.StatusCode {
 | 
			
		||||
			case http.StatusInternalServerError, http.StatusServiceUnavailable:
 | 
			
		||||
				// TODO: make sure this is a no leader response
 | 
			
		||||
				cerr.Errors = append(cerr.Errors, fmt.Errorf("client: etcd member %s has no leader", eps[k].String()))
 | 
			
		||||
			default:
 | 
			
		||||
				cerr.Errors = append(cerr.Errors, fmt.Errorf("client: etcd member %s returns server error [%s]", eps[k].String(), http.StatusText(resp.StatusCode)))
 | 
			
		||||
			}
 | 
			
		||||
			err = cerr.Errors[0]
 | 
			
		||||
		}
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			if !isOneShot {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			c.Lock()
 | 
			
		||||
			c.pinned = (k + 1) % leps
 | 
			
		||||
			c.Unlock()
 | 
			
		||||
			return nil, nil, err
 | 
			
		||||
		}
 | 
			
		||||
		if k != pinned {
 | 
			
		||||
			c.Lock()
 | 
			
		||||
			c.pinned = k
 | 
			
		||||
			c.Unlock()
 | 
			
		||||
		}
 | 
			
		||||
		return resp, body, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil, nil, cerr
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) Endpoints() []string {
 | 
			
		||||
	c.RLock()
 | 
			
		||||
	defer c.RUnlock()
 | 
			
		||||
 | 
			
		||||
	eps := make([]string, len(c.endpoints))
 | 
			
		||||
	for i, ep := range c.endpoints {
 | 
			
		||||
		eps[i] = ep.String()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return eps
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) Sync(ctx context.Context) error {
 | 
			
		||||
	mAPI := NewMembersAPI(c)
 | 
			
		||||
	ms, err := mAPI.List(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var eps []string
 | 
			
		||||
	for _, m := range ms {
 | 
			
		||||
		eps = append(eps, m.ClientURLs...)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	neps, err := c.parseEndpoints(eps)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	npin := 0
 | 
			
		||||
 | 
			
		||||
	switch c.selectionMode {
 | 
			
		||||
	case EndpointSelectionRandom:
 | 
			
		||||
		c.RLock()
 | 
			
		||||
		eq := endpointsEqual(c.endpoints, neps)
 | 
			
		||||
		c.RUnlock()
 | 
			
		||||
 | 
			
		||||
		if eq {
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
		// When items in the endpoint list changes, we choose a new pin
 | 
			
		||||
		neps = shuffleEndpoints(c.rand, neps)
 | 
			
		||||
	case EndpointSelectionPrioritizeLeader:
 | 
			
		||||
		nle, err := c.getLeaderEndpoint(ctx, neps)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return ErrNoLeaderEndpoint
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		for i, n := range neps {
 | 
			
		||||
			if n.String() == nle {
 | 
			
		||||
				npin = i
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	default:
 | 
			
		||||
		return fmt.Errorf("invalid endpoint selection mode: %d", c.selectionMode)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	c.Lock()
 | 
			
		||||
	defer c.Unlock()
 | 
			
		||||
	c.endpoints = neps
 | 
			
		||||
	c.pinned = npin
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) AutoSync(ctx context.Context, interval time.Duration) error {
 | 
			
		||||
	ticker := time.NewTicker(interval)
 | 
			
		||||
	defer ticker.Stop()
 | 
			
		||||
	for {
 | 
			
		||||
		err := c.Sync(ctx)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		select {
 | 
			
		||||
		case <-ctx.Done():
 | 
			
		||||
			return ctx.Err()
 | 
			
		||||
		case <-ticker.C:
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *httpClusterClient) GetVersion(ctx context.Context) (*version.Versions, error) {
 | 
			
		||||
	act := &getAction{Prefix: "/version"}
 | 
			
		||||
 | 
			
		||||
	resp, body, err := c.Do(ctx, act)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	switch resp.StatusCode {
 | 
			
		||||
	case http.StatusOK:
 | 
			
		||||
		if len(body) == 0 {
 | 
			
		||||
			return nil, ErrEmptyBody
 | 
			
		||||
		}
 | 
			
		||||
		var vresp version.Versions
 | 
			
		||||
		if err := json.Unmarshal(body, &vresp); err != nil {
 | 
			
		||||
			return nil, ErrInvalidJSON
 | 
			
		||||
		}
 | 
			
		||||
		return &vresp, nil
 | 
			
		||||
	default:
 | 
			
		||||
		var etcdErr Error
 | 
			
		||||
		if err := json.Unmarshal(body, &etcdErr); err != nil {
 | 
			
		||||
			return nil, ErrInvalidJSON
 | 
			
		||||
		}
 | 
			
		||||
		return nil, etcdErr
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type roundTripResponse struct {
 | 
			
		||||
	resp *http.Response
 | 
			
		||||
	err  error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type simpleHTTPClient struct {
 | 
			
		||||
	transport     CancelableTransport
 | 
			
		||||
	endpoint      url.URL
 | 
			
		||||
	headerTimeout time.Duration
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *simpleHTTPClient) Do(ctx context.Context, act httpAction) (*http.Response, []byte, error) {
 | 
			
		||||
	req := act.HTTPRequest(c.endpoint)
 | 
			
		||||
 | 
			
		||||
	if err := printcURL(req); err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	isWait := false
 | 
			
		||||
	if req != nil && req.URL != nil {
 | 
			
		||||
		ws := req.URL.Query().Get("wait")
 | 
			
		||||
		if len(ws) != 0 {
 | 
			
		||||
			var err error
 | 
			
		||||
			isWait, err = strconv.ParseBool(ws)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return nil, nil, fmt.Errorf("wrong wait value %s (%v for %+v)", ws, err, req)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var hctx context.Context
 | 
			
		||||
	var hcancel context.CancelFunc
 | 
			
		||||
	if !isWait && c.headerTimeout > 0 {
 | 
			
		||||
		hctx, hcancel = context.WithTimeout(ctx, c.headerTimeout)
 | 
			
		||||
	} else {
 | 
			
		||||
		hctx, hcancel = context.WithCancel(ctx)
 | 
			
		||||
	}
 | 
			
		||||
	defer hcancel()
 | 
			
		||||
 | 
			
		||||
	reqcancel := requestCanceler(c.transport, req)
 | 
			
		||||
 | 
			
		||||
	rtchan := make(chan roundTripResponse, 1)
 | 
			
		||||
	go func() {
 | 
			
		||||
		resp, err := c.transport.RoundTrip(req)
 | 
			
		||||
		rtchan <- roundTripResponse{resp: resp, err: err}
 | 
			
		||||
		close(rtchan)
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	var resp *http.Response
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	select {
 | 
			
		||||
	case rtresp := <-rtchan:
 | 
			
		||||
		resp, err = rtresp.resp, rtresp.err
 | 
			
		||||
	case <-hctx.Done():
 | 
			
		||||
		// cancel and wait for request to actually exit before continuing
 | 
			
		||||
		reqcancel()
 | 
			
		||||
		rtresp := <-rtchan
 | 
			
		||||
		resp = rtresp.resp
 | 
			
		||||
		switch {
 | 
			
		||||
		case ctx.Err() != nil:
 | 
			
		||||
			err = ctx.Err()
 | 
			
		||||
		case hctx.Err() != nil:
 | 
			
		||||
			err = fmt.Errorf("client: endpoint %s exceeded header timeout", c.endpoint.String())
 | 
			
		||||
		default:
 | 
			
		||||
			panic("failed to get error from context")
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// always check for resp nil-ness to deal with possible
 | 
			
		||||
	// race conditions between channels above
 | 
			
		||||
	defer func() {
 | 
			
		||||
		if resp != nil {
 | 
			
		||||
			resp.Body.Close()
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var body []byte
 | 
			
		||||
	done := make(chan struct{})
 | 
			
		||||
	go func() {
 | 
			
		||||
		body, err = ioutil.ReadAll(resp.Body)
 | 
			
		||||
		done <- struct{}{}
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	select {
 | 
			
		||||
	case <-ctx.Done():
 | 
			
		||||
		resp.Body.Close()
 | 
			
		||||
		<-done
 | 
			
		||||
		return nil, nil, ctx.Err()
 | 
			
		||||
	case <-done:
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return resp, body, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type authedAction struct {
 | 
			
		||||
	act         httpAction
 | 
			
		||||
	credentials credentials
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *authedAction) HTTPRequest(url url.URL) *http.Request {
 | 
			
		||||
	r := a.act.HTTPRequest(url)
 | 
			
		||||
	r.SetBasicAuth(a.credentials.username, a.credentials.password)
 | 
			
		||||
	return r
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type redirectFollowingHTTPClient struct {
 | 
			
		||||
	client        httpClient
 | 
			
		||||
	checkRedirect CheckRedirectFunc
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *redirectFollowingHTTPClient) Do(ctx context.Context, act httpAction) (*http.Response, []byte, error) {
 | 
			
		||||
	next := act
 | 
			
		||||
	for i := 0; i < 100; i++ {
 | 
			
		||||
		if i > 0 {
 | 
			
		||||
			if err := r.checkRedirect(i); err != nil {
 | 
			
		||||
				return nil, nil, err
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		resp, body, err := r.client.Do(ctx, next)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, nil, err
 | 
			
		||||
		}
 | 
			
		||||
		if resp.StatusCode/100 == 3 {
 | 
			
		||||
			hdr := resp.Header.Get("Location")
 | 
			
		||||
			if hdr == "" {
 | 
			
		||||
				return nil, nil, fmt.Errorf("Location header not set")
 | 
			
		||||
			}
 | 
			
		||||
			loc, err := url.Parse(hdr)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return nil, nil, fmt.Errorf("Location header not valid URL: %s", hdr)
 | 
			
		||||
			}
 | 
			
		||||
			next = &redirectedHTTPAction{
 | 
			
		||||
				action:   act,
 | 
			
		||||
				location: *loc,
 | 
			
		||||
			}
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		return resp, body, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil, nil, errTooManyRedirectChecks
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type redirectedHTTPAction struct {
 | 
			
		||||
	action   httpAction
 | 
			
		||||
	location url.URL
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *redirectedHTTPAction) HTTPRequest(ep url.URL) *http.Request {
 | 
			
		||||
	orig := r.action.HTTPRequest(ep)
 | 
			
		||||
	orig.URL = &r.location
 | 
			
		||||
	return orig
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func shuffleEndpoints(r *rand.Rand, eps []url.URL) []url.URL {
 | 
			
		||||
	// copied from Go 1.9<= rand.Rand.Perm
 | 
			
		||||
	n := len(eps)
 | 
			
		||||
	p := make([]int, n)
 | 
			
		||||
	for i := 0; i < n; i++ {
 | 
			
		||||
		j := r.Intn(i + 1)
 | 
			
		||||
		p[i] = p[j]
 | 
			
		||||
		p[j] = i
 | 
			
		||||
	}
 | 
			
		||||
	neps := make([]url.URL, n)
 | 
			
		||||
	for i, k := range p {
 | 
			
		||||
		neps[i] = eps[k]
 | 
			
		||||
	}
 | 
			
		||||
	return neps
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func endpointsEqual(left, right []url.URL) bool {
 | 
			
		||||
	if len(left) != len(right) {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sLeft := make([]string, len(left))
 | 
			
		||||
	sRight := make([]string, len(right))
 | 
			
		||||
	for i, l := range left {
 | 
			
		||||
		sLeft[i] = l.String()
 | 
			
		||||
	}
 | 
			
		||||
	for i, r := range right {
 | 
			
		||||
		sRight[i] = r.String()
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sort.Strings(sLeft)
 | 
			
		||||
	sort.Strings(sRight)
 | 
			
		||||
	for i := range sLeft {
 | 
			
		||||
		if sLeft[i] != sRight[i] {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										37
									
								
								vendor/github.com/coreos/etcd/client/cluster_error.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								vendor/github.com/coreos/etcd/client/cluster_error.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,37 @@
 | 
			
		||||
// Copyright 2015 The etcd Authors
 | 
			
		||||
//
 | 
			
		||||
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
// you may not use this file except in compliance with the License.
 | 
			
		||||
// You may obtain a copy of the License at
 | 
			
		||||
//
 | 
			
		||||
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
//
 | 
			
		||||
// Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
// See the License for the specific language governing permissions and
 | 
			
		||||
// limitations under the License.
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import "fmt"
 | 
			
		||||
 | 
			
		||||
type ClusterError struct {
 | 
			
		||||
	Errors []error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ce *ClusterError) Error() string {
 | 
			
		||||
	s := ErrClusterUnavailable.Error()
 | 
			
		||||
	for i, e := range ce.Errors {
 | 
			
		||||
		s += fmt.Sprintf("; error #%d: %s\n", i, e)
 | 
			
		||||
	}
 | 
			
		||||
	return s
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ce *ClusterError) Detail() string {
 | 
			
		||||
	s := ""
 | 
			
		||||
	for i, e := range ce.Errors {
 | 
			
		||||
		s += fmt.Sprintf("error #%d: %s\n", i, e)
 | 
			
		||||
	}
 | 
			
		||||
	return s
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										70
									
								
								vendor/github.com/coreos/etcd/client/curl.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								vendor/github.com/coreos/etcd/client/curl.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,70 @@
 | 
			
		||||
// Copyright 2015 The etcd Authors
 | 
			
		||||
//
 | 
			
		||||
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
// you may not use this file except in compliance with the License.
 | 
			
		||||
// You may obtain a copy of the License at
 | 
			
		||||
//
 | 
			
		||||
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
//
 | 
			
		||||
// Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
// See the License for the specific language governing permissions and
 | 
			
		||||
// limitations under the License.
 | 
			
		||||
 | 
			
		||||
package client
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"os"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	cURLDebug = false
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func EnablecURLDebug() {
 | 
			
		||||
	cURLDebug = true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func DisablecURLDebug() {
 | 
			
		||||
	cURLDebug = false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// printcURL prints the cURL equivalent request to stderr.
 | 
			
		||||
// It returns an error if the body of the request cannot
 | 
			
		||||
// be read.
 | 
			
		||||
// The caller MUST cancel the request if there is an error.
 | 
			
		||||
func printcURL(req *http.Request) error {
 | 
			
		||||
	if !cURLDebug {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	var (
 | 
			
		||||
		command string
 | 
			
		||||
		b       []byte
 | 
			
		||||
		err     error
 | 
			
		||||
	)
 | 
			
		||||
 | 
			
		||||
	if req.URL != nil {
 | 
			
		||||
		command = fmt.Sprintf("curl -X %s %s", req.Method, req.URL.String())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if req.Body != nil {
 | 
			
		||||
		b, err = ioutil.ReadAll(req.Body)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
		command += fmt.Sprintf(" -d %q", string(b))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	fmt.Fprintf(os.Stderr, "cURL Command: %s\n", command)
 | 
			
		||||
 | 
			
		||||
	// reset body
 | 
			
		||||
	body := bytes.NewBuffer(b)
 | 
			
		||||
	req.Body = ioutil.NopCloser(body)
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user