Compare commits
	
		
			667 Commits
		
	
	
		
			v1.14.0-rc
			...
			updatexmpp
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| bab3681ac2 | |||
| b74b884793 | |||
|   | c4157a4d5b | ||
|   | d16645c952 | ||
|   | 65d78e38af | ||
|   | 996e4a7fcf | ||
|   | a89267943c | ||
|   | 815d8b804f | ||
|   | 6edd5de3b7 | ||
|   | 2e9db32a83 | ||
|   | 4bf1c0450c | ||
|   | 0bb521512a | ||
|   | 13fd5c5d5b | ||
|   | d055b4530e | ||
|   | 6b528ffa4f | ||
|   | fa147c076f | ||
|   | b2df32bc81 | ||
|   | 733f4c71b8 | ||
|   | 70e8c6e9d3 | ||
|   | 2f33fe86f5 | ||
|   | 56e7bd01ca | ||
|   | 9459495484 | ||
|   | 89b0d362d2 | ||
|   | 574f25337d | ||
|   | 5bbe422161 | ||
|   | 6500714a93 | ||
|   | 5feafcddba | ||
|   | 3e20a3d180 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | 60639b6e50 | ||
|   | 839f384e45 | ||
|   | d42277979a | ||
|   | 24cf007a74 | ||
|   | 601f48a50e | ||
|   | c2b8e298d8 | ||
|   | 0917e17383 | ||
|   | 8587fa8585 | ||
|   | f345eeae55 | ||
|   | 89e2dbac15 | ||
|   | 356ada872c | ||
|   | a3deb48726 | ||
|   | 544cd5cd9e | ||
|   | 53b63adc71 | ||
|   | 2c349c50c7 | ||
|   | 08779c2909 | ||
|   | d5f9cdf912 | ||
|   | 3a4bdd7c56 | ||
|   | 2b236f3ff7 | ||
|   | 768fb791c5 | ||
|   | d00dcf3f58 | ||
|   | d5980303e5 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | f436ae7b31 | ||
|   | a9ccc18d73 | ||
|   | ac681687f8 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | 57ce19150f | ||
|   | 24f6747516 | ||
|   | 6dafebc7cc | ||
|   | d23f2560d5 | ||
|   | 880586bac4 | ||
|   | eac2a8c8dc | ||
|   | 4cf313c4c6 | ||
|   | 91353d0a4d | ||
|   | 0a6d64ab48 | ||
|   | 4fd0a76727 | ||
|   | 6da9d567dc | ||
|   | 6d5a3dff22 | ||
|   | 3ad5deaff1 | ||
|   | 9bbdf70e69 | ||
|   | 0c83946983 | ||
|   | fda05f2262 | ||
|   | 7abf1a5884 | ||
|   | 365acc36ea | ||
|   | 0482cd191d | ||
|   | 6a3fc71397 | ||
|   | 3c4192ebf6 | ||
|   | e450e1c447 | ||
|   | 20f841c513 | ||
|   | d07a3e09c9 | ||
|   | 4649876956 | ||
|   | 5604d140e3 | ||
|   | 8751fb4bb1 | ||
|   | 3819062574 | ||
|   | 051e6e76e9 | ||
|   | 1e55dd47f2 | ||
|   | 700b95546b | ||
|   | 2fa96ec0ed | ||
|   | 81e6f75aa4 | ||
|   | 888c8b9a84 | ||
|   | e775a8a22e | ||
|   | 99fbd9cae6 | ||
|   | 67adad3e08 | ||
|   | 2fca3c7563 | ||
|   | c3573f1a46 | ||
|   | ee932a9f8e | ||
|   | ce18c948e6 | ||
|   | 7bc93c5506 | ||
|   | d7cad3b404 | ||
|   | 7740a362c9 | ||
|   | 281ef53e7d | ||
|   | f044b948e2 | ||
|   | 32474a5f4d | ||
|   | 26596acf80 | ||
|   | e63870a631 | ||
|   | ce782ff6fb | ||
|   | c6716e030c | ||
|   | 4ab72acec6 | ||
|   | 30aae8e257 | ||
|   | d7b7ff7bb4 | ||
|   | 6fe0cff342 | ||
|   | 5f75f9886d | ||
|   | 5d9604cd15 | ||
|   | cc36ebf1c9 | ||
|   | e6adecfd81 | ||
|   | 5c8f224e3b | ||
|   | 952221d3b9 | ||
|   | 496d5b4ec7 | ||
|   | 2623a412c4 | ||
|   | d64eed49bc | ||
|   | fffa29c2f3 | ||
|   | 4da1444ffc | ||
|   | 21c4e56d16 | ||
|   | 5356b3856a | ||
|   | 320c996a21 | ||
|   | 69c74be7bb | ||
|   | aefa70891c | ||
|   | 1b9877fda4 | ||
|   | 0205a67309 | ||
|   | e3cafeaf92 | ||
|   | e7b193788a | ||
|   | 17da95b094 | ||
|   | c5e49eec96 | ||
|   | 24bc0f127b | ||
|   | f0f801402d | ||
|   | 663850a2b8 | ||
|   | c51753cab1 | ||
|   | b3be2e208c | ||
|   | c30e90ff3f | ||
|   | e4c0ca0f48 | ||
|   | 9c203327c0 | ||
|   | ccb5b1d075 | ||
|   | 0dbbd0414c | ||
|   | e7b3ebf98a | ||
|   | 5bc18fb780 | ||
|   | df30366072 | ||
|   | 65c7ac80b5 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | dd3fb32ec7 | ||
|   | 2a3f475ff5 | ||
|   | 7288f71201 | ||
|   | 9c43eff753 | ||
|   | c8d7fdeedc | ||
|   | c211152e23 | ||
|   | ab75d5097e | ||
|   | c3644c8d3b | ||
|   | 6438a3dba3 | ||
|   | 4b226a6a63 | ||
|   | 4801850013 | ||
|   | 6a7412bf2b | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | 5a1fd7dadd | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | ac06a26809 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | 61d56f26f8 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | 6aa05b3981 | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | aad60c882e | ||
| ![dependabot[bot]](/assets/img/avatar_default.png)  | fecca57507 | ||
|   | 2bcad846c0 | ||
|   | 15ad0165fc | ||
|   | 2e8ab11978 | ||
|   | 9a8ce9b17e | ||
|   | 16ab4c6fed | ||
|   | e3ee0df7ba | ||
|   | 8f7ab280e2 | ||
|   | dbedc99421 | ||
|   | 6cb359cb80 | ||
|   | ae2ad824a9 | ||
|   | 02e3d7852b | ||
|   | 3893a035be | ||
|   | 658bdd9faa | ||
|   | e1eebcd4e0 | ||
|   | 062b831e88 | ||
|   | b275efaeff | ||
|   | 80d3033456 | ||
|   | bd0516f09a | ||
|   | df4d76e466 | ||
|   | dcbd7f8cad | ||
|   | 73ec02ab9d | ||
|   | d1f8347071 | ||
|   | 8601eedada | ||
|   | 9afd33cdfc | ||
|   | 5e1be8e558 | ||
|   | 835dd2635a | ||
|   | f65b18c2f6 | ||
|   | b0e7b84f40 | ||
|   | 1635db93c7 | ||
|   | c4fe462d11 | ||
|   | b1f403165d | ||
|   | 46e4317b77 | ||
|   | e3ffbcadd8 | ||
|   | b7d73077e5 | ||
|   | 77f61ee20a | ||
|   | 8967f02fc9 | ||
|   | 831ff6d0a9 | ||
|   | 2199174def | ||
|   | 55f41ddaab | ||
|   | 21305d93bf | ||
|   | 4478d5d904 | ||
|   | cc6253a6b8 | ||
|   | 85f66853bc | ||
|   | 7464fd149c | ||
|   | 86f1a8019c | ||
|   | b98d56dcf6 | ||
|   | a3a8a5769d | ||
|   | 4dd8bae5c9 | ||
|   | 7ae45c42e7 | ||
|   | 7551b4e7a3 | ||
|   | 61bab22dde | ||
|   | 6dcc23ebb6 | ||
|   | b06a574cc5 | ||
|   | b56f80b1b8 | ||
|   | 20f6c05ec5 | ||
|   | 57fce93af7 | ||
|   | 110b6a1431 | ||
|   | 53cafa9f3d | ||
|   | d4195deb3a | ||
|   | 400ecfb79c | ||
|   | 86151da271 | ||
|   | 44f3e2557d | ||
|   | 1f365c716e | ||
|   | 9efcc41ab2 | ||
|   | 13bbeeaceb | ||
|   | da4dcec14d | ||
|   | 761c0b79c5 | ||
|   | d93ab0496f | ||
|   | 66b6f9749d | ||
|   | 17c2d1f26a | ||
|   | a79e632cdc | ||
|   | f36498421b | ||
|   | e45bbe4571 | ||
|   | fb5a84212c | ||
|   | dedc1c45a1 | ||
|   | 6a12f9ff84 | ||
|   | 641ed1873b | ||
|   | 1d50da4b1c | ||
|   | c7897cca5d | ||
|   | 4091b6f6b4 | ||
|   | 766f35554e | ||
|   | c86137449e | ||
|   | efec01a92f | ||
|   | 4fcad8e04b | ||
|   | 4b4b2d790e | ||
|   | ec6ae343dd | ||
|   | b9fb361959 | ||
|   | a189298ab0 | ||
|   | 714a2ad730 | ||
|   | fa8b96dfa1 | ||
|   | 01955a0df8 | ||
|   | ac4aee39e3 | ||
|   | a0bca42a7a | ||
|   | af543dcd05 | ||
|   | af77109a47 | ||
|   | b979aff270 | ||
|   | b293e3fa75 | ||
|   | 21eb37e471 | ||
|   | d3b60cc445 | ||
|   | 7466e1d014 | ||
|   | 2a7f28606c | ||
|   | 0450482e6e | ||
|   | ee5d9b43b5 | ||
|   | 3a8857c8c9 | ||
|   | be3dfb251d | ||
|   | 4e11e29f70 | ||
|   | 763bb95cea | ||
|   | 668e7407e6 | ||
|   | c147ba1da1 | ||
|   | 10f044c3dd | ||
|   | ce5140febd | ||
|   | 858cdc86f5 | ||
|   | 9a25297d51 | ||
|   | e24f7f5151 | ||
|   | eff5f1e119 | ||
|   | afcd362cd1 | ||
|   | 0452be0cb3 | ||
|   | 1624f10773 | ||
|   | 8764be7461 | ||
|   | 5dd15ef8e7 | ||
|   | 4ac6366706 | ||
|   | adc0912efa | ||
|   | 536823ce55 | ||
|   | 207cd24edb | ||
|   | b039da1eba | ||
|   | 8fcd0f3b6f | ||
|   | 16fde6935c | ||
|   | 9592cff9fa | ||
|   | 109148988c | ||
|   | cf13fff7d2 | ||
|   | a9d8ac8bc0 | ||
|   | 1a4717b366 | ||
|   | 6cadf12260 | ||
|   | 19d47784bd | ||
|   | b89102c5fc | ||
|   | 4f20ebead3 | ||
|   | a9f89dbc64 | ||
|   | 58ea1e07d2 | ||
|   | 6de4c7e971 | ||
|   | 03dc51ffa2 | ||
|   | aef2dcdfdd | ||
|   | 0494119bf4 | ||
|   | 0a17e21119 | ||
|   | 52e2f926f4 | ||
|   | 611fb279bc | ||
|   | 41b4e64be9 | ||
|   | 0d7315249d | ||
|   | 4913766d58 | ||
|   | 92da8c7044 | ||
|   | 9dba3d5385 | ||
|   | 2d3c26a4b2 | ||
|   | 8eba2d3e50 | ||
|   | a8d4a27de1 | ||
|   | c42167c6f4 | ||
|   | 44d182e2f9 | ||
|   | ad95e35687 | ||
|   | 640a9995f4 | ||
|   | 95625f6871 | ||
|   | 2c20f72a9c | ||
|   | 5ad788e768 | ||
|   | ed98c586c6 | ||
|   | 3e865708d6 | ||
|   | c3bcbd63c0 | ||
|   | 29e29439ee | ||
|   | 0c19716f44 | ||
|   | b24e1bafa1 | ||
|   | 64b899ac89 | ||
|   | aa274e5ab7 | ||
|   | 7b3eaf3ccf | ||
|   | 1a5353d768 | ||
|   | eff41759bc | ||
|   | c23252ab53 | ||
|   | 1a3c57a031 | ||
|   | 4cc2c914e6 | ||
|   | cbb46293ab | ||
|   | 765e00c949 | ||
|   | 662359908b | ||
|   | 0d99766686 | ||
|   | ae3bc3358b | ||
|   | 1e0b4532bd | ||
|   | 4f8b19c686 | ||
|   | 84ab223b81 | ||
|   | 2bb21262d4 | ||
|   | 3188a9ffe6 | ||
|   | 61569a8610 | ||
|   | 075a84427f | ||
|   | 950f2759bd | ||
|   | 25c82ddf02 | ||
|   | 2d98df6122 | ||
|   | 219a5453f9 | ||
|   | 214a6a1386 | ||
|   | e7781dc79c | ||
|   | 10c4bd1ac8 | ||
|   | a42e488e58 | ||
|   | 06eb89b05b | ||
|   | 91c58ec027 | ||
|   | 8b26e42a3a | ||
|   | acca011f15 | ||
|   | 2f59abdda7 | ||
|   | 17747a5c88 | ||
|   | cec63546ff | ||
|   | 75f67d2de4 | ||
|   | 712385ffd5 | ||
|   | ad90cf09fe | ||
|   | f9928c9e25 | ||
|   | a0741d99b8 | ||
|   | c63f08c811 | ||
|   | 58b6c4d277 | ||
|   | 27c02549c8 | ||
|   | 88d371c71c | ||
|   | b339524613 | ||
|   | d5feda5c8a | ||
|   | 2f506425c2 | ||
|   | e8167ee3d7 | ||
|   | 2f5e211065 | ||
|   | 39f4fb3446 | ||
|   | 56159b9bce | ||
|   | b2af76e7dc | ||
|   | 491fe35397 | ||
|   | b451285af7 | ||
|   | 63a1847cdc | ||
|   | 4e50fd8649 | ||
|   | dfdffa0027 | ||
|   | ebd2073144 | ||
|   | 1e94b716fb | ||
|   | 2a41abb3d1 | ||
|   | 2d2bebe976 | ||
|   | e1629994bd | ||
|   | e3d8fe4fd8 | ||
|   | 23d8742f0d | ||
|   | 3b6a8be07b | ||
|   | 71a5b72aff | ||
|   | 213bf349c3 | ||
|   | a94fe55886 | ||
|   | 9b22f16497 | ||
|   | 2977a5957e | ||
|   | f70d1c897a | ||
|   | a4a3525265 | ||
|   | a6dd8446e4 | ||
|   | 7bf9e1cfb3 | ||
|   | f291832a77 | ||
|   | 1fee323247 | ||
|   | a41accd033 | ||
|   | 37f7caf7f3 | ||
|   | 5847f7758c | ||
|   | bce736993e | ||
|   | 5636992446 | ||
|   | f996a2b7ae | ||
|   | 587de96ab3 | ||
|   | 80eb1cd202 | ||
|   | bbf594c815 | ||
|   | 2f0f2ee40d | ||
|   | 96022d3aaf | ||
|   | 8eb5e3cbf8 | ||
|   | ddc2625934 | ||
|   | 7f7ca697a0 | ||
|   | 900375679b | ||
|   | 9440b9e313 | ||
|   | 393f9e998b | ||
|   | ba0bfe70a8 | ||
|   | 3c4a3e3f75 | ||
|   | 274fb09ed4 | ||
|   | d44598a900 | ||
|   | c9cfa59f54 | ||
|   | 7062234331 | ||
|   | 9754569525 | ||
|   | 52a071e34d | ||
|   | 2d8f749e36 | ||
|   | a18cb74f03 | ||
|   | 6c442e239d | ||
|   | eaf92fca4d | ||
|   | 06b7bad714 | ||
|   | 19eec2ed03 | ||
|   | d99c54343a | ||
|   | 308a110000 | ||
|   | 4f406b2ce6 | ||
|   | e564c555d7 | ||
|   | f7ec9af9e8 | ||
|   | 4d93a774ce | ||
|   | 2595dd30bf | ||
|   | 9190365289 | ||
|   | 57794b3b9f | ||
|   | 3c36f651be | ||
|   | 8e6ddadba2 | ||
|   | 8a87a71927 | ||
|   | 0047e6f523 | ||
|   | 7183095a28 | ||
|   | 13c90893c7 | ||
|   | 976fbcd07f | ||
|   | d97b077e85 | ||
|   | 8950575bfb | ||
|   | 11fc4c286f | ||
|   | 8d08e348a9 | ||
|   | a18807f19e | ||
|   | 29f658fd3c | ||
|   | a30bb8fed0 | ||
|   | 092ca1cd67 | ||
|   | 0df2539641 | ||
|   | 0f2d8a599c | ||
|   | 54b3143a1d | ||
|   | 148f7d2a91 | ||
|   | 1aa662f763 | ||
|   | 0b86b88de7 | ||
|   | 98033b1ba7 | ||
|   | 2b7eab629d | ||
|   | 0e4973e15c | ||
|   | af0acf0dae | ||
|   | 76e5fe5a87 | ||
|   | 802c80f40c | ||
|   | a51c5bd905 | ||
|   | 8c68556f52 | ||
|   | cca1ea2404 | ||
|   | 281016a501 | ||
|   | d4acdf2f89 | ||
|   | 0951e75c85 | ||
|   | 6b017b226a | ||
|   | 9e3bd7398c | ||
|   | 79f764c7a8 | ||
|   | b5dc4353fb | ||
|   | 2fbac73c29 | ||
|   | 6616d105d1 | ||
|   | 6b4b19194e | ||
|   | 9785edd263 | ||
|   | 2a0bc11b68 | ||
|   | dd0325a88d | ||
|   | 20783c0978 | ||
|   | 3f06a40bd5 | ||
|   | 68f43985ad | ||
|   | 915ca8f817 | ||
|   | a65a81610b | ||
|   | 8eb6ed5639 | ||
|   | 795a8705c3 | ||
|   | 3af0dc3b3a | ||
|   | 9cf9b958a3 | ||
|   | 3ac2ba8d5a | ||
|   | d893421c7b | ||
|   | 250b3bb579 | ||
|   | e9edbfc051 | ||
|   | e343db6f72 | ||
|   | 4d57d66f85 | ||
|   | 54ed6320c2 | ||
|   | 23083f3ae0 | ||
|   | 1985873494 | ||
|   | 8ae5917659 | ||
|   | c91bfd08d8 | ||
|   | 49110a5872 | ||
|   | c01c8edeb8 | ||
|   | ff8cf067b8 | ||
|   | 1420f68050 | ||
|   | c0be3e585a | ||
|   | 3049ef9151 | ||
|   | 4be00bbe6b | ||
|   | 9382dde098 | ||
|   | 1bf46b7711 | ||
|   | b85bae31d9 | ||
|   | 0898829313 | ||
|   | f8ad877601 | ||
|   | 585d1556c1 | ||
|   | 7486555875 | ||
|   | fc30b1bacc | ||
|   | 0dd19af6e8 | ||
|   | 4c44515f9d | ||
|   | 9d84d6dd64 | ||
|   | 0f708daf2d | ||
|   | b9354de8fd | ||
|   | c9d5f4c898 | ||
|   | 810c150781 | ||
|   | 31dd538c0b | ||
|   | 62e38e7c45 | ||
|   | b9da28a29b | ||
|   | 84bfa8a6b1 | ||
|   | 1f830963f6 | ||
|   | 12d2c6fe89 | ||
|   | f43faf15f8 | ||
|   | 173a38a374 | ||
|   | 1604ff15b5 | ||
|   | 214fe502cd | ||
|   | aae45a8179 | ||
|   | 075ca9ca47 | ||
|   | d4253d7a55 | ||
|   | 0917dc8766 | ||
|   | aba86855b5 | ||
|   | ed5386c213 | ||
|   | 455e75e92f | ||
|   | c394de0c88 | ||
|   | bad1990173 | ||
|   | 0bc159341d | ||
|   | 45bf1fd63a | ||
|   | ff0de85817 | ||
|   | 727fa9f929 | ||
|   | 0b9bc18236 | ||
|   | bad3b83d33 | ||
|   | 00967a98ac | ||
|   | 1d708ab351 | ||
|   | ba6759010b | ||
|   | da3868c104 | ||
|   | 0abf4d5d5d | ||
|   | 9b320cd43f | ||
|   | 28783a4146 | ||
|   | f92927eae5 | ||
|   | 294139ce7a | ||
|   | 45becd2573 | ||
|   | a3bee01e0a | ||
|   | 1dc93ec4f0 | ||
|   | 3562d4220c | ||
|   | 1532f6e427 | ||
|   | 9327810bbf | ||
|   | f66d5f1e58 | ||
|   | cec086994e | ||
|   | 942d8f1ced | ||
|   | 1552dcb143 | ||
|   | d525f1c9e4 | ||
|   | 921f2dfcdf | ||
|   | 79a006c8de | ||
|   | ff27746c0c | ||
|   | 87788f354f | ||
|   | 7d2e440c83 | ||
|   | 5551f9d56f | ||
|   | 1fb91c6316 | ||
|   | e60949ff3f | ||
|   | 278a3c6890 | ||
|   | fcf734eb36 | ||
|   | cf3cddafab | ||
|   | c52664f22e | ||
|   | cb712ff37d | ||
|   | f4ae610448 | ||
|   | 601b8bc98d | ||
|   | 80b4cec87a | ||
|   | 76c7b69e4e | ||
|   | a5bd3c4dda | ||
|   | f06e9b5605 | ||
|   | 7a3bb0e55c | ||
|   | 6e8f535e8b | ||
|   | 5619a75b05 | ||
|   | 53dfb78215 | ||
|   | 8e97cbab1e | ||
|   | ce7b749fd5 | ||
|   | 6617bd6609 | ||
|   | e610fb3201 | ||
|   | 40f1d35415 | ||
|   | b79bf7d414 | ||
|   | 3724cc3a15 | ||
|   | 3418e8c9af | ||
|   | 9619dff334 | ||
|   | 1b2feb19e5 | ||
|   | 1829dc3d9f | ||
|   | bd0e81f5a0 | ||
|   | f04d360ee2 | ||
|   | 92f27281fa | ||
|   | 65781b9316 | ||
|   | 9be0be0316 | ||
|   | 9f5f004725 | ||
|   | fed77cccf3 | ||
|   | 9b520dfb78 | ||
|   | 8ad2be10b2 | ||
|   | 2d277a15f5 | ||
|   | d60468bb05 | ||
|   | 82d6210464 | ||
|   | ff198042d2 | ||
|   | 6b47e29583 | ||
|   | 380c38674c | ||
|   | 3c14a0891e | ||
|   | 8513a07416 | ||
|   | 220485a849 | ||
|   | 4db34b0506 | ||
|   | 5677c912a8 | ||
|   | 7a24de15e4 | ||
|   | 99d9ea283a | ||
|   | dac92a0e0a | ||
|   | a25efb16f3 | ||
|   | e4d73b29a1 | ||
|   | 8a875f292e | ||
|   | 60a85621ea | ||
|   | 115d20373c | ||
|   | cdf33e5748 | ||
|   | 01d0a9f412 | ||
|   | 8cc2d3b4fe | ||
|   | aba9e4f3be | ||
|   | 4d575ba13a | ||
|   | 7f0e4ad448 | ||
|   | 17cc14a9d2 | ||
|   | 1f8016182c | ||
|   | caf9ef2c4b | ||
|   | 64b57f2da3 | ||
|   | efd2c99862 | ||
|   | cc05ba8907 | ||
|   | 16763b715a | ||
|   | ffaa598796 | ||
|   | 858e16d34f | ||
|   | a60e62efb1 | ||
|   | 97f9d4be67 | ||
|   | fa4eec41f7 | ||
|   | 77516c97db | ||
|   | cba01f0865 | ||
|   | 8b754017ca | ||
|   | a27600046e | ||
|   | fb2667631d | ||
|   | b638f7037a | ||
|   | 74699a8262 | ||
|   | eabf2a4582 | ||
|   | 325d62b41c | ||
|   | e955a056e2 | ||
|   | 723f8c5fd5 | ||
|   | a16137f53f | ||
|   | d60b8b97f9 | 
							
								
								
									
										2
									
								
								.dockerignore
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								.dockerignore
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,2 @@ | ||||
| Dockerfile | ||||
| tgs.Dockerfile | ||||
							
								
								
									
										3
									
								
								.fixmie.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								.fixmie.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,3 @@ | ||||
| go: | ||||
|   comments: | ||||
|     disabled: true | ||||
							
								
								
									
										1
									
								
								.github/ISSUE_TEMPLATE/Bug_report.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.github/ISSUE_TEMPLATE/Bug_report.md
									
									
									
									
										vendored
									
									
								
							| @@ -1,6 +1,7 @@ | ||||
| --- | ||||
| name: Bug report | ||||
| about: Create a report to help us improve. (Check the FAQ on the wiki first) | ||||
| labels: bug | ||||
|  | ||||
| --- | ||||
|  | ||||
|   | ||||
							
								
								
									
										1
									
								
								.github/ISSUE_TEMPLATE/Feature_request.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.github/ISSUE_TEMPLATE/Feature_request.md
									
									
									
									
										vendored
									
									
								
							| @@ -1,6 +1,7 @@ | ||||
| --- | ||||
| name: Feature request | ||||
| about: Suggest an idea for this project | ||||
| labels: enhancement | ||||
|  | ||||
| --- | ||||
|  | ||||
|   | ||||
							
								
								
									
										71
									
								
								.github/workflows/codeql-analysis.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										71
									
								
								.github/workflows/codeql-analysis.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,71 @@ | ||||
| # For most projects, this workflow file will not need changing; you simply need | ||||
| # to commit it to your repository. | ||||
| # | ||||
| # You may wish to alter this file to override the set of languages analyzed, | ||||
| # or to provide custom queries or build logic. | ||||
| name: "CodeQL" | ||||
|  | ||||
| on: | ||||
|   push: | ||||
|     branches: [master] | ||||
|   pull_request: | ||||
|     # The branches below must be a subset of the branches above | ||||
|     branches: [master] | ||||
|   schedule: | ||||
|     - cron: '0 16 * * 1' | ||||
|  | ||||
| jobs: | ||||
|   analyze: | ||||
|     name: Analyze | ||||
|     runs-on: ubuntu-latest | ||||
|  | ||||
|     strategy: | ||||
|       fail-fast: false | ||||
|       matrix: | ||||
|         # Override automatic language detection by changing the below list | ||||
|         # Supported options are ['csharp', 'cpp', 'go', 'java', 'javascript', 'python'] | ||||
|         language: ['go'] | ||||
|         # Learn more... | ||||
|         # https://docs.github.com/en/github/finding-security-vulnerabilities-and-errors-in-your-code/configuring-code-scanning#overriding-automatic-language-detection | ||||
|  | ||||
|     steps: | ||||
|     - name: Checkout repository | ||||
|       uses: actions/checkout@v2 | ||||
|       with: | ||||
|         # We must fetch at least the immediate parents so that if this is | ||||
|         # a pull request then we can checkout the head. | ||||
|         fetch-depth: 2 | ||||
|  | ||||
|     # If this run was triggered by a pull request event, then checkout | ||||
|     # the head of the pull request instead of the merge commit. | ||||
|     - run: git checkout HEAD^2 | ||||
|       if: ${{ github.event_name == 'pull_request' }} | ||||
|  | ||||
|     # Initializes the CodeQL tools for scanning. | ||||
|     - name: Initialize CodeQL | ||||
|       uses: github/codeql-action/init@v1 | ||||
|       with: | ||||
|         languages: ${{ matrix.language }} | ||||
|         # If you wish to specify custom queries, you can do so here or in a config file. | ||||
|         # By default, queries listed here will override any specified in a config file.  | ||||
|         # Prefix the list here with "+" to use these queries and those in the config file. | ||||
|         # queries: ./path/to/local/query, your-org/your-repo/queries@main | ||||
|  | ||||
|     # Autobuild attempts to build any compiled languages  (C/C++, C#, or Java). | ||||
|     # If this step fails, then you should remove it and run the build manually (see below) | ||||
|     - name: Autobuild | ||||
|       uses: github/codeql-action/autobuild@v1 | ||||
|  | ||||
|     # ℹ️ Command-line programs to run using the OS shell. | ||||
|     # 📚 https://git.io/JvXDl | ||||
|  | ||||
|     # ✏️ If the Autobuild fails above, remove it and uncomment the following three lines | ||||
|     #    and modify them (or add more) to build your code if your project | ||||
|     #    uses a compiled language | ||||
|  | ||||
|     #- run: | | ||||
|     #   make bootstrap | ||||
|     #   make release | ||||
|  | ||||
|     - name: Perform CodeQL Analysis | ||||
|       uses: github/codeql-action/analyze@v1 | ||||
							
								
								
									
										58
									
								
								.github/workflows/development.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										58
									
								
								.github/workflows/development.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,58 @@ | ||||
| name: Development | ||||
| on: [push, pull_request] | ||||
| jobs: | ||||
|   lint: | ||||
|     name: golangci-lint | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - uses: actions/checkout@v3 | ||||
|         with: | ||||
|           fetch-depth: 20 | ||||
|       - name: Run golangci-lint | ||||
|         uses: golangci/golangci-lint-action@v3 | ||||
|         with: | ||||
|           version: latest | ||||
|           args: "-v --new-from-rev HEAD~5 --timeout=5m" | ||||
|   test-build-upload: | ||||
|     strategy: | ||||
|       matrix: | ||||
|         go-version: [1.22.x] | ||||
|         platform: [ubuntu-latest] | ||||
|     runs-on: ${{ matrix.platform }} | ||||
|     steps: | ||||
|     - name: Install Go | ||||
|       uses: actions/setup-go@v3 | ||||
|       with: | ||||
|         go-version: ${{ matrix.go-version }} | ||||
|         stable: false | ||||
|     - name: Checkout code | ||||
|       uses: actions/checkout@v3 | ||||
|       with: | ||||
|           fetch-depth: 0 | ||||
|     - name: Test | ||||
|       run: go test ./... -mod=vendor | ||||
|     - name: Build | ||||
|       run: | | ||||
|         mkdir -p output/{win,lin,arm,mac} | ||||
|         VERSION=$(git describe --tags) | ||||
|         CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags "-s -X github.com/42wim/matterbridge/version.GitHash=$(git log --pretty=format:'%h' -n 1)" -o output/lin/matterbridge-$VERSION-linux-amd64 | ||||
|         CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -ldflags "-s -X github.com/42wim/matterbridge/version.GitHash=$(git log --pretty=format:'%h' -n 1)" -o output/win/matterbridge-$VERSION-windows-amd64.exe | ||||
|         CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -ldflags "-s -X github.com/42wim/matterbridge/version.GitHash=$(git log --pretty=format:'%h' -n 1)" -o output/mac/matterbridge-$VERSION-darwin-amd64 | ||||
|     - name: Upload linux 64-bit | ||||
|       if: startsWith(matrix.go-version,'1.22') | ||||
|       uses: actions/upload-artifact@v3 | ||||
|       with: | ||||
|         name: matterbridge-linux-64bit | ||||
|         path: output/lin | ||||
|     - name: Upload windows 64-bit | ||||
|       if: startsWith(matrix.go-version,'1.22') | ||||
|       uses: actions/upload-artifact@v3 | ||||
|       with: | ||||
|         name: matterbridge-windows-64bit | ||||
|         path: output/win | ||||
|     - name: Upload darwin 64-bit | ||||
|       if: startsWith(matrix.go-version,'1.22') | ||||
|       uses: actions/upload-artifact@v3 | ||||
|       with: | ||||
|         name: matterbridge-darwin-64bit | ||||
|         path: output/mac | ||||
							
								
								
									
										68
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										68
									
								
								.github/workflows/docker.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,68 @@ | ||||
| name: docker | ||||
|  | ||||
| on: | ||||
|   push: | ||||
|     branches: | ||||
|       - 'master' | ||||
|     tags: | ||||
|       - 'v*' | ||||
|   pull_request: | ||||
|     branches: | ||||
|       - 'master' | ||||
|  | ||||
| jobs: | ||||
|   docker: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - | ||||
|         name: Checkout | ||||
|         uses: actions/checkout@v2 | ||||
|       - | ||||
|         name: Set up QEMU | ||||
|         uses: docker/setup-qemu-action@v1 | ||||
|         with: | ||||
|           platforms: amd64,arm64 | ||||
|       - | ||||
|         name: Set up Docker Buildx | ||||
|         uses: docker/setup-buildx-action@v1 | ||||
|       - | ||||
|         name: Docker meta | ||||
|         id: meta | ||||
|         uses: docker/metadata-action@v3 | ||||
|         with: | ||||
|           images: 42wim/matterbridge,ghcr.io/42wim/matterbridge | ||||
|           flavor: | | ||||
|             latest=true | ||||
|           tags: | | ||||
|             type=ref,event=branch | ||||
|             type=ref,event=pr | ||||
|             type=semver,pattern={{version}} | ||||
|             type=semver,pattern=stable | ||||
|             type=semver,pattern={{major}} | ||||
|             type=semver,pattern={{major}}.{{minor}} | ||||
|       - | ||||
|         name: Login to DockerHub | ||||
|         uses: docker/login-action@v1  | ||||
|         if: github.event_name != 'pull_request' | ||||
|         with: | ||||
|           username: ${{ secrets.DOCKERHUB_USERNAME }} | ||||
|           password: ${{ secrets.DOCKERHUB_TOKEN }} | ||||
|       - | ||||
|         name: Log into registry ghcr.io | ||||
|         uses: docker/login-action@v1 | ||||
|         if: github.event_name != 'pull_request' | ||||
|         with: | ||||
|           registry: ghcr.io | ||||
|           username: ${{ github.repository_owner }} | ||||
|           password: ${{ secrets.GITHUB_TOKEN }} | ||||
|       - | ||||
|         name: Build and push | ||||
|         id: docker_build | ||||
|         uses: docker/build-push-action@v2 | ||||
|         with: | ||||
|           context: . | ||||
|           platforms: linux/amd64,linux/arm64 | ||||
|           push: ${{ github.event_name != 'pull_request' }} | ||||
|           tags: ${{ steps.meta.outputs.tags }} | ||||
|           labels: ${{ steps.meta.outputs.labels }} | ||||
|  | ||||
							
								
								
									
										9
									
								
								.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| # Exclude matterbridge binary | ||||
| /matterbridge | ||||
| /matterbridge.exe | ||||
|  | ||||
| # Exclude configuration file | ||||
| matterbridge.toml | ||||
|  | ||||
| # Exclude IDE Files | ||||
| .vscode | ||||
| @@ -7,7 +7,7 @@ run: | ||||
|   # concurrency: 4 | ||||
|  | ||||
|   # timeout for analysis, e.g. 30s, 5m, default is 1m | ||||
|   deadline: 1m | ||||
|   deadline: 5m | ||||
|  | ||||
|   # exit code when at least one issue was found, default is 1 | ||||
|   issues-exit-code: 1 | ||||
| @@ -23,7 +23,7 @@ run: | ||||
|   # default value is empty list, but next dirs are always skipped independently | ||||
|   # from this option's value: | ||||
|   #   	vendor$, third_party$, testdata$, examples$, Godeps$, builtin$ | ||||
|   skip-dirs: | ||||
|   skip-dirs: gateway/bridgemap$ | ||||
|  | ||||
|   # which files to skip: they will be analyzed, but issues from them | ||||
|   # won't be reported. Default value is empty list, but there is | ||||
| @@ -91,7 +91,6 @@ linters-settings: | ||||
|     # Correct spellings using locale preferences for US or UK. | ||||
|     # Default is to use a neutral variety of English. | ||||
|     # Setting locale to US will correct the British spelling of 'colour' to 'color'. | ||||
|     locale: US | ||||
|   lll: | ||||
|     # max line length, lines longer will be reported. Default is 120. | ||||
|     # '\t' is counted as 1 character by default, and can be changed with the tab-width option | ||||
| @@ -105,10 +104,6 @@ linters-settings: | ||||
|     # with golangci-lint call it on a directory with the changed file. | ||||
|     check-exported: false | ||||
|   unparam: | ||||
|     # call graph construction algorithm (cha, rta). In general, use cha for libraries, | ||||
|     # and rta for programs with main packages. Default is cha. | ||||
|     algo: rta | ||||
|  | ||||
|     # Inspect exported functions, default is false. Set to true if no external program/library imports your code. | ||||
|     # XXX: if you enable this setting, unparam will report a lot of false-positives in text editors: | ||||
|     # if it's called for subdir of a project it can't find external interfaces. All text editor integrations | ||||
| @@ -158,7 +153,6 @@ linters-settings: | ||||
|       - regexpMust | ||||
|       - singleCaseSwitch | ||||
|       - sloppyLen | ||||
|       - sloppyReassign | ||||
|       - switchTrue | ||||
|       - typeSwitchVar | ||||
|       - typeUnparen | ||||
| @@ -179,8 +173,48 @@ linters: | ||||
|     - lll | ||||
|     - maligned | ||||
|     - prealloc | ||||
|  | ||||
|  | ||||
|     - wsl | ||||
|     - gomnd | ||||
|     - godox | ||||
|     - goerr113 | ||||
|     - testpackage | ||||
|     - godot | ||||
|     - interfacer | ||||
|     - goheader | ||||
|     - noctx | ||||
|     - gci | ||||
|     - errorlint | ||||
|     - nlreturn | ||||
|     - exhaustivestruct | ||||
|     - forbidigo | ||||
|     - wrapcheck | ||||
|     - varnamelen | ||||
|     - ireturn | ||||
|     - errorlint | ||||
|     - tparallel | ||||
|     - wrapcheck | ||||
|     - paralleltest | ||||
|     - makezero | ||||
|     - thelper | ||||
|     - cyclop | ||||
|     - revive | ||||
|     - importas | ||||
|     - gomoddirectives | ||||
|     - promlinter | ||||
|     - tagliatelle | ||||
|     - errname | ||||
|     - typecheck | ||||
|     - grouper | ||||
|     - decorder | ||||
|     - maintidx | ||||
|     - exhaustruct | ||||
|     - asasalint | ||||
|     - execinquery | ||||
|     - nosnakecase | ||||
|     - exhaustive | ||||
|     - testifylint | ||||
|     - mnd | ||||
|     - depguard | ||||
| # rules to deal with reported isues | ||||
| issues: | ||||
|   # List of regexps of issue texts to exclude, empty list by default. | ||||
|   | ||||
| @@ -18,17 +18,24 @@ builds: | ||||
|     - arm | ||||
|     - arm64 | ||||
|     - 386 | ||||
|   goarm: | ||||
|     - 6 | ||||
|     - 7 | ||||
|   ldflags: | ||||
|     - -s -w -X main.githash={{.ShortCommit}} | ||||
|     - -s -w -X github.com/42wim/matterbridge/version.GitHash={{.ShortCommit}} | ||||
|  | ||||
| archive: | ||||
|   name_template: "{{ .Binary }}-{{ .Version }}-{{ .Os }}-{{ .Arch }}{{ if .Arm }}v{{ .Arm }}{{ end }}" | ||||
|   format: binary | ||||
|   files: | ||||
|     - none* | ||||
|   replacements: | ||||
|     386: 32bit | ||||
|     amd64: 64bit | ||||
| archives: | ||||
|   - | ||||
|     id: matterbridge | ||||
|     builds: | ||||
|     - matterbridge | ||||
|     name_template: "{{ .Binary }}-{{ .Version }}-{{ .Os }}-{{ .Arch }}{{ if .Arm }}v{{ .Arm }}{{ end }}" | ||||
|     format: binary | ||||
|     files: | ||||
|       - none* | ||||
|     replacements: | ||||
|       386: 32bit | ||||
|       amd64: 64bit | ||||
|  | ||||
| checksum: | ||||
|   name_template: 'checksums.txt' | ||||
|   | ||||
							
								
								
									
										63
									
								
								.travis.yml
									
									
									
									
									
								
							
							
						
						
									
										63
									
								
								.travis.yml
									
									
									
									
									
								
							| @@ -1,63 +0,0 @@ | ||||
| language: go | ||||
| go: | ||||
|     - 1.11.x | ||||
| go_import_path: github.com/42wim/matterbridge | ||||
|  | ||||
| # we have everything vendored | ||||
| install: true | ||||
|  | ||||
| git: | ||||
|   depth: 200 | ||||
|  | ||||
| env: | ||||
|   global: | ||||
|     - GOOS=linux GOARCH=amd64 | ||||
|     - GOLANGCI_VERSION="v1.14.0" | ||||
|  | ||||
| matrix: | ||||
|   # It's ok if our code fails on unstable development versions of Go. | ||||
|   allow_failures: | ||||
|     - go: tip | ||||
|   # Don't wait for tip tests to finish. Mark the test run green if the | ||||
|   # tests pass on the stable versions of Go. | ||||
|   fast_finish: true | ||||
|  | ||||
| notifications: | ||||
|       email: false | ||||
|  | ||||
| before_script: | ||||
|   # Get version info from tags. | ||||
|   - MY_VERSION="$(git describe --tags)" | ||||
|   # Retrieve the golangci-lint linter binary. | ||||
|   - curl -sfL https://install.goreleaser.com/github.com/golangci/golangci-lint.sh | bash -s -- -b ${GOPATH}/bin ${GOLANGCI_VERSION} | ||||
|   # Retrieve and prepare CodeClimate's test coverage reporter. | ||||
|   - curl -L https://codeclimate.com/downloads/test-reporter/test-reporter-latest-linux-amd64 > ./cc-test-reporter | ||||
|   - chmod +x ./cc-test-reporter | ||||
|   - ./cc-test-reporter before-build | ||||
|  | ||||
| script: | ||||
|   # Run the linter. | ||||
|   - golangci-lint run | ||||
|   # Run all the tests with the race detector and generate coverage. | ||||
|   - go test -v -race -coverprofile c.out ./... | ||||
|   # Run the build script to generate the necessary binaries and images. | ||||
|   - /bin/bash ci/bintray.sh | ||||
|  | ||||
| after_script: | ||||
|   # Upload test coverage to CodeClimate. | ||||
|   - ./cc-test-reporter after-build --exit-code ${TRAVIS_TEST_RESULT} | ||||
|  | ||||
| deploy: | ||||
|   on: | ||||
|      all_branches: true | ||||
|   provider: bintray | ||||
|   on: | ||||
|     all_branches: true | ||||
|   edge: | ||||
|     branch: v1.8.47 | ||||
|   file: ci/deploy.json | ||||
|   user: 42wim | ||||
|   on: | ||||
|      all_branches: true | ||||
|   key: | ||||
|      secure: "CeXXe6JOmt7HYR81MdWLua0ltQHhDdkIeRGBFbgd7hkb1wi8eF9DgpAcQrTso8NIlHNZmSAP46uhFgsRvkuezzX0ygalZ7DCJyAyn3sAMEh+UQSHV1WGThRehTtidqRGjetzsIGSwdrJOWil+XTfbO1Z8DGzfakhSuAZka8CM4BAoe3YeP9rYK8h+84x0GHfczvsLtXZ3mWLvQuwe4pK6+ItBCUg0ae7O7ZUpWHy0xQQkkWztY/6RAzXfaG7DuGjIw+20fhx3WOXRNpHCtZ6Bc3qERCpk0s1HhlQWlrN9wDaFTBWYwlvSnNgvxxMbNXJ6RrRJ0l0bA7FUswYwyroxhzrGLdzWDg8dHaQkypocngdalfhpsnoO9j3ApJhomUFJ3UoEq5nOGRUrKn8MPi+dP0zE4kNQ3e4VNa1ufNrvfpWolMg3xh8OXuhQdD5wIM5zFAbRJLqWSCVAjPq4DDPecmvXBOlIial7oa312lN5qnBnUjvAcxszZ+FUyDHT1Grxzna4tMwxY9obPzZUzm7359AOCCwIQFVB8GLqD2nwIstcXS0zGRz+fhviPipHuBa02q5bGUZwmkvrSNab0s8Jo7pCrel2Rz3nWPKaiCfq2WjbW1CLheSMkOQrjsdUd1hhbqNWFPUjJPInTc77NAKCfm5runv5uyowRLh4NNd0sI=" | ||||
							
								
								
									
										23
									
								
								Dockerfile
									
									
									
									
									
								
							
							
						
						
									
										23
									
								
								Dockerfile
									
									
									
									
									
								
							| @@ -1,11 +1,14 @@ | ||||
| FROM alpine:edge | ||||
| ENTRYPOINT ["/bin/matterbridge"] | ||||
| FROM alpine AS builder | ||||
|  | ||||
| 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 | ||||
| COPY . /go/src/matterbridge | ||||
| RUN apk --no-cache add go git \ | ||||
|         && cd /go/src/matterbridge \ | ||||
|         && CGO_ENABLED=0 go build -mod vendor -ldflags "-X github.com/42wim/matterbridge/version.GitHash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge | ||||
|  | ||||
| FROM alpine | ||||
| RUN apk --no-cache add ca-certificates mailcap | ||||
| COPY --from=builder /bin/matterbridge /bin/matterbridge | ||||
| RUN mkdir /etc/matterbridge \ | ||||
|   && touch /etc/matterbridge/matterbridge.toml \ | ||||
|   && ln -sf /matterbridge.toml /etc/matterbridge/matterbridge.toml | ||||
| ENTRYPOINT ["/bin/matterbridge", "-conf", "/etc/matterbridge/matterbridge.toml"] | ||||
|   | ||||
							
								
								
									
										14
									
								
								Dockerfile_whatsappmulti
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								Dockerfile_whatsappmulti
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| FROM alpine AS builder | ||||
|  | ||||
| COPY . /go/src/matterbridge | ||||
| RUN apk --no-cache add go git \ | ||||
|         && cd /go/src/matterbridge \ | ||||
|         && CGO_ENABLED=0 go build -tags whatsappmulti -mod vendor -ldflags "-X github.com/42wim/matterbridge/version.GitHash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge | ||||
|  | ||||
| FROM alpine | ||||
| RUN apk --no-cache add ca-certificates mailcap | ||||
| COPY --from=builder /bin/matterbridge /bin/matterbridge | ||||
| RUN mkdir /etc/matterbridge \ | ||||
|   && touch /etc/matterbridge/matterbridge.toml \ | ||||
|   && ln -sf /matterbridge.toml /etc/matterbridge/matterbridge.toml | ||||
| ENTRYPOINT ["/bin/matterbridge", "-conf", "/etc/matterbridge/matterbridge.toml"] | ||||
							
								
								
									
										454
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										454
									
								
								README.md
									
									
									
									
									
								
							| @@ -3,160 +3,275 @@ | ||||
| # matterbridge | ||||
|  | ||||
| <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 or join the development chat.</sub> | ||||
| **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 or join the development chat.</sub> | ||||
|  | ||||
|    <sup> | ||||
|  | ||||
|    [Gitter][mb-gitter] | | ||||
|    [IRC][mb-irc] | | ||||
|       [Discord][mb-discord] | | ||||
|       [Matrix][mb-matrix] | | ||||
|       [Slack][mb-slack] | | ||||
|       [Mattermost][mb-mattermost] | | ||||
|       [Rocket.Chat][mb-rocketchat] | | ||||
|       [XMPP][mb-xmpp] | | ||||
|       [Twitch][mb-twitch] | | ||||
|       [WhatsApp][mb-whatsapp] | | ||||
|       [Zulip][mb-zulip] | | ||||
|       [Telegram][mb-telegram] | | ||||
|       And more... | ||||
|    </sup> | ||||
| [Discord][mb-discord] | | ||||
| [Gitter][mb-gitter] | | ||||
| [IRC][mb-irc] | | ||||
| [Keybase][mb-keybase] | | ||||
| [Matrix][mb-matrix] | | ||||
| [Mattermost][mb-mattermost] | | ||||
| [MSTeams][mb-msteams] | | ||||
| [Rocket.Chat][mb-rocketchat] | | ||||
| [Slack][mb-slack] | | ||||
| [Telegram][mb-telegram] | | ||||
| [Twitch][mb-twitch] | | ||||
| [WhatsApp][mb-whatsapp] | | ||||
| [XMPP][mb-xmpp] | | ||||
| [Zulip][mb-zulip] | | ||||
| And more... | ||||
| </sup> | ||||
|  | ||||
| --- | ||||
|  | ||||
| ---- | ||||
| [](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 /> | ||||
| [](https://codeclimate.com/github/42wim/matterbridge/maintainability) | ||||
| [](https://codeclimate.com/github/42wim/matterbridge/test_coverage)<br /> | ||||
|  | ||||
|   <hr /> | ||||
| </div> | ||||
| <div align="right"><sup> | ||||
|  | ||||
| **Note:** Matter<em>most</em> isn't required to run matter<em>bridge</em>.</sup></div> | ||||
|  | ||||
| ### Table of Contents | ||||
|  * [Features](https://github.com/42wim/matterbridge/wiki/Features) | ||||
|    * [Natively supported](#natively-supported) | ||||
|    * [3rd party via matterbridge api](#3rd-party-via-matterbridge-api) | ||||
|    * [API](#API) | ||||
|  * [Chat with us](#chat-with-us) | ||||
|  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) | ||||
|  * [Installing](#installing) | ||||
|    * [Binaries](#binaries) | ||||
|    * [Building](#building) | ||||
|  * [Configuration](#configuration) | ||||
|    * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) | ||||
|    * [Examples](#examples) | ||||
|  * [Running](#running) | ||||
|    * [Docker](#docker) | ||||
|  * [Changelog](#changelog) | ||||
|  * [FAQ](#faq) | ||||
|  * [Related projects](#related-projects) | ||||
|  * [Articles](#articles) | ||||
|  * [Thanks](#thanks) | ||||
| <p> | ||||
|   <a href="https://www.digitalocean.com/"> | ||||
|     <img src="https://opensource.nyc3.cdn.digitaloceanspaces.com/attribution/assets/PoweredByDO/DO_Powered_by_Badge_blue.svg" width="201px"> | ||||
|   </a> | ||||
| </p> | ||||
|  | ||||
| # Table of Contents | ||||
|  | ||||
| - [matterbridge](#matterbridge) | ||||
| - [Table of Contents](#table-of-contents) | ||||
|   - [Features](#features) | ||||
|     - [Natively supported](#natively-supported) | ||||
|     - [3rd party via matterbridge api](#3rd-party-via-matterbridge-api) | ||||
|     - [API](#api) | ||||
|   - [Chat with us](#chat-with-us) | ||||
|   - [Screenshots](#screenshots) | ||||
|   - [Installing / upgrading](#installing--upgrading) | ||||
|     - [Binaries](#binaries) | ||||
|     - [Packages](#packages) | ||||
|   - [Building](#building) | ||||
|   - [Building with whatsapp (beta) multidevice support](#building-with-whatsapp-beta-multidevice-support) | ||||
|   - [Configuration](#configuration) | ||||
|     - [Basic configuration](#basic-configuration) | ||||
|     - [Settings](#settings) | ||||
|     - [Advanced configuration](#advanced-configuration) | ||||
|       - [Examples](#examples) | ||||
|         - [Bridge mattermost (off-topic) - irc (#testing)](#bridge-mattermost-off-topic---irc-testing) | ||||
|         - [Bridge slack (#general) - discord (general)](#bridge-slack-general---discord-general) | ||||
|     - [Running](#running) | ||||
|       - [Docker](#docker) | ||||
|       - [Systemd](#systemd) | ||||
|     - [Changelog](#changelog) | ||||
|     - [FAQ](#faq) | ||||
|     - [Related projects](#related-projects) | ||||
|     - [Articles / Tutorials](#articles--tutorials) | ||||
|     - [Thanks](#thanks) | ||||
|  | ||||
| ## Features | ||||
| * [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols) | ||||
| * [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols) | ||||
| * [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes) | ||||
| * Preserves threading when possible | ||||
| * [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling) | ||||
| * [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing) | ||||
| * [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups) | ||||
| * [API](https://github.com/42wim/matterbridge/wiki/Features#api) | ||||
|  | ||||
| - [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols) | ||||
| - [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols) | ||||
| - [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes) | ||||
| - Preserves threading when possible | ||||
| - [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling) | ||||
| - [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing) | ||||
| - [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups) | ||||
| - [API](https://github.com/42wim/matterbridge/wiki/Features#api) | ||||
|  | ||||
| ### Natively supported | ||||
|  | ||||
| * [Mattermost](https://github.com/mattermost/mattermost-server/) 4.x, 5.x | ||||
| * [IRC](http://www.mirc.com/servers.html) | ||||
| * [XMPP](https://xmpp.org) | ||||
| * [Gitter](https://gitter.im) | ||||
| * [Slack](https://slack.com) | ||||
| * [Discord](https://discordapp.com) | ||||
| * [Telegram](https://telegram.org) | ||||
| * [Hipchat](https://www.hipchat.com) | ||||
| * [Rocket.chat](https://rocket.chat) | ||||
| * [Matrix](https://matrix.org) | ||||
| * [Steam](https://store.steampowered.com/) | ||||
| * [Twitch](https://twitch.tv) | ||||
| * [Ssh-chat](https://github.com/shazow/ssh-chat) | ||||
| * [WhatsApp](https://www.whatsapp.com/) | ||||
| * [Zulip](https://zulipchat.com) | ||||
| - [Discord](https://discordapp.com) | ||||
| - [Gitter](https://gitter.im) | ||||
| - [Harmony](https://harmonyapp.io) | ||||
| - [IRC](http://www.mirc.com/servers.html) | ||||
| - [Keybase](https://keybase.io) | ||||
| - [Matrix](https://matrix.org) | ||||
| - [Mattermost](https://github.com/mattermost/mattermost-server/) | ||||
| - [Microsoft Teams](https://teams.microsoft.com) | ||||
| - [Mumble](https://www.mumble.info/) | ||||
| - [Nextcloud Talk](https://nextcloud.com/talk/) | ||||
| - [Rocket.chat](https://rocket.chat) | ||||
| - [Slack](https://slack.com) | ||||
| - [Ssh-chat](https://github.com/shazow/ssh-chat) | ||||
| - ~~[Steam](https://store.steampowered.com/)~~ | ||||
|   - Not supported anymore, see [here](https://github.com/Philipp15b/go-steam/issues/94) for more info. | ||||
| - [Telegram](https://telegram.org) | ||||
| - [Twitch](https://twitch.tv) | ||||
| - [VK](https://vk.com/) | ||||
| - [WhatsApp](https://www.whatsapp.com/) | ||||
|   - Whatsapp legacy is natively supported | ||||
|   - Whatsapp multidevice beta is natively supported but you need to build yourself, see [here](#building-with-whatsapp-beta-multidevice-support) | ||||
| - [XMPP](https://xmpp.org) | ||||
| - [Zulip](https://zulipchat.com) | ||||
|  | ||||
| ### 3rd party via matterbridge api | ||||
| * [Minecraft](https://github.com/elytra/MatterLink) | ||||
| * [Reddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| * [Facebook messenger](https://github.com/VictorNine/fbridge) | ||||
|  | ||||
| - [Delta Chat](https://github.com/deltachat-bot/matterdelta) | ||||
| - [Minecraft](https://github.com/raws/mattercraft) | ||||
| - [Minecraft](https://gitlab.com/Programie/MatterBukkit) | ||||
|  | ||||
| #### Past 3rd party projects | ||||
| - [Discourse](https://github.com/DeclanHoare/matterbabble) | ||||
| - [Facebook messenger](https://github.com/powerjungle/fbridge-asyncio) | ||||
| - [Facebook messenger](https://github.com/VictorNine/fbridge) | ||||
| - [Minecraft](https://github.com/elytra/MatterLink) | ||||
| - [Reddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| - [MatterAMXX](https://github.com/andrewlindberg/MatterAMXX): [Counter-Strike, half-life and more](https://forums.alliedmods.net/showthread.php?t=319430) | ||||
| - [Vintage Story](https://github.com/NikkyAI/vs-matterbridge) | ||||
| - [Ultima Online Emulator](https://github.com/kuoushi/ServUO-Matterbridge) | ||||
| - [Teamspeak](https://github.com/Archeb/ts-matterbridge) | ||||
|  | ||||
|  | ||||
| ### API | ||||
| The API is very basic at the moment.    | ||||
|  | ||||
| The API is basic at the moment. | ||||
| More info and examples on the [wiki](https://github.com/42wim/matterbridge/wiki/Api). | ||||
|  | ||||
| Used by the projects below. 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) | ||||
| * [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
| - [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Forge server chat, archived) | ||||
| - [MatterCraft](https://github.com/raws/mattercraft) (Matterbridge link for Minecraft Forge server chat) | ||||
| - [MatterBukkit](https://gitlab.com/Programie/MatterBukkit) (Matterbridge link for Minecraft Bukkit/Spigot server chat) | ||||
| - [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| - [Mattereddit](https://github.com/bonehurtingjuice/mattereddit) (Reddit chat support) | ||||
| - [fbridge-asyncio](https://github.com/powerjungle/fbridge-asyncio) (Facebook messenger support) | ||||
| - [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
| - [matterbabble](https://github.com/DeclanHoare/matterbabble) (Discourse support) | ||||
| - [MatterAMXX](https://forums.alliedmods.net/showthread.php?t=319430) (Counter-Strike, half-life and more via AMXX mod) | ||||
| - [Vintage Story](https://github.com/NikkyAI/vs-matterbridge) | ||||
| - [ServUO-matterbridge](https://github.com/kuoushi/ServUO-Matterbridge) (A matterbridge connector for ServUO servers) | ||||
| - [ts-matterbridge](https://github.com/Archeb/ts-matterbridge) (Integrate teamspeak chat with matterbridge) | ||||
| - [beerchat](https://github.com/mt-mods/beerchat) (Matterbridge link for minetest) | ||||
|  | ||||
| ## Chat with us | ||||
|  | ||||
| Questions or want to test on your favorite platform? Join below: | ||||
|  | ||||
| * [Gitter][mb-gitter] | ||||
| * [IRC][mb-irc] | ||||
| * [Discord][mb-discord] | ||||
| * [Matrix][mb-matrix] | ||||
| * [Slack][mb-slack] | ||||
| * [Mattermost][mb-mattermost] | ||||
| * [Rocket.Chat][mb-rocketchat] | ||||
| * [XMPP][mb-xmpp] | ||||
| * [Twitch][mb-twitch] | ||||
| * [Zulip][mb-zulip] | ||||
| * [Telegram][mb-telegram] | ||||
| - [Discord][mb-discord] | ||||
| - [Gitter][mb-gitter] | ||||
| - [IRC][mb-irc] | ||||
| - [Keybase][mb-keybase] | ||||
| - [Matrix][mb-matrix] | ||||
| - [Mattermost][mb-mattermost] | ||||
| - [Rocket.Chat][mb-rocketchat] | ||||
| - [Slack][mb-slack] | ||||
| - [Telegram][mb-telegram] | ||||
| - [Twitch][mb-twitch] | ||||
| - [XMPP][mb-xmpp] (matterbridge@conference.jabber.de) | ||||
| - [Zulip][mb-zulip] | ||||
|  | ||||
| ## Screenshots | ||||
| See https://github.com/42wim/matterbridge/wiki | ||||
|  | ||||
| ## Installing | ||||
| See <https://github.com/42wim/matterbridge/wiki> | ||||
|  | ||||
| ## Installing / upgrading | ||||
|  | ||||
| ### Binaries | ||||
| * Latest stable release [v1.13.1](https://github.com/42wim/matterbridge/releases/latest) | ||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/) | ||||
|  | ||||
| - Latest stable release [v1.26.0](https://github.com/42wim/matterbridge/releases/latest) | ||||
| - Development releases (follows master) can be downloaded [here](https://github.com/42wim/matterbridge/actions) selecting the latest green build and then artifacts. | ||||
|  | ||||
| To install or upgrade just download the latest [binary](https://github.com/42wim/matterbridge/releases/latest). On \*nix platforms you may need to make the binary executable - you can do this by running `chmod a+x` on the binary (example: `chmod a+x matterbridge-1.24.1-linux-64bit`). After downloading (and making the binary executable, if necessary), follow the instructions on the [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||
|  | ||||
| ### Packages | ||||
| * [Overview](https://repology.org/metapackage/matterbridge/versions) | ||||
|  | ||||
| ### Building | ||||
| Go 1.9+ 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). | ||||
| - [Overview](https://repology.org/metapackage/matterbridge/versions) | ||||
| - [snap](https://snapcraft.io/matterbridge) | ||||
| - [scoop](https://github.com/42wim/scoop-bucket) | ||||
|  | ||||
| After Go is setup, download matterbridge to your $GOPATH directory. | ||||
| ## Building | ||||
|  | ||||
| ``` | ||||
| cd $GOPATH | ||||
| go get github.com/42wim/matterbridge | ||||
| Most people just want to use binaries, you can find those [here](https://github.com/42wim/matterbridge/releases/latest) | ||||
|  | ||||
| If you really want to build from source, follow these instructions: | ||||
| Go 1.18+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed. | ||||
|  | ||||
| Building the binary with **all** the bridges enabled needs about 3GB RAM to compile. | ||||
| You can reduce this memory requirement to 0,5GB RAM by adding the `nomsteams` tag if you don't need/use the Microsoft Teams bridge. | ||||
|  | ||||
| Matterbridge can be build without gcc/c-compiler: If you're running on windows first run `set CGO_ENABLED=0` on other platforms you prepend `CGO_ENABLED=0` to the `go build` command. (eg `CGO_ENABLED=0 go install github.com/42wim/matterbridge`) | ||||
|  | ||||
| To install the latest stable run: | ||||
|  | ||||
| ```bash | ||||
| go install github.com/42wim/matterbridge | ||||
| ``` | ||||
|  | ||||
| You should now have matterbridge binary in the bin directory: | ||||
| To install the latest dev run: | ||||
|  | ||||
| ```bash | ||||
| go install github.com/42wim/matterbridge@master | ||||
| ``` | ||||
| $ ls bin/ | ||||
|  | ||||
| To install the latest stable run without msteams or zulip bridge: | ||||
|  | ||||
| ```bash | ||||
| go install -tags nomsteams,nozulip github.com/42wim/matterbridge | ||||
| ``` | ||||
|  | ||||
| You should now have matterbridge binary in the ~/go/bin directory: | ||||
|  | ||||
| ```bash | ||||
| $ ls ~/go/bin/ | ||||
| matterbridge | ||||
| ``` | ||||
|  | ||||
| ## Building with whatsapp (beta) multidevice support | ||||
|  | ||||
| Because the library we use for Whatsapp multidevice support includes a GPL3 library we can not provide you binaries. | ||||
| (as this would require the Matterbridge to change it license to GPL) | ||||
|  | ||||
| Matterbridge can be build without gcc/c-compiler: If you're running on windows first run `set CGO_ENABLED=0` on other platforms you prepend `CGO_ENABLED=0` to the `go build` command. (eg `CGO_ENABLED=0 go install github.com/42wim/matterbridge`) | ||||
|  | ||||
| So this means you have to build it yourself using the instructions below: | ||||
|  | ||||
| ```bash | ||||
| go install -tags whatsappmulti github.com/42wim/matterbridge@master | ||||
| ``` | ||||
|  | ||||
| If you're low on memory and don't need msteams: | ||||
|  | ||||
| ```bash | ||||
| go install -tags nomsteams,whatsappmulti github.com/42wim/matterbridge@master | ||||
| ``` | ||||
|  | ||||
| You should now have matterbridge binary in the ~/go/bin directory: | ||||
|  | ||||
| ```bash | ||||
| $ ls ~/go/bin/ | ||||
| matterbridge | ||||
| ``` | ||||
|  | ||||
| ## 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. | ||||
|  | ||||
| ### Settings | ||||
|  | ||||
| All possible [settings](https://github.com/42wim/matterbridge/wiki/Settings) for each bridge. | ||||
|  | ||||
| ### 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 | ||||
|  | ||||
| #### Bridge mattermost (off-topic) - irc (#testing) | ||||
|  | ||||
| ```toml | ||||
| [irc] | ||||
|     [irc.freenode] | ||||
|     Server="irc.freenode.net:6667" | ||||
|     [irc.libera] | ||||
|     Server="irc.libera.chat:6667" | ||||
|     Nick="yourbotname" | ||||
|  | ||||
| [mattermost] | ||||
| @@ -172,7 +287,7 @@ See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config | ||||
| name="mygateway" | ||||
| enable=true | ||||
|     [[gateway.inout]] | ||||
|     account="irc.freenode" | ||||
|     account="irc.libera" | ||||
|     channel="#testing" | ||||
|  | ||||
|     [[gateway.inout]] | ||||
| @@ -181,6 +296,7 @@ enable=true | ||||
| ``` | ||||
|  | ||||
| #### Bridge slack (#general) - discord (general) | ||||
|  | ||||
| ```toml | ||||
| [slack] | ||||
| [slack.test] | ||||
| @@ -212,7 +328,7 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | ||||
|  | ||||
| See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||
|  | ||||
| ``` | ||||
| ```bash | ||||
| Usage of ./matterbridge: | ||||
|   -conf string | ||||
|         config file (default "matterbridge.toml") | ||||
| @@ -225,12 +341,15 @@ Usage of ./matterbridge: | ||||
| ``` | ||||
|  | ||||
| ### Docker | ||||
| Create your matterbridge.toml file locally eg in `/tmp/matterbridge.toml` | ||||
| ``` | ||||
| docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge | ||||
| ``` | ||||
|  | ||||
| Please take a look at the [Docker Wiki page](https://github.com/42wim/matterbridge/wiki/Deploy:-Docker) for more information. | ||||
|  | ||||
| ### Systemd | ||||
|  | ||||
| Please take a look at the [Service Files page](https://github.com/42wim/matterbridge/wiki/Service-files) for more information. | ||||
|  | ||||
| ## Changelog | ||||
|  | ||||
| See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md) | ||||
|  | ||||
| ## FAQ | ||||
| @@ -238,59 +357,88 @@ See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.m | ||||
| See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | ||||
|  | ||||
| ## Related projects | ||||
| * [FOSSRIT/infrastructure - roles/matterbridge](https://github.com/FOSSRIT/infrastructure/tree/master/roles/matterbridge) (Ansible role used to automate deployments of Matterbridge) | ||||
| * [matterbridge autoconfig](https://github.com/patcon/matterbridge-autoconfig) | ||||
| * [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||
| * [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||
| * [mattereddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| * [matterlink](https://github.com/elytra/MatterLink) | ||||
| * [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| * [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
|  | ||||
| ## Articles | ||||
| * [matterbridge on kubernetes](https://medium.freecodecamp.org/using-kubernetes-to-deploy-a-chat-gateway-or-when-technology-works-like-its-supposed-to-a169a8cd69a3) | ||||
| * 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 | ||||
| * https://daniele.tech/2019/02/how-to-use-matterbridge-to-connect-2-different-slack-workspaces/ | ||||
| - [jwflory/ansible-role-matterbridge](https://galaxy.ansible.com/jwflory/matterbridge) (Ansible role to simplify deploying Matterbridge) | ||||
| - [matterbridge autoconfig](https://github.com/patcon/matterbridge-autoconfig) | ||||
| - [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||
| - [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||
| - [mattereddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| - [matterlink](https://github.com/elytra/MatterLink) | ||||
| - [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost | ||||
| - [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| - [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
| - [isla](https://github.com/alphachung/isla) (Bot for Discord-Telegram groups used alongside matterbridge) | ||||
| - [matterbabble](https://github.com/DeclanHoare/matterbabble) (Connect Discourse threads to Matterbridge) | ||||
| - [nextcloud talk](https://github.com/nextcloud/talk_matterbridge) (Integrates matterbridge in Nextcloud Talk) | ||||
| - [mattercraft](https://github.com/raws/mattercraft) (Minecraft bridge) | ||||
| - [vs-matterbridge](https://github.com/NikkyAI/vs-matterbridge) (Vintage Story bridge) | ||||
| - [ServUO-matterbridge](https://github.com/kuoushi/ServUO-Matterbridge) (A matterbridge connector for ServUO servers) | ||||
| - [ts-matterbridge](https://github.com/Archeb/ts-matterbridge) (Integrate teamspeak chat with matterbridge) | ||||
|  | ||||
| ## Articles / Tutorials | ||||
|  | ||||
| - [matterbridge on kubernetes](https://medium.freecodecamp.org/using-kubernetes-to-deploy-a-chat-gateway-or-when-technology-works-like-its-supposed-to-a169a8cd69a3) | ||||
| - <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/> | ||||
| - <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> | ||||
| - <https://daniele.tech/2019/02/how-to-use-matterbridge-to-connect-2-different-slack-workspaces/> | ||||
| - <https://userlinux.net/mattermost-and-matterbridge.html> | ||||
| - <https://nextcloud.com/blog/bridging-chat-services-in-talk/> | ||||
| - <https://minecraftchest1.wordpress.com/2021/06/05/how-to-install-and-setup-matterbridge/> | ||||
| - Youtube: [whatsapp - telegram bridging](https://www.youtube.com/watch?v=W-VXISoKtNc) | ||||
|  | ||||
| ## Thanks | ||||
| [](https://www.digitalocean.com/) for sponsoring demo/testing droplets. | ||||
|  | ||||
| <p>This project is supported by:</p> | ||||
| <p> | ||||
|   <a href="https://www.digitalocean.com/"> | ||||
|     <img src="https://opensource.nyc3.cdn.digitaloceanspaces.com/attribution/assets/SVG/DO_Logo_horizontal_blue.svg" width="201px"> | ||||
|   </a> | ||||
| </p> | ||||
|  | ||||
| Matterbridge wouldn't exist without these libraries: | ||||
| * discord - https://github.com/bwmarrin/discordgo | ||||
| * echo - https://github.com/labstack/echo | ||||
| * gitter - https://github.com/sromku/go-gitter | ||||
| * gops - https://github.com/google/gops | ||||
| * gozulipbot - https://github.com/ifo/gozulipbot | ||||
| * irc - https://github.com/lrstanley/girc | ||||
| * mattermost - https://github.com/mattermost/mattermost-server | ||||
| * matrix - https://github.com/matrix-org/gomatrix | ||||
| * sshchat - https://github.com/shazow/ssh-chat | ||||
| * slack - https://github.com/nlopes/slack | ||||
| * steam - https://github.com/Philipp15b/go-steam | ||||
| * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | ||||
| * xmpp - https://github.com/mattn/go-xmpp | ||||
| * whatsapp - https://github.com/Rhymen/go-whatsapp/ | ||||
| * zulip - https://github.com/ifo/gozulipbot | ||||
| * tengo - https://github.com/d5/tengo | ||||
|  | ||||
| - discord - <https://github.com/bwmarrin/discordgo> | ||||
| - echo - <https://github.com/labstack/echo> | ||||
| - gops - <https://github.com/google/gops> | ||||
| - gozulipbot - <https://github.com/ifo/gozulipbot> | ||||
| - gumble - <https://github.com/layeh/gumble> | ||||
| - harmony - <https://github.com/harmony-development/shibshib> | ||||
| - irc - <https://github.com/lrstanley/girc> | ||||
| - keybase - <https://github.com/keybase/go-keybase-chat-bot> | ||||
| - matrix - <https://github.com/matrix-org/gomatrix> | ||||
| - mattermost - <https://github.com/mattermost/mattermost-server> | ||||
| - msgraph.go - <https://github.com/yaegashi/msgraph.go> | ||||
| - mumble - <https://github.com/layeh/gumble> | ||||
| - nctalk - <https://github.com/gary-kim/go-nc-talk> | ||||
| - rocketchat - <https://github.com/RocketChat/Rocket.Chat.Go.SDK> | ||||
| - slack - <https://github.com/nlopes/slack> | ||||
| - sshchat - <https://github.com/shazow/ssh-chat> | ||||
| - steam - <https://github.com/Philipp15b/go-steam> | ||||
| - telegram - <https://github.com/go-telegram-bot-api/telegram-bot-api> | ||||
| - tengo - <https://github.com/d5/tengo> | ||||
| - vk - <https://github.com/SevereCloud/vksdk> | ||||
| - whatsapp - <https://github.com/Rhymen/go-whatsapp> | ||||
| - whatsapp - <https://github.com/tulir/whatsmeow> | ||||
| - 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-rocketchat]: https://open.rocket.chat/channel/matterbridge | ||||
|    [mb-xmpp]: https://inverse.chat/ | ||||
|    [mb-twitch]: https://www.twitch.tv/matterbridge | ||||
|    [mb-whatsapp]: https://www.whatsapp.com/ | ||||
|    [mb-zulip]: https://matterbridge.zulipchat.com/register/ | ||||
|    [mb-telegram]: https://t.me/Matterbridge | ||||
| [mb-discord]: https://discord.gg/AkKPtrQ | ||||
| [mb-gitter]: https://gitter.im/42wim/matterbridge | ||||
| [mb-irc]: https://web.libera.chat/#matterbridge | ||||
| [mb-keybase]: https://keybase.io/team/matterbridge | ||||
| [mb-matrix]: https://riot.im/app/#/room/#matterbridge:matrix.org | ||||
| [mb-mattermost]: https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e | ||||
| [mb-msteams]: https://teams.microsoft.com/join/hj92x75gd3y7 | ||||
| [mb-rocketchat]: https://open.rocket.chat/channel/matterbridge | ||||
| [mb-slack]: https://join.slack.com/t/matterbridgechat/shared_invite/zt-2ourq2h2-7YvyYBq2WFGC~~zEzA68_Q | ||||
| [mb-telegram]: https://t.me/Matterbridge | ||||
| [mb-twitch]: https://www.twitch.tv/matterbridge | ||||
| [mb-whatsapp]: https://www.whatsapp.com/ | ||||
| [mb-xmpp]: https://inverse.chat/ | ||||
| [mb-zulip]: https://matterbridge.zulipchat.com/register/ | ||||
|   | ||||
| @@ -1,22 +1,28 @@ | ||||
| package api | ||||
|  | ||||
| import ( | ||||
| 	"encoding/base64" | ||||
| 	"encoding/json" | ||||
| 	"net/http" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/olahol/melody" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/labstack/echo/v4" | ||||
| 	"github.com/labstack/echo/v4/middleware" | ||||
| 	"github.com/zfjagann/golang-ring" | ||||
| 	"github.com/mitchellh/mapstructure" | ||||
| 	ring "github.com/zfjagann/golang-ring" | ||||
| ) | ||||
|  | ||||
| type API struct { | ||||
| 	Messages ring.Ring | ||||
| 	sync.RWMutex | ||||
| 	*bridge.Config | ||||
| 	mrouter *melody.Melody | ||||
| } | ||||
|  | ||||
| type Message struct { | ||||
| @@ -32,6 +38,32 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	e := echo.New() | ||||
| 	e.HideBanner = true | ||||
| 	e.HidePort = true | ||||
|  | ||||
| 	b.mrouter = melody.New() | ||||
| 	b.mrouter.HandleMessage(func(s *melody.Session, msg []byte) { | ||||
| 		message := config.Message{} | ||||
| 		err := json.Unmarshal(msg, &message) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("failed to decode message from byte[] '%s'", string(msg)) | ||||
| 			return | ||||
| 		} | ||||
| 		b.handleWebsocketMessage(message, s) | ||||
| 	}) | ||||
| 	b.mrouter.HandleConnect(func(session *melody.Session) { | ||||
| 		greet := b.getGreeting() | ||||
| 		data, err := json.Marshal(greet) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("failed to encode message '%v'", greet) | ||||
| 			return | ||||
| 		} | ||||
| 		err = session.Write(data) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("failed to write message '%s'", string(data)) | ||||
| 			return | ||||
| 		} | ||||
| 		// TODO: send message history buffer from `b.Messages` here | ||||
| 	}) | ||||
|  | ||||
| 	b.Messages = ring.Ring{} | ||||
| 	if b.GetInt("Buffer") != 0 { | ||||
| 		b.Messages.SetCapacity(b.GetInt("Buffer")) | ||||
| @@ -41,9 +73,17 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 			return key == b.GetString("Token"), nil | ||||
| 		})) | ||||
| 	} | ||||
|  | ||||
| 	// Set RemoteNickFormat to a sane default | ||||
| 	if !b.IsKeySet("RemoteNickFormat") { | ||||
| 		b.Log.Debugln("RemoteNickFormat is unset, defaulting to \"{NICK}\"") | ||||
| 		b.Config.Config.Viper().Set(b.GetConfigKey("RemoteNickFormat"), "{NICK}") | ||||
| 	} | ||||
|  | ||||
| 	e.GET("/api/health", b.handleHealthcheck) | ||||
| 	e.GET("/api/messages", b.handleMessages) | ||||
| 	e.GET("/api/stream", b.handleStream) | ||||
| 	e.GET("/api/websocket", b.handleWebsocket) | ||||
| 	e.POST("/api/message", b.handlePostMessage) | ||||
| 	go func() { | ||||
| 		if b.GetString("BindAddress") == "" { | ||||
| @@ -58,13 +98,13 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| func (b *API) Connect() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *API) Disconnect() error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *API) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *API) Send(msg config.Message) (string, error) { | ||||
| @@ -74,7 +114,14 @@ func (b *API) Send(msg config.Message) (string, error) { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	b.Messages.Enqueue(&msg) | ||||
| 	b.Log.Debugf("enqueueing message from %s on ring buffer", msg.Username) | ||||
| 	b.Messages.Enqueue(msg) | ||||
|  | ||||
| 	data, err := json.Marshal(msg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("failed to encode message  '%s'", msg) | ||||
| 	} | ||||
| 	_ = b.mrouter.Broadcast(data) | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| @@ -93,6 +140,36 @@ func (b *API) handlePostMessage(c echo.Context) error { | ||||
| 	message.Account = b.Account | ||||
| 	message.ID = "" | ||||
| 	message.Timestamp = time.Now() | ||||
|  | ||||
| 	var ( | ||||
| 		fm map[string]interface{} | ||||
| 		ds string | ||||
| 		ok bool | ||||
| 	) | ||||
|  | ||||
| 	for i, f := range message.Extra["file"] { | ||||
| 		fi := config.FileInfo{} | ||||
| 		if fm, ok = f.(map[string]interface{}); !ok { | ||||
| 			return echo.NewHTTPError(http.StatusInternalServerError, "invalid format for extra") | ||||
| 		} | ||||
| 		err := mapstructure.Decode(fm, &fi) | ||||
| 		if err != nil { | ||||
| 			if !strings.Contains(err.Error(), "got string") { | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
| 		// mapstructure doesn't decode base64 into []byte, so it must be done manually for fi.Data | ||||
| 		if ds, ok = fm["Data"].(string); !ok { | ||||
| 			return echo.NewHTTPError(http.StatusInternalServerError, "invalid format for data") | ||||
| 		} | ||||
|  | ||||
| 		data, err := base64.StdEncoding.DecodeString(ds) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		fi.Data = &data | ||||
| 		message.Extra["file"][i] = fi | ||||
| 	} | ||||
| 	b.Log.Debugf("Sending message from %s on %s to gateway", message.Username, "api") | ||||
| 	b.Remote <- message | ||||
| 	return c.JSON(http.StatusOK, message) | ||||
| @@ -106,25 +183,63 @@ func (b *API) handleMessages(c echo.Context) error { | ||||
| 	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{ | ||||
| func (b *API) getGreeting() config.Message { | ||||
| 	return config.Message{ | ||||
| 		Event:     config.EventAPIConnected, | ||||
| 		Timestamp: time.Now(), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *API) handleStream(c echo.Context) error { | ||||
| 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON) | ||||
| 	c.Response().WriteHeader(http.StatusOK) | ||||
| 	greet := b.getGreeting() | ||||
| 	if err := json.NewEncoder(c.Response()).Encode(greet); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	c.Response().Flush() | ||||
| 	for { | ||||
| 		msg := b.Messages.Dequeue() | ||||
| 		if msg != nil { | ||||
| 			if err := json.NewEncoder(c.Response()).Encode(msg); err != nil { | ||||
| 				return err | ||||
| 		select { | ||||
| 		// TODO: this causes issues, messages should be broadcasted to all connected clients | ||||
| 		default: | ||||
| 			msg := b.Messages.Dequeue() | ||||
| 			if msg != nil { | ||||
| 				if err := json.NewEncoder(c.Response()).Encode(msg); err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 				c.Response().Flush() | ||||
| 			} | ||||
| 			c.Response().Flush() | ||||
| 			time.Sleep(100 * time.Millisecond) | ||||
| 		case <-c.Request().Context().Done(): | ||||
| 			return nil | ||||
| 		} | ||||
| 		time.Sleep(200 * time.Millisecond) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *API) handleWebsocketMessage(message config.Message, s *melody.Session) { | ||||
| 	message.Channel = "api" | ||||
| 	message.Protocol = "api" | ||||
| 	message.Account = b.Account | ||||
| 	message.ID = "" | ||||
| 	message.Timestamp = time.Now() | ||||
|  | ||||
| 	data, err := json.Marshal(message) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("failed to encode message for loopback '%v'", message) | ||||
| 		return | ||||
| 	} | ||||
| 	_ = b.mrouter.BroadcastOthers(data, s) | ||||
|  | ||||
| 	b.Log.Debugf("Sending websocket message from %s on %s to gateway", message.Username, "api") | ||||
| 	b.Remote <- message | ||||
| } | ||||
|  | ||||
| func (b *API) handleWebsocket(c echo.Context) error { | ||||
| 	err := b.mrouter.HandleRequest(c.Response(), c.Request()) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("error in websocket handling  '%v'", err) | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|   | ||||
| @@ -1,8 +1,10 @@ | ||||
| package bridge | ||||
|  | ||||
| import ( | ||||
| 	"log" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| @@ -41,6 +43,10 @@ type Factory func(*Config) Bridger | ||||
|  | ||||
| func New(bridge *config.Bridge) *Bridge { | ||||
| 	accInfo := strings.Split(bridge.Account, ".") | ||||
| 	if len(accInfo) != 2 { | ||||
| 		log.Fatalf("config failure, account incorrect: %s", bridge.Account) | ||||
| 	} | ||||
|  | ||||
| 	protocol := accInfo[0] | ||||
| 	name := accInfo[1] | ||||
|  | ||||
| @@ -69,6 +75,7 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map | ||||
| 	for ID, channel := range channels { | ||||
| 		if !exists[ID] { | ||||
| 			b.Log.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID) | ||||
| 			time.Sleep(time.Duration(b.GetInt("JoinDelay")) * time.Millisecond) | ||||
| 			err := b.JoinChannel(channel) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| @@ -79,8 +86,16 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetConfigKey(key string) string { | ||||
| 	return b.Account + "." + key | ||||
| } | ||||
|  | ||||
| func (b *Bridge) IsKeySet(key string) bool { | ||||
| 	return b.Config.IsKeySet(b.GetConfigKey(key)) || b.Config.IsKeySet("general."+key) | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetBool(key string) bool { | ||||
| 	val, ok := b.Config.GetBool(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetBool(b.GetConfigKey(key)) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetBool("general." + key) | ||||
| 	} | ||||
| @@ -88,7 +103,7 @@ func (b *Bridge) GetBool(key string) bool { | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetInt(key string) int { | ||||
| 	val, ok := b.Config.GetInt(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetInt(b.GetConfigKey(key)) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetInt("general." + key) | ||||
| 	} | ||||
| @@ -96,7 +111,7 @@ func (b *Bridge) GetInt(key string) int { | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetString(key string) string { | ||||
| 	val, ok := b.Config.GetString(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetString(b.GetConfigKey(key)) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetString("general." + key) | ||||
| 	} | ||||
| @@ -104,7 +119,7 @@ func (b *Bridge) GetString(key string) string { | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetStringSlice(key string) []string { | ||||
| 	val, ok := b.Config.GetStringSlice(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetStringSlice(b.GetConfigKey(key)) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetStringSlice("general." + key) | ||||
| 	} | ||||
| @@ -112,7 +127,7 @@ func (b *Bridge) GetStringSlice(key string) []string { | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetStringSlice2D(key string) [][]string { | ||||
| 	val, ok := b.Config.GetStringSlice2D(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetStringSlice2D(b.GetConfigKey(key)) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetStringSlice2D("general." + key) | ||||
| 	} | ||||
|   | ||||
| @@ -3,6 +3,8 @@ package config | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"io/ioutil" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
| @@ -21,11 +23,15 @@ const ( | ||||
| 	EventRejoinChannels    = "rejoin_channels" | ||||
| 	EventUserAction        = "user_action" | ||||
| 	EventMsgDelete         = "msg_delete" | ||||
| 	EventFileDelete        = "file_delete" | ||||
| 	EventAPIConnected      = "api_connected" | ||||
| 	EventUserTyping        = "user_typing" | ||||
| 	EventGetChannelMembers = "get_channel_members" | ||||
| 	EventNoticeIRC         = "notice_irc" | ||||
| ) | ||||
|  | ||||
| const ParentIDNotFound = "msg-parent-not-found" | ||||
|  | ||||
| type Message struct { | ||||
| 	Text      string    `json:"text"` | ||||
| 	Channel   string    `json:"channel"` | ||||
| @@ -42,14 +48,23 @@ type Message struct { | ||||
| 	Extra     map[string][]interface{} | ||||
| } | ||||
|  | ||||
| func (m Message) ParentNotFound() bool { | ||||
| 	return m.ParentID == ParentIDNotFound | ||||
| } | ||||
|  | ||||
| func (m Message) ParentValid() bool { | ||||
| 	return m.ParentID != "" && !m.ParentNotFound() | ||||
| } | ||||
|  | ||||
| type FileInfo struct { | ||||
| 	Name    string | ||||
| 	Data    *[]byte | ||||
| 	Comment string | ||||
| 	URL     string | ||||
| 	Size    int64 | ||||
| 	Avatar  bool | ||||
| 	SHA     string | ||||
| 	Name     string | ||||
| 	Data     *[]byte | ||||
| 	Comment  string | ||||
| 	URL      string | ||||
| 	Size     int64 | ||||
| 	Avatar   bool | ||||
| 	SHA      string | ||||
| 	NativeID string | ||||
| } | ||||
|  | ||||
| type ChannelInfo struct { | ||||
| @@ -72,34 +87,43 @@ type ChannelMember struct { | ||||
| type ChannelMembers []ChannelMember | ||||
|  | ||||
| type Protocol struct { | ||||
| 	AuthCode               string // steam | ||||
| 	BindAddress            string // mattermost, slack // DEPRECATED | ||||
| 	Buffer                 int    // api | ||||
| 	Charset                string // irc | ||||
| 	ColorNicks             bool   // only irc for now | ||||
| 	Debug                  bool   // general | ||||
| 	DebugLevel             int    // only for irc now | ||||
| 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||
| 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||
| 	IconURL                string // mattermost, slack | ||||
| 	IgnoreFailureOnStart   bool   // general | ||||
| 	IgnoreNicks            string // all protocols | ||||
| 	IgnoreMessages         string // all protocols | ||||
| 	Jid                    string // xmpp | ||||
| 	Label                  string // all protocols | ||||
| 	Login                  string // mattermost, matrix | ||||
| 	AllowMention           []string // discord | ||||
| 	AuthCode               string   // steam | ||||
| 	BindAddress            string   // mattermost, slack // DEPRECATED | ||||
| 	Buffer                 int      // api | ||||
| 	Charset                string   // irc | ||||
| 	ClientID               string   // msteams | ||||
| 	ColorNicks             bool     // only irc for now | ||||
| 	Debug                  bool     // general | ||||
| 	DebugLevel             int      // only for irc now | ||||
| 	DisableWebPagePreview  bool     // telegram | ||||
| 	EditSuffix             string   // mattermost, slack, discord, telegram, gitter | ||||
| 	EditDisable            bool     // mattermost, slack, discord, telegram, gitter | ||||
| 	HTMLDisable            bool     // matrix | ||||
| 	IconURL                string   // mattermost, slack | ||||
| 	IgnoreFailureOnStart   bool     // general | ||||
| 	IgnoreNicks            string   // all protocols | ||||
| 	IgnoreMessages         string   // all protocols | ||||
| 	Jid                    string   // xmpp | ||||
| 	JoinDelay              string   // all protocols | ||||
| 	Label                  string   // all protocols | ||||
| 	Login                  string   // mattermost, matrix | ||||
| 	LogFile                string   // general | ||||
| 	MediaDownloadBlackList []string | ||||
| 	MediaDownloadPath      string // Basically MediaServerUpload, but instead of uploading it, just write it to a file on the same server. | ||||
| 	MediaDownloadSize      int    // all protocols | ||||
| 	MediaServerDownload    string | ||||
| 	MediaServerUpload      string | ||||
| 	MediaConvertTgs        string     // telegram | ||||
| 	MediaConvertWebPToPNG  bool       // telegram | ||||
| 	MessageDelay           int        // IRC, time in millisecond to wait between messages | ||||
| 	MessageFormat          string     // telegram | ||||
| 	MessageLength          int        // IRC, max length of a message allowed | ||||
| 	MessageQueue           int        // IRC, size of message queue for flood control | ||||
| 	MessageSplit           bool       // IRC, split long messages with newlines on MessageLength instead of clipping | ||||
| 	MessageSplitMaxCount   int        // discord, split long messages into at most this many messages instead of clipping (MessageLength=1950 cannot be configured) | ||||
| 	Muc                    string     // xmpp | ||||
| 	MxID                   string     // matrix | ||||
| 	Name                   string     // all protocols | ||||
| 	Nick                   string     // all protocols | ||||
| 	NickFormatter          string     // mattermost, slack | ||||
| @@ -109,38 +133,48 @@ type Protocol struct { | ||||
| 	NicksPerRow            int        // mattermost, slack | ||||
| 	NoHomeServerSuffix     bool       // matrix | ||||
| 	NoSendJoinPart         bool       // all protocols | ||||
| 	NoTLS                  bool       // mattermost | ||||
| 	NoTLS                  bool       // mattermost, xmpp | ||||
| 	Password               string     // IRC,mattermost,XMPP,matrix | ||||
| 	PrefixMessagesWithNick bool       // mattemost, slack | ||||
| 	PreserveThreading      bool       // slack | ||||
| 	Protocol               string     // all protocols | ||||
| 	QuoteDisable           bool       // telegram | ||||
| 	QuoteFormat            string     // telegram | ||||
| 	QuoteLengthLimit       int        // telegram | ||||
| 	RealName               string     // IRC | ||||
| 	RejoinDelay            int        // IRC | ||||
| 	ReplaceMessages        [][]string // all protocols | ||||
| 	ReplaceNicks           [][]string // all protocols | ||||
| 	RemoteNickFormat       string     // all protocols | ||||
| 	RunCommands            []string   // irc | ||||
| 	Server                 string     // IRC,mattermost,XMPP,discord | ||||
| 	RunCommands            []string   // IRC | ||||
| 	Server                 string     // IRC,mattermost,XMPP,discord,matrix | ||||
| 	SessionFile            string     // msteams,whatsapp | ||||
| 	ShowJoinPart           bool       // all protocols | ||||
| 	ShowTopicChange        bool       // slack | ||||
| 	ShowUserTyping         bool       // slack | ||||
| 	ShowEmbeds             bool       // discord | ||||
| 	SkipTLSVerify          bool       // IRC, mattermost | ||||
| 	SkipVersionCheck       bool       // mattermost | ||||
| 	StripNick              bool       // all protocols | ||||
| 	StripMarkdown          bool       // irc | ||||
| 	SyncTopic              bool       // slack | ||||
| 	TengoModifyMessage     string     // general | ||||
| 	Team                   string     // mattermost | ||||
| 	Token                  string     // gitter, slack, discord, api | ||||
| 	Team                   string     // mattermost, keybase | ||||
| 	TeamID                 string     // msteams | ||||
| 	TenantID               string     // msteams | ||||
| 	Token                  string     // gitter, slack, discord, api, matrix | ||||
| 	Topic                  string     // zulip | ||||
| 	URL                    string     // mattermost, slack // DEPRECATED | ||||
| 	UseAPI                 bool       // mattermost, slack | ||||
| 	UseLocalAvatar         []string   // discord | ||||
| 	UseSASL                bool       // IRC | ||||
| 	UseTLS                 bool       // IRC | ||||
| 	UseDiscriminator       bool       // discord | ||||
| 	UseFirstName           bool       // telegram | ||||
| 	UseUserName            bool       // discord | ||||
| 	UseUserName            bool       // discord, matrix, mattermost | ||||
| 	UseInsecureURL         bool       // telegram | ||||
| 	UserName               string     // IRC | ||||
| 	VerboseJoinPart        bool       // IRC | ||||
| 	WebhookBindAddress     string     // mattermost, slack | ||||
| 	WebhookURL             string     // mattermost, slack | ||||
| } | ||||
| @@ -166,6 +200,13 @@ type Gateway struct { | ||||
| 	InOut  []Bridge | ||||
| } | ||||
|  | ||||
| type Tengo struct { | ||||
| 	InMessage        string | ||||
| 	Message          string | ||||
| 	RemoteNickFormat string | ||||
| 	OutMessage       string | ||||
| } | ||||
|  | ||||
| type SameChannelGateway struct { | ||||
| 	Name     string | ||||
| 	Enable   bool | ||||
| @@ -189,13 +230,18 @@ type BridgeValues struct { | ||||
| 	SSHChat            map[string]Protocol | ||||
| 	WhatsApp           map[string]Protocol // TODO is this struct used? Search for "SlackLegacy" for example didn't return any results | ||||
| 	Zulip              map[string]Protocol | ||||
| 	Keybase            map[string]Protocol | ||||
| 	Mumble             map[string]Protocol | ||||
| 	General            Protocol | ||||
| 	Tengo              Tengo | ||||
| 	Gateway            []Gateway | ||||
| 	SameChannelGateway []SameChannelGateway | ||||
| } | ||||
|  | ||||
| type Config interface { | ||||
| 	Viper() *viper.Viper | ||||
| 	BridgeValues() *BridgeValues | ||||
| 	IsKeySet(key string) bool | ||||
| 	GetBool(key string) (bool, bool) | ||||
| 	GetInt(key string) (int, bool) | ||||
| 	GetString(key string) (string, bool) | ||||
| @@ -221,7 +267,17 @@ func NewConfig(rootLogger *logrus.Logger, cfgfile string) Config { | ||||
| 		logger.Fatalf("Failed to read configuration file: %#v", err) | ||||
| 	} | ||||
|  | ||||
| 	mycfg := newConfigFromString(logger, input) | ||||
| 	cfgtype := detectConfigType(cfgfile) | ||||
| 	mycfg := newConfigFromString(logger, input, cfgtype) | ||||
| 	if mycfg.cv.General.LogFile != "" { | ||||
| 		logfile, err := os.OpenFile(mycfg.cv.General.LogFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600) | ||||
| 		if err == nil { | ||||
| 			logger.Info("Opening log file ", mycfg.cv.General.LogFile) | ||||
| 			rootLogger.Out = logfile | ||||
| 		} else { | ||||
| 			logger.Warn("Failed to open ", mycfg.cv.General.LogFile) | ||||
| 		} | ||||
| 	} | ||||
| 	if mycfg.cv.General.MediaDownloadSize == 0 { | ||||
| 		mycfg.cv.General.MediaDownloadSize = 1000000 | ||||
| 	} | ||||
| @@ -232,25 +288,37 @@ func NewConfig(rootLogger *logrus.Logger, cfgfile string) Config { | ||||
| 	return mycfg | ||||
| } | ||||
|  | ||||
| // detectConfigType detects JSON and YAML formats, defaults to TOML. | ||||
| func detectConfigType(cfgfile string) string { | ||||
| 	fileExt := filepath.Ext(cfgfile) | ||||
| 	switch fileExt { | ||||
| 	case ".json": | ||||
| 		return "json" | ||||
| 	case ".yaml", ".yml": | ||||
| 		return "yaml" | ||||
| 	} | ||||
| 	return "toml" | ||||
| } | ||||
|  | ||||
| // NewConfigFromString instantiates a new configuration based on the specified string. | ||||
| func NewConfigFromString(rootLogger *logrus.Logger, input []byte) Config { | ||||
| 	logger := rootLogger.WithFields(logrus.Fields{"prefix": "config"}) | ||||
| 	return newConfigFromString(logger, input) | ||||
| 	return newConfigFromString(logger, input, "toml") | ||||
| } | ||||
|  | ||||
| func newConfigFromString(logger *logrus.Entry, input []byte) *config { | ||||
| 	viper.SetConfigType("toml") | ||||
| func newConfigFromString(logger *logrus.Entry, input []byte, cfgtype string) *config { | ||||
| 	viper.SetConfigType(cfgtype) | ||||
| 	viper.SetEnvPrefix("matterbridge") | ||||
| 	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_")) | ||||
| 	viper.AutomaticEnv() | ||||
|  | ||||
| 	if err := viper.ReadConfig(bytes.NewBuffer(input)); err != nil { | ||||
| 		logger.Fatalf("Failed to parse the configuration: %#v", err) | ||||
| 		logger.Fatalf("Failed to parse the configuration: %s", err) | ||||
| 	} | ||||
|  | ||||
| 	cfg := &BridgeValues{} | ||||
| 	if err := viper.Unmarshal(cfg); err != nil { | ||||
| 		logger.Fatalf("Failed to load the configuration: %#v", err) | ||||
| 		logger.Fatalf("Failed to load the configuration: %s", err) | ||||
| 	} | ||||
| 	return &config{ | ||||
| 		logger: logger, | ||||
| @@ -263,6 +331,16 @@ func (c *config) BridgeValues() *BridgeValues { | ||||
| 	return c.cv | ||||
| } | ||||
|  | ||||
| func (c *config) Viper() *viper.Viper { | ||||
| 	return c.v | ||||
| } | ||||
|  | ||||
| func (c *config) IsKeySet(key string) bool { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	return c.v.IsSet(key) | ||||
| } | ||||
|  | ||||
| func (c *config) GetBool(key string) (bool, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| @@ -322,6 +400,11 @@ type TestConfig struct { | ||||
| 	Overrides map[string]interface{} | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) IsKeySet(key string) bool { | ||||
| 	_, ok := c.Overrides[key] | ||||
| 	return ok || c.Config.IsKeySet(key) | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetBool(key string) (bool, bool) { | ||||
| 	val, ok := c.Overrides[key] | ||||
| 	if ok { | ||||
|   | ||||
| @@ -2,30 +2,31 @@ package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/discord/transmitter" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| 	lru "github.com/hashicorp/golang-lru" | ||||
| ) | ||||
|  | ||||
| const MessageLength = 1950 | ||||
| const ( | ||||
| 	MessageLength = 1950 | ||||
| 	cFileUpload   = "file_upload" | ||||
| ) | ||||
|  | ||||
| type Bdiscord struct { | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	c *discordgo.Session | ||||
|  | ||||
| 	nick            string | ||||
| 	useChannelID    bool | ||||
| 	guildID         string | ||||
| 	webhookID       string | ||||
| 	webhookToken    string | ||||
| 	canEditWebhooks bool | ||||
| 	nick    string | ||||
| 	userID  string | ||||
| 	guildID string | ||||
|  | ||||
| 	channelsMutex  sync.RWMutex | ||||
| 	channels       []*discordgo.Channel | ||||
| @@ -34,30 +35,39 @@ type Bdiscord struct { | ||||
| 	membersMutex  sync.RWMutex | ||||
| 	userMemberMap map[string]*discordgo.Member | ||||
| 	nickMemberMap map[string]*discordgo.Member | ||||
|  | ||||
| 	// Webhook specific logic | ||||
| 	useAutoWebhooks bool | ||||
| 	transmitter     *transmitter.Transmitter | ||||
| 	cache           *lru.Cache | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bdiscord{Config: cfg} | ||||
| 	newCache, err := lru.New(5000) | ||||
| 	if err != nil { | ||||
| 		cfg.Log.Fatalf("Could not create LRU cache: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	b := &Bdiscord{ | ||||
| 		Config: cfg, | ||||
| 		cache:  newCache, | ||||
| 	} | ||||
|  | ||||
| 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||
| 	b.nickMemberMap = make(map[string]*discordgo.Member) | ||||
| 	b.channelInfoMap = make(map[string]*config.ChannelInfo) | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		b.Log.Debug("Configuring Discord Incoming Webhook") | ||||
| 		b.webhookID, b.webhookToken = b.splitURL(b.GetString("WebhookURL")) | ||||
|  | ||||
| 	b.useAutoWebhooks = b.GetBool("AutoWebhooks") | ||||
| 	if b.useAutoWebhooks { | ||||
| 		b.Log.Debug("Using automatic webhooks") | ||||
| 	} | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) Connect() error { | ||||
| 	var err error | ||||
| 	var guildFound bool | ||||
| 	token := b.GetString("Token") | ||||
| 	b.Log.Info("Connecting") | ||||
| 	if b.GetString("WebhookURL") == "" { | ||||
| 		b.Log.Info("Connecting using token") | ||||
| 	} else { | ||||
| 		b.Log.Info("Connecting using webhookurl (for posting) and token") | ||||
| 	} | ||||
| 	if !strings.HasPrefix(b.GetString("Token"), "Bot ") { | ||||
| 		token = "Bot " + b.GetString("Token") | ||||
| 	} | ||||
| @@ -71,17 +81,16 @@ func (b *Bdiscord) Connect() error { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.c.AddHandler(b.messageCreate) | ||||
| 	b.c.AddHandler(b.memberUpdate) | ||||
| 	b.c.AddHandler(b.messageUpdate) | ||||
| 	b.c.AddHandler(b.messageDelete) | ||||
| 	b.c.AddHandler(b.memberAdd) | ||||
| 	b.c.AddHandler(b.memberRemove) | ||||
| 	// Add privileged intent for guild member tracking. This is needed to track nicks | ||||
| 	// for display names and @mention translation | ||||
| 	b.c.Identify.Intents = discordgo.MakeIntent(discordgo.IntentsAllWithoutPrivileged | | ||||
| 		discordgo.IntentsGuildMembers) | ||||
|  | ||||
| 	err = b.c.Open() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	guilds, err := b.c.UserGuilds(100, "", "") | ||||
| 	guilds, err := b.c.UserGuilds(100, "", "", false) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| @@ -91,55 +100,108 @@ func (b *Bdiscord) Connect() error { | ||||
| 	} | ||||
| 	serverName := strings.Replace(b.GetString("Server"), "ID:", "", -1) | ||||
| 	b.nick = userinfo.Username | ||||
| 	b.userID = userinfo.ID | ||||
|  | ||||
| 	// Try and find this account's guild, and populate channels | ||||
| 	b.channelsMutex.Lock() | ||||
| 	for _, guild := range guilds { | ||||
| 		if guild.Name == serverName || guild.ID == serverName { | ||||
| 			b.channels, err = b.c.GuildChannels(guild.ID) | ||||
| 			b.guildID = guild.ID | ||||
| 			guildFound = true | ||||
| 			if err != nil { | ||||
| 				break | ||||
| 			} | ||||
| 		// Skip, if the server name does not match the visible name or the ID | ||||
| 		if guild.Name != serverName && guild.ID != serverName { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// Complain about an ambiguous Server setting. Two Discord servers could have the same title! | ||||
| 		// For IDs, practically this will never happen. It would only trigger if some server's name is also an ID. | ||||
| 		if b.guildID != "" { | ||||
| 			return fmt.Errorf("found multiple Discord servers with the same name %#v, expected to see only one", serverName) | ||||
| 		} | ||||
|  | ||||
| 		// Getting this guild's channel could result in a permission error | ||||
| 		b.channels, err = b.c.GuildChannels(guild.ID) | ||||
| 		if err != nil { | ||||
| 			return fmt.Errorf("could not get %#v's channels: %w", b.GetString("Server"), err) | ||||
| 		} | ||||
|  | ||||
| 		b.guildID = guild.ID | ||||
| 	} | ||||
| 	b.channelsMutex.Unlock() | ||||
| 	if !guildFound { | ||||
| 		msg := fmt.Sprintf("Server \"%s\" not found", b.GetString("Server")) | ||||
| 		err = errors.New(msg) | ||||
| 		b.Log.Error(msg) | ||||
| 		b.Log.Info("Possible values:") | ||||
| 		for _, guild := range guilds { | ||||
| 			b.Log.Infof("Server=\"%s\" # Server name", guild.Name) | ||||
| 			b.Log.Infof("Server=\"%s\" # Server ID", guild.ID) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if err != nil { | ||||
| 	// If we couldn't find a guild, we print extra debug information and return a nice error | ||||
| 	if b.guildID == "" { | ||||
| 		err = fmt.Errorf("could not find Discord server %#v", b.GetString("Server")) | ||||
| 		b.Log.Error(err.Error()) | ||||
|  | ||||
| 		// Print all of the possible server values | ||||
| 		b.Log.Info("Possible server values:") | ||||
| 		for _, guild := range guilds { | ||||
| 			b.Log.Infof("\t- Server=%#v # by name", guild.Name) | ||||
| 			b.Log.Infof("\t- Server=%#v # by ID", guild.ID) | ||||
| 		} | ||||
|  | ||||
| 		// If there are no results, we should say that | ||||
| 		if len(guilds) == 0 { | ||||
| 			b.Log.Info("\t- (none found)") | ||||
| 		} | ||||
|  | ||||
| 		return err | ||||
| 	} | ||||
| 	b.channelsMutex.RLock() | ||||
| 	if b.GetString("WebhookURL") == "" { | ||||
| 		for _, channel := range b.channels { | ||||
| 			b.Log.Debugf("found channel %#v", channel) | ||||
| 		} | ||||
| 	} else { | ||||
| 		b.canEditWebhooks = true | ||||
| 		for _, channel := range b.channels { | ||||
| 			b.Log.Debugf("found channel %#v; verifying PermissionManageWebhooks", channel) | ||||
| 			perms, permsErr := b.c.State.UserChannelPermissions(userinfo.ID, channel.ID) | ||||
| 			manageWebhooks := discordgo.PermissionManageWebhooks | ||||
| 			if permsErr != nil || perms&manageWebhooks != manageWebhooks { | ||||
| 				b.Log.Warnf("Can't manage webhooks in channel \"%s\"", channel.Name) | ||||
| 				b.canEditWebhooks = false | ||||
|  | ||||
| 	// Legacy note: WebhookURL used to have an actual webhook URL that we would edit, | ||||
| 	// but we stopped doing that due to Discord making rate limits more aggressive. | ||||
| 	// | ||||
| 	// Even older: the same WebhookURL used to be used by every channel, which is usually unexpected. | ||||
| 	// This is no longer possible. | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		message := "The global WebhookURL setting has been removed. " | ||||
| 		message += "You can get similar \"webhook editing\" behaviour by replacing this line with `AutoWebhooks=true`. " | ||||
| 		message += "If you rely on the old-OLD (non-editing) behaviour, can move the WebhookURL to specific channel sections." | ||||
| 		b.Log.Errorln(message) | ||||
| 		return fmt.Errorf("use of removed WebhookURL setting") | ||||
| 	} | ||||
|  | ||||
| 	if b.GetInt("debuglevel") == 2 { | ||||
| 		b.Log.Debug("enabling even more discord debug") | ||||
| 		b.c.Debug = true | ||||
| 	} | ||||
|  | ||||
| 	// Initialise webhook management | ||||
| 	b.transmitter = transmitter.New(b.c, b.guildID, "matterbridge", b.useAutoWebhooks) | ||||
| 	b.transmitter.Log = b.Log | ||||
|  | ||||
| 	var webhookChannelIDs []string | ||||
| 	for _, channel := range b.Channels { | ||||
| 		channelID := b.getChannelID(channel.Name) // note(qaisjp): this readlocks channelsMutex | ||||
|  | ||||
| 		// If a WebhookURL was not explicitly provided for this channel, | ||||
| 		// there are two options: just a regular bot message (ugly) or this is should be webhook sent | ||||
| 		if channel.Options.WebhookURL == "" { | ||||
| 			// If it should be webhook sent, we should enforce this via the transmitter | ||||
| 			if b.useAutoWebhooks { | ||||
| 				webhookChannelIDs = append(webhookChannelIDs, channelID) | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
| 		if b.canEditWebhooks { | ||||
| 			b.Log.Info("Can manage webhooks; will edit channel for global webhook on send") | ||||
| 		} else { | ||||
| 			b.Log.Warn("Can't manage webhooks; won't edit channel for global webhook on send") | ||||
|  | ||||
| 		whID, whToken, ok := b.splitURL(channel.Options.WebhookURL) | ||||
| 		if !ok { | ||||
| 			return fmt.Errorf("failed to parse WebhookURL %#v for channel %#v", channel.Options.WebhookURL, channel.ID) | ||||
| 		} | ||||
|  | ||||
| 		b.transmitter.AddWebhook(channelID, &discordgo.Webhook{ | ||||
| 			ID:        whID, | ||||
| 			Token:     whToken, | ||||
| 			GuildID:   b.guildID, | ||||
| 			ChannelID: channelID, | ||||
| 		}) | ||||
| 	} | ||||
|  | ||||
| 	if b.useAutoWebhooks { | ||||
| 		err = b.transmitter.RefreshGuildWebhooks(webhookChannelIDs) | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Println("transmitter could not refresh guild webhooks") | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	// Obtaining guild members and initializing nickname mapping. | ||||
| 	b.membersMutex.Lock() | ||||
| @@ -160,6 +222,19 @@ func (b *Bdiscord) Connect() error { | ||||
| 			b.nickMemberMap[member.Nick] = member | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.c.AddHandler(b.messageCreate) | ||||
| 	b.c.AddHandler(b.messageTyping) | ||||
| 	b.c.AddHandler(b.messageUpdate) | ||||
| 	b.c.AddHandler(b.messageDelete) | ||||
| 	b.c.AddHandler(b.messageDeleteBulk) | ||||
| 	b.c.AddHandler(b.memberAdd) | ||||
| 	b.c.AddHandler(b.memberRemove) | ||||
| 	b.c.AddHandler(b.memberUpdate) | ||||
| 	if b.GetInt("debuglevel") == 1 { | ||||
| 		b.c.AddHandler(b.messageEvent) | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -172,10 +247,6 @@ func (b *Bdiscord) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	defer b.channelsMutex.Unlock() | ||||
|  | ||||
| 	b.channelInfoMap[channel.ID] = &channel | ||||
| 	idcheck := strings.Split(channel.Name, "ID:") | ||||
| 	if len(idcheck) > 1 { | ||||
| 		b.useChannelID = true | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -187,81 +258,35 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 		return "", fmt.Errorf("Could not find channelID for %v", msg.Channel) | ||||
| 	} | ||||
|  | ||||
| 	if msg.Event == config.EventUserTyping { | ||||
| 		if b.GetBool("ShowUserTyping") { | ||||
| 			err := b.c.ChannelTyping(channelID) | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "_" + msg.Text + "_" | ||||
| 	} | ||||
|  | ||||
| 	// use initial webhook configured for the entire Discord account | ||||
| 	isGlobalWebhook := true | ||||
| 	wID := b.webhookID | ||||
| 	wToken := b.webhookToken | ||||
|  | ||||
| 	// check if have a channel specific webhook | ||||
| 	b.channelsMutex.RLock() | ||||
| 	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok { | ||||
| 		if ci.Options.WebhookURL != "" { | ||||
| 			wID, wToken = b.splitURL(ci.Options.WebhookURL) | ||||
| 			isGlobalWebhook = false | ||||
| 		} | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 	} | ||||
| 	b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	// Use webhook to send the message | ||||
| 	if wID != "" { | ||||
| 		// skip events | ||||
| 		if msg.Event != "" && msg.Event != config.EventJoinLeave && msg.Event != config.EventTopicChange { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		b.Log.Debugf("Broadcasting using Webhook") | ||||
| 		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 | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		// 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] | ||||
| 		} | ||||
| 		// if we have a global webhook for this Discord account, and permission | ||||
| 		// to modify webhooks (previously verified), then set its channel to | ||||
| 		// the message channel before using it | ||||
| 		// TODO: this isn't necessary if the last message from this webhook was | ||||
| 		// sent to the current channel | ||||
| 		if isGlobalWebhook && b.canEditWebhooks { | ||||
| 			b.Log.Debugf("Setting webhook channel to \"%s\"", msg.Channel) | ||||
| 			_, err := b.c.WebhookEdit(wID, "", "", channelID) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("Could not set webhook channel: %s", err) | ||||
| 				return "", err | ||||
| 			} | ||||
| 		} | ||||
| 		err := b.c.WebhookExecute( | ||||
| 			wID, | ||||
| 			wToken, | ||||
| 			true, | ||||
| 			&discordgo.WebhookParams{ | ||||
| 				Content:   msg.Text, | ||||
| 				Username:  msg.Username, | ||||
| 				AvatarURL: msg.Avatar, | ||||
| 			}) | ||||
| 		return "", err | ||||
| 	useWebhooks := b.shouldMessageUseWebhooks(&msg) | ||||
| 	if useWebhooks && msg.Event != config.EventMsgDelete && msg.ParentID == "" { | ||||
| 		return b.handleEventWebhook(&msg, channelID) | ||||
| 	} | ||||
|  | ||||
| 	return b.handleEventBotUser(&msg, channelID) | ||||
| } | ||||
|  | ||||
| // handleEventDirect handles events via the bot user | ||||
| func (b *Bdiscord) handleEventBotUser(msg *config.Message, channelID string) (string, error) { | ||||
| 	b.Log.Debugf("Broadcasting using token (API)") | ||||
|  | ||||
| 	// Delete message | ||||
| @@ -273,80 +298,87 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// Delete a file | ||||
| 	if msg.Event == config.EventFileDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		if fi, ok := b.cache.Get(cFileUpload + msg.ID); ok { | ||||
| 			err := b.c.ChannelMessageDelete(channelID, fi.(string)) // nolint:forcetypeassert | ||||
| 			b.cache.Remove(cFileUpload + msg.ID) | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return "", fmt.Errorf("file %s not found", msg.ID) | ||||
| 	} | ||||
|  | ||||
| 	// 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) | ||||
| 		for _, rmsg := range helper.HandleExtra(msg, b.General) { | ||||
| 			// TODO: Use ClipOrSplitMessage | ||||
| 			rmsg.Text = helper.ClipMessage(rmsg.Text, MessageLength, b.GetString("MessageClipped")) | ||||
| 			if _, err := b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text); err != nil { | ||||
| 				b.Log.Errorf("Could not send message %#v: %s", rmsg, err) | ||||
| 			} | ||||
| 		} | ||||
| 		// check if we have files to upload (from slack, telegram or mattermost) | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			return b.handleUploadFile(&msg, channelID) | ||||
| 			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 | ||||
| } | ||||
|  | ||||
| // useWebhook returns true if we have a webhook defined somewhere | ||||
| func (b *Bdiscord) useWebhook() bool { | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, channel := range b.channelInfoMap { | ||||
| 		if channel.Options.WebhookURL != "" { | ||||
| 			return true | ||||
| 		// Exploit that a discord message ID is actually just a large number, and we encode a list of IDs by separating them with ";". | ||||
| 		msgIds := strings.Split(msg.ID, ";") | ||||
| 		msgParts := helper.ClipOrSplitMessage(b.replaceUserMentions(msg.Text), MessageLength, b.GetString("MessageClipped"), len(msgIds)) | ||||
| 		for len(msgParts) < len(msgIds) { | ||||
| 			msgParts = append(msgParts, "((obsoleted by edit))") | ||||
| 		} | ||||
| 	} | ||||
| 	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 | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, channel := range b.channelInfoMap { | ||||
| 		if channel.Options.WebhookURL != "" { | ||||
| 			wID, _ := b.splitURL(channel.Options.WebhookURL) | ||||
| 			if wID == id { | ||||
| 				return true | ||||
| 		for i := range msgParts { | ||||
| 			// In case of split-messages where some parts remain the same (i.e. only a typo-fix in a huge message), this causes some noop-updates. | ||||
| 			// TODO: Optimize away noop-updates of un-edited messages | ||||
| 			// TODO: Use RemoteNickFormat instead of this broken concatenation | ||||
| 			_, err := b.c.ChannelMessageEdit(channelID, msgIds[i], msg.Username+msgParts[i]) | ||||
| 			if err != nil { | ||||
| 				return "", err | ||||
| 			} | ||||
| 		} | ||||
| 		return msg.ID, nil | ||||
| 	} | ||||
| 	return false | ||||
|  | ||||
| 	msgParts := helper.ClipOrSplitMessage(b.replaceUserMentions(msg.Text), MessageLength, b.GetString("MessageClipped"), b.GetInt("MessageSplitMaxCount")) | ||||
| 	msgIds := []string{} | ||||
|  | ||||
| 	for _, msgPart := range msgParts { | ||||
| 		m := discordgo.MessageSend{ | ||||
| 			Content:         msg.Username + msgPart, | ||||
| 			AllowedMentions: b.getAllowedMentions(), | ||||
| 		} | ||||
|  | ||||
| 		if msg.ParentValid() { | ||||
| 			m.Reference = &discordgo.MessageReference{ | ||||
| 				MessageID: msg.ParentID, | ||||
| 				ChannelID: channelID, | ||||
| 				GuildID:   b.guildID, | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Post normal message | ||||
| 		res, err := b.c.ChannelMessageSendComplex(channelID, &m) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		msgIds = append(msgIds, res.ID) | ||||
| 	} | ||||
|  | ||||
| 	// Exploit that a discord message ID is actually just a large number, so we encode a list of IDs by separating them with ";". | ||||
| 	return strings.Join(msgIds, ";"), nil | ||||
| } | ||||
|  | ||||
| // 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{ | ||||
| @@ -355,13 +387,19 @@ func (b *Bdiscord) handleUploadFile(msg *config.Message, channelID string) (stri | ||||
| 			Reader:      bytes.NewReader(*fi.Data), | ||||
| 		} | ||||
| 		m := discordgo.MessageSend{ | ||||
| 			Content: msg.Username + fi.Comment, | ||||
| 			Files:   []*discordgo.File{&file}, | ||||
| 			Content:         msg.Username + fi.Comment, | ||||
| 			Files:           []*discordgo.File{&file}, | ||||
| 			AllowedMentions: b.getAllowedMentions(), | ||||
| 		} | ||||
| 		_, err = b.c.ChannelMessageSendComplex(channelID, &m) | ||||
| 		res, err := b.c.ChannelMessageSendComplex(channelID, &m) | ||||
| 		if err != nil { | ||||
| 			return "", fmt.Errorf("file upload failed: %s", err) | ||||
| 		} | ||||
|  | ||||
| 		// link file_upload_nativeID (file ID from the original bridge) to our upload id | ||||
| 		// so that we can remove this later when it eg needs to be deleted | ||||
| 		b.cache.Add(cFileUpload+fi.NativeID, res.ID) | ||||
| 	} | ||||
|  | ||||
| 	return "", nil | ||||
| } | ||||
|   | ||||
| @@ -3,32 +3,90 @@ package bdiscord | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| 	"github.com/davecgh/go-spew/spew" | ||||
| ) | ||||
|  | ||||
| func (b *Bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { //nolint:unparam | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring messageDelete because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	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 | ||||
| } | ||||
|  | ||||
| // TODO(qaisjp): if other bridges support bulk deletions, it could be fanned out centrally | ||||
| func (b *Bdiscord) messageDeleteBulk(s *discordgo.Session, m *discordgo.MessageDeleteBulk) { //nolint:unparam | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring messageDeleteBulk because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	for _, msgID := range m.Messages { | ||||
| 		rmsg := config.Message{ | ||||
| 			Account: b.Account, | ||||
| 			ID:      msgID, | ||||
| 			Event:   config.EventMsgDelete, | ||||
| 			Text:    config.EventMsgDelete, | ||||
| 			Channel: b.getChannelName(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) messageEvent(s *discordgo.Session, m *discordgo.Event) { | ||||
| 	b.Log.Debug(spew.Sdump(m.Struct)) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageTyping(s *discordgo.Session, m *discordgo.TypingStart) { | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring messageTyping because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	if !b.GetBool("ShowUserTyping") { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Ignore our own typing messages | ||||
| 	if m.UserID == b.userID { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{Account: b.Account, Event: config.EventUserTyping} | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { //nolint:unparam | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring messageUpdate because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	if b.GetBool("EditDisable") { | ||||
| 		return | ||||
| 	} | ||||
| 	// only when message is actually edited | ||||
| 	if m.Message.EditedTimestamp != "" { | ||||
| 	if m.Message.EditedTimestamp != nil { | ||||
| 		b.Log.Debugf("Sending edit message") | ||||
| 		m.Content += b.GetString("EditSuffix") | ||||
| 		b.messageCreate(s, (*discordgo.MessageCreate)(m)) | ||||
| 		msg := &discordgo.MessageCreate{ | ||||
| 			Message: m.Message, | ||||
| 		} | ||||
| 		b.messageCreate(s, msg) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { //nolint:unparam | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring messageCreate because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	var err error | ||||
|  | ||||
| 	// not relay our own messages | ||||
| @@ -36,7 +94,7 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | ||||
| 		return | ||||
| 	} | ||||
| 	// if using webhooks, do not relay if it's ours | ||||
| 	if b.useWebhook() && m.Author.Bot { // && b.isWebhookID(m.Author.ID) { | ||||
| 	if m.Author.Bot && b.transmitter.HasWebhook(m.Author.ID) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| @@ -49,9 +107,9 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | ||||
|  | ||||
| 	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} | ||||
|  | ||||
| 	b.Log.Debugf("== Receiving event %#v", m.Message) | ||||
|  | ||||
| 	if m.Content != "" { | ||||
| 		b.Log.Debugf("== Receiving event %#v", m.Message) | ||||
| 		m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||
| 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | ||||
| 		rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c) | ||||
| 		if err != nil { | ||||
| @@ -62,16 +120,13 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | ||||
|  | ||||
| 	// set channel name | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.useChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
|  | ||||
| 	// set username | ||||
| 	if !b.GetBool("UseUserName") { | ||||
| 		rmsg.Username = b.getNick(m.Author) | ||||
| 	fromWebhook := m.WebhookID != "" | ||||
| 	if !fromWebhook && !b.GetBool("UseUserName") { | ||||
| 		rmsg.Username = b.getNick(m.Author, m.GuildID) | ||||
| 	} else { | ||||
| 		rmsg.Username = m.Author.Username | ||||
| 		if b.GetBool("UseDiscriminator") { | ||||
| 		if !fromWebhook && b.GetBool("UseDiscriminator") { | ||||
| 			rmsg.Username += "#" + m.Author.Discriminator | ||||
| 		} | ||||
| 	} | ||||
| @@ -79,7 +134,7 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | ||||
| 	// if we have embedded content add it to text | ||||
| 	if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil { | ||||
| 		for _, embed := range m.Message.Embeds { | ||||
| 			rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | ||||
| 			rmsg.Text += handleEmbed(embed) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| @@ -95,12 +150,24 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat | ||||
| 		rmsg.Event = config.EventUserAction | ||||
| 	} | ||||
|  | ||||
| 	// Replace emotes | ||||
| 	rmsg.Text = replaceEmotes(rmsg.Text) | ||||
|  | ||||
| 	// Add our parent id if it exists, and if it's not referring to a message in another channel | ||||
| 	if ref := m.MessageReference; ref != nil && ref.ChannelID == m.ChannelID { | ||||
| 		rmsg.ParentID = ref.MessageID | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring memberUpdate because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	if m.Member == nil { | ||||
| 		b.Log.Warnf("Received member update with no member information: %#v", m) | ||||
| 	} | ||||
| @@ -128,6 +195,13 @@ func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUp | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) memberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) { | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring memberAdd because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	if b.GetBool("nosendjoinpart") { | ||||
| 		return | ||||
| 	} | ||||
| 	if m.Member == nil { | ||||
| 		b.Log.Warnf("Received member update with no member information: %#v", m) | ||||
| 		return | ||||
| @@ -149,6 +223,13 @@ func (b *Bdiscord) memberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) memberRemove(s *discordgo.Session, m *discordgo.GuildMemberRemove) { | ||||
| 	if m.GuildID != b.guildID { | ||||
| 		b.Log.Debugf("Ignoring memberRemove because it originates from a different guild") | ||||
| 		return | ||||
| 	} | ||||
| 	if b.GetBool("nosendjoinpart") { | ||||
| 		return | ||||
| 	} | ||||
| 	if m.Member == nil { | ||||
| 		b.Log.Warnf("Received member update with no member information: %#v", m) | ||||
| 		return | ||||
| @@ -168,3 +249,33 @@ func (b *Bdiscord) memberRemove(s *discordgo.Session, m *discordgo.GuildMemberRe | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func handleEmbed(embed *discordgo.MessageEmbed) string { | ||||
| 	var t []string | ||||
| 	var result string | ||||
|  | ||||
| 	t = append(t, embed.Title) | ||||
| 	t = append(t, embed.Description) | ||||
| 	t = append(t, embed.URL) | ||||
|  | ||||
| 	i := 0 | ||||
| 	for _, e := range t { | ||||
| 		if e == "" { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		i++ | ||||
| 		if i == 1 { | ||||
| 			result += " embed: " + e | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		result += " - " + e | ||||
| 	} | ||||
|  | ||||
| 	if result != "" { | ||||
| 		result += "\n" | ||||
| 	} | ||||
|  | ||||
| 	return result | ||||
| } | ||||
|   | ||||
							
								
								
									
										58
									
								
								bridge/discord/handlers_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										58
									
								
								bridge/discord/handlers_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,58 @@ | ||||
| package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
| ) | ||||
|  | ||||
| func TestHandleEmbed(t *testing.T) { | ||||
| 	testcases := map[string]struct { | ||||
| 		embed  *discordgo.MessageEmbed | ||||
| 		result string | ||||
| 	}{ | ||||
| 		"allempty": { | ||||
| 			embed:  &discordgo.MessageEmbed{}, | ||||
| 			result: "", | ||||
| 		}, | ||||
| 		"one": { | ||||
| 			embed: &discordgo.MessageEmbed{ | ||||
| 				Title: "blah", | ||||
| 			}, | ||||
| 			result: " embed: blah\n", | ||||
| 		}, | ||||
| 		"two": { | ||||
| 			embed: &discordgo.MessageEmbed{ | ||||
| 				Title:       "blah", | ||||
| 				Description: "blah2", | ||||
| 			}, | ||||
| 			result: " embed: blah - blah2\n", | ||||
| 		}, | ||||
| 		"three": { | ||||
| 			embed: &discordgo.MessageEmbed{ | ||||
| 				Title:       "blah", | ||||
| 				Description: "blah2", | ||||
| 				URL:         "blah3", | ||||
| 			}, | ||||
| 			result: " embed: blah - blah2 - blah3\n", | ||||
| 		}, | ||||
| 		"twob": { | ||||
| 			embed: &discordgo.MessageEmbed{ | ||||
| 				Description: "blah2", | ||||
| 				URL:         "blah3", | ||||
| 			}, | ||||
| 			result: " embed: blah2 - blah3\n", | ||||
| 		}, | ||||
| 		"oneb": { | ||||
| 			embed: &discordgo.MessageEmbed{ | ||||
| 				URL: "blah3", | ||||
| 			}, | ||||
| 			result: " embed: blah3\n", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for name, tc := range testcases { | ||||
| 		assert.Equalf(t, tc.result, handleEmbed(tc.embed), "Testcases %s", name) | ||||
| 	} | ||||
| } | ||||
| @@ -9,7 +9,31 @@ import ( | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| func (b *Bdiscord) getNick(user *discordgo.User) string { | ||||
| func (b *Bdiscord) getAllowedMentions() *discordgo.MessageAllowedMentions { | ||||
| 	// If AllowMention is not specified, then allow all mentions (default Discord behavior) | ||||
| 	if !b.IsKeySet("AllowMention") { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// Otherwise, allow only the mentions that are specified | ||||
| 	allowedMentionTypes := make([]discordgo.AllowedMentionType, 0, 3) | ||||
| 	for _, m := range b.GetStringSlice("AllowMention") { | ||||
| 		switch m { | ||||
| 		case "everyone": | ||||
| 			allowedMentionTypes = append(allowedMentionTypes, discordgo.AllowedMentionTypeEveryone) | ||||
| 		case "roles": | ||||
| 			allowedMentionTypes = append(allowedMentionTypes, discordgo.AllowedMentionTypeRoles) | ||||
| 		case "users": | ||||
| 			allowedMentionTypes = append(allowedMentionTypes, discordgo.AllowedMentionTypeUsers) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return &discordgo.MessageAllowedMentions{ | ||||
| 		Parse: allowedMentionTypes, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getNick(user *discordgo.User, guildID string) string { | ||||
| 	b.membersMutex.RLock() | ||||
| 	defer b.membersMutex.RUnlock() | ||||
|  | ||||
| @@ -23,9 +47,9 @@ func (b *Bdiscord) getNick(user *discordgo.User) string { | ||||
| 	} | ||||
|  | ||||
| 	// If we didn't find nick, search for it. | ||||
| 	member, err := b.c.GuildMember(b.guildID, user.ID) | ||||
| 	member, err := b.c.GuildMember(guildID, user.ID) | ||||
| 	if err != nil { | ||||
| 		b.Log.Warnf("Failed to fetch information for member %#v: %s", user, err) | ||||
| 		b.Log.Warnf("Failed to fetch information for member %#v on guild %#v: %s", user, guildID, err) | ||||
| 		return user.Username | ||||
| 	} else if member == nil { | ||||
| 		b.Log.Warnf("Got no information for member %#v", user) | ||||
| @@ -44,13 +68,16 @@ func (b *Bdiscord) getGuildMemberByNick(nick string) (*discordgo.Member, error) | ||||
| 	b.membersMutex.RLock() | ||||
| 	defer b.membersMutex.RUnlock() | ||||
|  | ||||
| 	if member, ok := b.nickMemberMap[nick]; ok { | ||||
| 	if member, ok := b.nickMemberMap[strings.TrimSpace(nick)]; ok { | ||||
| 		return member, 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 { | ||||
| 	if strings.Contains(name, "/") { | ||||
| 		return b.getCategoryChannelID(name) | ||||
| 	} | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| @@ -59,40 +86,92 @@ func (b *Bdiscord) getChannelID(name string) string { | ||||
| 		return idcheck[1] | ||||
| 	} | ||||
| 	for _, channel := range b.channels { | ||||
| 		if channel.Name == name { | ||||
| 		if channel.Name == name && channel.Type == discordgo.ChannelTypeGuildText { | ||||
| 			return channel.ID | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getCategoryChannelID(name string) string { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
| 	res := strings.Split(name, "/") | ||||
| 	// shouldn't happen because function should be only called from getChannelID | ||||
| 	if len(res) != 2 { | ||||
| 		return "" | ||||
| 	} | ||||
| 	catName, chanName := res[0], res[1] | ||||
| 	for _, channel := range b.channels { | ||||
| 		// if we have a parentID, lookup the name of that parent (category) | ||||
| 		// and if it matches return it | ||||
| 		if channel.Name == chanName && channel.ParentID != "" { | ||||
| 			for _, cat := range b.channels { | ||||
| 				if cat.ID == channel.ParentID && cat.Name == catName { | ||||
| 					return channel.ID | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelName(id string) string { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, c := range b.channelInfoMap { | ||||
| 		if c.Name == "ID:"+id { | ||||
| 			// if we have ID: specified in our gateway configuration return this | ||||
| 			return c.Name | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	for _, channel := range b.channels { | ||||
| 		if channel.ID == id { | ||||
| 			return channel.Name | ||||
| 			return b.getCategoryChannelName(channel.Name, channel.ParentID) | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getCategoryChannelName(name, parentID string) string { | ||||
| 	var usesCat bool | ||||
| 	// do we have a category configuration in the channel config | ||||
| 	for _, c := range b.channelInfoMap { | ||||
| 		if strings.Contains(c.Name, "/") { | ||||
| 			usesCat = true | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	// configuration without category, return the normal channel name | ||||
| 	if !usesCat { | ||||
| 		return name | ||||
| 	} | ||||
| 	// create a category/channel response | ||||
| 	for _, c := range b.channels { | ||||
| 		if c.ID == parentID { | ||||
| 			name = c.Name + "/" + name | ||||
| 		} | ||||
| 	} | ||||
| 	return name | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	// See https://discordapp.com/developers/docs/reference#message-formatting. | ||||
| 	channelMentionRE = regexp.MustCompile("<#[0-9]+>") | ||||
| 	emojiRE          = regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||
| 	userMentionRE    = regexp.MustCompile("@[^@\n]{1,32}") | ||||
| 	emoteRE          = regexp.MustCompile(`<a?(:\w+:)\d+>`) | ||||
| ) | ||||
|  | ||||
| func (b *Bdiscord) replaceChannelMentions(text string) string { | ||||
| 	replaceChannelMentionFunc := func(match string) string { | ||||
| 		var err error | ||||
| 		channelID := match[2 : len(match)-1] | ||||
|  | ||||
| 		channelName := b.getChannelName(channelID) | ||||
|  | ||||
| 		// If we don't have the channel refresh our list. | ||||
| 		if channelName == "" { | ||||
| 			var err error | ||||
| 			b.channels, err = b.c.GuildChannels(b.guildID) | ||||
| 			if err != nil { | ||||
| 				return "#unknownchannel" | ||||
| @@ -128,19 +207,20 @@ func (b *Bdiscord) replaceUserMentions(text string) string { | ||||
| 	return userMentionRE.ReplaceAllStringFunc(text, replaceUserMentionFunc) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) stripCustomoji(text string) string { | ||||
| 	return emojiRE.ReplaceAllString(text, `$1`) | ||||
| func replaceEmotes(text string) string { | ||||
| 	return emoteRE.ReplaceAllString(text, "$1") | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||
| 		return text[1:], true | ||||
| 	length := len(text) | ||||
| 	if length > 1 && text[0] == '_' && text[length-1] == '_' { | ||||
| 		return text[1 : length-1], true | ||||
| 	} | ||||
| 	return text, false | ||||
| } | ||||
|  | ||||
| // splitURL splits a webhookURL and returns the ID and token. | ||||
| func (b *Bdiscord) splitURL(url string) (string, string) { | ||||
| func (b *Bdiscord) splitURL(url string) (string, string, bool) { | ||||
| 	const ( | ||||
| 		expectedWebhookSplitCount = 7 | ||||
| 		webhookIdxID              = 5 | ||||
| @@ -148,9 +228,9 @@ func (b *Bdiscord) splitURL(url string) (string, string) { | ||||
| 	) | ||||
| 	webhookURLSplit := strings.Split(url, "/") | ||||
| 	if len(webhookURLSplit) != expectedWebhookSplitCount { | ||||
| 		b.Log.Fatalf("%s is no correct discord WebhookURL", url) | ||||
| 		return "", "", false | ||||
| 	} | ||||
| 	return webhookURLSplit[webhookIdxID], webhookURLSplit[webhookIdxToken] | ||||
| 	return webhookURLSplit[webhookIdxID], webhookURLSplit[webhookIdxToken], true | ||||
| } | ||||
|  | ||||
| func enumerateUsernames(s string) []string { | ||||
|   | ||||
							
								
								
									
										257
									
								
								bridge/discord/transmitter/transmitter.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										257
									
								
								bridge/discord/transmitter/transmitter.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,257 @@ | ||||
| // Package transmitter provides functionality for transmitting | ||||
| // arbitrary webhook messages to Discord. | ||||
| // | ||||
| // The package provides the following functionality: | ||||
| // | ||||
| // - Creating new webhooks, whenever necessary | ||||
| // - Loading webhooks that we have previously created | ||||
| // - Sending new messages | ||||
| // - Editing messages, via message ID | ||||
| // - Deleting messages, via message ID | ||||
| // | ||||
| // The package has been designed for matterbridge, but with other | ||||
| // Go bots in mind. The public API should be matterbridge-agnostic. | ||||
| package transmitter | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| // A Transmitter represents a message manager for a single guild. | ||||
| type Transmitter struct { | ||||
| 	session    *discordgo.Session | ||||
| 	guild      string | ||||
| 	title      string | ||||
| 	autoCreate bool | ||||
|  | ||||
| 	// channelWebhooks maps from a channel ID to a webhook instance | ||||
| 	channelWebhooks map[string]*discordgo.Webhook | ||||
|  | ||||
| 	mutex sync.RWMutex | ||||
|  | ||||
| 	Log *log.Entry | ||||
| } | ||||
|  | ||||
| // ErrWebhookNotFound is returned when a valid webhook for this channel/message combination does not exist | ||||
| var ErrWebhookNotFound = errors.New("webhook for this channel and message does not exist") | ||||
|  | ||||
| // ErrPermissionDenied is returned if the bot does not have permission to manage webhooks. | ||||
| // | ||||
| // Bots can be granted a guild-wide permission and channel-specific permissions to manage webhooks. | ||||
| // Despite potentially having guild-wide permission, channel specific overrides could deny a bot's permission to manage webhooks. | ||||
| var ErrPermissionDenied = errors.New("missing 'Manage Webhooks' permission") | ||||
|  | ||||
| // New returns a new Transmitter given a Discord session, guild ID, and title. | ||||
| func New(session *discordgo.Session, guild string, title string, autoCreate bool) *Transmitter { | ||||
| 	return &Transmitter{ | ||||
| 		session:    session, | ||||
| 		guild:      guild, | ||||
| 		title:      title, | ||||
| 		autoCreate: autoCreate, | ||||
|  | ||||
| 		channelWebhooks: make(map[string]*discordgo.Webhook), | ||||
|  | ||||
| 		Log: log.NewEntry(log.StandardLogger()), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Send transmits a message to the given channel with the provided webhook data, and waits until Discord responds with message data. | ||||
| func (t *Transmitter) Send(channelID string, params *discordgo.WebhookParams) (*discordgo.Message, error) { | ||||
| 	wh, err := t.getOrCreateWebhook(channelID) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	msg, err := t.session.WebhookExecute(wh.ID, wh.Token, true, params) | ||||
| 	if err != nil { | ||||
| 		return nil, fmt.Errorf("execute failed: %w", err) | ||||
| 	} | ||||
|  | ||||
| 	return msg, nil | ||||
| } | ||||
|  | ||||
| // Edit will edit a message in a channel, if possible. | ||||
| func (t *Transmitter) Edit(channelID string, messageID string, params *discordgo.WebhookParams) error { | ||||
| 	wh := t.getWebhook(channelID) | ||||
|  | ||||
| 	if wh == nil { | ||||
| 		return ErrWebhookNotFound | ||||
| 	} | ||||
|  | ||||
| 	uri := discordgo.EndpointWebhookToken(wh.ID, wh.Token) + "/messages/" + messageID | ||||
| 	_, err := t.session.RequestWithBucketID("PATCH", uri, params, discordgo.EndpointWebhookToken("", "")) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // HasWebhook checks whether the transmitter is using a particular webhook. | ||||
| func (t *Transmitter) HasWebhook(id string) bool { | ||||
| 	t.mutex.RLock() | ||||
| 	defer t.mutex.RUnlock() | ||||
|  | ||||
| 	for _, wh := range t.channelWebhooks { | ||||
| 		if wh.ID == id { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| // AddWebhook allows you to register a channel's webhook with the transmitter. | ||||
| func (t *Transmitter) AddWebhook(channelID string, webhook *discordgo.Webhook) bool { | ||||
| 	t.Log.Debugf("Manually added webhook %#v to channel %#v", webhook.ID, channelID) | ||||
| 	t.mutex.Lock() | ||||
| 	defer t.mutex.Unlock() | ||||
|  | ||||
| 	_, replaced := t.channelWebhooks[channelID] | ||||
| 	t.channelWebhooks[channelID] = webhook | ||||
| 	return replaced | ||||
| } | ||||
|  | ||||
| // RefreshGuildWebhooks loads "relevant" webhooks into the transmitter, with careful permission handling. | ||||
| // | ||||
| // Notes: | ||||
| // | ||||
| // - A webhook is "relevant" if it was created by this bot -- the ApplicationID should match the bot's ID. | ||||
| // - The term "having permission" means having the "Manage Webhooks" permission. See ErrPermissionDenied for more information. | ||||
| // - This function is additive and will not unload previously loaded webhooks. | ||||
| // - A nil channelIDs slice is treated the same as an empty one. | ||||
| // | ||||
| // If the bot has guild-wide permission: | ||||
| // | ||||
| // 1. it will load any "relevant" webhooks from the entire guild | ||||
| // 2. the given slice is ignored | ||||
| // | ||||
| // If the bot does not have guild-wide permission: | ||||
| // | ||||
| // 1. it will load any "relevant" webhooks in each channel | ||||
| // 2. a single error will be returned if any error occurs (incl. if there is no permission for any of these channels) | ||||
| // | ||||
| // If any channel has more than one "relevant" webhook, it will randomly pick one. | ||||
| func (t *Transmitter) RefreshGuildWebhooks(channelIDs []string) error { | ||||
| 	t.Log.Debugln("Refreshing guild webhooks") | ||||
|  | ||||
| 	botID, err := getDiscordUserID(t.session) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("could not get current user: %w", err) | ||||
| 	} | ||||
|  | ||||
| 	// Get all existing webhooks | ||||
| 	hooks, err := t.session.GuildWebhooks(t.guild) | ||||
| 	if err != nil { | ||||
| 		switch { | ||||
| 		case isDiscordPermissionError(err): | ||||
| 			// We fallback on manually fetching hooks from individual channels | ||||
| 			// if we don't have the "Manage Webhooks" permission globally. | ||||
| 			// We can only do this if we were provided channelIDs, though. | ||||
| 			if len(channelIDs) == 0 { | ||||
| 				return ErrPermissionDenied | ||||
| 			} | ||||
| 			t.Log.Debugln("Missing global 'Manage Webhooks' permission, falling back on per-channel permission") | ||||
| 			return t.fetchChannelsHooks(channelIDs, botID) | ||||
| 		default: | ||||
| 			return fmt.Errorf("could not get webhooks: %w", err) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	t.Log.Debugln("Refreshing guild webhooks using global permission") | ||||
| 	t.assignHooksByAppID(hooks, botID, false) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // createWebhook creates a webhook for a specific channel. | ||||
| func (t *Transmitter) createWebhook(channel string) (*discordgo.Webhook, error) { | ||||
| 	t.mutex.Lock() | ||||
| 	defer t.mutex.Unlock() | ||||
|  | ||||
| 	wh, err := t.session.WebhookCreate(channel, t.title+time.Now().Format(" 3:04:05PM"), "") | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	t.channelWebhooks[channel] = wh | ||||
| 	return wh, nil | ||||
| } | ||||
|  | ||||
| func (t *Transmitter) getWebhook(channel string) *discordgo.Webhook { | ||||
| 	t.mutex.RLock() | ||||
| 	defer t.mutex.RUnlock() | ||||
|  | ||||
| 	return t.channelWebhooks[channel] | ||||
| } | ||||
|  | ||||
| func (t *Transmitter) getOrCreateWebhook(channelID string) (*discordgo.Webhook, error) { | ||||
| 	// If we have a webhook for this channel, immediately return it | ||||
| 	wh := t.getWebhook(channelID) | ||||
| 	if wh != nil { | ||||
| 		return wh, nil | ||||
| 	} | ||||
|  | ||||
| 	// Early exit if we don't want to automatically create one | ||||
| 	if !t.autoCreate { | ||||
| 		return nil, ErrWebhookNotFound | ||||
| 	} | ||||
|  | ||||
| 	t.Log.Infof("Creating a webhook for %s\n", channelID) | ||||
| 	wh, err := t.createWebhook(channelID) | ||||
| 	if err != nil { | ||||
| 		return nil, fmt.Errorf("could not create webhook: %w", err) | ||||
| 	} | ||||
|  | ||||
| 	return wh, nil | ||||
| } | ||||
|  | ||||
| // fetchChannelsHooks fetches hooks for the given channelIDs and calls assignHooksByAppID for each channel's hooks | ||||
| func (t *Transmitter) fetchChannelsHooks(channelIDs []string, botID string) error { | ||||
| 	// For each channel, search for relevant hooks | ||||
| 	var failedHooks []string | ||||
| 	for _, channelID := range channelIDs { | ||||
| 		hooks, err := t.session.ChannelWebhooks(channelID) | ||||
| 		if err != nil { | ||||
| 			failedHooks = append(failedHooks, "\n- "+channelID+": "+err.Error()) | ||||
| 			continue | ||||
| 		} | ||||
| 		t.assignHooksByAppID(hooks, botID, true) | ||||
| 	} | ||||
|  | ||||
| 	// Compose an error if any hooks failed | ||||
| 	if len(failedHooks) > 0 { | ||||
| 		return errors.New("failed to fetch hooks:" + strings.Join(failedHooks, "")) | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (t *Transmitter) assignHooksByAppID(hooks []*discordgo.Webhook, appID string, channelTargeted bool) { | ||||
| 	logLine := "Picking up webhook" | ||||
| 	if channelTargeted { | ||||
| 		logLine += " (channel targeted)" | ||||
| 	} | ||||
|  | ||||
| 	t.mutex.Lock() | ||||
| 	defer t.mutex.Unlock() | ||||
|  | ||||
| 	for _, wh := range hooks { | ||||
| 		if wh.ApplicationID != appID { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		t.channelWebhooks[wh.ChannelID] = wh | ||||
| 		t.Log.WithFields(log.Fields{ | ||||
| 			"id":      wh.ID, | ||||
| 			"name":    wh.Name, | ||||
| 			"channel": wh.ChannelID, | ||||
| 		}).Println(logLine) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										32
									
								
								bridge/discord/transmitter/utils.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								bridge/discord/transmitter/utils.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | ||||
| package transmitter | ||||
|  | ||||
| import ( | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| // isDiscordPermissionError returns false for nil, and true if a Discord RESTError with code discordgo.ErrorCodeMissionPermissions | ||||
| func isDiscordPermissionError(err error) bool { | ||||
| 	if err == nil { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	restErr, ok := err.(*discordgo.RESTError) | ||||
| 	if !ok { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	return restErr.Message != nil && restErr.Message.Code == discordgo.ErrCodeMissingPermissions | ||||
| } | ||||
|  | ||||
| // getDiscordUserID gets own user ID from state, and fallback on API request | ||||
| func getDiscordUserID(session *discordgo.Session) (string, error) { | ||||
| 	if user := session.State.User; user != nil { | ||||
| 		return user.ID, nil | ||||
| 	} | ||||
|  | ||||
| 	user, err := session.User("@me") | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return user.ID, nil | ||||
| } | ||||
							
								
								
									
										179
									
								
								bridge/discord/webhook.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										179
									
								
								bridge/discord/webhook.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,179 @@ | ||||
| package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| // shouldMessageUseWebhooks checks if have a channel specific webhook, if we're not using auto webhooks | ||||
| func (b *Bdiscord) shouldMessageUseWebhooks(msg *config.Message) bool { | ||||
| 	if b.useAutoWebhooks { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
| 	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok { | ||||
| 		if ci.Options.WebhookURL != "" { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| // maybeGetLocalAvatar checks if UseLocalAvatar contains the message's | ||||
| // account or protocol, and if so, returns the Discord avatar (if exists) | ||||
| func (b *Bdiscord) maybeGetLocalAvatar(msg *config.Message) string { | ||||
| 	for _, val := range b.GetStringSlice("UseLocalAvatar") { | ||||
| 		if msg.Protocol != val && msg.Account != val { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		member, err := b.getGuildMemberByNick(msg.Username) | ||||
| 		if err != nil { | ||||
| 			return "" | ||||
| 		} | ||||
|  | ||||
| 		return member.User.AvatarURL("") | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) webhookSendTextOnly(msg *config.Message, channelID string) (string, error) { | ||||
| 	msgParts := helper.ClipOrSplitMessage(msg.Text, MessageLength, b.GetString("MessageClipped"), b.GetInt("MessageSplitMaxCount")) | ||||
| 	msgIds := []string{} | ||||
| 	for _, msgPart := range msgParts { | ||||
| 		res, err := b.transmitter.Send( | ||||
| 			channelID, | ||||
| 			&discordgo.WebhookParams{ | ||||
| 				Content:         msgPart, | ||||
| 				Username:        msg.Username, | ||||
| 				AvatarURL:       msg.Avatar, | ||||
| 				AllowedMentions: b.getAllowedMentions(), | ||||
| 			}, | ||||
| 		) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} else { | ||||
| 			msgIds = append(msgIds, res.ID) | ||||
| 		} | ||||
| 	} | ||||
| 	// Exploit that a discord message ID is actually just a large number, so we encode a list of IDs by separating them with ";". | ||||
| 	return strings.Join(msgIds, ";"), nil | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) webhookSendFilesOnly(msg *config.Message, channelID string) error { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) //nolint:forcetypeassert | ||||
| 		file := discordgo.File{ | ||||
| 			Name:        fi.Name, | ||||
| 			ContentType: "", | ||||
| 			Reader:      bytes.NewReader(*fi.Data), | ||||
| 		} | ||||
| 		content := fi.Comment | ||||
|  | ||||
| 		// Cannot use the resulting ID for any edits anyway, so throw it away. | ||||
| 		// This has to be re-enabled when we implement message deletion. | ||||
| 		_, err := b.transmitter.Send( | ||||
| 			channelID, | ||||
| 			&discordgo.WebhookParams{ | ||||
| 				Username:        msg.Username, | ||||
| 				AvatarURL:       msg.Avatar, | ||||
| 				Files:           []*discordgo.File{&file}, | ||||
| 				Content:         content, | ||||
| 				AllowedMentions: b.getAllowedMentions(), | ||||
| 			}, | ||||
| 		) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Could not send file %#v for message %#v: %s", file, msg, err) | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // webhookSend send one or more message via webhook, taking care of file | ||||
| // uploads (from slack, telegram or mattermost). | ||||
| // Returns messageID and error. | ||||
| func (b *Bdiscord) webhookSend(msg *config.Message, channelID string) (string, error) { | ||||
| 	var ( | ||||
| 		res string | ||||
| 		err error | ||||
| 	) | ||||
|  | ||||
| 	// If avatar is unset, mutate the message to include the local avatar (but only if settings say we should do this) | ||||
| 	if msg.Avatar == "" { | ||||
| 		msg.Avatar = b.maybeGetLocalAvatar(msg) | ||||
| 	} | ||||
|  | ||||
| 	// WebhookParams can have either `Content` or `File`. | ||||
|  | ||||
| 	// We can't send empty messages. | ||||
| 	if msg.Text != "" { | ||||
| 		res, err = b.webhookSendTextOnly(msg, channelID) | ||||
| 	} | ||||
|  | ||||
| 	if err == nil && msg.Extra != nil { | ||||
| 		err = b.webhookSendFilesOnly(msg, channelID) | ||||
| 	} | ||||
|  | ||||
| 	return res, err | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) handleEventWebhook(msg *config.Message, channelID string) (string, error) { | ||||
| 	// skip events | ||||
| 	if msg.Event != "" && msg.Event != config.EventUserAction && msg.Event != config.EventJoinLeave && msg.Event != config.EventTopicChange { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// skip empty messages | ||||
| 	if msg.Text == "" && (msg.Extra == nil || len(msg.Extra["file"]) == 0) { | ||||
| 		b.Log.Debugf("Skipping empty message %#v", msg) | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// discord username must be [0..32] max | ||||
| 	if len(msg.Username) > 32 { | ||||
| 		msg.Username = msg.Username[0:32] | ||||
| 	} | ||||
|  | ||||
| 	if msg.ID != "" { | ||||
| 		// Exploit that a discord message ID is actually just a large number, and we encode a list of IDs by separating them with ";". | ||||
| 		msgIds := strings.Split(msg.ID, ";") | ||||
| 		msgParts := helper.ClipOrSplitMessage(b.replaceUserMentions(msg.Text), MessageLength, b.GetString("MessageClipped"), len(msgIds)) | ||||
| 		for len(msgParts) < len(msgIds) { | ||||
| 			msgParts = append(msgParts, "((obsoleted by edit))") | ||||
| 		} | ||||
| 		b.Log.Debugf("Editing webhook message") | ||||
| 		var editErr error = nil | ||||
| 		for i := range msgParts { | ||||
| 			// In case of split-messages where some parts remain the same (i.e. only a typo-fix in a huge message), this causes some noop-updates. | ||||
| 			// TODO: Optimize away noop-updates of un-edited messages | ||||
| 			editErr = b.transmitter.Edit(channelID, msgIds[i], &discordgo.WebhookParams{ | ||||
| 				Content:         msgParts[i], | ||||
| 				Username:        msg.Username, | ||||
| 				AllowedMentions: b.getAllowedMentions(), | ||||
| 			}) | ||||
| 			if editErr != nil { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		if editErr == nil { | ||||
| 			return msg.ID, nil | ||||
| 		} | ||||
| 		b.Log.Errorf("Could not edit webhook message(s): %s; sending as new message(s) instead", editErr) | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("Processing webhook sending for message %#v", msg) | ||||
| 	msg.Text = b.replaceUserMentions(msg.Text) | ||||
| 	msgID, err := b.webhookSend(msg, channelID) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Could not broadcast via webhook for message %#v: %s", msgID, err) | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return msgID, nil | ||||
| } | ||||
| @@ -1,182 +0,0 @@ | ||||
| package bgitter | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"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 { | ||||
| 	c     *gitter.Gitter | ||||
| 	User  *gitter.User | ||||
| 	Users []gitter.User | ||||
| 	Rooms []gitter.Room | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Bgitter{Config: cfg} | ||||
| } | ||||
|  | ||||
| func (b *Bgitter) Connect() error { | ||||
| 	var err error | ||||
| 	b.Log.Info("Connecting") | ||||
| 	b.c = gitter.New(b.GetString("Token")) | ||||
| 	b.User, err = b.c.GetUser() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Rooms, err = b.c.GetRooms() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bgitter) Disconnect() error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	roomID, err := b.c.GetRoomId(channel.Name) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("Could not find roomID for %v. Please create the room on gitter.im", channel.Name) | ||||
| 	} | ||||
| 	room, err := b.c.GetRoom(roomID) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Rooms = append(b.Rooms, *room) | ||||
| 	user, err := b.c.GetUser() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = b.c.JoinRoom(roomID, user.ID) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	users, _ := b.c.GetUsersInRoom(roomID) | ||||
| 	b.Users = append(b.Users, users...) | ||||
| 	stream := b.c.Stream(roomID) | ||||
| 	go b.c.Listen(stream) | ||||
|  | ||||
| 	go func(stream *gitter.Stream, room string) { | ||||
| 		for event := range stream.Event { | ||||
| 			switch ev := event.Data.(type) { | ||||
| 			case *gitter.MessageReceived: | ||||
| 				// ignore message sent from ourselves | ||||
| 				if ev.Message.From.ID != b.User.ID { | ||||
| 					b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Message.From.Username, b.Account) | ||||
| 					rmsg := config.Message{Username: ev.Message.From.Username, Text: ev.Message.Text, Channel: room, | ||||
| 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID, | ||||
| 						ID: ev.Message.ID} | ||||
| 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | ||||
| 						rmsg.Event = config.EventUserAction | ||||
| 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | ||||
| 					} | ||||
| 					b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 					b.Remote <- rmsg | ||||
| 				} | ||||
| 			case *gitter.GitterConnectionClosed: | ||||
| 				b.Log.Errorf("connection with gitter closed for room %s", room) | ||||
| 			} | ||||
| 		} | ||||
| 	}(stream, room.URI) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bgitter) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
| 	roomID := b.getRoomID(msg.Channel) | ||||
| 	if roomID == "" { | ||||
| 		b.Log.Errorf("Could not find roomID for %v", msg.Channel) | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// 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 { | ||||
| 	for _, v := range b.Rooms { | ||||
| 		if v.URI == channel { | ||||
| 			return v.ID | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bgitter) getAvatar(user string) string { | ||||
| 	var avatar string | ||||
| 	if b.Users != nil { | ||||
| 		for _, u := range b.Users { | ||||
| 			if user == u.Username { | ||||
| 				return u.AvatarURLSmall | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return avatar | ||||
| } | ||||
|  | ||||
| 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 | ||||
| } | ||||
							
								
								
									
										252
									
								
								bridge/harmony/harmony.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										252
									
								
								bridge/harmony/harmony.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,252 @@ | ||||
| package harmony | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"log" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/harmony-development/shibshib" | ||||
| 	chatv1 "github.com/harmony-development/shibshib/gen/chat/v1" | ||||
| 	typesv1 "github.com/harmony-development/shibshib/gen/harmonytypes/v1" | ||||
| 	profilev1 "github.com/harmony-development/shibshib/gen/profile/v1" | ||||
| ) | ||||
|  | ||||
| type cachedProfile struct { | ||||
| 	data        *profilev1.GetProfileResponse | ||||
| 	lastUpdated time.Time | ||||
| } | ||||
|  | ||||
| type Bharmony struct { | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	c            *shibshib.Client | ||||
| 	profileCache map[uint64]cachedProfile | ||||
| } | ||||
|  | ||||
| func uToStr(in uint64) string { | ||||
| 	return strconv.FormatUint(in, 10) | ||||
| } | ||||
|  | ||||
| func strToU(in string) (uint64, error) { | ||||
| 	return strconv.ParseUint(in, 10, 64) | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bharmony{ | ||||
| 		Config:       cfg, | ||||
| 		profileCache: map[uint64]cachedProfile{}, | ||||
| 	} | ||||
|  | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) getProfile(u uint64) (*profilev1.GetProfileResponse, error) { | ||||
| 	if v, ok := b.profileCache[u]; ok && time.Since(v.lastUpdated) < time.Minute*10 { | ||||
| 		return v.data, nil | ||||
| 	} | ||||
|  | ||||
| 	resp, err := b.c.ProfileKit.GetProfile(&profilev1.GetProfileRequest{ | ||||
| 		UserId: u, | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		if v, ok := b.profileCache[u]; ok { | ||||
| 			return v.data, nil | ||||
| 		} | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	b.profileCache[u] = cachedProfile{ | ||||
| 		data:        resp, | ||||
| 		lastUpdated: time.Now(), | ||||
| 	} | ||||
| 	return resp, nil | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) avatarFor(m *chatv1.Message) string { | ||||
| 	if m.Overrides != nil { | ||||
| 		return m.Overrides.GetAvatar() | ||||
| 	} | ||||
|  | ||||
| 	profi, err := b.getProfile(m.AuthorId) | ||||
| 	if err != nil { | ||||
| 		return "" | ||||
| 	} | ||||
|  | ||||
| 	return b.c.TransformHMCURL(profi.Profile.GetUserAvatar()) | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) usernameFor(m *chatv1.Message) string { | ||||
| 	if m.Overrides != nil { | ||||
| 		return m.Overrides.GetUsername() | ||||
| 	} | ||||
|  | ||||
| 	profi, err := b.getProfile(m.AuthorId) | ||||
| 	if err != nil { | ||||
| 		return "" | ||||
| 	} | ||||
|  | ||||
| 	return profi.Profile.UserName | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) toMessage(msg *shibshib.LocatedMessage) config.Message { | ||||
| 	message := config.Message{} | ||||
| 	message.Account = b.Account | ||||
| 	message.UserID = uToStr(msg.Message.AuthorId) | ||||
| 	message.Avatar = b.avatarFor(msg.Message) | ||||
| 	message.Username = b.usernameFor(msg.Message) | ||||
| 	message.Channel = uToStr(msg.ChannelID) | ||||
| 	message.ID = uToStr(msg.MessageId) | ||||
|  | ||||
| 	switch content := msg.Message.Content.Content.(type) { | ||||
| 	case *chatv1.Content_EmbedMessage: | ||||
| 		message.Text = "Embed" | ||||
| 	case *chatv1.Content_AttachmentMessage: | ||||
| 		var s strings.Builder | ||||
| 		for idx, attach := range content.AttachmentMessage.Files { | ||||
| 			s.WriteString(b.c.TransformHMCURL(attach.Id)) | ||||
| 			if idx < len(content.AttachmentMessage.Files)-1 { | ||||
| 				s.WriteString(", ") | ||||
| 			} | ||||
| 		} | ||||
| 		message.Text = s.String() | ||||
| 	case *chatv1.Content_PhotoMessage: | ||||
| 		var s strings.Builder | ||||
| 		for idx, attach := range content.PhotoMessage.GetPhotos() { | ||||
| 			s.WriteString(attach.GetCaption().GetText()) | ||||
| 			s.WriteString("\n") | ||||
| 			s.WriteString(b.c.TransformHMCURL(attach.GetHmc())) | ||||
| 			if idx < len(content.PhotoMessage.GetPhotos())-1 { | ||||
| 				s.WriteString("\n\n") | ||||
| 			} | ||||
| 		} | ||||
| 		message.Text = s.String() | ||||
| 	case *chatv1.Content_TextMessage: | ||||
| 		message.Text = content.TextMessage.Content.Text | ||||
| 	} | ||||
|  | ||||
| 	return message | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) outputMessages() { | ||||
| 	for { | ||||
| 		msg := <-b.c.EventsStream() | ||||
|  | ||||
| 		if msg.Message.AuthorId == b.c.UserID { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		b.Remote <- b.toMessage(msg) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) GetUint64(conf string) uint64 { | ||||
| 	num, err := strToU(b.GetString(conf)) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	return num | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) Connect() (err error) { | ||||
| 	b.c, err = shibshib.NewClient(b.GetString("Homeserver"), b.GetString("Token"), b.GetUint64("UserID")) | ||||
| 	if err != nil { | ||||
| 		return | ||||
| 	} | ||||
| 	b.c.SubscribeToGuild(b.GetUint64("Community")) | ||||
|  | ||||
| 	go b.outputMessages() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) send(msg config.Message) (id string, err error) { | ||||
| 	msgChan, err := strToU(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	retID, err := b.c.ChatKit.SendMessage(&chatv1.SendMessageRequest{ | ||||
| 		GuildId:   b.GetUint64("Community"), | ||||
| 		ChannelId: msgChan, | ||||
| 		Content: &chatv1.Content{ | ||||
| 			Content: &chatv1.Content_TextMessage{ | ||||
| 				TextMessage: &chatv1.Content_TextContent{ | ||||
| 					Content: &chatv1.FormattedText{ | ||||
| 						Text: msg.Text, | ||||
| 					}, | ||||
| 				}, | ||||
| 			}, | ||||
| 		}, | ||||
| 		Overrides: &chatv1.Overrides{ | ||||
| 			Username: &msg.Username, | ||||
| 			Avatar:   &msg.Avatar, | ||||
| 			Reason:   &chatv1.Overrides_Bridge{Bridge: &typesv1.Empty{}}, | ||||
| 		}, | ||||
| 		InReplyTo: nil, | ||||
| 		EchoId:    nil, | ||||
| 		Metadata:  nil, | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		err = fmt.Errorf("send: error sending message: %w", err) | ||||
| 		log.Println(err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	return uToStr(retID.MessageId), err | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) delete(msg config.Message) (id string, err error) { | ||||
| 	msgChan, err := strToU(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	msgID, err := strToU(msg.ID) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	_, err = b.c.ChatKit.DeleteMessage(&chatv1.DeleteMessageRequest{ | ||||
| 		GuildId:   b.GetUint64("Community"), | ||||
| 		ChannelId: msgChan, | ||||
| 		MessageId: msgID, | ||||
| 	}) | ||||
| 	return "", err | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) typing(msg config.Message) (id string, err error) { | ||||
| 	msgChan, err := strToU(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	_, err = b.c.ChatKit.Typing(&chatv1.TypingRequest{ | ||||
| 		GuildId:   b.GetUint64("Community"), | ||||
| 		ChannelId: msgChan, | ||||
| 	}) | ||||
| 	return "", err | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) Send(msg config.Message) (id string, err error) { | ||||
| 	switch msg.Event { | ||||
| 	case "": | ||||
| 		return b.send(msg) | ||||
| 	case config.EventMsgDelete: | ||||
| 		return b.delete(msg) | ||||
| 	case config.EventUserTyping: | ||||
| 		return b.typing(msg) | ||||
| 	default: | ||||
| 		return "", nil | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bharmony) Disconnect() error { | ||||
| 	return nil | ||||
| } | ||||
| @@ -14,8 +14,10 @@ import ( | ||||
| 	"golang.org/x/image/webp" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/gomarkdown/markdown" | ||||
| 	"github.com/gomarkdown/markdown/html" | ||||
| 	"github.com/gomarkdown/markdown/parser" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"gitlab.com/golang-commonmark/markdown" | ||||
| ) | ||||
|  | ||||
| // DownloadFile downloads the given non-authenticated URL. | ||||
| @@ -46,6 +48,30 @@ func DownloadFileAuth(url string, auth string) (*[]byte, error) { | ||||
| 	return &data, nil | ||||
| } | ||||
|  | ||||
| // DownloadFileAuthRocket downloads the given URL using the specified Rocket user ID and authentication token. | ||||
| func DownloadFileAuthRocket(url, token, userID string) (*[]byte, error) { | ||||
| 	var buf bytes.Buffer | ||||
| 	client := &http.Client{ | ||||
| 		Timeout: time.Second * 5, | ||||
| 	} | ||||
| 	req, err := http.NewRequest("GET", url, nil) | ||||
|  | ||||
| 	req.Header.Add("X-Auth-Token", token) | ||||
| 	req.Header.Add("X-User-Id", userID) | ||||
|  | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	resp, err := client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	defer resp.Body.Close() | ||||
| 	_, err = io.Copy(&buf, resp.Body) | ||||
| 	data := buf.Bytes() | ||||
| 	return &data, err | ||||
| } | ||||
|  | ||||
| // GetSubLines splits messages in newline-delimited lines. If maxLineLength is | ||||
| // specified as non-zero GetSubLines will also clip long lines to the maximum | ||||
| // length and insert a warning marker that the line was clipped. | ||||
| @@ -53,11 +79,19 @@ func DownloadFileAuth(url string, auth string) (*[]byte, error) { | ||||
| // 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>" | ||||
| func GetSubLines(message string, maxLineLength int, clippingMessage string) []string { | ||||
| 	if clippingMessage == "" { | ||||
| 		clippingMessage = " <clipped message>" | ||||
| 	} | ||||
|  | ||||
| 	var lines []string | ||||
| 	for _, line := range strings.Split(strings.TrimSpace(message), "\n") { | ||||
| 		if line == "" { | ||||
| 			// Prevent sending empty messages, so we'll skip this line | ||||
| 			// if it has no content. | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if maxLineLength == 0 || len([]byte(line)) <= maxLineLength { | ||||
| 			lines = append(lines, line) | ||||
| 			continue | ||||
| @@ -140,17 +174,23 @@ func HandleDownloadSize(logger *logrus.Entry, msg *config.Message, name string, | ||||
|  | ||||
| // HandleDownloadData adds the data for a remote file into a Matterbridge gateway message. | ||||
| func HandleDownloadData(logger *logrus.Entry, msg *config.Message, name, comment, url string, data *[]byte, general *config.Protocol) { | ||||
| 	HandleDownloadData2(logger, msg, name, "", comment, url, data, general) | ||||
| } | ||||
|  | ||||
| // HandleDownloadData adds the data for a remote file into a Matterbridge gateway message. | ||||
| func HandleDownloadData2(logger *logrus.Entry, msg *config.Message, name, id, comment, url string, data *[]byte, general *config.Protocol) { | ||||
| 	var avatar bool | ||||
| 	logger.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, | ||||
| 		Name:     name, | ||||
| 		Data:     data, | ||||
| 		URL:      url, | ||||
| 		Comment:  comment, | ||||
| 		Avatar:   avatar, | ||||
| 		NativeID: id, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| @@ -164,24 +204,73 @@ func RemoveEmptyNewLines(msg string) string { | ||||
|  | ||||
| // ClipMessage trims a message to the specified length if it exceeds it and adds a warning | ||||
| // to the message in case it does so. | ||||
| func ClipMessage(text string, length int) string { | ||||
| 	const clippingMessage = " <clipped message>" | ||||
| func ClipMessage(text string, length int, clippingMessage string) string { | ||||
| 	if clippingMessage == "" { | ||||
| 		clippingMessage = " <clipped message>" | ||||
| 	} | ||||
|  | ||||
| 	if len(text) > length { | ||||
| 		text = text[:length-len(clippingMessage)] | ||||
| 		if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||
| 			text = text[:len(text)-size] | ||||
| 		for len(text) > 0 { | ||||
| 			if r, _ := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||
| 				text = text[:len(text)-1] | ||||
| 				// Note: DecodeLastRuneInString only returns the constant value "1" in | ||||
| 				// case of an error. We do not yet know whether the last rune is now | ||||
| 				// actually valid. Example: "€" is 0xE2 0x82 0xAC. If we happen to split | ||||
| 				// the string just before 0xAC, and go back only one byte, that would | ||||
| 				// leave us with a string that ends in the byte 0xE2, which is not a valid | ||||
| 				// rune, so we need to try again. | ||||
| 			} else { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		text += clippingMessage | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func ParseMarkdown(input string) string { | ||||
| 	md := markdown.New(markdown.XHTMLOutput(true), markdown.Breaks(true)) | ||||
| 	return (md.RenderToString([]byte(input))) | ||||
| func ClipOrSplitMessage(text string, length int, clippingMessage string, splitMax int) []string { | ||||
| 	var msgParts []string | ||||
| 	remainingText := text | ||||
| 	// Invariant of this splitting loop: No text is lost (msgParts+remainingText is the original text), | ||||
| 	// and all parts is guaranteed to satisfy the length requirement. | ||||
| 	for len(msgParts) < splitMax-1 && len(remainingText) > length { | ||||
| 		// Decision: The text needs to be split (again). | ||||
| 		var chunk string | ||||
| 		wasted := 0 | ||||
| 		// The longest UTF-8 encoding of a valid rune is 4 bytes (0xF4 0x8F 0xBF 0xBF, encoding U+10FFFF), | ||||
| 		// so we should never need to waste 4 or more bytes at a time. | ||||
| 		for wasted < 4 && wasted < length { | ||||
| 			chunk = remainingText[:length-wasted] | ||||
| 			if r, _ := utf8.DecodeLastRuneInString(chunk); r == utf8.RuneError { | ||||
| 				wasted += 1 | ||||
| 			} else { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		// Note: At this point, "chunk" might still be invalid, if "text" is very broken. | ||||
| 		msgParts = append(msgParts, chunk) | ||||
| 		remainingText = remainingText[len(chunk):] | ||||
| 	} | ||||
| 	msgParts = append(msgParts, ClipMessage(remainingText, length, clippingMessage)) | ||||
| 	return msgParts | ||||
| } | ||||
|  | ||||
| // ConvertWebPToPNG convert input data (which should be WebP format to PNG format) | ||||
| // ParseMarkdown takes in an input string as markdown and parses it to html | ||||
| func ParseMarkdown(input string) string { | ||||
| 	extensions := parser.HardLineBreak | parser.NoIntraEmphasis | parser.FencedCode | ||||
| 	markdownParser := parser.NewWithExtensions(extensions) | ||||
| 	renderer := html.NewRenderer(html.RendererOptions{ | ||||
| 		Flags: 0, | ||||
| 	}) | ||||
| 	parsedMarkdown := markdown.ToHTML([]byte(input), markdownParser, renderer) | ||||
| 	res := string(parsedMarkdown) | ||||
| 	res = strings.TrimPrefix(res, "<p>") | ||||
| 	res = strings.TrimSuffix(res, "</p>\n") | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| // ConvertWebPToPNG converts input data (which should be WebP format) to PNG format | ||||
| func ConvertWebPToPNG(data *[]byte) error { | ||||
| 	r := bytes.NewReader(*data) | ||||
| 	m, err := webp.Decode(r) | ||||
|   | ||||
| @@ -10,98 +10,105 @@ import ( | ||||
|  | ||||
| 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"}, | ||||
| 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.", | ||||
| 		}, | ||||
| 		"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."}, | ||||
| 		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!", | ||||
| 		}, | ||||
| 		"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!", | ||||
| 			}, | ||||
| 		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.", | ||||
| 			}, | ||||
| 	}, | ||||
| 	"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.", | ||||
| 		}, | ||||
| 		"Message ending with new-line.": { | ||||
| 			input:          "Newline ending\n", | ||||
| 			splitOutput:    []string{"Newline ending"}, | ||||
| 			nonSplitOutput: []string{"Newline ending"}, | ||||
| 		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.", | ||||
| 		}, | ||||
| 		"Long message containing UTF-8 multi-byte runes": { | ||||
| 			input: "不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說", | ||||
| 			splitOutput: []string{ | ||||
| 				"不布人個我此而及單石業喜資富下 <clipped message>", | ||||
| 				"我河下日沒一我臺空達的常景便物 <clipped message>", | ||||
| 				"沒為……子大我別名解成?生賣的 <clipped message>", | ||||
| 				"全直黑,我自我結毛分洲了世當, <clipped message>", | ||||
| 				"是政福那是東;斯說", | ||||
| 			}, | ||||
| 			nonSplitOutput: []string{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"}, | ||||
| 	}, | ||||
| 	"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{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"}, | ||||
| 	}, | ||||
| 	"Long message, clip three-byte rune after two bytes": { | ||||
| 		input: "x 人人生而自由,在尊嚴和權利上一律平等。 他們都具有理性和良知,應該以兄弟情誼的精神對待彼此。", | ||||
| 		splitOutput: []string{ | ||||
| 			"x 人人生而自由,在尊嚴和權利上 <clipped message>", | ||||
| 			"一律平等。 他們都具有理性和良知 <clipped message>", | ||||
| 			",應該以兄弟情誼的精神對待彼此。", | ||||
| 		}, | ||||
| 		nonSplitOutput: []string{"x 人人生而自由,在尊嚴和權利上一律平等。 他們都具有理性和良知,應該以兄弟情誼的精神對待彼此。"}, | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func TestGetSubLines(t *testing.T) { | ||||
| 	for testname, testcase := range lineSplittingTestCases { | ||||
| 		splitLines := GetSubLines(testcase.input, testLineLength) | ||||
| 		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) | ||||
| 		nonSplitLines := GetSubLines(testcase.input, 0, "") | ||||
| 		assert.Equalf(t, testcase.nonSplitOutput, nonSplitLines, "'%s' testcase should give expected lines without splitting.", testname) | ||||
| 	} | ||||
| } | ||||
| @@ -110,17 +117,122 @@ func TestConvertWebPToPNG(t *testing.T) { | ||||
| 	if os.Getenv("LOCAL_TEST") == "" { | ||||
| 		t.Skip() | ||||
| 	} | ||||
|  | ||||
| 	input, err := ioutil.ReadFile("test.webp") | ||||
| 	if err != nil { | ||||
| 		t.Fail() | ||||
| 	} | ||||
|  | ||||
| 	d := &input | ||||
| 	err = ConvertWebPToPNG(d) | ||||
| 	if err != nil { | ||||
| 		t.Fail() | ||||
| 	} | ||||
| 	err = ioutil.WriteFile("test.png", *d, 0644) | ||||
|  | ||||
| 	err = ioutil.WriteFile("test.png", *d, 0o644) // nolint:gosec | ||||
| 	if err != nil { | ||||
| 		t.Fail() | ||||
| 	} | ||||
| } | ||||
|  | ||||
| var clippingOrSplittingTestCases = map[string]struct { | ||||
| 	inputText       string | ||||
| 	clipSplitLength int | ||||
| 	clippingMessage string | ||||
| 	splitMax        int | ||||
| 	expectedOutput  []string | ||||
| }{ | ||||
| 	"Short single-line message, split 3": { | ||||
| 		inputText:       "short", | ||||
| 		clipSplitLength: 20, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        3, | ||||
| 		expectedOutput:  []string{"short"}, | ||||
| 	}, | ||||
| 	"Short single-line message, split 1": { | ||||
| 		inputText:       "short", | ||||
| 		clipSplitLength: 20, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        1, | ||||
| 		expectedOutput:  []string{"short"}, | ||||
| 	}, | ||||
| 	"Short single-line message, split 0": { | ||||
| 		// Mainly check that we don't crash. | ||||
| 		inputText:       "short", | ||||
| 		clipSplitLength: 20, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        0, | ||||
| 		expectedOutput:  []string{"short"}, | ||||
| 	}, | ||||
| 	"Long single-line message, noclip": { | ||||
| 		inputText:       "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        10, | ||||
| 		expectedOutput: []string{ | ||||
| 			"Lorem ipsum dolor sit amet, consectetur adipiscing", | ||||
| 			" elit, sed do eiusmod tempor incididunt ut labore ", | ||||
| 			"et dolore magna aliqua.", | ||||
| 		}, | ||||
| 	}, | ||||
| 	"Long single-line message, noclip tight": { | ||||
| 		inputText:       "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        3, | ||||
| 		expectedOutput: []string{ | ||||
| 			"Lorem ipsum dolor sit amet, consectetur adipiscing", | ||||
| 			" elit, sed do eiusmod tempor incididunt ut labore ", | ||||
| 			"et dolore magna aliqua.", | ||||
| 		}, | ||||
| 	}, | ||||
| 	"Long single-line message, clip custom": { | ||||
| 		inputText:       "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "?!?!", | ||||
| 		splitMax:        2, | ||||
| 		expectedOutput: []string{ | ||||
| 			"Lorem ipsum dolor sit amet, consectetur adipiscing", | ||||
| 			" elit, sed do eiusmod tempor incididunt ut lab?!?!", | ||||
| 		}, | ||||
| 	}, | ||||
| 	"Long single-line message, clip built-in": { | ||||
| 		inputText:       "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "", | ||||
| 		splitMax:        2, | ||||
| 		expectedOutput: []string{ | ||||
| 			"Lorem ipsum dolor sit amet, consectetur adipiscing", | ||||
| 			" elit, sed do eiusmod tempor inc <clipped message>", | ||||
| 		}, | ||||
| 	}, | ||||
| 	"Short multi-line message": { | ||||
| 		inputText:       "I\ncan't\nget\nno\nsatisfaction!", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "", | ||||
| 		splitMax:        2, | ||||
| 		expectedOutput:  []string{"I\ncan't\nget\nno\nsatisfaction!"}, | ||||
| 	}, | ||||
| 	"Long message containing UTF-8 multi-byte runes": { | ||||
| 		inputText:       "人人生而自由,在尊嚴和權利上一律平等。 他們都具有理性和良知,應該以兄弟情誼的精神對待彼此。", | ||||
| 		clipSplitLength: 50, | ||||
| 		clippingMessage: "", | ||||
| 		splitMax:        10, | ||||
| 		expectedOutput: []string{ | ||||
| 			"人人生而自由,在尊嚴和權利上一律",  // Note: only 48 bytes! | ||||
| 			"平等。 他們都具有理性和良知,應該", // Note: only 49 bytes! | ||||
| 			"以兄弟情誼的精神對待彼此。", | ||||
| 		}, | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func TestClipOrSplitMessage(t *testing.T) { | ||||
| 	for testname, testcase := range clippingOrSplittingTestCases { | ||||
| 		actualOutput := ClipOrSplitMessage(testcase.inputText, testcase.clipSplitLength, testcase.clippingMessage, testcase.splitMax) | ||||
| 		assert.Equalf(t, testcase.expectedOutput, actualOutput, "'%s' testcase should give expected lines with clipping+splitting.", testname) | ||||
| 		for _, splitLine := range testcase.expectedOutput { | ||||
| 			byteLength := len([]byte(splitLine)) | ||||
| 			assert.True(t, byteLength <= testcase.clipSplitLength, "Splitted line '%s' of testcase '%s' should not exceed the maximum byte-length (%d vs. %d).", splitLine, testname, testcase.clipSplitLength, byteLength) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										35
									
								
								bridge/helper/libtgsconverter.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								bridge/helper/libtgsconverter.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| //go:build cgolottie | ||||
|  | ||||
| package helper | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
|  | ||||
| 	"github.com/Benau/tgsconverter/libtgsconverter" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| func CanConvertTgsToX() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // ConvertTgsToX convert input data (which should be tgs format) to any format supported by libtgsconverter | ||||
| func ConvertTgsToX(data *[]byte, outputFormat string, logger *logrus.Entry) error { | ||||
| 	options := libtgsconverter.NewConverterOptions() | ||||
| 	options.SetExtension(outputFormat) | ||||
| 	blob, err := libtgsconverter.ImportFromData(*data, options) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("failed to run libtgsconverter.ImportFromData: %s", err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	*data = blob | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func SupportsFormat(format string) bool { | ||||
| 	return libtgsconverter.SupportsExtension(format) | ||||
| } | ||||
|  | ||||
| func LottieBackend() string { | ||||
| 	return "libtgsconverter" | ||||
| } | ||||
							
								
								
									
										90
									
								
								bridge/helper/lottie_convert.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										90
									
								
								bridge/helper/lottie_convert.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,90 @@ | ||||
| //go:build !cgolottie | ||||
|  | ||||
| package helper | ||||
|  | ||||
| import ( | ||||
| 	"io/ioutil" | ||||
| 	"os" | ||||
| 	"os/exec" | ||||
|  | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| // CanConvertTgsToX Checks whether the external command necessary for ConvertTgsToX works. | ||||
| func CanConvertTgsToX() error { | ||||
| 	// We depend on the fact that `lottie_convert.py --help` has exit status 0. | ||||
| 	// Hyrum's Law predicted this, and Murphy's Law predicts that this will break eventually. | ||||
| 	// However, there is no alternative like `lottie_convert.py --is-properly-installed` | ||||
| 	cmd := exec.Command("lottie_convert.py", "--help") | ||||
| 	return cmd.Run() | ||||
| } | ||||
|  | ||||
| // ConvertTgsToWebP convert input data (which should be tgs format) to WebP format | ||||
| // This relies on an external command, which is ugly, but works. | ||||
| func ConvertTgsToX(data *[]byte, outputFormat string, logger *logrus.Entry) error { | ||||
| 	// lottie can't handle input from a pipe, so write to a temporary file: | ||||
| 	tmpInFile, err := ioutil.TempFile(os.TempDir(), "matterbridge-lottie-input-*.tgs") | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	tmpInFileName := tmpInFile.Name() | ||||
| 	defer func() { | ||||
| 		if removeErr := os.Remove(tmpInFileName); removeErr != nil { | ||||
| 			logger.Errorf("Could not delete temporary (input) file %s: %v", tmpInFileName, removeErr) | ||||
| 		} | ||||
| 	}() | ||||
| 	// lottie can handle writing to a pipe, but there is no way to do that platform-independently. | ||||
| 	// "/dev/stdout" won't work on Windows, and "-" upsets Cairo for some reason. So we need another file: | ||||
| 	tmpOutFile, err := ioutil.TempFile(os.TempDir(), "matterbridge-lottie-output-*.data") | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	tmpOutFileName := tmpOutFile.Name() | ||||
| 	defer func() { | ||||
| 		if removeErr := os.Remove(tmpOutFileName); removeErr != nil { | ||||
| 			logger.Errorf("Could not delete temporary (output) file %s: %v", tmpOutFileName, removeErr) | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	if _, writeErr := tmpInFile.Write(*data); writeErr != nil { | ||||
| 		return writeErr | ||||
| 	} | ||||
| 	// Must close before calling lottie to avoid data races: | ||||
| 	if closeErr := tmpInFile.Close(); closeErr != nil { | ||||
| 		return closeErr | ||||
| 	} | ||||
|  | ||||
| 	// Call lottie to transform: | ||||
| 	cmd := exec.Command("lottie_convert.py", "--input-format", "lottie", "--output-format", outputFormat, tmpInFileName, tmpOutFileName) | ||||
| 	cmd.Stdout = nil | ||||
| 	cmd.Stderr = nil | ||||
| 	// NB: lottie writes progress into to stderr in all cases. | ||||
| 	_, stderr := cmd.Output() | ||||
| 	if stderr != nil { | ||||
| 		// 'stderr' already contains some parts of Stderr, because it was set to 'nil'. | ||||
| 		return stderr | ||||
| 	} | ||||
| 	dataContents, err := ioutil.ReadFile(tmpOutFileName) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	*data = dataContents | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func SupportsFormat(format string) bool { | ||||
| 	switch format { | ||||
| 	case "png": | ||||
| 		fallthrough | ||||
| 	case "webp": | ||||
| 		return true | ||||
| 	default: | ||||
| 		return false | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func LottieBackend() string { | ||||
| 	return "lottie_convert.py" | ||||
| } | ||||
							
								
								
									
										32
									
								
								bridge/irc/charset.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								bridge/irc/charset.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | ||||
| package birc | ||||
|  | ||||
| import ( | ||||
| 	"golang.org/x/text/encoding" | ||||
| 	"golang.org/x/text/encoding/japanese" | ||||
| 	"golang.org/x/text/encoding/korean" | ||||
| 	"golang.org/x/text/encoding/simplifiedchinese" | ||||
| 	"golang.org/x/text/encoding/traditionalchinese" | ||||
| 	"golang.org/x/text/encoding/unicode" | ||||
| ) | ||||
|  | ||||
| var encoders = map[string]encoding.Encoding{ | ||||
| 	"utf-8":       unicode.UTF8, | ||||
| 	"iso-2022-jp": japanese.ISO2022JP, | ||||
| 	"big5":        traditionalchinese.Big5, | ||||
| 	"gbk":         simplifiedchinese.GBK, | ||||
| 	"euc-kr":      korean.EUCKR, | ||||
| 	"gb2312":      simplifiedchinese.HZGB2312, | ||||
| 	"shift-jis":   japanese.ShiftJIS, | ||||
| 	"euc-jp":      japanese.EUCJP, | ||||
| 	"gb18030":     simplifiedchinese.GB18030, | ||||
| } | ||||
|  | ||||
| func toUTF8(from string, input string) string { | ||||
| 	enc, ok := encoders[from] | ||||
| 	if !ok { | ||||
| 		return input | ||||
| 	} | ||||
|  | ||||
| 	res, _ := enc.NewDecoder().String(input) | ||||
| 	return res | ||||
| } | ||||
| @@ -4,14 +4,12 @@ import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"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" | ||||
| @@ -25,12 +23,12 @@ func (b *Birc) handleCharset(msg *config.Message) error { | ||||
| 	if b.GetString("Charset") != "" { | ||||
| 		switch b.GetString("Charset") { | ||||
| 		case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 			msg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), msg.Text) | ||||
| 			msg.Text = toUTF8(b.GetString("Charset"), msg.Text) | ||||
| 		default: | ||||
| 			buf := new(bytes.Buffer) | ||||
| 			w, err := charset.NewWriter(b.GetString("Charset"), buf) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("charset from utf-8 conversion failed: %s", err) | ||||
| 				b.Log.Errorf("charset to utf-8 conversion failed: %s", err) | ||||
| 				return err | ||||
| 			} | ||||
| 			fmt.Fprint(w, msg.Text) | ||||
| @@ -55,12 +53,12 @@ func (b *Birc) handleFiles(msg *config.Message) bool { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		if fi.Comment != "" { | ||||
| 			msg.Text += fi.Comment + ": " | ||||
| 			msg.Text += fi.Comment + " : " | ||||
| 		} | ||||
| 		if fi.URL != "" { | ||||
| 			msg.Text = fi.URL | ||||
| 			if fi.Comment != "" { | ||||
| 				msg.Text = fi.Comment + ": " + fi.URL | ||||
| 				msg.Text = fi.Comment + " : " + fi.URL | ||||
| 			} | ||||
| 		} | ||||
| 		b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||
| @@ -68,6 +66,20 @@ func (b *Birc) handleFiles(msg *config.Message) bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleInvite(client *girc.Client, event girc.Event) { | ||||
| 	if len(event.Params) != 2 { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	channel := event.Params[1] | ||||
|  | ||||
| 	b.Log.Debugf("got invite for %s", channel) | ||||
|  | ||||
| 	if _, ok := b.channels[channel]; ok { | ||||
| 		b.i.Cmd.Join(channel) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||
| 	if len(event.Params) == 0 { | ||||
| 		b.Log.Debugf("handleJoinPart: empty Params? %#v", event) | ||||
| @@ -91,12 +103,13 @@ func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		// QUIT isn't channel bound, happens for all channels on the bridge | ||||
| 		if event.Command == "QUIT" { | ||||
| 			channel = "" | ||||
| 		} | ||||
| 		msg := config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EventJoinLeave} | ||||
| 		if b.GetBool("verbosejoinpart") { | ||||
| 			b.Log.Debugf("<= Sending verbose JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 			msg = config.Message{Username: "system", Text: event.Source.Name + " (" + event.Source.Ident + "@" + event.Source.Host + ") " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EventJoinLeave} | ||||
| 		} else { | ||||
| 			b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Message is %#v", msg) | ||||
| 		b.Remote <- msg | ||||
| 		return | ||||
| @@ -109,14 +122,25 @@ func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||
| 	i := b.i | ||||
| 	b.Nick = event.Params[0] | ||||
|  | ||||
| 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||
| 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||
| 	b.Log.Debug("Clearing handlers before adding in case of BNC reconnect") | ||||
| 	i.Handlers.Clear("PRIVMSG") | ||||
| 	i.Handlers.Clear("CTCP_ACTION") | ||||
| 	i.Handlers.Clear(girc.RPL_TOPICWHOTIME) | ||||
| 	i.Handlers.Clear(girc.NOTICE) | ||||
| 	i.Handlers.Clear("JOIN") | ||||
| 	i.Handlers.Clear("PART") | ||||
| 	i.Handlers.Clear("QUIT") | ||||
| 	i.Handlers.Clear("KICK") | ||||
| 	i.Handlers.Clear("INVITE") | ||||
|  | ||||
| 	i.Handlers.AddBg("PRIVMSG", b.handlePrivMsg) | ||||
| 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||
| 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||
| 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||
| 	i.Handlers.Add("PART", b.handleJoinPart) | ||||
| 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||
| 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||
| 	i.Handlers.AddBg(girc.NOTICE, b.handleNotice) | ||||
| 	i.Handlers.AddBg("JOIN", b.handleJoinPart) | ||||
| 	i.Handlers.AddBg("PART", b.handleJoinPart) | ||||
| 	i.Handlers.AddBg("QUIT", b.handleJoinPart) | ||||
| 	i.Handlers.AddBg("KICK", b.handleJoinPart) | ||||
| 	i.Handlers.Add("INVITE", b.handleInvite) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNickServ() { | ||||
| @@ -160,28 +184,43 @@ func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) { | ||||
| 	b.handleNickServ() | ||||
| 	b.handleRunCommands() | ||||
| 	// we are now fully connected | ||||
| 	b.connected <- nil | ||||
| 	// only send on first connection | ||||
| 	if b.FirstConnection { | ||||
| 		b.connected <- nil | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||
| 	if b.skipPrivMsg(event) { | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		Username: event.Source.Name, | ||||
| 		Channel:  strings.ToLower(event.Params[0]), | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   event.Source.Ident + "@" + event.Source.Host, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Last(), event) | ||||
|  | ||||
| 	// set action event | ||||
| 	if event.IsAction() { | ||||
| 	if ok, ctcp := event.IsCTCP(); ok { | ||||
| 		if ctcp.Command != girc.CTCP_ACTION { | ||||
| 			b.Log.Debugf("dropping user ctcp, command: %s", ctcp.Command) | ||||
| 			return | ||||
| 		} | ||||
| 		rmsg.Event = config.EventUserAction | ||||
| 	} | ||||
|  | ||||
| 	// set NOTICE event | ||||
| 	if event.Command == "NOTICE" { | ||||
| 		rmsg.Event = config.EventNoticeIRC | ||||
| 	} | ||||
|  | ||||
| 	// strip action, we made an event if it was an action | ||||
| 	rmsg.Text += event.StripAction() | ||||
|  | ||||
| 	// strip IRC colors | ||||
| 	re := regexp.MustCompile(`\x03(?:\d{1,2}(?:,\d{1,2})?)?|[[:cntrl:]]`) | ||||
| 	rmsg.Text = re.ReplaceAllString(rmsg.Text, "") | ||||
|  | ||||
| 	// start detecting the charset | ||||
| 	mycharset := b.GetString("Charset") | ||||
| 	if mycharset == "" { | ||||
| @@ -201,7 +240,7 @@ func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||
| 	} | ||||
| 	switch mycharset { | ||||
| 	case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 		rmsg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), rmsg.Text) | ||||
| 		rmsg.Text = toUTF8(b.GetString("Charset"), rmsg.Text) | ||||
| 	default: | ||||
| 		r, err := charset.NewReader(mycharset, strings.NewReader(rmsg.Text)) | ||||
| 		if err != nil { | ||||
| @@ -218,6 +257,7 @@ func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||
|  | ||||
| func (b *Birc) handleRunCommands() { | ||||
| 	for _, cmd := range b.GetStringSlice("RunCommands") { | ||||
| 		cmd = strings.ReplaceAll(cmd, "{BOTNICK}", b.Nick) | ||||
| 		if err := b.i.Cmd.SendRaw(cmd); err != nil { | ||||
| 			b.Log.Errorf("RunCommands %s failed: %s", cmd, err) | ||||
| 		} | ||||
|   | ||||
| @@ -2,8 +2,10 @@ package birc | ||||
|  | ||||
| import ( | ||||
| 	"crypto/tls" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"hash/crc32" | ||||
| 	"io/ioutil" | ||||
| 	"net" | ||||
| 	"sort" | ||||
| 	"strconv" | ||||
| @@ -14,6 +16,7 @@ import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/lrstanley/girc" | ||||
| 	stripmd "github.com/writeas/go-strip-markdown" | ||||
|  | ||||
| 	// We need to import the 'data' package as an implicit dependency. | ||||
| 	// See: https://godoc.org/github.com/paulrosania/go-charset/charset | ||||
| @@ -28,6 +31,7 @@ type Birc struct { | ||||
| 	Local                                     chan config.Message // local queue for flood control | ||||
| 	FirstConnection, authDone                 bool | ||||
| 	MessageDelay, MessageQueue, MessageLength int | ||||
| 	channels                                  map[string]bool | ||||
|  | ||||
| 	*bridge.Config | ||||
| } | ||||
| @@ -38,6 +42,8 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b.Nick = b.GetString("Nick") | ||||
| 	b.names = make(map[string][]string) | ||||
| 	b.connected = make(chan error) | ||||
| 	b.channels = make(map[string]bool) | ||||
|  | ||||
| 	if b.GetInt("MessageDelay") == 0 { | ||||
| 		b.MessageDelay = 1300 | ||||
| 	} else { | ||||
| @@ -67,6 +73,10 @@ func (b *Birc) Command(msg *config.Message) string { | ||||
| } | ||||
|  | ||||
| func (b *Birc) Connect() error { | ||||
| 	if b.GetBool("UseSASL") && b.GetString("TLSClientCertificate") != "" { | ||||
| 		return errors.New("you can't enable SASL and TLSClientCertificate at the same time") | ||||
| 	} | ||||
|  | ||||
| 	b.Local = make(chan config.Message, b.MessageQueue+10) | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
|  | ||||
| @@ -110,6 +120,7 @@ func (b *Birc) Disconnect() error { | ||||
| } | ||||
|  | ||||
| func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	b.channels[channel.Name] = true | ||||
| 	// need to check if we have nickserv auth done before joining channels | ||||
| 	for { | ||||
| 		if b.authDone { | ||||
| @@ -137,6 +148,7 @@ func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	// we can be in between reconnects #385 | ||||
| 	if !b.i.IsConnected() { | ||||
| 		b.Log.Error("Not connected to server, dropping message") | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Execute a command | ||||
| @@ -155,10 +167,14 @@ func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	var msgLines []string | ||||
| 	if b.GetBool("StripMarkdown") { | ||||
| 		msg.Text = stripmd.Strip(msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("MessageSplit") { | ||||
| 		msgLines = helper.GetSubLines(msg.Text, b.MessageLength) | ||||
| 		msgLines = helper.GetSubLines(msg.Text, b.MessageLength, b.GetString("MessageClipped")) | ||||
| 	} else { | ||||
| 		msgLines = helper.GetSubLines(msg.Text, 0) | ||||
| 		msgLines = helper.GetSubLines(msg.Text, 0, b.GetString("MessageClipped")) | ||||
| 	} | ||||
| 	for i := range msgLines { | ||||
| 		if len(b.Local) >= b.MessageQueue { | ||||
| @@ -166,12 +182,8 @@ func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		b.Local <- config.Message{ | ||||
| 			Text:     msgLines[i], | ||||
| 			Username: msg.Username, | ||||
| 			Channel:  msg.Channel, | ||||
| 			Event:    msg.Event, | ||||
| 		} | ||||
| 		msg.Text = msgLines[i] | ||||
| 		b.Local <- msg | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
| @@ -198,22 +210,58 @@ func (b *Birc) doConnect() { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Sanitize nicks for RELAYMSG: replace IRC characters with special meanings with "-" | ||||
| func sanitizeNick(nick string) string { | ||||
| 	sanitize := func(r rune) rune { | ||||
| 		if strings.ContainsRune("!+%@&#$:'\"?*,. ", r) { | ||||
| 			return '-' | ||||
| 		} | ||||
| 		return r | ||||
| 	} | ||||
| 	return strings.Map(sanitize, nick) | ||||
| } | ||||
|  | ||||
| func (b *Birc) doSend() { | ||||
| 	rate := time.Millisecond * time.Duration(b.MessageDelay) | ||||
| 	throttle := time.NewTicker(rate) | ||||
| 	for msg := range b.Local { | ||||
| 		<-throttle.C | ||||
| 		username := msg.Username | ||||
| 		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) | ||||
| 		// Optional support for the proposed RELAYMSG extension, described at | ||||
| 		// https://github.com/jlu5/ircv3-specifications/blob/master/extensions/relaymsg.md | ||||
| 		// nolint:nestif | ||||
| 		if (b.i.HasCapability("overdrivenetworks.com/relaymsg") || b.i.HasCapability("draft/relaymsg")) && | ||||
| 			b.GetBool("UseRelayMsg") { | ||||
| 			username = sanitizeNick(username) | ||||
| 			text := msg.Text | ||||
|  | ||||
| 			// Work around girc chomping leading commas on single word messages? | ||||
| 			if strings.HasPrefix(text, ":") && !strings.ContainsRune(text, ' ') { | ||||
| 				text = ":" + text | ||||
| 			} | ||||
|  | ||||
| 			if msg.Event == config.EventUserAction { | ||||
| 				b.i.Cmd.SendRawf("RELAYMSG %s %s :\x01ACTION %s\x01", msg.Channel, username, text) //nolint:errcheck | ||||
| 			} else { | ||||
| 				b.Log.Debugf("Sending RELAYMSG to channel %s: nick=%s", msg.Channel, username) | ||||
| 				b.i.Cmd.SendRawf("RELAYMSG %s %s :%s", msg.Channel, username, text) //nolint:errcheck | ||||
| 			} | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Sending to channel %s", msg.Channel) | ||||
| 			b.i.Cmd.Message(msg.Channel, 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) | ||||
| 			} | ||||
| 			switch msg.Event { | ||||
| 			case config.EventUserAction: | ||||
| 				b.i.Cmd.Action(msg.Channel, username+msg.Text) | ||||
| 			case config.EventNoticeIRC: | ||||
| 				b.Log.Debugf("Sending notice to channel %s", msg.Channel) | ||||
| 				b.i.Cmd.Notice(msg.Channel, username+msg.Text) | ||||
| 			default: | ||||
| 				b.Log.Debugf("Sending to channel %s", msg.Channel) | ||||
| 				b.i.Cmd.Message(msg.Channel, username+msg.Text) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -228,15 +276,39 @@ func (b *Birc) getClient() (*girc.Client, error) { | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	user := b.GetString("UserName") | ||||
| 	if user == "" { | ||||
| 		user = b.GetString("Nick") | ||||
| 	} | ||||
| 	// fix strict user handling of girc | ||||
| 	user := b.GetString("Nick") | ||||
| 	for !girc.IsValidUser(user) { | ||||
| 		if len(user) == 1 { | ||||
| 		if len(user) == 1 || len(user) == 0 { | ||||
| 			user = "matterbridge" | ||||
| 			break | ||||
| 		} | ||||
| 		user = user[1:] | ||||
| 	} | ||||
| 	realName := b.GetString("RealName") | ||||
| 	if realName == "" { | ||||
| 		realName = b.GetString("Nick") | ||||
| 	} | ||||
|  | ||||
| 	debug := ioutil.Discard | ||||
| 	if b.GetInt("DebugLevel") == 2 { | ||||
| 		debug = b.Log.Writer() | ||||
| 	} | ||||
|  | ||||
| 	pingDelay, err := time.ParseDuration(b.GetString("pingdelay")) | ||||
| 	if err != nil || pingDelay == 0 { | ||||
| 		pingDelay = time.Minute | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("setting pingdelay to %s", pingDelay) | ||||
|  | ||||
| 	tlsConfig, err := b.getTLSConfig() | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	i := girc.New(girc.Config{ | ||||
| 		Server:     server, | ||||
| @@ -244,10 +316,15 @@ func (b *Birc) getClient() (*girc.Client, error) { | ||||
| 		Port:       port, | ||||
| 		Nick:       b.GetString("Nick"), | ||||
| 		User:       user, | ||||
| 		Name:       b.GetString("Nick"), | ||||
| 		Name:       realName, | ||||
| 		SSL:        b.GetBool("UseTLS"), | ||||
| 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, //nolint:gosec | ||||
| 		PingDelay:  time.Minute, | ||||
| 		Bind:       b.GetString("Bind"), | ||||
| 		TLSConfig:  tlsConfig, | ||||
| 		PingDelay:  pingDelay, | ||||
| 		// skip gIRC internal rate limiting, since we have our own throttling | ||||
| 		AllowFlood:    true, | ||||
| 		Debug:         debug, | ||||
| 		SupportedCaps: map[string][]string{"overdrivenetworks.com/relaymsg": nil, "draft/relaymsg": nil}, | ||||
| 	}) | ||||
| 	return i, nil | ||||
| } | ||||
| @@ -257,12 +334,16 @@ func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||
| 	sort.Strings(b.names[channel]) | ||||
| 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | ||||
| 	for len(b.names[channel]) > maxNamesPerPost { | ||||
| 		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]), | ||||
| 			Channel: channel, Account: b.Account} | ||||
| 		b.Remote <- config.Message{ | ||||
| 			Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]), | ||||
| 			Channel: channel, Account: b.Account, | ||||
| 		} | ||||
| 		b.names[channel] = b.names[channel][maxNamesPerPost:] | ||||
| 	} | ||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel]), | ||||
| 		Channel: channel, Account: b.Account} | ||||
| 	b.Remote <- config.Message{ | ||||
| 		Username: b.Nick, Text: b.formatnicks(b.names[channel]), | ||||
| 		Channel: channel, Account: b.Account, | ||||
| 	} | ||||
| 	b.names[channel] = nil | ||||
| 	b.i.Handlers.Clear(girc.RPL_NAMREPLY) | ||||
| 	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES) | ||||
| @@ -273,7 +354,7 @@ func (b *Birc) skipPrivMsg(event girc.Event) bool { | ||||
| 	b.Nick = b.i.GetNick() | ||||
|  | ||||
| 	// freenode doesn't send 001 as first reply | ||||
| 	if event.Command == "NOTICE" { | ||||
| 	if event.Command == "NOTICE" && len(event.Params) != 2 { | ||||
| 		return true | ||||
| 	} | ||||
| 	// don't forward queries to the bot | ||||
| @@ -281,7 +362,18 @@ func (b *Birc) skipPrivMsg(event girc.Event) bool { | ||||
| 		return true | ||||
| 	} | ||||
| 	// don't forward message from ourself | ||||
| 	if event.Source.Name == b.Nick { | ||||
| 	if event.Source != nil { | ||||
| 		if event.Source.Name == b.Nick { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	// don't forward messages we sent via RELAYMSG | ||||
| 	if relayedNick, ok := event.Tags.Get("draft/relaymsg"); ok && relayedNick == b.Nick { | ||||
| 		return true | ||||
| 	} | ||||
| 	// This is the old name of the cap sent in spoofed messages; I've kept this in | ||||
| 	// for compatibility reasons | ||||
| 	if relayedNick, ok := event.Tags.Get("relaymsg"); ok && relayedNick == b.Nick { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| @@ -301,3 +393,23 @@ func (b *Birc) storeNames(client *girc.Client, event girc.Event) { | ||||
| func (b *Birc) formatnicks(nicks []string) string { | ||||
| 	return strings.Join(nicks, ", ") + " currently on IRC" | ||||
| } | ||||
|  | ||||
| func (b *Birc) getTLSConfig() (*tls.Config, error) { | ||||
| 	server, _, _ := net.SplitHostPort(b.GetString("server")) | ||||
|  | ||||
| 	tlsConfig := &tls.Config{ | ||||
| 		InsecureSkipVerify: b.GetBool("skiptlsverify"), //nolint:gosec | ||||
| 		ServerName:         server, | ||||
| 	} | ||||
|  | ||||
| 	if filename := b.GetString("TLSClientCertificate"); filename != "" { | ||||
| 		cert, err := tls.LoadX509KeyPair(filename, filename) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
|  | ||||
| 		tlsConfig.Certificates = []tls.Certificate{cert} | ||||
| 	} | ||||
|  | ||||
| 	return tlsConfig, nil | ||||
| } | ||||
|   | ||||
							
								
								
									
										59
									
								
								bridge/keybase/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								bridge/keybase/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,59 @@ | ||||
| package bkeybase | ||||
|  | ||||
| import ( | ||||
| 	"strconv" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/keybase/go-keybase-chat-bot/kbchat/types/chat1" | ||||
| ) | ||||
|  | ||||
| func (b *Bkeybase) handleKeybase() { | ||||
| 	sub, err := b.kbc.ListenForNewTextMessages() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Error listening: %s", err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	go func() { | ||||
| 		for { | ||||
| 			msg, err := sub.Read() | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("failed to read message: %s", err.Error()) | ||||
| 			} | ||||
|  | ||||
| 			if msg.Message.Content.TypeName != "text" { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			if msg.Message.Sender.Username == b.kbc.GetUsername() { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			b.handleMessage(msg.Message) | ||||
|  | ||||
| 		} | ||||
| 	}() | ||||
| } | ||||
|  | ||||
| func (b *Bkeybase) handleMessage(msg chat1.MsgSummary) { | ||||
| 	b.Log.Debugf("== Receiving event: %#v", msg) | ||||
| 	if msg.Channel.TopicName != b.channel || msg.Channel.Name != b.team { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if msg.Sender.Username != b.kbc.GetUsername() { | ||||
|  | ||||
| 		// TODO download avatar | ||||
|  | ||||
| 		// Create our message | ||||
| 		rmsg := config.Message{Username: msg.Sender.Username, Text: msg.Content.Text.Body, UserID: string(msg.Sender.Uid), Channel: msg.Channel.TopicName, ID: strconv.Itoa(int(msg.Id)), Account: b.Account} | ||||
|  | ||||
| 		// Text must be a string | ||||
| 		if msg.Content.TypeName != "text" { | ||||
| 			b.Log.Errorf("message is not text") | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		b.Log.Debugf("<= Sending message from %s on %s to gateway", msg.Sender.Username, msg.Channel.Name) | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										106
									
								
								bridge/keybase/keybase.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										106
									
								
								bridge/keybase/keybase.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,106 @@ | ||||
| package bkeybase | ||||
|  | ||||
| import ( | ||||
| 	"io/ioutil" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"strconv" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/keybase/go-keybase-chat-bot/kbchat" | ||||
| ) | ||||
|  | ||||
| // Bkeybase bridge structure | ||||
| type Bkeybase struct { | ||||
| 	kbc     *kbchat.API | ||||
| 	user    string | ||||
| 	channel string | ||||
| 	team    string | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| // New initializes Bkeybase object and sets team | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bkeybase{Config: cfg} | ||||
| 	b.team = b.Config.GetString("Team") | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| // Connect starts keybase API and listener loop | ||||
| func (b *Bkeybase) Connect() error { | ||||
| 	var err error | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Team")) | ||||
|  | ||||
| 	// use default keybase location (`keybase`) | ||||
| 	b.kbc, err = kbchat.Start(kbchat.RunOptions{}) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.user = b.kbc.GetUsername() | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	go b.handleKeybase() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Disconnect doesn't do anything for now | ||||
| func (b *Bkeybase) Disconnect() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // JoinChannel sets channel name in struct | ||||
| func (b *Bkeybase) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if _, err := b.kbc.JoinChannel(b.team, channel.Name); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.channel = channel.Name | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Send receives bridge messages and sends them to Keybase chat room | ||||
| func (b *Bkeybase) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// Handle /me events | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "_" + msg.Text + "_" | ||||
| 	} | ||||
|  | ||||
| 	// Delete message if we have an ID | ||||
| 	// Delete message not supported by keybase go library yet | ||||
|  | ||||
| 	// Edit message if we have an ID | ||||
| 	// kbchat lib does not support message editing yet | ||||
|  | ||||
| 	if len(msg.Extra["file"]) > 0 { | ||||
| 		// Upload a file | ||||
| 		dir, err := ioutil.TempDir("", "matterbridge") | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		defer os.RemoveAll(dir) | ||||
|  | ||||
| 		for _, f := range msg.Extra["file"] { | ||||
| 			fname := f.(config.FileInfo).Name | ||||
| 			fdata := *f.(config.FileInfo).Data | ||||
| 			fcaption := f.(config.FileInfo).Comment | ||||
| 			fpath := filepath.Join(dir, fname) | ||||
|  | ||||
| 			if err = ioutil.WriteFile(fpath, fdata, 0600); err != nil { | ||||
| 				return "", err | ||||
| 			} | ||||
|  | ||||
| 			_, _ = b.kbc.SendAttachmentByTeam(b.team, &b.channel, fpath, fcaption) | ||||
| 		} | ||||
|  | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Send regular message | ||||
| 	text := msg.Username + msg.Text | ||||
| 	resp, err := b.kbc.SendMessageByTeamName(b.team, &b.channel, text) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return strconv.Itoa(int(*resp.Result.MessageID)), err | ||||
| } | ||||
							
								
								
									
										215
									
								
								bridge/matrix/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										215
									
								
								bridge/matrix/helpers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,215 @@ | ||||
| package bmatrix | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	matrix "github.com/matterbridge/gomatrix" | ||||
| ) | ||||
|  | ||||
| func newMatrixUsername(username string) *matrixUsername { | ||||
| 	mUsername := new(matrixUsername) | ||||
|  | ||||
| 	// check if we have a </tag>. if we have, we don't escape HTML. #696 | ||||
| 	if htmlTag.MatchString(username) { | ||||
| 		mUsername.formatted = username | ||||
| 		// remove the HTML formatting for beautiful push messages #1188 | ||||
| 		mUsername.plain = htmlReplacementTag.ReplaceAllString(username, "") | ||||
| 	} else { | ||||
| 		mUsername.formatted = html.EscapeString(username) | ||||
| 		mUsername.plain = username | ||||
| 	} | ||||
|  | ||||
| 	return mUsername | ||||
| } | ||||
|  | ||||
| // getRoomID retrieves a matching room ID from the channel name. | ||||
| func (b *Bmatrix) getRoomID(channel string) string { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	for ID, name := range b.RoomMap { | ||||
| 		if name == channel { | ||||
| 			return ID | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| // interface2Struct marshals and immediately unmarshals an interface. | ||||
| // Useful for converting map[string]interface{} to a struct. | ||||
| func interface2Struct(in interface{}, out interface{}) error { | ||||
| 	jsonObj, err := json.Marshal(in) | ||||
| 	if err != nil { | ||||
| 		return err //nolint:wrapcheck | ||||
| 	} | ||||
|  | ||||
| 	return json.Unmarshal(jsonObj, out) | ||||
| } | ||||
|  | ||||
| // getDisplayName retrieves the displayName for mxid, querying the homeserver if the mxid is not in the cache. | ||||
| func (b *Bmatrix) getDisplayName(mxid string) string { | ||||
| 	if b.GetBool("UseUserName") { | ||||
| 		return mxid[1:] | ||||
| 	} | ||||
|  | ||||
| 	b.RLock() | ||||
| 	if val, present := b.NicknameMap[mxid]; present { | ||||
| 		b.RUnlock() | ||||
|  | ||||
| 		return val.displayName | ||||
| 	} | ||||
| 	b.RUnlock() | ||||
|  | ||||
| 	displayName, err := b.mc.GetDisplayName(mxid) | ||||
| 	var httpError *matrix.HTTPError | ||||
| 	if errors.As(err, &httpError) { | ||||
| 		b.Log.Warnf("Couldn't retrieve the display name for %s", mxid) | ||||
| 	} | ||||
|  | ||||
| 	if err != nil { | ||||
| 		return b.cacheDisplayName(mxid, mxid[1:]) | ||||
| 	} | ||||
|  | ||||
| 	return b.cacheDisplayName(mxid, displayName.DisplayName) | ||||
| } | ||||
|  | ||||
| // cacheDisplayName stores the mapping between a mxid and a display name, to be reused later without performing a query to the homserver. | ||||
| // Note that old entries are cleaned when this function is called. | ||||
| func (b *Bmatrix) cacheDisplayName(mxid string, displayName string) string { | ||||
| 	now := time.Now() | ||||
|  | ||||
| 	// scan to delete old entries, to stop memory usage from becoming too high with old entries. | ||||
| 	// In addition, we also detect if another user have the same username, and if so, we append their mxids to their usernames to differentiate them. | ||||
| 	toDelete := []string{} | ||||
| 	conflict := false | ||||
|  | ||||
| 	b.Lock() | ||||
| 	for mxid, v := range b.NicknameMap { | ||||
| 		// to prevent username reuse across matrix servers - or even on the same server, append | ||||
| 		// the mxid to the username when there is a conflict | ||||
| 		if v.displayName == displayName { | ||||
| 			conflict = true | ||||
| 			// TODO: it would be nice to be able to rename previous messages from this user. | ||||
| 			// The current behavior is that only users with clashing usernames and *that have spoken since the bridge last started* will get their mxids shown, and I don't know if that's the expected behavior. | ||||
| 			v.displayName = fmt.Sprintf("%s (%s)", displayName, mxid) | ||||
| 			b.NicknameMap[mxid] = v | ||||
| 		} | ||||
|  | ||||
| 		if now.Sub(v.lastUpdated) > 10*time.Minute { | ||||
| 			toDelete = append(toDelete, mxid) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if conflict { | ||||
| 		displayName = fmt.Sprintf("%s (%s)", displayName, mxid) | ||||
| 	} | ||||
|  | ||||
| 	for _, v := range toDelete { | ||||
| 		delete(b.NicknameMap, v) | ||||
| 	} | ||||
|  | ||||
| 	b.NicknameMap[mxid] = NicknameCacheEntry{ | ||||
| 		displayName: displayName, | ||||
| 		lastUpdated: now, | ||||
| 	} | ||||
| 	b.Unlock() | ||||
|  | ||||
| 	return displayName | ||||
| } | ||||
|  | ||||
| // handleError converts errors into httpError. | ||||
| //nolint:exhaustivestruct | ||||
| func handleError(err error) *httpError { | ||||
| 	var mErr matrix.HTTPError | ||||
| 	if !errors.As(err, &mErr) { | ||||
| 		return &httpError{ | ||||
| 			Err: "not a HTTPError", | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	var httpErr httpError | ||||
|  | ||||
| 	if err := json.Unmarshal(mErr.Contents, &httpErr); err != nil { | ||||
| 		return &httpError{ | ||||
| 			Err: "unmarshal failed", | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return &httpErr | ||||
| } | ||||
|  | ||||
| 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 | ||||
| } | ||||
|  | ||||
| // getAvatarURL returns the avatar URL of the specified sender. | ||||
| func (b *Bmatrix) getAvatarURL(sender string) string { | ||||
| 	urlPath := b.mc.BuildURL("profile", sender, "avatar_url") | ||||
|  | ||||
| 	s := struct { | ||||
| 		AvatarURL string `json:"avatar_url"` | ||||
| 	}{} | ||||
|  | ||||
| 	err := b.mc.MakeRequest("GET", urlPath, nil, &s) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("getAvatarURL failed: %s", err) | ||||
|  | ||||
| 		return "" | ||||
| 	} | ||||
|  | ||||
| 	url := strings.ReplaceAll(s.AvatarURL, "mxc://", b.GetString("Server")+"/_matrix/media/r0/thumbnail/") | ||||
| 	if url != "" { | ||||
| 		url += "?width=37&height=37&method=crop" | ||||
| 	} | ||||
|  | ||||
| 	return url | ||||
| } | ||||
|  | ||||
| // handleRatelimit handles the ratelimit errors and return if we're ratelimited and the amount of time to sleep | ||||
| func (b *Bmatrix) handleRatelimit(err error) (time.Duration, bool) { | ||||
| 	httpErr := handleError(err) | ||||
| 	if httpErr.Errcode != "M_LIMIT_EXCEEDED" { | ||||
| 		return 0, false | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("ratelimited: %s", httpErr.Err) | ||||
| 	b.Log.Infof("getting ratelimited by matrix, sleeping approx %d seconds before retrying", httpErr.RetryAfterMs/1000) | ||||
|  | ||||
| 	return time.Duration(httpErr.RetryAfterMs) * time.Millisecond, true | ||||
| } | ||||
|  | ||||
| // retry function will check if we're ratelimited and retries again when backoff time expired | ||||
| // returns original error if not 429 ratelimit | ||||
| func (b *Bmatrix) retry(f func() error) error { | ||||
| 	b.rateMutex.Lock() | ||||
| 	defer b.rateMutex.Unlock() | ||||
|  | ||||
| 	for { | ||||
| 		if err := f(); err != nil { | ||||
| 			if backoff, ok := b.handleRatelimit(err); ok { | ||||
| 				time.Sleep(backoff) | ||||
| 			} else { | ||||
| 				return err | ||||
| 			} | ||||
| 		} else { | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -3,11 +3,11 @@ package bmatrix | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"mime" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| @@ -15,40 +15,108 @@ import ( | ||||
| 	matrix "github.com/matterbridge/gomatrix" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	htmlTag            = regexp.MustCompile("</.*?>") | ||||
| 	htmlReplacementTag = regexp.MustCompile("<[^>]*>") | ||||
| ) | ||||
|  | ||||
| type NicknameCacheEntry struct { | ||||
| 	displayName string | ||||
| 	lastUpdated time.Time | ||||
| } | ||||
|  | ||||
| type Bmatrix struct { | ||||
| 	mc      *matrix.Client | ||||
| 	UserID  string | ||||
| 	RoomMap map[string]string | ||||
| 	mc          *matrix.Client | ||||
| 	UserID      string | ||||
| 	NicknameMap map[string]NicknameCacheEntry | ||||
| 	RoomMap     map[string]string | ||||
| 	rateMutex   sync.RWMutex | ||||
| 	sync.RWMutex | ||||
| 	htmlTag *regexp.Regexp | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| type httpError struct { | ||||
| 	Errcode      string `json:"errcode"` | ||||
| 	Err          string `json:"error"` | ||||
| 	RetryAfterMs int    `json:"retry_after_ms"` | ||||
| } | ||||
|  | ||||
| type matrixUsername struct { | ||||
| 	plain     string | ||||
| 	formatted string | ||||
| } | ||||
|  | ||||
| // SubTextMessage represents the new content of the message in edit messages. | ||||
| type SubTextMessage struct { | ||||
| 	MsgType       string `json:"msgtype"` | ||||
| 	Body          string `json:"body"` | ||||
| 	FormattedBody string `json:"formatted_body,omitempty"` | ||||
| 	Format        string `json:"format,omitempty"` | ||||
| } | ||||
|  | ||||
| // MessageRelation explains how the current message relates to a previous message. | ||||
| // Notably used for message edits. | ||||
| type MessageRelation struct { | ||||
| 	EventID string `json:"event_id"` | ||||
| 	Type    string `json:"rel_type"` | ||||
| } | ||||
|  | ||||
| type EditedMessage struct { | ||||
| 	NewContent SubTextMessage  `json:"m.new_content"` | ||||
| 	RelatedTo  MessageRelation `json:"m.relates_to"` | ||||
| 	matrix.TextMessage | ||||
| } | ||||
|  | ||||
| type InReplyToRelationContent struct { | ||||
| 	EventID string `json:"event_id"` | ||||
| } | ||||
|  | ||||
| type InReplyToRelation struct { | ||||
| 	InReplyTo InReplyToRelationContent `json:"m.in_reply_to"` | ||||
| } | ||||
|  | ||||
| type ReplyMessage struct { | ||||
| 	RelatedTo InReplyToRelation `json:"m.relates_to"` | ||||
| 	matrix.TextMessage | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bmatrix{Config: cfg} | ||||
| 	b.htmlTag = regexp.MustCompile("</.*?>") | ||||
| 	b.RoomMap = make(map[string]string) | ||||
| 	b.NicknameMap = make(map[string]NicknameCacheEntry) | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) Connect() error { | ||||
| 	var err error | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
| 	b.mc, err = matrix.NewClient(b.GetString("Server"), "", "") | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	if b.GetString("MxID") != "" && b.GetString("Token") != "" { | ||||
| 		b.mc, err = matrix.NewClient( | ||||
| 			b.GetString("Server"), b.GetString("MxID"), b.GetString("Token"), | ||||
| 		) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		b.UserID = b.GetString("MxID") | ||||
| 		b.Log.Info("Using existing Matrix credentials") | ||||
| 	} else { | ||||
| 		b.mc, err = matrix.NewClient(b.GetString("Server"), "", "") | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		resp, err := b.mc.Login(&matrix.ReqLogin{ | ||||
| 			Type:       "m.login.password", | ||||
| 			User:       b.GetString("Login"), | ||||
| 			Password:   b.GetString("Password"), | ||||
| 			Identifier: matrix.NewUserIdentifier(b.GetString("Login")), | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		b.mc.SetCredentials(resp.UserID, resp.AccessToken) | ||||
| 		b.UserID = resp.UserID | ||||
| 		b.Log.Info("Connection succeeded") | ||||
| 	} | ||||
| 	resp, err := b.mc.Login(&matrix.ReqLogin{ | ||||
| 		Type:     "m.login.password", | ||||
| 		User:     b.GetString("Login"), | ||||
| 		Password: b.GetString("Password"), | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.mc.SetCredentials(resp.UserID, resp.AccessToken) | ||||
| 	b.UserID = resp.UserID | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	go b.handlematrix() | ||||
| 	return nil | ||||
| } | ||||
| @@ -58,14 +126,18 @@ func (b *Bmatrix) Disconnect() error { | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	resp, err := b.mc.JoinRoom(channel.Name, "", nil) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Lock() | ||||
| 	b.RoomMap[resp.RoomID] = channel.Name | ||||
| 	b.Unlock() | ||||
| 	return err | ||||
| 	return b.retry(func() error { | ||||
| 		resp, err := b.mc.JoinRoom(channel.Name, "", nil) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
|  | ||||
| 		b.Lock() | ||||
| 		b.RoomMap[resp.RoomID] = channel.Name | ||||
| 		b.Unlock() | ||||
|  | ||||
| 		return nil | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| @@ -74,17 +146,61 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 	channel := b.getRoomID(msg.Channel) | ||||
| 	b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel) | ||||
|  | ||||
| 	username := newMatrixUsername(msg.Username) | ||||
|  | ||||
| 	body := username.plain + msg.Text | ||||
| 	formattedBody := username.formatted + helper.ParseMarkdown(msg.Text) | ||||
|  | ||||
| 	if b.GetBool("SpoofUsername") { | ||||
| 		// https://spec.matrix.org/v1.3/client-server-api/#mroommember | ||||
| 		type stateMember struct { | ||||
| 			AvatarURL   string `json:"avatar_url,omitempty"` | ||||
| 			DisplayName string `json:"displayname"` | ||||
| 			Membership  string `json:"membership"` | ||||
| 		} | ||||
|  | ||||
| 		// TODO: reset username afterwards with DisplayName: null ? | ||||
| 		m := stateMember{ | ||||
| 			AvatarURL:   "", | ||||
| 			DisplayName: username.plain, | ||||
| 			Membership:  "join", | ||||
| 		} | ||||
|  | ||||
| 		_, err := b.mc.SendStateEvent(channel, "m.room.member", b.UserID, m) | ||||
| 		if err == nil { | ||||
| 			body = msg.Text | ||||
| 			formattedBody = helper.ParseMarkdown(msg.Text) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		m := matrix.TextMessage{ | ||||
| 			MsgType: "m.emote", | ||||
| 			Body:    msg.Username + msg.Text, | ||||
| 			MsgType:       "m.emote", | ||||
| 			Body:          body, | ||||
| 			FormattedBody: formattedBody, | ||||
| 			Format:        "org.matrix.custom.html", | ||||
| 		} | ||||
| 		resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
|  | ||||
| 		if b.GetBool("HTMLDisable") { | ||||
| 			m.Format = "" | ||||
| 			m.FormattedBody = "" | ||||
| 		} | ||||
| 		return resp.EventID, err | ||||
|  | ||||
| 		msgID := "" | ||||
|  | ||||
| 		err := b.retry(func() error { | ||||
| 			resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
|  | ||||
| 			msgID = resp.EventID | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
|  | ||||
| 		return msgID, err | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| @@ -92,17 +208,34 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{}) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return resp.EventID, err | ||||
|  | ||||
| 		msgID := "" | ||||
|  | ||||
| 		err := b.retry(func() error { | ||||
| 			resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{}) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
|  | ||||
| 			msgID = resp.EventID | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
|  | ||||
| 		return msgID, err | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			if _, err := b.mc.SendText(channel, rmsg.Username+rmsg.Text); err != nil { | ||||
| 			rmsg := rmsg | ||||
|  | ||||
| 			err := b.retry(func() error { | ||||
| 				_, err := b.mc.SendText(channel, rmsg.Username+rmsg.Text) | ||||
|  | ||||
| 				return err | ||||
| 			}) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("sendText failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
| @@ -113,47 +246,162 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Edit message if we have an ID | ||||
| 	// matrix has no editing support | ||||
| 	if msg.ID != "" { | ||||
| 		rmsg := EditedMessage{ | ||||
| 			TextMessage: matrix.TextMessage{ | ||||
| 				Body:          body, | ||||
| 				MsgType:       "m.text", | ||||
| 				Format:        "org.matrix.custom.html", | ||||
| 				FormattedBody: formattedBody, | ||||
| 			}, | ||||
| 		} | ||||
|  | ||||
| 	// Use notices to send join/leave events | ||||
| 	if msg.Event == config.EventJoinLeave { | ||||
| 		resp, err := b.mc.SendNotice(channel, msg.Username+msg.Text) | ||||
| 		rmsg.NewContent = SubTextMessage{ | ||||
| 			Body:          rmsg.TextMessage.Body, | ||||
| 			FormattedBody: rmsg.TextMessage.FormattedBody, | ||||
| 			Format:        rmsg.TextMessage.Format, | ||||
| 			MsgType:       "m.text", | ||||
| 		} | ||||
|  | ||||
| 		if b.GetBool("HTMLDisable") { | ||||
| 			rmsg.TextMessage.Format = "" | ||||
| 			rmsg.TextMessage.FormattedBody = "" | ||||
| 			rmsg.NewContent.Format = "" | ||||
| 			rmsg.NewContent.FormattedBody = "" | ||||
| 		} | ||||
|  | ||||
| 		rmsg.RelatedTo = MessageRelation{ | ||||
| 			EventID: msg.ID, | ||||
| 			Type:    "m.replace", | ||||
| 		} | ||||
|  | ||||
| 		err := b.retry(func() error { | ||||
| 			_, err := b.mc.SendMessageEvent(channel, "m.room.message", rmsg) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return msg.ID, nil | ||||
| 	} | ||||
|  | ||||
| 	// Use notices to send join/leave events | ||||
| 	if msg.Event == config.EventJoinLeave { | ||||
| 		m := matrix.TextMessage{ | ||||
| 			MsgType:       "m.notice", | ||||
| 			Body:          body, | ||||
| 			FormattedBody: formattedBody, | ||||
| 			Format:        "org.matrix.custom.html", | ||||
| 		} | ||||
|  | ||||
| 		if b.GetBool("HTMLDisable") { | ||||
| 			m.Format = "" | ||||
| 			m.FormattedBody = "" | ||||
| 		} | ||||
|  | ||||
| 		var ( | ||||
| 			resp *matrix.RespSendEvent | ||||
| 			err  error | ||||
| 		) | ||||
|  | ||||
| 		err = b.retry(func() error { | ||||
| 			resp, err = b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return resp.EventID, err | ||||
| 	} | ||||
|  | ||||
| 	username := html.EscapeString(msg.Username) | ||||
| 	// check if we have a </tag>. if we have, we don't escape HTML. #696 | ||||
| 	if b.htmlTag.MatchString(msg.Username) { | ||||
| 		username = msg.Username | ||||
| 	if msg.ParentValid() { | ||||
| 		m := ReplyMessage{ | ||||
| 			TextMessage: matrix.TextMessage{ | ||||
| 				MsgType:       "m.text", | ||||
| 				Body:          body, | ||||
| 				FormattedBody: formattedBody, | ||||
| 				Format:        "org.matrix.custom.html", | ||||
| 			}, | ||||
| 		} | ||||
|  | ||||
| 		if b.GetBool("HTMLDisable") { | ||||
| 			m.TextMessage.Format = "" | ||||
| 			m.TextMessage.FormattedBody = "" | ||||
| 		} | ||||
|  | ||||
| 		m.RelatedTo = InReplyToRelation{ | ||||
| 			InReplyTo: InReplyToRelationContent{ | ||||
| 				EventID: msg.ParentID, | ||||
| 			}, | ||||
| 		} | ||||
|  | ||||
| 		var ( | ||||
| 			resp *matrix.RespSendEvent | ||||
| 			err  error | ||||
| 		) | ||||
|  | ||||
| 		err = b.retry(func() error { | ||||
| 			resp, err = b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return resp.EventID, err | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("HTMLDisable") { | ||||
| 		var ( | ||||
| 			resp *matrix.RespSendEvent | ||||
| 			err  error | ||||
| 		) | ||||
|  | ||||
| 		err = b.retry(func() error { | ||||
| 			resp, err = b.mc.SendText(channel, body) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return resp.EventID, err | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message with HTML support (eg riot.im) | ||||
| 	resp, err := b.mc.SendHTML(channel, msg.Username+msg.Text, username+helper.ParseMarkdown(msg.Text)) | ||||
| 	var ( | ||||
| 		resp *matrix.RespSendEvent | ||||
| 		err  error | ||||
| 	) | ||||
|  | ||||
| 	err = b.retry(func() error { | ||||
| 		resp, err = b.mc.SendFormattedText(channel, body, formattedBody) | ||||
|  | ||||
| 		return err | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return resp.EventID, err | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) getRoomID(channel string) string { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	for ID, name := range b.RoomMap { | ||||
| 		if name == channel { | ||||
| 			return ID | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| 	return resp.EventID, err | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handlematrix() { | ||||
| 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||
| 	syncer.OnEventType("m.room.redaction", b.handleEvent) | ||||
| 	syncer.OnEventType("m.room.message", b.handleEvent) | ||||
| 	syncer.OnEventType("m.room.member", b.handleMemberChange) | ||||
| 	go func() { | ||||
| 		for { | ||||
| 			if b == nil { | ||||
| 				return | ||||
| 			} | ||||
| 			if err := b.mc.Sync(); err != nil { | ||||
| 				b.Log.Println("Sync() returned ", err) | ||||
| 			} | ||||
| @@ -161,6 +409,77 @@ func (b *Bmatrix) handlematrix() { | ||||
| 	}() | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handleEdit(ev *matrix.Event, rmsg config.Message) bool { | ||||
| 	relationInterface, present := ev.Content["m.relates_to"] | ||||
| 	newContentInterface, present2 := ev.Content["m.new_content"] | ||||
| 	if !(present && present2) { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	var relation MessageRelation | ||||
| 	if err := interface2Struct(relationInterface, &relation); err != nil { | ||||
| 		b.Log.Warnf("Couldn't parse 'm.relates_to' object with value %#v", relationInterface) | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	var newContent SubTextMessage | ||||
| 	if err := interface2Struct(newContentInterface, &newContent); err != nil { | ||||
| 		b.Log.Warnf("Couldn't parse 'm.new_content' object with value %#v", newContentInterface) | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	if relation.Type != "m.replace" { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	rmsg.ID = relation.EventID | ||||
| 	rmsg.Text = newContent.Body | ||||
| 	b.Remote <- rmsg | ||||
|  | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handleReply(ev *matrix.Event, rmsg config.Message) bool { | ||||
| 	relationInterface, present := ev.Content["m.relates_to"] | ||||
| 	if !present { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	var relation InReplyToRelation | ||||
| 	if err := interface2Struct(relationInterface, &relation); err != nil { | ||||
| 		// probably fine | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	body := rmsg.Text | ||||
|  | ||||
| 	if !b.GetBool("keepquotedreply") { | ||||
| 		for strings.HasPrefix(body, "> ") { | ||||
| 			lineIdx := strings.IndexRune(body, '\n') | ||||
| 			if lineIdx == -1 { | ||||
| 				body = "" | ||||
| 			} else { | ||||
| 				body = body[(lineIdx + 1):] | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	rmsg.Text = body | ||||
| 	rmsg.ParentID = relation.InReplyTo.EventID | ||||
| 	b.Remote <- rmsg | ||||
|  | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handleMemberChange(ev *matrix.Event) { | ||||
| 	// Update the displayname on join messages, according to https://matrix.org/docs/spec/client_server/r0.6.1#events-on-change-of-profile-information | ||||
| 	if ev.Content["membership"] == "join" { | ||||
| 		if dn, ok := ev.Content["displayname"].(string); ok { | ||||
| 			b.cacheDisplayName(ev.Sender, dn) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
| 	b.Log.Debugf("== Receiving event: %#v", ev) | ||||
| 	if ev.Sender != b.UserID { | ||||
| @@ -172,16 +491,14 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
| 			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 | ||||
| 		rmsg := config.Message{ | ||||
| 			Username: b.getDisplayName(ev.Sender), | ||||
| 			Channel:  channel, | ||||
| 			Account:  b.Account, | ||||
| 			UserID:   ev.Sender, | ||||
| 			ID:       ev.ID, | ||||
| 			Avatar:   b.getAvatarURL(ev.Sender), | ||||
| 		} | ||||
|  | ||||
| 		// Remove homeserver suffix if configured | ||||
| @@ -199,11 +516,28 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		// 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 | ||||
| 		} | ||||
|  | ||||
| 		// Do we have a /me action | ||||
| 		if ev.Content["msgtype"].(string) == "m.emote" { | ||||
| 			rmsg.Event = config.EventUserAction | ||||
| 		} | ||||
|  | ||||
| 		// Is it an edit? | ||||
| 		if b.handleEdit(ev, rmsg) { | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		// Is it a reply? | ||||
| 		if b.handleReply(ev, rmsg) { | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		// Do we have attachments | ||||
| 		if b.containsAttachment(ev.Content) { | ||||
| 			err := b.handleDownloadFile(&rmsg, ev.Content) | ||||
| @@ -214,6 +548,11 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
|  | ||||
| 		b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Sender, b.Account) | ||||
| 		b.Remote <- rmsg | ||||
|  | ||||
| 		// not crucial, so no ratelimit check here | ||||
| 		if err := b.mc.MarkRead(ev.RoomID, ev.ID); err != nil { | ||||
| 			b.Log.Errorf("couldn't mark message as read %s", err.Error()) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| @@ -288,26 +627,30 @@ func (b *Bmatrix) handleUploadFiles(msg *config.Message, channel string) (string | ||||
|  | ||||
| // handleUploadFile handles native upload of a file. | ||||
| func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) { | ||||
| 	username := newMatrixUsername(msg.Username) | ||||
| 	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") { | ||||
| 		return | ||||
| 	} | ||||
| 	if fi.Comment != "" { | ||||
| 		_, err := b.mc.SendText(channel, msg.Username+fi.Comment) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file comment failed: %#v", err) | ||||
| 		} | ||||
| 	} else { | ||||
| 		// image and video uploads send no username, we have to do this ourself here #715 | ||||
| 		_, err := b.mc.SendText(channel, msg.Username) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file comment failed: %#v", err) | ||||
| 		} | ||||
| 	// image and video uploads send no username, we have to do this ourself here #715 | ||||
| 	err := b.retry(func() error { | ||||
| 		_, err := b.mc.SendFormattedText(channel, username.plain+fi.Comment, username.formatted+fi.Comment) | ||||
|  | ||||
| 		return err | ||||
| 	}) | ||||
| 	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))) | ||||
|  | ||||
| 	var res *matrix.RespMediaUpload | ||||
|  | ||||
| 	err = b.retry(func() error { | ||||
| 		res, err = b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) | ||||
|  | ||||
| 		return err | ||||
| 	}) | ||||
|  | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		return | ||||
| @@ -316,32 +659,60 @@ func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *conf | ||||
| 	switch { | ||||
| 	case strings.Contains(mtype, "video"): | ||||
| 		b.Log.Debugf("sendVideo %s", res.ContentURI) | ||||
| 		_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) | ||||
| 		err = b.retry(func() error { | ||||
| 			_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendVideo failed: %#v", err) | ||||
| 		} | ||||
| 	case strings.Contains(mtype, "image"): | ||||
| 		b.Log.Debugf("sendImage %s", res.ContentURI) | ||||
| 		_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) | ||||
| 		err = b.retry(func() error { | ||||
| 			_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendImage failed: %#v", err) | ||||
| 		} | ||||
| 	case strings.Contains(mtype, "audio"): | ||||
| 		b.Log.Debugf("sendAudio %s", res.ContentURI) | ||||
| 		err = b.retry(func() error { | ||||
| 			_, err = b.mc.SendMessageEvent(channel, "m.room.message", matrix.AudioMessage{ | ||||
| 				MsgType: "m.audio", | ||||
| 				Body:    fi.Name, | ||||
| 				URL:     res.ContentURI, | ||||
| 				Info: matrix.AudioInfo{ | ||||
| 					Mimetype: mtype, | ||||
| 					Size:     uint(len(*fi.Data)), | ||||
| 				}, | ||||
| 			}) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendAudio failed: %#v", err) | ||||
| 		} | ||||
| 	default: | ||||
| 		b.Log.Debugf("sendFile %s", res.ContentURI) | ||||
| 		err = b.retry(func() error { | ||||
| 			_, err = b.mc.SendMessageEvent(channel, "m.room.message", matrix.FileMessage{ | ||||
| 				MsgType: "m.file", | ||||
| 				Body:    fi.Name, | ||||
| 				URL:     res.ContentURI, | ||||
| 				Info: matrix.FileInfo{ | ||||
| 					Mimetype: mtype, | ||||
| 					Size:     uint(len(*fi.Data)), | ||||
| 				}, | ||||
| 			}) | ||||
|  | ||||
| 			return err | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendFile failed: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Debugf("result: %#v", res) | ||||
| } | ||||
|  | ||||
| // 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 | ||||
| } | ||||
|   | ||||
							
								
								
									
										28
									
								
								bridge/matrix/matrix_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								bridge/matrix/matrix_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | ||||
| package bmatrix | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/stretchr/testify/assert" | ||||
| ) | ||||
|  | ||||
| func TestPlainUsername(t *testing.T) { | ||||
| 	uut := newMatrixUsername("MyUser") | ||||
|  | ||||
| 	assert.Equal(t, "MyUser", uut.formatted) | ||||
| 	assert.Equal(t, "MyUser", uut.plain) | ||||
| } | ||||
|  | ||||
| func TestHTMLUsername(t *testing.T) { | ||||
| 	uut := newMatrixUsername("<b>MyUser</b>") | ||||
|  | ||||
| 	assert.Equal(t, "<b>MyUser</b>", uut.formatted) | ||||
| 	assert.Equal(t, "MyUser", uut.plain) | ||||
| } | ||||
|  | ||||
| func TestFancyUsername(t *testing.T) { | ||||
| 	uut := newMatrixUsername("<MyUser>") | ||||
|  | ||||
| 	assert.Equal(t, "<MyUser>", uut.formatted) | ||||
| 	assert.Equal(t, "<MyUser>", uut.plain) | ||||
| } | ||||
| @@ -1,10 +1,12 @@ | ||||
| package bmattermost | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| 	"github.com/matterbridge/matterclient" | ||||
| 	"github.com/mattermost/mattermost/server/public/model" | ||||
| ) | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| @@ -21,12 +23,17 @@ func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||
| 		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) | ||||
| 		var ( | ||||
| 			data []byte | ||||
| 			err  error | ||||
| 		) | ||||
| 		data, _, err = b.mc.Client.GetProfileImage(context.TODO(), userid, "") | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("ProfileImage download failed for %#v %s", userid, err) | ||||
| 			return | ||||
| 		} | ||||
| 		err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) | ||||
|  | ||||
| 		err = helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) | ||||
| 		if err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 			return | ||||
| @@ -36,20 +43,20 @@ func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| //nolint:wrapcheck | ||||
| 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) | ||||
| 	url, _, _ := b.mc.Client.GetFileLink(context.TODO(), id) | ||||
| 	finfo, _, err := b.mc.Client.GetFileInfo(context.TODO(), id) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, resp := b.mc.Client.DownloadFile(id, true) | ||||
| 	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, _, err := b.mc.Client.DownloadFile(context.TODO(), id, true) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General) | ||||
| 	return nil | ||||
| @@ -66,6 +73,10 @@ func (b *Bmattermost) handleMatter() { | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Choosing login/password based receiving") | ||||
| 		} | ||||
| 		// if for some reason we only want to sent stuff to mattermost but not receive, return | ||||
| 		if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") != "" && b.GetString("Token") == "" && b.GetString("Login") == "" { | ||||
| 			b.Log.Debugf("No WebhookBindAddress specified, only WebhookURL. You will not receive messages from mattermost, only sending is possible.") | ||||
| 		} | ||||
| 		go b.handleMatterClient(messages) | ||||
| 	} | ||||
| 	var ok bool | ||||
| @@ -82,18 +93,24 @@ func (b *Bmattermost) handleMatter() { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| //nolint:cyclop | ||||
| func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 	for message := range b.mc.MessageChan { | ||||
| 		b.Log.Debugf("%#v", message.Raw.Data) | ||||
| 		b.Log.Debugf("%#v %#v", message.Raw.GetData(), message.Raw.EventType()) | ||||
|  | ||||
| 		if b.skipMessage(message) { | ||||
| 			b.Log.Debugf("Skipped message: %#v", message) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		channelName := b.getChannelName(message.Post.ChannelId) | ||||
| 		if channelName == "" { | ||||
| 			channelName = message.Channel | ||||
| 		} | ||||
|  | ||||
| 		// 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) | ||||
| 			b.handleDownloadAvatar(message.UserID, channelName) | ||||
| 		} | ||||
|  | ||||
| 		b.Log.Debugf("== Receiving event %#v", message) | ||||
| @@ -101,10 +118,10 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 		rmsg := &config.Message{ | ||||
| 			Username: message.Username, | ||||
| 			UserID:   message.UserID, | ||||
| 			Channel:  message.Channel, | ||||
| 			Channel:  channelName, | ||||
| 			Text:     message.Text, | ||||
| 			ID:       message.Post.Id, | ||||
| 			ParentID: message.Post.ParentId, | ||||
| 			ParentID: message.Post.RootId, // ParentID is obsolete with mattermost | ||||
| 			Extra:    make(map[string][]interface{}), | ||||
| 		} | ||||
|  | ||||
| @@ -112,11 +129,11 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 		b.handleProps(rmsg, message) | ||||
|  | ||||
| 		// create a text for bridges that don't support native editing | ||||
| 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") { | ||||
| 		if message.Raw.EventType() == model.WebsocketEventPostEdited && !b.GetBool("EditDisable") { | ||||
| 			rmsg.Text = message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED { | ||||
| 		if message.Raw.EventType() == model.WebsocketEventPostDeleted { | ||||
| 			rmsg.Event = config.EventMsgDelete | ||||
| 		} | ||||
|  | ||||
| @@ -128,8 +145,10 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 		} | ||||
|  | ||||
| 		// Use nickname instead of username if defined | ||||
| 		if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||
| 			rmsg.Username = nick | ||||
| 		if !b.GetBool("useusername") { | ||||
| 			if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||
| 				rmsg.Username = nick | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		messages <- rmsg | ||||
| @@ -140,6 +159,7 @@ func (b *Bmattermost) handleMatterHook(messages chan *config.Message) { | ||||
| 	for { | ||||
| 		message := b.mh.Receive() | ||||
| 		b.Log.Debugf("Receiving from matterhook %#v", message) | ||||
|  | ||||
| 		messages <- &config.Message{ | ||||
| 			UserID:   message.UserID, | ||||
| 			Username: message.UserName, | ||||
| @@ -149,11 +169,10 @@ func (b *Bmattermost) handleMatterHook(messages chan *config.Message) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // 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) | ||||
| 	channelID := b.getChannelID(msg.Channel) | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name) | ||||
| @@ -169,6 +188,7 @@ func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	return res, err | ||||
| } | ||||
|  | ||||
| //nolint:forcetypeassert | ||||
| func (b *Bmattermost) handleProps(rmsg *config.Message, message *matterclient.Message) { | ||||
| 	props := message.Post.Props | ||||
| 	if props == nil { | ||||
| @@ -179,16 +199,18 @@ func (b *Bmattermost) handleProps(rmsg *config.Message, message *matterclient.Me | ||||
| 	} | ||||
| 	if _, ok := props["attachments"].([]interface{}); ok { | ||||
| 		rmsg.Extra["attachments"] = props["attachments"].([]interface{}) | ||||
| 		if rmsg.Text == "" { | ||||
| 			for _, attachment := range rmsg.Extra["attachments"] { | ||||
| 				attach := attachment.(map[string]interface{}) | ||||
| 				if attach["text"].(string) != "" { | ||||
| 					rmsg.Text += attach["text"].(string) | ||||
| 					continue | ||||
| 				} | ||||
| 				if attach["fallback"].(string) != "" { | ||||
| 					rmsg.Text += attach["fallback"].(string) | ||||
| 				} | ||||
| 		if rmsg.Text != "" { | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		for _, attachment := range rmsg.Extra["attachments"] { | ||||
| 			attach := attachment.(map[string]interface{}) | ||||
| 			if attach["text"].(string) != "" { | ||||
| 				rmsg.Text += attach["text"].(string) | ||||
| 				continue | ||||
| 			} | ||||
| 			if attach["fallback"].(string) != "" { | ||||
| 				rmsg.Text += attach["fallback"].(string) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|   | ||||
| @@ -1,13 +1,14 @@ | ||||
| package bmattermost | ||||
|  | ||||
| import ( | ||||
| 	"net/http" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| 	"github.com/matterbridge/matterclient" | ||||
| 	"github.com/mattermost/mattermost/server/public/model" | ||||
| ) | ||||
|  | ||||
| func (b *Bmattermost) doConnectWebhookBind() error { | ||||
| @@ -15,8 +16,10 @@ func (b *Bmattermost) doConnectWebhookBind() error { | ||||
| 	case b.GetString("WebhookURL") != "": | ||||
| 		b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||
| 		b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 			matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 				BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 			matterhook.Config{ | ||||
| 				InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 				BindAddress:        b.GetString("WebhookBindAddress"), | ||||
| 			}) | ||||
| 	case b.GetString("Token") != "": | ||||
| 		b.Log.Info("Connecting using token (sending)") | ||||
| 		err := b.apiLogin() | ||||
| @@ -32,8 +35,10 @@ func (b *Bmattermost) doConnectWebhookBind() error { | ||||
| 	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")}) | ||||
| 			matterhook.Config{ | ||||
| 				InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 				BindAddress:        b.GetString("WebhookBindAddress"), | ||||
| 			}) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @@ -41,8 +46,10 @@ func (b *Bmattermost) doConnectWebhookBind() error { | ||||
| func (b *Bmattermost) doConnectWebhookURL() error { | ||||
| 	b.Log.Info("Connecting using webhookurl (sending)") | ||||
| 	b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 		matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 			DisableServer: true}) | ||||
| 		matterhook.Config{ | ||||
| 			InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 			DisableServer:      true, | ||||
| 		}) | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		b.Log.Info("Connecting using token (receiving)") | ||||
| 		err := b.apiLogin() | ||||
| @@ -59,27 +66,28 @@ func (b *Bmattermost) doConnectWebhookURL() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| //nolint:wrapcheck | ||||
| func (b *Bmattermost) apiLogin() error { | ||||
| 	password := b.GetString("Password") | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		password = "token=" + b.GetString("Token") | ||||
| 	} | ||||
|  | ||||
| 	b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("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.SkipVersionCheck = b.GetBool("SkipVersionCheck") | ||||
| 	b.mc.NoTLS = b.GetBool("NoTLS") | ||||
| 	b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server")) | ||||
| 	err := b.mc.Login() | ||||
| 	if err != nil { | ||||
|  | ||||
| 	if err := b.mc.Login(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.TeamID = b.mc.GetTeamId() | ||||
| 	go b.mc.WsReceiver() | ||||
| 	go b.mc.StatusLoop() | ||||
| 	b.TeamID = b.mc.GetTeamID() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -112,6 +120,7 @@ func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| 	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) { | ||||
| @@ -135,7 +144,7 @@ func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text += fi.URL | ||||
| 					msg.Text += " " + fi.URL | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| @@ -162,19 +171,37 @@ func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| } | ||||
|  | ||||
| // skipMessages returns true if this message should not be handled | ||||
| // | ||||
| //nolint:gocyclo,cyclop | ||||
| 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" { | ||||
| 	skipJoinMessageTypes := map[string]struct{}{ | ||||
| 		"system_join_leave":          {}, // deprecated for system_add_to_channel | ||||
| 		"system_leave_channel":       {}, // deprecated for system_remove_from_channel | ||||
| 		"system_join_channel":        {}, | ||||
| 		"system_add_to_channel":      {}, | ||||
| 		"system_remove_from_channel": {}, | ||||
| 		"system_add_to_team":         {}, | ||||
| 		"system_remove_from_team":    {}, | ||||
| 	} | ||||
|  | ||||
| 	// dirty hack to efficiently check if this element is in the map without writing a contains func | ||||
| 	// can be replaced with native slice.contains with go 1.21 | ||||
| 	if _, ok := skipJoinMessageTypes[message.Type]; ok { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return true | ||||
| 		} | ||||
|  | ||||
| 		channelName := b.getChannelName(message.Post.ChannelId) | ||||
| 		if channelName == "" { | ||||
| 			channelName = message.Channel | ||||
| 		} | ||||
|  | ||||
| 		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, | ||||
| 			Channel:  channelName, | ||||
| 			Account:  b.Account, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 		} | ||||
| @@ -182,20 +209,27 @@ func (b *Bmattermost) skipMessage(message *matterclient.Message) bool { | ||||
| 	} | ||||
|  | ||||
| 	// Handle edited messages | ||||
| 	if (message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED) && b.GetBool("EditDisable") { | ||||
| 	if (message.Raw.EventType() == model.WebsocketEventPostEdited) && b.GetBool("EditDisable") { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Ignore non-post messages | ||||
| 	if message.Post == nil { | ||||
| 		b.Log.Debugf("ignoring nil message.Post: %#v", message) | ||||
| 		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") | ||||
| 			b.Log.Debug("sent by matterbridge, ignoring") | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from a user logged in as the bot | ||||
| 	if b.mc.User.Username == message.Username { | ||||
| 		b.Log.Debug("message from same user as bot, ignoring") | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| @@ -205,14 +239,56 @@ func (b *Bmattermost) skipMessage(message *matterclient.Message) bool { | ||||
| 	} | ||||
|  | ||||
| 	// ignore messages from other teams than ours | ||||
| 	if message.Raw.Data["team_id"].(string) != b.TeamID { | ||||
| 	if message.Raw.GetData()["team_id"].(string) != b.TeamID { | ||||
| 		b.Log.Debug("message from other team, ignoring") | ||||
| 		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) { | ||||
| 	if !(message.Raw.EventType() == "posted" || message.Raw.EventType() == model.WebsocketEventPostEdited || | ||||
| 		message.Raw.EventType() == model.WebsocketEventPostDeleted) { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) getVersion() string { | ||||
| 	proto := "https" | ||||
|  | ||||
| 	if b.GetBool("notls") { | ||||
| 		proto = "http" | ||||
| 	} | ||||
|  | ||||
| 	resp, err := http.Get(proto + "://" + b.GetString("server")) | ||||
| 	if err != nil { | ||||
| 		b.Log.Error("failed getting version") | ||||
| 		return "" | ||||
| 	} | ||||
|  | ||||
| 	defer resp.Body.Close() | ||||
|  | ||||
| 	return resp.Header.Get("X-Version-Id") | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) getChannelID(name string) string { | ||||
| 	idcheck := strings.Split(name, "ID:") | ||||
| 	if len(idcheck) > 1 { | ||||
| 		return idcheck[1] | ||||
| 	} | ||||
|  | ||||
| 	return b.mc.GetChannelID(name, b.TeamID) | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) getChannelName(id string) string { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, c := range b.channelInfoMap { | ||||
| 		if c.Name == "ID:"+id { | ||||
| 			// if we have ID: specified in our gateway configuration return this | ||||
| 			return c.Name | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return "" | ||||
| } | ||||
|   | ||||
| @@ -1,31 +1,44 @@ | ||||
| package bmattermost | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/matterbridge/matterclient" | ||||
| 	"github.com/rs/xid" | ||||
| ) | ||||
|  | ||||
| type Bmattermost struct { | ||||
| 	mh     *matterhook.Client | ||||
| 	mc     *matterclient.MMClient | ||||
| 	mc     *matterclient.Client | ||||
| 	v6     bool | ||||
| 	uuid   string | ||||
| 	TeamID string | ||||
| 	*bridge.Config | ||||
| 	avatarMap map[string]string | ||||
| 	avatarMap      map[string]string | ||||
| 	channelsMutex  sync.RWMutex | ||||
| 	channelInfoMap map[string]*config.ChannelInfo | ||||
| } | ||||
|  | ||||
| const mattermostPlugin = "mattermost.plugin" | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bmattermost{Config: cfg, avatarMap: make(map[string]string)} | ||||
| 	b := &Bmattermost{ | ||||
| 		Config:         cfg, | ||||
| 		avatarMap:      make(map[string]string), | ||||
| 		channelInfoMap: make(map[string]*config.ChannelInfo), | ||||
| 	} | ||||
|  | ||||
| 	b.v6 = b.GetBool("v6") | ||||
| 	b.uuid = xid.New().String() | ||||
|  | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| @@ -37,6 +50,13 @@ func (b *Bmattermost) Connect() error { | ||||
| 	if b.Account == mattermostPlugin { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	if strings.HasPrefix(b.getVersion(), "6.") || strings.HasPrefix(b.getVersion(), "7.") { | ||||
| 		if !b.v6 { | ||||
| 			b.v6 = true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		if err := b.doConnectWebhookBind(); err != nil { | ||||
| 			return err | ||||
| @@ -60,6 +80,7 @@ func (b *Bmattermost) Connect() error { | ||||
| 		go b.handleMatter() | ||||
| 	case b.GetString("Login") != "": | ||||
| 		b.Log.Info("Connecting using login/password (sending and receiving)") | ||||
| 		b.Log.Infof("Using mattermost v6 methods: %t", b.v6) | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| @@ -81,14 +102,21 @@ func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if b.Account == mattermostPlugin { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.Lock() | ||||
| 	b.channelInfoMap[channel.ID] = &channel | ||||
| 	b.channelsMutex.Unlock() | ||||
|  | ||||
| 	// we can only join channels using the API | ||||
| 	if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" { | ||||
| 		id := b.mc.GetChannelId(channel.Name, b.TeamID) | ||||
| 		id := b.getChannelID(channel.Name) | ||||
| 		if id == "" { | ||||
| 			return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) | ||||
| 		} | ||||
|  | ||||
| 		return b.mc.JoinChannel(id) | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -118,13 +146,31 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		return msg.ID, b.mc.DeleteMessage(msg.ID) | ||||
| 	} | ||||
|  | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 		msg.Text = fmt.Sprintf("[thread]: %s", msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	// we only can reply to the root of the thread, not to a specific ID (like discord for example does) | ||||
| 	if msg.ParentID != "" { | ||||
| 		post, _, err := b.mc.Client.GetPost(context.TODO(), msg.ParentID, "") | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("getting post %s failed: %s", msg.ParentID, err) | ||||
| 		} | ||||
| 		if post != nil && post.RootId != "" { | ||||
| 			msg.ParentID = post.RootId | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			if _, err := b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, b.TeamID), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { | ||||
| 			if _, err := b.mc.PostMessage(b.getChannelID(rmsg.Channel), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { | ||||
| 				b.Log.Errorf("PostMessage failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
| @@ -144,5 +190,5 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message | ||||
| 	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text, msg.ParentID) | ||||
| 	return b.mc.PostMessage(b.getChannelID(msg.Channel), msg.Text, msg.ParentID) | ||||
| } | ||||
|   | ||||
							
								
								
									
										101
									
								
								bridge/msteams/handler.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										101
									
								
								bridge/msteams/handler.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,101 @@ | ||||
| package bmsteams | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
|  | ||||
| 	msgraph "github.com/yaegashi/msgraph.go/beta" | ||||
| ) | ||||
|  | ||||
| func (b *Bmsteams) findFile(weburl string) (string, error) { | ||||
| 	itemRB, err := b.gc.GetDriveItemByURL(b.ctx, weburl) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	itemRB.Workbook().Worksheets() | ||||
| 	b.gc.Workbooks() | ||||
| 	item, err := itemRB.Request().Get(b.ctx) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	if url, ok := item.GetAdditionalData("@microsoft.graph.downloadUrl"); ok { | ||||
| 		return url.(string), nil | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bmsteams) handleDownloadFile(rmsg *config.Message, filename, weburl string) error { | ||||
| 	realURL, err := b.findFile(weburl) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	// Actually download the file. | ||||
| 	data, err := helper.DownloadFile(realURL) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("download %s failed %#v", weburl, err) | ||||
| 	} | ||||
|  | ||||
| 	// If a comment is attached to the file(s) it is in the 'Text' field of the teams 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, filename, comment, weburl, data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) handleAttachments(rmsg *config.Message, msg msgraph.ChatMessage) { | ||||
| 	for _, a := range msg.Attachments { | ||||
| 		//remove the attachment tags from the text | ||||
| 		rmsg.Text = attachRE.ReplaceAllString(rmsg.Text, "") | ||||
|  | ||||
| 		//handle a code snippet (code block) | ||||
| 		if *a.ContentType == "application/vnd.microsoft.card.codesnippet" { | ||||
| 			b.handleCodeSnippet(rmsg, a) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		//handle the download | ||||
| 		err := b.handleDownloadFile(rmsg, *a.Name, *a.ContentURL) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("download of %s failed: %s", *a.Name, err) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| type AttachContent struct { | ||||
| 	Language       string `json:"language"` | ||||
| 	CodeSnippetURL string `json:"codeSnippetUrl"` | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) handleCodeSnippet(rmsg *config.Message, attach msgraph.ChatMessageAttachment) { | ||||
| 	var content AttachContent | ||||
| 	err := json.Unmarshal([]byte(*attach.Content), &content) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("unmarshal codesnippet failed: %s", err) | ||||
| 		return | ||||
| 	} | ||||
| 	s := strings.Split(content.CodeSnippetURL, "/") | ||||
| 	if len(s) != 13 { | ||||
| 		b.Log.Errorf("codesnippetUrl has unexpected size: %s", content.CodeSnippetURL) | ||||
| 		return | ||||
| 	} | ||||
| 	resp, err := b.gc.Teams().Request().Client().Get(content.CodeSnippetURL) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("retrieving snippet content failed:%s", err) | ||||
| 		return | ||||
| 	} | ||||
| 	defer resp.Body.Close() | ||||
| 	res, err := ioutil.ReadAll(resp.Body) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("reading snippet data failed: %s", err) | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg.Text = rmsg.Text + "\n```" + content.Language + "\n" + string(res) + "\n```\n" | ||||
| } | ||||
							
								
								
									
										229
									
								
								bridge/msteams/msteams.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										229
									
								
								bridge/msteams/msteams.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,229 @@ | ||||
| package bmsteams | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/davecgh/go-spew/spew" | ||||
|  | ||||
| 	"github.com/mattn/godown" | ||||
| 	msgraph "github.com/yaegashi/msgraph.go/beta" | ||||
| 	"github.com/yaegashi/msgraph.go/msauth" | ||||
|  | ||||
| 	"golang.org/x/oauth2" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	defaultScopes = []string{"openid", "profile", "offline_access", "Group.Read.All", "Group.ReadWrite.All"} | ||||
| 	attachRE      = regexp.MustCompile(`<attachment id=.*?attachment>`) | ||||
| ) | ||||
|  | ||||
| type Bmsteams struct { | ||||
| 	gc    *msgraph.GraphServiceRequestBuilder | ||||
| 	ctx   context.Context | ||||
| 	botID string | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Bmsteams{Config: cfg} | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) Connect() error { | ||||
| 	tokenCachePath := b.GetString("sessionFile") | ||||
| 	if tokenCachePath == "" { | ||||
| 		tokenCachePath = "msteams_session.json" | ||||
| 	} | ||||
| 	ctx := context.Background() | ||||
| 	m := msauth.NewManager() | ||||
| 	m.LoadFile(tokenCachePath) //nolint:errcheck | ||||
| 	ts, err := m.DeviceAuthorizationGrant(ctx, b.GetString("TenantID"), b.GetString("ClientID"), defaultScopes, nil) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	err = m.SaveFile(tokenCachePath) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Couldn't save sessionfile in %s: %s", tokenCachePath, err) | ||||
| 	} | ||||
| 	// make file readable only for matterbridge user | ||||
| 	err = os.Chmod(tokenCachePath, 0o600) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Couldn't change permissions for %s: %s", tokenCachePath, err) | ||||
| 	} | ||||
| 	httpClient := oauth2.NewClient(ctx, ts) | ||||
| 	graphClient := msgraph.NewClient(httpClient) | ||||
| 	b.gc = graphClient | ||||
| 	b.ctx = ctx | ||||
|  | ||||
| 	err = b.setBotID() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) Disconnect() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	go func(name string) { | ||||
| 		for { | ||||
| 			err := b.poll(name) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("polling failed for %s: %s. retrying in 5 seconds", name, err) | ||||
| 			} | ||||
| 			time.Sleep(time.Second * 5) | ||||
| 		} | ||||
| 	}(channel.Name) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
| 	if msg.ParentValid() { | ||||
| 		return b.sendReply(msg) | ||||
| 	} | ||||
|  | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 		msg.Text = fmt.Sprintf("[thread]: %s", msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	ct := b.gc.Teams().ID(b.GetString("TeamID")).Channels().ID(msg.Channel).Messages().Request() | ||||
| 	text := msg.Username + msg.Text | ||||
| 	content := &msgraph.ItemBody{Content: &text} | ||||
| 	rmsg := &msgraph.ChatMessage{Body: content} | ||||
| 	res, err := ct.Add(b.ctx, rmsg) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return *res.ID, nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) sendReply(msg config.Message) (string, error) { | ||||
| 	ct := b.gc.Teams().ID(b.GetString("TeamID")).Channels().ID(msg.Channel).Messages().ID(msg.ParentID).Replies().Request() | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
|  | ||||
| 	text := msg.Username + msg.Text | ||||
| 	content := &msgraph.ItemBody{Content: &text} | ||||
| 	rmsg := &msgraph.ChatMessage{Body: content} | ||||
| 	res, err := ct.Add(b.ctx, rmsg) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return *res.ID, nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) getMessages(channel string) ([]msgraph.ChatMessage, error) { | ||||
| 	ct := b.gc.Teams().ID(b.GetString("TeamID")).Channels().ID(channel).Messages().Request() | ||||
| 	rct, err := ct.Get(b.ctx) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	b.Log.Debugf("got %#v messages", len(rct)) | ||||
| 	return rct, nil | ||||
| } | ||||
|  | ||||
| //nolint:gocognit | ||||
| func (b *Bmsteams) poll(channelName string) error { | ||||
| 	msgmap := make(map[string]time.Time) | ||||
| 	b.Log.Debug("getting initial messages") | ||||
| 	res, err := b.getMessages(channelName) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	for _, msg := range res { | ||||
| 		msgmap[*msg.ID] = *msg.CreatedDateTime | ||||
| 		if msg.LastModifiedDateTime != nil { | ||||
| 			msgmap[*msg.ID] = *msg.LastModifiedDateTime | ||||
| 		} | ||||
| 	} | ||||
| 	time.Sleep(time.Second * 5) | ||||
| 	b.Log.Debug("polling for messages") | ||||
| 	for { | ||||
| 		res, err := b.getMessages(channelName) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		for i := len(res) - 1; i >= 0; i-- { | ||||
| 			msg := res[i] | ||||
| 			if mtime, ok := msgmap[*msg.ID]; ok { | ||||
| 				if mtime == *msg.CreatedDateTime && msg.LastModifiedDateTime == nil { | ||||
| 					continue | ||||
| 				} | ||||
| 				if msg.LastModifiedDateTime != nil && mtime == *msg.LastModifiedDateTime { | ||||
| 					continue | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			if b.GetBool("debug") { | ||||
| 				b.Log.Debug("Msg dump: ", spew.Sdump(msg)) | ||||
| 			} | ||||
|  | ||||
| 			// skip non-user message for now. | ||||
| 			if msg.From == nil || msg.From.User == nil { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			if *msg.From.User.ID == b.botID { | ||||
| 				b.Log.Debug("skipping own message") | ||||
| 				msgmap[*msg.ID] = *msg.CreatedDateTime | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			msgmap[*msg.ID] = *msg.CreatedDateTime | ||||
| 			if msg.LastModifiedDateTime != nil { | ||||
| 				msgmap[*msg.ID] = *msg.LastModifiedDateTime | ||||
| 			} | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", *msg.From.User.DisplayName, b.Account) | ||||
| 			text := b.convertToMD(*msg.Body.Content) | ||||
| 			rmsg := config.Message{ | ||||
| 				Username: *msg.From.User.DisplayName, | ||||
| 				Text:     text, | ||||
| 				Channel:  channelName, | ||||
| 				Account:  b.Account, | ||||
| 				Avatar:   "", | ||||
| 				UserID:   *msg.From.User.ID, | ||||
| 				ID:       *msg.ID, | ||||
| 				Extra:    make(map[string][]interface{}), | ||||
| 			} | ||||
|  | ||||
| 			b.handleAttachments(&rmsg, msg) | ||||
| 			b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 		time.Sleep(time.Second * 5) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) setBotID() error { | ||||
| 	req := b.gc.Me().Request() | ||||
| 	r, err := req.Get(b.ctx) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.botID = *r.ID | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmsteams) convertToMD(text string) string { | ||||
| 	if !strings.Contains(text, "<div>") { | ||||
| 		return text | ||||
| 	} | ||||
| 	var sb strings.Builder | ||||
| 	err := godown.Convert(&sb, strings.NewReader(text), nil) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Couldn't convert message to markdown %s", text) | ||||
| 		return text | ||||
| 	} | ||||
| 	return sb.String() | ||||
| } | ||||
							
								
								
									
										70
									
								
								bridge/mumble/codec.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								bridge/mumble/codec.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,70 @@ | ||||
| package bmumble | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
|  | ||||
| 	"layeh.com/gumble/gumble" | ||||
| ) | ||||
|  | ||||
| // This is a dummy implementation of a Gumble audio codec which claims | ||||
| // to implement Opus, but does not actually do anything.  This serves | ||||
| // as a workaround until https://github.com/layeh/gumble/pull/61 is | ||||
| // merged. | ||||
| // See https://github.com/42wim/matterbridge/issues/1750 for details. | ||||
|  | ||||
| const ( | ||||
| 	audioCodecIDOpus = 4 | ||||
| ) | ||||
|  | ||||
| func registerNullCodecAsOpus() { | ||||
| 	codec := &NullCodec{ | ||||
| 		encoder: &NullAudioEncoder{}, | ||||
| 		decoder: &NullAudioDecoder{}, | ||||
| 	} | ||||
| 	gumble.RegisterAudioCodec(audioCodecIDOpus, codec) | ||||
| } | ||||
|  | ||||
| type NullCodec struct { | ||||
| 	encoder *NullAudioEncoder | ||||
| 	decoder *NullAudioDecoder | ||||
| } | ||||
|  | ||||
| func (c *NullCodec) ID() int { | ||||
| 	return audioCodecIDOpus | ||||
| } | ||||
|  | ||||
| func (c *NullCodec) NewEncoder() gumble.AudioEncoder { | ||||
| 	e := &NullAudioEncoder{} | ||||
| 	return e | ||||
| } | ||||
|  | ||||
| func (c *NullCodec) NewDecoder() gumble.AudioDecoder { | ||||
| 	d := &NullAudioDecoder{} | ||||
| 	return d | ||||
| } | ||||
|  | ||||
| type NullAudioEncoder struct{} | ||||
|  | ||||
| func (e *NullAudioEncoder) ID() int { | ||||
| 	return audioCodecIDOpus | ||||
| } | ||||
|  | ||||
| func (e *NullAudioEncoder) Encode(pcm []int16, mframeSize, maxDataBytes int) ([]byte, error) { | ||||
| 	return nil, fmt.Errorf("not implemented") | ||||
| } | ||||
|  | ||||
| func (e *NullAudioEncoder) Reset() { | ||||
| } | ||||
|  | ||||
| type NullAudioDecoder struct{} | ||||
|  | ||||
| func (d *NullAudioDecoder) ID() int { | ||||
| 	return audioCodecIDOpus | ||||
| } | ||||
|  | ||||
| func (d *NullAudioDecoder) Decode(data []byte, frameSize int) ([]int16, error) { | ||||
| 	return nil, fmt.Errorf("not implemented") | ||||
| } | ||||
|  | ||||
| func (d *NullAudioDecoder) Reset() { | ||||
| } | ||||
							
								
								
									
										158
									
								
								bridge/mumble/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										158
									
								
								bridge/mumble/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,158 @@ | ||||
| package bmumble | ||||
|  | ||||
| import ( | ||||
| 	"strconv" | ||||
| 	"time" | ||||
|  | ||||
| 	"layeh.com/gumble/gumble" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| ) | ||||
|  | ||||
| func (b *Bmumble) handleServerConfig(event *gumble.ServerConfigEvent) { | ||||
| 	b.serverConfigUpdate <- *event | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleTextMessage(event *gumble.TextMessageEvent) { | ||||
| 	sender := "unknown" | ||||
| 	if event.TextMessage.Sender != nil { | ||||
| 		sender = event.TextMessage.Sender.Name | ||||
| 	} | ||||
| 	// If the text message is received before receiving a ServerSync | ||||
| 	// and UserState, Client.Self or Self.Channel are nil | ||||
| 	if event.Client.Self == nil || event.Client.Self.Channel == nil { | ||||
| 		b.Log.Warn("Connection bootstrap not finished, discarding text message") | ||||
| 		return | ||||
| 	} | ||||
| 	// Convert Mumble HTML messages to markdown | ||||
| 	parts, err := b.convertHTMLtoMarkdown(event.TextMessage.Message) | ||||
| 	if err != nil { | ||||
| 		b.Log.Error(err) | ||||
| 	} | ||||
| 	now := time.Now().UTC() | ||||
| 	for i, part := range parts { | ||||
| 		// Construct matterbridge message and pass on to the gateway | ||||
| 		rmsg := config.Message{ | ||||
| 			Channel:  strconv.FormatUint(uint64(event.Client.Self.Channel.ID), 10), | ||||
| 			Username: sender, | ||||
| 			UserID:   sender + "@" + b.Host, | ||||
| 			Account:  b.Account, | ||||
| 		} | ||||
| 		if part.Image == nil { | ||||
| 			rmsg.Text = part.Text | ||||
| 		} else { | ||||
| 			fileExt := part.FileExtension | ||||
| 			if fileExt == ".jfif" { | ||||
| 				fileExt = ".jpg" | ||||
| 			} | ||||
| 			if fileExt == ".jpe" { | ||||
| 				fileExt = ".jpg" | ||||
| 			} | ||||
| 			fname := b.Account + "_" + strconv.FormatInt(now.UnixNano(), 10) + "_" + strconv.Itoa(i) + fileExt | ||||
| 			rmsg.Extra = make(map[string][]interface{}) | ||||
| 			if err = helper.HandleDownloadSize(b.Log, &rmsg, fname, int64(len(part.Image)), b.General); err != nil { | ||||
| 				b.Log.WithError(err).Warn("not including image in message") | ||||
| 				continue | ||||
| 			} | ||||
| 			helper.HandleDownloadData(b.Log, &rmsg, fname, "", "", &part.Image, b.General) | ||||
| 		} | ||||
| 		b.Log.Debugf("Sending message to gateway: %+v", rmsg) | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleConnect(event *gumble.ConnectEvent) { | ||||
| 	// Set the user's "bio"/comment | ||||
| 	if comment := b.GetString("UserComment"); comment != "" && event.Client.Self != nil { | ||||
| 		event.Client.Self.SetComment(comment) | ||||
| 	} | ||||
| 	// No need to talk or listen | ||||
| 	event.Client.Self.SetSelfDeafened(true) | ||||
| 	// if the Channel variable is set, this is a reconnect -> rejoin channel | ||||
| 	if b.Channel != nil { | ||||
| 		if err := b.doJoin(event.Client, *b.Channel); err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 		} | ||||
| 		b.Remote <- config.Message{ | ||||
| 			Username: "system", | ||||
| 			Text:     "rejoin", | ||||
| 			Channel:  "", | ||||
| 			Account:  b.Account, | ||||
| 			Event:    config.EventRejoinChannels, | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleJoinLeave(event *gumble.UserChangeEvent) { | ||||
| 	// Ignore events happening before setup is done | ||||
| 	if b.Channel == nil { | ||||
| 		return | ||||
| 	} | ||||
| 	if b.GetBool("nosendjoinpart") { | ||||
| 		return | ||||
| 	} | ||||
| 	b.Log.Debugf("Received gumble user change event: %+v", event) | ||||
|  | ||||
| 	text := "" | ||||
| 	switch { | ||||
| 	case event.Type&gumble.UserChangeKicked > 0: | ||||
| 		text = " was kicked" | ||||
| 	case event.Type&gumble.UserChangeBanned > 0: | ||||
| 		text = " was banned" | ||||
| 	case event.Type&gumble.UserChangeDisconnected > 0: | ||||
| 		if event.User.Channel != nil && event.User.Channel.ID == *b.Channel { | ||||
| 			text = " left" | ||||
| 		} | ||||
| 	case event.Type&gumble.UserChangeConnected > 0: | ||||
| 		if event.User.Channel != nil && event.User.Channel.ID == *b.Channel { | ||||
| 			text = " joined" | ||||
| 		} | ||||
| 	case event.Type&gumble.UserChangeChannel > 0: | ||||
| 		// Treat Mumble channel changes the same as connects/disconnects; as far as matterbridge is concerned, they are identical | ||||
| 		if event.User.Channel != nil && event.User.Channel.ID == *b.Channel { | ||||
| 			text = " joined" | ||||
| 		} else { | ||||
| 			text = " left" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if text != "" { | ||||
| 		b.Remote <- config.Message{ | ||||
| 			Username: "system", | ||||
| 			Text:     event.User.Name + text, | ||||
| 			Channel:  strconv.FormatUint(uint64(*b.Channel), 10), | ||||
| 			Account:  b.Account, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleUserModified(event *gumble.UserChangeEvent) { | ||||
| 	// Ignore events happening before setup is done | ||||
| 	if b.Channel == nil { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if event.Type&gumble.UserChangeChannel > 0 { | ||||
| 		// Someone attempted to move the user out of the configured channel; attempt to join back | ||||
| 		if err := b.doJoin(event.Client, *b.Channel); err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleUserChange(event *gumble.UserChangeEvent) { | ||||
| 	// The UserChangeEvent is used for both the gumble client itself as well as other clients | ||||
| 	if event.User != event.Client.Self { | ||||
| 		// other users | ||||
| 		b.handleJoinLeave(event) | ||||
| 	} else { | ||||
| 		// gumble user | ||||
| 		b.handleUserModified(event) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) handleDisconnect(event *gumble.DisconnectEvent) { | ||||
| 	b.connected <- *event | ||||
| } | ||||
							
								
								
									
										143
									
								
								bridge/mumble/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										143
									
								
								bridge/mumble/helpers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,143 @@ | ||||
| package bmumble | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"mime" | ||||
| 	"net/http" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/mattn/godown" | ||||
| 	"github.com/vincent-petithory/dataurl" | ||||
| ) | ||||
|  | ||||
| type MessagePart struct { | ||||
| 	Text          string | ||||
| 	FileExtension string | ||||
| 	Image         []byte | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) decodeImage(uri string, parts *[]MessagePart) error { | ||||
| 	// Decode the data:image/... URI | ||||
| 	image, err := dataurl.DecodeString(uri) | ||||
| 	if err != nil { | ||||
| 		b.Log.WithError(err).Info("No image extracted") | ||||
| 		return err | ||||
| 	} | ||||
| 	// Determine the file extensions for that image | ||||
| 	ext, err := mime.ExtensionsByType(image.MediaType.ContentType()) | ||||
| 	if err != nil || len(ext) == 0 { | ||||
| 		b.Log.WithError(err).Infof("No file extension registered for MIME type '%s'", image.MediaType.ContentType()) | ||||
| 		return err | ||||
| 	} | ||||
| 	// Add the image to the MessagePart slice | ||||
| 	*parts = append(*parts, MessagePart{"", ext[0], image.Data}) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) tokenize(t *string) ([]MessagePart, error) { | ||||
| 	// `^(.*?)` matches everything before the image | ||||
| 	// `!\[[^\]]*\]\(` matches the `]+)` matches the data: URI used by Mumble | ||||
| 	// `\)` matches the closing parenthesis after the URI | ||||
| 	// `(.*)$` matches the remaining text to be examined in the next iteration | ||||
| 	p := regexp.MustCompile(`^(?ms)(.*?)!\[[^\]]*\]\((data:image\/[^)]+)\)(.*)$`) | ||||
| 	remaining := *t | ||||
| 	var parts []MessagePart | ||||
| 	for { | ||||
| 		tokens := p.FindStringSubmatch(remaining) | ||||
| 		if tokens == nil { | ||||
| 			// no match -> remaining string is non-image text | ||||
| 			pre := strings.TrimSpace(remaining) | ||||
| 			if len(pre) > 0 { | ||||
| 				parts = append(parts, MessagePart{pre, "", nil}) | ||||
| 			} | ||||
| 			return parts, nil | ||||
| 		} | ||||
|  | ||||
| 		// tokens[1] is the text before the image | ||||
| 		if len(tokens[1]) > 0 { | ||||
| 			pre := strings.TrimSpace(tokens[1]) | ||||
| 			parts = append(parts, MessagePart{pre, "", nil}) | ||||
| 		} | ||||
| 		// tokens[2] is the image URL | ||||
| 		uri, err := dataurl.UnescapeToString(strings.TrimSpace(strings.ReplaceAll(tokens[2], " ", ""))) | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Info("URL unescaping failed") | ||||
| 			remaining = strings.TrimSpace(tokens[3]) | ||||
| 			continue | ||||
| 		} | ||||
| 		err = b.decodeImage(uri, &parts) | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Info("Decoding the image failed") | ||||
| 		} | ||||
| 		// tokens[3] is the text after the image, processed in the next iteration | ||||
| 		remaining = strings.TrimSpace(tokens[3]) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) convertHTMLtoMarkdown(html string) ([]MessagePart, error) { | ||||
| 	var sb strings.Builder | ||||
| 	err := godown.Convert(&sb, strings.NewReader(html), nil) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	markdown := sb.String() | ||||
| 	b.Log.Debugf("### to markdown: %s", markdown) | ||||
| 	return b.tokenize(&markdown) | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) extractFiles(msg *config.Message) []config.Message { | ||||
| 	var messages []config.Message | ||||
| 	if msg.Extra == nil || len(msg.Extra["file"]) == 0 { | ||||
| 		return messages | ||||
| 	} | ||||
| 	// Create a separate message for each file | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		imsg := config.Message{ | ||||
| 			Channel:   msg.Channel, | ||||
| 			Username:  msg.Username, | ||||
| 			UserID:    msg.UserID, | ||||
| 			Account:   msg.Account, | ||||
| 			Protocol:  msg.Protocol, | ||||
| 			Timestamp: msg.Timestamp, | ||||
| 			Event:     "mumble_image", | ||||
| 		} | ||||
| 		// If no data is present for the file, send a link instead | ||||
| 		if fi.Data == nil || len(*fi.Data) == 0 { | ||||
| 			if len(fi.URL) > 0 { | ||||
| 				imsg.Text = fmt.Sprintf(`<a href="%s">%s</a>`, fi.URL, fi.URL) | ||||
| 				messages = append(messages, imsg) | ||||
| 			} else { | ||||
| 				b.Log.Infof("Not forwarding file without local data") | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
| 		mimeType := http.DetectContentType(*fi.Data) | ||||
| 		// Mumble only supports images natively, send a link instead | ||||
| 		if !strings.HasPrefix(mimeType, "image/") { | ||||
| 			if len(fi.URL) > 0 { | ||||
| 				imsg.Text = fmt.Sprintf(`<a href="%s">%s</a>`, fi.URL, fi.URL) | ||||
| 				messages = append(messages, imsg) | ||||
| 			} else { | ||||
| 				b.Log.Infof("Not forwarding file of type %s", mimeType) | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
| 		mimeType = strings.TrimSpace(strings.Split(mimeType, ";")[0]) | ||||
| 		// Build data:image/...;base64,... style image URL and embed image directly into the message | ||||
| 		du := dataurl.New(*fi.Data, mimeType) | ||||
| 		dataURL, err := du.MarshalText() | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Infof("Image Serialization into data URL failed (type: %s, length: %d)", mimeType, len(*fi.Data)) | ||||
| 			continue | ||||
| 		} | ||||
| 		imsg.Text = fmt.Sprintf(`<img src="%s"/>`, dataURL) | ||||
| 		messages = append(messages, imsg) | ||||
| 	} | ||||
| 	// Remove files from original message | ||||
| 	msg.Extra["file"] = nil | ||||
| 	return messages | ||||
| } | ||||
							
								
								
									
										268
									
								
								bridge/mumble/mumble.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										268
									
								
								bridge/mumble/mumble.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,268 @@ | ||||
| package bmumble | ||||
|  | ||||
| import ( | ||||
| 	"crypto/tls" | ||||
| 	"crypto/x509" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"net" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"layeh.com/gumble/gumble" | ||||
| 	"layeh.com/gumble/gumbleutil" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	stripmd "github.com/writeas/go-strip-markdown" | ||||
|  | ||||
| 	// 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 Bmumble struct { | ||||
| 	client             *gumble.Client | ||||
| 	Nick               string | ||||
| 	Host               string | ||||
| 	Channel            *uint32 | ||||
| 	local              chan config.Message | ||||
| 	running            chan error | ||||
| 	connected          chan gumble.DisconnectEvent | ||||
| 	serverConfigUpdate chan gumble.ServerConfigEvent | ||||
| 	serverConfig       gumble.ServerConfigEvent | ||||
| 	tlsConfig          tls.Config | ||||
|  | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bmumble{} | ||||
| 	b.Config = cfg | ||||
| 	b.Nick = b.GetString("Nick") | ||||
| 	b.local = make(chan config.Message) | ||||
| 	b.running = make(chan error) | ||||
| 	b.connected = make(chan gumble.DisconnectEvent) | ||||
| 	b.serverConfigUpdate = make(chan gumble.ServerConfigEvent) | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) Connect() error { | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
| 	host, portstr, err := net.SplitHostPort(b.GetString("Server")) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Host = host | ||||
| 	_, err = strconv.Atoi(portstr) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	if err = b.buildTLSConfig(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	go b.doSend() | ||||
| 	go b.connectLoop() | ||||
| 	err = <-b.running | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) Disconnect() error { | ||||
| 	return b.client.Disconnect() | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	cid, err := strconv.ParseUint(channel.Name, 10, 32) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	channelID := uint32(cid) | ||||
| 	if b.Channel != nil && *b.Channel != channelID { | ||||
| 		b.Log.Fatalf("Cannot join channel ID '%d', already joined to channel ID %d", channelID, *b.Channel) | ||||
| 		return errors.New("the Mumble bridge can only join a single channel") | ||||
| 	} | ||||
| 	b.Channel = &channelID | ||||
| 	return b.doJoin(b.client, channelID) | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) Send(msg config.Message) (string, error) { | ||||
| 	// Only process text messages | ||||
| 	b.Log.Debugf("=> Received local message %#v", msg) | ||||
| 	if msg.Event != "" && msg.Event != config.EventUserAction && msg.Event != config.EventJoinLeave { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	attachments := b.extractFiles(&msg) | ||||
| 	b.local <- msg | ||||
| 	for _, a := range attachments { | ||||
| 		b.local <- a | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) buildTLSConfig() error { | ||||
| 	b.tlsConfig = tls.Config{} | ||||
| 	// Load TLS client certificate keypair required for registered user authentication | ||||
| 	if cpath := b.GetString("TLSClientCertificate"); cpath != "" { | ||||
| 		if ckey := b.GetString("TLSClientKey"); ckey != "" { | ||||
| 			cert, err := tls.LoadX509KeyPair(cpath, ckey) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			b.tlsConfig.Certificates = []tls.Certificate{cert} | ||||
| 		} | ||||
| 	} | ||||
| 	// Load TLS CA used for server verification.  If not provided, the Go system trust anchor is used | ||||
| 	if capath := b.GetString("TLSCACertificate"); capath != "" { | ||||
| 		ca, err := ioutil.ReadFile(capath) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		b.tlsConfig.RootCAs = x509.NewCertPool() | ||||
| 		b.tlsConfig.RootCAs.AppendCertsFromPEM(ca) | ||||
| 	} | ||||
| 	b.tlsConfig.InsecureSkipVerify = b.GetBool("SkipTLSVerify") | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) connectLoop() { | ||||
| 	firstConnect := true | ||||
| 	for { | ||||
| 		err := b.doConnect() | ||||
| 		if firstConnect { | ||||
| 			b.running <- err | ||||
| 		} | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Connection to server failed: %#v", err) | ||||
| 			if firstConnect { | ||||
| 				break | ||||
| 			} else { | ||||
| 				b.Log.Info("Retrying in 10s") | ||||
| 				time.Sleep(10 * time.Second) | ||||
| 				continue | ||||
| 			} | ||||
| 		} | ||||
| 		firstConnect = false | ||||
| 		d := <-b.connected | ||||
| 		switch d.Type { | ||||
| 		case gumble.DisconnectError: | ||||
| 			b.Log.Errorf("Lost connection to the server (%s), attempting reconnect", d.String) | ||||
| 			continue | ||||
| 		case gumble.DisconnectKicked: | ||||
| 			b.Log.Errorf("Kicked from the server (%s), attempting reconnect", d.String) | ||||
| 			continue | ||||
| 		case gumble.DisconnectBanned: | ||||
| 			b.Log.Errorf("Banned from the server (%s), not attempting reconnect", d.String) | ||||
| 			close(b.connected) | ||||
| 			close(b.running) | ||||
| 			return | ||||
| 		case gumble.DisconnectUser: | ||||
| 			b.Log.Infof("Disconnect successful") | ||||
| 			close(b.connected) | ||||
| 			close(b.running) | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) doConnect() error { | ||||
| 	// Create new gumble config and attach event handlers | ||||
| 	gumbleConfig := gumble.NewConfig() | ||||
| 	gumbleConfig.Attach(gumbleutil.Listener{ | ||||
| 		ServerConfig: b.handleServerConfig, | ||||
| 		TextMessage:  b.handleTextMessage, | ||||
| 		Connect:      b.handleConnect, | ||||
| 		Disconnect:   b.handleDisconnect, | ||||
| 		UserChange:   b.handleUserChange, | ||||
| 	}) | ||||
| 	gumbleConfig.Username = b.GetString("Nick") | ||||
| 	if password := b.GetString("Password"); password != "" { | ||||
| 		gumbleConfig.Password = password | ||||
| 	} | ||||
|  | ||||
| 	registerNullCodecAsOpus() | ||||
| 	client, err := gumble.DialWithDialer(new(net.Dialer), b.GetString("Server"), gumbleConfig, &b.tlsConfig) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.client = client | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) doJoin(client *gumble.Client, channelID uint32) error { | ||||
| 	channel, ok := client.Channels[channelID] | ||||
| 	if !ok { | ||||
| 		return fmt.Errorf("no channel with ID %d", channelID) | ||||
| 	} | ||||
| 	client.Self.Move(channel) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) doSend() { | ||||
| 	// Message sending loop that makes sure server-side | ||||
| 	// restrictions and client-side message traits don't conflict | ||||
| 	// with each other. | ||||
| 	for { | ||||
| 		select { | ||||
| 		case serverConfig := <-b.serverConfigUpdate: | ||||
| 			b.Log.Debugf("Received server config update: AllowHTML=%#v, MaximumMessageLength=%#v", serverConfig.AllowHTML, serverConfig.MaximumMessageLength) | ||||
| 			b.serverConfig = serverConfig | ||||
| 		case msg := <-b.local: | ||||
| 			b.processMessage(&msg) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmumble) processMessage(msg *config.Message) { | ||||
| 	b.Log.Debugf("Processing message %s", msg.Text) | ||||
|  | ||||
| 	allowHTML := true | ||||
| 	if b.serverConfig.AllowHTML != nil { | ||||
| 		allowHTML = *b.serverConfig.AllowHTML | ||||
| 	} | ||||
|  | ||||
| 	// If this is a specially generated image message, send it unmodified | ||||
| 	if msg.Event == "mumble_image" { | ||||
| 		if allowHTML { | ||||
| 			b.client.Self.Channel.Send(msg.Username+msg.Text, false) | ||||
| 		} else { | ||||
| 			b.Log.Info("Can't send image, server does not allow HTML messages") | ||||
| 		} | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Don't process empty messages | ||||
| 	if len(msg.Text) == 0 { | ||||
| 		return | ||||
| 	} | ||||
| 	// If HTML is allowed, convert markdown into HTML, otherwise strip markdown | ||||
| 	if allowHTML { | ||||
| 		msg.Text = helper.ParseMarkdown(msg.Text) | ||||
| 	} else { | ||||
| 		msg.Text = stripmd.Strip(msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	// If there is a maximum message length, split and truncate the lines | ||||
| 	var msgLines []string | ||||
| 	if maxLength := b.serverConfig.MaximumMessageLength; maxLength != nil { | ||||
| 		if *maxLength != 0 { // Some servers will have unlimited message lengths. | ||||
| 			// Not doing this makes underflows happen. | ||||
| 			msgLines = helper.GetSubLines(msg.Text, *maxLength-len(msg.Username), b.GetString("MessageClipped")) | ||||
| 		} else { | ||||
| 			msgLines = helper.GetSubLines(msg.Text, 0, b.GetString("MessageClipped")) | ||||
| 		} | ||||
| 	} else { | ||||
| 		msgLines = helper.GetSubLines(msg.Text, 0, b.GetString("MessageClipped")) | ||||
| 	} | ||||
| 	// Send the individual lines | ||||
| 	for i := range msgLines { | ||||
| 		// Remove unnecessary newline character, since either way we're sending it as individual lines | ||||
| 		msgLines[i] = strings.TrimSuffix(msgLines[i], "\n") | ||||
| 		b.client.Self.Channel.Send(msg.Username+msgLines[i], false) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										295
									
								
								bridge/nctalk/nctalk.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										295
									
								
								bridge/nctalk/nctalk.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,295 @@ | ||||
| package nctalk | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"crypto/tls" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
|  | ||||
| 	"gomod.garykim.dev/nc-talk/ocs" | ||||
| 	"gomod.garykim.dev/nc-talk/room" | ||||
| 	"gomod.garykim.dev/nc-talk/user" | ||||
| ) | ||||
|  | ||||
| type Btalk struct { | ||||
| 	user  *user.TalkUser | ||||
| 	rooms []Broom | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Btalk{Config: cfg} | ||||
| } | ||||
|  | ||||
| type Broom struct { | ||||
| 	room      *room.TalkRoom | ||||
| 	ctx       context.Context | ||||
| 	ctxCancel context.CancelFunc | ||||
| } | ||||
|  | ||||
| func (b *Btalk) Connect() error { | ||||
| 	b.Log.Info("Connecting") | ||||
| 	tconfig := &user.TalkUserConfig{ | ||||
| 		TLSConfig: &tls.Config{ | ||||
| 			InsecureSkipVerify: b.GetBool("SkipTLSVerify"), //nolint:gosec | ||||
| 		}, | ||||
| 	} | ||||
| 	var err error | ||||
| 	b.user, err = user.NewUser(b.GetString("Server"), b.GetString("Login"), b.GetString("Password"), tconfig) | ||||
| 	if err != nil { | ||||
| 		b.Log.Error("Config could not be used") | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = b.user.Capabilities() | ||||
| 	if err != nil { | ||||
| 		b.Log.Error("Cannot Connect") | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connected") | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) Disconnect() error { | ||||
| 	for _, r := range b.rooms { | ||||
| 		r.ctxCancel() | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	tr, err := room.NewTalkRoom(b.user, channel.Name) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	newRoom := Broom{ | ||||
| 		room: tr, | ||||
| 	} | ||||
| 	newRoom.ctx, newRoom.ctxCancel = context.WithCancel(context.Background()) | ||||
| 	c, err := newRoom.room.ReceiveMessages(newRoom.ctx) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.rooms = append(b.rooms, newRoom) | ||||
|  | ||||
| 	go func() { | ||||
| 		for msg := range c { | ||||
| 			msg := msg | ||||
|  | ||||
| 			if msg.Error != nil { | ||||
| 				b.Log.Errorf("Fatal message poll error: %s\n", msg.Error) | ||||
|  | ||||
| 				return | ||||
| 			} | ||||
|  | ||||
| 			// Ignore messages that are from the bot user | ||||
| 			if msg.ActorID == b.user.User || msg.ActorType == "bridged" { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			// Handle deleting messages | ||||
| 			if msg.MessageType == ocs.MessageSystem && msg.Parent != nil && msg.Parent.MessageType == ocs.MessageDelete { | ||||
| 				b.handleDeletingMessage(&msg, &newRoom) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			// Handle sending messages | ||||
| 			if msg.MessageType == ocs.MessageComment { | ||||
| 				b.handleSendingMessage(&msg, &newRoom) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 		} | ||||
| 	}() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) Send(msg config.Message) (string, error) { | ||||
| 	r := b.getRoom(msg.Channel) | ||||
| 	if r == nil { | ||||
| 		b.Log.Errorf("Could not find room for %v", msg.Channel) | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Standard Message Send | ||||
| 	if msg.Event == "" { | ||||
| 		// Handle sending files if they are included | ||||
| 		err := b.handleSendingFile(&msg, r) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Could not send files in message to room %v from %v: %v", msg.Channel, msg.Username, err) | ||||
|  | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		sentMessage, err := b.sendText(r, &msg, msg.Text) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Could not send message to room %v from %v: %v", msg.Channel, msg.Username, err) | ||||
|  | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		return strconv.Itoa(sentMessage.ID), nil | ||||
| 	} | ||||
|  | ||||
| 	// Message Deletion | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		messageID, err := strconv.Atoi(msg.ID) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		data, err := r.room.DeleteMessage(messageID) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return strconv.Itoa(data.ID), nil | ||||
| 	} | ||||
|  | ||||
| 	// Message is not a type that is currently supported | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) getRoom(token string) *Broom { | ||||
| 	for _, r := range b.rooms { | ||||
| 		if r.room.Token == token { | ||||
| 			return &r | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) sendText(r *Broom, msg *config.Message, text string) (*ocs.TalkRoomMessageData, error) { | ||||
| 	messageToSend := &room.Message{Message: msg.Username + text} | ||||
|  | ||||
| 	if b.GetBool("SeparateDisplayName") { | ||||
| 		messageToSend.Message = text | ||||
| 		messageToSend.ActorDisplayName = msg.Username | ||||
| 	} | ||||
|  | ||||
| 	return r.room.SendComplexMessage(messageToSend) | ||||
| } | ||||
|  | ||||
| func (b *Btalk) handleFiles(mmsg *config.Message, message *ocs.TalkRoomMessageData) error { | ||||
| 	for _, parameter := range message.MessageParameters { | ||||
| 		if parameter.Type == ocs.ROSTypeFile { | ||||
| 			// Get the file | ||||
| 			file, err := b.user.DownloadFile(parameter.Path) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
|  | ||||
| 			if mmsg.Extra == nil { | ||||
| 				mmsg.Extra = make(map[string][]interface{}) | ||||
| 			} | ||||
|  | ||||
| 			mmsg.Extra["file"] = append(mmsg.Extra["file"], config.FileInfo{ | ||||
| 				Name:   parameter.Name, | ||||
| 				Data:   file, | ||||
| 				Size:   int64(len(*file)), | ||||
| 				Avatar: false, | ||||
| 			}) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) handleSendingFile(msg *config.Message, r *Broom) error { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		if fi.URL == "" { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		message := "" | ||||
| 		if fi.Comment != "" { | ||||
| 			message += fi.Comment + " " | ||||
| 		} | ||||
| 		message += fi.URL | ||||
| 		_, err := b.sendText(r, msg, message) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btalk) handleSendingMessage(msg *ocs.TalkRoomMessageData, r *Broom) { | ||||
| 	remoteMessage := config.Message{ | ||||
| 		Text:     formatRichObjectString(msg.Message, msg.MessageParameters), | ||||
| 		Channel:  r.room.Token, | ||||
| 		Username: DisplayName(msg, b.guestSuffix()), | ||||
| 		UserID:   msg.ActorID, | ||||
| 		Account:  b.Account, | ||||
| 	} | ||||
| 	// It is possible for the ID to not be set on older versions of Talk so we only set it if | ||||
| 	// the ID is not blank | ||||
| 	if msg.ID != 0 { | ||||
| 		remoteMessage.ID = strconv.Itoa(msg.ID) | ||||
| 	} | ||||
|  | ||||
| 	// Handle Files | ||||
| 	err := b.handleFiles(&remoteMessage, msg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Error handling file: %#v", msg) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Message is %#v", remoteMessage) | ||||
| 	b.Remote <- remoteMessage | ||||
| } | ||||
|  | ||||
| func (b *Btalk) handleDeletingMessage(msg *ocs.TalkRoomMessageData, r *Broom) { | ||||
| 	remoteMessage := config.Message{ | ||||
| 		Event:   config.EventMsgDelete, | ||||
| 		Text:    config.EventMsgDelete, | ||||
| 		Channel: r.room.Token, | ||||
| 		ID:      strconv.Itoa(msg.Parent.ID), | ||||
| 		Account: b.Account, | ||||
| 	} | ||||
| 	b.Log.Debugf("<= Message being deleted is %#v", remoteMessage) | ||||
| 	b.Remote <- remoteMessage | ||||
| } | ||||
|  | ||||
| func (b *Btalk) guestSuffix() string { | ||||
| 	guestSuffix := " (Guest)" | ||||
| 	if b.IsKeySet("GuestSuffix") { | ||||
| 		guestSuffix = b.GetString("GuestSuffix") | ||||
| 	} | ||||
|  | ||||
| 	return guestSuffix | ||||
| } | ||||
|  | ||||
| // Spec: https://github.com/nextcloud/server/issues/1706#issue-182308785 | ||||
| func formatRichObjectString(message string, parameters map[string]ocs.RichObjectString) string { | ||||
| 	for id, parameter := range parameters { | ||||
| 		text := parameter.Name | ||||
|  | ||||
| 		switch parameter.Type { | ||||
| 		case ocs.ROSTypeUser, ocs.ROSTypeGroup: | ||||
| 			text = "@" + text | ||||
| 		case ocs.ROSTypeFile: | ||||
| 			if parameter.Link != "" { | ||||
| 				text = parameter.Name | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		message = strings.ReplaceAll(message, "{"+id+"}", text) | ||||
| 	} | ||||
|  | ||||
| 	return message | ||||
| } | ||||
|  | ||||
| func DisplayName(msg *ocs.TalkRoomMessageData, suffix string) string { | ||||
| 	if msg.ActorType == ocs.ActorGuest { | ||||
| 		if msg.ActorDisplayName == "" { | ||||
| 			return "Guest" | ||||
| 		} | ||||
|  | ||||
| 		return msg.ActorDisplayName + suffix | ||||
| 	} | ||||
|  | ||||
| 	return msg.ActorDisplayName | ||||
| } | ||||
| @@ -1,7 +1,11 @@ | ||||
| package brocketchat | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/matterbridge/Rocket.Chat.Go.SDK/models" | ||||
| ) | ||||
|  | ||||
| func (b *Brocketchat) handleRocket() { | ||||
| @@ -38,8 +42,26 @@ func (b *Brocketchat) handleRocketHook(messages chan *config.Message) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) handleStatusEvent(ev models.Message, rmsg *config.Message) bool { | ||||
| 	switch ev.Type { | ||||
| 	case "": | ||||
| 		// this is a normal message, no processing needed | ||||
| 		// return true so the message is not dropped | ||||
| 		return true | ||||
| 	case sUserJoined, sUserLeft: | ||||
| 		rmsg.Event = config.EventJoinLeave | ||||
| 		return true | ||||
| 	case sRoomChangedTopic: | ||||
| 		rmsg.Event = config.EventTopicChange | ||||
| 		return true | ||||
| 	} | ||||
| 	b.Log.Debugf("Dropping message with unknown type: %s", ev.Type) | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) handleRocketClient(messages chan *config.Message) { | ||||
| 	for message := range b.messageChan { | ||||
| 		message := message | ||||
| 		// skip messages with same ID, apparently messages get duplicated for an unknown reason | ||||
| 		if _, ok := b.cache.Get(message.ID); ok { | ||||
| 			continue | ||||
| @@ -58,11 +80,51 @@ func (b *Brocketchat) handleRocketClient(messages chan *config.Message) { | ||||
| 			Account:  b.Account, | ||||
| 			UserID:   message.User.ID, | ||||
| 			ID:       message.ID, | ||||
| 			Extra:    make(map[string][]interface{}), | ||||
| 		} | ||||
|  | ||||
| 		b.handleAttachments(&message, rmsg) | ||||
|  | ||||
| 		// handleStatusEvent returns false if the message should be dropped | ||||
| 		// in that case it is probably some modification to the channel we do not want to relay | ||||
| 		if b.handleStatusEvent(m, rmsg) { | ||||
| 			messages <- rmsg | ||||
| 		} | ||||
| 		messages <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) handleAttachments(message *models.Message, rmsg *config.Message) { | ||||
| 	if rmsg.Text == "" { | ||||
| 		for _, attachment := range message.Attachments { | ||||
| 			if attachment.Title != "" { | ||||
| 				rmsg.Text = attachment.Title + "\n" | ||||
| 			} | ||||
| 			if attachment.Title != "" && attachment.Text != "" { | ||||
| 				rmsg.Text += "\n" | ||||
| 			} | ||||
| 			if attachment.Text != "" { | ||||
| 				rmsg.Text += attachment.Text | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	for i := range message.Attachments { | ||||
| 		if err := b.handleDownloadFile(rmsg, &message.Attachments[i]); err != nil { | ||||
| 			b.Log.Errorf("Could not download incoming file: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) handleDownloadFile(rmsg *config.Message, file *models.Attachment) error { | ||||
| 	downloadURL := b.GetString("server") + file.TitleLink | ||||
| 	data, err := helper.DownloadFileAuthRocket(downloadURL, b.user.Token, b.user.ID) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("download %s failed %#v", downloadURL, err) | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, file.Title, rmsg.Text, downloadURL, data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) handleUploadFile(msg *config.Message) error { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
|   | ||||
| @@ -58,6 +58,9 @@ func (b *Brocketchat) doConnectWebhookURL() error { | ||||
| func (b *Brocketchat) apiLogin() error { | ||||
| 	b.Log.Debugf("handling apiLogin()") | ||||
| 	credentials := &models.UserCredentials{Email: b.GetString("login"), Password: b.GetString("password")} | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		credentials = &models.UserCredentials{ID: b.GetString("Login"), Token: b.GetString("Token")} | ||||
| 	} | ||||
| 	myURL, err := url.Parse(b.GetString("server")) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| @@ -95,7 +98,7 @@ func (b *Brocketchat) getChannelID(name string) string { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	for k, v := range b.channelMap { | ||||
| 		if v == name { | ||||
| 		if v == name || v == "#"+name { | ||||
| 			return k | ||||
| 		} | ||||
| 	} | ||||
| @@ -132,6 +135,7 @@ func (b *Brocketchat) uploadFile(fi *config.FileInfo, channel string) error { | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	defer resp.Body.Close() | ||||
| 	body, err := ioutil.ReadAll(resp.Body) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
|   | ||||
| @@ -2,6 +2,7 @@ package brocketchat | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| @@ -28,6 +29,12 @@ type Brocketchat struct { | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| const ( | ||||
| 	sUserJoined       = "uj" | ||||
| 	sUserLeft         = "ul" | ||||
| 	sRoomChangedTopic = "room_changed_topic" | ||||
| ) | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	newCache, err := lru.New(100) | ||||
| 	if err != nil { | ||||
| @@ -85,14 +92,14 @@ func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if b.c == nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	id, err := b.c.GetChannelId(channel.Name) | ||||
| 	id, err := b.c.GetChannelId(strings.TrimPrefix(channel.Name, "#")) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Lock() | ||||
| 	b.channelMap[id] = channel.Name | ||||
| 	b.Unlock() | ||||
| 	mychannel := &models.Channel{ID: id, Name: channel.Name} | ||||
| 	mychannel := &models.Channel{ID: id, Name: strings.TrimPrefix(channel.Name, "#")} | ||||
| 	if err := b.c.JoinChannel(id); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| @@ -103,8 +110,15 @@ func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||
| 	// strip the # if people has set this | ||||
| 	msg.Channel = strings.TrimPrefix(msg.Channel, "#") | ||||
| 	channel := &models.Channel{ID: b.getChannelID(msg.Channel), Name: msg.Channel} | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "_" + msg.Text + "_" | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| @@ -131,6 +145,8 @@ func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			// strip the # if people has set this | ||||
| 			rmsg.Channel = strings.TrimPrefix(rmsg.Channel, "#") | ||||
| 			smsg := &models.Message{ | ||||
| 				RoomID: b.getChannelID(rmsg.Channel), | ||||
| 				Msg:    rmsg.Username + rmsg.Text, | ||||
|   | ||||
| @@ -1,18 +1,22 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/nlopes/slack" | ||||
| 	"github.com/slack-go/slack" | ||||
| ) | ||||
|  | ||||
| // ErrEventIgnored is for events that should be ignored | ||||
| var ErrEventIgnored = errors.New("this event message should ignored") | ||||
|  | ||||
| func (b *Bslack) handleSlack() { | ||||
| 	messages := make(chan *config.Message) | ||||
| 	if b.GetString(incomingWebhookConfig) != "" { | ||||
| 	if b.GetString(incomingWebhookConfig) != "" && b.GetString(tokenConfig) == "" { | ||||
| 		b.Log.Debugf("Choosing webhooks based receiving") | ||||
| 		go b.handleMatterHook(messages) | ||||
| 	} else { | ||||
| @@ -22,20 +26,22 @@ func (b *Bslack) handleSlack() { | ||||
| 	time.Sleep(time.Second) | ||||
| 	b.Log.Debug("Start listening for Slack messages") | ||||
| 	for message := range messages { | ||||
| 		if message.Event != config.EventUserTyping { | ||||
| 		// don't do any action on deleted/typing messages | ||||
| 		if message.Event != config.EventUserTyping && message.Event != config.EventMsgDelete && | ||||
| 			message.Event != config.EventFileDelete { | ||||
| 			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 = b.replaceb0rkedMarkDown(message.Text) | ||||
| 			message.Text = html.UnescapeString(message.Text) | ||||
|  | ||||
| 			// Add the avatar | ||||
| 			message.Avatar = b.users.getAvatar(message.UserID) | ||||
| 		} | ||||
|  | ||||
| 		// 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 | ||||
| 	} | ||||
| @@ -43,7 +49,7 @@ func (b *Bslack) handleSlack() { | ||||
|  | ||||
| func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||
| 	for msg := range b.rtm.IncomingEvents { | ||||
| 		if msg.Type != sUserTyping && msg.Type != sLatencyReport { | ||||
| 		if msg.Type != sUserTyping && msg.Type != sHello && msg.Type != sLatencyReport { | ||||
| 			b.Log.Debugf("== Receiving event %#v", msg.Data) | ||||
| 		} | ||||
| 		switch ev := msg.Data.(type) { | ||||
| @@ -52,7 +58,9 @@ func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||
| 				continue | ||||
| 			} | ||||
| 			rmsg, err := b.handleTypingEvent(ev) | ||||
| 			if err != nil { | ||||
| 			if err == ErrEventIgnored { | ||||
| 				continue | ||||
| 			} else if err != nil { | ||||
| 				b.Log.Errorf("%#v", err) | ||||
| 				continue | ||||
| 			} | ||||
| @@ -69,28 +77,34 @@ func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||
| 				continue | ||||
| 			} | ||||
| 			messages <- rmsg | ||||
| 		case *slack.FileDeletedEvent: | ||||
| 			rmsg, err := b.handleFileDeletedEvent(ev) | ||||
| 			if err != nil { | ||||
| 				b.Log.Printf("%#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.channelsMutex.Lock() | ||||
| 			b.channelsByID[ev.Channel.ID] = &ev.Channel | ||||
| 			b.channelsByName[ev.Channel.Name] = &ev.Channel | ||||
| 			b.channelsMutex.Unlock() | ||||
| 			b.channels.registerChannel(ev.Channel) | ||||
| 		case *slack.ConnectedEvent: | ||||
| 			b.si = ev.Info | ||||
| 			b.populateChannels(true) | ||||
| 			b.populateUsers(true) | ||||
| 			b.channels.populateChannels(true) | ||||
| 			b.users.populateUsers(true) | ||||
| 		case *slack.InvalidAuthEvent: | ||||
| 			b.Log.Fatalf("Invalid Token %#v", ev) | ||||
| 		case *slack.ConnectionErrorEvent: | ||||
| 			b.Log.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj) | ||||
| 		case *slack.MemberJoinedChannelEvent: | ||||
| 			b.populateUser(ev.User) | ||||
| 		case *slack.LatencyReport: | ||||
| 			b.users.populateUser(ev.User) | ||||
| 		case *slack.HelloEvent, *slack.LatencyReport, *slack.ConnectingEvent: | ||||
| 			continue | ||||
| 		case *slack.UserChangeEvent: | ||||
| 			b.users.invalidateUser(ev.User.ID) | ||||
| 		default: | ||||
| 			b.Log.Debugf("Unhandled incoming event: %T", ev) | ||||
| 		} | ||||
| @@ -126,18 +140,34 @@ func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// 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 | ||||
| 	// Check for our callback ID | ||||
| 	hasOurCallbackID := false | ||||
| 	if len(ev.Blocks.BlockSet) == 1 { | ||||
| 		block, ok := ev.Blocks.BlockSet[0].(*slack.SectionBlock) | ||||
| 		hasOurCallbackID = ok && block.BlockID == "matterbridge_"+b.uuid | ||||
| 	} | ||||
|  | ||||
| 	// 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 { | ||||
| 	if ev.SubMessage != nil { | ||||
| 		// It seems ev.SubMessage.Edited == nil when slack unfurls. | ||||
| 		// Do not forward these messages. See Github issue #266. | ||||
| 		if ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp && | ||||
| 			ev.SubMessage.Edited == nil { | ||||
| 			return true | ||||
| 		} | ||||
| 		// see hidden subtypes at https://api.slack.com/events/message | ||||
| 		// these messages are sent when we add a message to a thread #709 | ||||
| 		if ev.SubType == "message_replied" && ev.Hidden { | ||||
| 			return true | ||||
| 		} | ||||
| 		if len(ev.SubMessage.Blocks.BlockSet) == 1 { | ||||
| 			block, ok := ev.SubMessage.Blocks.BlockSet[0].(*slack.SectionBlock) | ||||
| 			hasOurCallbackID = ok && block.BlockID == "matterbridge_"+b.uuid | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// 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) || hasOurCallbackID { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| @@ -200,6 +230,26 @@ func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, er | ||||
| 	return rmsg, nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleFileDeletedEvent(ev *slack.FileDeletedEvent) (*config.Message, error) { | ||||
| 	if rawChannel, ok := b.cache.Get(cfileDownloadChannel + ev.FileID); ok { | ||||
| 		channel, err := b.channels.getChannelByID(rawChannel.(string)) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
|  | ||||
| 		return &config.Message{ | ||||
| 			Event:    config.EventFileDelete, | ||||
| 			Text:     config.EventFileDelete, | ||||
| 			Channel:  channel.Name, | ||||
| 			Account:  b.Account, | ||||
| 			ID:       ev.FileID, | ||||
| 			Protocol: b.Protocol, | ||||
| 		}, nil | ||||
| 	} | ||||
|  | ||||
| 	return nil, fmt.Errorf("channel ID for file ID %s not found", ev.FileID) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) bool { | ||||
| 	switch ev.SubType { | ||||
| 	case sChannelJoined, sMemberJoined: | ||||
| @@ -210,7 +260,7 @@ func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) | ||||
| 		rmsg.Username = sSystemUser | ||||
| 		rmsg.Event = config.EventJoinLeave | ||||
| 	case sChannelTopic, sChannelPurpose: | ||||
| 		b.populateChannels(false) | ||||
| 		b.channels.populateChannels(false) | ||||
| 		rmsg.Event = config.EventTopicChange | ||||
| 	case sMessageChanged: | ||||
| 		rmsg.Text = ev.SubMessage.Text | ||||
| @@ -232,6 +282,13 @@ func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func getMessageTitle(attach *slack.Attachment) string { | ||||
| 	if attach.TitleLink != "" { | ||||
| 		return fmt.Sprintf("[%s](%s)\n", attach.Title, attach.TitleLink) | ||||
| 	} | ||||
| 	return attach.Title | ||||
| } | ||||
|  | ||||
| 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 { | ||||
| @@ -240,12 +297,15 @@ func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) | ||||
|  | ||||
| 	// See if we have some text in the attachments. | ||||
| 	if rmsg.Text == "" { | ||||
| 		for _, attach := range ev.Attachments { | ||||
| 		for i, attach := range ev.Attachments { | ||||
| 			if attach.Text != "" { | ||||
| 				if attach.Title != "" { | ||||
| 					rmsg.Text = attach.Title + "\n" | ||||
| 					rmsg.Text = getMessageTitle(&ev.Attachments[i]) | ||||
| 				} | ||||
| 				rmsg.Text += attach.Text | ||||
| 				if attach.Footer != "" { | ||||
| 					rmsg.Text += "\n\n" + attach.Footer | ||||
| 				} | ||||
| 			} else { | ||||
| 				rmsg.Text = attach.Fallback | ||||
| 			} | ||||
| @@ -259,6 +319,8 @@ func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) | ||||
|  | ||||
| 	// If we have files attached, download them (in memory) and put a pointer to it in msg.Extra. | ||||
| 	for i := range ev.Files { | ||||
| 		// keep reference in cache on which channel we added this file | ||||
| 		b.cache.Add(cfileDownloadChannel+ev.Files[i].ID, ev.Channel) | ||||
| 		if err := b.handleDownloadFile(rmsg, &ev.Files[i], false); err != nil { | ||||
| 			b.Log.Errorf("Could not download incoming file: %#v", err) | ||||
| 		} | ||||
| @@ -266,7 +328,10 @@ func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message, error) { | ||||
| 	channelInfo, err := b.getChannelByID(ev.Channel) | ||||
| 	if ev.User == b.si.User.ID { | ||||
| 		return nil, ErrEventIgnored | ||||
| 	} | ||||
| 	channelInfo, err := b.channels.getChannelByID(ev.Channel) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| @@ -305,7 +370,7 @@ func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File, retr | ||||
| 	// that the comment is not duplicated. | ||||
| 	comment := rmsg.Text | ||||
| 	rmsg.Text = "" | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, file.Name, comment, file.URLPrivateDownload, data, b.General) | ||||
| 	helper.HandleDownloadData2(b.Log, rmsg, file.Name, file.ID, comment, file.URLPrivateDownload, data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -316,36 +381,7 @@ func (b *Bslack) handleGetChannelMembers(rmsg *config.Message) bool { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	cMembers := config.ChannelMembers{} | ||||
|  | ||||
| 	b.channelMembersMutex.RLock() | ||||
|  | ||||
| 	for channelID, members := range b.channelMembers { | ||||
| 		for _, member := range members { | ||||
| 			channelName := "" | ||||
| 			userName := "" | ||||
| 			userNick := "" | ||||
| 			user := b.getUser(member) | ||||
| 			if user != nil { | ||||
| 				userName = user.Name | ||||
| 				userNick = user.Profile.DisplayName | ||||
| 			} | ||||
| 			channel, _ := b.getChannelByID(channelID) | ||||
| 			if channel != nil { | ||||
| 				channelName = channel.Name | ||||
| 			} | ||||
| 			cMember := config.ChannelMember{ | ||||
| 				Username:    userName, | ||||
| 				Nick:        userNick, | ||||
| 				UserID:      member, | ||||
| 				ChannelID:   channelID, | ||||
| 				ChannelName: channelName, | ||||
| 			} | ||||
| 			cMembers = append(cMembers, cMember) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.channelMembersMutex.RUnlock() | ||||
| 	cMembers := b.channels.getChannelMembers(b.users) | ||||
|  | ||||
| 	extra := make(map[string][]interface{}) | ||||
| 	extra[config.EventGetChannelMembers] = append(extra[config.EventGetChannelMembers], cMembers) | ||||
|   | ||||
| @@ -1,233 +1,21 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/nlopes/slack" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/slack-go/slack" | ||||
| ) | ||||
|  | ||||
| func (b *Bslack) getUser(id string) *slack.User { | ||||
| 	b.usersMutex.RLock() | ||||
| 	user, ok := b.users[id] | ||||
| 	b.usersMutex.RUnlock() | ||||
| 	if ok { | ||||
| 		return user | ||||
| 	} | ||||
| 	b.populateUser(id) | ||||
| 	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) { | ||||
| 	return b.getChannelBy(name, b.channelsByName) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) { | ||||
| 	return b.getChannelBy(ID, b.channelsByID) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelBy(lookupKey string, lookupMap map[string]*slack.Channel) (*slack.Channel, error) { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	if channel, ok := lookupMap[lookupKey]; ok { | ||||
| 		return channel, nil | ||||
| 	} | ||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, lookupKey) | ||||
| } | ||||
|  | ||||
| const minimumRefreshInterval = 10 * time.Second | ||||
|  | ||||
| func (b *Bslack) populateUser(userID string) { | ||||
| 	b.usersMutex.RLock() | ||||
| 	_, exists := b.users[userID] | ||||
| 	b.usersMutex.RUnlock() | ||||
| 	if exists { | ||||
| 		// already in cache | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	user, err := b.sc.GetUserInfo(userID) | ||||
| 	if err != nil { | ||||
| 		b.Log.Debugf("GetUserInfo failed for %v: %v", userID, err) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	b.usersMutex.Lock() | ||||
| 	b.users[userID] = user | ||||
| 	b.usersMutex.Unlock() | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateUsers(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (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 | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newUsers := map[string]*slack.User{} | ||||
| 	pagination := b.sc.GetUsersPaginated(slack.GetUsersOptionLimit(200)) | ||||
| 	count := 0 | ||||
| 	for { | ||||
| 		var err error | ||||
| 		pagination, err = pagination.Next(context.Background()) | ||||
| 		time.Sleep(time.Second) | ||||
| 		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.Log.Debugf("getting %d users", len(pagination.Users)) | ||||
| 		count++ | ||||
| 		// more > 2000 users, slack will complain and ratelimit. break | ||||
| 		if count > 10 { | ||||
| 			b.Log.Info("Large slack detected > 2000 users, skipping loading complete userlist.") | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	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(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (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 | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newChannelsByID := map[string]*slack.Channel{} | ||||
| 	newChannelsByName := map[string]*slack.Channel{} | ||||
| 	newChannelMembers := make(map[string][]string) | ||||
|  | ||||
| 	// 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] | ||||
| 			// also find all the members in every channel | ||||
| 			// comment for now, issues on big slacks | ||||
| 			/* | ||||
| 				members, err := b.getUsersInConversation(channels[i].ID) | ||||
| 				if err != nil { | ||||
| 					if err = b.handleRateLimit(err); err != nil { | ||||
| 						b.Log.Errorf("Could not retrieve channel members: %#v", err) | ||||
| 						return | ||||
| 					} | ||||
| 					continue | ||||
| 				} | ||||
| 				newChannelMembers[channels[i].ID] = members | ||||
| 			*/ | ||||
| 		} | ||||
|  | ||||
| 		if nextCursor == "" { | ||||
| 			break | ||||
| 		} | ||||
| 		queryParams.Cursor = nextCursor | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.Lock() | ||||
| 	defer b.channelsMutex.Unlock() | ||||
| 	b.channelsByID = newChannelsByID | ||||
| 	b.channelsByName = newChannelsByName | ||||
|  | ||||
| 	b.channelMembersMutex.Lock() | ||||
| 	defer b.channelMembersMutex.Unlock() | ||||
| 	b.channelMembers = newChannelMembers | ||||
|  | ||||
| 	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) | ||||
| 	channel, err := b.channels.getChannelByID(ev.Channel) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| @@ -289,7 +77,7 @@ func (b *Bslack) populateMessageWithUserInfo(ev *slack.MessageEvent, rmsg *confi | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	user := b.getUser(userID) | ||||
| 	user := b.users.getUser(userID) | ||||
| 	if user == nil { | ||||
| 		return fmt.Errorf("could not find information for user with id %s", ev.User) | ||||
| 	} | ||||
| @@ -299,6 +87,9 @@ func (b *Bslack) populateMessageWithUserInfo(ev *slack.MessageEvent, rmsg *confi | ||||
| 	if user.Profile.DisplayName != "" { | ||||
| 		rmsg.Username = user.Profile.DisplayName | ||||
| 	} | ||||
| 	if b.GetBool("UseFullName") && user.Profile.RealName != "" { | ||||
| 		rmsg.Username = user.Profile.RealName | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -310,12 +101,14 @@ func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config | ||||
| 	var err error | ||||
| 	var bot *slack.Bot | ||||
| 	for { | ||||
| 		bot, err = b.rtm.GetBotInfo(ev.BotID) | ||||
| 		bot, err = b.rtm.GetBotInfo(slack.GetBotInfoParameters{ | ||||
| 			Bot: ev.BotID, | ||||
| 		}) | ||||
| 		if err == nil { | ||||
| 			break | ||||
| 		} | ||||
|  | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 		if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 			b.Log.Errorf("Could not retrieve bot information: %#v", err) | ||||
| 			return err | ||||
| 		} | ||||
| @@ -336,7 +129,7 @@ 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(`<(.*?)(\|.*?)?>`) | ||||
| 	urlRE            = regexp.MustCompile(`<([^<\|]+)\|([^>]+)>`) | ||||
| 	codeFenceRE      = regexp.MustCompile(`(?m)^` + "```" + `\w+$`) | ||||
| 	topicOrPurposeRE = regexp.MustCompile(`(?s)(@.+) (cleared|set)(?: the)? channel (topic|purpose)(?:: (.*))?`) | ||||
| ) | ||||
| @@ -360,7 +153,7 @@ func (b *Bslack) extractTopicOrPurpose(text string) (string, string) { | ||||
| func (b *Bslack) replaceMention(text string) string { | ||||
| 	replaceFunc := func(match string) string { | ||||
| 		userID := strings.Trim(match, "@<>") | ||||
| 		if username := b.getUsername(userID); userID != "" { | ||||
| 		if username := b.users.getUsername(userID); userID != "" { | ||||
| 			return "@" + username | ||||
| 		} | ||||
| 		return match | ||||
| @@ -390,12 +183,35 @@ func (b *Bslack) replaceVariable(text string) string { | ||||
|  | ||||
| // @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 urlRE.ReplaceAllString(text, "[${2}](${1})") | ||||
| } | ||||
|  | ||||
| func (b *Bslack) replaceb0rkedMarkDown(text string) string { | ||||
| 	// taken from https://github.com/mattermost/mattermost-server/blob/master/app/slackimport.go | ||||
| 	// | ||||
| 	regexReplaceAllString := []struct { | ||||
| 		regex *regexp.Regexp | ||||
| 		rpl   string | ||||
| 	}{ | ||||
| 		// bold | ||||
| 		{ | ||||
| 			regexp.MustCompile(`(^|[\s.;,])\*(\S[^*\n]+)\*`), | ||||
| 			"$1**$2**", | ||||
| 		}, | ||||
| 		// strikethrough | ||||
| 		{ | ||||
| 			regexp.MustCompile(`(^|[\s.;,])\~(\S[^~\n]+)\~`), | ||||
| 			"$1~~$2~~", | ||||
| 		}, | ||||
| 		// single paragraph blockquote | ||||
| 		// Slack converts > character to > | ||||
| 		{ | ||||
| 			regexp.MustCompile(`(?sm)^>`), | ||||
| 			">", | ||||
| 		}, | ||||
| 	} | ||||
| 	for _, rule := range regexReplaceAllString { | ||||
| 		text = rule.regex.ReplaceAllString(text, rule.rpl) | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
| @@ -404,16 +220,6 @@ func (b *Bslack) replaceCodeFence(text string) string { | ||||
| 	return codeFenceRE.ReplaceAllString(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 | ||||
| } | ||||
|  | ||||
| // getUsersInConversation returns an array of userIDs that are members of channelID | ||||
| func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) { | ||||
| 	channelMembers := []string{} | ||||
| @@ -424,7 +230,7 @@ func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) { | ||||
|  | ||||
| 		members, nextCursor, err := b.sc.GetUsersInConversation(queryParams) | ||||
| 		if err != nil { | ||||
| 			if err = b.handleRateLimit(err); err != nil { | ||||
| 			if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 				return channelMembers, fmt.Errorf("Could not retrieve users in channels: %#v", err) | ||||
| 			} | ||||
| 			continue | ||||
| @@ -439,3 +245,13 @@ func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) { | ||||
| 	} | ||||
| 	return channelMembers, nil | ||||
| } | ||||
|  | ||||
| func handleRateLimit(log *logrus.Entry, err error) error { | ||||
| 	rateLimit, ok := err.(*slack.RateLimitedError) | ||||
| 	if !ok { | ||||
| 		return err | ||||
| 	} | ||||
| 	log.Infof("Rate-limited by Slack. Sleeping for %v", rateLimit.RetryAfter) | ||||
| 	time.Sleep(rateLimit.RetryAfter) | ||||
| 	return nil | ||||
| } | ||||
|   | ||||
| @@ -5,7 +5,7 @@ import ( | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/nlopes/slack" | ||||
| 	"github.com/slack-go/slack" | ||||
| ) | ||||
|  | ||||
| type BLegacy struct { | ||||
| @@ -13,7 +13,9 @@ type BLegacy struct { | ||||
| } | ||||
|  | ||||
| func NewLegacy(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &BLegacy{Bslack: newBridge(cfg)} | ||||
| 	b := &BLegacy{Bslack: newBridge(cfg)} | ||||
| 	b.legacy = true | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *BLegacy) Connect() error { | ||||
| @@ -55,14 +57,18 @@ func (b *BLegacy) Connect() error { | ||||
| 		}) | ||||
| 		if b.GetString(tokenConfig) != "" { | ||||
| 			b.Log.Info("Connecting using token (receiving)") | ||||
| 			b.sc = slack.New(b.GetString(tokenConfig)) | ||||
| 			b.sc = slack.New(b.GetString(tokenConfig), slack.OptionDebug(b.GetBool("debug"))) | ||||
| 			b.channels = newChannelManager(b.Log, b.sc) | ||||
| 			b.users = newUserManager(b.Log, b.sc) | ||||
| 			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.sc = slack.New(b.GetString(tokenConfig), slack.OptionDebug(b.GetBool("debug"))) | ||||
| 		b.channels = newChannelManager(b.Log, b.sc) | ||||
| 		b.users = newUserManager(b.Log, b.sc) | ||||
| 		b.rtm = b.sc.NewRTM() | ||||
| 		go b.rtm.ManageConnection() | ||||
| 		go b.handleSlack() | ||||
|   | ||||
| @@ -12,9 +12,9 @@ import ( | ||||
| 	"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" | ||||
| 	lru "github.com/hashicorp/golang-lru" | ||||
| 	"github.com/rs/xid" | ||||
| 	"github.com/slack-go/slack" | ||||
| ) | ||||
|  | ||||
| type Bslack struct { | ||||
| @@ -30,40 +30,31 @@ type Bslack struct { | ||||
| 	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 | ||||
|  | ||||
| 	channelMembers      map[string][]string | ||||
| 	channelMembersMutex sync.RWMutex | ||||
|  | ||||
| 	refreshInProgress      bool | ||||
| 	earliestChannelRefresh time.Time | ||||
| 	earliestUserRefresh    time.Time | ||||
| 	refreshMutex           sync.Mutex | ||||
| 	channels *channels | ||||
| 	users    *users | ||||
| 	legacy   bool | ||||
| } | ||||
|  | ||||
| const ( | ||||
| 	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" | ||||
| 	sHello               = "hello" | ||||
| 	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" | ||||
| 	cfileDownloadChannel = "file_download_channel" | ||||
|  | ||||
| 	tokenConfig           = "Token" | ||||
| 	incomingWebhookConfig = "WebhookBindAddress" | ||||
| @@ -74,6 +65,7 @@ const ( | ||||
| 	editSuffixConfig      = "EditSuffix" | ||||
| 	iconURLConfig         = "iconurl" | ||||
| 	noSendJoinConfig      = "nosendjoinpart" | ||||
| 	messageLength         = 3000 | ||||
| ) | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| @@ -94,14 +86,9 @@ func newBridge(cfg *bridge.Config) *Bslack { | ||||
| 		cfg.Log.Fatalf("Could not create LRU cache for Slack bridge: %v", err) | ||||
| 	} | ||||
| 	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(), | ||||
| 		Config: cfg, | ||||
| 		uuid:   xid.New().String(), | ||||
| 		cache:  newCache, | ||||
| 	} | ||||
| 	return b | ||||
| } | ||||
| @@ -121,7 +108,12 @@ func (b *Bslack) Connect() error { | ||||
| 	// If we have a token we use the Slack websocket-based RTM for both sending and receiving. | ||||
| 	if token := b.GetString(tokenConfig); token != "" { | ||||
| 		b.Log.Info("Connecting using token") | ||||
|  | ||||
| 		b.sc = slack.New(token, slack.OptionDebug(b.GetBool("Debug"))) | ||||
|  | ||||
| 		b.channels = newChannelManager(b.Log, b.sc) | ||||
| 		b.users = newUserManager(b.Log, b.sc) | ||||
|  | ||||
| 		b.rtm = b.sc.NewRTM() | ||||
| 		go b.rtm.ManageConnection() | ||||
| 		go b.handleSlack() | ||||
| @@ -163,9 +155,21 @@ func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	b.populateChannels(false) | ||||
| 	// try to join a channel when in legacy | ||||
| 	if b.legacy { | ||||
| 		_, _, _, err := b.sc.JoinConversation(channel.Name) | ||||
| 		if err != nil { | ||||
| 			switch err.Error() { | ||||
| 			case "name_taken", "restricted_action": | ||||
| 			case "default": | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	channelInfo, err := b.getChannel(channel.Name) | ||||
| 	b.channels.populateChannels(false) | ||||
|  | ||||
| 	channelInfo, err := b.channels.getChannel(channel.Name) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("could not join channel: %#v", err) | ||||
| 	} | ||||
| @@ -175,7 +179,8 @@ func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||
| 		channel.Name = channelInfo.Name | ||||
| 	} | ||||
|  | ||||
| 	if !channelInfo.IsMember { | ||||
| 	// we can't join a channel unless we are using legacy tokens #651 | ||||
| 	if !channelInfo.IsMember && !b.legacy { | ||||
| 		return fmt.Errorf("slack integration that matterbridge is using is not member of channel '%s', please add it manually", channelInfo.Name) | ||||
| 	} | ||||
| 	return nil | ||||
| @@ -191,6 +196,7 @@ func (b *Bslack) Send(msg config.Message) (string, error) { | ||||
| 		b.Log.Debugf("=> Receiving %#v", msg) | ||||
| 	} | ||||
|  | ||||
| 	msg.Text = helper.ClipMessage(msg.Text, messageLength, b.GetString("MessageClipped")) | ||||
| 	msg.Text = b.replaceCodeFence(msg.Text) | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| @@ -199,7 +205,7 @@ func (b *Bslack) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Use webhook to send the message | ||||
| 	if b.GetString(outgoingWebhookConfig) != "" { | ||||
| 	if b.GetString(outgoingWebhookConfig) != "" && b.GetString(tokenConfig) == "" { | ||||
| 		return "", b.sendWebhook(msg) | ||||
| 	} | ||||
| 	return b.sendRTM(msg) | ||||
| @@ -275,7 +281,7 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	channelInfo, err := b.getChannel(msg.Channel) | ||||
| 	channelInfo, err := b.channels.getChannel(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", fmt.Errorf("could not send message: %v", err) | ||||
| 	} | ||||
| @@ -294,7 +300,7 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
| 	if msg.ParentID == "msg-parent-not-found" { | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 		msg.Text = fmt.Sprintf("[thread]: %s", msg.Text) | ||||
| 	} | ||||
| @@ -315,7 +321,7 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file if it exists. | ||||
| 	if msg.Extra != nil { | ||||
| 	if len(msg.Extra) > 0 { | ||||
| 		extraMsgs := helper.HandleExtra(&msg, b.General) | ||||
| 		for i := range extraMsgs { | ||||
| 			rmsg := &extraMsgs[i] | ||||
| @@ -326,7 +332,7 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 			} | ||||
| 		} | ||||
| 		// Upload files if necessary (from Slack, Telegram or Mattermost). | ||||
| 		b.uploadFile(&msg, channelInfo.ID) | ||||
| 		return b.uploadFile(&msg, channelInfo.ID) | ||||
| 	} | ||||
|  | ||||
| 	// Post message. | ||||
| @@ -351,7 +357,7 @@ func (b *Bslack) updateTopicOrPurpose(msg *config.Message, channelInfo *slack.Ch | ||||
| 		if err == nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 		if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| @@ -392,7 +398,7 @@ func (b *Bslack) deleteMessage(msg *config.Message, channelInfo *slack.Channel) | ||||
| 			return true, nil | ||||
| 		} | ||||
|  | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 		if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 			b.Log.Errorf("Failed to delete user message from Slack: %#v", err) | ||||
| 			return true, err | ||||
| 		} | ||||
| @@ -405,13 +411,12 @@ func (b *Bslack) editMessage(msg *config.Message, channelInfo *slack.Channel) (b | ||||
| 	} | ||||
| 	messageOptions := b.prepareMessageOptions(msg) | ||||
| 	for { | ||||
| 		messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false)) | ||||
| 		_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, messageOptions...) | ||||
| 		if err == nil { | ||||
| 			return true, nil | ||||
| 		} | ||||
|  | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 		if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 			b.Log.Errorf("Failed to edit user message on Slack: %#v", err) | ||||
| 			return true, err | ||||
| 		} | ||||
| @@ -424,14 +429,13 @@ func (b *Bslack) postMessage(msg *config.Message, channelInfo *slack.Channel) (s | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	messageOptions := b.prepareMessageOptions(msg) | ||||
| 	messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false)) | ||||
| 	for { | ||||
| 		_, id, err := b.rtm.PostMessage(channelInfo.ID, messageOptions...) | ||||
| 		if err == nil { | ||||
| 			return id, nil | ||||
| 		} | ||||
|  | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 		if err = handleRateLimit(b.Log, err); err != nil { | ||||
| 			b.Log.Errorf("Failed to sent user message to Slack: %#v", err) | ||||
| 			return "", err | ||||
| 		} | ||||
| @@ -439,7 +443,8 @@ func (b *Bslack) postMessage(msg *config.Message, channelInfo *slack.Channel) (s | ||||
| } | ||||
|  | ||||
| // uploadFile handles native upload of files | ||||
| func (b *Bslack) uploadFile(msg *config.Message, channelID string) { | ||||
| func (b *Bslack) uploadFile(msg *config.Message, channelID string) (string, error) { | ||||
| 	var messageID string | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi, ok := f.(config.FileInfo) | ||||
| 		if !ok { | ||||
| @@ -456,7 +461,7 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID 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) | ||||
| 			initialComment += fmt.Sprintf(" with comment: %s", fi.Comment) | ||||
| 		} | ||||
| 		res, err := b.sc.UploadFile(slack.FileUploadParameters{ | ||||
| 			Reader:          bytes.NewReader(*fi.Data), | ||||
| @@ -467,13 +472,22 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID string) { | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("uploadfile %#v", err) | ||||
| 			return | ||||
| 			return "", err | ||||
| 		} | ||||
| 		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) | ||||
|  | ||||
| 			// search for message id by uploaded file in private/public channels, get thread timestamp from uploaded file | ||||
| 			if v, ok := res.Shares.Private[channelID]; ok && len(v) > 0 { | ||||
| 				messageID = v[0].Ts | ||||
| 			} | ||||
| 			if v, ok := res.Shares.Public[channelID]; ok && len(v) > 0 { | ||||
| 				messageID = v[0].Ts | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return messageID, nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption { | ||||
| @@ -490,8 +504,6 @@ func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption { | ||||
| 	} | ||||
|  | ||||
| 	var attachments []slack.Attachment | ||||
| 	// add a callback ID so we can see we created it | ||||
| 	attachments = append(attachments, slack.Attachment{CallbackID: "matterbridge_" + b.uuid}) | ||||
| 	// add file attachments | ||||
| 	attachments = append(attachments, b.createAttach(msg.Extra)...) | ||||
| 	// add slack attachments (from another slack bridge) | ||||
| @@ -502,6 +514,19 @@ func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption { | ||||
| 	} | ||||
|  | ||||
| 	var opts []slack.MsgOption | ||||
| 	opts = append(opts, | ||||
| 		// provide regular text field (fallback used in Slack notifications, etc.) | ||||
| 		slack.MsgOptionText(msg.Text, false), | ||||
|  | ||||
| 		// add a callback ID so we can see we created it | ||||
| 		slack.MsgOptionBlocks(slack.NewSectionBlock( | ||||
| 			slack.NewTextBlockObject(slack.MarkdownType, msg.Text, false, false), | ||||
| 			nil, nil, | ||||
| 			slack.SectionBlockOptionBlockID("matterbridge_"+b.uuid), | ||||
| 		)), | ||||
|  | ||||
| 		slack.MsgOptionEnableLinkUnfurl(), | ||||
| 	) | ||||
| 	opts = append(opts, slack.MsgOptionAttachments(attachments...)) | ||||
| 	opts = append(opts, slack.MsgOptionPostMessageParameters(params)) | ||||
| 	return opts | ||||
|   | ||||
							
								
								
									
										343
									
								
								bridge/slack/users_channels.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										343
									
								
								bridge/slack/users_channels.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,343 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/slack-go/slack" | ||||
| ) | ||||
|  | ||||
| const minimumRefreshInterval = 10 * time.Second | ||||
|  | ||||
| type users struct { | ||||
| 	log *logrus.Entry | ||||
| 	sc  *slack.Client | ||||
|  | ||||
| 	users           map[string]*slack.User | ||||
| 	usersMutex      sync.RWMutex | ||||
| 	usersSyncPoints map[string]chan struct{} | ||||
|  | ||||
| 	refreshInProgress bool | ||||
| 	earliestRefresh   time.Time | ||||
| 	refreshMutex      sync.Mutex | ||||
| } | ||||
|  | ||||
| func newUserManager(log *logrus.Entry, sc *slack.Client) *users { | ||||
| 	return &users{ | ||||
| 		log:             log, | ||||
| 		sc:              sc, | ||||
| 		users:           make(map[string]*slack.User), | ||||
| 		usersSyncPoints: make(map[string]chan struct{}), | ||||
| 		earliestRefresh: time.Now(), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *users) getUser(id string) *slack.User { | ||||
| 	b.usersMutex.RLock() | ||||
| 	user, ok := b.users[id] | ||||
| 	b.usersMutex.RUnlock() | ||||
| 	if ok { | ||||
| 		return user | ||||
| 	} | ||||
| 	b.populateUser(id) | ||||
| 	b.usersMutex.RLock() | ||||
| 	defer b.usersMutex.RUnlock() | ||||
|  | ||||
| 	return b.users[id] | ||||
| } | ||||
|  | ||||
| func (b *users) 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 *users) getAvatar(id string) string { | ||||
| 	if user := b.getUser(id); user != nil { | ||||
| 		return user.Profile.Image48 | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *users) populateUser(userID string) { | ||||
| 	for { | ||||
| 		b.usersMutex.Lock() | ||||
| 		_, exists := b.users[userID] | ||||
| 		if exists { | ||||
| 			// already in cache | ||||
| 			b.usersMutex.Unlock() | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		if syncPoint, ok := b.usersSyncPoints[userID]; ok { | ||||
| 			// Another goroutine is already populating this user for us so wait on it to finish. | ||||
| 			b.usersMutex.Unlock() | ||||
| 			<-syncPoint | ||||
| 			// We do not return and iterate again to check that the entry does indeed exist | ||||
| 			// in case the previous query failed for some reason. | ||||
| 		} else { | ||||
| 			b.usersSyncPoints[userID] = make(chan struct{}) | ||||
| 			defer func() { | ||||
| 				// Wake up any waiting goroutines and remove the synchronization point. | ||||
| 				close(b.usersSyncPoints[userID]) | ||||
| 				delete(b.usersSyncPoints, userID) | ||||
| 			}() | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Do not hold the lock while fetching information from Slack | ||||
| 	// as this might take an unbounded amount of time. | ||||
| 	b.usersMutex.Unlock() | ||||
|  | ||||
| 	user, err := b.sc.GetUserInfo(userID) | ||||
| 	if err != nil { | ||||
| 		b.log.Debugf("GetUserInfo failed for %v: %v", userID, err) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	b.usersMutex.Lock() | ||||
| 	defer b.usersMutex.Unlock() | ||||
|  | ||||
| 	// Register user information. | ||||
| 	b.users[userID] = user | ||||
| } | ||||
|  | ||||
| func (b *users) invalidateUser(userID string) { | ||||
| 	b.usersMutex.Lock() | ||||
| 	defer b.usersMutex.Unlock() | ||||
| 	delete(b.users, userID) | ||||
| } | ||||
|  | ||||
| func (b *users) populateUsers(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (time.Now().Before(b.earliestRefresh) || b.refreshInProgress) { | ||||
| 		b.log.Debugf("Not refreshing user list as it was done less than %v ago.", minimumRefreshInterval) | ||||
| 		b.refreshMutex.Unlock() | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newUsers := map[string]*slack.User{} | ||||
| 	pagination := b.sc.GetUsersPaginated(slack.GetUsersOptionLimit(200)) | ||||
| 	count := 0 | ||||
| 	for { | ||||
| 		var err error | ||||
| 		pagination, err = pagination.Next(context.Background()) | ||||
| 		time.Sleep(time.Second) | ||||
| 		if err != nil { | ||||
| 			if pagination.Done(err) { | ||||
| 				break | ||||
| 			} | ||||
|  | ||||
| 			if err = handleRateLimit(b.log, 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.log.Debugf("getting %d users", len(pagination.Users)) | ||||
| 		count++ | ||||
| 		// more > 2000 users, slack will complain and ratelimit. break | ||||
| 		if count > 10 { | ||||
| 			b.log.Info("Large slack detected > 2000 users, skipping loading complete userlist.") | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.usersMutex.Lock() | ||||
| 	defer b.usersMutex.Unlock() | ||||
| 	b.users = newUsers | ||||
|  | ||||
| 	b.refreshMutex.Lock() | ||||
| 	defer b.refreshMutex.Unlock() | ||||
| 	b.earliestRefresh = time.Now().Add(minimumRefreshInterval) | ||||
| 	b.refreshInProgress = false | ||||
| } | ||||
|  | ||||
| type channels struct { | ||||
| 	log *logrus.Entry | ||||
| 	sc  *slack.Client | ||||
|  | ||||
| 	channelsByID   map[string]*slack.Channel | ||||
| 	channelsByName map[string]*slack.Channel | ||||
| 	channelsMutex  sync.RWMutex | ||||
|  | ||||
| 	channelMembers      map[string][]string | ||||
| 	channelMembersMutex sync.RWMutex | ||||
|  | ||||
| 	refreshInProgress bool | ||||
| 	earliestRefresh   time.Time | ||||
| 	refreshMutex      sync.Mutex | ||||
| } | ||||
|  | ||||
| func newChannelManager(log *logrus.Entry, sc *slack.Client) *channels { | ||||
| 	return &channels{ | ||||
| 		log:             log, | ||||
| 		sc:              sc, | ||||
| 		channelsByID:    make(map[string]*slack.Channel), | ||||
| 		channelsByName:  make(map[string]*slack.Channel), | ||||
| 		earliestRefresh: time.Now(), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *channels) getChannel(channel string) (*slack.Channel, error) { | ||||
| 	if strings.HasPrefix(channel, "ID:") { | ||||
| 		return b.getChannelByID(strings.TrimPrefix(channel, "ID:")) | ||||
| 	} | ||||
| 	return b.getChannelByName(channel) | ||||
| } | ||||
|  | ||||
| func (b *channels) getChannelByName(name string) (*slack.Channel, error) { | ||||
| 	return b.getChannelBy(name, b.channelsByName) | ||||
| } | ||||
|  | ||||
| func (b *channels) getChannelByID(id string) (*slack.Channel, error) { | ||||
| 	return b.getChannelBy(id, b.channelsByID) | ||||
| } | ||||
|  | ||||
| func (b *channels) getChannelBy(lookupKey string, lookupMap map[string]*slack.Channel) (*slack.Channel, error) { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	if channel, ok := lookupMap[lookupKey]; ok { | ||||
| 		return channel, nil | ||||
| 	} | ||||
| 	return nil, fmt.Errorf("channel %s not found", lookupKey) | ||||
| } | ||||
|  | ||||
| func (b *channels) getChannelMembers(users *users) config.ChannelMembers { | ||||
| 	b.channelMembersMutex.RLock() | ||||
| 	defer b.channelMembersMutex.RUnlock() | ||||
|  | ||||
| 	membersInfo := config.ChannelMembers{} | ||||
| 	for channelID, members := range b.channelMembers { | ||||
| 		for _, member := range members { | ||||
| 			channelName := "" | ||||
| 			userName := "" | ||||
| 			userNick := "" | ||||
| 			user := users.getUser(member) | ||||
| 			if user != nil { | ||||
| 				userName = user.Name | ||||
| 				userNick = user.Profile.DisplayName | ||||
| 			} | ||||
| 			channel, _ := b.getChannelByID(channelID) | ||||
| 			if channel != nil { | ||||
| 				channelName = channel.Name | ||||
| 			} | ||||
| 			memberInfo := config.ChannelMember{ | ||||
| 				Username:    userName, | ||||
| 				Nick:        userNick, | ||||
| 				UserID:      member, | ||||
| 				ChannelID:   channelID, | ||||
| 				ChannelName: channelName, | ||||
| 			} | ||||
| 			membersInfo = append(membersInfo, memberInfo) | ||||
| 		} | ||||
| 	} | ||||
| 	return membersInfo | ||||
| } | ||||
|  | ||||
| func (b *channels) registerChannel(channel slack.Channel) { | ||||
| 	b.channelsMutex.Lock() | ||||
| 	defer b.channelsMutex.Unlock() | ||||
|  | ||||
| 	b.channelsByID[channel.ID] = &channel | ||||
| 	b.channelsByName[channel.Name] = &channel | ||||
| } | ||||
|  | ||||
| func (b *channels) populateChannels(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (time.Now().Before(b.earliestRefresh) || b.refreshInProgress) { | ||||
| 		b.log.Debugf("Not refreshing channel list as it was done less than %v seconds ago.", minimumRefreshInterval) | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		return | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newChannelsByID := map[string]*slack.Channel{} | ||||
| 	newChannelsByName := map[string]*slack.Channel{} | ||||
| 	newChannelMembers := make(map[string][]string) | ||||
|  | ||||
| 	// 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"}, | ||||
| 		Limit:           1000, | ||||
| 	} | ||||
| 	for { | ||||
| 		channels, nextCursor, err := b.sc.GetConversations(queryParams) | ||||
| 		if err != nil { | ||||
| 			if err = handleRateLimit(b.log, 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] | ||||
| 			// also find all the members in every channel | ||||
| 			// comment for now, issues on big slacks | ||||
| 			/* | ||||
| 				members, err := b.getUsersInConversation(channels[i].ID) | ||||
| 				if err != nil { | ||||
| 					if err = b.handleRateLimit(err); err != nil { | ||||
| 						b.Log.Errorf("Could not retrieve channel members: %#v", err) | ||||
| 						return | ||||
| 					} | ||||
| 					continue | ||||
| 				} | ||||
| 				newChannelMembers[channels[i].ID] = members | ||||
| 			*/ | ||||
| 		} | ||||
|  | ||||
| 		if nextCursor == "" { | ||||
| 			break | ||||
| 		} | ||||
| 		queryParams.Cursor = nextCursor | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.Lock() | ||||
| 	defer b.channelsMutex.Unlock() | ||||
| 	b.channelsByID = newChannelsByID | ||||
| 	b.channelsByName = newChannelsByName | ||||
|  | ||||
| 	b.channelMembersMutex.Lock() | ||||
| 	defer b.channelMembersMutex.Unlock() | ||||
| 	b.channelMembers = newChannelMembers | ||||
|  | ||||
| 	b.refreshMutex.Lock() | ||||
| 	defer b.refreshMutex.Unlock() | ||||
| 	b.earliestRefresh = time.Now().Add(minimumRefreshInterval) | ||||
| 	b.refreshInProgress = false | ||||
| } | ||||
| @@ -130,6 +130,10 @@ func (b *Bsshchat) handleSSHChat() error { | ||||
| 			if strings.Contains(b.r.Text(), "Rate limiting is in effect") { | ||||
| 				continue | ||||
| 			} | ||||
| 			// skip our own messages | ||||
| 			if !strings.HasPrefix(b.r.Text(), "["+b.GetString("Nick")+"] \x1b") { | ||||
| 				continue | ||||
| 			} | ||||
| 			res := strings.Split(stripPrompt(b.r.Text()), ":") | ||||
| 			if res[0] == "-> Set theme" { | ||||
| 				wait = false | ||||
|   | ||||
| @@ -85,7 +85,7 @@ func (b *Bsteam) handleEvents() { | ||||
|  | ||||
| func (b *Bsteam) handleLogOnFailed(e *steam.LogOnFailedEvent, myLoginInfo *steam.LogOnDetails) error { | ||||
| 	switch e.Result { | ||||
| 	case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode: | ||||
| 	case steamlang.EResult_AccountLoginDeniedNeedTwoFactor: | ||||
| 		b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:") | ||||
| 		var code string | ||||
| 		fmt.Scanf("%s", &code) | ||||
|   | ||||
| @@ -1,21 +1,41 @@ | ||||
| package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"regexp" | ||||
| 	"path/filepath" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"unicode/utf16" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| 	"github.com/davecgh/go-spew/spew" | ||||
| 	tgbotapi "github.com/matterbridge/telegram-bot-api/v6" | ||||
| ) | ||||
|  | ||||
| func (b *Btelegram) handleUpdate(rmsg *config.Message, message, posted, edited *tgbotapi.Message) *tgbotapi.Message { | ||||
| 	// handle channels | ||||
| 	if posted != nil { | ||||
| 		message = posted | ||||
| 		rmsg.Text = message.Text | ||||
| 		if posted.Text == "/chatId" { | ||||
| 			chatID := strconv.FormatInt(posted.Chat.ID, 10) | ||||
|  | ||||
| 			// Handle chat topics | ||||
| 			if posted.IsTopicMessage { | ||||
| 				chatID = chatID + "/" + strconv.Itoa(posted.MessageThreadID) | ||||
| 			} | ||||
|  | ||||
| 			_, err := b.Send(config.Message{ | ||||
| 				Channel: chatID, | ||||
| 				Text:    fmt.Sprintf("ID of this chat: %s", chatID), | ||||
| 			}) | ||||
| 			if err != nil { | ||||
| 				b.Log.Warnf("Unable to send chatID to %s", chatID) | ||||
| 			} | ||||
| 		} else { | ||||
| 			message = posted | ||||
| 			rmsg.Text = message.Text | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// edited channel message | ||||
| @@ -38,32 +58,54 @@ func (b *Btelegram) handleGroups(rmsg *config.Message, message *tgbotapi.Message | ||||
|  | ||||
| // handleForwarded handles forwarded messages | ||||
| func (b *Btelegram) handleForwarded(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.ForwardFrom != nil { | ||||
| 		usernameForward := "" | ||||
| 		if b.GetBool("UseFirstName") { | ||||
| 	if message.ForwardDate == 0 { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if message.ForwardFromChat != nil && message.ForwardFrom == nil { | ||||
| 		rmsg.Text = "Forwarded from " + message.ForwardFromChat.Title + ": " + rmsg.Text | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if message.ForwardFrom == nil { | ||||
| 		rmsg.Text = "Forwarded from " + unknownUser + ": " + rmsg.Text | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	usernameForward := "" | ||||
| 	if b.GetBool("UseFirstName") { | ||||
| 		usernameForward = message.ForwardFrom.FirstName | ||||
| 	} | ||||
| 	if b.GetBool("UseFullName") { | ||||
| 		usernameForward = message.ForwardFrom.FirstName + " " + message.ForwardFrom.LastName | ||||
| 	} | ||||
|  | ||||
| 	if usernameForward == "" { | ||||
| 		usernameForward = message.ForwardFrom.UserName | ||||
| 		if usernameForward == "" { | ||||
| 			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 usernameForward == "" { | ||||
| 		usernameForward = unknownUser | ||||
| 	} | ||||
|  | ||||
| 	rmsg.Text = "Forwarded from " + usernameForward + ": " + rmsg.Text | ||||
| } | ||||
|  | ||||
| // handleQuoting handles quoting of previous messages | ||||
| func (b *Btelegram) handleQuoting(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.ReplyToMessage != nil { | ||||
| 	// Used to check if the message was a reply to the root topic | ||||
| 	if message.ReplyToMessage != nil && (!message.IsTopicMessage || message.ReplyToMessage.MessageID != message.MessageThreadID) { //nolint:nestif | ||||
| 		usernameReply := "" | ||||
| 		if message.ReplyToMessage.From != nil { | ||||
| 			if b.GetBool("UseFirstName") { | ||||
| 				usernameReply = message.ReplyToMessage.From.FirstName | ||||
| 			} | ||||
| 			if b.GetBool("UseFullName") { | ||||
| 				usernameReply = message.ReplyToMessage.From.FirstName + " " + message.ReplyToMessage.From.LastName | ||||
| 			} | ||||
| 			if usernameReply == "" { | ||||
| 				usernameReply = message.ReplyToMessage.From.UserName | ||||
| 				if usernameReply == "" { | ||||
| @@ -75,7 +117,11 @@ func (b *Btelegram) handleQuoting(rmsg *config.Message, message *tgbotapi.Messag | ||||
| 			usernameReply = unknownUser | ||||
| 		} | ||||
| 		if !b.GetBool("QuoteDisable") { | ||||
| 			rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, message.ReplyToMessage.Text) | ||||
| 			quote := message.ReplyToMessage.Text | ||||
| 			if quote == "" { | ||||
| 				quote = message.ReplyToMessage.Caption | ||||
| 			} | ||||
| 			rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, quote) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -83,10 +129,15 @@ func (b *Btelegram) handleQuoting(rmsg *config.Message, message *tgbotapi.Messag | ||||
| // handleUsername handles the correct setting of the username | ||||
| func (b *Btelegram) handleUsername(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.From != nil { | ||||
| 		rmsg.UserID = strconv.Itoa(message.From.ID) | ||||
| 		rmsg.UserID = strconv.FormatInt(message.From.ID, 10) | ||||
| 		if b.GetBool("UseFirstName") { | ||||
| 			rmsg.Username = message.From.FirstName | ||||
| 		} | ||||
| 		if b.GetBool("UseFullName") { | ||||
| 			if message.From.FirstName != "" && message.From.LastName != "" { | ||||
| 				rmsg.Username = message.From.FirstName + " " + message.From.LastName | ||||
| 			} | ||||
| 		} | ||||
| 		if rmsg.Username == "" { | ||||
| 			rmsg.Username = message.From.UserName | ||||
| 			if rmsg.Username == "" { | ||||
| @@ -94,11 +145,40 @@ func (b *Btelegram) handleUsername(rmsg *config.Message, message *tgbotapi.Messa | ||||
| 			} | ||||
| 		} | ||||
| 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 		if b.General.MediaServerUpload != "" { | ||||
| 		if b.General.MediaServerUpload != "" || (b.General.MediaServerDownload != "" && b.General.MediaDownloadPath != "") { | ||||
| 			b.handleDownloadAvatar(message.From.ID, rmsg.Channel) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if message.SenderChat != nil { //nolint:nestif | ||||
| 		rmsg.UserID = strconv.FormatInt(message.SenderChat.ID, 10) | ||||
| 		if b.GetBool("UseFirstName") { | ||||
| 			rmsg.Username = message.SenderChat.FirstName | ||||
| 		} | ||||
| 		if b.GetBool("UseFullName") { | ||||
| 			if message.SenderChat.FirstName != "" && message.SenderChat.LastName != "" { | ||||
| 				rmsg.Username = message.SenderChat.FirstName + " " + message.SenderChat.LastName | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		if rmsg.Username == "" || rmsg.Username == "Channel_Bot" { | ||||
| 			rmsg.Username = message.SenderChat.UserName | ||||
|  | ||||
| 			if rmsg.Username == "" || rmsg.Username == "Channel_Bot" { | ||||
| 				rmsg.Username = message.SenderChat.FirstName | ||||
| 			} | ||||
| 		} | ||||
| 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 		if b.General.MediaServerUpload != "" || (b.General.MediaServerDownload != "" && b.General.MediaDownloadPath != "") { | ||||
| 			b.handleDownloadAvatar(message.SenderChat.ID, rmsg.Channel) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Fallback on author signature (used in "channel" type of chat) | ||||
| 	if rmsg.Username == "" && message.AuthorSignature != "" { | ||||
| 		rmsg.Username = message.AuthorSignature | ||||
| 	} | ||||
|  | ||||
| 	// if we really didn't find a username, set it to unknown | ||||
| 	if rmsg.Username == "" { | ||||
| 		rmsg.Username = unknownUser | ||||
| @@ -111,10 +191,16 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
|  | ||||
| 		if update.Message == nil && update.ChannelPost == nil && | ||||
| 			update.EditedMessage == nil && update.EditedChannelPost == nil { | ||||
| 			b.Log.Error("Getting nil messages, this shouldn't happen.") | ||||
| 			b.Log.Info("Received event without messages, skipping.") | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if b.GetInt("debuglevel") == 1 { | ||||
| 			spew.Dump(update.Message) | ||||
| 		} | ||||
|  | ||||
| 		b.handleGroupUpdate(update) | ||||
|  | ||||
| 		var message *tgbotapi.Message | ||||
|  | ||||
| 		rmsg := config.Message{Account: b.Account, Extra: make(map[string][]interface{})} | ||||
| @@ -125,9 +211,27 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
| 		// handle groups | ||||
| 		message = b.handleGroups(&rmsg, message, update) | ||||
|  | ||||
| 		if message == nil { | ||||
| 			b.Log.Error("message is nil, this shouldn't happen.") | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// set the ID's from the channel or group message | ||||
| 		rmsg.ID = strconv.Itoa(message.MessageID) | ||||
| 		rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10) | ||||
| 		if message.IsTopicMessage { | ||||
| 			rmsg.Channel += "/" + strconv.Itoa(message.MessageThreadID) | ||||
| 		} | ||||
|  | ||||
| 		// preserve threading from telegram reply | ||||
| 		if message.ReplyToMessage != nil && | ||||
| 			// Used to check if the message was a reply to the root topic | ||||
| 			(!message.IsTopicMessage || message.ReplyToMessage.MessageID != message.MessageThreadID) { | ||||
| 			rmsg.ParentID = strconv.Itoa(message.ReplyToMessage.MessageID) | ||||
| 		} | ||||
|  | ||||
| 		// handle entities (adding URLs) | ||||
| 		b.handleEntities(&rmsg, message) | ||||
|  | ||||
| 		// handle username | ||||
| 		b.handleUsername(&rmsg, message) | ||||
| @@ -144,14 +248,12 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
| 		// quote the previous message | ||||
| 		b.handleQuoting(&rmsg, message) | ||||
|  | ||||
| 		// handle entities (adding URLs) | ||||
| 		b.handleEntities(&rmsg, message) | ||||
|  | ||||
| 		if rmsg.Text != "" || len(rmsg.Extra) > 0 { | ||||
| 			rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text) | ||||
| 			// Comment the next line out due to avoid removing empty lines in Telegram | ||||
| 			// 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) | ||||
| 				rmsg.Avatar = helper.GetAvatar(b.avatarMap, strconv.FormatInt(message.From.ID, 10), b.General) | ||||
| 			} | ||||
|  | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account) | ||||
| @@ -161,54 +263,132 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleGroupUpdate(update tgbotapi.Update) { | ||||
| 	if msg := update.Message; msg != nil { | ||||
| 		switch { | ||||
| 		case msg.NewChatMembers != nil: | ||||
| 			b.handleUserJoin(update) | ||||
| 		case msg.LeftChatMember != nil: | ||||
| 			b.handleUserLeave(update) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleUserJoin(update tgbotapi.Update) { | ||||
| 	msg := update.Message | ||||
| 	for _, user := range msg.NewChatMembers { | ||||
| 		rmsg := config.Message{ | ||||
| 			UserID:   strconv.FormatInt(user.ID, 10), | ||||
| 			Username: user.FirstName, // for some reason all the other name felids are empty on this event (at least for me) | ||||
| 			Channel:  strconv.FormatInt(msg.Chat.ID, 10), | ||||
| 			Account:  b.Account, | ||||
| 			Protocol: b.Protocol, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 			Text:     "joined chat", | ||||
| 		} | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleUserLeave(update tgbotapi.Update) { | ||||
| 	msg := update.Message | ||||
| 	user := msg.LeftChatMember | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   strconv.FormatInt(user.ID, 10), | ||||
| 		Username: user.FirstName, // for some reason all the other name felids are empty on this event (at least for me) | ||||
| 		Channel:  strconv.FormatInt(msg.Chat.ID, 10), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Event:    config.EventJoinLeave, | ||||
| 		Text:     "left chat", | ||||
| 	} | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // 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{})} | ||||
| func (b *Btelegram) handleDownloadAvatar(userid int64, channel string) { | ||||
| 	rmsg := config.Message{ | ||||
| 		Username: "system", | ||||
| 		Text:     "avatar", | ||||
| 		Channel:  channel, | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   strconv.FormatInt(userid, 10), | ||||
| 		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 _, ok := b.avatarMap[strconv.FormatInt(userid, 10)]; ok { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	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.FormatInt(userid, 10) + ".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.Errorf("Userprofile download failed for %#v %s", userid, err) | ||||
| 			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 | ||||
| 	} | ||||
| } | ||||
|  | ||||
| 		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) | ||||
| func (b *Btelegram) maybeConvertTgs(name *string, data *[]byte) { | ||||
| 	format := b.GetString("MediaConvertTgs") | ||||
| 	if helper.SupportsFormat(format) { | ||||
| 		b.Log.Debugf("Format supported by %s, converting %v", helper.LottieBackend(), name) | ||||
| 	} else { | ||||
| 		// Otherwise, no conversion was requested. Trying to run the usual webp | ||||
| 		// converter would fail, because '.tgs.webp' is actually a gzipped JSON | ||||
| 		// file, and has nothing to do with WebP. | ||||
| 		return | ||||
| 	} | ||||
| 	err := helper.ConvertTgsToX(data, format, b.Log) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("conversion failed: %v", err) | ||||
| 	} else { | ||||
| 		*name = strings.Replace(*name, "tgs.webp", format, 1) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| 			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 | ||||
| func (b *Btelegram) maybeConvertWebp(name *string, data *[]byte) { | ||||
| 	if b.GetBool("MediaConvertWebPToPNG") { | ||||
| 		b.Log.Debugf("WebP to PNG conversion enabled, converting %v", name) | ||||
| 		err := helper.ConvertWebPToPNG(data) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("conversion failed: %v", err) | ||||
| 		} else { | ||||
| 			*name = strings.Replace(*name, ".webp", ".png", 1) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Message) error { | ||||
| 	size := 0 | ||||
| 	size := int64(0) | ||||
| 	var url, name, text string | ||||
| 	switch { | ||||
| 	case message.Sticker != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Sticker.FileID, ".webp", true) | ||||
| 		size = message.Sticker.FileSize | ||||
| 		size = int64(message.Sticker.FileSize) | ||||
| 	case message.Voice != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Voice.FileID, ".ogg", true) | ||||
| 		size = message.Voice.FileSize | ||||
| @@ -224,8 +404,8 @@ func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Messa | ||||
| 		name = message.Document.FileName | ||||
| 		text = " " + message.Document.FileName + " : " + url | ||||
| 	case message.Photo != nil: | ||||
| 		photos := *message.Photo | ||||
| 		size = photos[len(photos)-1].FileSize | ||||
| 		photos := message.Photo | ||||
| 		size = int64(photos[len(photos)-1].FileSize) | ||||
| 		text, name, url = b.getDownloadInfo(photos[len(photos)-1].FileID, "", true) | ||||
| 	} | ||||
|  | ||||
| @@ -248,15 +428,18 @@ func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Messa | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	if strings.HasSuffix(name, ".webp") && b.GetBool("MediaConvertWebPToPNG") { | ||||
| 		b.Log.Debugf("WebP to PNG conversion enabled, converting %s", name) | ||||
| 		err := helper.ConvertWebPToPNG(data) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("conversion failed: %s", err) | ||||
| 		} else { | ||||
| 			name = strings.Replace(name, ".webp", ".png", 1) | ||||
| 		} | ||||
|  | ||||
| 	if strings.HasSuffix(name, ".tgs.webp") { | ||||
| 		b.maybeConvertTgs(&name, data) | ||||
| 	} else if strings.HasSuffix(name, ".webp") { | ||||
| 		b.maybeConvertWebp(&name, data) | ||||
| 	} | ||||
|  | ||||
| 	// rename .oga to .ogg  https://github.com/42wim/matterbridge/issues/906#issuecomment-741793512 | ||||
| 	if strings.HasSuffix(name, ".oga") && message.Audio != nil { | ||||
| 		name = strings.Replace(name, ".oga", ".ogg", 1) | ||||
| 	} | ||||
|  | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General) | ||||
| 	return nil | ||||
| } | ||||
| @@ -268,7 +451,7 @@ func (b *Btelegram) getDownloadInfo(id string, suffix string, urlpart bool) (str | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 	} | ||||
| 	if suffix != "" && !strings.HasSuffix(name, suffix) { | ||||
| 	if suffix != "" && !strings.HasSuffix(name, suffix) && !strings.HasSuffix(name, ".webm") { | ||||
| 		name += suffix | ||||
| 	} | ||||
| 	text := " " + url | ||||
| @@ -280,11 +463,15 @@ func (b *Btelegram) handleDelete(msg *config.Message, chatid int64) (string, err | ||||
| 	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}) | ||||
|  | ||||
| 	cfg := tgbotapi.NewDeleteMessage(chatid, msgid) | ||||
| 	_, err = b.c.Request(cfg) | ||||
|  | ||||
| 	return "", err | ||||
| } | ||||
|  | ||||
| @@ -306,6 +493,9 @@ func (b *Btelegram) handleEdit(msg *config.Message, chatid int64) (string, error | ||||
| 	case "Markdown": | ||||
| 		b.Log.Debug("Using mode markdown") | ||||
| 		m.ParseMode = tgbotapi.ModeMarkdown | ||||
| 	case MarkdownV2: | ||||
| 		b.Log.Debug("Using mode MarkdownV2") | ||||
| 		m.ParseMode = MarkdownV2 | ||||
| 	} | ||||
| 	if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||
| 		b.Log.Debug("Using mode HTML - nick only") | ||||
| @@ -319,31 +509,57 @@ func (b *Btelegram) handleEdit(msg *config.Message, chatid int64) (string, error | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) string { | ||||
| 	var c tgbotapi.Chattable | ||||
| func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64, threadid int, parentID int) (string, error) { | ||||
| 	var media []interface{} | ||||
| 	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) | ||||
|  | ||||
| 		if b.GetString("MessageFormat") == HTMLFormat { | ||||
| 			fi.Comment = makeHTML(html.EscapeString(fi.Comment)) | ||||
| 		} | ||||
| 		_, err := b.c.Send(c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		} | ||||
| 		if fi.Comment != "" { | ||||
| 			if _, err := b.sendMessage(chatid, msg.Username, fi.Comment); err != nil { | ||||
| 				b.Log.Errorf("posting file comment %s failed: %s", fi.Comment, err) | ||||
|  | ||||
| 		switch filepath.Ext(fi.Name) { | ||||
| 		case ".jpg", ".jpe", ".png": | ||||
| 			pc := tgbotapi.NewInputMediaPhoto(file) | ||||
| 			if fi.Comment != "" { | ||||
| 				pc.Caption, pc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment) | ||||
| 			} | ||||
| 			media = append(media, pc) | ||||
| 		case ".mp4", ".m4v": | ||||
| 			vc := tgbotapi.NewInputMediaVideo(file) | ||||
| 			if fi.Comment != "" { | ||||
| 				vc.Caption, vc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment) | ||||
| 			} | ||||
| 			media = append(media, vc) | ||||
| 		case ".mp3", ".oga": | ||||
| 			ac := tgbotapi.NewInputMediaAudio(file) | ||||
| 			if fi.Comment != "" { | ||||
| 				ac.Caption, ac.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment) | ||||
| 			} | ||||
| 			media = append(media, ac) | ||||
| 		case ".ogg": | ||||
| 			voc := tgbotapi.NewVoice(chatid, file) | ||||
| 			voc.Caption, voc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment) | ||||
| 			voc.ReplyToMessageID = parentID | ||||
| 			res, err := b.c.Send(voc) | ||||
| 			if err != nil { | ||||
| 				return "", err | ||||
| 			} | ||||
| 			return strconv.Itoa(res.MessageID), nil | ||||
| 		default: | ||||
| 			dc := tgbotapi.NewInputMediaDocument(file) | ||||
| 			if fi.Comment != "" { | ||||
| 				dc.Caption, dc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment) | ||||
| 			} | ||||
| 			media = append(media, dc) | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
|  | ||||
| 	return b.sendMediaFiles(msg, chatid, threadid, parentID, media) | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string { | ||||
| @@ -351,6 +567,14 @@ func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string | ||||
| 	if format == "" { | ||||
| 		format = "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})" | ||||
| 	} | ||||
| 	quoteMessagelength := len([]rune(quoteMessage)) | ||||
| 	if b.GetInt("QuoteLengthLimit") != 0 && quoteMessagelength >= b.GetInt("QuoteLengthLimit") { | ||||
| 		runes := []rune(quoteMessage) | ||||
| 		quoteMessage = string(runes[0:b.GetInt("QuoteLengthLimit")]) | ||||
| 		if quoteMessagelength > b.GetInt("QuoteLengthLimit") { | ||||
| 			quoteMessage += "..." | ||||
| 		} | ||||
| 	} | ||||
| 	format = strings.Replace(format, "{MESSAGE}", message, -1) | ||||
| 	format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1) | ||||
| 	format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1) | ||||
| @@ -362,16 +586,61 @@ func (b *Btelegram) handleEntities(rmsg *config.Message, message *tgbotapi.Messa | ||||
| 	if message.Entities == nil { | ||||
| 		return | ||||
| 	} | ||||
| 	// for now only do URL replacements | ||||
| 	for _, e := range *message.Entities { | ||||
|  | ||||
| 	indexMovedBy := 0 | ||||
| 	prevLinkOffset := -1 | ||||
|  | ||||
| 	for _, e := range message.Entities { | ||||
|  | ||||
| 		asRunes := utf16.Encode([]rune(rmsg.Text)) | ||||
|  | ||||
| 		if e.Type == "text_link" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			url, err := e.ParseURL() | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("entity text_link url parse failed: %s", err) | ||||
| 				continue | ||||
| 			} | ||||
| 			link := rmsg.Text[e.Offset : e.Offset+e.Length] | ||||
| 			rmsg.Text = strings.Replace(rmsg.Text, link, url.String(), 1) | ||||
| 			utfEncodedString := utf16.Encode([]rune(rmsg.Text)) | ||||
| 			if offset+e.Length > len(utfEncodedString) { | ||||
| 				b.Log.Errorf("entity length is too long %d > %d", offset+e.Length, len(utfEncodedString)) | ||||
| 				continue | ||||
| 			} | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset+e.Length])) + " (" + url.String() + ")" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += len(url.String()) + 3 | ||||
| 			prevLinkOffset = e.Offset | ||||
| 		} | ||||
|  | ||||
| 		if e.Offset == prevLinkOffset { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if e.Type == "code" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset])) + "`" + string(utf16.Decode(asRunes[offset:offset+e.Length])) + "`" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += 2 | ||||
| 		} | ||||
|  | ||||
| 		if e.Type == "pre" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset])) + "```\n" + string(utf16.Decode(asRunes[offset:offset+e.Length])) + "```\n" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += 8 | ||||
| 		} | ||||
|  | ||||
| 		if e.Type == "bold" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset])) + "*" + string(utf16.Decode(asRunes[offset:offset+e.Length])) + "*" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += 2 | ||||
| 		} | ||||
| 		if e.Type == "italic" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset])) + "_" + string(utf16.Decode(asRunes[offset:offset+e.Length])) + "_" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += 2 | ||||
| 		} | ||||
| 		if e.Type == "strike" { | ||||
| 			offset := e.Offset + indexMovedBy | ||||
| 			rmsg.Text = string(utf16.Decode(asRunes[:offset])) + "~" + string(utf16.Decode(asRunes[offset:offset+e.Length])) + "~" + string(utf16.Decode(asRunes[offset+e.Length:])) | ||||
| 			indexMovedBy += 2 | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -2,7 +2,6 @@ package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"html" | ||||
|  | ||||
| 	"github.com/russross/blackfriday" | ||||
| ) | ||||
| @@ -24,10 +23,16 @@ func (options *customHTML) Paragraph(out *bytes.Buffer, text func() bool) { | ||||
| func (options *customHTML) BlockCode(out *bytes.Buffer, text []byte, lang string) { | ||||
| 	out.WriteString("<pre>") | ||||
|  | ||||
| 	out.WriteString(html.EscapeString(string(text))) | ||||
| 	out.WriteString(string(text)) | ||||
| 	out.WriteString("</pre>\n") | ||||
| } | ||||
|  | ||||
| func (options *customHTML) CodeSpan(out *bytes.Buffer, text []byte) { | ||||
| 	out.WriteString("<code>") | ||||
| 	out.WriteString(string(text)) | ||||
| 	out.WriteString("</code>") | ||||
| } | ||||
|  | ||||
| func (options *customHTML) Header(out *bytes.Buffer, text func() bool, level int, id string) { | ||||
| 	options.Paragraph(out, text) | ||||
| } | ||||
| @@ -42,6 +47,10 @@ func (options *customHTML) BlockQuote(out *bytes.Buffer, text []byte) { | ||||
| 	out.WriteByte('\n') | ||||
| } | ||||
|  | ||||
| func (options *customHTML) LineBreak(out *bytes.Buffer) { | ||||
| 	out.WriteByte('\n') | ||||
| } | ||||
|  | ||||
| func (options *customHTML) List(out *bytes.Buffer, text func() bool, flags int) { | ||||
| 	options.Paragraph(out, text) | ||||
| } | ||||
|   | ||||
| @@ -1,20 +1,23 @@ | ||||
| package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"log" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| 	tgbotapi "github.com/matterbridge/telegram-bot-api/v6" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	unknownUser = "unknown" | ||||
| 	HTMLFormat  = "HTML" | ||||
| 	HTMLNick    = "htmlnick" | ||||
| 	MarkdownV2  = "MarkdownV2" | ||||
| ) | ||||
|  | ||||
| type Btelegram struct { | ||||
| @@ -24,6 +27,16 @@ type Btelegram struct { | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	tgsConvertFormat := cfg.GetString("MediaConvertTgs") | ||||
| 	if tgsConvertFormat != "" { | ||||
| 		err := helper.CanConvertTgsToX() | ||||
| 		if err != nil { | ||||
| 			log.Fatalf("Telegram bridge configured to convert .tgs files to '%s', but %s does not appear to work:\n%#v", tgsConvertFormat, helper.LottieBackend(), err) | ||||
| 		} | ||||
| 		if !helper.SupportsFormat(tgsConvertFormat) { | ||||
| 			log.Fatalf("Telegram bridge configured to convert .tgs files to '%s', but %s doesn't support it.", tgsConvertFormat, helper.LottieBackend()) | ||||
| 		} | ||||
| 	} | ||||
| 	return &Btelegram{Config: cfg, avatarMap: make(map[string]string)} | ||||
| } | ||||
|  | ||||
| @@ -37,11 +50,7 @@ func (b *Btelegram) Connect() error { | ||||
| 	} | ||||
| 	u := tgbotapi.NewUpdate(0) | ||||
| 	u.Timeout = 60 | ||||
| 	updates, err := b.c.GetUpdatesChan(u) | ||||
| 	if err != nil { | ||||
| 		b.Log.Debugf("%#v", err) | ||||
| 		return err | ||||
| 	} | ||||
| 	updates := b.c.GetUpdatesChan(u) | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	go b.handleRecv(updates) | ||||
| 	return nil | ||||
| @@ -55,11 +64,63 @@ func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func TGGetParseMode(b *Btelegram, username string, text string) (textout string, parsemode string) { | ||||
| 	textout = username + text | ||||
| 	if b.GetString("MessageFormat") == HTMLFormat { | ||||
| 		b.Log.Debug("Using mode HTML") | ||||
| 		parsemode = tgbotapi.ModeHTML | ||||
| 	} | ||||
| 	if b.GetString("MessageFormat") == "Markdown" { | ||||
| 		b.Log.Debug("Using mode markdown") | ||||
| 		parsemode = tgbotapi.ModeMarkdown | ||||
| 	} | ||||
| 	if b.GetString("MessageFormat") == MarkdownV2 { | ||||
| 		b.Log.Debug("Using mode MarkdownV2") | ||||
| 		parsemode = MarkdownV2 | ||||
| 	} | ||||
| 	if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||
| 		b.Log.Debug("Using mode HTML - nick only") | ||||
| 		textout = username + html.EscapeString(text) | ||||
| 		parsemode = tgbotapi.ModeHTML | ||||
| 	} | ||||
| 	return textout, parsemode | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) getIds(channel string) (int64, int, error) { | ||||
| 	var chatid int64 | ||||
| 	topicid := 0 | ||||
|  | ||||
| 	// get the chatid | ||||
| 	if strings.Contains(channel, "/") { //nolint:nestif | ||||
| 		s := strings.Split(channel, "/") | ||||
| 		if len(s) < 2 { | ||||
| 			b.Log.Errorf("Invalid channel format: %#v\n", channel) | ||||
| 			return 0, 0, nil | ||||
| 		} | ||||
| 		id, err := strconv.ParseInt(s[0], 10, 64) | ||||
| 		if err != nil { | ||||
| 			return 0, 0, err | ||||
| 		} | ||||
| 		chatid = id | ||||
| 		tid, err := strconv.Atoi(s[1]) | ||||
| 		if err != nil { | ||||
| 			return 0, 0, err | ||||
| 		} | ||||
| 		topicid = tid | ||||
| 	} else { | ||||
| 		id, err := strconv.ParseInt(channel, 10, 64) | ||||
| 		if err != nil { | ||||
| 			return 0, 0, err | ||||
| 		} | ||||
| 		chatid = id | ||||
| 	} | ||||
| 	return chatid, topicid, nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// get the chatid | ||||
| 	chatid, err := strconv.ParseInt(msg.Channel, 10, 64) | ||||
| 	chatid, topicid, err := b.getIds(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| @@ -70,7 +131,7 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	if b.GetString("MessageFormat") == HTMLFormat { | ||||
| 		msg.Text = makeHTML(msg.Text) | ||||
| 		msg.Text = makeHTML(html.EscapeString(msg.Text)) | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| @@ -78,16 +139,27 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
| 		return b.handleDelete(&msg, chatid) | ||||
| 	} | ||||
|  | ||||
| 	// Handle prefix hint for unthreaded messages. | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 		msg.Text = fmt.Sprintf("[reply]: %s", msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	var parentID int | ||||
| 	if msg.ParentID != "" { | ||||
| 		parentID, _ = b.intParentID(msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			if _, err := b.sendMessage(chatid, rmsg.Username, rmsg.Text); err != nil { | ||||
| 				b.Log.Errorf("sendMessage failed: %s", err) | ||||
| 			if _, msgErr := b.sendMessage(chatid, topicid, rmsg.Username, rmsg.Text, parentID); msgErr != nil { | ||||
| 				b.Log.Errorf("sendMessage failed: %s", msgErr) | ||||
| 			} | ||||
| 		} | ||||
| 		// check if we have files to upload (from slack, telegram or mattermost) | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			b.handleUploadFile(&msg, chatid) | ||||
| 			return b.handleUploadFile(&msg, chatid, topicid, parentID) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| @@ -97,7 +169,14 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message | ||||
| 	return b.sendMessage(chatid, msg.Username, msg.Text) | ||||
| 	// TODO: recheck it. | ||||
| 	// Ignore empty text field needs for prevent double messages from whatsapp to telegram | ||||
| 	// when sending media with text caption | ||||
| 	if msg.Text != "" { | ||||
| 		return b.sendMessage(chatid, topicid, msg.Username, msg.Text, parentID) | ||||
| 	} | ||||
|  | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) getFileDirectURL(id string) string { | ||||
| @@ -108,22 +187,15 @@ func (b *Btelegram) getFileDirectURL(id string) string { | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) { | ||||
| func (b *Btelegram) sendMessage(chatid int64, topicid int, username, text string, parentID int) (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 | ||||
| 	m.Text, m.ParseMode = TGGetParseMode(b, username, text) | ||||
| 	if topicid != 0 { | ||||
| 		m.BaseChat.MessageThreadID = topicid | ||||
| 	} | ||||
| 	m.ReplyToMessageID = parentID | ||||
| 	m.DisableWebPagePreview = b.GetBool("DisableWebPagePreview") | ||||
|  | ||||
| 	res, err := b.c.Send(m) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| @@ -131,6 +203,37 @@ func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, er | ||||
| 	return strconv.Itoa(res.MessageID), nil | ||||
| } | ||||
|  | ||||
| // sendMediaFiles native upload media files via media group | ||||
| func (b *Btelegram) sendMediaFiles(msg *config.Message, chatid int64, threadid int, parentID int, media []interface{}) (string, error) { | ||||
| 	if len(media) == 0 { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	mg := tgbotapi.MediaGroupConfig{ | ||||
| 		BaseChat: tgbotapi.BaseChat{ | ||||
| 			ChatID:           chatid, | ||||
| 			MessageThreadID:  threadid, | ||||
| 			ChannelUsername:  msg.Username, | ||||
| 			ReplyToMessageID: parentID, | ||||
| 		}, | ||||
| 		Media: media, | ||||
| 	} | ||||
| 	messages, err := b.c.SendMediaGroup(mg) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	// return first message id | ||||
| 	return strconv.Itoa(messages[0].MessageID), nil | ||||
| } | ||||
|  | ||||
| // intParentID return integer parent id for telegram message | ||||
| func (b *Btelegram) intParentID(parentID string) (int, error) { | ||||
| 	pid, err := strconv.Atoi(parentID) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	return pid, 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, | ||||
|   | ||||
							
								
								
									
										333
									
								
								bridge/vk/vk.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										333
									
								
								bridge/vk/vk.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,333 @@ | ||||
| package bvk | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
|  | ||||
| 	"github.com/SevereCloud/vksdk/v2/api" | ||||
| 	"github.com/SevereCloud/vksdk/v2/events" | ||||
| 	longpoll "github.com/SevereCloud/vksdk/v2/longpoll-bot" | ||||
| 	"github.com/SevereCloud/vksdk/v2/object" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	audioMessage = "audio_message" | ||||
| 	document     = "doc" | ||||
| 	photo        = "photo" | ||||
| 	video        = "video" | ||||
| 	graffiti     = "graffiti" | ||||
| 	sticker      = "sticker" | ||||
| 	wall         = "wall" | ||||
| ) | ||||
|  | ||||
| type user struct { | ||||
| 	lastname, firstname, avatar string | ||||
| } | ||||
|  | ||||
| type Bvk struct { | ||||
| 	c            *api.VK | ||||
| 	lp           *longpoll.LongPoll | ||||
| 	usernamesMap map[int]user // cache of user names and avatar URLs | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Bvk{usernamesMap: make(map[int]user), Config: cfg} | ||||
| } | ||||
|  | ||||
| func (b *Bvk) Connect() error { | ||||
| 	b.Log.Info("Connecting") | ||||
| 	b.c = api.NewVK(b.GetString("Token")) | ||||
|  | ||||
| 	var err error | ||||
| 	b.lp, err = longpoll.NewLongPollCommunity(b.c) | ||||
| 	if err != nil { | ||||
| 		b.Log.Debugf("%#v", err) | ||||
|  | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	b.lp.MessageNew(func(ctx context.Context, obj events.MessageNewObject) { | ||||
| 		b.handleMessage(obj.Message, false) | ||||
| 	}) | ||||
|  | ||||
| 	b.Log.Info("Connection succeeded") | ||||
|  | ||||
| 	go func() { | ||||
| 		err := b.lp.Run() | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Fatal("Enable longpoll in group management") | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bvk) Disconnect() error { | ||||
| 	b.lp.Shutdown() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bvk) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bvk) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	peerID, err := strconv.Atoi(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	params := api.Params{} | ||||
|  | ||||
| 	text := msg.Username + msg.Text | ||||
|  | ||||
| 	if msg.Extra != nil { | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			// generate attachments string | ||||
| 			attachment, urls := b.uploadFiles(msg.Extra, peerID) | ||||
| 			params["attachment"] = attachment | ||||
| 			text += urls | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	params["message"] = text | ||||
|  | ||||
| 	if msg.ID == "" { | ||||
| 		// New message | ||||
| 		params["random_id"] = time.Now().Unix() | ||||
| 		params["peer_ids"] = msg.Channel | ||||
|  | ||||
| 		res, e := b.c.MessagesSendPeerIDs(params) | ||||
| 		if e != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return strconv.Itoa(res[0].ConversationMessageID), nil | ||||
| 	} | ||||
| 	// Edit message | ||||
| 	messageID, err := strconv.ParseInt(msg.ID, 10, 64) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	params["peer_id"] = peerID | ||||
| 	params["conversation_message_id"] = messageID | ||||
|  | ||||
| 	_, err = b.c.MessagesEdit(params) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	return msg.ID, nil | ||||
| } | ||||
|  | ||||
| func (b *Bvk) getUser(id int) user { | ||||
| 	u, found := b.usernamesMap[id] | ||||
| 	if !found { | ||||
| 		b.Log.Debug("Fetching username for ", id) | ||||
|  | ||||
| 		if id >= 0 { | ||||
| 			result, _ := b.c.UsersGet(api.Params{ | ||||
| 				"user_ids": id, | ||||
| 				"fields":   "photo_200", | ||||
| 			}) | ||||
|  | ||||
| 			resUser := result[0] | ||||
| 			u = user{lastname: resUser.LastName, firstname: resUser.FirstName, avatar: resUser.Photo200} | ||||
| 			b.usernamesMap[id] = u | ||||
| 		} else { | ||||
| 			result, _ := b.c.GroupsGetByID(api.Params{ | ||||
| 				"group_id": id * -1, | ||||
| 			}) | ||||
|  | ||||
| 			resGroup := result[0] | ||||
| 			u = user{lastname: resGroup.Name, avatar: resGroup.Photo200} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return u | ||||
| } | ||||
|  | ||||
| func (b *Bvk) handleMessage(msg object.MessagesMessage, isFwd bool) { | ||||
| 	b.Log.Debug("ChatID: ", msg.PeerID) | ||||
| 	// fetch user info | ||||
| 	u := b.getUser(msg.FromID) | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		Text:     msg.Text, | ||||
| 		Username: u.firstname + " " + u.lastname, | ||||
| 		Avatar:   u.avatar, | ||||
| 		Channel:  strconv.Itoa(msg.PeerID), | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   strconv.Itoa(msg.FromID), | ||||
| 		ID:       strconv.Itoa(msg.ConversationMessageID), | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 	} | ||||
|  | ||||
| 	if msg.ReplyMessage != nil { | ||||
| 		ur := b.getUser(msg.ReplyMessage.FromID) | ||||
| 		rmsg.Text = "Re: " + ur.firstname + " " + ur.lastname + "\n" + rmsg.Text | ||||
| 	} | ||||
|  | ||||
| 	if isFwd { | ||||
| 		rmsg.Username = "Fwd: " + rmsg.Username | ||||
| 	} | ||||
|  | ||||
| 	if len(msg.Attachments) > 0 { | ||||
| 		urls, text := b.getFiles(msg.Attachments) | ||||
|  | ||||
| 		if text != "" { | ||||
| 			rmsg.Text += "\n" + text | ||||
| 		} | ||||
|  | ||||
| 		// download | ||||
| 		b.downloadFiles(&rmsg, urls) | ||||
| 	} | ||||
|  | ||||
| 	if len(msg.FwdMessages) > 0 { | ||||
| 		rmsg.Text += strconv.Itoa(len(msg.FwdMessages)) + " forwarded messages" | ||||
| 	} | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
|  | ||||
| 	if len(msg.FwdMessages) > 0 { | ||||
| 		// recursive processing of forwarded messages | ||||
| 		for _, m := range msg.FwdMessages { | ||||
| 			m.PeerID = msg.PeerID | ||||
| 			b.handleMessage(m, true) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bvk) uploadFiles(extra map[string][]interface{}, peerID int) (string, string) { | ||||
| 	var attachments []string | ||||
| 	text := "" | ||||
|  | ||||
| 	for _, f := range extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
|  | ||||
| 		if fi.Comment != "" { | ||||
| 			text += fi.Comment + "\n" | ||||
| 		} | ||||
| 		a, err := b.uploadFile(fi, peerID) | ||||
| 		if err != nil { | ||||
| 			b.Log.WithError(err).Error("File upload error ", fi.Name) | ||||
| 		} | ||||
|  | ||||
| 		attachments = append(attachments, a) | ||||
| 	} | ||||
|  | ||||
| 	return strings.Join(attachments, ","), text | ||||
| } | ||||
|  | ||||
| func (b *Bvk) uploadFile(file config.FileInfo, peerID int) (string, error) { | ||||
| 	r := bytes.NewReader(*file.Data) | ||||
|  | ||||
| 	photoRE := regexp.MustCompile(".(jpg|jpe|png)$") | ||||
| 	if photoRE.MatchString(file.Name) { | ||||
| 		// BUG(VK): for community chat peerID=0 | ||||
| 		p, err := b.c.UploadMessagesPhoto(0, r) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return photo + strconv.Itoa(p[0].OwnerID) + "_" + strconv.Itoa(p[0].ID), nil | ||||
| 	} | ||||
|  | ||||
| 	var doctype string | ||||
| 	if strings.Contains(file.Name, ".ogg") { | ||||
| 		doctype = audioMessage | ||||
| 	} else { | ||||
| 		doctype = document | ||||
| 	} | ||||
|  | ||||
| 	doc, err := b.c.UploadMessagesDoc(peerID, doctype, file.Name, "", r) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	switch doc.Type { | ||||
| 	case audioMessage: | ||||
| 		return document + strconv.Itoa(doc.AudioMessage.OwnerID) + "_" + strconv.Itoa(doc.AudioMessage.ID), nil | ||||
| 	case document: | ||||
| 		return document + strconv.Itoa(doc.Doc.OwnerID) + "_" + strconv.Itoa(doc.Doc.ID), nil | ||||
| 	} | ||||
|  | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Bvk) getFiles(attachments []object.MessagesMessageAttachment) ([]string, string) { | ||||
| 	var urls []string | ||||
| 	var text []string | ||||
|  | ||||
| 	for _, a := range attachments { | ||||
| 		switch a.Type { | ||||
| 		case photo: | ||||
| 			var resolution float64 = 0 | ||||
| 			url := a.Photo.Sizes[0].URL | ||||
| 			for _, size := range a.Photo.Sizes { | ||||
| 				r := size.Height * size.Width | ||||
| 				if resolution < r { | ||||
| 					resolution = r | ||||
| 					url = size.URL | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			urls = append(urls, url) | ||||
|  | ||||
| 		case document: | ||||
| 			urls = append(urls, a.Doc.URL) | ||||
|  | ||||
| 		case graffiti: | ||||
| 			urls = append(urls, a.Graffiti.URL) | ||||
|  | ||||
| 		case audioMessage: | ||||
| 			urls = append(urls, a.AudioMessage.DocsDocPreviewAudioMessage.LinkOgg) | ||||
|  | ||||
| 		case sticker: | ||||
| 			var resolution float64 = 0 | ||||
| 			url := a.Sticker.Images[0].URL | ||||
| 			for _, size := range a.Sticker.Images { | ||||
| 				r := size.Height * size.Width | ||||
| 				if resolution < r { | ||||
| 					resolution = r | ||||
| 					url = size.URL | ||||
| 				} | ||||
| 			} | ||||
| 			urls = append(urls, url+".png") | ||||
| 		case video: | ||||
| 			text = append(text, "https://vk.com/video"+strconv.Itoa(a.Video.OwnerID)+"_"+strconv.Itoa(a.Video.ID)) | ||||
|  | ||||
| 		case wall: | ||||
| 			text = append(text, "https://vk.com/wall"+strconv.Itoa(a.Wall.FromID)+"_"+strconv.Itoa(a.Wall.ID)) | ||||
|  | ||||
| 		default: | ||||
| 			text = append(text, "This attachment is not supported ("+a.Type+")") | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return urls, strings.Join(text, "\n") | ||||
| } | ||||
|  | ||||
| func (b *Bvk) downloadFiles(rmsg *config.Message, urls []string) { | ||||
| 	for _, url := range urls { | ||||
| 		data, err := helper.DownloadFile(url) | ||||
| 		if err == nil { | ||||
| 			urlPart := strings.Split(url, "/") | ||||
| 			name := strings.Split(urlPart[len(urlPart)-1], "?")[0] | ||||
| 			helper.HandleDownloadData(b.Log, rmsg, name, "", url, data, b.General) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -1,14 +1,16 @@ | ||||
| // nolint:goconst | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"mime" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
|  | ||||
| 	"github.com/matterbridge/go-whatsapp" | ||||
|  | ||||
| 	whatsappExt "github.com/matterbridge/mautrix-whatsapp/whatsapp-ext" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
| 	"github.com/jpillora/backoff" | ||||
| ) | ||||
|  | ||||
| /* | ||||
| @@ -21,12 +23,56 @@ Check: | ||||
|  | ||||
| // HandleError received from WhatsApp | ||||
| func (b *Bwhatsapp) HandleError(err error) { | ||||
| 	b.Log.Errorf("%v", err) // TODO implement proper handling? at least respond to different error types | ||||
| 	// ignore received invalid data errors. https://github.com/42wim/matterbridge/issues/843 | ||||
| 	// ignore tag 174 errors. https://github.com/42wim/matterbridge/issues/1094 | ||||
| 	if strings.Contains(err.Error(), "error processing data: received invalid data") || | ||||
| 		strings.Contains(err.Error(), "invalid string with tag 174") { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	switch err.(type) { | ||||
| 	case *whatsapp.ErrConnectionClosed, *whatsapp.ErrConnectionFailed: | ||||
| 		b.reconnect(err) | ||||
| 	default: | ||||
| 		switch err { | ||||
| 		case whatsapp.ErrConnectionTimeout: | ||||
| 			b.reconnect(err) | ||||
| 		default: | ||||
| 			b.Log.Errorf("%v", err) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) reconnect(err error) { | ||||
| 	bf := &backoff.Backoff{ | ||||
| 		Min:    time.Second, | ||||
| 		Max:    5 * time.Minute, | ||||
| 		Jitter: true, | ||||
| 	} | ||||
|  | ||||
| 	for { | ||||
| 		d := bf.Duration() | ||||
|  | ||||
| 		b.Log.Errorf("Connection failed, underlying error: %v", err) | ||||
| 		b.Log.Infof("Waiting %s...", d) | ||||
|  | ||||
| 		time.Sleep(d) | ||||
|  | ||||
| 		b.Log.Info("Reconnecting...") | ||||
|  | ||||
| 		err := b.conn.Restore() | ||||
| 		if err == nil { | ||||
| 			bf.Reset() | ||||
| 			b.startedAt = uint64(time.Now().Unix()) | ||||
|  | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // HandleTextMessage sent from WhatsApp, relay it to the brige | ||||
| func (b *Bwhatsapp) HandleTextMessage(message whatsapp.TextMessage) { | ||||
| 	if message.Info.FromMe { // || !strings.Contains(strings.ToLower(message.Text), "@echo") { | ||||
| 	if message.Info.FromMe { | ||||
| 		return | ||||
| 	} | ||||
| 	// whatsapp sends last messages to show context , cut them | ||||
| @@ -34,17 +80,17 @@ func (b *Bwhatsapp) HandleTextMessage(message whatsapp.TextMessage) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	messageTime := time.Unix(int64(message.Info.Timestamp), 0) // TODO check how behaves between timezones | ||||
| 	groupJid := message.Info.RemoteJid | ||||
| 	groupJID := message.Info.RemoteJid | ||||
| 	senderJID := message.Info.SenderJid | ||||
|  | ||||
| 	senderJid := message.Info.SenderJid | ||||
| 	if len(senderJid) == 0 { | ||||
| 		// TODO workaround till https://github.com/Rhymen/go-whatsapp/issues/86 resolved | ||||
| 		senderJid = *message.Info.Source.Participant | ||||
| 	if len(senderJID) == 0 { | ||||
| 		if message.Info.Source != nil && message.Info.Source.Participant != nil { | ||||
| 			senderJID = *message.Info.Source.Participant | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// translate sender's Jid to the nicest username we can get | ||||
| 	senderName := b.getSenderName(senderJid) | ||||
| 	// translate sender's JID to the nicest username we can get | ||||
| 	senderName := b.getSenderName(senderJID) | ||||
| 	if senderName == "" { | ||||
| 		senderName = "Someone" // don't expose telephone number | ||||
| 	} | ||||
| @@ -52,53 +98,285 @@ func (b *Bwhatsapp) HandleTextMessage(message whatsapp.TextMessage) { | ||||
| 	extText := message.Info.Source.Message.ExtendedTextMessage | ||||
| 	if extText != nil && extText.ContextInfo != nil && extText.ContextInfo.MentionedJid != nil { | ||||
| 		// handle user mentions | ||||
| 		for _, mentionedJid := range extText.ContextInfo.MentionedJid { | ||||
| 			numberAndSuffix := strings.SplitN(mentionedJid, "@", 2) | ||||
| 		for _, mentionedJID := range extText.ContextInfo.MentionedJid { | ||||
| 			numberAndSuffix := strings.SplitN(mentionedJID, "@", 2) | ||||
|  | ||||
| 			// mentions comes as telephone numbers and we don't want to expose it to other bridges | ||||
| 			// replace it with something more meaninful to others | ||||
| 			mention := b.getSenderNotify(numberAndSuffix[0] + whatsappExt.NewUserSuffix) | ||||
| 			mention := b.getSenderNotify(numberAndSuffix[0] + "@s.whatsapp.net") | ||||
| 			if mention == "" { | ||||
| 				mention = "someone" | ||||
| 			} | ||||
|  | ||||
| 			message.Text = strings.Replace(message.Text, "@"+numberAndSuffix[0], "@"+mention, 1) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJid, b.Account) | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:    senderJid, | ||||
| 		Username:  senderName, | ||||
| 		Text:      message.Text, | ||||
| 		Timestamp: messageTime, | ||||
| 		Channel:   groupJid, | ||||
| 		Account:   b.Account, | ||||
| 		Protocol:  b.Protocol, | ||||
| 		Extra:     make(map[string][]interface{}), | ||||
| 		//		ParentID: TODO, // TODO handle thread replies  // map from Info.QuotedMessageID string | ||||
| 		//	Event     string    `json:"event"` | ||||
| 		//	Gateway   string  // will be added during message processing | ||||
| 		ID: message.Info.Id} | ||||
| 		UserID:   senderJID, | ||||
| 		Username: senderName, | ||||
| 		Text:     message.Text, | ||||
| 		Channel:  groupJID, | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		//	ParentID: TODO, // TODO handle thread replies  // map from Info.QuotedMessageID string | ||||
| 		ID: message.Info.Id, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJid]; exists { | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleImageMessage(message whatsapp.ImageMessage) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleVideoMessage(message whatsapp.VideoMessage) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleJsonMessage(message string) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // TODO HandleRawMessage | ||||
| // TODO HandleAudioMessage | ||||
| // HandleImageMessage sent from WhatsApp, relay it to the brige | ||||
| // nolint:funlen | ||||
| func (b *Bwhatsapp) HandleImageMessage(message whatsapp.ImageMessage) { | ||||
| 	if message.Info.FromMe || message.Info.Timestamp < b.startedAt { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	senderJID := message.Info.SenderJid | ||||
| 	if len(message.Info.SenderJid) == 0 && message.Info.Source != nil && message.Info.Source.Participant != nil { | ||||
| 		senderJID = *message.Info.Source.Participant | ||||
| 	} | ||||
|  | ||||
| 	senderName := b.getSenderName(message.Info.SenderJid) | ||||
| 	if senderName == "" { | ||||
| 		senderName = "Someone" // don't expose telephone number | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID, | ||||
| 		Username: senderName, | ||||
| 		Channel:  message.Info.RemoteJid, | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       message.Info.Id, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(message.Type) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// rename .jfif to .jpg https://github.com/42wim/matterbridge/issues/1292 | ||||
| 	if fileExt[0] == ".jfif" { | ||||
| 		fileExt[0] = ".jpg" | ||||
| 	} | ||||
|  | ||||
| 	// rename .jpe to .jpg https://github.com/42wim/matterbridge/issues/1463 | ||||
| 	if fileExt[0] == ".jpe" { | ||||
| 		fileExt[0] = ".jpg" | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", message.Info.Id, fileExt[0]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with type %s", filename, message.Type) | ||||
|  | ||||
| 	data, err := message.Download() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download image failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, message.Caption, "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleVideoMessage downloads video messages | ||||
| func (b *Bwhatsapp) HandleVideoMessage(message whatsapp.VideoMessage) { | ||||
| 	if message.Info.FromMe || message.Info.Timestamp < b.startedAt { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	senderJID := message.Info.SenderJid | ||||
| 	if len(message.Info.SenderJid) == 0 && message.Info.Source != nil && message.Info.Source.Participant != nil { | ||||
| 		senderJID = *message.Info.Source.Participant | ||||
| 	} | ||||
|  | ||||
| 	senderName := b.getSenderName(message.Info.SenderJid) | ||||
| 	if senderName == "" { | ||||
| 		senderName = "Someone" // don't expose telephone number | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID, | ||||
| 		Username: senderName, | ||||
| 		Channel:  message.Info.RemoteJid, | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       message.Info.Id, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(message.Type) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if len(fileExt) == 0 { | ||||
| 		fileExt = append(fileExt, ".mp4") | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", message.Info.Id, fileExt[0]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with size %#v and type %s", filename, message.Length, message.Type) | ||||
|  | ||||
| 	data, err := message.Download() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download video failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, message.Caption, "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleAudioMessage downloads audio messages | ||||
| func (b *Bwhatsapp) HandleAudioMessage(message whatsapp.AudioMessage) { | ||||
| 	if message.Info.FromMe || message.Info.Timestamp < b.startedAt { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	senderJID := message.Info.SenderJid | ||||
| 	if len(message.Info.SenderJid) == 0 && message.Info.Source != nil && message.Info.Source.Participant != nil { | ||||
| 		senderJID = *message.Info.Source.Participant | ||||
| 	} | ||||
|  | ||||
| 	senderName := b.getSenderName(message.Info.SenderJid) | ||||
| 	if senderName == "" { | ||||
| 		senderName = "Someone" // don't expose telephone number | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID, | ||||
| 		Username: senderName, | ||||
| 		Channel:  message.Info.RemoteJid, | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       message.Info.Id, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(message.Type) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if len(fileExt) == 0 { | ||||
| 		fileExt = append(fileExt, ".ogg") | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", message.Info.Id, fileExt[0]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with size %#v and type %s", filename, message.Length, message.Type) | ||||
|  | ||||
| 	data, err := message.Download() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download audio failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, "audio message", "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleDocumentMessage downloads documents | ||||
| func (b *Bwhatsapp) HandleDocumentMessage(message whatsapp.DocumentMessage) { | ||||
| 	if message.Info.FromMe || message.Info.Timestamp < b.startedAt { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	senderJID := message.Info.SenderJid | ||||
| 	if len(message.Info.SenderJid) == 0 && message.Info.Source != nil && message.Info.Source.Participant != nil { | ||||
| 		senderJID = *message.Info.Source.Participant | ||||
| 	} | ||||
|  | ||||
| 	senderName := b.getSenderName(message.Info.SenderJid) | ||||
| 	if senderName == "" { | ||||
| 		senderName = "Someone" // don't expose telephone number | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID, | ||||
| 		Username: senderName, | ||||
| 		Channel:  message.Info.RemoteJid, | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       message.Info.Id, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(message.Type) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v", message.FileName) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with extension %s and type %s", filename, fileExt, message.Type) | ||||
|  | ||||
| 	data, err := message.Download() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download document message failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, "document", "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|   | ||||
| @@ -2,17 +2,28 @@ package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"encoding/gob" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"strings" | ||||
|  | ||||
| 	qrcodeTerminal "github.com/Baozisoftware/qrcode-terminal-go" | ||||
| 	"github.com/matterbridge/go-whatsapp" | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
| ) | ||||
|  | ||||
| type ProfilePicInfo struct { | ||||
| 	URL    string `json:"eurl"` | ||||
| 	Tag    string `json:"tag"` | ||||
| 	Status int16  `json:"status"` | ||||
| } | ||||
|  | ||||
| func qrFromTerminal(invert bool) chan string { | ||||
| 	qr := make(chan string) | ||||
|  | ||||
| 	go func() { | ||||
| 		terminal := qrcodeTerminal.New() | ||||
|  | ||||
| 		if invert { | ||||
| 			terminal = qrcodeTerminal.New2(qrcodeTerminal.ConsoleColors.BrightWhite, qrcodeTerminal.ConsoleColors.BrightBlack, qrcodeTerminal.QRCodeRecoveryLevels.Medium) | ||||
| 		} | ||||
| @@ -35,13 +46,12 @@ func (b *Bwhatsapp) readSession() (whatsapp.Session, error) { | ||||
| 	if err != nil { | ||||
| 		return session, err | ||||
| 	} | ||||
|  | ||||
| 	defer file.Close() | ||||
|  | ||||
| 	decoder := gob.NewDecoder(file) | ||||
| 	err = decoder.Decode(&session) | ||||
| 	if err != nil { | ||||
| 		return session, err | ||||
| 	} | ||||
| 	return session, nil | ||||
|  | ||||
| 	return session, decoder.Decode(&session) | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) writeSession(session whatsapp.Session) error { | ||||
| @@ -56,11 +66,31 @@ func (b *Bwhatsapp) writeSession(session whatsapp.Session) error { | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	defer file.Close() | ||||
| 	encoder := gob.NewEncoder(file) | ||||
| 	err = encoder.Encode(session) | ||||
|  | ||||
| 	return err | ||||
| 	defer file.Close() | ||||
|  | ||||
| 	encoder := gob.NewEncoder(file) | ||||
|  | ||||
| 	return encoder.Encode(session) | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) restoreSession() (*whatsapp.Session, error) { | ||||
| 	session, err := b.readSession() | ||||
| 	if err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugln("Restoring WhatsApp session..") | ||||
|  | ||||
| 	session, err = b.conn.RestoreWithSession(session) | ||||
| 	if err != nil { | ||||
| 		// restore session connection timed out (I couldn't get over it without logging in again) | ||||
| 		return nil, errors.New("failed to restore session: " + err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugln("Session restored successfully!") | ||||
|  | ||||
| 	return &session, nil | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getSenderName(senderJid string) string { | ||||
| @@ -71,8 +101,32 @@ func (b *Bwhatsapp) getSenderName(senderJid string) string { | ||||
| 		// if user is not in phone contacts | ||||
| 		// it is the most obvious scenario unless you sync your phone contacts with some remote updated source | ||||
| 		// users can change it in their WhatsApp settings -> profile -> click on Avatar | ||||
| 		return sender.Notify | ||||
| 		if sender.Notify != "" { | ||||
| 			return sender.Notify | ||||
| 		} | ||||
|  | ||||
| 		if sender.Short != "" { | ||||
| 			return sender.Short | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// try to reload this contact | ||||
| 	if _, err := b.conn.Contacts(); err != nil { | ||||
| 		b.Log.Errorf("error on update of contacts: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	if contact, exists := b.conn.Store.Contacts[senderJid]; exists { | ||||
| 		// Add it to the user map | ||||
| 		b.users[senderJid] = contact | ||||
|  | ||||
| 		if contact.Name != "" { | ||||
| 			return contact.Name | ||||
| 		} | ||||
| 		// if user is not in phone contacts | ||||
| 		// same as above | ||||
| 		return contact.Notify | ||||
| 	} | ||||
|  | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| @@ -80,5 +134,29 @@ func (b *Bwhatsapp) getSenderNotify(senderJid string) string { | ||||
| 	if sender, exists := b.users[senderJid]; exists { | ||||
| 		return sender.Notify | ||||
| 	} | ||||
|  | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) GetProfilePicThumb(jid string) (*ProfilePicInfo, error) { | ||||
| 	data, err := b.conn.GetProfilePicThumb(jid) | ||||
| 	if err != nil { | ||||
| 		return nil, fmt.Errorf("failed to get avatar: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	content := <-data | ||||
| 	info := &ProfilePicInfo{} | ||||
|  | ||||
| 	err = json.Unmarshal([]byte(content), info) | ||||
| 	if err != nil { | ||||
| 		return info, fmt.Errorf("failed to unmarshal avatar info: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	return info, nil | ||||
| } | ||||
|  | ||||
| func isGroupJid(identifier string) bool { | ||||
| 	return strings.HasSuffix(identifier, "@g.us") || | ||||
| 		strings.HasSuffix(identifier, "@temp") || | ||||
| 		strings.HasSuffix(identifier, "@broadcast") | ||||
| } | ||||
|   | ||||
| @@ -1,20 +1,20 @@ | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/rand" | ||||
| 	"encoding/hex" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"mime" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
|  | ||||
| 	"github.com/matterbridge/go-whatsapp" | ||||
|  | ||||
| 	whatsappExt "github.com/matterbridge/mautrix-whatsapp/whatsapp-ext" | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| @@ -28,11 +28,8 @@ const ( | ||||
| type Bwhatsapp struct { | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	// https://github.com/Rhymen/go-whatsapp/blob/c31092027237441cffba1b9cb148eadf7c83c3d2/session.go#L18-L21 | ||||
| 	session *whatsapp.Session | ||||
| 	conn    *whatsapp.Conn | ||||
| 	// https://github.com/tulir/mautrix-whatsapp/blob/master/whatsapp-ext/whatsapp.go | ||||
| 	connExt   *whatsappExt.ExtendedConn | ||||
| 	session   *whatsapp.Session | ||||
| 	conn      *whatsapp.Conn | ||||
| 	startedAt uint64 | ||||
|  | ||||
| 	users       map[string]whatsapp.Contact | ||||
| @@ -42,6 +39,12 @@ type Bwhatsapp struct { | ||||
| // New Create a new WhatsApp bridge. This will be called for each [whatsapp.<server>] entry you have in the config file | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	number := cfg.GetString(cfgNumber) | ||||
|  | ||||
| 	cfg.Log.Warn("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") | ||||
| 	cfg.Log.Warn("This bridge is deprecated and not supported anymore. Use the new multidevice whatsapp bridge") | ||||
| 	cfg.Log.Warn("See https://github.com/42wim/matterbridge#building-with-whatsapp-beta-multidevice-support for more info") | ||||
| 	cfg.Log.Warn("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") | ||||
|  | ||||
| 	if number == "" { | ||||
| 		cfg.Log.Fatalf("Missing configuration for WhatsApp bridge: Number") | ||||
| 	} | ||||
| @@ -52,21 +55,17 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 		users:       make(map[string]whatsapp.Contact), | ||||
| 		userAvatars: make(map[string]string), | ||||
| 	} | ||||
|  | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| // Connect to WhatsApp. Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) Connect() error { | ||||
| 	b.RLock() // TODO do we need locking for Whatsapp? | ||||
| 	defer b.RUnlock() | ||||
|  | ||||
| 	number := b.GetString(cfgNumber) | ||||
| 	if number == "" { | ||||
| 		return errors.New("WhatsApp's telephone Number need to be configured") | ||||
| 		return errors.New("whatsapp's telephone number need to be configured") | ||||
| 	} | ||||
|  | ||||
| 	// https://github.com/Rhymen/go-whatsapp#creating-a-connection | ||||
| 	b.Log.Debugln("Connecting to WhatsApp..") | ||||
| 	conn, err := whatsapp.NewConn(20 * time.Second) | ||||
| 	if err != nil { | ||||
| @@ -74,42 +73,23 @@ func (b *Bwhatsapp) Connect() error { | ||||
| 	} | ||||
|  | ||||
| 	b.conn = conn | ||||
| 	b.connExt = whatsappExt.ExtendConn(b.conn) | ||||
| 	// TODO do we want to use it? b.connExt.SetClientName("Matterbridge WhatsApp bridge", "mb-wa") | ||||
|  | ||||
| 	b.conn.AddHandler(b) | ||||
| 	b.Log.Debugln("WhatsApp connection successful") | ||||
|  | ||||
| 	// load existing session in order to keep it between restarts | ||||
| 	if b.session == nil { | ||||
| 		var session whatsapp.Session | ||||
| 		session, err = b.readSession() | ||||
|  | ||||
| 		if err == nil { | ||||
| 			b.Log.Debugln("Restoring WhatsApp session..") | ||||
|  | ||||
| 			// https://github.com/Rhymen/go-whatsapp#restore | ||||
| 			session, err = b.conn.RestoreSession(session) | ||||
| 			if err != nil { | ||||
| 				// TODO return or continue to normal login? | ||||
| 				// restore session connection timed out (I couldn't get over it without logging in again) | ||||
| 				return errors.New("failed to restore session: " + err.Error()) | ||||
| 			} | ||||
|  | ||||
| 			b.session = &session | ||||
| 			b.Log.Debugln("Session restored successfully!") | ||||
| 		} else { | ||||
| 			b.Log.Warn(err.Error()) | ||||
| 		} | ||||
| 	b.session, err = b.restoreSession() | ||||
| 	if err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	// login to a new session | ||||
| 	if b.session == nil { | ||||
| 		err = b.Login() | ||||
| 		if err != nil { | ||||
| 		if err = b.Login(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.startedAt = uint64(time.Now().Unix()) | ||||
|  | ||||
| 	_, err = b.conn.Contacts() | ||||
| @@ -117,6 +97,13 @@ func (b *Bwhatsapp) Connect() error { | ||||
| 		return fmt.Errorf("error on update of contacts: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	// see https://github.com/Rhymen/go-whatsapp/issues/137#issuecomment-480316013 | ||||
| 	for len(b.conn.Store.Contacts) == 0 { | ||||
| 		b.conn.Contacts() // nolint:errcheck | ||||
|  | ||||
| 		<-time.After(1 * time.Second) | ||||
| 	} | ||||
|  | ||||
| 	// map all the users | ||||
| 	for id, contact := range b.conn.Store.Contacts { | ||||
| 		if !isGroupJid(id) && id != "status@broadcast" { | ||||
| @@ -130,15 +117,16 @@ func (b *Bwhatsapp) Connect() error { | ||||
| 		b.Log.Debug("Getting user avatars..") | ||||
|  | ||||
| 		for jid := range b.users { | ||||
| 			info, err := b.connExt.GetProfilePicThumb(jid) | ||||
| 			info, err := b.GetProfilePicThumb(jid) | ||||
| 			if err != nil { | ||||
| 				b.Log.Warnf("Could not get profile photo of %s: %v", jid, err) | ||||
|  | ||||
| 			} else { | ||||
| 				// TODO any race conditions here? | ||||
| 				b.Lock() | ||||
| 				b.userAvatars[jid] = info.URL | ||||
| 				b.Unlock() | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		b.Log.Debug("Finished getting avatars..") | ||||
| 	}() | ||||
|  | ||||
| @@ -155,8 +143,10 @@ func (b *Bwhatsapp) Login() error { | ||||
| 	session, err := b.conn.Login(qrChan) | ||||
| 	if err != nil { | ||||
| 		b.Log.Warnln("Failed to log in:", err) | ||||
|  | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	b.session = &session | ||||
|  | ||||
| 	b.Log.Infof("Logged into session: %#v", session) | ||||
| @@ -167,74 +157,122 @@ func (b *Bwhatsapp) Login() error { | ||||
| 		fmt.Fprintf(os.Stderr, "error saving session: %v\n", err) | ||||
| 	} | ||||
|  | ||||
| 	// TODO change connection strings to configured ones longClientName:"github.com/rhymen/go-whatsapp", shortClientName:"go-whatsapp"}" prefix=whatsapp | ||||
| 	// TODO get also a nice logo | ||||
|  | ||||
| 	// TODO notification about unplugged and dead battery | ||||
| 	// conn.Info: Wid, Pushname, Connected, Battery, Plugged | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Disconnect is called while reconnecting to the bridge | ||||
| // TODO 42wim Documentation would be helpful on when reconnects happen and what should be done in this function | ||||
| // Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) Disconnect() error { | ||||
| 	// We could Logout, but that would close the session completely and would require a new QR code scan | ||||
| 	// https://github.com/Rhymen/go-whatsapp/blob/c31092027237441cffba1b9cb148eadf7c83c3d2/session.go#L377-L381 | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func isGroupJid(identifier string) bool { | ||||
| 	return strings.HasSuffix(identifier, "@g.us") || strings.HasSuffix(identifier, "@temp") | ||||
| } | ||||
|  | ||||
| // JoinChannel Join a WhatsApp group specified in gateway config as channel='number-id@g.us' or channel='Channel name' | ||||
| // Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	byJid := isGroupJid(channel.Name) | ||||
|  | ||||
| 	// see https://github.com/Rhymen/go-whatsapp/issues/137#issuecomment-480316013 | ||||
| 	for len(b.conn.Store.Contacts) == 0 { | ||||
| 		b.conn.Contacts() // nolint:errcheck | ||||
| 		<-time.After(1 * time.Second) | ||||
| 	} | ||||
|  | ||||
| 	// verify if we are member of the given group | ||||
| 	if byJid { | ||||
| 		// channel.Name specifies static group jID, not the name | ||||
| 		if _, exists := b.conn.Store.Contacts[channel.Name]; !exists { | ||||
| 			return fmt.Errorf("account doesn't belong to group with jid %s", channel.Name) | ||||
| 		} | ||||
| 	} else { | ||||
| 		// channel.Name specifies group name that might change, warn about it | ||||
| 		var jids []string | ||||
| 		for id, contact := range b.conn.Store.Contacts { | ||||
| 			if isGroupJid(id) && contact.Name == channel.Name { | ||||
| 				jids = append(jids, id) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		switch len(jids) { | ||||
| 		case 0: | ||||
| 			// didn't match any group - print out possibilites | ||||
| 			// TODO sort | ||||
| 			// copy b; | ||||
| 			//sort.Slice(people, func(i, j int) bool { | ||||
| 			//	return people[i].Age > people[j].Age | ||||
| 			//}) | ||||
| 			for id, contact := range b.conn.Store.Contacts { | ||||
| 				if isGroupJid(id) { | ||||
| 					b.Log.Infof("%s %s", contact.Jid, contact.Name) | ||||
| 				} | ||||
| 			} | ||||
| 			return fmt.Errorf("please specify group's JID from the list above instead of the name '%s'", channel.Name) | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 		case 1: | ||||
| 			return fmt.Errorf("group name might change. Please configure gateway with channel=\"%v\" instead of channel=\"%v\"", jids[0], channel.Name) | ||||
|  | ||||
| 		default: | ||||
| 			return fmt.Errorf("there is more than one group with name '%s'. Please specify one of JIDs as channel name: %v", channel.Name, jids) | ||||
| 	// channel.Name specifies group name that might change, warn about it | ||||
| 	var jids []string | ||||
| 	for id, contact := range b.conn.Store.Contacts { | ||||
| 		if isGroupJid(id) && contact.Name == channel.Name { | ||||
| 			jids = append(jids, id) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| 	switch len(jids) { | ||||
| 	case 0: | ||||
| 		// didn't match any group - print out possibilites | ||||
| 		for id, contact := range b.conn.Store.Contacts { | ||||
| 			if isGroupJid(id) { | ||||
| 				b.Log.Infof("%s %s", contact.Jid, contact.Name) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		return fmt.Errorf("please specify group's JID from the list above instead of the name '%s'", channel.Name) | ||||
| 	case 1: | ||||
| 		return fmt.Errorf("group name might change. Please configure gateway with channel=\"%v\" instead of channel=\"%v\"", jids[0], channel.Name) | ||||
| 	default: | ||||
| 		return fmt.Errorf("there is more than one group with name '%s'. Please specify one of JIDs as channel name: %v", channel.Name, jids) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Post a document message from the bridge to WhatsApp | ||||
| func (b *Bwhatsapp) PostDocumentMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	// Post document message | ||||
| 	message := whatsapp.DocumentMessage{ | ||||
| 		Info: whatsapp.MessageInfo{ | ||||
| 			RemoteJid: msg.Channel, | ||||
| 		}, | ||||
| 		Title:    fi.Name, | ||||
| 		FileName: fi.Name, | ||||
| 		Type:     filetype, | ||||
| 		Content:  bytes.NewReader(*fi.Data), | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v", msg) | ||||
|  | ||||
| 	// create message ID | ||||
| 	// TODO follow and act if https://github.com/Rhymen/go-whatsapp/issues/101 implemented | ||||
| 	idBytes := make([]byte, 10) | ||||
| 	if _, err := rand.Read(idBytes); err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	message.Info.Id = strings.ToUpper(hex.EncodeToString(idBytes)) | ||||
| 	_, err := b.conn.Send(message) | ||||
|  | ||||
| 	return message.Info.Id, err | ||||
| } | ||||
|  | ||||
| // Post an image message from the bridge to WhatsApp | ||||
| // Handle, for sure image/jpeg, image/png and image/gif MIME types | ||||
| func (b *Bwhatsapp) PostImageMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	// Post image message | ||||
| 	message := whatsapp.ImageMessage{ | ||||
| 		Info: whatsapp.MessageInfo{ | ||||
| 			RemoteJid: msg.Channel, | ||||
| 		}, | ||||
| 		Type:    filetype, | ||||
| 		Caption: msg.Username + fi.Comment, | ||||
| 		Content: bytes.NewReader(*fi.Data), | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v", msg) | ||||
|  | ||||
| 	// create message ID | ||||
| 	// TODO follow and act if https://github.com/Rhymen/go-whatsapp/issues/101 implemented | ||||
| 	idBytes := make([]byte, 10) | ||||
| 	if _, err := rand.Read(idBytes); err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	message.Info.Id = strings.ToUpper(hex.EncodeToString(idBytes)) | ||||
| 	_, err := b.conn.Send(message) | ||||
|  | ||||
| 	return message.Info.Id, err | ||||
| } | ||||
|  | ||||
| // Send a message from the bridge to WhatsApp | ||||
| @@ -248,36 +286,44 @@ func (b *Bwhatsapp) Send(msg config.Message) (string, error) { | ||||
| 		if msg.ID == "" { | ||||
| 			// No message ID in case action is executed on a message sent before the bridge was started | ||||
| 			// and then the bridge cache doesn't have this message ID mapped | ||||
|  | ||||
| 			// TODO 42wim Doesn't the app get clogged with a ton of IDs after some time of running? | ||||
| 			// WhatsApp allows to set any ID so in that case we could use external IDs and don't do mapping | ||||
| 			// but external IDs are not set | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		// TODO delete message on WhatsApp https://github.com/Rhymen/go-whatsapp/issues/100 | ||||
| 		return "", nil | ||||
|  | ||||
| 		_, err := b.conn.RevokeMessage(msg.Channel, msg.ID, true) | ||||
|  | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// Edit message | ||||
| 	if msg.ID != "" { | ||||
| 		b.Log.Debugf("updating message with id %s", msg.ID) | ||||
|  | ||||
| 		msg.Text += " (edited)" | ||||
| 		// TODO handle edit as a message reply with updated text | ||||
| 		if b.GetString("editsuffix") != "" { | ||||
| 			msg.Text += b.GetString("EditSuffix") | ||||
| 		} else { | ||||
| 			msg.Text += " (edited)" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	//// TODO Handle Upload a file | ||||
| 	//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) | ||||
| 	//	} | ||||
| 	//} | ||||
| 	// Handle Upload a file | ||||
| 	if msg.Extra["file"] != nil { | ||||
| 		fi := msg.Extra["file"][0].(config.FileInfo) | ||||
| 		filetype := mime.TypeByExtension(filepath.Ext(fi.Name)) | ||||
|  | ||||
| 		b.Log.Debugf("Extra file is %#v", filetype) | ||||
|  | ||||
| 		// TODO: add different types | ||||
| 		// TODO: add webp conversion | ||||
| 		switch filetype { | ||||
| 		case "image/jpeg", "image/png", "image/gif": | ||||
| 			return b.PostImageMessage(msg, filetype) | ||||
| 		default: | ||||
| 			return b.PostDocumentMessage(msg, filetype) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Post text message | ||||
| 	text := whatsapp.TextMessage{ | ||||
| 	message := whatsapp.TextMessage{ | ||||
| 		Info: whatsapp.MessageInfo{ | ||||
| 			RemoteJid: msg.Channel, // which equals to group id | ||||
| 		}, | ||||
| @@ -286,17 +332,7 @@ func (b *Bwhatsapp) Send(msg config.Message) (string, error) { | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v", msg) | ||||
|  | ||||
| 	// create message ID | ||||
| 	// TODO follow and act if https://github.com/Rhymen/go-whatsapp/issues/101 implemented | ||||
| 	bytes := make([]byte, 10) | ||||
| 	if _, err := rand.Read(bytes); err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
| 	text.Info.Id = strings.ToUpper(hex.EncodeToString(bytes)) | ||||
|  | ||||
| 	err := b.conn.Send(text) | ||||
|  | ||||
| 	return text.Info.Id, err | ||||
| 	return b.conn.Send(message) | ||||
| } | ||||
|  | ||||
| // TODO do we want that? to allow login with QR code from a bridged channel? https://github.com/tulir/mautrix-whatsapp/blob/513eb18e2d59bada0dd515ee1abaaf38a3bfe3d5/commands.go#L76 | ||||
|   | ||||
							
								
								
									
										454
									
								
								bridge/whatsappmulti/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										454
									
								
								bridge/whatsappmulti/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,454 @@ | ||||
| //go:build whatsappmulti | ||||
| // +build whatsappmulti | ||||
|  | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"mime" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
|  | ||||
| 	"go.mau.fi/whatsmeow/binary/proto" | ||||
| 	"go.mau.fi/whatsmeow/types" | ||||
| 	"go.mau.fi/whatsmeow/types/events" | ||||
| ) | ||||
|  | ||||
| // nolint:gocritic | ||||
| func (b *Bwhatsapp) eventHandler(evt interface{}) { | ||||
| 	switch e := evt.(type) { | ||||
| 	case *events.Message: | ||||
| 		b.handleMessage(e) | ||||
| 	case *events.GroupInfo: | ||||
| 		b.handleGroupInfo(e) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleGroupInfo(event *events.GroupInfo) { | ||||
| 	b.Log.Debugf("Receiving event %#v", event) | ||||
|  | ||||
| 	switch { | ||||
| 	case event.Join != nil: | ||||
| 		b.handleUserJoin(event) | ||||
| 	case event.Leave != nil: | ||||
| 		b.handleUserLeave(event) | ||||
| 	case event.Topic != nil: | ||||
| 		b.handleTopicChange(event) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleUserJoin(event *events.GroupInfo) { | ||||
| 	for _, joinedJid := range event.Join { | ||||
| 		senderName := b.getSenderNameFromJID(joinedJid) | ||||
|  | ||||
| 		rmsg := config.Message{ | ||||
| 			UserID:   joinedJid.String(), | ||||
| 			Username: senderName, | ||||
| 			Channel:  event.JID.String(), | ||||
| 			Account:  b.Account, | ||||
| 			Protocol: b.Protocol, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 			Text:     "joined chat", | ||||
| 		} | ||||
|  | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleUserLeave(event *events.GroupInfo) { | ||||
| 	for _, leftJid := range event.Leave { | ||||
| 		senderName := b.getSenderNameFromJID(leftJid) | ||||
|  | ||||
| 		rmsg := config.Message{ | ||||
| 			UserID:   leftJid.String(), | ||||
| 			Username: senderName, | ||||
| 			Channel:  event.JID.String(), | ||||
| 			Account:  b.Account, | ||||
| 			Protocol: b.Protocol, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 			Text:     "left chat", | ||||
| 		} | ||||
|  | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleTopicChange(event *events.GroupInfo) { | ||||
| 	msg := event.Topic | ||||
| 	senderJid := msg.TopicSetBy | ||||
| 	senderName := b.getSenderNameFromJID(senderJid) | ||||
|  | ||||
| 	text := msg.Topic | ||||
| 	if text == "" { | ||||
| 		text = "removed topic" | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJid.String(), | ||||
| 		Username: senderName, | ||||
| 		Channel:  event.JID.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Event:    config.EventTopicChange, | ||||
| 		Text:     "Topic changed: " + text, | ||||
| 	} | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleMessage(message *events.Message) { | ||||
| 	msg := message.Message | ||||
| 	switch { | ||||
| 	case msg == nil, message.Info.IsFromMe, message.Info.Timestamp.Before(b.startedAt): | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("Receiving message %#v", msg) | ||||
|  | ||||
| 	switch { | ||||
| 	case msg.Conversation != nil || msg.ExtendedTextMessage != nil: | ||||
| 		b.handleTextMessage(message.Info, msg) | ||||
| 	case msg.VideoMessage != nil: | ||||
| 		b.handleVideoMessage(message) | ||||
| 	case msg.AudioMessage != nil: | ||||
| 		b.handleAudioMessage(message) | ||||
| 	case msg.DocumentMessage != nil: | ||||
| 		b.handleDocumentMessage(message) | ||||
| 	case msg.ImageMessage != nil: | ||||
| 		b.handleImageMessage(message) | ||||
| 	case msg.ProtocolMessage != nil && *msg.ProtocolMessage.Type == proto.ProtocolMessage_REVOKE: | ||||
| 		b.handleDelete(msg.ProtocolMessage) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // nolint:funlen | ||||
| func (b *Bwhatsapp) handleTextMessage(messageInfo types.MessageInfo, msg *proto.Message) { | ||||
| 	senderJID := messageInfo.Sender | ||||
| 	channel := messageInfo.Chat | ||||
|  | ||||
| 	senderName := b.getSenderName(messageInfo) | ||||
|  | ||||
| 	if msg.GetExtendedTextMessage() == nil && msg.GetConversation() == "" { | ||||
| 		b.Log.Debugf("message without text content? %#v", msg) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	var text string | ||||
|  | ||||
| 	// nolint:nestif | ||||
| 	if msg.GetExtendedTextMessage() == nil { | ||||
| 		text = msg.GetConversation() | ||||
| 	} else if msg.GetExtendedTextMessage().GetContextInfo() == nil { | ||||
| 		// Handle pure text message with a link preview | ||||
| 		// A pure text message with a link preview acts as an extended text message but will not contain any context info | ||||
| 		text = msg.GetExtendedTextMessage().GetText() | ||||
| 	} else { | ||||
| 		text = msg.GetExtendedTextMessage().GetText() | ||||
| 		ci := msg.GetExtendedTextMessage().GetContextInfo() | ||||
|  | ||||
| 		if senderJID == (types.JID{}) && ci.Participant != nil { | ||||
| 			senderJID = types.NewJID(ci.GetParticipant(), types.DefaultUserServer) | ||||
| 		} | ||||
|  | ||||
| 		if ci.MentionedJID != nil { | ||||
| 			// handle user mentions | ||||
| 			for _, mentionedJID := range ci.MentionedJID { | ||||
| 				numberAndSuffix := strings.SplitN(mentionedJID, "@", 2) | ||||
|  | ||||
| 				// mentions comes as telephone numbers and we don't want to expose it to other bridges | ||||
| 				// replace it with something more meaninful to others | ||||
| 				mention := b.getSenderNotify(types.NewJID(numberAndSuffix[0], types.DefaultUserServer)) | ||||
|  | ||||
| 				text = strings.Replace(text, "@"+numberAndSuffix[0], "@"+mention, 1) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	parentID := "" | ||||
| 	if msg.GetExtendedTextMessage() != nil { | ||||
| 		ci := msg.GetExtendedTextMessage().GetContextInfo() | ||||
| 		parentID = getParentIdFromCtx(ci) | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID.String(), | ||||
| 		Username: senderName, | ||||
| 		Text:     text, | ||||
| 		Channel:  channel.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       getMessageIdFormat(senderJID, messageInfo.ID), | ||||
| 		ParentID: parentID, | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID.String()]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleImageMessage sent from WhatsApp, relay it to the brige | ||||
| func (b *Bwhatsapp) handleImageMessage(msg *events.Message) { | ||||
| 	imsg := msg.Message.GetImageMessage() | ||||
|  | ||||
| 	senderJID := msg.Info.Sender | ||||
| 	senderName := b.getSenderName(msg.Info) | ||||
| 	ci := imsg.GetContextInfo() | ||||
|  | ||||
| 	if senderJID == (types.JID{}) && ci.Participant != nil { | ||||
| 		senderJID = types.NewJID(ci.GetParticipant(), types.DefaultUserServer) | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID.String(), | ||||
| 		Username: senderName, | ||||
| 		Channel:  msg.Info.Chat.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       getMessageIdFormat(senderJID, msg.Info.ID), | ||||
| 		ParentID: getParentIdFromCtx(ci), | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID.String()]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(imsg.GetMimetype()) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// rename .jfif to .jpg https://github.com/42wim/matterbridge/issues/1292 | ||||
| 	if fileExt[0] == ".jfif" { | ||||
| 		fileExt[0] = ".jpg" | ||||
| 	} | ||||
|  | ||||
| 	// rename .jpe to .jpg https://github.com/42wim/matterbridge/issues/1463 | ||||
| 	if fileExt[0] == ".jpe" { | ||||
| 		fileExt[0] = ".jpg" | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", msg.Info.ID, fileExt[0]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with type %s", filename, imsg.GetMimetype()) | ||||
|  | ||||
| 	data, err := b.wc.Download(imsg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download image failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, imsg.GetCaption(), "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleVideoMessage downloads video messages | ||||
| func (b *Bwhatsapp) handleVideoMessage(msg *events.Message) { | ||||
| 	imsg := msg.Message.GetVideoMessage() | ||||
|  | ||||
| 	senderJID := msg.Info.Sender | ||||
| 	senderName := b.getSenderName(msg.Info) | ||||
| 	ci := imsg.GetContextInfo() | ||||
|  | ||||
| 	if senderJID == (types.JID{}) && ci.Participant != nil { | ||||
| 		senderJID = types.NewJID(ci.GetParticipant(), types.DefaultUserServer) | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID.String(), | ||||
| 		Username: senderName, | ||||
| 		Channel:  msg.Info.Chat.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       getMessageIdFormat(senderJID, msg.Info.ID), | ||||
| 		ParentID: getParentIdFromCtx(ci), | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID.String()]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(imsg.GetMimetype()) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if len(fileExt) == 0 { | ||||
| 		fileExt = append(fileExt, ".mp4") | ||||
| 	} | ||||
|  | ||||
| 	// Prefer .mp4 extension, otherwise fallback to first index | ||||
| 	fileExtIndex := 0 | ||||
| 	for i, n := range fileExt { | ||||
| 		if ".mp4" == n { | ||||
| 			fileExtIndex = i | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", msg.Info.ID, fileExt[fileExtIndex]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with size %#v and type %s", filename, imsg.GetFileLength(), imsg.GetMimetype()) | ||||
|  | ||||
| 	data, err := b.wc.Download(imsg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download video failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, imsg.GetCaption(), "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleAudioMessage downloads audio messages | ||||
| func (b *Bwhatsapp) handleAudioMessage(msg *events.Message) { | ||||
| 	imsg := msg.Message.GetAudioMessage() | ||||
|  | ||||
| 	senderJID := msg.Info.Sender | ||||
| 	senderName := b.getSenderName(msg.Info) | ||||
| 	ci := imsg.GetContextInfo() | ||||
|  | ||||
| 	if senderJID == (types.JID{}) && ci.Participant != nil { | ||||
| 		senderJID = types.NewJID(ci.GetParticipant(), types.DefaultUserServer) | ||||
| 	} | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID.String(), | ||||
| 		Username: senderName, | ||||
| 		Channel:  msg.Info.Chat.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       getMessageIdFormat(senderJID, msg.Info.ID), | ||||
| 		ParentID: getParentIdFromCtx(ci), | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID.String()]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(imsg.GetMimetype()) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if len(fileExt) == 0 { | ||||
| 		fileExt = append(fileExt, ".ogg") | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v%v", msg.Info.ID, fileExt[0]) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with size %#v and type %s", filename, imsg.GetFileLength(), imsg.GetMimetype()) | ||||
|  | ||||
| 	data, err := b.wc.Download(imsg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download video failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, "audio message", "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // HandleDocumentMessage downloads documents | ||||
| func (b *Bwhatsapp) handleDocumentMessage(msg *events.Message) { | ||||
| 	imsg := msg.Message.GetDocumentMessage() | ||||
|  | ||||
| 	senderJID := msg.Info.Sender | ||||
| 	senderName := b.getSenderName(msg.Info) | ||||
| 	ci := imsg.GetContextInfo() | ||||
|  | ||||
| 	if senderJID == (types.JID{}) && ci.Participant != nil { | ||||
| 		senderJID = types.NewJID(ci.GetParticipant(), types.DefaultUserServer) | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:   senderJID.String(), | ||||
| 		Username: senderName, | ||||
| 		Channel:  msg.Info.Chat.String(), | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ID:       getMessageIdFormat(senderJID, msg.Info.ID), | ||||
| 		ParentID: getParentIdFromCtx(ci), | ||||
| 	} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJID.String()]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	fileExt, err := mime.ExtensionsByType(imsg.GetMimetype()) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Mimetype detection error: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	filename := fmt.Sprintf("%v", imsg.GetFileName()) | ||||
|  | ||||
| 	b.Log.Debugf("Trying to download %s with extension %s and type %s", filename, fileExt, imsg.GetMimetype()) | ||||
|  | ||||
| 	data, err := b.wc.Download(imsg) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Download document message failed: %s", err) | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Move file to bridge storage | ||||
| 	helper.HandleDownloadData(b.Log, &rmsg, filename, imsg.GetCaption(), "", &data, b.General) | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJID, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
|  | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) handleDelete(messageInfo *proto.ProtocolMessage) { | ||||
| 	sender, _ := types.ParseJID(*messageInfo.Key.Participant) | ||||
|  | ||||
| 	rmsg := config.Message{ | ||||
| 		Account:  b.Account, | ||||
| 		Protocol: b.Protocol, | ||||
| 		ID:       getMessageIdFormat(sender, *messageInfo.Key.ID), | ||||
| 		Event:    config.EventMsgDelete, | ||||
| 		Text:     config.EventMsgDelete, | ||||
| 		Channel:  *messageInfo.Key.RemoteJID, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s to gateway", b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
							
								
								
									
										209
									
								
								bridge/whatsappmulti/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										209
									
								
								bridge/whatsappmulti/helpers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,209 @@ | ||||
| //go:build whatsappmulti | ||||
| // +build whatsappmulti | ||||
|  | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
|  | ||||
| 	goproto "google.golang.org/protobuf/proto" | ||||
|  | ||||
| 	"go.mau.fi/whatsmeow" | ||||
| 	"go.mau.fi/whatsmeow/binary/proto" | ||||
| 	"go.mau.fi/whatsmeow/store" | ||||
| 	"go.mau.fi/whatsmeow/store/sqlstore" | ||||
| 	"go.mau.fi/whatsmeow/types" | ||||
| ) | ||||
|  | ||||
| type ProfilePicInfo struct { | ||||
| 	URL    string `json:"eurl"` | ||||
| 	Tag    string `json:"tag"` | ||||
| 	Status int16  `json:"status"` | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) reloadContacts() { | ||||
| 	if _, err := b.wc.Store.Contacts.GetAllContacts(); err != nil { | ||||
| 		b.Log.Errorf("error on update of contacts: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	allcontacts, err := b.wc.Store.Contacts.GetAllContacts() | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("error on update of contacts: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	if len(allcontacts) > 0 { | ||||
| 		b.contacts = allcontacts | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getSenderName(info types.MessageInfo) string { | ||||
| 	// Parse AD JID | ||||
| 	var senderJid types.JID | ||||
| 	senderJid.User, senderJid.Server = info.Sender.User, info.Sender.Server | ||||
|  | ||||
| 	sender, exists := b.contacts[senderJid] | ||||
|  | ||||
| 	if !exists || (sender.FullName == "" && sender.FirstName == "") { | ||||
| 		b.reloadContacts() // Contacts may need to be reloaded | ||||
| 		sender, exists = b.contacts[senderJid] | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FullName != "" { | ||||
| 		return sender.FullName | ||||
| 	} | ||||
|  | ||||
| 	if info.PushName != "" { | ||||
| 		return info.PushName | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FirstName != "" { | ||||
| 		return sender.FirstName | ||||
| 	} | ||||
|  | ||||
| 	return "Someone" | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getSenderNameFromJID(senderJid types.JID) string { | ||||
| 	sender, exists := b.contacts[senderJid] | ||||
|  | ||||
| 	if !exists || (sender.FullName == "" && sender.FirstName == "") { | ||||
| 		b.reloadContacts() // Contacts may need to be reloaded | ||||
| 		sender, exists = b.contacts[senderJid] | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FullName != "" { | ||||
| 		return sender.FullName | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FirstName != "" { | ||||
| 		return sender.FirstName | ||||
| 	} | ||||
|  | ||||
| 	if sender.PushName != "" { | ||||
| 		return sender.PushName | ||||
| 	} | ||||
|  | ||||
| 	return "Someone" | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getSenderNotify(senderJid types.JID) string { | ||||
| 	sender, exists := b.contacts[senderJid] | ||||
|  | ||||
| 	if !exists || (sender.FullName == "" && sender.PushName == "" && sender.FirstName == "") { | ||||
| 		b.reloadContacts() // Contacts may need to be reloaded | ||||
| 		sender, exists = b.contacts[senderJid] | ||||
| 	} | ||||
|  | ||||
| 	if !exists { | ||||
| 		return "someone" | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FullName != "" { | ||||
| 		return sender.FullName | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.PushName != "" { | ||||
| 		return sender.PushName | ||||
| 	} | ||||
|  | ||||
| 	if exists && sender.FirstName != "" { | ||||
| 		return sender.FirstName | ||||
| 	} | ||||
|  | ||||
| 	return "someone" | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) GetProfilePicThumb(jid string) (*types.ProfilePictureInfo, error) { | ||||
| 	pjid, _ := types.ParseJID(jid) | ||||
|  | ||||
| 	info, err := b.wc.GetProfilePictureInfo(pjid, &whatsmeow.GetProfilePictureParams{ | ||||
| 		Preview: true, | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		return nil, fmt.Errorf("failed to get avatar: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	return info, nil | ||||
| } | ||||
|  | ||||
| func isGroupJid(identifier string) bool { | ||||
| 	return strings.HasSuffix(identifier, "@g.us") || | ||||
| 		strings.HasSuffix(identifier, "@temp") || | ||||
| 		strings.HasSuffix(identifier, "@broadcast") | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getDevice() (*store.Device, error) { | ||||
| 	device := &store.Device{} | ||||
|  | ||||
| 	storeContainer, err := sqlstore.New("sqlite", "file:"+b.Config.GetString("sessionfile")+".db?_pragma=foreign_keys(1)&_pragma=busy_timeout=10000", nil) | ||||
| 	if err != nil { | ||||
| 		return device, fmt.Errorf("failed to connect to database: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	device, err = storeContainer.GetFirstDevice() | ||||
| 	if err != nil { | ||||
| 		return device, fmt.Errorf("failed to get device: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	return device, nil | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getNewReplyContext(parentID string) (*proto.ContextInfo, error) { | ||||
| 	replyInfo, err := b.parseMessageID(parentID) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	sender := fmt.Sprintf("%s@%s", replyInfo.Sender.User, replyInfo.Sender.Server) | ||||
| 	ctx := &proto.ContextInfo{ | ||||
| 		StanzaID:      &replyInfo.MessageID, | ||||
| 		Participant:   &sender, | ||||
| 		QuotedMessage: &proto.Message{Conversation: goproto.String("")}, | ||||
| 	} | ||||
|  | ||||
| 	return ctx, nil | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) parseMessageID(id string) (*Replyable, error) { | ||||
| 	// No message ID in case action is executed on a message sent before the bridge was started | ||||
| 	// and then the bridge cache doesn't have this message ID mapped | ||||
| 	if id == "" { | ||||
| 		return &Replyable{MessageID: id}, nil | ||||
| 	} | ||||
|  | ||||
| 	replyInfo := strings.Split(id, "/") | ||||
|  | ||||
| 	if len(replyInfo) == 2 { | ||||
| 		sender, err := types.ParseJID(replyInfo[0]) | ||||
|  | ||||
| 		if err == nil { | ||||
| 			return &Replyable{ | ||||
| 				MessageID: types.MessageID(replyInfo[1]), | ||||
| 				Sender:    sender, | ||||
| 			}, nil | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	err := fmt.Errorf("MessageID does not match format of {senderJID}:{messageID} : \"%s\"", id) | ||||
|  | ||||
| 	return &Replyable{MessageID: id}, err | ||||
| } | ||||
|  | ||||
| func getParentIdFromCtx(ci *proto.ContextInfo) string { | ||||
| 	if ci != nil && ci.StanzaID != nil { | ||||
| 		senderJid, err := types.ParseJID(*ci.Participant) | ||||
|  | ||||
| 		if err == nil { | ||||
| 			return getMessageIdFormat(senderJid, *ci.StanzaID) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func getMessageIdFormat(jid types.JID, messageID string) string { | ||||
| 	// we're crafting our own JID str as AD JID format messes with how stuff looks on a webclient | ||||
| 	jidStr := fmt.Sprintf("%s@%s", jid.User, jid.Server) | ||||
| 	return fmt.Sprintf("%s/%s", jidStr, messageID) | ||||
| } | ||||
							
								
								
									
										456
									
								
								bridge/whatsappmulti/whatsapp.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										456
									
								
								bridge/whatsappmulti/whatsapp.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,456 @@ | ||||
| //go:build whatsappmulti | ||||
| // +build whatsappmulti | ||||
|  | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"mime" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/mdp/qrterminal" | ||||
|  | ||||
| 	"go.mau.fi/whatsmeow" | ||||
| 	"go.mau.fi/whatsmeow/binary/proto" | ||||
| 	"go.mau.fi/whatsmeow/types" | ||||
| 	waLog "go.mau.fi/whatsmeow/util/log" | ||||
|  | ||||
| 	goproto "google.golang.org/protobuf/proto" | ||||
|  | ||||
| 	_ "modernc.org/sqlite" // needed for sqlite | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	// Account config parameters | ||||
| 	cfgNumber = "Number" | ||||
| ) | ||||
|  | ||||
| // Bwhatsapp Bridge structure keeping all the information needed for relying | ||||
| type Bwhatsapp struct { | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	startedAt    time.Time | ||||
| 	wc           *whatsmeow.Client | ||||
| 	contacts     map[types.JID]types.ContactInfo | ||||
| 	users        map[string]types.ContactInfo | ||||
| 	userAvatars  map[string]string | ||||
| 	joinedGroups []*types.GroupInfo | ||||
| } | ||||
|  | ||||
| type Replyable struct { | ||||
| 	MessageID types.MessageID | ||||
| 	Sender    types.JID | ||||
| } | ||||
|  | ||||
| // New Create a new WhatsApp bridge. This will be called for each [whatsapp.<server>] entry you have in the config file | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	number := cfg.GetString(cfgNumber) | ||||
|  | ||||
| 	if number == "" { | ||||
| 		cfg.Log.Fatalf("Missing configuration for WhatsApp bridge: Number") | ||||
| 	} | ||||
|  | ||||
| 	b := &Bwhatsapp{ | ||||
| 		Config: cfg, | ||||
|  | ||||
| 		users:       make(map[string]types.ContactInfo), | ||||
| 		userAvatars: make(map[string]string), | ||||
| 	} | ||||
|  | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| // Connect to WhatsApp. Required implementation of the Bridger interface | ||||
| func (b *Bwhatsapp) Connect() error { | ||||
| 	device, err := b.getDevice() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	number := b.GetString(cfgNumber) | ||||
| 	if number == "" { | ||||
| 		return errors.New("whatsapp's telephone number need to be configured") | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugln("Connecting to WhatsApp..") | ||||
|  | ||||
| 	b.wc = whatsmeow.NewClient(device, waLog.Stdout("Client", "INFO", true)) | ||||
| 	b.wc.AddEventHandler(b.eventHandler) | ||||
|  | ||||
| 	firstlogin := false | ||||
| 	var qrChan <-chan whatsmeow.QRChannelItem | ||||
| 	if b.wc.Store.ID == nil { | ||||
| 		firstlogin = true | ||||
| 		qrChan, err = b.wc.GetQRChannel(context.Background()) | ||||
| 		if err != nil && !errors.Is(err, whatsmeow.ErrQRStoreContainsID) { | ||||
| 			return errors.New("failed to to get QR channel:" + err.Error()) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	err = b.wc.Connect() | ||||
| 	if err != nil { | ||||
| 		return errors.New("failed to connect to WhatsApp: " + err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	if b.wc.Store.ID == nil { | ||||
| 		for evt := range qrChan { | ||||
| 			if evt.Event == "code" { | ||||
| 				qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout) | ||||
| 			} else { | ||||
| 				b.Log.Infof("QR channel result: %s", evt.Event) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// disconnect and reconnect on our first login/pairing | ||||
| 	// for some reason the GetJoinedGroups in JoinChannel doesn't work on first login | ||||
| 	if firstlogin { | ||||
| 		b.wc.Disconnect() | ||||
| 		time.Sleep(time.Second) | ||||
|  | ||||
| 		err = b.wc.Connect() | ||||
| 		if err != nil { | ||||
| 			return errors.New("failed to connect to WhatsApp: " + err.Error()) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Infoln("WhatsApp connection successful") | ||||
|  | ||||
| 	b.contacts, err = b.wc.Store.Contacts.GetAllContacts() | ||||
| 	if err != nil { | ||||
| 		return errors.New("failed to get contacts: " + err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	b.joinedGroups, err = b.wc.GetJoinedGroups() | ||||
| 	if err != nil { | ||||
| 		return errors.New("failed to get list of joined groups: " + err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	b.startedAt = time.Now() | ||||
|  | ||||
| 	// map all the users | ||||
| 	for id, contact := range b.contacts { | ||||
| 		if !isGroupJid(id.String()) && id.String() != "status@broadcast" { | ||||
| 			// it is user | ||||
| 			b.users[id.String()] = contact | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// get user avatar asynchronously | ||||
| 	b.Log.Info("Getting user avatars..") | ||||
|  | ||||
| 	for jid := range b.users { | ||||
| 		info, err := b.GetProfilePicThumb(jid) | ||||
| 		if err != nil { | ||||
| 			b.Log.Warnf("Could not get profile photo of %s: %v", jid, err) | ||||
| 		} else { | ||||
| 			b.Lock() | ||||
| 			if info != nil { | ||||
| 				b.userAvatars[jid] = info.URL | ||||
| 			} | ||||
| 			b.Unlock() | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Info("Finished getting avatars..") | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Disconnect is called while reconnecting to the bridge | ||||
| // Required implementation of the Bridger interface | ||||
| func (b *Bwhatsapp) Disconnect() error { | ||||
| 	b.wc.Disconnect() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // JoinChannel Join a WhatsApp group specified in gateway config as channel='number-id@g.us' or channel='Channel name' | ||||
| // Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	byJid := isGroupJid(channel.Name) | ||||
|  | ||||
| 	// verify if we are member of the given group | ||||
| 	if byJid { | ||||
| 		gJID, err := types.ParseJID(channel.Name) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
|  | ||||
| 		for _, group := range b.joinedGroups { | ||||
| 			if group.JID == gJID { | ||||
| 				return nil | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	foundGroups := []string{} | ||||
|  | ||||
| 	for _, group := range b.joinedGroups { | ||||
| 		if group.Name == channel.Name { | ||||
| 			foundGroups = append(foundGroups, group.Name) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	switch len(foundGroups) { | ||||
| 	case 0: | ||||
| 		// didn't match any group - print out possibilites | ||||
| 		for _, group := range b.joinedGroups { | ||||
| 			b.Log.Infof("%s %s", group.JID, group.Name) | ||||
| 		} | ||||
| 		return fmt.Errorf("please specify group's JID from the list above instead of the name '%s'", channel.Name) | ||||
| 	case 1: | ||||
| 		return fmt.Errorf("group name might change. Please configure gateway with channel=\"%v\" instead of channel=\"%v\"", foundGroups[0], channel.Name) | ||||
| 	default: | ||||
| 		return fmt.Errorf("there is more than one group with name '%s'. Please specify one of JIDs as channel name: %v", channel.Name, foundGroups) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Post a document message from the bridge to WhatsApp | ||||
| func (b *Bwhatsapp) PostDocumentMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	groupJID, _ := types.ParseJID(msg.Channel) | ||||
|  | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	caption := msg.Username + fi.Comment | ||||
|  | ||||
| 	resp, err := b.wc.Upload(context.Background(), *fi.Data, whatsmeow.MediaDocument) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// Post document message | ||||
| 	var message proto.Message | ||||
| 	var ctx *proto.ContextInfo | ||||
| 	if msg.ParentID != "" { | ||||
| 		ctx, _ = b.getNewReplyContext(msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	message.DocumentMessage = &proto.DocumentMessage{ | ||||
| 		Title:         &fi.Name, | ||||
| 		FileName:      &fi.Name, | ||||
| 		Mimetype:      &filetype, | ||||
| 		Caption:       &caption, | ||||
| 		MediaKey:      resp.MediaKey, | ||||
| 		FileEncSHA256: resp.FileEncSHA256, | ||||
| 		FileSHA256:    resp.FileSHA256, | ||||
| 		FileLength:    goproto.Uint64(resp.FileLength), | ||||
| 		URL:           &resp.URL, | ||||
| 		DirectPath:    &resp.DirectPath, | ||||
| 		ContextInfo:   ctx, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v as a document", msg) | ||||
|  | ||||
| 	ID := whatsmeow.GenerateMessageID() | ||||
| 	_, err = b.wc.SendMessage(context.TODO(), groupJID, &message, whatsmeow.SendRequestExtra{ID: ID}) | ||||
|  | ||||
| 	return ID, err | ||||
| } | ||||
|  | ||||
| // Post an image message from the bridge to WhatsApp | ||||
| // Handle, for sure image/jpeg, image/png and image/gif MIME types | ||||
| func (b *Bwhatsapp) PostImageMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	caption := msg.Username + fi.Comment | ||||
|  | ||||
| 	resp, err := b.wc.Upload(context.Background(), *fi.Data, whatsmeow.MediaImage) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	var message proto.Message | ||||
| 	var ctx *proto.ContextInfo | ||||
| 	if msg.ParentID != "" { | ||||
| 		ctx, _ = b.getNewReplyContext(msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	message.ImageMessage = &proto.ImageMessage{ | ||||
| 		Mimetype:      &filetype, | ||||
| 		Caption:       &caption, | ||||
| 		MediaKey:      resp.MediaKey, | ||||
| 		FileEncSHA256: resp.FileEncSHA256, | ||||
| 		FileSHA256:    resp.FileSHA256, | ||||
| 		FileLength:    goproto.Uint64(resp.FileLength), | ||||
| 		URL:           &resp.URL, | ||||
| 		DirectPath:    &resp.DirectPath, | ||||
| 		ContextInfo:   ctx, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v as an image", msg) | ||||
|  | ||||
| 	return b.sendMessage(msg, &message) | ||||
| } | ||||
|  | ||||
| // Post a video message from the bridge to WhatsApp | ||||
| func (b *Bwhatsapp) PostVideoMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	caption := msg.Username + fi.Comment | ||||
|  | ||||
| 	resp, err := b.wc.Upload(context.Background(), *fi.Data, whatsmeow.MediaVideo) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	var message proto.Message | ||||
| 	var ctx *proto.ContextInfo | ||||
| 	if msg.ParentID != "" { | ||||
| 		ctx, _ = b.getNewReplyContext(msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	message.VideoMessage = &proto.VideoMessage{ | ||||
| 		Mimetype:      &filetype, | ||||
| 		Caption:       &caption, | ||||
| 		MediaKey:      resp.MediaKey, | ||||
| 		FileEncSHA256: resp.FileEncSHA256, | ||||
| 		FileSHA256:    resp.FileSHA256, | ||||
| 		FileLength:    goproto.Uint64(resp.FileLength), | ||||
| 		URL:           &resp.URL, | ||||
| 		DirectPath:    &resp.DirectPath, | ||||
| 		ContextInfo:   ctx, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v as a video", msg) | ||||
|  | ||||
| 	return b.sendMessage(msg, &message) | ||||
| } | ||||
|  | ||||
| // Post audio inline | ||||
| func (b *Bwhatsapp) PostAudioMessage(msg config.Message, filetype string) (string, error) { | ||||
| 	groupJID, _ := types.ParseJID(msg.Channel) | ||||
|  | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
|  | ||||
| 	resp, err := b.wc.Upload(context.Background(), *fi.Data, whatsmeow.MediaAudio) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	var message proto.Message | ||||
| 	var ctx *proto.ContextInfo | ||||
| 	if msg.ParentID != "" { | ||||
| 		ctx, _ = b.getNewReplyContext(msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	message.AudioMessage = &proto.AudioMessage{ | ||||
| 		Mimetype:      &filetype, | ||||
| 		MediaKey:      resp.MediaKey, | ||||
| 		FileEncSHA256: resp.FileEncSHA256, | ||||
| 		FileSHA256:    resp.FileSHA256, | ||||
| 		FileLength:    goproto.Uint64(resp.FileLength), | ||||
| 		URL:           &resp.URL, | ||||
| 		DirectPath:    &resp.DirectPath, | ||||
| 		ContextInfo:   ctx, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v as audio", msg) | ||||
|  | ||||
| 	ID, err := b.sendMessage(msg, &message) | ||||
|  | ||||
| 	var captionMessage proto.Message | ||||
| 	caption := msg.Username + fi.Comment + "\u2B06" // the char on the end is upwards arrow emoji | ||||
| 	captionMessage.Conversation = &caption | ||||
|  | ||||
| 	captionID := whatsmeow.GenerateMessageID() | ||||
| 	_, err = b.wc.SendMessage(context.TODO(), groupJID, &captionMessage, whatsmeow.SendRequestExtra{ID: captionID}) | ||||
|  | ||||
| 	return ID, err | ||||
| } | ||||
|  | ||||
| // Send a message from the bridge to WhatsApp | ||||
| func (b *Bwhatsapp) Send(msg config.Message) (string, error) { | ||||
| 	groupJID, _ := types.ParseJID(msg.Channel) | ||||
|  | ||||
| 	extendedMsgID, _ := b.parseMessageID(msg.ID) | ||||
| 	msg.ID = extendedMsgID.MessageID | ||||
|  | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			// No message ID in case action is executed on a message sent before the bridge was started | ||||
| 			// and then the bridge cache doesn't have this message ID mapped | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		_, err := b.wc.RevokeMessage(groupJID, msg.ID) | ||||
|  | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// Edit message | ||||
| 	if msg.ID != "" { | ||||
| 		b.Log.Debugf("updating message with id %s", msg.ID) | ||||
|  | ||||
| 		if b.GetString("editsuffix") != "" { | ||||
| 			msg.Text += b.GetString("EditSuffix") | ||||
| 		} else { | ||||
| 			msg.Text += " (edited)" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Handle Upload a file | ||||
| 	if msg.Extra["file"] != nil { | ||||
| 		fi := msg.Extra["file"][0].(config.FileInfo) | ||||
| 		filetype := mime.TypeByExtension(filepath.Ext(fi.Name)) | ||||
|  | ||||
| 		b.Log.Debugf("Extra file is %#v", filetype) | ||||
|  | ||||
| 		// TODO: add different types | ||||
| 		// TODO: add webp conversion | ||||
| 		switch filetype { | ||||
| 		case "image/jpeg", "image/png", "image/gif": | ||||
| 			return b.PostImageMessage(msg, filetype) | ||||
| 		case "video/mp4", "video/3gpp": // TODO: Check if codecs are supported by WA | ||||
| 			return b.PostVideoMessage(msg, filetype) | ||||
| 		case "audio/ogg": | ||||
| 			return b.PostAudioMessage(msg, "audio/ogg; codecs=opus") // TODO: Detect if it is actually OPUS | ||||
| 		case "audio/aac", "audio/mp4", "audio/amr", "audio/mpeg": | ||||
| 			return b.PostAudioMessage(msg, filetype) | ||||
| 		default: | ||||
| 			return b.PostDocumentMessage(msg, filetype) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	var message proto.Message | ||||
| 	text := msg.Username + msg.Text | ||||
|  | ||||
| 	// If we have a parent ID send an extended message | ||||
| 	if msg.ParentID != "" { | ||||
| 		replyContext, err := b.getNewReplyContext(msg.ParentID) | ||||
|  | ||||
| 		if err == nil { | ||||
| 			message = proto.Message{ | ||||
| 				ExtendedTextMessage: &proto.ExtendedTextMessage{ | ||||
| 					Text:        &text, | ||||
| 					ContextInfo: replyContext, | ||||
| 				}, | ||||
| 			} | ||||
|  | ||||
| 			return b.sendMessage(msg, &message) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	message.Conversation = &text | ||||
|  | ||||
| 	return b.sendMessage(msg, &message) | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) sendMessage(rmsg config.Message, message *proto.Message) (string, error) { | ||||
| 	groupJID, _ := types.ParseJID(rmsg.Channel) | ||||
| 	ID := whatsmeow.GenerateMessageID() | ||||
|  | ||||
| 	_, err := b.wc.SendMessage(context.Background(), groupJID, message, whatsmeow.SendRequestExtra{ID: ID}) | ||||
|  | ||||
| 	return getMessageIdFormat(*b.wc.Store.ID, ID), err | ||||
| } | ||||
							
								
								
									
										34
									
								
								bridge/xmpp/handler.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								bridge/xmpp/handler.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | ||||
| package bxmpp | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/matterbridge/go-xmpp" | ||||
| ) | ||||
|  | ||||
| // 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 *Bxmpp) handleDownloadAvatar(avatar xmpp.AvatarData) { | ||||
| 	rmsg := config.Message{ | ||||
| 		Username: "system", | ||||
| 		Text:     "avatar", | ||||
| 		Channel:  b.parseChannel(avatar.From), | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   avatar.From, | ||||
| 		Event:    config.EventAvatarDownload, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 	} | ||||
| 	if _, ok := b.avatarMap[avatar.From]; !ok { | ||||
| 		b.Log.Debugf("Avatar.From: %s", avatar.From) | ||||
|  | ||||
| 		err := helper.HandleDownloadSize(b.Log, &rmsg, avatar.From+".png", int64(len(avatar.Data)), b.General) | ||||
| 		if err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 			return | ||||
| 		} | ||||
| 		helper.HandleDownloadData(b.Log, &rmsg, avatar.From+".png", rmsg.Text, "", &avatar.Data, b.General) | ||||
| 		b.Log.Debugf("Avatar download complete") | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										30
									
								
								bridge/xmpp/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								bridge/xmpp/helpers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,30 @@ | ||||
| package bxmpp | ||||
|  | ||||
| import ( | ||||
| 	"regexp" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| ) | ||||
|  | ||||
| var pathRegex = regexp.MustCompile("[^a-zA-Z0-9]+") | ||||
|  | ||||
| // GetAvatar constructs a URL for a given user-avatar if it is available in the cache. | ||||
| func getAvatar(av map[string]string, userid string, general *config.Protocol) string { | ||||
| 	if hash, ok := av[userid]; ok { | ||||
| 		// NOTE: This does not happen in bridge/helper/helper.go but messes up XMPP | ||||
| 		id := pathRegex.ReplaceAllString(userid, "_") | ||||
| 		return general.MediaServerDownload + "/" + hash + "/" + id + ".png" | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) cacheAvatar(msg *config.Message) string { | ||||
| 	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 "" | ||||
| } | ||||
| @@ -1,63 +1,63 @@ | ||||
| package bxmpp | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/tls" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net/http" | ||||
| 	"net/url" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	lru "github.com/hashicorp/golang-lru" | ||||
| 	"github.com/jpillora/backoff" | ||||
| 	"github.com/matterbridge/go-xmpp" | ||||
| 	"github.com/rs/xid" | ||||
| ) | ||||
|  | ||||
| type Bxmpp struct { | ||||
| 	xc      *xmpp.Client | ||||
| 	xmppMap map[string]string | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	startTime time.Time | ||||
| 	xc        *xmpp.Client | ||||
| 	xmppMap   map[string]string | ||||
| 	connected bool | ||||
| 	sync.RWMutex | ||||
|  | ||||
| 	StanzaIDs          *lru.Cache | ||||
| 	OriginIDs          *lru.Cache | ||||
|  | ||||
| 	avatarAvailability map[string]bool | ||||
| 	avatarMap          map[string]string | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bxmpp{Config: cfg} | ||||
| 	b.xmppMap = make(map[string]string) | ||||
| 	return b | ||||
| 	stanzaIDs, _ := lru.New(5000) | ||||
| 	originIDs, _ := lru.New(5000) | ||||
| 	return &Bxmpp{ | ||||
| 		Config:             cfg, | ||||
| 		StanzaIDs:          stanzaIDs, | ||||
| 		OriginIDs:          originIDs, | ||||
| 		xmppMap:            make(map[string]string), | ||||
| 		avatarAvailability: make(map[string]bool), | ||||
| 		avatarMap:          make(map[string]string), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) Connect() error { | ||||
| 	var err error | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
| 	b.xc, err = b.createXMPP() | ||||
| 	if err != nil { | ||||
| 	if err := b.createXMPP(); err != nil { | ||||
| 		b.Log.Debugf("%#v", err) | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	go func() { | ||||
| 		initial := true | ||||
| 		bf := &backoff.Backoff{ | ||||
| 			Min:    time.Second, | ||||
| 			Max:    5 * time.Minute, | ||||
| 			Jitter: true, | ||||
| 		} | ||||
| 		for { | ||||
| 			if initial { | ||||
| 				b.handleXMPP() | ||||
| 				initial = false | ||||
| 			} | ||||
| 			d := bf.Duration() | ||||
| 			b.Log.Infof("Disconnected. Reconnecting in %s", d) | ||||
| 			time.Sleep(d) | ||||
| 			b.xc, err = b.createXMPP() | ||||
| 			if err == nil { | ||||
| 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EventRejoinChannels} | ||||
| 				b.handleXMPP() | ||||
| 				bf.Reset() | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
| 	go b.manageConnection() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -76,58 +76,199 @@ func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error { | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||
| 	// should be fixed by using a cache instead of dropping | ||||
| 	if !b.Connected() { | ||||
| 		return "", fmt.Errorf("bridge %s not connected, dropping message %#v to bridge", b.Account, msg) | ||||
| 	} | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		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.Event == config.EventAvatarDownload { | ||||
| 		return b.cacheAvatar(&msg), nil | ||||
| 	} | ||||
|  | ||||
| 	// Make a action /me of the message, prepend the username with it. | ||||
| 	// https://xmpp.org/extensions/xep-0245.html | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Username = "/me " + msg.Username | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file (in XMPP case send the upload URL because XMPP has no native upload support). | ||||
| 	var err error | ||||
| 	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}) | ||||
| 			b.Log.Debugf("=> Sending attachement message %#v", rmsg) | ||||
| 			if b.GetString("WebhookURL") != "" { | ||||
| 				err = b.postSlackCompatibleWebhook(msg) | ||||
| 			} else { | ||||
| 				_, err = b.xc.Send(xmpp.Chat{ | ||||
| 					Type:   "groupchat", | ||||
| 					Remote: rmsg.Channel + "@" + b.GetString("Muc"), | ||||
| 					Text:   rmsg.Username + rmsg.Text, | ||||
| 				}) | ||||
| 			} | ||||
|  | ||||
| 			if err != nil { | ||||
| 				b.Log.WithError(err).Error("Unable to send message with share URL.") | ||||
| 			} | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			return b.handleUploadFile(&msg) | ||||
| 			return "", b.handleUploadFile(&msg) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	var msgreplaceid string | ||||
| 	msgid := xid.New().String() | ||||
| 	if msg.ID != "" { | ||||
| 		msgid = msg.ID | ||||
| 		msgreplaceid = msg.ID | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		b.Log.Debugf("Sending message using Webhook") | ||||
| 		err := b.postSlackCompatibleWebhook(msg) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Failed to send message using webhook: %s", err) | ||||
| 			return "", err | ||||
| 		} | ||||
|  | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	// 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 { | ||||
|  | ||||
| 	if msg.ParentNotFound() { | ||||
| 		msg.ParentID = "" | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message. | ||||
| 	var msgReplaceID string | ||||
| 	msgID := xid.New().String() | ||||
| 	if msg.ID != "" { | ||||
| 		msgReplaceID = msg.ID | ||||
| 	} | ||||
| 	var replyID string | ||||
| 	if res, ok := b.StanzaIDs.Get(msg.ParentID); ok { | ||||
| 		replyID, _ = res.(string) | ||||
| 	} | ||||
| 	b.Log.Debugf("=> Sending message %#v", msg) | ||||
| 	if _, err := b.xc.Send(xmpp.Chat{ | ||||
| 		Type:      "groupchat", | ||||
| 		Remote:    msg.Channel + "@" + b.GetString("Muc"), | ||||
| 		Text:      msg.Username + msg.Text, | ||||
| 		ID:        msgID, | ||||
| 		ReplaceID: msgReplaceID, | ||||
| 		ReplyID:   replyID, | ||||
| 	}); err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return msgid, nil | ||||
| 	return msgID, nil | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) createXMPP() (*xmpp.Client, error) { | ||||
| 	tc := new(tls.Config) | ||||
| 	tc.InsecureSkipVerify = b.GetBool("SkipTLSVerify") | ||||
| 	tc.ServerName = strings.Split(b.GetString("Server"), ":")[0] | ||||
| func (b *Bxmpp) postSlackCompatibleWebhook(msg config.Message) error { | ||||
| 	type XMPPWebhook struct { | ||||
| 		Username string `json:"username"` | ||||
| 		Text     string `json:"text"` | ||||
| 	} | ||||
| 	webhookBody, err := json.Marshal(XMPPWebhook{ | ||||
| 		Username: msg.Username, | ||||
| 		Text:     msg.Text, | ||||
| 	}) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Failed to marshal webhook: %s", err) | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	resp, err := http.Post(b.GetString("WebhookURL")+"/"+url.QueryEscape(msg.Channel), "application/json", bytes.NewReader(webhookBody)) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Failed to POST webhook: %s", err) | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	resp.Body.Close() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) createXMPP() error { | ||||
| 	var serverName string | ||||
| 	switch { | ||||
| 	case !b.GetBool("Anonymous"): | ||||
| 		if !strings.Contains(b.GetString("Jid"), "@") { | ||||
| 			return fmt.Errorf("the Jid %s doesn't contain an @", b.GetString("Jid")) | ||||
| 		} | ||||
| 		serverName = strings.Split(b.GetString("Jid"), "@")[1] | ||||
| 	case !strings.Contains(b.GetString("Server"), ":"): | ||||
| 		serverName = strings.Split(b.GetString("Server"), ":")[0] | ||||
| 	default: | ||||
| 		serverName = b.GetString("Server") | ||||
| 	} | ||||
|  | ||||
| 	tc := &tls.Config{ | ||||
| 		ServerName:         serverName, | ||||
| 		InsecureSkipVerify: b.GetBool("SkipTLSVerify"), // nolint: gosec | ||||
| 	} | ||||
|  | ||||
| 	xmpp.DebugWriter = b.Log.Writer() | ||||
|  | ||||
| 	options := xmpp.Options{ | ||||
| 		Host:                         b.GetString("Server"), | ||||
| 		User:                         b.GetString("Jid"), | ||||
| 		Password:                     b.GetString("Password"), | ||||
| 		NoTLS:                        true, | ||||
| 		StartTLS:                     true, | ||||
| 		StartTLS:                     !b.GetBool("NoTLS"), | ||||
| 		TLSConfig:                    tc, | ||||
| 		Debug:                        b.GetBool("debug"), | ||||
| 		Logger:                       b.Log.Writer(), | ||||
| 		Session:                      true, | ||||
| 		Status:                       "", | ||||
| 		StatusMessage:                "", | ||||
| 		Resource:                     "", | ||||
| 		InsecureAllowUnencryptedAuth: false, | ||||
| 		InsecureAllowUnencryptedAuth: b.GetBool("NoTLS"), | ||||
| 	} | ||||
| 	var err error | ||||
| 	b.xc, err = options.NewClient() | ||||
| 	return b.xc, err | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) manageConnection() { | ||||
| 	b.setConnected(true) | ||||
| 	initial := true | ||||
| 	bf := &backoff.Backoff{ | ||||
| 		Min:    time.Second, | ||||
| 		Max:    5 * time.Minute, | ||||
| 		Jitter: true, | ||||
| 	} | ||||
|  | ||||
| 	// Main connection loop. Each iteration corresponds to a successful | ||||
| 	// connection attempt and the subsequent handling of the connection. | ||||
| 	for { | ||||
| 		if initial { | ||||
| 			initial = false | ||||
| 		} else { | ||||
| 			b.Remote <- config.Message{ | ||||
| 				Username: "system", | ||||
| 				Text:     "rejoin", | ||||
| 				Channel:  "", | ||||
| 				Account:  b.Account, | ||||
| 				Event:    config.EventRejoinChannels, | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		if err := b.handleXMPP(); err != nil { | ||||
| 			b.Log.WithError(err).Error("Disconnected.") | ||||
| 			b.setConnected(false) | ||||
| 		} | ||||
|  | ||||
| 		// Reconnection loop using an exponential back-off strategy. We | ||||
| 		// only break out of the loop if we have successfully reconnected. | ||||
| 		for { | ||||
| 			d := bf.Duration() | ||||
| 			b.Log.Infof("Reconnecting in %s.", d) | ||||
| 			time.Sleep(d) | ||||
|  | ||||
| 			b.Log.Infof("Reconnecting now.") | ||||
| 			if err := b.createXMPP(); err == nil { | ||||
| 				b.setConnected(true) | ||||
| 				bf.Reset() | ||||
| 				break | ||||
| 			} | ||||
| 			b.Log.Warn("Failed to reconnect.") | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) xmppKeepAlive() chan bool { | ||||
| @@ -139,8 +280,7 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | ||||
| 			select { | ||||
| 			case <-ticker.C: | ||||
| 				b.Log.Debugf("PING") | ||||
| 				err := b.xc.PingC2S("", "") | ||||
| 				if err != nil { | ||||
| 				if err := b.xc.PingC2S("", ""); err != nil { | ||||
| 					b.Log.Debugf("PING failed %#v", err) | ||||
| 				} | ||||
| 			case <-done: | ||||
| @@ -152,53 +292,92 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) handleXMPP() error { | ||||
| 	var ok bool | ||||
| 	var msgid string | ||||
| 	b.startTime = time.Now() | ||||
|  | ||||
| 	done := b.xmppKeepAlive() | ||||
| 	defer close(done) | ||||
|  | ||||
| 	for { | ||||
| 		m, err := b.xc.Recv() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 			// An error together with AvatarData is non-fatal | ||||
| 			switch m.(type) { | ||||
| 			case xmpp.AvatarData: | ||||
| 				continue | ||||
| 			default: | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		switch v := m.(type) { | ||||
| 		case xmpp.Chat: | ||||
| 			if v.Type == "groupchat" { | ||||
| 				b.Log.Debugf("== Receiving %#v", v) | ||||
| 				event := "" | ||||
| 				// skip invalid messages | ||||
|  | ||||
| 				if v.ID != "" && v.StanzaID != "" { | ||||
| 					b.StanzaIDs.Add(v.ID, v.StanzaID) | ||||
| 					b.OriginIDs.Add(v.StanzaID, v.ID) | ||||
| 				} | ||||
|  | ||||
| 				// Skip invalid messages. | ||||
| 				if b.skipMessage(v) { | ||||
| 					continue | ||||
| 				} | ||||
|  | ||||
| 				var event string | ||||
| 				if strings.Contains(v.Text, "has set the subject to:") { | ||||
| 					event = config.EventTopicChange | ||||
| 				} | ||||
| 				msgid = v.ID | ||||
| 				if v.ReplaceID != "" { | ||||
| 					msgid = v.ReplaceID | ||||
|  | ||||
| 				available, sok := b.avatarAvailability[v.Remote] | ||||
| 				avatar := "" | ||||
| 				if !sok { | ||||
| 					b.Log.Debugf("Requesting avatar data") | ||||
| 					b.avatarAvailability[v.Remote] = false | ||||
| 					b.xc.AvatarRequestData(v.Remote) | ||||
| 				} else if available { | ||||
| 					avatar = getAvatar(b.avatarMap, v.Remote, b.General) | ||||
| 				} | ||||
|  | ||||
| 				msgID := v.ID | ||||
| 				if v.ReplaceID != "" { | ||||
| 					msgID = v.ReplaceID | ||||
| 				} | ||||
|  | ||||
| 				var parentID string | ||||
| 				if res, ok := b.OriginIDs.Get(v.ReplyID); ok { | ||||
| 					parentID, _ = res.(string) | ||||
| 				} | ||||
|  | ||||
| 				rmsg := config.Message{ | ||||
| 					Username: b.parseNick(v.Remote), | ||||
| 					Text:     v.Text, | ||||
| 					Channel:  b.parseChannel(v.Remote), | ||||
| 					Account:  b.Account, | ||||
| 					Avatar:   avatar, | ||||
| 					UserID:   v.Remote, | ||||
| 					ID:       msgid, | ||||
| 					ParentID: parentID, | ||||
| 					ID:       msgID, | ||||
| 					Event:    event, | ||||
| 				} | ||||
|  | ||||
| 				// check if we have an action event | ||||
| 				// Check if we have an action event. | ||||
| 				var ok bool | ||||
| 				rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||
| 				if ok { | ||||
| 					rmsg.Event = config.EventUserAction | ||||
| 				} | ||||
|  | ||||
| 				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.AvatarData: | ||||
| 			b.handleDownloadAvatar(v) | ||||
| 			b.avatarAvailability[v.From] = true | ||||
| 			b.Log.Debugf("Avatar for %s is now available", v.From) | ||||
| 		case xmpp.Presence: | ||||
| 			// do nothing | ||||
| 			// Do nothing. | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -211,35 +390,46 @@ func (b *Bxmpp) replaceAction(text string) (string, bool) { | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Bxmpp) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	var urldesc = "" | ||||
| func (b *Bxmpp) handleUploadFile(msg *config.Message) error { | ||||
| 	var urlDesc string | ||||
|  | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		if fi.Comment != "" { | ||||
| 			msg.Text += fi.Comment + ": " | ||||
| 	for _, file := range msg.Extra["file"] { | ||||
| 		fileInfo := file.(config.FileInfo) | ||||
| 		if fileInfo.Comment != "" { | ||||
| 			msg.Text += fileInfo.Comment + ": " | ||||
| 		} | ||||
| 		if fi.URL != "" { | ||||
| 			msg.Text = fi.URL | ||||
| 			if fi.Comment != "" { | ||||
| 				msg.Text = fi.Comment + ": " + fi.URL | ||||
| 				urldesc = fi.Comment | ||||
| 		if fileInfo.URL != "" { | ||||
| 			msg.Text = fileInfo.URL | ||||
| 			if fileInfo.Comment != "" { | ||||
| 				msg.Text = fileInfo.Comment + ": " + fileInfo.URL | ||||
| 				urlDesc = fileInfo.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 _, err := b.xc.Send(xmpp.Chat{ | ||||
| 			Type:   "groupchat", | ||||
| 			Remote: msg.Channel + "@" + b.GetString("Muc"), | ||||
| 			Text:   msg.Username + msg.Text, | ||||
| 		}); 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}) | ||||
|  | ||||
| 		if fileInfo.URL != "" { | ||||
| 			if _, err := b.xc.SendOOB(xmpp.Chat{ | ||||
| 				Type:    "groupchat", | ||||
| 				Remote:  msg.Channel + "@" + b.GetString("Muc"), | ||||
| 				Ooburl:  fileInfo.URL, | ||||
| 				Oobdesc: urlDesc, | ||||
| 			}); err != nil { | ||||
| 				b.Log.WithError(err).Warn("Failed to send share URL.") | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) parseNick(remote string) string { | ||||
| 	s := strings.Split(remote, "@") | ||||
| 	if len(s) > 0 { | ||||
| 	if len(s) > 1 { | ||||
| 		s = strings.Split(s[1], "/") | ||||
| 		if len(s) == 2 { | ||||
| 			return s[1] // nick | ||||
| @@ -278,7 +468,23 @@ func (b *Bxmpp) skipMessage(message xmpp.Chat) bool { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages posted by our webhook | ||||
| 	if b.GetString("WebhookURL") != "" && strings.Contains(message.ID, "webhookbot") { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// skip delayed messages | ||||
| 	t := time.Time{} | ||||
| 	return message.Stamp != t | ||||
| 	return !message.Stamp.IsZero() && time.Since(message.Stamp).Minutes() > 5 | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) setConnected(state bool) { | ||||
| 	b.Lock() | ||||
| 	b.connected = state | ||||
| 	defer b.Unlock() | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) Connected() bool { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	return b.connected | ||||
| } | ||||
|   | ||||
| @@ -2,6 +2,7 @@ package bzulip | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| @@ -11,6 +12,7 @@ import ( | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/version" | ||||
| 	gzb "github.com/matterbridge/gozulipbot" | ||||
| ) | ||||
|  | ||||
| @@ -27,7 +29,7 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| } | ||||
|  | ||||
| func (b *Bzulip) Connect() error { | ||||
| 	bot := gzb.Bot{APIKey: b.GetString("token"), APIURL: b.GetString("server") + "/api/v1/", Email: b.GetString("login")} | ||||
| 	bot := gzb.Bot{APIKey: b.GetString("token"), APIURL: b.GetString("server") + "/api/v1/", Email: b.GetString("login"), UserAgent: fmt.Sprintf("matterbridge/%s", version.Release)} | ||||
| 	bot.Init() | ||||
| 	q, err := bot.RegisterAll() | ||||
| 	b.q = q | ||||
| @@ -104,29 +106,31 @@ func (b *Bzulip) getChannel(id int) string { | ||||
| func (b *Bzulip) handleQueue() error { | ||||
| 	for { | ||||
| 		messages, err := b.q.GetEvents() | ||||
| 		switch err { | ||||
| 		case gzb.BackoffError: | ||||
| 			time.Sleep(time.Second * 5) | ||||
| 		case gzb.NoJSONError: | ||||
| 			b.Log.Error("Response wasn't JSON, server down or restarting? sleeping 10 seconds") | ||||
| 			time.Sleep(time.Second * 10) | ||||
| 		case gzb.BadEventQueueError: | ||||
| 			b.Log.Info("got a bad event queue id error, reconnecting") | ||||
| 			b.bot.Queues = nil | ||||
| 			for { | ||||
| 				b.q, err = b.bot.RegisterAll() | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("reconnecting failed: %s. Sleeping 10 seconds", err) | ||||
| 					time.Sleep(time.Second * 10) | ||||
| 				} | ||||
| 				break | ||||
| 			} | ||||
| 		case gzb.HeartbeatError: | ||||
| 			b.Log.Debug("heartbeat received.") | ||||
| 		default: | ||||
| 			b.Log.Debugf("receiving error: %#v", err) | ||||
| 		} | ||||
| 		if err != nil { | ||||
| 			switch err { | ||||
| 			case gzb.BackoffError: | ||||
| 				time.Sleep(time.Second * 5) | ||||
| 			case gzb.NoJSONError: | ||||
| 				b.Log.Error("Response wasn't JSON, server down or restarting? sleeping 10 seconds") | ||||
| 				time.Sleep(time.Second * 10) | ||||
| 			case gzb.BadEventQueueError: | ||||
| 				b.Log.Info("got a bad event queue id error, reconnecting") | ||||
| 				b.bot.Queues = nil | ||||
| 				for { | ||||
| 					b.q, err = b.bot.RegisterAll() | ||||
| 					if err != nil { | ||||
| 						b.Log.Errorf("reconnecting failed: %s. Sleeping 10 seconds", err) | ||||
| 						time.Sleep(time.Second * 10) | ||||
| 					} | ||||
| 					break | ||||
| 				} | ||||
| 			case gzb.HeartbeatError: | ||||
| 				b.Log.Debug("heartbeat received.") | ||||
| 			default: | ||||
| 				b.Log.Debugf("receiving error: %#v", err) | ||||
| 				time.Sleep(time.Second * 10) | ||||
| 			} | ||||
|  | ||||
| 			continue | ||||
| 		} | ||||
| 		for _, m := range messages { | ||||
| @@ -135,19 +139,25 @@ func (b *Bzulip) handleQueue() error { | ||||
| 			if m.SenderEmail == b.GetString("login") { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			avatarURL := m.AvatarURL | ||||
| 			if !strings.HasPrefix(avatarURL, "http") { | ||||
| 				avatarURL = b.GetString("server") + avatarURL | ||||
| 			} | ||||
|  | ||||
| 			rmsg := config.Message{ | ||||
| 				Username: m.SenderFullName, | ||||
| 				Text:     m.Content, | ||||
| 				Channel:  b.getChannel(m.StreamID) + "/topic:" + m.Subject, | ||||
| 				Account:  b.Account, | ||||
| 				UserID:   strconv.Itoa(m.SenderID), | ||||
| 				Avatar:   m.AvatarURL, | ||||
| 				Avatar:   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) | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										1846
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						
									
										1846
									
								
								changelog.md
									
									
									
									
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							| @@ -1,27 +0,0 @@ | ||||
| #!/bin/bash | ||||
| go version | grep go1.11 || exit | ||||
| VERSION=$(git describe --tags) | ||||
| mkdir ci/binaries | ||||
| GOOS=windows GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-windows-amd64.exe | ||||
| GOOS=linux GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-amd64 | ||||
| GOOS=linux GOARCH=arm go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-linux-arm | ||||
| GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o ci/binaries/matterbridge-$VERSION-darwin-amd64 | ||||
| cd ci | ||||
| cat > deploy.json <<EOF | ||||
| { | ||||
|     "package": { | ||||
|         "name": "Matterbridge", | ||||
|         "repo": "nightly", | ||||
|         "subject": "42wim" | ||||
|     }, | ||||
|     "version": { | ||||
|         "name": "$VERSION" | ||||
|     }, | ||||
|     "files": | ||||
|         [ | ||||
|         {"includePattern": "ci/binaries/(.*)", "uploadPattern":"\$1"} | ||||
|         ], | ||||
|     "publish": true | ||||
| } | ||||
| EOF | ||||
|  | ||||
							
								
								
									
										15
									
								
								contrib/inmessage-logger.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								contrib/inmessage-logger.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,15 @@ | ||||
| fmt := import("fmt") | ||||
| os := import("os") | ||||
| times := import("times") | ||||
|  | ||||
| if msgText != "" && msgUsername != "system" { | ||||
|     os.chdir("/var/www/matterbridge") | ||||
|     file := os.open_file("inmessage.log", os.o_append|os.o_wronly|os.o_create, 0644) | ||||
|     file.write_string(fmt.sprintf( | ||||
|         "[%s] <%s> %s\n", | ||||
|         times.time_format(times.now(), times.format_rfc1123), | ||||
|         msgUsername, | ||||
|         msgText | ||||
|     )) | ||||
|     file.close() | ||||
| } | ||||
							
								
								
									
										19
									
								
								contrib/matterbridge.openrc
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										19
									
								
								contrib/matterbridge.openrc
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,19 @@ | ||||
| #!/sbin/openrc-run | ||||
| # Copyright 2021-2022 Gentoo Authors | ||||
| # Distributed under the terms of the GNU General Public License v2 | ||||
|  | ||||
| command=/usr/bin/matterbridge | ||||
| command_args="-conf ${MATTERBRIDGE_CONF:-/etc/matterbridge/bridge.toml} ${MATTERBRIDGE_ARGS}" | ||||
| command_user="matterbridge:matterbridge" | ||||
| pidfile="/run/${RC_SVCNAME}.pid" | ||||
| command_background=1 | ||||
| output_log="/var/log/${RC_SVCNAME}.log" | ||||
| error_log="${output_log}" | ||||
|  | ||||
| depend() { | ||||
| 	need net | ||||
| } | ||||
|  | ||||
| start_pre() { | ||||
| 	checkpath -f "${output_log}" -o "${command_user}" || return 1 | ||||
| } | ||||
							
								
								
									
										6
									
								
								contrib/mumble-autolink.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								contrib/mumble-autolink.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,6 @@ | ||||
| text := import("text") | ||||
|  | ||||
| if outProtocol == "mumble" { | ||||
|     urlRE := text.re_compile(`(?is)((http|https):\/\/)?([a-z0-9-]+\.)?[a-z0-9-]+(\.[a-z]{2,6}){1,3}(\/[a-z0-9.,_\/~#&=;%+?-]*)?`) | ||||
|     msgText = urlRE.replace(msgText,`<a href="$0">$0</a>`) | ||||
| } | ||||
							
								
								
									
										10
									
								
								contrib/outmessage-discordemoji.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								contrib/outmessage-discordemoji.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,10 @@ | ||||
| text := import("text") | ||||
|  | ||||
| // if we're not sending to a discord bridge, | ||||
| // then convert custom emoji tags into url's | ||||
| if (inProtocol == "discord" && outProtocol != "discord") { | ||||
|     rePNG := text.re_compile(`<:.*?:([0-9]+)>`) | ||||
|     msgText=rePNG.replace(msgText,"https://cdn.discordapp.com/emojis/$1.png") | ||||
|     reGIF := text.re_compile(`<a:.*?:([0-9]+)>`) | ||||
|     msgText=reGIF.replace(msgText,"https://cdn.discordapp.com/emojis/$1.gif") | ||||
| } | ||||
							
								
								
									
										14
									
								
								contrib/outmessage-irccolornick.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								contrib/outmessage-irccolornick.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| // See https://github.com/42wim/matterbridge/issues/881 | ||||
| // Generates a colored nick for each msgUsername, with example to filter specific codes  | ||||
|  | ||||
| text := import("text") | ||||
| fmt := import("fmt") | ||||
| if outProtocol == "irc" { | ||||
|     // generate a color for a nick, make sure it isn't 0 or 15 | ||||
|     colorCode := len(msgUsername)+bytes(msgUsername)[0]%14 + 2 | ||||
|     // example if we want to use colorCode 3 when we have calculated colorcode 14 | ||||
|     if colorCode == 14 { | ||||
|         colorCode = 3 | ||||
|     } | ||||
|     msgUsername=fmt.sprintf("\x03%02d%s\x0F", colorCode, msgUsername) | ||||
| } | ||||
							
								
								
									
										7
									
								
								contrib/outmessage-irccolors.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								contrib/outmessage-irccolors.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,7 @@ | ||||
| // See https://github.com/42wim/matterbridge/issues/798 | ||||
|  | ||||
| // if we're not sending to an irc bridge we strip the IRC colors | ||||
| if outProtocol != "irc" { | ||||
|     re := text.re_compile(`\x03(?:\d{1,2}(?:,\d{1,2})?)?|[[:cntrl:]]`) | ||||
|     msgText=re.replace(msgText,"") | ||||
| } | ||||
							
								
								
									
										16
									
								
								contrib/remotenickformat-zerowidth.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										16
									
								
								contrib/remotenickformat-zerowidth.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,16 @@ | ||||
| /* | ||||
| This script will return the nick except with multi-character usernames | ||||
| containing a zero-width space between the first and second character letter. | ||||
|  | ||||
| Single character usernames will be left untouched. | ||||
|  | ||||
| This is useful to prevent remote users from nickalerting | ||||
| IRC users of the same name when the remote user speaks. | ||||
|  | ||||
| This result can be used in {TENGO} in RemoteNickFormat. | ||||
| */ | ||||
|  | ||||
| result = nick | ||||
| if len(nick) > 1 { | ||||
|     result = string(nick[0]) + "" + nick[1:] | ||||
| } | ||||
							
								
								
									
										9
									
								
								contrib/remotenickformat.tengo
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								contrib/remotenickformat.tengo
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| /* | ||||
| This script will return the current time in kitchen format if the protocol (of the remote bridge) isn't irc | ||||
| See https://github.com/d5/tengo/blob/master/docs/stdlib-times.md | ||||
| This result can be used in {TENGO} in RemoteNickFormat | ||||
| */ | ||||
| times := import("times") | ||||
| if protocol != "irc" { | ||||
|    result=times.time_format(times.now(),times.format_kitchen) | ||||
| } | ||||
| @@ -1,9 +1,10 @@ | ||||
| FROM alpine:edge as certs | ||||
| RUN apk --update add ca-certificates | ||||
| ARG VERSION=1.22.3 | ||||
| ADD https://github.com/42wim/matterbridge/releases/download/v${VERSION}/matterbridge-${VERSION}-linux-arm64 /bin/matterbridge | ||||
| RUN chmod +x /bin/matterbridge | ||||
|  | ||||
| FROM scratch | ||||
| ARG VERSION=1.12.3 | ||||
| COPY --from=certs /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt | ||||
| ADD https://github.com/42wim/matterbridge/releases/download/v${VERSION}/matterbridge-linux-arm /bin/matterbridge | ||||
| RUN chmod +x /bin/matterbridge | ||||
| COPY --from=certs /bin/matterbridge /bin/matterbridge | ||||
| ENTRYPOINT ["/bin/matterbridge"] | ||||
|   | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/api.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/api.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !noapi | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/api" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["api"] = api.New | ||||
| } | ||||
							
								
								
									
										12
									
								
								gateway/bridgemap/bdiscord.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								gateway/bridgemap/bdiscord.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | ||||
| // +build !nodiscord | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bdiscord "github.com/42wim/matterbridge/bridge/discord" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["discord"] = bdiscord.New | ||||
| 	UserTypingSupport["discord"] = struct{}{} | ||||
| } | ||||
							
								
								
									
										12
									
								
								gateway/bridgemap/bharmony.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								gateway/bridgemap/bharmony.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | ||||
| //go:build !noharmony | ||||
| // +build !noharmony | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bharmony "github.com/42wim/matterbridge/bridge/harmony" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["harmony"] = bharmony.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/birc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/birc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !noirc | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	birc "github.com/42wim/matterbridge/bridge/irc" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["irc"] = birc.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bkeybase.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bkeybase.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nokeybase | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bkeybase "github.com/42wim/matterbridge/bridge/keybase" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["keybase"] = bkeybase.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bmatrix.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bmatrix.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nomatrix | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bmatrix "github.com/42wim/matterbridge/bridge/matrix" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["matrix"] = bmatrix.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bmattermost.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bmattermost.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nomattermost | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bmattermost "github.com/42wim/matterbridge/bridge/mattermost" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["mattermost"] = bmattermost.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bmsteams.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bmsteams.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nomsteams | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bmsteams "github.com/42wim/matterbridge/bridge/msteams" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["msteams"] = bmsteams.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bmumble.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bmumble.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nomumble | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bmumble "github.com/42wim/matterbridge/bridge/mumble" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["mumble"] = bmumble.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bnctalk.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bnctalk.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nonctalk | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	btalk "github.com/42wim/matterbridge/bridge/nctalk" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["nctalk"] = btalk.New | ||||
| } | ||||
| @@ -2,36 +2,9 @@ package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/api" | ||||
| 	"github.com/42wim/matterbridge/bridge/discord" | ||||
| 	"github.com/42wim/matterbridge/bridge/gitter" | ||||
| 	"github.com/42wim/matterbridge/bridge/irc" | ||||
| 	"github.com/42wim/matterbridge/bridge/matrix" | ||||
| 	"github.com/42wim/matterbridge/bridge/mattermost" | ||||
| 	"github.com/42wim/matterbridge/bridge/rocketchat" | ||||
| 	"github.com/42wim/matterbridge/bridge/slack" | ||||
| 	"github.com/42wim/matterbridge/bridge/sshchat" | ||||
| 	"github.com/42wim/matterbridge/bridge/steam" | ||||
| 	"github.com/42wim/matterbridge/bridge/telegram" | ||||
| 	"github.com/42wim/matterbridge/bridge/whatsapp" | ||||
| 	"github.com/42wim/matterbridge/bridge/xmpp" | ||||
| 	"github.com/42wim/matterbridge/bridge/zulip" | ||||
| ) | ||||
|  | ||||
| var FullMap = 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, | ||||
| 	"whatsapp":     bwhatsapp.New, | ||||
| 	"xmpp":         bxmpp.New, | ||||
| 	"zulip":        bzulip.New, | ||||
| } | ||||
| var ( | ||||
| 	FullMap           = map[string]bridge.Factory{} | ||||
| 	UserTypingSupport = map[string]struct{}{} | ||||
| ) | ||||
|   | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/brocketchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/brocketchat.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !norocketchat | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	brocketchat "github.com/42wim/matterbridge/bridge/rocketchat" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["rocketchat"] = brocketchat.New | ||||
| } | ||||
							
								
								
									
										13
									
								
								gateway/bridgemap/bslack.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								gateway/bridgemap/bslack.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,13 @@ | ||||
| // +build !noslack | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bslack "github.com/42wim/matterbridge/bridge/slack" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["slack-legacy"] = bslack.NewLegacy | ||||
| 	FullMap["slack"] = bslack.New | ||||
| 	UserTypingSupport["slack"] = struct{}{} | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bsshchat.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bsshchat.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nosshchat | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bsshchat "github.com/42wim/matterbridge/bridge/sshchat" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["sshchat"] = bsshchat.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bsteam.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bsteam.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !nosteam | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bsteam "github.com/42wim/matterbridge/bridge/steam" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["steam"] = bsteam.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/btelegram.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/btelegram.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !notelegram | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	btelegram "github.com/42wim/matterbridge/bridge/telegram" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["telegram"] = btelegram.New | ||||
| } | ||||
							
								
								
									
										11
									
								
								gateway/bridgemap/bvk.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								gateway/bridgemap/bvk.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| // +build !novk | ||||
|  | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	bvk "github.com/42wim/matterbridge/bridge/vk" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	FullMap["vk"] = bvk.New | ||||
| } | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user