Compare commits
	
		
			466 Commits
		
	
	
		
			v1.0.0-rc1
			...
			v1.12.1
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | d2cfd235ef | ||
|   | f8bf7f8d76 | ||
|   | 8432330cb2 | ||
|   | 02577a2b5c | ||
|   | 73501739d5 | ||
|   | ba674af5d4 | ||
|   | 3c85d937c6 | ||
|   | c6fd65d1d7 | ||
|   | 0795906533 | ||
|   | a2b45bc799 | ||
|   | 757657f29c | ||
|   | 219c7659e1 | ||
|   | ae32bae791 | ||
|   | 57eba77561 | ||
|   | d5bc7c4343 | ||
|   | 32f57b7c26 | ||
|   | 692bb8faa7 | ||
|   | 455a0fc239 | ||
|   | b2cbd13251 | ||
|   | ce21ba1545 | ||
|   | c89085bf44 | ||
|   | 4254ed3c63 | ||
|   | 85564a35fd | ||
|   | 09713d40ba | ||
|   | 16d5aeac7c | ||
|   | e19ba5a06a | ||
|   | f7a5077d5d | ||
|   | f8dc24bc09 | ||
|   | e9419f10d3 | ||
|   | cded603c27 | ||
|   | d2ae3ebf9e | ||
|   | 730ccdd456 | ||
|   | 2f042ad915 | ||
|   | ba70691877 | ||
|   | ed11686a99 | ||
|   | 5c50d86908 | ||
|   | fea31753b0 | ||
|   | 0d64cd8bab | ||
|   | 9be0f8f000 | ||
|   | 78401214b0 | ||
|   | b2a07aba3a | ||
|   | 1e0bb3da95 | ||
|   | 59994da176 | ||
|   | 3d281b3316 | ||
|   | ea86849a58 | ||
|   | 399789811e | ||
|   | 8d117cb0a4 | ||
|   | 588b8e0303 | ||
|   | 1794922263 | ||
|   | 0ededb8863 | ||
|   | aa59bb1a41 | ||
|   | f2703979a4 | ||
|   | d2a1dc792f | ||
|   | 06d66a0b2b | ||
|   | 0e2522279e | ||
|   | 141a42a75b | ||
|   | a1bf37e457 | ||
|   | a20b7895a9 | ||
|   | 5666821e7b | ||
|   | 5132d8f097 | ||
|   | b81ff9c008 | ||
|   | 7e62bc4819 | ||
|   | d058be25ad | ||
|   | 1269be1d04 | ||
|   | 3b8837a16b | ||
|   | 32f478e4a0 | ||
|   | e2b50d6194 | ||
|   | 74e33b0a51 | ||
|   | 107969c09a | ||
|   | d379118772 | ||
|   | 291594b99c | ||
|   | f2cdda7278 | ||
|   | 6911458d15 | ||
|   | 6238effdc2 | ||
|   | 498377a230 | ||
|   | 3dd4ec57ff | ||
|   | e15b0e04b8 | ||
|   | 97b1fc813b | ||
|   | 917040b044 | ||
|   | 69646a160d | ||
|   | 54adb0509e | ||
|   | bd3a3b6eaf | ||
|   | 296428d53e | ||
|   | e0ca876de2 | ||
|   | a431a4fa04 | ||
|   | cc2bd03ec9 | ||
|   | 1fe81b7d1e | ||
|   | 0bd5a0d92d | ||
|   | 330ddb6a30 | ||
|   | 52dbd702ad | ||
|   | d7c3570ba3 | ||
|   | ab4d51b40b | ||
|   | 1665c93d3b | ||
|   | b51fdbce9f | ||
|   | 351b423e15 | ||
|   | 7690be1647 | ||
|   | 68aeb93afa | ||
|   | 51062863a5 | ||
|   | 4fb4b7aa6c | ||
|   | 7f3cbcedc0 | ||
|   | 6ef09def81 | ||
|   | c4c6aff9a5 | ||
|   | d71850cef6 | ||
|   | 2597c9bfac | ||
|   | 93307b57aa | ||
|   | 618953c865 | ||
|   | e04dd78624 | ||
|   | fa0c4025f7 | ||
|   | 2d2d185200 | ||
|   | cb7278eb50 | ||
|   | 89aa114192 | ||
|   | ed062e0ce5 | ||
|   | a69ef8402b | ||
|   | 8779f67d2d | ||
|   | e4b72136b8 | ||
|   | 4ff5091bc2 | ||
|   | 6f131250f1 | ||
|   | 221a63d980 | ||
|   | d02eda147c | ||
|   | 9b25716136 | ||
|   | 6628a47f23 | ||
|   | ec0e6bc3f8 | ||
|   | d2c02be3a0 | ||
|   | 594492fbdd | ||
|   | bd9ea7a88d | ||
|   | 51327a4056 | ||
|   | 33bd60528b | ||
|   | 7e54474111 | ||
|   | e307069d62 | ||
|   | 91db63294c | ||
|   | fd04e08c9c | ||
|   | 6576409d60 | ||
|   | 045cb2058c | ||
|   | d03afc12fd | ||
|   | 48799a3cff | ||
|   | 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 | 
							
								
								
									
										28
									
								
								.github/ISSUE_TEMPLATE.md
									
									
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,22 +1,36 @@ | |||||||
| 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. | <!-- 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. | ||||||
|  |  | ||||||
| Please answer the following questions.  | You can also preview your report before submitting it. | ||||||
|  |  | ||||||
| ### Which version of matterbridge are you using? | Text between <!-- and --> marks will be invisible in the report. | ||||||
| run ```matterbridge -version``` | --> | ||||||
|  |  | ||||||
| ### If you're having problems with mattermost please specify mattermost version.  | <!-- If you have a configuration problem, please first try to create a basic configuration following the instructions on [the wiki](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) before filing an issue. --> | ||||||
|  |  | ||||||
|  |  | ||||||
|  | ### Environment | ||||||
|  | <!-- run `matterbridge -version` --> | ||||||
|  | <!-- If you're having problems with mattermost also specify the mattermost version. --> | ||||||
|  | Version: | ||||||
|  |  | ||||||
|  | <!-- What operating system are you using ? (be as specific as possible) --> | ||||||
|  | Operating system: | ||||||
|  |  | ||||||
|  | <!-- If you compiled matterbridge yourself: | ||||||
|  |        * Specify the output of `go version`  | ||||||
|  |        * Specify the output of `git rev-parse HEAD` --> | ||||||
|  |  | ||||||
| ### Please describe the expected behavior. | ### Please describe the expected behavior. | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Please describe the actual behavior.  | ### Please describe the actual behavior.  | ||||||
| #### Use logs from running ```matterbridge -debug``` if possible. | <!-- Use logs from running `matterbridge -debug` if possible. --> | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Any steps to reproduce the behavior? | ### Any steps to reproduce the behavior? | ||||||
|  |  | ||||||
|  |  | ||||||
| ### Please add your configuration file  | ### Please add your configuration file  | ||||||
| #### (be sure to exclude or anonymize private data (tokens/passwords)) | <!-- (be sure to exclude or anonymize private data (tokens/passwords)) --> | ||||||
|   | |||||||
							
								
								
									
										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
									
								
							
							
						
						| @@ -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. | ||||||
							
								
								
									
										22
									
								
								.travis.yml
									
									
									
									
									
								
							
							
						
						| @@ -1,12 +1,13 @@ | |||||||
| language: go | language: go | ||||||
| go: | go: | ||||||
|     #- 1.7.x |     - 1.11.x | ||||||
|     - 1.8.x |  | ||||||
|       # - tip |  | ||||||
|  |  | ||||||
| # we have everything vendored | # we have everything vendored | ||||||
| install: true | install: true | ||||||
|  |  | ||||||
|  | git: | ||||||
|  |   depth: 200 | ||||||
|  |  | ||||||
| env: | env: | ||||||
|     - GOOS=linux GOARCH=amd64 |     - GOOS=linux GOARCH=amd64 | ||||||
|       #    - GOOS=windows GOARCH=amd64 |       #    - GOOS=windows GOARCH=amd64 | ||||||
| @@ -25,24 +26,29 @@ notifications: | |||||||
|  |  | ||||||
| before_script: | before_script: | ||||||
|   - MY_VERSION=$(git describe --tags) |   - MY_VERSION=$(git describe --tags) | ||||||
|   - GO_FILES=$(find . -iname '*.go' | grep -v /vendor/)  # All the .go files, excluding vendor/ | #  - 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/ |   - PKGS=$(go list ./... | grep -v /vendor/)             # All the import paths, excluding vendor/ | ||||||
| #  - go get github.com/golang/lint/golint                 # Linter | #  - go get github.com/golang/lint/golint                 # Linter | ||||||
|   - go get honnef.co/go/tools/cmd/megacheck              # Badass static analyzer/linter | #- go get honnef.co/go/tools/cmd/megacheck              # Badass static analyzer/linter | ||||||
|  |   - curl -sfL https://install.goreleaser.com/github.com/golangci/golangci-lint.sh | bash -s -- -b $GOPATH/bin v1.12.2 | ||||||
|  |  | ||||||
|  |  | ||||||
| # Anything in before_script: that returns a nonzero exit code will | # Anything in before_script: that returns a nonzero exit code will | ||||||
| # flunk the build and immediately stop. It's sorta like having | # flunk the build and immediately stop. It's sorta like having | ||||||
| # set -e enabled in bash.  | # set -e enabled in bash.  | ||||||
| script: | script: | ||||||
|   - test -z $(gofmt -s -l $GO_FILES)  # Fail if a .go file hasn't been formatted with gofmt |   #- test -z "$(go fmt ./...)"         # Fail if a .go file hasn't been formatted with gofmt | ||||||
|   - go test -v -race $PKGS            # Run all the tests with the race detector enabled |   - go test -v -race $PKGS            # Run all the tests with the race detector enabled | ||||||
|   - go vet $PKGS                      # go vet is the official Go static analyzer |   #- go vet $PKGS                      # go vet is the official Go static analyzer | ||||||
|   - megacheck $PKGS                   # "go vet on steroids" + linter |   - golangci-lint run --enable-all -D lll -D errcheck -D gosec -D maligned -D prealloc -D gocyclo -D gochecknoglobals | ||||||
|  |   #- megacheck $PKGS                   # "go vet on steroids" + linter | ||||||
|   - /bin/bash ci/bintray.sh |   - /bin/bash ci/bintray.sh | ||||||
|   #- golint -set_exit_status $PKGS     # one last linter |   #- golint -set_exit_status $PKGS     # one last linter | ||||||
|  |  | ||||||
| deploy: | deploy: | ||||||
|   provider: bintray |   provider: bintray | ||||||
|  |   edge: | ||||||
|  |     branch: v1.8.47 | ||||||
|   file: ci/deploy.json |   file: ci/deploy.json | ||||||
|   user: 42wim |   user: 42wim | ||||||
|   key: |   key: | ||||||
|   | |||||||
							
								
								
									
										157
									
								
								README.md
									
									
									
									
									
								
							
							
						
						| @@ -1,17 +1,40 @@ | |||||||
|  | <div align="center"> | ||||||
|  |  | ||||||
| # matterbridge | # matterbridge | ||||||
| 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)  | <br /> | ||||||
|  |    **A simple chat bridge**<br /> | ||||||
|  |    Letting people be where they want to be.<br /> | ||||||
|  |    <sub>Bridges between a growing number of protocols. Click below to demo.</sub> | ||||||
|  |  | ||||||
| [](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion) |    <sup> | ||||||
|  |  | ||||||
|  |    [Gitter][mb-gitter] | | ||||||
|  |    [IRC][mb-irc] | | ||||||
|  |       [Discord][mb-discord] | | ||||||
|  |       [Matrix][mb-matrix] | | ||||||
|  |       [Slack][mb-slack] | | ||||||
|  |       [Mattermost][mb-mattermost] | | ||||||
|  |       [XMPP][mb-xmpp] | | ||||||
|  |       [Twitch][mb-twitch] | | ||||||
|  |       [Zulip][mb-zulip] | | ||||||
|  |       And more... | ||||||
|  |    </sup> | ||||||
|  |  | ||||||
| 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) | ||||||
|  |    [](https://codeclimate.com/github/42wim/matterbridge/maintainability) | ||||||
|  |    [](https://codeclimate.com/github/42wim/matterbridge/test_coverage)<br /> | ||||||
|  |   <hr /> | ||||||
|  | </div> | ||||||
|  | <div align="right"><sup> | ||||||
|  |  | ||||||
| # Table of Contents | **Note:** Matter<em>most</em> isn't required to run matter<em>bridge</em>.</sup></div> | ||||||
|  * [Features](#features) |  | ||||||
|  | ### Table of Contents | ||||||
|  |  * [Features](https://github.com/42wim/matterbridge/wiki/Features) | ||||||
|  |    * [API](#API) | ||||||
|  * [Requirements](#requirements) |  * [Requirements](#requirements) | ||||||
|  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) |  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) | ||||||
|  * [Installing](#installing) |  * [Installing](#installing) | ||||||
| @@ -19,24 +42,37 @@ Has a REST API. | |||||||
|    * [Building](#building) |    * [Building](#building) | ||||||
|  * [Configuration](#configuration) |  * [Configuration](#configuration) | ||||||
|    * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) |    * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) | ||||||
|    * [Examples](#examples)  |    * [Examples](#examples) | ||||||
|  * [Running](#running) |  * [Running](#running) | ||||||
|    * [Docker](#docker) |    * [Docker](#docker) | ||||||
|  * [Changelog](#changelog) |  * [Changelog](#changelog) | ||||||
|  * [FAQ](#faq) |  * [FAQ](#faq) | ||||||
|  |  * [Related projects](#related-projects) | ||||||
|  |  * [Articles](#articles) | ||||||
|  * [Thanks](#thanks) |  * [Thanks](#thanks) | ||||||
|  |  | ||||||
| # Features | ## Features | ||||||
| * Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam.  | * [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols) | ||||||
|   Pick and mix. | * [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols) | ||||||
| * Matterbridge can also work with private groups on your mattermost/slack. | * [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes) | ||||||
| * Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts. | * Preserves threading when possible | ||||||
| * The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways). | * [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling) | ||||||
| * REST API to read/post messages to bridges (WIP). | * [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) | ||||||
|  |  | ||||||
| # Requirements | ### API | ||||||
|  | The API is very basic at the moment and rather undocumented. | ||||||
|  |  | ||||||
|  | Used by at least 3 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) | ||||||
|  | * [Mattereddit](https://github.com/bonehurtingjuice/mattereddit) (Reddit chat support) | ||||||
|  |  | ||||||
|  | ## Requirements | ||||||
| Accounts to one of the supported bridges | Accounts to one of the supported bridges | ||||||
| * [Mattermost](https://github.com/mattermost/platform/) 3.5.x - 3.10.x, 4.0.x | * [Mattermost](https://github.com/mattermost/platform/) 3.8.x - 3.10.x, 4.x, 5.x | ||||||
| * [IRC](http://www.mirc.com/servers.html) | * [IRC](http://www.mirc.com/servers.html) | ||||||
| * [XMPP](https://jabber.org) | * [XMPP](https://jabber.org) | ||||||
| * [Gitter](https://gitter.im) | * [Gitter](https://gitter.im) | ||||||
| @@ -47,17 +83,22 @@ Accounts to one of the supported bridges | |||||||
| * [Rocket.chat](https://rocket.chat) | * [Rocket.chat](https://rocket.chat) | ||||||
| * [Matrix](https://matrix.org) | * [Matrix](https://matrix.org) | ||||||
| * [Steam](https://store.steampowered.com/) | * [Steam](https://store.steampowered.com/) | ||||||
|  | * [Twitch](https://twitch.tv) | ||||||
|  | * [Ssh-chat](https://github.com/shazow/ssh-chat) | ||||||
|  | * [Zulip](https://zulipchat.com) | ||||||
|  |  | ||||||
| # Screenshots | ## Screenshots | ||||||
| See https://github.com/42wim/matterbridge/wiki | See https://github.com/42wim/matterbridge/wiki | ||||||
|  |  | ||||||
| # Installing | ## Installing | ||||||
| ## Binaries | ### Binaries | ||||||
| * Latest stable release [v1.0.0-rc1](https://github.com/42wim/matterbridge/releases/latest) | * Latest stable release [v1.12.0](https://github.com/42wim/matterbridge/releases/latest) | ||||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)   | * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/) | ||||||
|  |  | ||||||
| ## Building | ### Building | ||||||
| Go 1.7+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH] (https://golang.org/doc/code.html#GOPATH) | Go 1.8+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH](https://golang.org/doc/code.html#GOPATH). | ||||||
|  |  | ||||||
|  | After Go is setup, download matterbridge to your $GOPATH directory. | ||||||
|  |  | ||||||
| ``` | ``` | ||||||
| cd $GOPATH | cd $GOPATH | ||||||
| @@ -71,16 +112,16 @@ $ ls bin/ | |||||||
| matterbridge | matterbridge | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| # Configuration | ## Configuration | ||||||
| ## Basic 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. | 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 | ### Advanced configuration | ||||||
| * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | ||||||
|  |  | ||||||
| ## Examples  | ### Examples | ||||||
| ### Bridge mattermost (off-topic) - irc (#testing) | #### Bridge mattermost (off-topic) - irc (#testing) | ||||||
| ``` | ```toml | ||||||
| [irc] | [irc] | ||||||
|     [irc.freenode] |     [irc.freenode] | ||||||
|     Server="irc.freenode.net:6667" |     Server="irc.freenode.net:6667" | ||||||
| @@ -107,8 +148,8 @@ enable=true | |||||||
|     channel="off-topic" |     channel="off-topic" | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| ### Bridge slack (#general) - discord (general) | #### Bridge slack (#general) - discord (general) | ||||||
| ``` | ```toml | ||||||
| [slack] | [slack] | ||||||
| [slack.test] | [slack.test] | ||||||
| Token="yourslacktoken" | Token="yourslacktoken" | ||||||
| @@ -135,7 +176,7 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | |||||||
|     channel = "general" |     channel = "general" | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| # Running | ## Running | ||||||
|  |  | ||||||
| See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||||
|  |  | ||||||
| @@ -151,33 +192,67 @@ Usage of ./matterbridge: | |||||||
|         show version |         show version | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| ## Docker | ### Docker | ||||||
| Create your matterbridge.toml file locally eg in ```/tmp/matterbridge.toml``` | Create your matterbridge.toml file locally eg in `/tmp/matterbridge.toml` | ||||||
| ``` | ``` | ||||||
| docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge | docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| # Changelog | ## Changelog | ||||||
| See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md) | See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md) | ||||||
|  |  | ||||||
| # FAQ | ## FAQ | ||||||
|  |  | ||||||
| See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | ||||||
|  |  | ||||||
| Want to tip ?  | Want to tip ? | ||||||
| * eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f | * eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f | ||||||
| * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | ||||||
|  |  | ||||||
| # Thanks | ## Related projects | ||||||
|  | * [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||||
|  | * [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||||
|  | * [matterbridge autoconfig](https://github.com/patcon/matterbridge-autoconfig) | ||||||
|  | * [matterlink](https://github.com/elytra/MatterLink) | ||||||
|  | * [mattereddit](https://github.com/bonehurtingjuice/mattereddit) | ||||||
|  | * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||||
|  | * [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost | ||||||
|  |  | ||||||
|  | ## Articles | ||||||
|  | * https://mattermost.com/blog/connect-irc-to-mattermost/ | ||||||
|  | * https://blog.valvin.fr/2016/09/17/mattermost-et-un-channel-irc-cest-possible/ | ||||||
|  | * https://blog.brightscout.com/top-10-mattermost-integrations/ | ||||||
|  | * http://bencey.co.nz/2018/09/17/bridge/ | ||||||
|  | * https://www.algoo.fr/blog/2018/01/19/recouvrez-votre-liberte-en-quittant-slack-pour-un-mattermost-auto-heberge/ | ||||||
|  | * https://kopano.com/blog/matterbridge-bridging-mattermost-chat/ | ||||||
|  | * https://www.stitcher.com/s/?eid=52382713 | ||||||
|  |  | ||||||
|  | ## Thanks | ||||||
|  | [](https://www.digitalocean.com/) for sponsoring demo/testing droplets. | ||||||
|  |  | ||||||
| Matterbridge wouldn't exist without these libraries: | Matterbridge wouldn't exist without these libraries: | ||||||
| * discord - https://github.com/bwmarrin/discordgo | * discord - https://github.com/bwmarrin/discordgo | ||||||
| * echo - https://github.com/labstack/echo | * echo - https://github.com/labstack/echo | ||||||
| * gitter - https://github.com/sromku/go-gitter | * gitter - https://github.com/sromku/go-gitter | ||||||
| * gops - https://github.com/google/gops | * gops - https://github.com/google/gops | ||||||
| * irc - https://github.com/thoj/go-ircevent | * gozulipbot - https://github.com/ifo/gozulipbot | ||||||
|  | * irc - https://github.com/lrstanley/girc | ||||||
| * mattermost - https://github.com/mattermost/platform | * mattermost - https://github.com/mattermost/platform | ||||||
| * matrix - https://github.com/matrix-org/gomatrix | * matrix - https://github.com/matrix-org/gomatrix | ||||||
| * slack - https://github.com/nlopes/slack | * slack - https://github.com/nlopes/slack | ||||||
| * steam - https://github.com/Philipp15b/go-steam | * steam - https://github.com/Philipp15b/go-steam | ||||||
| * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | ||||||
| * xmpp - https://github.com/mattn/go-xmpp | * xmpp - https://github.com/mattn/go-xmpp | ||||||
|  | * zulip - https://github.com/ifo/gozulipbot | ||||||
|  |  | ||||||
|  | <!-- Links --> | ||||||
|  |  | ||||||
|  |    [mb-gitter]: https://gitter.im/42wim/matterbridge | ||||||
|  |    [mb-irc]: https://webchat.freenode.net/?channels=matterbridgechat | ||||||
|  |    [mb-discord]: https://discord.gg/AkKPtrQ | ||||||
|  |    [mb-matrix]: https://riot.im/app/#/room/#matterbridge:matrix.org | ||||||
|  |    [mb-slack]: https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA | ||||||
|  |    [mb-mattermost]: https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e | ||||||
|  |    [mb-xmpp]: https://inverse.chat/ | ||||||
|  |    [mb-twitch]: https://www.twitch.tv/matterbridge | ||||||
|  |    [mb-zulip]: https://matterbridge.zulipchat.com/register/ | ||||||
|   | |||||||
| @@ -1,24 +1,25 @@ | |||||||
| package api | package api | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"net/http" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/labstack/echo" | 	"github.com/labstack/echo" | ||||||
| 	"github.com/labstack/echo/middleware" | 	"github.com/labstack/echo/middleware" | ||||||
| 	"github.com/zfjagann/golang-ring" | 	"github.com/zfjagann/golang-ring" | ||||||
| 	"net/http" |  | ||||||
| 	"sync" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Api struct { | type API struct { | ||||||
| 	Config   *config.Protocol |  | ||||||
| 	Remote   chan config.Message |  | ||||||
| 	Account  string |  | ||||||
| 	Messages ring.Ring | 	Messages ring.Ring | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| type ApiMessage struct { | type Message struct { | ||||||
| 	Text     string `json:"text"` | 	Text     string `json:"text"` | ||||||
| 	Username string `json:"username"` | 	Username string `json:"username"` | ||||||
| 	UserID   string `json:"userid"` | 	UserID   string `json:"userid"` | ||||||
| @@ -26,76 +27,110 @@ type ApiMessage struct { | |||||||
| 	Gateway  string `json:"gateway"` | 	Gateway  string `json:"gateway"` | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "api" | 	b := &API{Config: cfg} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Api { |  | ||||||
| 	b := &Api{} |  | ||||||
| 	e := echo.New() | 	e := echo.New() | ||||||
|  | 	e.HideBanner = true | ||||||
|  | 	e.HidePort = true | ||||||
| 	b.Messages = ring.Ring{} | 	b.Messages = ring.Ring{} | ||||||
| 	b.Messages.SetCapacity(cfg.Buffer) | 	if b.GetInt("Buffer") != 0 { | ||||||
| 	b.Config = &cfg | 		b.Messages.SetCapacity(b.GetInt("Buffer")) | ||||||
| 	b.Account = account | 	} | ||||||
| 	b.Remote = c | 	if b.GetString("Token") != "" { | ||||||
| 	if b.Config.Token != "" { |  | ||||||
| 		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) { | 		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) { | ||||||
| 			return key == b.Config.Token, nil | 			return key == b.GetString("Token"), nil | ||||||
| 		})) | 		})) | ||||||
| 	} | 	} | ||||||
|  | 	e.GET("/api/health", b.handleHealthcheck) | ||||||
| 	e.GET("/api/messages", b.handleMessages) | 	e.GET("/api/messages", b.handleMessages) | ||||||
|  | 	e.GET("/api/stream", b.handleStream) | ||||||
| 	e.POST("/api/message", b.handlePostMessage) | 	e.POST("/api/message", b.handlePostMessage) | ||||||
| 	go func() { | 	go func() { | ||||||
| 		flog.Fatal(e.Start(cfg.BindAddress)) | 		if b.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 | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) Connect() error { | func (b *API) Connect() error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| func (b *Api) Disconnect() error { | func (b *API) Disconnect() error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
| func (b *Api) JoinChannel(channel string) error { | func (b *API) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) Send(msg config.Message) error { | func (b *API) Send(msg config.Message) (string, error) { | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	b.Messages.Enqueue(&msg) | 	b.Messages.Enqueue(&msg) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) handlePostMessage(c echo.Context) error { | func (b *API) handleHealthcheck(c echo.Context) error { | ||||||
| 	message := &ApiMessage{} | 	return c.String(http.StatusOK, "OK") | ||||||
| 	if err := c.Bind(message); err != nil { | } | ||||||
|  |  | ||||||
|  | func (b *API) handlePostMessage(c echo.Context) error { | ||||||
|  | 	message := config.Message{} | ||||||
|  | 	if err := c.Bind(&message); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api") | 	// these values are fixed | ||||||
| 	b.Remote <- config.Message{ | 	message.Channel = "api" | ||||||
| 		Text:     message.Text, | 	message.Protocol = "api" | ||||||
| 		Username: message.Username, | 	message.Account = b.Account | ||||||
| 		UserID:   message.UserID, | 	message.ID = "" | ||||||
| 		Channel:  "api", | 	message.Timestamp = time.Now() | ||||||
| 		Avatar:   message.Avatar, | 	b.Log.Debugf("Sending message from %s on %s to gateway", message.Username, "api") | ||||||
| 		Account:  b.Account, | 	b.Remote <- message | ||||||
| 		Gateway:  message.Gateway, |  | ||||||
| 		Protocol: "api", |  | ||||||
| 	} |  | ||||||
| 	return c.JSON(http.StatusOK, message) | 	return c.JSON(http.StatusOK, message) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Api) handleMessages(c echo.Context) error { | func (b *API) handleMessages(c echo.Context) error { | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
| 	c.JSONPretty(http.StatusOK, b.Messages.Values(), " ") | 	c.JSONPretty(http.StatusOK, b.Messages.Values(), " ") | ||||||
| 	b.Messages = ring.Ring{} | 	b.Messages = ring.Ring{} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *API) handleStream(c echo.Context) error { | ||||||
|  | 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON) | ||||||
|  | 	c.Response().WriteHeader(http.StatusOK) | ||||||
|  | 	greet := config.Message{ | ||||||
|  | 		Event:     config.EventAPIConnected, | ||||||
|  | 		Timestamp: time.Now(), | ||||||
|  | 	} | ||||||
|  | 	if err := json.NewEncoder(c.Response()).Encode(greet); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	c.Response().Flush() | ||||||
|  | 	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
									
									
									
									
									
								
							
							
						
						| @@ -1,41 +1,42 @@ | |||||||
| package bridge | package bridge | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/42wim/matterbridge/bridge/api" |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/bridge/discord" | 	log "github.com/sirupsen/logrus" | ||||||
| 	"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" |  | ||||||
|  |  | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bridger interface { | type Bridger interface { | ||||||
| 	Send(msg config.Message) error | 	Send(msg config.Message) (string, error) | ||||||
| 	Connect() error | 	Connect() error | ||||||
| 	JoinChannel(channel string) error | 	JoinChannel(channel config.ChannelInfo) error | ||||||
| 	Disconnect() error | 	Disconnect() error | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bridge struct { | type Bridge struct { | ||||||
| 	Config config.Protocol |  | ||||||
| 	Bridger | 	Bridger | ||||||
| 	Name     string | 	Name     string | ||||||
| 	Account  string | 	Account  string | ||||||
| 	Protocol string | 	Protocol string | ||||||
| 	Channels map[string]config.ChannelInfo | 	Channels map[string]config.ChannelInfo | ||||||
| 	Joined   map[string]bool | 	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 := new(Bridge) | ||||||
| 	b.Channels = make(map[string]config.ChannelInfo) | 	b.Channels = make(map[string]config.ChannelInfo) | ||||||
| 	accInfo := strings.Split(bridge.Account, ".") | 	accInfo := strings.Split(bridge.Account, ".") | ||||||
| @@ -45,44 +46,6 @@ func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Brid | |||||||
| 	b.Protocol = protocol | 	b.Protocol = protocol | ||||||
| 	b.Account = bridge.Account | 	b.Account = bridge.Account | ||||||
| 	b.Joined = make(map[string]bool) | 	b.Joined = make(map[string]bool) | ||||||
|  |  | ||||||
| 	// 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 | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -92,16 +55,10 @@ func (b *Bridge) JoinChannels() error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | ||||||
| 	mychannel := "" |  | ||||||
| 	for ID, channel := range channels { | 	for ID, channel := range channels { | ||||||
| 		if !exists[ID] { | 		if !exists[ID] { | ||||||
| 			mychannel = channel.Name | 			b.Log.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID) | ||||||
| 			log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID) | 			err := b.JoinChannel(channel) | ||||||
| 			if b.Protocol == "irc" && channel.Options.Key != "" { |  | ||||||
| 				log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) |  | ||||||
| 				mychannel = mychannel + " " + channel.Options.Key |  | ||||||
| 			} |  | ||||||
| 			err := b.JoinChannel(mychannel) |  | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| @@ -110,3 +67,43 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map | |||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) GetBool(key string) bool { | ||||||
|  | 	val, ok := b.Config.GetBool(b.Account + "." + key) | ||||||
|  | 	if !ok { | ||||||
|  | 		val, _ = b.Config.GetBool("general." + key) | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) GetInt(key string) int { | ||||||
|  | 	val, ok := b.Config.GetInt(b.Account + "." + key) | ||||||
|  | 	if !ok { | ||||||
|  | 		val, _ = b.Config.GetInt("general." + key) | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) GetString(key string) string { | ||||||
|  | 	val, ok := b.Config.GetString(b.Account + "." + key) | ||||||
|  | 	if !ok { | ||||||
|  | 		val, _ = b.Config.GetString("general." + key) | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) GetStringSlice(key string) []string { | ||||||
|  | 	val, ok := b.Config.GetStringSlice(b.Account + "." + key) | ||||||
|  | 	if !ok { | ||||||
|  | 		val, _ = b.Config.GetStringSlice("general." + key) | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bridge) GetStringSlice2D(key string) [][]string { | ||||||
|  | 	val, ok := b.Config.GetStringSlice2D(b.Account + "." + key) | ||||||
|  | 	if !ok { | ||||||
|  | 		val, _ = b.Config.GetStringSlice2D("general." + key) | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,19 +1,29 @@ | |||||||
| package config | package config | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/BurntSushi/toml" | 	"bytes" | ||||||
| 	"log" | 	"io/ioutil" | ||||||
| 	"os" |  | ||||||
| 	"reflect" |  | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/fsnotify/fsnotify" | ||||||
|  | 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
|  | 	"github.com/spf13/viper" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	EVENT_JOIN_LEAVE      = "join_leave" | 	EventJoinLeave       = "join_leave" | ||||||
| 	EVENT_FAILURE         = "failure" | 	EventTopicChange     = "topic_change" | ||||||
| 	EVENT_REJOIN_CHANNELS = "rejoin_channels" | 	EventFailure         = "failure" | ||||||
| 	EVENT_USER_ACTION     = "user_action" | 	EventFileFailureSize = "file_failure_size" | ||||||
|  | 	EventAvatarDownload  = "avatar_download" | ||||||
|  | 	EventRejoinChannels  = "rejoin_channels" | ||||||
|  | 	EventUserAction      = "user_action" | ||||||
|  | 	EventMsgDelete       = "msg_delete" | ||||||
|  | 	EventAPIConnected    = "api_connected" | ||||||
|  | 	EventUserTyping      = "user_typing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Message struct { | type Message struct { | ||||||
| @@ -26,7 +36,20 @@ type Message struct { | |||||||
| 	Event     string    `json:"event"` | 	Event     string    `json:"event"` | ||||||
| 	Protocol  string    `json:"protocol"` | 	Protocol  string    `json:"protocol"` | ||||||
| 	Gateway   string    `json:"gateway"` | 	Gateway   string    `json:"gateway"` | ||||||
|  | 	ParentID  string    `json:"parent_id"` | ||||||
| 	Timestamp time.Time `json:"timestamp"` | 	Timestamp time.Time `json:"timestamp"` | ||||||
|  | 	ID        string    `json:"id"` | ||||||
|  | 	Extra     map[string][]interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type FileInfo struct { | ||||||
|  | 	Name    string | ||||||
|  | 	Data    *[]byte | ||||||
|  | 	Comment string | ||||||
|  | 	URL     string | ||||||
|  | 	Size    int64 | ||||||
|  | 	Avatar  bool | ||||||
|  | 	SHA     string | ||||||
| } | } | ||||||
|  |  | ||||||
| type ChannelInfo struct { | type ChannelInfo struct { | ||||||
| @@ -42,49 +65,74 @@ type Protocol struct { | |||||||
| 	AuthCode               string // steam | 	AuthCode               string // steam | ||||||
| 	BindAddress            string // mattermost, slack // DEPRECATED | 	BindAddress            string // mattermost, slack // DEPRECATED | ||||||
| 	Buffer                 int    // api | 	Buffer                 int    // api | ||||||
|  | 	Charset                string // irc | ||||||
|  | 	ColorNicks             bool   // only irc for now | ||||||
|  | 	Debug                  bool   // general | ||||||
|  | 	DebugLevel             int    // only for irc now | ||||||
| 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||||
| 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||||
| 	IconURL                string // mattermost, slack | 	IconURL                string // mattermost, slack | ||||||
| 	IgnoreNicks            string // all protocols | 	IgnoreNicks            string // all protocols | ||||||
| 	IgnoreMessages         string // all protocols | 	IgnoreMessages         string // all protocols | ||||||
| 	Jid                    string // xmpp | 	Jid                    string // xmpp | ||||||
|  | 	Label                  string // all protocols | ||||||
| 	Login                  string // mattermost, matrix | 	Login                  string // mattermost, matrix | ||||||
| 	Muc                    string // xmpp | 	MediaDownloadBlackList []string | ||||||
| 	Name                   string // all protocols | 	MediaDownloadPath      string // Basically MediaServerUpload, but instead of uploading it, just write it to a file on the same server. | ||||||
| 	Nick                   string // all protocols | 	MediaDownloadSize      int    // all protocols | ||||||
| 	NickFormatter          string // mattermost, slack | 	MediaServerDownload    string | ||||||
| 	NickServNick           string // IRC | 	MediaServerUpload      string | ||||||
| 	NickServPassword       string // IRC | 	MessageDelay           int        // IRC, time in millisecond to wait between messages | ||||||
| 	NicksPerRow            int    // mattermost, slack | 	MessageFormat          string     // telegram | ||||||
| 	NoHomeServerSuffix     bool   // matrix | 	MessageLength          int        // IRC, max length of a message allowed | ||||||
| 	NoTLS                  bool   // mattermost | 	MessageQueue           int        // IRC, size of message queue for flood control | ||||||
| 	Password               string // IRC,mattermost,XMPP,matrix | 	MessageSplit           bool       // IRC, split long messages with newlines on MessageLength instead of clipping | ||||||
| 	PrefixMessagesWithNick bool   // mattemost, slack | 	Muc                    string     // xmpp | ||||||
| 	Protocol               string //all protocols | 	Name                   string     // all protocols | ||||||
| 	MessageQueue           int    // IRC, size of message queue for flood control | 	Nick                   string     // all protocols | ||||||
| 	MessageDelay           int    // IRC, time in millisecond to wait between messages | 	NickFormatter          string     // mattermost, slack | ||||||
| 	MessageLength          int    // IRC, max length of a message allowed | 	NickServNick           string     // IRC | ||||||
| 	MessageFormat          string // telegram | 	NickServUsername       string     // IRC | ||||||
| 	RemoteNickFormat       string // all protocols | 	NickServPassword       string     // IRC | ||||||
| 	Server                 string // IRC,mattermost,XMPP,discord | 	NicksPerRow            int        // mattermost, slack | ||||||
| 	ShowJoinPart           bool   // all protocols | 	NoHomeServerSuffix     bool       // matrix | ||||||
| 	ShowEmbeds             bool   // discord | 	NoSendJoinPart         bool       // all protocols | ||||||
| 	SkipTLSVerify          bool   // IRC, mattermost | 	NoTLS                  bool       // mattermost | ||||||
| 	Team                   string // mattermost | 	Password               string     // IRC,mattermost,XMPP,matrix | ||||||
| 	Token                  string // gitter, slack, discord, api | 	PrefixMessagesWithNick bool       // mattemost, slack | ||||||
| 	URL                    string // mattermost, slack // DEPRECATED | 	PreserveThreading      bool       // slack | ||||||
| 	UseAPI                 bool   // mattermost, slack | 	Protocol               string     // all protocols | ||||||
| 	UseSASL                bool   // IRC | 	QuoteDisable           bool       // telegram | ||||||
| 	UseTLS                 bool   // IRC | 	QuoteFormat            string     // telegram | ||||||
| 	UseFirstName           bool   // telegram | 	RejoinDelay            int        // IRC | ||||||
| 	UseInsecureURL         bool   // telegram | 	ReplaceMessages        [][]string // all protocols | ||||||
| 	WebhookBindAddress     string // mattermost, slack | 	ReplaceNicks           [][]string // all protocols | ||||||
| 	WebhookURL             string // mattermost, slack | 	RemoteNickFormat       string     // all protocols | ||||||
| 	WebhookUse             string // mattermost, slack, discord | 	Server                 string     // IRC,mattermost,XMPP,discord | ||||||
|  | 	ShowJoinPart           bool       // all protocols | ||||||
|  | 	ShowTopicChange        bool       // slack | ||||||
|  | 	ShowUserTyping         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 { | type ChannelOptions struct { | ||||||
| 	Key string // irc | 	Key        string // irc, xmpp | ||||||
|  | 	WebhookURL string // discord | ||||||
| } | } | ||||||
|  |  | ||||||
| type Bridge struct { | type Bridge struct { | ||||||
| @@ -109,92 +157,146 @@ type SameChannelGateway struct { | |||||||
| 	Accounts []string | 	Accounts []string | ||||||
| } | } | ||||||
|  |  | ||||||
| type Config struct { | type BridgeValues struct { | ||||||
| 	Api                map[string]Protocol | 	API                map[string]Protocol | ||||||
| 	IRC                map[string]Protocol | 	IRC                map[string]Protocol | ||||||
| 	Mattermost         map[string]Protocol | 	Mattermost         map[string]Protocol | ||||||
| 	Matrix             map[string]Protocol | 	Matrix             map[string]Protocol | ||||||
| 	Slack              map[string]Protocol | 	Slack              map[string]Protocol | ||||||
|  | 	SlackLegacy        map[string]Protocol | ||||||
| 	Steam              map[string]Protocol | 	Steam              map[string]Protocol | ||||||
| 	Gitter             map[string]Protocol | 	Gitter             map[string]Protocol | ||||||
| 	Xmpp               map[string]Protocol | 	XMPP               map[string]Protocol | ||||||
| 	Discord            map[string]Protocol | 	Discord            map[string]Protocol | ||||||
| 	Telegram           map[string]Protocol | 	Telegram           map[string]Protocol | ||||||
| 	Rocketchat         map[string]Protocol | 	Rocketchat         map[string]Protocol | ||||||
|  | 	SSHChat            map[string]Protocol | ||||||
|  | 	Zulip              map[string]Protocol | ||||||
| 	General            Protocol | 	General            Protocol | ||||||
| 	Gateway            []Gateway | 	Gateway            []Gateway | ||||||
| 	SameChannelGateway []SameChannelGateway | 	SameChannelGateway []SameChannelGateway | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewConfig(cfgfile string) *Config { | type Config interface { | ||||||
| 	var cfg Config | 	BridgeValues() *BridgeValues | ||||||
| 	if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil { | 	GetBool(key string) (bool, bool) | ||||||
|  | 	GetInt(key string) (int, bool) | ||||||
|  | 	GetString(key string) (string, bool) | ||||||
|  | 	GetStringSlice(key string) ([]string, bool) | ||||||
|  | 	GetStringSlice2D(key string) ([][]string, bool) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type config struct { | ||||||
|  | 	v *viper.Viper | ||||||
|  | 	sync.RWMutex | ||||||
|  |  | ||||||
|  | 	cv *BridgeValues | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewConfig(cfgfile string) Config { | ||||||
|  | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||||
|  | 	flog := log.WithFields(log.Fields{"prefix": "config"}) | ||||||
|  | 	viper.SetConfigFile(cfgfile) | ||||||
|  | 	input, err := getFileContents(cfgfile) | ||||||
|  | 	if err != nil { | ||||||
| 		log.Fatal(err) | 		log.Fatal(err) | ||||||
| 	} | 	} | ||||||
| 	fail := false | 	mycfg := newConfigFromString(input) | ||||||
| 	for k, v := range cfg.Mattermost { | 	if mycfg.cv.General.MediaDownloadSize == 0 { | ||||||
| 		res := Deprecated(v, "mattermost."+k) | 		mycfg.cv.General.MediaDownloadSize = 1000000 | ||||||
| 		if res { |  | ||||||
| 			fail = res |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| 	for k, v := range cfg.Slack { | 	viper.WatchConfig() | ||||||
| 		res := Deprecated(v, "slack."+k) | 	viper.OnConfigChange(func(e fsnotify.Event) { | ||||||
| 		if res { | 		flog.Println("Config file changed:", e.Name) | ||||||
| 			fail = res | 	}) | ||||||
| 		} | 	return mycfg | ||||||
| 	} |  | ||||||
| 	for k, v := range cfg.Rocketchat { |  | ||||||
| 		res := Deprecated(v, "rocketchat."+k) |  | ||||||
| 		if res { |  | ||||||
| 			fail = res |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	if fail { |  | ||||||
| 		log.Fatalf("Fix your config. Please see changelog for more information") |  | ||||||
| 	} |  | ||||||
| 	return &cfg |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func OverrideCfgFromEnv(cfg *Config, protocol string, account string) { | func getFileContents(filename string) ([]byte, error) { | ||||||
| 	var protoCfg Protocol | 	input, err := ioutil.ReadFile(filename) | ||||||
| 	val := reflect.ValueOf(cfg).Elem() | 	if err != nil { | ||||||
| 	// loop over the Config struct | 		log.Fatal(err) | ||||||
| 	for i := 0; i < val.NumField(); i++ { | 		return []byte(nil), err | ||||||
| 		typeField := val.Type().Field(i) | 	} | ||||||
| 		// look for the protocol map (both lowercase) | 	return input, nil | ||||||
| 		if strings.ToLower(typeField.Name) == protocol { | } | ||||||
| 			// get the Protocol struct from the map |  | ||||||
| 			data := val.Field(i).MapIndex(reflect.ValueOf(account)) | func NewConfigFromString(input []byte) Config { | ||||||
| 			protoCfg = data.Interface().(Protocol) | 	return newConfigFromString(input) | ||||||
| 			protoStruct := reflect.ValueOf(&protoCfg).Elem() | } | ||||||
| 			// loop over the found protocol struct |  | ||||||
| 			for i := 0; i < protoStruct.NumField(); i++ { | func newConfigFromString(input []byte) *config { | ||||||
| 				typeField := protoStruct.Type().Field(i) | 	viper.SetConfigType("toml") | ||||||
| 				// build our environment key (eg MATTERBRIDGE_MATTERMOST_WORK_LOGIN) | 	viper.SetEnvPrefix("matterbridge") | ||||||
| 				key := "matterbridge_" + protocol + "_" + account + "_" + typeField.Name | 	viper.AddConfigPath(".") | ||||||
| 				key = strings.ToUpper(key) | 	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_")) | ||||||
| 				// search the environment | 	viper.AutomaticEnv() | ||||||
| 				res := os.Getenv(key) | 	err := viper.ReadConfig(bytes.NewBuffer(input)) | ||||||
| 				// if it exists and the current field is a string | 	if err != nil { | ||||||
| 				// then update the current field | 		log.Fatal(err) | ||||||
| 				if res != "" { | 	} | ||||||
| 					fieldVal := protoStruct.Field(i) |  | ||||||
| 					if fieldVal.Kind() == reflect.String { | 	cfg := &BridgeValues{} | ||||||
| 						log.Printf("config: overriding %s from env with %s\n", key, res) | 	err = viper.Unmarshal(cfg) | ||||||
| 						fieldVal.Set(reflect.ValueOf(res)) | 	if err != nil { | ||||||
| 					} | 		log.Fatal(err) | ||||||
| 				} | 	} | ||||||
|  | 	return &config{ | ||||||
|  | 		v:  viper.GetViper(), | ||||||
|  | 		cv: cfg, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) BridgeValues() *BridgeValues { | ||||||
|  | 	return c.cv | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) GetBool(key string) (bool, bool) { | ||||||
|  | 	c.RLock() | ||||||
|  | 	defer c.RUnlock() | ||||||
|  | 	//	log.Debugf("getting bool %s = %#v", key, c.v.GetBool(key)) | ||||||
|  | 	return c.v.GetBool(key), c.v.IsSet(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) GetInt(key string) (int, bool) { | ||||||
|  | 	c.RLock() | ||||||
|  | 	defer c.RUnlock() | ||||||
|  | 	//	log.Debugf("getting int %s = %d", key, c.v.GetInt(key)) | ||||||
|  | 	return c.v.GetInt(key), c.v.IsSet(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) GetString(key string) (string, bool) { | ||||||
|  | 	c.RLock() | ||||||
|  | 	defer c.RUnlock() | ||||||
|  | 	//	log.Debugf("getting String %s = %s", key, c.v.GetString(key)) | ||||||
|  | 	return c.v.GetString(key), c.v.IsSet(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) GetStringSlice(key string) ([]string, bool) { | ||||||
|  | 	c.RLock() | ||||||
|  | 	defer c.RUnlock() | ||||||
|  | 	// log.Debugf("getting StringSlice %s = %#v", key, c.v.GetStringSlice(key)) | ||||||
|  | 	return c.v.GetStringSlice(key), c.v.IsSet(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *config) GetStringSlice2D(key string) ([][]string, bool) { | ||||||
|  | 	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) | 			result = append(result, result2) | ||||||
| 			val.Field(i).SetMapIndex(reflect.ValueOf(account), reflect.ValueOf(protoCfg)) |  | ||||||
| 			break |  | ||||||
| 		} | 		} | ||||||
|  | 		return result, true | ||||||
| 	} | 	} | ||||||
|  | 	return result, false | ||||||
| } | } | ||||||
|  |  | ||||||
| func GetIconURL(msg *Message, cfg *Protocol) string { | func GetIconURL(msg *Message, iconURL string) string { | ||||||
| 	iconURL := cfg.IconURL |  | ||||||
| 	info := strings.Split(msg.Account, ".") | 	info := strings.Split(msg.Account, ".") | ||||||
| 	protocol := info[0] | 	protocol := info[0] | ||||||
| 	name := info[1] | 	name := info[1] | ||||||
| @@ -204,16 +306,44 @@ func GetIconURL(msg *Message, cfg *Protocol) string { | |||||||
| 	return iconURL | 	return iconURL | ||||||
| } | } | ||||||
|  |  | ||||||
| func Deprecated(cfg Protocol, account string) bool { | type TestConfig struct { | ||||||
| 	if cfg.BindAddress != "" { | 	Config | ||||||
| 		log.Printf("ERROR: %s BindAddress is deprecated, you need to change it to WebhookBindAddress.", account) |  | ||||||
| 	} else if cfg.URL != "" { | 	Overrides map[string]interface{} | ||||||
| 		log.Printf("ERROR: %s URL is deprecated, you need to change it to WebhookURL.", account) | } | ||||||
| 	} else if cfg.UseAPI { |  | ||||||
| 		log.Printf("ERROR: %s UseAPI is deprecated, it's enabled by default, please remove it from your config file.", account) | func (c *TestConfig) GetBool(key string) (bool, bool) { | ||||||
| 	} else { | 	val, ok := c.Overrides[key] | ||||||
| 		return false | 	if ok { | ||||||
| 	} | 		return val.(bool), true | ||||||
| 	return true | 	} | ||||||
| 	//log.Fatalf("ERROR: Fix your config: %s", account) | 	return c.Config.GetBool(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TestConfig) GetInt(key string) (int, bool) { | ||||||
|  | 	if val, ok := c.Overrides[key]; ok { | ||||||
|  | 		return val.(int), true | ||||||
|  | 	} | ||||||
|  | 	return c.Config.GetInt(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TestConfig) GetString(key string) (string, bool) { | ||||||
|  | 	if val, ok := c.Overrides[key]; ok { | ||||||
|  | 		return val.(string), true | ||||||
|  | 	} | ||||||
|  | 	return c.Config.GetString(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TestConfig) GetStringSlice(key string) ([]string, bool) { | ||||||
|  | 	if val, ok := c.Overrides[key]; ok { | ||||||
|  | 		return val.([]string), true | ||||||
|  | 	} | ||||||
|  | 	return c.Config.GetStringSlice(key) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *TestConfig) GetStringSlice2D(key string) ([][]string, bool) { | ||||||
|  | 	if val, ok := c.Overrides[key]; ok { | ||||||
|  | 		return val.([][]string), true | ||||||
|  | 	} | ||||||
|  | 	return c.Config.GetStringSlice2D(key) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,223 +1,332 @@ | |||||||
| package bdiscord | package bdiscord | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"bytes" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"errors" | ||||||
| 	"github.com/bwmarrin/discordgo" | 	"fmt" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	"github.com/bwmarrin/discordgo" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type bdiscord struct { | const MessageLength = 1950 | ||||||
| 	c             *discordgo.Session |  | ||||||
| 	Config        *config.Protocol | type Bdiscord struct { | ||||||
| 	Remote        chan config.Message | 	c              *discordgo.Session | ||||||
| 	Account       string | 	Channels       []*discordgo.Channel | ||||||
| 	Channels      []*discordgo.Channel | 	Nick           string | ||||||
| 	Nick          string | 	UseChannelID   bool | ||||||
| 	UseChannelID  bool | 	userMemberMap  map[string]*discordgo.Member | ||||||
| 	userMemberMap map[string]*discordgo.Member | 	nickMemberMap  map[string]*discordgo.Member | ||||||
| 	guildID       string | 	guildID        string | ||||||
| 	webhookID     string | 	webhookID      string | ||||||
| 	webhookToken  string | 	webhookToken   string | ||||||
|  | 	channelInfoMap map[string]*config.ChannelInfo | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "discord" | 	b := &Bdiscord{Config: cfg} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord { |  | ||||||
| 	b := &bdiscord{} |  | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.userMemberMap = make(map[string]*discordgo.Member) | 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||||
| 	if b.Config.WebhookURL != "" { | 	b.nickMemberMap = make(map[string]*discordgo.Member) | ||||||
| 		flog.Debug("Configuring Discord Incoming Webhook") | 	b.channelInfoMap = make(map[string]*config.ChannelInfo) | ||||||
| 		webhookURLSplit := strings.Split(b.Config.WebhookURL, "/") | 	if b.GetString("WebhookURL") != "" { | ||||||
| 		b.webhookToken = webhookURLSplit[len(webhookURLSplit)-1] | 		b.Log.Debug("Configuring Discord Incoming Webhook") | ||||||
| 		b.webhookID = webhookURLSplit[len(webhookURLSplit)-2] | 		b.webhookID, b.webhookToken = b.splitURL(b.GetString("WebhookURL")) | ||||||
| 	} | 	} | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) Connect() error { | func (b *Bdiscord) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Info("Connecting") | 	var token string | ||||||
| 	if b.Config.WebhookURL == "" { | 	b.Log.Info("Connecting") | ||||||
| 		flog.Info("Connecting using token") | 	if b.GetString("WebhookURL") == "" { | ||||||
|  | 		b.Log.Info("Connecting using token") | ||||||
| 	} else { | 	} 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 ") { | 	if !strings.HasPrefix(b.GetString("Token"), "Bot ") { | ||||||
| 		b.Config.Token = "Bot " + b.Config.Token | 		token = "Bot " + b.GetString("Token") | ||||||
| 	} | 	} | ||||||
| 	b.c, err = discordgo.New(b.Config.Token) | 	b.c, err = discordgo.New(token) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	b.Log.Info("Connection succeeded") | ||||||
| 	b.c.AddHandler(b.messageCreate) | 	b.c.AddHandler(b.messageCreate) | ||||||
| 	b.c.AddHandler(b.memberUpdate) | 	b.c.AddHandler(b.memberUpdate) | ||||||
| 	b.c.AddHandler(b.messageUpdate) | 	b.c.AddHandler(b.messageUpdate) | ||||||
|  | 	b.c.AddHandler(b.messageDelete) | ||||||
| 	err = b.c.Open() | 	err = b.c.Open() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	guilds, err := b.c.UserGuilds(100, "", "") | 	guilds, err := b.c.UserGuilds(100, "", "") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	userinfo, err := b.c.User("@me") | 	userinfo, err := b.c.User("@me") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	serverName := strings.Replace(b.GetString("Server"), "ID:", "", -1) | ||||||
| 	b.Nick = userinfo.Username | 	b.Nick = userinfo.Username | ||||||
| 	for _, guild := range guilds { | 	for _, guild := range guilds { | ||||||
| 		if guild.Name == b.Config.Server { | 		if guild.Name == serverName || guild.ID == serverName { | ||||||
| 			b.Channels, err = b.c.GuildChannels(guild.ID) | 			b.Channels, err = b.c.GuildChannels(guild.ID) | ||||||
| 			b.guildID = guild.ID | 			b.guildID = guild.ID | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				flog.Debugf("%#v", err) |  | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  | 	for _, channel := range b.Channels { | ||||||
|  | 		b.Log.Debugf("found channel %#v", channel) | ||||||
|  | 	} | ||||||
|  | 	// obtaining guild members and initializing nickname mapping | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  | 	members, err := b.c.GuildMembers(b.guildID, "", 1000) | ||||||
|  | 	if err != nil { | ||||||
|  | 		b.Log.Error("Error obtaining guild members", err) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	for _, member := range members { | ||||||
|  | 		b.userMemberMap[member.User.ID] = member | ||||||
|  | 		b.nickMemberMap[member.User.Username] = member | ||||||
|  | 		if member.Nick != "" { | ||||||
|  | 			b.nickMemberMap[member.Nick] = member | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) Disconnect() error { | func (b *Bdiscord) Disconnect() error { | ||||||
| 	return nil | 	return b.c.Close() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) JoinChannel(channel string) error { | func (b *Bdiscord) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	idcheck := strings.Split(channel, "ID:") | 	b.channelInfoMap[channel.ID] = &channel | ||||||
|  | 	idcheck := strings.Split(channel.Name, "ID:") | ||||||
| 	if len(idcheck) > 1 { | 	if len(idcheck) > 1 { | ||||||
| 		b.UseChannelID = true | 		b.UseChannelID = true | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) Send(msg config.Message) error { | func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  |  | ||||||
| 	channelID := b.getChannelID(msg.Channel) | 	channelID := b.getChannelID(msg.Channel) | ||||||
| 	if channelID == "" { | 	if channelID == "" { | ||||||
| 		flog.Errorf("Could not find channelID for %v", msg.Channel) | 		return "", fmt.Errorf("Could not find channelID for %v", msg.Channel) | ||||||
| 		return nil |  | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL == "" { |  | ||||||
| 		flog.Debugf("Broadcasting using token (API)") | 	// Make a action /me of the message | ||||||
| 		if msg.Event == config.EVENT_USER_ACTION { | 	if msg.Event == config.EventUserAction { | ||||||
| 			msg.Username = "_" + msg.Username | 		msg.Text = "_" + msg.Text + "_" | ||||||
| 			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) | ||||||
| 		} | 		} | ||||||
| 		b.c.ChannelMessageSend(channelID, msg.Username+msg.Text) | 	} | ||||||
| 	} else { |  | ||||||
| 		flog.Debugf("Broadcasting using Webhook") | 	// Use webhook to send the message | ||||||
| 		if msg.Event == config.EVENT_USER_ACTION { | 	if wID != "" { | ||||||
| 			msg.Text = "_" + msg.Text + "_" | 		// skip events | ||||||
|  | 		if msg.Event != "" && msg.Event != config.EventJoinLeave && msg.Event != config.EventTopicChange { | ||||||
|  | 			return "", nil | ||||||
| 		} | 		} | ||||||
| 		b.c.WebhookExecute( | 		b.Log.Debugf("Broadcasting using Webhook") | ||||||
| 			b.webhookID, | 		for _, f := range msg.Extra["file"] { | ||||||
| 			b.webhookToken, | 			fi := f.(config.FileInfo) | ||||||
|  | 			if fi.URL != "" { | ||||||
|  | 				msg.Text += " " + fi.URL | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		// skip empty messages | ||||||
|  | 		if msg.Text == "" { | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		msg.Text = helper.ClipMessage(msg.Text, MessageLength) | ||||||
|  | 		msg.Text = b.replaceUserMentions(msg.Text) | ||||||
|  | 		// discord username must be [0..32] max | ||||||
|  | 		if len(msg.Username) > 32 { | ||||||
|  | 			msg.Username = msg.Username[0:32] | ||||||
|  | 		} | ||||||
|  | 		err := b.c.WebhookExecute( | ||||||
|  | 			wID, | ||||||
|  | 			wToken, | ||||||
| 			true, | 			true, | ||||||
| 			&discordgo.WebhookParams{ | 			&discordgo.WebhookParams{ | ||||||
| 				Content:   msg.Text, | 				Content:   msg.Text, | ||||||
| 				Username:  msg.Username, | 				Username:  msg.Username, | ||||||
| 				AvatarURL: msg.Avatar, | 				AvatarURL: msg.Avatar, | ||||||
| 			}) | 			}) | ||||||
|  | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	return nil |  | ||||||
|  | 	b.Log.Debugf("Broadcasting using token (API)") | ||||||
|  |  | ||||||
|  | 	// Delete message | ||||||
|  | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		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) { | ||||||
|  | 			rmsg.Text = helper.ClipMessage(rmsg.Text, MessageLength) | ||||||
|  | 			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) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	msg.Text = helper.ClipMessage(msg.Text, MessageLength) | ||||||
|  | 	msg.Text = b.replaceUserMentions(msg.Text) | ||||||
|  |  | ||||||
|  | 	// 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) { | func (b *Bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { | ||||||
| 	if b.Config.EditDisable { | 	rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EventMsgDelete, Text: config.EventMsgDelete} | ||||||
|  | 	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 | 		return | ||||||
| 	} | 	} | ||||||
| 	// only when message is actually edited | 	// only when message is actually edited | ||||||
| 	if m.Message.EditedTimestamp != "" { | 	if m.Message.EditedTimestamp != "" { | ||||||
| 		flog.Debugf("Sending edit message") | 		b.Log.Debugf("Sending edit message") | ||||||
| 		m.Content = m.Content + b.Config.EditSuffix | 		m.Content += b.GetString("EditSuffix") | ||||||
| 		b.messageCreate(s, (*discordgo.MessageCreate)(m)) | 		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 | 	// not relay our own messages | ||||||
| 	if m.Author.Username == b.Nick { | 	if m.Author.Username == b.Nick { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	// if using webhooks, do not relay if it's ours | 	// if using webhooks, do not relay if it's ours | ||||||
| 	if b.Config.WebhookURL != "" && m.Author.Bot && m.Author.ID == b.webhookID { | 	if b.useWebhook() && m.Author.Bot && b.isWebhookID(m.Author.ID) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// add the url of the attachments to content | ||||||
| 	if len(m.Attachments) > 0 { | 	if len(m.Attachments) > 0 { | ||||||
| 		for _, attach := range m.Attachments { | 		for _, attach := range m.Attachments { | ||||||
| 			m.Content = m.Content + "\n" + attach.URL | 			m.Content = m.Content + "\n" + attach.URL | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var text string | 	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 != "" { | 	if m.Content != "" { | ||||||
| 		flog.Debugf("Receiving message %#v", m.Message) | 		b.Log.Debugf("== Receiving event %#v", m.Message) | ||||||
| 		if len(m.MentionRoles) > 0 { |  | ||||||
| 			m.Message.Content = b.replaceRoleMentions(m.Message.Content) |  | ||||||
| 		} |  | ||||||
| 		m.Message.Content = b.stripCustomoji(m.Message.Content) | 		m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||||
| 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | ||||||
| 		text = m.ContentWithMentionsReplaced() | 		rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c) | ||||||
|  | 		if err != nil { | ||||||
|  | 			b.Log.Errorf("ContentWithMoreMentionsReplaced failed: %s", err) | ||||||
|  | 			rmsg.Text = m.ContentWithMentionsReplaced() | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", | 	// set channel name | ||||||
| 		UserID: m.Author.ID} |  | ||||||
|  |  | ||||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||||
| 	if b.UseChannelID { | 	if b.UseChannelID { | ||||||
| 		rmsg.Channel = "ID:" + m.ChannelID | 		rmsg.Channel = "ID:" + m.ChannelID | ||||||
| 	} | 	} | ||||||
| 	rmsg.Username = b.getNick(m.Author) |  | ||||||
|  |  | ||||||
| 	if b.Config.ShowEmbeds && m.Message.Embeds != nil { | 	// 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 { | 		for _, embed := range m.Message.Embeds { | ||||||
| 			text = text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | 			rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// no empty messages | 	// no empty messages | ||||||
| 	if text == "" { | 	if rmsg.Text == "" { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	text, ok := b.replaceAction(text) | 	// do we have a /me action | ||||||
|  | 	var ok bool | ||||||
|  | 	rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||||
| 	if ok { | 	if ok { | ||||||
| 		rmsg.Event = config.EVENT_USER_ACTION | 		rmsg.Event = config.EventUserAction | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	rmsg.Text = text | 	b.Log.Debugf("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.Account) | 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||||
| 	b.Remote <- 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() | 	b.Lock() | ||||||
| 	if _, ok := b.userMemberMap[m.Member.User.ID]; ok { | 	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.userMemberMap[m.Member.User.ID] = m.Member | ||||||
|  | 	b.nickMemberMap[m.Member.User.Username] = m.Member | ||||||
|  | 	if m.Member.Nick != "" { | ||||||
|  | 		b.nickMemberMap[m.Member.Nick] = m.Member | ||||||
|  | 	} | ||||||
| 	b.Unlock() | 	b.Unlock() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) getNick(user *discordgo.User) string { | func (b *Bdiscord) getNick(user *discordgo.User) string { | ||||||
| 	var err error | 	var err error | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	defer b.Unlock() | 	defer b.Unlock() | ||||||
| @@ -244,7 +353,19 @@ func (b *bdiscord) getNick(user *discordgo.User) string { | |||||||
| 	return user.Username | 	return user.Username | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) getChannelID(name string) string { | func (b *Bdiscord) getGuildMemberByNick(nick string) (*discordgo.Member, error) { | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  | 	if _, ok := b.nickMemberMap[nick]; ok { | ||||||
|  | 		if b.nickMemberMap[nick] != nil { | ||||||
|  | 			return b.nickMemberMap[nick], nil | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil, errors.New("Couldn't find guild member with nick " + nick) // This will most likely get ignored by the caller | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bdiscord) getChannelID(name string) string { | ||||||
| 	idcheck := strings.Split(name, "ID:") | 	idcheck := strings.Split(name, "ID:") | ||||||
| 	if len(idcheck) > 1 { | 	if len(idcheck) > 1 { | ||||||
| 		return idcheck[1] | 		return idcheck[1] | ||||||
| @@ -257,7 +378,7 @@ func (b *bdiscord) getChannelID(name string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) getChannelName(id string) string { | func (b *Bdiscord) getChannelName(id string) string { | ||||||
| 	for _, channel := range b.Channels { | 	for _, channel := range b.Channels { | ||||||
| 		if channel.ID == id { | 		if channel.ID == id { | ||||||
| 			return channel.Name | 			return channel.Name | ||||||
| @@ -266,19 +387,7 @@ func (b *bdiscord) getChannelName(id string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) replaceRoleMentions(text string) string { | func (b *Bdiscord) replaceChannelMentions(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 { |  | ||||||
| 	var err error | 	var err error | ||||||
| 	re := regexp.MustCompile("<#[0-9]+>") | 	re := regexp.MustCompile("<#[0-9]+>") | ||||||
| 	text = re.ReplaceAllStringFunc(text, func(m string) string { | 	text = re.ReplaceAllStringFunc(text, func(m string) string { | ||||||
| @@ -297,15 +406,106 @@ func (b *bdiscord) replaceChannelMentions(text string) string { | |||||||
| 	return text | 	return text | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) replaceAction(text string) (string, bool) { | func (b *Bdiscord) replaceUserMentions(text string) string { | ||||||
|  | 	re := regexp.MustCompile("@[^@]{1,32}") | ||||||
|  | 	text = re.ReplaceAllStringFunc(text, func(m string) string { | ||||||
|  | 		mention := strings.TrimSpace(m[1:]) | ||||||
|  | 		var member *discordgo.Member | ||||||
|  | 		var err error | ||||||
|  | 		for { | ||||||
|  | 			b.Log.Debugf("Testing mention: '%s'", mention) | ||||||
|  | 			member, err = b.getGuildMemberByNick(mention) | ||||||
|  | 			if err != nil { | ||||||
|  | 				lastSpace := strings.LastIndex(mention, " ") | ||||||
|  | 				if lastSpace == -1 { | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 				mention = strings.TrimSpace(mention[0:lastSpace]) | ||||||
|  | 			} else { | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if err != nil { | ||||||
|  | 			return m | ||||||
|  | 		} | ||||||
|  | 		return strings.Replace(m, "@"+mention, member.User.Mention(), -1) | ||||||
|  | 	}) | ||||||
|  | 	b.Log.Debugf("Message with mention replaced: %s", text) | ||||||
|  | 	return text | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bdiscord) replaceAction(text string) (string, bool) { | ||||||
| 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||||
| 		return strings.Replace(text, "_", "", -1), true | 		return strings.Replace(text, "_", "", -1), true | ||||||
| 	} | 	} | ||||||
| 	return text, false | 	return text, false | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *bdiscord) stripCustomoji(text string) string { | func (b *Bdiscord) stripCustomoji(text string) string { | ||||||
| 	// <:doge:302803592035958784> | 	// <:doge:302803592035958784> | ||||||
| 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||||
| 	return re.ReplaceAllString(text, `$1`) | 	return re.ReplaceAllString(text, `$1`) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // splitURL splits a webhookURL and returns the id and token | ||||||
|  | func (b *Bdiscord) splitURL(url string) (string, string) { | ||||||
|  | 	webhookURLSplit := strings.Split(url, "/") | ||||||
|  | 	if len(webhookURLSplit) != 7 { | ||||||
|  | 		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) | ||||||
|  | 		file := discordgo.File{ | ||||||
|  | 			Name:        fi.Name, | ||||||
|  | 			ContentType: "", | ||||||
|  | 			Reader:      bytes.NewReader(*fi.Data), | ||||||
|  | 		} | ||||||
|  | 		m := discordgo.MessageSend{ | ||||||
|  | 			Content: msg.Username + fi.Comment, | ||||||
|  | 			Files:   []*discordgo.File{&file}, | ||||||
|  | 		} | ||||||
|  | 		_, err = b.c.ChannelMessageSendComplex(channelID, &m) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", fmt.Errorf("file upload failed: %#v", err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return "", nil | ||||||
|  | } | ||||||
|   | |||||||
| @@ -2,47 +2,39 @@ package bgitter | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/sromku/go-gitter" |  | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/go-gitter" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bgitter struct { | type Bgitter struct { | ||||||
| 	c       *gitter.Gitter | 	c     *gitter.Gitter | ||||||
| 	Config  *config.Protocol | 	User  *gitter.User | ||||||
| 	Remote  chan config.Message | 	Users []gitter.User | ||||||
| 	Account string | 	Rooms []gitter.Room | ||||||
| 	Users   []gitter.User | 	*bridge.Config | ||||||
| 	Rooms   []gitter.Room |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "gitter" | 	return &Bgitter{Config: cfg} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bgitter { |  | ||||||
| 	b := &Bgitter{} |  | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) Connect() error { | func (b *Bgitter) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Info("Connecting") | 	b.Log.Info("Connecting") | ||||||
| 	b.c = gitter.New(b.Config.Token) | 	b.c = gitter.New(b.GetString("Token")) | ||||||
| 	_, err = b.c.GetUser() | 	b.User, err = b.c.GetUser() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	b.Rooms, err = b.c.GetRooms() | ||||||
| 	b.Rooms, _ = b.c.GetRooms() | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.Log.Info("Connection succeeded") | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -51,10 +43,10 @@ func (b *Bgitter) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) JoinChannel(channel string) error { | func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	roomID, err := b.c.GetRoomId(channel) | 	roomID, err := b.c.GetRoomId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel) | 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel.Name) | ||||||
| 	} | 	} | ||||||
| 	room, err := b.c.GetRoom(roomID) | 	room, err := b.c.GetRoom(roomID) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -78,34 +70,74 @@ func (b *Bgitter) JoinChannel(channel string) error { | |||||||
| 		for event := range stream.Event { | 		for event := range stream.Event { | ||||||
| 			switch ev := event.Data.(type) { | 			switch ev := event.Data.(type) { | ||||||
| 			case *gitter.MessageReceived: | 			case *gitter.MessageReceived: | ||||||
| 				// check for ZWSP to see if it's not an echo | 				// ignore message sent from ourselves | ||||||
| 				if !strings.HasSuffix(ev.Message.Text, "") { | 				if ev.Message.From.ID != b.User.ID { | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | 					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, | 					rmsg := config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | ||||||
| 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID} | 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID, | ||||||
|  | 						ID: ev.Message.ID} | ||||||
| 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | ||||||
| 						rmsg.Event = config.EVENT_USER_ACTION | 						rmsg.Event = config.EventUserAction | ||||||
| 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | ||||||
| 					} | 					} | ||||||
|  | 					b.Log.Debugf("<= Message is %#v", rmsg) | ||||||
| 					b.Remote <- rmsg | 					b.Remote <- rmsg | ||||||
| 				} | 				} | ||||||
| 			case *gitter.GitterConnectionClosed: | 			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 | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bgitter) Send(msg config.Message) error { | func (b *Bgitter) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
| 	roomID := b.getRoomID(msg.Channel) | 	roomID := b.getRoomID(msg.Channel) | ||||||
| 	if roomID == "" { | 	if roomID == "" { | ||||||
| 		flog.Errorf("Could not find roomID for %v", msg.Channel) | 		b.Log.Errorf("Could not find roomID for %v", msg.Channel) | ||||||
| 		return nil | 		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.EventMsgDelete { | ||||||
|  | 		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 { | func (b *Bgitter) getRoomID(channel string) string { | ||||||
| @@ -128,3 +160,23 @@ func (b *Bgitter) getAvatar(user string) string { | |||||||
| 	} | 	} | ||||||
| 	return avatar | 	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 | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										153
									
								
								bridge/helper/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,153 @@ | |||||||
|  | package helper | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | 	"unicode/utf8" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetSubLines splits messages in newline-delimited lines. If maxLineLength is | ||||||
|  | // specified as non-zero GetSubLines will and also clip long lines to the | ||||||
|  | // maximum length and insert a warning marker that the line was clipped. | ||||||
|  | // | ||||||
|  | // TODO: The current implementation has the inconvenient that it disregards | ||||||
|  | // word boundaries when splitting but this is hard to solve without potentially | ||||||
|  | // breaking formatting and other stylistic effects. | ||||||
|  | func GetSubLines(message string, maxLineLength int) []string { | ||||||
|  | 	const clippingMessage = " <clipped message>" | ||||||
|  |  | ||||||
|  | 	var lines []string | ||||||
|  | 	for _, line := range strings.Split(strings.TrimSpace(message), "\n") { | ||||||
|  | 		if maxLineLength == 0 || len([]byte(line)) <= maxLineLength { | ||||||
|  | 			lines = append(lines, line) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// !!! WARNING !!! | ||||||
|  | 		// Before touching the splitting logic below please ensure that you PROPERLY | ||||||
|  | 		// understand how strings, runes and range loops over strings work in Go. | ||||||
|  | 		// A good place to start is to read https://blog.golang.org/strings. :-) | ||||||
|  | 		var splitStart int | ||||||
|  | 		var startOfPreviousRune int | ||||||
|  | 		for i := range line { | ||||||
|  | 			if i-splitStart > maxLineLength-len([]byte(clippingMessage)) { | ||||||
|  | 				lines = append(lines, line[splitStart:startOfPreviousRune]+clippingMessage) | ||||||
|  | 				splitStart = startOfPreviousRune | ||||||
|  | 			} | ||||||
|  | 			startOfPreviousRune = i | ||||||
|  | 		} | ||||||
|  | 		// This last append is safe to do without looking at the remaining byte-length | ||||||
|  | 		// as we assume that the byte-length of the last rune will never exceed that of | ||||||
|  | 		// the byte-length of the clipping message. | ||||||
|  | 		lines = append(lines, line[splitStart:]) | ||||||
|  | 	} | ||||||
|  | 	return lines | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handle all the stuff we put into extra | ||||||
|  | func HandleExtra(msg *config.Message, general *config.Protocol) []config.Message { | ||||||
|  | 	extra := msg.Extra | ||||||
|  | 	rmsg := []config.Message{} | ||||||
|  | 	for _, f := range extra[config.EventFileFailureSize] { | ||||||
|  | 		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, Account: msg.Account}) | ||||||
|  | 	} | ||||||
|  | 	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 { | ||||||
|  | 	// check blacklist here | ||||||
|  | 	for _, entry := range general.MediaDownloadBlackList { | ||||||
|  | 		if entry != "" { | ||||||
|  | 			re, err := regexp.Compile(entry) | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if re.MatchString(name) { | ||||||
|  | 				return fmt.Errorf("Matching blacklist %s. Not downloading %s", entry, name) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	flog.Debugf("Trying to download %#v with size %#v", name, size) | ||||||
|  | 	if int(size) > general.MediaDownloadSize { | ||||||
|  | 		msg.Event = config.EventFileFailureSize | ||||||
|  | 		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.EventAvatarDownload { | ||||||
|  | 		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 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ClipMessage(text string, length int) string { | ||||||
|  | 	// clip too long messages | ||||||
|  | 	if len(text) > length { | ||||||
|  | 		text = text[:length-len(" *message clipped*")] | ||||||
|  | 		if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||||
|  | 			text = text[:len(text)-size] | ||||||
|  | 		} | ||||||
|  | 		text += " *message clipped*" | ||||||
|  | 	} | ||||||
|  | 	return text | ||||||
|  | } | ||||||
							
								
								
									
										105
									
								
								bridge/helper/helper_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,105 @@ | |||||||
|  | package helper | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const testLineLength = 64 | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	lineSplittingTestCases = map[string]struct { | ||||||
|  | 		input          string | ||||||
|  | 		splitOutput    []string | ||||||
|  | 		nonSplitOutput []string | ||||||
|  | 	}{ | ||||||
|  | 		"Short single-line message": { | ||||||
|  | 			input:          "short", | ||||||
|  | 			splitOutput:    []string{"short"}, | ||||||
|  | 			nonSplitOutput: []string{"short"}, | ||||||
|  | 		}, | ||||||
|  | 		"Long single-line message": { | ||||||
|  | 			input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||||
|  | 			splitOutput: []string{ | ||||||
|  | 				"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>", | ||||||
|  | 				"cing elit, sed do eiusmod tempor incididunt ut <clipped message>", | ||||||
|  | 				" labore et dolore magna aliqua.", | ||||||
|  | 			}, | ||||||
|  | 			nonSplitOutput: []string{"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."}, | ||||||
|  | 		}, | ||||||
|  | 		"Short multi-line message": { | ||||||
|  | 			input: "I\ncan't\nget\nno\nsatisfaction!", | ||||||
|  | 			splitOutput: []string{ | ||||||
|  | 				"I", | ||||||
|  | 				"can't", | ||||||
|  | 				"get", | ||||||
|  | 				"no", | ||||||
|  | 				"satisfaction!", | ||||||
|  | 			}, | ||||||
|  | 			nonSplitOutput: []string{ | ||||||
|  | 				"I", | ||||||
|  | 				"can't", | ||||||
|  | 				"get", | ||||||
|  | 				"no", | ||||||
|  | 				"satisfaction!", | ||||||
|  | 			}, | ||||||
|  | 		}, | ||||||
|  | 		"Long multi-line message": { | ||||||
|  | 			input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.\n" + | ||||||
|  | 				"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.\n" + | ||||||
|  | 				"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.\n" + | ||||||
|  | 				"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.", | ||||||
|  | 			splitOutput: []string{ | ||||||
|  | 				"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>", | ||||||
|  | 				"cing elit, sed do eiusmod tempor incididunt ut <clipped message>", | ||||||
|  | 				" labore et dolore magna aliqua.", | ||||||
|  | 				"Ut enim ad minim veniam, quis nostrud exercita <clipped message>", | ||||||
|  | 				"tion ullamco laboris nisi ut aliquip ex ea com <clipped message>", | ||||||
|  | 				"modo consequat.", | ||||||
|  | 				"Duis aute irure dolor in reprehenderit in volu <clipped message>", | ||||||
|  | 				"ptate velit esse cillum dolore eu fugiat nulla <clipped message>", | ||||||
|  | 				" pariatur.", | ||||||
|  | 				"Excepteur sint occaecat cupidatat non proident <clipped message>", | ||||||
|  | 				", sunt in culpa qui officia deserunt mollit an <clipped message>", | ||||||
|  | 				"im id est laborum.", | ||||||
|  | 			}, | ||||||
|  | 			nonSplitOutput: []string{ | ||||||
|  | 				"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||||
|  | 				"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.", | ||||||
|  | 				"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.", | ||||||
|  | 				"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.", | ||||||
|  | 			}, | ||||||
|  | 		}, | ||||||
|  | 		"Message ending with new-line.": { | ||||||
|  | 			input:          "Newline ending\n", | ||||||
|  | 			splitOutput:    []string{"Newline ending"}, | ||||||
|  | 			nonSplitOutput: []string{"Newline ending"}, | ||||||
|  | 		}, | ||||||
|  | 		"Long message containing UTF-8 multi-byte runes": { | ||||||
|  | 			input: "不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說", | ||||||
|  | 			splitOutput: []string{ | ||||||
|  | 				"不布人個我此而及單石業喜資富下 <clipped message>", | ||||||
|  | 				"我河下日沒一我臺空達的常景便物 <clipped message>", | ||||||
|  | 				"沒為……子大我別名解成?生賣的 <clipped message>", | ||||||
|  | 				"全直黑,我自我結毛分洲了世當, <clipped message>", | ||||||
|  | 				"是政福那是東;斯說", | ||||||
|  | 			}, | ||||||
|  | 			nonSplitOutput: []string{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"}, | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestGetSubLines(t *testing.T) { | ||||||
|  | 	for testname, testcase := range lineSplittingTestCases { | ||||||
|  | 		splitLines := GetSubLines(testcase.input, testLineLength) | ||||||
|  | 		assert.Equalf(t, testcase.splitOutput, splitLines, "'%s' testcase should give expected lines with splitting.", testname) | ||||||
|  | 		for _, splitLine := range splitLines { | ||||||
|  | 			byteLength := len([]byte(splitLine)) | ||||||
|  | 			assert.True(t, byteLength <= testLineLength, "Splitted line '%s' of testcase '%s' should not exceed the maximum byte-length (%d vs. %d).", splitLine, testcase, byteLength, testLineLength) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		nonSplitLines := GetSubLines(testcase.input, 0) | ||||||
|  | 		assert.Equalf(t, testcase.nonSplitOutput, nonSplitLines, "'%s' testcase should give expected lines without splitting.", testname) | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,61 +0,0 @@ | |||||||
| package birc |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"strings" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| /* |  | ||||||
| func tableformatter(nicks []string, nicksPerRow int, continued bool) string { |  | ||||||
| 	result := "|IRC users" |  | ||||||
| 	if continued { |  | ||||||
| 		result = "|(continued)" |  | ||||||
| 	} |  | ||||||
| 	for i := 0; i < 2; i++ { |  | ||||||
| 		for j := 1; j <= nicksPerRow && j <= len(nicks); j++ { |  | ||||||
| 			if i == 0 { |  | ||||||
| 				result += "|" |  | ||||||
| 			} else { |  | ||||||
| 				result += ":-|" |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		result += "\r\n|" |  | ||||||
| 	} |  | ||||||
| 	result += nicks[0] + "|" |  | ||||||
| 	for i := 1; i < len(nicks); i++ { |  | ||||||
| 		if i%nicksPerRow == 0 { |  | ||||||
| 			result += "\r\n|" + nicks[i] + "|" |  | ||||||
| 		} else { |  | ||||||
| 			result += nicks[i] + "|" |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return result |  | ||||||
| } |  | ||||||
| */ |  | ||||||
|  |  | ||||||
| func plainformatter(nicks []string, nicksPerRow int) string { |  | ||||||
| 	return strings.Join(nicks, ", ") + " currently on IRC" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func IsMarkup(message string) bool { |  | ||||||
| 	switch message[0] { |  | ||||||
| 	case '|': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '#': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '_': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '*': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '~': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '-': |  | ||||||
| 		fallthrough |  | ||||||
| 	case ':': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '>': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '=': |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
| @@ -1,329 +1,486 @@ | |||||||
| package birc | package birc | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/go-ircevent" | 	"hash/crc32" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/paulrosania/go-charset/charset" |  | ||||||
| 	_ "github.com/paulrosania/go-charset/data" |  | ||||||
| 	"github.com/saintfish/chardet" |  | ||||||
| 	ircm "github.com/sorcix/irc" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
|  | 	"net" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"sort" | 	"sort" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	"github.com/dfordsoft/golib/ic" | ||||||
|  | 	"github.com/lrstanley/girc" | ||||||
|  | 	"github.com/paulrosania/go-charset/charset" | ||||||
|  | 	"github.com/saintfish/chardet" | ||||||
|  |  | ||||||
|  | 	// We need to import the 'data' package as an implicit dependency. | ||||||
|  | 	// See: https://godoc.org/github.com/paulrosania/go-charset/charset | ||||||
|  | 	_ "github.com/paulrosania/go-charset/data" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Birc struct { | type Birc struct { | ||||||
| 	i               *irc.Connection | 	i                                         *girc.Client | ||||||
| 	Nick            string | 	Nick                                      string | ||||||
| 	names           map[string][]string | 	names                                     map[string][]string | ||||||
| 	Config          *config.Protocol | 	connected                                 chan error | ||||||
| 	Remote          chan config.Message | 	Local                                     chan config.Message // local queue for flood control | ||||||
| 	connected       chan struct{} | 	FirstConnection, authDone                 bool | ||||||
| 	Local           chan config.Message // local queue for flood control | 	MessageDelay, MessageQueue, MessageLength int | ||||||
| 	Account         string |  | ||||||
| 	FirstConnection bool | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "irc" |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Birc { |  | ||||||
| 	b := &Birc{} | 	b := &Birc{} | ||||||
| 	b.Config = &cfg | 	b.Config = cfg | ||||||
| 	b.Nick = b.Config.Nick | 	b.Nick = b.GetString("Nick") | ||||||
| 	b.Remote = c |  | ||||||
| 	b.names = make(map[string][]string) | 	b.names = make(map[string][]string) | ||||||
| 	b.Account = account | 	b.connected = make(chan error) | ||||||
| 	b.connected = make(chan struct{}) | 	if b.GetInt("MessageDelay") == 0 { | ||||||
| 	if b.Config.MessageDelay == 0 { | 		b.MessageDelay = 1300 | ||||||
| 		b.Config.MessageDelay = 1300 | 	} else { | ||||||
|  | 		b.MessageDelay = b.GetInt("MessageDelay") | ||||||
| 	} | 	} | ||||||
| 	if b.Config.MessageQueue == 0 { | 	if b.GetInt("MessageQueue") == 0 { | ||||||
| 		b.Config.MessageQueue = 30 | 		b.MessageQueue = 30 | ||||||
|  | 	} else { | ||||||
|  | 		b.MessageQueue = b.GetInt("MessageQueue") | ||||||
| 	} | 	} | ||||||
| 	if b.Config.MessageLength == 0 { | 	if b.GetInt("MessageLength") == 0 { | ||||||
| 		b.Config.MessageLength = 400 | 		b.MessageLength = 400 | ||||||
|  | 	} else { | ||||||
|  | 		b.MessageLength = b.GetInt("MessageLength") | ||||||
| 	} | 	} | ||||||
| 	b.FirstConnection = true | 	b.FirstConnection = true | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Command(msg *config.Message) string { | func (b *Birc) Command(msg *config.Message) string { | ||||||
| 	switch msg.Text { | 	if msg.Text == "!users" { | ||||||
| 	case "!users": | 		b.i.Handlers.Add(girc.RPL_NAMREPLY, b.storeNames) | ||||||
| 		b.i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames) | 		b.i.Handlers.Add(girc.RPL_ENDOFNAMES, b.endNames) | ||||||
| 		b.i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames) | 		b.i.Cmd.SendRaw("NAMES " + msg.Channel) | ||||||
| 		b.i.SendRaw("NAMES " + msg.Channel) |  | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Connect() error { | func (b *Birc) Connect() error { | ||||||
| 	b.Local = make(chan config.Message, b.Config.MessageQueue+10) | 	b.Local = make(chan config.Message, b.MessageQueue+10) | ||||||
| 	flog.Infof("Connecting %s", b.Config.Server) | 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||||
| 	i := irc.IRC(b.Config.Nick, b.Config.Nick) | 	server, portstr, err := net.SplitHostPort(b.GetString("Server")) | ||||||
| 	if log.GetLevel() == log.DebugLevel { |  | ||||||
| 		i.Debug = true |  | ||||||
| 	} |  | ||||||
| 	i.UseTLS = b.Config.UseTLS |  | ||||||
| 	i.UseSASL = b.Config.UseSASL |  | ||||||
| 	i.SASLLogin = b.Config.NickServNick |  | ||||||
| 	i.SASLPassword = b.Config.NickServPassword |  | ||||||
| 	i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify} |  | ||||||
| 	i.KeepAlive = time.Minute |  | ||||||
| 	i.PingFreq = time.Minute |  | ||||||
| 	if b.Config.Password != "" { |  | ||||||
| 		i.Password = b.Config.Password |  | ||||||
| 	} |  | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, b.handleNewConnection) |  | ||||||
| 	err := i.Connect(b.Config.Server) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	b.i = i | 	port, err := strconv.Atoi(portstr) | ||||||
| 	select { | 	if err != nil { | ||||||
| 	case <-b.connected: | 		return err | ||||||
| 		flog.Info("Connection succeeded") |  | ||||||
| 	case <-time.After(time.Second * 30): |  | ||||||
| 		return fmt.Errorf("connection timed out") |  | ||||||
| 	} | 	} | ||||||
| 	i.Debug = false | 	// fix strict user handling of girc | ||||||
| 	// clear on reconnects | 	user := b.GetString("Nick") | ||||||
| 	i.ClearCallback(ircm.RPL_WELCOME) | 	for !girc.IsValidUser(user) { | ||||||
| 	i.AddCallback(ircm.RPL_WELCOME, func(event *irc.Event) { | 		if len(user) == 1 { | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | 			user = "matterbridge" | ||||||
| 		// set our correct nick on reconnect if necessary | 			break | ||||||
| 		b.Nick = event.Nick | 		} | ||||||
|  | 		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, | ||||||
| 	}) | 	}) | ||||||
| 	go i.Loop() |  | ||||||
|  | 	if b.GetBool("UseSASL") { | ||||||
|  | 		i.Config.SASL = &girc.SASLPlain{ | ||||||
|  | 			User: b.GetString("NickServNick"), | ||||||
|  | 			Pass: 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("disconnect: error: %s", err) | ||||||
|  | 				if b.FirstConnection { | ||||||
|  | 					b.connected <- err | ||||||
|  | 					return | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				b.Log.Info("disconnect: client requested quit") | ||||||
|  | 			} | ||||||
|  | 			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.EventRejoinChannels} | ||||||
|  | 				// set our correct nick on reconnect if necessary | ||||||
|  | 				b.Nick = event.Source.Name | ||||||
|  | 			}) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 	b.i = i | ||||||
|  | 	err = <-b.connected | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("connection failed %s", err) | ||||||
|  | 	} | ||||||
|  | 	b.Log.Info("Connection succeeded") | ||||||
|  | 	b.FirstConnection = false | ||||||
|  | 	if b.GetInt("DebugLevel") == 0 { | ||||||
|  | 		i.Handlers.Clear(girc.ALL_EVENTS) | ||||||
|  | 	} | ||||||
| 	go b.doSend() | 	go b.doSend() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Disconnect() error { | func (b *Birc) Disconnect() error { | ||||||
| 	//b.i.Disconnect() | 	b.i.Close() | ||||||
| 	close(b.Local) | 	close(b.Local) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) JoinChannel(channel string) error { | func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.i.Join(channel) | 	// need to check if we have nickserv auth done before joining channels | ||||||
|  | 	for { | ||||||
|  | 		if b.authDone { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		time.Sleep(time.Second) | ||||||
|  | 	} | ||||||
|  | 	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) | ||||||
|  | 	} | ||||||
|  | 	b.authDone = false | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) Send(msg config.Message) error { | func (b *Birc) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		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, "!") { | 	if strings.HasPrefix(msg.Text, "!") { | ||||||
| 		b.Command(&msg) | 		b.Command(&msg) | ||||||
| 	} | 	} | ||||||
| 	for _, text := range strings.Split(msg.Text, "\n") { |  | ||||||
| 		if len(text) > b.Config.MessageLength { | 	// convert to specified charset | ||||||
| 			text = text[:b.Config.MessageLength] + " <message clipped>" | 	if b.GetString("Charset") != "" { | ||||||
| 		} | 		switch b.GetString("Charset") { | ||||||
| 		if len(b.Local) < b.Config.MessageQueue { | 		case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||||
| 			if len(b.Local) == b.Config.MessageQueue-1 { | 			msg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), msg.Text) | ||||||
| 				text = text + " <message clipped>" | 		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, Event: msg.Event} | 			fmt.Fprint(w, msg.Text) | ||||||
| 		} else { | 			w.Close() | ||||||
| 			flog.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | 			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 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var msgLines []string | ||||||
|  | 	if b.GetBool("MessageSplit") { | ||||||
|  | 		msgLines = helper.GetSubLines(msg.Text, b.MessageLength) | ||||||
|  | 	} else { | ||||||
|  | 		msgLines = helper.GetSubLines(msg.Text, 0) | ||||||
|  | 	} | ||||||
|  | 	for i := range msgLines { | ||||||
|  | 		if len(b.Local) >= b.MessageQueue { | ||||||
|  | 			b.Log.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | ||||||
|  | 			return "", nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		b.Local <- config.Message{ | ||||||
|  | 			Text:     msgLines[i], | ||||||
|  | 			Username: msg.Username, | ||||||
|  | 			Channel:  msg.Channel, | ||||||
|  | 			Event:    msg.Event, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) doSend() { | func (b *Birc) doSend() { | ||||||
| 	rate := time.Millisecond * time.Duration(b.Config.MessageDelay) | 	rate := time.Millisecond * time.Duration(b.MessageDelay) | ||||||
| 	throttle := time.NewTicker(rate) | 	throttle := time.NewTicker(rate) | ||||||
| 	for msg := range b.Local { | 	for msg := range b.Local { | ||||||
| 		<-throttle.C | 		<-throttle.C | ||||||
| 		if msg.Event == config.EVENT_USER_ACTION { | 		username := msg.Username | ||||||
| 			b.i.Action(msg.Channel, msg.Username+msg.Text) | 		if b.GetBool("Colornicks") { | ||||||
|  | 			checksum := crc32.ChecksumIEEE([]byte(msg.Username)) | ||||||
|  | 			colorCode := checksum%14 + 2 // quick fix - prevent white or black color codes | ||||||
|  | 			username = fmt.Sprintf("\x03%02d%s\x0F", colorCode, msg.Username) | ||||||
|  | 		} | ||||||
|  | 		if msg.Event == config.EventUserAction { | ||||||
|  | 			b.i.Cmd.Action(msg.Channel, username+msg.Text) | ||||||
| 		} else { | 		} else { | ||||||
| 			b.i.Privmsg(msg.Channel, msg.Username+msg.Text) | 			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) { | func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[1] | 	channel := event.Params[1] | ||||||
| 	sort.Strings(b.names[channel]) | 	sort.Strings(b.names[channel]) | ||||||
| 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | ||||||
| 	continued := false |  | ||||||
| 	for len(b.names[channel]) > maxNamesPerPost { | 	for len(b.names[channel]) > maxNamesPerPost { | ||||||
| 		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost], continued), | 		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]), | ||||||
| 			Channel: channel, Account: b.Account} | 			Channel: channel, Account: b.Account} | ||||||
| 		b.names[channel] = b.names[channel][maxNamesPerPost:] | 		b.names[channel] = b.names[channel][maxNamesPerPost:] | ||||||
| 		continued = true |  | ||||||
| 	} | 	} | ||||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel]), | ||||||
| 		Channel: channel, Account: b.Account} | 		Channel: channel, Account: b.Account} | ||||||
| 	b.names[channel] = nil | 	b.names[channel] = nil | ||||||
| 	b.i.ClearCallback(ircm.RPL_NAMREPLY) | 	b.i.Handlers.Clear(girc.RPL_NAMREPLY) | ||||||
| 	b.i.ClearCallback(ircm.RPL_ENDOFNAMES) | 	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNewConnection(event *irc.Event) { | func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||||
| 	flog.Debug("Registering callbacks") | 	b.Log.Debug("Registering callbacks") | ||||||
| 	i := b.i | 	i := b.i | ||||||
| 	b.Nick = event.Arguments[0] | 	b.Nick = event.Params[0] | ||||||
| 	i.AddCallback("PRIVMSG", b.handlePrivMsg) |  | ||||||
| 	i.AddCallback("CTCP_ACTION", b.handlePrivMsg) | 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||||
| 	i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||||
| 	i.AddCallback(ircm.NOTICE, b.handleNotice) | 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||||
| 	//i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) }) | 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||||
| 	i.AddCallback("PING", func(e *irc.Event) { | 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||||
| 		i.SendRaw("PONG :" + e.Message()) | 	i.Handlers.Add("PART", b.handleJoinPart) | ||||||
| 		flog.Debugf("PING/PONG") | 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||||
| 	}) | 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||||
| 	i.AddCallback("JOIN", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("PART", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("QUIT", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("KICK", b.handleJoinPart) |  | ||||||
| 	i.AddCallback("*", b.handleOther) |  | ||||||
| 	// we are now fully connected |  | ||||||
| 	b.connected <- struct{}{} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleJoinPart(event *irc.Event) { | func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[0] | 	if len(event.Params) == 0 { | ||||||
| 	if event.Code == "KICK" { | 		b.Log.Debugf("handleJoinPart: empty Params? %#v", event) | ||||||
| 		flog.Infof("Got kicked from %s by %s", channel, event.Nick) |  | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} |  | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if event.Code == "QUIT" { | 	channel := strings.ToLower(event.Params[0]) | ||||||
| 		if event.Nick == b.Nick && strings.Contains(event.Raw, "Ping timeout") { | 	if event.Command == "KICK" && event.Params[1] == b.Nick { | ||||||
| 			flog.Infof("%s reconnecting ..", b.Account) | 		b.Log.Infof("Got kicked from %s by %s", channel, event.Source.Name) | ||||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | 		time.Sleep(time.Duration(b.GetInt("RejoinDelay")) * time.Second) | ||||||
|  | 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EventRejoinChannels} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	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.EventFailure} | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	if event.Nick != b.Nick { | 	if event.Source.Name != b.Nick { | ||||||
| 		flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | 		if b.GetBool("nosendjoinpart") { | ||||||
| 		b.Remote <- config.Message{Username: "system", Text: event.Nick + " " + strings.ToLower(event.Code) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | 			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.EventJoinLeave} | ||||||
|  | 		b.Log.Debugf("<= Message is %#v", msg) | ||||||
|  | 		b.Remote <- msg | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("handle %#v", event) | 	b.Log.Debugf("handle %#v", event) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleNotice(event *irc.Event) { | func (b *Birc) handleNotice(client *girc.Client, event girc.Event) { | ||||||
| 	if strings.Contains(event.Message(), "This nickname is registered") && event.Nick == b.Config.NickServNick { | 	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.GetString("NickServNick") { | ||||||
| 		b.i.Privmsg(b.Config.NickServNick, "IDENTIFY "+b.Config.NickServPassword) | 		b.handleNickServ() | ||||||
| 	} else { | 	} else { | ||||||
| 		b.handlePrivMsg(event) | 		b.handlePrivMsg(client, event) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleOther(event *irc.Event) { | func (b *Birc) handleOther(client *girc.Client, event girc.Event) { | ||||||
| 	switch event.Code { | 	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": | 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | ||||||
| 		return | 		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) { | ||||||
|  | 	b.handleNickServ() | ||||||
|  | 	// we are now fully connected | ||||||
|  | 	b.connected <- nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Birc) skipPrivMsg(event girc.Event) bool { | ||||||
|  | 	// Our nick can be changed | ||||||
| 	b.Nick = b.i.GetNick() | 	b.Nick = b.i.GetNick() | ||||||
|  |  | ||||||
| 	// freenode doesn't send 001 as first reply | 	// freenode doesn't send 001 as first reply | ||||||
| 	if event.Code == "NOTICE" { | 	if event.Command == "NOTICE" { | ||||||
| 		return | 		return true | ||||||
| 	} | 	} | ||||||
| 	// don't forward queries to the bot | 	// don't forward queries to the bot | ||||||
| 	if event.Arguments[0] == b.Nick { | 	if event.Params[0] == b.Nick { | ||||||
| 		return | 		return true | ||||||
| 	} | 	} | ||||||
| 	// don't forward message from ourself | 	// don't forward message from ourself | ||||||
| 	if event.Nick == b.Nick { | 	if event.Source.Name == b.Nick { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||||
|  | 	if b.skipPrivMsg(event) { | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	rmsg := config.Message{Username: event.Nick, Channel: event.Arguments[0], Account: b.Account, UserID: event.User + "@" + event.Host} | 	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||||
| 	flog.Debugf("handlePrivMsg() %s %s %#v", event.Nick, event.Message(), event) | 	b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Trailing, event) | ||||||
| 	msg := "" |  | ||||||
| 	if event.Code == "CTCP_ACTION" { | 	// set action event | ||||||
| 		//	msg = event.Nick + " " | 	if event.IsAction() { | ||||||
| 		rmsg.Event = config.EVENT_USER_ACTION | 		rmsg.Event = config.EventUserAction | ||||||
| 	} | 	} | ||||||
| 	msg += event.Message() |  | ||||||
|  | 	// strip action, we made an event if it was an action | ||||||
|  | 	rmsg.Text += event.StripAction() | ||||||
|  |  | ||||||
| 	// strip IRC colors | 	// strip IRC colors | ||||||
| 	re := regexp.MustCompile(`[[:cntrl:]](\d+,|)\d+`) | 	re := regexp.MustCompile(`\x03(?:\d{1,2}(?:,\d{1,2})?)?|[[:cntrl:]]`) | ||||||
| 	msg = re.ReplaceAllString(msg, "") | 	rmsg.Text = re.ReplaceAllString(rmsg.Text, "") | ||||||
|  |  | ||||||
| 	// detect what were sending so that we convert it to utf-8 | 	// start detecting the charset | ||||||
| 	detector := chardet.NewTextDetector() |  | ||||||
| 	result, err := detector.DetectBest([]byte(msg)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		flog.Infof("detection failed for msg: %#v", msg) |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	flog.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) |  | ||||||
| 	var r io.Reader | 	var r io.Reader | ||||||
| 	r, err = charset.NewReader(result.Charset, strings.NewReader(msg)) | 	var err error | ||||||
| 	// if we're not sure, just pick ISO-8859-1 | 	mycharset := b.GetString("Charset") | ||||||
| 	if result.Confidence < 80 { | 	if mycharset == "" { | ||||||
| 		r, err = charset.NewReader("ISO-8859-1", strings.NewReader(msg)) | 		// detect what were sending so that we convert it to utf-8 | ||||||
|  | 		detector := chardet.NewTextDetector() | ||||||
|  | 		result, err := detector.DetectBest([]byte(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" | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| 	if err != nil { | 	switch mycharset { | ||||||
| 		flog.Errorf("charset to utf-8 conversion failed: %s", err) | 	case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||||
| 		return | 		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) | ||||||
| 	} | 	} | ||||||
| 	output, _ := ioutil.ReadAll(r) |  | ||||||
| 	msg = string(output) |  | ||||||
|  |  | ||||||
| 	flog.Debugf("Sending message from %s on %s to gateway", event.Arguments[0], b.Account) | 	b.Log.Debugf("<= Sending message from %s on %s to gateway", event.Params[0], b.Account) | ||||||
| 	rmsg.Text = msg |  | ||||||
| 	b.Remote <- rmsg | 	b.Remote <- rmsg | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) handleTopicWhoTime(event *irc.Event) { | func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) { | ||||||
| 	parts := strings.Split(event.Arguments[2], "!") | 	parts := strings.Split(event.Params[2], "!") | ||||||
| 	t, err := strconv.ParseInt(event.Arguments[3], 10, 64) | 	t, err := strconv.ParseInt(event.Params[3], 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Errorf("Invalid time stamp: %s", event.Arguments[3]) | 		b.Log.Errorf("Invalid time stamp: %s", event.Params[3]) | ||||||
| 	} | 	} | ||||||
| 	user := parts[0] | 	user := parts[0] | ||||||
| 	if len(parts) > 1 { | 	if len(parts) > 1 { | ||||||
| 		user += " [" + parts[1] + "]" | 		user += " [" + parts[1] + "]" | ||||||
| 	} | 	} | ||||||
| 	flog.Debugf("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0)) | 	b.Log.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0)) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) nicksPerRow() int { | func (b *Birc) nicksPerRow() int { | ||||||
| 	return 4 | 	return 4 | ||||||
| 	/* |  | ||||||
| 		if b.Config.Mattermost.NicksPerRow < 1 { |  | ||||||
| 			return 4 |  | ||||||
| 		} |  | ||||||
| 		return b.Config.Mattermost.NicksPerRow |  | ||||||
| 	*/ |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) storeNames(event *irc.Event) { | func (b *Birc) storeNames(client *girc.Client, event girc.Event) { | ||||||
| 	channel := event.Arguments[2] | 	channel := event.Params[2] | ||||||
| 	b.names[channel] = append( | 	b.names[channel] = append( | ||||||
| 		b.names[channel], | 		b.names[channel], | ||||||
| 		strings.Split(strings.TrimSpace(event.Message()), " ")...) | 		strings.Split(strings.TrimSpace(event.Trailing), " ")...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Birc) formatnicks(nicks []string, continued bool) string { | func (b *Birc) formatnicks(nicks []string) string { | ||||||
| 	return plainformatter(nicks, b.nicksPerRow()) | 	return strings.Join(nicks, ", ") + " currently on IRC" | ||||||
| 	/* | } | ||||||
| 		switch b.Config.Mattermost.NickFormatter { |  | ||||||
| 		case "table": | func (b *Birc) handleNickServ() { | ||||||
| 			return tableformatter(nicks, b.nicksPerRow(), continued) | 	if !b.GetBool("UseSASL") && b.GetString("NickServNick") != "" && b.GetString("NickServPassword") != "" { | ||||||
| 		default: | 		b.Log.Debugf("Sending identify to nickserv %s", b.GetString("NickServNick")) | ||||||
| 			return plainformatter(nicks, b.nicksPerRow()) | 		b.i.Cmd.Message(b.GetString("NickServNick"), "IDENTIFY "+b.GetString("NickServPassword")) | ||||||
| 		} | 	} | ||||||
| 	*/ | 	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")) | ||||||
|  | 	} | ||||||
|  | 	// give nickserv some slack | ||||||
|  | 	time.Sleep(time.Second * 5) | ||||||
|  | 	b.authDone = true | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,60 +1,51 @@ | |||||||
| package bmatrix | package bmatrix | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"mime" | ||||||
| 	"regexp" | 	"regexp" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	matrix "github.com/matrix-org/gomatrix" | 	matrix "github.com/matterbridge/gomatrix" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bmatrix struct { | type Bmatrix struct { | ||||||
| 	mc      *matrix.Client | 	mc      *matrix.Client | ||||||
| 	Config  *config.Protocol |  | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| 	UserID  string | 	UserID  string | ||||||
| 	RoomMap map[string]string | 	RoomMap map[string]string | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "matrix" | 	b := &Bmatrix{Config: cfg} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bmatrix { |  | ||||||
| 	b := &Bmatrix{} |  | ||||||
| 	b.RoomMap = make(map[string]string) | 	b.RoomMap = make(map[string]string) | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.Remote = c |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) Connect() error { | func (b *Bmatrix) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Infof("Connecting %s", b.Config.Server) | 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||||
| 	b.mc, err = matrix.NewClient(b.Config.Server, "", "") | 	b.mc, err = matrix.NewClient(b.GetString("Server"), "", "") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	resp, err := b.mc.Login(&matrix.ReqLogin{ | 	resp, err := b.mc.Login(&matrix.ReqLogin{ | ||||||
| 		Type:     "m.login.password", | 		Type:     "m.login.password", | ||||||
| 		User:     b.Config.Login, | 		User:     b.GetString("Login"), | ||||||
| 		Password: b.Config.Password, | 		Password: b.GetString("Password"), | ||||||
| 	}) | 	}) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	b.mc.SetCredentials(resp.UserID, resp.AccessToken) | 	b.mc.SetCredentials(resp.UserID, resp.AccessToken) | ||||||
| 	b.UserID = resp.UserID | 	b.UserID = resp.UserID | ||||||
| 	flog.Info("Connection succeeded") | 	b.Log.Info("Connection succeeded") | ||||||
| 	go b.handlematrix() | 	go b.handlematrix() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -63,28 +54,68 @@ func (b *Bmatrix) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) JoinChannel(channel string) error { | func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	resp, err := b.mc.JoinRoom(channel, "", nil) | 	resp, err := b.mc.JoinRoom(channel.Name, "", nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	b.Lock() | 	b.Lock() | ||||||
| 	b.RoomMap[resp.RoomID] = channel | 	b.RoomMap[resp.RoomID] = channel.Name | ||||||
| 	b.Unlock() | 	b.Unlock() | ||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmatrix) Send(msg config.Message) error { | func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  |  | ||||||
| 	channel := b.getRoomID(msg.Channel) | 	channel := b.getRoomID(msg.Channel) | ||||||
| 	flog.Debugf("Sending to channel %s", channel) | 	b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel) | ||||||
| 	if msg.Event == config.EVENT_USER_ACTION { |  | ||||||
| 		b.mc.SendMessageEvent(channel, "m.room.message", | 	// Make a action /me of the message | ||||||
| 			matrix.TextMessage{"m.emote", msg.Username + msg.Text}) | 	if msg.Event == config.EventUserAction { | ||||||
| 		return nil | 		m := matrix.TextMessage{ | ||||||
|  | 			MsgType: "m.emote", | ||||||
|  | 			Body:    msg.Username + msg.Text, | ||||||
|  | 		} | ||||||
|  | 		resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		return resp.EventID, err | ||||||
| 	} | 	} | ||||||
| 	b.mc.SendText(channel, msg.Username+msg.Text) |  | ||||||
| 	return nil | 	// Delete message | ||||||
|  | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		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 { | func (b *Bmatrix) getRoomID(channel string) string { | ||||||
| @@ -97,37 +128,188 @@ func (b *Bmatrix) getRoomID(channel string) string { | |||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
| func (b *Bmatrix) handlematrix() error { |  | ||||||
|  | func (b *Bmatrix) handlematrix() { | ||||||
| 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||||
| 	syncer.OnEventType("m.room.message", func(ev *matrix.Event) { | 	syncer.OnEventType("m.room.redaction", b.handleEvent) | ||||||
| 		if (ev.Content["msgtype"].(string) == "m.text" || ev.Content["msgtype"].(string) == "m.emote") && ev.Sender != b.UserID { | 	syncer.OnEventType("m.room.message", b.handleEvent) | ||||||
| 			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`) |  | ||||||
| 			} |  | ||||||
| 			rmsg := config.Message{Username: username, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account, UserID: ev.Sender} |  | ||||||
| 			if ev.Content["msgtype"].(string) == "m.emote" { |  | ||||||
| 				rmsg.Event = config.EVENT_USER_ACTION |  | ||||||
| 			} |  | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) |  | ||||||
| 			b.Remote <- rmsg |  | ||||||
| 		} |  | ||||||
| 		flog.Debugf("Received: %#v", ev) |  | ||||||
| 	}) |  | ||||||
| 	go func() { | 	go func() { | ||||||
| 		for { | 		for { | ||||||
| 			if err := b.mc.Sync(); err != nil { | 			if err := b.mc.Sync(); err != nil { | ||||||
| 				flog.Println("Sync() returned ", err) | 				b.Log.Println("Sync() returned ", err) | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	}() | 	}() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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] is not a string: %T\n%#v", | ||||||
|  | 				ev.Content["body"], ev.Content) | ||||||
|  | 			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.EventMsgDelete | ||||||
|  | 			rmsg.ID = ev.Redacts | ||||||
|  | 			rmsg.Text = config.EventMsgDelete | ||||||
|  | 			b.Remote <- rmsg | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Do we have a /me action | ||||||
|  | 		if ev.Content["msgtype"].(string) == "m.emote" { | ||||||
|  | 			rmsg.Event = config.EventUserAction | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// 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 += mext[0] | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			// just a default .png extension if we don't have mime info | ||||||
|  | 			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 | 	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 | ||||||
|  | } | ||||||
|   | |||||||
| @@ -2,51 +2,32 @@ package bmattermost | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/42wim/matterbridge/matterclient" | 	"github.com/42wim/matterbridge/matterclient" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/mattermost/platform/model" | ||||||
| 	"strings" | 	"github.com/rs/xid" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type MMhook struct { |  | ||||||
| 	mh *matterhook.Client |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type MMapi struct { |  | ||||||
| 	mc    *matterclient.MMClient |  | ||||||
| 	mmMap map[string]string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type MMMessage struct { |  | ||||||
| 	Text     string |  | ||||||
| 	Channel  string |  | ||||||
| 	Username string |  | ||||||
| 	UserID   string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type Bmattermost struct { | type Bmattermost struct { | ||||||
| 	MMhook | 	mh     *matterhook.Client | ||||||
| 	MMapi | 	mc     *matterclient.MMClient | ||||||
| 	Config  *config.Protocol | 	uuid   string | ||||||
| 	Remote  chan config.Message | 	TeamID string | ||||||
| 	TeamId  string | 	*bridge.Config | ||||||
| 	Account string | 	avatarMap map[string]string | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | const mattermostPlugin = "mattermost.plugin" | ||||||
| var protocol = "mattermost" |  | ||||||
|  |  | ||||||
| func init() { | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	b := &Bmattermost{Config: cfg, avatarMap: make(map[string]string)} | ||||||
| } | 	b.uuid = xid.New().String() | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bmattermost { |  | ||||||
| 	b := &Bmattermost{} |  | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.mmMap = make(map[string]string) |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -55,34 +36,52 @@ func (b *Bmattermost) Command(cmd string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Connect() error { | func (b *Bmattermost) Connect() error { | ||||||
| 	if b.Config.WebhookBindAddress != "" { | 	if b.Account == mattermostPlugin { | ||||||
| 		if b.Config.WebhookURL != "" { | 		return nil | ||||||
| 			flog.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | 	} | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 	if b.GetString("WebhookBindAddress") != "" { | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 		switch { | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) | 		case b.GetString("WebhookURL") != "": | ||||||
| 		} else if b.Config.Login != "" { | 			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||||
| 			flog.Info("Connecting using login/password (sending)") | 			b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||||
|  | 				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||||
|  | 					BindAddress: b.GetString("WebhookBindAddress")}) | ||||||
|  | 		case b.GetString("Token") != "": | ||||||
|  | 			b.Log.Info("Connecting using token (sending)") | ||||||
| 			err := b.apiLogin() | 			err := b.apiLogin() | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 		} else { | 		case b.GetString("Login") != "": | ||||||
| 			flog.Info("Connecting using webhookbindaddress (receiving)") | 			b.Log.Info("Connecting using login/password (sending)") | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 			err := b.apiLogin() | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 			if err != nil { | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) | 				return err | ||||||
|  | 			} | ||||||
|  | 		default: | ||||||
|  | 			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() | 		go b.handleMatter() | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL != "" { | 	switch { | ||||||
| 		flog.Info("Connecting using webhookurl (sending)") | 	case b.GetString("WebhookURL") != "": | ||||||
| 		b.mh = matterhook.New(b.Config.WebhookURL, | 		b.Log.Info("Connecting using webhookurl (sending)") | ||||||
| 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 		b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||||
|  | 			matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||||
| 				DisableServer: true}) | 				DisableServer: true}) | ||||||
| 		if b.Config.Login != "" { | 		if b.GetString("Token") != "" { | ||||||
| 			flog.Info("Connecting using login/password (receiving)") | 			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() | 			err := b.apiLogin() | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
| @@ -90,16 +89,23 @@ func (b *Bmattermost) Connect() error { | |||||||
| 			go b.handleMatter() | 			go b.handleMatter() | ||||||
| 		} | 		} | ||||||
| 		return nil | 		return nil | ||||||
| 	} else if b.Config.Login != "" { | 	case b.GetString("Token") != "": | ||||||
| 		flog.Info("Connecting using login/password (sending and receiving)") | 		b.Log.Info("Connecting using token (sending and receiving)") | ||||||
|  | 		err := b.apiLogin() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		go b.handleMatter() | ||||||
|  | 	case b.GetString("Login") != "": | ||||||
|  | 		b.Log.Info("Connecting using login/password (sending and receiving)") | ||||||
| 		err := b.apiLogin() | 		err := b.apiLogin() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		go b.handleMatter() | 		go b.handleMatter() | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Login == "" { | 	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 Login/Password/Server/Team configured.") | 		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured") | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -108,138 +114,369 @@ func (b *Bmattermost) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) JoinChannel(channel string) error { | func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||||
|  | 	if b.Account == mattermostPlugin { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
| 	// we can only join channels using the API | 	// we can only join channels using the API | ||||||
| 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | 	if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" { | ||||||
| 		return b.mc.JoinChannel(b.mc.GetChannelId(channel, "")) | 		id := b.mc.GetChannelId(channel.Name, b.TeamID) | ||||||
|  | 		if id == "" { | ||||||
|  | 			return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) | ||||||
|  | 		} | ||||||
|  | 		return b.mc.JoinChannel(id) | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) Send(msg config.Message) error { | func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	if b.Account == mattermostPlugin { | ||||||
| 	if msg.Event == config.EVENT_USER_ACTION { | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  |  | ||||||
|  | 	// Make a action /me of the message | ||||||
|  | 	if msg.Event == config.EventUserAction { | ||||||
| 		msg.Text = "*" + msg.Text + "*" | 		msg.Text = "*" + msg.Text + "*" | ||||||
| 	} | 	} | ||||||
| 	nick := msg.Username |  | ||||||
| 	message := msg.Text |  | ||||||
| 	channel := msg.Channel |  | ||||||
|  |  | ||||||
| 	if b.Config.PrefixMessagesWithNick { | 	// map the file SHA to our user (caches the avatar) | ||||||
| 		message = nick + message | 	if msg.Event == config.EventAvatarDownload { | ||||||
|  | 		return b.cacheAvatar(&msg) | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL != "" { |  | ||||||
| 		matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 	// Use webhook to send the message | ||||||
| 		matterMessage.IconURL = msg.Avatar | 	if b.GetString("WebhookURL") != "" { | ||||||
| 		matterMessage.Channel = channel | 		return b.sendWebhook(msg) | ||||||
| 		matterMessage.UserName = nick | 	} | ||||||
| 		matterMessage.Type = "" |  | ||||||
| 		matterMessage.Text = message | 	// Delete message | ||||||
| 		err := b.mh.Send(matterMessage) | 	if msg.Event == config.EventMsgDelete { | ||||||
| 		if err != nil { | 		if msg.ID == "" { | ||||||
| 			flog.Info(err) | 			return "", nil | ||||||
| 			return err |  | ||||||
| 		} | 		} | ||||||
| 		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, b.TeamID), 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, b.TeamID), msg.Text) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) handleMatter() { | func (b *Bmattermost) handleMatter() { | ||||||
| 	mchan := make(chan *MMMessage) | 	messages := make(chan *config.Message) | ||||||
| 	if b.Config.WebhookBindAddress != "" { | 	if b.GetString("WebhookBindAddress") != "" { | ||||||
| 		flog.Debugf("Choosing webhooks based receiving") | 		b.Log.Debugf("Choosing webhooks based receiving") | ||||||
| 		go b.handleMatterHook(mchan) | 		go b.handleMatterHook(messages) | ||||||
| 	} else { | 	} else { | ||||||
| 		flog.Debugf("Choosing login/password based receiving") | 		if b.GetString("Token") != "" { | ||||||
| 		go b.handleMatterClient(mchan) | 			b.Log.Debugf("Choosing token based receiving") | ||||||
| 	} | 		} else { | ||||||
| 	for message := range mchan { | 			b.Log.Debugf("Choosing login/password based receiving") | ||||||
| 		rmsg := config.Message{Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID} |  | ||||||
| 		text, ok := b.replaceAction(message.Text) |  | ||||||
| 		if ok { |  | ||||||
| 			rmsg.Event = config.EVENT_USER_ACTION |  | ||||||
| 		} | 		} | ||||||
| 		rmsg.Text = text | 		go b.handleMatterClient(messages) | ||||||
| 		flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) | 	} | ||||||
| 		b.Remote <- rmsg | 	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.EventUserAction | ||||||
|  | 		} | ||||||
|  | 		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 { | 	for message := range b.mc.MessageChan { | ||||||
| 		flog.Debugf("%#v", message.Raw.Data) | 		b.Log.Debugf("%#v", message.Raw.Data) | ||||||
| 		if message.Type == "system_join_leave" || |  | ||||||
| 			message.Type == "system_join_channel" || | 		if b.skipMessage(message) { | ||||||
| 			message.Type == "system_leave_channel" { | 			b.Log.Debugf("Skipped message: %#v", message) | ||||||
| 			flog.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) |  | ||||||
| 			b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} |  | ||||||
| 			continue | 			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.General.MediaDownloadPath != "" { | ||||||
|  | 			b.handleDownloadAvatar(message.UserID, message.Channel) | ||||||
| 		} | 		} | ||||||
| 		// do not post our own messages back to irc |  | ||||||
| 		// only listen to message from our team | 		b.Log.Debugf("== Receiving event %#v", message) | ||||||
| 		if (message.Raw.Event == "posted" || message.Raw.Event == "post_edited") && |  | ||||||
| 			b.mc.User.Username != message.Username && message.Raw.Data["team_id"].(string) == b.TeamId { | 		rmsg := &config.Message{Username: message.Username, UserID: message.UserID, Channel: message.Channel, Text: message.Text, ID: message.Post.Id, Extra: make(map[string][]interface{})} | ||||||
| 			// if the message has reactions don't repost it (for now, until we can correlate reaction with message) |  | ||||||
| 			if message.Post.HasReactions { | 		// handle mattermost post properties (override username and attachments) | ||||||
| 				continue | 		props := message.Post.Props | ||||||
|  | 		if props != nil { | ||||||
|  | 			if _, ok := props["override_username"].(string); ok { | ||||||
|  | 				rmsg.Username = props["override_username"].(string) | ||||||
| 			} | 			} | ||||||
| 			flog.Debugf("Receiving from matterclient %#v", message) | 			if _, ok := props["attachments"].([]interface{}); ok { | ||||||
| 			m := &MMMessage{} | 				rmsg.Extra["attachments"] = props["attachments"].([]interface{}) | ||||||
| 			m.UserID = message.UserID | 				if rmsg.Text == "" { | ||||||
| 			m.Username = message.Username | 					for _, attachment := range rmsg.Extra["attachments"] { | ||||||
| 			m.Channel = message.Channel | 						attach := attachment.(map[string]interface{}) | ||||||
| 			m.Text = message.Text | 						if attach["text"].(string) != "" { | ||||||
| 			if message.Raw.Event == "post_edited" && !b.Config.EditDisable { | 							rmsg.Text += attach["text"].(string) | ||||||
| 				m.Text = message.Text + b.Config.EditSuffix | 							continue | ||||||
| 			} | 						} | ||||||
| 			if len(message.Post.FileIds) > 0 { | 						if attach["fallback"].(string) != "" { | ||||||
| 				for _, link := range b.mc.GetFileLinks(message.Post.FileIds) { | 							rmsg.Text += attach["fallback"].(string) | ||||||
| 					m.Text = m.Text + "\n" + link | 						} | ||||||
|  | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 			mchan <- m |  | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		// create a text for bridges that don't support native editing | ||||||
|  | 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") { | ||||||
|  | 			rmsg.Text = message.Text + b.GetString("EditSuffix") | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED { | ||||||
|  | 			rmsg.Event = config.EventMsgDelete | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		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) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		// Use nickname instead of username if defined | ||||||
|  | 		if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||||
|  | 			rmsg.Username = nick | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		messages <- rmsg | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) handleMatterHook(mchan chan *MMMessage) { | func (b *Bmattermost) handleMatterHook(messages chan *config.Message) { | ||||||
| 	for { | 	for { | ||||||
| 		message := b.mh.Receive() | 		message := b.mh.Receive() | ||||||
| 		flog.Debugf("Receiving from matterhook %#v", message) | 		b.Log.Debugf("Receiving from matterhook %#v", message) | ||||||
| 		m := &MMMessage{} | 		messages <- &config.Message{UserID: message.UserID, Username: message.UserName, Text: message.Text, Channel: message.ChannelName} | ||||||
| 		m.UserID = message.UserID |  | ||||||
| 		m.Username = message.UserName |  | ||||||
| 		m.Text = message.Text |  | ||||||
| 		m.Channel = message.ChannelName |  | ||||||
| 		mchan <- m |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bmattermost) apiLogin() error { | func (b *Bmattermost) apiLogin() error { | ||||||
| 	b.mc = matterclient.New(b.Config.Login, b.Config.Password, | 	password := b.GetString("Password") | ||||||
| 		b.Config.Team, b.Config.Server) | 	if b.GetString("Token") != "" { | ||||||
| 	b.mc.SkipTLSVerify = b.Config.SkipTLSVerify | 		password = "token=" + b.GetString("Token") | ||||||
| 	b.mc.NoTLS = b.Config.NoTLS | 	} | ||||||
| 	flog.Infof("Connecting %s (team: %s) on %s", b.Config.Login, b.Config.Team, b.Config.Server) |  | ||||||
|  | 	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() | 	err := b.mc.Login() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	b.Log.Info("Connection succeeded") | ||||||
| 	b.TeamId = b.mc.GetTeamId() | 	b.TeamID = b.mc.GetTeamId() | ||||||
| 	go b.mc.WsReceiver() | 	go b.mc.WsReceiver() | ||||||
| 	go b.mc.StatusLoop() | 	go b.mc.StatusLoop() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // replaceAction replace the message with the correct action (/me) code | ||||||
| func (b *Bmattermost) replaceAction(text string) (string, bool) { | func (b *Bmattermost) replaceAction(text string) (string, bool) { | ||||||
| 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | ||||||
| 		return strings.Replace(text, "*", "", -1), true | 		return strings.Replace(text, "*", "", -1), true | ||||||
| 	} | 	} | ||||||
| 	return text, false | 	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.EventAvatarDownload, 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, b.TeamID) | ||||||
|  | 	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) { | ||||||
|  | 			rmsg := rmsg // scopelint | ||||||
|  | 			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.EventJoinLeave} | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Handle edited messages | ||||||
|  | 	if (message.Raw.Event == model.WEBSOCKET_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 == model.WEBSOCKET_EVENT_POST_EDITED || message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED) { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,37 +1,21 @@ | |||||||
| package brocketchat | package brocketchat | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/42wim/matterbridge/hook/rockethook" | 	"github.com/42wim/matterbridge/hook/rockethook" | ||||||
| 	"github.com/42wim/matterbridge/matterhook" | 	"github.com/42wim/matterbridge/matterhook" | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type MMhook struct { | type Brocketchat struct { | ||||||
| 	mh *matterhook.Client | 	mh *matterhook.Client | ||||||
| 	rh *rockethook.Client | 	rh *rockethook.Client | ||||||
|  | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| type Brocketchat struct { | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| 	MMhook | 	return &Brocketchat{Config: cfg} | ||||||
| 	Config  *config.Protocol |  | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var flog *log.Entry |  | ||||||
| var protocol = "rocketchat" |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Brocketchat { |  | ||||||
| 	b := &Brocketchat{} |  | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Command(cmd string) string { | func (b *Brocketchat) Command(cmd string) string { | ||||||
| @@ -39,11 +23,11 @@ func (b *Brocketchat) Command(cmd string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Connect() error { | func (b *Brocketchat) Connect() error { | ||||||
| 	flog.Info("Connecting webhooks") | 	b.Log.Info("Connecting webhooks") | ||||||
| 	b.mh = matterhook.New(b.Config.WebhookURL, | 	b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||||
| 		matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 		matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||||
| 			DisableServer: true}) | 			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() | 	go b.handleRocketHook() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -53,34 +37,56 @@ func (b *Brocketchat) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) JoinChannel(channel string) error { | func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) Send(msg config.Message) error { | func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	// ignore delete messages | ||||||
| 	matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL} | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
|  | 			rmsg := rmsg // scopelint | ||||||
|  | 			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.Channel = msg.Channel | ||||||
| 	matterMessage.UserName = msg.Username | 	matterMessage.UserName = msg.Username | ||||||
| 	matterMessage.Type = "" | 	matterMessage.Type = "" | ||||||
| 	matterMessage.Text = msg.Text | 	matterMessage.Text = msg.Text | ||||||
| 	err := b.mh.Send(matterMessage) | 	err := b.mh.Send(matterMessage) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Info(err) | 		b.Log.Info(err) | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Brocketchat) handleRocketHook() { | func (b *Brocketchat) handleRocketHook() { | ||||||
| 	for { | 	for { | ||||||
| 		message := b.rh.Receive() | 		message := b.rh.Receive() | ||||||
| 		flog.Debugf("Receiving from rockethook %#v", message) | 		b.Log.Debugf("Receiving from rockethook %#v", message) | ||||||
| 		// do not loop | 		// do not loop | ||||||
| 		if message.UserName == b.Config.Nick { | 		if message.UserName == b.GetString("Nick") { | ||||||
| 			continue | 			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} | 		b.Remote <- config.Message{Text: message.Text, Username: message.UserName, Channel: message.ChannelName, Account: b.Account, UserID: message.UserID} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										311
									
								
								bridge/slack/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,311 @@ | |||||||
|  | package bslack | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"html" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	"github.com/nlopes/slack" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func (b *Bslack) handleSlack() { | ||||||
|  | 	messages := make(chan *config.Message) | ||||||
|  | 	if b.GetString(incomingWebhookConfig) != "" { | ||||||
|  | 		b.Log.Debugf("Choosing webhooks based receiving") | ||||||
|  | 		go b.handleMatterHook(messages) | ||||||
|  | 	} else { | ||||||
|  | 		b.Log.Debugf("Choosing token based receiving") | ||||||
|  | 		go b.handleSlackClient(messages) | ||||||
|  | 	} | ||||||
|  | 	time.Sleep(time.Second) | ||||||
|  | 	b.Log.Debug("Start listening for Slack messages") | ||||||
|  | 	for message := range messages { | ||||||
|  | 		if message.Event != config.EventUserTyping { | ||||||
|  | 			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(message.UserID) | ||||||
|  |  | ||||||
|  | 		b.Log.Debugf("<= Message is %#v", message) | ||||||
|  | 		b.Remote <- *message | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||||
|  | 	for msg := range b.rtm.IncomingEvents { | ||||||
|  | 		if msg.Type != sUserTyping && msg.Type != sLatencyReport { | ||||||
|  | 			b.Log.Debugf("== Receiving event %#v", msg.Data) | ||||||
|  | 		} | ||||||
|  | 		switch ev := msg.Data.(type) { | ||||||
|  | 		case *slack.UserTypingEvent: | ||||||
|  | 			if !b.GetBool("ShowUserTyping") { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			rmsg, err := b.handleTypingEvent(ev) | ||||||
|  | 			if err != nil { | ||||||
|  | 				b.Log.Errorf("%#v", err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			messages <- rmsg | ||||||
|  | 		case *slack.MessageEvent: | ||||||
|  | 			if b.skipMessageEvent(ev) { | ||||||
|  | 				b.Log.Debugf("Skipped message: %#v", ev) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			rmsg, err := b.handleMessageEvent(ev) | ||||||
|  | 			if err != nil { | ||||||
|  | 				b.Log.Errorf("%#v", err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			messages <- rmsg | ||||||
|  | 		case *slack.OutgoingErrorEvent: | ||||||
|  | 			b.Log.Debugf("%#v", ev.Error()) | ||||||
|  | 		case *slack.ChannelJoinedEvent: | ||||||
|  | 			// When we join a channel we update the full list of users as | ||||||
|  | 			// well as the information for the channel that we joined as this | ||||||
|  | 			// should now tell that we are a member of it. | ||||||
|  | 			b.populateUsers() | ||||||
|  |  | ||||||
|  | 			b.channelsMutex.Lock() | ||||||
|  | 			b.channelsByID[ev.Channel.ID] = &ev.Channel | ||||||
|  | 			b.channelsByName[ev.Channel.Name] = &ev.Channel | ||||||
|  | 			b.channelsMutex.Unlock() | ||||||
|  | 		case *slack.ConnectedEvent: | ||||||
|  | 			b.si = ev.Info | ||||||
|  | 			b.populateChannels() | ||||||
|  | 			b.populateUsers() | ||||||
|  | 		case *slack.InvalidAuthEvent: | ||||||
|  | 			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(messages chan *config.Message) { | ||||||
|  | 	for { | ||||||
|  | 		message := b.mh.Receive() | ||||||
|  | 		b.Log.Debugf("receiving from matterhook (slack) %#v", message) | ||||||
|  | 		if message.UserName == "slackbot" { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		messages <- &config.Message{ | ||||||
|  | 			Username: message.UserName, | ||||||
|  | 			Text:     message.Text, | ||||||
|  | 			Channel:  message.ChannelName, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // skipMessageEvent skips event that need to be skipped :-) | ||||||
|  | func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool { | ||||||
|  | 	switch ev.SubType { | ||||||
|  | 	case sChannelLeave, sChannelJoin: | ||||||
|  | 		return b.GetBool(noSendJoinConfig) | ||||||
|  | 	case sPinnedItem, sUnpinnedItem: | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Skip any messages that we made ourselves or from 'slackbot' (see #527). | ||||||
|  | 	if ev.Username == sSlackBotUser || | ||||||
|  | 		(b.rtm != nil && ev.Username == b.si.User.Name) || | ||||||
|  | 		(len(ev.Attachments) > 0 && ev.Attachments[0].CallbackID == "matterbridge_"+b.uuid) { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// It seems ev.SubMessage.Edited == nil when slack unfurls. | ||||||
|  | 	// Do not forward these messages. See Github issue #266. | ||||||
|  | 	if ev.SubMessage != nil && | ||||||
|  | 		ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp && | ||||||
|  | 		ev.SubMessage.Edited == nil { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(ev.Files) > 0 { | ||||||
|  | 		return b.filesCached(ev.Files) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) filesCached(files []slack.File) bool { | ||||||
|  | 	for i := range files { | ||||||
|  | 		if !b.fileCached(&files[i]) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleMessageEvent handles the message events. Together with any called sub-methods, | ||||||
|  | // this method implements the following event processing pipeline: | ||||||
|  | // | ||||||
|  | // 1. Check if the message should be ignored. | ||||||
|  | //    NOTE: This is not actually part of the method below but is done just before it | ||||||
|  | //          is called via the 'skipMessageEvent()' method. | ||||||
|  | // 2. Populate the Matterbridge message that will be sent to the router based on the | ||||||
|  | //    received event and logic that is common to all events that are not skipped. | ||||||
|  | // 3. Detect and handle any message that is "status" related (think join channel, etc.). | ||||||
|  | //    This might result in an early exit from the pipeline and passing of the | ||||||
|  | //    pre-populated message to the Matterbridge router. | ||||||
|  | // 4. Handle the specific case of messages that edit existing messages depending on | ||||||
|  | //    configuration. | ||||||
|  | // 5. Handle any attachments of the received event. | ||||||
|  | // 6. Check that the Matterbridge message that we end up with after at the end of the | ||||||
|  | //    pipeline is valid before sending it to the Matterbridge router. | ||||||
|  | func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, error) { | ||||||
|  | 	rmsg, err := b.populateReceivedMessage(ev) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Handle some message types early. | ||||||
|  | 	if b.handleStatusEvent(ev, rmsg) { | ||||||
|  | 		return rmsg, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.handleAttachments(ev, rmsg) | ||||||
|  |  | ||||||
|  | 	// Verify that we have the right information and the message | ||||||
|  | 	// is well-formed before sending it out to the router. | ||||||
|  | 	if len(ev.Files) == 0 && (rmsg.Text == "" || rmsg.Username == "") { | ||||||
|  | 		if ev.BotID != "" { | ||||||
|  | 			// This is probably a webhook we couldn't resolve. | ||||||
|  | 			return nil, fmt.Errorf("message handling resulted in an empty bot message (probably an incoming webhook we couldn't resolve): %#v", ev) | ||||||
|  | 		} | ||||||
|  | 		return nil, fmt.Errorf("message handling resulted in an empty message: %#v", ev) | ||||||
|  | 	} | ||||||
|  | 	return rmsg, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) bool { | ||||||
|  | 	switch ev.SubType { | ||||||
|  | 	case sChannelJoined, sMemberJoined: | ||||||
|  | 		b.populateUsers() | ||||||
|  | 		// There's no further processing needed on channel events | ||||||
|  | 		// so we return 'true'. | ||||||
|  | 		return true | ||||||
|  | 	case sChannelJoin, sChannelLeave: | ||||||
|  | 		rmsg.Username = sSystemUser | ||||||
|  | 		rmsg.Event = config.EventJoinLeave | ||||||
|  | 	case sChannelTopic, sChannelPurpose: | ||||||
|  | 		rmsg.Event = config.EventTopicChange | ||||||
|  | 	case sMessageChanged: | ||||||
|  | 		rmsg.Text = ev.SubMessage.Text | ||||||
|  | 		// handle deleted thread starting messages | ||||||
|  | 		if ev.SubMessage.Text == "This message was deleted." { | ||||||
|  | 			rmsg.Event = config.EventMsgDelete | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	case sMessageDeleted: | ||||||
|  | 		rmsg.Text = config.EventMsgDelete | ||||||
|  | 		rmsg.Event = config.EventMsgDelete | ||||||
|  | 		rmsg.ID = ev.DeletedTimestamp | ||||||
|  | 		// If a message is being deleted we do not need to process | ||||||
|  | 		// the event any further so we return 'true'. | ||||||
|  | 		return true | ||||||
|  | 	case sMeMessage: | ||||||
|  | 		rmsg.Event = config.EventUserAction | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) { | ||||||
|  | 	// File comments are set by the system (because there is no username given). | ||||||
|  | 	if ev.SubType == sFileComment { | ||||||
|  | 		rmsg.Username = sSystemUser | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// See if we have some text in the attachments. | ||||||
|  | 	if rmsg.Text == "" { | ||||||
|  | 		for _, attach := range ev.Attachments { | ||||||
|  | 			if attach.Text != "" { | ||||||
|  | 				if attach.Title != "" { | ||||||
|  | 					rmsg.Text = attach.Title + "\n" | ||||||
|  | 				} | ||||||
|  | 				rmsg.Text += attach.Text | ||||||
|  | 			} else { | ||||||
|  | 				rmsg.Text = attach.Fallback | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Save the attachments, so that we can send them to other slack (compatible) bridges. | ||||||
|  | 	if len(ev.Attachments) > 0 { | ||||||
|  | 		rmsg.Extra[sSlackAttachment] = append(rmsg.Extra[sSlackAttachment], ev.Attachments) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If we have files attached, download them (in memory) and put a pointer to it in msg.Extra. | ||||||
|  | 	for i := range ev.Files { | ||||||
|  | 		if err := b.handleDownloadFile(rmsg, &ev.Files[i]); err != nil { | ||||||
|  | 			b.Log.Errorf("Could not download incoming file: %#v", err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message, error) { | ||||||
|  | 	channelInfo, err := b.getChannelByID(ev.Channel) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return &config.Message{ | ||||||
|  | 		Channel: channelInfo.Name, | ||||||
|  | 		Account: b.Account, | ||||||
|  | 		Event:   config.EventUserTyping, | ||||||
|  | 	}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleDownloadFile handles file download | ||||||
|  | func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) error { | ||||||
|  | 	if b.fileCached(file) { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	// Check that the file is neither too large nor blacklisted. | ||||||
|  | 	if err := helper.HandleDownloadSize(b.Log, rmsg, file.Name, int64(file.Size), b.General); err != nil { | ||||||
|  | 		b.Log.WithError(err).Infof("Skipping download of incoming file.") | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Actually download the file. | ||||||
|  | 	data, err := helper.DownloadFileAuth(file.URLPrivateDownload, "Bearer "+b.GetString(tokenConfig)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("download %s failed %#v", file.URLPrivateDownload, err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If a comment is attached to the file(s) it is in the 'Text' field of the Slack messge event | ||||||
|  | 	// and should be added as comment to only one of the files. We reset the 'Text' field to ensure | ||||||
|  | 	// that the comment is not duplicated. | ||||||
|  | 	comment := rmsg.Text | ||||||
|  | 	rmsg.Text = "" | ||||||
|  | 	helper.HandleDownloadData(b.Log, rmsg, file.Name, comment, file.URLPrivateDownload, data, b.General) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // fileCached implements Matterbridge's caching logic for files | ||||||
|  | // shared via Slack. | ||||||
|  | // | ||||||
|  | // We consider that a file was cached if its ID was added in the last minute or | ||||||
|  | // it's name was registered in the last 10 seconds. This ensures that an | ||||||
|  | // identically named file but with different content will be uploaded correctly | ||||||
|  | // (the assumption is that such name collisions will not occur within the given | ||||||
|  | // timeframes). | ||||||
|  | func (b *Bslack) fileCached(file *slack.File) bool { | ||||||
|  | 	if ts, ok := b.cache.Get("file" + file.ID); ok && time.Since(ts.(time.Time)) < time.Minute { | ||||||
|  | 		return true | ||||||
|  | 	} else if ts, ok = b.cache.Get("filename" + file.Name); ok && time.Since(ts.(time.Time)) < 10*time.Second { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
							
								
								
									
										323
									
								
								bridge/slack/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,323 @@ | |||||||
|  | package bslack | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/nlopes/slack" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func (b *Bslack) getUser(id string) *slack.User { | ||||||
|  | 	b.usersMutex.RLock() | ||||||
|  | 	defer b.usersMutex.RUnlock() | ||||||
|  |  | ||||||
|  | 	return b.users[id] | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) getUsername(id string) string { | ||||||
|  | 	if user := b.getUser(id); user != nil { | ||||||
|  | 		if user.Profile.DisplayName != "" { | ||||||
|  | 			return user.Profile.DisplayName | ||||||
|  | 		} | ||||||
|  | 		return user.Name | ||||||
|  | 	} | ||||||
|  | 	b.Log.Warnf("Could not find user with ID '%s'", id) | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) getAvatar(id string) string { | ||||||
|  | 	if user := b.getUser(id); user != nil { | ||||||
|  | 		return user.Profile.Image48 | ||||||
|  | 	} | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) getChannel(channel string) (*slack.Channel, error) { | ||||||
|  | 	if strings.HasPrefix(channel, "ID:") { | ||||||
|  | 		return b.getChannelByID(strings.TrimPrefix(channel, "ID:")) | ||||||
|  | 	} | ||||||
|  | 	return b.getChannelByName(channel) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) getChannelByName(name string) (*slack.Channel, error) { | ||||||
|  | 	b.channelsMutex.RLock() | ||||||
|  | 	defer b.channelsMutex.RUnlock() | ||||||
|  |  | ||||||
|  | 	if channel, ok := b.channelsByName[name]; ok { | ||||||
|  | 		return channel, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, name) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) { | ||||||
|  | 	b.channelsMutex.RLock() | ||||||
|  | 	defer b.channelsMutex.RUnlock() | ||||||
|  |  | ||||||
|  | 	if channel, ok := b.channelsByID[ID]; ok { | ||||||
|  | 		return channel, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, ID) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const minimumRefreshInterval = 10 * time.Second | ||||||
|  |  | ||||||
|  | func (b *Bslack) populateUsers() { | ||||||
|  | 	b.refreshMutex.Lock() | ||||||
|  | 	if time.Now().Before(b.earliestUserRefresh) || b.refreshInProgress { | ||||||
|  | 		b.Log.Debugf("Not refreshing user list as it was done less than %v ago.", | ||||||
|  | 			minimumRefreshInterval) | ||||||
|  | 		b.refreshMutex.Unlock() | ||||||
|  |  | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	b.refreshInProgress = true | ||||||
|  | 	b.refreshMutex.Unlock() | ||||||
|  |  | ||||||
|  | 	newUsers := map[string]*slack.User{} | ||||||
|  | 	pagination := b.sc.GetUsersPaginated(slack.GetUsersOptionLimit(200)) | ||||||
|  | 	for { | ||||||
|  | 		var err error | ||||||
|  | 		pagination, err = pagination.Next(context.Background()) | ||||||
|  | 		if err != nil { | ||||||
|  | 			if pagination.Done(err) { | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if err = b.handleRateLimit(err); err != nil { | ||||||
|  | 				b.Log.Errorf("Could not retrieve users: %#v", err) | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i := range pagination.Users { | ||||||
|  | 			newUsers[pagination.Users[i].ID] = &pagination.Users[i] | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.usersMutex.Lock() | ||||||
|  | 	defer b.usersMutex.Unlock() | ||||||
|  | 	b.users = newUsers | ||||||
|  |  | ||||||
|  | 	b.refreshMutex.Lock() | ||||||
|  | 	defer b.refreshMutex.Unlock() | ||||||
|  | 	b.earliestUserRefresh = time.Now().Add(minimumRefreshInterval) | ||||||
|  | 	b.refreshInProgress = false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) populateChannels() { | ||||||
|  | 	b.refreshMutex.Lock() | ||||||
|  | 	if time.Now().Before(b.earliestChannelRefresh) || b.refreshInProgress { | ||||||
|  | 		b.Log.Debugf("Not refreshing channel list as it was done less than %v seconds ago.", | ||||||
|  | 			minimumRefreshInterval) | ||||||
|  | 		b.refreshMutex.Unlock() | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	b.refreshInProgress = true | ||||||
|  | 	b.refreshMutex.Unlock() | ||||||
|  |  | ||||||
|  | 	newChannelsByID := map[string]*slack.Channel{} | ||||||
|  | 	newChannelsByName := map[string]*slack.Channel{} | ||||||
|  |  | ||||||
|  | 	// We only retrieve public and private channels, not IMs | ||||||
|  | 	// and MPIMs as those do not have a channel name. | ||||||
|  | 	queryParams := &slack.GetConversationsParameters{ | ||||||
|  | 		ExcludeArchived: "true", | ||||||
|  | 		Types:           []string{"public_channel,private_channel"}, | ||||||
|  | 	} | ||||||
|  | 	for { | ||||||
|  | 		channels, nextCursor, err := b.sc.GetConversations(queryParams) | ||||||
|  | 		if err != nil { | ||||||
|  | 			if err = b.handleRateLimit(err); err != nil { | ||||||
|  | 				b.Log.Errorf("Could not retrieve channels: %#v", err) | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i := range channels { | ||||||
|  | 			newChannelsByID[channels[i].ID] = &channels[i] | ||||||
|  | 			newChannelsByName[channels[i].Name] = &channels[i] | ||||||
|  | 		} | ||||||
|  | 		if nextCursor == "" { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		queryParams.Cursor = nextCursor | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.channelsMutex.Lock() | ||||||
|  | 	defer b.channelsMutex.Unlock() | ||||||
|  | 	b.channelsByID = newChannelsByID | ||||||
|  | 	b.channelsByName = newChannelsByName | ||||||
|  |  | ||||||
|  | 	b.refreshMutex.Lock() | ||||||
|  | 	defer b.refreshMutex.Unlock() | ||||||
|  | 	b.earliestChannelRefresh = time.Now().Add(minimumRefreshInterval) | ||||||
|  | 	b.refreshInProgress = false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // populateReceivedMessage shapes the initial Matterbridge message that we will forward to the | ||||||
|  | // router before we apply message-dependent modifications. | ||||||
|  | func (b *Bslack) populateReceivedMessage(ev *slack.MessageEvent) (*config.Message, error) { | ||||||
|  | 	// 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:       ev.Timestamp, | ||||||
|  | 		Extra:    make(map[string][]interface{}), | ||||||
|  | 		ParentID: ev.ThreadTimestamp, | ||||||
|  | 		Protocol: b.Protocol, | ||||||
|  | 	} | ||||||
|  | 	if b.useChannelID { | ||||||
|  | 		rmsg.Channel = "ID:" + channel.ID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Handle 'edit' messages. | ||||||
|  | 	if ev.SubMessage != nil && !b.GetBool(editDisableConfig) { | ||||||
|  | 		rmsg.ID = ev.SubMessage.Timestamp | ||||||
|  | 		if ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { | ||||||
|  | 			b.Log.Debugf("SubMessage %#v", ev.SubMessage) | ||||||
|  | 			rmsg.Text = ev.SubMessage.Text + b.GetString(editSuffixConfig) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err = b.populateMessageWithUserInfo(ev, rmsg); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return rmsg, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) populateMessageWithUserInfo(ev *slack.MessageEvent, rmsg *config.Message) error { | ||||||
|  | 	if ev.SubType == sMessageDeleted || ev.SubType == sFileComment { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// First, deal with bot-originating messages but only do so when not using webhooks: we | ||||||
|  | 	// would not be able to distinguish which bot would be sending them. | ||||||
|  | 	if err := b.populateMessageWithBotInfo(ev, rmsg); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Second, deal with "real" users if we have the necessary information. | ||||||
|  | 	var userID string | ||||||
|  | 	switch { | ||||||
|  | 	case ev.User != "": | ||||||
|  | 		userID = ev.User | ||||||
|  | 	case ev.SubMessage != nil && ev.SubMessage.User != "": | ||||||
|  | 		userID = ev.SubMessage.User | ||||||
|  | 	default: | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	user := b.getUser(userID) | ||||||
|  | 	if user == nil { | ||||||
|  | 		return fmt.Errorf("could not find information for user with id %s", ev.User) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rmsg.UserID = user.ID | ||||||
|  | 	rmsg.Username = user.Name | ||||||
|  | 	if user.Profile.DisplayName != "" { | ||||||
|  | 		rmsg.Username = user.Profile.DisplayName | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config.Message) error { | ||||||
|  | 	if ev.BotID == "" || b.GetString(outgoingWebhookConfig) != "" { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  | 	var bot *slack.Bot | ||||||
|  | 	for { | ||||||
|  | 		bot, err = b.rtm.GetBotInfo(ev.BotID) | ||||||
|  | 		if err == nil { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if err = b.handleRateLimit(err); err != nil { | ||||||
|  | 			b.Log.Errorf("Could not retrieve bot information: %#v", err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if bot.Name != "" && bot.Name != "Slack API Tester" { | ||||||
|  | 		rmsg.Username = bot.Name | ||||||
|  | 		if ev.Username != "" { | ||||||
|  | 			rmsg.Username = ev.Username | ||||||
|  | 		} | ||||||
|  | 		rmsg.UserID = bot.ID | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	mentionRE  = regexp.MustCompile(`<@([a-zA-Z0-9]+)>`) | ||||||
|  | 	channelRE  = regexp.MustCompile(`<#[a-zA-Z0-9]+\|(.+?)>`) | ||||||
|  | 	variableRE = regexp.MustCompile(`<!((?:subteam\^)?[a-zA-Z0-9]+)(?:\|@?(.+?))?>`) | ||||||
|  | 	urlRE      = regexp.MustCompile(`<(.*?)(\|.*?)?>`) | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
|  | func (b *Bslack) replaceMention(text string) string { | ||||||
|  | 	replaceFunc := func(match string) string { | ||||||
|  | 		userID := strings.Trim(match, "@<>") | ||||||
|  | 		if username := b.getUsername(userID); userID != "" { | ||||||
|  | 			return "@" + username | ||||||
|  | 		} | ||||||
|  | 		return match | ||||||
|  | 	} | ||||||
|  | 	return mentionRE.ReplaceAllStringFunc(text, replaceFunc) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||||
|  | func (b *Bslack) replaceChannel(text string) string { | ||||||
|  | 	for _, r := range channelRE.FindAllStringSubmatch(text, -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 { | ||||||
|  | 	for _, r := range variableRE.FindAllStringSubmatch(text, -1) { | ||||||
|  | 		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 { | ||||||
|  | 	for _, r := range urlRE.FindAllStringSubmatch(text, -1) { | ||||||
|  | 		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) handleRateLimit(err error) error { | ||||||
|  | 	rateLimit, ok := err.(*slack.RateLimitedError) | ||||||
|  | 	if !ok { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	b.Log.Infof("Rate-limited by Slack. Sleeping for %v", rateLimit.RetryAfter) | ||||||
|  | 	time.Sleep(rateLimit.RetryAfter) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										74
									
								
								bridge/slack/legacy.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,74 @@ | |||||||
|  | package bslack | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/matterhook" | ||||||
|  | 	"github.com/nlopes/slack" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type BLegacy struct { | ||||||
|  | 	*Bslack | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewLegacy(cfg *bridge.Config) bridge.Bridger { | ||||||
|  | 	return &BLegacy{Bslack: newBridge(cfg)} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *BLegacy) Connect() error { | ||||||
|  | 	b.RLock() | ||||||
|  | 	defer b.RUnlock() | ||||||
|  | 	if b.GetString(incomingWebhookConfig) != "" { | ||||||
|  | 		switch { | ||||||
|  | 		case b.GetString(outgoingWebhookConfig) != "": | ||||||
|  | 			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||||
|  | 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||||
|  | 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||||
|  | 			}) | ||||||
|  | 		case b.GetString(tokenConfig) != "": | ||||||
|  | 			b.Log.Info("Connecting using token (sending)") | ||||||
|  | 			b.sc = slack.New(b.GetString(tokenConfig)) | ||||||
|  | 			b.rtm = b.sc.NewRTM() | ||||||
|  | 			go b.rtm.ManageConnection() | ||||||
|  | 			b.Log.Info("Connecting using webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||||
|  | 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||||
|  | 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||||
|  | 			}) | ||||||
|  | 		default: | ||||||
|  | 			b.Log.Info("Connecting using webhookbindaddress (receiving)") | ||||||
|  | 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||||
|  | 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||||
|  | 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||||
|  | 			}) | ||||||
|  | 		} | ||||||
|  | 		go b.handleSlack() | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	if b.GetString(outgoingWebhookConfig) != "" { | ||||||
|  | 		b.Log.Info("Connecting using webhookurl (sending)") | ||||||
|  | 		b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||||
|  | 			InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||||
|  | 			DisableServer:      true, | ||||||
|  | 		}) | ||||||
|  | 		if b.GetString(tokenConfig) != "" { | ||||||
|  | 			b.Log.Info("Connecting using token (receiving)") | ||||||
|  | 			b.sc = slack.New(b.GetString(tokenConfig)) | ||||||
|  | 			b.rtm = b.sc.NewRTM() | ||||||
|  | 			go b.rtm.ManageConnection() | ||||||
|  | 			go b.handleSlack() | ||||||
|  | 		} | ||||||
|  | 	} else if b.GetString(tokenConfig) != "" { | ||||||
|  | 		b.Log.Info("Connecting using token (sending and receiving)") | ||||||
|  | 		b.sc = slack.New(b.GetString(tokenConfig)) | ||||||
|  | 		b.rtm = b.sc.NewRTM() | ||||||
|  | 		go b.rtm.ManageConnection() | ||||||
|  | 		go b.handleSlack() | ||||||
|  | 	} | ||||||
|  | 	if b.GetString(incomingWebhookConfig) == "" && b.GetString(outgoingWebhookConfig) == "" && b.GetString(tokenConfig) == "" { | ||||||
|  | 		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token configured") | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
| @@ -1,51 +1,110 @@ | |||||||
| package bslack | package bslack | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	"github.com/42wim/matterbridge/matterhook" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/nlopes/slack" |  | ||||||
| 	"html" |  | ||||||
| 	"regexp" |  | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"sync" | ||||||
| 	"time" | 	"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/hashicorp/golang-lru" | ||||||
|  | 	"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 { | type Bslack struct { | ||||||
| 	mh       *matterhook.Client | 	sync.RWMutex | ||||||
| 	sc       *slack.Client | 	*bridge.Config | ||||||
| 	Config   *config.Protocol |  | ||||||
| 	rtm      *slack.RTM | 	mh  *matterhook.Client | ||||||
| 	Plus     bool | 	sc  *slack.Client | ||||||
| 	Remote   chan config.Message | 	rtm *slack.RTM | ||||||
| 	Users    []slack.User | 	si  *slack.Info | ||||||
| 	Account  string |  | ||||||
| 	si       *slack.Info | 	cache        *lru.Cache | ||||||
| 	channels []slack.Channel | 	uuid         string | ||||||
|  | 	useChannelID bool | ||||||
|  |  | ||||||
|  | 	users      map[string]*slack.User | ||||||
|  | 	usersMutex sync.RWMutex | ||||||
|  |  | ||||||
|  | 	channelsByID   map[string]*slack.Channel | ||||||
|  | 	channelsByName map[string]*slack.Channel | ||||||
|  | 	channelsMutex  sync.RWMutex | ||||||
|  |  | ||||||
|  | 	refreshInProgress      bool | ||||||
|  | 	earliestChannelRefresh time.Time | ||||||
|  | 	earliestUserRefresh    time.Time | ||||||
|  | 	refreshMutex           sync.Mutex | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | const ( | ||||||
| var protocol = "slack" | 	sChannelJoin     = "channel_join" | ||||||
|  | 	sChannelLeave    = "channel_leave" | ||||||
|  | 	sChannelJoined   = "channel_joined" | ||||||
|  | 	sMemberJoined    = "member_joined_channel" | ||||||
|  | 	sMessageChanged  = "message_changed" | ||||||
|  | 	sMessageDeleted  = "message_deleted" | ||||||
|  | 	sSlackAttachment = "slack_attachment" | ||||||
|  | 	sPinnedItem      = "pinned_item" | ||||||
|  | 	sUnpinnedItem    = "unpinned_item" | ||||||
|  | 	sChannelTopic    = "channel_topic" | ||||||
|  | 	sChannelPurpose  = "channel_purpose" | ||||||
|  | 	sFileComment     = "file_comment" | ||||||
|  | 	sMeMessage       = "me_message" | ||||||
|  | 	sUserTyping      = "user_typing" | ||||||
|  | 	sLatencyReport   = "latency_report" | ||||||
|  | 	sSystemUser      = "system" | ||||||
|  | 	sSlackBotUser    = "slackbot" | ||||||
|  |  | ||||||
| func init() { | 	tokenConfig           = "Token" | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) | 	incomingWebhookConfig = "WebhookBindAddress" | ||||||
|  | 	outgoingWebhookConfig = "WebhookURL" | ||||||
|  | 	skipTLSConfig         = "SkipTLSVerify" | ||||||
|  | 	useNickPrefixConfig   = "PrefixMessagesWithNick" | ||||||
|  | 	editDisableConfig     = "EditDisable" | ||||||
|  | 	editSuffixConfig      = "EditSuffix" | ||||||
|  | 	iconURLConfig         = "iconurl" | ||||||
|  | 	noSendJoinConfig      = "nosendjoinpart" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
|  | 	// Print a deprecation warning for legacy non-bot tokens (#527). | ||||||
|  | 	token := cfg.GetString(tokenConfig) | ||||||
|  | 	if token != "" && !strings.HasPrefix(token, "xoxb") { | ||||||
|  | 		cfg.Log.Error("Non-bot token detected. It is STRONGLY recommended to use a proper bot-token instead.") | ||||||
|  | 		cfg.Log.Error("Legacy tokens may be deprecated by Slack at short notice. See the Matterbridge GitHub wiki for a migration guide.") | ||||||
|  | 		cfg.Log.Error("See https://github.com/42wim/matterbridge/wiki/Slack-bot-setup") | ||||||
|  | 		cfg.Log.Error("") | ||||||
|  | 		cfg.Log.Error("To continue using a legacy token please move your configuration to a \"slack-legacy\" bridge instead.") | ||||||
|  | 		cfg.Log.Error("See https://github.com/42wim/matterbridge/wiki/Section-Slack-(basic)#legacy-configuration)") | ||||||
|  | 		cfg.Log.Error("Delaying start of bridge by 30 seconds. Future Matterbridge release will fail here unless you use a \"slack-legacy\" bridge.") | ||||||
|  | 		time.Sleep(30 * time.Second) | ||||||
|  | 		return NewLegacy(cfg) | ||||||
|  | 	} | ||||||
|  | 	return newBridge(cfg) | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bslack { | func newBridge(cfg *bridge.Config) *Bslack { | ||||||
| 	b := &Bslack{} | 	newCache, err := lru.New(5000) | ||||||
| 	b.Config = &cfg | 	if err != nil { | ||||||
| 	b.Remote = c | 		cfg.Log.Fatalf("Could not create LRU cache for Slack bridge: %v", err) | ||||||
| 	b.Account = account | 	} | ||||||
|  | 	b := &Bslack{ | ||||||
|  | 		Config:                 cfg, | ||||||
|  | 		uuid:                   xid.New().String(), | ||||||
|  | 		cache:                  newCache, | ||||||
|  | 		users:                  map[string]*slack.User{}, | ||||||
|  | 		channelsByID:           map[string]*slack.Channel{}, | ||||||
|  | 		channelsByName:         map[string]*slack.Channel{}, | ||||||
|  | 		earliestChannelRefresh: time.Now(), | ||||||
|  | 		earliestUserRefresh:    time.Now(), | ||||||
|  | 	} | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -54,315 +113,359 @@ func (b *Bslack) Command(cmd string) string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Connect() error { | func (b *Bslack) Connect() error { | ||||||
| 	if b.Config.WebhookBindAddress != "" { | 	b.RLock() | ||||||
| 		if b.Config.WebhookURL != "" { | 	defer b.RUnlock() | ||||||
| 			flog.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") |  | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, | 	if b.GetString(incomingWebhookConfig) == "" && b.GetString(outgoingWebhookConfig) == "" && b.GetString(tokenConfig) == "" { | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 		return errors.New("no connection method found: WebhookBindAddress, WebhookURL or Token need to be configured") | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) |  | ||||||
| 		} else if b.Config.Token != "" { |  | ||||||
| 			flog.Info("Connecting using token (sending)") |  | ||||||
| 			b.sc = slack.New(b.Config.Token) |  | ||||||
| 			b.rtm = b.sc.NewRTM() |  | ||||||
| 			go b.rtm.ManageConnection() |  | ||||||
| 			flog.Info("Connecting using webhookbindaddress (receiving)") |  | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, |  | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, |  | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) |  | ||||||
| 		} else { |  | ||||||
| 			flog.Info("Connecting using webhookbindaddress (receiving)") |  | ||||||
| 			b.mh = matterhook.New(b.Config.WebhookURL, |  | ||||||
| 				matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, |  | ||||||
| 					BindAddress: b.Config.WebhookBindAddress}) |  | ||||||
| 		} |  | ||||||
| 		go b.handleSlack() |  | ||||||
| 		return nil |  | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookURL != "" { |  | ||||||
| 		flog.Info("Connecting using webhookurl (sending)") | 	// If we have a token we use the Slack websocket-based RTM for both sending and receiving. | ||||||
| 		b.mh = matterhook.New(b.Config.WebhookURL, | 	if token := b.GetString(tokenConfig); token != "" { | ||||||
| 			matterhook.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, | 		b.Log.Info("Connecting using token") | ||||||
| 				DisableServer: true}) | 		b.sc = slack.New(token) | ||||||
| 		if b.Config.Token != "" { |  | ||||||
| 			flog.Info("Connecting using token (receiving)") |  | ||||||
| 			b.sc = slack.New(b.Config.Token) |  | ||||||
| 			b.rtm = b.sc.NewRTM() |  | ||||||
| 			go b.rtm.ManageConnection() |  | ||||||
| 			go b.handleSlack() |  | ||||||
| 		} |  | ||||||
| 	} else if b.Config.Token != "" { |  | ||||||
| 		flog.Info("Connecting using token (sending and receiving)") |  | ||||||
| 		b.sc = slack.New(b.Config.Token) |  | ||||||
| 		b.rtm = b.sc.NewRTM() | 		b.rtm = b.sc.NewRTM() | ||||||
| 		go b.rtm.ManageConnection() | 		go b.rtm.ManageConnection() | ||||||
| 		go b.handleSlack() | 		go b.handleSlack() | ||||||
|  | 		return nil | ||||||
| 	} | 	} | ||||||
| 	if b.Config.WebhookBindAddress == "" && b.Config.WebhookURL == "" && b.Config.Token == "" { |  | ||||||
| 		return errors.New("No connection method found. See that you have WebhookBindAddress, WebhookURL or Token configured.") | 	// In absence of a token we fall back to incoming and outgoing Webhooks. | ||||||
|  | 	b.mh = matterhook.New( | ||||||
|  | 		"", | ||||||
|  | 		matterhook.Config{ | ||||||
|  | 			InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||||
|  | 			DisableServer:      true, | ||||||
|  | 		}, | ||||||
|  | 	) | ||||||
|  | 	if b.GetString(outgoingWebhookConfig) != "" { | ||||||
|  | 		b.Log.Info("Using specified webhook for outgoing messages.") | ||||||
|  | 		b.mh.Url = b.GetString(outgoingWebhookConfig) | ||||||
|  | 	} | ||||||
|  | 	if b.GetString(incomingWebhookConfig) != "" { | ||||||
|  | 		b.Log.Info("Setting up local webhook for incoming messages.") | ||||||
|  | 		b.mh.BindAddress = b.GetString(incomingWebhookConfig) | ||||||
|  | 		b.mh.DisableServer = false | ||||||
|  | 		go b.handleSlack() | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) Disconnect() error { | func (b *Bslack) Disconnect() error { | ||||||
| 	return nil | 	return b.rtm.Disconnect() | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) JoinChannel(channel string) error { | // JoinChannel only acts as a verification method that checks whether Matterbridge's | ||||||
| 	// we can only join channels using the API | // Slack integration is already member of the channel. This is because Slack does not | ||||||
| 	if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" { | // allow apps or bots to join channels themselves and they need to be invited | ||||||
| 		if strings.HasPrefix(b.Config.Token, "xoxb") { | // manually by a user. | ||||||
| 			// TODO check if bot has already joined channel | func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 			return nil | 	// We can only join a channel through the Slack API. | ||||||
| 		} | 	if b.sc == nil { | ||||||
| 		_, err := b.sc.JoinChannel(channel) |  | ||||||
| 		if err != nil { |  | ||||||
| 			if err.Error() != "name_taken" { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bslack) Send(msg config.Message) error { |  | ||||||
| 	flog.Debugf("Receiving %#v", msg) |  | ||||||
| 	if msg.Event == config.EVENT_USER_ACTION { |  | ||||||
| 		msg.Text = "_" + msg.Text + "_" |  | ||||||
| 	} |  | ||||||
| 	nick := msg.Username |  | ||||||
| 	message := msg.Text |  | ||||||
| 	channel := msg.Channel |  | ||||||
| 	if b.Config.PrefixMessagesWithNick { |  | ||||||
| 		message = nick + " " + message |  | ||||||
| 	} |  | ||||||
| 	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 | 		return nil | ||||||
| 	} | 	} | ||||||
| 	schannel, err := b.getChannelByName(channel) |  | ||||||
|  | 	b.populateChannels() | ||||||
|  |  | ||||||
|  | 	channelInfo, err := b.getChannel(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return fmt.Errorf("could not join channel: %#v", err) | ||||||
| 	} | 	} | ||||||
| 	np := slack.NewPostMessageParameters() |  | ||||||
| 	if b.Config.PrefixMessagesWithNick { |  | ||||||
| 		np.AsUser = true |  | ||||||
| 	} |  | ||||||
| 	np.Username = nick |  | ||||||
| 	np.IconURL = config.GetIconURL(&msg, b.Config) |  | ||||||
| 	if msg.Avatar != "" { |  | ||||||
| 		np.IconURL = msg.Avatar |  | ||||||
| 	} |  | ||||||
| 	np.Attachments = append(np.Attachments, slack.Attachment{CallbackID: "matterbridge"}) |  | ||||||
| 	b.sc.PostMessage(schannel.ID, message, np) |  | ||||||
|  |  | ||||||
| 	/* | 	if strings.HasPrefix(channel.Name, "ID:") { | ||||||
| 	   newmsg := b.rtm.NewOutgoingMessage(message, schannel.ID) | 		b.useChannelID = true | ||||||
| 	   b.rtm.SendMessage(newmsg) | 		channel.Name = channelInfo.Name | ||||||
| 	*/ | 	} | ||||||
|  |  | ||||||
|  | 	if !channelInfo.IsMember { | ||||||
|  | 		return fmt.Errorf("slack integration that matterbridge is using is not member of channel '%s', please add it manually", channelInfo.Name) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) getAvatar(user string) string { | func (b *Bslack) Reload(cfg *bridge.Config) (string, error) { | ||||||
| 	var avatar string | 	return "", nil | ||||||
| 	if b.Users != nil { | } | ||||||
| 		for _, u := range b.Users { |  | ||||||
| 			if user == u.Name { | func (b *Bslack) Send(msg config.Message) (string, error) { | ||||||
| 				return u.Profile.Image48 | 	// Too noisy to log like other events | ||||||
|  | 	if msg.Event != config.EventUserTyping { | ||||||
|  | 		b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Make a action /me of the message | ||||||
|  | 	if msg.Event == config.EventUserAction { | ||||||
|  | 		msg.Text = "_" + msg.Text + "_" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Use webhook to send the message | ||||||
|  | 	if b.GetString(outgoingWebhookConfig) != "" { | ||||||
|  | 		return b.sendWebhook(msg) | ||||||
|  | 	} | ||||||
|  | 	return b.sendRTM(msg) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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(useNickPrefixConfig) { | ||||||
|  | 		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) { | ||||||
|  | 			rmsg := rmsg // scopelint | ||||||
|  | 			iconURL := config.GetIconURL(&rmsg, b.GetString(iconURLConfig)) | ||||||
|  | 			matterMessage := matterhook.OMessage{ | ||||||
|  | 				IconURL:  iconURL, | ||||||
|  | 				Channel:  msg.Channel, | ||||||
|  | 				UserName: rmsg.Username, | ||||||
|  | 				Text:     rmsg.Text, | ||||||
|  | 			} | ||||||
|  | 			if err := b.mh.Send(matterMessage); err != nil { | ||||||
|  | 				b.Log.Errorf("Failed to send message: %v", err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Webhook doesn't support file uploads, so we add the URL manually. | ||||||
|  | 		for _, f := range msg.Extra["file"] { | ||||||
|  | 			fi, ok := f.(config.FileInfo) | ||||||
|  | 			if !ok { | ||||||
|  | 				b.Log.Errorf("Received a file with unexpected content: %#v", f) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if fi.URL != "" { | ||||||
|  | 				msg.Text += " " + fi.URL | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return avatar |  | ||||||
|  | 	// If we have native slack_attachments add them. | ||||||
|  | 	var attachs []slack.Attachment | ||||||
|  | 	for _, attach := range msg.Extra[sSlackAttachment] { | ||||||
|  | 		attachs = append(attachs, attach.([]slack.Attachment)...) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	iconURL := config.GetIconURL(&msg, b.GetString(iconURLConfig)) | ||||||
|  | 	matterMessage := matterhook.OMessage{ | ||||||
|  | 		IconURL:     iconURL, | ||||||
|  | 		Attachments: attachs, | ||||||
|  | 		Channel:     msg.Channel, | ||||||
|  | 		UserName:    msg.Username, | ||||||
|  | 		Text:        msg.Text, | ||||||
|  | 	} | ||||||
|  | 	if msg.Avatar != "" { | ||||||
|  | 		matterMessage.IconURL = msg.Avatar | ||||||
|  | 	} | ||||||
|  | 	if err := b.mh.Send(matterMessage); err != nil { | ||||||
|  | 		b.Log.Errorf("Failed to send message via webhook: %#v", err) | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) getChannelByName(name string) (*slack.Channel, error) { | func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||||
| 	if b.channels == nil { | 	channelInfo, err := b.getChannel(msg.Channel) | ||||||
| 		return nil, fmt.Errorf("%s: channel %s not found (no channels found)", b.Account, name) | 	if err != nil { | ||||||
|  | 		return "", fmt.Errorf("could not send message: %v", err) | ||||||
| 	} | 	} | ||||||
| 	for _, channel := range b.channels { | 	if msg.Event == config.EventUserTyping { | ||||||
| 		if channel.Name == name { | 		if b.GetBool("ShowUserTyping") { | ||||||
| 			return &channel, nil | 			b.rtm.SendMessage(b.rtm.NewTypingMessage(channelInfo.ID)) | ||||||
| 		} | 		} | ||||||
|  | 		return "", nil | ||||||
| 	} | 	} | ||||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, name) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) { | 	// Handle message deletions. | ||||||
| 	if b.channels == nil { | 	var handled bool | ||||||
| 		return nil, fmt.Errorf("%s: channel %s not found (no channels found)", b.Account, ID) | 	if handled, err = b.deleteMessage(&msg, channelInfo); handled { | ||||||
|  | 		return msg.ID, err | ||||||
| 	} | 	} | ||||||
| 	for _, channel := range b.channels { |  | ||||||
| 		if channel.ID == ID { |  | ||||||
| 			return &channel, nil |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, ID) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bslack) handleSlack() { | 	// Prepend nickname if configured. | ||||||
| 	mchan := make(chan *MMMessage) | 	if b.GetBool(useNickPrefixConfig) { | ||||||
| 	if b.Config.WebhookBindAddress != "" { | 		msg.Text = msg.Username + msg.Text | ||||||
| 		flog.Debugf("Choosing webhooks based receiving") |  | ||||||
| 		go b.handleMatterHook(mchan) |  | ||||||
| 	} else { |  | ||||||
| 		flog.Debugf("Choosing token based receiving") |  | ||||||
| 		go b.handleSlackClient(mchan) |  | ||||||
| 	} | 	} | ||||||
| 	time.Sleep(time.Second) |  | ||||||
| 	flog.Debug("Start listening for Slack messages") | 	// Handle message edits. | ||||||
| 	for message := range mchan { | 	if handled, err = b.editMessage(&msg, channelInfo); handled { | ||||||
| 		// do not send messages from ourself | 		return msg.ID, err | ||||||
| 		if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" && message.Username == b.si.User.Name { | 	} | ||||||
| 			continue |  | ||||||
| 		} | 	messageParameters := b.prepareMessageParameters(&msg) | ||||||
| 		if message.Text == "" || message.Username == "" { |  | ||||||
| 			continue | 	// Upload a file if it exists. | ||||||
| 		} | 	if msg.Extra != nil { | ||||||
| 		texts := strings.Split(message.Text, "\n") | 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||||
| 		for _, text := range texts { | 			_, _, err = b.rtm.PostMessage(channelInfo.ID, rmsg.Username+rmsg.Text, *messageParameters) | ||||||
| 			text = b.replaceURL(text) | 			if err != nil { | ||||||
| 			text = html.UnescapeString(text) | 				b.Log.Error(err) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", message.Username, b.Account) |  | ||||||
| 			msg := config.Message{Text: text, Username: message.Username, Channel: message.Channel, Account: b.Account, Avatar: b.getAvatar(message.Username), UserID: message.UserID} |  | ||||||
| 			if message.Raw.SubType == "me_message" { |  | ||||||
| 				msg.Event = config.EVENT_USER_ACTION |  | ||||||
| 			} | 			} | ||||||
| 			b.Remote <- msg |  | ||||||
| 		} | 		} | ||||||
|  | 		// Upload files if necessary (from Slack, Telegram or Mattermost). | ||||||
|  | 		b.uploadFile(&msg, channelInfo.ID) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Post message. | ||||||
|  | 	return b.postMessage(&msg, messageParameters, channelInfo) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) handleSlackClient(mchan chan *MMMessage) { | func (b *Bslack) deleteMessage(msg *config.Message, channelInfo *slack.Channel) (bool, error) { | ||||||
| 	count := 0 | 	if msg.Event != config.EventMsgDelete { | ||||||
| 	for msg := range b.rtm.IncomingEvents { | 		return false, nil | ||||||
| 		switch ev := msg.Data.(type) { | 	} | ||||||
| 		case *slack.MessageEvent: |  | ||||||
| 			// ignore first message | 	// Some protocols echo deletes, but with an empty ID. | ||||||
| 			if count > 0 { | 	if msg.ID == "" { | ||||||
| 				flog.Debugf("Receiving from slackclient %#v", ev) | 		return true, nil | ||||||
| 				if len(ev.Attachments) > 0 { |  | ||||||
| 					// skip messages we made ourselves |  | ||||||
| 					if ev.Attachments[0].CallbackID == "matterbridge" { |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				if !b.Config.EditDisable && ev.SubMessage != nil && ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { |  | ||||||
| 					flog.Debugf("SubMessage %#v", ev.SubMessage) |  | ||||||
| 					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 |  | ||||||
| 				} |  | ||||||
| 				m := &MMMessage{} |  | ||||||
| 				if ev.BotID == "" { |  | ||||||
| 					user, err := b.rtm.GetUserInfo(ev.User) |  | ||||||
| 					if err != nil { |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 					m.UserID = user.ID |  | ||||||
| 					m.Username = user.Name |  | ||||||
| 				} |  | ||||||
| 				m.Channel = channel.Name |  | ||||||
| 				m.Text = ev.Text |  | ||||||
| 				if m.Text == "" { |  | ||||||
| 					for _, attach := range ev.Attachments { |  | ||||||
| 						if attach.Text != "" { |  | ||||||
| 							m.Text = attach.Text |  | ||||||
| 						} else { |  | ||||||
| 							m.Text = attach.Fallback |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				m.Raw = ev |  | ||||||
| 				m.Text = b.replaceMention(m.Text) |  | ||||||
| 				// when using webhookURL we can't check if it's our webhook or not for now |  | ||||||
| 				if ev.BotID != "" && b.Config.WebhookURL == "" { |  | ||||||
| 					bot, err := b.rtm.GetBotInfo(ev.BotID) |  | ||||||
| 					if err != nil { |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 					if bot.Name != "" { |  | ||||||
| 						m.Username = bot.Name |  | ||||||
| 						m.UserID = bot.ID |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				mchan <- m |  | ||||||
| 			} |  | ||||||
| 			count++ |  | ||||||
| 		case *slack.OutgoingErrorEvent: |  | ||||||
| 			flog.Debugf("%#v", ev.Error()) |  | ||||||
| 		case *slack.ChannelJoinedEvent: |  | ||||||
| 			b.Users, _ = b.sc.GetUsers() |  | ||||||
| 		case *slack.ConnectedEvent: |  | ||||||
| 			b.channels = ev.Info.Channels |  | ||||||
| 			b.si = ev.Info |  | ||||||
| 			b.Users, _ = b.sc.GetUsers() |  | ||||||
| 			// add private channels |  | ||||||
| 			groups, _ := b.sc.GetGroups(true) |  | ||||||
| 			for _, g := range groups { |  | ||||||
| 				channel := new(slack.Channel) |  | ||||||
| 				channel.ID = g.ID |  | ||||||
| 				channel.Name = g.Name |  | ||||||
| 				b.channels = append(b.channels, *channel) |  | ||||||
| 			} |  | ||||||
| 		case *slack.InvalidAuthEvent: |  | ||||||
| 			flog.Fatalf("Invalid Token %#v", ev) |  | ||||||
| 		default: |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bslack) handleMatterHook(mchan chan *MMMessage) { |  | ||||||
| 	for { | 	for { | ||||||
| 		message := b.mh.Receive() | 		_, _, err := b.rtm.DeleteMessage(channelInfo.ID, msg.ID) | ||||||
| 		flog.Debugf("receiving from matterhook (slack) %#v", message) | 		if err == nil { | ||||||
| 		m := &MMMessage{} | 			return true, nil | ||||||
| 		m.Username = message.UserName | 		} | ||||||
| 		m.Text = message.Text |  | ||||||
| 		m.Text = b.replaceMention(m.Text) | 		if err = b.handleRateLimit(err); err != nil { | ||||||
| 		m.Channel = message.ChannelName | 			b.Log.Errorf("Failed to delete user message from Slack: %#v", err) | ||||||
| 		if m.Username == "slackbot" { | 			return true, err | ||||||
| 			continue |  | ||||||
| 		} | 		} | ||||||
| 		mchan <- m |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) userName(id string) string { | func (b *Bslack) editMessage(msg *config.Message, channelInfo *slack.Channel) (bool, error) { | ||||||
| 	for _, u := range b.Users { | 	if msg.ID == "" { | ||||||
| 		if u.ID == id { | 		return false, nil | ||||||
| 			return u.Name | 	} | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, msg.Text) | ||||||
|  | 		if err == nil { | ||||||
|  | 			return true, nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if err = b.handleRateLimit(err); err != nil { | ||||||
|  | 			b.Log.Errorf("Failed to edit user message on Slack: %#v", err) | ||||||
|  | 			return true, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) postMessage(msg *config.Message, messageParameters *slack.PostMessageParameters, channelInfo *slack.Channel) (string, error) { | ||||||
|  | 	// don't post empty messages | ||||||
|  | 	if msg.Text == "" { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
|  | 	for { | ||||||
|  | 		_, id, err := b.rtm.PostMessage(channelInfo.ID, msg.Text, *messageParameters) | ||||||
|  | 		if err == nil { | ||||||
|  | 			return id, nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if err = b.handleRateLimit(err); err != nil { | ||||||
|  | 			b.Log.Errorf("Failed to sent user message to Slack: %#v", err) | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // uploadFile handles native upload of files | ||||||
|  | func (b *Bslack) uploadFile(msg *config.Message, channelID string) { | ||||||
|  | 	for _, f := range msg.Extra["file"] { | ||||||
|  | 		fi, ok := f.(config.FileInfo) | ||||||
|  | 		if !ok { | ||||||
|  | 			b.Log.Errorf("Received a file with unexpected content: %#v", f) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		if msg.Text == fi.Comment { | ||||||
|  | 			msg.Text = "" | ||||||
|  | 		} | ||||||
|  | 		// Because the result of the UploadFile is slower than the MessageEvent from slack | ||||||
|  | 		// we can't match on the file ID yet, so we have to match on the filename too. | ||||||
|  | 		ts := time.Now() | ||||||
|  | 		b.Log.Debugf("Adding file %s to cache at %s with timestamp", fi.Name, ts.String()) | ||||||
|  | 		b.cache.Add("filename"+fi.Name, ts) | ||||||
|  | 		initialComment := fmt.Sprintf("File from %s", msg.Username) | ||||||
|  | 		if fi.Comment != "" { | ||||||
|  | 			initialComment += fmt.Sprintf("with comment: %s", fi.Comment) | ||||||
|  | 		} | ||||||
|  | 		res, err := b.sc.UploadFile(slack.FileUploadParameters{ | ||||||
|  | 			Reader:          bytes.NewReader(*fi.Data), | ||||||
|  | 			Filename:        fi.Name, | ||||||
|  | 			Channels:        []string{channelID}, | ||||||
|  | 			InitialComment:  initialComment, | ||||||
|  | 			ThreadTimestamp: msg.ParentID, | ||||||
|  | 		}) | ||||||
|  | 		if err != nil { | ||||||
|  | 			b.Log.Errorf("uploadfile %#v", err) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		if res.ID != "" { | ||||||
|  | 			b.Log.Debugf("Adding file ID %s to cache with timestamp %s", res.ID, ts.String()) | ||||||
|  | 			b.cache.Add("file"+res.ID, ts) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) prepareMessageParameters(msg *config.Message) *slack.PostMessageParameters { | ||||||
|  | 	params := slack.NewPostMessageParameters() | ||||||
|  | 	if b.GetBool(useNickPrefixConfig) { | ||||||
|  | 		params.AsUser = true | ||||||
|  | 	} | ||||||
|  | 	params.Username = msg.Username | ||||||
|  | 	params.LinkNames = 1 // replace mentions | ||||||
|  | 	params.IconURL = config.GetIconURL(msg, b.GetString(iconURLConfig)) | ||||||
|  | 	params.ThreadTimestamp = msg.ParentID | ||||||
|  | 	if msg.Avatar != "" { | ||||||
|  | 		params.IconURL = msg.Avatar | ||||||
|  | 	} | ||||||
|  | 	// add a callback ID so we can see we created it | ||||||
|  | 	params.Attachments = append(params.Attachments, slack.Attachment{CallbackID: "matterbridge_" + b.uuid}) | ||||||
|  | 	// add file attachments | ||||||
|  | 	params.Attachments = append(params.Attachments, b.createAttach(msg.Extra)...) | ||||||
|  | 	// add slack attachments (from another slack bridge) | ||||||
|  | 	if msg.Extra != nil { | ||||||
|  | 		for _, attach := range msg.Extra[sSlackAttachment] { | ||||||
|  | 			params.Attachments = append(params.Attachments, attach.([]slack.Attachment)...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return ¶ms | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Bslack) createAttach(extra map[string][]interface{}) []slack.Attachment { | ||||||
|  | 	var attachements []slack.Attachment | ||||||
|  | 	for _, v := range extra["attachments"] { | ||||||
|  | 		entry := v.(map[string]interface{}) | ||||||
|  | 		s := slack.Attachment{ | ||||||
|  | 			Fallback:   extractStringField(entry, "fallback"), | ||||||
|  | 			Color:      extractStringField(entry, "color"), | ||||||
|  | 			Pretext:    extractStringField(entry, "pretext"), | ||||||
|  | 			AuthorName: extractStringField(entry, "author_name"), | ||||||
|  | 			AuthorLink: extractStringField(entry, "author_link"), | ||||||
|  | 			AuthorIcon: extractStringField(entry, "author_icon"), | ||||||
|  | 			Title:      extractStringField(entry, "title"), | ||||||
|  | 			TitleLink:  extractStringField(entry, "title_link"), | ||||||
|  | 			Text:       extractStringField(entry, "text"), | ||||||
|  | 			ImageURL:   extractStringField(entry, "image_url"), | ||||||
|  | 			ThumbURL:   extractStringField(entry, "thumb_url"), | ||||||
|  | 			Footer:     extractStringField(entry, "footer"), | ||||||
|  | 			FooterIcon: extractStringField(entry, "footer_icon"), | ||||||
|  | 		} | ||||||
|  | 		attachements = append(attachements, s) | ||||||
|  | 	} | ||||||
|  | 	return attachements | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func extractStringField(data map[string]interface{}, field string) string { | ||||||
|  | 	if rawValue, found := data[field]; found { | ||||||
|  | 		if value, ok := rawValue.(string); ok { | ||||||
|  | 			return value | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bslack) replaceMention(text string) string { |  | ||||||
| 	results := regexp.MustCompile(`<@([a-zA-z0-9]+)>`).FindAllStringSubmatch(text, -1) |  | ||||||
| 	for _, r := range results { |  | ||||||
| 		text = strings.Replace(text, "<@"+r[1]+">", "@"+b.userName(r[1]), -1) |  | ||||||
|  |  | ||||||
| 	} |  | ||||||
| 	return text |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Bslack) replaceURL(text string) string { |  | ||||||
| 	results := regexp.MustCompile(`<(.*?)\|.*?>`).FindAllStringSubmatch(text, -1) |  | ||||||
| 	for _, r := range results { |  | ||||||
| 		text = strings.Replace(text, r[0], r[1], -1) |  | ||||||
| 	} |  | ||||||
| 	return text |  | ||||||
| } |  | ||||||
|   | |||||||
							
								
								
									
										141
									
								
								bridge/sshchat/sshchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,141 @@ | |||||||
|  | package bsshchat | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bufio" | ||||||
|  | 	"io" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
|  | 	"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" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | 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.EventMsgDelete { | ||||||
|  | 		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() { | ||||||
|  | 			// ignore messages from ourselves | ||||||
|  | 			if !strings.Contains(b.r.Text(), "\033[K") { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			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,52 +2,41 @@ package bsteam | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	"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" | 	"strconv" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"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" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bsteam struct { | type Bsteam struct { | ||||||
| 	c         *steam.Client | 	c         *steam.Client | ||||||
| 	connected chan struct{} | 	connected chan struct{} | ||||||
| 	Config    *config.Protocol |  | ||||||
| 	Remote    chan config.Message |  | ||||||
| 	Account   string |  | ||||||
| 	userMap   map[steamid.SteamId]string | 	userMap   map[steamid.SteamId]string | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
|  | 	*bridge.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "steam" | 	b := &Bsteam{Config: cfg} | ||||||
|  |  | ||||||
| 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 |  | ||||||
| 	b.userMap = make(map[steamid.SteamId]string) | 	b.userMap = make(map[steamid.SteamId]string) | ||||||
| 	b.connected = make(chan struct{}) | 	b.connected = make(chan struct{}) | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) Connect() error { | func (b *Bsteam) Connect() error { | ||||||
| 	flog.Info("Connecting") | 	b.Log.Info("Connecting") | ||||||
| 	b.c = steam.NewClient() | 	b.c = steam.NewClient() | ||||||
| 	go b.handleEvents() | 	go b.handleEvents() | ||||||
| 	go b.c.Connect() | 	go b.c.Connect() | ||||||
| 	select { | 	select { | ||||||
| 	case <-b.connected: | 	case <-b.connected: | ||||||
| 		flog.Info("Connection succeeded") | 		b.Log.Info("Connection succeeded") | ||||||
| 	case <-time.After(time.Second * 30): | 	case <-time.After(time.Second * 30): | ||||||
| 		return fmt.Errorf("connection timed out") | 		return fmt.Errorf("connection timed out") | ||||||
| 	} | 	} | ||||||
| @@ -60,8 +49,8 @@ func (b *Bsteam) Disconnect() error { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) JoinChannel(channel string) error { | func (b *Bsteam) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	id, err := steamid.NewId(channel) | 	id, err := steamid.NewId(channel.Name) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -69,13 +58,41 @@ func (b *Bsteam) JoinChannel(channel string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) Send(msg config.Message) error { | func (b *Bsteam) Send(msg config.Message) (string, error) { | ||||||
|  | 	// ignore delete messages | ||||||
|  | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		return "", nil | ||||||
|  | 	} | ||||||
| 	id, err := steamid.NewId(msg.Channel) | 	id, err := steamid.NewId(msg.Channel) | ||||||
| 	if err != nil { | 	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) | 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||||
| 	return nil | 	return "", nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bsteam) getNick(id steamid.SteamId) string { | func (b *Bsteam) getNick(id steamid.SteamId) string { | ||||||
| @@ -89,24 +106,29 @@ func (b *Bsteam) getNick(id steamid.SteamId) string { | |||||||
|  |  | ||||||
| func (b *Bsteam) handleEvents() { | func (b *Bsteam) handleEvents() { | ||||||
| 	myLoginInfo := new(steam.LogOnDetails) | 	myLoginInfo := new(steam.LogOnDetails) | ||||||
| 	myLoginInfo.Username = b.Config.Login | 	myLoginInfo.Username = b.GetString("Login") | ||||||
| 	myLoginInfo.Password = b.Config.Password | 	myLoginInfo.Password = b.GetString("Password") | ||||||
| 	myLoginInfo.AuthCode = b.Config.AuthCode | 	myLoginInfo.AuthCode = b.GetString("AuthCode") | ||||||
| 	// Attempt to read existing auth hash to avoid steam guard. | 	// Attempt to read existing auth hash to avoid steam guard. | ||||||
| 	// Maybe works | 	// Maybe works | ||||||
| 	//myLoginInfo.SentryFileHash, _ = ioutil.ReadFile("sentry") | 	//myLoginInfo.SentryFileHash, _ = ioutil.ReadFile("sentry") | ||||||
| 	for event := range b.c.Events() { | 	for event := range b.c.Events() { | ||||||
| 		//flog.Info(event) | 		//b.Log.Info(event) | ||||||
| 		switch e := event.(type) { | 		switch e := event.(type) { | ||||||
| 		case *steam.ChatMsgEvent: | 		case *steam.ChatMsgEvent: | ||||||
| 			flog.Debugf("Receiving ChatMsgEvent: %#v", e) | 			b.Log.Debugf("Receiving ChatMsgEvent: %#v", e) | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | 			b.Log.Debugf("<= Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | ||||||
| 			// for some reason we have to remove 0x18000000000000 | 			var channel int64 | ||||||
| 			channel := int64(e.ChatRoomId) - 0x18000000000000 | 			if e.ChatRoomId == 0 { | ||||||
|  | 				channel = int64(e.ChatterId) | ||||||
|  | 			} else { | ||||||
|  | 				// for some reason we have to remove 0x18000000000000 | ||||||
|  | 				channel = int64(e.ChatRoomId) - 0x18000000000000 | ||||||
|  | 			} | ||||||
| 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | ||||||
| 			b.Remote <- msg | 			b.Remote <- msg | ||||||
| 		case *steam.PersonaStateEvent: | 		case *steam.PersonaStateEvent: | ||||||
| 			flog.Debugf("PersonaStateEvent: %#v\n", e) | 			b.Log.Debugf("PersonaStateEvent: %#v\n", e) | ||||||
| 			b.Lock() | 			b.Lock() | ||||||
| 			b.userMap[e.FriendId] = e.Name | 			b.userMap[e.FriendId] = e.Name | ||||||
| 			b.Unlock() | 			b.Unlock() | ||||||
| @@ -114,47 +136,45 @@ func (b *Bsteam) handleEvents() { | |||||||
| 			b.c.Auth.LogOn(myLoginInfo) | 			b.c.Auth.LogOn(myLoginInfo) | ||||||
| 		case *steam.MachineAuthUpdateEvent: | 		case *steam.MachineAuthUpdateEvent: | ||||||
| 			/* | 			/* | ||||||
| 				flog.Info("authupdate", e) | 				b.Log.Info("authupdate", e) | ||||||
| 				flog.Info("hash", e.Hash) | 				b.Log.Info("hash", e.Hash) | ||||||
| 				ioutil.WriteFile("sentry", e.Hash, 0666) | 				ioutil.WriteFile("sentry", e.Hash, 0666) | ||||||
| 			*/ | 			*/ | ||||||
| 		case *steam.LogOnFailedEvent: | 		case *steam.LogOnFailedEvent: | ||||||
| 			flog.Info("Logon failed", e) | 			b.Log.Info("Logon failed", e) | ||||||
| 			switch e.Result { | 			switch e.Result { | ||||||
| 			case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode: | 			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 | 					var code string | ||||||
| 					fmt.Scanf("%s", &code) | 					fmt.Scanf("%s", &code) | ||||||
| 					myLoginInfo.TwoFactorCode = code | 					myLoginInfo.TwoFactorCode = code | ||||||
| 				} | 				} | ||||||
| 			case steamlang.EResult_AccountLogonDenied: | 			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 | 					var code string | ||||||
| 					fmt.Scanf("%s", &code) | 					fmt.Scanf("%s", &code) | ||||||
| 					myLoginInfo.AuthCode = code | 					myLoginInfo.AuthCode = code | ||||||
| 				} | 				} | ||||||
| 			default: | 			default: | ||||||
| 				log.Errorf("LogOnFailedEvent: %#v ", e.Result) | 				b.Log.Errorf("LogOnFailedEvent: %#v ", e.Result) | ||||||
| 				// TODO: Handle EResult_InvalidLoginAuthCode | 				// TODO: Handle EResult_InvalidLoginAuthCode | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
| 		case *steam.LoggedOnEvent: | 		case *steam.LoggedOnEvent: | ||||||
| 			flog.Debugf("LoggedOnEvent: %#v", e) | 			b.Log.Debugf("LoggedOnEvent: %#v", e) | ||||||
| 			b.connected <- struct{}{} | 			b.connected <- struct{}{} | ||||||
| 			flog.Debugf("setting online") | 			b.Log.Debugf("setting online") | ||||||
| 			b.c.Social.SetPersonaState(steamlang.EPersonaState_Online) | 			b.c.Social.SetPersonaState(steamlang.EPersonaState_Online) | ||||||
| 		case *steam.DisconnectedEvent: | 		case *steam.DisconnectedEvent: | ||||||
| 			flog.Info("Disconnected") | 			b.Log.Info("Disconnected") | ||||||
| 			flog.Info("Attempting to reconnect...") | 			b.Log.Info("Attempting to reconnect...") | ||||||
| 			b.c.Connect() | 			b.c.Connect() | ||||||
| 		case steam.FatalErrorEvent: | 		case steam.FatalErrorEvent: | ||||||
| 			flog.Error(e) | 			b.Log.Error(e) | ||||||
| 		case error: |  | ||||||
| 			flog.Error(e) |  | ||||||
| 		default: | 		default: | ||||||
| 			flog.Debugf("unknown event %#v", e) | 			b.Log.Debugf("unknown event %#v", e) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -2,15 +2,17 @@ package btelegram | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
| 	"github.com/russross/blackfriday" |  | ||||||
| 	"html" | 	"html" | ||||||
|  | 	"io" | ||||||
|  |  | ||||||
|  | 	"github.com/russross/blackfriday" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type customHtml struct { | type customHTML struct { | ||||||
| 	blackfriday.Renderer | 	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() | 	marker := out.Len() | ||||||
|  |  | ||||||
| 	if !text() { | 	if !text() { | ||||||
| @@ -20,45 +22,48 @@ func (options *customHtml) Paragraph(out *bytes.Buffer, text func() bool) { | |||||||
| 	out.WriteString("\n") | 	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("<pre>") | ||||||
|  |  | ||||||
| 	out.WriteString(html.EscapeString(string(text))) | 	out.WriteString(html.EscapeString(string(text))) | ||||||
| 	out.WriteString("</pre>\n") | 	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) | 	options.Paragraph(out, text) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (options *customHtml) HRule(out *bytes.Buffer) { | func (options *customHTML) HRule(out io.ByteWriter) { | ||||||
| 	out.WriteByte('\n') | 	out.WriteByte('\n') | ||||||
| } | } | ||||||
|  |  | ||||||
| func (options *customHtml) BlockQuote(out *bytes.Buffer, text []byte) { | func (options *customHTML) BlockQuote(out *bytes.Buffer, text []byte) { | ||||||
| 	out.WriteString("> ") | 	out.WriteString("> ") | ||||||
| 	out.Write(text) | 	out.Write(text) | ||||||
| 	out.WriteByte('\n') | 	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) | 	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.WriteString("- ") | ||||||
| 	out.Write(text) | 	out.Write(text) | ||||||
| 	out.WriteByte('\n') | 	out.WriteByte('\n') | ||||||
| } | } | ||||||
|  |  | ||||||
| func makeHTML(input string) string { | func makeHTML(input string) string { | ||||||
| 	return string(blackfriday.Markdown([]byte(input), | 	extensions := blackfriday.NoIntraEmphasis | | ||||||
| 		&customHtml{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")}, | 		blackfriday.FencedCode | | ||||||
| 		blackfriday.EXTENSION_NO_INTRA_EMPHASIS| | 		blackfriday.Autolink | | ||||||
| 			blackfriday.EXTENSION_FENCED_CODE| | 		blackfriday.SpaceHeadings | | ||||||
| 			blackfriday.EXTENSION_AUTOLINK| | 		blackfriday.HeadingIDs | | ||||||
| 			blackfriday.EXTENSION_SPACE_HEADERS| | 		blackfriday.BackslashLineBreak | | ||||||
| 			blackfriday.EXTENSION_HEADER_IDS| | 		blackfriday.DefinitionLists | ||||||
| 			blackfriday.EXTENSION_BACKSLASH_LINE_BREAK| |  | ||||||
| 			blackfriday.EXTENSION_DEFINITION_LISTS)) | 	renderer := &customHTML{blackfriday.NewHTMLRenderer(blackfriday.HTMLRendererParameters{ | ||||||
|  | 		Flags: blackfriday.UseXHTML | blackfriday.SkipImages, | ||||||
|  | 	})} | ||||||
|  | 	return string(blackfriday.Run([]byte(input), blackfriday.WithExtensions(extensions), blackfriday.WithRenderer(renderer))) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,137 +1,257 @@ | |||||||
| package btelegram | package btelegram | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"html" | ||||||
|  | 	"regexp" | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	unknownUser = "unknown" | ||||||
|  | 	HTMLFormat  = "HTML" | ||||||
|  | 	HTMLNick    = "htmlnick" | ||||||
|  | ) | ||||||
|  |  | ||||||
| type Btelegram struct { | type Btelegram struct { | ||||||
| 	c       *tgbotapi.BotAPI | 	c *tgbotapi.BotAPI | ||||||
| 	Config  *config.Protocol | 	*bridge.Config | ||||||
| 	Remote  chan config.Message | 	avatarMap map[string]string // keep cache of userid and avatar sha | ||||||
| 	Account string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "telegram" | 	return &Btelegram{Config: cfg, avatarMap: make(map[string]string)} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Btelegram { |  | ||||||
| 	b := &Btelegram{} |  | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Remote = c |  | ||||||
| 	b.Account = account |  | ||||||
| 	return b |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Connect() error { | func (b *Btelegram) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Info("Connecting") | 	b.Log.Info("Connecting") | ||||||
| 	b.c, err = tgbotapi.NewBotAPI(b.Config.Token) | 	b.c, err = tgbotapi.NewBotAPI(b.GetString("Token")) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		b.Log.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	updates, err := b.c.GetUpdatesChan(tgbotapi.NewUpdate(0)) | 	u := tgbotapi.NewUpdate(0) | ||||||
|  | 	u.Timeout = 60 | ||||||
|  | 	updates, err := b.c.GetUpdatesChan(u) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		b.Log.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	b.Log.Info("Connection succeeded") | ||||||
| 	go b.handleRecv(updates) | 	go b.handleRecv(updates) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Disconnect() error { | func (b *Btelegram) Disconnect() error { | ||||||
| 	return nil | 	return nil | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) JoinChannel(channel string) error { | func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Btelegram) Send(msg config.Message) error { | func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	b.Log.Debugf("=> Receiving %#v", msg) | ||||||
|  |  | ||||||
|  | 	// get the chatid | ||||||
| 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if b.Config.MessageFormat == "HTML" { | 	// map the file SHA to our user (caches the avatar) | ||||||
|  | 	if msg.Event == config.EventAvatarDownload { | ||||||
|  | 		return b.cacheAvatar(&msg) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if b.GetString("MessageFormat") == HTMLFormat { | ||||||
| 		msg.Text = makeHTML(msg.Text) | 		msg.Text = makeHTML(msg.Text) | ||||||
| 	} | 	} | ||||||
| 	m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text) |  | ||||||
| 	if b.Config.MessageFormat == "HTML" { | 	// Delete message | ||||||
| 		m.ParseMode = tgbotapi.ModeHTML | 	if msg.Event == config.EventMsgDelete { | ||||||
|  | 		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 | ||||||
|  | 		} | ||||||
|  | 		if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||||
|  | 			b.Log.Debug("Using mode HTML - nick only") | ||||||
|  | 			msg.Text = html.EscapeString(msg.Text) | ||||||
|  | 		} | ||||||
|  | 		m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text) | ||||||
|  | 		if b.GetString("MessageFormat") == HTMLFormat { | ||||||
|  | 			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 | ||||||
|  | 		} | ||||||
|  | 		if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||||
|  | 			b.Log.Debug("Using mode HTML - nick only") | ||||||
|  | 			m.ParseMode = tgbotapi.ModeHTML | ||||||
|  | 		} | ||||||
|  | 		_, 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) { | func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||||
| 	for update := range updates { | 	for update := range updates { | ||||||
| 		flog.Debugf("Receiving from telegram: %#v", update.Message) | 		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 | 		var message *tgbotapi.Message | ||||||
| 		username := "" |  | ||||||
| 		channel := "" | 		rmsg := config.Message{Account: b.Account, Extra: make(map[string][]interface{})} | ||||||
| 		text := "" |  | ||||||
| 		// handle channels | 		// handle channels | ||||||
| 		if update.ChannelPost != nil { | 		if update.ChannelPost != nil { | ||||||
| 			message = update.ChannelPost | 			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 = update.EditedChannelPost | ||||||
| 			message.Text = message.Text + b.Config.EditSuffix | 			rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix") | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// handle groups | 		// handle groups | ||||||
| 		if update.Message != nil { | 		if update.Message != nil { | ||||||
| 			message = update.Message | 			message = update.Message | ||||||
| 		} | 			rmsg.Text = message.Text | ||||||
| 		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) |  | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if username == "" { | 		// edited group message | ||||||
| 			username = "unknown" | 		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 = unknownUser | ||||||
| 		} | 		} | ||||||
| 		if message.Photo != nil && b.Config.UseInsecureURL { |  | ||||||
| 			photos := *message.Photo | 		// handle any downloads | ||||||
| 			// last photo is the biggest | 		err := b.handleDownload(message, &rmsg) | ||||||
| 			text = text + " " + b.getFileDirectURL(photos[len(photos)-1].FileID) | 		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 = unknownUser | ||||||
|  | 			} | ||||||
|  | 			rmsg.Text = "Forwarded from " + usernameForward + ": " + rmsg.Text | ||||||
| 		} | 		} | ||||||
| 		if text != "" { |  | ||||||
| 			flog.Debugf("Sending message from %s on %s to gateway", username, b.Account) | 		// quote the previous message | ||||||
| 			b.Remote <- config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID)} | 		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 = unknownUser | ||||||
|  | 			} | ||||||
|  | 			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 | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -143,3 +263,188 @@ func (b *Btelegram) getFileDirectURL(id string) string { | |||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // handleDownloadAvatar downloads the avatar of userid from channel | ||||||
|  | // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||||
|  | // logs an error message if it fails | ||||||
|  | func (b *Btelegram) handleDownloadAvatar(userid int, channel string) { | ||||||
|  | 	rmsg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: strconv.Itoa(userid), Event: config.EventAvatarDownload, 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 += ".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 += ".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 += 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{ | ||||||
|  | 			Name:  fi.Name, | ||||||
|  | 			Bytes: *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") == HTMLFormat { | ||||||
|  | 		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 | ||||||
|  | 	} | ||||||
|  | 	if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||||
|  | 		b.Log.Debug("Using mode HTML - nick only") | ||||||
|  | 		m.Text = username + html.EscapeString(text) | ||||||
|  | 		m.ParseMode = tgbotapi.ModeHTML | ||||||
|  | 	} | ||||||
|  | 	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,48 +2,38 @@ package bxmpp | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/jpillora/backoff" |  | ||||||
| 	"github.com/mattn/go-xmpp" |  | ||||||
|  |  | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/helper" | ||||||
|  | 	"github.com/jpillora/backoff" | ||||||
|  | 	"github.com/matterbridge/go-xmpp" | ||||||
|  | 	"github.com/rs/xid" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Bxmpp struct { | type Bxmpp struct { | ||||||
| 	xc      *xmpp.Client | 	xc      *xmpp.Client | ||||||
| 	xmppMap map[string]string | 	xmppMap map[string]string | ||||||
| 	Config  *config.Protocol | 	*bridge.Config | ||||||
| 	Remote  chan config.Message |  | ||||||
| 	Account string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| var flog *log.Entry | func New(cfg *bridge.Config) bridge.Bridger { | ||||||
| var protocol = "xmpp" | 	b := &Bxmpp{Config: cfg} | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	flog = log.WithFields(log.Fields{"module": protocol}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func New(cfg config.Protocol, account string, c chan config.Message) *Bxmpp { |  | ||||||
| 	b := &Bxmpp{} |  | ||||||
| 	b.xmppMap = make(map[string]string) | 	b.xmppMap = make(map[string]string) | ||||||
| 	b.Config = &cfg |  | ||||||
| 	b.Account = account |  | ||||||
| 	b.Remote = c |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) Connect() error { | func (b *Bxmpp) Connect() error { | ||||||
| 	var err error | 	var err error | ||||||
| 	flog.Infof("Connecting %s", b.Config.Server) | 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||||
| 	b.xc, err = b.createXMPP() | 	b.xc, err = b.createXMPP() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		flog.Debugf("%#v", err) | 		b.Log.Debugf("%#v", err) | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	flog.Info("Connection succeeded") | 	b.Log.Info("Connection succeeded") | ||||||
| 	go func() { | 	go func() { | ||||||
| 		initial := true | 		initial := true | ||||||
| 		bf := &backoff.Backoff{ | 		bf := &backoff.Backoff{ | ||||||
| @@ -53,16 +43,16 @@ func (b *Bxmpp) Connect() error { | |||||||
| 		} | 		} | ||||||
| 		for { | 		for { | ||||||
| 			if initial { | 			if initial { | ||||||
| 				b.handleXmpp() | 				b.handleXMPP() | ||||||
| 				initial = false | 				initial = false | ||||||
| 			} | 			} | ||||||
| 			d := bf.Duration() | 			d := bf.Duration() | ||||||
| 			flog.Infof("Disconnected. Reconnecting in %s", d) | 			b.Log.Infof("Disconnected. Reconnecting in %s", d) | ||||||
| 			time.Sleep(d) | 			time.Sleep(d) | ||||||
| 			b.xc, err = b.createXMPP() | 			b.xc, err = b.createXMPP() | ||||||
| 			if err == nil { | 			if err == nil { | ||||||
| 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EventRejoinChannels} | ||||||
| 				b.handleXmpp() | 				b.handleXMPP() | ||||||
| 				bf.Reset() | 				bf.Reset() | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -74,37 +64,65 @@ func (b *Bxmpp) Disconnect() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) JoinChannel(channel string) error { | func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error { | ||||||
| 	b.xc.JoinMUCNoHistory(channel+"@"+b.Config.Muc, b.Config.Nick) | 	if channel.Options.Key != "" { | ||||||
|  | 		b.Log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||||
|  | 		b.xc.JoinProtectedMUC(channel.Name+"@"+b.GetString("Muc"), b.GetString("Nick"), channel.Options.Key, xmpp.NoHistory, 0, nil) | ||||||
|  | 	} else { | ||||||
|  | 		b.xc.JoinMUCNoHistory(channel.Name+"@"+b.GetString("Muc"), b.GetString("Nick")) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) Send(msg config.Message) error { | func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||||
| 	flog.Debugf("Receiving %#v", msg) | 	// ignore delete messages | ||||||
| 	b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text}) | 	if msg.Event == config.EventMsgDelete { | ||||||
| 	return nil | 		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) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var msgreplaceid string | ||||||
|  | 	msgid := xid.New().String() | ||||||
|  | 	if msg.ID != "" { | ||||||
|  | 		msgid = msg.ID | ||||||
|  | 		msgreplaceid = msg.ID | ||||||
|  | 	} | ||||||
|  | 	// Post normal message | ||||||
|  | 	_, err := b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.GetString("Muc"), Text: msg.Username + msg.Text, ID: msgid, ReplaceID: msgreplaceid}) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return msgid, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | ||||||
| 	tc := new(tls.Config) | 	tc := new(tls.Config) | ||||||
| 	tc.InsecureSkipVerify = b.Config.SkipTLSVerify | 	tc.InsecureSkipVerify = b.GetBool("SkipTLSVerify") | ||||||
| 	tc.ServerName = strings.Split(b.Config.Server, ":")[0] | 	tc.ServerName = strings.Split(b.GetString("Server"), ":")[0] | ||||||
| 	options := xmpp.Options{ | 	options := xmpp.Options{ | ||||||
| 		Host:      b.Config.Server, | 		Host:                         b.GetString("Server"), | ||||||
| 		User:      b.Config.Jid, | 		User:                         b.GetString("Jid"), | ||||||
| 		Password:  b.Config.Password, | 		Password:                     b.GetString("Password"), | ||||||
| 		NoTLS:     true, | 		NoTLS:                        true, | ||||||
| 		StartTLS:  true, | 		StartTLS:                     true, | ||||||
| 		TLSConfig: tc, | 		TLSConfig:                    tc, | ||||||
|  | 		Debug:                        b.GetBool("debug"), | ||||||
| 		//StartTLS:      false, | 		Logger:                       b.Log.Writer(), | ||||||
| 		Debug:                        true, |  | ||||||
| 		Session:                      true, | 		Session:                      true, | ||||||
| 		Status:                       "", | 		Status:                       "", | ||||||
| 		StatusMessage:                "", | 		StatusMessage:                "", | ||||||
| 		Resource:                     "", | 		Resource:                     "", | ||||||
| 		InsecureAllowUnencryptedAuth: false, | 		InsecureAllowUnencryptedAuth: false, | ||||||
| 		//InsecureAllowUnencryptedAuth: true, |  | ||||||
| 	} | 	} | ||||||
| 	var err error | 	var err error | ||||||
| 	b.xc, err = options.NewClient() | 	b.xc, err = options.NewClient() | ||||||
| @@ -119,10 +137,10 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | |||||||
| 		for { | 		for { | ||||||
| 			select { | 			select { | ||||||
| 			case <-ticker.C: | 			case <-ticker.C: | ||||||
| 				flog.Debugf("PING") | 				b.Log.Debugf("PING") | ||||||
| 				err := b.xc.PingC2S("", "") | 				err := b.xc.PingC2S("", "") | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| 					flog.Debugf("PING failed %#v", err) | 					b.Log.Debugf("PING failed %#v", err) | ||||||
| 				} | 				} | ||||||
| 			case <-done: | 			case <-done: | ||||||
| 				return | 				return | ||||||
| @@ -132,11 +150,11 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | |||||||
| 	return done | 	return done | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Bxmpp) handleXmpp() error { | func (b *Bxmpp) handleXMPP() error { | ||||||
| 	var ok bool | 	var ok bool | ||||||
|  | 	var msgid string | ||||||
| 	done := b.xmppKeepAlive() | 	done := b.xmppKeepAlive() | ||||||
| 	defer close(done) | 	defer close(done) | ||||||
| 	nodelay := time.Time{} |  | ||||||
| 	for { | 	for { | ||||||
| 		m, err := b.xc.Recv() | 		m, err := b.xc.Recv() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| @@ -144,25 +162,26 @@ func (b *Bxmpp) handleXmpp() error { | |||||||
| 		} | 		} | ||||||
| 		switch v := m.(type) { | 		switch v := m.(type) { | ||||||
| 		case xmpp.Chat: | 		case xmpp.Chat: | ||||||
| 			var channel, nick string |  | ||||||
| 			if v.Type == "groupchat" { | 			if v.Type == "groupchat" { | ||||||
| 				s := strings.Split(v.Remote, "@") | 				b.Log.Debugf("== Receiving %#v", v) | ||||||
| 				if len(s) >= 2 { | 				// skip invalid messages | ||||||
| 					channel = s[0] | 				if b.skipMessage(v) { | ||||||
|  | 					continue | ||||||
| 				} | 				} | ||||||
| 				s = strings.Split(s[1], "/") | 				msgid = v.ID | ||||||
| 				if len(s) == 2 { | 				if v.ReplaceID != "" { | ||||||
| 					nick = s[1] | 					msgid = v.ReplaceID | ||||||
| 				} | 				} | ||||||
| 				if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" { | 				rmsg := config.Message{Username: b.parseNick(v.Remote), Text: v.Text, Channel: b.parseChannel(v.Remote), Account: b.Account, UserID: v.Remote, ID: msgid} | ||||||
| 					rmsg := config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote} |  | ||||||
| 					rmsg.Text, ok = b.replaceAction(rmsg.Text) | 				// check if we have an action event | ||||||
| 					if ok { | 				rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||||
| 						rmsg.Event = config.EVENT_USER_ACTION | 				if ok { | ||||||
| 					} | 					rmsg.Event = config.EventUserAction | ||||||
| 					flog.Debugf("Sending message from %s on %s to gateway", nick, b.Account) |  | ||||||
| 					b.Remote <- rmsg |  | ||||||
| 				} | 				} | ||||||
|  | 				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: | 		case xmpp.Presence: | ||||||
| 			// do nothing | 			// do nothing | ||||||
| @@ -176,3 +195,71 @@ func (b *Bxmpp) replaceAction(text string) (string, bool) { | |||||||
| 	} | 	} | ||||||
| 	return text, false | 	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
									
								
							
							
						
						| @@ -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.EventMsgDelete { | ||||||
|  | 		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 | ||||||
|  | } | ||||||
							
								
								
									
										343
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						| @@ -1,3 +1,346 @@ | |||||||
|  | # v1.12.1 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * discord: fix regression on server ID connection #619 #617 | ||||||
|  | * discord: Limit discord username via webhook to 32 chars | ||||||
|  | * slack: Make sure threaded files stay in thread (slack). Fixes #590 | ||||||
|  | * slack: Do not post empty messages (slack). Fixes #574 | ||||||
|  | * slack: Handle deleted/edited thread starting messages (slack). Fixes #600 (#605) | ||||||
|  | * irc: Rework connection logic (irc) | ||||||
|  | * irc: Fix Nickserv logic (irc) #602 | ||||||
|  |  | ||||||
|  | # v1.12.0 | ||||||
|  |  | ||||||
|  | ## Breaking changes | ||||||
|  | The slack bridge has been split in a `slack-legacy` and `slack` bridge. | ||||||
|  | If you're still using `legacy tokens` and want to keep using them you'll have to rename `slack` to `slack-legacy` in your configuration. See [wiki](https://github.com/42wim/matterbridge/wiki/Section-Slack-(basic)#legacy-configuration) for more information. | ||||||
|  |  | ||||||
|  | To migrate to the new bot-token based setup you can follow the instructions [here](https://github.com/42wim/matterbridge/wiki/Slack-bot-setup). | ||||||
|  |  | ||||||
|  | Slack legacy tokens may be deprecated by Slack at short notice, so it is STRONGLY recommended to use a proper bot-token instead. | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * general: New {GATEWAY} variable for `RemoteNickFormat` #501. See `RemoteNickFormat` in matterbridge.toml.sample. | ||||||
|  | * general: New {CHANNEL} variable for `RemoteNickFormat` #515. See `RemoteNickFormat` in matterbridge.toml.sample. | ||||||
|  | * general: Remove hyphens when auto-loading envvars from viper config #545 | ||||||
|  | * discord: You can mention discord-users from other bridges. | ||||||
|  | * slack: Preserve threading between Slack instances #529. See `PreserveThreading` in matterbridge.toml.sample. | ||||||
|  | * slack: Add ability to show when user is typing across Slack bridges #559 | ||||||
|  | * slack: Add rate-limiting | ||||||
|  | * mattermost: Add support for mattermost [matterbridge plugin](https://github.com/matterbridge/mattermost-plugin) | ||||||
|  | * api: Respond with message on connect. #550 | ||||||
|  | * api: Add a health endpoint to API #554 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Refactoring and making it better. | ||||||
|  | * slack: Restore file comments coming from Slack. #583 | ||||||
|  | * irc: Fix IRC line splitting. #587 | ||||||
|  | * mattermost: Fix cookie and personal token behaviour. #530 | ||||||
|  | * mattermost: Check for expiring sessions and reconnect. | ||||||
|  |  | ||||||
|  |  | ||||||
|  | ## Contributors | ||||||
|  | This release couldn't exist without the following contributors: | ||||||
|  | @jheiselman, @NikkyAI, @dajohi, @NetwideRogue, @patcon and @Helcaraxan | ||||||
|  |  | ||||||
|  | Special thanks to @Helcaraxan and @patcon for their work on improving/refactoring slack. | ||||||
|  |  | ||||||
|  | # v1.11.3 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * mattermost: fix panic when using webhooks #491 | ||||||
|  | * slack: fix issues regarding API changes and lots of channels #489 | ||||||
|  | * irc: fix rejoin on kick problem #488 | ||||||
|  |  | ||||||
|  | # v1.11.2 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: fix slack API changes regarding to files/images | ||||||
|  |  | ||||||
|  | # v1.11.1 | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * slack: Add support for slack channels by ID. Closes #436 | ||||||
|  | * discord: Clip too long messages sent to discord (discord). Closes #440 | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * general: fix possible panic on downloads that are too big #448 | ||||||
|  | * general: Fix avatar uploads to work with MediaDownloadPath. Closes #454 | ||||||
|  | * discord: allow receiving of topic changes/channel leave/joins from other bridges through the webhook | ||||||
|  | * discord: Add a space before url in file uploads (discord). Closes #461 | ||||||
|  | * discord:  Skip empty messages being sent with the webhook (discord). #469 | ||||||
|  | * mattermost: Use nickname instead of username if defined (mattermost). Closes #452 | ||||||
|  | * irc: Stop numbers being stripped after non-color control codes (irc) (#465) | ||||||
|  | * slack: Use UserID to look for avatar instead of username (slack). Closes #472 | ||||||
|  |  | ||||||
|  | # v1.11.0 | ||||||
|  |  | ||||||
|  | ## New features | ||||||
|  | * general: Add config option MediaDownloadPath (#443). See `MediaDownloadPath` in matterbridge.toml.sample | ||||||
|  | * general: Add MediaDownloadBlacklist option. Closes #442. See `MediaDownloadBlacklist` in matterbridge.toml.sample | ||||||
|  | * xmpp: Add channel password support for XMPP (#451) | ||||||
|  | * xmpp: Add message correction support for XMPP (#437) | ||||||
|  | * telegram: Add support for MessageFormat=htmlnick (telegram). #444 | ||||||
|  | * mattermost: Add support for mattermost 5.x | ||||||
|  |  | ||||||
|  | ## Enhancements | ||||||
|  | * slack: Add Title from attachment slack message (#446) | ||||||
|  | * irc: Prevent white or black color codes (irc) (#434) | ||||||
|  |  | ||||||
|  | ## Bugfix | ||||||
|  | * slack: Fix regexp in replaceMention (slack). (#435) | ||||||
|  | * irc: Reconnect on quit. (irc) See #431 (#445) | ||||||
|  | * sshchat: Ignore messages from ourself. (sshchat) Closes #439 | ||||||
|  |  | ||||||
|  | # v1.10.1 | ||||||
|  | ## New features | ||||||
|  | * 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 #413. See `QuoteFormat` in matterbridge.toml.sample | ||||||
|  | * xmpp: Send attached files to XMPP in different message with OOB data and without body (#421) | ||||||
|  |  | ||||||
|  | ## 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) | ||||||
|  |  | ||||||
|  | # 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 | # v1.0.0-rc1 | ||||||
| ## New features | ## New features | ||||||
| * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | * general: Add action support for slack,mattermost,irc,gitter,matrix,xmpp,discord. #199 | ||||||
|   | |||||||
| @@ -1,10 +1,11 @@ | |||||||
| #!/bin/bash | #!/bin/bash | ||||||
| go version |grep go1.8 || exit | go version | grep go1.11 || exit | ||||||
| VERSION=$(git describe --tags) | VERSION=$(git describe --tags) | ||||||
| mkdir ci/binaries | mkdir ci/binaries | ||||||
| GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-win64.exe | GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-windows-amd64.exe | ||||||
| GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux64 | GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-amd64 | ||||||
| GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | ||||||
|  | GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-darwin-amd64 | ||||||
| cd ci | cd ci | ||||||
| cat > deploy.json <<EOF | cat > deploy.json <<EOF | ||||||
| { | { | ||||||
|   | |||||||
							
								
								
									
										11
									
								
								contrib/matterbridge.service
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -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
									
								
							
							
						
						| @@ -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,19 +1,40 @@ | |||||||
| package gateway | package gateway | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"crypto/sha1" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge" | 	"io/ioutil" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"net/http" | ||||||
| 	log "github.com/Sirupsen/logrus" | 	"os" | ||||||
| 	//	"github.com/davecgh/go-spew/spew" | 	"path/filepath" | ||||||
| 	"github.com/peterhellberg/emojilib" |  | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/api" | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	bdiscord "github.com/42wim/matterbridge/bridge/discord" | ||||||
|  | 	bgitter "github.com/42wim/matterbridge/bridge/gitter" | ||||||
|  | 	birc "github.com/42wim/matterbridge/bridge/irc" | ||||||
|  | 	bmatrix "github.com/42wim/matterbridge/bridge/matrix" | ||||||
|  | 	bmattermost "github.com/42wim/matterbridge/bridge/mattermost" | ||||||
|  | 	brocketchat "github.com/42wim/matterbridge/bridge/rocketchat" | ||||||
|  | 	bslack "github.com/42wim/matterbridge/bridge/slack" | ||||||
|  | 	bsshchat "github.com/42wim/matterbridge/bridge/sshchat" | ||||||
|  | 	bsteam "github.com/42wim/matterbridge/bridge/steam" | ||||||
|  | 	btelegram "github.com/42wim/matterbridge/bridge/telegram" | ||||||
|  | 	bxmpp "github.com/42wim/matterbridge/bridge/xmpp" | ||||||
|  | 	bzulip "github.com/42wim/matterbridge/bridge/zulip" | ||||||
|  | 	"github.com/hashicorp/golang-lru" | ||||||
|  | 	"github.com/peterhellberg/emojilib" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Gateway struct { | type Gateway struct { | ||||||
| 	*config.Config | 	config.Config | ||||||
|  |  | ||||||
| 	Router         *Router | 	Router         *Router | ||||||
| 	MyConfig       *config.Gateway | 	MyConfig       *config.Gateway | ||||||
| 	Bridges        map[string]*bridge.Bridge | 	Bridges        map[string]*bridge.Bridge | ||||||
| @@ -21,19 +42,79 @@ type Gateway struct { | |||||||
| 	ChannelOptions map[string]config.ChannelOptions | 	ChannelOptions map[string]config.ChannelOptions | ||||||
| 	Message        chan config.Message | 	Message        chan config.Message | ||||||
| 	Name           string | 	Name           string | ||||||
|  | 	Messages       *lru.Cache | ||||||
| } | } | ||||||
|  |  | ||||||
|  | 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-legacy": bslack.NewLegacy, | ||||||
|  | 	"slack":        bslack.New, | ||||||
|  | 	"sshchat":      bsshchat.New, | ||||||
|  | 	"steam":        bsteam.New, | ||||||
|  | 	"telegram":     btelegram.New, | ||||||
|  | 	"xmpp":         bxmpp.New, | ||||||
|  | 	"zulip":        bzulip.New, | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	apiProtocol = "api" | ||||||
|  | ) | ||||||
|  |  | ||||||
| func New(cfg config.Gateway, r *Router) *Gateway { | func New(cfg config.Gateway, r *Router) *Gateway { | ||||||
|  | 	flog = log.WithFields(log.Fields{"prefix": "gateway"}) | ||||||
| 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | ||||||
| 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | ||||||
|  | 	cache, _ := lru.New(5000) | ||||||
|  | 	gw.Messages = cache | ||||||
| 	gw.AddConfig(&cfg) | 	gw.AddConfig(&cfg) | ||||||
| 	return gw | 	return gw | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Find the canonical ID that the message is keyed under in cache | ||||||
|  | func (gw *Gateway) FindCanonicalMsgID(protocol string, mID string) string { | ||||||
|  | 	ID := protocol + " " + mID | ||||||
|  | 	if gw.Messages.Contains(ID) { | ||||||
|  | 		return mID | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If not keyed, iterate through cache for downstream, and infer upstream. | ||||||
|  | 	for _, mid := range gw.Messages.Keys() { | ||||||
|  | 		v, _ := gw.Messages.Peek(mid) | ||||||
|  | 		ids := v.([]*BrMsgID) | ||||||
|  | 		for _, downstreamMsgObj := range ids { | ||||||
|  | 			if ID == downstreamMsgObj.ID { | ||||||
|  | 				return strings.Replace(mid.(string), protocol+" ", "", 1) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
| func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||||
| 	br := gw.Router.getBridge(cfg.Account) | 	br := gw.Router.getBridge(cfg.Account) | ||||||
| 	if br == nil { | 	if br == nil { | ||||||
| 		br = bridge.New(gw.Config, cfg, gw.Message) | 		br = bridge.New(cfg) | ||||||
|  | 		br.Config = gw.Router.Config | ||||||
|  | 		br.General = &gw.BridgeValues().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) | ||||||
| 	} | 	} | ||||||
| 	gw.mapChannelsToBridge(br) | 	gw.mapChannelsToBridge(br) | ||||||
| 	gw.Bridges[cfg.Account] = br | 	gw.Bridges[cfg.Account] = br | ||||||
| @@ -45,6 +126,7 @@ func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | |||||||
| 	gw.MyConfig = cfg | 	gw.MyConfig = cfg | ||||||
| 	gw.mapChannels() | 	gw.mapChannels() | ||||||
| 	for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) { | 	for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) { | ||||||
|  | 		br := br //scopelint | ||||||
| 		err := gw.AddBridge(&br) | 		err := gw.AddBridge(&br) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| @@ -65,10 +147,10 @@ func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | |||||||
| 	br.Disconnect() | 	br.Disconnect() | ||||||
| 	time.Sleep(time.Second * 5) | 	time.Sleep(time.Second * 5) | ||||||
| RECONNECT: | RECONNECT: | ||||||
| 	log.Infof("Reconnecting %s", br.Account) | 	flog.Infof("Reconnecting %s", br.Account) | ||||||
| 	err := br.Connect() | 	err := br.Connect() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err) | 		flog.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err) | ||||||
| 		time.Sleep(time.Second * 60) | 		time.Sleep(time.Second * 60) | ||||||
| 		goto RECONNECT | 		goto RECONNECT | ||||||
| 	} | 	} | ||||||
| @@ -78,8 +160,12 @@ RECONNECT: | |||||||
|  |  | ||||||
| func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | ||||||
| 	for _, br := range cfg { | 	for _, br := range cfg { | ||||||
| 		if isApi(br.Account) { | 		if isAPI(br.Account) { | ||||||
| 			br.Channel = "api" | 			br.Channel = apiProtocol | ||||||
|  | 		} | ||||||
|  | 		// make sure to lowercase irc channels in config #348 | ||||||
|  | 		if strings.HasPrefix(br.Account, "irc.") { | ||||||
|  | 			br.Channel = strings.ToLower(br.Channel) | ||||||
| 		} | 		} | ||||||
| 		ID := br.Channel + br.Account | 		ID := br.Channel + br.Account | ||||||
| 		if _, ok := gw.Channels[ID]; !ok { | 		if _, ok := gw.Channels[ID]; !ok { | ||||||
| @@ -106,6 +192,12 @@ func (gw *Gateway) mapChannels() error { | |||||||
|  |  | ||||||
| func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo { | func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo { | ||||||
| 	var channels []config.ChannelInfo | 	var channels []config.ChannelInfo | ||||||
|  |  | ||||||
|  | 	// for messages received from the api check that the gateway is the specified one | ||||||
|  | 	if msg.Protocol == apiProtocol && gw.Name != msg.Gateway { | ||||||
|  | 		return channels | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// if source channel is in only, do nothing | 	// if source channel is in only, do nothing | ||||||
| 	for _, channel := range gw.Channels { | 	for _, channel := range gw.Channels { | ||||||
| 		// lookup the channel from the message | 		// lookup the channel from the message | ||||||
| @@ -122,51 +214,131 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con | |||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// do samechannelgateway logic | 		// do samechannelgateway flogic | ||||||
| 		if channel.SameChannel[msg.Gateway] { | 		if channel.SameChannel[msg.Gateway] { | ||||||
| 			if msg.Channel == channel.Name && msg.Account != dest.Account { | 			if msg.Channel == channel.Name && msg.Account != dest.Account { | ||||||
| 				channels = append(channels, *channel) | 				channels = append(channels, *channel) | ||||||
| 			} | 			} | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) { | 		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg) { | ||||||
| 			channels = append(channels, *channel) | 			channels = append(channels, *channel) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) { | func (gw *Gateway) getDestMsgID(msgID string, dest *bridge.Bridge, channel config.ChannelInfo) string { | ||||||
| 	// only relay join/part when configged | 	if res, ok := gw.Messages.Get(msgID); ok { | ||||||
| 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart { | 		IDs := res.([]*BrMsgID) | ||||||
| 		return | 		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 { | ||||||
|  | 				return strings.Replace(id.ID, dest.Protocol+" ", "", 1) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  | 	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.EventFileFailureSize]) != 0 { | ||||||
|  | 			if msg.Text == "" { | ||||||
|  | 				return brMsgIDs | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Avatar downloads are only relevant for telegram and mattermost for now | ||||||
|  | 	if msg.Event == config.EventAvatarDownload { | ||||||
|  | 		if dest.Protocol != "mattermost" && | ||||||
|  | 			dest.Protocol != "telegram" { | ||||||
|  | 			return brMsgIDs | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// only relay join/part when configured | ||||||
|  | 	if msg.Event == config.EventJoinLeave && !gw.Bridges[dest.Account].GetBool("ShowJoinPart") { | ||||||
|  | 		return brMsgIDs | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// only relay topic change when configured | ||||||
|  | 	if msg.Event == config.EventTopicChange && !gw.Bridges[dest.Account].GetBool("ShowTopicChange") { | ||||||
|  | 		return brMsgIDs | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// broadcast to every out channel (irc QUIT) | 	// broadcast to every out channel (irc QUIT) | ||||||
| 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | 	if msg.Channel == "" && msg.Event != config.EventJoinLeave { | ||||||
| 		log.Debug("empty channel") | 		flog.Debug("empty channel") | ||||||
| 		return | 		return brMsgIDs | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// Get the ID of the parent message in thread | ||||||
|  | 	var canonicalParentMsgID string | ||||||
|  | 	if msg.ParentID != "" && (gw.BridgeValues().General.PreserveThreading || dest.GetBool("PreserveThreading")) { | ||||||
|  | 		canonicalParentMsgID = gw.FindCanonicalMsgID(msg.Protocol, msg.ParentID) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	originchannel := msg.Channel | 	originchannel := msg.Channel | ||||||
| 	origmsg := msg | 	origmsg := msg | ||||||
| 	channels := gw.getDestChannel(&msg, *dest) | 	channels := gw.getDestChannel(&msg, *dest) | ||||||
| 	for _, channel := range channels { | 	for _, channel := range channels { | ||||||
| 		// do not send to ourself | 		// Only send the avatar download event to ourselves. | ||||||
| 		if channel.ID == getChannelID(origmsg) { | 		if msg.Event == config.EventAvatarDownload { | ||||||
| 			continue | 			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) |  | ||||||
|  | 		// Too noisy to log like other events | ||||||
|  | 		if msg.Event != config.EventUserTyping { | ||||||
|  | 			flog.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name) | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		msg.Channel = channel.Name | 		msg.Channel = channel.Name | ||||||
| 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||||
| 		msg.Username = gw.modifyUsername(origmsg, dest) | 		msg.Username = gw.modifyUsername(origmsg, dest) | ||||||
|  |  | ||||||
|  | 		msg.ID = gw.getDestMsgID(origmsg.Protocol+" "+origmsg.ID, dest, channel) | ||||||
|  |  | ||||||
| 		// for api we need originchannel as channel | 		// for api we need originchannel as channel | ||||||
| 		if dest.Protocol == "api" { | 		if dest.Protocol == apiProtocol { | ||||||
| 			msg.Channel = originchannel | 			msg.Channel = originchannel | ||||||
| 		} | 		} | ||||||
| 		err := dest.Send(msg) |  | ||||||
|  | 		msg.ParentID = gw.getDestMsgID(origmsg.Protocol+" "+canonicalParentMsgID, dest, channel) | ||||||
|  | 		if msg.ParentID == "" { | ||||||
|  | 			msg.ParentID = canonicalParentMsgID | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// if we are using mattermost plugin account, send messages to MattermostPlugin channel | ||||||
|  | 		// that can be picked up by the mattermost matterbridge plugin | ||||||
|  | 		if dest.Account == "mattermost.plugin" { | ||||||
|  | 			gw.Router.MattermostPlugin <- msg | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		mID, err := dest.Send(msg) | ||||||
| 		if err != nil { | 		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, dest.Protocol + " " + mID, channel.ID}) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  | 	return brMsgIDs | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||||
| @@ -174,26 +346,42 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | |||||||
| 	if _, ok := gw.Bridges[msg.Account]; !ok { | 	if _, ok := gw.Bridges[msg.Account]; !ok { | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// check if we need to ignore a empty message | ||||||
| 	if msg.Text == "" { | 	if msg.Text == "" { | ||||||
| 		log.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | 		if msg.Event == config.EventUserTyping { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 		// 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.EventFileFailureSize]) > 0) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 		flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) { |  | ||||||
|  | 	// is the username in IgnoreNicks field | ||||||
|  | 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks")) { | ||||||
| 		if msg.Username == entry { | 		if msg.Username == entry { | ||||||
| 			log.Debugf("ignoring %s from %s", msg.Username, msg.Account) | 			flog.Debugf("ignoring %s from %s", msg.Username, msg.Account) | ||||||
| 			return true | 			return true | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// does the message match regex in IgnoreMessages field | ||||||
| 	// TODO do not compile regexps everytime | 	// 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 != "" { | 		if entry != "" { | ||||||
| 			re, err := regexp.Compile(entry) | 			re, err := regexp.Compile(entry) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				log.Errorf("incorrect regexp %s for %s", entry, msg.Account) | 				flog.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
| 			if re.MatchString(msg.Text) { | 			if re.MatchString(msg.Text) { | ||||||
| 				log.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account) | 				flog.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account) | ||||||
| 				return true | 				return true | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -204,10 +392,28 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | |||||||
| func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string { | func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string { | ||||||
| 	br := gw.Bridges[msg.Account] | 	br := gw.Bridges[msg.Account] | ||||||
| 	msg.Protocol = br.Protocol | 	msg.Protocol = br.Protocol | ||||||
| 	nick := gw.Config.General.RemoteNickFormat | 	if gw.BridgeValues().General.StripNick || dest.GetBool("StripNick") { | ||||||
| 	if nick == "" { | 		re := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||||
| 		nick = dest.Config.RemoteNickFormat | 		msg.Username = re.ReplaceAllString(msg.Username, "") | ||||||
| 	} | 	} | ||||||
|  | 	nick := dest.GetString("RemoteNickFormat") | ||||||
|  | 	if nick == "" { | ||||||
|  | 		nick = gw.BridgeValues().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 { | 	if len(msg.Username) > 0 { | ||||||
| 		// fix utf-8 issue #193 | 		// fix utf-8 issue #193 | ||||||
| 		i := 0 | 		i := 0 | ||||||
| @@ -220,16 +426,20 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin | |||||||
| 		} | 		} | ||||||
| 		nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1) | 		nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1) | ||||||
| 	} | 	} | ||||||
| 	nick = strings.Replace(nick, "{NICK}", msg.Username, -1) |  | ||||||
| 	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1) | 	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1) | ||||||
| 	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1) | 	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1) | ||||||
|  | 	nick = strings.Replace(nick, "{GATEWAY}", gw.Name, -1) | ||||||
|  | 	nick = strings.Replace(nick, "{LABEL}", br.GetString("Label"), -1) | ||||||
|  | 	nick = strings.Replace(nick, "{NICK}", msg.Username, -1) | ||||||
|  | 	nick = strings.Replace(nick, "{CHANNEL}", msg.Channel, -1) | ||||||
| 	return nick | 	return nick | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string { | func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string { | ||||||
| 	iconurl := gw.Config.General.IconURL | 	iconurl := gw.BridgeValues().General.IconURL | ||||||
| 	if iconurl == "" { | 	if iconurl == "" { | ||||||
| 		iconurl = dest.Config.IconURL | 		iconurl = dest.GetString("IconURL") | ||||||
| 	} | 	} | ||||||
| 	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1) | 	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1) | ||||||
| 	if msg.Avatar == "" { | 	if msg.Avatar == "" { | ||||||
| @@ -241,17 +451,117 @@ func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string | |||||||
| func (gw *Gateway) modifyMessage(msg *config.Message) { | func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||||
| 	// replace :emoji: to unicode | 	// replace :emoji: to unicode | ||||||
| 	msg.Text = emojilib.Replace(msg.Text) | 	msg.Text = emojilib.Replace(msg.Text) | ||||||
| 	msg.Gateway = gw.Name |  | ||||||
|  | 	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 != apiProtocol { | ||||||
|  | 		msg.Gateway = gw.Name | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // handleFiles uploads or places all files on the given msg to the MediaServer and | ||||||
|  | // adds the new URL of the file on the MediaServer onto the given msg. | ||||||
|  | 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.BridgeValues().General.MediaServerUpload == "" && | ||||||
|  | 			gw.BridgeValues().General.MediaDownloadPath == "") { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// If we don't have files, nothing to upload. | ||||||
|  | 	if len(msg.Extra["file"]) == 0 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	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 += ext | ||||||
|  |  | ||||||
|  | 		sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] | ||||||
|  |  | ||||||
|  | 		if gw.BridgeValues().General.MediaServerUpload != "" { | ||||||
|  | 			// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth. | ||||||
|  |  | ||||||
|  | 			url := gw.BridgeValues().General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name | ||||||
|  |  | ||||||
|  | 			req, err := http.NewRequest("PUT", url, bytes.NewReader(*fi.Data)) | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Errorf("mediaserver upload failed, could not create request: %#v", err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			flog.Debugf("mediaserver upload url: %s", url) | ||||||
|  |  | ||||||
|  | 			req.Header.Set("Content-Type", "binary/octet-stream") | ||||||
|  | 			_, err = client.Do(req) | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Errorf("mediaserver upload failed, could not Do request: %#v", err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			// Use MediaServerPath. Place the file on the current filesystem. | ||||||
|  |  | ||||||
|  | 			dir := gw.BridgeValues().General.MediaDownloadPath + "/" + sha1sum | ||||||
|  | 			err := os.Mkdir(dir, os.ModePerm) | ||||||
|  | 			if err != nil && !os.IsExist(err) { | ||||||
|  | 				flog.Errorf("mediaserver path failed, could not mkdir: %s %#v", err, err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			path := dir + "/" + fi.Name | ||||||
|  | 			flog.Debugf("mediaserver path placing file: %s", path) | ||||||
|  |  | ||||||
|  | 			err = ioutil.WriteFile(path, *fi.Data, os.ModePerm) | ||||||
|  | 			if err != nil { | ||||||
|  | 				flog.Errorf("mediaserver path failed, could not writefile: %s %#v", err, err) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Download URL. | ||||||
|  | 		durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||||
|  |  | ||||||
|  | 		flog.Debugf("mediaserver download URL = %s", durl) | ||||||
|  |  | ||||||
|  | 		// We uploaded/placed the file successfully. Add the SHA and URL. | ||||||
|  | 		extra := msg.Extra["file"][i].(config.FileInfo) | ||||||
|  | 		extra.URL = durl | ||||||
|  | 		extra.SHA = sha1sum | ||||||
|  | 		msg.Extra["file"][i] = extra | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (gw *Gateway) validGatewayDest(msg *config.Message) bool { | ||||||
|  | 	return msg.Gateway == gw.Name | ||||||
| } | } | ||||||
|  |  | ||||||
| func getChannelID(msg config.Message) string { | func getChannelID(msg config.Message) string { | ||||||
| 	return msg.Channel + msg.Account | 	return msg.Channel + msg.Account | ||||||
| } | } | ||||||
|  |  | ||||||
| func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | func isAPI(account string) bool { | ||||||
| 	return msg.Gateway == gw.Name |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isApi(account string) bool { |  | ||||||
| 	return strings.HasPrefix(account, "api.") | 	return strings.HasPrefix(account, "api.") | ||||||
| } | } | ||||||
|   | |||||||
| @@ -2,15 +2,15 @@ package gateway | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" |  | ||||||
| 	"github.com/BurntSushi/toml" |  | ||||||
| 	"github.com/stretchr/testify/assert" |  | ||||||
| 	"strconv" | 	"strconv" | ||||||
|  |  | ||||||
|  | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
|  | 	"github.com/stretchr/testify/assert" | ||||||
|  |  | ||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var testconfig = ` | var testconfig = []byte(` | ||||||
| [irc.freenode] | [irc.freenode] | ||||||
| [mattermost.test] | [mattermost.test] | ||||||
| [gitter.42wim] | [gitter.42wim] | ||||||
| @@ -37,9 +37,9 @@ var testconfig = ` | |||||||
|     [[gateway.inout]] |     [[gateway.inout]] | ||||||
|     account="slack.test" |     account="slack.test" | ||||||
|     channel="testing" |     channel="testing" | ||||||
| 	` | 	`) | ||||||
|  |  | ||||||
| var testconfig2 = ` | var testconfig2 = []byte(` | ||||||
| [irc.freenode] | [irc.freenode] | ||||||
| [mattermost.test] | [mattermost.test] | ||||||
| [gitter.42wim] | [gitter.42wim] | ||||||
| @@ -80,8 +80,9 @@ var testconfig2 = ` | |||||||
|     [[gateway.out]] |     [[gateway.out]] | ||||||
|     account = "discord.test" |     account = "discord.test" | ||||||
|     channel = "general2" |     channel = "general2" | ||||||
| 	` | 	`) | ||||||
| var testconfig3 = ` |  | ||||||
|  | var testconfig3 = []byte(` | ||||||
| [irc.zzz] | [irc.zzz] | ||||||
| [telegram.zzz] | [telegram.zzz] | ||||||
| [slack.zzz] | [slack.zzz] | ||||||
| @@ -149,13 +150,16 @@ enable=true | |||||||
|     [[gateway.inout]] |     [[gateway.inout]] | ||||||
|     account="telegram.zzz" |     account="telegram.zzz" | ||||||
|     channel="--333333333333" |     channel="--333333333333" | ||||||
| ` | `) | ||||||
|  |  | ||||||
| func maketestRouter(input string) *Router { | const ( | ||||||
| 	var cfg *config.Config | 	ircTestAccount   = "irc.zzz" | ||||||
| 	if _, err := toml.Decode(input, &cfg); err != nil { | 	tgTestAccount    = "telegram.zzz" | ||||||
| 		fmt.Println(err) | 	slackTestAccount = "slack.zzz" | ||||||
| 	} | ) | ||||||
|  |  | ||||||
|  | func maketestRouter(input []byte) *Router { | ||||||
|  | 	cfg := config.NewConfigFromString(input) | ||||||
| 	r, err := NewRouter(cfg) | 	r, err := NewRouter(cfg) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		fmt.Println(err) | 		fmt.Println(err) | ||||||
| @@ -163,14 +167,7 @@ func maketestRouter(input string) *Router { | |||||||
| 	return r | 	return r | ||||||
| } | } | ||||||
| func TestNewRouter(t *testing.T) { | func TestNewRouter(t *testing.T) { | ||||||
| 	var cfg *config.Config | 	r := maketestRouter(testconfig) | ||||||
| 	if _, err := toml.Decode(testconfig, &cfg); err != nil { |  | ||||||
| 		fmt.Println(err) |  | ||||||
| 	} |  | ||||||
| 	r, err := NewRouter(cfg) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Println(err) |  | ||||||
| 	} |  | ||||||
| 	assert.Equal(t, 1, len(r.Gateways)) | 	assert.Equal(t, 1, len(r.Gateways)) | ||||||
| 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges)) | ||||||
| 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
| @@ -181,18 +178,27 @@ func TestNewRouter(t *testing.T) { | |||||||
| 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) | ||||||
| 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||||
| 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) | 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) | ||||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "out", | 	assert.Equal(t, &config.ChannelInfo{ | ||||||
| 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | 		Name:        "42wim/testroom", | ||||||
| 		SameChannel: map[string]bool{"bridge2": false}}, | 		Direction:   "out", | ||||||
| 		r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | 		ID:          "42wim/testroomgitter.42wim", | ||||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "in", | 		Account:     "gitter.42wim", | ||||||
| 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | 		SameChannel: map[string]bool{"bridge2": false}, | ||||||
| 		SameChannel: map[string]bool{"bridge1": false}}, | 	}, r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | ||||||
| 		r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) | 	assert.Equal(t, &config.ChannelInfo{ | ||||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "general", Direction: "inout", | 		Name:        "42wim/testroom", | ||||||
| 		ID: "generaldiscord.test", Account: "discord.test", | 		Direction:   "in", | ||||||
| 		SameChannel: map[string]bool{"bridge1": false}}, | 		ID:          "42wim/testroomgitter.42wim", | ||||||
| 		r.Gateways["bridge1"].Channels["generaldiscord.test"]) | 		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) { | func TestGetDestChannel(t *testing.T) { | ||||||
| @@ -201,11 +207,23 @@ func TestGetDestChannel(t *testing.T) { | |||||||
| 	for _, br := range r.Gateways["bridge1"].Bridges { | 	for _, br := range r.Gateways["bridge1"].Bridges { | ||||||
| 		switch br.Account { | 		switch br.Account { | ||||||
| 		case "discord.test": | 		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: ""}}}, | 			assert.Equal(t, []config.ChannelInfo{{ | ||||||
| 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | 				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": | 		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: ""}}}, | 			assert.Equal(t, []config.ChannelInfo{{ | ||||||
| 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | 				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": | 		case "gitter.42wim": | ||||||
| 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||||
| 		case "irc.freenode": | 		case "irc.freenode": | ||||||
| @@ -235,35 +253,87 @@ func TestGetDestChannelAdvanced(t *testing.T) { | |||||||
| 				} | 				} | ||||||
| 				switch gw.Name { | 				switch gw.Name { | ||||||
| 				case "bridge": | 				case "bridge": | ||||||
| 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz" || msg.Account == "slack.zzz") { | 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && | ||||||
|  | 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount || msg.Account == slackTestAccount) { | ||||||
| 						hits[gw.Name]++ | 						hits[gw.Name]++ | ||||||
| 						switch br.Account { | 						switch br.Account { | ||||||
| 						case "irc.zzz": | 						case ircTestAccount: | ||||||
| 							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) | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
| 						case "telegram.zzz": | 								Name:        "#main", | ||||||
| 							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) | 								Account:     ircTestAccount, | ||||||
| 						case "slack.zzz": | 								Direction:   "inout", | ||||||
| 							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) | 								ID:          "#mainirc.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
|  | 						case tgTestAccount: | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 								Name:        "-1111111111111", | ||||||
|  | 								Account:     tgTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "-1111111111111telegram.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
|  | 						case slackTestAccount: | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 								Name:        "irc", | ||||||
|  | 								Account:     slackTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "ircslack.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				case "bridge2": | 				case "bridge2": | ||||||
| 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && | ||||||
|  | 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount) { | ||||||
| 						hits[gw.Name]++ | 						hits[gw.Name]++ | ||||||
| 						switch br.Account { | 						switch br.Account { | ||||||
| 						case "irc.zzz": | 						case ircTestAccount: | ||||||
| 							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) | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
| 						case "telegram.zzz": | 								Name:        "#main-help", | ||||||
| 							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) | 								Account:     ircTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "#main-helpirc.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge2": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
|  | 						case tgTestAccount: | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 								Name:        "--444444444444", | ||||||
|  | 								Account:     tgTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "--444444444444telegram.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge2": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				case "bridge3": | 				case "bridge3": | ||||||
| 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && | ||||||
|  | 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount) { | ||||||
| 						hits[gw.Name]++ | 						hits[gw.Name]++ | ||||||
| 						switch br.Account { | 						switch br.Account { | ||||||
| 						case "irc.zzz": | 						case ircTestAccount: | ||||||
| 							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) | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
| 						case "telegram.zzz": | 								Name:        "#main-telegram", | ||||||
| 							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) | 								Account:     ircTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "#main-telegramirc.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge3": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
|  | 						case tgTestAccount: | ||||||
|  | 							assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 								Name:        "--333333333333", | ||||||
|  | 								Account:     tgTestAccount, | ||||||
|  | 								Direction:   "inout", | ||||||
|  | 								ID:          "--333333333333telegram.zzz", | ||||||
|  | 								SameChannel: map[string]bool{"bridge3": false}, | ||||||
|  | 								Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 							}}, channels) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				case "announcements": | 				case "announcements": | ||||||
| @@ -273,12 +343,42 @@ func TestGetDestChannelAdvanced(t *testing.T) { | |||||||
| 					} | 					} | ||||||
| 					hits[gw.Name]++ | 					hits[gw.Name]++ | ||||||
| 					switch br.Account { | 					switch br.Account { | ||||||
| 					case "irc.zzz": | 					case ircTestAccount: | ||||||
| 						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) | 						assert.Len(t, channels, 2) | ||||||
| 					case "slack.zzz": | 						assert.Contains(t, channels, config.ChannelInfo{ | ||||||
| 						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) | 							Name:        "#main", | ||||||
| 					case "telegram.zzz": | 							Account:     ircTestAccount, | ||||||
| 						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) | 							Direction:   "out", | ||||||
|  | 							ID:          "#mainirc.zzz", | ||||||
|  | 							SameChannel: map[string]bool{"announcements": false}, | ||||||
|  | 							Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 						}) | ||||||
|  | 						assert.Contains(t, channels, config.ChannelInfo{ | ||||||
|  | 							Name:        "#main-help", | ||||||
|  | 							Account:     ircTestAccount, | ||||||
|  | 							Direction:   "out", | ||||||
|  | 							ID:          "#main-helpirc.zzz", | ||||||
|  | 							SameChannel: map[string]bool{"announcements": false}, | ||||||
|  | 							Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 						}) | ||||||
|  | 					case slackTestAccount: | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 							Name:        "general", | ||||||
|  | 							Account:     slackTestAccount, | ||||||
|  | 							Direction:   "out", | ||||||
|  | 							ID:          "generalslack.zzz", | ||||||
|  | 							SameChannel: map[string]bool{"announcements": false}, | ||||||
|  | 							Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 						}}, channels) | ||||||
|  | 					case tgTestAccount: | ||||||
|  | 						assert.Equal(t, []config.ChannelInfo{{ | ||||||
|  | 							Name:        "--333333333333", | ||||||
|  | 							Account:     tgTestAccount, | ||||||
|  | 							Direction:   "out", | ||||||
|  | 							ID:          "--333333333333telegram.zzz", | ||||||
|  | 							SameChannel: map[string]bool{"announcements": false}, | ||||||
|  | 							Options:     config.ChannelOptions{Key: ""}, | ||||||
|  | 						}}, channels) | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
|   | |||||||
| @@ -2,29 +2,32 @@ package gateway | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/42wim/matterbridge/bridge" | 	"github.com/42wim/matterbridge/bridge" | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/gateway/samechannel" | 	samechannelgateway "github.com/42wim/matterbridge/gateway/samechannel" | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	//	"github.com/davecgh/go-spew/spew" |  | ||||||
| 	"time" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Router struct { | type Router struct { | ||||||
| 	Gateways map[string]*Gateway | 	config.Config | ||||||
| 	Message  chan config.Message |  | ||||||
| 	*config.Config | 	Gateways         map[string]*Gateway | ||||||
|  | 	Message          chan config.Message | ||||||
|  | 	MattermostPlugin chan config.Message | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewRouter(cfg *config.Config) (*Router, error) { | func NewRouter(cfg config.Config) (*Router, error) { | ||||||
| 	r := &Router{} | 	r := &Router{ | ||||||
| 	r.Config = cfg | 		Config:           cfg, | ||||||
| 	r.Message = make(chan config.Message) | 		Message:          make(chan config.Message), | ||||||
| 	r.Gateways = make(map[string]*Gateway) | 		MattermostPlugin: make(chan config.Message), | ||||||
|  | 		Gateways:         make(map[string]*Gateway), | ||||||
|  | 	} | ||||||
| 	sgw := samechannelgateway.New(cfg) | 	sgw := samechannelgateway.New(cfg) | ||||||
| 	gwconfigs := sgw.GetConfig() | 	gwconfigs := sgw.GetConfig() | ||||||
|  |  | ||||||
| 	for _, entry := range append(gwconfigs, cfg.Gateway...) { | 	for _, entry := range append(gwconfigs, cfg.BridgeValues().Gateway...) { | ||||||
| 		if !entry.Enable { | 		if !entry.Enable { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| @@ -42,12 +45,13 @@ func NewRouter(cfg *config.Config) (*Router, error) { | |||||||
| func (r *Router) Start() error { | func (r *Router) Start() error { | ||||||
| 	m := make(map[string]*bridge.Bridge) | 	m := make(map[string]*bridge.Bridge) | ||||||
| 	for _, gw := range r.Gateways { | 	for _, gw := range r.Gateways { | ||||||
|  | 		flog.Infof("Parsing gateway %s", gw.Name) | ||||||
| 		for _, br := range gw.Bridges { | 		for _, br := range gw.Bridges { | ||||||
| 			m[br.Account] = br | 			m[br.Account] = br | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	for _, br := range m { | 	for _, br := range m { | ||||||
| 		log.Infof("Starting bridge: %s ", br.Account) | 		flog.Infof("Starting bridge: %s ", br.Account) | ||||||
| 		err := br.Connect() | 		err := br.Connect() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||||
| @@ -72,7 +76,8 @@ func (r *Router) getBridge(account string) *bridge.Bridge { | |||||||
|  |  | ||||||
| func (r *Router) handleReceive() { | func (r *Router) handleReceive() { | ||||||
| 	for msg := range r.Message { | 	for msg := range r.Message { | ||||||
| 		if msg.Event == config.EVENT_FAILURE { | 		msg := msg // scopelint | ||||||
|  | 		if msg.Event == config.EventFailure { | ||||||
| 		Loop: | 		Loop: | ||||||
| 			for _, gw := range r.Gateways { | 			for _, gw := range r.Gateways { | ||||||
| 				for _, br := range gw.Bridges { | 				for _, br := range gw.Bridges { | ||||||
| @@ -83,7 +88,7 @@ func (r *Router) handleReceive() { | |||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		if msg.Event == config.EVENT_REJOIN_CHANNELS { | 		if msg.Event == config.EventRejoinChannels { | ||||||
| 			for _, gw := range r.Gateways { | 			for _, gw := range r.Gateways { | ||||||
| 				for _, br := range gw.Bridges { | 				for _, br := range gw.Bridges { | ||||||
| 					if msg.Account == br.Account { | 					if msg.Account == br.Account { | ||||||
| @@ -94,11 +99,18 @@ func (r *Router) handleReceive() { | |||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		for _, gw := range r.Gateways { | 		for _, gw := range r.Gateways { | ||||||
|  | 			// record all the message ID's of the different bridges | ||||||
|  | 			var msgIDs []*BrMsgID | ||||||
| 			if !gw.ignoreMessage(&msg) { | 			if !gw.ignoreMessage(&msg) { | ||||||
| 				msg.Timestamp = time.Now() | 				msg.Timestamp = time.Now() | ||||||
| 				gw.modifyMessage(&msg) | 				gw.modifyMessage(&msg) | ||||||
|  | 				gw.handleFiles(&msg) | ||||||
| 				for _, br := range gw.Bridges { | 				for _, br := range gw.Bridges { | ||||||
| 					gw.handleMessage(msg, br) | 					msgIDs = append(msgIDs, gw.handleMessage(msg, br)...) | ||||||
|  | 				} | ||||||
|  | 				// only add the message ID if it doesn't already exists | ||||||
|  | 				if _, ok := gw.Messages.Get(msg.Protocol + " " + msg.ID); !ok && msg.ID != "" { | ||||||
|  | 					gw.Messages.Add(msg.Protocol+" "+msg.ID, msgIDs) | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|   | |||||||
| @@ -5,17 +5,17 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| type SameChannelGateway struct { | type SameChannelGateway struct { | ||||||
| 	*config.Config | 	config.Config | ||||||
| } | } | ||||||
|  |  | ||||||
| func New(cfg *config.Config) *SameChannelGateway { | func New(cfg config.Config) *SameChannelGateway { | ||||||
| 	return &SameChannelGateway{Config: cfg} | 	return &SameChannelGateway{Config: cfg} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (sgw *SameChannelGateway) GetConfig() []config.Gateway { | func (sgw *SameChannelGateway) GetConfig() []config.Gateway { | ||||||
| 	var gwconfigs []config.Gateway | 	var gwconfigs []config.Gateway | ||||||
| 	cfg := sgw.Config | 	cfg := sgw.Config | ||||||
| 	for _, gw := range cfg.SameChannelGateway { | 	for _, gw := range cfg.BridgeValues().SameChannelGateway { | ||||||
| 		gwconfig := config.Gateway{Name: gw.Name, Enable: gw.Enable} | 		gwconfig := config.Gateway{Name: gw.Name, Enable: gw.Enable} | ||||||
| 		for _, account := range gw.Accounts { | 		for _, account := range gw.Accounts { | ||||||
| 			for _, channel := range gw.Channels { | 			for _, channel := range gw.Channels { | ||||||
|   | |||||||
| @@ -1,15 +1,13 @@ | |||||||
| package samechannelgateway | package samechannelgateway | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"fmt" |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/BurntSushi/toml" |  | ||||||
| 	"github.com/stretchr/testify/assert" | 	"github.com/stretchr/testify/assert" | ||||||
|  |  | ||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var testconfig = ` | const testConfig = ` | ||||||
| [mattermost.test] | [mattermost.test] | ||||||
| [slack.test] | [slack.test] | ||||||
|  |  | ||||||
| @@ -20,12 +18,56 @@ var testconfig = ` | |||||||
|       channels = [ "testing","testing2","testing10"] |       channels = [ "testing","testing2","testing10"] | ||||||
| ` | ` | ||||||
|  |  | ||||||
| func TestGetConfig(t *testing.T) { | var ( | ||||||
| 	var cfg *config.Config | 	expectedConfig = config.Gateway{ | ||||||
| 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | 		Name:   "blah", | ||||||
| 		fmt.Println(err) | 		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, | ||||||
|  | 			}, | ||||||
|  | 		}, | ||||||
| 	} | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestGetConfig(t *testing.T) { | ||||||
|  | 	cfg := config.NewConfigFromString([]byte(testConfig)) | ||||||
| 	sgw := New(cfg) | 	sgw := New(cfg) | ||||||
| 	configs := sgw.GetConfig() | 	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) | 	assert.Equal(t, []config.Gateway{expectedConfig}, configs) | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										80
									
								
								go.mod
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,80 @@ | |||||||
|  | module github.com/42wim/matterbridge | ||||||
|  |  | ||||||
|  | require ( | ||||||
|  | 	github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 | ||||||
|  | 	github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 // indirect | ||||||
|  | 	github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 | ||||||
|  | 	github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b // indirect | ||||||
|  | 	github.com/bwmarrin/discordgo v0.19.0 | ||||||
|  | 	github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d | ||||||
|  | 	github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a // indirect | ||||||
|  | 	github.com/fsnotify/fsnotify v1.4.7 | ||||||
|  | 	github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c | ||||||
|  | 	github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc // indirect | ||||||
|  | 	github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c | ||||||
|  | 	github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f // indirect | ||||||
|  | 	github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c | ||||||
|  | 	github.com/gorilla/websocket v1.4.0 | ||||||
|  | 	github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad | ||||||
|  | 	github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb // indirect | ||||||
|  | 	github.com/hpcloud/tail v1.0.0 // indirect | ||||||
|  | 	github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d | ||||||
|  | 	github.com/jtolds/gls v4.2.1+incompatible // indirect | ||||||
|  | 	github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462 // indirect | ||||||
|  | 	github.com/kr/pretty v0.1.0 // indirect | ||||||
|  | 	github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1 | ||||||
|  | 	github.com/labstack/gommon v0.2.1 // indirect | ||||||
|  | 	github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e | ||||||
|  | 	github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 // indirect | ||||||
|  | 	github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 // indirect | ||||||
|  | 	github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 // indirect | ||||||
|  | 	github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 | ||||||
|  | 	github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f | ||||||
|  | 	github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544 | ||||||
|  | 	github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 | ||||||
|  | 	github.com/mattermost/platform v4.6.2+incompatible | ||||||
|  | 	github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597 // indirect | ||||||
|  | 	github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc // indirect | ||||||
|  | 	github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect | ||||||
|  | 	github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 // indirect | ||||||
|  | 	github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 // indirect | ||||||
|  | 	github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff // indirect | ||||||
|  | 	github.com/nicksnyder/go-i18n v1.4.0 // indirect | ||||||
|  | 	github.com/nlopes/slack v0.4.0 | ||||||
|  | 	github.com/onsi/ginkgo v1.6.0 // indirect | ||||||
|  | 	github.com/onsi/gomega v1.4.1 // indirect | ||||||
|  | 	github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83 | ||||||
|  | 	github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606 // indirect | ||||||
|  | 	github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e // indirect | ||||||
|  | 	github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271 | ||||||
|  | 	github.com/pkg/errors v0.8.0 // indirect | ||||||
|  | 	github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a | ||||||
|  | 	github.com/russross/blackfriday v2.0.0+incompatible | ||||||
|  | 	github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca | ||||||
|  | 	github.com/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1 // indirect | ||||||
|  | 	github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991 | ||||||
|  | 	github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95 // indirect | ||||||
|  | 	github.com/sirupsen/logrus v1.2.0 | ||||||
|  | 	github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 // indirect | ||||||
|  | 	github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a // indirect | ||||||
|  | 	github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff // indirect | ||||||
|  | 	github.com/spf13/cast v1.2.0 // indirect | ||||||
|  | 	github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec // indirect | ||||||
|  | 	github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac // indirect | ||||||
|  | 	github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7 | ||||||
|  | 	github.com/stretchr/testify v1.2.2 | ||||||
|  | 	github.com/technoweenie/multipartstreamer v1.0.1 // indirect | ||||||
|  | 	github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a // indirect | ||||||
|  | 	github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4 // indirect | ||||||
|  | 	github.com/x-cray/logrus-prefixed-formatter v0.5.2 // indirect | ||||||
|  | 	github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6 | ||||||
|  | 	golang.org/x/crypto v0.0.0-20181112202954-3d3f9f413869 // indirect | ||||||
|  | 	golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 // indirect | ||||||
|  | 	golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f // indirect | ||||||
|  | 	golang.org/x/sys v0.0.0-20181116161606-93218def8b18 // indirect | ||||||
|  | 	golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 // indirect | ||||||
|  | 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect | ||||||
|  | 	gopkg.in/fsnotify.v1 v1.4.7 // indirect | ||||||
|  | 	gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 // indirect | ||||||
|  | 	gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129 // indirect | ||||||
|  | ) | ||||||
							
								
								
									
										167
									
								
								go.sum
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,167 @@ | |||||||
|  | github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 h1:IZtuWGfzQnKnCSu+vl8WGLhpVQ5Uvy3rlSwqXSg+sQg= | ||||||
|  | github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557/go.mod h1:jL0YSXMs/txjtGJ4PWrmETOk6KUHMDPMshgQZlTeB3Y= | ||||||
|  | github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 h1:v/zr4ns/4sSahF9KBm4Uc933bLsEEv7LuT63CJ019yo= | ||||||
|  | github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= | ||||||
|  | github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 h1:V4+1E1SRYUySqwOoI3ZphFADtabbF568zTHa5ix/zU0= | ||||||
|  | github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3/go.mod h1:HuVM+sZFzumUdKPWiz+IlCMb4RdsKdT3T+nQBKL+sYg= | ||||||
|  | github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b h1:1OpGXps6UOY5HtQaQcLowsV1qMWCNBzhFvK7q4fgXtc= | ||||||
|  | github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b/go.mod h1:iCVmQ9g4TfaRX5m5jq5sXY7RXYWPv9/PynM/GocbG3w= | ||||||
|  | github.com/bwmarrin/discordgo v0.19.0 h1:kMED/DB0NR1QhRcalb85w0Cu3Ep2OrGAqZH1R5awQiY= | ||||||
|  | github.com/bwmarrin/discordgo v0.19.0/go.mod h1:O9S4p+ofTFwB02em7jkpkV8M3R0/PUVOwN61zSZ0r4Q= | ||||||
|  | github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||||||
|  | github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
|  | github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d h1:rONNnZDE5CYuaSFQk+gP4GEQTXEUcyQ5p6p/dgxIHas= | ||||||
|  | github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d/go.mod h1:UGa5M2Sz/Uh13AMse4+RELKCDw7kqgqlTjeGae+7vUY= | ||||||
|  | github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a h1:MuHMeSsXbNEeUyxjB7T9P8s1+5k8OLTC/M27qsVwixM= | ||||||
|  | github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= | ||||||
|  | github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I= | ||||||
|  | github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= | ||||||
|  | github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c h1:3gMh737vMGqAkkkSfNbwjO8VRHOSaCjYRG4y9xVMEIQ= | ||||||
|  | github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM= | ||||||
|  | github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc h1:wdhDSKrkYy24mcfzuA3oYm58h0QkyXjwERCkzJDP5kA= | ||||||
|  | github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | ||||||
|  | github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c h1:MrMA1vhRTNidtgENqmsmLOIUS6ixMBOU/g10rm7IUe8= | ||||||
|  | github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c/go.mod h1:pMQgrscwEK/aUSW1IFSaBPbJX82FPHWaSoJw1axQfD0= | ||||||
|  | github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f h1:FDM3EtwZLyhW48YRiyqjivNlNZjAObv4xt4NnJaU+NQ= | ||||||
|  | github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= | ||||||
|  | github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c h1:mORYpib1aLu3M2Oi50Z1pNTXuDJEHcoLb6oo6VdOutk= | ||||||
|  | github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c/go.mod h1:kgLaKoK1FELgZqMAVxx/5cbj0kT+57qxUrAlIO2eleU= | ||||||
|  | github.com/gorilla/websocket v1.4.0 h1:WDFjx/TMzVgy9VdMMQi2K2Emtwi2QcUQsztZ/zLaH/Q= | ||||||
|  | github.com/gorilla/websocket v1.4.0/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ= | ||||||
|  | github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad h1:eMxs9EL0PvIGS9TTtxg4R+JxuPGav82J8rA+GFnY7po= | ||||||
|  | github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= | ||||||
|  | github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb h1:1OvvPvZkn/yCQ3xBcM8y4020wdkMXPHLB4+NfoGWh4U= | ||||||
|  | github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb/go.mod h1:oZtUIOe8dh44I2q6ScRibXws4Ajl+d+nod3AaR9vL5w= | ||||||
|  | github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI= | ||||||
|  | github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= | ||||||
|  | github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d h1:ETeT81zgLgSNc4BWdDO2Fg9ekVItYErbNtE8mKD2pJA= | ||||||
|  | github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d/go.mod h1:2iMrUgbbvHEiQClaW2NsSzMyGHqN+rDFqY705q49KG0= | ||||||
|  | github.com/jtolds/gls v4.2.1+incompatible h1:fSuqC+Gmlu6l/ZYAoZzx2pyucC8Xza35fpRVWLVmUEE= | ||||||
|  | github.com/jtolds/gls v4.2.1+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= | ||||||
|  | github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462 h1:oSOOTPHkCzMeu1vJ0nHxg5+XZBdMMjNa+6NPnm8arok= | ||||||
|  | github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462/go.mod h1:1NbS8ALrpOvjt0rHPNLyCIeMtbizbir8U//inJ+zuB8= | ||||||
|  | github.com/konsorten/go-windows-terminal-sequences v1.0.1 h1:mweAR1A6xJ3oS2pRaGiHgQ4OO8tzTaLawm8vnODuwDk= | ||||||
|  | github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= | ||||||
|  | github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= | ||||||
|  | github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= | ||||||
|  | github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | ||||||
|  | github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= | ||||||
|  | github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||||
|  | github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1 h1:cOIt0LZKdfeirAfTP4VtIJuWbjVTGtd1suuPXp/J+dE= | ||||||
|  | github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1/go.mod h1:0INS7j/VjnFxD4E2wkz67b8cVwCLbBmJyDaka6Cmk1s= | ||||||
|  | github.com/labstack/gommon v0.2.1 h1:C+I4NYknueQncqKYZQ34kHsLZJVeB5KwPUhnO0nmbpU= | ||||||
|  | github.com/labstack/gommon v0.2.1/go.mod h1:/tj9csK2iPSBvn+3NLM9e52usepMtrd5ilFYA+wQNJ4= | ||||||
|  | github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e h1:RpktB2igr6nS1EN7bCvjldAEfngrM5GyAbmOa4/cafU= | ||||||
|  | github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e/go.mod h1:7cRs1SIBfKQ7e3Tam6GKTILSNHzR862JD0JpINaZoJk= | ||||||
|  | github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 h1:AsEBgzv3DhuYHI/GiQh2HxvTP71HCCE9E/tzGUzGdtU= | ||||||
|  | github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5/go.mod h1:c2mYKRyMb1BPkO5St0c/ps62L4S0W2NAkaTXj9qEI+0= | ||||||
|  | github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 h1:iOAVXzZyXtW408TMYejlUPo6BIn92HmOacWtIfNyYns= | ||||||
|  | github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6/go.mod h1:sFlOUpQL1YcjhFVXhg1CG8ZASEs/Mf1oVb6H75JL/zg= | ||||||
|  | github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 h1:HWxJJvF+QceKcql4r9PC93NtMEgEBfBxlQrZPvbcQvs= | ||||||
|  | github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= | ||||||
|  | github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 h1:KzDEcy8eDbTx881giW8a6llsAck3e2bJvMyKvh1IK+k= | ||||||
|  | github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91/go.mod h1:ECDRehsR9TYTKCAsRS8/wLeOk6UUqDydw47ln7wG41Q= | ||||||
|  | github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f h1:2eKh6Qi/sJ8bXvYMoyVfQxHgR8UcCDWjOmhV1oCstMU= | ||||||
|  | github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f/go.mod h1:+jWeaaUtXQbBRdKYWfjW6JDDYiI2XXE+3NnTjW5kg8g= | ||||||
|  | github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544 h1:A8lLG3DAu75B5jITHs9z4JBmU6oCq1WiUNnDAmqKCZc= | ||||||
|  | github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544/go.mod h1:yAjnZ34DuDyPHMPHHjOsTk/FefW4JJjoMMCGt/8uuQA= | ||||||
|  | github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 h1:R/MgM/eUyRBQx2FiH6JVmXck8PaAuKfe2M1tWIzW7nE= | ||||||
|  | github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61/go.mod h1:iXGEotOvwI1R1SjLxRc+BF5rUORTMtE0iMZBT2lxqAU= | ||||||
|  | github.com/mattermost/platform v4.6.2+incompatible h1:9WqKNuJFIp6SDYn5wl1RF5urdhEw8d7o5tAOwT1MW0A= | ||||||
|  | github.com/mattermost/platform v4.6.2+incompatible/go.mod h1:HjGKtkQNu3HXTOykPMQckMnH11WHvNvQqDBNnVXVbfM= | ||||||
|  | github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597 h1:hGizH4aMDFFt1iOA4HNKC13lqIBoCyxIjWcAnWIy7aU= | ||||||
|  | github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= | ||||||
|  | github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc h1:pK7tzC30erKOTfEDCYGvPZQCkmM9X5iSmmAR5m9x3Yc= | ||||||
|  | github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= | ||||||
|  | github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4= | ||||||
|  | github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE= | ||||||
|  | github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 h1:+MZW2uvHgN8kYvksEN3f7eFL2wpzk0GxmlFsMybWc7E= | ||||||
|  | github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= | ||||||
|  | github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 h1:oKIteTqeSpenyTrOVj5zkiyCaflLa8B+CD0324otT+o= | ||||||
|  | github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474/go.mod h1:OQA4XLvDbMgS8P0CevmM4m9Q3Jq4phKUzcocxuGJ5m8= | ||||||
|  | github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff h1:HLGD5/9UxxfEuO9DtP8gnTmNtMxbPyhYltfxsITel8g= | ||||||
|  | github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff/go.mod h1:B8jLfIIPn2sKyWr0D7cL2v7tnrDD5z291s2Zypdu89E= | ||||||
|  | github.com/nicksnyder/go-i18n v1.4.0 h1:AgLl+Yq7kg5OYlzCgu9cKTZOyI4tD/NgukKqLqC8E+I= | ||||||
|  | github.com/nicksnyder/go-i18n v1.4.0/go.mod h1:HrK7VCrbOvQoUAQ7Vpy7i87N7JZZZ7R2xBGjv0j365Q= | ||||||
|  | github.com/nlopes/slack v0.4.0 h1:OVnHm7lv5gGT5gkcHsZAyw++oHVFihbjWbL3UceUpiA= | ||||||
|  | github.com/nlopes/slack v0.4.0/go.mod h1:jVI4BBK3lSktibKahxBF74txcK2vyvkza1z/+rRnVAM= | ||||||
|  | github.com/onsi/ginkgo v1.6.0 h1:Ix8l273rp3QzYgXSR+c8d1fTG7UPgYkOSELPhiY/YGw= | ||||||
|  | github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= | ||||||
|  | github.com/onsi/gomega v1.4.1 h1:PZSj/UFNaVp3KxrzHOcS7oyuWA7LoOY/77yCTEFu21U= | ||||||
|  | github.com/onsi/gomega v1.4.1/go.mod h1:C1qb7wdrVGGVU+Z6iS04AVkA3Q65CEZX59MT0QO5uiA= | ||||||
|  | github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83 h1:XQonH5Iv5rbyIkMJOQ4xKmKHQTh8viXtRSmep5Ca5I4= | ||||||
|  | github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83/go.mod h1:YnNlZP7l4MhyGQ4CBRwv6ohZTPrUJJZtEv4ZgADkbs4= | ||||||
|  | github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606 h1:/CPgDYrfeK2LMK6xcUhvI17yO9SlpAdDIJGkhDEgO8A= | ||||||
|  | github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34= | ||||||
|  | github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e h1:ZW8599OjioQsmBbkGpyruHUlRVQceYFWnJsGr4NCkiA= | ||||||
|  | github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic= | ||||||
|  | github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271 h1:wQ9lVx75za6AT2kI0S9QID0uWuwTWnvcTfN+uw1F8vg= | ||||||
|  | github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271/go.mod h1:G7LufuPajuIvdt9OitkNt2qh0mmvD4bfRgRM7bhDIOA= | ||||||
|  | github.com/pkg/errors v0.8.0 h1:WdK/asTD0HN+q6hsWO3/vpuAkAr+tw6aNJNDFFf0+qw= | ||||||
|  | github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= | ||||||
|  | github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | ||||||
|  | github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
|  | github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a h1:UWKek6MK3K6/TpbsFcv+8rrO6rSc6KKSp2FbMOHWsq4= | ||||||
|  | github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a/go.mod h1:+uKXf+4Djp6Md1KODXJxgGQPKngRmWyn10oCKFzNHOQ= | ||||||
|  | github.com/russross/blackfriday v2.0.0+incompatible h1:cBXrhZNUf9C+La9/YpS+UHpUT8YD6Td9ZMSU9APFcsk= | ||||||
|  | github.com/russross/blackfriday v2.0.0+incompatible/go.mod h1:JO/DiYxRf+HjHt06OyowR9PTA263kcR/rfWxYHBV53g= | ||||||
|  | github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca h1:NugYot0LIVPxTvN8n+Kvkn6TrbMyxQiuvKdEwFdR9vI= | ||||||
|  | github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca/go.mod h1:uugorj2VCxiV1x+LzaIdVa9b4S4qGAcH6cbhh4qVxOU= | ||||||
|  | github.com/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1 h1:Lx3BlDGFElJt4u/zKc9A3BuGYbQAGlEFyPuUA3jeMD0= | ||||||
|  | github.com/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1/go.mod h1:vt2jWY/3Qw1bIzle5thrJWucsLuuX9iUNnp20CqCciI= | ||||||
|  | github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991 h1:PQiUTDzUC5EUh0vNurK7KQS22zlKqLLOFn+K9nJXDQQ= | ||||||
|  | github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991/go.mod h1:KwtnpMClmrXsHCKTbRui5xBUNt17n1GGrGhdiw2KcoY= | ||||||
|  | github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95 h1:/vdW8Cb7EXrkqWGufVMES1OH2sU9gKVb2n9/1y5NMBY= | ||||||
|  | github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= | ||||||
|  | github.com/sirupsen/logrus v1.2.0 h1:juTguoYk5qI21pwyTXY3B3Y5cOTH3ZUyZCg1v/mihuo= | ||||||
|  | github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= | ||||||
|  | github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 h1:lXQ+j+KwZcbwrbgU0Rp4Eglg3EJLHbuZU3BbOqAGBmg= | ||||||
|  | github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= | ||||||
|  | github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a h1:JSvGDIbmil4Ui/dDdFBExb7/cmkNjyX5F97oglmvCDo= | ||||||
|  | github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s= | ||||||
|  | github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff h1:HLvGWId7M56TfuxTeZ6aoiTAcrWO5Mnq/ArwVRgV62I= | ||||||
|  | github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ= | ||||||
|  | github.com/spf13/cast v1.2.0 h1:HHl1DSRbEQN2i8tJmtS6ViPyHx35+p51amrdsiTCrkg= | ||||||
|  | github.com/spf13/cast v1.2.0/go.mod h1:r2rcYCSwa1IExKTDiTfzaxqT2FNHs8hODu4LnUfgKEg= | ||||||
|  | github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec h1:2ZXvIUGghLpdTVHR1UfvfrzoVlZaE/yOWC5LueIHZig= | ||||||
|  | github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo= | ||||||
|  | github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac h1:+uzyQ0TQ3aKorQxsOjcDDgE7CuUXwpkKnK19LULQALQ= | ||||||
|  | github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= | ||||||
|  | github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7 h1:Wj4cg2M6Um7j1N7yD/mxsdy1/wrsdjzVha2eWdOhti8= | ||||||
|  | github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7/go.mod h1:A8kyI5cUJhb8N+3pkfONlcEcZbueH6nhAm0Fq7SrnBM= | ||||||
|  | github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A= | ||||||
|  | github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||||
|  | github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w= | ||||||
|  | github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= | ||||||
|  | github.com/technoweenie/multipartstreamer v1.0.1 h1:XRztA5MXiR1TIRHxH2uNxXxaIkKQDeX7m2XsSOlQEnM= | ||||||
|  | github.com/technoweenie/multipartstreamer v1.0.1/go.mod h1:jNVxdtShOxzAsukZwTSw6MDx5eUJoiEBsSvzDU9uzog= | ||||||
|  | github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a h1:AOcehBWpFhYPYw0ioDTppQzgI8pAAahVCiMSKTp9rbo= | ||||||
|  | github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= | ||||||
|  | github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4 h1:gKMu1Bf6QINDnvyZuTaACm9ofY+PRh+5vFz4oxBZeF8= | ||||||
|  | github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4/go.mod h1:50wTf68f99/Zt14pr046Tgt3Lp2vLyFZKzbFXTOabXw= | ||||||
|  | github.com/x-cray/logrus-prefixed-formatter v0.5.2 h1:00txxvfBM9muc0jiLIEAkAcIMJzfthRT6usrui8uGmg= | ||||||
|  | github.com/x-cray/logrus-prefixed-formatter v0.5.2/go.mod h1:2duySbKsL6M18s5GU7VPsoEPHyzalCE06qoARUCeBBE= | ||||||
|  | github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6 h1:/WULP+6asFz569UbOwg87f3iDT7T+GF5/vjLmL51Pdk= | ||||||
|  | github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6/go.mod h1:0MsIttMJIF/8Y7x0XjonJP7K99t3sR6bjj4m5S4JmqU= | ||||||
|  | golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||||
|  | golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16 h1:y6ce7gCWtnH+m3dCjzQ1PCuwl28DDIc3VNnvY29DlIA= | ||||||
|  | golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||||
|  | golang.org/x/crypto v0.0.0-20181112202954-3d3f9f413869 h1:kkXA53yGe04D0adEYJwEVQjeBppL01Exg+fnMjfUraU= | ||||||
|  | golang.org/x/crypto v0.0.0-20181112202954-3d3f9f413869/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||||
|  | golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 h1:BkNcmLtAVeWe9h5k0jt24CQgaG5vb4x/doFbAiEC/Ho= | ||||||
|  | golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||||
|  | golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f h1:wMNYb4v58l5UBM7MYRLPG6ZhfOqbKu7X5eyFl8ZhKvA= | ||||||
|  | golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||||
|  | golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||||
|  | golang.org/x/sys v0.0.0-20181116161606-93218def8b18 h1:Wh+XCfg3kNpjhdq2LXrsiOProjtQZKme5XUx7VcxwAw= | ||||||
|  | golang.org/x/sys v0.0.0-20181116161606-93218def8b18/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||||
|  | golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 h1:WNm0tmiuBMW4FJRuXKWOqaQfmKptHs0n8nTCyG0ayjc= | ||||||
|  | golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||||
|  | gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY= | ||||||
|  | gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||||
|  | gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4= | ||||||
|  | gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= | ||||||
|  | gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= | ||||||
|  | gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw= | ||||||
|  | gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129 h1:RBgb9aPUbZ9nu66ecQNIBNsA7j3mB5h8PNDIfhPjaJg= | ||||||
|  | gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74= | ||||||
							
								
								
									
										
											BIN
										
									
								
								img/matterbridge-notext.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 70 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/matterbridge.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 77 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-add-scopes.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 270 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-app-page.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 170 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-create-app.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 282 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-create-bot.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 204 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-finished.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 48 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-install-app.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 73 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-invite-bot.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 62 KiB | 
| @@ -3,24 +3,24 @@ package main | |||||||
| import ( | import ( | ||||||
| 	"flag" | 	"flag" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
| 	"github.com/42wim/matterbridge/bridge/config" | 	"github.com/42wim/matterbridge/bridge/config" | ||||||
| 	"github.com/42wim/matterbridge/gateway" | 	"github.com/42wim/matterbridge/gateway" | ||||||
| 	//"github.com/42wim/matterbridge/gateway/samechannel" |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
| 	"github.com/google/gops/agent" | 	"github.com/google/gops/agent" | ||||||
| 	"strings" | 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	version = "1.0.0-rc1" | 	version = "1.12.1" | ||||||
| 	githash string | 	githash string | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
|  | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true}) | ||||||
|  | 	flog := log.WithFields(log.Fields{"prefix": "main"}) | ||||||
| 	flagConfig := flag.String("conf", "matterbridge.toml", "config file") | 	flagConfig := flag.String("conf", "matterbridge.toml", "config file") | ||||||
| 	flagDebug := flag.Bool("debug", false, "enable debug") | 	flagDebug := flag.Bool("debug", false, "enable debug") | ||||||
| 	flagVersion := flag.Bool("version", false, "show version") | 	flagVersion := flag.Bool("version", false, "show version") | ||||||
| @@ -34,23 +34,25 @@ func main() { | |||||||
| 		fmt.Printf("version: %s %s\n", version, githash) | 		fmt.Printf("version: %s %s\n", version, githash) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	if *flagDebug { | 	if *flagDebug || os.Getenv("DEBUG") == "1" { | ||||||
| 		log.Info("Enabling debug") | 		log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||||
|  | 		flog.Info("Enabling debug") | ||||||
| 		log.SetLevel(log.DebugLevel) | 		log.SetLevel(log.DebugLevel) | ||||||
| 	} | 	} | ||||||
| 	log.Printf("Running version %s %s", version, githash) | 	flog.Printf("Running version %s %s", version, githash) | ||||||
| 	if strings.Contains(version, "-dev") { | 	if strings.Contains(version, "-dev") { | ||||||
| 		log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | 		flog.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||||
| 	} | 	} | ||||||
| 	cfg := config.NewConfig(*flagConfig) | 	cfg := config.NewConfig(*flagConfig) | ||||||
|  | 	cfg.BridgeValues().General.Debug = *flagDebug | ||||||
| 	r, err := gateway.NewRouter(cfg) | 	r, err := gateway.NewRouter(cfg) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Fatalf("Starting gateway failed: %s", err) | 		flog.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	} | 	} | ||||||
| 	err = r.Start() | 	err = r.Start() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Fatalf("Starting gateway failed: %s", err) | 		flog.Fatalf("Starting gateway failed: %s", err) | ||||||
| 	} | 	} | ||||||
| 	log.Printf("Gateway(s) started succesfully. Now relaying messages") | 	flog.Printf("Gateway(s) started succesfully. Now relaying messages") | ||||||
| 	select {} | 	select {} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -6,7 +6,6 @@ | |||||||
|  |  | ||||||
| [mattermost] | [mattermost] | ||||||
|     [mattermost.work] |     [mattermost.work] | ||||||
|     useAPI=true |  | ||||||
|     #do not prefix it wit http:// or https:// |     #do not prefix it wit http:// or https:// | ||||||
|     Server="yourmattermostserver.domain"  |     Server="yourmattermostserver.domain"  | ||||||
|     Team="yourteam" |     Team="yourteam" | ||||||
|   | |||||||
| @@ -9,23 +9,24 @@ import ( | |||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"net/http/cookiejar" | 	"net/http/cookiejar" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 	"strconv" |  | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	log "github.com/Sirupsen/logrus" |  | ||||||
|  |  | ||||||
| 	"github.com/gorilla/websocket" | 	"github.com/gorilla/websocket" | ||||||
| 	"github.com/hashicorp/golang-lru" | 	"github.com/hashicorp/golang-lru" | ||||||
| 	"github.com/jpillora/backoff" | 	"github.com/jpillora/backoff" | ||||||
|  | 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||||
| 	"github.com/mattermost/platform/model" | 	"github.com/mattermost/platform/model" | ||||||
|  | 	log "github.com/sirupsen/logrus" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Credentials struct { | type Credentials struct { | ||||||
| 	Login         string | 	Login         string | ||||||
| 	Team          string | 	Team          string | ||||||
| 	Pass          string | 	Pass          string | ||||||
|  | 	Token         string | ||||||
|  | 	CookieToken   bool | ||||||
| 	Server        string | 	Server        string | ||||||
| 	NoTLS         bool | 	NoTLS         bool | ||||||
| 	SkipTLSVerify bool | 	SkipTLSVerify bool | ||||||
| @@ -42,11 +43,12 @@ type Message struct { | |||||||
| 	UserID   string | 	UserID   string | ||||||
| } | } | ||||||
|  |  | ||||||
|  | //nolint:golint | ||||||
| type Team struct { | type Team struct { | ||||||
| 	Team         *model.Team | 	Team         *model.Team | ||||||
| 	Id           string | 	Id           string | ||||||
| 	Channels     *model.ChannelList | 	Channels     []*model.Channel | ||||||
| 	MoreChannels *model.ChannelList | 	MoreChannels []*model.Channel | ||||||
| 	Users        map[string]*model.User | 	Users        map[string]*model.User | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -55,7 +57,7 @@ type MMClient struct { | |||||||
| 	*Credentials | 	*Credentials | ||||||
| 	Team          *Team | 	Team          *Team | ||||||
| 	OtherTeams    []*Team | 	OtherTeams    []*Team | ||||||
| 	Client        *model.Client | 	Client        *model.Client4 | ||||||
| 	User          *model.User | 	User          *model.User | ||||||
| 	Users         map[string]*model.User | 	Users         map[string]*model.User | ||||||
| 	MessageChan   chan *Message | 	MessageChan   chan *Message | ||||||
| @@ -74,12 +76,16 @@ type MMClient struct { | |||||||
| func New(login, pass, team, server string) *MMClient { | func New(login, pass, team, server string) *MMClient { | ||||||
| 	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server} | 	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server} | ||||||
| 	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | 	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | ||||||
| 	mmclient.log = log.WithFields(log.Fields{"module": "matterclient"}) | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true}) | ||||||
| 	log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) | 	mmclient.log = log.WithFields(log.Fields{"prefix": "matterclient"}) | ||||||
| 	mmclient.lruCache, _ = lru.New(500) | 	mmclient.lruCache, _ = lru.New(500) | ||||||
| 	return mmclient | 	return mmclient | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) SetDebugLog() { | ||||||
|  | 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) SetLogLevel(level string) { | func (m *MMClient) SetLogLevel(level string) { | ||||||
| 	l, err := log.ParseLevel(level) | 	l, err := log.ParseLevel(level) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -109,21 +115,38 @@ func (m *MMClient) Login() error { | |||||||
| 		uriScheme = "http://" | 		uriScheme = "http://" | ||||||
| 	} | 	} | ||||||
| 	// login to mattermost | 	// login to mattermost | ||||||
| 	m.Client = model.NewClient(uriScheme + m.Credentials.Server) | 	m.Client = model.NewAPIv4Client(uriScheme + m.Credentials.Server) | ||||||
| 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | ||||||
| 	m.Client.HttpClient.Timeout = time.Second * 10 | 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||||
|  |  | ||||||
|  | 	if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||||
|  | 		token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||||
|  | 		if len(token) != 2 { | ||||||
|  | 			return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||||
|  | 		} | ||||||
|  | 		m.Credentials.Token = token[1] | ||||||
|  | 		m.Credentials.CookieToken = true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if strings.Contains(m.Credentials.Pass, "token=") { | ||||||
|  | 		token := strings.Split(m.Credentials.Pass, "token=") | ||||||
|  | 		if len(token) != 2 { | ||||||
|  | 			return errors.New("incorrect personal token. valid input is token=yourtoken") | ||||||
|  | 		} | ||||||
|  | 		m.Credentials.Token = token[1] | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
| 		d := b.Duration() | 		d := b.Duration() | ||||||
| 		// bogus call to get the serverversion | 		// bogus call to get the serverversion | ||||||
| 		_, err := m.Client.GetClientProperties() | 		_, resp := m.Client.Logout() | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return fmt.Errorf("%#v", err.Error()) | 			return fmt.Errorf("%#v", resp.Error.Error()) | ||||||
| 		} | 		} | ||||||
| 		if firstConnection && !supportedVersion(m.Client.ServerVersion) { | 		if firstConnection && !supportedVersion(resp.ServerVersion) { | ||||||
| 			return fmt.Errorf("unsupported mattermost version: %s", m.Client.ServerVersion) | 			return fmt.Errorf("unsupported mattermost version: %s", resp.ServerVersion) | ||||||
| 		} | 		} | ||||||
| 		m.ServerVersion = m.Client.ServerVersion | 		m.ServerVersion = resp.ServerVersion | ||||||
| 		if m.ServerVersion == "" { | 		if m.ServerVersion == "" { | ||||||
| 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||||
| 			time.Sleep(d) | 			time.Sleep(d) | ||||||
| @@ -134,30 +157,33 @@ func (m *MMClient) Login() error { | |||||||
| 	} | 	} | ||||||
| 	b.Reset() | 	b.Reset() | ||||||
|  |  | ||||||
| 	var myinfo *model.Result | 	var resp *model.Response | ||||||
|  | 	//var myinfo *model.Result | ||||||
| 	var appErr *model.AppError | 	var appErr *model.AppError | ||||||
| 	var logmsg = "trying login" | 	var logmsg = "trying login" | ||||||
| 	for { | 	for { | ||||||
| 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | ||||||
| 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | 		if m.Credentials.Token != "" { | ||||||
| 			m.log.Debugf(logmsg+" with %s", model.SESSION_COOKIE_TOKEN) | 			m.Client.AuthType = model.HEADER_BEARER | ||||||
| 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | 			m.Client.AuthToken = m.Credentials.Token | ||||||
| 			if len(token) != 2 { | 			if m.Credentials.CookieToken { | ||||||
| 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | 				m.log.Debugf(logmsg + " with cookie (MMAUTH) token") | ||||||
|  | 				m.Client.HttpClient.Jar = m.createCookieJar(m.Credentials.Token) | ||||||
|  | 			} else { | ||||||
|  | 				m.log.Debugf(logmsg + " with personal token") | ||||||
| 			} | 			} | ||||||
| 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | 			m.User, resp = m.Client.GetMe("") | ||||||
| 			m.Client.MockSession(token[1]) | 			if resp.Error != nil { | ||||||
| 			myinfo, appErr = m.Client.GetMe("") | 				return resp.Error | ||||||
| 			if appErr != nil { |  | ||||||
| 				return errors.New(appErr.DetailedError) |  | ||||||
| 			} | 			} | ||||||
| 			if myinfo.Data.(*model.User) == nil { | 			if m.User == nil { | ||||||
| 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | ||||||
| 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | 				return errors.New("invalid token") | ||||||
| 			} | 			} | ||||||
| 		} else { | 		} else { | ||||||
| 			_, appErr = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | 			m.User, resp = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | ||||||
| 		} | 		} | ||||||
|  | 		appErr = resp.Error | ||||||
| 		if appErr != nil { | 		if appErr != nil { | ||||||
| 			d := b.Duration() | 			d := b.Duration() | ||||||
| 			m.log.Debug(appErr.DetailedError) | 			m.log.Debug(appErr.DetailedError) | ||||||
| @@ -183,10 +209,12 @@ func (m *MMClient) Login() error { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if m.Team == nil { | 	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() | 	m.wsConnect() | ||||||
|  |  | ||||||
| @@ -207,7 +235,7 @@ func (m *MMClient) wsConnect() { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// setup websocket connection | 	// setup websocket connection | ||||||
| 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V3 + "/users/websocket" | 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V4 + "/websocket" | ||||||
| 	header := http.Header{} | 	header := http.Header{} | ||||||
| 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||||
|  |  | ||||||
| @@ -241,9 +269,9 @@ func (m *MMClient) Logout() error { | |||||||
| 		m.log.Debug("Not invalidating session in logout, credential is a token") | 		m.log.Debug("Not invalidating session in logout, credential is a token") | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	_, err := m.Client.Logout() | 	_, resp := m.Client.Logout() | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return err | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -277,6 +305,13 @@ func (m *MMClient) WsReceiver() { | |||||||
| 			// check if we didn't empty the message | 			// check if we didn't empty the message | ||||||
| 			if msg.Text != "" { | 			if msg.Text != "" { | ||||||
| 				m.MessageChan <- msg | 				m.MessageChan <- msg | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// if we have file attached but the message is empty, also send it | ||||||
|  | 			if msg.Post != nil { | ||||||
|  | 				if msg.Text != "" || len(msg.Post.FileIds) > 0 || msg.Post.Type == "slack_attachment" { | ||||||
|  | 					m.MessageChan <- msg | ||||||
|  | 				} | ||||||
| 			} | 			} | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| @@ -292,8 +327,15 @@ func (m *MMClient) WsReceiver() { | |||||||
|  |  | ||||||
| func (m *MMClient) parseMessage(rmsg *Message) { | func (m *MMClient) parseMessage(rmsg *Message) { | ||||||
| 	switch rmsg.Raw.Event { | 	switch rmsg.Raw.Event { | ||||||
| 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED: | 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED, model.WEBSOCKET_EVENT_POST_DELETED: | ||||||
| 		m.parseActionPost(rmsg) | 		m.parseActionPost(rmsg) | ||||||
|  | 	case "user_updated": | ||||||
|  | 		user := rmsg.Raw.Data["user"].(map[string]interface{}) | ||||||
|  | 		if _, ok := user["id"].(string); ok { | ||||||
|  | 			m.UpdateUser(user["id"].(string)) | ||||||
|  | 		} | ||||||
|  | 	case "group_added": | ||||||
|  | 		m.UpdateChannels() | ||||||
| 		/* | 		/* | ||||||
| 			case model.ACTION_USER_REMOVED: | 			case model.ACTION_USER_REMOVED: | ||||||
| 				m.handleWsActionUserRemoved(&rmsg) | 				m.handleWsActionUserRemoved(&rmsg) | ||||||
| @@ -323,7 +365,8 @@ func (m *MMClient) parseActionPost(rmsg *Message) { | |||||||
| 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | ||||||
| 	// we don't have the user, refresh the userlist | 	// we don't have the user, refresh the userlist | ||||||
| 	if m.GetUser(data.UserId) == nil { | 	if m.GetUser(data.UserId) == nil { | ||||||
| 		m.log.Infof("User %s is not known, ignoring message %s", data) | 		m.log.Infof("User '%v' is not known, ignoring message '%#v'", | ||||||
|  | 			data.UserId, data) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	rmsg.Username = m.GetUserName(data.UserId) | 	rmsg.Username = m.GetUserName(data.UserId) | ||||||
| @@ -349,38 +392,39 @@ func (m *MMClient) parseActionPost(rmsg *Message) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateUsers() error { | func (m *MMClient) UpdateUsers() error { | ||||||
| 	mmusers, err := m.Client.GetProfiles(0, 50000, "") | 	mmusers, resp := m.Client.GetUsers(0, 50000, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Users = mmusers.Data.(map[string]*model.User) | 	for _, user := range mmusers { | ||||||
|  | 		m.Users[user.Id] = user | ||||||
|  | 	} | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannels() error { | func (m *MMClient) UpdateChannels() error { | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New(err.DetailedError) | 		return errors.New(resp.Error.DetailedError) | ||||||
| 	} |  | ||||||
| 	var mmchannels2 *model.Result |  | ||||||
| 	if m.mmVersion() >= 3.08 { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannelsPage(0, 5000) |  | ||||||
| 	} else { |  | ||||||
| 		mmchannels2, err = m.Client.GetMoreChannels("") |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return errors.New(err.DetailedError) |  | ||||||
| 	} | 	} | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) | 	m.Team.Channels = mmchannels | ||||||
| 	m.Team.MoreChannels = mmchannels2.Data.(*model.ChannelList) | 	m.Unlock() | ||||||
|  |  | ||||||
|  | 	mmchannels, resp = m.Client.GetPublicChannelsForTeam(m.Team.Id, 0, 5000, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return errors.New(resp.Error.DetailedError) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	m.Team.MoreChannels = mmchannels | ||||||
| 	m.Unlock() | 	m.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetChannelName(channelId string) string { | func (m *MMClient) GetChannelName(channelId string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| @@ -388,15 +432,25 @@ func (m *MMClient) GetChannelName(channelId string) string { | |||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		if t.Channels != nil { | 		if t.Channels != nil { | ||||||
| 			for _, channel := range *t.Channels { | 			for _, channel := range t.Channels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
|  | 					if channel.Type == model.CHANNEL_GROUP { | ||||||
|  | 						res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||||
|  | 						res = strings.Replace(res, " ", "_", -1) | ||||||
|  | 						return res | ||||||
|  | 					} | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			for _, channel := range *t.MoreChannels { | 			for _, channel := range t.MoreChannels { | ||||||
| 				if channel.Id == channelId { | 				if channel.Id == channelId { | ||||||
|  | 					if channel.Type == model.CHANNEL_GROUP { | ||||||
|  | 						res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||||
|  | 						res = strings.Replace(res, " ", "_", -1) | ||||||
|  | 						return res | ||||||
|  | 					} | ||||||
| 					return channel.Name | 					return channel.Name | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| @@ -405,15 +459,27 @@ func (m *MMClient) GetChannelName(channelId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetChannelId(name string, teamId string) string { | func (m *MMClient) GetChannelId(name string, teamId string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	if teamId == "" { | 	if teamId == "" { | ||||||
| 		teamId = m.Team.Id | 		for _, t := range m.OtherTeams { | ||||||
|  | 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
|  | 				if channel.Type == model.CHANNEL_GROUP { | ||||||
|  | 					res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||||
|  | 					res = strings.Replace(res, " ", "_", -1) | ||||||
|  | 					if res == name { | ||||||
|  | 						return channel.Id | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id == teamId { | 		if t.Id == teamId { | ||||||
| 			for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 				if channel.Name == name { | 				if channel.Name == name { | ||||||
| 					return channel.Id | 					return channel.Id | ||||||
| 				} | 				} | ||||||
| @@ -423,11 +489,11 @@ func (m *MMClient) GetChannelId(name string, teamId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetChannelTeamId(id string) string { | func (m *MMClient) GetChannelTeamId(id string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range append(m.OtherTeams, m.Team) { | 	for _, t := range append(m.OtherTeams, m.Team) { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == id { | 			if channel.Id == id { | ||||||
| 				return channel.TeamId | 				return channel.TeamId | ||||||
| 			} | 			} | ||||||
| @@ -436,11 +502,11 @@ func (m *MMClient) GetChannelTeamId(id string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetChannelHeader(channelId string) string { | func (m *MMClient) GetChannelHeader(channelId string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		for _, channel := range append(*t.Channels, *t.MoreChannels...) { | 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||||
| 			if channel.Id == channelId { | 			if channel.Id == channelId { | ||||||
| 				return channel.Header | 				return channel.Header | ||||||
| 			} | 			} | ||||||
| @@ -450,55 +516,85 @@ func (m *MMClient) GetChannelHeader(channelId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) PostMessage(channelId string, text string) { | func (m *MMClient) PostMessage(channelId string, text string) (string, error) { //nolint:golint | ||||||
| 	post := &model.Post{ChannelId: channelId, Message: text} | 	post := &model.Post{ChannelId: channelId, Message: text} | ||||||
| 	m.Client.CreatePost(post) | 	res, resp := m.Client.CreatePost(post) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return res.Id, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) JoinChannel(channelId string) error { | func (m *MMClient) PostMessageWithFiles(channelId string, text string, fileIds []string) (string, error) { //nolint:golint | ||||||
|  | 	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) { //nolint:golint | ||||||
|  | 	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 { //nolint:golint | ||||||
|  | 	_, resp := m.Client.DeletePost(postId) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) JoinChannel(channelId string) error { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, c := range *m.Team.Channels { | 	for _, c := range m.Team.Channels { | ||||||
| 		if c.Id == channelId { | 		if c.Id == channelId { | ||||||
| 			m.log.Debug("Not joining ", channelId, " already joined.") | 			m.log.Debug("Not joining ", channelId, " already joined.") | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	m.log.Debug("Joining ", channelId) | 	m.log.Debug("Joining ", channelId) | ||||||
| 	_, err := m.Client.JoinChannel(channelId) | 	_, resp := m.Client.AddChannelMember(channelId, m.User.Id) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return errors.New("failed to join") | 		return resp.Error | ||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { //nolint:golint | ||||||
| 	res, err := m.Client.GetPostsSince(channelId, time) | 	res, resp := m.Client.GetPostsSince(channelId, time) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) SearchPosts(query string) *model.PostList { | func (m *MMClient) SearchPosts(query string) *model.PostList { | ||||||
| 	res, err := m.Client.SearchPosts(query, false) | 	res, resp := m.Client.SearchPosts(m.Team.Id, query, false) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { //nolint:golint | ||||||
| 	res, err := m.Client.GetPosts(channelId, 0, limit, "") | 	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 	return res.Data.(*model.PostList) | 	return res | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetPublicLink(filename string) string { | func (m *MMClient) GetPublicLink(filename string) string { | ||||||
| 	res, err := m.Client.GetPublicLink(filename) | 	res, resp := m.Client.GetFileLink(filename) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	return res | 	return res | ||||||
| @@ -507,8 +603,8 @@ func (m *MMClient) GetPublicLink(filename string) string { | |||||||
| func (m *MMClient) GetPublicLinks(filenames []string) []string { | func (m *MMClient) GetPublicLinks(filenames []string) []string { | ||||||
| 	var output []string | 	var output []string | ||||||
| 	for _, f := range filenames { | 	for _, f := range filenames { | ||||||
| 		res, err := m.Client.GetPublicLink(f) | 		res, resp := m.Client.GetFileLink(f) | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		output = append(output, res) | 		output = append(output, res) | ||||||
| @@ -524,10 +620,10 @@ func (m *MMClient) GetFileLinks(filenames []string) []string { | |||||||
|  |  | ||||||
| 	var output []string | 	var output []string | ||||||
| 	for _, f := range filenames { | 	for _, f := range filenames { | ||||||
| 		res, err := m.Client.GetPublicLink(f) | 		res, resp := m.Client.GetFileLink(f) | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			// public links is probably disabled, create the link ourselves | 			// public links is probably disabled, create the link ourselves | ||||||
| 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V3+"/files/"+f+"/get") | 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V4+"/files/"+f) | ||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
| 		output = append(output, res) | 		output = append(output, res) | ||||||
| @@ -535,43 +631,46 @@ func (m *MMClient) GetFileLinks(filenames []string) []string { | |||||||
| 	return output | 	return output | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | func (m *MMClient) UpdateChannelHeader(channelId string, header string) { //nolint:golint | ||||||
| 	data := make(map[string]string) | 	channel := &model.Channel{Id: channelId, Header: header} | ||||||
| 	data["channel_id"] = channelId |  | ||||||
| 	data["channel_header"] = header |  | ||||||
| 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||||
| 	_, err := m.Client.UpdateChannelHeader(data) | 	_, resp := m.Client.UpdateChannel(channel) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		log.Error(err) | 		log.Error(resp.Error) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UpdateLastViewed(channelId string) { | func (m *MMClient) UpdateLastViewed(channelId string) error { //nolint:golint | ||||||
| 	m.log.Debugf("posting lastview %#v", channelId) | 	m.log.Debugf("posting lastview %#v", channelId) | ||||||
| 	if m.mmVersion() >= 3.08 { | 	view := &model.ChannelView{ChannelId: channelId} | ||||||
| 		view := model.ChannelView{ChannelId: channelId} | 	_, resp := m.Client.ViewChannel(m.User.Id, view) | ||||||
| 		res, _ := m.Client.ViewChannel(view) | 	if resp.Error != nil { | ||||||
| 		if !res { | 		m.log.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error) | ||||||
| 			m.log.Errorf("ChannelView update for %s failed", channelId) | 		return resp.Error | ||||||
| 		} |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	_, err := m.Client.UpdateLastViewedAt(channelId, true) |  | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Error(err) |  | ||||||
| 	} | 	} | ||||||
|  | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) UsernamesInChannel(channelId string) []string { | func (m *MMClient) UpdateUserNick(nick string) error { | ||||||
| 	res, err := m.Client.GetProfilesInChannel(channelId, 0, 50000, "") | 	user := m.User | ||||||
| 	if err != nil { | 	user.Nickname = nick | ||||||
| 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, err) | 	_, resp := m.Client.UpdateUser(user) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UsernamesInChannel(channelId string) []string { //nolint:golint | ||||||
|  | 	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error) | ||||||
| 		return []string{} | 		return []string{} | ||||||
| 	} | 	} | ||||||
| 	members := res.Data.(map[string]*model.User) | 	allusers := m.GetUsers() | ||||||
| 	result := []string{} | 	result := []string{} | ||||||
| 	for _, member := range members { | 	for _, member := range *res { | ||||||
| 		result = append(result, member.Nickname) | 		result = append(result, allusers[member.UserId].Nickname) | ||||||
| 	} | 	} | ||||||
| 	return result | 	return result | ||||||
| } | } | ||||||
| @@ -592,34 +691,31 @@ func (m *MMClient) createCookieJar(token string) *cookiejar.Jar { | |||||||
| } | } | ||||||
|  |  | ||||||
| // SendDirectMessage sends a direct message to specified user | // SendDirectMessage sends a direct message to specified user | ||||||
| func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | func (m *MMClient) SendDirectMessage(toUserId string, msg string) { //nolint:golint | ||||||
|  | 	m.SendDirectMessageProps(toUserId, msg, nil) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) SendDirectMessageProps(toUserId string, msg string, props map[string]interface{}) { //nolint:golint | ||||||
| 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||||
| 	// create DM channel (only happens on first message) | 	// create DM channel (only happens on first message) | ||||||
| 	_, err := m.Client.CreateDirectChannel(toUserId) | 	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId) | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, err) | 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
| 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||||
|  |  | ||||||
| 	// update our channels | 	// update our channels | ||||||
| 	mmchannels, err := m.Client.GetChannels("") | 	m.UpdateChannels() | ||||||
| 	if err != nil { |  | ||||||
| 		m.log.Debug("SendDirectMessage: Couldn't update channels") |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.Team.Channels = mmchannels.Data.(*model.ChannelList) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	// build & send the message | 	// build & send the message | ||||||
| 	msg = strings.Replace(msg, "\r", "", -1) | 	msg = strings.Replace(msg, "\r", "", -1) | ||||||
| 	post := &model.Post{ChannelId: m.GetChannelId(channelName, ""), Message: msg} | 	post := &model.Post{ChannelId: m.GetChannelId(channelName, m.Team.Id), Message: msg, Props: props} | ||||||
| 	m.Client.CreatePost(post) | 	m.Client.CreatePost(post) | ||||||
| } | } | ||||||
|  |  | ||||||
| // GetTeamName returns the name of the specified teamId | // GetTeamName returns the name of the specified teamId | ||||||
| func (m *MMClient) GetTeamName(teamId string) string { | func (m *MMClient) GetTeamName(teamId string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| @@ -636,10 +732,10 @@ func (m *MMClient) GetChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	// our primary team channels first | 	// our primary team channels first | ||||||
| 	channels = append(channels, *m.Team.Channels...) | 	channels = append(channels, m.Team.Channels...) | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		if t.Id != m.Team.Id { | 		if t.Id != m.Team.Id { | ||||||
| 			channels = append(channels, *t.Channels...) | 			channels = append(channels, t.Channels...) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| @@ -651,39 +747,42 @@ func (m *MMClient) GetMoreChannels() []*model.Channel { | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.MoreChannels...) | 		channels = append(channels, t.MoreChannels...) | ||||||
| 	} | 	} | ||||||
| 	return channels | 	return channels | ||||||
| } | } | ||||||
|  |  | ||||||
| // GetTeamFromChannel returns teamId belonging to channel (DM channels have no teamId). | // GetTeamFromChannel returns teamId belonging to channel (DM channels have no teamId). | ||||||
| func (m *MMClient) GetTeamFromChannel(channelId string) string { | func (m *MMClient) GetTeamFromChannel(channelId string) string { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	var channels []*model.Channel | 	var channels []*model.Channel | ||||||
| 	for _, t := range m.OtherTeams { | 	for _, t := range m.OtherTeams { | ||||||
| 		channels = append(channels, *t.Channels...) | 		channels = append(channels, t.Channels...) | ||||||
| 		if t.MoreChannels != nil { | 		if t.MoreChannels != nil { | ||||||
| 			channels = append(channels, *t.MoreChannels...) | 			channels = append(channels, t.MoreChannels...) | ||||||
| 		} | 		} | ||||||
| 		for _, c := range channels { | 		for _, c := range channels { | ||||||
| 			if c.Id == channelId { | 			if c.Id == channelId { | ||||||
|  | 				if c.Type == model.CHANNEL_GROUP { | ||||||
|  | 					return "G" | ||||||
|  | 				} | ||||||
| 				return t.Id | 				return t.Id | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | 		channels = nil | ||||||
| 	} | 	} | ||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetLastViewedAt(channelId string) int64 { | func (m *MMClient) GetLastViewedAt(channelId string) int64 { //nolint:golint | ||||||
| 	m.RLock() | 	m.RLock() | ||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
| 	res, err := m.Client.GetChannel(channelId, "") | 	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "") | ||||||
| 	if err != nil { | 	if resp.Error != nil { | ||||||
| 		return model.GetMillis() | 		return model.GetMillis() | ||||||
| 	} | 	} | ||||||
| 	data := res.Data.(*model.ChannelData) | 	return res.LastViewedAt | ||||||
| 	return data.Member.LastViewedAt |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetUsers() map[string]*model.User { | func (m *MMClient) GetUsers() map[string]*model.User { | ||||||
| @@ -696,22 +795,31 @@ func (m *MMClient) GetUsers() map[string]*model.User { | |||||||
| 	return users | 	return users | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetUser(userId string) *model.User { | func (m *MMClient) GetUser(userId string) *model.User { //nolint:golint | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	_, ok := m.Users[userId] | 	_, ok := m.Users[userId] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		res, err := m.Client.GetProfilesByIds([]string{userId}) | 		res, resp := m.Client.GetUser(userId, "") | ||||||
| 		if err != nil { | 		if resp.Error != nil { | ||||||
| 			return nil | 			return nil | ||||||
| 		} | 		} | ||||||
| 		u := res.Data.(map[string]*model.User)[userId] | 		m.Users[userId] = res | ||||||
| 		m.Users[userId] = u |  | ||||||
| 	} | 	} | ||||||
| 	return m.Users[userId] | 	return m.Users[userId] | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetUserName(userId string) string { | func (m *MMClient) UpdateUser(userId string) { //nolint:golint | ||||||
|  | 	m.Lock() | ||||||
|  | 	defer m.Unlock() | ||||||
|  | 	res, resp := m.Client.GetUser(userId, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	m.Users[userId] = res | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) GetUserName(userId string) string { //nolint:golint | ||||||
| 	user := m.GetUser(userId) | 	user := m.GetUser(userId) | ||||||
| 	if user != nil { | 	if user != nil { | ||||||
| 		return user.Username | 		return user.Username | ||||||
| @@ -719,69 +827,97 @@ func (m *MMClient) GetUserName(userId string) string { | |||||||
| 	return "" | 	return "" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetStatus(userId string) string { | func (m *MMClient) GetNickName(userId string) string { //nolint:golint | ||||||
| 	res, err := m.Client.GetStatusesByIds([]string{userId}) | 	user := m.GetUser(userId) | ||||||
| 	if err != nil { | 	if user != nil { | ||||||
|  | 		return user.Nickname | ||||||
|  | 	} | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) GetStatus(userId string) string { //nolint:golint | ||||||
|  | 	res, resp := m.Client.GetUserStatus(userId, "") | ||||||
|  | 	if resp.Error != nil { | ||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| 	status := res.Data.(map[string]string) | 	if res.Status == model.STATUS_AWAY { | ||||||
| 	if status[userId] == model.STATUS_AWAY { |  | ||||||
| 		return "away" | 		return "away" | ||||||
| 	} | 	} | ||||||
| 	if status[userId] == model.STATUS_ONLINE { | 	if res.Status == model.STATUS_ONLINE { | ||||||
| 		return "online" | 		return "online" | ||||||
| 	} | 	} | ||||||
| 	return "offline" | 	return "offline" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UpdateStatus(userId string, status string) error { //nolint:golint | ||||||
|  | 	_, 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 { | func (m *MMClient) GetStatuses() map[string]string { | ||||||
| 	var ok bool | 	var ids []string | ||||||
| 	statuses := make(map[string]string) | 	statuses := make(map[string]string) | ||||||
| 	res, err := m.Client.GetStatuses() | 	for id := range m.Users { | ||||||
| 	if err != nil { | 		ids = append(ids, id) | ||||||
|  | 	} | ||||||
|  | 	res, resp := m.Client.GetUsersStatusesByIds(ids) | ||||||
|  | 	if resp.Error != nil { | ||||||
| 		return statuses | 		return statuses | ||||||
| 	} | 	} | ||||||
| 	if statuses, ok = res.Data.(map[string]string); ok { | 	for _, status := range res { | ||||||
| 		for userId, status := range statuses { | 		statuses[status.UserId] = "offline" | ||||||
| 			statuses[userId] = "offline" | 		if status.Status == model.STATUS_AWAY { | ||||||
| 			if status == model.STATUS_AWAY { | 			statuses[status.UserId] = "away" | ||||||
| 				statuses[userId] = "away" | 		} | ||||||
| 			} | 		if status.Status == model.STATUS_ONLINE { | ||||||
| 			if status == model.STATUS_ONLINE { | 			statuses[status.UserId] = "online" | ||||||
| 				statuses[userId] = "online" |  | ||||||
| 			} |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	return statuses | 	return statuses | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) GetTeamId() string { | func (m *MMClient) GetTeamId() string { //nolint:golint | ||||||
| 	return m.Team.Id | 	return m.Team.Id | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { //nolint:golint | ||||||
|  | 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return "", resp.Error | ||||||
|  | 	} | ||||||
|  | 	return f.FileInfos[0].Id, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) StatusLoop() { | func (m *MMClient) StatusLoop() { | ||||||
| 	retries := 0 | 	retries := 0 | ||||||
| 	backoff := time.Second * 60 | 	backoff := time.Second * 60 | ||||||
| 	if m.OnWsConnect != nil { | 	if m.OnWsConnect != nil { | ||||||
| 		m.OnWsConnect() | 		m.OnWsConnect() | ||||||
| 	} | 	} | ||||||
| 	m.log.Debug("StatusLoop:", m.OnWsConnect) | 	m.log.Debug("StatusLoop:", m.OnWsConnect != nil) | ||||||
| 	for { | 	for { | ||||||
| 		if m.WsQuit { | 		if m.WsQuit { | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		if m.WsConnected { | 		if m.WsConnected { | ||||||
| 			m.log.Debug("WS PING") | 			m.checkAlive() | ||||||
| 			m.sendWSRequest("ping", nil) |  | ||||||
| 			select { | 			select { | ||||||
| 			case <-m.WsPingChan: | 			case <-m.WsPingChan: | ||||||
| 				m.log.Debug("WS PONG received") | 				m.log.Debug("WS PONG received") | ||||||
| 				backoff = time.Second * 60 | 				backoff = time.Second * 60 | ||||||
| 			case <-time.After(time.Second * 5): | 			case <-time.After(time.Second * 5): | ||||||
| 				if retries > 3 { | 				if retries > 3 { | ||||||
|  | 					m.log.Debug("StatusLoop() timeout") | ||||||
| 					m.Logout() | 					m.Logout() | ||||||
| 					m.WsQuit = false | 					m.WsQuit = false | ||||||
| 					m.Login() | 					err := m.Login() | ||||||
|  | 					if err != nil { | ||||||
|  | 						log.Errorf("Login failed: %#v", err) | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
| 					if m.OnWsConnect != nil { | 					if m.OnWsConnect != nil { | ||||||
| 						m.OnWsConnect() | 						m.OnWsConnect() | ||||||
| 					} | 					} | ||||||
| @@ -800,40 +936,39 @@ func (m *MMClient) StatusLoop() { | |||||||
| func (m *MMClient) initUser() error { | func (m *MMClient) initUser() error { | ||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
| 	initLoad, err := m.Client.GetInitialLoad() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	initData := initLoad.Data.(*model.InitialLoad) |  | ||||||
| 	m.User = initData.User |  | ||||||
| 	// we only load all team data on initial login. | 	// we only load all team data on initial login. | ||||||
| 	// all other updates are for channels from our (primary) team only. | 	// all other updates are for channels from our (primary) team only. | ||||||
| 	//m.log.Debug("initUser(): loading all team data") | 	//m.log.Debug("initUser(): loading all team data") | ||||||
| 	for _, v := range initData.Teams { | 	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "") | ||||||
| 		m.Client.SetTeamId(v.Id) | 	if resp.Error != nil { | ||||||
| 		mmusers, err := m.Client.GetProfiles(0, 50000, "") | 		return resp.Error | ||||||
| 		if err != nil { | 	} | ||||||
| 			return errors.New(err.DetailedError) | 	for _, team := range teams { | ||||||
|  | 		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return errors.New(resp.Error.DetailedError) | ||||||
| 		} | 		} | ||||||
| 		t := &Team{Team: v, Users: mmusers.Data.(map[string]*model.User), Id: v.Id} | 		usermap := make(map[string]*model.User) | ||||||
| 		mmchannels, err := m.Client.GetChannels("") | 		for _, user := range mmusers { | ||||||
| 		if err != nil { | 			usermap[user.Id] = user | ||||||
| 			return errors.New(err.DetailedError) |  | ||||||
| 		} | 		} | ||||||
| 		t.Channels = mmchannels.Data.(*model.ChannelList) |  | ||||||
| 		if m.mmVersion() >= 3.08 { | 		t := &Team{Team: team, Users: usermap, Id: team.Id} | ||||||
| 			mmchannels, err = m.Client.GetMoreChannelsPage(0, 5000) |  | ||||||
| 		} else { | 		mmchannels, resp := m.Client.GetChannelsForTeamForUser(team.Id, m.User.Id, "") | ||||||
| 			mmchannels, err = m.Client.GetMoreChannels("") | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
| 		} | 		} | ||||||
| 		if err != nil { | 		t.Channels = mmchannels | ||||||
| 			return errors.New(err.DetailedError) | 		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "") | ||||||
|  | 		if resp.Error != nil { | ||||||
|  | 			return resp.Error | ||||||
| 		} | 		} | ||||||
| 		t.MoreChannels = mmchannels.Data.(*model.ChannelList) | 		t.MoreChannels = mmchannels | ||||||
| 		m.OtherTeams = append(m.OtherTeams, t) | 		m.OtherTeams = append(m.OtherTeams, t) | ||||||
| 		if v.Name == m.Credentials.Team { | 		if team.Name == m.Credentials.Team { | ||||||
| 			m.Team = t | 			m.Team = t | ||||||
| 			m.log.Debugf("initUser(): found our team %s (id: %s)", v.Name, v.Id) | 			m.log.Debugf("initUser(): found our team %s (id: %s)", team.Name, team.Id) | ||||||
| 		} | 		} | ||||||
| 		// add all users | 		// add all users | ||||||
| 		for k, v := range t.Users { | 		for k, v := range t.Users { | ||||||
| @@ -843,6 +978,16 @@ func (m *MMClient) initUser() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (m *MMClient) checkAlive() error { | ||||||
|  | 	// check if session still is valid | ||||||
|  | 	_, resp := m.Client.GetMe("") | ||||||
|  | 	if resp.Error != nil { | ||||||
|  | 		return resp.Error | ||||||
|  | 	} | ||||||
|  | 	m.log.Debug("WS PING") | ||||||
|  | 	return m.sendWSRequest("ping", nil) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error { | func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error { | ||||||
| 	req := &model.WebSocketRequest{} | 	req := &model.WebSocketRequest{} | ||||||
| 	req.Seq = m.WsSequence | 	req.Seq = m.WsSequence | ||||||
| @@ -854,22 +999,12 @@ func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) err | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *MMClient) mmVersion() float64 { |  | ||||||
| 	v, _ := strconv.ParseFloat(string(m.ServerVersion[0:2])+"0"+string(m.ServerVersion[2]), 64) |  | ||||||
| 	if string(m.ServerVersion[4]) == "." { |  | ||||||
| 		v, _ = strconv.ParseFloat(m.ServerVersion[0:4], 64) |  | ||||||
| 	} |  | ||||||
| 	return v |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func supportedVersion(version string) bool { | func supportedVersion(version string) bool { | ||||||
| 	if strings.HasPrefix(version, "3.5.0") || | 	if strings.HasPrefix(version, "3.8.0") || | ||||||
| 		strings.HasPrefix(version, "3.6.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.7.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.8.0") || |  | ||||||
| 		strings.HasPrefix(version, "3.9.0") || | 		strings.HasPrefix(version, "3.9.0") || | ||||||
| 		strings.HasPrefix(version, "3.10.0") || | 		strings.HasPrefix(version, "3.10.0") || | ||||||
| 		strings.HasPrefix(version, "4.0") { | 		strings.HasPrefix(version, "4.") || | ||||||
|  | 		strings.HasPrefix(version, "5.") { | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
| 	return false | 	return false | ||||||
|   | |||||||
| @@ -6,24 +6,27 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"github.com/gorilla/schema" |  | ||||||
| 	"io" | 	"io" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
| 	"log" | 	"log" | ||||||
| 	"net" | 	"net" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/gorilla/schema" | ||||||
|  | 	"github.com/nlopes/slack" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // OMessage for mattermost incoming webhook. (send to mattermost) | // OMessage for mattermost incoming webhook. (send to mattermost) | ||||||
| type OMessage struct { | type OMessage struct { | ||||||
| 	Channel     string      `json:"channel,omitempty"` | 	Channel     string                 `json:"channel,omitempty"` | ||||||
| 	IconURL     string      `json:"icon_url,omitempty"` | 	IconURL     string                 `json:"icon_url,omitempty"` | ||||||
| 	IconEmoji   string      `json:"icon_emoji,omitempty"` | 	IconEmoji   string                 `json:"icon_emoji,omitempty"` | ||||||
| 	UserName    string      `json:"username,omitempty"` | 	UserName    string                 `json:"username,omitempty"` | ||||||
| 	Text        string      `json:"text"` | 	Text        string                 `json:"text"` | ||||||
| 	Attachments interface{} `json:"attachments,omitempty"` | 	Attachments []slack.Attachment     `json:"attachments,omitempty"` | ||||||
| 	Type        string      `json:"type,omitempty"` | 	Type        string                 `json:"type,omitempty"` | ||||||
|  | 	Props       map[string]interface{} `json:"props"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // IMessage for mattermost outgoing webhook. (received from mattermost) | // IMessage for mattermost outgoing webhook. (received from mattermost) | ||||||
| @@ -38,16 +41,18 @@ type IMessage struct { | |||||||
| 	Timestamp   string `schema:"timestamp"` | 	Timestamp   string `schema:"timestamp"` | ||||||
| 	UserID      string `schema:"user_id"` | 	UserID      string `schema:"user_id"` | ||||||
| 	UserName    string `schema:"user_name"` | 	UserName    string `schema:"user_name"` | ||||||
| 	PostId      string `schema:"post_id"` | 	PostId      string `schema:"post_id"` //nolint:golint | ||||||
| 	RawText     string `schema:"raw_text"` | 	RawText     string `schema:"raw_text"` | ||||||
| 	ServiceId   string `schema:"service_id"` | 	ServiceId   string `schema:"service_id"` //nolint:golint | ||||||
| 	Text        string `schema:"text"` | 	Text        string `schema:"text"` | ||||||
| 	TriggerWord string `schema:"trigger_word"` | 	TriggerWord string `schema:"trigger_word"` | ||||||
|  | 	FileIDs     string `schema:"file_ids"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Client for Mattermost. | // Client for Mattermost. | ||||||
| type Client struct { | type Client struct { | ||||||
| 	Url        string // URL for incoming webhooks on mattermost. | 	// URL for incoming webhooks on mattermost. | ||||||
|  | 	Url        string // nolint:golint | ||||||
| 	In         chan IMessage | 	In         chan IMessage | ||||||
| 	Out        chan OMessage | 	Out        chan OMessage | ||||||
| 	httpclient *http.Client | 	httpclient *http.Client | ||||||
|   | |||||||
							
								
								
									
										3
									
								
								vendor/github.com/42wim/go-gitter/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,3 @@ | |||||||
|  | .idea | ||||||
|  | /test | ||||||
|  | app.yaml | ||||||
							
								
								
									
										154
									
								
								vendor/github.com/42wim/go-gitter/README.md
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,154 @@ | |||||||
|  | # gitter | ||||||
|  | Gitter API in Go | ||||||
|  | https://developer.gitter.im | ||||||
|  |  | ||||||
|  | #### Install | ||||||
|  |  | ||||||
|  | `go get github.com/sromku/go-gitter` | ||||||
|  |  | ||||||
|  | - [Initialize](#initialize) | ||||||
|  | - [Users](#users) | ||||||
|  | - [Rooms](#rooms) | ||||||
|  | - [Messages](#messages) | ||||||
|  | - [Stream](#stream) | ||||||
|  | - [Faye (Experimental)](#faye-experimental) | ||||||
|  | - [Debug](#debug) | ||||||
|  | - [App Engine](#app-engine) | ||||||
|  |  | ||||||
|  | ##### Initialize | ||||||
|  | ``` Go | ||||||
|  | api := gitter.New("YOUR_ACCESS_TOKEN") | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | ##### Users | ||||||
|  |  | ||||||
|  | - Get current user | ||||||
|  |  | ||||||
|  | 	``` Go | ||||||
|  | 	user, err := api.GetUser() | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | ##### Rooms | ||||||
|  |  | ||||||
|  | - Get all rooms | ||||||
|  | 	``` Go | ||||||
|  | 	rooms, err := api.GetRooms() | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Get room by id | ||||||
|  | 	``` Go | ||||||
|  | 	room, err := api.GetRoom("roomID") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Get rooms of some user | ||||||
|  | 	``` Go | ||||||
|  | 	rooms, err := api.GetRooms("userID") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Join room | ||||||
|  | 	``` Go | ||||||
|  | 	room, err := api.JoinRoom("roomID", "userID") | ||||||
|  | 	``` | ||||||
|  | 	 | ||||||
|  | - Leave room | ||||||
|  | 	``` Go | ||||||
|  | 	room, err := api.LeaveRoom("roomID", "userID") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Get room id | ||||||
|  | 	``` Go | ||||||
|  | 	id, err := api.GetRoomId("room/uri") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Search gitter rooms | ||||||
|  | 	``` Go | ||||||
|  | 	rooms, err := api.SearchRooms("search/string") | ||||||
|  | 	``` | ||||||
|  | ##### Messages | ||||||
|  |  | ||||||
|  | - Get messages of room | ||||||
|  | 	``` Go | ||||||
|  | 	messages, err := api.GetMessages("roomID", nil) | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Get one message | ||||||
|  | 	``` Go | ||||||
|  | 	message, err := api.GetMessage("roomID", "messageID") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | - Send message | ||||||
|  | 	``` Go | ||||||
|  | 	err := api.SendMessage("roomID", "free chat text") | ||||||
|  | 	``` | ||||||
|  |  | ||||||
|  | ##### Stream | ||||||
|  |  | ||||||
|  | Create stream to the room and start listening to incoming messages | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | stream := api.Stream(room.Id) | ||||||
|  | go api.Listen(stream) | ||||||
|  |  | ||||||
|  | for { | ||||||
|  |     event := <-stream.Event | ||||||
|  |     switch ev := event.Data.(type) { | ||||||
|  |     case *gitter.MessageReceived: | ||||||
|  |         fmt.Println(ev.Message.From.Username + ": " + ev.Message.Text) | ||||||
|  |     case *gitter.GitterConnectionClosed: | ||||||
|  |         // connection was closed | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | Close stream connection | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | stream.Close() | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | ##### Faye (Experimental) | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | faye := api.Faye(room.ID) | ||||||
|  | go faye.Listen() | ||||||
|  |  | ||||||
|  | for { | ||||||
|  |     event := <-faye.Event | ||||||
|  |     switch ev := event.Data.(type) { | ||||||
|  |     case *gitter.MessageReceived: | ||||||
|  |         fmt.Println(ev.Message.From.Username + ": " + ev.Message.Text) | ||||||
|  |     case *gitter.GitterConnectionClosed: //this one is never called in Faye | ||||||
|  |         // connection was closed | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | ##### Debug | ||||||
|  |  | ||||||
|  | You can print the internal errors by enabling debug to true | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | api.SetDebug(true, nil) | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | You can also define your own `io.Writer` in case you want to persist the logs somewhere. | ||||||
|  | For example keeping the errors on file | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | logFile, err := os.Create("gitter.log") | ||||||
|  | api.SetDebug(true, logFile) | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | ##### App Engine | ||||||
|  |  | ||||||
|  | Initialize app engine client and continue as usual | ||||||
|  |  | ||||||
|  | ``` Go | ||||||
|  | c := appengine.NewContext(r) | ||||||
|  | client := urlfetch.Client(c) | ||||||
|  |  | ||||||
|  | api := gitter.New("YOUR_ACCESS_TOKEN") | ||||||
|  | api.SetClient(client) | ||||||
|  | ``` | ||||||
|  |  | ||||||
|  | [Documentation](https://godoc.org/github.com/sromku/go-gitter) | ||||||
| @@ -205,17 +205,43 @@ func (gitter *Gitter) GetMessage(roomID, messageID string) (*Message, error) { | |||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // SendMessage sends a message to a room | // SendMessage sends a message to a room | ||||||
| func (gitter *Gitter) SendMessage(roomID, text string) error { | func (gitter *Gitter) SendMessage(roomID, text string) (*Message, error) { | ||||||
| 
 | 
 | ||||||
| 	message := Message{Text: text} | 	message := Message{Text: text} | ||||||
| 	body, _ := json.Marshal(message) | 	body, _ := json.Marshal(message) | ||||||
| 	_, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | 	response, err := gitter.post(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages", body) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		gitter.log(err) | 		gitter.log(err) | ||||||
| 		return err | 		return nil, err | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	return nil | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // UpdateMessage updates a message in a room | ||||||
|  | func (gitter *Gitter) UpdateMessage(roomID, msgID, text string) (*Message, error) { | ||||||
|  | 
 | ||||||
|  | 	message := Message{Text: text} | ||||||
|  | 	body, _ := json.Marshal(message) | ||||||
|  | 	response, err := gitter.put(gitter.config.apiBaseURL+"rooms/"+roomID+"/chatMessages/"+msgID, body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	err = json.Unmarshal(response, &message) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &message, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // JoinRoom joins a room | // JoinRoom joins a room | ||||||
| @@ -265,7 +291,7 @@ func (gitter *Gitter) SearchRooms(room string) ([]Room, error) { | |||||||
| 		Results []Room `json:"results"` | 		Results []Room `json:"results"` | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room ) | 	response, err := gitter.get(gitter.config.apiBaseURL + "rooms?q=" + room) | ||||||
| 
 | 
 | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		gitter.log(err) | 		gitter.log(err) | ||||||
| @@ -414,6 +440,39 @@ func (gitter *Gitter) post(url string, body []byte) ([]byte, error) { | |||||||
| 	return result, nil | 	return result, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | func (gitter *Gitter) put(url string, body []byte) ([]byte, error) { | ||||||
|  | 	r, err := http.NewRequest("PUT", url, bytes.NewBuffer(body)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	r.Header.Set("Content-Type", "application/json") | ||||||
|  | 	r.Header.Set("Accept", "application/json") | ||||||
|  | 	r.Header.Set("Authorization", "Bearer "+gitter.config.token) | ||||||
|  | 
 | ||||||
|  | 	resp, err := gitter.config.client.Do(r) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer resp.Body.Close() | ||||||
|  | 
 | ||||||
|  | 	if resp.StatusCode != http.StatusOK { | ||||||
|  | 		err = APIError{What: fmt.Sprintf("Status code: %v", resp.StatusCode)} | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	result, err := ioutil.ReadAll(resp.Body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		gitter.log(err) | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return result, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| func (gitter *Gitter) delete(url string) ([]byte, error) { | func (gitter *Gitter) delete(url string) ([]byte, error) { | ||||||
| 	r, err := http.NewRequest("delete", url, nil) | 	r, err := http.NewRequest("delete", url, nil) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
							
								
								
									
										27
									
								
								vendor/github.com/42wim/go-ircevent/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,27 +0,0 @@ | |||||||
| // Copyright (c) 2009 Thomas Jager. All rights reserved. |  | ||||||
| // |  | ||||||
| // Redistribution and use in source and binary forms, with or without |  | ||||||
| // modification, are permitted provided that the following conditions are |  | ||||||
| // met: |  | ||||||
| // |  | ||||||
| //    * Redistributions of source code must retain the above copyright |  | ||||||
| // notice, this list of conditions and the following disclaimer. |  | ||||||
| //    * Redistributions in binary form must reproduce the above |  | ||||||
| // copyright notice, this list of conditions and the following disclaimer |  | ||||||
| // in the documentation and/or other materials provided with the |  | ||||||
| // distribution. |  | ||||||
| //    * Neither the name of Google Inc. nor the names of its |  | ||||||
| // contributors may be used to endorse or promote products derived from |  | ||||||
| // this software without specific prior written permission. |  | ||||||
| // |  | ||||||
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |  | ||||||
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |  | ||||||
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |  | ||||||
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |  | ||||||
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |  | ||||||
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |  | ||||||
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |  | ||||||
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |  | ||||||
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |  | ||||||
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |  | ||||||
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |  | ||||||
							
								
								
									
										572
									
								
								vendor/github.com/42wim/go-ircevent/irc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,572 +0,0 @@ | |||||||
| // Copyright 2009 Thomas Jager <mail@jager.no>  All rights reserved. |  | ||||||
| // Use of this source code is governed by a BSD-style |  | ||||||
| // license that can be found in the LICENSE file. |  | ||||||
|  |  | ||||||
| /* |  | ||||||
| This package provides an event based IRC client library. It allows to |  | ||||||
| register callbacks for the events you need to handle. Its features |  | ||||||
| include handling standard CTCP, reconnecting on errors and detecting |  | ||||||
| stones servers. |  | ||||||
| Details of the IRC protocol can be found in the following RFCs: |  | ||||||
| https://tools.ietf.org/html/rfc1459 |  | ||||||
| https://tools.ietf.org/html/rfc2810 |  | ||||||
| https://tools.ietf.org/html/rfc2811 |  | ||||||
| https://tools.ietf.org/html/rfc2812 |  | ||||||
| https://tools.ietf.org/html/rfc2813 |  | ||||||
| The details of the client-to-client protocol (CTCP) can be found here: http://www.irchelp.org/irchelp/rfc/ctcpspec.html |  | ||||||
| */ |  | ||||||
|  |  | ||||||
| package irc |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bufio" |  | ||||||
| 	"bytes" |  | ||||||
| 	"crypto/tls" |  | ||||||
| 	"errors" |  | ||||||
| 	"fmt" |  | ||||||
| 	"log" |  | ||||||
| 	"net" |  | ||||||
| 	"os" |  | ||||||
| 	"strconv" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	VERSION = "go-ircevent v2.1" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var ErrDisconnected = errors.New("Disconnect Called") |  | ||||||
|  |  | ||||||
| // Read data from a connection. To be used as a goroutine. |  | ||||||
| func (irc *Connection) readLoop() { |  | ||||||
| 	defer irc.Done() |  | ||||||
| 	br := bufio.NewReaderSize(irc.socket, 512) |  | ||||||
|  |  | ||||||
| 	errChan := irc.ErrorChan() |  | ||||||
|  |  | ||||||
| 	for { |  | ||||||
| 		select { |  | ||||||
| 		case <-irc.end: |  | ||||||
| 			return |  | ||||||
| 		default: |  | ||||||
| 			// Set a read deadline based on the combined timeout and ping frequency |  | ||||||
| 			// We should ALWAYS have received a response from the server within the timeout |  | ||||||
| 			// after our own pings |  | ||||||
| 			if irc.socket != nil { |  | ||||||
| 				irc.socket.SetReadDeadline(time.Now().Add(irc.Timeout + irc.PingFreq)) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			msg, err := br.ReadString('\n') |  | ||||||
|  |  | ||||||
| 			// We got past our blocking read, so bin timeout |  | ||||||
| 			if irc.socket != nil { |  | ||||||
| 				var zero time.Time |  | ||||||
| 				irc.socket.SetReadDeadline(zero) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if err != nil { |  | ||||||
| 				errChan <- err |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if irc.Debug { |  | ||||||
| 				irc.Log.Printf("<-- %s\n", strings.TrimSpace(msg)) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			irc.Lock() |  | ||||||
| 			irc.lastMessage = time.Now() |  | ||||||
| 			irc.Unlock() |  | ||||||
| 			event, err := parseToEvent(msg) |  | ||||||
| 			event.Connection = irc |  | ||||||
| 			if err == nil { |  | ||||||
| 				/* XXX: len(args) == 0: args should be empty */ |  | ||||||
| 				irc.RunCallbacks(event) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // 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 |  | ||||||
| 	msg = strings.TrimSuffix(msg, "\r") |  | ||||||
| 	event := &Event{Raw: msg} |  | ||||||
| 	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] |  | ||||||
| 			msg = msg[i+1 : len(msg)] |  | ||||||
|  |  | ||||||
| 		} else { |  | ||||||
| 			return nil, errors.New("Malformed msg from server") |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if i, j := strings.Index(event.Source, "!"), strings.Index(event.Source, "@"); i > -1 && j > -1 && i < j { |  | ||||||
| 			event.Nick = event.Source[0:i] |  | ||||||
| 			event.User = event.Source[i+1 : j] |  | ||||||
| 			event.Host = event.Source[j+1 : len(event.Source)] |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	split := strings.SplitN(msg, " :", 2) |  | ||||||
| 	args := strings.Split(split[0], " ") |  | ||||||
| 	event.Code = strings.ToUpper(args[0]) |  | ||||||
| 	event.Arguments = args[1:] |  | ||||||
| 	if len(split) > 1 { |  | ||||||
| 		event.Arguments = append(event.Arguments, split[1]) |  | ||||||
| 	} |  | ||||||
| 	return event, nil |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Loop to write to a connection. To be used as a goroutine. |  | ||||||
| func (irc *Connection) writeLoop() { |  | ||||||
| 	defer irc.Done() |  | ||||||
| 	errChan := irc.ErrorChan() |  | ||||||
| 	for { |  | ||||||
| 		select { |  | ||||||
| 		case <-irc.end: |  | ||||||
| 			return |  | ||||||
| 		case b, ok := <-irc.pwrite: |  | ||||||
| 			if !ok || b == "" || irc.socket == nil { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if irc.Debug { |  | ||||||
| 				irc.Log.Printf("--> %s\n", strings.TrimSpace(b)) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// Set a write deadline based on the time out |  | ||||||
| 			irc.socket.SetWriteDeadline(time.Now().Add(irc.Timeout)) |  | ||||||
|  |  | ||||||
| 			_, err := irc.socket.Write([]byte(b)) |  | ||||||
|  |  | ||||||
| 			// Past blocking write, bin timeout |  | ||||||
| 			var zero time.Time |  | ||||||
| 			irc.socket.SetWriteDeadline(zero) |  | ||||||
|  |  | ||||||
| 			if err != nil { |  | ||||||
| 				errChan <- err |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Pings the server if we have not received any messages for 5 minutes |  | ||||||
| // to keep the connection alive. To be used as a goroutine. |  | ||||||
| func (irc *Connection) pingLoop() { |  | ||||||
| 	defer irc.Done() |  | ||||||
| 	ticker := time.NewTicker(1 * time.Minute) // Tick every minute for monitoring |  | ||||||
| 	ticker2 := time.NewTicker(irc.PingFreq)   // Tick at the ping frequency. |  | ||||||
| 	for { |  | ||||||
| 		select { |  | ||||||
| 		case <-ticker.C: |  | ||||||
| 			//Ping if we haven't received anything from the server within the keep alive period |  | ||||||
| 			if time.Since(irc.lastMessage) >= irc.KeepAlive { |  | ||||||
| 				irc.SendRawf("PING %d", time.Now().UnixNano()) |  | ||||||
| 			} |  | ||||||
| 		case <-ticker2.C: |  | ||||||
| 			//Ping at the ping frequency |  | ||||||
| 			irc.SendRawf("PING %d", time.Now().UnixNano()) |  | ||||||
| 			//Try to recapture nickname if it's not as configured. |  | ||||||
| 			irc.Lock() |  | ||||||
| 			if irc.nick != irc.nickcurrent { |  | ||||||
| 				irc.nickcurrent = irc.nick |  | ||||||
| 				irc.SendRawf("NICK %s", irc.nick) |  | ||||||
| 			} |  | ||||||
| 			irc.Unlock() |  | ||||||
| 		case <-irc.end: |  | ||||||
| 			ticker.Stop() |  | ||||||
| 			ticker2.Stop() |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (irc *Connection) isQuitting() bool { |  | ||||||
| 	irc.Lock() |  | ||||||
| 	defer irc.Unlock() |  | ||||||
| 	return irc.quit |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Main loop to control the connection. |  | ||||||
| func (irc *Connection) Loop() { |  | ||||||
| 	errChan := irc.ErrorChan() |  | ||||||
| 	for !irc.isQuitting() { |  | ||||||
| 		err := <-errChan |  | ||||||
| 		close(irc.end) |  | ||||||
| 		irc.Wait() |  | ||||||
| 		for !irc.isQuitting() { |  | ||||||
| 			irc.Log.Printf("Error, disconnected: %s\n", err) |  | ||||||
| 			if err = irc.Reconnect(); err != nil { |  | ||||||
| 				irc.Log.Printf("Error while reconnecting: %s\n", err) |  | ||||||
| 				time.Sleep(60 * time.Second) |  | ||||||
| 			} else { |  | ||||||
| 				errChan = irc.ErrorChan() |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Quit the current connection and disconnect from the server |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.1.6 |  | ||||||
| func (irc *Connection) Quit() { |  | ||||||
| 	quit := "QUIT" |  | ||||||
|  |  | ||||||
| 	if irc.QuitMessage != "" { |  | ||||||
| 		quit = fmt.Sprintf("QUIT :%s", irc.QuitMessage) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	irc.SendRaw(quit) |  | ||||||
| 	irc.Lock() |  | ||||||
| 	irc.stopped = true |  | ||||||
| 	irc.quit = true |  | ||||||
| 	irc.Unlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Use the connection to join a given channel. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.2.1 |  | ||||||
| func (irc *Connection) Join(channel string) { |  | ||||||
| 	irc.pwrite <- fmt.Sprintf("JOIN %s\r\n", channel) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Leave a given channel. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.2.2 |  | ||||||
| func (irc *Connection) Part(channel string) { |  | ||||||
| 	irc.pwrite <- fmt.Sprintf("PART %s\r\n", channel) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send a notification to a nickname. This is similar to Privmsg but must not receive replies. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.4.2 |  | ||||||
| func (irc *Connection) Notice(target, message string) { |  | ||||||
| 	irc.pwrite <- fmt.Sprintf("NOTICE %s :%s\r\n", target, message) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send a formated notification to a nickname. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.4.2 |  | ||||||
| func (irc *Connection) Noticef(target, format string, a ...interface{}) { |  | ||||||
| 	irc.Notice(target, fmt.Sprintf(format, a...)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send (action) message to a target (channel or nickname). |  | ||||||
| // No clear RFC on this one... |  | ||||||
| func (irc *Connection) Action(target, message string) { |  | ||||||
| 	irc.pwrite <- fmt.Sprintf("PRIVMSG %s :\001ACTION %s\001\r\n", target, message) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send formatted (action) message to a target (channel or nickname). |  | ||||||
| func (irc *Connection) Actionf(target, format string, a ...interface{}) { |  | ||||||
| 	irc.Action(target, fmt.Sprintf(format, a...)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send (private) message to a target (channel or nickname). |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.4.1 |  | ||||||
| func (irc *Connection) Privmsg(target, message string) { |  | ||||||
| 	irc.pwrite <- fmt.Sprintf("PRIVMSG %s :%s\r\n", target, message) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send formated string to specified target (channel or nickname). |  | ||||||
| func (irc *Connection) Privmsgf(target, format string, a ...interface{}) { |  | ||||||
| 	irc.Privmsg(target, fmt.Sprintf(format, a...)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Kick <user> from <channel> with <msg>. For no message, pass empty string ("") |  | ||||||
| func (irc *Connection) Kick(user, channel, msg string) { |  | ||||||
| 	var cmd bytes.Buffer |  | ||||||
| 	cmd.WriteString(fmt.Sprintf("KICK %s %s", channel, user)) |  | ||||||
| 	if msg != "" { |  | ||||||
| 		cmd.WriteString(fmt.Sprintf(" :%s", msg)) |  | ||||||
| 	} |  | ||||||
| 	cmd.WriteString("\r\n") |  | ||||||
| 	irc.pwrite <- cmd.String() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Kick all <users> from <channel> with <msg>. For no message, pass |  | ||||||
| // empty string ("") |  | ||||||
| func (irc *Connection) MultiKick(users []string, channel string, msg string) { |  | ||||||
| 	var cmd bytes.Buffer |  | ||||||
| 	cmd.WriteString(fmt.Sprintf("KICK %s %s", channel, strings.Join(users, ","))) |  | ||||||
| 	if msg != "" { |  | ||||||
| 		cmd.WriteString(fmt.Sprintf(" :%s", msg)) |  | ||||||
| 	} |  | ||||||
| 	cmd.WriteString("\r\n") |  | ||||||
| 	irc.pwrite <- cmd.String() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send raw string. |  | ||||||
| func (irc *Connection) SendRaw(message string) { |  | ||||||
| 	irc.pwrite <- message + "\r\n" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Send raw formated string. |  | ||||||
| func (irc *Connection) SendRawf(format string, a ...interface{}) { |  | ||||||
| 	irc.SendRaw(fmt.Sprintf(format, a...)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set (new) nickname. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.1.2 |  | ||||||
| func (irc *Connection) Nick(n string) { |  | ||||||
| 	irc.nick = n |  | ||||||
| 	irc.SendRawf("NICK %s", n) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Determine nick currently used with the connection. |  | ||||||
| func (irc *Connection) GetNick() string { |  | ||||||
| 	return irc.nickcurrent |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Query information about a particular nickname. |  | ||||||
| // RFC 1459: https://tools.ietf.org/html/rfc1459#section-4.5.2 |  | ||||||
| func (irc *Connection) Whois(nick string) { |  | ||||||
| 	irc.SendRawf("WHOIS %s", nick) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Query information about a given nickname in the server. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.5.1 |  | ||||||
| func (irc *Connection) Who(nick string) { |  | ||||||
| 	irc.SendRawf("WHO %s", nick) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set different modes for a target (channel or nickname). |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.2.3 |  | ||||||
| func (irc *Connection) Mode(target string, modestring ...string) { |  | ||||||
| 	if len(modestring) > 0 { |  | ||||||
| 		mode := strings.Join(modestring, " ") |  | ||||||
| 		irc.SendRawf("MODE %s %s", target, mode) |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	irc.SendRawf("MODE %s", target) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (irc *Connection) ErrorChan() chan error { |  | ||||||
| 	return irc.Error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Returns true if the connection is connected to an IRC server. |  | ||||||
| func (irc *Connection) Connected() bool { |  | ||||||
| 	return !irc.stopped |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // A disconnect sends all buffered messages (if possible), |  | ||||||
| // stops all goroutines and then closes the socket. |  | ||||||
| func (irc *Connection) Disconnect() { |  | ||||||
| 	if irc.socket != nil { |  | ||||||
| 		irc.socket.Close() |  | ||||||
| 	} |  | ||||||
| 	irc.ErrorChan() <- ErrDisconnected |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Reconnect to a server using the current connection. |  | ||||||
| func (irc *Connection) Reconnect() error { |  | ||||||
| 	irc.end = make(chan struct{}) |  | ||||||
| 	return irc.Connect(irc.Server) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Connect to a given server using the current connection configuration. |  | ||||||
| // This function also takes care of identification if a password is provided. |  | ||||||
| // RFC 1459 details: https://tools.ietf.org/html/rfc1459#section-4.1 |  | ||||||
| func (irc *Connection) Connect(server string) error { |  | ||||||
| 	irc.Server = server |  | ||||||
| 	// mark Server as stopped since there can be an error during connect |  | ||||||
| 	irc.stopped = true |  | ||||||
|  |  | ||||||
| 	// make sure everything is ready for connection |  | ||||||
| 	if len(irc.Server) == 0 { |  | ||||||
| 		return errors.New("empty 'server'") |  | ||||||
| 	} |  | ||||||
| 	if strings.Count(irc.Server, ":") != 1 { |  | ||||||
| 		return errors.New("wrong number of ':' in address") |  | ||||||
| 	} |  | ||||||
| 	if strings.Index(irc.Server, ":") == 0 { |  | ||||||
| 		return errors.New("hostname is missing") |  | ||||||
| 	} |  | ||||||
| 	if strings.Index(irc.Server, ":") == len(irc.Server)-1 { |  | ||||||
| 		return errors.New("port missing") |  | ||||||
| 	} |  | ||||||
| 	// check for valid range |  | ||||||
| 	ports := strings.Split(irc.Server, ":")[1] |  | ||||||
| 	port, err := strconv.Atoi(ports) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return errors.New("extracting port failed") |  | ||||||
| 	} |  | ||||||
| 	if !((port >= 0) && (port <= 65535)) { |  | ||||||
| 		return errors.New("port number outside valid range") |  | ||||||
| 	} |  | ||||||
| 	if irc.Log == nil { |  | ||||||
| 		return errors.New("'Log' points to nil") |  | ||||||
| 	} |  | ||||||
| 	if len(irc.nick) == 0 { |  | ||||||
| 		return errors.New("empty 'nick'") |  | ||||||
| 	} |  | ||||||
| 	if len(irc.user) == 0 { |  | ||||||
| 		return errors.New("empty 'user'") |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if irc.UseTLS { |  | ||||||
| 		dialer := &net.Dialer{Timeout: irc.Timeout} |  | ||||||
| 		irc.socket, err = tls.DialWithDialer(dialer, "tcp", irc.Server, irc.TLSConfig) |  | ||||||
| 	} else { |  | ||||||
| 		irc.socket, err = net.DialTimeout("tcp", irc.Server, irc.Timeout) |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	irc.stopped = false |  | ||||||
| 	irc.Log.Printf("Connected to %s (%s)\n", irc.Server, irc.socket.RemoteAddr()) |  | ||||||
|  |  | ||||||
| 	irc.pwrite = make(chan string, 10) |  | ||||||
| 	irc.Error = make(chan error, 2) |  | ||||||
| 	irc.Add(3) |  | ||||||
| 	go irc.readLoop() |  | ||||||
| 	go irc.writeLoop() |  | ||||||
| 	go irc.pingLoop() |  | ||||||
| 	if len(irc.Password) > 0 { |  | ||||||
| 		irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	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 { |  | ||||||
| 		select { |  | ||||||
| 		case res := <-saslResChan: |  | ||||||
| 			if res.Failed { |  | ||||||
| 				close(saslResChan) |  | ||||||
| 				return res.Err |  | ||||||
| 			} |  | ||||||
| 		case <-time.After(time.Second * 15): |  | ||||||
| 			close(saslResChan) |  | ||||||
| 			return errors.New("SASL setup timed out. This shouldn't happen.") |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// 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 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Create a connection with the (publicly visible) nickname and username. |  | ||||||
| // The nickname is later used to address the user. Returns nil if nick |  | ||||||
| // or user are empty. |  | ||||||
| func IRC(nick, user string) *Connection { |  | ||||||
| 	// catch invalid values |  | ||||||
| 	if len(nick) == 0 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	if len(user) == 0 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	irc := &Connection{ |  | ||||||
| 		nick:        nick, |  | ||||||
| 		nickcurrent: nick, |  | ||||||
| 		user:        user, |  | ||||||
| 		Log:         log.New(os.Stdout, "", log.LstdFlags), |  | ||||||
| 		end:         make(chan struct{}), |  | ||||||
| 		Version:     VERSION, |  | ||||||
| 		KeepAlive:   4 * time.Minute, |  | ||||||
| 		Timeout:     1 * time.Minute, |  | ||||||
| 		PingFreq:    15 * time.Minute, |  | ||||||
| 		SASLMech:    "PLAIN", |  | ||||||
| 		QuitMessage: "", |  | ||||||
| 	} |  | ||||||
| 	irc.setupCallbacks() |  | ||||||
| 	return irc |  | ||||||
| } |  | ||||||
							
								
								
									
										222
									
								
								vendor/github.com/42wim/go-ircevent/irc_callback.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,222 +0,0 @@ | |||||||
| package irc |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"strconv" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Register a callback to a connection and event code. A callback is a function |  | ||||||
| // which takes only an Event pointer as parameter. Valid event codes are all |  | ||||||
| // IRC/CTCP commands and error/response codes. This function returns the ID of |  | ||||||
| // the registered callback for later management. |  | ||||||
| func (irc *Connection) AddCallback(eventcode string, callback func(*Event)) int { |  | ||||||
| 	eventcode = strings.ToUpper(eventcode) |  | ||||||
| 	id := 0 |  | ||||||
| 	if _, ok := irc.events[eventcode]; !ok { |  | ||||||
| 		irc.events[eventcode] = make(map[int]func(*Event)) |  | ||||||
| 		id = 0 |  | ||||||
| 	} else { |  | ||||||
| 		id = len(irc.events[eventcode]) |  | ||||||
| 	} |  | ||||||
| 	irc.events[eventcode][id] = callback |  | ||||||
| 	return id |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Remove callback i (ID) from the given event code. This functions returns |  | ||||||
| // true upon success, false if any error occurs. |  | ||||||
| func (irc *Connection) RemoveCallback(eventcode string, i int) bool { |  | ||||||
| 	eventcode = strings.ToUpper(eventcode) |  | ||||||
|  |  | ||||||
| 	if event, ok := irc.events[eventcode]; ok { |  | ||||||
| 		if _, ok := event[i]; ok { |  | ||||||
| 			delete(irc.events[eventcode], i) |  | ||||||
| 			return true |  | ||||||
| 		} |  | ||||||
| 		irc.Log.Printf("Event found, but no callback found at id %d\n", i) |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	irc.Log.Println("Event not found") |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Remove all callbacks from a given event code. It returns true |  | ||||||
| // if given event code is found and cleared. |  | ||||||
| func (irc *Connection) ClearCallback(eventcode string) bool { |  | ||||||
| 	eventcode = strings.ToUpper(eventcode) |  | ||||||
|  |  | ||||||
| 	if _, ok := irc.events[eventcode]; ok { |  | ||||||
| 		irc.events[eventcode] = make(map[int]func(*Event)) |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	irc.Log.Println("Event not found") |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Replace callback i (ID) associated with a given event code with a new callback function. |  | ||||||
| func (irc *Connection) ReplaceCallback(eventcode string, i int, callback func(*Event)) { |  | ||||||
| 	eventcode = strings.ToUpper(eventcode) |  | ||||||
|  |  | ||||||
| 	if event, ok := irc.events[eventcode]; ok { |  | ||||||
| 		if _, ok := event[i]; ok { |  | ||||||
| 			event[i] = callback |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 		irc.Log.Printf("Event found, but no callback found at id %d\n", i) |  | ||||||
| 	} |  | ||||||
| 	irc.Log.Printf("Event not found. Use AddCallBack\n") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Execute all callbacks associated with a given event. |  | ||||||
| func (irc *Connection) RunCallbacks(event *Event) { |  | ||||||
| 	msg := event.Message() |  | ||||||
| 	if event.Code == "PRIVMSG" && len(msg) > 2 && msg[0] == '\x01' { |  | ||||||
| 		event.Code = "CTCP" //Unknown CTCP |  | ||||||
|  |  | ||||||
| 		if i := strings.LastIndex(msg, "\x01"); i > 0 { |  | ||||||
| 			msg = msg[1:i] |  | ||||||
| 		} else { |  | ||||||
| 			irc.Log.Printf("Invalid CTCP Message: %s\n", strconv.Quote(msg)) |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if msg == "VERSION" { |  | ||||||
| 			event.Code = "CTCP_VERSION" |  | ||||||
|  |  | ||||||
| 		} else if msg == "TIME" { |  | ||||||
| 			event.Code = "CTCP_TIME" |  | ||||||
|  |  | ||||||
| 		} else if strings.HasPrefix(msg, "PING") { |  | ||||||
| 			event.Code = "CTCP_PING" |  | ||||||
|  |  | ||||||
| 		} else if msg == "USERINFO" { |  | ||||||
| 			event.Code = "CTCP_USERINFO" |  | ||||||
|  |  | ||||||
| 		} else if msg == "CLIENTINFO" { |  | ||||||
| 			event.Code = "CTCP_CLIENTINFO" |  | ||||||
|  |  | ||||||
| 		} else if strings.HasPrefix(msg, "ACTION") { |  | ||||||
| 			event.Code = "CTCP_ACTION" |  | ||||||
| 			if len(msg) > 6 { |  | ||||||
| 				msg = msg[7:] |  | ||||||
| 			} else { |  | ||||||
| 				msg = "" |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		event.Arguments[len(event.Arguments)-1] = msg |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if callbacks, ok := irc.events[event.Code]; ok { |  | ||||||
| 		if irc.VerboseCallbackHandler { |  | ||||||
| 			irc.Log.Printf("%v (%v) >> %#v\n", event.Code, len(callbacks), event) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		for _, callback := range callbacks { |  | ||||||
| 			callback(event) |  | ||||||
| 		} |  | ||||||
| 	} else if irc.VerboseCallbackHandler { |  | ||||||
| 		irc.Log.Printf("%v (0) >> %#v\n", event.Code, event) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if callbacks, ok := irc.events["*"]; ok { |  | ||||||
| 		if irc.VerboseCallbackHandler { |  | ||||||
| 			irc.Log.Printf("%v (0) >> %#v\n", event.Code, event) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		for _, callback := range callbacks { |  | ||||||
| 			callback(event) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set up some initial callbacks to handle the IRC/CTCP protocol. |  | ||||||
| func (irc *Connection) setupCallbacks() { |  | ||||||
| 	irc.events = make(map[string]map[int]func(*Event)) |  | ||||||
|  |  | ||||||
| 	//Handle error events. |  | ||||||
| 	irc.AddCallback("ERROR", func(e *Event) { irc.Disconnect() }) |  | ||||||
|  |  | ||||||
| 	//Handle ping events |  | ||||||
| 	irc.AddCallback("PING", func(e *Event) { irc.SendRaw("PONG :" + e.Message()) }) |  | ||||||
|  |  | ||||||
| 	//Version handler |  | ||||||
| 	irc.AddCallback("CTCP_VERSION", func(e *Event) { |  | ||||||
| 		irc.SendRawf("NOTICE %s :\x01VERSION %s\x01", e.Nick, irc.Version) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	irc.AddCallback("CTCP_USERINFO", func(e *Event) { |  | ||||||
| 		irc.SendRawf("NOTICE %s :\x01USERINFO %s\x01", e.Nick, irc.user) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	irc.AddCallback("CTCP_CLIENTINFO", func(e *Event) { |  | ||||||
| 		irc.SendRawf("NOTICE %s :\x01CLIENTINFO PING VERSION TIME USERINFO CLIENTINFO\x01", e.Nick) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	irc.AddCallback("CTCP_TIME", func(e *Event) { |  | ||||||
| 		ltime := time.Now() |  | ||||||
| 		irc.SendRawf("NOTICE %s :\x01TIME %s\x01", e.Nick, ltime.String()) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	irc.AddCallback("CTCP_PING", func(e *Event) { irc.SendRawf("NOTICE %s :\x01%s\x01", e.Nick, e.Message()) }) |  | ||||||
|  |  | ||||||
| 	// 437: ERR_UNAVAILRESOURCE "<nick/channel> :Nick/channel is temporarily unavailable" |  | ||||||
| 	// Add a _ to current nick. If irc.nickcurrent is empty this cannot |  | ||||||
| 	// work. It has to be set somewhere first in case the nick is already |  | ||||||
| 	// taken or unavailable from the beginning. |  | ||||||
| 	irc.AddCallback("437", func(e *Event) { |  | ||||||
| 		// If irc.nickcurrent hasn't been set yet, set to irc.nick |  | ||||||
| 		if irc.nickcurrent == "" { |  | ||||||
| 			irc.nickcurrent = irc.nick |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if len(irc.nickcurrent) > 8 { |  | ||||||
| 			irc.nickcurrent = "_" + irc.nickcurrent |  | ||||||
| 		} else { |  | ||||||
| 			irc.nickcurrent = irc.nickcurrent + "_" |  | ||||||
| 		} |  | ||||||
| 		irc.SendRawf("NICK %s", irc.nickcurrent) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	// 433: ERR_NICKNAMEINUSE "<nick> :Nickname is already in use" |  | ||||||
| 	// Add a _ to current nick. |  | ||||||
| 	irc.AddCallback("433", func(e *Event) { |  | ||||||
| 		// If irc.nickcurrent hasn't been set yet, set to irc.nick |  | ||||||
| 		if irc.nickcurrent == "" { |  | ||||||
| 			irc.nickcurrent = irc.nick |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if len(irc.nickcurrent) > 8 { |  | ||||||
| 			irc.nickcurrent = "_" + irc.nickcurrent |  | ||||||
| 		} else { |  | ||||||
| 			irc.nickcurrent = irc.nickcurrent + "_" |  | ||||||
| 		} |  | ||||||
| 		irc.SendRawf("NICK %s", irc.nickcurrent) |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	irc.AddCallback("PONG", func(e *Event) { |  | ||||||
| 		ns, _ := strconv.ParseInt(e.Message(), 10, 64) |  | ||||||
| 		delta := time.Duration(time.Now().UnixNano() - ns) |  | ||||||
| 		if irc.Debug { |  | ||||||
| 			irc.Log.Printf("Lag: %.3f s\n", delta.Seconds()) |  | ||||||
| 		} |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	// NICK Define a nickname. |  | ||||||
| 	// Set irc.nickcurrent to the new nick actually used in this connection. |  | ||||||
| 	irc.AddCallback("NICK", func(e *Event) { |  | ||||||
| 		if e.Nick == irc.nick { |  | ||||||
| 			irc.nickcurrent = e.Message() |  | ||||||
| 		} |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	// 1: RPL_WELCOME "Welcome to the Internet Relay Network <nick>!<user>@<host>" |  | ||||||
| 	// Set irc.nickcurrent to the actually used nick in this connection. |  | ||||||
| 	irc.AddCallback("001", func(e *Event) { |  | ||||||
| 		irc.Lock() |  | ||||||
| 		irc.nickcurrent = e.Arguments[0] |  | ||||||
| 		irc.Unlock() |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
							
								
								
									
										53
									
								
								vendor/github.com/42wim/go-ircevent/irc_sasl.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,53 +0,0 @@ | |||||||
| package irc |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"encoding/base64" |  | ||||||
| 	"errors" |  | ||||||
| 	"fmt" |  | ||||||
| 	"strings" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type SASLResult struct { |  | ||||||
| 	Failed bool |  | ||||||
| 	Err    error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (irc *Connection) setupSASLCallbacks(result chan<- *SASLResult) { |  | ||||||
| 	irc.AddCallback("CAP", func(e *Event) { |  | ||||||
| 		if len(e.Arguments) == 3 { |  | ||||||
| 			if e.Arguments[1] == "LS" { |  | ||||||
| 				if !strings.Contains(e.Arguments[2], "sasl") { |  | ||||||
| 					result <- &SASLResult{true, errors.New("no SASL capability " + e.Arguments[2])} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			if e.Arguments[1] == "ACK" { |  | ||||||
| 				if irc.SASLMech != "PLAIN" { |  | ||||||
| 					result <- &SASLResult{true, errors.New("only PLAIN is supported")} |  | ||||||
| 				} |  | ||||||
| 				irc.SendRaw("AUTHENTICATE " + irc.SASLMech) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	}) |  | ||||||
| 	irc.AddCallback("AUTHENTICATE", func(e *Event) { |  | ||||||
| 		str := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s\x00%s\x00%s", irc.SASLLogin, irc.SASLLogin, irc.SASLPassword))) |  | ||||||
| 		irc.SendRaw("AUTHENTICATE " + str) |  | ||||||
| 	}) |  | ||||||
| 	irc.AddCallback("901", func(e *Event) { |  | ||||||
| 		irc.SendRaw("CAP END") |  | ||||||
| 		irc.SendRaw("QUIT") |  | ||||||
| 		result <- &SASLResult{true, errors.New(e.Arguments[1])} |  | ||||||
| 	}) |  | ||||||
| 	irc.AddCallback("902", func(e *Event) { |  | ||||||
| 		irc.SendRaw("CAP END") |  | ||||||
| 		irc.SendRaw("QUIT") |  | ||||||
| 		result <- &SASLResult{true, errors.New(e.Arguments[1])} |  | ||||||
| 	}) |  | ||||||
| 	irc.AddCallback("903", func(e *Event) { |  | ||||||
| 		result <- &SASLResult{false, nil} |  | ||||||
| 	}) |  | ||||||
| 	irc.AddCallback("904", func(e *Event) { |  | ||||||
| 		irc.SendRaw("CAP END") |  | ||||||
| 		irc.SendRaw("QUIT") |  | ||||||
| 		result <- &SASLResult{true, errors.New(e.Arguments[1])} |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
							
								
								
									
										76
									
								
								vendor/github.com/42wim/go-ircevent/irc_struct.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,76 +0,0 @@ | |||||||
| // Copyright 2009 Thomas Jager <mail@jager.no>  All rights reserved. |  | ||||||
| // Use of this source code is governed by a BSD-style |  | ||||||
| // license that can be found in the LICENSE file. |  | ||||||
|  |  | ||||||
| package irc |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"crypto/tls" |  | ||||||
| 	"log" |  | ||||||
| 	"net" |  | ||||||
| 	"sync" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type Connection struct { |  | ||||||
| 	sync.Mutex |  | ||||||
| 	sync.WaitGroup |  | ||||||
| 	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 |  | ||||||
| 	end    chan struct{} |  | ||||||
|  |  | ||||||
| 	nick        string //The nickname we want. |  | ||||||
| 	nickcurrent string //The nickname we currently have. |  | ||||||
| 	user        string |  | ||||||
| 	registered  bool |  | ||||||
| 	events      map[string]map[int]func(*Event) |  | ||||||
|  |  | ||||||
| 	QuitMessage string |  | ||||||
| 	lastMessage time.Time |  | ||||||
|  |  | ||||||
| 	VerboseCallbackHandler bool |  | ||||||
| 	Log                    *log.Logger |  | ||||||
|  |  | ||||||
| 	stopped bool |  | ||||||
| 	quit    bool //User called Quit, do not reconnect. |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // A struct to represent an event. |  | ||||||
| type Event struct { |  | ||||||
| 	Code       string |  | ||||||
| 	Raw        string |  | ||||||
| 	Nick       string //<nick> |  | ||||||
| 	Host       string //<nick>!<usr>@<host> |  | ||||||
| 	Source     string //<host> |  | ||||||
| 	User       string //<usr> |  | ||||||
| 	Arguments  []string |  | ||||||
| 	Tags       map[string]string |  | ||||||
| 	Connection *Connection |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Retrieve the last message from Event arguments. |  | ||||||
| // This function  leaves the arguments untouched and |  | ||||||
| // returns an empty string if there are none. |  | ||||||
| func (e *Event) Message() string { |  | ||||||
| 	if len(e.Arguments) == 0 { |  | ||||||
| 		return "" |  | ||||||
| 	} |  | ||||||
| 	return e.Arguments[len(e.Arguments)-1] |  | ||||||
| } |  | ||||||
							
								
								
									
										14
									
								
								vendor/github.com/42wim/go-ircevent/irc_test_fuzz.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,14 +0,0 @@ | |||||||
| // +build gofuzz |  | ||||||
|  |  | ||||||
| package irc |  | ||||||
|  |  | ||||||
| func Fuzz(data []byte) int { |  | ||||||
| 	b := bytes.NewBuffer(data) |  | ||||||
| 	event, err := parseToEvent(b.String()) |  | ||||||
| 	if err == nil { |  | ||||||
| 		irc := IRC("go-eventirc", "go-eventirc") |  | ||||||
| 		irc.RunCallbacks(event) |  | ||||||
| 		return 1 |  | ||||||
| 	} |  | ||||||
| 	return 0 |  | ||||||
| } |  | ||||||
							
								
								
									
										14
									
								
								vendor/github.com/BurntSushi/toml/COPYING
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,14 +0,0 @@ | |||||||
|             DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE |  | ||||||
|                     Version 2, December 2004 |  | ||||||
|  |  | ||||||
|  Copyright (C) 2004 Sam Hocevar <sam@hocevar.net> |  | ||||||
|  |  | ||||||
|  Everyone is permitted to copy and distribute verbatim or modified |  | ||||||
|  copies of this license document, and changing it is allowed as long |  | ||||||
|  as the name is changed. |  | ||||||
|  |  | ||||||
|             DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE |  | ||||||
|    TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION |  | ||||||
|  |  | ||||||
|   0. You just DO WHAT THE FUCK YOU WANT TO. |  | ||||||
|  |  | ||||||
							
								
								
									
										90
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-decoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,90 +0,0 @@ | |||||||
| // Command toml-test-decoder satisfies the toml-test interface for testing |  | ||||||
| // TOML decoders. Namely, it accepts TOML on stdin and outputs JSON on stdout. |  | ||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"encoding/json" |  | ||||||
| 	"flag" |  | ||||||
| 	"fmt" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| 	"path" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/BurntSushi/toml" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.SetFlags(0) |  | ||||||
|  |  | ||||||
| 	flag.Usage = usage |  | ||||||
| 	flag.Parse() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func usage() { |  | ||||||
| 	log.Printf("Usage: %s < toml-file\n", path.Base(os.Args[0])) |  | ||||||
| 	flag.PrintDefaults() |  | ||||||
|  |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	if flag.NArg() != 0 { |  | ||||||
| 		flag.Usage() |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var tmp interface{} |  | ||||||
| 	if _, err := toml.DecodeReader(os.Stdin, &tmp); err != nil { |  | ||||||
| 		log.Fatalf("Error decoding TOML: %s", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	typedTmp := translate(tmp) |  | ||||||
| 	if err := json.NewEncoder(os.Stdout).Encode(typedTmp); err != nil { |  | ||||||
| 		log.Fatalf("Error encoding JSON: %s", err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func translate(tomlData interface{}) interface{} { |  | ||||||
| 	switch orig := tomlData.(type) { |  | ||||||
| 	case map[string]interface{}: |  | ||||||
| 		typed := make(map[string]interface{}, len(orig)) |  | ||||||
| 		for k, v := range orig { |  | ||||||
| 			typed[k] = translate(v) |  | ||||||
| 		} |  | ||||||
| 		return typed |  | ||||||
| 	case []map[string]interface{}: |  | ||||||
| 		typed := make([]map[string]interface{}, len(orig)) |  | ||||||
| 		for i, v := range orig { |  | ||||||
| 			typed[i] = translate(v).(map[string]interface{}) |  | ||||||
| 		} |  | ||||||
| 		return typed |  | ||||||
| 	case []interface{}: |  | ||||||
| 		typed := make([]interface{}, len(orig)) |  | ||||||
| 		for i, v := range orig { |  | ||||||
| 			typed[i] = translate(v) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// We don't really need to tag arrays, but let's be future proof. |  | ||||||
| 		// (If TOML ever supports tuples, we'll need this.) |  | ||||||
| 		return tag("array", typed) |  | ||||||
| 	case time.Time: |  | ||||||
| 		return tag("datetime", orig.Format("2006-01-02T15:04:05Z")) |  | ||||||
| 	case bool: |  | ||||||
| 		return tag("bool", fmt.Sprintf("%v", orig)) |  | ||||||
| 	case int64: |  | ||||||
| 		return tag("integer", fmt.Sprintf("%d", orig)) |  | ||||||
| 	case float64: |  | ||||||
| 		return tag("float", fmt.Sprintf("%v", orig)) |  | ||||||
| 	case string: |  | ||||||
| 		return tag("string", orig) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	panic(fmt.Sprintf("Unknown type: %T", tomlData)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func tag(typeName string, data interface{}) map[string]interface{} { |  | ||||||
| 	return map[string]interface{}{ |  | ||||||
| 		"type":  typeName, |  | ||||||
| 		"value": data, |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										131
									
								
								vendor/github.com/BurntSushi/toml/cmd/toml-test-encoder/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,131 +0,0 @@ | |||||||
| // Command toml-test-encoder satisfies the toml-test interface for testing |  | ||||||
| // TOML encoders. Namely, it accepts JSON on stdin and outputs TOML on stdout. |  | ||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"encoding/json" |  | ||||||
| 	"flag" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| 	"path" |  | ||||||
| 	"strconv" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/BurntSushi/toml" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.SetFlags(0) |  | ||||||
|  |  | ||||||
| 	flag.Usage = usage |  | ||||||
| 	flag.Parse() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func usage() { |  | ||||||
| 	log.Printf("Usage: %s < json-file\n", path.Base(os.Args[0])) |  | ||||||
| 	flag.PrintDefaults() |  | ||||||
|  |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	if flag.NArg() != 0 { |  | ||||||
| 		flag.Usage() |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var tmp interface{} |  | ||||||
| 	if err := json.NewDecoder(os.Stdin).Decode(&tmp); err != nil { |  | ||||||
| 		log.Fatalf("Error decoding JSON: %s", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	tomlData := translate(tmp) |  | ||||||
| 	if err := toml.NewEncoder(os.Stdout).Encode(tomlData); err != nil { |  | ||||||
| 		log.Fatalf("Error encoding TOML: %s", err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func translate(typedJson interface{}) interface{} { |  | ||||||
| 	switch v := typedJson.(type) { |  | ||||||
| 	case map[string]interface{}: |  | ||||||
| 		if len(v) == 2 && in("type", v) && in("value", v) { |  | ||||||
| 			return untag(v) |  | ||||||
| 		} |  | ||||||
| 		m := make(map[string]interface{}, len(v)) |  | ||||||
| 		for k, v2 := range v { |  | ||||||
| 			m[k] = translate(v2) |  | ||||||
| 		} |  | ||||||
| 		return m |  | ||||||
| 	case []interface{}: |  | ||||||
| 		tabArray := make([]map[string]interface{}, len(v)) |  | ||||||
| 		for i := range v { |  | ||||||
| 			if m, ok := translate(v[i]).(map[string]interface{}); ok { |  | ||||||
| 				tabArray[i] = m |  | ||||||
| 			} else { |  | ||||||
| 				log.Fatalf("JSON arrays may only contain objects. This " + |  | ||||||
| 					"corresponds to only tables being allowed in " + |  | ||||||
| 					"TOML table arrays.") |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return tabArray |  | ||||||
| 	} |  | ||||||
| 	log.Fatalf("Unrecognized JSON format '%T'.", typedJson) |  | ||||||
| 	panic("unreachable") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func untag(typed map[string]interface{}) interface{} { |  | ||||||
| 	t := typed["type"].(string) |  | ||||||
| 	v := typed["value"] |  | ||||||
| 	switch t { |  | ||||||
| 	case "string": |  | ||||||
| 		return v.(string) |  | ||||||
| 	case "integer": |  | ||||||
| 		v := v.(string) |  | ||||||
| 		n, err := strconv.Atoi(v) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("Could not parse '%s' as integer: %s", v, err) |  | ||||||
| 		} |  | ||||||
| 		return n |  | ||||||
| 	case "float": |  | ||||||
| 		v := v.(string) |  | ||||||
| 		f, err := strconv.ParseFloat(v, 64) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("Could not parse '%s' as float64: %s", v, err) |  | ||||||
| 		} |  | ||||||
| 		return f |  | ||||||
| 	case "datetime": |  | ||||||
| 		v := v.(string) |  | ||||||
| 		t, err := time.Parse("2006-01-02T15:04:05Z", v) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("Could not parse '%s' as a datetime: %s", v, err) |  | ||||||
| 		} |  | ||||||
| 		return t |  | ||||||
| 	case "bool": |  | ||||||
| 		v := v.(string) |  | ||||||
| 		switch v { |  | ||||||
| 		case "true": |  | ||||||
| 			return true |  | ||||||
| 		case "false": |  | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 		log.Fatalf("Could not parse '%s' as a boolean.", v) |  | ||||||
| 	case "array": |  | ||||||
| 		v := v.([]interface{}) |  | ||||||
| 		array := make([]interface{}, len(v)) |  | ||||||
| 		for i := range v { |  | ||||||
| 			if m, ok := v[i].(map[string]interface{}); ok { |  | ||||||
| 				array[i] = untag(m) |  | ||||||
| 			} else { |  | ||||||
| 				log.Fatalf("Arrays may only contain other arrays or "+ |  | ||||||
| 					"primitive values, but found a '%T'.", m) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return array |  | ||||||
| 	} |  | ||||||
| 	log.Fatalf("Unrecognized tag type '%s'.", t) |  | ||||||
| 	panic("unreachable") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func in(key string, m map[string]interface{}) bool { |  | ||||||
| 	_, ok := m[key] |  | ||||||
| 	return ok |  | ||||||
| } |  | ||||||
							
								
								
									
										61
									
								
								vendor/github.com/BurntSushi/toml/cmd/tomlv/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,61 +0,0 @@ | |||||||
| // Command tomlv validates TOML documents and prints each key's type. |  | ||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"flag" |  | ||||||
| 	"fmt" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| 	"path" |  | ||||||
| 	"strings" |  | ||||||
| 	"text/tabwriter" |  | ||||||
|  |  | ||||||
| 	"github.com/BurntSushi/toml" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	flagTypes = false |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	log.SetFlags(0) |  | ||||||
|  |  | ||||||
| 	flag.BoolVar(&flagTypes, "types", flagTypes, |  | ||||||
| 		"When set, the types of every defined key will be shown.") |  | ||||||
|  |  | ||||||
| 	flag.Usage = usage |  | ||||||
| 	flag.Parse() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func usage() { |  | ||||||
| 	log.Printf("Usage: %s toml-file [ toml-file ... ]\n", |  | ||||||
| 		path.Base(os.Args[0])) |  | ||||||
| 	flag.PrintDefaults() |  | ||||||
|  |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	if flag.NArg() < 1 { |  | ||||||
| 		flag.Usage() |  | ||||||
| 	} |  | ||||||
| 	for _, f := range flag.Args() { |  | ||||||
| 		var tmp interface{} |  | ||||||
| 		md, err := toml.DecodeFile(f, &tmp) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("Error in '%s': %s", f, err) |  | ||||||
| 		} |  | ||||||
| 		if flagTypes { |  | ||||||
| 			printTypes(md) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func printTypes(md toml.MetaData) { |  | ||||||
| 	tabw := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0) |  | ||||||
| 	for _, key := range md.Keys() { |  | ||||||
| 		fmt.Fprintf(tabw, "%s%s\t%s\n", |  | ||||||
| 			strings.Repeat("    ", len(key)-1), key, md.Type(key...)) |  | ||||||
| 	} |  | ||||||
| 	tabw.Flush() |  | ||||||
| } |  | ||||||
							
								
								
									
										509
									
								
								vendor/github.com/BurntSushi/toml/decode.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,509 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"io" |  | ||||||
| 	"io/ioutil" |  | ||||||
| 	"math" |  | ||||||
| 	"reflect" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func e(format string, args ...interface{}) error { |  | ||||||
| 	return fmt.Errorf("toml: "+format, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Unmarshaler is the interface implemented by objects that can unmarshal a |  | ||||||
| // TOML description of themselves. |  | ||||||
| type Unmarshaler interface { |  | ||||||
| 	UnmarshalTOML(interface{}) error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Unmarshal decodes the contents of `p` in TOML format into a pointer `v`. |  | ||||||
| func Unmarshal(p []byte, v interface{}) error { |  | ||||||
| 	_, err := Decode(string(p), v) |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Primitive is a TOML value that hasn't been decoded into a Go value. |  | ||||||
| // When using the various `Decode*` functions, the type `Primitive` may |  | ||||||
| // be given to any value, and its decoding will be delayed. |  | ||||||
| // |  | ||||||
| // A `Primitive` value can be decoded using the `PrimitiveDecode` function. |  | ||||||
| // |  | ||||||
| // The underlying representation of a `Primitive` value is subject to change. |  | ||||||
| // Do not rely on it. |  | ||||||
| // |  | ||||||
| // N.B. Primitive values are still parsed, so using them will only avoid |  | ||||||
| // the overhead of reflection. They can be useful when you don't know the |  | ||||||
| // exact type of TOML data until run time. |  | ||||||
| type Primitive struct { |  | ||||||
| 	undecoded interface{} |  | ||||||
| 	context   Key |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // DEPRECATED! |  | ||||||
| // |  | ||||||
| // Use MetaData.PrimitiveDecode instead. |  | ||||||
| func PrimitiveDecode(primValue Primitive, v interface{}) error { |  | ||||||
| 	md := MetaData{decoded: make(map[string]bool)} |  | ||||||
| 	return md.unify(primValue.undecoded, rvalue(v)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PrimitiveDecode is just like the other `Decode*` functions, except it |  | ||||||
| // decodes a TOML value that has already been parsed. Valid primitive values |  | ||||||
| // can *only* be obtained from values filled by the decoder functions, |  | ||||||
| // including this method. (i.e., `v` may contain more `Primitive` |  | ||||||
| // values.) |  | ||||||
| // |  | ||||||
| // Meta data for primitive values is included in the meta data returned by |  | ||||||
| // the `Decode*` functions with one exception: keys returned by the Undecoded |  | ||||||
| // method will only reflect keys that were decoded. Namely, any keys hidden |  | ||||||
| // behind a Primitive will be considered undecoded. Executing this method will |  | ||||||
| // update the undecoded keys in the meta data. (See the example.) |  | ||||||
| func (md *MetaData) PrimitiveDecode(primValue Primitive, v interface{}) error { |  | ||||||
| 	md.context = primValue.context |  | ||||||
| 	defer func() { md.context = nil }() |  | ||||||
| 	return md.unify(primValue.undecoded, rvalue(v)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Decode will decode the contents of `data` in TOML format into a pointer |  | ||||||
| // `v`. |  | ||||||
| // |  | ||||||
| // TOML hashes correspond to Go structs or maps. (Dealer's choice. They can be |  | ||||||
| // used interchangeably.) |  | ||||||
| // |  | ||||||
| // TOML arrays of tables correspond to either a slice of structs or a slice |  | ||||||
| // of maps. |  | ||||||
| // |  | ||||||
| // TOML datetimes correspond to Go `time.Time` values. |  | ||||||
| // |  | ||||||
| // All other TOML types (float, string, int, bool and array) correspond |  | ||||||
| // to the obvious Go types. |  | ||||||
| // |  | ||||||
| // An exception to the above rules is if a type implements the |  | ||||||
| // encoding.TextUnmarshaler interface. In this case, any primitive TOML value |  | ||||||
| // (floats, strings, integers, booleans and datetimes) will be converted to |  | ||||||
| // a byte string and given to the value's UnmarshalText method. See the |  | ||||||
| // Unmarshaler example for a demonstration with time duration strings. |  | ||||||
| // |  | ||||||
| // Key mapping |  | ||||||
| // |  | ||||||
| // TOML keys can map to either keys in a Go map or field names in a Go |  | ||||||
| // struct. The special `toml` struct tag may be used to map TOML keys to |  | ||||||
| // struct fields that don't match the key name exactly. (See the example.) |  | ||||||
| // A case insensitive match to struct names will be tried if an exact match |  | ||||||
| // can't be found. |  | ||||||
| // |  | ||||||
| // The mapping between TOML values and Go values is loose. That is, there |  | ||||||
| // may exist TOML values that cannot be placed into your representation, and |  | ||||||
| // there may be parts of your representation that do not correspond to |  | ||||||
| // TOML values. This loose mapping can be made stricter by using the IsDefined |  | ||||||
| // and/or Undecoded methods on the MetaData returned. |  | ||||||
| // |  | ||||||
| // This decoder will not handle cyclic types. If a cyclic type is passed, |  | ||||||
| // `Decode` will not terminate. |  | ||||||
| func Decode(data string, v interface{}) (MetaData, error) { |  | ||||||
| 	rv := reflect.ValueOf(v) |  | ||||||
| 	if rv.Kind() != reflect.Ptr { |  | ||||||
| 		return MetaData{}, e("Decode of non-pointer %s", reflect.TypeOf(v)) |  | ||||||
| 	} |  | ||||||
| 	if rv.IsNil() { |  | ||||||
| 		return MetaData{}, e("Decode of nil %s", reflect.TypeOf(v)) |  | ||||||
| 	} |  | ||||||
| 	p, err := parse(data) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return MetaData{}, err |  | ||||||
| 	} |  | ||||||
| 	md := MetaData{ |  | ||||||
| 		p.mapping, p.types, p.ordered, |  | ||||||
| 		make(map[string]bool, len(p.ordered)), nil, |  | ||||||
| 	} |  | ||||||
| 	return md, md.unify(p.mapping, indirect(rv)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // DecodeFile is just like Decode, except it will automatically read the |  | ||||||
| // contents of the file at `fpath` and decode it for you. |  | ||||||
| func DecodeFile(fpath string, v interface{}) (MetaData, error) { |  | ||||||
| 	bs, err := ioutil.ReadFile(fpath) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return MetaData{}, err |  | ||||||
| 	} |  | ||||||
| 	return Decode(string(bs), v) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // DecodeReader is just like Decode, except it will consume all bytes |  | ||||||
| // from the reader and decode it for you. |  | ||||||
| func DecodeReader(r io.Reader, v interface{}) (MetaData, error) { |  | ||||||
| 	bs, err := ioutil.ReadAll(r) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return MetaData{}, err |  | ||||||
| 	} |  | ||||||
| 	return Decode(string(bs), v) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // unify performs a sort of type unification based on the structure of `rv`, |  | ||||||
| // which is the client representation. |  | ||||||
| // |  | ||||||
| // Any type mismatch produces an error. Finding a type that we don't know |  | ||||||
| // how to handle produces an unsupported type error. |  | ||||||
| func (md *MetaData) unify(data interface{}, rv reflect.Value) error { |  | ||||||
|  |  | ||||||
| 	// Special case. Look for a `Primitive` value. |  | ||||||
| 	if rv.Type() == reflect.TypeOf((*Primitive)(nil)).Elem() { |  | ||||||
| 		// Save the undecoded data and the key context into the primitive |  | ||||||
| 		// value. |  | ||||||
| 		context := make(Key, len(md.context)) |  | ||||||
| 		copy(context, md.context) |  | ||||||
| 		rv.Set(reflect.ValueOf(Primitive{ |  | ||||||
| 			undecoded: data, |  | ||||||
| 			context:   context, |  | ||||||
| 		})) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Special case. Unmarshaler Interface support. |  | ||||||
| 	if rv.CanAddr() { |  | ||||||
| 		if v, ok := rv.Addr().Interface().(Unmarshaler); ok { |  | ||||||
| 			return v.UnmarshalTOML(data) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Special case. Handle time.Time values specifically. |  | ||||||
| 	// TODO: Remove this code when we decide to drop support for Go 1.1. |  | ||||||
| 	// This isn't necessary in Go 1.2 because time.Time satisfies the encoding |  | ||||||
| 	// interfaces. |  | ||||||
| 	if rv.Type().AssignableTo(rvalue(time.Time{}).Type()) { |  | ||||||
| 		return md.unifyDatetime(data, rv) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Special case. Look for a value satisfying the TextUnmarshaler interface. |  | ||||||
| 	if v, ok := rv.Interface().(TextUnmarshaler); ok { |  | ||||||
| 		return md.unifyText(data, v) |  | ||||||
| 	} |  | ||||||
| 	// BUG(burntsushi) |  | ||||||
| 	// The behavior here is incorrect whenever a Go type satisfies the |  | ||||||
| 	// encoding.TextUnmarshaler interface but also corresponds to a TOML |  | ||||||
| 	// hash or array. In particular, the unmarshaler should only be applied |  | ||||||
| 	// to primitive TOML values. But at this point, it will be applied to |  | ||||||
| 	// all kinds of values and produce an incorrect error whenever those values |  | ||||||
| 	// are hashes or arrays (including arrays of tables). |  | ||||||
|  |  | ||||||
| 	k := rv.Kind() |  | ||||||
|  |  | ||||||
| 	// laziness |  | ||||||
| 	if k >= reflect.Int && k <= reflect.Uint64 { |  | ||||||
| 		return md.unifyInt(data, rv) |  | ||||||
| 	} |  | ||||||
| 	switch k { |  | ||||||
| 	case reflect.Ptr: |  | ||||||
| 		elem := reflect.New(rv.Type().Elem()) |  | ||||||
| 		err := md.unify(data, reflect.Indirect(elem)) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		rv.Set(elem) |  | ||||||
| 		return nil |  | ||||||
| 	case reflect.Struct: |  | ||||||
| 		return md.unifyStruct(data, rv) |  | ||||||
| 	case reflect.Map: |  | ||||||
| 		return md.unifyMap(data, rv) |  | ||||||
| 	case reflect.Array: |  | ||||||
| 		return md.unifyArray(data, rv) |  | ||||||
| 	case reflect.Slice: |  | ||||||
| 		return md.unifySlice(data, rv) |  | ||||||
| 	case reflect.String: |  | ||||||
| 		return md.unifyString(data, rv) |  | ||||||
| 	case reflect.Bool: |  | ||||||
| 		return md.unifyBool(data, rv) |  | ||||||
| 	case reflect.Interface: |  | ||||||
| 		// we only support empty interfaces. |  | ||||||
| 		if rv.NumMethod() > 0 { |  | ||||||
| 			return e("unsupported type %s", rv.Type()) |  | ||||||
| 		} |  | ||||||
| 		return md.unifyAnything(data, rv) |  | ||||||
| 	case reflect.Float32: |  | ||||||
| 		fallthrough |  | ||||||
| 	case reflect.Float64: |  | ||||||
| 		return md.unifyFloat64(data, rv) |  | ||||||
| 	} |  | ||||||
| 	return e("unsupported type %s", rv.Kind()) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyStruct(mapping interface{}, rv reflect.Value) error { |  | ||||||
| 	tmap, ok := mapping.(map[string]interface{}) |  | ||||||
| 	if !ok { |  | ||||||
| 		if mapping == nil { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		return e("type mismatch for %s: expected table but found %T", |  | ||||||
| 			rv.Type().String(), mapping) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for key, datum := range tmap { |  | ||||||
| 		var f *field |  | ||||||
| 		fields := cachedTypeFields(rv.Type()) |  | ||||||
| 		for i := range fields { |  | ||||||
| 			ff := &fields[i] |  | ||||||
| 			if ff.name == key { |  | ||||||
| 				f = ff |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 			if f == nil && strings.EqualFold(ff.name, key) { |  | ||||||
| 				f = ff |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if f != nil { |  | ||||||
| 			subv := rv |  | ||||||
| 			for _, i := range f.index { |  | ||||||
| 				subv = indirect(subv.Field(i)) |  | ||||||
| 			} |  | ||||||
| 			if isUnifiable(subv) { |  | ||||||
| 				md.decoded[md.context.add(key).String()] = true |  | ||||||
| 				md.context = append(md.context, key) |  | ||||||
| 				if err := md.unify(datum, subv); err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				md.context = md.context[0 : len(md.context)-1] |  | ||||||
| 			} else if f.name != "" { |  | ||||||
| 				// Bad user! No soup for you! |  | ||||||
| 				return e("cannot write unexported field %s.%s", |  | ||||||
| 					rv.Type().String(), f.name) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyMap(mapping interface{}, rv reflect.Value) error { |  | ||||||
| 	tmap, ok := mapping.(map[string]interface{}) |  | ||||||
| 	if !ok { |  | ||||||
| 		if tmap == nil { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		return badtype("map", mapping) |  | ||||||
| 	} |  | ||||||
| 	if rv.IsNil() { |  | ||||||
| 		rv.Set(reflect.MakeMap(rv.Type())) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range tmap { |  | ||||||
| 		md.decoded[md.context.add(k).String()] = true |  | ||||||
| 		md.context = append(md.context, k) |  | ||||||
|  |  | ||||||
| 		rvkey := indirect(reflect.New(rv.Type().Key())) |  | ||||||
| 		rvval := reflect.Indirect(reflect.New(rv.Type().Elem())) |  | ||||||
| 		if err := md.unify(v, rvval); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		md.context = md.context[0 : len(md.context)-1] |  | ||||||
|  |  | ||||||
| 		rvkey.SetString(k) |  | ||||||
| 		rv.SetMapIndex(rvkey, rvval) |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyArray(data interface{}, rv reflect.Value) error { |  | ||||||
| 	datav := reflect.ValueOf(data) |  | ||||||
| 	if datav.Kind() != reflect.Slice { |  | ||||||
| 		if !datav.IsValid() { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		return badtype("slice", data) |  | ||||||
| 	} |  | ||||||
| 	sliceLen := datav.Len() |  | ||||||
| 	if sliceLen != rv.Len() { |  | ||||||
| 		return e("expected array length %d; got TOML array of length %d", |  | ||||||
| 			rv.Len(), sliceLen) |  | ||||||
| 	} |  | ||||||
| 	return md.unifySliceArray(datav, rv) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifySlice(data interface{}, rv reflect.Value) error { |  | ||||||
| 	datav := reflect.ValueOf(data) |  | ||||||
| 	if datav.Kind() != reflect.Slice { |  | ||||||
| 		if !datav.IsValid() { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		return badtype("slice", data) |  | ||||||
| 	} |  | ||||||
| 	n := datav.Len() |  | ||||||
| 	if rv.IsNil() || rv.Cap() < n { |  | ||||||
| 		rv.Set(reflect.MakeSlice(rv.Type(), n, n)) |  | ||||||
| 	} |  | ||||||
| 	rv.SetLen(n) |  | ||||||
| 	return md.unifySliceArray(datav, rv) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifySliceArray(data, rv reflect.Value) error { |  | ||||||
| 	sliceLen := data.Len() |  | ||||||
| 	for i := 0; i < sliceLen; i++ { |  | ||||||
| 		v := data.Index(i).Interface() |  | ||||||
| 		sliceval := indirect(rv.Index(i)) |  | ||||||
| 		if err := md.unify(v, sliceval); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyDatetime(data interface{}, rv reflect.Value) error { |  | ||||||
| 	if _, ok := data.(time.Time); ok { |  | ||||||
| 		rv.Set(reflect.ValueOf(data)) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return badtype("time.Time", data) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyString(data interface{}, rv reflect.Value) error { |  | ||||||
| 	if s, ok := data.(string); ok { |  | ||||||
| 		rv.SetString(s) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return badtype("string", data) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyFloat64(data interface{}, rv reflect.Value) error { |  | ||||||
| 	if num, ok := data.(float64); ok { |  | ||||||
| 		switch rv.Kind() { |  | ||||||
| 		case reflect.Float32: |  | ||||||
| 			fallthrough |  | ||||||
| 		case reflect.Float64: |  | ||||||
| 			rv.SetFloat(num) |  | ||||||
| 		default: |  | ||||||
| 			panic("bug") |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return badtype("float", data) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyInt(data interface{}, rv reflect.Value) error { |  | ||||||
| 	if num, ok := data.(int64); ok { |  | ||||||
| 		if rv.Kind() >= reflect.Int && rv.Kind() <= reflect.Int64 { |  | ||||||
| 			switch rv.Kind() { |  | ||||||
| 			case reflect.Int, reflect.Int64: |  | ||||||
| 				// No bounds checking necessary. |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				if num < math.MinInt8 || num > math.MaxInt8 { |  | ||||||
| 					return e("value %d is out of range for int8", num) |  | ||||||
| 				} |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				if num < math.MinInt16 || num > math.MaxInt16 { |  | ||||||
| 					return e("value %d is out of range for int16", num) |  | ||||||
| 				} |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				if num < math.MinInt32 || num > math.MaxInt32 { |  | ||||||
| 					return e("value %d is out of range for int32", num) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			rv.SetInt(num) |  | ||||||
| 		} else if rv.Kind() >= reflect.Uint && rv.Kind() <= reflect.Uint64 { |  | ||||||
| 			unum := uint64(num) |  | ||||||
| 			switch rv.Kind() { |  | ||||||
| 			case reflect.Uint, reflect.Uint64: |  | ||||||
| 				// No bounds checking necessary. |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				if num < 0 || unum > math.MaxUint8 { |  | ||||||
| 					return e("value %d is out of range for uint8", num) |  | ||||||
| 				} |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				if num < 0 || unum > math.MaxUint16 { |  | ||||||
| 					return e("value %d is out of range for uint16", num) |  | ||||||
| 				} |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				if num < 0 || unum > math.MaxUint32 { |  | ||||||
| 					return e("value %d is out of range for uint32", num) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			rv.SetUint(unum) |  | ||||||
| 		} else { |  | ||||||
| 			panic("unreachable") |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return badtype("integer", data) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyBool(data interface{}, rv reflect.Value) error { |  | ||||||
| 	if b, ok := data.(bool); ok { |  | ||||||
| 		rv.SetBool(b) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return badtype("boolean", data) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyAnything(data interface{}, rv reflect.Value) error { |  | ||||||
| 	rv.Set(reflect.ValueOf(data)) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (md *MetaData) unifyText(data interface{}, v TextUnmarshaler) error { |  | ||||||
| 	var s string |  | ||||||
| 	switch sdata := data.(type) { |  | ||||||
| 	case TextMarshaler: |  | ||||||
| 		text, err := sdata.MarshalText() |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		s = string(text) |  | ||||||
| 	case fmt.Stringer: |  | ||||||
| 		s = sdata.String() |  | ||||||
| 	case string: |  | ||||||
| 		s = sdata |  | ||||||
| 	case bool: |  | ||||||
| 		s = fmt.Sprintf("%v", sdata) |  | ||||||
| 	case int64: |  | ||||||
| 		s = fmt.Sprintf("%d", sdata) |  | ||||||
| 	case float64: |  | ||||||
| 		s = fmt.Sprintf("%f", sdata) |  | ||||||
| 	default: |  | ||||||
| 		return badtype("primitive (string-like)", data) |  | ||||||
| 	} |  | ||||||
| 	if err := v.UnmarshalText([]byte(s)); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // rvalue returns a reflect.Value of `v`. All pointers are resolved. |  | ||||||
| func rvalue(v interface{}) reflect.Value { |  | ||||||
| 	return indirect(reflect.ValueOf(v)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // indirect returns the value pointed to by a pointer. |  | ||||||
| // Pointers are followed until the value is not a pointer. |  | ||||||
| // New values are allocated for each nil pointer. |  | ||||||
| // |  | ||||||
| // An exception to this rule is if the value satisfies an interface of |  | ||||||
| // interest to us (like encoding.TextUnmarshaler). |  | ||||||
| func indirect(v reflect.Value) reflect.Value { |  | ||||||
| 	if v.Kind() != reflect.Ptr { |  | ||||||
| 		if v.CanSet() { |  | ||||||
| 			pv := v.Addr() |  | ||||||
| 			if _, ok := pv.Interface().(TextUnmarshaler); ok { |  | ||||||
| 				return pv |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return v |  | ||||||
| 	} |  | ||||||
| 	if v.IsNil() { |  | ||||||
| 		v.Set(reflect.New(v.Type().Elem())) |  | ||||||
| 	} |  | ||||||
| 	return indirect(reflect.Indirect(v)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isUnifiable(rv reflect.Value) bool { |  | ||||||
| 	if rv.CanSet() { |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	if _, ok := rv.Interface().(TextUnmarshaler); ok { |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func badtype(expected string, data interface{}) error { |  | ||||||
| 	return e("cannot load TOML value of type %T into a Go %s", data, expected) |  | ||||||
| } |  | ||||||
							
								
								
									
										121
									
								
								vendor/github.com/BurntSushi/toml/decode_meta.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,121 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| import "strings" |  | ||||||
|  |  | ||||||
| // MetaData allows access to meta information about TOML data that may not |  | ||||||
| // be inferrable via reflection. In particular, whether a key has been defined |  | ||||||
| // and the TOML type of a key. |  | ||||||
| type MetaData struct { |  | ||||||
| 	mapping map[string]interface{} |  | ||||||
| 	types   map[string]tomlType |  | ||||||
| 	keys    []Key |  | ||||||
| 	decoded map[string]bool |  | ||||||
| 	context Key // Used only during decoding. |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // IsDefined returns true if the key given exists in the TOML data. The key |  | ||||||
| // should be specified hierarchially. e.g., |  | ||||||
| // |  | ||||||
| //	// access the TOML key 'a.b.c' |  | ||||||
| //	IsDefined("a", "b", "c") |  | ||||||
| // |  | ||||||
| // IsDefined will return false if an empty key given. Keys are case sensitive. |  | ||||||
| func (md *MetaData) IsDefined(key ...string) bool { |  | ||||||
| 	if len(key) == 0 { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var hash map[string]interface{} |  | ||||||
| 	var ok bool |  | ||||||
| 	var hashOrVal interface{} = md.mapping |  | ||||||
| 	for _, k := range key { |  | ||||||
| 		if hash, ok = hashOrVal.(map[string]interface{}); !ok { |  | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 		if hashOrVal, ok = hash[k]; !ok { |  | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Type returns a string representation of the type of the key specified. |  | ||||||
| // |  | ||||||
| // Type will return the empty string if given an empty key or a key that |  | ||||||
| // does not exist. Keys are case sensitive. |  | ||||||
| func (md *MetaData) Type(key ...string) string { |  | ||||||
| 	fullkey := strings.Join(key, ".") |  | ||||||
| 	if typ, ok := md.types[fullkey]; ok { |  | ||||||
| 		return typ.typeString() |  | ||||||
| 	} |  | ||||||
| 	return "" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Key is the type of any TOML key, including key groups. Use (MetaData).Keys |  | ||||||
| // to get values of this type. |  | ||||||
| type Key []string |  | ||||||
|  |  | ||||||
| func (k Key) String() string { |  | ||||||
| 	return strings.Join(k, ".") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (k Key) maybeQuotedAll() string { |  | ||||||
| 	var ss []string |  | ||||||
| 	for i := range k { |  | ||||||
| 		ss = append(ss, k.maybeQuoted(i)) |  | ||||||
| 	} |  | ||||||
| 	return strings.Join(ss, ".") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (k Key) maybeQuoted(i int) string { |  | ||||||
| 	quote := false |  | ||||||
| 	for _, c := range k[i] { |  | ||||||
| 		if !isBareKeyChar(c) { |  | ||||||
| 			quote = true |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	if quote { |  | ||||||
| 		return "\"" + strings.Replace(k[i], "\"", "\\\"", -1) + "\"" |  | ||||||
| 	} |  | ||||||
| 	return k[i] |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (k Key) add(piece string) Key { |  | ||||||
| 	newKey := make(Key, len(k)+1) |  | ||||||
| 	copy(newKey, k) |  | ||||||
| 	newKey[len(k)] = piece |  | ||||||
| 	return newKey |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Keys returns a slice of every key in the TOML data, including key groups. |  | ||||||
| // Each key is itself a slice, where the first element is the top of the |  | ||||||
| // hierarchy and the last is the most specific. |  | ||||||
| // |  | ||||||
| // The list will have the same order as the keys appeared in the TOML data. |  | ||||||
| // |  | ||||||
| // All keys returned are non-empty. |  | ||||||
| func (md *MetaData) Keys() []Key { |  | ||||||
| 	return md.keys |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Undecoded returns all keys that have not been decoded in the order in which |  | ||||||
| // they appear in the original TOML document. |  | ||||||
| // |  | ||||||
| // This includes keys that haven't been decoded because of a Primitive value. |  | ||||||
| // Once the Primitive value is decoded, the keys will be considered decoded. |  | ||||||
| // |  | ||||||
| // Also note that decoding into an empty interface will result in no decoding, |  | ||||||
| // and so no keys will be considered decoded. |  | ||||||
| // |  | ||||||
| // In this sense, the Undecoded keys correspond to keys in the TOML document |  | ||||||
| // that do not have a concrete type in your representation. |  | ||||||
| func (md *MetaData) Undecoded() []Key { |  | ||||||
| 	undecoded := make([]Key, 0, len(md.keys)) |  | ||||||
| 	for _, key := range md.keys { |  | ||||||
| 		if !md.decoded[key.String()] { |  | ||||||
| 			undecoded = append(undecoded, key) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return undecoded |  | ||||||
| } |  | ||||||
							
								
								
									
										27
									
								
								vendor/github.com/BurntSushi/toml/doc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,27 +0,0 @@ | |||||||
| /* |  | ||||||
| Package toml provides facilities for decoding and encoding TOML configuration |  | ||||||
| files via reflection. There is also support for delaying decoding with |  | ||||||
| the Primitive type, and querying the set of keys in a TOML document with the |  | ||||||
| MetaData type. |  | ||||||
|  |  | ||||||
| The specification implemented: https://github.com/toml-lang/toml |  | ||||||
|  |  | ||||||
| The sub-command github.com/BurntSushi/toml/cmd/tomlv can be used to verify |  | ||||||
| whether a file is a valid TOML document. It can also be used to print the |  | ||||||
| type of each key in a TOML document. |  | ||||||
|  |  | ||||||
| Testing |  | ||||||
|  |  | ||||||
| There are two important types of tests used for this package. The first is |  | ||||||
| contained inside '*_test.go' files and uses the standard Go unit testing |  | ||||||
| framework. These tests are primarily devoted to holistically testing the |  | ||||||
| decoder and encoder. |  | ||||||
|  |  | ||||||
| The second type of testing is used to verify the implementation's adherence |  | ||||||
| to the TOML specification. These tests have been factored into their own |  | ||||||
| project: https://github.com/BurntSushi/toml-test |  | ||||||
|  |  | ||||||
| The reason the tests are in a separate project is so that they can be used by |  | ||||||
| any implementation of TOML. Namely, it is language agnostic. |  | ||||||
| */ |  | ||||||
| package toml |  | ||||||
							
								
								
									
										568
									
								
								vendor/github.com/BurntSushi/toml/encode.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,568 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bufio" |  | ||||||
| 	"errors" |  | ||||||
| 	"fmt" |  | ||||||
| 	"io" |  | ||||||
| 	"reflect" |  | ||||||
| 	"sort" |  | ||||||
| 	"strconv" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type tomlEncodeError struct{ error } |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	errArrayMixedElementTypes = errors.New( |  | ||||||
| 		"toml: cannot encode array with mixed element types") |  | ||||||
| 	errArrayNilElement = errors.New( |  | ||||||
| 		"toml: cannot encode array with nil element") |  | ||||||
| 	errNonString = errors.New( |  | ||||||
| 		"toml: cannot encode a map with non-string key type") |  | ||||||
| 	errAnonNonStruct = errors.New( |  | ||||||
| 		"toml: cannot encode an anonymous field that is not a struct") |  | ||||||
| 	errArrayNoTable = errors.New( |  | ||||||
| 		"toml: TOML array element cannot contain a table") |  | ||||||
| 	errNoKey = errors.New( |  | ||||||
| 		"toml: top-level values must be Go maps or structs") |  | ||||||
| 	errAnything = errors.New("") // used in testing |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var quotedReplacer = strings.NewReplacer( |  | ||||||
| 	"\t", "\\t", |  | ||||||
| 	"\n", "\\n", |  | ||||||
| 	"\r", "\\r", |  | ||||||
| 	"\"", "\\\"", |  | ||||||
| 	"\\", "\\\\", |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Encoder controls the encoding of Go values to a TOML document to some |  | ||||||
| // io.Writer. |  | ||||||
| // |  | ||||||
| // The indentation level can be controlled with the Indent field. |  | ||||||
| type Encoder struct { |  | ||||||
| 	// A single indentation level. By default it is two spaces. |  | ||||||
| 	Indent string |  | ||||||
|  |  | ||||||
| 	// hasWritten is whether we have written any output to w yet. |  | ||||||
| 	hasWritten bool |  | ||||||
| 	w          *bufio.Writer |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewEncoder returns a TOML encoder that encodes Go values to the io.Writer |  | ||||||
| // given. By default, a single indentation level is 2 spaces. |  | ||||||
| func NewEncoder(w io.Writer) *Encoder { |  | ||||||
| 	return &Encoder{ |  | ||||||
| 		w:      bufio.NewWriter(w), |  | ||||||
| 		Indent: "  ", |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Encode writes a TOML representation of the Go value to the underlying |  | ||||||
| // io.Writer. If the value given cannot be encoded to a valid TOML document, |  | ||||||
| // then an error is returned. |  | ||||||
| // |  | ||||||
| // The mapping between Go values and TOML values should be precisely the same |  | ||||||
| // as for the Decode* functions. Similarly, the TextMarshaler interface is |  | ||||||
| // supported by encoding the resulting bytes as strings. (If you want to write |  | ||||||
| // arbitrary binary data then you will need to use something like base64 since |  | ||||||
| // TOML does not have any binary types.) |  | ||||||
| // |  | ||||||
| // When encoding TOML hashes (i.e., Go maps or structs), keys without any |  | ||||||
| // sub-hashes are encoded first. |  | ||||||
| // |  | ||||||
| // If a Go map is encoded, then its keys are sorted alphabetically for |  | ||||||
| // deterministic output. More control over this behavior may be provided if |  | ||||||
| // there is demand for it. |  | ||||||
| // |  | ||||||
| // Encoding Go values without a corresponding TOML representation---like map |  | ||||||
| // types with non-string keys---will cause an error to be returned. Similarly |  | ||||||
| // for mixed arrays/slices, arrays/slices with nil elements, embedded |  | ||||||
| // non-struct types and nested slices containing maps or structs. |  | ||||||
| // (e.g., [][]map[string]string is not allowed but []map[string]string is OK |  | ||||||
| // and so is []map[string][]string.) |  | ||||||
| func (enc *Encoder) Encode(v interface{}) error { |  | ||||||
| 	rv := eindirect(reflect.ValueOf(v)) |  | ||||||
| 	if err := enc.safeEncode(Key([]string{}), rv); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	return enc.w.Flush() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) safeEncode(key Key, rv reflect.Value) (err error) { |  | ||||||
| 	defer func() { |  | ||||||
| 		if r := recover(); r != nil { |  | ||||||
| 			if terr, ok := r.(tomlEncodeError); ok { |  | ||||||
| 				err = terr.error |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 			panic(r) |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
| 	enc.encode(key, rv) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) encode(key Key, rv reflect.Value) { |  | ||||||
| 	// Special case. Time needs to be in ISO8601 format. |  | ||||||
| 	// Special case. If we can marshal the type to text, then we used that. |  | ||||||
| 	// Basically, this prevents the encoder for handling these types as |  | ||||||
| 	// generic structs (or whatever the underlying type of a TextMarshaler is). |  | ||||||
| 	switch rv.Interface().(type) { |  | ||||||
| 	case time.Time, TextMarshaler: |  | ||||||
| 		enc.keyEqElement(key, rv) |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	k := rv.Kind() |  | ||||||
| 	switch k { |  | ||||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, |  | ||||||
| 		reflect.Int64, |  | ||||||
| 		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, |  | ||||||
| 		reflect.Uint64, |  | ||||||
| 		reflect.Float32, reflect.Float64, reflect.String, reflect.Bool: |  | ||||||
| 		enc.keyEqElement(key, rv) |  | ||||||
| 	case reflect.Array, reflect.Slice: |  | ||||||
| 		if typeEqual(tomlArrayHash, tomlTypeOfGo(rv)) { |  | ||||||
| 			enc.eArrayOfTables(key, rv) |  | ||||||
| 		} else { |  | ||||||
| 			enc.keyEqElement(key, rv) |  | ||||||
| 		} |  | ||||||
| 	case reflect.Interface: |  | ||||||
| 		if rv.IsNil() { |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 		enc.encode(key, rv.Elem()) |  | ||||||
| 	case reflect.Map: |  | ||||||
| 		if rv.IsNil() { |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 		enc.eTable(key, rv) |  | ||||||
| 	case reflect.Ptr: |  | ||||||
| 		if rv.IsNil() { |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
| 		enc.encode(key, rv.Elem()) |  | ||||||
| 	case reflect.Struct: |  | ||||||
| 		enc.eTable(key, rv) |  | ||||||
| 	default: |  | ||||||
| 		panic(e("unsupported type for key '%s': %s", key, k)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // eElement encodes any value that can be an array element (primitives and |  | ||||||
| // arrays). |  | ||||||
| func (enc *Encoder) eElement(rv reflect.Value) { |  | ||||||
| 	switch v := rv.Interface().(type) { |  | ||||||
| 	case time.Time: |  | ||||||
| 		// Special case time.Time as a primitive. Has to come before |  | ||||||
| 		// TextMarshaler below because time.Time implements |  | ||||||
| 		// encoding.TextMarshaler, but we need to always use UTC. |  | ||||||
| 		enc.wf(v.UTC().Format("2006-01-02T15:04:05Z")) |  | ||||||
| 		return |  | ||||||
| 	case TextMarshaler: |  | ||||||
| 		// Special case. Use text marshaler if it's available for this value. |  | ||||||
| 		if s, err := v.MarshalText(); err != nil { |  | ||||||
| 			encPanic(err) |  | ||||||
| 		} else { |  | ||||||
| 			enc.writeQuoted(string(s)) |  | ||||||
| 		} |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	switch rv.Kind() { |  | ||||||
| 	case reflect.Bool: |  | ||||||
| 		enc.wf(strconv.FormatBool(rv.Bool())) |  | ||||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, |  | ||||||
| 		reflect.Int64: |  | ||||||
| 		enc.wf(strconv.FormatInt(rv.Int(), 10)) |  | ||||||
| 	case reflect.Uint, reflect.Uint8, reflect.Uint16, |  | ||||||
| 		reflect.Uint32, reflect.Uint64: |  | ||||||
| 		enc.wf(strconv.FormatUint(rv.Uint(), 10)) |  | ||||||
| 	case reflect.Float32: |  | ||||||
| 		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 32))) |  | ||||||
| 	case reflect.Float64: |  | ||||||
| 		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 64))) |  | ||||||
| 	case reflect.Array, reflect.Slice: |  | ||||||
| 		enc.eArrayOrSliceElement(rv) |  | ||||||
| 	case reflect.Interface: |  | ||||||
| 		enc.eElement(rv.Elem()) |  | ||||||
| 	case reflect.String: |  | ||||||
| 		enc.writeQuoted(rv.String()) |  | ||||||
| 	default: |  | ||||||
| 		panic(e("unexpected primitive type: %s", rv.Kind())) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // By the TOML spec, all floats must have a decimal with at least one |  | ||||||
| // number on either side. |  | ||||||
| func floatAddDecimal(fstr string) string { |  | ||||||
| 	if !strings.Contains(fstr, ".") { |  | ||||||
| 		return fstr + ".0" |  | ||||||
| 	} |  | ||||||
| 	return fstr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) writeQuoted(s string) { |  | ||||||
| 	enc.wf("\"%s\"", quotedReplacer.Replace(s)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eArrayOrSliceElement(rv reflect.Value) { |  | ||||||
| 	length := rv.Len() |  | ||||||
| 	enc.wf("[") |  | ||||||
| 	for i := 0; i < length; i++ { |  | ||||||
| 		elem := rv.Index(i) |  | ||||||
| 		enc.eElement(elem) |  | ||||||
| 		if i != length-1 { |  | ||||||
| 			enc.wf(", ") |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	enc.wf("]") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eArrayOfTables(key Key, rv reflect.Value) { |  | ||||||
| 	if len(key) == 0 { |  | ||||||
| 		encPanic(errNoKey) |  | ||||||
| 	} |  | ||||||
| 	for i := 0; i < rv.Len(); i++ { |  | ||||||
| 		trv := rv.Index(i) |  | ||||||
| 		if isNil(trv) { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		panicIfInvalidKey(key) |  | ||||||
| 		enc.newline() |  | ||||||
| 		enc.wf("%s[[%s]]", enc.indentStr(key), key.maybeQuotedAll()) |  | ||||||
| 		enc.newline() |  | ||||||
| 		enc.eMapOrStruct(key, trv) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eTable(key Key, rv reflect.Value) { |  | ||||||
| 	panicIfInvalidKey(key) |  | ||||||
| 	if len(key) == 1 { |  | ||||||
| 		// Output an extra newline between top-level tables. |  | ||||||
| 		// (The newline isn't written if nothing else has been written though.) |  | ||||||
| 		enc.newline() |  | ||||||
| 	} |  | ||||||
| 	if len(key) > 0 { |  | ||||||
| 		enc.wf("%s[%s]", enc.indentStr(key), key.maybeQuotedAll()) |  | ||||||
| 		enc.newline() |  | ||||||
| 	} |  | ||||||
| 	enc.eMapOrStruct(key, rv) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eMapOrStruct(key Key, rv reflect.Value) { |  | ||||||
| 	switch rv := eindirect(rv); rv.Kind() { |  | ||||||
| 	case reflect.Map: |  | ||||||
| 		enc.eMap(key, rv) |  | ||||||
| 	case reflect.Struct: |  | ||||||
| 		enc.eStruct(key, rv) |  | ||||||
| 	default: |  | ||||||
| 		panic("eTable: unhandled reflect.Value Kind: " + rv.Kind().String()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eMap(key Key, rv reflect.Value) { |  | ||||||
| 	rt := rv.Type() |  | ||||||
| 	if rt.Key().Kind() != reflect.String { |  | ||||||
| 		encPanic(errNonString) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Sort keys so that we have deterministic output. And write keys directly |  | ||||||
| 	// underneath this key first, before writing sub-structs or sub-maps. |  | ||||||
| 	var mapKeysDirect, mapKeysSub []string |  | ||||||
| 	for _, mapKey := range rv.MapKeys() { |  | ||||||
| 		k := mapKey.String() |  | ||||||
| 		if typeIsHash(tomlTypeOfGo(rv.MapIndex(mapKey))) { |  | ||||||
| 			mapKeysSub = append(mapKeysSub, k) |  | ||||||
| 		} else { |  | ||||||
| 			mapKeysDirect = append(mapKeysDirect, k) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var writeMapKeys = func(mapKeys []string) { |  | ||||||
| 		sort.Strings(mapKeys) |  | ||||||
| 		for _, mapKey := range mapKeys { |  | ||||||
| 			mrv := rv.MapIndex(reflect.ValueOf(mapKey)) |  | ||||||
| 			if isNil(mrv) { |  | ||||||
| 				// Don't write anything for nil fields. |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			enc.encode(key.add(mapKey), mrv) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	writeMapKeys(mapKeysDirect) |  | ||||||
| 	writeMapKeys(mapKeysSub) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) eStruct(key Key, rv reflect.Value) { |  | ||||||
| 	// Write keys for fields directly under this key first, because if we write |  | ||||||
| 	// a field that creates a new table, then all keys under it will be in that |  | ||||||
| 	// table (not the one we're writing here). |  | ||||||
| 	rt := rv.Type() |  | ||||||
| 	var fieldsDirect, fieldsSub [][]int |  | ||||||
| 	var addFields func(rt reflect.Type, rv reflect.Value, start []int) |  | ||||||
| 	addFields = func(rt reflect.Type, rv reflect.Value, start []int) { |  | ||||||
| 		for i := 0; i < rt.NumField(); i++ { |  | ||||||
| 			f := rt.Field(i) |  | ||||||
| 			// skip unexported fields |  | ||||||
| 			if f.PkgPath != "" && !f.Anonymous { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			frv := rv.Field(i) |  | ||||||
| 			if f.Anonymous { |  | ||||||
| 				t := f.Type |  | ||||||
| 				switch t.Kind() { |  | ||||||
| 				case reflect.Struct: |  | ||||||
| 					// Treat anonymous struct fields with |  | ||||||
| 					// tag names as though they are not |  | ||||||
| 					// anonymous, like encoding/json does. |  | ||||||
| 					if getOptions(f.Tag).name == "" { |  | ||||||
| 						addFields(t, frv, f.Index) |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 				case reflect.Ptr: |  | ||||||
| 					if t.Elem().Kind() == reflect.Struct && |  | ||||||
| 						getOptions(f.Tag).name == "" { |  | ||||||
| 						if !frv.IsNil() { |  | ||||||
| 							addFields(t.Elem(), frv.Elem(), f.Index) |  | ||||||
| 						} |  | ||||||
| 						continue |  | ||||||
| 					} |  | ||||||
| 					// Fall through to the normal field encoding logic below |  | ||||||
| 					// for non-struct anonymous fields. |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if typeIsHash(tomlTypeOfGo(frv)) { |  | ||||||
| 				fieldsSub = append(fieldsSub, append(start, f.Index...)) |  | ||||||
| 			} else { |  | ||||||
| 				fieldsDirect = append(fieldsDirect, append(start, f.Index...)) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	addFields(rt, rv, nil) |  | ||||||
|  |  | ||||||
| 	var writeFields = func(fields [][]int) { |  | ||||||
| 		for _, fieldIndex := range fields { |  | ||||||
| 			sft := rt.FieldByIndex(fieldIndex) |  | ||||||
| 			sf := rv.FieldByIndex(fieldIndex) |  | ||||||
| 			if isNil(sf) { |  | ||||||
| 				// Don't write anything for nil fields. |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			opts := getOptions(sft.Tag) |  | ||||||
| 			if opts.skip { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			keyName := sft.Name |  | ||||||
| 			if opts.name != "" { |  | ||||||
| 				keyName = opts.name |  | ||||||
| 			} |  | ||||||
| 			if opts.omitempty && isEmpty(sf) { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			if opts.omitzero && isZero(sf) { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			enc.encode(key.add(keyName), sf) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	writeFields(fieldsDirect) |  | ||||||
| 	writeFields(fieldsSub) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // tomlTypeName returns the TOML type name of the Go value's type. It is |  | ||||||
| // used to determine whether the types of array elements are mixed (which is |  | ||||||
| // forbidden). If the Go value is nil, then it is illegal for it to be an array |  | ||||||
| // element, and valueIsNil is returned as true. |  | ||||||
|  |  | ||||||
| // Returns the TOML type of a Go value. The type may be `nil`, which means |  | ||||||
| // no concrete TOML type could be found. |  | ||||||
| func tomlTypeOfGo(rv reflect.Value) tomlType { |  | ||||||
| 	if isNil(rv) || !rv.IsValid() { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	switch rv.Kind() { |  | ||||||
| 	case reflect.Bool: |  | ||||||
| 		return tomlBool |  | ||||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, |  | ||||||
| 		reflect.Int64, |  | ||||||
| 		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, |  | ||||||
| 		reflect.Uint64: |  | ||||||
| 		return tomlInteger |  | ||||||
| 	case reflect.Float32, reflect.Float64: |  | ||||||
| 		return tomlFloat |  | ||||||
| 	case reflect.Array, reflect.Slice: |  | ||||||
| 		if typeEqual(tomlHash, tomlArrayType(rv)) { |  | ||||||
| 			return tomlArrayHash |  | ||||||
| 		} |  | ||||||
| 		return tomlArray |  | ||||||
| 	case reflect.Ptr, reflect.Interface: |  | ||||||
| 		return tomlTypeOfGo(rv.Elem()) |  | ||||||
| 	case reflect.String: |  | ||||||
| 		return tomlString |  | ||||||
| 	case reflect.Map: |  | ||||||
| 		return tomlHash |  | ||||||
| 	case reflect.Struct: |  | ||||||
| 		switch rv.Interface().(type) { |  | ||||||
| 		case time.Time: |  | ||||||
| 			return tomlDatetime |  | ||||||
| 		case TextMarshaler: |  | ||||||
| 			return tomlString |  | ||||||
| 		default: |  | ||||||
| 			return tomlHash |  | ||||||
| 		} |  | ||||||
| 	default: |  | ||||||
| 		panic("unexpected reflect.Kind: " + rv.Kind().String()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // tomlArrayType returns the element type of a TOML array. The type returned |  | ||||||
| // may be nil if it cannot be determined (e.g., a nil slice or a zero length |  | ||||||
| // slize). This function may also panic if it finds a type that cannot be |  | ||||||
| // expressed in TOML (such as nil elements, heterogeneous arrays or directly |  | ||||||
| // nested arrays of tables). |  | ||||||
| func tomlArrayType(rv reflect.Value) tomlType { |  | ||||||
| 	if isNil(rv) || !rv.IsValid() || rv.Len() == 0 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	firstType := tomlTypeOfGo(rv.Index(0)) |  | ||||||
| 	if firstType == nil { |  | ||||||
| 		encPanic(errArrayNilElement) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	rvlen := rv.Len() |  | ||||||
| 	for i := 1; i < rvlen; i++ { |  | ||||||
| 		elem := rv.Index(i) |  | ||||||
| 		switch elemType := tomlTypeOfGo(elem); { |  | ||||||
| 		case elemType == nil: |  | ||||||
| 			encPanic(errArrayNilElement) |  | ||||||
| 		case !typeEqual(firstType, elemType): |  | ||||||
| 			encPanic(errArrayMixedElementTypes) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	// If we have a nested array, then we must make sure that the nested |  | ||||||
| 	// array contains ONLY primitives. |  | ||||||
| 	// This checks arbitrarily nested arrays. |  | ||||||
| 	if typeEqual(firstType, tomlArray) || typeEqual(firstType, tomlArrayHash) { |  | ||||||
| 		nest := tomlArrayType(eindirect(rv.Index(0))) |  | ||||||
| 		if typeEqual(nest, tomlHash) || typeEqual(nest, tomlArrayHash) { |  | ||||||
| 			encPanic(errArrayNoTable) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return firstType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type tagOptions struct { |  | ||||||
| 	skip      bool // "-" |  | ||||||
| 	name      string |  | ||||||
| 	omitempty bool |  | ||||||
| 	omitzero  bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func getOptions(tag reflect.StructTag) tagOptions { |  | ||||||
| 	t := tag.Get("toml") |  | ||||||
| 	if t == "-" { |  | ||||||
| 		return tagOptions{skip: true} |  | ||||||
| 	} |  | ||||||
| 	var opts tagOptions |  | ||||||
| 	parts := strings.Split(t, ",") |  | ||||||
| 	opts.name = parts[0] |  | ||||||
| 	for _, s := range parts[1:] { |  | ||||||
| 		switch s { |  | ||||||
| 		case "omitempty": |  | ||||||
| 			opts.omitempty = true |  | ||||||
| 		case "omitzero": |  | ||||||
| 			opts.omitzero = true |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isZero(rv reflect.Value) bool { |  | ||||||
| 	switch rv.Kind() { |  | ||||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |  | ||||||
| 		return rv.Int() == 0 |  | ||||||
| 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: |  | ||||||
| 		return rv.Uint() == 0 |  | ||||||
| 	case reflect.Float32, reflect.Float64: |  | ||||||
| 		return rv.Float() == 0.0 |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isEmpty(rv reflect.Value) bool { |  | ||||||
| 	switch rv.Kind() { |  | ||||||
| 	case reflect.Array, reflect.Slice, reflect.Map, reflect.String: |  | ||||||
| 		return rv.Len() == 0 |  | ||||||
| 	case reflect.Bool: |  | ||||||
| 		return !rv.Bool() |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) newline() { |  | ||||||
| 	if enc.hasWritten { |  | ||||||
| 		enc.wf("\n") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) keyEqElement(key Key, val reflect.Value) { |  | ||||||
| 	if len(key) == 0 { |  | ||||||
| 		encPanic(errNoKey) |  | ||||||
| 	} |  | ||||||
| 	panicIfInvalidKey(key) |  | ||||||
| 	enc.wf("%s%s = ", enc.indentStr(key), key.maybeQuoted(len(key)-1)) |  | ||||||
| 	enc.eElement(val) |  | ||||||
| 	enc.newline() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) wf(format string, v ...interface{}) { |  | ||||||
| 	if _, err := fmt.Fprintf(enc.w, format, v...); err != nil { |  | ||||||
| 		encPanic(err) |  | ||||||
| 	} |  | ||||||
| 	enc.hasWritten = true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (enc *Encoder) indentStr(key Key) string { |  | ||||||
| 	return strings.Repeat(enc.Indent, len(key)-1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func encPanic(err error) { |  | ||||||
| 	panic(tomlEncodeError{err}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func eindirect(v reflect.Value) reflect.Value { |  | ||||||
| 	switch v.Kind() { |  | ||||||
| 	case reflect.Ptr, reflect.Interface: |  | ||||||
| 		return eindirect(v.Elem()) |  | ||||||
| 	default: |  | ||||||
| 		return v |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isNil(rv reflect.Value) bool { |  | ||||||
| 	switch rv.Kind() { |  | ||||||
| 	case reflect.Interface, reflect.Map, reflect.Ptr, reflect.Slice: |  | ||||||
| 		return rv.IsNil() |  | ||||||
| 	default: |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func panicIfInvalidKey(key Key) { |  | ||||||
| 	for _, k := range key { |  | ||||||
| 		if len(k) == 0 { |  | ||||||
| 			encPanic(e("Key '%s' is not a valid table name. Key names "+ |  | ||||||
| 				"cannot be empty.", key.maybeQuotedAll())) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isValidKeyName(s string) bool { |  | ||||||
| 	return len(s) != 0 |  | ||||||
| } |  | ||||||
							
								
								
									
										19
									
								
								vendor/github.com/BurntSushi/toml/encoding_types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,19 +0,0 @@ | |||||||
| // +build go1.2 |  | ||||||
|  |  | ||||||
| package toml |  | ||||||
|  |  | ||||||
| // In order to support Go 1.1, we define our own TextMarshaler and |  | ||||||
| // TextUnmarshaler types. For Go 1.2+, we just alias them with the |  | ||||||
| // standard library interfaces. |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"encoding" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here |  | ||||||
| // so that Go 1.1 can be supported. |  | ||||||
| type TextMarshaler encoding.TextMarshaler |  | ||||||
|  |  | ||||||
| // TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined |  | ||||||
| // here so that Go 1.1 can be supported. |  | ||||||
| type TextUnmarshaler encoding.TextUnmarshaler |  | ||||||
							
								
								
									
										18
									
								
								vendor/github.com/BurntSushi/toml/encoding_types_1.1.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,18 +0,0 @@ | |||||||
| // +build !go1.2 |  | ||||||
|  |  | ||||||
| package toml |  | ||||||
|  |  | ||||||
| // These interfaces were introduced in Go 1.2, so we add them manually when |  | ||||||
| // compiling for Go 1.1. |  | ||||||
|  |  | ||||||
| // TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here |  | ||||||
| // so that Go 1.1 can be supported. |  | ||||||
| type TextMarshaler interface { |  | ||||||
| 	MarshalText() (text []byte, err error) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined |  | ||||||
| // here so that Go 1.1 can be supported. |  | ||||||
| type TextUnmarshaler interface { |  | ||||||
| 	UnmarshalText(text []byte) error |  | ||||||
| } |  | ||||||
							
								
								
									
										953
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,953 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"strings" |  | ||||||
| 	"unicode" |  | ||||||
| 	"unicode/utf8" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type itemType int |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	itemError itemType = iota |  | ||||||
| 	itemNIL            // used in the parser to indicate no type |  | ||||||
| 	itemEOF |  | ||||||
| 	itemText |  | ||||||
| 	itemString |  | ||||||
| 	itemRawString |  | ||||||
| 	itemMultilineString |  | ||||||
| 	itemRawMultilineString |  | ||||||
| 	itemBool |  | ||||||
| 	itemInteger |  | ||||||
| 	itemFloat |  | ||||||
| 	itemDatetime |  | ||||||
| 	itemArray // the start of an array |  | ||||||
| 	itemArrayEnd |  | ||||||
| 	itemTableStart |  | ||||||
| 	itemTableEnd |  | ||||||
| 	itemArrayTableStart |  | ||||||
| 	itemArrayTableEnd |  | ||||||
| 	itemKeyStart |  | ||||||
| 	itemCommentStart |  | ||||||
| 	itemInlineTableStart |  | ||||||
| 	itemInlineTableEnd |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	eof              = 0 |  | ||||||
| 	comma            = ',' |  | ||||||
| 	tableStart       = '[' |  | ||||||
| 	tableEnd         = ']' |  | ||||||
| 	arrayTableStart  = '[' |  | ||||||
| 	arrayTableEnd    = ']' |  | ||||||
| 	tableSep         = '.' |  | ||||||
| 	keySep           = '=' |  | ||||||
| 	arrayStart       = '[' |  | ||||||
| 	arrayEnd         = ']' |  | ||||||
| 	commentStart     = '#' |  | ||||||
| 	stringStart      = '"' |  | ||||||
| 	stringEnd        = '"' |  | ||||||
| 	rawStringStart   = '\'' |  | ||||||
| 	rawStringEnd     = '\'' |  | ||||||
| 	inlineTableStart = '{' |  | ||||||
| 	inlineTableEnd   = '}' |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type stateFn func(lx *lexer) stateFn |  | ||||||
|  |  | ||||||
| type lexer struct { |  | ||||||
| 	input string |  | ||||||
| 	start int |  | ||||||
| 	pos   int |  | ||||||
| 	line  int |  | ||||||
| 	state stateFn |  | ||||||
| 	items chan item |  | ||||||
|  |  | ||||||
| 	// Allow for backing up up to three runes. |  | ||||||
| 	// This is necessary because TOML contains 3-rune tokens (""" and '''). |  | ||||||
| 	prevWidths [3]int |  | ||||||
| 	nprev      int // how many of prevWidths are in use |  | ||||||
| 	// If we emit an eof, we can still back up, but it is not OK to call |  | ||||||
| 	// next again. |  | ||||||
| 	atEOF bool |  | ||||||
|  |  | ||||||
| 	// A stack of state functions used to maintain context. |  | ||||||
| 	// The idea is to reuse parts of the state machine in various places. |  | ||||||
| 	// For example, values can appear at the top level or within arbitrarily |  | ||||||
| 	// nested arrays. The last state on the stack is used after a value has |  | ||||||
| 	// been lexed. Similarly for comments. |  | ||||||
| 	stack []stateFn |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type item struct { |  | ||||||
| 	typ  itemType |  | ||||||
| 	val  string |  | ||||||
| 	line int |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) nextItem() item { |  | ||||||
| 	for { |  | ||||||
| 		select { |  | ||||||
| 		case item := <-lx.items: |  | ||||||
| 			return item |  | ||||||
| 		default: |  | ||||||
| 			lx.state = lx.state(lx) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lex(input string) *lexer { |  | ||||||
| 	lx := &lexer{ |  | ||||||
| 		input: input, |  | ||||||
| 		state: lexTop, |  | ||||||
| 		line:  1, |  | ||||||
| 		items: make(chan item, 10), |  | ||||||
| 		stack: make([]stateFn, 0, 10), |  | ||||||
| 	} |  | ||||||
| 	return lx |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) push(state stateFn) { |  | ||||||
| 	lx.stack = append(lx.stack, state) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) pop() stateFn { |  | ||||||
| 	if len(lx.stack) == 0 { |  | ||||||
| 		return lx.errorf("BUG in lexer: no states to pop") |  | ||||||
| 	} |  | ||||||
| 	last := lx.stack[len(lx.stack)-1] |  | ||||||
| 	lx.stack = lx.stack[0 : len(lx.stack)-1] |  | ||||||
| 	return last |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) current() string { |  | ||||||
| 	return lx.input[lx.start:lx.pos] |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) emit(typ itemType) { |  | ||||||
| 	lx.items <- item{typ, lx.current(), lx.line} |  | ||||||
| 	lx.start = lx.pos |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) emitTrim(typ itemType) { |  | ||||||
| 	lx.items <- item{typ, strings.TrimSpace(lx.current()), lx.line} |  | ||||||
| 	lx.start = lx.pos |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (lx *lexer) next() (r rune) { |  | ||||||
| 	if lx.atEOF { |  | ||||||
| 		panic("next called after EOF") |  | ||||||
| 	} |  | ||||||
| 	if lx.pos >= len(lx.input) { |  | ||||||
| 		lx.atEOF = true |  | ||||||
| 		return eof |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if lx.input[lx.pos] == '\n' { |  | ||||||
| 		lx.line++ |  | ||||||
| 	} |  | ||||||
| 	lx.prevWidths[2] = lx.prevWidths[1] |  | ||||||
| 	lx.prevWidths[1] = lx.prevWidths[0] |  | ||||||
| 	if lx.nprev < 3 { |  | ||||||
| 		lx.nprev++ |  | ||||||
| 	} |  | ||||||
| 	r, w := utf8.DecodeRuneInString(lx.input[lx.pos:]) |  | ||||||
| 	lx.prevWidths[0] = w |  | ||||||
| 	lx.pos += w |  | ||||||
| 	return r |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ignore skips over the pending input before this point. |  | ||||||
| func (lx *lexer) ignore() { |  | ||||||
| 	lx.start = lx.pos |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // backup steps back one rune. Can be called only twice between calls to next. |  | ||||||
| func (lx *lexer) backup() { |  | ||||||
| 	if lx.atEOF { |  | ||||||
| 		lx.atEOF = false |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	if lx.nprev < 1 { |  | ||||||
| 		panic("backed up too far") |  | ||||||
| 	} |  | ||||||
| 	w := lx.prevWidths[0] |  | ||||||
| 	lx.prevWidths[0] = lx.prevWidths[1] |  | ||||||
| 	lx.prevWidths[1] = lx.prevWidths[2] |  | ||||||
| 	lx.nprev-- |  | ||||||
| 	lx.pos -= w |  | ||||||
| 	if lx.pos < len(lx.input) && lx.input[lx.pos] == '\n' { |  | ||||||
| 		lx.line-- |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // accept consumes the next rune if it's equal to `valid`. |  | ||||||
| func (lx *lexer) accept(valid rune) bool { |  | ||||||
| 	if lx.next() == valid { |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	lx.backup() |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // peek returns but does not consume the next rune in the input. |  | ||||||
| func (lx *lexer) peek() rune { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	lx.backup() |  | ||||||
| 	return r |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // skip ignores all input that matches the given predicate. |  | ||||||
| func (lx *lexer) skip(pred func(rune) bool) { |  | ||||||
| 	for { |  | ||||||
| 		r := lx.next() |  | ||||||
| 		if pred(r) { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		lx.backup() |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // errorf stops all lexing by emitting an error and returning `nil`. |  | ||||||
| // Note that any value that is a character is escaped if it's a special |  | ||||||
| // character (newlines, tabs, etc.). |  | ||||||
| func (lx *lexer) errorf(format string, values ...interface{}) stateFn { |  | ||||||
| 	lx.items <- item{ |  | ||||||
| 		itemError, |  | ||||||
| 		fmt.Sprintf(format, values...), |  | ||||||
| 		lx.line, |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexTop consumes elements at the top level of TOML data. |  | ||||||
| func lexTop(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isWhitespace(r) || isNL(r) { |  | ||||||
| 		return lexSkip(lx, lexTop) |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case commentStart: |  | ||||||
| 		lx.push(lexTop) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case tableStart: |  | ||||||
| 		return lexTableStart |  | ||||||
| 	case eof: |  | ||||||
| 		if lx.pos > lx.start { |  | ||||||
| 			return lx.errorf("unexpected EOF") |  | ||||||
| 		} |  | ||||||
| 		lx.emit(itemEOF) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// At this point, the only valid item can be a key, so we back up |  | ||||||
| 	// and let the key lexer do the rest. |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.push(lexTopEnd) |  | ||||||
| 	return lexKeyStart |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexTopEnd is entered whenever a top-level item has been consumed. (A value |  | ||||||
| // or a table.) It must see only whitespace, and will turn back to lexTop |  | ||||||
| // upon a newline. If it sees EOF, it will quit the lexer successfully. |  | ||||||
| func lexTopEnd(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case r == commentStart: |  | ||||||
| 		// a comment will read to a newline for us. |  | ||||||
| 		lx.push(lexTop) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexTopEnd |  | ||||||
| 	case isNL(r): |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lexTop |  | ||||||
| 	case r == eof: |  | ||||||
| 		lx.emit(itemEOF) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("expected a top-level item to end with a newline, "+ |  | ||||||
| 		"comment, or EOF, but got %q instead", r) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexTable lexes the beginning of a table. Namely, it makes sure that |  | ||||||
| // it starts with a character other than '.' and ']'. |  | ||||||
| // It assumes that '[' has already been consumed. |  | ||||||
| // It also handles the case that this is an item in an array of tables. |  | ||||||
| // e.g., '[[name]]'. |  | ||||||
| func lexTableStart(lx *lexer) stateFn { |  | ||||||
| 	if lx.peek() == arrayTableStart { |  | ||||||
| 		lx.next() |  | ||||||
| 		lx.emit(itemArrayTableStart) |  | ||||||
| 		lx.push(lexArrayTableEnd) |  | ||||||
| 	} else { |  | ||||||
| 		lx.emit(itemTableStart) |  | ||||||
| 		lx.push(lexTableEnd) |  | ||||||
| 	} |  | ||||||
| 	return lexTableNameStart |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexTableEnd(lx *lexer) stateFn { |  | ||||||
| 	lx.emit(itemTableEnd) |  | ||||||
| 	return lexTopEnd |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexArrayTableEnd(lx *lexer) stateFn { |  | ||||||
| 	if r := lx.next(); r != arrayTableEnd { |  | ||||||
| 		return lx.errorf("expected end of table array name delimiter %q, "+ |  | ||||||
| 			"but got %q instead", arrayTableEnd, r) |  | ||||||
| 	} |  | ||||||
| 	lx.emit(itemArrayTableEnd) |  | ||||||
| 	return lexTopEnd |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexTableNameStart(lx *lexer) stateFn { |  | ||||||
| 	lx.skip(isWhitespace) |  | ||||||
| 	switch r := lx.peek(); { |  | ||||||
| 	case r == tableEnd || r == eof: |  | ||||||
| 		return lx.errorf("unexpected end of table name " + |  | ||||||
| 			"(table names cannot be empty)") |  | ||||||
| 	case r == tableSep: |  | ||||||
| 		return lx.errorf("unexpected table separator " + |  | ||||||
| 			"(table names cannot be empty)") |  | ||||||
| 	case r == stringStart || r == rawStringStart: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		lx.push(lexTableNameEnd) |  | ||||||
| 		return lexValue // reuse string lexing |  | ||||||
| 	default: |  | ||||||
| 		return lexBareTableName |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexBareTableName lexes the name of a table. It assumes that at least one |  | ||||||
| // valid character for the table has already been read. |  | ||||||
| func lexBareTableName(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isBareKeyChar(r) { |  | ||||||
| 		return lexBareTableName |  | ||||||
| 	} |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.emit(itemText) |  | ||||||
| 	return lexTableNameEnd |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexTableNameEnd reads the end of a piece of a table name, optionally |  | ||||||
| // consuming whitespace. |  | ||||||
| func lexTableNameEnd(lx *lexer) stateFn { |  | ||||||
| 	lx.skip(isWhitespace) |  | ||||||
| 	switch r := lx.next(); { |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexTableNameEnd |  | ||||||
| 	case r == tableSep: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lexTableNameStart |  | ||||||
| 	case r == tableEnd: |  | ||||||
| 		return lx.pop() |  | ||||||
| 	default: |  | ||||||
| 		return lx.errorf("expected '.' or ']' to end table name, "+ |  | ||||||
| 			"but got %q instead", r) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexKeyStart consumes a key name up until the first non-whitespace character. |  | ||||||
| // lexKeyStart will ignore whitespace. |  | ||||||
| func lexKeyStart(lx *lexer) stateFn { |  | ||||||
| 	r := lx.peek() |  | ||||||
| 	switch { |  | ||||||
| 	case r == keySep: |  | ||||||
| 		return lx.errorf("unexpected key separator %q", keySep) |  | ||||||
| 	case isWhitespace(r) || isNL(r): |  | ||||||
| 		lx.next() |  | ||||||
| 		return lexSkip(lx, lexKeyStart) |  | ||||||
| 	case r == stringStart || r == rawStringStart: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		lx.emit(itemKeyStart) |  | ||||||
| 		lx.push(lexKeyEnd) |  | ||||||
| 		return lexValue // reuse string lexing |  | ||||||
| 	default: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		lx.emit(itemKeyStart) |  | ||||||
| 		return lexBareKey |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexBareKey consumes the text of a bare key. Assumes that the first character |  | ||||||
| // (which is not whitespace) has not yet been consumed. |  | ||||||
| func lexBareKey(lx *lexer) stateFn { |  | ||||||
| 	switch r := lx.next(); { |  | ||||||
| 	case isBareKeyChar(r): |  | ||||||
| 		return lexBareKey |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		lx.backup() |  | ||||||
| 		lx.emit(itemText) |  | ||||||
| 		return lexKeyEnd |  | ||||||
| 	case r == keySep: |  | ||||||
| 		lx.backup() |  | ||||||
| 		lx.emit(itemText) |  | ||||||
| 		return lexKeyEnd |  | ||||||
| 	default: |  | ||||||
| 		return lx.errorf("bare keys cannot contain %q", r) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexKeyEnd consumes the end of a key and trims whitespace (up to the key |  | ||||||
| // separator). |  | ||||||
| func lexKeyEnd(lx *lexer) stateFn { |  | ||||||
| 	switch r := lx.next(); { |  | ||||||
| 	case r == keySep: |  | ||||||
| 		return lexSkip(lx, lexValue) |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexSkip(lx, lexKeyEnd) |  | ||||||
| 	default: |  | ||||||
| 		return lx.errorf("expected key separator %q, but got %q instead", |  | ||||||
| 			keySep, r) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexValue starts the consumption of a value anywhere a value is expected. |  | ||||||
| // lexValue will ignore whitespace. |  | ||||||
| // After a value is lexed, the last state on the next is popped and returned. |  | ||||||
| func lexValue(lx *lexer) stateFn { |  | ||||||
| 	// We allow whitespace to precede a value, but NOT newlines. |  | ||||||
| 	// In array syntax, the array states are responsible for ignoring newlines. |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexSkip(lx, lexValue) |  | ||||||
| 	case isDigit(r): |  | ||||||
| 		lx.backup() // avoid an extra state and use the same as above |  | ||||||
| 		return lexNumberOrDateStart |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case arrayStart: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		lx.emit(itemArray) |  | ||||||
| 		return lexArrayValue |  | ||||||
| 	case inlineTableStart: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		lx.emit(itemInlineTableStart) |  | ||||||
| 		return lexInlineTableValue |  | ||||||
| 	case stringStart: |  | ||||||
| 		if lx.accept(stringStart) { |  | ||||||
| 			if lx.accept(stringStart) { |  | ||||||
| 				lx.ignore() // Ignore """ |  | ||||||
| 				return lexMultilineString |  | ||||||
| 			} |  | ||||||
| 			lx.backup() |  | ||||||
| 		} |  | ||||||
| 		lx.ignore() // ignore the '"' |  | ||||||
| 		return lexString |  | ||||||
| 	case rawStringStart: |  | ||||||
| 		if lx.accept(rawStringStart) { |  | ||||||
| 			if lx.accept(rawStringStart) { |  | ||||||
| 				lx.ignore() // Ignore """ |  | ||||||
| 				return lexMultilineRawString |  | ||||||
| 			} |  | ||||||
| 			lx.backup() |  | ||||||
| 		} |  | ||||||
| 		lx.ignore() // ignore the "'" |  | ||||||
| 		return lexRawString |  | ||||||
| 	case '+', '-': |  | ||||||
| 		return lexNumberStart |  | ||||||
| 	case '.': // special error case, be kind to users |  | ||||||
| 		return lx.errorf("floats must start with a digit, not '.'") |  | ||||||
| 	} |  | ||||||
| 	if unicode.IsLetter(r) { |  | ||||||
| 		// Be permissive here; lexBool will give a nice error if the |  | ||||||
| 		// user wrote something like |  | ||||||
| 		//   x = foo |  | ||||||
| 		// (i.e. not 'true' or 'false' but is something else word-like.) |  | ||||||
| 		lx.backup() |  | ||||||
| 		return lexBool |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("expected value but found %q instead", r) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexArrayValue consumes one value in an array. It assumes that '[' or ',' |  | ||||||
| // have already been consumed. All whitespace and newlines are ignored. |  | ||||||
| func lexArrayValue(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case isWhitespace(r) || isNL(r): |  | ||||||
| 		return lexSkip(lx, lexArrayValue) |  | ||||||
| 	case r == commentStart: |  | ||||||
| 		lx.push(lexArrayValue) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case r == comma: |  | ||||||
| 		return lx.errorf("unexpected comma") |  | ||||||
| 	case r == arrayEnd: |  | ||||||
| 		// NOTE(caleb): The spec isn't clear about whether you can have |  | ||||||
| 		// a trailing comma or not, so we'll allow it. |  | ||||||
| 		return lexArrayEnd |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.push(lexArrayValueEnd) |  | ||||||
| 	return lexValue |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexArrayValueEnd consumes everything between the end of an array value and |  | ||||||
| // the next value (or the end of the array): it ignores whitespace and newlines |  | ||||||
| // and expects either a ',' or a ']'. |  | ||||||
| func lexArrayValueEnd(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case isWhitespace(r) || isNL(r): |  | ||||||
| 		return lexSkip(lx, lexArrayValueEnd) |  | ||||||
| 	case r == commentStart: |  | ||||||
| 		lx.push(lexArrayValueEnd) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case r == comma: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lexArrayValue // move on to the next value |  | ||||||
| 	case r == arrayEnd: |  | ||||||
| 		return lexArrayEnd |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf( |  | ||||||
| 		"expected a comma or array terminator %q, but got %q instead", |  | ||||||
| 		arrayEnd, r, |  | ||||||
| 	) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexArrayEnd finishes the lexing of an array. |  | ||||||
| // It assumes that a ']' has just been consumed. |  | ||||||
| func lexArrayEnd(lx *lexer) stateFn { |  | ||||||
| 	lx.ignore() |  | ||||||
| 	lx.emit(itemArrayEnd) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexInlineTableValue consumes one key/value pair in an inline table. |  | ||||||
| // It assumes that '{' or ',' have already been consumed. Whitespace is ignored. |  | ||||||
| func lexInlineTableValue(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexSkip(lx, lexInlineTableValue) |  | ||||||
| 	case isNL(r): |  | ||||||
| 		return lx.errorf("newlines not allowed within inline tables") |  | ||||||
| 	case r == commentStart: |  | ||||||
| 		lx.push(lexInlineTableValue) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case r == comma: |  | ||||||
| 		return lx.errorf("unexpected comma") |  | ||||||
| 	case r == inlineTableEnd: |  | ||||||
| 		return lexInlineTableEnd |  | ||||||
| 	} |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.push(lexInlineTableValueEnd) |  | ||||||
| 	return lexKeyStart |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexInlineTableValueEnd consumes everything between the end of an inline table |  | ||||||
| // key/value pair and the next pair (or the end of the table): |  | ||||||
| // it ignores whitespace and expects either a ',' or a '}'. |  | ||||||
| func lexInlineTableValueEnd(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case isWhitespace(r): |  | ||||||
| 		return lexSkip(lx, lexInlineTableValueEnd) |  | ||||||
| 	case isNL(r): |  | ||||||
| 		return lx.errorf("newlines not allowed within inline tables") |  | ||||||
| 	case r == commentStart: |  | ||||||
| 		lx.push(lexInlineTableValueEnd) |  | ||||||
| 		return lexCommentStart |  | ||||||
| 	case r == comma: |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lexInlineTableValue |  | ||||||
| 	case r == inlineTableEnd: |  | ||||||
| 		return lexInlineTableEnd |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("expected a comma or an inline table terminator %q, "+ |  | ||||||
| 		"but got %q instead", inlineTableEnd, r) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexInlineTableEnd finishes the lexing of an inline table. |  | ||||||
| // It assumes that a '}' has just been consumed. |  | ||||||
| func lexInlineTableEnd(lx *lexer) stateFn { |  | ||||||
| 	lx.ignore() |  | ||||||
| 	lx.emit(itemInlineTableEnd) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexString consumes the inner contents of a string. It assumes that the |  | ||||||
| // beginning '"' has already been consumed and ignored. |  | ||||||
| func lexString(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case r == eof: |  | ||||||
| 		return lx.errorf("unexpected EOF") |  | ||||||
| 	case isNL(r): |  | ||||||
| 		return lx.errorf("strings cannot contain newlines") |  | ||||||
| 	case r == '\\': |  | ||||||
| 		lx.push(lexString) |  | ||||||
| 		return lexStringEscape |  | ||||||
| 	case r == stringEnd: |  | ||||||
| 		lx.backup() |  | ||||||
| 		lx.emit(itemString) |  | ||||||
| 		lx.next() |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lx.pop() |  | ||||||
| 	} |  | ||||||
| 	return lexString |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexMultilineString consumes the inner contents of a string. It assumes that |  | ||||||
| // the beginning '"""' has already been consumed and ignored. |  | ||||||
| func lexMultilineString(lx *lexer) stateFn { |  | ||||||
| 	switch lx.next() { |  | ||||||
| 	case eof: |  | ||||||
| 		return lx.errorf("unexpected EOF") |  | ||||||
| 	case '\\': |  | ||||||
| 		return lexMultilineStringEscape |  | ||||||
| 	case stringEnd: |  | ||||||
| 		if lx.accept(stringEnd) { |  | ||||||
| 			if lx.accept(stringEnd) { |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.emit(itemMultilineString) |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.ignore() |  | ||||||
| 				return lx.pop() |  | ||||||
| 			} |  | ||||||
| 			lx.backup() |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return lexMultilineString |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexRawString consumes a raw string. Nothing can be escaped in such a string. |  | ||||||
| // It assumes that the beginning "'" has already been consumed and ignored. |  | ||||||
| func lexRawString(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch { |  | ||||||
| 	case r == eof: |  | ||||||
| 		return lx.errorf("unexpected EOF") |  | ||||||
| 	case isNL(r): |  | ||||||
| 		return lx.errorf("strings cannot contain newlines") |  | ||||||
| 	case r == rawStringEnd: |  | ||||||
| 		lx.backup() |  | ||||||
| 		lx.emit(itemRawString) |  | ||||||
| 		lx.next() |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return lx.pop() |  | ||||||
| 	} |  | ||||||
| 	return lexRawString |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexMultilineRawString consumes a raw string. Nothing can be escaped in such |  | ||||||
| // a string. It assumes that the beginning "'''" has already been consumed and |  | ||||||
| // ignored. |  | ||||||
| func lexMultilineRawString(lx *lexer) stateFn { |  | ||||||
| 	switch lx.next() { |  | ||||||
| 	case eof: |  | ||||||
| 		return lx.errorf("unexpected EOF") |  | ||||||
| 	case rawStringEnd: |  | ||||||
| 		if lx.accept(rawStringEnd) { |  | ||||||
| 			if lx.accept(rawStringEnd) { |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.backup() |  | ||||||
| 				lx.emit(itemRawMultilineString) |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.next() |  | ||||||
| 				lx.ignore() |  | ||||||
| 				return lx.pop() |  | ||||||
| 			} |  | ||||||
| 			lx.backup() |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return lexMultilineRawString |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexMultilineStringEscape consumes an escaped character. It assumes that the |  | ||||||
| // preceding '\\' has already been consumed. |  | ||||||
| func lexMultilineStringEscape(lx *lexer) stateFn { |  | ||||||
| 	// Handle the special case first: |  | ||||||
| 	if isNL(lx.next()) { |  | ||||||
| 		return lexMultilineString |  | ||||||
| 	} |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.push(lexMultilineString) |  | ||||||
| 	return lexStringEscape(lx) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexStringEscape(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	switch r { |  | ||||||
| 	case 'b': |  | ||||||
| 		fallthrough |  | ||||||
| 	case 't': |  | ||||||
| 		fallthrough |  | ||||||
| 	case 'n': |  | ||||||
| 		fallthrough |  | ||||||
| 	case 'f': |  | ||||||
| 		fallthrough |  | ||||||
| 	case 'r': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '"': |  | ||||||
| 		fallthrough |  | ||||||
| 	case '\\': |  | ||||||
| 		return lx.pop() |  | ||||||
| 	case 'u': |  | ||||||
| 		return lexShortUnicodeEscape |  | ||||||
| 	case 'U': |  | ||||||
| 		return lexLongUnicodeEscape |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("invalid escape character %q; only the following "+ |  | ||||||
| 		"escape characters are allowed: "+ |  | ||||||
| 		`\b, \t, \n, \f, \r, \", \\, \uXXXX, and \UXXXXXXXX`, r) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexShortUnicodeEscape(lx *lexer) stateFn { |  | ||||||
| 	var r rune |  | ||||||
| 	for i := 0; i < 4; i++ { |  | ||||||
| 		r = lx.next() |  | ||||||
| 		if !isHexadecimal(r) { |  | ||||||
| 			return lx.errorf(`expected four hexadecimal digits after '\u', `+ |  | ||||||
| 				"but got %q instead", lx.current()) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func lexLongUnicodeEscape(lx *lexer) stateFn { |  | ||||||
| 	var r rune |  | ||||||
| 	for i := 0; i < 8; i++ { |  | ||||||
| 		r = lx.next() |  | ||||||
| 		if !isHexadecimal(r) { |  | ||||||
| 			return lx.errorf(`expected eight hexadecimal digits after '\U', `+ |  | ||||||
| 				"but got %q instead", lx.current()) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexNumberOrDateStart consumes either an integer, a float, or datetime. |  | ||||||
| func lexNumberOrDateStart(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isDigit(r) { |  | ||||||
| 		return lexNumberOrDate |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case '_': |  | ||||||
| 		return lexNumber |  | ||||||
| 	case 'e', 'E': |  | ||||||
| 		return lexFloat |  | ||||||
| 	case '.': |  | ||||||
| 		return lx.errorf("floats must start with a digit, not '.'") |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("expected a digit but got %q", r) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexNumberOrDate consumes either an integer, float or datetime. |  | ||||||
| func lexNumberOrDate(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isDigit(r) { |  | ||||||
| 		return lexNumberOrDate |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case '-': |  | ||||||
| 		return lexDatetime |  | ||||||
| 	case '_': |  | ||||||
| 		return lexNumber |  | ||||||
| 	case '.', 'e', 'E': |  | ||||||
| 		return lexFloat |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.emit(itemInteger) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexDatetime consumes a Datetime, to a first approximation. |  | ||||||
| // The parser validates that it matches one of the accepted formats. |  | ||||||
| func lexDatetime(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isDigit(r) { |  | ||||||
| 		return lexDatetime |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case '-', 'T', ':', '.', 'Z': |  | ||||||
| 		return lexDatetime |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.emit(itemDatetime) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexNumberStart consumes either an integer or a float. It assumes that a sign |  | ||||||
| // has already been read, but that *no* digits have been consumed. |  | ||||||
| // lexNumberStart will move to the appropriate integer or float states. |  | ||||||
| func lexNumberStart(lx *lexer) stateFn { |  | ||||||
| 	// We MUST see a digit. Even floats have to start with a digit. |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if !isDigit(r) { |  | ||||||
| 		if r == '.' { |  | ||||||
| 			return lx.errorf("floats must start with a digit, not '.'") |  | ||||||
| 		} |  | ||||||
| 		return lx.errorf("expected a digit but got %q", r) |  | ||||||
| 	} |  | ||||||
| 	return lexNumber |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexNumber consumes an integer or a float after seeing the first digit. |  | ||||||
| func lexNumber(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isDigit(r) { |  | ||||||
| 		return lexNumber |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case '_': |  | ||||||
| 		return lexNumber |  | ||||||
| 	case '.', 'e', 'E': |  | ||||||
| 		return lexFloat |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.emit(itemInteger) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexFloat consumes the elements of a float. It allows any sequence of |  | ||||||
| // float-like characters, so floats emitted by the lexer are only a first |  | ||||||
| // approximation and must be validated by the parser. |  | ||||||
| func lexFloat(lx *lexer) stateFn { |  | ||||||
| 	r := lx.next() |  | ||||||
| 	if isDigit(r) { |  | ||||||
| 		return lexFloat |  | ||||||
| 	} |  | ||||||
| 	switch r { |  | ||||||
| 	case '_', '.', '-', '+', 'e', 'E': |  | ||||||
| 		return lexFloat |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	lx.backup() |  | ||||||
| 	lx.emit(itemFloat) |  | ||||||
| 	return lx.pop() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexBool consumes a bool string: 'true' or 'false. |  | ||||||
| func lexBool(lx *lexer) stateFn { |  | ||||||
| 	var rs []rune |  | ||||||
| 	for { |  | ||||||
| 		r := lx.next() |  | ||||||
| 		if r == eof || isWhitespace(r) || isNL(r) { |  | ||||||
| 			lx.backup() |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 		rs = append(rs, r) |  | ||||||
| 	} |  | ||||||
| 	s := string(rs) |  | ||||||
| 	switch s { |  | ||||||
| 	case "true", "false": |  | ||||||
| 		lx.emit(itemBool) |  | ||||||
| 		return lx.pop() |  | ||||||
| 	} |  | ||||||
| 	return lx.errorf("expected value but found %q instead", s) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexCommentStart begins the lexing of a comment. It will emit |  | ||||||
| // itemCommentStart and consume no characters, passing control to lexComment. |  | ||||||
| func lexCommentStart(lx *lexer) stateFn { |  | ||||||
| 	lx.ignore() |  | ||||||
| 	lx.emit(itemCommentStart) |  | ||||||
| 	return lexComment |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexComment lexes an entire comment. It assumes that '#' has been consumed. |  | ||||||
| // It will consume *up to* the first newline character, and pass control |  | ||||||
| // back to the last state on the stack. |  | ||||||
| func lexComment(lx *lexer) stateFn { |  | ||||||
| 	r := lx.peek() |  | ||||||
| 	if isNL(r) || r == eof { |  | ||||||
| 		lx.emit(itemText) |  | ||||||
| 		return lx.pop() |  | ||||||
| 	} |  | ||||||
| 	lx.next() |  | ||||||
| 	return lexComment |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // lexSkip ignores all slurped input and moves on to the next state. |  | ||||||
| func lexSkip(lx *lexer, nextState stateFn) stateFn { |  | ||||||
| 	return func(lx *lexer) stateFn { |  | ||||||
| 		lx.ignore() |  | ||||||
| 		return nextState |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // isWhitespace returns true if `r` is a whitespace character according |  | ||||||
| // to the spec. |  | ||||||
| func isWhitespace(r rune) bool { |  | ||||||
| 	return r == '\t' || r == ' ' |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isNL(r rune) bool { |  | ||||||
| 	return r == '\n' || r == '\r' |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isDigit(r rune) bool { |  | ||||||
| 	return r >= '0' && r <= '9' |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isHexadecimal(r rune) bool { |  | ||||||
| 	return (r >= '0' && r <= '9') || |  | ||||||
| 		(r >= 'a' && r <= 'f') || |  | ||||||
| 		(r >= 'A' && r <= 'F') |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isBareKeyChar(r rune) bool { |  | ||||||
| 	return (r >= 'A' && r <= 'Z') || |  | ||||||
| 		(r >= 'a' && r <= 'z') || |  | ||||||
| 		(r >= '0' && r <= '9') || |  | ||||||
| 		r == '_' || |  | ||||||
| 		r == '-' |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (itype itemType) String() string { |  | ||||||
| 	switch itype { |  | ||||||
| 	case itemError: |  | ||||||
| 		return "Error" |  | ||||||
| 	case itemNIL: |  | ||||||
| 		return "NIL" |  | ||||||
| 	case itemEOF: |  | ||||||
| 		return "EOF" |  | ||||||
| 	case itemText: |  | ||||||
| 		return "Text" |  | ||||||
| 	case itemString, itemRawString, itemMultilineString, itemRawMultilineString: |  | ||||||
| 		return "String" |  | ||||||
| 	case itemBool: |  | ||||||
| 		return "Bool" |  | ||||||
| 	case itemInteger: |  | ||||||
| 		return "Integer" |  | ||||||
| 	case itemFloat: |  | ||||||
| 		return "Float" |  | ||||||
| 	case itemDatetime: |  | ||||||
| 		return "DateTime" |  | ||||||
| 	case itemTableStart: |  | ||||||
| 		return "TableStart" |  | ||||||
| 	case itemTableEnd: |  | ||||||
| 		return "TableEnd" |  | ||||||
| 	case itemKeyStart: |  | ||||||
| 		return "KeyStart" |  | ||||||
| 	case itemArray: |  | ||||||
| 		return "Array" |  | ||||||
| 	case itemArrayEnd: |  | ||||||
| 		return "ArrayEnd" |  | ||||||
| 	case itemCommentStart: |  | ||||||
| 		return "CommentStart" |  | ||||||
| 	} |  | ||||||
| 	panic(fmt.Sprintf("BUG: Unknown type '%d'.", int(itype))) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (item item) String() string { |  | ||||||
| 	return fmt.Sprintf("(%s, %s)", item.typ.String(), item.val) |  | ||||||
| } |  | ||||||
							
								
								
									
										592
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,592 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"strconv" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| 	"unicode" |  | ||||||
| 	"unicode/utf8" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type parser struct { |  | ||||||
| 	mapping map[string]interface{} |  | ||||||
| 	types   map[string]tomlType |  | ||||||
| 	lx      *lexer |  | ||||||
|  |  | ||||||
| 	// A list of keys in the order that they appear in the TOML data. |  | ||||||
| 	ordered []Key |  | ||||||
|  |  | ||||||
| 	// the full key for the current hash in scope |  | ||||||
| 	context Key |  | ||||||
|  |  | ||||||
| 	// the base key name for everything except hashes |  | ||||||
| 	currentKey string |  | ||||||
|  |  | ||||||
| 	// rough approximation of line number |  | ||||||
| 	approxLine int |  | ||||||
|  |  | ||||||
| 	// A map of 'key.group.names' to whether they were created implicitly. |  | ||||||
| 	implicits map[string]bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type parseError string |  | ||||||
|  |  | ||||||
| func (pe parseError) Error() string { |  | ||||||
| 	return string(pe) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func parse(data string) (p *parser, err error) { |  | ||||||
| 	defer func() { |  | ||||||
| 		if r := recover(); r != nil { |  | ||||||
| 			var ok bool |  | ||||||
| 			if err, ok = r.(parseError); ok { |  | ||||||
| 				return |  | ||||||
| 			} |  | ||||||
| 			panic(r) |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	p = &parser{ |  | ||||||
| 		mapping:   make(map[string]interface{}), |  | ||||||
| 		types:     make(map[string]tomlType), |  | ||||||
| 		lx:        lex(data), |  | ||||||
| 		ordered:   make([]Key, 0), |  | ||||||
| 		implicits: make(map[string]bool), |  | ||||||
| 	} |  | ||||||
| 	for { |  | ||||||
| 		item := p.next() |  | ||||||
| 		if item.typ == itemEOF { |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 		p.topLevel(item) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return p, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) panicf(format string, v ...interface{}) { |  | ||||||
| 	msg := fmt.Sprintf("Near line %d (last key parsed '%s'): %s", |  | ||||||
| 		p.approxLine, p.current(), fmt.Sprintf(format, v...)) |  | ||||||
| 	panic(parseError(msg)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) next() item { |  | ||||||
| 	it := p.lx.nextItem() |  | ||||||
| 	if it.typ == itemError { |  | ||||||
| 		p.panicf("%s", it.val) |  | ||||||
| 	} |  | ||||||
| 	return it |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) bug(format string, v ...interface{}) { |  | ||||||
| 	panic(fmt.Sprintf("BUG: "+format+"\n\n", v...)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) expect(typ itemType) item { |  | ||||||
| 	it := p.next() |  | ||||||
| 	p.assertEqual(typ, it.typ) |  | ||||||
| 	return it |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) assertEqual(expected, got itemType) { |  | ||||||
| 	if expected != got { |  | ||||||
| 		p.bug("Expected '%s' but got '%s'.", expected, got) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) topLevel(item item) { |  | ||||||
| 	switch item.typ { |  | ||||||
| 	case itemCommentStart: |  | ||||||
| 		p.approxLine = item.line |  | ||||||
| 		p.expect(itemText) |  | ||||||
| 	case itemTableStart: |  | ||||||
| 		kg := p.next() |  | ||||||
| 		p.approxLine = kg.line |  | ||||||
|  |  | ||||||
| 		var key Key |  | ||||||
| 		for ; kg.typ != itemTableEnd && kg.typ != itemEOF; kg = p.next() { |  | ||||||
| 			key = append(key, p.keyString(kg)) |  | ||||||
| 		} |  | ||||||
| 		p.assertEqual(itemTableEnd, kg.typ) |  | ||||||
|  |  | ||||||
| 		p.establishContext(key, false) |  | ||||||
| 		p.setType("", tomlHash) |  | ||||||
| 		p.ordered = append(p.ordered, key) |  | ||||||
| 	case itemArrayTableStart: |  | ||||||
| 		kg := p.next() |  | ||||||
| 		p.approxLine = kg.line |  | ||||||
|  |  | ||||||
| 		var key Key |  | ||||||
| 		for ; kg.typ != itemArrayTableEnd && kg.typ != itemEOF; kg = p.next() { |  | ||||||
| 			key = append(key, p.keyString(kg)) |  | ||||||
| 		} |  | ||||||
| 		p.assertEqual(itemArrayTableEnd, kg.typ) |  | ||||||
|  |  | ||||||
| 		p.establishContext(key, true) |  | ||||||
| 		p.setType("", tomlArrayHash) |  | ||||||
| 		p.ordered = append(p.ordered, key) |  | ||||||
| 	case itemKeyStart: |  | ||||||
| 		kname := p.next() |  | ||||||
| 		p.approxLine = kname.line |  | ||||||
| 		p.currentKey = p.keyString(kname) |  | ||||||
|  |  | ||||||
| 		val, typ := p.value(p.next()) |  | ||||||
| 		p.setValue(p.currentKey, val) |  | ||||||
| 		p.setType(p.currentKey, typ) |  | ||||||
| 		p.ordered = append(p.ordered, p.context.add(p.currentKey)) |  | ||||||
| 		p.currentKey = "" |  | ||||||
| 	default: |  | ||||||
| 		p.bug("Unexpected type at top level: %s", item.typ) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Gets a string for a key (or part of a key in a table name). |  | ||||||
| func (p *parser) keyString(it item) string { |  | ||||||
| 	switch it.typ { |  | ||||||
| 	case itemText: |  | ||||||
| 		return it.val |  | ||||||
| 	case itemString, itemMultilineString, |  | ||||||
| 		itemRawString, itemRawMultilineString: |  | ||||||
| 		s, _ := p.value(it) |  | ||||||
| 		return s.(string) |  | ||||||
| 	default: |  | ||||||
| 		p.bug("Unexpected key type: %s", it.typ) |  | ||||||
| 		panic("unreachable") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // value translates an expected value from the lexer into a Go value wrapped |  | ||||||
| // as an empty interface. |  | ||||||
| func (p *parser) value(it item) (interface{}, tomlType) { |  | ||||||
| 	switch it.typ { |  | ||||||
| 	case itemString: |  | ||||||
| 		return p.replaceEscapes(it.val), p.typeOfPrimitive(it) |  | ||||||
| 	case itemMultilineString: |  | ||||||
| 		trimmed := stripFirstNewline(stripEscapedWhitespace(it.val)) |  | ||||||
| 		return p.replaceEscapes(trimmed), p.typeOfPrimitive(it) |  | ||||||
| 	case itemRawString: |  | ||||||
| 		return it.val, p.typeOfPrimitive(it) |  | ||||||
| 	case itemRawMultilineString: |  | ||||||
| 		return stripFirstNewline(it.val), p.typeOfPrimitive(it) |  | ||||||
| 	case itemBool: |  | ||||||
| 		switch it.val { |  | ||||||
| 		case "true": |  | ||||||
| 			return true, p.typeOfPrimitive(it) |  | ||||||
| 		case "false": |  | ||||||
| 			return false, p.typeOfPrimitive(it) |  | ||||||
| 		} |  | ||||||
| 		p.bug("Expected boolean value, but got '%s'.", it.val) |  | ||||||
| 	case itemInteger: |  | ||||||
| 		if !numUnderscoresOK(it.val) { |  | ||||||
| 			p.panicf("Invalid integer %q: underscores must be surrounded by digits", |  | ||||||
| 				it.val) |  | ||||||
| 		} |  | ||||||
| 		val := strings.Replace(it.val, "_", "", -1) |  | ||||||
| 		num, err := strconv.ParseInt(val, 10, 64) |  | ||||||
| 		if err != nil { |  | ||||||
| 			// Distinguish integer values. Normally, it'd be a bug if the lexer |  | ||||||
| 			// provides an invalid integer, but it's possible that the number is |  | ||||||
| 			// out of range of valid values (which the lexer cannot determine). |  | ||||||
| 			// So mark the former as a bug but the latter as a legitimate user |  | ||||||
| 			// error. |  | ||||||
| 			if e, ok := err.(*strconv.NumError); ok && |  | ||||||
| 				e.Err == strconv.ErrRange { |  | ||||||
|  |  | ||||||
| 				p.panicf("Integer '%s' is out of the range of 64-bit "+ |  | ||||||
| 					"signed integers.", it.val) |  | ||||||
| 			} else { |  | ||||||
| 				p.bug("Expected integer value, but got '%s'.", it.val) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return num, p.typeOfPrimitive(it) |  | ||||||
| 	case itemFloat: |  | ||||||
| 		parts := strings.FieldsFunc(it.val, func(r rune) bool { |  | ||||||
| 			switch r { |  | ||||||
| 			case '.', 'e', 'E': |  | ||||||
| 				return true |  | ||||||
| 			} |  | ||||||
| 			return false |  | ||||||
| 		}) |  | ||||||
| 		for _, part := range parts { |  | ||||||
| 			if !numUnderscoresOK(part) { |  | ||||||
| 				p.panicf("Invalid float %q: underscores must be "+ |  | ||||||
| 					"surrounded by digits", it.val) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if !numPeriodsOK(it.val) { |  | ||||||
| 			// As a special case, numbers like '123.' or '1.e2', |  | ||||||
| 			// which are valid as far as Go/strconv are concerned, |  | ||||||
| 			// must be rejected because TOML says that a fractional |  | ||||||
| 			// part consists of '.' followed by 1+ digits. |  | ||||||
| 			p.panicf("Invalid float %q: '.' must be followed "+ |  | ||||||
| 				"by one or more digits", it.val) |  | ||||||
| 		} |  | ||||||
| 		val := strings.Replace(it.val, "_", "", -1) |  | ||||||
| 		num, err := strconv.ParseFloat(val, 64) |  | ||||||
| 		if err != nil { |  | ||||||
| 			if e, ok := err.(*strconv.NumError); ok && |  | ||||||
| 				e.Err == strconv.ErrRange { |  | ||||||
|  |  | ||||||
| 				p.panicf("Float '%s' is out of the range of 64-bit "+ |  | ||||||
| 					"IEEE-754 floating-point numbers.", it.val) |  | ||||||
| 			} else { |  | ||||||
| 				p.panicf("Invalid float value: %q", it.val) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return num, p.typeOfPrimitive(it) |  | ||||||
| 	case itemDatetime: |  | ||||||
| 		var t time.Time |  | ||||||
| 		var ok bool |  | ||||||
| 		var err error |  | ||||||
| 		for _, format := range []string{ |  | ||||||
| 			"2006-01-02T15:04:05Z07:00", |  | ||||||
| 			"2006-01-02T15:04:05", |  | ||||||
| 			"2006-01-02", |  | ||||||
| 		} { |  | ||||||
| 			t, err = time.ParseInLocation(format, it.val, time.Local) |  | ||||||
| 			if err == nil { |  | ||||||
| 				ok = true |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if !ok { |  | ||||||
| 			p.panicf("Invalid TOML Datetime: %q.", it.val) |  | ||||||
| 		} |  | ||||||
| 		return t, p.typeOfPrimitive(it) |  | ||||||
| 	case itemArray: |  | ||||||
| 		array := make([]interface{}, 0) |  | ||||||
| 		types := make([]tomlType, 0) |  | ||||||
|  |  | ||||||
| 		for it = p.next(); it.typ != itemArrayEnd; it = p.next() { |  | ||||||
| 			if it.typ == itemCommentStart { |  | ||||||
| 				p.expect(itemText) |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			val, typ := p.value(it) |  | ||||||
| 			array = append(array, val) |  | ||||||
| 			types = append(types, typ) |  | ||||||
| 		} |  | ||||||
| 		return array, p.typeOfArray(types) |  | ||||||
| 	case itemInlineTableStart: |  | ||||||
| 		var ( |  | ||||||
| 			hash         = make(map[string]interface{}) |  | ||||||
| 			outerContext = p.context |  | ||||||
| 			outerKey     = p.currentKey |  | ||||||
| 		) |  | ||||||
|  |  | ||||||
| 		p.context = append(p.context, p.currentKey) |  | ||||||
| 		p.currentKey = "" |  | ||||||
| 		for it := p.next(); it.typ != itemInlineTableEnd; it = p.next() { |  | ||||||
| 			if it.typ != itemKeyStart { |  | ||||||
| 				p.bug("Expected key start but instead found %q, around line %d", |  | ||||||
| 					it.val, p.approxLine) |  | ||||||
| 			} |  | ||||||
| 			if it.typ == itemCommentStart { |  | ||||||
| 				p.expect(itemText) |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// retrieve key |  | ||||||
| 			k := p.next() |  | ||||||
| 			p.approxLine = k.line |  | ||||||
| 			kname := p.keyString(k) |  | ||||||
|  |  | ||||||
| 			// retrieve value |  | ||||||
| 			p.currentKey = kname |  | ||||||
| 			val, typ := p.value(p.next()) |  | ||||||
| 			// make sure we keep metadata up to date |  | ||||||
| 			p.setType(kname, typ) |  | ||||||
| 			p.ordered = append(p.ordered, p.context.add(p.currentKey)) |  | ||||||
| 			hash[kname] = val |  | ||||||
| 		} |  | ||||||
| 		p.context = outerContext |  | ||||||
| 		p.currentKey = outerKey |  | ||||||
| 		return hash, tomlHash |  | ||||||
| 	} |  | ||||||
| 	p.bug("Unexpected value type: %s", it.typ) |  | ||||||
| 	panic("unreachable") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // numUnderscoresOK checks whether each underscore in s is surrounded by |  | ||||||
| // characters that are not underscores. |  | ||||||
| func numUnderscoresOK(s string) bool { |  | ||||||
| 	accept := false |  | ||||||
| 	for _, r := range s { |  | ||||||
| 		if r == '_' { |  | ||||||
| 			if !accept { |  | ||||||
| 				return false |  | ||||||
| 			} |  | ||||||
| 			accept = false |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		accept = true |  | ||||||
| 	} |  | ||||||
| 	return accept |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // numPeriodsOK checks whether every period in s is followed by a digit. |  | ||||||
| func numPeriodsOK(s string) bool { |  | ||||||
| 	period := false |  | ||||||
| 	for _, r := range s { |  | ||||||
| 		if period && !isDigit(r) { |  | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 		period = r == '.' |  | ||||||
| 	} |  | ||||||
| 	return !period |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // establishContext sets the current context of the parser, |  | ||||||
| // where the context is either a hash or an array of hashes. Which one is |  | ||||||
| // set depends on the value of the `array` parameter. |  | ||||||
| // |  | ||||||
| // Establishing the context also makes sure that the key isn't a duplicate, and |  | ||||||
| // will create implicit hashes automatically. |  | ||||||
| func (p *parser) establishContext(key Key, array bool) { |  | ||||||
| 	var ok bool |  | ||||||
|  |  | ||||||
| 	// Always start at the top level and drill down for our context. |  | ||||||
| 	hashContext := p.mapping |  | ||||||
| 	keyContext := make(Key, 0) |  | ||||||
|  |  | ||||||
| 	// We only need implicit hashes for key[0:-1] |  | ||||||
| 	for _, k := range key[0 : len(key)-1] { |  | ||||||
| 		_, ok = hashContext[k] |  | ||||||
| 		keyContext = append(keyContext, k) |  | ||||||
|  |  | ||||||
| 		// No key? Make an implicit hash and move on. |  | ||||||
| 		if !ok { |  | ||||||
| 			p.addImplicit(keyContext) |  | ||||||
| 			hashContext[k] = make(map[string]interface{}) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// If the hash context is actually an array of tables, then set |  | ||||||
| 		// the hash context to the last element in that array. |  | ||||||
| 		// |  | ||||||
| 		// Otherwise, it better be a table, since this MUST be a key group (by |  | ||||||
| 		// virtue of it not being the last element in a key). |  | ||||||
| 		switch t := hashContext[k].(type) { |  | ||||||
| 		case []map[string]interface{}: |  | ||||||
| 			hashContext = t[len(t)-1] |  | ||||||
| 		case map[string]interface{}: |  | ||||||
| 			hashContext = t |  | ||||||
| 		default: |  | ||||||
| 			p.panicf("Key '%s' was already created as a hash.", keyContext) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	p.context = keyContext |  | ||||||
| 	if array { |  | ||||||
| 		// If this is the first element for this array, then allocate a new |  | ||||||
| 		// list of tables for it. |  | ||||||
| 		k := key[len(key)-1] |  | ||||||
| 		if _, ok := hashContext[k]; !ok { |  | ||||||
| 			hashContext[k] = make([]map[string]interface{}, 0, 5) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Add a new table. But make sure the key hasn't already been used |  | ||||||
| 		// for something else. |  | ||||||
| 		if hash, ok := hashContext[k].([]map[string]interface{}); ok { |  | ||||||
| 			hashContext[k] = append(hash, make(map[string]interface{})) |  | ||||||
| 		} else { |  | ||||||
| 			p.panicf("Key '%s' was already created and cannot be used as "+ |  | ||||||
| 				"an array.", keyContext) |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		p.setValue(key[len(key)-1], make(map[string]interface{})) |  | ||||||
| 	} |  | ||||||
| 	p.context = append(p.context, key[len(key)-1]) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // setValue sets the given key to the given value in the current context. |  | ||||||
| // It will make sure that the key hasn't already been defined, account for |  | ||||||
| // implicit key groups. |  | ||||||
| func (p *parser) setValue(key string, value interface{}) { |  | ||||||
| 	var tmpHash interface{} |  | ||||||
| 	var ok bool |  | ||||||
|  |  | ||||||
| 	hash := p.mapping |  | ||||||
| 	keyContext := make(Key, 0) |  | ||||||
| 	for _, k := range p.context { |  | ||||||
| 		keyContext = append(keyContext, k) |  | ||||||
| 		if tmpHash, ok = hash[k]; !ok { |  | ||||||
| 			p.bug("Context for key '%s' has not been established.", keyContext) |  | ||||||
| 		} |  | ||||||
| 		switch t := tmpHash.(type) { |  | ||||||
| 		case []map[string]interface{}: |  | ||||||
| 			// The context is a table of hashes. Pick the most recent table |  | ||||||
| 			// defined as the current hash. |  | ||||||
| 			hash = t[len(t)-1] |  | ||||||
| 		case map[string]interface{}: |  | ||||||
| 			hash = t |  | ||||||
| 		default: |  | ||||||
| 			p.bug("Expected hash to have type 'map[string]interface{}', but "+ |  | ||||||
| 				"it has '%T' instead.", tmpHash) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	keyContext = append(keyContext, key) |  | ||||||
|  |  | ||||||
| 	if _, ok := hash[key]; ok { |  | ||||||
| 		// Typically, if the given key has already been set, then we have |  | ||||||
| 		// to raise an error since duplicate keys are disallowed. However, |  | ||||||
| 		// it's possible that a key was previously defined implicitly. In this |  | ||||||
| 		// case, it is allowed to be redefined concretely. (See the |  | ||||||
| 		// `tests/valid/implicit-and-explicit-after.toml` test in `toml-test`.) |  | ||||||
| 		// |  | ||||||
| 		// But we have to make sure to stop marking it as an implicit. (So that |  | ||||||
| 		// another redefinition provokes an error.) |  | ||||||
| 		// |  | ||||||
| 		// Note that since it has already been defined (as a hash), we don't |  | ||||||
| 		// want to overwrite it. So our business is done. |  | ||||||
| 		if p.isImplicit(keyContext) { |  | ||||||
| 			p.removeImplicit(keyContext) |  | ||||||
| 			return |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Otherwise, we have a concrete key trying to override a previous |  | ||||||
| 		// key, which is *always* wrong. |  | ||||||
| 		p.panicf("Key '%s' has already been defined.", keyContext) |  | ||||||
| 	} |  | ||||||
| 	hash[key] = value |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // setType sets the type of a particular value at a given key. |  | ||||||
| // It should be called immediately AFTER setValue. |  | ||||||
| // |  | ||||||
| // Note that if `key` is empty, then the type given will be applied to the |  | ||||||
| // current context (which is either a table or an array of tables). |  | ||||||
| func (p *parser) setType(key string, typ tomlType) { |  | ||||||
| 	keyContext := make(Key, 0, len(p.context)+1) |  | ||||||
| 	for _, k := range p.context { |  | ||||||
| 		keyContext = append(keyContext, k) |  | ||||||
| 	} |  | ||||||
| 	if len(key) > 0 { // allow type setting for hashes |  | ||||||
| 		keyContext = append(keyContext, key) |  | ||||||
| 	} |  | ||||||
| 	p.types[keyContext.String()] = typ |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // addImplicit sets the given Key as having been created implicitly. |  | ||||||
| func (p *parser) addImplicit(key Key) { |  | ||||||
| 	p.implicits[key.String()] = true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // removeImplicit stops tagging the given key as having been implicitly |  | ||||||
| // created. |  | ||||||
| func (p *parser) removeImplicit(key Key) { |  | ||||||
| 	p.implicits[key.String()] = false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // isImplicit returns true if the key group pointed to by the key was created |  | ||||||
| // implicitly. |  | ||||||
| func (p *parser) isImplicit(key Key) bool { |  | ||||||
| 	return p.implicits[key.String()] |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // current returns the full key name of the current context. |  | ||||||
| func (p *parser) current() string { |  | ||||||
| 	if len(p.currentKey) == 0 { |  | ||||||
| 		return p.context.String() |  | ||||||
| 	} |  | ||||||
| 	if len(p.context) == 0 { |  | ||||||
| 		return p.currentKey |  | ||||||
| 	} |  | ||||||
| 	return fmt.Sprintf("%s.%s", p.context, p.currentKey) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func stripFirstNewline(s string) string { |  | ||||||
| 	if len(s) == 0 || s[0] != '\n' { |  | ||||||
| 		return s |  | ||||||
| 	} |  | ||||||
| 	return s[1:] |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func stripEscapedWhitespace(s string) string { |  | ||||||
| 	esc := strings.Split(s, "\\\n") |  | ||||||
| 	if len(esc) > 1 { |  | ||||||
| 		for i := 1; i < len(esc); i++ { |  | ||||||
| 			esc[i] = strings.TrimLeftFunc(esc[i], unicode.IsSpace) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return strings.Join(esc, "") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) replaceEscapes(str string) string { |  | ||||||
| 	var replaced []rune |  | ||||||
| 	s := []byte(str) |  | ||||||
| 	r := 0 |  | ||||||
| 	for r < len(s) { |  | ||||||
| 		if s[r] != '\\' { |  | ||||||
| 			c, size := utf8.DecodeRune(s[r:]) |  | ||||||
| 			r += size |  | ||||||
| 			replaced = append(replaced, c) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		r += 1 |  | ||||||
| 		if r >= len(s) { |  | ||||||
| 			p.bug("Escape sequence at end of string.") |  | ||||||
| 			return "" |  | ||||||
| 		} |  | ||||||
| 		switch s[r] { |  | ||||||
| 		default: |  | ||||||
| 			p.bug("Expected valid escape code after \\, but got %q.", s[r]) |  | ||||||
| 			return "" |  | ||||||
| 		case 'b': |  | ||||||
| 			replaced = append(replaced, rune(0x0008)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case 't': |  | ||||||
| 			replaced = append(replaced, rune(0x0009)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case 'n': |  | ||||||
| 			replaced = append(replaced, rune(0x000A)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case 'f': |  | ||||||
| 			replaced = append(replaced, rune(0x000C)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case 'r': |  | ||||||
| 			replaced = append(replaced, rune(0x000D)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case '"': |  | ||||||
| 			replaced = append(replaced, rune(0x0022)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case '\\': |  | ||||||
| 			replaced = append(replaced, rune(0x005C)) |  | ||||||
| 			r += 1 |  | ||||||
| 		case 'u': |  | ||||||
| 			// At this point, we know we have a Unicode escape of the form |  | ||||||
| 			// `uXXXX` at [r, r+5). (Because the lexer guarantees this |  | ||||||
| 			// for us.) |  | ||||||
| 			escaped := p.asciiEscapeToUnicode(s[r+1 : r+5]) |  | ||||||
| 			replaced = append(replaced, escaped) |  | ||||||
| 			r += 5 |  | ||||||
| 		case 'U': |  | ||||||
| 			// At this point, we know we have a Unicode escape of the form |  | ||||||
| 			// `uXXXX` at [r, r+9). (Because the lexer guarantees this |  | ||||||
| 			// for us.) |  | ||||||
| 			escaped := p.asciiEscapeToUnicode(s[r+1 : r+9]) |  | ||||||
| 			replaced = append(replaced, escaped) |  | ||||||
| 			r += 9 |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return string(replaced) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (p *parser) asciiEscapeToUnicode(bs []byte) rune { |  | ||||||
| 	s := string(bs) |  | ||||||
| 	hex, err := strconv.ParseUint(strings.ToLower(s), 16, 32) |  | ||||||
| 	if err != nil { |  | ||||||
| 		p.bug("Could not parse '%s' as a hexadecimal number, but the "+ |  | ||||||
| 			"lexer claims it's OK: %s", s, err) |  | ||||||
| 	} |  | ||||||
| 	if !utf8.ValidRune(rune(hex)) { |  | ||||||
| 		p.panicf("Escaped character '\\u%s' is not valid UTF-8.", s) |  | ||||||
| 	} |  | ||||||
| 	return rune(hex) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isStringType(ty itemType) bool { |  | ||||||
| 	return ty == itemString || ty == itemMultilineString || |  | ||||||
| 		ty == itemRawString || ty == itemRawMultilineString |  | ||||||
| } |  | ||||||
							
								
								
									
										91
									
								
								vendor/github.com/BurntSushi/toml/type_check.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,91 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| // tomlType represents any Go type that corresponds to a TOML type. |  | ||||||
| // While the first draft of the TOML spec has a simplistic type system that |  | ||||||
| // probably doesn't need this level of sophistication, we seem to be militating |  | ||||||
| // toward adding real composite types. |  | ||||||
| type tomlType interface { |  | ||||||
| 	typeString() string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // typeEqual accepts any two types and returns true if they are equal. |  | ||||||
| func typeEqual(t1, t2 tomlType) bool { |  | ||||||
| 	if t1 == nil || t2 == nil { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| 	return t1.typeString() == t2.typeString() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func typeIsHash(t tomlType) bool { |  | ||||||
| 	return typeEqual(t, tomlHash) || typeEqual(t, tomlArrayHash) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type tomlBaseType string |  | ||||||
|  |  | ||||||
| func (btype tomlBaseType) typeString() string { |  | ||||||
| 	return string(btype) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (btype tomlBaseType) String() string { |  | ||||||
| 	return btype.typeString() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	tomlInteger   tomlBaseType = "Integer" |  | ||||||
| 	tomlFloat     tomlBaseType = "Float" |  | ||||||
| 	tomlDatetime  tomlBaseType = "Datetime" |  | ||||||
| 	tomlString    tomlBaseType = "String" |  | ||||||
| 	tomlBool      tomlBaseType = "Bool" |  | ||||||
| 	tomlArray     tomlBaseType = "Array" |  | ||||||
| 	tomlHash      tomlBaseType = "Hash" |  | ||||||
| 	tomlArrayHash tomlBaseType = "ArrayHash" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // typeOfPrimitive returns a tomlType of any primitive value in TOML. |  | ||||||
| // Primitive values are: Integer, Float, Datetime, String and Bool. |  | ||||||
| // |  | ||||||
| // Passing a lexer item other than the following will cause a BUG message |  | ||||||
| // to occur: itemString, itemBool, itemInteger, itemFloat, itemDatetime. |  | ||||||
| func (p *parser) typeOfPrimitive(lexItem item) tomlType { |  | ||||||
| 	switch lexItem.typ { |  | ||||||
| 	case itemInteger: |  | ||||||
| 		return tomlInteger |  | ||||||
| 	case itemFloat: |  | ||||||
| 		return tomlFloat |  | ||||||
| 	case itemDatetime: |  | ||||||
| 		return tomlDatetime |  | ||||||
| 	case itemString: |  | ||||||
| 		return tomlString |  | ||||||
| 	case itemMultilineString: |  | ||||||
| 		return tomlString |  | ||||||
| 	case itemRawString: |  | ||||||
| 		return tomlString |  | ||||||
| 	case itemRawMultilineString: |  | ||||||
| 		return tomlString |  | ||||||
| 	case itemBool: |  | ||||||
| 		return tomlBool |  | ||||||
| 	} |  | ||||||
| 	p.bug("Cannot infer primitive type of lex item '%s'.", lexItem) |  | ||||||
| 	panic("unreachable") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // typeOfArray returns a tomlType for an array given a list of types of its |  | ||||||
| // values. |  | ||||||
| // |  | ||||||
| // In the current spec, if an array is homogeneous, then its type is always |  | ||||||
| // "Array". If the array is not homogeneous, an error is generated. |  | ||||||
| func (p *parser) typeOfArray(types []tomlType) tomlType { |  | ||||||
| 	// Empty arrays are cool. |  | ||||||
| 	if len(types) == 0 { |  | ||||||
| 		return tomlArray |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	theType := types[0] |  | ||||||
| 	for _, t := range types[1:] { |  | ||||||
| 		if !typeEqual(theType, t) { |  | ||||||
| 			p.panicf("Array contains values of type '%s' and '%s', but "+ |  | ||||||
| 				"arrays must be homogeneous.", theType, t) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return tomlArray |  | ||||||
| } |  | ||||||
							
								
								
									
										242
									
								
								vendor/github.com/BurntSushi/toml/type_fields.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,242 +0,0 @@ | |||||||
| package toml |  | ||||||
|  |  | ||||||
| // Struct field handling is adapted from code in encoding/json: |  | ||||||
| // |  | ||||||
| // Copyright 2010 The Go Authors.  All rights reserved. |  | ||||||
| // Use of this source code is governed by a BSD-style |  | ||||||
| // license that can be found in the Go distribution. |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"reflect" |  | ||||||
| 	"sort" |  | ||||||
| 	"sync" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // A field represents a single field found in a struct. |  | ||||||
| type field struct { |  | ||||||
| 	name  string       // the name of the field (`toml` tag included) |  | ||||||
| 	tag   bool         // whether field has a `toml` tag |  | ||||||
| 	index []int        // represents the depth of an anonymous field |  | ||||||
| 	typ   reflect.Type // the type of the field |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // byName sorts field by name, breaking ties with depth, |  | ||||||
| // then breaking ties with "name came from toml tag", then |  | ||||||
| // breaking ties with index sequence. |  | ||||||
| type byName []field |  | ||||||
|  |  | ||||||
| func (x byName) Len() int { return len(x) } |  | ||||||
|  |  | ||||||
| func (x byName) Swap(i, j int) { x[i], x[j] = x[j], x[i] } |  | ||||||
|  |  | ||||||
| func (x byName) Less(i, j int) bool { |  | ||||||
| 	if x[i].name != x[j].name { |  | ||||||
| 		return x[i].name < x[j].name |  | ||||||
| 	} |  | ||||||
| 	if len(x[i].index) != len(x[j].index) { |  | ||||||
| 		return len(x[i].index) < len(x[j].index) |  | ||||||
| 	} |  | ||||||
| 	if x[i].tag != x[j].tag { |  | ||||||
| 		return x[i].tag |  | ||||||
| 	} |  | ||||||
| 	return byIndex(x).Less(i, j) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // byIndex sorts field by index sequence. |  | ||||||
| type byIndex []field |  | ||||||
|  |  | ||||||
| func (x byIndex) Len() int { return len(x) } |  | ||||||
|  |  | ||||||
| func (x byIndex) Swap(i, j int) { x[i], x[j] = x[j], x[i] } |  | ||||||
|  |  | ||||||
| func (x byIndex) Less(i, j int) bool { |  | ||||||
| 	for k, xik := range x[i].index { |  | ||||||
| 		if k >= len(x[j].index) { |  | ||||||
| 			return false |  | ||||||
| 		} |  | ||||||
| 		if xik != x[j].index[k] { |  | ||||||
| 			return xik < x[j].index[k] |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return len(x[i].index) < len(x[j].index) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // typeFields returns a list of fields that TOML should recognize for the given |  | ||||||
| // type. The algorithm is breadth-first search over the set of structs to |  | ||||||
| // include - the top struct and then any reachable anonymous structs. |  | ||||||
| func typeFields(t reflect.Type) []field { |  | ||||||
| 	// Anonymous fields to explore at the current level and the next. |  | ||||||
| 	current := []field{} |  | ||||||
| 	next := []field{{typ: t}} |  | ||||||
|  |  | ||||||
| 	// Count of queued names for current level and the next. |  | ||||||
| 	count := map[reflect.Type]int{} |  | ||||||
| 	nextCount := map[reflect.Type]int{} |  | ||||||
|  |  | ||||||
| 	// Types already visited at an earlier level. |  | ||||||
| 	visited := map[reflect.Type]bool{} |  | ||||||
|  |  | ||||||
| 	// Fields found. |  | ||||||
| 	var fields []field |  | ||||||
|  |  | ||||||
| 	for len(next) > 0 { |  | ||||||
| 		current, next = next, current[:0] |  | ||||||
| 		count, nextCount = nextCount, map[reflect.Type]int{} |  | ||||||
|  |  | ||||||
| 		for _, f := range current { |  | ||||||
| 			if visited[f.typ] { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			visited[f.typ] = true |  | ||||||
|  |  | ||||||
| 			// Scan f.typ for fields to include. |  | ||||||
| 			for i := 0; i < f.typ.NumField(); i++ { |  | ||||||
| 				sf := f.typ.Field(i) |  | ||||||
| 				if sf.PkgPath != "" && !sf.Anonymous { // unexported |  | ||||||
| 					continue |  | ||||||
| 				} |  | ||||||
| 				opts := getOptions(sf.Tag) |  | ||||||
| 				if opts.skip { |  | ||||||
| 					continue |  | ||||||
| 				} |  | ||||||
| 				index := make([]int, len(f.index)+1) |  | ||||||
| 				copy(index, f.index) |  | ||||||
| 				index[len(f.index)] = i |  | ||||||
|  |  | ||||||
| 				ft := sf.Type |  | ||||||
| 				if ft.Name() == "" && ft.Kind() == reflect.Ptr { |  | ||||||
| 					// Follow pointer. |  | ||||||
| 					ft = ft.Elem() |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				// Record found field and index sequence. |  | ||||||
| 				if opts.name != "" || !sf.Anonymous || ft.Kind() != reflect.Struct { |  | ||||||
| 					tagged := opts.name != "" |  | ||||||
| 					name := opts.name |  | ||||||
| 					if name == "" { |  | ||||||
| 						name = sf.Name |  | ||||||
| 					} |  | ||||||
| 					fields = append(fields, field{name, tagged, index, ft}) |  | ||||||
| 					if count[f.typ] > 1 { |  | ||||||
| 						// If there were multiple instances, add a second, |  | ||||||
| 						// so that the annihilation code will see a duplicate. |  | ||||||
| 						// It only cares about the distinction between 1 or 2, |  | ||||||
| 						// so don't bother generating any more copies. |  | ||||||
| 						fields = append(fields, fields[len(fields)-1]) |  | ||||||
| 					} |  | ||||||
| 					continue |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				// Record new anonymous struct to explore in next round. |  | ||||||
| 				nextCount[ft]++ |  | ||||||
| 				if nextCount[ft] == 1 { |  | ||||||
| 					f := field{name: ft.Name(), index: index, typ: ft} |  | ||||||
| 					next = append(next, f) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sort.Sort(byName(fields)) |  | ||||||
|  |  | ||||||
| 	// Delete all fields that are hidden by the Go rules for embedded fields, |  | ||||||
| 	// except that fields with TOML tags are promoted. |  | ||||||
|  |  | ||||||
| 	// The fields are sorted in primary order of name, secondary order |  | ||||||
| 	// of field index length. Loop over names; for each name, delete |  | ||||||
| 	// hidden fields by choosing the one dominant field that survives. |  | ||||||
| 	out := fields[:0] |  | ||||||
| 	for advance, i := 0, 0; i < len(fields); i += advance { |  | ||||||
| 		// One iteration per name. |  | ||||||
| 		// Find the sequence of fields with the name of this first field. |  | ||||||
| 		fi := fields[i] |  | ||||||
| 		name := fi.name |  | ||||||
| 		for advance = 1; i+advance < len(fields); advance++ { |  | ||||||
| 			fj := fields[i+advance] |  | ||||||
| 			if fj.name != name { |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if advance == 1 { // Only one field with this name |  | ||||||
| 			out = append(out, fi) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		dominant, ok := dominantField(fields[i : i+advance]) |  | ||||||
| 		if ok { |  | ||||||
| 			out = append(out, dominant) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	fields = out |  | ||||||
| 	sort.Sort(byIndex(fields)) |  | ||||||
|  |  | ||||||
| 	return fields |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // dominantField looks through the fields, all of which are known to |  | ||||||
| // have the same name, to find the single field that dominates the |  | ||||||
| // others using Go's embedding rules, modified by the presence of |  | ||||||
| // TOML tags. If there are multiple top-level fields, the boolean |  | ||||||
| // will be false: This condition is an error in Go and we skip all |  | ||||||
| // the fields. |  | ||||||
| func dominantField(fields []field) (field, bool) { |  | ||||||
| 	// The fields are sorted in increasing index-length order. The winner |  | ||||||
| 	// must therefore be one with the shortest index length. Drop all |  | ||||||
| 	// longer entries, which is easy: just truncate the slice. |  | ||||||
| 	length := len(fields[0].index) |  | ||||||
| 	tagged := -1 // Index of first tagged field. |  | ||||||
| 	for i, f := range fields { |  | ||||||
| 		if len(f.index) > length { |  | ||||||
| 			fields = fields[:i] |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
| 		if f.tag { |  | ||||||
| 			if tagged >= 0 { |  | ||||||
| 				// Multiple tagged fields at the same level: conflict. |  | ||||||
| 				// Return no field. |  | ||||||
| 				return field{}, false |  | ||||||
| 			} |  | ||||||
| 			tagged = i |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	if tagged >= 0 { |  | ||||||
| 		return fields[tagged], true |  | ||||||
| 	} |  | ||||||
| 	// All remaining fields have the same length. If there's more than one, |  | ||||||
| 	// we have a conflict (two fields named "X" at the same level) and we |  | ||||||
| 	// return no field. |  | ||||||
| 	if len(fields) > 1 { |  | ||||||
| 		return field{}, false |  | ||||||
| 	} |  | ||||||
| 	return fields[0], true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var fieldCache struct { |  | ||||||
| 	sync.RWMutex |  | ||||||
| 	m map[reflect.Type][]field |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // cachedTypeFields is like typeFields but uses a cache to avoid repeated work. |  | ||||||
| func cachedTypeFields(t reflect.Type) []field { |  | ||||||
| 	fieldCache.RLock() |  | ||||||
| 	f := fieldCache.m[t] |  | ||||||
| 	fieldCache.RUnlock() |  | ||||||
| 	if f != nil { |  | ||||||
| 		return f |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Compute fields without lock. |  | ||||||
| 	// Might duplicate effort but won't hold other computations back. |  | ||||||
| 	f = typeFields(t) |  | ||||||
| 	if f == nil { |  | ||||||
| 		f = []field{} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	fieldCache.Lock() |  | ||||||
| 	if fieldCache.m == nil { |  | ||||||
| 		fieldCache.m = map[reflect.Type][]field{} |  | ||||||
| 	} |  | ||||||
| 	fieldCache.m[t] = f |  | ||||||
| 	fieldCache.Unlock() |  | ||||||
| 	return f |  | ||||||
| } |  | ||||||
							
								
								
									
										138
									
								
								vendor/github.com/GeertJohan/go.rice/appended.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,138 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"archive/zip" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/daaku/go.zipexe" |  | ||||||
| 	"github.com/kardianos/osext" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // appendedBox defines an appended box |  | ||||||
| type appendedBox struct { |  | ||||||
| 	Name  string                   // box name |  | ||||||
| 	Files map[string]*appendedFile // appended files (*zip.File) by full path |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type appendedFile struct { |  | ||||||
| 	zipFile  *zip.File |  | ||||||
| 	dir      bool |  | ||||||
| 	dirInfo  *appendedDirInfo |  | ||||||
| 	children []*appendedFile |  | ||||||
| 	content  []byte |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // appendedBoxes is a public register of appendes boxes |  | ||||||
| var appendedBoxes = make(map[string]*appendedBox) |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	// find if exec is appended |  | ||||||
| 	thisFile, err := osext.Executable() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return // not appended or cant find self executable |  | ||||||
| 	} |  | ||||||
| 	closer, rd, err := zipexe.OpenCloser(thisFile) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return // not appended |  | ||||||
| 	} |  | ||||||
| 	defer closer.Close() |  | ||||||
|  |  | ||||||
| 	for _, f := range rd.File { |  | ||||||
| 		// get box and file name from f.Name |  | ||||||
| 		fileParts := strings.SplitN(strings.TrimLeft(filepath.ToSlash(f.Name), "/"), "/", 2) |  | ||||||
| 		boxName := fileParts[0] |  | ||||||
| 		var fileName string |  | ||||||
| 		if len(fileParts) > 1 { |  | ||||||
| 			fileName = fileParts[1] |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// find box or create new one if doesn't exist |  | ||||||
| 		box := appendedBoxes[boxName] |  | ||||||
| 		if box == nil { |  | ||||||
| 			box = &appendedBox{ |  | ||||||
| 				Name:  boxName, |  | ||||||
| 				Files: make(map[string]*appendedFile), |  | ||||||
| 			} |  | ||||||
| 			appendedBoxes[boxName] = box |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// create and add file to box |  | ||||||
| 		af := &appendedFile{ |  | ||||||
| 			zipFile: f, |  | ||||||
| 		} |  | ||||||
| 		if f.Comment == "dir" { |  | ||||||
| 			af.dir = true |  | ||||||
| 			af.dirInfo = &appendedDirInfo{ |  | ||||||
| 				name: filepath.Base(af.zipFile.Name), |  | ||||||
| 				//++ TODO: use zip modtime when that is set correctly: af.zipFile.ModTime() |  | ||||||
| 				time: time.Now(), |  | ||||||
| 			} |  | ||||||
| 		} else { |  | ||||||
| 			// this is a file, we need it's contents so we can create a bytes.Reader when the file is opened |  | ||||||
| 			// make a new byteslice |  | ||||||
| 			af.content = make([]byte, af.zipFile.FileInfo().Size()) |  | ||||||
| 			// ignore reading empty files from zip (empty file still is a valid file to be read though!) |  | ||||||
| 			if len(af.content) > 0 { |  | ||||||
| 				// open io.ReadCloser |  | ||||||
| 				rc, err := af.zipFile.Open() |  | ||||||
| 				if err != nil { |  | ||||||
| 					af.content = nil // this will cause an error when the file is being opened or seeked (which is good) |  | ||||||
| 					// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet.. |  | ||||||
| 					log.Printf("error opening appended file %s: %v", af.zipFile.Name, err) |  | ||||||
| 				} else { |  | ||||||
| 					_, err = rc.Read(af.content) |  | ||||||
| 					rc.Close() |  | ||||||
| 					if err != nil { |  | ||||||
| 						af.content = nil // this will cause an error when the file is being opened or seeked (which is good) |  | ||||||
| 						// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet.. |  | ||||||
| 						log.Printf("error reading data for appended file %s: %v", af.zipFile.Name, err) |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// add appendedFile to box file list |  | ||||||
| 		box.Files[fileName] = af |  | ||||||
|  |  | ||||||
| 		// add to parent dir (if any) |  | ||||||
| 		dirName := filepath.Dir(fileName) |  | ||||||
| 		if dirName == "." { |  | ||||||
| 			dirName = "" |  | ||||||
| 		} |  | ||||||
| 		if fileName != "" { // don't make box root dir a child of itself |  | ||||||
| 			if dir := box.Files[dirName]; dir != nil { |  | ||||||
| 				dir.children = append(dir.children, af) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // implements os.FileInfo. |  | ||||||
| // used for Readdir() |  | ||||||
| type appendedDirInfo struct { |  | ||||||
| 	name string |  | ||||||
| 	time time.Time |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (adi *appendedDirInfo) Name() string { |  | ||||||
| 	return adi.name |  | ||||||
| } |  | ||||||
| func (adi *appendedDirInfo) Size() int64 { |  | ||||||
| 	return 0 |  | ||||||
| } |  | ||||||
| func (adi *appendedDirInfo) Mode() os.FileMode { |  | ||||||
| 	return os.ModeDir |  | ||||||
| } |  | ||||||
| func (adi *appendedDirInfo) ModTime() time.Time { |  | ||||||
| 	return adi.time |  | ||||||
| } |  | ||||||
| func (adi *appendedDirInfo) IsDir() bool { |  | ||||||
| 	return true |  | ||||||
| } |  | ||||||
| func (adi *appendedDirInfo) Sys() interface{} { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
							
								
								
									
										337
									
								
								vendor/github.com/GeertJohan/go.rice/box.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,337 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"errors" |  | ||||||
| 	"fmt" |  | ||||||
| 	"io/ioutil" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"runtime" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/GeertJohan/go.rice/embedded" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Box abstracts a directory for resources/files. |  | ||||||
| // It can either load files from disk, or from embedded code (when `rice --embed` was ran). |  | ||||||
| type Box struct { |  | ||||||
| 	name         string |  | ||||||
| 	absolutePath string |  | ||||||
| 	embed        *embedded.EmbeddedBox |  | ||||||
| 	appendd      *appendedBox |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var defaultLocateOrder = []LocateMethod{LocateEmbedded, LocateAppended, LocateFS} |  | ||||||
|  |  | ||||||
| func findBox(name string, order []LocateMethod) (*Box, error) { |  | ||||||
| 	b := &Box{name: name} |  | ||||||
|  |  | ||||||
| 	// no support for absolute paths since gopath can be different on different machines. |  | ||||||
| 	// therefore, required box must be located relative to package requiring it. |  | ||||||
| 	if filepath.IsAbs(name) { |  | ||||||
| 		return nil, errors.New("given name/path is absolute") |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var err error |  | ||||||
| 	for _, method := range order { |  | ||||||
| 		switch method { |  | ||||||
| 		case LocateEmbedded: |  | ||||||
| 			if embed := embedded.EmbeddedBoxes[name]; embed != nil { |  | ||||||
| 				b.embed = embed |  | ||||||
| 				return b, nil |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 		case LocateAppended: |  | ||||||
| 			appendedBoxName := strings.Replace(name, `/`, `-`, -1) |  | ||||||
| 			if appendd := appendedBoxes[appendedBoxName]; appendd != nil { |  | ||||||
| 				b.appendd = appendd |  | ||||||
| 				return b, nil |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 		case LocateFS: |  | ||||||
| 			// resolve absolute directory path |  | ||||||
| 			err := b.resolveAbsolutePathFromCaller() |  | ||||||
| 			if err != nil { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			// check if absolutePath exists on filesystem |  | ||||||
| 			info, err := os.Stat(b.absolutePath) |  | ||||||
| 			if err != nil { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			// check if absolutePath is actually a directory |  | ||||||
| 			if !info.IsDir() { |  | ||||||
| 				err = errors.New("given name/path is not a directory") |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			return b, nil |  | ||||||
| 		case LocateWorkingDirectory: |  | ||||||
| 			// resolve absolute directory path |  | ||||||
| 			err := b.resolveAbsolutePathFromWorkingDirectory() |  | ||||||
| 			if err != nil { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			// check if absolutePath exists on filesystem |  | ||||||
| 			info, err := os.Stat(b.absolutePath) |  | ||||||
| 			if err != nil { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			// check if absolutePath is actually a directory |  | ||||||
| 			if !info.IsDir() { |  | ||||||
| 				err = errors.New("given name/path is not a directory") |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			return b, nil |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err == nil { |  | ||||||
| 		err = fmt.Errorf("could not locate box %q", name) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil, err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FindBox returns a Box instance for given name. |  | ||||||
| // When the given name is a relative path, it's base path will be the calling pkg/cmd's source root. |  | ||||||
| // When the given name is absolute, it's absolute. derp. |  | ||||||
| // Make sure the path doesn't contain any sensitive information as it might be placed into generated go source (embedded). |  | ||||||
| func FindBox(name string) (*Box, error) { |  | ||||||
| 	return findBox(name, defaultLocateOrder) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustFindBox returns a Box instance for given name, like FindBox does. |  | ||||||
| // It does not return an error, instead it panics when an error occurs. |  | ||||||
| func MustFindBox(name string) *Box { |  | ||||||
| 	box, err := findBox(name, defaultLocateOrder) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic(err) |  | ||||||
| 	} |  | ||||||
| 	return box |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // This is injected as a mutable function literal so that we can mock it out in |  | ||||||
| // tests and return a fixed test file. |  | ||||||
| var resolveAbsolutePathFromCaller = func(name string, nStackFrames int) (string, error) { |  | ||||||
| 	_, callingGoFile, _, ok := runtime.Caller(nStackFrames) |  | ||||||
| 	if !ok { |  | ||||||
| 		return "", errors.New("couldn't find caller on stack") |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// resolve to proper path |  | ||||||
| 	pkgDir := filepath.Dir(callingGoFile) |  | ||||||
| 	// fix for go cover |  | ||||||
| 	const coverPath = "_test/_obj_test" |  | ||||||
| 	if !filepath.IsAbs(pkgDir) { |  | ||||||
| 		if i := strings.Index(pkgDir, coverPath); i >= 0 { |  | ||||||
| 			pkgDir = pkgDir[:i] + pkgDir[i+len(coverPath):]            // remove coverPath |  | ||||||
| 			pkgDir = filepath.Join(os.Getenv("GOPATH"), "src", pkgDir) // make absolute |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return filepath.Join(pkgDir, name), nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Box) resolveAbsolutePathFromCaller() error { |  | ||||||
| 	path, err := resolveAbsolutePathFromCaller(b.name, 4) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	b.absolutePath = path |  | ||||||
| 	return nil |  | ||||||
|  |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Box) resolveAbsolutePathFromWorkingDirectory() error { |  | ||||||
| 	path, err := os.Getwd() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	b.absolutePath = filepath.Join(path, b.name) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // IsEmbedded indicates wether this box was embedded into the application |  | ||||||
| func (b *Box) IsEmbedded() bool { |  | ||||||
| 	return b.embed != nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // IsAppended indicates wether this box was appended to the application |  | ||||||
| func (b *Box) IsAppended() bool { |  | ||||||
| 	return b.appendd != nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Time returns how actual the box is. |  | ||||||
| // When the box is embedded, it's value is saved in the embedding code. |  | ||||||
| // When the box is live, this methods returns time.Now() |  | ||||||
| func (b *Box) Time() time.Time { |  | ||||||
| 	if b.IsEmbedded() { |  | ||||||
| 		return b.embed.Time |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	//++ TODO: return time for appended box |  | ||||||
|  |  | ||||||
| 	return time.Now() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Open opens a File from the box |  | ||||||
| // If there is an error, it will be of type *os.PathError. |  | ||||||
| func (b *Box) Open(name string) (*File, error) { |  | ||||||
| 	if Debug { |  | ||||||
| 		fmt.Printf("Open(%s)\n", name) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if b.IsEmbedded() { |  | ||||||
| 		if Debug { |  | ||||||
| 			fmt.Println("Box is embedded") |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// trim prefix (paths are relative to box) |  | ||||||
| 		name = strings.TrimLeft(name, "/") |  | ||||||
| 		if Debug { |  | ||||||
| 			fmt.Printf("Trying %s\n", name) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// search for file |  | ||||||
| 		ef := b.embed.Files[name] |  | ||||||
| 		if ef == nil { |  | ||||||
| 			if Debug { |  | ||||||
| 				fmt.Println("Didn't find file in embed") |  | ||||||
| 			} |  | ||||||
| 			// file not found, try dir |  | ||||||
| 			ed := b.embed.Dirs[name] |  | ||||||
| 			if ed == nil { |  | ||||||
| 				if Debug { |  | ||||||
| 					fmt.Println("Didn't find dir in embed") |  | ||||||
| 				} |  | ||||||
| 				// dir not found, error out |  | ||||||
| 				return nil, &os.PathError{ |  | ||||||
| 					Op:   "open", |  | ||||||
| 					Path: name, |  | ||||||
| 					Err:  os.ErrNotExist, |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			if Debug { |  | ||||||
| 				fmt.Println("Found dir. Returning virtual dir") |  | ||||||
| 			} |  | ||||||
| 			vd := newVirtualDir(ed) |  | ||||||
| 			return &File{virtualD: vd}, nil |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// box is embedded |  | ||||||
| 		if Debug { |  | ||||||
| 			fmt.Println("Found file. Returning virtual file") |  | ||||||
| 		} |  | ||||||
| 		vf := newVirtualFile(ef) |  | ||||||
| 		return &File{virtualF: vf}, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if b.IsAppended() { |  | ||||||
| 		// trim prefix (paths are relative to box) |  | ||||||
| 		name = strings.TrimLeft(name, "/") |  | ||||||
|  |  | ||||||
| 		// search for file |  | ||||||
| 		appendedFile := b.appendd.Files[name] |  | ||||||
| 		if appendedFile == nil { |  | ||||||
| 			return nil, &os.PathError{ |  | ||||||
| 				Op:   "open", |  | ||||||
| 				Path: name, |  | ||||||
| 				Err:  os.ErrNotExist, |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// create new file |  | ||||||
| 		f := &File{ |  | ||||||
| 			appendedF: appendedFile, |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// if this file is a directory, we want to be able to read and seek |  | ||||||
| 		if !appendedFile.dir { |  | ||||||
| 			// looks like malformed data in zip, error now |  | ||||||
| 			if appendedFile.content == nil { |  | ||||||
| 				return nil, &os.PathError{ |  | ||||||
| 					Op:   "open", |  | ||||||
| 					Path: "name", |  | ||||||
| 					Err:  errors.New("error reading data from zip file"), |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			// create new bytes.Reader |  | ||||||
| 			f.appendedFileReader = bytes.NewReader(appendedFile.content) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// all done |  | ||||||
| 		return f, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// perform os open |  | ||||||
| 	if Debug { |  | ||||||
| 		fmt.Printf("Using os.Open(%s)", filepath.Join(b.absolutePath, name)) |  | ||||||
| 	} |  | ||||||
| 	file, err := os.Open(filepath.Join(b.absolutePath, name)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
| 	return &File{realF: file}, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Bytes returns the content of the file with given name as []byte. |  | ||||||
| func (b *Box) Bytes(name string) ([]byte, error) { |  | ||||||
| 	file, err := b.Open(name) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
| 	defer file.Close() |  | ||||||
|  |  | ||||||
| 	content, err := ioutil.ReadAll(file) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return content, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustBytes returns the content of the file with given name as []byte. |  | ||||||
| // panic's on error. |  | ||||||
| func (b *Box) MustBytes(name string) []byte { |  | ||||||
| 	bts, err := b.Bytes(name) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic(err) |  | ||||||
| 	} |  | ||||||
| 	return bts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // String returns the content of the file with given name as string. |  | ||||||
| func (b *Box) String(name string) (string, error) { |  | ||||||
| 	// check if box is embedded, optimized fast path |  | ||||||
| 	if b.IsEmbedded() { |  | ||||||
| 		// find file in embed |  | ||||||
| 		ef := b.embed.Files[name] |  | ||||||
| 		if ef == nil { |  | ||||||
| 			return "", os.ErrNotExist |  | ||||||
| 		} |  | ||||||
| 		// return as string |  | ||||||
| 		return ef.Content, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	bts, err := b.Bytes(name) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return "", err |  | ||||||
| 	} |  | ||||||
| 	return string(bts), nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustString returns the content of the file with given name as string. |  | ||||||
| // panic's on error. |  | ||||||
| func (b *Box) MustString(name string) string { |  | ||||||
| 	str, err := b.String(name) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic(err) |  | ||||||
| 	} |  | ||||||
| 	return str |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Name returns the name of the box |  | ||||||
| func (b *Box) Name() string { |  | ||||||
| 	return b.name |  | ||||||
| } |  | ||||||
							
								
								
									
										39
									
								
								vendor/github.com/GeertJohan/go.rice/config.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,39 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| // LocateMethod defines how a box is located. |  | ||||||
| type LocateMethod int |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	LocateFS               = LocateMethod(iota) // Locate on the filesystem according to package path. |  | ||||||
| 	LocateAppended                              // Locate boxes appended to the executable. |  | ||||||
| 	LocateEmbedded                              // Locate embedded boxes. |  | ||||||
| 	LocateWorkingDirectory                      // Locate on the binary working directory |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Config allows customizing the box lookup behavior. |  | ||||||
| type Config struct { |  | ||||||
| 	// LocateOrder defines the priority order that boxes are searched for. By |  | ||||||
| 	// default, the package global FindBox searches for embedded boxes first, |  | ||||||
| 	// then appended boxes, and then finally boxes on the filesystem.  That |  | ||||||
| 	// search order may be customized by provided the ordered list here. Leaving |  | ||||||
| 	// out a particular method will omit that from the search space. For |  | ||||||
| 	// example, []LocateMethod{LocateEmbedded, LocateAppended} will never search |  | ||||||
| 	// the filesystem for boxes. |  | ||||||
| 	LocateOrder []LocateMethod |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FindBox searches for boxes using the LocateOrder of the config. |  | ||||||
| func (c *Config) FindBox(boxName string) (*Box, error) { |  | ||||||
| 	return findBox(boxName, c.LocateOrder) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustFindBox searches for boxes using the LocateOrder of the config, like |  | ||||||
| // FindBox does.  It does not return an error, instead it panics when an error |  | ||||||
| // occurs. |  | ||||||
| func (c *Config) MustFindBox(boxName string) *Box { |  | ||||||
| 	box, err := findBox(boxName, c.LocateOrder) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic(err) |  | ||||||
| 	} |  | ||||||
| 	return box |  | ||||||
| } |  | ||||||
							
								
								
									
										4
									
								
								vendor/github.com/GeertJohan/go.rice/debug.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,4 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| // Debug can be set to true to enable debugging. |  | ||||||
| var Debug = false |  | ||||||
							
								
								
									
										90
									
								
								vendor/github.com/GeertJohan/go.rice/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,90 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"os" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/GeertJohan/go.rice/embedded" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // re-type to make exported methods invisible to user (godoc) |  | ||||||
| // they're not required for the user |  | ||||||
| // embeddedDirInfo implements os.FileInfo |  | ||||||
| type embeddedDirInfo embedded.EmbeddedDir |  | ||||||
|  |  | ||||||
| // Name returns the base name of the directory |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) Name() string { |  | ||||||
| 	return ed.Filename |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Size always returns 0 |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) Size() int64 { |  | ||||||
| 	return 0 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Mode returns the file mode bits |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) Mode() os.FileMode { |  | ||||||
| 	return os.FileMode(0555 | os.ModeDir) // dr-xr-xr-x |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ModTime returns the modification time |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) ModTime() time.Time { |  | ||||||
| 	return ed.DirModTime |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // IsDir returns the abbreviation for Mode().IsDir() (always true) |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) IsDir() bool { |  | ||||||
| 	return true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Sys returns the underlying data source (always nil) |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ed *embeddedDirInfo) Sys() interface{} { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // re-type to make exported methods invisible to user (godoc) |  | ||||||
| // they're not required for the user |  | ||||||
| // embeddedFileInfo implements os.FileInfo |  | ||||||
| type embeddedFileInfo embedded.EmbeddedFile |  | ||||||
|  |  | ||||||
| // Name returns the base name of the file |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) Name() string { |  | ||||||
| 	return ef.Filename |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Size returns the length in bytes for regular files; system-dependent for others |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) Size() int64 { |  | ||||||
| 	return int64(len(ef.Content)) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Mode returns the file mode bits |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) Mode() os.FileMode { |  | ||||||
| 	return os.FileMode(0555) // r-xr-xr-x |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ModTime returns the modification time |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) ModTime() time.Time { |  | ||||||
| 	return ef.FileModTime |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // IsDir returns the abbreviation for Mode().IsDir() (always false) |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) IsDir() bool { |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Sys returns the underlying data source (always nil) |  | ||||||
| // (implementing os.FileInfo) |  | ||||||
| func (ef *embeddedFileInfo) Sys() interface{} { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
							
								
								
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/embedded/embedded.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,80 +0,0 @@ | |||||||
| // Package embedded defines embedded data types that are shared between the go.rice package and generated code. |  | ||||||
| package embedded |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	EmbedTypeGo   = 0 |  | ||||||
| 	EmbedTypeSyso = 1 |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // EmbeddedBox defines an embedded box |  | ||||||
| type EmbeddedBox struct { |  | ||||||
| 	Name      string                   // box name |  | ||||||
| 	Time      time.Time                // embed time |  | ||||||
| 	EmbedType int                      // kind of embedding |  | ||||||
| 	Files     map[string]*EmbeddedFile // ALL embedded files by full path |  | ||||||
| 	Dirs      map[string]*EmbeddedDir  // ALL embedded dirs by full path |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Link creates the ChildDirs and ChildFiles links in all EmbeddedDir's |  | ||||||
| func (e *EmbeddedBox) Link() { |  | ||||||
| 	for path, ed := range e.Dirs { |  | ||||||
| 		fmt.Println(path) |  | ||||||
| 		ed.ChildDirs = make([]*EmbeddedDir, 0) |  | ||||||
| 		ed.ChildFiles = make([]*EmbeddedFile, 0) |  | ||||||
| 	} |  | ||||||
| 	for path, ed := range e.Dirs { |  | ||||||
| 		parentDirpath, _ := filepath.Split(path) |  | ||||||
| 		if strings.HasSuffix(parentDirpath, "/") { |  | ||||||
| 			parentDirpath = parentDirpath[:len(parentDirpath)-1] |  | ||||||
| 		} |  | ||||||
| 		parentDir := e.Dirs[parentDirpath] |  | ||||||
| 		if parentDir == nil { |  | ||||||
| 			panic("parentDir `" + parentDirpath + "` is missing in embedded box") |  | ||||||
| 		} |  | ||||||
| 		parentDir.ChildDirs = append(parentDir.ChildDirs, ed) |  | ||||||
| 	} |  | ||||||
| 	for path, ef := range e.Files { |  | ||||||
| 		dirpath, _ := filepath.Split(path) |  | ||||||
| 		if strings.HasSuffix(dirpath, "/") { |  | ||||||
| 			dirpath = dirpath[:len(dirpath)-1] |  | ||||||
| 		} |  | ||||||
| 		dir := e.Dirs[dirpath] |  | ||||||
| 		if dir == nil { |  | ||||||
| 			panic("dir `" + dirpath + "` is missing in embedded box") |  | ||||||
| 		} |  | ||||||
| 		dir.ChildFiles = append(dir.ChildFiles, ef) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // EmbeddedDir is instanced in the code generated by the rice tool and contains all necicary information about an embedded file |  | ||||||
| type EmbeddedDir struct { |  | ||||||
| 	Filename   string |  | ||||||
| 	DirModTime time.Time |  | ||||||
| 	ChildDirs  []*EmbeddedDir  // direct childs, as returned by virtualDir.Readdir() |  | ||||||
| 	ChildFiles []*EmbeddedFile // direct childs, as returned by virtualDir.Readdir() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // EmbeddedFile is instanced in the code generated by the rice tool and contains all necicary information about an embedded file |  | ||||||
| type EmbeddedFile struct { |  | ||||||
| 	Filename    string // filename |  | ||||||
| 	FileModTime time.Time |  | ||||||
| 	Content     string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // EmbeddedBoxes is a public register of embedded boxes |  | ||||||
| var EmbeddedBoxes = make(map[string]*EmbeddedBox) |  | ||||||
|  |  | ||||||
| // RegisterEmbeddedBox registers an EmbeddedBox |  | ||||||
| func RegisterEmbeddedBox(name string, box *EmbeddedBox) { |  | ||||||
| 	if _, exists := EmbeddedBoxes[name]; exists { |  | ||||||
| 		panic(fmt.Sprintf("EmbeddedBox with name `%s` exists already", name)) |  | ||||||
| 	} |  | ||||||
| 	EmbeddedBoxes[name] = box |  | ||||||
| } |  | ||||||
							
								
								
									
										69
									
								
								vendor/github.com/GeertJohan/go.rice/example/example.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,69 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"encoding/hex" |  | ||||||
| 	"fmt" |  | ||||||
| 	"log" |  | ||||||
| 	"net/http" |  | ||||||
| 	"os" |  | ||||||
| 	"text/template" |  | ||||||
|  |  | ||||||
| 	"github.com/GeertJohan/go.rice" |  | ||||||
| 	"github.com/davecgh/go-spew/spew" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	conf := rice.Config{ |  | ||||||
| 		LocateOrder: []rice.LocateMethod{rice.LocateEmbedded, rice.LocateAppended, rice.LocateFS}, |  | ||||||
| 	} |  | ||||||
| 	box, err := conf.FindBox("example-files") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatalf("error opening rice.Box: %s\n", err) |  | ||||||
| 	} |  | ||||||
| 	// spew.Dump(box) |  | ||||||
|  |  | ||||||
| 	contentString, err := box.String("file.txt") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatalf("could not read file contents as string: %s\n", err) |  | ||||||
| 	} |  | ||||||
| 	log.Printf("Read some file contents as string:\n%s\n", contentString) |  | ||||||
|  |  | ||||||
| 	contentBytes, err := box.Bytes("file.txt") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatalf("could not read file contents as byteSlice: %s\n", err) |  | ||||||
| 	} |  | ||||||
| 	log.Printf("Read some file contents as byteSlice:\n%s\n", hex.Dump(contentBytes)) |  | ||||||
|  |  | ||||||
| 	file, err := box.Open("file.txt") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatalf("could not open file: %s\n", err) |  | ||||||
| 	} |  | ||||||
| 	spew.Dump(file) |  | ||||||
|  |  | ||||||
| 	// find/create a rice.Box |  | ||||||
| 	templateBox, err := rice.FindBox("example-templates") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	// get file contents as string |  | ||||||
| 	templateString, err := templateBox.String("message.tmpl") |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	// parse and execute the template |  | ||||||
| 	tmplMessage, err := template.New("message").Parse(templateString) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	tmplMessage.Execute(os.Stdout, map[string]string{"Message": "Hello, world!"}) |  | ||||||
|  |  | ||||||
| 	http.Handle("/", http.FileServer(box.HTTPBox())) |  | ||||||
| 	go func() { |  | ||||||
| 		fmt.Println("Serving files on :8080, press ctrl-C to exit") |  | ||||||
| 		err := http.ListenAndServe(":8080", nil) |  | ||||||
| 		if err != nil { |  | ||||||
| 			log.Fatalf("error serving files: %v", err) |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
| 	select {} |  | ||||||
| } |  | ||||||
							
								
								
									
										144
									
								
								vendor/github.com/GeertJohan/go.rice/file.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,144 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"errors" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // File implements the io.Reader, io.Seeker, io.Closer and http.File interfaces |  | ||||||
| type File struct { |  | ||||||
| 	// File abstracts file methods so the user doesn't see the difference between rice.virtualFile, rice.virtualDir and os.File |  | ||||||
| 	// TODO: maybe use internal File interface and four implementations: *os.File, appendedFile, virtualFile, virtualDir |  | ||||||
|  |  | ||||||
| 	// real file on disk |  | ||||||
| 	realF *os.File |  | ||||||
|  |  | ||||||
| 	// when embedded (go) |  | ||||||
| 	virtualF *virtualFile |  | ||||||
| 	virtualD *virtualDir |  | ||||||
|  |  | ||||||
| 	// when appended (zip) |  | ||||||
| 	appendedF          *appendedFile |  | ||||||
| 	appendedFileReader *bytes.Reader |  | ||||||
| 	// TODO: is appendedFileReader subject of races? Might need a lock here.. |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Close is like (*os.File).Close() |  | ||||||
| // Visit http://golang.org/pkg/os/#File.Close for more information |  | ||||||
| func (f *File) Close() error { |  | ||||||
| 	if f.appendedF != nil { |  | ||||||
| 		if f.appendedFileReader == nil { |  | ||||||
| 			return errors.New("already closed") |  | ||||||
| 		} |  | ||||||
| 		f.appendedFileReader = nil |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	if f.virtualF != nil { |  | ||||||
| 		return f.virtualF.close() |  | ||||||
| 	} |  | ||||||
| 	if f.virtualD != nil { |  | ||||||
| 		return f.virtualD.close() |  | ||||||
| 	} |  | ||||||
| 	return f.realF.Close() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Stat is like (*os.File).Stat() |  | ||||||
| // Visit http://golang.org/pkg/os/#File.Stat for more information |  | ||||||
| func (f *File) Stat() (os.FileInfo, error) { |  | ||||||
| 	if f.appendedF != nil { |  | ||||||
| 		if f.appendedF.dir { |  | ||||||
| 			return f.appendedF.dirInfo, nil |  | ||||||
| 		} |  | ||||||
| 		if f.appendedFileReader == nil { |  | ||||||
| 			return nil, errors.New("file is closed") |  | ||||||
| 		} |  | ||||||
| 		return f.appendedF.zipFile.FileInfo(), nil |  | ||||||
| 	} |  | ||||||
| 	if f.virtualF != nil { |  | ||||||
| 		return f.virtualF.stat() |  | ||||||
| 	} |  | ||||||
| 	if f.virtualD != nil { |  | ||||||
| 		return f.virtualD.stat() |  | ||||||
| 	} |  | ||||||
| 	return f.realF.Stat() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Readdir is like (*os.File).Readdir() |  | ||||||
| // Visit http://golang.org/pkg/os/#File.Readdir for more information |  | ||||||
| func (f *File) Readdir(count int) ([]os.FileInfo, error) { |  | ||||||
| 	if f.appendedF != nil { |  | ||||||
| 		if f.appendedF.dir { |  | ||||||
| 			fi := make([]os.FileInfo, 0, len(f.appendedF.children)) |  | ||||||
| 			for _, childAppendedFile := range f.appendedF.children { |  | ||||||
| 				if childAppendedFile.dir { |  | ||||||
| 					fi = append(fi, childAppendedFile.dirInfo) |  | ||||||
| 				} else { |  | ||||||
| 					fi = append(fi, childAppendedFile.zipFile.FileInfo()) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			return fi, nil |  | ||||||
| 		} |  | ||||||
| 		//++ TODO: is os.ErrInvalid the correct error for Readdir on file? |  | ||||||
| 		return nil, os.ErrInvalid |  | ||||||
| 	} |  | ||||||
| 	if f.virtualF != nil { |  | ||||||
| 		return f.virtualF.readdir(count) |  | ||||||
| 	} |  | ||||||
| 	if f.virtualD != nil { |  | ||||||
| 		return f.virtualD.readdir(count) |  | ||||||
| 	} |  | ||||||
| 	return f.realF.Readdir(count) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Read is like (*os.File).Read() |  | ||||||
| // Visit http://golang.org/pkg/os/#File.Read for more information |  | ||||||
| func (f *File) Read(bts []byte) (int, error) { |  | ||||||
| 	if f.appendedF != nil { |  | ||||||
| 		if f.appendedFileReader == nil { |  | ||||||
| 			return 0, &os.PathError{ |  | ||||||
| 				Op:   "read", |  | ||||||
| 				Path: filepath.Base(f.appendedF.zipFile.Name), |  | ||||||
| 				Err:  errors.New("file is closed"), |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if f.appendedF.dir { |  | ||||||
| 			return 0, &os.PathError{ |  | ||||||
| 				Op:   "read", |  | ||||||
| 				Path: filepath.Base(f.appendedF.zipFile.Name), |  | ||||||
| 				Err:  errors.New("is a directory"), |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return f.appendedFileReader.Read(bts) |  | ||||||
| 	} |  | ||||||
| 	if f.virtualF != nil { |  | ||||||
| 		return f.virtualF.read(bts) |  | ||||||
| 	} |  | ||||||
| 	if f.virtualD != nil { |  | ||||||
| 		return f.virtualD.read(bts) |  | ||||||
| 	} |  | ||||||
| 	return f.realF.Read(bts) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Seek is like (*os.File).Seek() |  | ||||||
| // Visit http://golang.org/pkg/os/#File.Seek for more information |  | ||||||
| func (f *File) Seek(offset int64, whence int) (int64, error) { |  | ||||||
| 	if f.appendedF != nil { |  | ||||||
| 		if f.appendedFileReader == nil { |  | ||||||
| 			return 0, &os.PathError{ |  | ||||||
| 				Op:   "seek", |  | ||||||
| 				Path: filepath.Base(f.appendedF.zipFile.Name), |  | ||||||
| 				Err:  errors.New("file is closed"), |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return f.appendedFileReader.Seek(offset, whence) |  | ||||||
| 	} |  | ||||||
| 	if f.virtualF != nil { |  | ||||||
| 		return f.virtualF.seek(offset, whence) |  | ||||||
| 	} |  | ||||||
| 	if f.virtualD != nil { |  | ||||||
| 		return f.virtualD.seek(offset, whence) |  | ||||||
| 	} |  | ||||||
| 	return f.realF.Seek(offset, whence) |  | ||||||
| } |  | ||||||
							
								
								
									
										21
									
								
								vendor/github.com/GeertJohan/go.rice/http.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,21 +0,0 @@ | |||||||
| package rice |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"net/http" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // HTTPBox implements http.FileSystem which allows the use of Box with a http.FileServer. |  | ||||||
| //   e.g.: http.Handle("/", http.FileServer(rice.MustFindBox("http-files").HTTPBox())) |  | ||||||
| type HTTPBox struct { |  | ||||||
| 	*Box |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // HTTPBox creates a new HTTPBox from an existing Box |  | ||||||
| func (b *Box) HTTPBox() *HTTPBox { |  | ||||||
| 	return &HTTPBox{b} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Open returns a File using the http.File interface |  | ||||||
| func (hb *HTTPBox) Open(name string) (http.File, error) { |  | ||||||
| 	return hb.Box.Open(name) |  | ||||||
| } |  | ||||||
							
								
								
									
										172
									
								
								vendor/github.com/GeertJohan/go.rice/rice/append.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,172 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"archive/zip" |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"io" |  | ||||||
| 	"os" |  | ||||||
| 	"os/exec" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"runtime" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/daaku/go.zipexe" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func operationAppend(pkgs []*build.Package) { |  | ||||||
| 	if runtime.GOOS == "windows" { |  | ||||||
| 		_, err := exec.LookPath("zip") |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Println("#### WARNING ! ####") |  | ||||||
| 			fmt.Println("`rice append` is known not to work under windows because the `zip` command is not available. Please let me know if you got this to work (and how).") |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// MARKED FOR DELETION |  | ||||||
| 	// This is actually not required, the append command now has the option --exec required. |  | ||||||
| 	// // check if package is a command |  | ||||||
| 	// if !pkg.IsCommand() { |  | ||||||
| 	// 	fmt.Println("Error: can not append to non-main package. Please follow instructions at github.com/GeertJohan/go.rice") |  | ||||||
| 	// 	os.Exit(1) |  | ||||||
| 	// } |  | ||||||
|  |  | ||||||
| 	// create tmp zipfile |  | ||||||
| 	tmpZipfileName := filepath.Join(os.TempDir(), fmt.Sprintf("ricebox-%d-%s.zip", time.Now().Unix(), randomString(10))) |  | ||||||
| 	verbosef("Will create tmp zipfile: %s\n", tmpZipfileName) |  | ||||||
| 	tmpZipfile, err := os.Create(tmpZipfileName) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error creating tmp zipfile: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	defer func() { |  | ||||||
| 		tmpZipfile.Close() |  | ||||||
| 		os.Remove(tmpZipfileName) |  | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	// find abs path for binary file |  | ||||||
| 	binfileName, err := filepath.Abs(flags.Append.Executable) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error finding absolute path for executable to append: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	verbosef("Will append to file: %s\n", binfileName) |  | ||||||
|  |  | ||||||
| 	// check that command doesn't already have zip appended |  | ||||||
| 	if rd, _ := zipexe.Open(binfileName); rd != nil { |  | ||||||
| 		fmt.Printf("Cannot append to already appended executable. Please remove %s and build a fresh one.\n", binfileName) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// open binfile |  | ||||||
| 	binfile, err := os.OpenFile(binfileName, os.O_WRONLY, os.ModeAppend) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error: unable to open executable file: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// create zip.Writer |  | ||||||
| 	zipWriter := zip.NewWriter(tmpZipfile) |  | ||||||
|  |  | ||||||
| 	for _, pkg := range pkgs { |  | ||||||
| 		// find boxes for this command |  | ||||||
| 		boxMap := findBoxes(pkg) |  | ||||||
|  |  | ||||||
| 		// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? |  | ||||||
| 		if len(boxMap) == 0 { |  | ||||||
| 			fmt.Printf("no calls to rice.FindBox() or rice.MustFindBox() found in import path `%s`\n", pkg.ImportPath) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		verbosef("\n") |  | ||||||
|  |  | ||||||
| 		for boxname := range boxMap { |  | ||||||
| 			appendedBoxName := strings.Replace(boxname, `/`, `-`, -1) |  | ||||||
|  |  | ||||||
| 			// walk box path's and insert files |  | ||||||
| 			boxPath := filepath.Clean(filepath.Join(pkg.Dir, boxname)) |  | ||||||
| 			filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { |  | ||||||
| 				if info == nil { |  | ||||||
| 					fmt.Printf("Error: box \"%s\" not found on disk\n", path) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				// create zipFilename |  | ||||||
| 				zipFileName := filepath.Join(appendedBoxName, strings.TrimPrefix(path, boxPath)) |  | ||||||
| 				// write directories as empty file with comment "dir" |  | ||||||
| 				if info.IsDir() { |  | ||||||
| 					_, err := zipWriter.CreateHeader(&zip.FileHeader{ |  | ||||||
| 						Name:    zipFileName, |  | ||||||
| 						Comment: "dir", |  | ||||||
| 					}) |  | ||||||
| 					if err != nil { |  | ||||||
| 						fmt.Printf("Error creating dir in tmp zip: %s\n", err) |  | ||||||
| 						os.Exit(1) |  | ||||||
| 					} |  | ||||||
| 					return nil |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				// create zipFileWriter |  | ||||||
| 				zipFileHeader, err := zip.FileInfoHeader(info) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("Error creating zip FileHeader: %v\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				zipFileHeader.Name = zipFileName |  | ||||||
| 				zipFileWriter, err := zipWriter.CreateHeader(zipFileHeader) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("Error creating file in tmp zip: %s\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				srcFile, err := os.Open(path) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("Error opening file to append: %s\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				_, err = io.Copy(zipFileWriter, srcFile) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("Error copying file contents to zip: %s\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				srcFile.Close() |  | ||||||
|  |  | ||||||
| 				return nil |  | ||||||
| 			}) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = zipWriter.Close() |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error closing tmp zipfile: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	err = tmpZipfile.Sync() |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error syncing tmp zipfile: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	_, err = tmpZipfile.Seek(0, 0) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error seeking tmp zipfile: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	_, err = binfile.Seek(0, 2) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error seeking bin file: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	_, err = io.Copy(binfile, tmpZipfile) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error appending zipfile to executable: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	zipA := exec.Command("zip", "-A", binfileName) |  | ||||||
| 	err = zipA.Run() |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("Error setting zip offset: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										33
									
								
								vendor/github.com/GeertJohan/go.rice/rice/clean.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,33 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"strings" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func operationClean(pkg *build.Package) { |  | ||||||
| 	filepath.Walk(pkg.Dir, func(filename string, info os.FileInfo, err error) error { |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error walking pkg dir to clean files: %v\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		if info.IsDir() { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		verbosef("checking file '%s'\n", filename) |  | ||||||
| 		if filepath.Base(filename) == "rice-box.go" || |  | ||||||
| 			strings.HasSuffix(filename, ".rice-box.go") || |  | ||||||
| 			strings.HasSuffix(filename, ".rice-box.syso") { |  | ||||||
| 			err := os.Remove(filename) |  | ||||||
| 			if err != nil { |  | ||||||
| 				fmt.Printf("error removing file (%s): %s\n", filename, err) |  | ||||||
| 				os.Exit(-1) |  | ||||||
| 			} |  | ||||||
| 			verbosef("removed file '%s'\n", filename) |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
							
								
								
									
										158
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-go.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,158 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"go/format" |  | ||||||
| 	"io" |  | ||||||
| 	"io/ioutil" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"strings" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| const boxFilename = "rice-box.go" |  | ||||||
|  |  | ||||||
| func operationEmbedGo(pkg *build.Package) { |  | ||||||
|  |  | ||||||
| 	boxMap := findBoxes(pkg) |  | ||||||
|  |  | ||||||
| 	// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? |  | ||||||
| 	if len(boxMap) == 0 { |  | ||||||
| 		fmt.Println("no calls to rice.FindBox() found") |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	verbosef("\n") |  | ||||||
| 	var boxes []*boxDataType |  | ||||||
|  |  | ||||||
| 	for boxname := range boxMap { |  | ||||||
| 		// find path and filename for this box |  | ||||||
| 		boxPath := filepath.Join(pkg.Dir, boxname) |  | ||||||
|  |  | ||||||
| 		// Check to see if the path for the box is a symbolic link.  If so, simply |  | ||||||
| 		// box what the symbolic link points to.  Note: the filepath.Walk function |  | ||||||
| 		// will NOT follow any nested symbolic links.  This only handles the case |  | ||||||
| 		// where the root of the box is a symbolic link. |  | ||||||
| 		symPath, serr := os.Readlink(boxPath) |  | ||||||
| 		if serr == nil { |  | ||||||
| 			boxPath = symPath |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// verbose info |  | ||||||
| 		verbosef("embedding box '%s' to '%s'\n", boxname, boxFilename) |  | ||||||
|  |  | ||||||
| 		// read box metadata |  | ||||||
| 		boxInfo, ierr := os.Stat(boxPath) |  | ||||||
| 		if ierr != nil { |  | ||||||
| 			fmt.Printf("Error: unable to access box at %s\n", boxPath) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// create box datastructure (used by template) |  | ||||||
| 		box := &boxDataType{ |  | ||||||
| 			BoxName: boxname, |  | ||||||
| 			UnixNow: boxInfo.ModTime().Unix(), |  | ||||||
| 			Files:   make([]*fileDataType, 0), |  | ||||||
| 			Dirs:    make(map[string]*dirDataType), |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if !boxInfo.IsDir() { |  | ||||||
| 			fmt.Printf("Error: Box %s must point to a directory but points to %s instead\n", |  | ||||||
| 				boxname, boxPath) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// fill box datastructure with file data |  | ||||||
| 		filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { |  | ||||||
| 			if err != nil { |  | ||||||
| 				fmt.Printf("error walking box: %s\n", err) |  | ||||||
| 				os.Exit(1) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			filename := strings.TrimPrefix(path, boxPath) |  | ||||||
| 			filename = strings.Replace(filename, "\\", "/", -1) |  | ||||||
| 			filename = strings.TrimPrefix(filename, "/") |  | ||||||
| 			if info.IsDir() { |  | ||||||
| 				dirData := &dirDataType{ |  | ||||||
| 					Identifier: "dir" + nextIdentifier(), |  | ||||||
| 					FileName:   filename, |  | ||||||
| 					ModTime:    info.ModTime().Unix(), |  | ||||||
| 					ChildFiles: make([]*fileDataType, 0), |  | ||||||
| 					ChildDirs:  make([]*dirDataType, 0), |  | ||||||
| 				} |  | ||||||
| 				verbosef("\tincludes dir: '%s'\n", dirData.FileName) |  | ||||||
| 				box.Dirs[dirData.FileName] = dirData |  | ||||||
|  |  | ||||||
| 				// add tree entry (skip for root, it'll create a recursion) |  | ||||||
| 				if dirData.FileName != "" { |  | ||||||
| 					pathParts := strings.Split(dirData.FileName, "/") |  | ||||||
| 					parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] |  | ||||||
| 					parentDir.ChildDirs = append(parentDir.ChildDirs, dirData) |  | ||||||
| 				} |  | ||||||
| 			} else { |  | ||||||
| 				fileData := &fileDataType{ |  | ||||||
| 					Identifier: "file" + nextIdentifier(), |  | ||||||
| 					FileName:   filename, |  | ||||||
| 					ModTime:    info.ModTime().Unix(), |  | ||||||
| 				} |  | ||||||
| 				verbosef("\tincludes file: '%s'\n", fileData.FileName) |  | ||||||
| 				fileData.Content, err = ioutil.ReadFile(path) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("error reading file content while walking box: %s\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				box.Files = append(box.Files, fileData) |  | ||||||
|  |  | ||||||
| 				// add tree entry |  | ||||||
| 				pathParts := strings.Split(fileData.FileName, "/") |  | ||||||
| 				parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] |  | ||||||
| 				if parentDir == nil { |  | ||||||
| 					fmt.Printf("Error: parent of %s is not within the box\n", path) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				parentDir.ChildFiles = append(parentDir.ChildFiles, fileData) |  | ||||||
| 			} |  | ||||||
| 			return nil |  | ||||||
| 		}) |  | ||||||
| 		boxes = append(boxes, box) |  | ||||||
|  |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	embedSourceUnformated := bytes.NewBuffer(make([]byte, 0)) |  | ||||||
|  |  | ||||||
| 	// execute template to buffer |  | ||||||
| 	err := tmplEmbeddedBox.Execute( |  | ||||||
| 		embedSourceUnformated, |  | ||||||
| 		embedFileDataType{pkg.Name, boxes}, |  | ||||||
| 	) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Printf("error writing embedded box to file (template execute): %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// format the source code |  | ||||||
| 	embedSource, err := format.Source(embedSourceUnformated.Bytes()) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Printf("error formatting embedSource: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// create go file for box |  | ||||||
| 	boxFile, err := os.Create(filepath.Join(pkg.Dir, boxFilename)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Printf("error creating embedded box file: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	defer boxFile.Close() |  | ||||||
|  |  | ||||||
| 	// write source to file |  | ||||||
| 	_, err = io.Copy(boxFile, bytes.NewBuffer(embedSource)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		log.Printf("error writing embedSource to file: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| } |  | ||||||
							
								
								
									
										204
									
								
								vendor/github.com/GeertJohan/go.rice/rice/embed-syso.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,204 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"encoding/gob" |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"io" |  | ||||||
| 	"io/ioutil" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"regexp" |  | ||||||
| 	"strings" |  | ||||||
| 	"text/template" |  | ||||||
|  |  | ||||||
| 	"github.com/GeertJohan/go.rice/embedded" |  | ||||||
| 	"github.com/akavel/rsrc/coff" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type sizedReader struct { |  | ||||||
| 	*bytes.Reader |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (s sizedReader) Size() int64 { |  | ||||||
| 	return int64(s.Len()) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var tmplEmbeddedSysoHelper *template.Template |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	var err error |  | ||||||
| 	tmplEmbeddedSysoHelper, err = template.New("embeddedSysoHelper").Parse(`package {{.Package}} |  | ||||||
| // ############# GENERATED CODE ##################### |  | ||||||
| // ## This file was generated by the rice tool. |  | ||||||
| // ## Do not edit unless you know what you're doing. |  | ||||||
| // ################################################## |  | ||||||
|  |  | ||||||
| // extern char _bricebox_{{.Symname}}[], _ericebox_{{.Symname}}; |  | ||||||
| // int get_{{.Symname}}_length() { |  | ||||||
| // 	return &_ericebox_{{.Symname}} - _bricebox_{{.Symname}}; |  | ||||||
| // } |  | ||||||
| import "C" |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"encoding/gob" |  | ||||||
| 	"github.com/GeertJohan/go.rice/embedded" |  | ||||||
| 	"unsafe" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	ptr := unsafe.Pointer(&C._bricebox_{{.Symname}}) |  | ||||||
| 	bts := C.GoBytes(ptr, C.get_{{.Symname}}_length()) |  | ||||||
| 	embeddedBox := &embedded.EmbeddedBox{} |  | ||||||
| 	err := gob.NewDecoder(bytes.NewReader(bts)).Decode(embeddedBox) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic("error decoding embedded box: "+err.Error()) |  | ||||||
| 	} |  | ||||||
| 	embeddedBox.Link() |  | ||||||
| 	embedded.RegisterEmbeddedBox(embeddedBox.Name, embeddedBox) |  | ||||||
| }`) |  | ||||||
| 	if err != nil { |  | ||||||
| 		panic("could not parse template embeddedSysoHelper: " + err.Error()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type embeddedSysoHelperData struct { |  | ||||||
| 	Package string |  | ||||||
| 	Symname string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func operationEmbedSyso(pkg *build.Package) { |  | ||||||
|  |  | ||||||
| 	regexpSynameReplacer := regexp.MustCompile(`[^a-z0-9_]`) |  | ||||||
|  |  | ||||||
| 	boxMap := findBoxes(pkg) |  | ||||||
|  |  | ||||||
| 	// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ? |  | ||||||
| 	if len(boxMap) == 0 { |  | ||||||
| 		fmt.Println("no calls to rice.FindBox() found") |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	verbosef("\n") |  | ||||||
|  |  | ||||||
| 	for boxname := range boxMap { |  | ||||||
| 		// find path and filename for this box |  | ||||||
| 		boxPath := filepath.Join(pkg.Dir, boxname) |  | ||||||
| 		boxFilename := strings.Replace(boxname, "/", "-", -1) |  | ||||||
| 		boxFilename = strings.Replace(boxFilename, "..", "back", -1) |  | ||||||
| 		boxFilename = strings.Replace(boxFilename, ".", "-", -1) |  | ||||||
|  |  | ||||||
| 		// verbose info |  | ||||||
| 		verbosef("embedding box '%s'\n", boxname) |  | ||||||
| 		verbosef("\tto file %s\n", boxFilename) |  | ||||||
|  |  | ||||||
| 		// read box metadata |  | ||||||
| 		boxInfo, ierr := os.Stat(boxPath) |  | ||||||
| 		if ierr != nil { |  | ||||||
| 			fmt.Printf("Error: unable to access box at %s\n", boxPath) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// create box datastructure (used by template) |  | ||||||
| 		box := &embedded.EmbeddedBox{ |  | ||||||
| 			Name:      boxname, |  | ||||||
| 			Time:      boxInfo.ModTime(), |  | ||||||
| 			EmbedType: embedded.EmbedTypeSyso, |  | ||||||
| 			Files:     make(map[string]*embedded.EmbeddedFile), |  | ||||||
| 			Dirs:      make(map[string]*embedded.EmbeddedDir), |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// fill box datastructure with file data |  | ||||||
| 		filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error { |  | ||||||
| 			if err != nil { |  | ||||||
| 				fmt.Printf("error walking box: %s\n", err) |  | ||||||
| 				os.Exit(1) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			filename := strings.TrimPrefix(path, boxPath) |  | ||||||
| 			filename = strings.Replace(filename, "\\", "/", -1) |  | ||||||
| 			filename = strings.TrimPrefix(filename, "/") |  | ||||||
| 			if info.IsDir() { |  | ||||||
| 				embeddedDir := &embedded.EmbeddedDir{ |  | ||||||
| 					Filename:   filename, |  | ||||||
| 					DirModTime: info.ModTime(), |  | ||||||
| 				} |  | ||||||
| 				verbosef("\tincludes dir: '%s'\n", embeddedDir.Filename) |  | ||||||
| 				box.Dirs[embeddedDir.Filename] = embeddedDir |  | ||||||
|  |  | ||||||
| 				// add tree entry (skip for root, it'll create a recursion) |  | ||||||
| 				if embeddedDir.Filename != "" { |  | ||||||
| 					pathParts := strings.Split(embeddedDir.Filename, "/") |  | ||||||
| 					parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")] |  | ||||||
| 					parentDir.ChildDirs = append(parentDir.ChildDirs, embeddedDir) |  | ||||||
| 				} |  | ||||||
| 			} else { |  | ||||||
| 				embeddedFile := &embedded.EmbeddedFile{ |  | ||||||
| 					Filename:    filename, |  | ||||||
| 					FileModTime: info.ModTime(), |  | ||||||
| 					Content:     "", |  | ||||||
| 				} |  | ||||||
| 				verbosef("\tincludes file: '%s'\n", embeddedFile.Filename) |  | ||||||
| 				contentBytes, err := ioutil.ReadFile(path) |  | ||||||
| 				if err != nil { |  | ||||||
| 					fmt.Printf("error reading file content while walking box: %s\n", err) |  | ||||||
| 					os.Exit(1) |  | ||||||
| 				} |  | ||||||
| 				embeddedFile.Content = string(contentBytes) |  | ||||||
| 				box.Files[embeddedFile.Filename] = embeddedFile |  | ||||||
| 			} |  | ||||||
| 			return nil |  | ||||||
| 		}) |  | ||||||
|  |  | ||||||
| 		// encode embedded box to gob file |  | ||||||
| 		boxGobBuf := &bytes.Buffer{} |  | ||||||
| 		err := gob.NewEncoder(boxGobBuf).Encode(box) |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error encoding box to gob: %v\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		verbosef("gob-encoded embeddedBox is %d bytes large\n", boxGobBuf.Len()) |  | ||||||
|  |  | ||||||
| 		// write coff |  | ||||||
| 		symname := regexpSynameReplacer.ReplaceAllString(boxname, "_") |  | ||||||
| 		createCoffSyso(boxname, symname, "386", boxGobBuf.Bytes()) |  | ||||||
| 		createCoffSyso(boxname, symname, "amd64", boxGobBuf.Bytes()) |  | ||||||
|  |  | ||||||
| 		// write go |  | ||||||
| 		sysoHelperData := embeddedSysoHelperData{ |  | ||||||
| 			Package: pkg.Name, |  | ||||||
| 			Symname: symname, |  | ||||||
| 		} |  | ||||||
| 		fileSysoHelper, err := os.Create(boxFilename + ".rice-box.go") |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error creating syso helper: %v\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		err = tmplEmbeddedSysoHelper.Execute(fileSysoHelper, sysoHelperData) |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error executing tmplEmbeddedSysoHelper: %v\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func createCoffSyso(boxFilename string, symname string, arch string, data []byte) { |  | ||||||
| 	boxCoff := coff.NewRDATA() |  | ||||||
| 	switch arch { |  | ||||||
| 	case "386": |  | ||||||
| 	case "amd64": |  | ||||||
| 		boxCoff.FileHeader.Machine = 0x8664 |  | ||||||
| 	default: |  | ||||||
| 		panic("invalid arch") |  | ||||||
| 	} |  | ||||||
| 	boxCoff.AddData("_bricebox_"+symname, sizedReader{bytes.NewReader(data)}) |  | ||||||
| 	boxCoff.AddData("_ericebox_"+symname, io.NewSectionReader(strings.NewReader("\000\000"), 0, 2)) // TODO: why? copied from rsrc, which copied it from as-generated |  | ||||||
| 	boxCoff.Freeze() |  | ||||||
| 	err := writeCoff(boxCoff, boxFilename+"_"+arch+".rice-box.syso") |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("error writing %s coff/.syso: %v\n", arch, err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										150
									
								
								vendor/github.com/GeertJohan/go.rice/rice/find.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,150 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/ast" |  | ||||||
| 	"go/build" |  | ||||||
| 	"go/parser" |  | ||||||
| 	"go/token" |  | ||||||
| 	"os" |  | ||||||
| 	"path/filepath" |  | ||||||
| 	"strings" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func badArgument(fileset *token.FileSet, p token.Pos) { |  | ||||||
| 	pos := fileset.Position(p) |  | ||||||
| 	filename := pos.Filename |  | ||||||
| 	base, err := os.Getwd() |  | ||||||
| 	if err == nil { |  | ||||||
| 		rpath, perr := filepath.Rel(base, pos.Filename) |  | ||||||
| 		if perr == nil { |  | ||||||
| 			filename = rpath |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	msg := fmt.Sprintf("%s:%d: Error: found call to rice.FindBox, "+ |  | ||||||
| 		"but argument must be a string literal.\n", filename, pos.Line) |  | ||||||
| 	fmt.Println(msg) |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func findBoxes(pkg *build.Package) map[string]bool { |  | ||||||
| 	// create map of boxes to embed |  | ||||||
| 	var boxMap = make(map[string]bool) |  | ||||||
|  |  | ||||||
| 	// create one list of files for this package |  | ||||||
| 	filenames := make([]string, 0, len(pkg.GoFiles)+len(pkg.CgoFiles)) |  | ||||||
| 	filenames = append(filenames, pkg.GoFiles...) |  | ||||||
| 	filenames = append(filenames, pkg.CgoFiles...) |  | ||||||
|  |  | ||||||
| 	// loop over files, search for rice.FindBox(..) calls |  | ||||||
| 	for _, filename := range filenames { |  | ||||||
| 		// find full filepath |  | ||||||
| 		fullpath := filepath.Join(pkg.Dir, filename) |  | ||||||
| 		if strings.HasSuffix(filename, "rice-box.go") { |  | ||||||
| 			// Ignore *.rice-box.go files |  | ||||||
| 			verbosef("skipping file %q\n", fullpath) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		verbosef("scanning file %q\n", fullpath) |  | ||||||
|  |  | ||||||
| 		fset := token.NewFileSet() |  | ||||||
| 		f, err := parser.ParseFile(fset, fullpath, nil, 0) |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Println(err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		var riceIsImported bool |  | ||||||
| 		ricePkgName := "rice" |  | ||||||
| 		for _, imp := range f.Imports { |  | ||||||
| 			if strings.HasSuffix(imp.Path.Value, "go.rice\"") { |  | ||||||
| 				if imp.Name != nil { |  | ||||||
| 					ricePkgName = imp.Name.Name |  | ||||||
| 				} |  | ||||||
| 				riceIsImported = true |  | ||||||
| 				break |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		if !riceIsImported { |  | ||||||
| 			// Rice wasn't imported, so we won't find a box. |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
| 		if ricePkgName == "_" { |  | ||||||
| 			// Rice pkg is unnamed, so we won't find a box. |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Inspect AST, looking for calls to (Must)?FindBox. |  | ||||||
| 		// First parameter of the func must be a basic literal. |  | ||||||
| 		// Identifiers won't be resolved. |  | ||||||
| 		var nextIdentIsBoxFunc bool |  | ||||||
| 		var nextBasicLitParamIsBoxName bool |  | ||||||
| 		var boxCall token.Pos |  | ||||||
| 		var variableToRemember string |  | ||||||
| 		var validVariablesForBoxes map[string]bool = make(map[string]bool) |  | ||||||
|  |  | ||||||
| 		ast.Inspect(f, func(node ast.Node) bool { |  | ||||||
| 			if node == nil { |  | ||||||
| 				return false |  | ||||||
| 			} |  | ||||||
| 			switch x := node.(type) { |  | ||||||
| 			// this case fixes the var := func() style assignments, not assignments to vars declared separately from the assignment. |  | ||||||
| 			case *ast.AssignStmt: |  | ||||||
| 				var assign = node.(*ast.AssignStmt) |  | ||||||
| 				name, found := assign.Lhs[0].(*ast.Ident) |  | ||||||
| 				if found { |  | ||||||
| 					variableToRemember = name.Name |  | ||||||
| 					composite, first := assign.Rhs[0].(*ast.CompositeLit) |  | ||||||
| 					if first { |  | ||||||
| 						riceSelector, second := composite.Type.(*ast.SelectorExpr) |  | ||||||
|  |  | ||||||
| 						if second { |  | ||||||
| 							callCorrect := riceSelector.Sel.Name == "Config" |  | ||||||
| 							packageName, third := riceSelector.X.(*ast.Ident) |  | ||||||
|  |  | ||||||
| 							if third && callCorrect && packageName.Name == ricePkgName { |  | ||||||
| 								validVariablesForBoxes[name.Name] = true |  | ||||||
| 								verbosef("\tfound variable, saving to scan for boxes: %q\n", name.Name) |  | ||||||
| 							} |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			case *ast.Ident: |  | ||||||
| 				if nextIdentIsBoxFunc || ricePkgName == "." { |  | ||||||
| 					nextIdentIsBoxFunc = false |  | ||||||
| 					if x.Name == "FindBox" || x.Name == "MustFindBox" { |  | ||||||
| 						nextBasicLitParamIsBoxName = true |  | ||||||
| 						boxCall = x.Pos() |  | ||||||
| 					} |  | ||||||
| 				} else { |  | ||||||
| 					if x.Name == ricePkgName || validVariablesForBoxes[x.Name] { |  | ||||||
| 						nextIdentIsBoxFunc = true |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			case *ast.BasicLit: |  | ||||||
| 				if nextBasicLitParamIsBoxName { |  | ||||||
| 					if x.Kind == token.STRING { |  | ||||||
| 						nextBasicLitParamIsBoxName = false |  | ||||||
| 						// trim "" or `` |  | ||||||
| 						name := x.Value[1 : len(x.Value)-1] |  | ||||||
| 						boxMap[name] = true |  | ||||||
| 						verbosef("\tfound box %q\n", name) |  | ||||||
| 					} else { |  | ||||||
| 						badArgument(fset, boxCall) |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 			default: |  | ||||||
| 				if nextIdentIsBoxFunc { |  | ||||||
| 					nextIdentIsBoxFunc = false |  | ||||||
| 				} |  | ||||||
| 				if nextBasicLitParamIsBoxName { |  | ||||||
| 					badArgument(fset, boxCall) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			return true |  | ||||||
| 		}) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return boxMap |  | ||||||
| } |  | ||||||
							
								
								
									
										80
									
								
								vendor/github.com/GeertJohan/go.rice/rice/flags.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,80 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"os" |  | ||||||
|  |  | ||||||
| 	goflags "github.com/jessevdk/go-flags" // rename import to `goflags` (file scope) so we can use `var flags` (package scope) |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // flags |  | ||||||
| var flags struct { |  | ||||||
| 	Verbose     bool     `long:"verbose" short:"v" description:"Show verbose debug information"` |  | ||||||
| 	ImportPaths []string `long:"import-path" short:"i" description:"Import path(s) to use. Using PWD when left empty. Specify multiple times for more import paths to append"` |  | ||||||
|  |  | ||||||
| 	Append struct { |  | ||||||
| 		Executable string `long:"exec" description:"Executable to append" required:"true"` |  | ||||||
| 	} `command:"append"` |  | ||||||
|  |  | ||||||
| 	EmbedGo   struct{} `command:"embed-go" alias:"embed"` |  | ||||||
| 	EmbedSyso struct{} `command:"embed-syso"` |  | ||||||
| 	Clean     struct{} `command:"clean"` |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // flags parser |  | ||||||
| var flagsParser *goflags.Parser |  | ||||||
|  |  | ||||||
| // initFlags parses the given flags. |  | ||||||
| // when the user asks for help (-h or --help): the application exists with status 0 |  | ||||||
| // when unexpected flags is given: the application exits with status 1 |  | ||||||
| func parseArguments() { |  | ||||||
| 	// create flags parser in global var, for flagsParser.Active.Name (operation) |  | ||||||
| 	flagsParser = goflags.NewParser(&flags, goflags.Default) |  | ||||||
|  |  | ||||||
| 	// parse flags |  | ||||||
| 	args, err := flagsParser.Parse() |  | ||||||
| 	if err != nil { |  | ||||||
| 		// assert the err to be a flags.Error |  | ||||||
| 		flagError := err.(*goflags.Error) |  | ||||||
| 		if flagError.Type == goflags.ErrHelp { |  | ||||||
| 			// user asked for help on flags. |  | ||||||
| 			// program can exit successfully |  | ||||||
| 			os.Exit(0) |  | ||||||
| 		} |  | ||||||
| 		if flagError.Type == goflags.ErrUnknownFlag { |  | ||||||
| 			fmt.Println("Use --help to view available options.") |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		if flagError.Type == goflags.ErrRequired { |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		fmt.Printf("Error parsing flags: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// error on left-over arguments |  | ||||||
| 	if len(args) > 0 { |  | ||||||
| 		fmt.Printf("Unexpected arguments: %s\nUse --help to view available options.", args) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// default ImportPath to pwd when not set |  | ||||||
| 	if len(flags.ImportPaths) == 0 { |  | ||||||
| 		pwd, err := os.Getwd() |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error getting pwd: %s\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		verbosef("using pwd as import path\n") |  | ||||||
| 		// find non-absolute path for this pwd |  | ||||||
| 		pkg, err := build.ImportDir(pwd, build.FindOnly) |  | ||||||
| 		if err != nil { |  | ||||||
| 			fmt.Printf("error using current directory as import path: %s\n", err) |  | ||||||
| 			os.Exit(1) |  | ||||||
| 		} |  | ||||||
| 		flags.ImportPaths = append(flags.ImportPaths, pkg.ImportPath) |  | ||||||
| 		verbosef("using import paths: %s\n", flags.ImportPaths) |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										14
									
								
								vendor/github.com/GeertJohan/go.rice/rice/identifier.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,14 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"strconv" |  | ||||||
|  |  | ||||||
| 	"github.com/GeertJohan/go.incremental" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var identifierCount incremental.Uint64 |  | ||||||
|  |  | ||||||
| func nextIdentifier() string { |  | ||||||
| 	num := identifierCount.Next() |  | ||||||
| 	return strconv.FormatUint(num, 36) // 0123456789abcdefghijklmnopqrstuvwxyz |  | ||||||
| } |  | ||||||
							
								
								
									
										68
									
								
								vendor/github.com/GeertJohan/go.rice/rice/main.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,68 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"go/build" |  | ||||||
| 	"log" |  | ||||||
| 	"os" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func main() { |  | ||||||
| 	// parser arguments |  | ||||||
| 	parseArguments() |  | ||||||
|  |  | ||||||
| 	// find package for path |  | ||||||
| 	var pkgs []*build.Package |  | ||||||
| 	for _, importPath := range flags.ImportPaths { |  | ||||||
| 		pkg := pkgForPath(importPath) |  | ||||||
| 		pkgs = append(pkgs, pkg) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// switch on the operation to perform |  | ||||||
| 	switch flagsParser.Active.Name { |  | ||||||
| 	case "embed", "embed-go": |  | ||||||
| 		for _, pkg := range pkgs { |  | ||||||
| 			operationEmbedGo(pkg) |  | ||||||
| 		} |  | ||||||
| 	case "embed-syso": |  | ||||||
| 		log.Println("WARNING: embedding .syso is experimental..") |  | ||||||
| 		for _, pkg := range pkgs { |  | ||||||
| 			operationEmbedSyso(pkg) |  | ||||||
| 		} |  | ||||||
| 	case "append": |  | ||||||
| 		operationAppend(pkgs) |  | ||||||
| 	case "clean": |  | ||||||
| 		for _, pkg := range pkgs { |  | ||||||
| 			operationClean(pkg) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// all done |  | ||||||
| 	verbosef("\n") |  | ||||||
| 	verbosef("rice finished successfully\n") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // helper function to get *build.Package for given path |  | ||||||
| func pkgForPath(path string) *build.Package { |  | ||||||
| 	// get pwd for relative imports |  | ||||||
| 	pwd, err := os.Getwd() |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("error getting pwd (required for relative imports): %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// read full package information |  | ||||||
| 	pkg, err := build.Import(path, pwd, 0) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("error reading package: %s\n", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return pkg |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func verbosef(format string, stuff ...interface{}) { |  | ||||||
| 	if flags.Verbose { |  | ||||||
| 		log.Printf(format, stuff...) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										98
									
								
								vendor/github.com/GeertJohan/go.rice/rice/templates.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,98 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"os" |  | ||||||
| 	"text/template" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var tmplEmbeddedBox *template.Template |  | ||||||
|  |  | ||||||
| func init() { |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	// parse embedded box template |  | ||||||
| 	tmplEmbeddedBox, err = template.New("embeddedBox").Parse(`package {{.Package}} |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"github.com/GeertJohan/go.rice/embedded" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| {{range .Boxes}} |  | ||||||
| func init() { |  | ||||||
|  |  | ||||||
| 	// define files |  | ||||||
| 	{{range .Files}}{{.Identifier}} := &embedded.EmbeddedFile{ |  | ||||||
| 		Filename:    ` + "`" + `{{.FileName}}` + "`" + `, |  | ||||||
| 		FileModTime: time.Unix({{.ModTime}}, 0), |  | ||||||
| 		Content:     string({{.Content | printf "%q"}}),  |  | ||||||
| 	} |  | ||||||
| 	{{end}} |  | ||||||
|  |  | ||||||
| 	// define dirs |  | ||||||
| 	{{range .Dirs}}{{.Identifier}} := &embedded.EmbeddedDir{ |  | ||||||
| 		Filename:    ` + "`" + `{{.FileName}}` + "`" + `, |  | ||||||
| 		DirModTime: time.Unix({{.ModTime}}, 0), |  | ||||||
| 		ChildFiles:  []*embedded.EmbeddedFile{ |  | ||||||
| 			{{range .ChildFiles}}{{.Identifier}}, // {{.FileName}} |  | ||||||
| 			{{end}} |  | ||||||
| 		}, |  | ||||||
| 	} |  | ||||||
| 	{{end}} |  | ||||||
|  |  | ||||||
| 	// link ChildDirs |  | ||||||
| 	{{range .Dirs}}{{.Identifier}}.ChildDirs = []*embedded.EmbeddedDir{ |  | ||||||
| 		{{range .ChildDirs}}{{.Identifier}}, // {{.FileName}} |  | ||||||
| 		{{end}} |  | ||||||
| 	} |  | ||||||
| 	{{end}} |  | ||||||
|  |  | ||||||
| 	// register embeddedBox |  | ||||||
| 	embedded.RegisterEmbeddedBox(` + "`" + `{{.BoxName}}` + "`" + `, &embedded.EmbeddedBox{ |  | ||||||
| 		Name: ` + "`" + `{{.BoxName}}` + "`" + `, |  | ||||||
| 		Time: time.Unix({{.UnixNow}}, 0), |  | ||||||
| 		Dirs: map[string]*embedded.EmbeddedDir{ |  | ||||||
| 			{{range .Dirs}}"{{.FileName}}": {{.Identifier}}, |  | ||||||
| 			{{end}} |  | ||||||
| 		}, |  | ||||||
| 		Files: map[string]*embedded.EmbeddedFile{ |  | ||||||
| 			{{range .Files}}"{{.FileName}}": {{.Identifier}}, |  | ||||||
| 			{{end}} |  | ||||||
| 		}, |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
| {{end}}`) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Printf("error parsing embedded box template: %s\n", err) |  | ||||||
| 		os.Exit(-1) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type embedFileDataType struct { |  | ||||||
| 	Package string |  | ||||||
| 	Boxes   []*boxDataType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type boxDataType struct { |  | ||||||
| 	BoxName string |  | ||||||
| 	UnixNow int64 |  | ||||||
| 	Files   []*fileDataType |  | ||||||
| 	Dirs    map[string]*dirDataType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type fileDataType struct { |  | ||||||
| 	Identifier string |  | ||||||
| 	FileName   string |  | ||||||
| 	Content    []byte |  | ||||||
| 	ModTime    int64 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type dirDataType struct { |  | ||||||
| 	Identifier string |  | ||||||
| 	FileName   string |  | ||||||
| 	Content    []byte |  | ||||||
| 	ModTime    int64 |  | ||||||
| 	ChildDirs  []*dirDataType |  | ||||||
| 	ChildFiles []*fileDataType |  | ||||||
| } |  | ||||||
							
								
								
									
										22
									
								
								vendor/github.com/GeertJohan/go.rice/rice/util.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,22 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"math/rand" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // randomString generates a pseudo-random alpha-numeric string with given length. |  | ||||||
| func randomString(length int) string { |  | ||||||
| 	rand.Seed(time.Now().UnixNano()) |  | ||||||
| 	k := make([]rune, length) |  | ||||||
| 	for i := 0; i < length; i++ { |  | ||||||
| 		c := rand.Intn(35) |  | ||||||
| 		if c < 10 { |  | ||||||
| 			c += 48 // numbers (0-9) (0+48 == 48 == '0', 9+48 == 57 == '9') |  | ||||||
| 		} else { |  | ||||||
| 			c += 87 // lower case alphabets (a-z) (10+87 == 97 == 'a', 35+87 == 122 = 'z') |  | ||||||
| 		} |  | ||||||
| 		k[i] = rune(c) |  | ||||||
| 	} |  | ||||||
| 	return string(k) |  | ||||||
| } |  | ||||||
							
								
								
									
										42
									
								
								vendor/github.com/GeertJohan/go.rice/rice/writecoff.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,42 +0,0 @@ | |||||||
| package main |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"os" |  | ||||||
| 	"reflect" |  | ||||||
|  |  | ||||||
| 	"github.com/akavel/rsrc/binutil" |  | ||||||
| 	"github.com/akavel/rsrc/coff" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // copied from github.com/akavel/rsrc |  | ||||||
| // LICENSE: MIT |  | ||||||
| // Copyright 2013-2014 The rsrc Authors. (https://github.com/akavel/rsrc/blob/master/AUTHORS) |  | ||||||
| func writeCoff(coff *coff.Coff, fnameout string) error { |  | ||||||
| 	out, err := os.Create(fnameout) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	defer out.Close() |  | ||||||
| 	w := binutil.Writer{W: out} |  | ||||||
|  |  | ||||||
| 	// write the resulting file to disk |  | ||||||
| 	binutil.Walk(coff, func(v reflect.Value, path string) error { |  | ||||||
| 		if binutil.Plain(v.Kind()) { |  | ||||||
| 			w.WriteLE(v.Interface()) |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		vv, ok := v.Interface().(binutil.SizedReader) |  | ||||||
| 		if ok { |  | ||||||
| 			w.WriteFromSized(vv) |  | ||||||
| 			return binutil.WALK_SKIP |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	if w.Err != nil { |  | ||||||
| 		return fmt.Errorf("Error writing output file: %s", w.Err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||